xref: /freebsd/contrib/llvm-project/llvm/lib/CodeGen/SelectionDAG/SelectionDAGISel.cpp (revision 0eae32dcef82f6f06de6419a0d623d7def0cc8f6)
10b57cec5SDimitry Andric //===- SelectionDAGISel.cpp - Implement the SelectionDAGISel class --------===//
20b57cec5SDimitry Andric //
30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
60b57cec5SDimitry Andric //
70b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
80b57cec5SDimitry Andric //
90b57cec5SDimitry Andric // This implements the SelectionDAGISel class.
100b57cec5SDimitry Andric //
110b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
120b57cec5SDimitry Andric 
130b57cec5SDimitry Andric #include "llvm/CodeGen/SelectionDAGISel.h"
140b57cec5SDimitry Andric #include "ScheduleDAGSDNodes.h"
150b57cec5SDimitry Andric #include "SelectionDAGBuilder.h"
160b57cec5SDimitry Andric #include "llvm/ADT/APInt.h"
170b57cec5SDimitry Andric #include "llvm/ADT/DenseMap.h"
180b57cec5SDimitry Andric #include "llvm/ADT/None.h"
190b57cec5SDimitry Andric #include "llvm/ADT/PostOrderIterator.h"
200b57cec5SDimitry Andric #include "llvm/ADT/STLExtras.h"
210b57cec5SDimitry Andric #include "llvm/ADT/SmallPtrSet.h"
220b57cec5SDimitry Andric #include "llvm/ADT/SmallSet.h"
230b57cec5SDimitry Andric #include "llvm/ADT/SmallVector.h"
240b57cec5SDimitry Andric #include "llvm/ADT/Statistic.h"
250b57cec5SDimitry Andric #include "llvm/ADT/StringRef.h"
260b57cec5SDimitry Andric #include "llvm/Analysis/AliasAnalysis.h"
270b57cec5SDimitry Andric #include "llvm/Analysis/BranchProbabilityInfo.h"
280b57cec5SDimitry Andric #include "llvm/Analysis/CFG.h"
290b57cec5SDimitry Andric #include "llvm/Analysis/EHPersonalities.h"
30480093f4SDimitry Andric #include "llvm/Analysis/LazyBlockFrequencyInfo.h"
318bcb0991SDimitry Andric #include "llvm/Analysis/LegacyDivergenceAnalysis.h"
320b57cec5SDimitry Andric #include "llvm/Analysis/OptimizationRemarkEmitter.h"
33480093f4SDimitry Andric #include "llvm/Analysis/ProfileSummaryInfo.h"
340b57cec5SDimitry Andric #include "llvm/Analysis/TargetLibraryInfo.h"
350b57cec5SDimitry Andric #include "llvm/Analysis/TargetTransformInfo.h"
36349cc55cSDimitry Andric #include "llvm/CodeGen/CodeGenCommonISel.h"
370b57cec5SDimitry Andric #include "llvm/CodeGen/FastISel.h"
380b57cec5SDimitry Andric #include "llvm/CodeGen/FunctionLoweringInfo.h"
390b57cec5SDimitry Andric #include "llvm/CodeGen/GCMetadata.h"
400b57cec5SDimitry Andric #include "llvm/CodeGen/ISDOpcodes.h"
410b57cec5SDimitry Andric #include "llvm/CodeGen/MachineBasicBlock.h"
420b57cec5SDimitry Andric #include "llvm/CodeGen/MachineFrameInfo.h"
430b57cec5SDimitry Andric #include "llvm/CodeGen/MachineFunction.h"
440b57cec5SDimitry Andric #include "llvm/CodeGen/MachineFunctionPass.h"
450b57cec5SDimitry Andric #include "llvm/CodeGen/MachineInstr.h"
460b57cec5SDimitry Andric #include "llvm/CodeGen/MachineInstrBuilder.h"
470b57cec5SDimitry Andric #include "llvm/CodeGen/MachineMemOperand.h"
480b57cec5SDimitry Andric #include "llvm/CodeGen/MachineModuleInfo.h"
490b57cec5SDimitry Andric #include "llvm/CodeGen/MachineOperand.h"
500b57cec5SDimitry Andric #include "llvm/CodeGen/MachinePassRegistry.h"
510b57cec5SDimitry Andric #include "llvm/CodeGen/MachineRegisterInfo.h"
520b57cec5SDimitry Andric #include "llvm/CodeGen/SchedulerRegistry.h"
530b57cec5SDimitry Andric #include "llvm/CodeGen/SelectionDAG.h"
540b57cec5SDimitry Andric #include "llvm/CodeGen/SelectionDAGNodes.h"
550b57cec5SDimitry Andric #include "llvm/CodeGen/StackProtector.h"
560b57cec5SDimitry Andric #include "llvm/CodeGen/SwiftErrorValueTracking.h"
570b57cec5SDimitry Andric #include "llvm/CodeGen/TargetInstrInfo.h"
580b57cec5SDimitry Andric #include "llvm/CodeGen/TargetLowering.h"
590b57cec5SDimitry Andric #include "llvm/CodeGen/TargetRegisterInfo.h"
600b57cec5SDimitry Andric #include "llvm/CodeGen/TargetSubtargetInfo.h"
610b57cec5SDimitry Andric #include "llvm/CodeGen/ValueTypes.h"
620b57cec5SDimitry Andric #include "llvm/IR/BasicBlock.h"
630b57cec5SDimitry Andric #include "llvm/IR/Constants.h"
640b57cec5SDimitry Andric #include "llvm/IR/DataLayout.h"
650b57cec5SDimitry Andric #include "llvm/IR/DebugInfoMetadata.h"
660b57cec5SDimitry Andric #include "llvm/IR/DebugLoc.h"
670b57cec5SDimitry Andric #include "llvm/IR/DiagnosticInfo.h"
680b57cec5SDimitry Andric #include "llvm/IR/Dominators.h"
690b57cec5SDimitry Andric #include "llvm/IR/Function.h"
700b57cec5SDimitry Andric #include "llvm/IR/InlineAsm.h"
710b57cec5SDimitry Andric #include "llvm/IR/InstIterator.h"
720b57cec5SDimitry Andric #include "llvm/IR/InstrTypes.h"
730b57cec5SDimitry Andric #include "llvm/IR/Instruction.h"
740b57cec5SDimitry Andric #include "llvm/IR/Instructions.h"
750b57cec5SDimitry Andric #include "llvm/IR/IntrinsicInst.h"
760b57cec5SDimitry Andric #include "llvm/IR/Intrinsics.h"
77480093f4SDimitry Andric #include "llvm/IR/IntrinsicsWebAssembly.h"
780b57cec5SDimitry Andric #include "llvm/IR/Metadata.h"
79e8d8bef9SDimitry Andric #include "llvm/IR/Statepoint.h"
800b57cec5SDimitry Andric #include "llvm/IR/Type.h"
810b57cec5SDimitry Andric #include "llvm/IR/User.h"
820b57cec5SDimitry Andric #include "llvm/IR/Value.h"
83480093f4SDimitry Andric #include "llvm/InitializePasses.h"
840b57cec5SDimitry Andric #include "llvm/MC/MCInstrDesc.h"
850b57cec5SDimitry Andric #include "llvm/MC/MCRegisterInfo.h"
860b57cec5SDimitry Andric #include "llvm/Pass.h"
870b57cec5SDimitry Andric #include "llvm/Support/BranchProbability.h"
880b57cec5SDimitry Andric #include "llvm/Support/Casting.h"
890b57cec5SDimitry Andric #include "llvm/Support/CodeGen.h"
900b57cec5SDimitry Andric #include "llvm/Support/CommandLine.h"
910b57cec5SDimitry Andric #include "llvm/Support/Compiler.h"
920b57cec5SDimitry Andric #include "llvm/Support/Debug.h"
930b57cec5SDimitry Andric #include "llvm/Support/ErrorHandling.h"
940b57cec5SDimitry Andric #include "llvm/Support/KnownBits.h"
950b57cec5SDimitry Andric #include "llvm/Support/MachineValueType.h"
960b57cec5SDimitry Andric #include "llvm/Support/Timer.h"
970b57cec5SDimitry Andric #include "llvm/Support/raw_ostream.h"
980b57cec5SDimitry Andric #include "llvm/Target/TargetIntrinsicInfo.h"
990b57cec5SDimitry Andric #include "llvm/Target/TargetMachine.h"
1000b57cec5SDimitry Andric #include "llvm/Target/TargetOptions.h"
1010b57cec5SDimitry Andric #include "llvm/Transforms/Utils/BasicBlockUtils.h"
1020b57cec5SDimitry Andric #include <algorithm>
1030b57cec5SDimitry Andric #include <cassert>
1040b57cec5SDimitry Andric #include <cstdint>
1050b57cec5SDimitry Andric #include <iterator>
1060b57cec5SDimitry Andric #include <limits>
1070b57cec5SDimitry Andric #include <memory>
1080b57cec5SDimitry Andric #include <string>
1090b57cec5SDimitry Andric #include <utility>
1100b57cec5SDimitry Andric #include <vector>
1110b57cec5SDimitry Andric 
1120b57cec5SDimitry Andric using namespace llvm;
1130b57cec5SDimitry Andric 
1140b57cec5SDimitry Andric #define DEBUG_TYPE "isel"
1150b57cec5SDimitry Andric 
1160b57cec5SDimitry Andric STATISTIC(NumFastIselFailures, "Number of instructions fast isel failed on");
1170b57cec5SDimitry Andric STATISTIC(NumFastIselSuccess, "Number of instructions fast isel selected");
1180b57cec5SDimitry Andric STATISTIC(NumFastIselBlocks, "Number of blocks selected entirely by fast isel");
1190b57cec5SDimitry Andric STATISTIC(NumDAGBlocks, "Number of blocks selected using DAG");
1200b57cec5SDimitry Andric STATISTIC(NumDAGIselRetries,"Number of times dag isel has to try another path");
1210b57cec5SDimitry Andric STATISTIC(NumEntryBlocks, "Number of entry blocks encountered");
1220b57cec5SDimitry Andric STATISTIC(NumFastIselFailLowerArguments,
1230b57cec5SDimitry Andric           "Number of entry blocks where fast isel failed to lower arguments");
1240b57cec5SDimitry Andric 
1250b57cec5SDimitry Andric static cl::opt<int> EnableFastISelAbort(
1260b57cec5SDimitry Andric     "fast-isel-abort", cl::Hidden,
1270b57cec5SDimitry Andric     cl::desc("Enable abort calls when \"fast\" instruction selection "
1280b57cec5SDimitry Andric              "fails to lower an instruction: 0 disable the abort, 1 will "
1290b57cec5SDimitry Andric              "abort but for args, calls and terminators, 2 will also "
1300b57cec5SDimitry Andric              "abort for argument lowering, and 3 will never fallback "
1310b57cec5SDimitry Andric              "to SelectionDAG."));
1320b57cec5SDimitry Andric 
1330b57cec5SDimitry Andric static cl::opt<bool> EnableFastISelFallbackReport(
1340b57cec5SDimitry Andric     "fast-isel-report-on-fallback", cl::Hidden,
1350b57cec5SDimitry Andric     cl::desc("Emit a diagnostic when \"fast\" instruction selection "
1360b57cec5SDimitry Andric              "falls back to SelectionDAG."));
1370b57cec5SDimitry Andric 
1380b57cec5SDimitry Andric static cl::opt<bool>
1390b57cec5SDimitry Andric UseMBPI("use-mbpi",
1400b57cec5SDimitry Andric         cl::desc("use Machine Branch Probability Info"),
1410b57cec5SDimitry Andric         cl::init(true), cl::Hidden);
1420b57cec5SDimitry Andric 
1430b57cec5SDimitry Andric #ifndef NDEBUG
1440b57cec5SDimitry Andric static cl::opt<std::string>
1450b57cec5SDimitry Andric FilterDAGBasicBlockName("filter-view-dags", cl::Hidden,
1460b57cec5SDimitry Andric                         cl::desc("Only display the basic block whose name "
1470b57cec5SDimitry Andric                                  "matches this for all view-*-dags options"));
1480b57cec5SDimitry Andric static cl::opt<bool>
1490b57cec5SDimitry Andric ViewDAGCombine1("view-dag-combine1-dags", cl::Hidden,
1500b57cec5SDimitry Andric           cl::desc("Pop up a window to show dags before the first "
1510b57cec5SDimitry Andric                    "dag combine pass"));
1520b57cec5SDimitry Andric static cl::opt<bool>
1530b57cec5SDimitry Andric ViewLegalizeTypesDAGs("view-legalize-types-dags", cl::Hidden,
1540b57cec5SDimitry Andric           cl::desc("Pop up a window to show dags before legalize types"));
1550b57cec5SDimitry Andric static cl::opt<bool>
156480093f4SDimitry Andric     ViewDAGCombineLT("view-dag-combine-lt-dags", cl::Hidden,
157480093f4SDimitry Andric                      cl::desc("Pop up a window to show dags before the post "
158480093f4SDimitry Andric                               "legalize types dag combine pass"));
159480093f4SDimitry Andric static cl::opt<bool>
1600b57cec5SDimitry Andric     ViewLegalizeDAGs("view-legalize-dags", cl::Hidden,
1610b57cec5SDimitry Andric                      cl::desc("Pop up a window to show dags before legalize"));
1620b57cec5SDimitry Andric static cl::opt<bool>
1630b57cec5SDimitry Andric ViewDAGCombine2("view-dag-combine2-dags", cl::Hidden,
1640b57cec5SDimitry Andric           cl::desc("Pop up a window to show dags before the second "
1650b57cec5SDimitry Andric                    "dag combine pass"));
1660b57cec5SDimitry Andric static cl::opt<bool>
1670b57cec5SDimitry Andric ViewISelDAGs("view-isel-dags", cl::Hidden,
1680b57cec5SDimitry Andric           cl::desc("Pop up a window to show isel dags as they are selected"));
1690b57cec5SDimitry Andric static cl::opt<bool>
1700b57cec5SDimitry Andric ViewSchedDAGs("view-sched-dags", cl::Hidden,
1710b57cec5SDimitry Andric           cl::desc("Pop up a window to show sched dags as they are processed"));
1720b57cec5SDimitry Andric static cl::opt<bool>
1730b57cec5SDimitry Andric ViewSUnitDAGs("view-sunit-dags", cl::Hidden,
1740b57cec5SDimitry Andric       cl::desc("Pop up a window to show SUnit dags after they are processed"));
1750b57cec5SDimitry Andric #else
176480093f4SDimitry Andric static const bool ViewDAGCombine1 = false, ViewLegalizeTypesDAGs = false,
177480093f4SDimitry Andric                   ViewDAGCombineLT = false, ViewLegalizeDAGs = false,
178480093f4SDimitry Andric                   ViewDAGCombine2 = false, ViewISelDAGs = false,
179480093f4SDimitry Andric                   ViewSchedDAGs = false, ViewSUnitDAGs = false;
1800b57cec5SDimitry Andric #endif
1810b57cec5SDimitry Andric 
1820b57cec5SDimitry Andric //===---------------------------------------------------------------------===//
1830b57cec5SDimitry Andric ///
1840b57cec5SDimitry Andric /// RegisterScheduler class - Track the registration of instruction schedulers.
1850b57cec5SDimitry Andric ///
1860b57cec5SDimitry Andric //===---------------------------------------------------------------------===//
1870b57cec5SDimitry Andric MachinePassRegistry<RegisterScheduler::FunctionPassCtor>
1880b57cec5SDimitry Andric     RegisterScheduler::Registry;
1890b57cec5SDimitry Andric 
1900b57cec5SDimitry Andric //===---------------------------------------------------------------------===//
1910b57cec5SDimitry Andric ///
1920b57cec5SDimitry Andric /// ISHeuristic command line option for instruction schedulers.
1930b57cec5SDimitry Andric ///
1940b57cec5SDimitry Andric //===---------------------------------------------------------------------===//
1950b57cec5SDimitry Andric static cl::opt<RegisterScheduler::FunctionPassCtor, false,
1960b57cec5SDimitry Andric                RegisterPassParser<RegisterScheduler>>
1970b57cec5SDimitry Andric ISHeuristic("pre-RA-sched",
1980b57cec5SDimitry Andric             cl::init(&createDefaultScheduler), cl::Hidden,
1990b57cec5SDimitry Andric             cl::desc("Instruction schedulers available (before register"
2000b57cec5SDimitry Andric                      " allocation):"));
2010b57cec5SDimitry Andric 
2020b57cec5SDimitry Andric static RegisterScheduler
2030b57cec5SDimitry Andric defaultListDAGScheduler("default", "Best scheduler for the target",
2040b57cec5SDimitry Andric                         createDefaultScheduler);
2050b57cec5SDimitry Andric 
2060b57cec5SDimitry Andric namespace llvm {
2070b57cec5SDimitry Andric 
2080b57cec5SDimitry Andric   //===--------------------------------------------------------------------===//
2090b57cec5SDimitry Andric   /// This class is used by SelectionDAGISel to temporarily override
2100b57cec5SDimitry Andric   /// the optimization level on a per-function basis.
2110b57cec5SDimitry Andric   class OptLevelChanger {
2120b57cec5SDimitry Andric     SelectionDAGISel &IS;
2130b57cec5SDimitry Andric     CodeGenOpt::Level SavedOptLevel;
2140b57cec5SDimitry Andric     bool SavedFastISel;
2150b57cec5SDimitry Andric 
2160b57cec5SDimitry Andric   public:
2170b57cec5SDimitry Andric     OptLevelChanger(SelectionDAGISel &ISel,
2180b57cec5SDimitry Andric                     CodeGenOpt::Level NewOptLevel) : IS(ISel) {
2190b57cec5SDimitry Andric       SavedOptLevel = IS.OptLevel;
2205ffd83dbSDimitry Andric       SavedFastISel = IS.TM.Options.EnableFastISel;
2210b57cec5SDimitry Andric       if (NewOptLevel == SavedOptLevel)
2220b57cec5SDimitry Andric         return;
2230b57cec5SDimitry Andric       IS.OptLevel = NewOptLevel;
2240b57cec5SDimitry Andric       IS.TM.setOptLevel(NewOptLevel);
2250b57cec5SDimitry Andric       LLVM_DEBUG(dbgs() << "\nChanging optimization level for Function "
2260b57cec5SDimitry Andric                         << IS.MF->getFunction().getName() << "\n");
2270b57cec5SDimitry Andric       LLVM_DEBUG(dbgs() << "\tBefore: -O" << SavedOptLevel << " ; After: -O"
2280b57cec5SDimitry Andric                         << NewOptLevel << "\n");
2290b57cec5SDimitry Andric       if (NewOptLevel == CodeGenOpt::None) {
2300b57cec5SDimitry Andric         IS.TM.setFastISel(IS.TM.getO0WantsFastISel());
2310b57cec5SDimitry Andric         LLVM_DEBUG(
2320b57cec5SDimitry Andric             dbgs() << "\tFastISel is "
2330b57cec5SDimitry Andric                    << (IS.TM.Options.EnableFastISel ? "enabled" : "disabled")
2340b57cec5SDimitry Andric                    << "\n");
2350b57cec5SDimitry Andric       }
2360b57cec5SDimitry Andric     }
2370b57cec5SDimitry Andric 
2380b57cec5SDimitry Andric     ~OptLevelChanger() {
2390b57cec5SDimitry Andric       if (IS.OptLevel == SavedOptLevel)
2400b57cec5SDimitry Andric         return;
2410b57cec5SDimitry Andric       LLVM_DEBUG(dbgs() << "\nRestoring optimization level for Function "
2420b57cec5SDimitry Andric                         << IS.MF->getFunction().getName() << "\n");
2430b57cec5SDimitry Andric       LLVM_DEBUG(dbgs() << "\tBefore: -O" << IS.OptLevel << " ; After: -O"
2440b57cec5SDimitry Andric                         << SavedOptLevel << "\n");
2450b57cec5SDimitry Andric       IS.OptLevel = SavedOptLevel;
2460b57cec5SDimitry Andric       IS.TM.setOptLevel(SavedOptLevel);
2470b57cec5SDimitry Andric       IS.TM.setFastISel(SavedFastISel);
2480b57cec5SDimitry Andric     }
2490b57cec5SDimitry Andric   };
2500b57cec5SDimitry Andric 
2510b57cec5SDimitry Andric   //===--------------------------------------------------------------------===//
2520b57cec5SDimitry Andric   /// createDefaultScheduler - This creates an instruction scheduler appropriate
2530b57cec5SDimitry Andric   /// for the target.
2540b57cec5SDimitry Andric   ScheduleDAGSDNodes* createDefaultScheduler(SelectionDAGISel *IS,
2550b57cec5SDimitry Andric                                              CodeGenOpt::Level OptLevel) {
2560b57cec5SDimitry Andric     const TargetLowering *TLI = IS->TLI;
2570b57cec5SDimitry Andric     const TargetSubtargetInfo &ST = IS->MF->getSubtarget();
2580b57cec5SDimitry Andric 
2590b57cec5SDimitry Andric     // Try first to see if the Target has its own way of selecting a scheduler
2600b57cec5SDimitry Andric     if (auto *SchedulerCtor = ST.getDAGScheduler(OptLevel)) {
2610b57cec5SDimitry Andric       return SchedulerCtor(IS, OptLevel);
2620b57cec5SDimitry Andric     }
2630b57cec5SDimitry Andric 
2640b57cec5SDimitry Andric     if (OptLevel == CodeGenOpt::None ||
2650b57cec5SDimitry Andric         (ST.enableMachineScheduler() && ST.enableMachineSchedDefaultSched()) ||
2660b57cec5SDimitry Andric         TLI->getSchedulingPreference() == Sched::Source)
2670b57cec5SDimitry Andric       return createSourceListDAGScheduler(IS, OptLevel);
2680b57cec5SDimitry Andric     if (TLI->getSchedulingPreference() == Sched::RegPressure)
2690b57cec5SDimitry Andric       return createBURRListDAGScheduler(IS, OptLevel);
2700b57cec5SDimitry Andric     if (TLI->getSchedulingPreference() == Sched::Hybrid)
2710b57cec5SDimitry Andric       return createHybridListDAGScheduler(IS, OptLevel);
2720b57cec5SDimitry Andric     if (TLI->getSchedulingPreference() == Sched::VLIW)
2730b57cec5SDimitry Andric       return createVLIWDAGScheduler(IS, OptLevel);
274fe6060f1SDimitry Andric     if (TLI->getSchedulingPreference() == Sched::Fast)
275fe6060f1SDimitry Andric       return createFastDAGScheduler(IS, OptLevel);
276fe6060f1SDimitry Andric     if (TLI->getSchedulingPreference() == Sched::Linearize)
277fe6060f1SDimitry Andric       return createDAGLinearizer(IS, OptLevel);
2780b57cec5SDimitry Andric     assert(TLI->getSchedulingPreference() == Sched::ILP &&
2790b57cec5SDimitry Andric            "Unknown sched type!");
2800b57cec5SDimitry Andric     return createILPListDAGScheduler(IS, OptLevel);
2810b57cec5SDimitry Andric   }
2820b57cec5SDimitry Andric 
2830b57cec5SDimitry Andric } // end namespace llvm
2840b57cec5SDimitry Andric 
2850b57cec5SDimitry Andric // EmitInstrWithCustomInserter - This method should be implemented by targets
2860b57cec5SDimitry Andric // that mark instructions with the 'usesCustomInserter' flag.  These
2870b57cec5SDimitry Andric // instructions are special in various ways, which require special support to
2880b57cec5SDimitry Andric // insert.  The specified MachineInstr is created but not inserted into any
2890b57cec5SDimitry Andric // basic blocks, and this method is called to expand it into a sequence of
2900b57cec5SDimitry Andric // instructions, potentially also creating new basic blocks and control flow.
2910b57cec5SDimitry Andric // When new basic blocks are inserted and the edges from MBB to its successors
2920b57cec5SDimitry Andric // are modified, the method should insert pairs of <OldSucc, NewSucc> into the
2930b57cec5SDimitry Andric // DenseMap.
2940b57cec5SDimitry Andric MachineBasicBlock *
2950b57cec5SDimitry Andric TargetLowering::EmitInstrWithCustomInserter(MachineInstr &MI,
2960b57cec5SDimitry Andric                                             MachineBasicBlock *MBB) const {
2970b57cec5SDimitry Andric #ifndef NDEBUG
2980b57cec5SDimitry Andric   dbgs() << "If a target marks an instruction with "
2990b57cec5SDimitry Andric           "'usesCustomInserter', it must implement "
300*0eae32dcSDimitry Andric           "TargetLowering::EmitInstrWithCustomInserter!\n";
3010b57cec5SDimitry Andric #endif
3020b57cec5SDimitry Andric   llvm_unreachable(nullptr);
3030b57cec5SDimitry Andric }
3040b57cec5SDimitry Andric 
3050b57cec5SDimitry Andric void TargetLowering::AdjustInstrPostInstrSelection(MachineInstr &MI,
3060b57cec5SDimitry Andric                                                    SDNode *Node) const {
3070b57cec5SDimitry Andric   assert(!MI.hasPostISelHook() &&
3080b57cec5SDimitry Andric          "If a target marks an instruction with 'hasPostISelHook', "
3090b57cec5SDimitry Andric          "it must implement TargetLowering::AdjustInstrPostInstrSelection!");
3100b57cec5SDimitry Andric }
3110b57cec5SDimitry Andric 
3120b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
3130b57cec5SDimitry Andric // SelectionDAGISel code
3140b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
3150b57cec5SDimitry Andric 
316480093f4SDimitry Andric SelectionDAGISel::SelectionDAGISel(TargetMachine &tm, CodeGenOpt::Level OL)
317480093f4SDimitry Andric     : MachineFunctionPass(ID), TM(tm), FuncInfo(new FunctionLoweringInfo()),
3180b57cec5SDimitry Andric       SwiftError(new SwiftErrorValueTracking()),
3190b57cec5SDimitry Andric       CurDAG(new SelectionDAG(tm, OL)),
320480093f4SDimitry Andric       SDB(std::make_unique<SelectionDAGBuilder>(*CurDAG, *FuncInfo, *SwiftError,
321480093f4SDimitry Andric                                                 OL)),
322480093f4SDimitry Andric       AA(), GFI(), OptLevel(OL), DAGSize(0) {
3230b57cec5SDimitry Andric   initializeGCModuleInfoPass(*PassRegistry::getPassRegistry());
3240b57cec5SDimitry Andric   initializeBranchProbabilityInfoWrapperPassPass(
3250b57cec5SDimitry Andric       *PassRegistry::getPassRegistry());
3260b57cec5SDimitry Andric   initializeAAResultsWrapperPassPass(*PassRegistry::getPassRegistry());
327480093f4SDimitry Andric   initializeTargetLibraryInfoWrapperPassPass(*PassRegistry::getPassRegistry());
3280b57cec5SDimitry Andric }
3290b57cec5SDimitry Andric 
3300b57cec5SDimitry Andric SelectionDAGISel::~SelectionDAGISel() {
3310b57cec5SDimitry Andric   delete CurDAG;
3320b57cec5SDimitry Andric   delete SwiftError;
3330b57cec5SDimitry Andric }
3340b57cec5SDimitry Andric 
3350b57cec5SDimitry Andric void SelectionDAGISel::getAnalysisUsage(AnalysisUsage &AU) const {
3360b57cec5SDimitry Andric   if (OptLevel != CodeGenOpt::None)
3370b57cec5SDimitry Andric     AU.addRequired<AAResultsWrapperPass>();
3380b57cec5SDimitry Andric   AU.addRequired<GCModuleInfo>();
3390b57cec5SDimitry Andric   AU.addRequired<StackProtector>();
3400b57cec5SDimitry Andric   AU.addPreserved<GCModuleInfo>();
3410b57cec5SDimitry Andric   AU.addRequired<TargetLibraryInfoWrapperPass>();
3420b57cec5SDimitry Andric   AU.addRequired<TargetTransformInfoWrapperPass>();
3430b57cec5SDimitry Andric   if (UseMBPI && OptLevel != CodeGenOpt::None)
3440b57cec5SDimitry Andric     AU.addRequired<BranchProbabilityInfoWrapperPass>();
345480093f4SDimitry Andric   AU.addRequired<ProfileSummaryInfoWrapperPass>();
3465ffd83dbSDimitry Andric   if (OptLevel != CodeGenOpt::None)
347480093f4SDimitry Andric     LazyBlockFrequencyInfoPass::getLazyBFIAnalysisUsage(AU);
3480b57cec5SDimitry Andric   MachineFunctionPass::getAnalysisUsage(AU);
3490b57cec5SDimitry Andric }
3500b57cec5SDimitry Andric 
3510b57cec5SDimitry Andric /// SplitCriticalSideEffectEdges - Look for critical edges with a PHI value that
3520b57cec5SDimitry Andric /// may trap on it.  In this case we have to split the edge so that the path
3530b57cec5SDimitry Andric /// through the predecessor block that doesn't go to the phi block doesn't
3540b57cec5SDimitry Andric /// execute the possibly trapping instruction. If available, we pass domtree
3550b57cec5SDimitry Andric /// and loop info to be updated when we split critical edges. This is because
3560b57cec5SDimitry Andric /// SelectionDAGISel preserves these analyses.
3570b57cec5SDimitry Andric /// This is required for correctness, so it must be done at -O0.
3580b57cec5SDimitry Andric ///
3590b57cec5SDimitry Andric static void SplitCriticalSideEffectEdges(Function &Fn, DominatorTree *DT,
3600b57cec5SDimitry Andric                                          LoopInfo *LI) {
3610b57cec5SDimitry Andric   // Loop for blocks with phi nodes.
3620b57cec5SDimitry Andric   for (BasicBlock &BB : Fn) {
3630b57cec5SDimitry Andric     PHINode *PN = dyn_cast<PHINode>(BB.begin());
3640b57cec5SDimitry Andric     if (!PN) continue;
3650b57cec5SDimitry Andric 
3660b57cec5SDimitry Andric   ReprocessBlock:
3670b57cec5SDimitry Andric     // For each block with a PHI node, check to see if any of the input values
3680b57cec5SDimitry Andric     // are potentially trapping constant expressions.  Constant expressions are
3690b57cec5SDimitry Andric     // the only potentially trapping value that can occur as the argument to a
3700b57cec5SDimitry Andric     // PHI.
3710b57cec5SDimitry Andric     for (BasicBlock::iterator I = BB.begin(); (PN = dyn_cast<PHINode>(I)); ++I)
3720b57cec5SDimitry Andric       for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i) {
3730b57cec5SDimitry Andric         ConstantExpr *CE = dyn_cast<ConstantExpr>(PN->getIncomingValue(i));
3740b57cec5SDimitry Andric         if (!CE || !CE->canTrap()) continue;
3750b57cec5SDimitry Andric 
3760b57cec5SDimitry Andric         // The only case we have to worry about is when the edge is critical.
3770b57cec5SDimitry Andric         // Since this block has a PHI Node, we assume it has multiple input
3780b57cec5SDimitry Andric         // edges: check to see if the pred has multiple successors.
3790b57cec5SDimitry Andric         BasicBlock *Pred = PN->getIncomingBlock(i);
3800b57cec5SDimitry Andric         if (Pred->getTerminator()->getNumSuccessors() == 1)
3810b57cec5SDimitry Andric           continue;
3820b57cec5SDimitry Andric 
3830b57cec5SDimitry Andric         // Okay, we have to split this edge.
3840b57cec5SDimitry Andric         SplitCriticalEdge(
3850b57cec5SDimitry Andric             Pred->getTerminator(), GetSuccessorNumber(Pred, &BB),
3860b57cec5SDimitry Andric             CriticalEdgeSplittingOptions(DT, LI).setMergeIdenticalEdges());
3870b57cec5SDimitry Andric         goto ReprocessBlock;
3880b57cec5SDimitry Andric       }
3890b57cec5SDimitry Andric   }
3900b57cec5SDimitry Andric }
3910b57cec5SDimitry Andric 
3920b57cec5SDimitry Andric static void computeUsesMSVCFloatingPoint(const Triple &TT, const Function &F,
3930b57cec5SDimitry Andric                                          MachineModuleInfo &MMI) {
3940b57cec5SDimitry Andric   // Only needed for MSVC
3950b57cec5SDimitry Andric   if (!TT.isWindowsMSVCEnvironment())
3960b57cec5SDimitry Andric     return;
3970b57cec5SDimitry Andric 
3980b57cec5SDimitry Andric   // If it's already set, nothing to do.
3990b57cec5SDimitry Andric   if (MMI.usesMSVCFloatingPoint())
4000b57cec5SDimitry Andric     return;
4010b57cec5SDimitry Andric 
4020b57cec5SDimitry Andric   for (const Instruction &I : instructions(F)) {
4030b57cec5SDimitry Andric     if (I.getType()->isFPOrFPVectorTy()) {
4040b57cec5SDimitry Andric       MMI.setUsesMSVCFloatingPoint(true);
4050b57cec5SDimitry Andric       return;
4060b57cec5SDimitry Andric     }
4070b57cec5SDimitry Andric     for (const auto &Op : I.operands()) {
4080b57cec5SDimitry Andric       if (Op->getType()->isFPOrFPVectorTy()) {
4090b57cec5SDimitry Andric         MMI.setUsesMSVCFloatingPoint(true);
4100b57cec5SDimitry Andric         return;
4110b57cec5SDimitry Andric       }
4120b57cec5SDimitry Andric     }
4130b57cec5SDimitry Andric   }
4140b57cec5SDimitry Andric }
4150b57cec5SDimitry Andric 
4160b57cec5SDimitry Andric bool SelectionDAGISel::runOnMachineFunction(MachineFunction &mf) {
4170b57cec5SDimitry Andric   // If we already selected that function, we do not need to run SDISel.
4180b57cec5SDimitry Andric   if (mf.getProperties().hasProperty(
4190b57cec5SDimitry Andric           MachineFunctionProperties::Property::Selected))
4200b57cec5SDimitry Andric     return false;
4210b57cec5SDimitry Andric   // Do some sanity-checking on the command-line options.
4220b57cec5SDimitry Andric   assert((!EnableFastISelAbort || TM.Options.EnableFastISel) &&
4230b57cec5SDimitry Andric          "-fast-isel-abort > 0 requires -fast-isel");
4240b57cec5SDimitry Andric 
4250b57cec5SDimitry Andric   const Function &Fn = mf.getFunction();
4260b57cec5SDimitry Andric   MF = &mf;
4270b57cec5SDimitry Andric 
4280b57cec5SDimitry Andric   // Reset the target options before resetting the optimization
4290b57cec5SDimitry Andric   // level below.
4300b57cec5SDimitry Andric   // FIXME: This is a horrible hack and should be processed via
4310b57cec5SDimitry Andric   // codegen looking at the optimization level explicitly when
4320b57cec5SDimitry Andric   // it wants to look at it.
4330b57cec5SDimitry Andric   TM.resetTargetOptions(Fn);
4340b57cec5SDimitry Andric   // Reset OptLevel to None for optnone functions.
4350b57cec5SDimitry Andric   CodeGenOpt::Level NewOptLevel = OptLevel;
4360b57cec5SDimitry Andric   if (OptLevel != CodeGenOpt::None && skipFunction(Fn))
4370b57cec5SDimitry Andric     NewOptLevel = CodeGenOpt::None;
4380b57cec5SDimitry Andric   OptLevelChanger OLC(*this, NewOptLevel);
4390b57cec5SDimitry Andric 
4400b57cec5SDimitry Andric   TII = MF->getSubtarget().getInstrInfo();
4410b57cec5SDimitry Andric   TLI = MF->getSubtarget().getTargetLowering();
4420b57cec5SDimitry Andric   RegInfo = &MF->getRegInfo();
4438bcb0991SDimitry Andric   LibInfo = &getAnalysis<TargetLibraryInfoWrapperPass>().getTLI(Fn);
4440b57cec5SDimitry Andric   GFI = Fn.hasGC() ? &getAnalysis<GCModuleInfo>().getFunctionInfo(Fn) : nullptr;
4458bcb0991SDimitry Andric   ORE = std::make_unique<OptimizationRemarkEmitter>(&Fn);
4460b57cec5SDimitry Andric   auto *DTWP = getAnalysisIfAvailable<DominatorTreeWrapperPass>();
4470b57cec5SDimitry Andric   DominatorTree *DT = DTWP ? &DTWP->getDomTree() : nullptr;
4480b57cec5SDimitry Andric   auto *LIWP = getAnalysisIfAvailable<LoopInfoWrapperPass>();
4490b57cec5SDimitry Andric   LoopInfo *LI = LIWP ? &LIWP->getLoopInfo() : nullptr;
450480093f4SDimitry Andric   auto *PSI = &getAnalysis<ProfileSummaryInfoWrapperPass>().getPSI();
4515ffd83dbSDimitry Andric   BlockFrequencyInfo *BFI = nullptr;
4525ffd83dbSDimitry Andric   if (PSI && PSI->hasProfileSummary() && OptLevel != CodeGenOpt::None)
4535ffd83dbSDimitry Andric     BFI = &getAnalysis<LazyBlockFrequencyInfoPass>().getBFI();
4540b57cec5SDimitry Andric 
4550b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "\n\n\n=== " << Fn.getName() << "\n");
4560b57cec5SDimitry Andric 
4570b57cec5SDimitry Andric   SplitCriticalSideEffectEdges(const_cast<Function &>(Fn), DT, LI);
4580b57cec5SDimitry Andric 
4590b57cec5SDimitry Andric   CurDAG->init(*MF, *ORE, this, LibInfo,
460480093f4SDimitry Andric                getAnalysisIfAvailable<LegacyDivergenceAnalysis>(), PSI, BFI);
4610b57cec5SDimitry Andric   FuncInfo->set(Fn, *MF, CurDAG);
4620b57cec5SDimitry Andric   SwiftError->setFunction(*MF);
4630b57cec5SDimitry Andric 
4640b57cec5SDimitry Andric   // Now get the optional analyzes if we want to.
4650b57cec5SDimitry Andric   // This is based on the possibly changed OptLevel (after optnone is taken
4660b57cec5SDimitry Andric   // into account).  That's unfortunate but OK because it just means we won't
4670b57cec5SDimitry Andric   // ask for passes that have been required anyway.
4680b57cec5SDimitry Andric 
4690b57cec5SDimitry Andric   if (UseMBPI && OptLevel != CodeGenOpt::None)
4700b57cec5SDimitry Andric     FuncInfo->BPI = &getAnalysis<BranchProbabilityInfoWrapperPass>().getBPI();
4710b57cec5SDimitry Andric   else
4720b57cec5SDimitry Andric     FuncInfo->BPI = nullptr;
4730b57cec5SDimitry Andric 
4740b57cec5SDimitry Andric   if (OptLevel != CodeGenOpt::None)
4750b57cec5SDimitry Andric     AA = &getAnalysis<AAResultsWrapperPass>().getAAResults();
4760b57cec5SDimitry Andric   else
4770b57cec5SDimitry Andric     AA = nullptr;
4780b57cec5SDimitry Andric 
4790b57cec5SDimitry Andric   SDB->init(GFI, AA, LibInfo);
4800b57cec5SDimitry Andric 
4810b57cec5SDimitry Andric   MF->setHasInlineAsm(false);
4820b57cec5SDimitry Andric 
4830b57cec5SDimitry Andric   FuncInfo->SplitCSR = false;
4840b57cec5SDimitry Andric 
4850b57cec5SDimitry Andric   // We split CSR if the target supports it for the given function
4860b57cec5SDimitry Andric   // and the function has only return exits.
4870b57cec5SDimitry Andric   if (OptLevel != CodeGenOpt::None && TLI->supportSplitCSR(MF)) {
4880b57cec5SDimitry Andric     FuncInfo->SplitCSR = true;
4890b57cec5SDimitry Andric 
4900b57cec5SDimitry Andric     // Collect all the return blocks.
4910b57cec5SDimitry Andric     for (const BasicBlock &BB : Fn) {
4920b57cec5SDimitry Andric       if (!succ_empty(&BB))
4930b57cec5SDimitry Andric         continue;
4940b57cec5SDimitry Andric 
4950b57cec5SDimitry Andric       const Instruction *Term = BB.getTerminator();
4960b57cec5SDimitry Andric       if (isa<UnreachableInst>(Term) || isa<ReturnInst>(Term))
4970b57cec5SDimitry Andric         continue;
4980b57cec5SDimitry Andric 
4990b57cec5SDimitry Andric       // Bail out if the exit block is not Return nor Unreachable.
5000b57cec5SDimitry Andric       FuncInfo->SplitCSR = false;
5010b57cec5SDimitry Andric       break;
5020b57cec5SDimitry Andric     }
5030b57cec5SDimitry Andric   }
5040b57cec5SDimitry Andric 
5050b57cec5SDimitry Andric   MachineBasicBlock *EntryMBB = &MF->front();
5060b57cec5SDimitry Andric   if (FuncInfo->SplitCSR)
5070b57cec5SDimitry Andric     // This performs initialization so lowering for SplitCSR will be correct.
5080b57cec5SDimitry Andric     TLI->initializeSplitCSR(EntryMBB);
5090b57cec5SDimitry Andric 
5100b57cec5SDimitry Andric   SelectAllBasicBlocks(Fn);
5110b57cec5SDimitry Andric   if (FastISelFailed && EnableFastISelFallbackReport) {
5120b57cec5SDimitry Andric     DiagnosticInfoISelFallback DiagFallback(Fn);
5130b57cec5SDimitry Andric     Fn.getContext().diagnose(DiagFallback);
5140b57cec5SDimitry Andric   }
5150b57cec5SDimitry Andric 
5160b57cec5SDimitry Andric   // Replace forward-declared registers with the registers containing
5170b57cec5SDimitry Andric   // the desired value.
5180b57cec5SDimitry Andric   // Note: it is important that this happens **before** the call to
5190b57cec5SDimitry Andric   // EmitLiveInCopies, since implementations can skip copies of unused
5200b57cec5SDimitry Andric   // registers. If we don't apply the reg fixups before, some registers may
5210b57cec5SDimitry Andric   // appear as unused and will be skipped, resulting in bad MI.
5220b57cec5SDimitry Andric   MachineRegisterInfo &MRI = MF->getRegInfo();
5235ffd83dbSDimitry Andric   for (DenseMap<Register, Register>::iterator I = FuncInfo->RegFixups.begin(),
5240b57cec5SDimitry Andric                                               E = FuncInfo->RegFixups.end();
5250b57cec5SDimitry Andric        I != E; ++I) {
5265ffd83dbSDimitry Andric     Register From = I->first;
5275ffd83dbSDimitry Andric     Register To = I->second;
5280b57cec5SDimitry Andric     // If To is also scheduled to be replaced, find what its ultimate
5290b57cec5SDimitry Andric     // replacement is.
5300b57cec5SDimitry Andric     while (true) {
5315ffd83dbSDimitry Andric       DenseMap<Register, Register>::iterator J = FuncInfo->RegFixups.find(To);
5320b57cec5SDimitry Andric       if (J == E)
5330b57cec5SDimitry Andric         break;
5340b57cec5SDimitry Andric       To = J->second;
5350b57cec5SDimitry Andric     }
5360b57cec5SDimitry Andric     // Make sure the new register has a sufficiently constrained register class.
5378bcb0991SDimitry Andric     if (Register::isVirtualRegister(From) && Register::isVirtualRegister(To))
5380b57cec5SDimitry Andric       MRI.constrainRegClass(To, MRI.getRegClass(From));
5390b57cec5SDimitry Andric     // Replace it.
5400b57cec5SDimitry Andric 
5410b57cec5SDimitry Andric     // Replacing one register with another won't touch the kill flags.
5420b57cec5SDimitry Andric     // We need to conservatively clear the kill flags as a kill on the old
5430b57cec5SDimitry Andric     // register might dominate existing uses of the new register.
5440b57cec5SDimitry Andric     if (!MRI.use_empty(To))
5450b57cec5SDimitry Andric       MRI.clearKillFlags(From);
5460b57cec5SDimitry Andric     MRI.replaceRegWith(From, To);
5470b57cec5SDimitry Andric   }
5480b57cec5SDimitry Andric 
5490b57cec5SDimitry Andric   // If the first basic block in the function has live ins that need to be
5500b57cec5SDimitry Andric   // copied into vregs, emit the copies into the top of the block before
5510b57cec5SDimitry Andric   // emitting the code for the block.
5520b57cec5SDimitry Andric   const TargetRegisterInfo &TRI = *MF->getSubtarget().getRegisterInfo();
5530b57cec5SDimitry Andric   RegInfo->EmitLiveInCopies(EntryMBB, TRI, *TII);
5540b57cec5SDimitry Andric 
5550b57cec5SDimitry Andric   // Insert copies in the entry block and the return blocks.
5560b57cec5SDimitry Andric   if (FuncInfo->SplitCSR) {
5570b57cec5SDimitry Andric     SmallVector<MachineBasicBlock*, 4> Returns;
5580b57cec5SDimitry Andric     // Collect all the return blocks.
5590b57cec5SDimitry Andric     for (MachineBasicBlock &MBB : mf) {
5600b57cec5SDimitry Andric       if (!MBB.succ_empty())
5610b57cec5SDimitry Andric         continue;
5620b57cec5SDimitry Andric 
5630b57cec5SDimitry Andric       MachineBasicBlock::iterator Term = MBB.getFirstTerminator();
5640b57cec5SDimitry Andric       if (Term != MBB.end() && Term->isReturn()) {
5650b57cec5SDimitry Andric         Returns.push_back(&MBB);
5660b57cec5SDimitry Andric         continue;
5670b57cec5SDimitry Andric       }
5680b57cec5SDimitry Andric     }
5690b57cec5SDimitry Andric     TLI->insertCopiesSplitCSR(EntryMBB, Returns);
5700b57cec5SDimitry Andric   }
5710b57cec5SDimitry Andric 
5720b57cec5SDimitry Andric   DenseMap<unsigned, unsigned> LiveInMap;
5730b57cec5SDimitry Andric   if (!FuncInfo->ArgDbgValues.empty())
5740b57cec5SDimitry Andric     for (std::pair<unsigned, unsigned> LI : RegInfo->liveins())
5750b57cec5SDimitry Andric       if (LI.second)
5760b57cec5SDimitry Andric         LiveInMap.insert(LI);
5770b57cec5SDimitry Andric 
5780b57cec5SDimitry Andric   // Insert DBG_VALUE instructions for function arguments to the entry block.
579349cc55cSDimitry Andric   bool InstrRef = MF->useDebugInstrRef();
5800b57cec5SDimitry Andric   for (unsigned i = 0, e = FuncInfo->ArgDbgValues.size(); i != e; ++i) {
5810b57cec5SDimitry Andric     MachineInstr *MI = FuncInfo->ArgDbgValues[e - i - 1];
582fe6060f1SDimitry Andric     assert(MI->getOpcode() != TargetOpcode::DBG_VALUE_LIST &&
583fe6060f1SDimitry Andric            "Function parameters should not be described by DBG_VALUE_LIST.");
5840b57cec5SDimitry Andric     bool hasFI = MI->getOperand(0).isFI();
5850b57cec5SDimitry Andric     Register Reg =
5860b57cec5SDimitry Andric         hasFI ? TRI.getFrameRegister(*MF) : MI->getOperand(0).getReg();
5878bcb0991SDimitry Andric     if (Register::isPhysicalRegister(Reg))
5880b57cec5SDimitry Andric       EntryMBB->insert(EntryMBB->begin(), MI);
5890b57cec5SDimitry Andric     else {
5900b57cec5SDimitry Andric       MachineInstr *Def = RegInfo->getVRegDef(Reg);
5910b57cec5SDimitry Andric       if (Def) {
5920b57cec5SDimitry Andric         MachineBasicBlock::iterator InsertPos = Def;
5930b57cec5SDimitry Andric         // FIXME: VR def may not be in entry block.
5940b57cec5SDimitry Andric         Def->getParent()->insert(std::next(InsertPos), MI);
5950b57cec5SDimitry Andric       } else
5960b57cec5SDimitry Andric         LLVM_DEBUG(dbgs() << "Dropping debug info for dead vreg"
5978bcb0991SDimitry Andric                           << Register::virtReg2Index(Reg) << "\n");
5980b57cec5SDimitry Andric     }
5990b57cec5SDimitry Andric 
600fe6060f1SDimitry Andric     // Don't try and extend through copies in instruction referencing mode.
601fe6060f1SDimitry Andric     if (InstrRef)
602fe6060f1SDimitry Andric       continue;
603fe6060f1SDimitry Andric 
6040b57cec5SDimitry Andric     // If Reg is live-in then update debug info to track its copy in a vreg.
6050b57cec5SDimitry Andric     DenseMap<unsigned, unsigned>::iterator LDI = LiveInMap.find(Reg);
6060b57cec5SDimitry Andric     if (LDI != LiveInMap.end()) {
6070b57cec5SDimitry Andric       assert(!hasFI && "There's no handling of frame pointer updating here yet "
6080b57cec5SDimitry Andric                        "- add if needed");
6090b57cec5SDimitry Andric       MachineInstr *Def = RegInfo->getVRegDef(LDI->second);
6100b57cec5SDimitry Andric       MachineBasicBlock::iterator InsertPos = Def;
6110b57cec5SDimitry Andric       const MDNode *Variable = MI->getDebugVariable();
6120b57cec5SDimitry Andric       const MDNode *Expr = MI->getDebugExpression();
6130b57cec5SDimitry Andric       DebugLoc DL = MI->getDebugLoc();
6140b57cec5SDimitry Andric       bool IsIndirect = MI->isIndirectDebugValue();
6150b57cec5SDimitry Andric       if (IsIndirect)
6160b57cec5SDimitry Andric         assert(MI->getOperand(1).getImm() == 0 &&
6170b57cec5SDimitry Andric                "DBG_VALUE with nonzero offset");
6180b57cec5SDimitry Andric       assert(cast<DILocalVariable>(Variable)->isValidLocationForIntrinsic(DL) &&
6190b57cec5SDimitry Andric              "Expected inlined-at fields to agree");
620fe6060f1SDimitry Andric       assert(MI->getOpcode() != TargetOpcode::DBG_VALUE_LIST &&
621fe6060f1SDimitry Andric              "Didn't expect to see a DBG_VALUE_LIST here");
6220b57cec5SDimitry Andric       // Def is never a terminator here, so it is ok to increment InsertPos.
6230b57cec5SDimitry Andric       BuildMI(*EntryMBB, ++InsertPos, DL, TII->get(TargetOpcode::DBG_VALUE),
6240b57cec5SDimitry Andric               IsIndirect, LDI->second, Variable, Expr);
6250b57cec5SDimitry Andric 
6260b57cec5SDimitry Andric       // If this vreg is directly copied into an exported register then
6270b57cec5SDimitry Andric       // that COPY instructions also need DBG_VALUE, if it is the only
6280b57cec5SDimitry Andric       // user of LDI->second.
6290b57cec5SDimitry Andric       MachineInstr *CopyUseMI = nullptr;
6300b57cec5SDimitry Andric       for (MachineRegisterInfo::use_instr_iterator
6310b57cec5SDimitry Andric            UI = RegInfo->use_instr_begin(LDI->second),
6320b57cec5SDimitry Andric            E = RegInfo->use_instr_end(); UI != E; ) {
6330b57cec5SDimitry Andric         MachineInstr *UseMI = &*(UI++);
6340b57cec5SDimitry Andric         if (UseMI->isDebugValue()) continue;
6350b57cec5SDimitry Andric         if (UseMI->isCopy() && !CopyUseMI && UseMI->getParent() == EntryMBB) {
6360b57cec5SDimitry Andric           CopyUseMI = UseMI; continue;
6370b57cec5SDimitry Andric         }
6380b57cec5SDimitry Andric         // Otherwise this is another use or second copy use.
6390b57cec5SDimitry Andric         CopyUseMI = nullptr; break;
6400b57cec5SDimitry Andric       }
6415ffd83dbSDimitry Andric       if (CopyUseMI &&
6425ffd83dbSDimitry Andric           TRI.getRegSizeInBits(LDI->second, MRI) ==
6435ffd83dbSDimitry Andric               TRI.getRegSizeInBits(CopyUseMI->getOperand(0).getReg(), MRI)) {
6440b57cec5SDimitry Andric         // Use MI's debug location, which describes where Variable was
6450b57cec5SDimitry Andric         // declared, rather than whatever is attached to CopyUseMI.
6460b57cec5SDimitry Andric         MachineInstr *NewMI =
6470b57cec5SDimitry Andric             BuildMI(*MF, DL, TII->get(TargetOpcode::DBG_VALUE), IsIndirect,
6480b57cec5SDimitry Andric                     CopyUseMI->getOperand(0).getReg(), Variable, Expr);
6490b57cec5SDimitry Andric         MachineBasicBlock::iterator Pos = CopyUseMI;
6500b57cec5SDimitry Andric         EntryMBB->insertAfter(Pos, NewMI);
6510b57cec5SDimitry Andric       }
6520b57cec5SDimitry Andric     }
6530b57cec5SDimitry Andric   }
6540b57cec5SDimitry Andric 
655fe6060f1SDimitry Andric   // For debug-info, in instruction referencing mode, we need to perform some
656fe6060f1SDimitry Andric   // post-isel maintenence.
657fe6060f1SDimitry Andric   MF->finalizeDebugInstrRefs();
658fe6060f1SDimitry Andric 
6590b57cec5SDimitry Andric   // Determine if there are any calls in this machine function.
6600b57cec5SDimitry Andric   MachineFrameInfo &MFI = MF->getFrameInfo();
6610b57cec5SDimitry Andric   for (const auto &MBB : *MF) {
6620b57cec5SDimitry Andric     if (MFI.hasCalls() && MF->hasInlineAsm())
6630b57cec5SDimitry Andric       break;
6640b57cec5SDimitry Andric 
6650b57cec5SDimitry Andric     for (const auto &MI : MBB) {
6660b57cec5SDimitry Andric       const MCInstrDesc &MCID = TII->get(MI.getOpcode());
6670b57cec5SDimitry Andric       if ((MCID.isCall() && !MCID.isReturn()) ||
6680b57cec5SDimitry Andric           MI.isStackAligningInlineAsm()) {
6690b57cec5SDimitry Andric         MFI.setHasCalls(true);
6700b57cec5SDimitry Andric       }
6710b57cec5SDimitry Andric       if (MI.isInlineAsm()) {
6720b57cec5SDimitry Andric         MF->setHasInlineAsm(true);
6730b57cec5SDimitry Andric       }
6740b57cec5SDimitry Andric     }
6750b57cec5SDimitry Andric   }
6760b57cec5SDimitry Andric 
6770b57cec5SDimitry Andric   // Determine if there is a call to setjmp in the machine function.
6780b57cec5SDimitry Andric   MF->setExposesReturnsTwice(Fn.callsFunctionThatReturnsTwice());
6790b57cec5SDimitry Andric 
6800b57cec5SDimitry Andric   // Determine if floating point is used for msvc
6810b57cec5SDimitry Andric   computeUsesMSVCFloatingPoint(TM.getTargetTriple(), Fn, MF->getMMI());
6820b57cec5SDimitry Andric 
6830b57cec5SDimitry Andric   // Release function-specific state. SDB and CurDAG are already cleared
6840b57cec5SDimitry Andric   // at this point.
6850b57cec5SDimitry Andric   FuncInfo->clear();
6860b57cec5SDimitry Andric 
6870b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "*** MachineFunction at end of ISel ***\n");
6880b57cec5SDimitry Andric   LLVM_DEBUG(MF->print(dbgs()));
6890b57cec5SDimitry Andric 
6900b57cec5SDimitry Andric   return true;
6910b57cec5SDimitry Andric }
6920b57cec5SDimitry Andric 
6930b57cec5SDimitry Andric static void reportFastISelFailure(MachineFunction &MF,
6940b57cec5SDimitry Andric                                   OptimizationRemarkEmitter &ORE,
6950b57cec5SDimitry Andric                                   OptimizationRemarkMissed &R,
6960b57cec5SDimitry Andric                                   bool ShouldAbort) {
6970b57cec5SDimitry Andric   // Print the function name explicitly if we don't have a debug location (which
6980b57cec5SDimitry Andric   // makes the diagnostic less useful) or if we're going to emit a raw error.
6990b57cec5SDimitry Andric   if (!R.getLocation().isValid() || ShouldAbort)
7000b57cec5SDimitry Andric     R << (" (in function: " + MF.getName() + ")").str();
7010b57cec5SDimitry Andric 
7020b57cec5SDimitry Andric   if (ShouldAbort)
703349cc55cSDimitry Andric     report_fatal_error(Twine(R.getMsg()));
7040b57cec5SDimitry Andric 
7050b57cec5SDimitry Andric   ORE.emit(R);
7060b57cec5SDimitry Andric }
7070b57cec5SDimitry Andric 
7080b57cec5SDimitry Andric void SelectionDAGISel::SelectBasicBlock(BasicBlock::const_iterator Begin,
7090b57cec5SDimitry Andric                                         BasicBlock::const_iterator End,
7100b57cec5SDimitry Andric                                         bool &HadTailCall) {
7110b57cec5SDimitry Andric   // Allow creating illegal types during DAG building for the basic block.
7120b57cec5SDimitry Andric   CurDAG->NewNodesMustHaveLegalTypes = false;
7130b57cec5SDimitry Andric 
7140b57cec5SDimitry Andric   // Lower the instructions. If a call is emitted as a tail call, cease emitting
7150b57cec5SDimitry Andric   // nodes for this block.
7160b57cec5SDimitry Andric   for (BasicBlock::const_iterator I = Begin; I != End && !SDB->HasTailCall; ++I) {
7170b57cec5SDimitry Andric     if (!ElidedArgCopyInstrs.count(&*I))
7180b57cec5SDimitry Andric       SDB->visit(*I);
7190b57cec5SDimitry Andric   }
7200b57cec5SDimitry Andric 
7210b57cec5SDimitry Andric   // Make sure the root of the DAG is up-to-date.
7220b57cec5SDimitry Andric   CurDAG->setRoot(SDB->getControlRoot());
7230b57cec5SDimitry Andric   HadTailCall = SDB->HasTailCall;
7240b57cec5SDimitry Andric   SDB->resolveOrClearDbgInfo();
7250b57cec5SDimitry Andric   SDB->clear();
7260b57cec5SDimitry Andric 
7270b57cec5SDimitry Andric   // Final step, emit the lowered DAG as machine code.
7280b57cec5SDimitry Andric   CodeGenAndEmitDAG();
7290b57cec5SDimitry Andric }
7300b57cec5SDimitry Andric 
7310b57cec5SDimitry Andric void SelectionDAGISel::ComputeLiveOutVRegInfo() {
732480093f4SDimitry Andric   SmallPtrSet<SDNode *, 16> Added;
7330b57cec5SDimitry Andric   SmallVector<SDNode*, 128> Worklist;
7340b57cec5SDimitry Andric 
7350b57cec5SDimitry Andric   Worklist.push_back(CurDAG->getRoot().getNode());
736480093f4SDimitry Andric   Added.insert(CurDAG->getRoot().getNode());
7370b57cec5SDimitry Andric 
7380b57cec5SDimitry Andric   KnownBits Known;
7390b57cec5SDimitry Andric 
7400b57cec5SDimitry Andric   do {
7410b57cec5SDimitry Andric     SDNode *N = Worklist.pop_back_val();
7420b57cec5SDimitry Andric 
7430b57cec5SDimitry Andric     // Otherwise, add all chain operands to the worklist.
7440b57cec5SDimitry Andric     for (const SDValue &Op : N->op_values())
745480093f4SDimitry Andric       if (Op.getValueType() == MVT::Other && Added.insert(Op.getNode()).second)
7460b57cec5SDimitry Andric         Worklist.push_back(Op.getNode());
7470b57cec5SDimitry Andric 
7480b57cec5SDimitry Andric     // If this is a CopyToReg with a vreg dest, process it.
7490b57cec5SDimitry Andric     if (N->getOpcode() != ISD::CopyToReg)
7500b57cec5SDimitry Andric       continue;
7510b57cec5SDimitry Andric 
7520b57cec5SDimitry Andric     unsigned DestReg = cast<RegisterSDNode>(N->getOperand(1))->getReg();
7538bcb0991SDimitry Andric     if (!Register::isVirtualRegister(DestReg))
7540b57cec5SDimitry Andric       continue;
7550b57cec5SDimitry Andric 
7560b57cec5SDimitry Andric     // Ignore non-integer values.
7570b57cec5SDimitry Andric     SDValue Src = N->getOperand(2);
7580b57cec5SDimitry Andric     EVT SrcVT = Src.getValueType();
7590b57cec5SDimitry Andric     if (!SrcVT.isInteger())
7600b57cec5SDimitry Andric       continue;
7610b57cec5SDimitry Andric 
7620b57cec5SDimitry Andric     unsigned NumSignBits = CurDAG->ComputeNumSignBits(Src);
7630b57cec5SDimitry Andric     Known = CurDAG->computeKnownBits(Src);
7640b57cec5SDimitry Andric     FuncInfo->AddLiveOutRegInfo(DestReg, NumSignBits, Known);
7650b57cec5SDimitry Andric   } while (!Worklist.empty());
7660b57cec5SDimitry Andric }
7670b57cec5SDimitry Andric 
7680b57cec5SDimitry Andric void SelectionDAGISel::CodeGenAndEmitDAG() {
7690b57cec5SDimitry Andric   StringRef GroupName = "sdag";
7700b57cec5SDimitry Andric   StringRef GroupDescription = "Instruction Selection and Scheduling";
7710b57cec5SDimitry Andric   std::string BlockName;
7720b57cec5SDimitry Andric   bool MatchFilterBB = false; (void)MatchFilterBB;
7730b57cec5SDimitry Andric #ifndef NDEBUG
7740b57cec5SDimitry Andric   TargetTransformInfo &TTI =
7750b57cec5SDimitry Andric       getAnalysis<TargetTransformInfoWrapperPass>().getTTI(*FuncInfo->Fn);
7760b57cec5SDimitry Andric #endif
7770b57cec5SDimitry Andric 
7780b57cec5SDimitry Andric   // Pre-type legalization allow creation of any node types.
7790b57cec5SDimitry Andric   CurDAG->NewNodesMustHaveLegalTypes = false;
7800b57cec5SDimitry Andric 
7810b57cec5SDimitry Andric #ifndef NDEBUG
7820b57cec5SDimitry Andric   MatchFilterBB = (FilterDAGBasicBlockName.empty() ||
7830b57cec5SDimitry Andric                    FilterDAGBasicBlockName ==
7840b57cec5SDimitry Andric                        FuncInfo->MBB->getBasicBlock()->getName());
7850b57cec5SDimitry Andric #endif
7860b57cec5SDimitry Andric #ifdef NDEBUG
787480093f4SDimitry Andric   if (ViewDAGCombine1 || ViewLegalizeTypesDAGs || ViewDAGCombineLT ||
788480093f4SDimitry Andric       ViewLegalizeDAGs || ViewDAGCombine2 || ViewISelDAGs || ViewSchedDAGs ||
7890b57cec5SDimitry Andric       ViewSUnitDAGs)
7900b57cec5SDimitry Andric #endif
7910b57cec5SDimitry Andric   {
7920b57cec5SDimitry Andric     BlockName =
7930b57cec5SDimitry Andric         (MF->getName() + ":" + FuncInfo->MBB->getBasicBlock()->getName()).str();
7940b57cec5SDimitry Andric   }
7950b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "Initial selection DAG: "
7960b57cec5SDimitry Andric                     << printMBBReference(*FuncInfo->MBB) << " '" << BlockName
7970b57cec5SDimitry Andric                     << "'\n";
7980b57cec5SDimitry Andric              CurDAG->dump());
7990b57cec5SDimitry Andric 
800e8d8bef9SDimitry Andric #ifndef NDEBUG
801e8d8bef9SDimitry Andric   if (TTI.hasBranchDivergence())
802349cc55cSDimitry Andric     CurDAG->VerifyDAGDivergence();
803e8d8bef9SDimitry Andric #endif
804e8d8bef9SDimitry Andric 
8050b57cec5SDimitry Andric   if (ViewDAGCombine1 && MatchFilterBB)
8060b57cec5SDimitry Andric     CurDAG->viewGraph("dag-combine1 input for " + BlockName);
8070b57cec5SDimitry Andric 
8080b57cec5SDimitry Andric   // Run the DAG combiner in pre-legalize mode.
8090b57cec5SDimitry Andric   {
8100b57cec5SDimitry Andric     NamedRegionTimer T("combine1", "DAG Combining 1", GroupName,
8110b57cec5SDimitry Andric                        GroupDescription, TimePassesIsEnabled);
8120b57cec5SDimitry Andric     CurDAG->Combine(BeforeLegalizeTypes, AA, OptLevel);
8130b57cec5SDimitry Andric   }
8140b57cec5SDimitry Andric 
8150b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "Optimized lowered selection DAG: "
8160b57cec5SDimitry Andric                     << printMBBReference(*FuncInfo->MBB) << " '" << BlockName
8170b57cec5SDimitry Andric                     << "'\n";
8180b57cec5SDimitry Andric              CurDAG->dump());
8190b57cec5SDimitry Andric 
820e8d8bef9SDimitry Andric #ifndef NDEBUG
821e8d8bef9SDimitry Andric   if (TTI.hasBranchDivergence())
822349cc55cSDimitry Andric     CurDAG->VerifyDAGDivergence();
823e8d8bef9SDimitry Andric #endif
824e8d8bef9SDimitry Andric 
8250b57cec5SDimitry Andric   // Second step, hack on the DAG until it only uses operations and types that
8260b57cec5SDimitry Andric   // the target supports.
8270b57cec5SDimitry Andric   if (ViewLegalizeTypesDAGs && MatchFilterBB)
8280b57cec5SDimitry Andric     CurDAG->viewGraph("legalize-types input for " + BlockName);
8290b57cec5SDimitry Andric 
8300b57cec5SDimitry Andric   bool Changed;
8310b57cec5SDimitry Andric   {
8320b57cec5SDimitry Andric     NamedRegionTimer T("legalize_types", "Type Legalization", GroupName,
8330b57cec5SDimitry Andric                        GroupDescription, TimePassesIsEnabled);
8340b57cec5SDimitry Andric     Changed = CurDAG->LegalizeTypes();
8350b57cec5SDimitry Andric   }
8360b57cec5SDimitry Andric 
8370b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "Type-legalized selection DAG: "
8380b57cec5SDimitry Andric                     << printMBBReference(*FuncInfo->MBB) << " '" << BlockName
8390b57cec5SDimitry Andric                     << "'\n";
8400b57cec5SDimitry Andric              CurDAG->dump());
8410b57cec5SDimitry Andric 
842e8d8bef9SDimitry Andric #ifndef NDEBUG
843e8d8bef9SDimitry Andric   if (TTI.hasBranchDivergence())
844349cc55cSDimitry Andric     CurDAG->VerifyDAGDivergence();
845e8d8bef9SDimitry Andric #endif
846e8d8bef9SDimitry Andric 
8470b57cec5SDimitry Andric   // Only allow creation of legal node types.
8480b57cec5SDimitry Andric   CurDAG->NewNodesMustHaveLegalTypes = true;
8490b57cec5SDimitry Andric 
8500b57cec5SDimitry Andric   if (Changed) {
8510b57cec5SDimitry Andric     if (ViewDAGCombineLT && MatchFilterBB)
8520b57cec5SDimitry Andric       CurDAG->viewGraph("dag-combine-lt input for " + BlockName);
8530b57cec5SDimitry Andric 
8540b57cec5SDimitry Andric     // Run the DAG combiner in post-type-legalize mode.
8550b57cec5SDimitry Andric     {
8560b57cec5SDimitry Andric       NamedRegionTimer T("combine_lt", "DAG Combining after legalize types",
8570b57cec5SDimitry Andric                          GroupName, GroupDescription, TimePassesIsEnabled);
8580b57cec5SDimitry Andric       CurDAG->Combine(AfterLegalizeTypes, AA, OptLevel);
8590b57cec5SDimitry Andric     }
8600b57cec5SDimitry Andric 
8610b57cec5SDimitry Andric     LLVM_DEBUG(dbgs() << "Optimized type-legalized selection DAG: "
8620b57cec5SDimitry Andric                       << printMBBReference(*FuncInfo->MBB) << " '" << BlockName
8630b57cec5SDimitry Andric                       << "'\n";
8640b57cec5SDimitry Andric                CurDAG->dump());
865e8d8bef9SDimitry Andric 
866e8d8bef9SDimitry Andric #ifndef NDEBUG
867e8d8bef9SDimitry Andric     if (TTI.hasBranchDivergence())
868349cc55cSDimitry Andric       CurDAG->VerifyDAGDivergence();
869e8d8bef9SDimitry Andric #endif
8700b57cec5SDimitry Andric   }
8710b57cec5SDimitry Andric 
8720b57cec5SDimitry Andric   {
8730b57cec5SDimitry Andric     NamedRegionTimer T("legalize_vec", "Vector Legalization", GroupName,
8740b57cec5SDimitry Andric                        GroupDescription, TimePassesIsEnabled);
8750b57cec5SDimitry Andric     Changed = CurDAG->LegalizeVectors();
8760b57cec5SDimitry Andric   }
8770b57cec5SDimitry Andric 
8780b57cec5SDimitry Andric   if (Changed) {
8790b57cec5SDimitry Andric     LLVM_DEBUG(dbgs() << "Vector-legalized selection DAG: "
8800b57cec5SDimitry Andric                       << printMBBReference(*FuncInfo->MBB) << " '" << BlockName
8810b57cec5SDimitry Andric                       << "'\n";
8820b57cec5SDimitry Andric                CurDAG->dump());
8830b57cec5SDimitry Andric 
884e8d8bef9SDimitry Andric #ifndef NDEBUG
885e8d8bef9SDimitry Andric     if (TTI.hasBranchDivergence())
886349cc55cSDimitry Andric       CurDAG->VerifyDAGDivergence();
887e8d8bef9SDimitry Andric #endif
888e8d8bef9SDimitry Andric 
8890b57cec5SDimitry Andric     {
8900b57cec5SDimitry Andric       NamedRegionTimer T("legalize_types2", "Type Legalization 2", GroupName,
8910b57cec5SDimitry Andric                          GroupDescription, TimePassesIsEnabled);
8920b57cec5SDimitry Andric       CurDAG->LegalizeTypes();
8930b57cec5SDimitry Andric     }
8940b57cec5SDimitry Andric 
8950b57cec5SDimitry Andric     LLVM_DEBUG(dbgs() << "Vector/type-legalized selection DAG: "
8960b57cec5SDimitry Andric                       << printMBBReference(*FuncInfo->MBB) << " '" << BlockName
8970b57cec5SDimitry Andric                       << "'\n";
8980b57cec5SDimitry Andric                CurDAG->dump());
8990b57cec5SDimitry Andric 
900e8d8bef9SDimitry Andric #ifndef NDEBUG
901e8d8bef9SDimitry Andric     if (TTI.hasBranchDivergence())
902349cc55cSDimitry Andric       CurDAG->VerifyDAGDivergence();
903e8d8bef9SDimitry Andric #endif
904e8d8bef9SDimitry Andric 
9050b57cec5SDimitry Andric     if (ViewDAGCombineLT && MatchFilterBB)
9060b57cec5SDimitry Andric       CurDAG->viewGraph("dag-combine-lv input for " + BlockName);
9070b57cec5SDimitry Andric 
9080b57cec5SDimitry Andric     // Run the DAG combiner in post-type-legalize mode.
9090b57cec5SDimitry Andric     {
9100b57cec5SDimitry Andric       NamedRegionTimer T("combine_lv", "DAG Combining after legalize vectors",
9110b57cec5SDimitry Andric                          GroupName, GroupDescription, TimePassesIsEnabled);
9120b57cec5SDimitry Andric       CurDAG->Combine(AfterLegalizeVectorOps, AA, OptLevel);
9130b57cec5SDimitry Andric     }
9140b57cec5SDimitry Andric 
9150b57cec5SDimitry Andric     LLVM_DEBUG(dbgs() << "Optimized vector-legalized selection DAG: "
9160b57cec5SDimitry Andric                       << printMBBReference(*FuncInfo->MBB) << " '" << BlockName
9170b57cec5SDimitry Andric                       << "'\n";
9180b57cec5SDimitry Andric                CurDAG->dump());
9190b57cec5SDimitry Andric 
9200b57cec5SDimitry Andric #ifndef NDEBUG
9210b57cec5SDimitry Andric     if (TTI.hasBranchDivergence())
922349cc55cSDimitry Andric       CurDAG->VerifyDAGDivergence();
9230b57cec5SDimitry Andric #endif
9240b57cec5SDimitry Andric   }
9250b57cec5SDimitry Andric 
9260b57cec5SDimitry Andric   if (ViewLegalizeDAGs && MatchFilterBB)
9270b57cec5SDimitry Andric     CurDAG->viewGraph("legalize input for " + BlockName);
9280b57cec5SDimitry Andric 
9290b57cec5SDimitry Andric   {
9300b57cec5SDimitry Andric     NamedRegionTimer T("legalize", "DAG Legalization", GroupName,
9310b57cec5SDimitry Andric                        GroupDescription, TimePassesIsEnabled);
9320b57cec5SDimitry Andric     CurDAG->Legalize();
9330b57cec5SDimitry Andric   }
9340b57cec5SDimitry Andric 
9350b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "Legalized selection DAG: "
9360b57cec5SDimitry Andric                     << printMBBReference(*FuncInfo->MBB) << " '" << BlockName
9370b57cec5SDimitry Andric                     << "'\n";
9380b57cec5SDimitry Andric              CurDAG->dump());
9390b57cec5SDimitry Andric 
940e8d8bef9SDimitry Andric #ifndef NDEBUG
941e8d8bef9SDimitry Andric   if (TTI.hasBranchDivergence())
942349cc55cSDimitry Andric     CurDAG->VerifyDAGDivergence();
943e8d8bef9SDimitry Andric #endif
944e8d8bef9SDimitry Andric 
9450b57cec5SDimitry Andric   if (ViewDAGCombine2 && MatchFilterBB)
9460b57cec5SDimitry Andric     CurDAG->viewGraph("dag-combine2 input for " + BlockName);
9470b57cec5SDimitry Andric 
9480b57cec5SDimitry Andric   // Run the DAG combiner in post-legalize mode.
9490b57cec5SDimitry Andric   {
9500b57cec5SDimitry Andric     NamedRegionTimer T("combine2", "DAG Combining 2", GroupName,
9510b57cec5SDimitry Andric                        GroupDescription, TimePassesIsEnabled);
9520b57cec5SDimitry Andric     CurDAG->Combine(AfterLegalizeDAG, AA, OptLevel);
9530b57cec5SDimitry Andric   }
9540b57cec5SDimitry Andric 
9550b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "Optimized legalized selection DAG: "
9560b57cec5SDimitry Andric                     << printMBBReference(*FuncInfo->MBB) << " '" << BlockName
9570b57cec5SDimitry Andric                     << "'\n";
9580b57cec5SDimitry Andric              CurDAG->dump());
9590b57cec5SDimitry Andric 
960e8d8bef9SDimitry Andric #ifndef NDEBUG
961e8d8bef9SDimitry Andric   if (TTI.hasBranchDivergence())
962349cc55cSDimitry Andric     CurDAG->VerifyDAGDivergence();
963e8d8bef9SDimitry Andric #endif
964e8d8bef9SDimitry Andric 
9650b57cec5SDimitry Andric   if (OptLevel != CodeGenOpt::None)
9660b57cec5SDimitry Andric     ComputeLiveOutVRegInfo();
9670b57cec5SDimitry Andric 
9680b57cec5SDimitry Andric   if (ViewISelDAGs && MatchFilterBB)
9690b57cec5SDimitry Andric     CurDAG->viewGraph("isel input for " + BlockName);
9700b57cec5SDimitry Andric 
9710b57cec5SDimitry Andric   // Third, instruction select all of the operations to machine code, adding the
9720b57cec5SDimitry Andric   // code to the MachineBasicBlock.
9730b57cec5SDimitry Andric   {
9740b57cec5SDimitry Andric     NamedRegionTimer T("isel", "Instruction Selection", GroupName,
9750b57cec5SDimitry Andric                        GroupDescription, TimePassesIsEnabled);
9760b57cec5SDimitry Andric     DoInstructionSelection();
9770b57cec5SDimitry Andric   }
9780b57cec5SDimitry Andric 
9790b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "Selected selection DAG: "
9800b57cec5SDimitry Andric                     << printMBBReference(*FuncInfo->MBB) << " '" << BlockName
9810b57cec5SDimitry Andric                     << "'\n";
9820b57cec5SDimitry Andric              CurDAG->dump());
9830b57cec5SDimitry Andric 
9840b57cec5SDimitry Andric   if (ViewSchedDAGs && MatchFilterBB)
9850b57cec5SDimitry Andric     CurDAG->viewGraph("scheduler input for " + BlockName);
9860b57cec5SDimitry Andric 
9870b57cec5SDimitry Andric   // Schedule machine code.
9880b57cec5SDimitry Andric   ScheduleDAGSDNodes *Scheduler = CreateScheduler();
9890b57cec5SDimitry Andric   {
9900b57cec5SDimitry Andric     NamedRegionTimer T("sched", "Instruction Scheduling", GroupName,
9910b57cec5SDimitry Andric                        GroupDescription, TimePassesIsEnabled);
9920b57cec5SDimitry Andric     Scheduler->Run(CurDAG, FuncInfo->MBB);
9930b57cec5SDimitry Andric   }
9940b57cec5SDimitry Andric 
9950b57cec5SDimitry Andric   if (ViewSUnitDAGs && MatchFilterBB)
9960b57cec5SDimitry Andric     Scheduler->viewGraph();
9970b57cec5SDimitry Andric 
9980b57cec5SDimitry Andric   // Emit machine code to BB.  This can change 'BB' to the last block being
9990b57cec5SDimitry Andric   // inserted into.
10000b57cec5SDimitry Andric   MachineBasicBlock *FirstMBB = FuncInfo->MBB, *LastMBB;
10010b57cec5SDimitry Andric   {
10020b57cec5SDimitry Andric     NamedRegionTimer T("emit", "Instruction Creation", GroupName,
10030b57cec5SDimitry Andric                        GroupDescription, TimePassesIsEnabled);
10040b57cec5SDimitry Andric 
10050b57cec5SDimitry Andric     // FuncInfo->InsertPt is passed by reference and set to the end of the
10060b57cec5SDimitry Andric     // scheduled instructions.
10070b57cec5SDimitry Andric     LastMBB = FuncInfo->MBB = Scheduler->EmitSchedule(FuncInfo->InsertPt);
10080b57cec5SDimitry Andric   }
10090b57cec5SDimitry Andric 
10100b57cec5SDimitry Andric   // If the block was split, make sure we update any references that are used to
10110b57cec5SDimitry Andric   // update PHI nodes later on.
10120b57cec5SDimitry Andric   if (FirstMBB != LastMBB)
10130b57cec5SDimitry Andric     SDB->UpdateSplitBlock(FirstMBB, LastMBB);
10140b57cec5SDimitry Andric 
10150b57cec5SDimitry Andric   // Free the scheduler state.
10160b57cec5SDimitry Andric   {
10170b57cec5SDimitry Andric     NamedRegionTimer T("cleanup", "Instruction Scheduling Cleanup", GroupName,
10180b57cec5SDimitry Andric                        GroupDescription, TimePassesIsEnabled);
10190b57cec5SDimitry Andric     delete Scheduler;
10200b57cec5SDimitry Andric   }
10210b57cec5SDimitry Andric 
10220b57cec5SDimitry Andric   // Free the SelectionDAG state, now that we're finished with it.
10230b57cec5SDimitry Andric   CurDAG->clear();
10240b57cec5SDimitry Andric }
10250b57cec5SDimitry Andric 
10260b57cec5SDimitry Andric namespace {
10270b57cec5SDimitry Andric 
10280b57cec5SDimitry Andric /// ISelUpdater - helper class to handle updates of the instruction selection
10290b57cec5SDimitry Andric /// graph.
10300b57cec5SDimitry Andric class ISelUpdater : public SelectionDAG::DAGUpdateListener {
10310b57cec5SDimitry Andric   SelectionDAG::allnodes_iterator &ISelPosition;
10320b57cec5SDimitry Andric 
10330b57cec5SDimitry Andric public:
10340b57cec5SDimitry Andric   ISelUpdater(SelectionDAG &DAG, SelectionDAG::allnodes_iterator &isp)
10350b57cec5SDimitry Andric     : SelectionDAG::DAGUpdateListener(DAG), ISelPosition(isp) {}
10360b57cec5SDimitry Andric 
10370b57cec5SDimitry Andric   /// NodeDeleted - Handle nodes deleted from the graph. If the node being
10380b57cec5SDimitry Andric   /// deleted is the current ISelPosition node, update ISelPosition.
10390b57cec5SDimitry Andric   ///
10400b57cec5SDimitry Andric   void NodeDeleted(SDNode *N, SDNode *E) override {
10410b57cec5SDimitry Andric     if (ISelPosition == SelectionDAG::allnodes_iterator(N))
10420b57cec5SDimitry Andric       ++ISelPosition;
10430b57cec5SDimitry Andric   }
10440b57cec5SDimitry Andric };
10450b57cec5SDimitry Andric 
10460b57cec5SDimitry Andric } // end anonymous namespace
10470b57cec5SDimitry Andric 
10480b57cec5SDimitry Andric // This function is used to enforce the topological node id property
1049349cc55cSDimitry Andric // leveraged during instruction selection. Before the selection process all
1050349cc55cSDimitry Andric // nodes are given a non-negative id such that all nodes have a greater id than
10510b57cec5SDimitry Andric // their operands. As this holds transitively we can prune checks that a node N
10520b57cec5SDimitry Andric // is a predecessor of M another by not recursively checking through M's
1053349cc55cSDimitry Andric // operands if N's ID is larger than M's ID. This significantly improves
1054349cc55cSDimitry Andric // performance of various legality checks (e.g. IsLegalToFold / UpdateChains).
10550b57cec5SDimitry Andric 
1056349cc55cSDimitry Andric // However, when we fuse multiple nodes into a single node during the
1057349cc55cSDimitry Andric // selection we may induce a predecessor relationship between inputs and
1058349cc55cSDimitry Andric // outputs of distinct nodes being merged, violating the topological property.
1059349cc55cSDimitry Andric // Should a fused node have a successor which has yet to be selected,
1060349cc55cSDimitry Andric // our legality checks would be incorrect. To avoid this we mark all unselected
1061349cc55cSDimitry Andric // successor nodes, i.e. id != -1, as invalid for pruning by bit-negating (x =>
10620b57cec5SDimitry Andric // (-(x+1))) the ids and modify our pruning check to ignore negative Ids of M.
10630b57cec5SDimitry Andric // We use bit-negation to more clearly enforce that node id -1 can only be
1064349cc55cSDimitry Andric // achieved by selected nodes. As the conversion is reversable to the original
1065349cc55cSDimitry Andric // Id, topological pruning can still be leveraged when looking for unselected
1066349cc55cSDimitry Andric // nodes. This method is called internally in all ISel replacement related
1067349cc55cSDimitry Andric // functions.
10680b57cec5SDimitry Andric void SelectionDAGISel::EnforceNodeIdInvariant(SDNode *Node) {
10690b57cec5SDimitry Andric   SmallVector<SDNode *, 4> Nodes;
10700b57cec5SDimitry Andric   Nodes.push_back(Node);
10710b57cec5SDimitry Andric 
10720b57cec5SDimitry Andric   while (!Nodes.empty()) {
10730b57cec5SDimitry Andric     SDNode *N = Nodes.pop_back_val();
10740b57cec5SDimitry Andric     for (auto *U : N->uses()) {
10750b57cec5SDimitry Andric       auto UId = U->getNodeId();
10760b57cec5SDimitry Andric       if (UId > 0) {
10770b57cec5SDimitry Andric         InvalidateNodeId(U);
10780b57cec5SDimitry Andric         Nodes.push_back(U);
10790b57cec5SDimitry Andric       }
10800b57cec5SDimitry Andric     }
10810b57cec5SDimitry Andric   }
10820b57cec5SDimitry Andric }
10830b57cec5SDimitry Andric 
1084349cc55cSDimitry Andric // InvalidateNodeId - As explained in EnforceNodeIdInvariant, mark a
10850b57cec5SDimitry Andric // NodeId with the equivalent node id which is invalid for topological
10860b57cec5SDimitry Andric // pruning.
10870b57cec5SDimitry Andric void SelectionDAGISel::InvalidateNodeId(SDNode *N) {
10880b57cec5SDimitry Andric   int InvalidId = -(N->getNodeId() + 1);
10890b57cec5SDimitry Andric   N->setNodeId(InvalidId);
10900b57cec5SDimitry Andric }
10910b57cec5SDimitry Andric 
10920b57cec5SDimitry Andric // getUninvalidatedNodeId - get original uninvalidated node id.
10930b57cec5SDimitry Andric int SelectionDAGISel::getUninvalidatedNodeId(SDNode *N) {
10940b57cec5SDimitry Andric   int Id = N->getNodeId();
10950b57cec5SDimitry Andric   if (Id < -1)
10960b57cec5SDimitry Andric     return -(Id + 1);
10970b57cec5SDimitry Andric   return Id;
10980b57cec5SDimitry Andric }
10990b57cec5SDimitry Andric 
11000b57cec5SDimitry Andric void SelectionDAGISel::DoInstructionSelection() {
11010b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "===== Instruction selection begins: "
11020b57cec5SDimitry Andric                     << printMBBReference(*FuncInfo->MBB) << " '"
11030b57cec5SDimitry Andric                     << FuncInfo->MBB->getName() << "'\n");
11040b57cec5SDimitry Andric 
11050b57cec5SDimitry Andric   PreprocessISelDAG();
11060b57cec5SDimitry Andric 
11070b57cec5SDimitry Andric   // Select target instructions for the DAG.
11080b57cec5SDimitry Andric   {
11090b57cec5SDimitry Andric     // Number all nodes with a topological order and set DAGSize.
11100b57cec5SDimitry Andric     DAGSize = CurDAG->AssignTopologicalOrder();
11110b57cec5SDimitry Andric 
11120b57cec5SDimitry Andric     // Create a dummy node (which is not added to allnodes), that adds
11130b57cec5SDimitry Andric     // a reference to the root node, preventing it from being deleted,
11140b57cec5SDimitry Andric     // and tracking any changes of the root.
11150b57cec5SDimitry Andric     HandleSDNode Dummy(CurDAG->getRoot());
11160b57cec5SDimitry Andric     SelectionDAG::allnodes_iterator ISelPosition (CurDAG->getRoot().getNode());
11170b57cec5SDimitry Andric     ++ISelPosition;
11180b57cec5SDimitry Andric 
11190b57cec5SDimitry Andric     // Make sure that ISelPosition gets properly updated when nodes are deleted
11200b57cec5SDimitry Andric     // in calls made from this function.
11210b57cec5SDimitry Andric     ISelUpdater ISU(*CurDAG, ISelPosition);
11220b57cec5SDimitry Andric 
11230b57cec5SDimitry Andric     // The AllNodes list is now topological-sorted. Visit the
11240b57cec5SDimitry Andric     // nodes by starting at the end of the list (the root of the
11250b57cec5SDimitry Andric     // graph) and preceding back toward the beginning (the entry
11260b57cec5SDimitry Andric     // node).
11270b57cec5SDimitry Andric     while (ISelPosition != CurDAG->allnodes_begin()) {
11280b57cec5SDimitry Andric       SDNode *Node = &*--ISelPosition;
11290b57cec5SDimitry Andric       // Skip dead nodes. DAGCombiner is expected to eliminate all dead nodes,
11300b57cec5SDimitry Andric       // but there are currently some corner cases that it misses. Also, this
11310b57cec5SDimitry Andric       // makes it theoretically possible to disable the DAGCombiner.
11320b57cec5SDimitry Andric       if (Node->use_empty())
11330b57cec5SDimitry Andric         continue;
11340b57cec5SDimitry Andric 
11350b57cec5SDimitry Andric #ifndef NDEBUG
11360b57cec5SDimitry Andric       SmallVector<SDNode *, 4> Nodes;
11370b57cec5SDimitry Andric       Nodes.push_back(Node);
11380b57cec5SDimitry Andric 
11390b57cec5SDimitry Andric       while (!Nodes.empty()) {
11400b57cec5SDimitry Andric         auto N = Nodes.pop_back_val();
11410b57cec5SDimitry Andric         if (N->getOpcode() == ISD::TokenFactor || N->getNodeId() < 0)
11420b57cec5SDimitry Andric           continue;
11430b57cec5SDimitry Andric         for (const SDValue &Op : N->op_values()) {
11440b57cec5SDimitry Andric           if (Op->getOpcode() == ISD::TokenFactor)
11450b57cec5SDimitry Andric             Nodes.push_back(Op.getNode());
11460b57cec5SDimitry Andric           else {
11470b57cec5SDimitry Andric             // We rely on topological ordering of node ids for checking for
11480b57cec5SDimitry Andric             // cycles when fusing nodes during selection. All unselected nodes
11490b57cec5SDimitry Andric             // successors of an already selected node should have a negative id.
11500b57cec5SDimitry Andric             // This assertion will catch such cases. If this assertion triggers
11510b57cec5SDimitry Andric             // it is likely you using DAG-level Value/Node replacement functions
11520b57cec5SDimitry Andric             // (versus equivalent ISEL replacement) in backend-specific
11530b57cec5SDimitry Andric             // selections. See comment in EnforceNodeIdInvariant for more
11540b57cec5SDimitry Andric             // details.
11550b57cec5SDimitry Andric             assert(Op->getNodeId() != -1 &&
11560b57cec5SDimitry Andric                    "Node has already selected predecessor node");
11570b57cec5SDimitry Andric           }
11580b57cec5SDimitry Andric         }
11590b57cec5SDimitry Andric       }
11600b57cec5SDimitry Andric #endif
11610b57cec5SDimitry Andric 
11620b57cec5SDimitry Andric       // When we are using non-default rounding modes or FP exception behavior
11630b57cec5SDimitry Andric       // FP operations are represented by StrictFP pseudo-operations.  For
11640b57cec5SDimitry Andric       // targets that do not (yet) understand strict FP operations directly,
11650b57cec5SDimitry Andric       // we convert them to normal FP opcodes instead at this point.  This
11660b57cec5SDimitry Andric       // will allow them to be handled by existing target-specific instruction
11670b57cec5SDimitry Andric       // selectors.
1168480093f4SDimitry Andric       if (!TLI->isStrictFPEnabled() && Node->isStrictFPOpcode()) {
1169480093f4SDimitry Andric         // For some opcodes, we need to call TLI->getOperationAction using
1170480093f4SDimitry Andric         // the first operand type instead of the result type.  Note that this
1171480093f4SDimitry Andric         // must match what SelectionDAGLegalize::LegalizeOp is doing.
1172480093f4SDimitry Andric         EVT ActionVT;
1173480093f4SDimitry Andric         switch (Node->getOpcode()) {
1174480093f4SDimitry Andric         case ISD::STRICT_SINT_TO_FP:
1175480093f4SDimitry Andric         case ISD::STRICT_UINT_TO_FP:
1176480093f4SDimitry Andric         case ISD::STRICT_LRINT:
1177480093f4SDimitry Andric         case ISD::STRICT_LLRINT:
1178480093f4SDimitry Andric         case ISD::STRICT_LROUND:
1179480093f4SDimitry Andric         case ISD::STRICT_LLROUND:
1180480093f4SDimitry Andric         case ISD::STRICT_FSETCC:
1181480093f4SDimitry Andric         case ISD::STRICT_FSETCCS:
1182480093f4SDimitry Andric           ActionVT = Node->getOperand(1).getValueType();
1183480093f4SDimitry Andric           break;
1184480093f4SDimitry Andric         default:
1185480093f4SDimitry Andric           ActionVT = Node->getValueType(0);
1186480093f4SDimitry Andric           break;
1187480093f4SDimitry Andric         }
1188480093f4SDimitry Andric         if (TLI->getOperationAction(Node->getOpcode(), ActionVT)
1189480093f4SDimitry Andric             == TargetLowering::Expand)
11900b57cec5SDimitry Andric           Node = CurDAG->mutateStrictFPToFP(Node);
1191480093f4SDimitry Andric       }
11920b57cec5SDimitry Andric 
11930b57cec5SDimitry Andric       LLVM_DEBUG(dbgs() << "\nISEL: Starting selection on root node: ";
11940b57cec5SDimitry Andric                  Node->dump(CurDAG));
11950b57cec5SDimitry Andric 
11960b57cec5SDimitry Andric       Select(Node);
11970b57cec5SDimitry Andric     }
11980b57cec5SDimitry Andric 
11990b57cec5SDimitry Andric     CurDAG->setRoot(Dummy.getValue());
12000b57cec5SDimitry Andric   }
12010b57cec5SDimitry Andric 
12020b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "\n===== Instruction selection ends:\n");
12030b57cec5SDimitry Andric 
12040b57cec5SDimitry Andric   PostprocessISelDAG();
12050b57cec5SDimitry Andric }
12060b57cec5SDimitry Andric 
12070b57cec5SDimitry Andric static bool hasExceptionPointerOrCodeUser(const CatchPadInst *CPI) {
12080b57cec5SDimitry Andric   for (const User *U : CPI->users()) {
12090b57cec5SDimitry Andric     if (const IntrinsicInst *EHPtrCall = dyn_cast<IntrinsicInst>(U)) {
12100b57cec5SDimitry Andric       Intrinsic::ID IID = EHPtrCall->getIntrinsicID();
12110b57cec5SDimitry Andric       if (IID == Intrinsic::eh_exceptionpointer ||
12120b57cec5SDimitry Andric           IID == Intrinsic::eh_exceptioncode)
12130b57cec5SDimitry Andric         return true;
12140b57cec5SDimitry Andric     }
12150b57cec5SDimitry Andric   }
12160b57cec5SDimitry Andric   return false;
12170b57cec5SDimitry Andric }
12180b57cec5SDimitry Andric 
12190b57cec5SDimitry Andric // wasm.landingpad.index intrinsic is for associating a landing pad index number
12200b57cec5SDimitry Andric // with a catchpad instruction. Retrieve the landing pad index in the intrinsic
12210b57cec5SDimitry Andric // and store the mapping in the function.
12220b57cec5SDimitry Andric static void mapWasmLandingPadIndex(MachineBasicBlock *MBB,
12230b57cec5SDimitry Andric                                    const CatchPadInst *CPI) {
12240b57cec5SDimitry Andric   MachineFunction *MF = MBB->getParent();
12250b57cec5SDimitry Andric   // In case of single catch (...), we don't emit LSDA, so we don't need
12260b57cec5SDimitry Andric   // this information.
12270b57cec5SDimitry Andric   bool IsSingleCatchAllClause =
12280b57cec5SDimitry Andric       CPI->getNumArgOperands() == 1 &&
12290b57cec5SDimitry Andric       cast<Constant>(CPI->getArgOperand(0))->isNullValue();
1230349cc55cSDimitry Andric   // cathchpads for longjmp use an empty type list, e.g. catchpad within %0 []
1231349cc55cSDimitry Andric   // and they don't need LSDA info
1232349cc55cSDimitry Andric   bool IsCatchLongjmp = CPI->getNumArgOperands() == 0;
1233349cc55cSDimitry Andric   if (!IsSingleCatchAllClause && !IsCatchLongjmp) {
12340b57cec5SDimitry Andric     // Create a mapping from landing pad label to landing pad index.
12350b57cec5SDimitry Andric     bool IntrFound = false;
12360b57cec5SDimitry Andric     for (const User *U : CPI->users()) {
12370b57cec5SDimitry Andric       if (const auto *Call = dyn_cast<IntrinsicInst>(U)) {
12380b57cec5SDimitry Andric         Intrinsic::ID IID = Call->getIntrinsicID();
12390b57cec5SDimitry Andric         if (IID == Intrinsic::wasm_landingpad_index) {
12400b57cec5SDimitry Andric           Value *IndexArg = Call->getArgOperand(1);
12410b57cec5SDimitry Andric           int Index = cast<ConstantInt>(IndexArg)->getZExtValue();
12420b57cec5SDimitry Andric           MF->setWasmLandingPadIndex(MBB, Index);
12430b57cec5SDimitry Andric           IntrFound = true;
12440b57cec5SDimitry Andric           break;
12450b57cec5SDimitry Andric         }
12460b57cec5SDimitry Andric       }
12470b57cec5SDimitry Andric     }
12480b57cec5SDimitry Andric     assert(IntrFound && "wasm.landingpad.index intrinsic not found!");
12490b57cec5SDimitry Andric     (void)IntrFound;
12500b57cec5SDimitry Andric   }
12510b57cec5SDimitry Andric }
12520b57cec5SDimitry Andric 
12530b57cec5SDimitry Andric /// PrepareEHLandingPad - Emit an EH_LABEL, set up live-in registers, and
12540b57cec5SDimitry Andric /// do other setup for EH landing-pad blocks.
12550b57cec5SDimitry Andric bool SelectionDAGISel::PrepareEHLandingPad() {
12560b57cec5SDimitry Andric   MachineBasicBlock *MBB = FuncInfo->MBB;
12570b57cec5SDimitry Andric   const Constant *PersonalityFn = FuncInfo->Fn->getPersonalityFn();
12580b57cec5SDimitry Andric   const BasicBlock *LLVMBB = MBB->getBasicBlock();
12590b57cec5SDimitry Andric   const TargetRegisterClass *PtrRC =
12600b57cec5SDimitry Andric       TLI->getRegClassFor(TLI->getPointerTy(CurDAG->getDataLayout()));
12610b57cec5SDimitry Andric 
12620b57cec5SDimitry Andric   auto Pers = classifyEHPersonality(PersonalityFn);
12630b57cec5SDimitry Andric 
12640b57cec5SDimitry Andric   // Catchpads have one live-in register, which typically holds the exception
12650b57cec5SDimitry Andric   // pointer or code.
12660b57cec5SDimitry Andric   if (isFuncletEHPersonality(Pers)) {
12670b57cec5SDimitry Andric     if (const auto *CPI = dyn_cast<CatchPadInst>(LLVMBB->getFirstNonPHI())) {
12680b57cec5SDimitry Andric       if (hasExceptionPointerOrCodeUser(CPI)) {
12690b57cec5SDimitry Andric         // Get or create the virtual register to hold the pointer or code.  Mark
12700b57cec5SDimitry Andric         // the live in physreg and copy into the vreg.
12710b57cec5SDimitry Andric         MCPhysReg EHPhysReg = TLI->getExceptionPointerRegister(PersonalityFn);
12720b57cec5SDimitry Andric         assert(EHPhysReg && "target lacks exception pointer register");
12730b57cec5SDimitry Andric         MBB->addLiveIn(EHPhysReg);
12740b57cec5SDimitry Andric         unsigned VReg = FuncInfo->getCatchPadExceptionPointerVReg(CPI, PtrRC);
12750b57cec5SDimitry Andric         BuildMI(*MBB, FuncInfo->InsertPt, SDB->getCurDebugLoc(),
12760b57cec5SDimitry Andric                 TII->get(TargetOpcode::COPY), VReg)
12770b57cec5SDimitry Andric             .addReg(EHPhysReg, RegState::Kill);
12780b57cec5SDimitry Andric       }
12790b57cec5SDimitry Andric     }
12800b57cec5SDimitry Andric     return true;
12810b57cec5SDimitry Andric   }
12820b57cec5SDimitry Andric 
12830b57cec5SDimitry Andric   // Add a label to mark the beginning of the landing pad.  Deletion of the
12840b57cec5SDimitry Andric   // landing pad can thus be detected via the MachineModuleInfo.
12850b57cec5SDimitry Andric   MCSymbol *Label = MF->addLandingPad(MBB);
12860b57cec5SDimitry Andric 
12870b57cec5SDimitry Andric   const MCInstrDesc &II = TII->get(TargetOpcode::EH_LABEL);
12880b57cec5SDimitry Andric   BuildMI(*MBB, FuncInfo->InsertPt, SDB->getCurDebugLoc(), II)
12890b57cec5SDimitry Andric     .addSym(Label);
12900b57cec5SDimitry Andric 
1291e8d8bef9SDimitry Andric   // If the unwinder does not preserve all registers, ensure that the
1292e8d8bef9SDimitry Andric   // function marks the clobbered registers as used.
1293e8d8bef9SDimitry Andric   const TargetRegisterInfo &TRI = *MF->getSubtarget().getRegisterInfo();
1294e8d8bef9SDimitry Andric   if (auto *RegMask = TRI.getCustomEHPadPreservedMask(*MF))
1295e8d8bef9SDimitry Andric     MF->getRegInfo().addPhysRegsUsedFromRegMask(RegMask);
1296e8d8bef9SDimitry Andric 
12970b57cec5SDimitry Andric   if (Pers == EHPersonality::Wasm_CXX) {
12980b57cec5SDimitry Andric     if (const auto *CPI = dyn_cast<CatchPadInst>(LLVMBB->getFirstNonPHI()))
12990b57cec5SDimitry Andric       mapWasmLandingPadIndex(MBB, CPI);
13000b57cec5SDimitry Andric   } else {
13010b57cec5SDimitry Andric     // Assign the call site to the landing pad's begin label.
13020b57cec5SDimitry Andric     MF->setCallSiteLandingPad(Label, SDB->LPadToCallSiteMap[MBB]);
13030b57cec5SDimitry Andric     // Mark exception register as live in.
13040b57cec5SDimitry Andric     if (unsigned Reg = TLI->getExceptionPointerRegister(PersonalityFn))
13050b57cec5SDimitry Andric       FuncInfo->ExceptionPointerVirtReg = MBB->addLiveIn(Reg, PtrRC);
13060b57cec5SDimitry Andric     // Mark exception selector register as live in.
13070b57cec5SDimitry Andric     if (unsigned Reg = TLI->getExceptionSelectorRegister(PersonalityFn))
13080b57cec5SDimitry Andric       FuncInfo->ExceptionSelectorVirtReg = MBB->addLiveIn(Reg, PtrRC);
13090b57cec5SDimitry Andric   }
13100b57cec5SDimitry Andric 
13110b57cec5SDimitry Andric   return true;
13120b57cec5SDimitry Andric }
13130b57cec5SDimitry Andric 
13140b57cec5SDimitry Andric /// isFoldedOrDeadInstruction - Return true if the specified instruction is
13150b57cec5SDimitry Andric /// side-effect free and is either dead or folded into a generated instruction.
13160b57cec5SDimitry Andric /// Return false if it needs to be emitted.
13170b57cec5SDimitry Andric static bool isFoldedOrDeadInstruction(const Instruction *I,
1318480093f4SDimitry Andric                                       const FunctionLoweringInfo &FuncInfo) {
13190b57cec5SDimitry Andric   return !I->mayWriteToMemory() && // Side-effecting instructions aren't folded.
13200b57cec5SDimitry Andric          !I->isTerminator() &&     // Terminators aren't folded.
13210b57cec5SDimitry Andric          !isa<DbgInfoIntrinsic>(I) && // Debug instructions aren't folded.
13220b57cec5SDimitry Andric          !I->isEHPad() &&             // EH pad instructions aren't folded.
1323480093f4SDimitry Andric          !FuncInfo.isExportedInst(I); // Exported instrs must be computed.
13240b57cec5SDimitry Andric }
13250b57cec5SDimitry Andric 
13260b57cec5SDimitry Andric /// Collect llvm.dbg.declare information. This is done after argument lowering
13270b57cec5SDimitry Andric /// in case the declarations refer to arguments.
1328480093f4SDimitry Andric static void processDbgDeclares(FunctionLoweringInfo &FuncInfo) {
1329480093f4SDimitry Andric   MachineFunction *MF = FuncInfo.MF;
13300b57cec5SDimitry Andric   const DataLayout &DL = MF->getDataLayout();
1331480093f4SDimitry Andric   for (const BasicBlock &BB : *FuncInfo.Fn) {
13320b57cec5SDimitry Andric     for (const Instruction &I : BB) {
13330b57cec5SDimitry Andric       const DbgDeclareInst *DI = dyn_cast<DbgDeclareInst>(&I);
13340b57cec5SDimitry Andric       if (!DI)
13350b57cec5SDimitry Andric         continue;
13360b57cec5SDimitry Andric 
13370b57cec5SDimitry Andric       assert(DI->getVariable() && "Missing variable");
13380b57cec5SDimitry Andric       assert(DI->getDebugLoc() && "Missing location");
13390b57cec5SDimitry Andric       const Value *Address = DI->getAddress();
13405ffd83dbSDimitry Andric       if (!Address) {
13415ffd83dbSDimitry Andric         LLVM_DEBUG(dbgs() << "processDbgDeclares skipping " << *DI
13425ffd83dbSDimitry Andric                           << " (bad address)\n");
13430b57cec5SDimitry Andric         continue;
13445ffd83dbSDimitry Andric       }
13450b57cec5SDimitry Andric 
13460b57cec5SDimitry Andric       // Look through casts and constant offset GEPs. These mostly come from
13470b57cec5SDimitry Andric       // inalloca.
13480b57cec5SDimitry Andric       APInt Offset(DL.getTypeSizeInBits(Address->getType()), 0);
13490b57cec5SDimitry Andric       Address = Address->stripAndAccumulateInBoundsConstantOffsets(DL, Offset);
13500b57cec5SDimitry Andric 
13510b57cec5SDimitry Andric       // Check if the variable is a static alloca or a byval or inalloca
13520b57cec5SDimitry Andric       // argument passed in memory. If it is not, then we will ignore this
13530b57cec5SDimitry Andric       // intrinsic and handle this during isel like dbg.value.
13540b57cec5SDimitry Andric       int FI = std::numeric_limits<int>::max();
13550b57cec5SDimitry Andric       if (const auto *AI = dyn_cast<AllocaInst>(Address)) {
1356480093f4SDimitry Andric         auto SI = FuncInfo.StaticAllocaMap.find(AI);
1357480093f4SDimitry Andric         if (SI != FuncInfo.StaticAllocaMap.end())
13580b57cec5SDimitry Andric           FI = SI->second;
13590b57cec5SDimitry Andric       } else if (const auto *Arg = dyn_cast<Argument>(Address))
1360480093f4SDimitry Andric         FI = FuncInfo.getArgumentFrameIndex(Arg);
13610b57cec5SDimitry Andric 
13620b57cec5SDimitry Andric       if (FI == std::numeric_limits<int>::max())
13630b57cec5SDimitry Andric         continue;
13640b57cec5SDimitry Andric 
13650b57cec5SDimitry Andric       DIExpression *Expr = DI->getExpression();
13660b57cec5SDimitry Andric       if (Offset.getBoolValue())
13670b57cec5SDimitry Andric         Expr = DIExpression::prepend(Expr, DIExpression::ApplyOffset,
13680b57cec5SDimitry Andric                                      Offset.getZExtValue());
13695ffd83dbSDimitry Andric       LLVM_DEBUG(dbgs() << "processDbgDeclares: setVariableDbgInfo FI=" << FI
13705ffd83dbSDimitry Andric                         << ", " << *DI << "\n");
13710b57cec5SDimitry Andric       MF->setVariableDbgInfo(DI->getVariable(), Expr, FI, DI->getDebugLoc());
13720b57cec5SDimitry Andric     }
13730b57cec5SDimitry Andric   }
13740b57cec5SDimitry Andric }
13750b57cec5SDimitry Andric 
13760b57cec5SDimitry Andric void SelectionDAGISel::SelectAllBasicBlocks(const Function &Fn) {
13770b57cec5SDimitry Andric   FastISelFailed = false;
13780b57cec5SDimitry Andric   // Initialize the Fast-ISel state, if needed.
13790b57cec5SDimitry Andric   FastISel *FastIS = nullptr;
13800b57cec5SDimitry Andric   if (TM.Options.EnableFastISel) {
13810b57cec5SDimitry Andric     LLVM_DEBUG(dbgs() << "Enabling fast-isel\n");
13820b57cec5SDimitry Andric     FastIS = TLI->createFastISel(*FuncInfo, LibInfo);
13830b57cec5SDimitry Andric   }
13840b57cec5SDimitry Andric 
13850b57cec5SDimitry Andric   ReversePostOrderTraversal<const Function*> RPOT(&Fn);
13860b57cec5SDimitry Andric 
13870b57cec5SDimitry Andric   // Lower arguments up front. An RPO iteration always visits the entry block
13880b57cec5SDimitry Andric   // first.
13890b57cec5SDimitry Andric   assert(*RPOT.begin() == &Fn.getEntryBlock());
13900b57cec5SDimitry Andric   ++NumEntryBlocks;
13910b57cec5SDimitry Andric 
13920b57cec5SDimitry Andric   // Set up FuncInfo for ISel. Entry blocks never have PHIs.
13930b57cec5SDimitry Andric   FuncInfo->MBB = FuncInfo->MBBMap[&Fn.getEntryBlock()];
13940b57cec5SDimitry Andric   FuncInfo->InsertPt = FuncInfo->MBB->begin();
13950b57cec5SDimitry Andric 
1396480093f4SDimitry Andric   CurDAG->setFunctionLoweringInfo(FuncInfo.get());
13970b57cec5SDimitry Andric 
13980b57cec5SDimitry Andric   if (!FastIS) {
13990b57cec5SDimitry Andric     LowerArguments(Fn);
14000b57cec5SDimitry Andric   } else {
14010b57cec5SDimitry Andric     // See if fast isel can lower the arguments.
14020b57cec5SDimitry Andric     FastIS->startNewBlock();
14030b57cec5SDimitry Andric     if (!FastIS->lowerArguments()) {
14040b57cec5SDimitry Andric       FastISelFailed = true;
14050b57cec5SDimitry Andric       // Fast isel failed to lower these arguments
14060b57cec5SDimitry Andric       ++NumFastIselFailLowerArguments;
14070b57cec5SDimitry Andric 
14080b57cec5SDimitry Andric       OptimizationRemarkMissed R("sdagisel", "FastISelFailure",
14090b57cec5SDimitry Andric                                  Fn.getSubprogram(),
14100b57cec5SDimitry Andric                                  &Fn.getEntryBlock());
14110b57cec5SDimitry Andric       R << "FastISel didn't lower all arguments: "
14120b57cec5SDimitry Andric         << ore::NV("Prototype", Fn.getType());
14130b57cec5SDimitry Andric       reportFastISelFailure(*MF, *ORE, R, EnableFastISelAbort > 1);
14140b57cec5SDimitry Andric 
14150b57cec5SDimitry Andric       // Use SelectionDAG argument lowering
14160b57cec5SDimitry Andric       LowerArguments(Fn);
14170b57cec5SDimitry Andric       CurDAG->setRoot(SDB->getControlRoot());
14180b57cec5SDimitry Andric       SDB->clear();
14190b57cec5SDimitry Andric       CodeGenAndEmitDAG();
14200b57cec5SDimitry Andric     }
14210b57cec5SDimitry Andric 
14220b57cec5SDimitry Andric     // If we inserted any instructions at the beginning, make a note of
14230b57cec5SDimitry Andric     // where they are, so we can be sure to emit subsequent instructions
14240b57cec5SDimitry Andric     // after them.
14250b57cec5SDimitry Andric     if (FuncInfo->InsertPt != FuncInfo->MBB->begin())
14260b57cec5SDimitry Andric       FastIS->setLastLocalValue(&*std::prev(FuncInfo->InsertPt));
14270b57cec5SDimitry Andric     else
14280b57cec5SDimitry Andric       FastIS->setLastLocalValue(nullptr);
14290b57cec5SDimitry Andric   }
14300b57cec5SDimitry Andric 
14310b57cec5SDimitry Andric   bool Inserted = SwiftError->createEntriesInEntryBlock(SDB->getCurDebugLoc());
14320b57cec5SDimitry Andric 
14330b57cec5SDimitry Andric   if (FastIS && Inserted)
14340b57cec5SDimitry Andric     FastIS->setLastLocalValue(&*std::prev(FuncInfo->InsertPt));
14350b57cec5SDimitry Andric 
1436480093f4SDimitry Andric   processDbgDeclares(*FuncInfo);
14370b57cec5SDimitry Andric 
14380b57cec5SDimitry Andric   // Iterate over all basic blocks in the function.
14390b57cec5SDimitry Andric   StackProtector &SP = getAnalysis<StackProtector>();
14400b57cec5SDimitry Andric   for (const BasicBlock *LLVMBB : RPOT) {
14410b57cec5SDimitry Andric     if (OptLevel != CodeGenOpt::None) {
14420b57cec5SDimitry Andric       bool AllPredsVisited = true;
1443fe6060f1SDimitry Andric       for (const BasicBlock *Pred : predecessors(LLVMBB)) {
1444fe6060f1SDimitry Andric         if (!FuncInfo->VisitedBBs.count(Pred)) {
14450b57cec5SDimitry Andric           AllPredsVisited = false;
14460b57cec5SDimitry Andric           break;
14470b57cec5SDimitry Andric         }
14480b57cec5SDimitry Andric       }
14490b57cec5SDimitry Andric 
14500b57cec5SDimitry Andric       if (AllPredsVisited) {
14510b57cec5SDimitry Andric         for (const PHINode &PN : LLVMBB->phis())
14520b57cec5SDimitry Andric           FuncInfo->ComputePHILiveOutRegInfo(&PN);
14530b57cec5SDimitry Andric       } else {
14540b57cec5SDimitry Andric         for (const PHINode &PN : LLVMBB->phis())
14550b57cec5SDimitry Andric           FuncInfo->InvalidatePHILiveOutRegInfo(&PN);
14560b57cec5SDimitry Andric       }
14570b57cec5SDimitry Andric 
14580b57cec5SDimitry Andric       FuncInfo->VisitedBBs.insert(LLVMBB);
14590b57cec5SDimitry Andric     }
14600b57cec5SDimitry Andric 
14610b57cec5SDimitry Andric     BasicBlock::const_iterator const Begin =
14620b57cec5SDimitry Andric         LLVMBB->getFirstNonPHI()->getIterator();
14630b57cec5SDimitry Andric     BasicBlock::const_iterator const End = LLVMBB->end();
14640b57cec5SDimitry Andric     BasicBlock::const_iterator BI = End;
14650b57cec5SDimitry Andric 
14660b57cec5SDimitry Andric     FuncInfo->MBB = FuncInfo->MBBMap[LLVMBB];
14670b57cec5SDimitry Andric     if (!FuncInfo->MBB)
14680b57cec5SDimitry Andric       continue; // Some blocks like catchpads have no code or MBB.
14690b57cec5SDimitry Andric 
14700b57cec5SDimitry Andric     // Insert new instructions after any phi or argument setup code.
14710b57cec5SDimitry Andric     FuncInfo->InsertPt = FuncInfo->MBB->end();
14720b57cec5SDimitry Andric 
14730b57cec5SDimitry Andric     // Setup an EH landing-pad block.
14740b57cec5SDimitry Andric     FuncInfo->ExceptionPointerVirtReg = 0;
14750b57cec5SDimitry Andric     FuncInfo->ExceptionSelectorVirtReg = 0;
14760b57cec5SDimitry Andric     if (LLVMBB->isEHPad())
14770b57cec5SDimitry Andric       if (!PrepareEHLandingPad())
14780b57cec5SDimitry Andric         continue;
14790b57cec5SDimitry Andric 
14800b57cec5SDimitry Andric     // Before doing SelectionDAG ISel, see if FastISel has been requested.
14810b57cec5SDimitry Andric     if (FastIS) {
14820b57cec5SDimitry Andric       if (LLVMBB != &Fn.getEntryBlock())
14830b57cec5SDimitry Andric         FastIS->startNewBlock();
14840b57cec5SDimitry Andric 
14850b57cec5SDimitry Andric       unsigned NumFastIselRemaining = std::distance(Begin, End);
14860b57cec5SDimitry Andric 
14870b57cec5SDimitry Andric       // Pre-assign swifterror vregs.
14880b57cec5SDimitry Andric       SwiftError->preassignVRegs(FuncInfo->MBB, Begin, End);
14890b57cec5SDimitry Andric 
14900b57cec5SDimitry Andric       // Do FastISel on as many instructions as possible.
14910b57cec5SDimitry Andric       for (; BI != Begin; --BI) {
14920b57cec5SDimitry Andric         const Instruction *Inst = &*std::prev(BI);
14930b57cec5SDimitry Andric 
14940b57cec5SDimitry Andric         // If we no longer require this instruction, skip it.
1495480093f4SDimitry Andric         if (isFoldedOrDeadInstruction(Inst, *FuncInfo) ||
14960b57cec5SDimitry Andric             ElidedArgCopyInstrs.count(Inst)) {
14970b57cec5SDimitry Andric           --NumFastIselRemaining;
14980b57cec5SDimitry Andric           continue;
14990b57cec5SDimitry Andric         }
15000b57cec5SDimitry Andric 
15010b57cec5SDimitry Andric         // Bottom-up: reset the insert pos at the top, after any local-value
15020b57cec5SDimitry Andric         // instructions.
15030b57cec5SDimitry Andric         FastIS->recomputeInsertPt();
15040b57cec5SDimitry Andric 
15050b57cec5SDimitry Andric         // Try to select the instruction with FastISel.
15060b57cec5SDimitry Andric         if (FastIS->selectInstruction(Inst)) {
15070b57cec5SDimitry Andric           --NumFastIselRemaining;
15080b57cec5SDimitry Andric           ++NumFastIselSuccess;
15090b57cec5SDimitry Andric           // If fast isel succeeded, skip over all the folded instructions, and
15100b57cec5SDimitry Andric           // then see if there is a load right before the selected instructions.
15110b57cec5SDimitry Andric           // Try to fold the load if so.
15120b57cec5SDimitry Andric           const Instruction *BeforeInst = Inst;
15130b57cec5SDimitry Andric           while (BeforeInst != &*Begin) {
15140b57cec5SDimitry Andric             BeforeInst = &*std::prev(BasicBlock::const_iterator(BeforeInst));
1515480093f4SDimitry Andric             if (!isFoldedOrDeadInstruction(BeforeInst, *FuncInfo))
15160b57cec5SDimitry Andric               break;
15170b57cec5SDimitry Andric           }
15180b57cec5SDimitry Andric           if (BeforeInst != Inst && isa<LoadInst>(BeforeInst) &&
15190b57cec5SDimitry Andric               BeforeInst->hasOneUse() &&
15200b57cec5SDimitry Andric               FastIS->tryToFoldLoad(cast<LoadInst>(BeforeInst), Inst)) {
15210b57cec5SDimitry Andric             // If we succeeded, don't re-select the load.
15220b57cec5SDimitry Andric             BI = std::next(BasicBlock::const_iterator(BeforeInst));
15230b57cec5SDimitry Andric             --NumFastIselRemaining;
15240b57cec5SDimitry Andric             ++NumFastIselSuccess;
15250b57cec5SDimitry Andric           }
15260b57cec5SDimitry Andric           continue;
15270b57cec5SDimitry Andric         }
15280b57cec5SDimitry Andric 
15290b57cec5SDimitry Andric         FastISelFailed = true;
15300b57cec5SDimitry Andric 
15310b57cec5SDimitry Andric         // Then handle certain instructions as single-LLVM-Instruction blocks.
15320b57cec5SDimitry Andric         // We cannot separate out GCrelocates to their own blocks since we need
15330b57cec5SDimitry Andric         // to keep track of gc-relocates for a particular gc-statepoint. This is
15340b57cec5SDimitry Andric         // done by SelectionDAGBuilder::LowerAsSTATEPOINT, called before
15350b57cec5SDimitry Andric         // visitGCRelocate.
15365ffd83dbSDimitry Andric         if (isa<CallInst>(Inst) && !isa<GCStatepointInst>(Inst) &&
15375ffd83dbSDimitry Andric             !isa<GCRelocateInst>(Inst) && !isa<GCResultInst>(Inst)) {
15380b57cec5SDimitry Andric           OptimizationRemarkMissed R("sdagisel", "FastISelFailure",
15390b57cec5SDimitry Andric                                      Inst->getDebugLoc(), LLVMBB);
15400b57cec5SDimitry Andric 
15410b57cec5SDimitry Andric           R << "FastISel missed call";
15420b57cec5SDimitry Andric 
15430b57cec5SDimitry Andric           if (R.isEnabled() || EnableFastISelAbort) {
15440b57cec5SDimitry Andric             std::string InstStrStorage;
15450b57cec5SDimitry Andric             raw_string_ostream InstStr(InstStrStorage);
15460b57cec5SDimitry Andric             InstStr << *Inst;
15470b57cec5SDimitry Andric 
15480b57cec5SDimitry Andric             R << ": " << InstStr.str();
15490b57cec5SDimitry Andric           }
15500b57cec5SDimitry Andric 
15510b57cec5SDimitry Andric           reportFastISelFailure(*MF, *ORE, R, EnableFastISelAbort > 2);
15520b57cec5SDimitry Andric 
15530b57cec5SDimitry Andric           if (!Inst->getType()->isVoidTy() && !Inst->getType()->isTokenTy() &&
15540b57cec5SDimitry Andric               !Inst->use_empty()) {
15555ffd83dbSDimitry Andric             Register &R = FuncInfo->ValueMap[Inst];
15560b57cec5SDimitry Andric             if (!R)
15570b57cec5SDimitry Andric               R = FuncInfo->CreateRegs(Inst);
15580b57cec5SDimitry Andric           }
15590b57cec5SDimitry Andric 
15600b57cec5SDimitry Andric           bool HadTailCall = false;
15610b57cec5SDimitry Andric           MachineBasicBlock::iterator SavedInsertPt = FuncInfo->InsertPt;
15620b57cec5SDimitry Andric           SelectBasicBlock(Inst->getIterator(), BI, HadTailCall);
15630b57cec5SDimitry Andric 
15640b57cec5SDimitry Andric           // If the call was emitted as a tail call, we're done with the block.
15650b57cec5SDimitry Andric           // We also need to delete any previously emitted instructions.
15660b57cec5SDimitry Andric           if (HadTailCall) {
15670b57cec5SDimitry Andric             FastIS->removeDeadCode(SavedInsertPt, FuncInfo->MBB->end());
15680b57cec5SDimitry Andric             --BI;
15690b57cec5SDimitry Andric             break;
15700b57cec5SDimitry Andric           }
15710b57cec5SDimitry Andric 
15720b57cec5SDimitry Andric           // Recompute NumFastIselRemaining as Selection DAG instruction
15730b57cec5SDimitry Andric           // selection may have handled the call, input args, etc.
15740b57cec5SDimitry Andric           unsigned RemainingNow = std::distance(Begin, BI);
15750b57cec5SDimitry Andric           NumFastIselFailures += NumFastIselRemaining - RemainingNow;
15760b57cec5SDimitry Andric           NumFastIselRemaining = RemainingNow;
15770b57cec5SDimitry Andric           continue;
15780b57cec5SDimitry Andric         }
15790b57cec5SDimitry Andric 
15800b57cec5SDimitry Andric         OptimizationRemarkMissed R("sdagisel", "FastISelFailure",
15810b57cec5SDimitry Andric                                    Inst->getDebugLoc(), LLVMBB);
15820b57cec5SDimitry Andric 
15830b57cec5SDimitry Andric         bool ShouldAbort = EnableFastISelAbort;
15840b57cec5SDimitry Andric         if (Inst->isTerminator()) {
15850b57cec5SDimitry Andric           // Use a different message for terminator misses.
15860b57cec5SDimitry Andric           R << "FastISel missed terminator";
15870b57cec5SDimitry Andric           // Don't abort for terminator unless the level is really high
15880b57cec5SDimitry Andric           ShouldAbort = (EnableFastISelAbort > 2);
15890b57cec5SDimitry Andric         } else {
15900b57cec5SDimitry Andric           R << "FastISel missed";
15910b57cec5SDimitry Andric         }
15920b57cec5SDimitry Andric 
15930b57cec5SDimitry Andric         if (R.isEnabled() || EnableFastISelAbort) {
15940b57cec5SDimitry Andric           std::string InstStrStorage;
15950b57cec5SDimitry Andric           raw_string_ostream InstStr(InstStrStorage);
15960b57cec5SDimitry Andric           InstStr << *Inst;
15970b57cec5SDimitry Andric           R << ": " << InstStr.str();
15980b57cec5SDimitry Andric         }
15990b57cec5SDimitry Andric 
16000b57cec5SDimitry Andric         reportFastISelFailure(*MF, *ORE, R, ShouldAbort);
16010b57cec5SDimitry Andric 
16020b57cec5SDimitry Andric         NumFastIselFailures += NumFastIselRemaining;
16030b57cec5SDimitry Andric         break;
16040b57cec5SDimitry Andric       }
16050b57cec5SDimitry Andric 
16060b57cec5SDimitry Andric       FastIS->recomputeInsertPt();
16070b57cec5SDimitry Andric     }
16080b57cec5SDimitry Andric 
16090b57cec5SDimitry Andric     if (SP.shouldEmitSDCheck(*LLVMBB)) {
16100b57cec5SDimitry Andric       bool FunctionBasedInstrumentation =
16110b57cec5SDimitry Andric           TLI->getSSPStackGuardCheck(*Fn.getParent());
16120b57cec5SDimitry Andric       SDB->SPDescriptor.initialize(LLVMBB, FuncInfo->MBBMap[LLVMBB],
16130b57cec5SDimitry Andric                                    FunctionBasedInstrumentation);
16140b57cec5SDimitry Andric     }
16150b57cec5SDimitry Andric 
16160b57cec5SDimitry Andric     if (Begin != BI)
16170b57cec5SDimitry Andric       ++NumDAGBlocks;
16180b57cec5SDimitry Andric     else
16190b57cec5SDimitry Andric       ++NumFastIselBlocks;
16200b57cec5SDimitry Andric 
16210b57cec5SDimitry Andric     if (Begin != BI) {
16220b57cec5SDimitry Andric       // Run SelectionDAG instruction selection on the remainder of the block
16230b57cec5SDimitry Andric       // not handled by FastISel. If FastISel is not run, this is the entire
16240b57cec5SDimitry Andric       // block.
16250b57cec5SDimitry Andric       bool HadTailCall;
16260b57cec5SDimitry Andric       SelectBasicBlock(Begin, BI, HadTailCall);
16270b57cec5SDimitry Andric 
16280b57cec5SDimitry Andric       // But if FastISel was run, we already selected some of the block.
16290b57cec5SDimitry Andric       // If we emitted a tail-call, we need to delete any previously emitted
16300b57cec5SDimitry Andric       // instruction that follows it.
1631480093f4SDimitry Andric       if (FastIS && HadTailCall && FuncInfo->InsertPt != FuncInfo->MBB->end())
16320b57cec5SDimitry Andric         FastIS->removeDeadCode(FuncInfo->InsertPt, FuncInfo->MBB->end());
16330b57cec5SDimitry Andric     }
16340b57cec5SDimitry Andric 
16350b57cec5SDimitry Andric     if (FastIS)
16360b57cec5SDimitry Andric       FastIS->finishBasicBlock();
16370b57cec5SDimitry Andric     FinishBasicBlock();
16380b57cec5SDimitry Andric     FuncInfo->PHINodesToUpdate.clear();
16390b57cec5SDimitry Andric     ElidedArgCopyInstrs.clear();
16400b57cec5SDimitry Andric   }
16410b57cec5SDimitry Andric 
16420b57cec5SDimitry Andric   SP.copyToMachineFrameInfo(MF->getFrameInfo());
16430b57cec5SDimitry Andric 
16440b57cec5SDimitry Andric   SwiftError->propagateVRegs();
16450b57cec5SDimitry Andric 
16460b57cec5SDimitry Andric   delete FastIS;
16470b57cec5SDimitry Andric   SDB->clearDanglingDebugInfo();
16480b57cec5SDimitry Andric   SDB->SPDescriptor.resetPerFunctionState();
16490b57cec5SDimitry Andric }
16500b57cec5SDimitry Andric 
16510b57cec5SDimitry Andric void
16520b57cec5SDimitry Andric SelectionDAGISel::FinishBasicBlock() {
16530b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "Total amount of phi nodes to update: "
16540b57cec5SDimitry Andric                     << FuncInfo->PHINodesToUpdate.size() << "\n";
16550b57cec5SDimitry Andric              for (unsigned i = 0, e = FuncInfo->PHINodesToUpdate.size(); i != e;
16560b57cec5SDimitry Andric                   ++i) dbgs()
16570b57cec5SDimitry Andric              << "Node " << i << " : (" << FuncInfo->PHINodesToUpdate[i].first
16580b57cec5SDimitry Andric              << ", " << FuncInfo->PHINodesToUpdate[i].second << ")\n");
16590b57cec5SDimitry Andric 
16600b57cec5SDimitry Andric   // Next, now that we know what the last MBB the LLVM BB expanded is, update
16610b57cec5SDimitry Andric   // PHI nodes in successors.
16620b57cec5SDimitry Andric   for (unsigned i = 0, e = FuncInfo->PHINodesToUpdate.size(); i != e; ++i) {
16630b57cec5SDimitry Andric     MachineInstrBuilder PHI(*MF, FuncInfo->PHINodesToUpdate[i].first);
16640b57cec5SDimitry Andric     assert(PHI->isPHI() &&
16650b57cec5SDimitry Andric            "This is not a machine PHI node that we are updating!");
16660b57cec5SDimitry Andric     if (!FuncInfo->MBB->isSuccessor(PHI->getParent()))
16670b57cec5SDimitry Andric       continue;
16680b57cec5SDimitry Andric     PHI.addReg(FuncInfo->PHINodesToUpdate[i].second).addMBB(FuncInfo->MBB);
16690b57cec5SDimitry Andric   }
16700b57cec5SDimitry Andric 
16710b57cec5SDimitry Andric   // Handle stack protector.
16720b57cec5SDimitry Andric   if (SDB->SPDescriptor.shouldEmitFunctionBasedCheckStackProtector()) {
16730b57cec5SDimitry Andric     // The target provides a guard check function. There is no need to
16740b57cec5SDimitry Andric     // generate error handling code or to split current basic block.
16750b57cec5SDimitry Andric     MachineBasicBlock *ParentMBB = SDB->SPDescriptor.getParentMBB();
16760b57cec5SDimitry Andric 
16770b57cec5SDimitry Andric     // Add load and check to the basicblock.
16780b57cec5SDimitry Andric     FuncInfo->MBB = ParentMBB;
16790b57cec5SDimitry Andric     FuncInfo->InsertPt =
1680349cc55cSDimitry Andric         findSplitPointForStackProtector(ParentMBB, *TII);
16810b57cec5SDimitry Andric     SDB->visitSPDescriptorParent(SDB->SPDescriptor, ParentMBB);
16820b57cec5SDimitry Andric     CurDAG->setRoot(SDB->getRoot());
16830b57cec5SDimitry Andric     SDB->clear();
16840b57cec5SDimitry Andric     CodeGenAndEmitDAG();
16850b57cec5SDimitry Andric 
16860b57cec5SDimitry Andric     // Clear the Per-BB State.
16870b57cec5SDimitry Andric     SDB->SPDescriptor.resetPerBBState();
16880b57cec5SDimitry Andric   } else if (SDB->SPDescriptor.shouldEmitStackProtector()) {
16890b57cec5SDimitry Andric     MachineBasicBlock *ParentMBB = SDB->SPDescriptor.getParentMBB();
16900b57cec5SDimitry Andric     MachineBasicBlock *SuccessMBB = SDB->SPDescriptor.getSuccessMBB();
16910b57cec5SDimitry Andric 
16920b57cec5SDimitry Andric     // Find the split point to split the parent mbb. At the same time copy all
16930b57cec5SDimitry Andric     // physical registers used in the tail of parent mbb into virtual registers
16940b57cec5SDimitry Andric     // before the split point and back into physical registers after the split
16950b57cec5SDimitry Andric     // point. This prevents us needing to deal with Live-ins and many other
16960b57cec5SDimitry Andric     // register allocation issues caused by us splitting the parent mbb. The
16970b57cec5SDimitry Andric     // register allocator will clean up said virtual copies later on.
16980b57cec5SDimitry Andric     MachineBasicBlock::iterator SplitPoint =
1699349cc55cSDimitry Andric         findSplitPointForStackProtector(ParentMBB, *TII);
17000b57cec5SDimitry Andric 
17010b57cec5SDimitry Andric     // Splice the terminator of ParentMBB into SuccessMBB.
17020b57cec5SDimitry Andric     SuccessMBB->splice(SuccessMBB->end(), ParentMBB,
17030b57cec5SDimitry Andric                        SplitPoint,
17040b57cec5SDimitry Andric                        ParentMBB->end());
17050b57cec5SDimitry Andric 
17060b57cec5SDimitry Andric     // Add compare/jump on neq/jump to the parent BB.
17070b57cec5SDimitry Andric     FuncInfo->MBB = ParentMBB;
17080b57cec5SDimitry Andric     FuncInfo->InsertPt = ParentMBB->end();
17090b57cec5SDimitry Andric     SDB->visitSPDescriptorParent(SDB->SPDescriptor, ParentMBB);
17100b57cec5SDimitry Andric     CurDAG->setRoot(SDB->getRoot());
17110b57cec5SDimitry Andric     SDB->clear();
17120b57cec5SDimitry Andric     CodeGenAndEmitDAG();
17130b57cec5SDimitry Andric 
17140b57cec5SDimitry Andric     // CodeGen Failure MBB if we have not codegened it yet.
17150b57cec5SDimitry Andric     MachineBasicBlock *FailureMBB = SDB->SPDescriptor.getFailureMBB();
17160b57cec5SDimitry Andric     if (FailureMBB->empty()) {
17170b57cec5SDimitry Andric       FuncInfo->MBB = FailureMBB;
17180b57cec5SDimitry Andric       FuncInfo->InsertPt = FailureMBB->end();
17190b57cec5SDimitry Andric       SDB->visitSPDescriptorFailure(SDB->SPDescriptor);
17200b57cec5SDimitry Andric       CurDAG->setRoot(SDB->getRoot());
17210b57cec5SDimitry Andric       SDB->clear();
17220b57cec5SDimitry Andric       CodeGenAndEmitDAG();
17230b57cec5SDimitry Andric     }
17240b57cec5SDimitry Andric 
17250b57cec5SDimitry Andric     // Clear the Per-BB State.
17260b57cec5SDimitry Andric     SDB->SPDescriptor.resetPerBBState();
17270b57cec5SDimitry Andric   }
17280b57cec5SDimitry Andric 
17290b57cec5SDimitry Andric   // Lower each BitTestBlock.
17300b57cec5SDimitry Andric   for (auto &BTB : SDB->SL->BitTestCases) {
17310b57cec5SDimitry Andric     // Lower header first, if it wasn't already lowered
17320b57cec5SDimitry Andric     if (!BTB.Emitted) {
17330b57cec5SDimitry Andric       // Set the current basic block to the mbb we wish to insert the code into
17340b57cec5SDimitry Andric       FuncInfo->MBB = BTB.Parent;
17350b57cec5SDimitry Andric       FuncInfo->InsertPt = FuncInfo->MBB->end();
17360b57cec5SDimitry Andric       // Emit the code
17370b57cec5SDimitry Andric       SDB->visitBitTestHeader(BTB, FuncInfo->MBB);
17380b57cec5SDimitry Andric       CurDAG->setRoot(SDB->getRoot());
17390b57cec5SDimitry Andric       SDB->clear();
17400b57cec5SDimitry Andric       CodeGenAndEmitDAG();
17410b57cec5SDimitry Andric     }
17420b57cec5SDimitry Andric 
17430b57cec5SDimitry Andric     BranchProbability UnhandledProb = BTB.Prob;
17440b57cec5SDimitry Andric     for (unsigned j = 0, ej = BTB.Cases.size(); j != ej; ++j) {
17450b57cec5SDimitry Andric       UnhandledProb -= BTB.Cases[j].ExtraProb;
17460b57cec5SDimitry Andric       // Set the current basic block to the mbb we wish to insert the code into
17470b57cec5SDimitry Andric       FuncInfo->MBB = BTB.Cases[j].ThisBB;
17480b57cec5SDimitry Andric       FuncInfo->InsertPt = FuncInfo->MBB->end();
17490b57cec5SDimitry Andric       // Emit the code
17500b57cec5SDimitry Andric 
17510b57cec5SDimitry Andric       // If all cases cover a contiguous range, it is not necessary to jump to
17520b57cec5SDimitry Andric       // the default block after the last bit test fails. This is because the
17530b57cec5SDimitry Andric       // range check during bit test header creation has guaranteed that every
17540b57cec5SDimitry Andric       // case here doesn't go outside the range. In this case, there is no need
17550b57cec5SDimitry Andric       // to perform the last bit test, as it will always be true. Instead, make
17560b57cec5SDimitry Andric       // the second-to-last bit-test fall through to the target of the last bit
17570b57cec5SDimitry Andric       // test, and delete the last bit test.
17580b57cec5SDimitry Andric 
17590b57cec5SDimitry Andric       MachineBasicBlock *NextMBB;
1760349cc55cSDimitry Andric       if ((BTB.ContiguousRange || BTB.FallthroughUnreachable) && j + 2 == ej) {
1761349cc55cSDimitry Andric         // Second-to-last bit-test with contiguous range or omitted range
1762349cc55cSDimitry Andric         // check: fall through to the target of the final bit test.
17630b57cec5SDimitry Andric         NextMBB = BTB.Cases[j + 1].TargetBB;
17640b57cec5SDimitry Andric       } else if (j + 1 == ej) {
17650b57cec5SDimitry Andric         // For the last bit test, fall through to Default.
17660b57cec5SDimitry Andric         NextMBB = BTB.Default;
17670b57cec5SDimitry Andric       } else {
17680b57cec5SDimitry Andric         // Otherwise, fall through to the next bit test.
17690b57cec5SDimitry Andric         NextMBB = BTB.Cases[j + 1].ThisBB;
17700b57cec5SDimitry Andric       }
17710b57cec5SDimitry Andric 
17720b57cec5SDimitry Andric       SDB->visitBitTestCase(BTB, NextMBB, UnhandledProb, BTB.Reg, BTB.Cases[j],
17730b57cec5SDimitry Andric                             FuncInfo->MBB);
17740b57cec5SDimitry Andric 
17750b57cec5SDimitry Andric       CurDAG->setRoot(SDB->getRoot());
17760b57cec5SDimitry Andric       SDB->clear();
17770b57cec5SDimitry Andric       CodeGenAndEmitDAG();
17780b57cec5SDimitry Andric 
1779349cc55cSDimitry Andric       if ((BTB.ContiguousRange || BTB.FallthroughUnreachable) && j + 2 == ej) {
17800b57cec5SDimitry Andric         // Since we're not going to use the final bit test, remove it.
17810b57cec5SDimitry Andric         BTB.Cases.pop_back();
17820b57cec5SDimitry Andric         break;
17830b57cec5SDimitry Andric       }
17840b57cec5SDimitry Andric     }
17850b57cec5SDimitry Andric 
17860b57cec5SDimitry Andric     // Update PHI Nodes
1787*0eae32dcSDimitry Andric     for (const std::pair<MachineInstr *, unsigned> &P :
1788*0eae32dcSDimitry Andric          FuncInfo->PHINodesToUpdate) {
1789*0eae32dcSDimitry Andric       MachineInstrBuilder PHI(*MF, P.first);
17900b57cec5SDimitry Andric       MachineBasicBlock *PHIBB = PHI->getParent();
17910b57cec5SDimitry Andric       assert(PHI->isPHI() &&
17920b57cec5SDimitry Andric              "This is not a machine PHI node that we are updating!");
17930b57cec5SDimitry Andric       // This is "default" BB. We have two jumps to it. From "header" BB and
17940b57cec5SDimitry Andric       // from last "case" BB, unless the latter was skipped.
17950b57cec5SDimitry Andric       if (PHIBB == BTB.Default) {
1796*0eae32dcSDimitry Andric         PHI.addReg(P.second).addMBB(BTB.Parent);
17970b57cec5SDimitry Andric         if (!BTB.ContiguousRange) {
1798*0eae32dcSDimitry Andric           PHI.addReg(P.second).addMBB(BTB.Cases.back().ThisBB);
17990b57cec5SDimitry Andric          }
18000b57cec5SDimitry Andric       }
18010b57cec5SDimitry Andric       // One of "cases" BB.
1802*0eae32dcSDimitry Andric       for (const SwitchCG::BitTestCase &BT : BTB.Cases) {
1803*0eae32dcSDimitry Andric         MachineBasicBlock* cBB = BT.ThisBB;
18040b57cec5SDimitry Andric         if (cBB->isSuccessor(PHIBB))
1805*0eae32dcSDimitry Andric           PHI.addReg(P.second).addMBB(cBB);
18060b57cec5SDimitry Andric       }
18070b57cec5SDimitry Andric     }
18080b57cec5SDimitry Andric   }
18090b57cec5SDimitry Andric   SDB->SL->BitTestCases.clear();
18100b57cec5SDimitry Andric 
18110b57cec5SDimitry Andric   // If the JumpTable record is filled in, then we need to emit a jump table.
18120b57cec5SDimitry Andric   // Updating the PHI nodes is tricky in this case, since we need to determine
18130b57cec5SDimitry Andric   // whether the PHI is a successor of the range check MBB or the jump table MBB
18140b57cec5SDimitry Andric   for (unsigned i = 0, e = SDB->SL->JTCases.size(); i != e; ++i) {
18150b57cec5SDimitry Andric     // Lower header first, if it wasn't already lowered
18160b57cec5SDimitry Andric     if (!SDB->SL->JTCases[i].first.Emitted) {
18170b57cec5SDimitry Andric       // Set the current basic block to the mbb we wish to insert the code into
18180b57cec5SDimitry Andric       FuncInfo->MBB = SDB->SL->JTCases[i].first.HeaderBB;
18190b57cec5SDimitry Andric       FuncInfo->InsertPt = FuncInfo->MBB->end();
18200b57cec5SDimitry Andric       // Emit the code
18210b57cec5SDimitry Andric       SDB->visitJumpTableHeader(SDB->SL->JTCases[i].second,
18220b57cec5SDimitry Andric                                 SDB->SL->JTCases[i].first, FuncInfo->MBB);
18230b57cec5SDimitry Andric       CurDAG->setRoot(SDB->getRoot());
18240b57cec5SDimitry Andric       SDB->clear();
18250b57cec5SDimitry Andric       CodeGenAndEmitDAG();
18260b57cec5SDimitry Andric     }
18270b57cec5SDimitry Andric 
18280b57cec5SDimitry Andric     // Set the current basic block to the mbb we wish to insert the code into
18290b57cec5SDimitry Andric     FuncInfo->MBB = SDB->SL->JTCases[i].second.MBB;
18300b57cec5SDimitry Andric     FuncInfo->InsertPt = FuncInfo->MBB->end();
18310b57cec5SDimitry Andric     // Emit the code
18320b57cec5SDimitry Andric     SDB->visitJumpTable(SDB->SL->JTCases[i].second);
18330b57cec5SDimitry Andric     CurDAG->setRoot(SDB->getRoot());
18340b57cec5SDimitry Andric     SDB->clear();
18350b57cec5SDimitry Andric     CodeGenAndEmitDAG();
18360b57cec5SDimitry Andric 
18370b57cec5SDimitry Andric     // Update PHI Nodes
18380b57cec5SDimitry Andric     for (unsigned pi = 0, pe = FuncInfo->PHINodesToUpdate.size();
18390b57cec5SDimitry Andric          pi != pe; ++pi) {
18400b57cec5SDimitry Andric       MachineInstrBuilder PHI(*MF, FuncInfo->PHINodesToUpdate[pi].first);
18410b57cec5SDimitry Andric       MachineBasicBlock *PHIBB = PHI->getParent();
18420b57cec5SDimitry Andric       assert(PHI->isPHI() &&
18430b57cec5SDimitry Andric              "This is not a machine PHI node that we are updating!");
18440b57cec5SDimitry Andric       // "default" BB. We can go there only from header BB.
18450b57cec5SDimitry Andric       if (PHIBB == SDB->SL->JTCases[i].second.Default)
18460b57cec5SDimitry Andric         PHI.addReg(FuncInfo->PHINodesToUpdate[pi].second)
18470b57cec5SDimitry Andric            .addMBB(SDB->SL->JTCases[i].first.HeaderBB);
18480b57cec5SDimitry Andric       // JT BB. Just iterate over successors here
18490b57cec5SDimitry Andric       if (FuncInfo->MBB->isSuccessor(PHIBB))
18500b57cec5SDimitry Andric         PHI.addReg(FuncInfo->PHINodesToUpdate[pi].second).addMBB(FuncInfo->MBB);
18510b57cec5SDimitry Andric     }
18520b57cec5SDimitry Andric   }
18530b57cec5SDimitry Andric   SDB->SL->JTCases.clear();
18540b57cec5SDimitry Andric 
18550b57cec5SDimitry Andric   // If we generated any switch lowering information, build and codegen any
18560b57cec5SDimitry Andric   // additional DAGs necessary.
18570b57cec5SDimitry Andric   for (unsigned i = 0, e = SDB->SL->SwitchCases.size(); i != e; ++i) {
18580b57cec5SDimitry Andric     // Set the current basic block to the mbb we wish to insert the code into
18590b57cec5SDimitry Andric     FuncInfo->MBB = SDB->SL->SwitchCases[i].ThisBB;
18600b57cec5SDimitry Andric     FuncInfo->InsertPt = FuncInfo->MBB->end();
18610b57cec5SDimitry Andric 
18620b57cec5SDimitry Andric     // Determine the unique successors.
18630b57cec5SDimitry Andric     SmallVector<MachineBasicBlock *, 2> Succs;
18640b57cec5SDimitry Andric     Succs.push_back(SDB->SL->SwitchCases[i].TrueBB);
18650b57cec5SDimitry Andric     if (SDB->SL->SwitchCases[i].TrueBB != SDB->SL->SwitchCases[i].FalseBB)
18660b57cec5SDimitry Andric       Succs.push_back(SDB->SL->SwitchCases[i].FalseBB);
18670b57cec5SDimitry Andric 
18680b57cec5SDimitry Andric     // Emit the code. Note that this could result in FuncInfo->MBB being split.
18690b57cec5SDimitry Andric     SDB->visitSwitchCase(SDB->SL->SwitchCases[i], FuncInfo->MBB);
18700b57cec5SDimitry Andric     CurDAG->setRoot(SDB->getRoot());
18710b57cec5SDimitry Andric     SDB->clear();
18720b57cec5SDimitry Andric     CodeGenAndEmitDAG();
18730b57cec5SDimitry Andric 
18740b57cec5SDimitry Andric     // Remember the last block, now that any splitting is done, for use in
18750b57cec5SDimitry Andric     // populating PHI nodes in successors.
18760b57cec5SDimitry Andric     MachineBasicBlock *ThisBB = FuncInfo->MBB;
18770b57cec5SDimitry Andric 
18780b57cec5SDimitry Andric     // Handle any PHI nodes in successors of this chunk, as if we were coming
18790b57cec5SDimitry Andric     // from the original BB before switch expansion.  Note that PHI nodes can
18800b57cec5SDimitry Andric     // occur multiple times in PHINodesToUpdate.  We have to be very careful to
18810b57cec5SDimitry Andric     // handle them the right number of times.
18820b57cec5SDimitry Andric     for (unsigned i = 0, e = Succs.size(); i != e; ++i) {
18830b57cec5SDimitry Andric       FuncInfo->MBB = Succs[i];
18840b57cec5SDimitry Andric       FuncInfo->InsertPt = FuncInfo->MBB->end();
18850b57cec5SDimitry Andric       // FuncInfo->MBB may have been removed from the CFG if a branch was
18860b57cec5SDimitry Andric       // constant folded.
18870b57cec5SDimitry Andric       if (ThisBB->isSuccessor(FuncInfo->MBB)) {
18880b57cec5SDimitry Andric         for (MachineBasicBlock::iterator
18890b57cec5SDimitry Andric              MBBI = FuncInfo->MBB->begin(), MBBE = FuncInfo->MBB->end();
18900b57cec5SDimitry Andric              MBBI != MBBE && MBBI->isPHI(); ++MBBI) {
18910b57cec5SDimitry Andric           MachineInstrBuilder PHI(*MF, MBBI);
18920b57cec5SDimitry Andric           // This value for this PHI node is recorded in PHINodesToUpdate.
18930b57cec5SDimitry Andric           for (unsigned pn = 0; ; ++pn) {
18940b57cec5SDimitry Andric             assert(pn != FuncInfo->PHINodesToUpdate.size() &&
18950b57cec5SDimitry Andric                    "Didn't find PHI entry!");
18960b57cec5SDimitry Andric             if (FuncInfo->PHINodesToUpdate[pn].first == PHI) {
18970b57cec5SDimitry Andric               PHI.addReg(FuncInfo->PHINodesToUpdate[pn].second).addMBB(ThisBB);
18980b57cec5SDimitry Andric               break;
18990b57cec5SDimitry Andric             }
19000b57cec5SDimitry Andric           }
19010b57cec5SDimitry Andric         }
19020b57cec5SDimitry Andric       }
19030b57cec5SDimitry Andric     }
19040b57cec5SDimitry Andric   }
19050b57cec5SDimitry Andric   SDB->SL->SwitchCases.clear();
19060b57cec5SDimitry Andric }
19070b57cec5SDimitry Andric 
19080b57cec5SDimitry Andric /// Create the scheduler. If a specific scheduler was specified
19090b57cec5SDimitry Andric /// via the SchedulerRegistry, use it, otherwise select the
19100b57cec5SDimitry Andric /// one preferred by the target.
19110b57cec5SDimitry Andric ///
19120b57cec5SDimitry Andric ScheduleDAGSDNodes *SelectionDAGISel::CreateScheduler() {
19130b57cec5SDimitry Andric   return ISHeuristic(this, OptLevel);
19140b57cec5SDimitry Andric }
19150b57cec5SDimitry Andric 
19160b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
19170b57cec5SDimitry Andric // Helper functions used by the generated instruction selector.
19180b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
19190b57cec5SDimitry Andric // Calls to these methods are generated by tblgen.
19200b57cec5SDimitry Andric 
19210b57cec5SDimitry Andric /// CheckAndMask - The isel is trying to match something like (and X, 255).  If
19220b57cec5SDimitry Andric /// the dag combiner simplified the 255, we still want to match.  RHS is the
19230b57cec5SDimitry Andric /// actual value in the DAG on the RHS of an AND, and DesiredMaskS is the value
19240b57cec5SDimitry Andric /// specified in the .td file (e.g. 255).
19250b57cec5SDimitry Andric bool SelectionDAGISel::CheckAndMask(SDValue LHS, ConstantSDNode *RHS,
19260b57cec5SDimitry Andric                                     int64_t DesiredMaskS) const {
19270b57cec5SDimitry Andric   const APInt &ActualMask = RHS->getAPIntValue();
19280b57cec5SDimitry Andric   const APInt &DesiredMask = APInt(LHS.getValueSizeInBits(), DesiredMaskS);
19290b57cec5SDimitry Andric 
19300b57cec5SDimitry Andric   // If the actual mask exactly matches, success!
19310b57cec5SDimitry Andric   if (ActualMask == DesiredMask)
19320b57cec5SDimitry Andric     return true;
19330b57cec5SDimitry Andric 
19340b57cec5SDimitry Andric   // If the actual AND mask is allowing unallowed bits, this doesn't match.
19350b57cec5SDimitry Andric   if (!ActualMask.isSubsetOf(DesiredMask))
19360b57cec5SDimitry Andric     return false;
19370b57cec5SDimitry Andric 
19380b57cec5SDimitry Andric   // Otherwise, the DAG Combiner may have proven that the value coming in is
19390b57cec5SDimitry Andric   // either already zero or is not demanded.  Check for known zero input bits.
19400b57cec5SDimitry Andric   APInt NeededMask = DesiredMask & ~ActualMask;
19410b57cec5SDimitry Andric   if (CurDAG->MaskedValueIsZero(LHS, NeededMask))
19420b57cec5SDimitry Andric     return true;
19430b57cec5SDimitry Andric 
19440b57cec5SDimitry Andric   // TODO: check to see if missing bits are just not demanded.
19450b57cec5SDimitry Andric 
19460b57cec5SDimitry Andric   // Otherwise, this pattern doesn't match.
19470b57cec5SDimitry Andric   return false;
19480b57cec5SDimitry Andric }
19490b57cec5SDimitry Andric 
19500b57cec5SDimitry Andric /// CheckOrMask - The isel is trying to match something like (or X, 255).  If
19510b57cec5SDimitry Andric /// the dag combiner simplified the 255, we still want to match.  RHS is the
19520b57cec5SDimitry Andric /// actual value in the DAG on the RHS of an OR, and DesiredMaskS is the value
19530b57cec5SDimitry Andric /// specified in the .td file (e.g. 255).
19540b57cec5SDimitry Andric bool SelectionDAGISel::CheckOrMask(SDValue LHS, ConstantSDNode *RHS,
19550b57cec5SDimitry Andric                                    int64_t DesiredMaskS) const {
19560b57cec5SDimitry Andric   const APInt &ActualMask = RHS->getAPIntValue();
19570b57cec5SDimitry Andric   const APInt &DesiredMask = APInt(LHS.getValueSizeInBits(), DesiredMaskS);
19580b57cec5SDimitry Andric 
19590b57cec5SDimitry Andric   // If the actual mask exactly matches, success!
19600b57cec5SDimitry Andric   if (ActualMask == DesiredMask)
19610b57cec5SDimitry Andric     return true;
19620b57cec5SDimitry Andric 
19630b57cec5SDimitry Andric   // If the actual AND mask is allowing unallowed bits, this doesn't match.
19640b57cec5SDimitry Andric   if (!ActualMask.isSubsetOf(DesiredMask))
19650b57cec5SDimitry Andric     return false;
19660b57cec5SDimitry Andric 
19670b57cec5SDimitry Andric   // Otherwise, the DAG Combiner may have proven that the value coming in is
19680b57cec5SDimitry Andric   // either already zero or is not demanded.  Check for known zero input bits.
19690b57cec5SDimitry Andric   APInt NeededMask = DesiredMask & ~ActualMask;
19700b57cec5SDimitry Andric   KnownBits Known = CurDAG->computeKnownBits(LHS);
19710b57cec5SDimitry Andric 
19720b57cec5SDimitry Andric   // If all the missing bits in the or are already known to be set, match!
19730b57cec5SDimitry Andric   if (NeededMask.isSubsetOf(Known.One))
19740b57cec5SDimitry Andric     return true;
19750b57cec5SDimitry Andric 
19760b57cec5SDimitry Andric   // TODO: check to see if missing bits are just not demanded.
19770b57cec5SDimitry Andric 
19780b57cec5SDimitry Andric   // Otherwise, this pattern doesn't match.
19790b57cec5SDimitry Andric   return false;
19800b57cec5SDimitry Andric }
19810b57cec5SDimitry Andric 
19820b57cec5SDimitry Andric /// SelectInlineAsmMemoryOperands - Calls to this are automatically generated
19830b57cec5SDimitry Andric /// by tblgen.  Others should not call it.
19840b57cec5SDimitry Andric void SelectionDAGISel::SelectInlineAsmMemoryOperands(std::vector<SDValue> &Ops,
19850b57cec5SDimitry Andric                                                      const SDLoc &DL) {
19860b57cec5SDimitry Andric   std::vector<SDValue> InOps;
19870b57cec5SDimitry Andric   std::swap(InOps, Ops);
19880b57cec5SDimitry Andric 
19890b57cec5SDimitry Andric   Ops.push_back(InOps[InlineAsm::Op_InputChain]); // 0
19900b57cec5SDimitry Andric   Ops.push_back(InOps[InlineAsm::Op_AsmString]);  // 1
19910b57cec5SDimitry Andric   Ops.push_back(InOps[InlineAsm::Op_MDNode]);     // 2, !srcloc
19920b57cec5SDimitry Andric   Ops.push_back(InOps[InlineAsm::Op_ExtraInfo]);  // 3 (SideEffect, AlignStack)
19930b57cec5SDimitry Andric 
19940b57cec5SDimitry Andric   unsigned i = InlineAsm::Op_FirstOperand, e = InOps.size();
19950b57cec5SDimitry Andric   if (InOps[e-1].getValueType() == MVT::Glue)
19960b57cec5SDimitry Andric     --e;  // Don't process a glue operand if it is here.
19970b57cec5SDimitry Andric 
19980b57cec5SDimitry Andric   while (i != e) {
19990b57cec5SDimitry Andric     unsigned Flags = cast<ConstantSDNode>(InOps[i])->getZExtValue();
20000b57cec5SDimitry Andric     if (!InlineAsm::isMemKind(Flags)) {
20010b57cec5SDimitry Andric       // Just skip over this operand, copying the operands verbatim.
20020b57cec5SDimitry Andric       Ops.insert(Ops.end(), InOps.begin()+i,
20030b57cec5SDimitry Andric                  InOps.begin()+i+InlineAsm::getNumOperandRegisters(Flags) + 1);
20040b57cec5SDimitry Andric       i += InlineAsm::getNumOperandRegisters(Flags) + 1;
20050b57cec5SDimitry Andric     } else {
20060b57cec5SDimitry Andric       assert(InlineAsm::getNumOperandRegisters(Flags) == 1 &&
20070b57cec5SDimitry Andric              "Memory operand with multiple values?");
20080b57cec5SDimitry Andric 
20090b57cec5SDimitry Andric       unsigned TiedToOperand;
20100b57cec5SDimitry Andric       if (InlineAsm::isUseOperandTiedToDef(Flags, TiedToOperand)) {
20110b57cec5SDimitry Andric         // We need the constraint ID from the operand this is tied to.
20120b57cec5SDimitry Andric         unsigned CurOp = InlineAsm::Op_FirstOperand;
20130b57cec5SDimitry Andric         Flags = cast<ConstantSDNode>(InOps[CurOp])->getZExtValue();
20140b57cec5SDimitry Andric         for (; TiedToOperand; --TiedToOperand) {
20150b57cec5SDimitry Andric           CurOp += InlineAsm::getNumOperandRegisters(Flags)+1;
20160b57cec5SDimitry Andric           Flags = cast<ConstantSDNode>(InOps[CurOp])->getZExtValue();
20170b57cec5SDimitry Andric         }
20180b57cec5SDimitry Andric       }
20190b57cec5SDimitry Andric 
20200b57cec5SDimitry Andric       // Otherwise, this is a memory operand.  Ask the target to select it.
20210b57cec5SDimitry Andric       std::vector<SDValue> SelOps;
20220b57cec5SDimitry Andric       unsigned ConstraintID = InlineAsm::getMemoryConstraintID(Flags);
20230b57cec5SDimitry Andric       if (SelectInlineAsmMemoryOperand(InOps[i+1], ConstraintID, SelOps))
20240b57cec5SDimitry Andric         report_fatal_error("Could not match memory address.  Inline asm"
20250b57cec5SDimitry Andric                            " failure!");
20260b57cec5SDimitry Andric 
20270b57cec5SDimitry Andric       // Add this to the output node.
20280b57cec5SDimitry Andric       unsigned NewFlags =
20290b57cec5SDimitry Andric         InlineAsm::getFlagWord(InlineAsm::Kind_Mem, SelOps.size());
20300b57cec5SDimitry Andric       NewFlags = InlineAsm::getFlagWordForMem(NewFlags, ConstraintID);
20310b57cec5SDimitry Andric       Ops.push_back(CurDAG->getTargetConstant(NewFlags, DL, MVT::i32));
2032e8d8bef9SDimitry Andric       llvm::append_range(Ops, SelOps);
20330b57cec5SDimitry Andric       i += 2;
20340b57cec5SDimitry Andric     }
20350b57cec5SDimitry Andric   }
20360b57cec5SDimitry Andric 
20370b57cec5SDimitry Andric   // Add the glue input back if present.
20380b57cec5SDimitry Andric   if (e != InOps.size())
20390b57cec5SDimitry Andric     Ops.push_back(InOps.back());
20400b57cec5SDimitry Andric }
20410b57cec5SDimitry Andric 
20420b57cec5SDimitry Andric /// findGlueUse - Return use of MVT::Glue value produced by the specified
20430b57cec5SDimitry Andric /// SDNode.
20440b57cec5SDimitry Andric ///
20450b57cec5SDimitry Andric static SDNode *findGlueUse(SDNode *N) {
20460b57cec5SDimitry Andric   unsigned FlagResNo = N->getNumValues()-1;
20470b57cec5SDimitry Andric   for (SDNode::use_iterator I = N->use_begin(), E = N->use_end(); I != E; ++I) {
20480b57cec5SDimitry Andric     SDUse &Use = I.getUse();
20490b57cec5SDimitry Andric     if (Use.getResNo() == FlagResNo)
20500b57cec5SDimitry Andric       return Use.getUser();
20510b57cec5SDimitry Andric   }
20520b57cec5SDimitry Andric   return nullptr;
20530b57cec5SDimitry Andric }
20540b57cec5SDimitry Andric 
20550b57cec5SDimitry Andric /// findNonImmUse - Return true if "Def" is a predecessor of "Root" via a path
20560b57cec5SDimitry Andric /// beyond "ImmedUse".  We may ignore chains as they are checked separately.
20570b57cec5SDimitry Andric static bool findNonImmUse(SDNode *Root, SDNode *Def, SDNode *ImmedUse,
20580b57cec5SDimitry Andric                           bool IgnoreChains) {
20590b57cec5SDimitry Andric   SmallPtrSet<const SDNode *, 16> Visited;
20600b57cec5SDimitry Andric   SmallVector<const SDNode *, 16> WorkList;
20610b57cec5SDimitry Andric   // Only check if we have non-immediate uses of Def.
20620b57cec5SDimitry Andric   if (ImmedUse->isOnlyUserOf(Def))
20630b57cec5SDimitry Andric     return false;
20640b57cec5SDimitry Andric 
20650b57cec5SDimitry Andric   // We don't care about paths to Def that go through ImmedUse so mark it
20660b57cec5SDimitry Andric   // visited and mark non-def operands as used.
20670b57cec5SDimitry Andric   Visited.insert(ImmedUse);
20680b57cec5SDimitry Andric   for (const SDValue &Op : ImmedUse->op_values()) {
20690b57cec5SDimitry Andric     SDNode *N = Op.getNode();
20700b57cec5SDimitry Andric     // Ignore chain deps (they are validated by
20710b57cec5SDimitry Andric     // HandleMergeInputChains) and immediate uses
20720b57cec5SDimitry Andric     if ((Op.getValueType() == MVT::Other && IgnoreChains) || N == Def)
20730b57cec5SDimitry Andric       continue;
20740b57cec5SDimitry Andric     if (!Visited.insert(N).second)
20750b57cec5SDimitry Andric       continue;
20760b57cec5SDimitry Andric     WorkList.push_back(N);
20770b57cec5SDimitry Andric   }
20780b57cec5SDimitry Andric 
20790b57cec5SDimitry Andric   // Initialize worklist to operands of Root.
20800b57cec5SDimitry Andric   if (Root != ImmedUse) {
20810b57cec5SDimitry Andric     for (const SDValue &Op : Root->op_values()) {
20820b57cec5SDimitry Andric       SDNode *N = Op.getNode();
20830b57cec5SDimitry Andric       // Ignore chains (they are validated by HandleMergeInputChains)
20840b57cec5SDimitry Andric       if ((Op.getValueType() == MVT::Other && IgnoreChains) || N == Def)
20850b57cec5SDimitry Andric         continue;
20860b57cec5SDimitry Andric       if (!Visited.insert(N).second)
20870b57cec5SDimitry Andric         continue;
20880b57cec5SDimitry Andric       WorkList.push_back(N);
20890b57cec5SDimitry Andric     }
20900b57cec5SDimitry Andric   }
20910b57cec5SDimitry Andric 
20920b57cec5SDimitry Andric   return SDNode::hasPredecessorHelper(Def, Visited, WorkList, 0, true);
20930b57cec5SDimitry Andric }
20940b57cec5SDimitry Andric 
20950b57cec5SDimitry Andric /// IsProfitableToFold - Returns true if it's profitable to fold the specific
20960b57cec5SDimitry Andric /// operand node N of U during instruction selection that starts at Root.
20970b57cec5SDimitry Andric bool SelectionDAGISel::IsProfitableToFold(SDValue N, SDNode *U,
20980b57cec5SDimitry Andric                                           SDNode *Root) const {
20990b57cec5SDimitry Andric   if (OptLevel == CodeGenOpt::None) return false;
21000b57cec5SDimitry Andric   return N.hasOneUse();
21010b57cec5SDimitry Andric }
21020b57cec5SDimitry Andric 
21030b57cec5SDimitry Andric /// IsLegalToFold - Returns true if the specific operand node N of
21040b57cec5SDimitry Andric /// U can be folded during instruction selection that starts at Root.
21050b57cec5SDimitry Andric bool SelectionDAGISel::IsLegalToFold(SDValue N, SDNode *U, SDNode *Root,
21060b57cec5SDimitry Andric                                      CodeGenOpt::Level OptLevel,
21070b57cec5SDimitry Andric                                      bool IgnoreChains) {
21080b57cec5SDimitry Andric   if (OptLevel == CodeGenOpt::None) return false;
21090b57cec5SDimitry Andric 
21100b57cec5SDimitry Andric   // If Root use can somehow reach N through a path that that doesn't contain
21110b57cec5SDimitry Andric   // U then folding N would create a cycle. e.g. In the following
21120b57cec5SDimitry Andric   // diagram, Root can reach N through X. If N is folded into Root, then
21130b57cec5SDimitry Andric   // X is both a predecessor and a successor of U.
21140b57cec5SDimitry Andric   //
21150b57cec5SDimitry Andric   //          [N*]           //
21160b57cec5SDimitry Andric   //         ^   ^           //
21170b57cec5SDimitry Andric   //        /     \          //
21180b57cec5SDimitry Andric   //      [U*]    [X]?       //
21190b57cec5SDimitry Andric   //        ^     ^          //
21200b57cec5SDimitry Andric   //         \   /           //
21210b57cec5SDimitry Andric   //          \ /            //
21220b57cec5SDimitry Andric   //         [Root*]         //
21230b57cec5SDimitry Andric   //
21240b57cec5SDimitry Andric   // * indicates nodes to be folded together.
21250b57cec5SDimitry Andric   //
21260b57cec5SDimitry Andric   // If Root produces glue, then it gets (even more) interesting. Since it
21270b57cec5SDimitry Andric   // will be "glued" together with its glue use in the scheduler, we need to
21280b57cec5SDimitry Andric   // check if it might reach N.
21290b57cec5SDimitry Andric   //
21300b57cec5SDimitry Andric   //          [N*]           //
21310b57cec5SDimitry Andric   //         ^   ^           //
21320b57cec5SDimitry Andric   //        /     \          //
21330b57cec5SDimitry Andric   //      [U*]    [X]?       //
21340b57cec5SDimitry Andric   //        ^       ^        //
21350b57cec5SDimitry Andric   //         \       \       //
21360b57cec5SDimitry Andric   //          \      |       //
21370b57cec5SDimitry Andric   //         [Root*] |       //
21380b57cec5SDimitry Andric   //          ^      |       //
21390b57cec5SDimitry Andric   //          f      |       //
21400b57cec5SDimitry Andric   //          |      /       //
21410b57cec5SDimitry Andric   //         [Y]    /        //
21420b57cec5SDimitry Andric   //           ^   /         //
21430b57cec5SDimitry Andric   //           f  /          //
21440b57cec5SDimitry Andric   //           | /           //
21450b57cec5SDimitry Andric   //          [GU]           //
21460b57cec5SDimitry Andric   //
21470b57cec5SDimitry Andric   // If GU (glue use) indirectly reaches N (the load), and Root folds N
21480b57cec5SDimitry Andric   // (call it Fold), then X is a predecessor of GU and a successor of
21490b57cec5SDimitry Andric   // Fold. But since Fold and GU are glued together, this will create
21500b57cec5SDimitry Andric   // a cycle in the scheduling graph.
21510b57cec5SDimitry Andric 
21520b57cec5SDimitry Andric   // If the node has glue, walk down the graph to the "lowest" node in the
21530b57cec5SDimitry Andric   // glueged set.
21540b57cec5SDimitry Andric   EVT VT = Root->getValueType(Root->getNumValues()-1);
21550b57cec5SDimitry Andric   while (VT == MVT::Glue) {
21560b57cec5SDimitry Andric     SDNode *GU = findGlueUse(Root);
21570b57cec5SDimitry Andric     if (!GU)
21580b57cec5SDimitry Andric       break;
21590b57cec5SDimitry Andric     Root = GU;
21600b57cec5SDimitry Andric     VT = Root->getValueType(Root->getNumValues()-1);
21610b57cec5SDimitry Andric 
21620b57cec5SDimitry Andric     // If our query node has a glue result with a use, we've walked up it.  If
21630b57cec5SDimitry Andric     // the user (which has already been selected) has a chain or indirectly uses
21640b57cec5SDimitry Andric     // the chain, HandleMergeInputChains will not consider it.  Because of
21650b57cec5SDimitry Andric     // this, we cannot ignore chains in this predicate.
21660b57cec5SDimitry Andric     IgnoreChains = false;
21670b57cec5SDimitry Andric   }
21680b57cec5SDimitry Andric 
21690b57cec5SDimitry Andric   return !findNonImmUse(Root, N.getNode(), U, IgnoreChains);
21700b57cec5SDimitry Andric }
21710b57cec5SDimitry Andric 
21725ffd83dbSDimitry Andric void SelectionDAGISel::Select_INLINEASM(SDNode *N) {
21730b57cec5SDimitry Andric   SDLoc DL(N);
21740b57cec5SDimitry Andric 
21750b57cec5SDimitry Andric   std::vector<SDValue> Ops(N->op_begin(), N->op_end());
21760b57cec5SDimitry Andric   SelectInlineAsmMemoryOperands(Ops, DL);
21770b57cec5SDimitry Andric 
21780b57cec5SDimitry Andric   const EVT VTs[] = {MVT::Other, MVT::Glue};
21795ffd83dbSDimitry Andric   SDValue New = CurDAG->getNode(N->getOpcode(), DL, VTs, Ops);
21800b57cec5SDimitry Andric   New->setNodeId(-1);
21810b57cec5SDimitry Andric   ReplaceUses(N, New.getNode());
21820b57cec5SDimitry Andric   CurDAG->RemoveDeadNode(N);
21830b57cec5SDimitry Andric }
21840b57cec5SDimitry Andric 
21850b57cec5SDimitry Andric void SelectionDAGISel::Select_READ_REGISTER(SDNode *Op) {
21860b57cec5SDimitry Andric   SDLoc dl(Op);
2187480093f4SDimitry Andric   MDNodeSDNode *MD = cast<MDNodeSDNode>(Op->getOperand(1));
2188480093f4SDimitry Andric   const MDString *RegStr = cast<MDString>(MD->getMD()->getOperand(0));
2189480093f4SDimitry Andric 
2190480093f4SDimitry Andric   EVT VT = Op->getValueType(0);
2191480093f4SDimitry Andric   LLT Ty = VT.isSimple() ? getLLTForMVT(VT.getSimpleVT()) : LLT();
21928bcb0991SDimitry Andric   Register Reg =
2193480093f4SDimitry Andric       TLI->getRegisterByName(RegStr->getString().data(), Ty,
21948bcb0991SDimitry Andric                              CurDAG->getMachineFunction());
21950b57cec5SDimitry Andric   SDValue New = CurDAG->getCopyFromReg(
21960b57cec5SDimitry Andric                         Op->getOperand(0), dl, Reg, Op->getValueType(0));
21970b57cec5SDimitry Andric   New->setNodeId(-1);
21980b57cec5SDimitry Andric   ReplaceUses(Op, New.getNode());
21990b57cec5SDimitry Andric   CurDAG->RemoveDeadNode(Op);
22000b57cec5SDimitry Andric }
22010b57cec5SDimitry Andric 
22020b57cec5SDimitry Andric void SelectionDAGISel::Select_WRITE_REGISTER(SDNode *Op) {
22030b57cec5SDimitry Andric   SDLoc dl(Op);
2204480093f4SDimitry Andric   MDNodeSDNode *MD = cast<MDNodeSDNode>(Op->getOperand(1));
2205480093f4SDimitry Andric   const MDString *RegStr = cast<MDString>(MD->getMD()->getOperand(0));
2206480093f4SDimitry Andric 
2207480093f4SDimitry Andric   EVT VT = Op->getOperand(2).getValueType();
2208480093f4SDimitry Andric   LLT Ty = VT.isSimple() ? getLLTForMVT(VT.getSimpleVT()) : LLT();
2209480093f4SDimitry Andric 
2210480093f4SDimitry Andric   Register Reg = TLI->getRegisterByName(RegStr->getString().data(), Ty,
22118bcb0991SDimitry Andric                                         CurDAG->getMachineFunction());
22120b57cec5SDimitry Andric   SDValue New = CurDAG->getCopyToReg(
22130b57cec5SDimitry Andric                         Op->getOperand(0), dl, Reg, Op->getOperand(2));
22140b57cec5SDimitry Andric   New->setNodeId(-1);
22150b57cec5SDimitry Andric   ReplaceUses(Op, New.getNode());
22160b57cec5SDimitry Andric   CurDAG->RemoveDeadNode(Op);
22170b57cec5SDimitry Andric }
22180b57cec5SDimitry Andric 
22190b57cec5SDimitry Andric void SelectionDAGISel::Select_UNDEF(SDNode *N) {
22200b57cec5SDimitry Andric   CurDAG->SelectNodeTo(N, TargetOpcode::IMPLICIT_DEF, N->getValueType(0));
22210b57cec5SDimitry Andric }
22220b57cec5SDimitry Andric 
22235ffd83dbSDimitry Andric void SelectionDAGISel::Select_FREEZE(SDNode *N) {
22245ffd83dbSDimitry Andric   // TODO: We don't have FREEZE pseudo-instruction in MachineInstr-level now.
22255ffd83dbSDimitry Andric   // If FREEZE instruction is added later, the code below must be changed as
22265ffd83dbSDimitry Andric   // well.
22275ffd83dbSDimitry Andric   CurDAG->SelectNodeTo(N, TargetOpcode::COPY, N->getValueType(0),
22285ffd83dbSDimitry Andric                        N->getOperand(0));
22295ffd83dbSDimitry Andric }
22305ffd83dbSDimitry Andric 
2231fe6060f1SDimitry Andric void SelectionDAGISel::Select_ARITH_FENCE(SDNode *N) {
2232fe6060f1SDimitry Andric   CurDAG->SelectNodeTo(N, TargetOpcode::ARITH_FENCE, N->getValueType(0),
2233fe6060f1SDimitry Andric                        N->getOperand(0));
2234fe6060f1SDimitry Andric }
2235fe6060f1SDimitry Andric 
22360b57cec5SDimitry Andric /// GetVBR - decode a vbr encoding whose top bit is set.
2237e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static uint64_t
22380b57cec5SDimitry Andric GetVBR(uint64_t Val, const unsigned char *MatcherTable, unsigned &Idx) {
22390b57cec5SDimitry Andric   assert(Val >= 128 && "Not a VBR");
22400b57cec5SDimitry Andric   Val &= 127;  // Remove first vbr bit.
22410b57cec5SDimitry Andric 
22420b57cec5SDimitry Andric   unsigned Shift = 7;
22430b57cec5SDimitry Andric   uint64_t NextBits;
22440b57cec5SDimitry Andric   do {
22450b57cec5SDimitry Andric     NextBits = MatcherTable[Idx++];
22460b57cec5SDimitry Andric     Val |= (NextBits&127) << Shift;
22470b57cec5SDimitry Andric     Shift += 7;
22480b57cec5SDimitry Andric   } while (NextBits & 128);
22490b57cec5SDimitry Andric 
22500b57cec5SDimitry Andric   return Val;
22510b57cec5SDimitry Andric }
22520b57cec5SDimitry Andric 
22530b57cec5SDimitry Andric /// When a match is complete, this method updates uses of interior chain results
22540b57cec5SDimitry Andric /// to use the new results.
22550b57cec5SDimitry Andric void SelectionDAGISel::UpdateChains(
22560b57cec5SDimitry Andric     SDNode *NodeToMatch, SDValue InputChain,
22570b57cec5SDimitry Andric     SmallVectorImpl<SDNode *> &ChainNodesMatched, bool isMorphNodeTo) {
22580b57cec5SDimitry Andric   SmallVector<SDNode*, 4> NowDeadNodes;
22590b57cec5SDimitry Andric 
22600b57cec5SDimitry Andric   // Now that all the normal results are replaced, we replace the chain and
22610b57cec5SDimitry Andric   // glue results if present.
22620b57cec5SDimitry Andric   if (!ChainNodesMatched.empty()) {
22630b57cec5SDimitry Andric     assert(InputChain.getNode() &&
22640b57cec5SDimitry Andric            "Matched input chains but didn't produce a chain");
22650b57cec5SDimitry Andric     // Loop over all of the nodes we matched that produced a chain result.
22660b57cec5SDimitry Andric     // Replace all the chain results with the final chain we ended up with.
22670b57cec5SDimitry Andric     for (unsigned i = 0, e = ChainNodesMatched.size(); i != e; ++i) {
22680b57cec5SDimitry Andric       SDNode *ChainNode = ChainNodesMatched[i];
22690b57cec5SDimitry Andric       // If ChainNode is null, it's because we replaced it on a previous
22700b57cec5SDimitry Andric       // iteration and we cleared it out of the map. Just skip it.
22710b57cec5SDimitry Andric       if (!ChainNode)
22720b57cec5SDimitry Andric         continue;
22730b57cec5SDimitry Andric 
22740b57cec5SDimitry Andric       assert(ChainNode->getOpcode() != ISD::DELETED_NODE &&
22750b57cec5SDimitry Andric              "Deleted node left in chain");
22760b57cec5SDimitry Andric 
22770b57cec5SDimitry Andric       // Don't replace the results of the root node if we're doing a
22780b57cec5SDimitry Andric       // MorphNodeTo.
22790b57cec5SDimitry Andric       if (ChainNode == NodeToMatch && isMorphNodeTo)
22800b57cec5SDimitry Andric         continue;
22810b57cec5SDimitry Andric 
22820b57cec5SDimitry Andric       SDValue ChainVal = SDValue(ChainNode, ChainNode->getNumValues()-1);
22830b57cec5SDimitry Andric       if (ChainVal.getValueType() == MVT::Glue)
22840b57cec5SDimitry Andric         ChainVal = ChainVal.getValue(ChainVal->getNumValues()-2);
22850b57cec5SDimitry Andric       assert(ChainVal.getValueType() == MVT::Other && "Not a chain?");
22860b57cec5SDimitry Andric       SelectionDAG::DAGNodeDeletedListener NDL(
22870b57cec5SDimitry Andric           *CurDAG, [&](SDNode *N, SDNode *E) {
22880b57cec5SDimitry Andric             std::replace(ChainNodesMatched.begin(), ChainNodesMatched.end(), N,
22890b57cec5SDimitry Andric                          static_cast<SDNode *>(nullptr));
22900b57cec5SDimitry Andric           });
22910b57cec5SDimitry Andric       if (ChainNode->getOpcode() != ISD::TokenFactor)
22920b57cec5SDimitry Andric         ReplaceUses(ChainVal, InputChain);
22930b57cec5SDimitry Andric 
22940b57cec5SDimitry Andric       // If the node became dead and we haven't already seen it, delete it.
22950b57cec5SDimitry Andric       if (ChainNode != NodeToMatch && ChainNode->use_empty() &&
2296e8d8bef9SDimitry Andric           !llvm::is_contained(NowDeadNodes, ChainNode))
22970b57cec5SDimitry Andric         NowDeadNodes.push_back(ChainNode);
22980b57cec5SDimitry Andric     }
22990b57cec5SDimitry Andric   }
23000b57cec5SDimitry Andric 
23010b57cec5SDimitry Andric   if (!NowDeadNodes.empty())
23020b57cec5SDimitry Andric     CurDAG->RemoveDeadNodes(NowDeadNodes);
23030b57cec5SDimitry Andric 
23040b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "ISEL: Match complete!\n");
23050b57cec5SDimitry Andric }
23060b57cec5SDimitry Andric 
23070b57cec5SDimitry Andric /// HandleMergeInputChains - This implements the OPC_EmitMergeInputChains
23080b57cec5SDimitry Andric /// operation for when the pattern matched at least one node with a chains.  The
23090b57cec5SDimitry Andric /// input vector contains a list of all of the chained nodes that we match.  We
23100b57cec5SDimitry Andric /// must determine if this is a valid thing to cover (i.e. matching it won't
23110b57cec5SDimitry Andric /// induce cycles in the DAG) and if so, creating a TokenFactor node. that will
23120b57cec5SDimitry Andric /// be used as the input node chain for the generated nodes.
23130b57cec5SDimitry Andric static SDValue
23140b57cec5SDimitry Andric HandleMergeInputChains(SmallVectorImpl<SDNode*> &ChainNodesMatched,
23150b57cec5SDimitry Andric                        SelectionDAG *CurDAG) {
23160b57cec5SDimitry Andric 
23170b57cec5SDimitry Andric   SmallPtrSet<const SDNode *, 16> Visited;
23180b57cec5SDimitry Andric   SmallVector<const SDNode *, 8> Worklist;
23190b57cec5SDimitry Andric   SmallVector<SDValue, 3> InputChains;
23200b57cec5SDimitry Andric   unsigned int Max = 8192;
23210b57cec5SDimitry Andric 
23220b57cec5SDimitry Andric   // Quick exit on trivial merge.
23230b57cec5SDimitry Andric   if (ChainNodesMatched.size() == 1)
23240b57cec5SDimitry Andric     return ChainNodesMatched[0]->getOperand(0);
23250b57cec5SDimitry Andric 
23260b57cec5SDimitry Andric   // Add chains that aren't already added (internal). Peek through
23270b57cec5SDimitry Andric   // token factors.
23280b57cec5SDimitry Andric   std::function<void(const SDValue)> AddChains = [&](const SDValue V) {
23290b57cec5SDimitry Andric     if (V.getValueType() != MVT::Other)
23300b57cec5SDimitry Andric       return;
23310b57cec5SDimitry Andric     if (V->getOpcode() == ISD::EntryToken)
23320b57cec5SDimitry Andric       return;
23330b57cec5SDimitry Andric     if (!Visited.insert(V.getNode()).second)
23340b57cec5SDimitry Andric       return;
23350b57cec5SDimitry Andric     if (V->getOpcode() == ISD::TokenFactor) {
23360b57cec5SDimitry Andric       for (const SDValue &Op : V->op_values())
23370b57cec5SDimitry Andric         AddChains(Op);
23380b57cec5SDimitry Andric     } else
23390b57cec5SDimitry Andric       InputChains.push_back(V);
23400b57cec5SDimitry Andric   };
23410b57cec5SDimitry Andric 
23420b57cec5SDimitry Andric   for (auto *N : ChainNodesMatched) {
23430b57cec5SDimitry Andric     Worklist.push_back(N);
23440b57cec5SDimitry Andric     Visited.insert(N);
23450b57cec5SDimitry Andric   }
23460b57cec5SDimitry Andric 
23470b57cec5SDimitry Andric   while (!Worklist.empty())
23480b57cec5SDimitry Andric     AddChains(Worklist.pop_back_val()->getOperand(0));
23490b57cec5SDimitry Andric 
23500b57cec5SDimitry Andric   // Skip the search if there are no chain dependencies.
23510b57cec5SDimitry Andric   if (InputChains.size() == 0)
23520b57cec5SDimitry Andric     return CurDAG->getEntryNode();
23530b57cec5SDimitry Andric 
23540b57cec5SDimitry Andric   // If one of these chains is a successor of input, we must have a
23550b57cec5SDimitry Andric   // node that is both the predecessor and successor of the
23560b57cec5SDimitry Andric   // to-be-merged nodes. Fail.
23570b57cec5SDimitry Andric   Visited.clear();
23580b57cec5SDimitry Andric   for (SDValue V : InputChains)
23590b57cec5SDimitry Andric     Worklist.push_back(V.getNode());
23600b57cec5SDimitry Andric 
23610b57cec5SDimitry Andric   for (auto *N : ChainNodesMatched)
23620b57cec5SDimitry Andric     if (SDNode::hasPredecessorHelper(N, Visited, Worklist, Max, true))
23630b57cec5SDimitry Andric       return SDValue();
23640b57cec5SDimitry Andric 
23650b57cec5SDimitry Andric   // Return merged chain.
23660b57cec5SDimitry Andric   if (InputChains.size() == 1)
23670b57cec5SDimitry Andric     return InputChains[0];
23680b57cec5SDimitry Andric   return CurDAG->getNode(ISD::TokenFactor, SDLoc(ChainNodesMatched[0]),
23690b57cec5SDimitry Andric                          MVT::Other, InputChains);
23700b57cec5SDimitry Andric }
23710b57cec5SDimitry Andric 
23720b57cec5SDimitry Andric /// MorphNode - Handle morphing a node in place for the selector.
23730b57cec5SDimitry Andric SDNode *SelectionDAGISel::
23740b57cec5SDimitry Andric MorphNode(SDNode *Node, unsigned TargetOpc, SDVTList VTList,
23750b57cec5SDimitry Andric           ArrayRef<SDValue> Ops, unsigned EmitNodeInfo) {
23760b57cec5SDimitry Andric   // It is possible we're using MorphNodeTo to replace a node with no
23770b57cec5SDimitry Andric   // normal results with one that has a normal result (or we could be
23780b57cec5SDimitry Andric   // adding a chain) and the input could have glue and chains as well.
23790b57cec5SDimitry Andric   // In this case we need to shift the operands down.
23800b57cec5SDimitry Andric   // FIXME: This is a horrible hack and broken in obscure cases, no worse
23810b57cec5SDimitry Andric   // than the old isel though.
23820b57cec5SDimitry Andric   int OldGlueResultNo = -1, OldChainResultNo = -1;
23830b57cec5SDimitry Andric 
23840b57cec5SDimitry Andric   unsigned NTMNumResults = Node->getNumValues();
23850b57cec5SDimitry Andric   if (Node->getValueType(NTMNumResults-1) == MVT::Glue) {
23860b57cec5SDimitry Andric     OldGlueResultNo = NTMNumResults-1;
23870b57cec5SDimitry Andric     if (NTMNumResults != 1 &&
23880b57cec5SDimitry Andric         Node->getValueType(NTMNumResults-2) == MVT::Other)
23890b57cec5SDimitry Andric       OldChainResultNo = NTMNumResults-2;
23900b57cec5SDimitry Andric   } else if (Node->getValueType(NTMNumResults-1) == MVT::Other)
23910b57cec5SDimitry Andric     OldChainResultNo = NTMNumResults-1;
23920b57cec5SDimitry Andric 
23930b57cec5SDimitry Andric   // Call the underlying SelectionDAG routine to do the transmogrification. Note
23940b57cec5SDimitry Andric   // that this deletes operands of the old node that become dead.
23950b57cec5SDimitry Andric   SDNode *Res = CurDAG->MorphNodeTo(Node, ~TargetOpc, VTList, Ops);
23960b57cec5SDimitry Andric 
23970b57cec5SDimitry Andric   // MorphNodeTo can operate in two ways: if an existing node with the
23980b57cec5SDimitry Andric   // specified operands exists, it can just return it.  Otherwise, it
23990b57cec5SDimitry Andric   // updates the node in place to have the requested operands.
24000b57cec5SDimitry Andric   if (Res == Node) {
24010b57cec5SDimitry Andric     // If we updated the node in place, reset the node ID.  To the isel,
24020b57cec5SDimitry Andric     // this should be just like a newly allocated machine node.
24030b57cec5SDimitry Andric     Res->setNodeId(-1);
24040b57cec5SDimitry Andric   }
24050b57cec5SDimitry Andric 
24060b57cec5SDimitry Andric   unsigned ResNumResults = Res->getNumValues();
24070b57cec5SDimitry Andric   // Move the glue if needed.
24080b57cec5SDimitry Andric   if ((EmitNodeInfo & OPFL_GlueOutput) && OldGlueResultNo != -1 &&
24090b57cec5SDimitry Andric       (unsigned)OldGlueResultNo != ResNumResults-1)
24100b57cec5SDimitry Andric     ReplaceUses(SDValue(Node, OldGlueResultNo),
24110b57cec5SDimitry Andric                 SDValue(Res, ResNumResults - 1));
24120b57cec5SDimitry Andric 
24130b57cec5SDimitry Andric   if ((EmitNodeInfo & OPFL_GlueOutput) != 0)
24140b57cec5SDimitry Andric     --ResNumResults;
24150b57cec5SDimitry Andric 
24160b57cec5SDimitry Andric   // Move the chain reference if needed.
24170b57cec5SDimitry Andric   if ((EmitNodeInfo & OPFL_Chain) && OldChainResultNo != -1 &&
24180b57cec5SDimitry Andric       (unsigned)OldChainResultNo != ResNumResults-1)
24190b57cec5SDimitry Andric     ReplaceUses(SDValue(Node, OldChainResultNo),
24200b57cec5SDimitry Andric                 SDValue(Res, ResNumResults - 1));
24210b57cec5SDimitry Andric 
24220b57cec5SDimitry Andric   // Otherwise, no replacement happened because the node already exists. Replace
24230b57cec5SDimitry Andric   // Uses of the old node with the new one.
24240b57cec5SDimitry Andric   if (Res != Node) {
24250b57cec5SDimitry Andric     ReplaceNode(Node, Res);
24260b57cec5SDimitry Andric   } else {
24270b57cec5SDimitry Andric     EnforceNodeIdInvariant(Res);
24280b57cec5SDimitry Andric   }
24290b57cec5SDimitry Andric 
24300b57cec5SDimitry Andric   return Res;
24310b57cec5SDimitry Andric }
24320b57cec5SDimitry Andric 
24330b57cec5SDimitry Andric /// CheckSame - Implements OP_CheckSame.
2434e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
2435e8d8bef9SDimitry Andric CheckSame(const unsigned char *MatcherTable, unsigned &MatcherIndex, SDValue N,
24360b57cec5SDimitry Andric           const SmallVectorImpl<std::pair<SDValue, SDNode *>> &RecordedNodes) {
24370b57cec5SDimitry Andric   // Accept if it is exactly the same as a previously recorded node.
24380b57cec5SDimitry Andric   unsigned RecNo = MatcherTable[MatcherIndex++];
24390b57cec5SDimitry Andric   assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
24400b57cec5SDimitry Andric   return N == RecordedNodes[RecNo].first;
24410b57cec5SDimitry Andric }
24420b57cec5SDimitry Andric 
24430b57cec5SDimitry Andric /// CheckChildSame - Implements OP_CheckChildXSame.
2444e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool CheckChildSame(
2445e8d8bef9SDimitry Andric     const unsigned char *MatcherTable, unsigned &MatcherIndex, SDValue N,
24460b57cec5SDimitry Andric     const SmallVectorImpl<std::pair<SDValue, SDNode *>> &RecordedNodes,
24470b57cec5SDimitry Andric     unsigned ChildNo) {
24480b57cec5SDimitry Andric   if (ChildNo >= N.getNumOperands())
24490b57cec5SDimitry Andric     return false;  // Match fails if out of range child #.
24500b57cec5SDimitry Andric   return ::CheckSame(MatcherTable, MatcherIndex, N.getOperand(ChildNo),
24510b57cec5SDimitry Andric                      RecordedNodes);
24520b57cec5SDimitry Andric }
24530b57cec5SDimitry Andric 
24540b57cec5SDimitry Andric /// CheckPatternPredicate - Implements OP_CheckPatternPredicate.
2455e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
24560b57cec5SDimitry Andric CheckPatternPredicate(const unsigned char *MatcherTable, unsigned &MatcherIndex,
24570b57cec5SDimitry Andric                       const SelectionDAGISel &SDISel) {
24580b57cec5SDimitry Andric   return SDISel.CheckPatternPredicate(MatcherTable[MatcherIndex++]);
24590b57cec5SDimitry Andric }
24600b57cec5SDimitry Andric 
24610b57cec5SDimitry Andric /// CheckNodePredicate - Implements OP_CheckNodePredicate.
2462e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
24630b57cec5SDimitry Andric CheckNodePredicate(const unsigned char *MatcherTable, unsigned &MatcherIndex,
24640b57cec5SDimitry Andric                    const SelectionDAGISel &SDISel, SDNode *N) {
24650b57cec5SDimitry Andric   return SDISel.CheckNodePredicate(N, MatcherTable[MatcherIndex++]);
24660b57cec5SDimitry Andric }
24670b57cec5SDimitry Andric 
2468e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
24690b57cec5SDimitry Andric CheckOpcode(const unsigned char *MatcherTable, unsigned &MatcherIndex,
24700b57cec5SDimitry Andric             SDNode *N) {
24710b57cec5SDimitry Andric   uint16_t Opc = MatcherTable[MatcherIndex++];
24720b57cec5SDimitry Andric   Opc |= (unsigned short)MatcherTable[MatcherIndex++] << 8;
24730b57cec5SDimitry Andric   return N->getOpcode() == Opc;
24740b57cec5SDimitry Andric }
24750b57cec5SDimitry Andric 
2476e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
24770b57cec5SDimitry Andric CheckType(const unsigned char *MatcherTable, unsigned &MatcherIndex, SDValue N,
24780b57cec5SDimitry Andric           const TargetLowering *TLI, const DataLayout &DL) {
24790b57cec5SDimitry Andric   MVT::SimpleValueType VT = (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
24800b57cec5SDimitry Andric   if (N.getValueType() == VT) return true;
24810b57cec5SDimitry Andric 
24820b57cec5SDimitry Andric   // Handle the case when VT is iPTR.
24830b57cec5SDimitry Andric   return VT == MVT::iPTR && N.getValueType() == TLI->getPointerTy(DL);
24840b57cec5SDimitry Andric }
24850b57cec5SDimitry Andric 
2486e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
24870b57cec5SDimitry Andric CheckChildType(const unsigned char *MatcherTable, unsigned &MatcherIndex,
24880b57cec5SDimitry Andric                SDValue N, const TargetLowering *TLI, const DataLayout &DL,
24890b57cec5SDimitry Andric                unsigned ChildNo) {
24900b57cec5SDimitry Andric   if (ChildNo >= N.getNumOperands())
24910b57cec5SDimitry Andric     return false;  // Match fails if out of range child #.
24920b57cec5SDimitry Andric   return ::CheckType(MatcherTable, MatcherIndex, N.getOperand(ChildNo), TLI,
24930b57cec5SDimitry Andric                      DL);
24940b57cec5SDimitry Andric }
24950b57cec5SDimitry Andric 
2496e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
24970b57cec5SDimitry Andric CheckCondCode(const unsigned char *MatcherTable, unsigned &MatcherIndex,
24980b57cec5SDimitry Andric               SDValue N) {
24990b57cec5SDimitry Andric   return cast<CondCodeSDNode>(N)->get() ==
25000b57cec5SDimitry Andric       (ISD::CondCode)MatcherTable[MatcherIndex++];
25010b57cec5SDimitry Andric }
25020b57cec5SDimitry Andric 
2503e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
25040b57cec5SDimitry Andric CheckChild2CondCode(const unsigned char *MatcherTable, unsigned &MatcherIndex,
25050b57cec5SDimitry Andric                     SDValue N) {
25060b57cec5SDimitry Andric   if (2 >= N.getNumOperands())
25070b57cec5SDimitry Andric     return false;
25080b57cec5SDimitry Andric   return ::CheckCondCode(MatcherTable, MatcherIndex, N.getOperand(2));
25090b57cec5SDimitry Andric }
25100b57cec5SDimitry Andric 
2511e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
25120b57cec5SDimitry Andric CheckValueType(const unsigned char *MatcherTable, unsigned &MatcherIndex,
25130b57cec5SDimitry Andric                SDValue N, const TargetLowering *TLI, const DataLayout &DL) {
25140b57cec5SDimitry Andric   MVT::SimpleValueType VT = (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
25150b57cec5SDimitry Andric   if (cast<VTSDNode>(N)->getVT() == VT)
25160b57cec5SDimitry Andric     return true;
25170b57cec5SDimitry Andric 
25180b57cec5SDimitry Andric   // Handle the case when VT is iPTR.
25190b57cec5SDimitry Andric   return VT == MVT::iPTR && cast<VTSDNode>(N)->getVT() == TLI->getPointerTy(DL);
25200b57cec5SDimitry Andric }
25210b57cec5SDimitry Andric 
2522fe6060f1SDimitry Andric // Bit 0 stores the sign of the immediate. The upper bits contain the magnitude
2523fe6060f1SDimitry Andric // shifted left by 1.
2524fe6060f1SDimitry Andric static uint64_t decodeSignRotatedValue(uint64_t V) {
2525fe6060f1SDimitry Andric   if ((V & 1) == 0)
2526fe6060f1SDimitry Andric     return V >> 1;
2527fe6060f1SDimitry Andric   if (V != 1)
2528fe6060f1SDimitry Andric     return -(V >> 1);
2529fe6060f1SDimitry Andric   // There is no such thing as -0 with integers.  "-0" really means MININT.
2530fe6060f1SDimitry Andric   return 1ULL << 63;
2531fe6060f1SDimitry Andric }
2532fe6060f1SDimitry Andric 
2533e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
25340b57cec5SDimitry Andric CheckInteger(const unsigned char *MatcherTable, unsigned &MatcherIndex,
25350b57cec5SDimitry Andric              SDValue N) {
25360b57cec5SDimitry Andric   int64_t Val = MatcherTable[MatcherIndex++];
25370b57cec5SDimitry Andric   if (Val & 128)
25380b57cec5SDimitry Andric     Val = GetVBR(Val, MatcherTable, MatcherIndex);
25390b57cec5SDimitry Andric 
2540fe6060f1SDimitry Andric   Val = decodeSignRotatedValue(Val);
2541fe6060f1SDimitry Andric 
25420b57cec5SDimitry Andric   ConstantSDNode *C = dyn_cast<ConstantSDNode>(N);
25430b57cec5SDimitry Andric   return C && C->getSExtValue() == Val;
25440b57cec5SDimitry Andric }
25450b57cec5SDimitry Andric 
2546e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
25470b57cec5SDimitry Andric CheckChildInteger(const unsigned char *MatcherTable, unsigned &MatcherIndex,
25480b57cec5SDimitry Andric                   SDValue N, unsigned ChildNo) {
25490b57cec5SDimitry Andric   if (ChildNo >= N.getNumOperands())
25500b57cec5SDimitry Andric     return false;  // Match fails if out of range child #.
25510b57cec5SDimitry Andric   return ::CheckInteger(MatcherTable, MatcherIndex, N.getOperand(ChildNo));
25520b57cec5SDimitry Andric }
25530b57cec5SDimitry Andric 
2554e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
25550b57cec5SDimitry Andric CheckAndImm(const unsigned char *MatcherTable, unsigned &MatcherIndex,
25560b57cec5SDimitry Andric             SDValue N, const SelectionDAGISel &SDISel) {
25570b57cec5SDimitry Andric   int64_t Val = MatcherTable[MatcherIndex++];
25580b57cec5SDimitry Andric   if (Val & 128)
25590b57cec5SDimitry Andric     Val = GetVBR(Val, MatcherTable, MatcherIndex);
25600b57cec5SDimitry Andric 
25610b57cec5SDimitry Andric   if (N->getOpcode() != ISD::AND) return false;
25620b57cec5SDimitry Andric 
25630b57cec5SDimitry Andric   ConstantSDNode *C = dyn_cast<ConstantSDNode>(N->getOperand(1));
25640b57cec5SDimitry Andric   return C && SDISel.CheckAndMask(N.getOperand(0), C, Val);
25650b57cec5SDimitry Andric }
25660b57cec5SDimitry Andric 
2567e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
2568e8d8bef9SDimitry Andric CheckOrImm(const unsigned char *MatcherTable, unsigned &MatcherIndex, SDValue N,
2569e8d8bef9SDimitry Andric            const SelectionDAGISel &SDISel) {
25700b57cec5SDimitry Andric   int64_t Val = MatcherTable[MatcherIndex++];
25710b57cec5SDimitry Andric   if (Val & 128)
25720b57cec5SDimitry Andric     Val = GetVBR(Val, MatcherTable, MatcherIndex);
25730b57cec5SDimitry Andric 
25740b57cec5SDimitry Andric   if (N->getOpcode() != ISD::OR) return false;
25750b57cec5SDimitry Andric 
25760b57cec5SDimitry Andric   ConstantSDNode *C = dyn_cast<ConstantSDNode>(N->getOperand(1));
25770b57cec5SDimitry Andric   return C && SDISel.CheckOrMask(N.getOperand(0), C, Val);
25780b57cec5SDimitry Andric }
25790b57cec5SDimitry Andric 
25800b57cec5SDimitry Andric /// IsPredicateKnownToFail - If we know how and can do so without pushing a
25810b57cec5SDimitry Andric /// scope, evaluate the current node.  If the current predicate is known to
25820b57cec5SDimitry Andric /// fail, set Result=true and return anything.  If the current predicate is
25830b57cec5SDimitry Andric /// known to pass, set Result=false and return the MatcherIndex to continue
25840b57cec5SDimitry Andric /// with.  If the current predicate is unknown, set Result=false and return the
25850b57cec5SDimitry Andric /// MatcherIndex to continue with.
25860b57cec5SDimitry Andric static unsigned IsPredicateKnownToFail(const unsigned char *Table,
25870b57cec5SDimitry Andric                                        unsigned Index, SDValue N,
25880b57cec5SDimitry Andric                                        bool &Result,
25890b57cec5SDimitry Andric                                        const SelectionDAGISel &SDISel,
25900b57cec5SDimitry Andric                   SmallVectorImpl<std::pair<SDValue, SDNode*>> &RecordedNodes) {
25910b57cec5SDimitry Andric   switch (Table[Index++]) {
25920b57cec5SDimitry Andric   default:
25930b57cec5SDimitry Andric     Result = false;
25940b57cec5SDimitry Andric     return Index-1;  // Could not evaluate this predicate.
25950b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckSame:
25960b57cec5SDimitry Andric     Result = !::CheckSame(Table, Index, N, RecordedNodes);
25970b57cec5SDimitry Andric     return Index;
25980b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild0Same:
25990b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild1Same:
26000b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild2Same:
26010b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild3Same:
26020b57cec5SDimitry Andric     Result = !::CheckChildSame(Table, Index, N, RecordedNodes,
26030b57cec5SDimitry Andric                         Table[Index-1] - SelectionDAGISel::OPC_CheckChild0Same);
26040b57cec5SDimitry Andric     return Index;
26050b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckPatternPredicate:
26060b57cec5SDimitry Andric     Result = !::CheckPatternPredicate(Table, Index, SDISel);
26070b57cec5SDimitry Andric     return Index;
26080b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckPredicate:
26090b57cec5SDimitry Andric     Result = !::CheckNodePredicate(Table, Index, SDISel, N.getNode());
26100b57cec5SDimitry Andric     return Index;
26110b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckOpcode:
26120b57cec5SDimitry Andric     Result = !::CheckOpcode(Table, Index, N.getNode());
26130b57cec5SDimitry Andric     return Index;
26140b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckType:
26150b57cec5SDimitry Andric     Result = !::CheckType(Table, Index, N, SDISel.TLI,
26160b57cec5SDimitry Andric                           SDISel.CurDAG->getDataLayout());
26170b57cec5SDimitry Andric     return Index;
26180b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckTypeRes: {
26190b57cec5SDimitry Andric     unsigned Res = Table[Index++];
26200b57cec5SDimitry Andric     Result = !::CheckType(Table, Index, N.getValue(Res), SDISel.TLI,
26210b57cec5SDimitry Andric                           SDISel.CurDAG->getDataLayout());
26220b57cec5SDimitry Andric     return Index;
26230b57cec5SDimitry Andric   }
26240b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild0Type:
26250b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild1Type:
26260b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild2Type:
26270b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild3Type:
26280b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild4Type:
26290b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild5Type:
26300b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild6Type:
26310b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild7Type:
26320b57cec5SDimitry Andric     Result = !::CheckChildType(
26330b57cec5SDimitry Andric                  Table, Index, N, SDISel.TLI, SDISel.CurDAG->getDataLayout(),
26340b57cec5SDimitry Andric                  Table[Index - 1] - SelectionDAGISel::OPC_CheckChild0Type);
26350b57cec5SDimitry Andric     return Index;
26360b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckCondCode:
26370b57cec5SDimitry Andric     Result = !::CheckCondCode(Table, Index, N);
26380b57cec5SDimitry Andric     return Index;
26390b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild2CondCode:
26400b57cec5SDimitry Andric     Result = !::CheckChild2CondCode(Table, Index, N);
26410b57cec5SDimitry Andric     return Index;
26420b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckValueType:
26430b57cec5SDimitry Andric     Result = !::CheckValueType(Table, Index, N, SDISel.TLI,
26440b57cec5SDimitry Andric                                SDISel.CurDAG->getDataLayout());
26450b57cec5SDimitry Andric     return Index;
26460b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckInteger:
26470b57cec5SDimitry Andric     Result = !::CheckInteger(Table, Index, N);
26480b57cec5SDimitry Andric     return Index;
26490b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild0Integer:
26500b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild1Integer:
26510b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild2Integer:
26520b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild3Integer:
26530b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild4Integer:
26540b57cec5SDimitry Andric     Result = !::CheckChildInteger(Table, Index, N,
26550b57cec5SDimitry Andric                      Table[Index-1] - SelectionDAGISel::OPC_CheckChild0Integer);
26560b57cec5SDimitry Andric     return Index;
26570b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckAndImm:
26580b57cec5SDimitry Andric     Result = !::CheckAndImm(Table, Index, N, SDISel);
26590b57cec5SDimitry Andric     return Index;
26600b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckOrImm:
26610b57cec5SDimitry Andric     Result = !::CheckOrImm(Table, Index, N, SDISel);
26620b57cec5SDimitry Andric     return Index;
26630b57cec5SDimitry Andric   }
26640b57cec5SDimitry Andric }
26650b57cec5SDimitry Andric 
26660b57cec5SDimitry Andric namespace {
26670b57cec5SDimitry Andric 
26680b57cec5SDimitry Andric struct MatchScope {
26690b57cec5SDimitry Andric   /// FailIndex - If this match fails, this is the index to continue with.
26700b57cec5SDimitry Andric   unsigned FailIndex;
26710b57cec5SDimitry Andric 
26720b57cec5SDimitry Andric   /// NodeStack - The node stack when the scope was formed.
26730b57cec5SDimitry Andric   SmallVector<SDValue, 4> NodeStack;
26740b57cec5SDimitry Andric 
26750b57cec5SDimitry Andric   /// NumRecordedNodes - The number of recorded nodes when the scope was formed.
26760b57cec5SDimitry Andric   unsigned NumRecordedNodes;
26770b57cec5SDimitry Andric 
26780b57cec5SDimitry Andric   /// NumMatchedMemRefs - The number of matched memref entries.
26790b57cec5SDimitry Andric   unsigned NumMatchedMemRefs;
26800b57cec5SDimitry Andric 
26810b57cec5SDimitry Andric   /// InputChain/InputGlue - The current chain/glue
26820b57cec5SDimitry Andric   SDValue InputChain, InputGlue;
26830b57cec5SDimitry Andric 
26840b57cec5SDimitry Andric   /// HasChainNodesMatched - True if the ChainNodesMatched list is non-empty.
26850b57cec5SDimitry Andric   bool HasChainNodesMatched;
26860b57cec5SDimitry Andric };
26870b57cec5SDimitry Andric 
26880b57cec5SDimitry Andric /// \A DAG update listener to keep the matching state
26890b57cec5SDimitry Andric /// (i.e. RecordedNodes and MatchScope) uptodate if the target is allowed to
26900b57cec5SDimitry Andric /// change the DAG while matching.  X86 addressing mode matcher is an example
26910b57cec5SDimitry Andric /// for this.
26920b57cec5SDimitry Andric class MatchStateUpdater : public SelectionDAG::DAGUpdateListener
26930b57cec5SDimitry Andric {
26940b57cec5SDimitry Andric   SDNode **NodeToMatch;
26950b57cec5SDimitry Andric   SmallVectorImpl<std::pair<SDValue, SDNode *>> &RecordedNodes;
26960b57cec5SDimitry Andric   SmallVectorImpl<MatchScope> &MatchScopes;
26970b57cec5SDimitry Andric 
26980b57cec5SDimitry Andric public:
26990b57cec5SDimitry Andric   MatchStateUpdater(SelectionDAG &DAG, SDNode **NodeToMatch,
27000b57cec5SDimitry Andric                     SmallVectorImpl<std::pair<SDValue, SDNode *>> &RN,
27010b57cec5SDimitry Andric                     SmallVectorImpl<MatchScope> &MS)
27020b57cec5SDimitry Andric       : SelectionDAG::DAGUpdateListener(DAG), NodeToMatch(NodeToMatch),
27030b57cec5SDimitry Andric         RecordedNodes(RN), MatchScopes(MS) {}
27040b57cec5SDimitry Andric 
27050b57cec5SDimitry Andric   void NodeDeleted(SDNode *N, SDNode *E) override {
27060b57cec5SDimitry Andric     // Some early-returns here to avoid the search if we deleted the node or
27070b57cec5SDimitry Andric     // if the update comes from MorphNodeTo (MorphNodeTo is the last thing we
27080b57cec5SDimitry Andric     // do, so it's unnecessary to update matching state at that point).
27090b57cec5SDimitry Andric     // Neither of these can occur currently because we only install this
27100b57cec5SDimitry Andric     // update listener during matching a complex patterns.
27110b57cec5SDimitry Andric     if (!E || E->isMachineOpcode())
27120b57cec5SDimitry Andric       return;
27130b57cec5SDimitry Andric     // Check if NodeToMatch was updated.
27140b57cec5SDimitry Andric     if (N == *NodeToMatch)
27150b57cec5SDimitry Andric       *NodeToMatch = E;
27160b57cec5SDimitry Andric     // Performing linear search here does not matter because we almost never
27170b57cec5SDimitry Andric     // run this code.  You'd have to have a CSE during complex pattern
27180b57cec5SDimitry Andric     // matching.
27190b57cec5SDimitry Andric     for (auto &I : RecordedNodes)
27200b57cec5SDimitry Andric       if (I.first.getNode() == N)
27210b57cec5SDimitry Andric         I.first.setNode(E);
27220b57cec5SDimitry Andric 
27230b57cec5SDimitry Andric     for (auto &I : MatchScopes)
27240b57cec5SDimitry Andric       for (auto &J : I.NodeStack)
27250b57cec5SDimitry Andric         if (J.getNode() == N)
27260b57cec5SDimitry Andric           J.setNode(E);
27270b57cec5SDimitry Andric   }
27280b57cec5SDimitry Andric };
27290b57cec5SDimitry Andric 
27300b57cec5SDimitry Andric } // end anonymous namespace
27310b57cec5SDimitry Andric 
27320b57cec5SDimitry Andric void SelectionDAGISel::SelectCodeCommon(SDNode *NodeToMatch,
27330b57cec5SDimitry Andric                                         const unsigned char *MatcherTable,
27340b57cec5SDimitry Andric                                         unsigned TableSize) {
27350b57cec5SDimitry Andric   // FIXME: Should these even be selected?  Handle these cases in the caller?
27360b57cec5SDimitry Andric   switch (NodeToMatch->getOpcode()) {
27370b57cec5SDimitry Andric   default:
27380b57cec5SDimitry Andric     break;
27390b57cec5SDimitry Andric   case ISD::EntryToken:       // These nodes remain the same.
27400b57cec5SDimitry Andric   case ISD::BasicBlock:
27410b57cec5SDimitry Andric   case ISD::Register:
27420b57cec5SDimitry Andric   case ISD::RegisterMask:
27430b57cec5SDimitry Andric   case ISD::HANDLENODE:
27440b57cec5SDimitry Andric   case ISD::MDNODE_SDNODE:
27450b57cec5SDimitry Andric   case ISD::TargetConstant:
27460b57cec5SDimitry Andric   case ISD::TargetConstantFP:
27470b57cec5SDimitry Andric   case ISD::TargetConstantPool:
27480b57cec5SDimitry Andric   case ISD::TargetFrameIndex:
27490b57cec5SDimitry Andric   case ISD::TargetExternalSymbol:
27500b57cec5SDimitry Andric   case ISD::MCSymbol:
27510b57cec5SDimitry Andric   case ISD::TargetBlockAddress:
27520b57cec5SDimitry Andric   case ISD::TargetJumpTable:
27530b57cec5SDimitry Andric   case ISD::TargetGlobalTLSAddress:
27540b57cec5SDimitry Andric   case ISD::TargetGlobalAddress:
27550b57cec5SDimitry Andric   case ISD::TokenFactor:
27560b57cec5SDimitry Andric   case ISD::CopyFromReg:
27570b57cec5SDimitry Andric   case ISD::CopyToReg:
27580b57cec5SDimitry Andric   case ISD::EH_LABEL:
27590b57cec5SDimitry Andric   case ISD::ANNOTATION_LABEL:
27600b57cec5SDimitry Andric   case ISD::LIFETIME_START:
27610b57cec5SDimitry Andric   case ISD::LIFETIME_END:
2762e8d8bef9SDimitry Andric   case ISD::PSEUDO_PROBE:
27630b57cec5SDimitry Andric     NodeToMatch->setNodeId(-1); // Mark selected.
27640b57cec5SDimitry Andric     return;
27650b57cec5SDimitry Andric   case ISD::AssertSext:
27660b57cec5SDimitry Andric   case ISD::AssertZext:
27675ffd83dbSDimitry Andric   case ISD::AssertAlign:
27680b57cec5SDimitry Andric     ReplaceUses(SDValue(NodeToMatch, 0), NodeToMatch->getOperand(0));
27690b57cec5SDimitry Andric     CurDAG->RemoveDeadNode(NodeToMatch);
27700b57cec5SDimitry Andric     return;
27710b57cec5SDimitry Andric   case ISD::INLINEASM:
27720b57cec5SDimitry Andric   case ISD::INLINEASM_BR:
27735ffd83dbSDimitry Andric     Select_INLINEASM(NodeToMatch);
27740b57cec5SDimitry Andric     return;
27750b57cec5SDimitry Andric   case ISD::READ_REGISTER:
27760b57cec5SDimitry Andric     Select_READ_REGISTER(NodeToMatch);
27770b57cec5SDimitry Andric     return;
27780b57cec5SDimitry Andric   case ISD::WRITE_REGISTER:
27790b57cec5SDimitry Andric     Select_WRITE_REGISTER(NodeToMatch);
27800b57cec5SDimitry Andric     return;
27810b57cec5SDimitry Andric   case ISD::UNDEF:
27820b57cec5SDimitry Andric     Select_UNDEF(NodeToMatch);
27830b57cec5SDimitry Andric     return;
27845ffd83dbSDimitry Andric   case ISD::FREEZE:
27855ffd83dbSDimitry Andric     Select_FREEZE(NodeToMatch);
27865ffd83dbSDimitry Andric     return;
2787fe6060f1SDimitry Andric   case ISD::ARITH_FENCE:
2788fe6060f1SDimitry Andric     Select_ARITH_FENCE(NodeToMatch);
2789fe6060f1SDimitry Andric     return;
27900b57cec5SDimitry Andric   }
27910b57cec5SDimitry Andric 
27920b57cec5SDimitry Andric   assert(!NodeToMatch->isMachineOpcode() && "Node already selected!");
27930b57cec5SDimitry Andric 
27940b57cec5SDimitry Andric   // Set up the node stack with NodeToMatch as the only node on the stack.
27950b57cec5SDimitry Andric   SmallVector<SDValue, 8> NodeStack;
27960b57cec5SDimitry Andric   SDValue N = SDValue(NodeToMatch, 0);
27970b57cec5SDimitry Andric   NodeStack.push_back(N);
27980b57cec5SDimitry Andric 
27990b57cec5SDimitry Andric   // MatchScopes - Scopes used when matching, if a match failure happens, this
28000b57cec5SDimitry Andric   // indicates where to continue checking.
28010b57cec5SDimitry Andric   SmallVector<MatchScope, 8> MatchScopes;
28020b57cec5SDimitry Andric 
28030b57cec5SDimitry Andric   // RecordedNodes - This is the set of nodes that have been recorded by the
28040b57cec5SDimitry Andric   // state machine.  The second value is the parent of the node, or null if the
28050b57cec5SDimitry Andric   // root is recorded.
28060b57cec5SDimitry Andric   SmallVector<std::pair<SDValue, SDNode*>, 8> RecordedNodes;
28070b57cec5SDimitry Andric 
28080b57cec5SDimitry Andric   // MatchedMemRefs - This is the set of MemRef's we've seen in the input
28090b57cec5SDimitry Andric   // pattern.
28100b57cec5SDimitry Andric   SmallVector<MachineMemOperand*, 2> MatchedMemRefs;
28110b57cec5SDimitry Andric 
28120b57cec5SDimitry Andric   // These are the current input chain and glue for use when generating nodes.
28130b57cec5SDimitry Andric   // Various Emit operations change these.  For example, emitting a copytoreg
28140b57cec5SDimitry Andric   // uses and updates these.
28150b57cec5SDimitry Andric   SDValue InputChain, InputGlue;
28160b57cec5SDimitry Andric 
28170b57cec5SDimitry Andric   // ChainNodesMatched - If a pattern matches nodes that have input/output
28180b57cec5SDimitry Andric   // chains, the OPC_EmitMergeInputChains operation is emitted which indicates
28190b57cec5SDimitry Andric   // which ones they are.  The result is captured into this list so that we can
28200b57cec5SDimitry Andric   // update the chain results when the pattern is complete.
28210b57cec5SDimitry Andric   SmallVector<SDNode*, 3> ChainNodesMatched;
28220b57cec5SDimitry Andric 
28230b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "ISEL: Starting pattern match\n");
28240b57cec5SDimitry Andric 
28250b57cec5SDimitry Andric   // Determine where to start the interpreter.  Normally we start at opcode #0,
28260b57cec5SDimitry Andric   // but if the state machine starts with an OPC_SwitchOpcode, then we
28270b57cec5SDimitry Andric   // accelerate the first lookup (which is guaranteed to be hot) with the
28280b57cec5SDimitry Andric   // OpcodeOffset table.
28290b57cec5SDimitry Andric   unsigned MatcherIndex = 0;
28300b57cec5SDimitry Andric 
28310b57cec5SDimitry Andric   if (!OpcodeOffset.empty()) {
28320b57cec5SDimitry Andric     // Already computed the OpcodeOffset table, just index into it.
28330b57cec5SDimitry Andric     if (N.getOpcode() < OpcodeOffset.size())
28340b57cec5SDimitry Andric       MatcherIndex = OpcodeOffset[N.getOpcode()];
28350b57cec5SDimitry Andric     LLVM_DEBUG(dbgs() << "  Initial Opcode index to " << MatcherIndex << "\n");
28360b57cec5SDimitry Andric 
28370b57cec5SDimitry Andric   } else if (MatcherTable[0] == OPC_SwitchOpcode) {
28380b57cec5SDimitry Andric     // Otherwise, the table isn't computed, but the state machine does start
28390b57cec5SDimitry Andric     // with an OPC_SwitchOpcode instruction.  Populate the table now, since this
28400b57cec5SDimitry Andric     // is the first time we're selecting an instruction.
28410b57cec5SDimitry Andric     unsigned Idx = 1;
28420b57cec5SDimitry Andric     while (true) {
28430b57cec5SDimitry Andric       // Get the size of this case.
28440b57cec5SDimitry Andric       unsigned CaseSize = MatcherTable[Idx++];
28450b57cec5SDimitry Andric       if (CaseSize & 128)
28460b57cec5SDimitry Andric         CaseSize = GetVBR(CaseSize, MatcherTable, Idx);
28470b57cec5SDimitry Andric       if (CaseSize == 0) break;
28480b57cec5SDimitry Andric 
28490b57cec5SDimitry Andric       // Get the opcode, add the index to the table.
28500b57cec5SDimitry Andric       uint16_t Opc = MatcherTable[Idx++];
28510b57cec5SDimitry Andric       Opc |= (unsigned short)MatcherTable[Idx++] << 8;
28520b57cec5SDimitry Andric       if (Opc >= OpcodeOffset.size())
28530b57cec5SDimitry Andric         OpcodeOffset.resize((Opc+1)*2);
28540b57cec5SDimitry Andric       OpcodeOffset[Opc] = Idx;
28550b57cec5SDimitry Andric       Idx += CaseSize;
28560b57cec5SDimitry Andric     }
28570b57cec5SDimitry Andric 
28580b57cec5SDimitry Andric     // Okay, do the lookup for the first opcode.
28590b57cec5SDimitry Andric     if (N.getOpcode() < OpcodeOffset.size())
28600b57cec5SDimitry Andric       MatcherIndex = OpcodeOffset[N.getOpcode()];
28610b57cec5SDimitry Andric   }
28620b57cec5SDimitry Andric 
28630b57cec5SDimitry Andric   while (true) {
28640b57cec5SDimitry Andric     assert(MatcherIndex < TableSize && "Invalid index");
28650b57cec5SDimitry Andric #ifndef NDEBUG
28660b57cec5SDimitry Andric     unsigned CurrentOpcodeIndex = MatcherIndex;
28670b57cec5SDimitry Andric #endif
28680b57cec5SDimitry Andric     BuiltinOpcodes Opcode = (BuiltinOpcodes)MatcherTable[MatcherIndex++];
28690b57cec5SDimitry Andric     switch (Opcode) {
28700b57cec5SDimitry Andric     case OPC_Scope: {
28710b57cec5SDimitry Andric       // Okay, the semantics of this operation are that we should push a scope
28720b57cec5SDimitry Andric       // then evaluate the first child.  However, pushing a scope only to have
28730b57cec5SDimitry Andric       // the first check fail (which then pops it) is inefficient.  If we can
28740b57cec5SDimitry Andric       // determine immediately that the first check (or first several) will
28750b57cec5SDimitry Andric       // immediately fail, don't even bother pushing a scope for them.
28760b57cec5SDimitry Andric       unsigned FailIndex;
28770b57cec5SDimitry Andric 
28780b57cec5SDimitry Andric       while (true) {
28790b57cec5SDimitry Andric         unsigned NumToSkip = MatcherTable[MatcherIndex++];
28800b57cec5SDimitry Andric         if (NumToSkip & 128)
28810b57cec5SDimitry Andric           NumToSkip = GetVBR(NumToSkip, MatcherTable, MatcherIndex);
28820b57cec5SDimitry Andric         // Found the end of the scope with no match.
28830b57cec5SDimitry Andric         if (NumToSkip == 0) {
28840b57cec5SDimitry Andric           FailIndex = 0;
28850b57cec5SDimitry Andric           break;
28860b57cec5SDimitry Andric         }
28870b57cec5SDimitry Andric 
28880b57cec5SDimitry Andric         FailIndex = MatcherIndex+NumToSkip;
28890b57cec5SDimitry Andric 
28900b57cec5SDimitry Andric         unsigned MatcherIndexOfPredicate = MatcherIndex;
28910b57cec5SDimitry Andric         (void)MatcherIndexOfPredicate; // silence warning.
28920b57cec5SDimitry Andric 
28930b57cec5SDimitry Andric         // If we can't evaluate this predicate without pushing a scope (e.g. if
28940b57cec5SDimitry Andric         // it is a 'MoveParent') or if the predicate succeeds on this node, we
28950b57cec5SDimitry Andric         // push the scope and evaluate the full predicate chain.
28960b57cec5SDimitry Andric         bool Result;
28970b57cec5SDimitry Andric         MatcherIndex = IsPredicateKnownToFail(MatcherTable, MatcherIndex, N,
28980b57cec5SDimitry Andric                                               Result, *this, RecordedNodes);
28990b57cec5SDimitry Andric         if (!Result)
29000b57cec5SDimitry Andric           break;
29010b57cec5SDimitry Andric 
29020b57cec5SDimitry Andric         LLVM_DEBUG(
29030b57cec5SDimitry Andric             dbgs() << "  Skipped scope entry (due to false predicate) at "
29040b57cec5SDimitry Andric                    << "index " << MatcherIndexOfPredicate << ", continuing at "
29050b57cec5SDimitry Andric                    << FailIndex << "\n");
29060b57cec5SDimitry Andric         ++NumDAGIselRetries;
29070b57cec5SDimitry Andric 
29080b57cec5SDimitry Andric         // Otherwise, we know that this case of the Scope is guaranteed to fail,
29090b57cec5SDimitry Andric         // move to the next case.
29100b57cec5SDimitry Andric         MatcherIndex = FailIndex;
29110b57cec5SDimitry Andric       }
29120b57cec5SDimitry Andric 
29130b57cec5SDimitry Andric       // If the whole scope failed to match, bail.
29140b57cec5SDimitry Andric       if (FailIndex == 0) break;
29150b57cec5SDimitry Andric 
29160b57cec5SDimitry Andric       // Push a MatchScope which indicates where to go if the first child fails
29170b57cec5SDimitry Andric       // to match.
29180b57cec5SDimitry Andric       MatchScope NewEntry;
29190b57cec5SDimitry Andric       NewEntry.FailIndex = FailIndex;
29200b57cec5SDimitry Andric       NewEntry.NodeStack.append(NodeStack.begin(), NodeStack.end());
29210b57cec5SDimitry Andric       NewEntry.NumRecordedNodes = RecordedNodes.size();
29220b57cec5SDimitry Andric       NewEntry.NumMatchedMemRefs = MatchedMemRefs.size();
29230b57cec5SDimitry Andric       NewEntry.InputChain = InputChain;
29240b57cec5SDimitry Andric       NewEntry.InputGlue = InputGlue;
29250b57cec5SDimitry Andric       NewEntry.HasChainNodesMatched = !ChainNodesMatched.empty();
29260b57cec5SDimitry Andric       MatchScopes.push_back(NewEntry);
29270b57cec5SDimitry Andric       continue;
29280b57cec5SDimitry Andric     }
29290b57cec5SDimitry Andric     case OPC_RecordNode: {
29300b57cec5SDimitry Andric       // Remember this node, it may end up being an operand in the pattern.
29310b57cec5SDimitry Andric       SDNode *Parent = nullptr;
29320b57cec5SDimitry Andric       if (NodeStack.size() > 1)
29330b57cec5SDimitry Andric         Parent = NodeStack[NodeStack.size()-2].getNode();
29340b57cec5SDimitry Andric       RecordedNodes.push_back(std::make_pair(N, Parent));
29350b57cec5SDimitry Andric       continue;
29360b57cec5SDimitry Andric     }
29370b57cec5SDimitry Andric 
29380b57cec5SDimitry Andric     case OPC_RecordChild0: case OPC_RecordChild1:
29390b57cec5SDimitry Andric     case OPC_RecordChild2: case OPC_RecordChild3:
29400b57cec5SDimitry Andric     case OPC_RecordChild4: case OPC_RecordChild5:
29410b57cec5SDimitry Andric     case OPC_RecordChild6: case OPC_RecordChild7: {
29420b57cec5SDimitry Andric       unsigned ChildNo = Opcode-OPC_RecordChild0;
29430b57cec5SDimitry Andric       if (ChildNo >= N.getNumOperands())
29440b57cec5SDimitry Andric         break;  // Match fails if out of range child #.
29450b57cec5SDimitry Andric 
29460b57cec5SDimitry Andric       RecordedNodes.push_back(std::make_pair(N->getOperand(ChildNo),
29470b57cec5SDimitry Andric                                              N.getNode()));
29480b57cec5SDimitry Andric       continue;
29490b57cec5SDimitry Andric     }
29500b57cec5SDimitry Andric     case OPC_RecordMemRef:
29510b57cec5SDimitry Andric       if (auto *MN = dyn_cast<MemSDNode>(N))
29520b57cec5SDimitry Andric         MatchedMemRefs.push_back(MN->getMemOperand());
29530b57cec5SDimitry Andric       else {
29540b57cec5SDimitry Andric         LLVM_DEBUG(dbgs() << "Expected MemSDNode "; N->dump(CurDAG);
29550b57cec5SDimitry Andric                    dbgs() << '\n');
29560b57cec5SDimitry Andric       }
29570b57cec5SDimitry Andric 
29580b57cec5SDimitry Andric       continue;
29590b57cec5SDimitry Andric 
29600b57cec5SDimitry Andric     case OPC_CaptureGlueInput:
29610b57cec5SDimitry Andric       // If the current node has an input glue, capture it in InputGlue.
29620b57cec5SDimitry Andric       if (N->getNumOperands() != 0 &&
29630b57cec5SDimitry Andric           N->getOperand(N->getNumOperands()-1).getValueType() == MVT::Glue)
29640b57cec5SDimitry Andric         InputGlue = N->getOperand(N->getNumOperands()-1);
29650b57cec5SDimitry Andric       continue;
29660b57cec5SDimitry Andric 
29670b57cec5SDimitry Andric     case OPC_MoveChild: {
29680b57cec5SDimitry Andric       unsigned ChildNo = MatcherTable[MatcherIndex++];
29690b57cec5SDimitry Andric       if (ChildNo >= N.getNumOperands())
29700b57cec5SDimitry Andric         break;  // Match fails if out of range child #.
29710b57cec5SDimitry Andric       N = N.getOperand(ChildNo);
29720b57cec5SDimitry Andric       NodeStack.push_back(N);
29730b57cec5SDimitry Andric       continue;
29740b57cec5SDimitry Andric     }
29750b57cec5SDimitry Andric 
29760b57cec5SDimitry Andric     case OPC_MoveChild0: case OPC_MoveChild1:
29770b57cec5SDimitry Andric     case OPC_MoveChild2: case OPC_MoveChild3:
29780b57cec5SDimitry Andric     case OPC_MoveChild4: case OPC_MoveChild5:
29790b57cec5SDimitry Andric     case OPC_MoveChild6: case OPC_MoveChild7: {
29800b57cec5SDimitry Andric       unsigned ChildNo = Opcode-OPC_MoveChild0;
29810b57cec5SDimitry Andric       if (ChildNo >= N.getNumOperands())
29820b57cec5SDimitry Andric         break;  // Match fails if out of range child #.
29830b57cec5SDimitry Andric       N = N.getOperand(ChildNo);
29840b57cec5SDimitry Andric       NodeStack.push_back(N);
29850b57cec5SDimitry Andric       continue;
29860b57cec5SDimitry Andric     }
29870b57cec5SDimitry Andric 
29880b57cec5SDimitry Andric     case OPC_MoveParent:
29890b57cec5SDimitry Andric       // Pop the current node off the NodeStack.
29900b57cec5SDimitry Andric       NodeStack.pop_back();
29910b57cec5SDimitry Andric       assert(!NodeStack.empty() && "Node stack imbalance!");
29920b57cec5SDimitry Andric       N = NodeStack.back();
29930b57cec5SDimitry Andric       continue;
29940b57cec5SDimitry Andric 
29950b57cec5SDimitry Andric     case OPC_CheckSame:
29960b57cec5SDimitry Andric       if (!::CheckSame(MatcherTable, MatcherIndex, N, RecordedNodes)) break;
29970b57cec5SDimitry Andric       continue;
29980b57cec5SDimitry Andric 
29990b57cec5SDimitry Andric     case OPC_CheckChild0Same: case OPC_CheckChild1Same:
30000b57cec5SDimitry Andric     case OPC_CheckChild2Same: case OPC_CheckChild3Same:
30010b57cec5SDimitry Andric       if (!::CheckChildSame(MatcherTable, MatcherIndex, N, RecordedNodes,
30020b57cec5SDimitry Andric                             Opcode-OPC_CheckChild0Same))
30030b57cec5SDimitry Andric         break;
30040b57cec5SDimitry Andric       continue;
30050b57cec5SDimitry Andric 
30060b57cec5SDimitry Andric     case OPC_CheckPatternPredicate:
30070b57cec5SDimitry Andric       if (!::CheckPatternPredicate(MatcherTable, MatcherIndex, *this)) break;
30080b57cec5SDimitry Andric       continue;
30090b57cec5SDimitry Andric     case OPC_CheckPredicate:
30100b57cec5SDimitry Andric       if (!::CheckNodePredicate(MatcherTable, MatcherIndex, *this,
30110b57cec5SDimitry Andric                                 N.getNode()))
30120b57cec5SDimitry Andric         break;
30130b57cec5SDimitry Andric       continue;
30140b57cec5SDimitry Andric     case OPC_CheckPredicateWithOperands: {
30150b57cec5SDimitry Andric       unsigned OpNum = MatcherTable[MatcherIndex++];
30160b57cec5SDimitry Andric       SmallVector<SDValue, 8> Operands;
30170b57cec5SDimitry Andric 
30180b57cec5SDimitry Andric       for (unsigned i = 0; i < OpNum; ++i)
30190b57cec5SDimitry Andric         Operands.push_back(RecordedNodes[MatcherTable[MatcherIndex++]].first);
30200b57cec5SDimitry Andric 
30210b57cec5SDimitry Andric       unsigned PredNo = MatcherTable[MatcherIndex++];
30220b57cec5SDimitry Andric       if (!CheckNodePredicateWithOperands(N.getNode(), PredNo, Operands))
30230b57cec5SDimitry Andric         break;
30240b57cec5SDimitry Andric       continue;
30250b57cec5SDimitry Andric     }
30260b57cec5SDimitry Andric     case OPC_CheckComplexPat: {
30270b57cec5SDimitry Andric       unsigned CPNum = MatcherTable[MatcherIndex++];
30280b57cec5SDimitry Andric       unsigned RecNo = MatcherTable[MatcherIndex++];
30290b57cec5SDimitry Andric       assert(RecNo < RecordedNodes.size() && "Invalid CheckComplexPat");
30300b57cec5SDimitry Andric 
30310b57cec5SDimitry Andric       // If target can modify DAG during matching, keep the matching state
30320b57cec5SDimitry Andric       // consistent.
30330b57cec5SDimitry Andric       std::unique_ptr<MatchStateUpdater> MSU;
30340b57cec5SDimitry Andric       if (ComplexPatternFuncMutatesDAG())
30350b57cec5SDimitry Andric         MSU.reset(new MatchStateUpdater(*CurDAG, &NodeToMatch, RecordedNodes,
30360b57cec5SDimitry Andric                                         MatchScopes));
30370b57cec5SDimitry Andric 
30380b57cec5SDimitry Andric       if (!CheckComplexPattern(NodeToMatch, RecordedNodes[RecNo].second,
30390b57cec5SDimitry Andric                                RecordedNodes[RecNo].first, CPNum,
30400b57cec5SDimitry Andric                                RecordedNodes))
30410b57cec5SDimitry Andric         break;
30420b57cec5SDimitry Andric       continue;
30430b57cec5SDimitry Andric     }
30440b57cec5SDimitry Andric     case OPC_CheckOpcode:
30450b57cec5SDimitry Andric       if (!::CheckOpcode(MatcherTable, MatcherIndex, N.getNode())) break;
30460b57cec5SDimitry Andric       continue;
30470b57cec5SDimitry Andric 
30480b57cec5SDimitry Andric     case OPC_CheckType:
30490b57cec5SDimitry Andric       if (!::CheckType(MatcherTable, MatcherIndex, N, TLI,
30500b57cec5SDimitry Andric                        CurDAG->getDataLayout()))
30510b57cec5SDimitry Andric         break;
30520b57cec5SDimitry Andric       continue;
30530b57cec5SDimitry Andric 
30540b57cec5SDimitry Andric     case OPC_CheckTypeRes: {
30550b57cec5SDimitry Andric       unsigned Res = MatcherTable[MatcherIndex++];
30560b57cec5SDimitry Andric       if (!::CheckType(MatcherTable, MatcherIndex, N.getValue(Res), TLI,
30570b57cec5SDimitry Andric                        CurDAG->getDataLayout()))
30580b57cec5SDimitry Andric         break;
30590b57cec5SDimitry Andric       continue;
30600b57cec5SDimitry Andric     }
30610b57cec5SDimitry Andric 
30620b57cec5SDimitry Andric     case OPC_SwitchOpcode: {
30630b57cec5SDimitry Andric       unsigned CurNodeOpcode = N.getOpcode();
30640b57cec5SDimitry Andric       unsigned SwitchStart = MatcherIndex-1; (void)SwitchStart;
30650b57cec5SDimitry Andric       unsigned CaseSize;
30660b57cec5SDimitry Andric       while (true) {
30670b57cec5SDimitry Andric         // Get the size of this case.
30680b57cec5SDimitry Andric         CaseSize = MatcherTable[MatcherIndex++];
30690b57cec5SDimitry Andric         if (CaseSize & 128)
30700b57cec5SDimitry Andric           CaseSize = GetVBR(CaseSize, MatcherTable, MatcherIndex);
30710b57cec5SDimitry Andric         if (CaseSize == 0) break;
30720b57cec5SDimitry Andric 
30730b57cec5SDimitry Andric         uint16_t Opc = MatcherTable[MatcherIndex++];
30740b57cec5SDimitry Andric         Opc |= (unsigned short)MatcherTable[MatcherIndex++] << 8;
30750b57cec5SDimitry Andric 
30760b57cec5SDimitry Andric         // If the opcode matches, then we will execute this case.
30770b57cec5SDimitry Andric         if (CurNodeOpcode == Opc)
30780b57cec5SDimitry Andric           break;
30790b57cec5SDimitry Andric 
30800b57cec5SDimitry Andric         // Otherwise, skip over this case.
30810b57cec5SDimitry Andric         MatcherIndex += CaseSize;
30820b57cec5SDimitry Andric       }
30830b57cec5SDimitry Andric 
30840b57cec5SDimitry Andric       // If no cases matched, bail out.
30850b57cec5SDimitry Andric       if (CaseSize == 0) break;
30860b57cec5SDimitry Andric 
30870b57cec5SDimitry Andric       // Otherwise, execute the case we found.
30880b57cec5SDimitry Andric       LLVM_DEBUG(dbgs() << "  OpcodeSwitch from " << SwitchStart << " to "
30890b57cec5SDimitry Andric                         << MatcherIndex << "\n");
30900b57cec5SDimitry Andric       continue;
30910b57cec5SDimitry Andric     }
30920b57cec5SDimitry Andric 
30930b57cec5SDimitry Andric     case OPC_SwitchType: {
30940b57cec5SDimitry Andric       MVT CurNodeVT = N.getSimpleValueType();
30950b57cec5SDimitry Andric       unsigned SwitchStart = MatcherIndex-1; (void)SwitchStart;
30960b57cec5SDimitry Andric       unsigned CaseSize;
30970b57cec5SDimitry Andric       while (true) {
30980b57cec5SDimitry Andric         // Get the size of this case.
30990b57cec5SDimitry Andric         CaseSize = MatcherTable[MatcherIndex++];
31000b57cec5SDimitry Andric         if (CaseSize & 128)
31010b57cec5SDimitry Andric           CaseSize = GetVBR(CaseSize, MatcherTable, MatcherIndex);
31020b57cec5SDimitry Andric         if (CaseSize == 0) break;
31030b57cec5SDimitry Andric 
31040b57cec5SDimitry Andric         MVT CaseVT = (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
31050b57cec5SDimitry Andric         if (CaseVT == MVT::iPTR)
31060b57cec5SDimitry Andric           CaseVT = TLI->getPointerTy(CurDAG->getDataLayout());
31070b57cec5SDimitry Andric 
31080b57cec5SDimitry Andric         // If the VT matches, then we will execute this case.
31090b57cec5SDimitry Andric         if (CurNodeVT == CaseVT)
31100b57cec5SDimitry Andric           break;
31110b57cec5SDimitry Andric 
31120b57cec5SDimitry Andric         // Otherwise, skip over this case.
31130b57cec5SDimitry Andric         MatcherIndex += CaseSize;
31140b57cec5SDimitry Andric       }
31150b57cec5SDimitry Andric 
31160b57cec5SDimitry Andric       // If no cases matched, bail out.
31170b57cec5SDimitry Andric       if (CaseSize == 0) break;
31180b57cec5SDimitry Andric 
31190b57cec5SDimitry Andric       // Otherwise, execute the case we found.
31200b57cec5SDimitry Andric       LLVM_DEBUG(dbgs() << "  TypeSwitch[" << EVT(CurNodeVT).getEVTString()
31210b57cec5SDimitry Andric                         << "] from " << SwitchStart << " to " << MatcherIndex
31220b57cec5SDimitry Andric                         << '\n');
31230b57cec5SDimitry Andric       continue;
31240b57cec5SDimitry Andric     }
31250b57cec5SDimitry Andric     case OPC_CheckChild0Type: case OPC_CheckChild1Type:
31260b57cec5SDimitry Andric     case OPC_CheckChild2Type: case OPC_CheckChild3Type:
31270b57cec5SDimitry Andric     case OPC_CheckChild4Type: case OPC_CheckChild5Type:
31280b57cec5SDimitry Andric     case OPC_CheckChild6Type: case OPC_CheckChild7Type:
31290b57cec5SDimitry Andric       if (!::CheckChildType(MatcherTable, MatcherIndex, N, TLI,
31300b57cec5SDimitry Andric                             CurDAG->getDataLayout(),
31310b57cec5SDimitry Andric                             Opcode - OPC_CheckChild0Type))
31320b57cec5SDimitry Andric         break;
31330b57cec5SDimitry Andric       continue;
31340b57cec5SDimitry Andric     case OPC_CheckCondCode:
31350b57cec5SDimitry Andric       if (!::CheckCondCode(MatcherTable, MatcherIndex, N)) break;
31360b57cec5SDimitry Andric       continue;
31370b57cec5SDimitry Andric     case OPC_CheckChild2CondCode:
31380b57cec5SDimitry Andric       if (!::CheckChild2CondCode(MatcherTable, MatcherIndex, N)) break;
31390b57cec5SDimitry Andric       continue;
31400b57cec5SDimitry Andric     case OPC_CheckValueType:
31410b57cec5SDimitry Andric       if (!::CheckValueType(MatcherTable, MatcherIndex, N, TLI,
31420b57cec5SDimitry Andric                             CurDAG->getDataLayout()))
31430b57cec5SDimitry Andric         break;
31440b57cec5SDimitry Andric       continue;
31450b57cec5SDimitry Andric     case OPC_CheckInteger:
31460b57cec5SDimitry Andric       if (!::CheckInteger(MatcherTable, MatcherIndex, N)) break;
31470b57cec5SDimitry Andric       continue;
31480b57cec5SDimitry Andric     case OPC_CheckChild0Integer: case OPC_CheckChild1Integer:
31490b57cec5SDimitry Andric     case OPC_CheckChild2Integer: case OPC_CheckChild3Integer:
31500b57cec5SDimitry Andric     case OPC_CheckChild4Integer:
31510b57cec5SDimitry Andric       if (!::CheckChildInteger(MatcherTable, MatcherIndex, N,
31520b57cec5SDimitry Andric                                Opcode-OPC_CheckChild0Integer)) break;
31530b57cec5SDimitry Andric       continue;
31540b57cec5SDimitry Andric     case OPC_CheckAndImm:
31550b57cec5SDimitry Andric       if (!::CheckAndImm(MatcherTable, MatcherIndex, N, *this)) break;
31560b57cec5SDimitry Andric       continue;
31570b57cec5SDimitry Andric     case OPC_CheckOrImm:
31580b57cec5SDimitry Andric       if (!::CheckOrImm(MatcherTable, MatcherIndex, N, *this)) break;
31590b57cec5SDimitry Andric       continue;
31600b57cec5SDimitry Andric     case OPC_CheckImmAllOnesV:
3161e8d8bef9SDimitry Andric       if (!ISD::isConstantSplatVectorAllOnes(N.getNode()))
3162e8d8bef9SDimitry Andric         break;
31630b57cec5SDimitry Andric       continue;
31640b57cec5SDimitry Andric     case OPC_CheckImmAllZerosV:
3165e8d8bef9SDimitry Andric       if (!ISD::isConstantSplatVectorAllZeros(N.getNode()))
3166e8d8bef9SDimitry Andric         break;
31670b57cec5SDimitry Andric       continue;
31680b57cec5SDimitry Andric 
31690b57cec5SDimitry Andric     case OPC_CheckFoldableChainNode: {
31700b57cec5SDimitry Andric       assert(NodeStack.size() != 1 && "No parent node");
31710b57cec5SDimitry Andric       // Verify that all intermediate nodes between the root and this one have
3172480093f4SDimitry Andric       // a single use (ignoring chains, which are handled in UpdateChains).
31730b57cec5SDimitry Andric       bool HasMultipleUses = false;
3174480093f4SDimitry Andric       for (unsigned i = 1, e = NodeStack.size()-1; i != e; ++i) {
3175480093f4SDimitry Andric         unsigned NNonChainUses = 0;
3176480093f4SDimitry Andric         SDNode *NS = NodeStack[i].getNode();
3177480093f4SDimitry Andric         for (auto UI = NS->use_begin(), UE = NS->use_end(); UI != UE; ++UI)
3178480093f4SDimitry Andric           if (UI.getUse().getValueType() != MVT::Other)
3179480093f4SDimitry Andric             if (++NNonChainUses > 1) {
31800b57cec5SDimitry Andric               HasMultipleUses = true;
31810b57cec5SDimitry Andric               break;
31820b57cec5SDimitry Andric             }
31830b57cec5SDimitry Andric         if (HasMultipleUses) break;
3184480093f4SDimitry Andric       }
3185480093f4SDimitry Andric       if (HasMultipleUses) break;
31860b57cec5SDimitry Andric 
31870b57cec5SDimitry Andric       // Check to see that the target thinks this is profitable to fold and that
31880b57cec5SDimitry Andric       // we can fold it without inducing cycles in the graph.
31890b57cec5SDimitry Andric       if (!IsProfitableToFold(N, NodeStack[NodeStack.size()-2].getNode(),
31900b57cec5SDimitry Andric                               NodeToMatch) ||
31910b57cec5SDimitry Andric           !IsLegalToFold(N, NodeStack[NodeStack.size()-2].getNode(),
31920b57cec5SDimitry Andric                          NodeToMatch, OptLevel,
31930b57cec5SDimitry Andric                          true/*We validate our own chains*/))
31940b57cec5SDimitry Andric         break;
31950b57cec5SDimitry Andric 
31960b57cec5SDimitry Andric       continue;
31970b57cec5SDimitry Andric     }
3198fe6060f1SDimitry Andric     case OPC_EmitInteger:
3199fe6060f1SDimitry Andric     case OPC_EmitStringInteger: {
32000b57cec5SDimitry Andric       MVT::SimpleValueType VT =
32010b57cec5SDimitry Andric         (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
32020b57cec5SDimitry Andric       int64_t Val = MatcherTable[MatcherIndex++];
32030b57cec5SDimitry Andric       if (Val & 128)
32040b57cec5SDimitry Andric         Val = GetVBR(Val, MatcherTable, MatcherIndex);
3205fe6060f1SDimitry Andric       if (Opcode == OPC_EmitInteger)
3206fe6060f1SDimitry Andric         Val = decodeSignRotatedValue(Val);
32070b57cec5SDimitry Andric       RecordedNodes.push_back(std::pair<SDValue, SDNode*>(
32080b57cec5SDimitry Andric                               CurDAG->getTargetConstant(Val, SDLoc(NodeToMatch),
32090b57cec5SDimitry Andric                                                         VT), nullptr));
32100b57cec5SDimitry Andric       continue;
32110b57cec5SDimitry Andric     }
32120b57cec5SDimitry Andric     case OPC_EmitRegister: {
32130b57cec5SDimitry Andric       MVT::SimpleValueType VT =
32140b57cec5SDimitry Andric         (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
32150b57cec5SDimitry Andric       unsigned RegNo = MatcherTable[MatcherIndex++];
32160b57cec5SDimitry Andric       RecordedNodes.push_back(std::pair<SDValue, SDNode*>(
32170b57cec5SDimitry Andric                               CurDAG->getRegister(RegNo, VT), nullptr));
32180b57cec5SDimitry Andric       continue;
32190b57cec5SDimitry Andric     }
32200b57cec5SDimitry Andric     case OPC_EmitRegister2: {
32210b57cec5SDimitry Andric       // For targets w/ more than 256 register names, the register enum
32220b57cec5SDimitry Andric       // values are stored in two bytes in the matcher table (just like
32230b57cec5SDimitry Andric       // opcodes).
32240b57cec5SDimitry Andric       MVT::SimpleValueType VT =
32250b57cec5SDimitry Andric         (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
32260b57cec5SDimitry Andric       unsigned RegNo = MatcherTable[MatcherIndex++];
32270b57cec5SDimitry Andric       RegNo |= MatcherTable[MatcherIndex++] << 8;
32280b57cec5SDimitry Andric       RecordedNodes.push_back(std::pair<SDValue, SDNode*>(
32290b57cec5SDimitry Andric                               CurDAG->getRegister(RegNo, VT), nullptr));
32300b57cec5SDimitry Andric       continue;
32310b57cec5SDimitry Andric     }
32320b57cec5SDimitry Andric 
32330b57cec5SDimitry Andric     case OPC_EmitConvertToTarget:  {
32340b57cec5SDimitry Andric       // Convert from IMM/FPIMM to target version.
32350b57cec5SDimitry Andric       unsigned RecNo = MatcherTable[MatcherIndex++];
32360b57cec5SDimitry Andric       assert(RecNo < RecordedNodes.size() && "Invalid EmitConvertToTarget");
32370b57cec5SDimitry Andric       SDValue Imm = RecordedNodes[RecNo].first;
32380b57cec5SDimitry Andric 
32390b57cec5SDimitry Andric       if (Imm->getOpcode() == ISD::Constant) {
32400b57cec5SDimitry Andric         const ConstantInt *Val=cast<ConstantSDNode>(Imm)->getConstantIntValue();
32410b57cec5SDimitry Andric         Imm = CurDAG->getTargetConstant(*Val, SDLoc(NodeToMatch),
32420b57cec5SDimitry Andric                                         Imm.getValueType());
32430b57cec5SDimitry Andric       } else if (Imm->getOpcode() == ISD::ConstantFP) {
32440b57cec5SDimitry Andric         const ConstantFP *Val=cast<ConstantFPSDNode>(Imm)->getConstantFPValue();
32450b57cec5SDimitry Andric         Imm = CurDAG->getTargetConstantFP(*Val, SDLoc(NodeToMatch),
32460b57cec5SDimitry Andric                                           Imm.getValueType());
32470b57cec5SDimitry Andric       }
32480b57cec5SDimitry Andric 
32490b57cec5SDimitry Andric       RecordedNodes.push_back(std::make_pair(Imm, RecordedNodes[RecNo].second));
32500b57cec5SDimitry Andric       continue;
32510b57cec5SDimitry Andric     }
32520b57cec5SDimitry Andric 
32530b57cec5SDimitry Andric     case OPC_EmitMergeInputChains1_0:    // OPC_EmitMergeInputChains, 1, 0
32540b57cec5SDimitry Andric     case OPC_EmitMergeInputChains1_1:    // OPC_EmitMergeInputChains, 1, 1
32550b57cec5SDimitry Andric     case OPC_EmitMergeInputChains1_2: {  // OPC_EmitMergeInputChains, 1, 2
32560b57cec5SDimitry Andric       // These are space-optimized forms of OPC_EmitMergeInputChains.
32570b57cec5SDimitry Andric       assert(!InputChain.getNode() &&
32580b57cec5SDimitry Andric              "EmitMergeInputChains should be the first chain producing node");
32590b57cec5SDimitry Andric       assert(ChainNodesMatched.empty() &&
32600b57cec5SDimitry Andric              "Should only have one EmitMergeInputChains per match");
32610b57cec5SDimitry Andric 
32620b57cec5SDimitry Andric       // Read all of the chained nodes.
32630b57cec5SDimitry Andric       unsigned RecNo = Opcode - OPC_EmitMergeInputChains1_0;
32640b57cec5SDimitry Andric       assert(RecNo < RecordedNodes.size() && "Invalid EmitMergeInputChains");
32650b57cec5SDimitry Andric       ChainNodesMatched.push_back(RecordedNodes[RecNo].first.getNode());
32660b57cec5SDimitry Andric 
32670b57cec5SDimitry Andric       // FIXME: What if other value results of the node have uses not matched
32680b57cec5SDimitry Andric       // by this pattern?
32690b57cec5SDimitry Andric       if (ChainNodesMatched.back() != NodeToMatch &&
32700b57cec5SDimitry Andric           !RecordedNodes[RecNo].first.hasOneUse()) {
32710b57cec5SDimitry Andric         ChainNodesMatched.clear();
32720b57cec5SDimitry Andric         break;
32730b57cec5SDimitry Andric       }
32740b57cec5SDimitry Andric 
32750b57cec5SDimitry Andric       // Merge the input chains if they are not intra-pattern references.
32760b57cec5SDimitry Andric       InputChain = HandleMergeInputChains(ChainNodesMatched, CurDAG);
32770b57cec5SDimitry Andric 
32780b57cec5SDimitry Andric       if (!InputChain.getNode())
32790b57cec5SDimitry Andric         break;  // Failed to merge.
32800b57cec5SDimitry Andric       continue;
32810b57cec5SDimitry Andric     }
32820b57cec5SDimitry Andric 
32830b57cec5SDimitry Andric     case OPC_EmitMergeInputChains: {
32840b57cec5SDimitry Andric       assert(!InputChain.getNode() &&
32850b57cec5SDimitry Andric              "EmitMergeInputChains should be the first chain producing node");
32860b57cec5SDimitry Andric       // This node gets a list of nodes we matched in the input that have
32870b57cec5SDimitry Andric       // chains.  We want to token factor all of the input chains to these nodes
32880b57cec5SDimitry Andric       // together.  However, if any of the input chains is actually one of the
32890b57cec5SDimitry Andric       // nodes matched in this pattern, then we have an intra-match reference.
32900b57cec5SDimitry Andric       // Ignore these because the newly token factored chain should not refer to
32910b57cec5SDimitry Andric       // the old nodes.
32920b57cec5SDimitry Andric       unsigned NumChains = MatcherTable[MatcherIndex++];
32930b57cec5SDimitry Andric       assert(NumChains != 0 && "Can't TF zero chains");
32940b57cec5SDimitry Andric 
32950b57cec5SDimitry Andric       assert(ChainNodesMatched.empty() &&
32960b57cec5SDimitry Andric              "Should only have one EmitMergeInputChains per match");
32970b57cec5SDimitry Andric 
32980b57cec5SDimitry Andric       // Read all of the chained nodes.
32990b57cec5SDimitry Andric       for (unsigned i = 0; i != NumChains; ++i) {
33000b57cec5SDimitry Andric         unsigned RecNo = MatcherTable[MatcherIndex++];
33010b57cec5SDimitry Andric         assert(RecNo < RecordedNodes.size() && "Invalid EmitMergeInputChains");
33020b57cec5SDimitry Andric         ChainNodesMatched.push_back(RecordedNodes[RecNo].first.getNode());
33030b57cec5SDimitry Andric 
33040b57cec5SDimitry Andric         // FIXME: What if other value results of the node have uses not matched
33050b57cec5SDimitry Andric         // by this pattern?
33060b57cec5SDimitry Andric         if (ChainNodesMatched.back() != NodeToMatch &&
33070b57cec5SDimitry Andric             !RecordedNodes[RecNo].first.hasOneUse()) {
33080b57cec5SDimitry Andric           ChainNodesMatched.clear();
33090b57cec5SDimitry Andric           break;
33100b57cec5SDimitry Andric         }
33110b57cec5SDimitry Andric       }
33120b57cec5SDimitry Andric 
33130b57cec5SDimitry Andric       // If the inner loop broke out, the match fails.
33140b57cec5SDimitry Andric       if (ChainNodesMatched.empty())
33150b57cec5SDimitry Andric         break;
33160b57cec5SDimitry Andric 
33170b57cec5SDimitry Andric       // Merge the input chains if they are not intra-pattern references.
33180b57cec5SDimitry Andric       InputChain = HandleMergeInputChains(ChainNodesMatched, CurDAG);
33190b57cec5SDimitry Andric 
33200b57cec5SDimitry Andric       if (!InputChain.getNode())
33210b57cec5SDimitry Andric         break;  // Failed to merge.
33220b57cec5SDimitry Andric 
33230b57cec5SDimitry Andric       continue;
33240b57cec5SDimitry Andric     }
33250b57cec5SDimitry Andric 
33268bcb0991SDimitry Andric     case OPC_EmitCopyToReg:
33278bcb0991SDimitry Andric     case OPC_EmitCopyToReg2: {
33280b57cec5SDimitry Andric       unsigned RecNo = MatcherTable[MatcherIndex++];
33290b57cec5SDimitry Andric       assert(RecNo < RecordedNodes.size() && "Invalid EmitCopyToReg");
33300b57cec5SDimitry Andric       unsigned DestPhysReg = MatcherTable[MatcherIndex++];
33318bcb0991SDimitry Andric       if (Opcode == OPC_EmitCopyToReg2)
33328bcb0991SDimitry Andric         DestPhysReg |= MatcherTable[MatcherIndex++] << 8;
33330b57cec5SDimitry Andric 
33340b57cec5SDimitry Andric       if (!InputChain.getNode())
33350b57cec5SDimitry Andric         InputChain = CurDAG->getEntryNode();
33360b57cec5SDimitry Andric 
33370b57cec5SDimitry Andric       InputChain = CurDAG->getCopyToReg(InputChain, SDLoc(NodeToMatch),
33380b57cec5SDimitry Andric                                         DestPhysReg, RecordedNodes[RecNo].first,
33390b57cec5SDimitry Andric                                         InputGlue);
33400b57cec5SDimitry Andric 
33410b57cec5SDimitry Andric       InputGlue = InputChain.getValue(1);
33420b57cec5SDimitry Andric       continue;
33430b57cec5SDimitry Andric     }
33440b57cec5SDimitry Andric 
33450b57cec5SDimitry Andric     case OPC_EmitNodeXForm: {
33460b57cec5SDimitry Andric       unsigned XFormNo = MatcherTable[MatcherIndex++];
33470b57cec5SDimitry Andric       unsigned RecNo = MatcherTable[MatcherIndex++];
33480b57cec5SDimitry Andric       assert(RecNo < RecordedNodes.size() && "Invalid EmitNodeXForm");
33490b57cec5SDimitry Andric       SDValue Res = RunSDNodeXForm(RecordedNodes[RecNo].first, XFormNo);
33500b57cec5SDimitry Andric       RecordedNodes.push_back(std::pair<SDValue,SDNode*>(Res, nullptr));
33510b57cec5SDimitry Andric       continue;
33520b57cec5SDimitry Andric     }
33530b57cec5SDimitry Andric     case OPC_Coverage: {
33540b57cec5SDimitry Andric       // This is emitted right before MorphNode/EmitNode.
33550b57cec5SDimitry Andric       // So it should be safe to assume that this node has been selected
33560b57cec5SDimitry Andric       unsigned index = MatcherTable[MatcherIndex++];
33570b57cec5SDimitry Andric       index |= (MatcherTable[MatcherIndex++] << 8);
33580b57cec5SDimitry Andric       dbgs() << "COVERED: " << getPatternForIndex(index) << "\n";
33590b57cec5SDimitry Andric       dbgs() << "INCLUDED: " << getIncludePathForIndex(index) << "\n";
33600b57cec5SDimitry Andric       continue;
33610b57cec5SDimitry Andric     }
33620b57cec5SDimitry Andric 
33630b57cec5SDimitry Andric     case OPC_EmitNode:     case OPC_MorphNodeTo:
33640b57cec5SDimitry Andric     case OPC_EmitNode0:    case OPC_EmitNode1:    case OPC_EmitNode2:
33650b57cec5SDimitry Andric     case OPC_MorphNodeTo0: case OPC_MorphNodeTo1: case OPC_MorphNodeTo2: {
33660b57cec5SDimitry Andric       uint16_t TargetOpc = MatcherTable[MatcherIndex++];
33670b57cec5SDimitry Andric       TargetOpc |= (unsigned short)MatcherTable[MatcherIndex++] << 8;
33680b57cec5SDimitry Andric       unsigned EmitNodeInfo = MatcherTable[MatcherIndex++];
33690b57cec5SDimitry Andric       // Get the result VT list.
33700b57cec5SDimitry Andric       unsigned NumVTs;
33710b57cec5SDimitry Andric       // If this is one of the compressed forms, get the number of VTs based
33720b57cec5SDimitry Andric       // on the Opcode. Otherwise read the next byte from the table.
33730b57cec5SDimitry Andric       if (Opcode >= OPC_MorphNodeTo0 && Opcode <= OPC_MorphNodeTo2)
33740b57cec5SDimitry Andric         NumVTs = Opcode - OPC_MorphNodeTo0;
33750b57cec5SDimitry Andric       else if (Opcode >= OPC_EmitNode0 && Opcode <= OPC_EmitNode2)
33760b57cec5SDimitry Andric         NumVTs = Opcode - OPC_EmitNode0;
33770b57cec5SDimitry Andric       else
33780b57cec5SDimitry Andric         NumVTs = MatcherTable[MatcherIndex++];
33790b57cec5SDimitry Andric       SmallVector<EVT, 4> VTs;
33800b57cec5SDimitry Andric       for (unsigned i = 0; i != NumVTs; ++i) {
33810b57cec5SDimitry Andric         MVT::SimpleValueType VT =
33820b57cec5SDimitry Andric           (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
33830b57cec5SDimitry Andric         if (VT == MVT::iPTR)
33840b57cec5SDimitry Andric           VT = TLI->getPointerTy(CurDAG->getDataLayout()).SimpleTy;
33850b57cec5SDimitry Andric         VTs.push_back(VT);
33860b57cec5SDimitry Andric       }
33870b57cec5SDimitry Andric 
33880b57cec5SDimitry Andric       if (EmitNodeInfo & OPFL_Chain)
33890b57cec5SDimitry Andric         VTs.push_back(MVT::Other);
33900b57cec5SDimitry Andric       if (EmitNodeInfo & OPFL_GlueOutput)
33910b57cec5SDimitry Andric         VTs.push_back(MVT::Glue);
33920b57cec5SDimitry Andric 
33930b57cec5SDimitry Andric       // This is hot code, so optimize the two most common cases of 1 and 2
33940b57cec5SDimitry Andric       // results.
33950b57cec5SDimitry Andric       SDVTList VTList;
33960b57cec5SDimitry Andric       if (VTs.size() == 1)
33970b57cec5SDimitry Andric         VTList = CurDAG->getVTList(VTs[0]);
33980b57cec5SDimitry Andric       else if (VTs.size() == 2)
33990b57cec5SDimitry Andric         VTList = CurDAG->getVTList(VTs[0], VTs[1]);
34000b57cec5SDimitry Andric       else
34010b57cec5SDimitry Andric         VTList = CurDAG->getVTList(VTs);
34020b57cec5SDimitry Andric 
34030b57cec5SDimitry Andric       // Get the operand list.
34040b57cec5SDimitry Andric       unsigned NumOps = MatcherTable[MatcherIndex++];
34050b57cec5SDimitry Andric       SmallVector<SDValue, 8> Ops;
34060b57cec5SDimitry Andric       for (unsigned i = 0; i != NumOps; ++i) {
34070b57cec5SDimitry Andric         unsigned RecNo = MatcherTable[MatcherIndex++];
34080b57cec5SDimitry Andric         if (RecNo & 128)
34090b57cec5SDimitry Andric           RecNo = GetVBR(RecNo, MatcherTable, MatcherIndex);
34100b57cec5SDimitry Andric 
34110b57cec5SDimitry Andric         assert(RecNo < RecordedNodes.size() && "Invalid EmitNode");
34120b57cec5SDimitry Andric         Ops.push_back(RecordedNodes[RecNo].first);
34130b57cec5SDimitry Andric       }
34140b57cec5SDimitry Andric 
34150b57cec5SDimitry Andric       // If there are variadic operands to add, handle them now.
34160b57cec5SDimitry Andric       if (EmitNodeInfo & OPFL_VariadicInfo) {
34170b57cec5SDimitry Andric         // Determine the start index to copy from.
34180b57cec5SDimitry Andric         unsigned FirstOpToCopy = getNumFixedFromVariadicInfo(EmitNodeInfo);
34190b57cec5SDimitry Andric         FirstOpToCopy += (EmitNodeInfo & OPFL_Chain) ? 1 : 0;
34200b57cec5SDimitry Andric         assert(NodeToMatch->getNumOperands() >= FirstOpToCopy &&
34210b57cec5SDimitry Andric                "Invalid variadic node");
34220b57cec5SDimitry Andric         // Copy all of the variadic operands, not including a potential glue
34230b57cec5SDimitry Andric         // input.
34240b57cec5SDimitry Andric         for (unsigned i = FirstOpToCopy, e = NodeToMatch->getNumOperands();
34250b57cec5SDimitry Andric              i != e; ++i) {
34260b57cec5SDimitry Andric           SDValue V = NodeToMatch->getOperand(i);
34270b57cec5SDimitry Andric           if (V.getValueType() == MVT::Glue) break;
34280b57cec5SDimitry Andric           Ops.push_back(V);
34290b57cec5SDimitry Andric         }
34300b57cec5SDimitry Andric       }
34310b57cec5SDimitry Andric 
34320b57cec5SDimitry Andric       // If this has chain/glue inputs, add them.
34330b57cec5SDimitry Andric       if (EmitNodeInfo & OPFL_Chain)
34340b57cec5SDimitry Andric         Ops.push_back(InputChain);
34350b57cec5SDimitry Andric       if ((EmitNodeInfo & OPFL_GlueInput) && InputGlue.getNode() != nullptr)
34360b57cec5SDimitry Andric         Ops.push_back(InputGlue);
34370b57cec5SDimitry Andric 
3438480093f4SDimitry Andric       // Check whether any matched node could raise an FP exception.  Since all
3439480093f4SDimitry Andric       // such nodes must have a chain, it suffices to check ChainNodesMatched.
3440480093f4SDimitry Andric       // We need to perform this check before potentially modifying one of the
3441480093f4SDimitry Andric       // nodes via MorphNode.
3442480093f4SDimitry Andric       bool MayRaiseFPException = false;
3443480093f4SDimitry Andric       for (auto *N : ChainNodesMatched)
3444480093f4SDimitry Andric         if (mayRaiseFPException(N) && !N->getFlags().hasNoFPExcept()) {
3445480093f4SDimitry Andric           MayRaiseFPException = true;
3446480093f4SDimitry Andric           break;
3447480093f4SDimitry Andric         }
3448480093f4SDimitry Andric 
34490b57cec5SDimitry Andric       // Create the node.
34500b57cec5SDimitry Andric       MachineSDNode *Res = nullptr;
34510b57cec5SDimitry Andric       bool IsMorphNodeTo = Opcode == OPC_MorphNodeTo ||
34520b57cec5SDimitry Andric                      (Opcode >= OPC_MorphNodeTo0 && Opcode <= OPC_MorphNodeTo2);
34530b57cec5SDimitry Andric       if (!IsMorphNodeTo) {
34540b57cec5SDimitry Andric         // If this is a normal EmitNode command, just create the new node and
34550b57cec5SDimitry Andric         // add the results to the RecordedNodes list.
34560b57cec5SDimitry Andric         Res = CurDAG->getMachineNode(TargetOpc, SDLoc(NodeToMatch),
34570b57cec5SDimitry Andric                                      VTList, Ops);
34580b57cec5SDimitry Andric 
34590b57cec5SDimitry Andric         // Add all the non-glue/non-chain results to the RecordedNodes list.
34600b57cec5SDimitry Andric         for (unsigned i = 0, e = VTs.size(); i != e; ++i) {
34610b57cec5SDimitry Andric           if (VTs[i] == MVT::Other || VTs[i] == MVT::Glue) break;
34620b57cec5SDimitry Andric           RecordedNodes.push_back(std::pair<SDValue,SDNode*>(SDValue(Res, i),
34630b57cec5SDimitry Andric                                                              nullptr));
34640b57cec5SDimitry Andric         }
34650b57cec5SDimitry Andric       } else {
34660b57cec5SDimitry Andric         assert(NodeToMatch->getOpcode() != ISD::DELETED_NODE &&
34670b57cec5SDimitry Andric                "NodeToMatch was removed partway through selection");
34680b57cec5SDimitry Andric         SelectionDAG::DAGNodeDeletedListener NDL(*CurDAG, [&](SDNode *N,
34690b57cec5SDimitry Andric                                                               SDNode *E) {
34700b57cec5SDimitry Andric           CurDAG->salvageDebugInfo(*N);
34710b57cec5SDimitry Andric           auto &Chain = ChainNodesMatched;
34720b57cec5SDimitry Andric           assert((!E || !is_contained(Chain, N)) &&
34730b57cec5SDimitry Andric                  "Chain node replaced during MorphNode");
3474e8d8bef9SDimitry Andric           llvm::erase_value(Chain, N);
34750b57cec5SDimitry Andric         });
34760b57cec5SDimitry Andric         Res = cast<MachineSDNode>(MorphNode(NodeToMatch, TargetOpc, VTList,
34770b57cec5SDimitry Andric                                             Ops, EmitNodeInfo));
34780b57cec5SDimitry Andric       }
34790b57cec5SDimitry Andric 
3480480093f4SDimitry Andric       // Set the NoFPExcept flag when no original matched node could
3481480093f4SDimitry Andric       // raise an FP exception, but the new node potentially might.
3482480093f4SDimitry Andric       if (!MayRaiseFPException && mayRaiseFPException(Res)) {
3483480093f4SDimitry Andric         SDNodeFlags Flags = Res->getFlags();
3484480093f4SDimitry Andric         Flags.setNoFPExcept(true);
3485480093f4SDimitry Andric         Res->setFlags(Flags);
3486480093f4SDimitry Andric       }
3487480093f4SDimitry Andric 
34880b57cec5SDimitry Andric       // If the node had chain/glue results, update our notion of the current
34890b57cec5SDimitry Andric       // chain and glue.
34900b57cec5SDimitry Andric       if (EmitNodeInfo & OPFL_GlueOutput) {
34910b57cec5SDimitry Andric         InputGlue = SDValue(Res, VTs.size()-1);
34920b57cec5SDimitry Andric         if (EmitNodeInfo & OPFL_Chain)
34930b57cec5SDimitry Andric           InputChain = SDValue(Res, VTs.size()-2);
34940b57cec5SDimitry Andric       } else if (EmitNodeInfo & OPFL_Chain)
34950b57cec5SDimitry Andric         InputChain = SDValue(Res, VTs.size()-1);
34960b57cec5SDimitry Andric 
34970b57cec5SDimitry Andric       // If the OPFL_MemRefs glue is set on this node, slap all of the
34980b57cec5SDimitry Andric       // accumulated memrefs onto it.
34990b57cec5SDimitry Andric       //
35000b57cec5SDimitry Andric       // FIXME: This is vastly incorrect for patterns with multiple outputs
35010b57cec5SDimitry Andric       // instructions that access memory and for ComplexPatterns that match
35020b57cec5SDimitry Andric       // loads.
35030b57cec5SDimitry Andric       if (EmitNodeInfo & OPFL_MemRefs) {
35040b57cec5SDimitry Andric         // Only attach load or store memory operands if the generated
35050b57cec5SDimitry Andric         // instruction may load or store.
35060b57cec5SDimitry Andric         const MCInstrDesc &MCID = TII->get(TargetOpc);
35070b57cec5SDimitry Andric         bool mayLoad = MCID.mayLoad();
35080b57cec5SDimitry Andric         bool mayStore = MCID.mayStore();
35090b57cec5SDimitry Andric 
35100b57cec5SDimitry Andric         // We expect to have relatively few of these so just filter them into a
35110b57cec5SDimitry Andric         // temporary buffer so that we can easily add them to the instruction.
35120b57cec5SDimitry Andric         SmallVector<MachineMemOperand *, 4> FilteredMemRefs;
35130b57cec5SDimitry Andric         for (MachineMemOperand *MMO : MatchedMemRefs) {
35140b57cec5SDimitry Andric           if (MMO->isLoad()) {
35150b57cec5SDimitry Andric             if (mayLoad)
35160b57cec5SDimitry Andric               FilteredMemRefs.push_back(MMO);
35170b57cec5SDimitry Andric           } else if (MMO->isStore()) {
35180b57cec5SDimitry Andric             if (mayStore)
35190b57cec5SDimitry Andric               FilteredMemRefs.push_back(MMO);
35200b57cec5SDimitry Andric           } else {
35210b57cec5SDimitry Andric             FilteredMemRefs.push_back(MMO);
35220b57cec5SDimitry Andric           }
35230b57cec5SDimitry Andric         }
35240b57cec5SDimitry Andric 
35250b57cec5SDimitry Andric         CurDAG->setNodeMemRefs(Res, FilteredMemRefs);
35260b57cec5SDimitry Andric       }
35270b57cec5SDimitry Andric 
35280b57cec5SDimitry Andric       LLVM_DEBUG(if (!MatchedMemRefs.empty() && Res->memoperands_empty()) dbgs()
35290b57cec5SDimitry Andric                      << "  Dropping mem operands\n";
35300b57cec5SDimitry Andric                  dbgs() << "  " << (IsMorphNodeTo ? "Morphed" : "Created")
35310b57cec5SDimitry Andric                         << " node: ";
35320b57cec5SDimitry Andric                  Res->dump(CurDAG););
35330b57cec5SDimitry Andric 
35340b57cec5SDimitry Andric       // If this was a MorphNodeTo then we're completely done!
35350b57cec5SDimitry Andric       if (IsMorphNodeTo) {
35360b57cec5SDimitry Andric         // Update chain uses.
35370b57cec5SDimitry Andric         UpdateChains(Res, InputChain, ChainNodesMatched, true);
35380b57cec5SDimitry Andric         return;
35390b57cec5SDimitry Andric       }
35400b57cec5SDimitry Andric       continue;
35410b57cec5SDimitry Andric     }
35420b57cec5SDimitry Andric 
35430b57cec5SDimitry Andric     case OPC_CompleteMatch: {
35440b57cec5SDimitry Andric       // The match has been completed, and any new nodes (if any) have been
35450b57cec5SDimitry Andric       // created.  Patch up references to the matched dag to use the newly
35460b57cec5SDimitry Andric       // created nodes.
35470b57cec5SDimitry Andric       unsigned NumResults = MatcherTable[MatcherIndex++];
35480b57cec5SDimitry Andric 
35490b57cec5SDimitry Andric       for (unsigned i = 0; i != NumResults; ++i) {
35500b57cec5SDimitry Andric         unsigned ResSlot = MatcherTable[MatcherIndex++];
35510b57cec5SDimitry Andric         if (ResSlot & 128)
35520b57cec5SDimitry Andric           ResSlot = GetVBR(ResSlot, MatcherTable, MatcherIndex);
35530b57cec5SDimitry Andric 
35540b57cec5SDimitry Andric         assert(ResSlot < RecordedNodes.size() && "Invalid CompleteMatch");
35550b57cec5SDimitry Andric         SDValue Res = RecordedNodes[ResSlot].first;
35560b57cec5SDimitry Andric 
35570b57cec5SDimitry Andric         assert(i < NodeToMatch->getNumValues() &&
35580b57cec5SDimitry Andric                NodeToMatch->getValueType(i) != MVT::Other &&
35590b57cec5SDimitry Andric                NodeToMatch->getValueType(i) != MVT::Glue &&
35600b57cec5SDimitry Andric                "Invalid number of results to complete!");
35610b57cec5SDimitry Andric         assert((NodeToMatch->getValueType(i) == Res.getValueType() ||
35620b57cec5SDimitry Andric                 NodeToMatch->getValueType(i) == MVT::iPTR ||
35630b57cec5SDimitry Andric                 Res.getValueType() == MVT::iPTR ||
35640b57cec5SDimitry Andric                 NodeToMatch->getValueType(i).getSizeInBits() ==
35650b57cec5SDimitry Andric                     Res.getValueSizeInBits()) &&
35660b57cec5SDimitry Andric                "invalid replacement");
35670b57cec5SDimitry Andric         ReplaceUses(SDValue(NodeToMatch, i), Res);
35680b57cec5SDimitry Andric       }
35690b57cec5SDimitry Andric 
35700b57cec5SDimitry Andric       // Update chain uses.
35710b57cec5SDimitry Andric       UpdateChains(NodeToMatch, InputChain, ChainNodesMatched, false);
35720b57cec5SDimitry Andric 
35730b57cec5SDimitry Andric       // If the root node defines glue, we need to update it to the glue result.
35740b57cec5SDimitry Andric       // TODO: This never happens in our tests and I think it can be removed /
35750b57cec5SDimitry Andric       // replaced with an assert, but if we do it this the way the change is
35760b57cec5SDimitry Andric       // NFC.
35770b57cec5SDimitry Andric       if (NodeToMatch->getValueType(NodeToMatch->getNumValues() - 1) ==
35780b57cec5SDimitry Andric               MVT::Glue &&
35790b57cec5SDimitry Andric           InputGlue.getNode())
35800b57cec5SDimitry Andric         ReplaceUses(SDValue(NodeToMatch, NodeToMatch->getNumValues() - 1),
35810b57cec5SDimitry Andric                     InputGlue);
35820b57cec5SDimitry Andric 
35830b57cec5SDimitry Andric       assert(NodeToMatch->use_empty() &&
35840b57cec5SDimitry Andric              "Didn't replace all uses of the node?");
35850b57cec5SDimitry Andric       CurDAG->RemoveDeadNode(NodeToMatch);
35860b57cec5SDimitry Andric 
35870b57cec5SDimitry Andric       return;
35880b57cec5SDimitry Andric     }
35890b57cec5SDimitry Andric     }
35900b57cec5SDimitry Andric 
35910b57cec5SDimitry Andric     // If the code reached this point, then the match failed.  See if there is
35920b57cec5SDimitry Andric     // another child to try in the current 'Scope', otherwise pop it until we
35930b57cec5SDimitry Andric     // find a case to check.
35940b57cec5SDimitry Andric     LLVM_DEBUG(dbgs() << "  Match failed at index " << CurrentOpcodeIndex
35950b57cec5SDimitry Andric                       << "\n");
35960b57cec5SDimitry Andric     ++NumDAGIselRetries;
35970b57cec5SDimitry Andric     while (true) {
35980b57cec5SDimitry Andric       if (MatchScopes.empty()) {
35990b57cec5SDimitry Andric         CannotYetSelect(NodeToMatch);
36000b57cec5SDimitry Andric         return;
36010b57cec5SDimitry Andric       }
36020b57cec5SDimitry Andric 
36030b57cec5SDimitry Andric       // Restore the interpreter state back to the point where the scope was
36040b57cec5SDimitry Andric       // formed.
36050b57cec5SDimitry Andric       MatchScope &LastScope = MatchScopes.back();
36060b57cec5SDimitry Andric       RecordedNodes.resize(LastScope.NumRecordedNodes);
36070b57cec5SDimitry Andric       NodeStack.clear();
36080b57cec5SDimitry Andric       NodeStack.append(LastScope.NodeStack.begin(), LastScope.NodeStack.end());
36090b57cec5SDimitry Andric       N = NodeStack.back();
36100b57cec5SDimitry Andric 
36110b57cec5SDimitry Andric       if (LastScope.NumMatchedMemRefs != MatchedMemRefs.size())
36120b57cec5SDimitry Andric         MatchedMemRefs.resize(LastScope.NumMatchedMemRefs);
36130b57cec5SDimitry Andric       MatcherIndex = LastScope.FailIndex;
36140b57cec5SDimitry Andric 
36150b57cec5SDimitry Andric       LLVM_DEBUG(dbgs() << "  Continuing at " << MatcherIndex << "\n");
36160b57cec5SDimitry Andric 
36170b57cec5SDimitry Andric       InputChain = LastScope.InputChain;
36180b57cec5SDimitry Andric       InputGlue = LastScope.InputGlue;
36190b57cec5SDimitry Andric       if (!LastScope.HasChainNodesMatched)
36200b57cec5SDimitry Andric         ChainNodesMatched.clear();
36210b57cec5SDimitry Andric 
36220b57cec5SDimitry Andric       // Check to see what the offset is at the new MatcherIndex.  If it is zero
36230b57cec5SDimitry Andric       // we have reached the end of this scope, otherwise we have another child
36240b57cec5SDimitry Andric       // in the current scope to try.
36250b57cec5SDimitry Andric       unsigned NumToSkip = MatcherTable[MatcherIndex++];
36260b57cec5SDimitry Andric       if (NumToSkip & 128)
36270b57cec5SDimitry Andric         NumToSkip = GetVBR(NumToSkip, MatcherTable, MatcherIndex);
36280b57cec5SDimitry Andric 
36290b57cec5SDimitry Andric       // If we have another child in this scope to match, update FailIndex and
36300b57cec5SDimitry Andric       // try it.
36310b57cec5SDimitry Andric       if (NumToSkip != 0) {
36320b57cec5SDimitry Andric         LastScope.FailIndex = MatcherIndex+NumToSkip;
36330b57cec5SDimitry Andric         break;
36340b57cec5SDimitry Andric       }
36350b57cec5SDimitry Andric 
36360b57cec5SDimitry Andric       // End of this scope, pop it and try the next child in the containing
36370b57cec5SDimitry Andric       // scope.
36380b57cec5SDimitry Andric       MatchScopes.pop_back();
36390b57cec5SDimitry Andric     }
36400b57cec5SDimitry Andric   }
36410b57cec5SDimitry Andric }
36420b57cec5SDimitry Andric 
3643480093f4SDimitry Andric /// Return whether the node may raise an FP exception.
3644480093f4SDimitry Andric bool SelectionDAGISel::mayRaiseFPException(SDNode *N) const {
3645480093f4SDimitry Andric   // For machine opcodes, consult the MCID flag.
3646480093f4SDimitry Andric   if (N->isMachineOpcode()) {
3647480093f4SDimitry Andric     const MCInstrDesc &MCID = TII->get(N->getMachineOpcode());
3648480093f4SDimitry Andric     return MCID.mayRaiseFPException();
3649480093f4SDimitry Andric   }
3650480093f4SDimitry Andric 
3651480093f4SDimitry Andric   // For ISD opcodes, only StrictFP opcodes may raise an FP
3652480093f4SDimitry Andric   // exception.
3653480093f4SDimitry Andric   if (N->isTargetOpcode())
3654480093f4SDimitry Andric     return N->isTargetStrictFPOpcode();
3655480093f4SDimitry Andric   return N->isStrictFPOpcode();
3656480093f4SDimitry Andric }
3657480093f4SDimitry Andric 
36580b57cec5SDimitry Andric bool SelectionDAGISel::isOrEquivalentToAdd(const SDNode *N) const {
36590b57cec5SDimitry Andric   assert(N->getOpcode() == ISD::OR && "Unexpected opcode");
36600b57cec5SDimitry Andric   auto *C = dyn_cast<ConstantSDNode>(N->getOperand(1));
36610b57cec5SDimitry Andric   if (!C)
36620b57cec5SDimitry Andric     return false;
36630b57cec5SDimitry Andric 
36640b57cec5SDimitry Andric   // Detect when "or" is used to add an offset to a stack object.
36650b57cec5SDimitry Andric   if (auto *FN = dyn_cast<FrameIndexSDNode>(N->getOperand(0))) {
36660b57cec5SDimitry Andric     MachineFrameInfo &MFI = MF->getFrameInfo();
36675ffd83dbSDimitry Andric     Align A = MFI.getObjectAlign(FN->getIndex());
36680b57cec5SDimitry Andric     int32_t Off = C->getSExtValue();
36690b57cec5SDimitry Andric     // If the alleged offset fits in the zero bits guaranteed by
36700b57cec5SDimitry Andric     // the alignment, then this or is really an add.
36715ffd83dbSDimitry Andric     return (Off >= 0) && (((A.value() - 1) & Off) == unsigned(Off));
36720b57cec5SDimitry Andric   }
36730b57cec5SDimitry Andric   return false;
36740b57cec5SDimitry Andric }
36750b57cec5SDimitry Andric 
36760b57cec5SDimitry Andric void SelectionDAGISel::CannotYetSelect(SDNode *N) {
36770b57cec5SDimitry Andric   std::string msg;
36780b57cec5SDimitry Andric   raw_string_ostream Msg(msg);
36790b57cec5SDimitry Andric   Msg << "Cannot select: ";
36800b57cec5SDimitry Andric 
36810b57cec5SDimitry Andric   if (N->getOpcode() != ISD::INTRINSIC_W_CHAIN &&
36820b57cec5SDimitry Andric       N->getOpcode() != ISD::INTRINSIC_WO_CHAIN &&
36830b57cec5SDimitry Andric       N->getOpcode() != ISD::INTRINSIC_VOID) {
36840b57cec5SDimitry Andric     N->printrFull(Msg, CurDAG);
36850b57cec5SDimitry Andric     Msg << "\nIn function: " << MF->getName();
36860b57cec5SDimitry Andric   } else {
36870b57cec5SDimitry Andric     bool HasInputChain = N->getOperand(0).getValueType() == MVT::Other;
36880b57cec5SDimitry Andric     unsigned iid =
36890b57cec5SDimitry Andric       cast<ConstantSDNode>(N->getOperand(HasInputChain))->getZExtValue();
36900b57cec5SDimitry Andric     if (iid < Intrinsic::num_intrinsics)
3691fe6060f1SDimitry Andric       Msg << "intrinsic %" << Intrinsic::getBaseName((Intrinsic::ID)iid);
36920b57cec5SDimitry Andric     else if (const TargetIntrinsicInfo *TII = TM.getIntrinsicInfo())
36930b57cec5SDimitry Andric       Msg << "target intrinsic %" << TII->getName(iid);
36940b57cec5SDimitry Andric     else
36950b57cec5SDimitry Andric       Msg << "unknown intrinsic #" << iid;
36960b57cec5SDimitry Andric   }
3697349cc55cSDimitry Andric   report_fatal_error(Twine(Msg.str()));
36980b57cec5SDimitry Andric }
36990b57cec5SDimitry Andric 
37000b57cec5SDimitry Andric char SelectionDAGISel::ID = 0;
3701