xref: /freebsd/contrib/llvm-project/llvm/lib/CodeGen/SelectionDAG/SelectionDAGISel.cpp (revision 5f757f3ff9144b609b3c433dfd370cc6bdc191ad)
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/PostOrderIterator.h"
190b57cec5SDimitry Andric #include "llvm/ADT/STLExtras.h"
200b57cec5SDimitry Andric #include "llvm/ADT/SmallPtrSet.h"
210b57cec5SDimitry Andric #include "llvm/ADT/SmallVector.h"
220b57cec5SDimitry Andric #include "llvm/ADT/Statistic.h"
230b57cec5SDimitry Andric #include "llvm/ADT/StringRef.h"
240b57cec5SDimitry Andric #include "llvm/Analysis/AliasAnalysis.h"
25bdd1243dSDimitry Andric #include "llvm/Analysis/AssumptionCache.h"
260b57cec5SDimitry Andric #include "llvm/Analysis/BranchProbabilityInfo.h"
270b57cec5SDimitry Andric #include "llvm/Analysis/CFG.h"
28480093f4SDimitry Andric #include "llvm/Analysis/LazyBlockFrequencyInfo.h"
290b57cec5SDimitry Andric #include "llvm/Analysis/OptimizationRemarkEmitter.h"
30480093f4SDimitry Andric #include "llvm/Analysis/ProfileSummaryInfo.h"
310b57cec5SDimitry Andric #include "llvm/Analysis/TargetLibraryInfo.h"
320b57cec5SDimitry Andric #include "llvm/Analysis/TargetTransformInfo.h"
3306c3fb27SDimitry Andric #include "llvm/Analysis/UniformityAnalysis.h"
34bdd1243dSDimitry Andric #include "llvm/CodeGen/AssignmentTrackingAnalysis.h"
35349cc55cSDimitry Andric #include "llvm/CodeGen/CodeGenCommonISel.h"
360b57cec5SDimitry Andric #include "llvm/CodeGen/FastISel.h"
370b57cec5SDimitry Andric #include "llvm/CodeGen/FunctionLoweringInfo.h"
380b57cec5SDimitry Andric #include "llvm/CodeGen/GCMetadata.h"
390b57cec5SDimitry Andric #include "llvm/CodeGen/ISDOpcodes.h"
400b57cec5SDimitry Andric #include "llvm/CodeGen/MachineBasicBlock.h"
410b57cec5SDimitry Andric #include "llvm/CodeGen/MachineFrameInfo.h"
420b57cec5SDimitry Andric #include "llvm/CodeGen/MachineFunction.h"
430b57cec5SDimitry Andric #include "llvm/CodeGen/MachineFunctionPass.h"
440b57cec5SDimitry Andric #include "llvm/CodeGen/MachineInstr.h"
450b57cec5SDimitry Andric #include "llvm/CodeGen/MachineInstrBuilder.h"
460b57cec5SDimitry Andric #include "llvm/CodeGen/MachineMemOperand.h"
470b57cec5SDimitry Andric #include "llvm/CodeGen/MachineModuleInfo.h"
480b57cec5SDimitry Andric #include "llvm/CodeGen/MachineOperand.h"
490b57cec5SDimitry Andric #include "llvm/CodeGen/MachinePassRegistry.h"
500b57cec5SDimitry Andric #include "llvm/CodeGen/MachineRegisterInfo.h"
5106c3fb27SDimitry Andric #include "llvm/CodeGen/MachineValueType.h"
520b57cec5SDimitry Andric #include "llvm/CodeGen/SchedulerRegistry.h"
530b57cec5SDimitry Andric #include "llvm/CodeGen/SelectionDAG.h"
540b57cec5SDimitry Andric #include "llvm/CodeGen/SelectionDAGNodes.h"
55753f127fSDimitry Andric #include "llvm/CodeGen/StackMaps.h"
560b57cec5SDimitry Andric #include "llvm/CodeGen/StackProtector.h"
570b57cec5SDimitry Andric #include "llvm/CodeGen/SwiftErrorValueTracking.h"
580b57cec5SDimitry Andric #include "llvm/CodeGen/TargetInstrInfo.h"
590b57cec5SDimitry Andric #include "llvm/CodeGen/TargetLowering.h"
600b57cec5SDimitry Andric #include "llvm/CodeGen/TargetRegisterInfo.h"
610b57cec5SDimitry Andric #include "llvm/CodeGen/TargetSubtargetInfo.h"
620b57cec5SDimitry Andric #include "llvm/CodeGen/ValueTypes.h"
6306c3fb27SDimitry Andric #include "llvm/CodeGen/WinEHFuncInfo.h"
640b57cec5SDimitry Andric #include "llvm/IR/BasicBlock.h"
650b57cec5SDimitry Andric #include "llvm/IR/Constants.h"
660b57cec5SDimitry Andric #include "llvm/IR/DataLayout.h"
67bdd1243dSDimitry Andric #include "llvm/IR/DebugInfo.h"
680b57cec5SDimitry Andric #include "llvm/IR/DebugInfoMetadata.h"
690b57cec5SDimitry Andric #include "llvm/IR/DebugLoc.h"
700b57cec5SDimitry Andric #include "llvm/IR/DiagnosticInfo.h"
7106c3fb27SDimitry Andric #include "llvm/IR/EHPersonalities.h"
720b57cec5SDimitry Andric #include "llvm/IR/Function.h"
730b57cec5SDimitry Andric #include "llvm/IR/InlineAsm.h"
740b57cec5SDimitry Andric #include "llvm/IR/InstIterator.h"
750b57cec5SDimitry Andric #include "llvm/IR/Instruction.h"
760b57cec5SDimitry Andric #include "llvm/IR/Instructions.h"
770b57cec5SDimitry Andric #include "llvm/IR/IntrinsicInst.h"
780b57cec5SDimitry Andric #include "llvm/IR/Intrinsics.h"
79480093f4SDimitry Andric #include "llvm/IR/IntrinsicsWebAssembly.h"
800b57cec5SDimitry Andric #include "llvm/IR/Metadata.h"
81*5f757f3fSDimitry Andric #include "llvm/IR/PrintPasses.h"
82e8d8bef9SDimitry Andric #include "llvm/IR/Statepoint.h"
830b57cec5SDimitry Andric #include "llvm/IR/Type.h"
840b57cec5SDimitry Andric #include "llvm/IR/User.h"
850b57cec5SDimitry Andric #include "llvm/IR/Value.h"
86480093f4SDimitry Andric #include "llvm/InitializePasses.h"
870b57cec5SDimitry Andric #include "llvm/MC/MCInstrDesc.h"
880b57cec5SDimitry Andric #include "llvm/Pass.h"
890b57cec5SDimitry Andric #include "llvm/Support/BranchProbability.h"
900b57cec5SDimitry Andric #include "llvm/Support/Casting.h"
910b57cec5SDimitry Andric #include "llvm/Support/CodeGen.h"
920b57cec5SDimitry Andric #include "llvm/Support/CommandLine.h"
930b57cec5SDimitry Andric #include "llvm/Support/Compiler.h"
940b57cec5SDimitry Andric #include "llvm/Support/Debug.h"
950b57cec5SDimitry Andric #include "llvm/Support/ErrorHandling.h"
960b57cec5SDimitry Andric #include "llvm/Support/KnownBits.h"
970b57cec5SDimitry Andric #include "llvm/Support/Timer.h"
980b57cec5SDimitry Andric #include "llvm/Support/raw_ostream.h"
990b57cec5SDimitry Andric #include "llvm/Target/TargetIntrinsicInfo.h"
1000b57cec5SDimitry Andric #include "llvm/Target/TargetMachine.h"
1010b57cec5SDimitry Andric #include "llvm/Target/TargetOptions.h"
1020b57cec5SDimitry Andric #include "llvm/Transforms/Utils/BasicBlockUtils.h"
1030b57cec5SDimitry Andric #include <algorithm>
1040b57cec5SDimitry Andric #include <cassert>
1050b57cec5SDimitry Andric #include <cstdint>
1060b57cec5SDimitry Andric #include <iterator>
1070b57cec5SDimitry Andric #include <limits>
1080b57cec5SDimitry Andric #include <memory>
109bdd1243dSDimitry Andric #include <optional>
1100b57cec5SDimitry Andric #include <string>
1110b57cec5SDimitry Andric #include <utility>
1120b57cec5SDimitry Andric #include <vector>
1130b57cec5SDimitry Andric 
1140b57cec5SDimitry Andric using namespace llvm;
1150b57cec5SDimitry Andric 
1160b57cec5SDimitry Andric #define DEBUG_TYPE "isel"
117*5f757f3fSDimitry Andric #define ISEL_DUMP_DEBUG_TYPE DEBUG_TYPE "-dump"
1180b57cec5SDimitry Andric 
1190b57cec5SDimitry Andric STATISTIC(NumFastIselFailures, "Number of instructions fast isel failed on");
1200b57cec5SDimitry Andric STATISTIC(NumFastIselSuccess, "Number of instructions fast isel selected");
1210b57cec5SDimitry Andric STATISTIC(NumFastIselBlocks, "Number of blocks selected entirely by fast isel");
1220b57cec5SDimitry Andric STATISTIC(NumDAGBlocks, "Number of blocks selected using DAG");
1230b57cec5SDimitry Andric STATISTIC(NumDAGIselRetries,"Number of times dag isel has to try another path");
1240b57cec5SDimitry Andric STATISTIC(NumEntryBlocks, "Number of entry blocks encountered");
1250b57cec5SDimitry Andric STATISTIC(NumFastIselFailLowerArguments,
1260b57cec5SDimitry Andric           "Number of entry blocks where fast isel failed to lower arguments");
1270b57cec5SDimitry Andric 
1280b57cec5SDimitry Andric static cl::opt<int> EnableFastISelAbort(
1290b57cec5SDimitry Andric     "fast-isel-abort", cl::Hidden,
1300b57cec5SDimitry Andric     cl::desc("Enable abort calls when \"fast\" instruction selection "
1310b57cec5SDimitry Andric              "fails to lower an instruction: 0 disable the abort, 1 will "
1320b57cec5SDimitry Andric              "abort but for args, calls and terminators, 2 will also "
1330b57cec5SDimitry Andric              "abort for argument lowering, and 3 will never fallback "
1340b57cec5SDimitry Andric              "to SelectionDAG."));
1350b57cec5SDimitry Andric 
1360b57cec5SDimitry Andric static cl::opt<bool> EnableFastISelFallbackReport(
1370b57cec5SDimitry Andric     "fast-isel-report-on-fallback", cl::Hidden,
1380b57cec5SDimitry Andric     cl::desc("Emit a diagnostic when \"fast\" instruction selection "
1390b57cec5SDimitry Andric              "falls back to SelectionDAG."));
1400b57cec5SDimitry Andric 
1410b57cec5SDimitry Andric static cl::opt<bool>
1420b57cec5SDimitry Andric UseMBPI("use-mbpi",
1430b57cec5SDimitry Andric         cl::desc("use Machine Branch Probability Info"),
1440b57cec5SDimitry Andric         cl::init(true), cl::Hidden);
1450b57cec5SDimitry Andric 
1460b57cec5SDimitry Andric #ifndef NDEBUG
1470b57cec5SDimitry Andric static cl::opt<std::string>
1480b57cec5SDimitry Andric FilterDAGBasicBlockName("filter-view-dags", cl::Hidden,
1490b57cec5SDimitry Andric                         cl::desc("Only display the basic block whose name "
1500b57cec5SDimitry Andric                                  "matches this for all view-*-dags options"));
1510b57cec5SDimitry Andric static cl::opt<bool>
1520b57cec5SDimitry Andric ViewDAGCombine1("view-dag-combine1-dags", cl::Hidden,
1530b57cec5SDimitry Andric           cl::desc("Pop up a window to show dags before the first "
1540b57cec5SDimitry Andric                    "dag combine pass"));
1550b57cec5SDimitry Andric static cl::opt<bool>
1560b57cec5SDimitry Andric ViewLegalizeTypesDAGs("view-legalize-types-dags", cl::Hidden,
1570b57cec5SDimitry Andric           cl::desc("Pop up a window to show dags before legalize types"));
1580b57cec5SDimitry Andric static cl::opt<bool>
159480093f4SDimitry Andric     ViewDAGCombineLT("view-dag-combine-lt-dags", cl::Hidden,
160480093f4SDimitry Andric                      cl::desc("Pop up a window to show dags before the post "
161480093f4SDimitry Andric                               "legalize types dag combine pass"));
162480093f4SDimitry Andric static cl::opt<bool>
1630b57cec5SDimitry Andric     ViewLegalizeDAGs("view-legalize-dags", cl::Hidden,
1640b57cec5SDimitry Andric                      cl::desc("Pop up a window to show dags before legalize"));
1650b57cec5SDimitry Andric static cl::opt<bool>
1660b57cec5SDimitry Andric ViewDAGCombine2("view-dag-combine2-dags", cl::Hidden,
1670b57cec5SDimitry Andric           cl::desc("Pop up a window to show dags before the second "
1680b57cec5SDimitry Andric                    "dag combine pass"));
1690b57cec5SDimitry Andric static cl::opt<bool>
1700b57cec5SDimitry Andric ViewISelDAGs("view-isel-dags", cl::Hidden,
1710b57cec5SDimitry Andric           cl::desc("Pop up a window to show isel dags as they are selected"));
1720b57cec5SDimitry Andric static cl::opt<bool>
1730b57cec5SDimitry Andric ViewSchedDAGs("view-sched-dags", cl::Hidden,
1740b57cec5SDimitry Andric           cl::desc("Pop up a window to show sched dags as they are processed"));
1750b57cec5SDimitry Andric static cl::opt<bool>
1760b57cec5SDimitry Andric ViewSUnitDAGs("view-sunit-dags", cl::Hidden,
1770b57cec5SDimitry Andric       cl::desc("Pop up a window to show SUnit dags after they are processed"));
1780b57cec5SDimitry Andric #else
179480093f4SDimitry Andric static const bool ViewDAGCombine1 = false, ViewLegalizeTypesDAGs = false,
180480093f4SDimitry Andric                   ViewDAGCombineLT = false, ViewLegalizeDAGs = false,
181480093f4SDimitry Andric                   ViewDAGCombine2 = false, ViewISelDAGs = false,
182480093f4SDimitry Andric                   ViewSchedDAGs = false, ViewSUnitDAGs = false;
1830b57cec5SDimitry Andric #endif
1840b57cec5SDimitry Andric 
185*5f757f3fSDimitry Andric #ifndef NDEBUG
186*5f757f3fSDimitry Andric #define ISEL_DUMP(X)                                                           \
187*5f757f3fSDimitry Andric   do {                                                                         \
188*5f757f3fSDimitry Andric     if (llvm::DebugFlag &&                                                     \
189*5f757f3fSDimitry Andric         (isCurrentDebugType(DEBUG_TYPE) ||                                     \
190*5f757f3fSDimitry Andric          (isCurrentDebugType(ISEL_DUMP_DEBUG_TYPE) && MatchFilterFuncName))) { \
191*5f757f3fSDimitry Andric       X;                                                                       \
192*5f757f3fSDimitry Andric     }                                                                          \
193*5f757f3fSDimitry Andric   } while (false)
194*5f757f3fSDimitry Andric #else
195*5f757f3fSDimitry Andric #define ISEL_DUMP(X) do { } while (false)
196*5f757f3fSDimitry Andric #endif
197*5f757f3fSDimitry Andric 
1980b57cec5SDimitry Andric //===---------------------------------------------------------------------===//
1990b57cec5SDimitry Andric ///
2000b57cec5SDimitry Andric /// RegisterScheduler class - Track the registration of instruction schedulers.
2010b57cec5SDimitry Andric ///
2020b57cec5SDimitry Andric //===---------------------------------------------------------------------===//
2030b57cec5SDimitry Andric MachinePassRegistry<RegisterScheduler::FunctionPassCtor>
2040b57cec5SDimitry Andric     RegisterScheduler::Registry;
2050b57cec5SDimitry Andric 
2060b57cec5SDimitry Andric //===---------------------------------------------------------------------===//
2070b57cec5SDimitry Andric ///
2080b57cec5SDimitry Andric /// ISHeuristic command line option for instruction schedulers.
2090b57cec5SDimitry Andric ///
2100b57cec5SDimitry Andric //===---------------------------------------------------------------------===//
2110b57cec5SDimitry Andric static cl::opt<RegisterScheduler::FunctionPassCtor, false,
2120b57cec5SDimitry Andric                RegisterPassParser<RegisterScheduler>>
2130b57cec5SDimitry Andric ISHeuristic("pre-RA-sched",
2140b57cec5SDimitry Andric             cl::init(&createDefaultScheduler), cl::Hidden,
2150b57cec5SDimitry Andric             cl::desc("Instruction schedulers available (before register"
2160b57cec5SDimitry Andric                      " allocation):"));
2170b57cec5SDimitry Andric 
2180b57cec5SDimitry Andric static RegisterScheduler
2190b57cec5SDimitry Andric defaultListDAGScheduler("default", "Best scheduler for the target",
2200b57cec5SDimitry Andric                         createDefaultScheduler);
2210b57cec5SDimitry Andric 
222*5f757f3fSDimitry Andric static bool dontUseFastISelFor(const Function &Fn) {
223*5f757f3fSDimitry Andric   // Don't enable FastISel for functions with swiftasync Arguments.
224*5f757f3fSDimitry Andric   // Debug info on those is reliant on good Argument lowering, and FastISel is
225*5f757f3fSDimitry Andric   // not capable of lowering the entire function. Mixing the two selectors tend
226*5f757f3fSDimitry Andric   // to result in poor lowering of Arguments.
227*5f757f3fSDimitry Andric   return any_of(Fn.args(), [](const Argument &Arg) {
228*5f757f3fSDimitry Andric     return Arg.hasAttribute(Attribute::AttrKind::SwiftAsync);
229*5f757f3fSDimitry Andric   });
230*5f757f3fSDimitry Andric }
231*5f757f3fSDimitry Andric 
2320b57cec5SDimitry Andric namespace llvm {
2330b57cec5SDimitry Andric 
2340b57cec5SDimitry Andric   //===--------------------------------------------------------------------===//
2350b57cec5SDimitry Andric   /// This class is used by SelectionDAGISel to temporarily override
2360b57cec5SDimitry Andric   /// the optimization level on a per-function basis.
2370b57cec5SDimitry Andric   class OptLevelChanger {
2380b57cec5SDimitry Andric     SelectionDAGISel &IS;
239*5f757f3fSDimitry Andric     CodeGenOptLevel SavedOptLevel;
2400b57cec5SDimitry Andric     bool SavedFastISel;
2410b57cec5SDimitry Andric 
2420b57cec5SDimitry Andric   public:
243*5f757f3fSDimitry Andric     OptLevelChanger(SelectionDAGISel &ISel, CodeGenOptLevel NewOptLevel)
244*5f757f3fSDimitry Andric         : IS(ISel) {
2450b57cec5SDimitry Andric       SavedOptLevel = IS.OptLevel;
2465ffd83dbSDimitry Andric       SavedFastISel = IS.TM.Options.EnableFastISel;
247*5f757f3fSDimitry Andric       if (NewOptLevel != SavedOptLevel) {
2480b57cec5SDimitry Andric         IS.OptLevel = NewOptLevel;
2490b57cec5SDimitry Andric         IS.TM.setOptLevel(NewOptLevel);
2500b57cec5SDimitry Andric         LLVM_DEBUG(dbgs() << "\nChanging optimization level for Function "
2510b57cec5SDimitry Andric                           << IS.MF->getFunction().getName() << "\n");
252*5f757f3fSDimitry Andric         LLVM_DEBUG(dbgs() << "\tBefore: -O" << static_cast<int>(SavedOptLevel)
253*5f757f3fSDimitry Andric                           << " ; After: -O" << static_cast<int>(NewOptLevel)
254*5f757f3fSDimitry Andric                           << "\n");
255*5f757f3fSDimitry Andric         if (NewOptLevel == CodeGenOptLevel::None)
2560b57cec5SDimitry Andric           IS.TM.setFastISel(IS.TM.getO0WantsFastISel());
257*5f757f3fSDimitry Andric       }
258*5f757f3fSDimitry Andric       if (dontUseFastISelFor(IS.MF->getFunction()))
259*5f757f3fSDimitry Andric         IS.TM.setFastISel(false);
2600b57cec5SDimitry Andric       LLVM_DEBUG(
2610b57cec5SDimitry Andric           dbgs() << "\tFastISel is "
2620b57cec5SDimitry Andric                  << (IS.TM.Options.EnableFastISel ? "enabled" : "disabled")
2630b57cec5SDimitry Andric                  << "\n");
2640b57cec5SDimitry Andric     }
2650b57cec5SDimitry Andric 
2660b57cec5SDimitry Andric     ~OptLevelChanger() {
2670b57cec5SDimitry Andric       if (IS.OptLevel == SavedOptLevel)
2680b57cec5SDimitry Andric         return;
2690b57cec5SDimitry Andric       LLVM_DEBUG(dbgs() << "\nRestoring optimization level for Function "
2700b57cec5SDimitry Andric                         << IS.MF->getFunction().getName() << "\n");
271*5f757f3fSDimitry Andric       LLVM_DEBUG(dbgs() << "\tBefore: -O" << static_cast<int>(IS.OptLevel)
272*5f757f3fSDimitry Andric                         << " ; After: -O" << static_cast<int>(SavedOptLevel) << "\n");
2730b57cec5SDimitry Andric       IS.OptLevel = SavedOptLevel;
2740b57cec5SDimitry Andric       IS.TM.setOptLevel(SavedOptLevel);
2750b57cec5SDimitry Andric       IS.TM.setFastISel(SavedFastISel);
2760b57cec5SDimitry Andric     }
2770b57cec5SDimitry Andric   };
2780b57cec5SDimitry Andric 
2790b57cec5SDimitry Andric   //===--------------------------------------------------------------------===//
2800b57cec5SDimitry Andric   /// createDefaultScheduler - This creates an instruction scheduler appropriate
2810b57cec5SDimitry Andric   /// for the target.
2820b57cec5SDimitry Andric   ScheduleDAGSDNodes *createDefaultScheduler(SelectionDAGISel *IS,
283*5f757f3fSDimitry Andric                                              CodeGenOptLevel OptLevel) {
2840b57cec5SDimitry Andric     const TargetLowering *TLI = IS->TLI;
2850b57cec5SDimitry Andric     const TargetSubtargetInfo &ST = IS->MF->getSubtarget();
2860b57cec5SDimitry Andric 
2870b57cec5SDimitry Andric     // Try first to see if the Target has its own way of selecting a scheduler
2880b57cec5SDimitry Andric     if (auto *SchedulerCtor = ST.getDAGScheduler(OptLevel)) {
2890b57cec5SDimitry Andric       return SchedulerCtor(IS, OptLevel);
2900b57cec5SDimitry Andric     }
2910b57cec5SDimitry Andric 
292*5f757f3fSDimitry Andric     if (OptLevel == CodeGenOptLevel::None ||
2930b57cec5SDimitry Andric         (ST.enableMachineScheduler() && ST.enableMachineSchedDefaultSched()) ||
2940b57cec5SDimitry Andric         TLI->getSchedulingPreference() == Sched::Source)
2950b57cec5SDimitry Andric       return createSourceListDAGScheduler(IS, OptLevel);
2960b57cec5SDimitry Andric     if (TLI->getSchedulingPreference() == Sched::RegPressure)
2970b57cec5SDimitry Andric       return createBURRListDAGScheduler(IS, OptLevel);
2980b57cec5SDimitry Andric     if (TLI->getSchedulingPreference() == Sched::Hybrid)
2990b57cec5SDimitry Andric       return createHybridListDAGScheduler(IS, OptLevel);
3000b57cec5SDimitry Andric     if (TLI->getSchedulingPreference() == Sched::VLIW)
3010b57cec5SDimitry Andric       return createVLIWDAGScheduler(IS, OptLevel);
302fe6060f1SDimitry Andric     if (TLI->getSchedulingPreference() == Sched::Fast)
303fe6060f1SDimitry Andric       return createFastDAGScheduler(IS, OptLevel);
304fe6060f1SDimitry Andric     if (TLI->getSchedulingPreference() == Sched::Linearize)
305fe6060f1SDimitry Andric       return createDAGLinearizer(IS, OptLevel);
3060b57cec5SDimitry Andric     assert(TLI->getSchedulingPreference() == Sched::ILP &&
3070b57cec5SDimitry Andric            "Unknown sched type!");
3080b57cec5SDimitry Andric     return createILPListDAGScheduler(IS, OptLevel);
3090b57cec5SDimitry Andric   }
3100b57cec5SDimitry Andric 
3110b57cec5SDimitry Andric } // end namespace llvm
3120b57cec5SDimitry Andric 
3130b57cec5SDimitry Andric // EmitInstrWithCustomInserter - This method should be implemented by targets
3140b57cec5SDimitry Andric // that mark instructions with the 'usesCustomInserter' flag.  These
3150b57cec5SDimitry Andric // instructions are special in various ways, which require special support to
3160b57cec5SDimitry Andric // insert.  The specified MachineInstr is created but not inserted into any
3170b57cec5SDimitry Andric // basic blocks, and this method is called to expand it into a sequence of
3180b57cec5SDimitry Andric // instructions, potentially also creating new basic blocks and control flow.
3190b57cec5SDimitry Andric // When new basic blocks are inserted and the edges from MBB to its successors
3200b57cec5SDimitry Andric // are modified, the method should insert pairs of <OldSucc, NewSucc> into the
3210b57cec5SDimitry Andric // DenseMap.
3220b57cec5SDimitry Andric MachineBasicBlock *
3230b57cec5SDimitry Andric TargetLowering::EmitInstrWithCustomInserter(MachineInstr &MI,
3240b57cec5SDimitry Andric                                             MachineBasicBlock *MBB) const {
3250b57cec5SDimitry Andric #ifndef NDEBUG
3260b57cec5SDimitry Andric   dbgs() << "If a target marks an instruction with "
3270b57cec5SDimitry Andric           "'usesCustomInserter', it must implement "
3280eae32dcSDimitry Andric           "TargetLowering::EmitInstrWithCustomInserter!\n";
3290b57cec5SDimitry Andric #endif
3300b57cec5SDimitry Andric   llvm_unreachable(nullptr);
3310b57cec5SDimitry Andric }
3320b57cec5SDimitry Andric 
3330b57cec5SDimitry Andric void TargetLowering::AdjustInstrPostInstrSelection(MachineInstr &MI,
3340b57cec5SDimitry Andric                                                    SDNode *Node) const {
3350b57cec5SDimitry Andric   assert(!MI.hasPostISelHook() &&
3360b57cec5SDimitry Andric          "If a target marks an instruction with 'hasPostISelHook', "
3370b57cec5SDimitry Andric          "it must implement TargetLowering::AdjustInstrPostInstrSelection!");
3380b57cec5SDimitry Andric }
3390b57cec5SDimitry Andric 
3400b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
3410b57cec5SDimitry Andric // SelectionDAGISel code
3420b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
3430b57cec5SDimitry Andric 
344bdd1243dSDimitry Andric SelectionDAGISel::SelectionDAGISel(char &ID, TargetMachine &tm,
345*5f757f3fSDimitry Andric                                    CodeGenOptLevel OL)
346480093f4SDimitry Andric     : MachineFunctionPass(ID), TM(tm), FuncInfo(new FunctionLoweringInfo()),
3470b57cec5SDimitry Andric       SwiftError(new SwiftErrorValueTracking()),
3480b57cec5SDimitry Andric       CurDAG(new SelectionDAG(tm, OL)),
349480093f4SDimitry Andric       SDB(std::make_unique<SelectionDAGBuilder>(*CurDAG, *FuncInfo, *SwiftError,
350480093f4SDimitry Andric                                                 OL)),
3511fd87a68SDimitry Andric       OptLevel(OL) {
3520b57cec5SDimitry Andric   initializeGCModuleInfoPass(*PassRegistry::getPassRegistry());
3530b57cec5SDimitry Andric   initializeBranchProbabilityInfoWrapperPassPass(
3540b57cec5SDimitry Andric       *PassRegistry::getPassRegistry());
3550b57cec5SDimitry Andric   initializeAAResultsWrapperPassPass(*PassRegistry::getPassRegistry());
356480093f4SDimitry Andric   initializeTargetLibraryInfoWrapperPassPass(*PassRegistry::getPassRegistry());
3570b57cec5SDimitry Andric }
3580b57cec5SDimitry Andric 
3590b57cec5SDimitry Andric SelectionDAGISel::~SelectionDAGISel() {
3600b57cec5SDimitry Andric   delete CurDAG;
3610b57cec5SDimitry Andric   delete SwiftError;
3620b57cec5SDimitry Andric }
3630b57cec5SDimitry Andric 
3640b57cec5SDimitry Andric void SelectionDAGISel::getAnalysisUsage(AnalysisUsage &AU) const {
365*5f757f3fSDimitry Andric   if (OptLevel != CodeGenOptLevel::None)
3660b57cec5SDimitry Andric       AU.addRequired<AAResultsWrapperPass>();
3670b57cec5SDimitry Andric   AU.addRequired<GCModuleInfo>();
3680b57cec5SDimitry Andric   AU.addRequired<StackProtector>();
3690b57cec5SDimitry Andric   AU.addPreserved<GCModuleInfo>();
3700b57cec5SDimitry Andric   AU.addRequired<TargetLibraryInfoWrapperPass>();
3710b57cec5SDimitry Andric   AU.addRequired<TargetTransformInfoWrapperPass>();
372bdd1243dSDimitry Andric   AU.addRequired<AssumptionCacheTracker>();
373*5f757f3fSDimitry Andric   if (UseMBPI && OptLevel != CodeGenOptLevel::None)
3740b57cec5SDimitry Andric       AU.addRequired<BranchProbabilityInfoWrapperPass>();
375480093f4SDimitry Andric   AU.addRequired<ProfileSummaryInfoWrapperPass>();
376bdd1243dSDimitry Andric   // AssignmentTrackingAnalysis only runs if assignment tracking is enabled for
377bdd1243dSDimitry Andric   // the module.
378bdd1243dSDimitry Andric   AU.addRequired<AssignmentTrackingAnalysis>();
379bdd1243dSDimitry Andric   AU.addPreserved<AssignmentTrackingAnalysis>();
380*5f757f3fSDimitry Andric   if (OptLevel != CodeGenOptLevel::None)
381480093f4SDimitry Andric       LazyBlockFrequencyInfoPass::getLazyBFIAnalysisUsage(AU);
3820b57cec5SDimitry Andric   MachineFunctionPass::getAnalysisUsage(AU);
3830b57cec5SDimitry Andric }
3840b57cec5SDimitry Andric 
3850b57cec5SDimitry Andric static void computeUsesMSVCFloatingPoint(const Triple &TT, const Function &F,
3860b57cec5SDimitry Andric                                          MachineModuleInfo &MMI) {
3870b57cec5SDimitry Andric   // Only needed for MSVC
3880b57cec5SDimitry Andric   if (!TT.isWindowsMSVCEnvironment())
3890b57cec5SDimitry Andric     return;
3900b57cec5SDimitry Andric 
3910b57cec5SDimitry Andric   // If it's already set, nothing to do.
3920b57cec5SDimitry Andric   if (MMI.usesMSVCFloatingPoint())
3930b57cec5SDimitry Andric     return;
3940b57cec5SDimitry Andric 
3950b57cec5SDimitry Andric   for (const Instruction &I : instructions(F)) {
3960b57cec5SDimitry Andric     if (I.getType()->isFPOrFPVectorTy()) {
3970b57cec5SDimitry Andric       MMI.setUsesMSVCFloatingPoint(true);
3980b57cec5SDimitry Andric       return;
3990b57cec5SDimitry Andric     }
4000b57cec5SDimitry Andric     for (const auto &Op : I.operands()) {
4010b57cec5SDimitry Andric       if (Op->getType()->isFPOrFPVectorTy()) {
4020b57cec5SDimitry Andric         MMI.setUsesMSVCFloatingPoint(true);
4030b57cec5SDimitry Andric         return;
4040b57cec5SDimitry Andric       }
4050b57cec5SDimitry Andric     }
4060b57cec5SDimitry Andric   }
4070b57cec5SDimitry Andric }
4080b57cec5SDimitry Andric 
4090b57cec5SDimitry Andric bool SelectionDAGISel::runOnMachineFunction(MachineFunction &mf) {
4100b57cec5SDimitry Andric   // If we already selected that function, we do not need to run SDISel.
4110b57cec5SDimitry Andric   if (mf.getProperties().hasProperty(
4120b57cec5SDimitry Andric           MachineFunctionProperties::Property::Selected))
4130b57cec5SDimitry Andric     return false;
4140b57cec5SDimitry Andric   // Do some sanity-checking on the command-line options.
4150b57cec5SDimitry Andric   assert((!EnableFastISelAbort || TM.Options.EnableFastISel) &&
4160b57cec5SDimitry Andric          "-fast-isel-abort > 0 requires -fast-isel");
4170b57cec5SDimitry Andric 
4180b57cec5SDimitry Andric   const Function &Fn = mf.getFunction();
4190b57cec5SDimitry Andric   MF = &mf;
4200b57cec5SDimitry Andric 
421*5f757f3fSDimitry Andric #ifndef NDEBUG
422*5f757f3fSDimitry Andric   StringRef FuncName = Fn.getName();
423*5f757f3fSDimitry Andric   MatchFilterFuncName = isFunctionInPrintList(FuncName);
424*5f757f3fSDimitry Andric #else
425*5f757f3fSDimitry Andric   (void)MatchFilterFuncName;
426*5f757f3fSDimitry Andric #endif
427*5f757f3fSDimitry Andric 
4283a9a9c0cSDimitry Andric   // Decide what flavour of variable location debug-info will be used, before
4293a9a9c0cSDimitry Andric   // we change the optimisation level.
430bdd1243dSDimitry Andric   bool InstrRef = mf.shouldUseDebugInstrRef();
431bdd1243dSDimitry Andric   mf.setUseDebugInstrRef(InstrRef);
4323a9a9c0cSDimitry Andric 
4330b57cec5SDimitry Andric   // Reset the target options before resetting the optimization
4340b57cec5SDimitry Andric   // level below.
4350b57cec5SDimitry Andric   // FIXME: This is a horrible hack and should be processed via
4360b57cec5SDimitry Andric   // codegen looking at the optimization level explicitly when
4370b57cec5SDimitry Andric   // it wants to look at it.
4380b57cec5SDimitry Andric   TM.resetTargetOptions(Fn);
4390b57cec5SDimitry Andric   // Reset OptLevel to None for optnone functions.
440*5f757f3fSDimitry Andric   CodeGenOptLevel NewOptLevel = OptLevel;
441*5f757f3fSDimitry Andric   if (OptLevel != CodeGenOptLevel::None && skipFunction(Fn))
442*5f757f3fSDimitry Andric     NewOptLevel = CodeGenOptLevel::None;
4430b57cec5SDimitry Andric   OptLevelChanger OLC(*this, NewOptLevel);
4440b57cec5SDimitry Andric 
4450b57cec5SDimitry Andric   TII = MF->getSubtarget().getInstrInfo();
4460b57cec5SDimitry Andric   TLI = MF->getSubtarget().getTargetLowering();
4470b57cec5SDimitry Andric   RegInfo = &MF->getRegInfo();
4488bcb0991SDimitry Andric   LibInfo = &getAnalysis<TargetLibraryInfoWrapperPass>().getTLI(Fn);
4490b57cec5SDimitry Andric   GFI = Fn.hasGC() ? &getAnalysis<GCModuleInfo>().getFunctionInfo(Fn) : nullptr;
4508bcb0991SDimitry Andric   ORE = std::make_unique<OptimizationRemarkEmitter>(&Fn);
451bdd1243dSDimitry Andric   AC = &getAnalysis<AssumptionCacheTracker>().getAssumptionCache(mf.getFunction());
452480093f4SDimitry Andric   auto *PSI = &getAnalysis<ProfileSummaryInfoWrapperPass>().getPSI();
4535ffd83dbSDimitry Andric   BlockFrequencyInfo *BFI = nullptr;
454*5f757f3fSDimitry Andric   if (PSI && PSI->hasProfileSummary() && OptLevel != CodeGenOptLevel::None)
4555ffd83dbSDimitry Andric     BFI = &getAnalysis<LazyBlockFrequencyInfoPass>().getBFI();
4560b57cec5SDimitry Andric 
457bdd1243dSDimitry Andric   FunctionVarLocs const *FnVarLocs = nullptr;
458bdd1243dSDimitry Andric   if (isAssignmentTrackingEnabled(*Fn.getParent()))
459bdd1243dSDimitry Andric     FnVarLocs = getAnalysis<AssignmentTrackingAnalysis>().getResults();
460bdd1243dSDimitry Andric 
461*5f757f3fSDimitry Andric   ISEL_DUMP(dbgs() << "\n\n\n=== " << FuncName << "\n");
4620b57cec5SDimitry Andric 
46306c3fb27SDimitry Andric   UniformityInfo *UA = nullptr;
46406c3fb27SDimitry Andric   if (auto *UAPass = getAnalysisIfAvailable<UniformityInfoWrapperPass>())
46506c3fb27SDimitry Andric     UA = &UAPass->getUniformityInfo();
46606c3fb27SDimitry Andric   CurDAG->init(*MF, *ORE, this, LibInfo, UA, PSI, BFI, FnVarLocs);
4670b57cec5SDimitry Andric   FuncInfo->set(Fn, *MF, CurDAG);
4680b57cec5SDimitry Andric   SwiftError->setFunction(*MF);
4690b57cec5SDimitry Andric 
4700b57cec5SDimitry Andric   // Now get the optional analyzes if we want to.
4710b57cec5SDimitry Andric   // This is based on the possibly changed OptLevel (after optnone is taken
4720b57cec5SDimitry Andric   // into account).  That's unfortunate but OK because it just means we won't
4730b57cec5SDimitry Andric   // ask for passes that have been required anyway.
4740b57cec5SDimitry Andric 
475*5f757f3fSDimitry Andric   if (UseMBPI && OptLevel != CodeGenOptLevel::None)
4760b57cec5SDimitry Andric     FuncInfo->BPI = &getAnalysis<BranchProbabilityInfoWrapperPass>().getBPI();
4770b57cec5SDimitry Andric   else
4780b57cec5SDimitry Andric     FuncInfo->BPI = nullptr;
4790b57cec5SDimitry Andric 
480*5f757f3fSDimitry Andric   if (OptLevel != CodeGenOptLevel::None)
4810b57cec5SDimitry Andric     AA = &getAnalysis<AAResultsWrapperPass>().getAAResults();
4820b57cec5SDimitry Andric   else
4830b57cec5SDimitry Andric     AA = nullptr;
4840b57cec5SDimitry Andric 
485bdd1243dSDimitry Andric   SDB->init(GFI, AA, AC, LibInfo);
4860b57cec5SDimitry Andric 
4870b57cec5SDimitry Andric   MF->setHasInlineAsm(false);
4880b57cec5SDimitry Andric 
4890b57cec5SDimitry Andric   FuncInfo->SplitCSR = false;
4900b57cec5SDimitry Andric 
4910b57cec5SDimitry Andric   // We split CSR if the target supports it for the given function
4920b57cec5SDimitry Andric   // and the function has only return exits.
493*5f757f3fSDimitry Andric   if (OptLevel != CodeGenOptLevel::None && TLI->supportSplitCSR(MF)) {
4940b57cec5SDimitry Andric     FuncInfo->SplitCSR = true;
4950b57cec5SDimitry Andric 
4960b57cec5SDimitry Andric     // Collect all the return blocks.
4970b57cec5SDimitry Andric     for (const BasicBlock &BB : Fn) {
4980b57cec5SDimitry Andric       if (!succ_empty(&BB))
4990b57cec5SDimitry Andric         continue;
5000b57cec5SDimitry Andric 
5010b57cec5SDimitry Andric       const Instruction *Term = BB.getTerminator();
5020b57cec5SDimitry Andric       if (isa<UnreachableInst>(Term) || isa<ReturnInst>(Term))
5030b57cec5SDimitry Andric         continue;
5040b57cec5SDimitry Andric 
5050b57cec5SDimitry Andric       // Bail out if the exit block is not Return nor Unreachable.
5060b57cec5SDimitry Andric       FuncInfo->SplitCSR = false;
5070b57cec5SDimitry Andric       break;
5080b57cec5SDimitry Andric     }
5090b57cec5SDimitry Andric   }
5100b57cec5SDimitry Andric 
5110b57cec5SDimitry Andric   MachineBasicBlock *EntryMBB = &MF->front();
5120b57cec5SDimitry Andric   if (FuncInfo->SplitCSR)
5130b57cec5SDimitry Andric     // This performs initialization so lowering for SplitCSR will be correct.
5140b57cec5SDimitry Andric     TLI->initializeSplitCSR(EntryMBB);
5150b57cec5SDimitry Andric 
5160b57cec5SDimitry Andric   SelectAllBasicBlocks(Fn);
5170b57cec5SDimitry Andric   if (FastISelFailed && EnableFastISelFallbackReport) {
5180b57cec5SDimitry Andric     DiagnosticInfoISelFallback DiagFallback(Fn);
5190b57cec5SDimitry Andric     Fn.getContext().diagnose(DiagFallback);
5200b57cec5SDimitry Andric   }
5210b57cec5SDimitry Andric 
5220b57cec5SDimitry Andric   // Replace forward-declared registers with the registers containing
5230b57cec5SDimitry Andric   // the desired value.
5240b57cec5SDimitry Andric   // Note: it is important that this happens **before** the call to
5250b57cec5SDimitry Andric   // EmitLiveInCopies, since implementations can skip copies of unused
5260b57cec5SDimitry Andric   // registers. If we don't apply the reg fixups before, some registers may
5270b57cec5SDimitry Andric   // appear as unused and will be skipped, resulting in bad MI.
5280b57cec5SDimitry Andric   MachineRegisterInfo &MRI = MF->getRegInfo();
5295ffd83dbSDimitry Andric   for (DenseMap<Register, Register>::iterator I = FuncInfo->RegFixups.begin(),
5300b57cec5SDimitry Andric                                               E = FuncInfo->RegFixups.end();
5310b57cec5SDimitry Andric        I != E; ++I) {
5325ffd83dbSDimitry Andric     Register From = I->first;
5335ffd83dbSDimitry Andric     Register To = I->second;
5340b57cec5SDimitry Andric     // If To is also scheduled to be replaced, find what its ultimate
5350b57cec5SDimitry Andric     // replacement is.
5360b57cec5SDimitry Andric     while (true) {
5375ffd83dbSDimitry Andric       DenseMap<Register, Register>::iterator J = FuncInfo->RegFixups.find(To);
5380b57cec5SDimitry Andric       if (J == E)
5390b57cec5SDimitry Andric         break;
5400b57cec5SDimitry Andric       To = J->second;
5410b57cec5SDimitry Andric     }
5420b57cec5SDimitry Andric     // Make sure the new register has a sufficiently constrained register class.
543bdd1243dSDimitry Andric     if (From.isVirtual() && To.isVirtual())
5440b57cec5SDimitry Andric       MRI.constrainRegClass(To, MRI.getRegClass(From));
5450b57cec5SDimitry Andric     // Replace it.
5460b57cec5SDimitry Andric 
5470b57cec5SDimitry Andric     // Replacing one register with another won't touch the kill flags.
5480b57cec5SDimitry Andric     // We need to conservatively clear the kill flags as a kill on the old
5490b57cec5SDimitry Andric     // register might dominate existing uses of the new register.
5500b57cec5SDimitry Andric     if (!MRI.use_empty(To))
5510b57cec5SDimitry Andric       MRI.clearKillFlags(From);
5520b57cec5SDimitry Andric     MRI.replaceRegWith(From, To);
5530b57cec5SDimitry Andric   }
5540b57cec5SDimitry Andric 
5550b57cec5SDimitry Andric   // If the first basic block in the function has live ins that need to be
5560b57cec5SDimitry Andric   // copied into vregs, emit the copies into the top of the block before
5570b57cec5SDimitry Andric   // emitting the code for the block.
5580b57cec5SDimitry Andric   const TargetRegisterInfo &TRI = *MF->getSubtarget().getRegisterInfo();
5590b57cec5SDimitry Andric   RegInfo->EmitLiveInCopies(EntryMBB, TRI, *TII);
5600b57cec5SDimitry Andric 
5610b57cec5SDimitry Andric   // Insert copies in the entry block and the return blocks.
5620b57cec5SDimitry Andric   if (FuncInfo->SplitCSR) {
5630b57cec5SDimitry Andric     SmallVector<MachineBasicBlock*, 4> Returns;
5640b57cec5SDimitry Andric     // Collect all the return blocks.
5650b57cec5SDimitry Andric     for (MachineBasicBlock &MBB : mf) {
5660b57cec5SDimitry Andric       if (!MBB.succ_empty())
5670b57cec5SDimitry Andric         continue;
5680b57cec5SDimitry Andric 
5690b57cec5SDimitry Andric       MachineBasicBlock::iterator Term = MBB.getFirstTerminator();
5700b57cec5SDimitry Andric       if (Term != MBB.end() && Term->isReturn()) {
5710b57cec5SDimitry Andric         Returns.push_back(&MBB);
5720b57cec5SDimitry Andric         continue;
5730b57cec5SDimitry Andric       }
5740b57cec5SDimitry Andric     }
5750b57cec5SDimitry Andric     TLI->insertCopiesSplitCSR(EntryMBB, Returns);
5760b57cec5SDimitry Andric   }
5770b57cec5SDimitry Andric 
5780b57cec5SDimitry Andric   DenseMap<unsigned, unsigned> LiveInMap;
5790b57cec5SDimitry Andric   if (!FuncInfo->ArgDbgValues.empty())
5800b57cec5SDimitry Andric     for (std::pair<unsigned, unsigned> LI : RegInfo->liveins())
5810b57cec5SDimitry Andric       if (LI.second)
5820b57cec5SDimitry Andric         LiveInMap.insert(LI);
5830b57cec5SDimitry Andric 
5840b57cec5SDimitry Andric   // Insert DBG_VALUE instructions for function arguments to the entry block.
5850b57cec5SDimitry Andric   for (unsigned i = 0, e = FuncInfo->ArgDbgValues.size(); i != e; ++i) {
5860b57cec5SDimitry Andric     MachineInstr *MI = FuncInfo->ArgDbgValues[e - i - 1];
587fe6060f1SDimitry Andric     assert(MI->getOpcode() != TargetOpcode::DBG_VALUE_LIST &&
588fe6060f1SDimitry Andric            "Function parameters should not be described by DBG_VALUE_LIST.");
589bdd1243dSDimitry Andric     bool hasFI = MI->getDebugOperand(0).isFI();
5900b57cec5SDimitry Andric     Register Reg =
591bdd1243dSDimitry Andric         hasFI ? TRI.getFrameRegister(*MF) : MI->getDebugOperand(0).getReg();
592bdd1243dSDimitry Andric     if (Reg.isPhysical())
5930b57cec5SDimitry Andric       EntryMBB->insert(EntryMBB->begin(), MI);
5940b57cec5SDimitry Andric     else {
5950b57cec5SDimitry Andric       MachineInstr *Def = RegInfo->getVRegDef(Reg);
5960b57cec5SDimitry Andric       if (Def) {
5970b57cec5SDimitry Andric         MachineBasicBlock::iterator InsertPos = Def;
5980b57cec5SDimitry Andric         // FIXME: VR def may not be in entry block.
5990b57cec5SDimitry Andric         Def->getParent()->insert(std::next(InsertPos), MI);
6000b57cec5SDimitry Andric       } else
6010b57cec5SDimitry Andric         LLVM_DEBUG(dbgs() << "Dropping debug info for dead vreg"
6028bcb0991SDimitry Andric                           << Register::virtReg2Index(Reg) << "\n");
6030b57cec5SDimitry Andric     }
6040b57cec5SDimitry Andric 
605fe6060f1SDimitry Andric     // Don't try and extend through copies in instruction referencing mode.
606fe6060f1SDimitry Andric     if (InstrRef)
607fe6060f1SDimitry Andric       continue;
608fe6060f1SDimitry Andric 
6090b57cec5SDimitry Andric     // If Reg is live-in then update debug info to track its copy in a vreg.
6100b57cec5SDimitry Andric     DenseMap<unsigned, unsigned>::iterator LDI = LiveInMap.find(Reg);
6110b57cec5SDimitry Andric     if (LDI != LiveInMap.end()) {
6120b57cec5SDimitry Andric       assert(!hasFI && "There's no handling of frame pointer updating here yet "
6130b57cec5SDimitry Andric                        "- add if needed");
6140b57cec5SDimitry Andric       MachineInstr *Def = RegInfo->getVRegDef(LDI->second);
6150b57cec5SDimitry Andric       MachineBasicBlock::iterator InsertPos = Def;
6160b57cec5SDimitry Andric       const MDNode *Variable = MI->getDebugVariable();
6170b57cec5SDimitry Andric       const MDNode *Expr = MI->getDebugExpression();
6180b57cec5SDimitry Andric       DebugLoc DL = MI->getDebugLoc();
6190b57cec5SDimitry Andric       bool IsIndirect = MI->isIndirectDebugValue();
6200b57cec5SDimitry Andric       if (IsIndirect)
621bdd1243dSDimitry Andric         assert(MI->getDebugOffset().getImm() == 0 &&
6220b57cec5SDimitry Andric                "DBG_VALUE with nonzero offset");
6230b57cec5SDimitry Andric       assert(cast<DILocalVariable>(Variable)->isValidLocationForIntrinsic(DL) &&
6240b57cec5SDimitry Andric              "Expected inlined-at fields to agree");
625fe6060f1SDimitry Andric       assert(MI->getOpcode() != TargetOpcode::DBG_VALUE_LIST &&
626fe6060f1SDimitry Andric              "Didn't expect to see a DBG_VALUE_LIST here");
6270b57cec5SDimitry Andric       // Def is never a terminator here, so it is ok to increment InsertPos.
6280b57cec5SDimitry Andric       BuildMI(*EntryMBB, ++InsertPos, DL, TII->get(TargetOpcode::DBG_VALUE),
6290b57cec5SDimitry Andric               IsIndirect, LDI->second, Variable, Expr);
6300b57cec5SDimitry Andric 
6310b57cec5SDimitry Andric       // If this vreg is directly copied into an exported register then
6320b57cec5SDimitry Andric       // that COPY instructions also need DBG_VALUE, if it is the only
6330b57cec5SDimitry Andric       // user of LDI->second.
6340b57cec5SDimitry Andric       MachineInstr *CopyUseMI = nullptr;
6350b57cec5SDimitry Andric       for (MachineRegisterInfo::use_instr_iterator
6360b57cec5SDimitry Andric            UI = RegInfo->use_instr_begin(LDI->second),
6370b57cec5SDimitry Andric            E = RegInfo->use_instr_end(); UI != E; ) {
6380b57cec5SDimitry Andric         MachineInstr *UseMI = &*(UI++);
6390b57cec5SDimitry Andric         if (UseMI->isDebugValue()) continue;
6400b57cec5SDimitry Andric         if (UseMI->isCopy() && !CopyUseMI && UseMI->getParent() == EntryMBB) {
6410b57cec5SDimitry Andric           CopyUseMI = UseMI; continue;
6420b57cec5SDimitry Andric         }
6430b57cec5SDimitry Andric         // Otherwise this is another use or second copy use.
6440b57cec5SDimitry Andric         CopyUseMI = nullptr; break;
6450b57cec5SDimitry Andric       }
6465ffd83dbSDimitry Andric       if (CopyUseMI &&
6475ffd83dbSDimitry Andric           TRI.getRegSizeInBits(LDI->second, MRI) ==
6485ffd83dbSDimitry Andric               TRI.getRegSizeInBits(CopyUseMI->getOperand(0).getReg(), MRI)) {
6490b57cec5SDimitry Andric         // Use MI's debug location, which describes where Variable was
6500b57cec5SDimitry Andric         // declared, rather than whatever is attached to CopyUseMI.
6510b57cec5SDimitry Andric         MachineInstr *NewMI =
6520b57cec5SDimitry Andric             BuildMI(*MF, DL, TII->get(TargetOpcode::DBG_VALUE), IsIndirect,
6530b57cec5SDimitry Andric                     CopyUseMI->getOperand(0).getReg(), Variable, Expr);
6540b57cec5SDimitry Andric         MachineBasicBlock::iterator Pos = CopyUseMI;
6550b57cec5SDimitry Andric         EntryMBB->insertAfter(Pos, NewMI);
6560b57cec5SDimitry Andric       }
6570b57cec5SDimitry Andric     }
6580b57cec5SDimitry Andric   }
6590b57cec5SDimitry Andric 
660fe6060f1SDimitry Andric   // For debug-info, in instruction referencing mode, we need to perform some
661fe6060f1SDimitry Andric   // post-isel maintenence.
662bdd1243dSDimitry Andric   if (MF->useDebugInstrRef())
663fe6060f1SDimitry Andric     MF->finalizeDebugInstrRefs();
664fe6060f1SDimitry Andric 
6650b57cec5SDimitry Andric   // Determine if there are any calls in this machine function.
6660b57cec5SDimitry Andric   MachineFrameInfo &MFI = MF->getFrameInfo();
6670b57cec5SDimitry Andric   for (const auto &MBB : *MF) {
6680b57cec5SDimitry Andric     if (MFI.hasCalls() && MF->hasInlineAsm())
6690b57cec5SDimitry Andric       break;
6700b57cec5SDimitry Andric 
6710b57cec5SDimitry Andric     for (const auto &MI : MBB) {
6720b57cec5SDimitry Andric       const MCInstrDesc &MCID = TII->get(MI.getOpcode());
6730b57cec5SDimitry Andric       if ((MCID.isCall() && !MCID.isReturn()) ||
6740b57cec5SDimitry Andric           MI.isStackAligningInlineAsm()) {
6750b57cec5SDimitry Andric         MFI.setHasCalls(true);
6760b57cec5SDimitry Andric       }
6770b57cec5SDimitry Andric       if (MI.isInlineAsm()) {
6780b57cec5SDimitry Andric         MF->setHasInlineAsm(true);
6790b57cec5SDimitry Andric       }
6800b57cec5SDimitry Andric     }
6810b57cec5SDimitry Andric   }
6820b57cec5SDimitry Andric 
6830b57cec5SDimitry Andric   // Determine if there is a call to setjmp in the machine function.
6840b57cec5SDimitry Andric   MF->setExposesReturnsTwice(Fn.callsFunctionThatReturnsTwice());
6850b57cec5SDimitry Andric 
6860b57cec5SDimitry Andric   // Determine if floating point is used for msvc
6870b57cec5SDimitry Andric   computeUsesMSVCFloatingPoint(TM.getTargetTriple(), Fn, MF->getMMI());
6880b57cec5SDimitry Andric 
6890b57cec5SDimitry Andric   // Release function-specific state. SDB and CurDAG are already cleared
6900b57cec5SDimitry Andric   // at this point.
6910b57cec5SDimitry Andric   FuncInfo->clear();
6920b57cec5SDimitry Andric 
693*5f757f3fSDimitry Andric   ISEL_DUMP(dbgs() << "*** MachineFunction at end of ISel ***\n");
694*5f757f3fSDimitry Andric   ISEL_DUMP(MF->print(dbgs()));
6950b57cec5SDimitry Andric 
6960b57cec5SDimitry Andric   return true;
6970b57cec5SDimitry Andric }
6980b57cec5SDimitry Andric 
6990b57cec5SDimitry Andric static void reportFastISelFailure(MachineFunction &MF,
7000b57cec5SDimitry Andric                                   OptimizationRemarkEmitter &ORE,
7010b57cec5SDimitry Andric                                   OptimizationRemarkMissed &R,
7020b57cec5SDimitry Andric                                   bool ShouldAbort) {
7030b57cec5SDimitry Andric   // Print the function name explicitly if we don't have a debug location (which
7040b57cec5SDimitry Andric   // makes the diagnostic less useful) or if we're going to emit a raw error.
7050b57cec5SDimitry Andric   if (!R.getLocation().isValid() || ShouldAbort)
7060b57cec5SDimitry Andric     R << (" (in function: " + MF.getName() + ")").str();
7070b57cec5SDimitry Andric 
7080b57cec5SDimitry Andric   if (ShouldAbort)
709349cc55cSDimitry Andric     report_fatal_error(Twine(R.getMsg()));
7100b57cec5SDimitry Andric 
7110b57cec5SDimitry Andric   ORE.emit(R);
71281ad6265SDimitry Andric   LLVM_DEBUG(dbgs() << R.getMsg() << "\n");
7130b57cec5SDimitry Andric }
7140b57cec5SDimitry Andric 
7150b57cec5SDimitry Andric void SelectionDAGISel::SelectBasicBlock(BasicBlock::const_iterator Begin,
7160b57cec5SDimitry Andric                                         BasicBlock::const_iterator End,
7170b57cec5SDimitry Andric                                         bool &HadTailCall) {
7180b57cec5SDimitry Andric   // Allow creating illegal types during DAG building for the basic block.
7190b57cec5SDimitry Andric   CurDAG->NewNodesMustHaveLegalTypes = false;
7200b57cec5SDimitry Andric 
7210b57cec5SDimitry Andric   // Lower the instructions. If a call is emitted as a tail call, cease emitting
722*5f757f3fSDimitry Andric   // nodes for this block. If an instruction is elided, don't emit it, but do
723*5f757f3fSDimitry Andric   // handle any debug-info attached to it.
7240b57cec5SDimitry Andric   for (BasicBlock::const_iterator I = Begin; I != End && !SDB->HasTailCall; ++I) {
7250b57cec5SDimitry Andric     if (!ElidedArgCopyInstrs.count(&*I))
7260b57cec5SDimitry Andric       SDB->visit(*I);
727*5f757f3fSDimitry Andric     else
728*5f757f3fSDimitry Andric       SDB->visitDbgInfo(*I);
7290b57cec5SDimitry Andric   }
7300b57cec5SDimitry Andric 
7310b57cec5SDimitry Andric   // Make sure the root of the DAG is up-to-date.
7320b57cec5SDimitry Andric   CurDAG->setRoot(SDB->getControlRoot());
7330b57cec5SDimitry Andric   HadTailCall = SDB->HasTailCall;
7340b57cec5SDimitry Andric   SDB->resolveOrClearDbgInfo();
7350b57cec5SDimitry Andric   SDB->clear();
7360b57cec5SDimitry Andric 
7370b57cec5SDimitry Andric   // Final step, emit the lowered DAG as machine code.
7380b57cec5SDimitry Andric   CodeGenAndEmitDAG();
7390b57cec5SDimitry Andric }
7400b57cec5SDimitry Andric 
7410b57cec5SDimitry Andric void SelectionDAGISel::ComputeLiveOutVRegInfo() {
742480093f4SDimitry Andric   SmallPtrSet<SDNode *, 16> Added;
7430b57cec5SDimitry Andric   SmallVector<SDNode*, 128> Worklist;
7440b57cec5SDimitry Andric 
7450b57cec5SDimitry Andric   Worklist.push_back(CurDAG->getRoot().getNode());
746480093f4SDimitry Andric   Added.insert(CurDAG->getRoot().getNode());
7470b57cec5SDimitry Andric 
7480b57cec5SDimitry Andric   KnownBits Known;
7490b57cec5SDimitry Andric 
7500b57cec5SDimitry Andric   do {
7510b57cec5SDimitry Andric     SDNode *N = Worklist.pop_back_val();
7520b57cec5SDimitry Andric 
7530b57cec5SDimitry Andric     // Otherwise, add all chain operands to the worklist.
7540b57cec5SDimitry Andric     for (const SDValue &Op : N->op_values())
755480093f4SDimitry Andric       if (Op.getValueType() == MVT::Other && Added.insert(Op.getNode()).second)
7560b57cec5SDimitry Andric         Worklist.push_back(Op.getNode());
7570b57cec5SDimitry Andric 
7580b57cec5SDimitry Andric     // If this is a CopyToReg with a vreg dest, process it.
7590b57cec5SDimitry Andric     if (N->getOpcode() != ISD::CopyToReg)
7600b57cec5SDimitry Andric       continue;
7610b57cec5SDimitry Andric 
7620b57cec5SDimitry Andric     unsigned DestReg = cast<RegisterSDNode>(N->getOperand(1))->getReg();
7638bcb0991SDimitry Andric     if (!Register::isVirtualRegister(DestReg))
7640b57cec5SDimitry Andric       continue;
7650b57cec5SDimitry Andric 
7660b57cec5SDimitry Andric     // Ignore non-integer values.
7670b57cec5SDimitry Andric     SDValue Src = N->getOperand(2);
7680b57cec5SDimitry Andric     EVT SrcVT = Src.getValueType();
7690b57cec5SDimitry Andric     if (!SrcVT.isInteger())
7700b57cec5SDimitry Andric       continue;
7710b57cec5SDimitry Andric 
7720b57cec5SDimitry Andric     unsigned NumSignBits = CurDAG->ComputeNumSignBits(Src);
7730b57cec5SDimitry Andric     Known = CurDAG->computeKnownBits(Src);
7740b57cec5SDimitry Andric     FuncInfo->AddLiveOutRegInfo(DestReg, NumSignBits, Known);
7750b57cec5SDimitry Andric   } while (!Worklist.empty());
7760b57cec5SDimitry Andric }
7770b57cec5SDimitry Andric 
7780b57cec5SDimitry Andric void SelectionDAGISel::CodeGenAndEmitDAG() {
7790b57cec5SDimitry Andric   StringRef GroupName = "sdag";
7800b57cec5SDimitry Andric   StringRef GroupDescription = "Instruction Selection and Scheduling";
7810b57cec5SDimitry Andric   std::string BlockName;
7820b57cec5SDimitry Andric   bool MatchFilterBB = false; (void)MatchFilterBB;
7830b57cec5SDimitry Andric #ifndef NDEBUG
7840b57cec5SDimitry Andric   TargetTransformInfo &TTI =
7850b57cec5SDimitry Andric       getAnalysis<TargetTransformInfoWrapperPass>().getTTI(*FuncInfo->Fn);
7860b57cec5SDimitry Andric #endif
7870b57cec5SDimitry Andric 
7880b57cec5SDimitry Andric   // Pre-type legalization allow creation of any node types.
7890b57cec5SDimitry Andric   CurDAG->NewNodesMustHaveLegalTypes = false;
7900b57cec5SDimitry Andric 
7910b57cec5SDimitry Andric #ifndef NDEBUG
7920b57cec5SDimitry Andric   MatchFilterBB = (FilterDAGBasicBlockName.empty() ||
7930b57cec5SDimitry Andric                    FilterDAGBasicBlockName ==
7940b57cec5SDimitry Andric                        FuncInfo->MBB->getBasicBlock()->getName());
7950b57cec5SDimitry Andric #endif
7960b57cec5SDimitry Andric #ifdef NDEBUG
797480093f4SDimitry Andric   if (ViewDAGCombine1 || ViewLegalizeTypesDAGs || ViewDAGCombineLT ||
798480093f4SDimitry Andric       ViewLegalizeDAGs || ViewDAGCombine2 || ViewISelDAGs || ViewSchedDAGs ||
7990b57cec5SDimitry Andric       ViewSUnitDAGs)
8000b57cec5SDimitry Andric #endif
8010b57cec5SDimitry Andric   {
8020b57cec5SDimitry Andric     BlockName =
8030b57cec5SDimitry Andric         (MF->getName() + ":" + FuncInfo->MBB->getBasicBlock()->getName()).str();
8040b57cec5SDimitry Andric   }
805*5f757f3fSDimitry Andric   ISEL_DUMP(dbgs() << "\nInitial selection DAG: "
8060b57cec5SDimitry Andric                    << printMBBReference(*FuncInfo->MBB) << " '" << BlockName
8070b57cec5SDimitry Andric                    << "'\n";
8080b57cec5SDimitry Andric             CurDAG->dump());
8090b57cec5SDimitry Andric 
810e8d8bef9SDimitry Andric #ifndef NDEBUG
811e8d8bef9SDimitry Andric   if (TTI.hasBranchDivergence())
812349cc55cSDimitry Andric     CurDAG->VerifyDAGDivergence();
813e8d8bef9SDimitry Andric #endif
814e8d8bef9SDimitry Andric 
8150b57cec5SDimitry Andric   if (ViewDAGCombine1 && MatchFilterBB)
8160b57cec5SDimitry Andric     CurDAG->viewGraph("dag-combine1 input for " + BlockName);
8170b57cec5SDimitry Andric 
8180b57cec5SDimitry Andric   // Run the DAG combiner in pre-legalize mode.
8190b57cec5SDimitry Andric   {
8200b57cec5SDimitry Andric     NamedRegionTimer T("combine1", "DAG Combining 1", GroupName,
8210b57cec5SDimitry Andric                        GroupDescription, TimePassesIsEnabled);
8220b57cec5SDimitry Andric     CurDAG->Combine(BeforeLegalizeTypes, AA, OptLevel);
8230b57cec5SDimitry Andric   }
8240b57cec5SDimitry Andric 
825*5f757f3fSDimitry Andric   ISEL_DUMP(dbgs() << "\nOptimized lowered selection DAG: "
8260b57cec5SDimitry Andric                    << printMBBReference(*FuncInfo->MBB) << " '" << BlockName
8270b57cec5SDimitry Andric                    << "'\n";
8280b57cec5SDimitry Andric             CurDAG->dump());
8290b57cec5SDimitry Andric 
830e8d8bef9SDimitry Andric #ifndef NDEBUG
831e8d8bef9SDimitry Andric   if (TTI.hasBranchDivergence())
832349cc55cSDimitry Andric     CurDAG->VerifyDAGDivergence();
833e8d8bef9SDimitry Andric #endif
834e8d8bef9SDimitry Andric 
8350b57cec5SDimitry Andric   // Second step, hack on the DAG until it only uses operations and types that
8360b57cec5SDimitry Andric   // the target supports.
8370b57cec5SDimitry Andric   if (ViewLegalizeTypesDAGs && MatchFilterBB)
8380b57cec5SDimitry Andric     CurDAG->viewGraph("legalize-types input for " + BlockName);
8390b57cec5SDimitry Andric 
8400b57cec5SDimitry Andric   bool Changed;
8410b57cec5SDimitry Andric   {
8420b57cec5SDimitry Andric     NamedRegionTimer T("legalize_types", "Type Legalization", GroupName,
8430b57cec5SDimitry Andric                        GroupDescription, TimePassesIsEnabled);
8440b57cec5SDimitry Andric     Changed = CurDAG->LegalizeTypes();
8450b57cec5SDimitry Andric   }
8460b57cec5SDimitry Andric 
847*5f757f3fSDimitry Andric   ISEL_DUMP(dbgs() << "\nType-legalized selection DAG: "
8480b57cec5SDimitry Andric                    << printMBBReference(*FuncInfo->MBB) << " '" << BlockName
8490b57cec5SDimitry Andric                    << "'\n";
8500b57cec5SDimitry Andric             CurDAG->dump());
8510b57cec5SDimitry Andric 
852e8d8bef9SDimitry Andric #ifndef NDEBUG
853e8d8bef9SDimitry Andric   if (TTI.hasBranchDivergence())
854349cc55cSDimitry Andric     CurDAG->VerifyDAGDivergence();
855e8d8bef9SDimitry Andric #endif
856e8d8bef9SDimitry Andric 
8570b57cec5SDimitry Andric   // Only allow creation of legal node types.
8580b57cec5SDimitry Andric   CurDAG->NewNodesMustHaveLegalTypes = true;
8590b57cec5SDimitry Andric 
8600b57cec5SDimitry Andric   if (Changed) {
8610b57cec5SDimitry Andric     if (ViewDAGCombineLT && MatchFilterBB)
8620b57cec5SDimitry Andric       CurDAG->viewGraph("dag-combine-lt input for " + BlockName);
8630b57cec5SDimitry Andric 
8640b57cec5SDimitry Andric     // Run the DAG combiner in post-type-legalize mode.
8650b57cec5SDimitry Andric     {
8660b57cec5SDimitry Andric       NamedRegionTimer T("combine_lt", "DAG Combining after legalize types",
8670b57cec5SDimitry Andric                          GroupName, GroupDescription, TimePassesIsEnabled);
8680b57cec5SDimitry Andric       CurDAG->Combine(AfterLegalizeTypes, AA, OptLevel);
8690b57cec5SDimitry Andric     }
8700b57cec5SDimitry Andric 
871*5f757f3fSDimitry Andric     ISEL_DUMP(dbgs() << "\nOptimized type-legalized selection DAG: "
8720b57cec5SDimitry Andric                      << printMBBReference(*FuncInfo->MBB) << " '" << BlockName
8730b57cec5SDimitry Andric                      << "'\n";
8740b57cec5SDimitry Andric               CurDAG->dump());
875e8d8bef9SDimitry Andric 
876e8d8bef9SDimitry Andric #ifndef NDEBUG
877e8d8bef9SDimitry Andric     if (TTI.hasBranchDivergence())
878349cc55cSDimitry Andric       CurDAG->VerifyDAGDivergence();
879e8d8bef9SDimitry Andric #endif
8800b57cec5SDimitry Andric   }
8810b57cec5SDimitry Andric 
8820b57cec5SDimitry Andric   {
8830b57cec5SDimitry Andric     NamedRegionTimer T("legalize_vec", "Vector Legalization", GroupName,
8840b57cec5SDimitry Andric                        GroupDescription, TimePassesIsEnabled);
8850b57cec5SDimitry Andric     Changed = CurDAG->LegalizeVectors();
8860b57cec5SDimitry Andric   }
8870b57cec5SDimitry Andric 
8880b57cec5SDimitry Andric   if (Changed) {
889*5f757f3fSDimitry Andric     ISEL_DUMP(dbgs() << "\nVector-legalized selection DAG: "
8900b57cec5SDimitry Andric                      << printMBBReference(*FuncInfo->MBB) << " '" << BlockName
8910b57cec5SDimitry Andric                      << "'\n";
8920b57cec5SDimitry Andric               CurDAG->dump());
8930b57cec5SDimitry Andric 
894e8d8bef9SDimitry Andric #ifndef NDEBUG
895e8d8bef9SDimitry Andric     if (TTI.hasBranchDivergence())
896349cc55cSDimitry Andric       CurDAG->VerifyDAGDivergence();
897e8d8bef9SDimitry Andric #endif
898e8d8bef9SDimitry Andric 
8990b57cec5SDimitry Andric     {
9000b57cec5SDimitry Andric       NamedRegionTimer T("legalize_types2", "Type Legalization 2", GroupName,
9010b57cec5SDimitry Andric                          GroupDescription, TimePassesIsEnabled);
9020b57cec5SDimitry Andric       CurDAG->LegalizeTypes();
9030b57cec5SDimitry Andric     }
9040b57cec5SDimitry Andric 
905*5f757f3fSDimitry Andric     ISEL_DUMP(dbgs() << "\nVector/type-legalized selection DAG: "
9060b57cec5SDimitry Andric                      << printMBBReference(*FuncInfo->MBB) << " '" << BlockName
9070b57cec5SDimitry Andric                      << "'\n";
9080b57cec5SDimitry Andric               CurDAG->dump());
9090b57cec5SDimitry Andric 
910e8d8bef9SDimitry Andric #ifndef NDEBUG
911e8d8bef9SDimitry Andric     if (TTI.hasBranchDivergence())
912349cc55cSDimitry Andric       CurDAG->VerifyDAGDivergence();
913e8d8bef9SDimitry Andric #endif
914e8d8bef9SDimitry Andric 
9150b57cec5SDimitry Andric     if (ViewDAGCombineLT && MatchFilterBB)
9160b57cec5SDimitry Andric       CurDAG->viewGraph("dag-combine-lv input for " + BlockName);
9170b57cec5SDimitry Andric 
9180b57cec5SDimitry Andric     // Run the DAG combiner in post-type-legalize mode.
9190b57cec5SDimitry Andric     {
9200b57cec5SDimitry Andric       NamedRegionTimer T("combine_lv", "DAG Combining after legalize vectors",
9210b57cec5SDimitry Andric                          GroupName, GroupDescription, TimePassesIsEnabled);
9220b57cec5SDimitry Andric       CurDAG->Combine(AfterLegalizeVectorOps, AA, OptLevel);
9230b57cec5SDimitry Andric     }
9240b57cec5SDimitry Andric 
925*5f757f3fSDimitry Andric     ISEL_DUMP(dbgs() << "\nOptimized vector-legalized selection DAG: "
9260b57cec5SDimitry Andric                      << printMBBReference(*FuncInfo->MBB) << " '" << BlockName
9270b57cec5SDimitry Andric                      << "'\n";
9280b57cec5SDimitry Andric               CurDAG->dump());
9290b57cec5SDimitry Andric 
9300b57cec5SDimitry Andric #ifndef NDEBUG
9310b57cec5SDimitry Andric     if (TTI.hasBranchDivergence())
932349cc55cSDimitry Andric       CurDAG->VerifyDAGDivergence();
9330b57cec5SDimitry Andric #endif
9340b57cec5SDimitry Andric   }
9350b57cec5SDimitry Andric 
9360b57cec5SDimitry Andric   if (ViewLegalizeDAGs && MatchFilterBB)
9370b57cec5SDimitry Andric     CurDAG->viewGraph("legalize input for " + BlockName);
9380b57cec5SDimitry Andric 
9390b57cec5SDimitry Andric   {
9400b57cec5SDimitry Andric     NamedRegionTimer T("legalize", "DAG Legalization", GroupName,
9410b57cec5SDimitry Andric                        GroupDescription, TimePassesIsEnabled);
9420b57cec5SDimitry Andric     CurDAG->Legalize();
9430b57cec5SDimitry Andric   }
9440b57cec5SDimitry Andric 
945*5f757f3fSDimitry Andric   ISEL_DUMP(dbgs() << "\nLegalized selection DAG: "
9460b57cec5SDimitry Andric                    << printMBBReference(*FuncInfo->MBB) << " '" << BlockName
9470b57cec5SDimitry Andric                    << "'\n";
9480b57cec5SDimitry Andric             CurDAG->dump());
9490b57cec5SDimitry Andric 
950e8d8bef9SDimitry Andric #ifndef NDEBUG
951e8d8bef9SDimitry Andric   if (TTI.hasBranchDivergence())
952349cc55cSDimitry Andric     CurDAG->VerifyDAGDivergence();
953e8d8bef9SDimitry Andric #endif
954e8d8bef9SDimitry Andric 
9550b57cec5SDimitry Andric   if (ViewDAGCombine2 && MatchFilterBB)
9560b57cec5SDimitry Andric     CurDAG->viewGraph("dag-combine2 input for " + BlockName);
9570b57cec5SDimitry Andric 
9580b57cec5SDimitry Andric   // Run the DAG combiner in post-legalize mode.
9590b57cec5SDimitry Andric   {
9600b57cec5SDimitry Andric     NamedRegionTimer T("combine2", "DAG Combining 2", GroupName,
9610b57cec5SDimitry Andric                        GroupDescription, TimePassesIsEnabled);
9620b57cec5SDimitry Andric     CurDAG->Combine(AfterLegalizeDAG, AA, OptLevel);
9630b57cec5SDimitry Andric   }
9640b57cec5SDimitry Andric 
965*5f757f3fSDimitry Andric   ISEL_DUMP(dbgs() << "\nOptimized legalized selection DAG: "
9660b57cec5SDimitry Andric                    << printMBBReference(*FuncInfo->MBB) << " '" << BlockName
9670b57cec5SDimitry Andric                    << "'\n";
9680b57cec5SDimitry Andric             CurDAG->dump());
9690b57cec5SDimitry Andric 
970e8d8bef9SDimitry Andric #ifndef NDEBUG
971e8d8bef9SDimitry Andric   if (TTI.hasBranchDivergence())
972349cc55cSDimitry Andric     CurDAG->VerifyDAGDivergence();
973e8d8bef9SDimitry Andric #endif
974e8d8bef9SDimitry Andric 
975*5f757f3fSDimitry Andric   if (OptLevel != CodeGenOptLevel::None)
9760b57cec5SDimitry Andric     ComputeLiveOutVRegInfo();
9770b57cec5SDimitry Andric 
9780b57cec5SDimitry Andric   if (ViewISelDAGs && MatchFilterBB)
9790b57cec5SDimitry Andric     CurDAG->viewGraph("isel input for " + BlockName);
9800b57cec5SDimitry Andric 
9810b57cec5SDimitry Andric   // Third, instruction select all of the operations to machine code, adding the
9820b57cec5SDimitry Andric   // code to the MachineBasicBlock.
9830b57cec5SDimitry Andric   {
9840b57cec5SDimitry Andric     NamedRegionTimer T("isel", "Instruction Selection", GroupName,
9850b57cec5SDimitry Andric                        GroupDescription, TimePassesIsEnabled);
9860b57cec5SDimitry Andric     DoInstructionSelection();
9870b57cec5SDimitry Andric   }
9880b57cec5SDimitry Andric 
989*5f757f3fSDimitry Andric   ISEL_DUMP(dbgs() << "\nSelected selection DAG: "
9900b57cec5SDimitry Andric                    << printMBBReference(*FuncInfo->MBB) << " '" << BlockName
9910b57cec5SDimitry Andric                    << "'\n";
9920b57cec5SDimitry Andric             CurDAG->dump());
9930b57cec5SDimitry Andric 
9940b57cec5SDimitry Andric   if (ViewSchedDAGs && MatchFilterBB)
9950b57cec5SDimitry Andric     CurDAG->viewGraph("scheduler input for " + BlockName);
9960b57cec5SDimitry Andric 
9970b57cec5SDimitry Andric   // Schedule machine code.
9980b57cec5SDimitry Andric   ScheduleDAGSDNodes *Scheduler = CreateScheduler();
9990b57cec5SDimitry Andric   {
10000b57cec5SDimitry Andric     NamedRegionTimer T("sched", "Instruction Scheduling", GroupName,
10010b57cec5SDimitry Andric                        GroupDescription, TimePassesIsEnabled);
10020b57cec5SDimitry Andric     Scheduler->Run(CurDAG, FuncInfo->MBB);
10030b57cec5SDimitry Andric   }
10040b57cec5SDimitry Andric 
10050b57cec5SDimitry Andric   if (ViewSUnitDAGs && MatchFilterBB)
10060b57cec5SDimitry Andric     Scheduler->viewGraph();
10070b57cec5SDimitry Andric 
10080b57cec5SDimitry Andric   // Emit machine code to BB.  This can change 'BB' to the last block being
10090b57cec5SDimitry Andric   // inserted into.
10100b57cec5SDimitry Andric   MachineBasicBlock *FirstMBB = FuncInfo->MBB, *LastMBB;
10110b57cec5SDimitry Andric   {
10120b57cec5SDimitry Andric     NamedRegionTimer T("emit", "Instruction Creation", GroupName,
10130b57cec5SDimitry Andric                        GroupDescription, TimePassesIsEnabled);
10140b57cec5SDimitry Andric 
10150b57cec5SDimitry Andric     // FuncInfo->InsertPt is passed by reference and set to the end of the
10160b57cec5SDimitry Andric     // scheduled instructions.
10170b57cec5SDimitry Andric     LastMBB = FuncInfo->MBB = Scheduler->EmitSchedule(FuncInfo->InsertPt);
10180b57cec5SDimitry Andric   }
10190b57cec5SDimitry Andric 
10200b57cec5SDimitry Andric   // If the block was split, make sure we update any references that are used to
10210b57cec5SDimitry Andric   // update PHI nodes later on.
10220b57cec5SDimitry Andric   if (FirstMBB != LastMBB)
10230b57cec5SDimitry Andric     SDB->UpdateSplitBlock(FirstMBB, LastMBB);
10240b57cec5SDimitry Andric 
10250b57cec5SDimitry Andric   // Free the scheduler state.
10260b57cec5SDimitry Andric   {
10270b57cec5SDimitry Andric     NamedRegionTimer T("cleanup", "Instruction Scheduling Cleanup", GroupName,
10280b57cec5SDimitry Andric                        GroupDescription, TimePassesIsEnabled);
10290b57cec5SDimitry Andric     delete Scheduler;
10300b57cec5SDimitry Andric   }
10310b57cec5SDimitry Andric 
10320b57cec5SDimitry Andric   // Free the SelectionDAG state, now that we're finished with it.
10330b57cec5SDimitry Andric   CurDAG->clear();
10340b57cec5SDimitry Andric }
10350b57cec5SDimitry Andric 
10360b57cec5SDimitry Andric namespace {
10370b57cec5SDimitry Andric 
10380b57cec5SDimitry Andric /// ISelUpdater - helper class to handle updates of the instruction selection
10390b57cec5SDimitry Andric /// graph.
10400b57cec5SDimitry Andric class ISelUpdater : public SelectionDAG::DAGUpdateListener {
10410b57cec5SDimitry Andric   SelectionDAG::allnodes_iterator &ISelPosition;
10420b57cec5SDimitry Andric 
10430b57cec5SDimitry Andric public:
10440b57cec5SDimitry Andric   ISelUpdater(SelectionDAG &DAG, SelectionDAG::allnodes_iterator &isp)
10450b57cec5SDimitry Andric     : SelectionDAG::DAGUpdateListener(DAG), ISelPosition(isp) {}
10460b57cec5SDimitry Andric 
10470b57cec5SDimitry Andric   /// NodeDeleted - Handle nodes deleted from the graph. If the node being
10480b57cec5SDimitry Andric   /// deleted is the current ISelPosition node, update ISelPosition.
10490b57cec5SDimitry Andric   ///
10500b57cec5SDimitry Andric   void NodeDeleted(SDNode *N, SDNode *E) override {
10510b57cec5SDimitry Andric     if (ISelPosition == SelectionDAG::allnodes_iterator(N))
10520b57cec5SDimitry Andric       ++ISelPosition;
10530b57cec5SDimitry Andric   }
1054bdd1243dSDimitry Andric 
1055bdd1243dSDimitry Andric   /// NodeInserted - Handle new nodes inserted into the graph: propagate
1056bdd1243dSDimitry Andric   /// metadata from root nodes that also applies to new nodes, in case the root
1057bdd1243dSDimitry Andric   /// is later deleted.
1058bdd1243dSDimitry Andric   void NodeInserted(SDNode *N) override {
1059bdd1243dSDimitry Andric     SDNode *CurNode = &*ISelPosition;
1060bdd1243dSDimitry Andric     if (MDNode *MD = DAG.getPCSections(CurNode))
1061bdd1243dSDimitry Andric       DAG.addPCSections(N, MD);
1062bdd1243dSDimitry Andric   }
10630b57cec5SDimitry Andric };
10640b57cec5SDimitry Andric 
10650b57cec5SDimitry Andric } // end anonymous namespace
10660b57cec5SDimitry Andric 
10670b57cec5SDimitry Andric // This function is used to enforce the topological node id property
1068349cc55cSDimitry Andric // leveraged during instruction selection. Before the selection process all
1069349cc55cSDimitry Andric // nodes are given a non-negative id such that all nodes have a greater id than
10700b57cec5SDimitry Andric // their operands. As this holds transitively we can prune checks that a node N
10710b57cec5SDimitry Andric // is a predecessor of M another by not recursively checking through M's
1072349cc55cSDimitry Andric // operands if N's ID is larger than M's ID. This significantly improves
1073349cc55cSDimitry Andric // performance of various legality checks (e.g. IsLegalToFold / UpdateChains).
10740b57cec5SDimitry Andric 
1075349cc55cSDimitry Andric // However, when we fuse multiple nodes into a single node during the
1076349cc55cSDimitry Andric // selection we may induce a predecessor relationship between inputs and
1077349cc55cSDimitry Andric // outputs of distinct nodes being merged, violating the topological property.
1078349cc55cSDimitry Andric // Should a fused node have a successor which has yet to be selected,
1079349cc55cSDimitry Andric // our legality checks would be incorrect. To avoid this we mark all unselected
1080349cc55cSDimitry Andric // successor nodes, i.e. id != -1, as invalid for pruning by bit-negating (x =>
10810b57cec5SDimitry Andric // (-(x+1))) the ids and modify our pruning check to ignore negative Ids of M.
10820b57cec5SDimitry Andric // We use bit-negation to more clearly enforce that node id -1 can only be
1083349cc55cSDimitry Andric // achieved by selected nodes. As the conversion is reversable to the original
1084349cc55cSDimitry Andric // Id, topological pruning can still be leveraged when looking for unselected
1085349cc55cSDimitry Andric // nodes. This method is called internally in all ISel replacement related
1086349cc55cSDimitry Andric // functions.
10870b57cec5SDimitry Andric void SelectionDAGISel::EnforceNodeIdInvariant(SDNode *Node) {
10880b57cec5SDimitry Andric   SmallVector<SDNode *, 4> Nodes;
10890b57cec5SDimitry Andric   Nodes.push_back(Node);
10900b57cec5SDimitry Andric 
10910b57cec5SDimitry Andric   while (!Nodes.empty()) {
10920b57cec5SDimitry Andric     SDNode *N = Nodes.pop_back_val();
10930b57cec5SDimitry Andric     for (auto *U : N->uses()) {
10940b57cec5SDimitry Andric       auto UId = U->getNodeId();
10950b57cec5SDimitry Andric       if (UId > 0) {
10960b57cec5SDimitry Andric         InvalidateNodeId(U);
10970b57cec5SDimitry Andric         Nodes.push_back(U);
10980b57cec5SDimitry Andric       }
10990b57cec5SDimitry Andric     }
11000b57cec5SDimitry Andric   }
11010b57cec5SDimitry Andric }
11020b57cec5SDimitry Andric 
1103349cc55cSDimitry Andric // InvalidateNodeId - As explained in EnforceNodeIdInvariant, mark a
11040b57cec5SDimitry Andric // NodeId with the equivalent node id which is invalid for topological
11050b57cec5SDimitry Andric // pruning.
11060b57cec5SDimitry Andric void SelectionDAGISel::InvalidateNodeId(SDNode *N) {
11070b57cec5SDimitry Andric   int InvalidId = -(N->getNodeId() + 1);
11080b57cec5SDimitry Andric   N->setNodeId(InvalidId);
11090b57cec5SDimitry Andric }
11100b57cec5SDimitry Andric 
11110b57cec5SDimitry Andric // getUninvalidatedNodeId - get original uninvalidated node id.
11120b57cec5SDimitry Andric int SelectionDAGISel::getUninvalidatedNodeId(SDNode *N) {
11130b57cec5SDimitry Andric   int Id = N->getNodeId();
11140b57cec5SDimitry Andric   if (Id < -1)
11150b57cec5SDimitry Andric     return -(Id + 1);
11160b57cec5SDimitry Andric   return Id;
11170b57cec5SDimitry Andric }
11180b57cec5SDimitry Andric 
11190b57cec5SDimitry Andric void SelectionDAGISel::DoInstructionSelection() {
11200b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "===== Instruction selection begins: "
11210b57cec5SDimitry Andric                     << printMBBReference(*FuncInfo->MBB) << " '"
11220b57cec5SDimitry Andric                     << FuncInfo->MBB->getName() << "'\n");
11230b57cec5SDimitry Andric 
11240b57cec5SDimitry Andric   PreprocessISelDAG();
11250b57cec5SDimitry Andric 
11260b57cec5SDimitry Andric   // Select target instructions for the DAG.
11270b57cec5SDimitry Andric   {
11280b57cec5SDimitry Andric     // Number all nodes with a topological order and set DAGSize.
11290b57cec5SDimitry Andric     DAGSize = CurDAG->AssignTopologicalOrder();
11300b57cec5SDimitry Andric 
11310b57cec5SDimitry Andric     // Create a dummy node (which is not added to allnodes), that adds
11320b57cec5SDimitry Andric     // a reference to the root node, preventing it from being deleted,
11330b57cec5SDimitry Andric     // and tracking any changes of the root.
11340b57cec5SDimitry Andric     HandleSDNode Dummy(CurDAG->getRoot());
11350b57cec5SDimitry Andric     SelectionDAG::allnodes_iterator ISelPosition (CurDAG->getRoot().getNode());
11360b57cec5SDimitry Andric     ++ISelPosition;
11370b57cec5SDimitry Andric 
11380b57cec5SDimitry Andric     // Make sure that ISelPosition gets properly updated when nodes are deleted
1139bdd1243dSDimitry Andric     // in calls made from this function. New nodes inherit relevant metadata.
11400b57cec5SDimitry Andric     ISelUpdater ISU(*CurDAG, ISelPosition);
11410b57cec5SDimitry Andric 
11420b57cec5SDimitry Andric     // The AllNodes list is now topological-sorted. Visit the
11430b57cec5SDimitry Andric     // nodes by starting at the end of the list (the root of the
11440b57cec5SDimitry Andric     // graph) and preceding back toward the beginning (the entry
11450b57cec5SDimitry Andric     // node).
11460b57cec5SDimitry Andric     while (ISelPosition != CurDAG->allnodes_begin()) {
11470b57cec5SDimitry Andric       SDNode *Node = &*--ISelPosition;
11480b57cec5SDimitry Andric       // Skip dead nodes. DAGCombiner is expected to eliminate all dead nodes,
11490b57cec5SDimitry Andric       // but there are currently some corner cases that it misses. Also, this
11500b57cec5SDimitry Andric       // makes it theoretically possible to disable the DAGCombiner.
11510b57cec5SDimitry Andric       if (Node->use_empty())
11520b57cec5SDimitry Andric         continue;
11530b57cec5SDimitry Andric 
11540b57cec5SDimitry Andric #ifndef NDEBUG
11550b57cec5SDimitry Andric       SmallVector<SDNode *, 4> Nodes;
11560b57cec5SDimitry Andric       Nodes.push_back(Node);
11570b57cec5SDimitry Andric 
11580b57cec5SDimitry Andric       while (!Nodes.empty()) {
11590b57cec5SDimitry Andric         auto N = Nodes.pop_back_val();
11600b57cec5SDimitry Andric         if (N->getOpcode() == ISD::TokenFactor || N->getNodeId() < 0)
11610b57cec5SDimitry Andric           continue;
11620b57cec5SDimitry Andric         for (const SDValue &Op : N->op_values()) {
11630b57cec5SDimitry Andric           if (Op->getOpcode() == ISD::TokenFactor)
11640b57cec5SDimitry Andric             Nodes.push_back(Op.getNode());
11650b57cec5SDimitry Andric           else {
11660b57cec5SDimitry Andric             // We rely on topological ordering of node ids for checking for
11670b57cec5SDimitry Andric             // cycles when fusing nodes during selection. All unselected nodes
11680b57cec5SDimitry Andric             // successors of an already selected node should have a negative id.
11690b57cec5SDimitry Andric             // This assertion will catch such cases. If this assertion triggers
11700b57cec5SDimitry Andric             // it is likely you using DAG-level Value/Node replacement functions
11710b57cec5SDimitry Andric             // (versus equivalent ISEL replacement) in backend-specific
11720b57cec5SDimitry Andric             // selections. See comment in EnforceNodeIdInvariant for more
11730b57cec5SDimitry Andric             // details.
11740b57cec5SDimitry Andric             assert(Op->getNodeId() != -1 &&
11750b57cec5SDimitry Andric                    "Node has already selected predecessor node");
11760b57cec5SDimitry Andric           }
11770b57cec5SDimitry Andric         }
11780b57cec5SDimitry Andric       }
11790b57cec5SDimitry Andric #endif
11800b57cec5SDimitry Andric 
11810b57cec5SDimitry Andric       // When we are using non-default rounding modes or FP exception behavior
11820b57cec5SDimitry Andric       // FP operations are represented by StrictFP pseudo-operations.  For
11830b57cec5SDimitry Andric       // targets that do not (yet) understand strict FP operations directly,
11840b57cec5SDimitry Andric       // we convert them to normal FP opcodes instead at this point.  This
11850b57cec5SDimitry Andric       // will allow them to be handled by existing target-specific instruction
11860b57cec5SDimitry Andric       // selectors.
1187480093f4SDimitry Andric       if (!TLI->isStrictFPEnabled() && Node->isStrictFPOpcode()) {
1188480093f4SDimitry Andric         // For some opcodes, we need to call TLI->getOperationAction using
1189480093f4SDimitry Andric         // the first operand type instead of the result type.  Note that this
1190480093f4SDimitry Andric         // must match what SelectionDAGLegalize::LegalizeOp is doing.
1191480093f4SDimitry Andric         EVT ActionVT;
1192480093f4SDimitry Andric         switch (Node->getOpcode()) {
1193480093f4SDimitry Andric         case ISD::STRICT_SINT_TO_FP:
1194480093f4SDimitry Andric         case ISD::STRICT_UINT_TO_FP:
1195480093f4SDimitry Andric         case ISD::STRICT_LRINT:
1196480093f4SDimitry Andric         case ISD::STRICT_LLRINT:
1197480093f4SDimitry Andric         case ISD::STRICT_LROUND:
1198480093f4SDimitry Andric         case ISD::STRICT_LLROUND:
1199480093f4SDimitry Andric         case ISD::STRICT_FSETCC:
1200480093f4SDimitry Andric         case ISD::STRICT_FSETCCS:
1201480093f4SDimitry Andric           ActionVT = Node->getOperand(1).getValueType();
1202480093f4SDimitry Andric           break;
1203480093f4SDimitry Andric         default:
1204480093f4SDimitry Andric           ActionVT = Node->getValueType(0);
1205480093f4SDimitry Andric           break;
1206480093f4SDimitry Andric         }
1207480093f4SDimitry Andric         if (TLI->getOperationAction(Node->getOpcode(), ActionVT)
1208480093f4SDimitry Andric             == TargetLowering::Expand)
12090b57cec5SDimitry Andric           Node = CurDAG->mutateStrictFPToFP(Node);
1210480093f4SDimitry Andric       }
12110b57cec5SDimitry Andric 
12120b57cec5SDimitry Andric       LLVM_DEBUG(dbgs() << "\nISEL: Starting selection on root node: ";
12130b57cec5SDimitry Andric                  Node->dump(CurDAG));
12140b57cec5SDimitry Andric 
12150b57cec5SDimitry Andric       Select(Node);
12160b57cec5SDimitry Andric     }
12170b57cec5SDimitry Andric 
12180b57cec5SDimitry Andric     CurDAG->setRoot(Dummy.getValue());
12190b57cec5SDimitry Andric   }
12200b57cec5SDimitry Andric 
12210b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "\n===== Instruction selection ends:\n");
12220b57cec5SDimitry Andric 
12230b57cec5SDimitry Andric   PostprocessISelDAG();
12240b57cec5SDimitry Andric }
12250b57cec5SDimitry Andric 
12260b57cec5SDimitry Andric static bool hasExceptionPointerOrCodeUser(const CatchPadInst *CPI) {
12270b57cec5SDimitry Andric   for (const User *U : CPI->users()) {
12280b57cec5SDimitry Andric     if (const IntrinsicInst *EHPtrCall = dyn_cast<IntrinsicInst>(U)) {
12290b57cec5SDimitry Andric       Intrinsic::ID IID = EHPtrCall->getIntrinsicID();
12300b57cec5SDimitry Andric       if (IID == Intrinsic::eh_exceptionpointer ||
12310b57cec5SDimitry Andric           IID == Intrinsic::eh_exceptioncode)
12320b57cec5SDimitry Andric         return true;
12330b57cec5SDimitry Andric     }
12340b57cec5SDimitry Andric   }
12350b57cec5SDimitry Andric   return false;
12360b57cec5SDimitry Andric }
12370b57cec5SDimitry Andric 
12380b57cec5SDimitry Andric // wasm.landingpad.index intrinsic is for associating a landing pad index number
12390b57cec5SDimitry Andric // with a catchpad instruction. Retrieve the landing pad index in the intrinsic
12400b57cec5SDimitry Andric // and store the mapping in the function.
12410b57cec5SDimitry Andric static void mapWasmLandingPadIndex(MachineBasicBlock *MBB,
12420b57cec5SDimitry Andric                                    const CatchPadInst *CPI) {
12430b57cec5SDimitry Andric   MachineFunction *MF = MBB->getParent();
12440b57cec5SDimitry Andric   // In case of single catch (...), we don't emit LSDA, so we don't need
12450b57cec5SDimitry Andric   // this information.
12460b57cec5SDimitry Andric   bool IsSingleCatchAllClause =
1247bdd1243dSDimitry Andric       CPI->arg_size() == 1 &&
12480b57cec5SDimitry Andric       cast<Constant>(CPI->getArgOperand(0))->isNullValue();
1249349cc55cSDimitry Andric   // cathchpads for longjmp use an empty type list, e.g. catchpad within %0 []
1250349cc55cSDimitry Andric   // and they don't need LSDA info
1251bdd1243dSDimitry Andric   bool IsCatchLongjmp = CPI->arg_size() == 0;
1252349cc55cSDimitry Andric   if (!IsSingleCatchAllClause && !IsCatchLongjmp) {
12530b57cec5SDimitry Andric     // Create a mapping from landing pad label to landing pad index.
12540b57cec5SDimitry Andric     bool IntrFound = false;
12550b57cec5SDimitry Andric     for (const User *U : CPI->users()) {
12560b57cec5SDimitry Andric       if (const auto *Call = dyn_cast<IntrinsicInst>(U)) {
12570b57cec5SDimitry Andric         Intrinsic::ID IID = Call->getIntrinsicID();
12580b57cec5SDimitry Andric         if (IID == Intrinsic::wasm_landingpad_index) {
12590b57cec5SDimitry Andric           Value *IndexArg = Call->getArgOperand(1);
12600b57cec5SDimitry Andric           int Index = cast<ConstantInt>(IndexArg)->getZExtValue();
12610b57cec5SDimitry Andric           MF->setWasmLandingPadIndex(MBB, Index);
12620b57cec5SDimitry Andric           IntrFound = true;
12630b57cec5SDimitry Andric           break;
12640b57cec5SDimitry Andric         }
12650b57cec5SDimitry Andric       }
12660b57cec5SDimitry Andric     }
12670b57cec5SDimitry Andric     assert(IntrFound && "wasm.landingpad.index intrinsic not found!");
12680b57cec5SDimitry Andric     (void)IntrFound;
12690b57cec5SDimitry Andric   }
12700b57cec5SDimitry Andric }
12710b57cec5SDimitry Andric 
12720b57cec5SDimitry Andric /// PrepareEHLandingPad - Emit an EH_LABEL, set up live-in registers, and
12730b57cec5SDimitry Andric /// do other setup for EH landing-pad blocks.
12740b57cec5SDimitry Andric bool SelectionDAGISel::PrepareEHLandingPad() {
12750b57cec5SDimitry Andric   MachineBasicBlock *MBB = FuncInfo->MBB;
12760b57cec5SDimitry Andric   const Constant *PersonalityFn = FuncInfo->Fn->getPersonalityFn();
12770b57cec5SDimitry Andric   const BasicBlock *LLVMBB = MBB->getBasicBlock();
12780b57cec5SDimitry Andric   const TargetRegisterClass *PtrRC =
12790b57cec5SDimitry Andric       TLI->getRegClassFor(TLI->getPointerTy(CurDAG->getDataLayout()));
12800b57cec5SDimitry Andric 
12810b57cec5SDimitry Andric   auto Pers = classifyEHPersonality(PersonalityFn);
12820b57cec5SDimitry Andric 
12830b57cec5SDimitry Andric   // Catchpads have one live-in register, which typically holds the exception
12840b57cec5SDimitry Andric   // pointer or code.
12850b57cec5SDimitry Andric   if (isFuncletEHPersonality(Pers)) {
12860b57cec5SDimitry Andric     if (const auto *CPI = dyn_cast<CatchPadInst>(LLVMBB->getFirstNonPHI())) {
12870b57cec5SDimitry Andric       if (hasExceptionPointerOrCodeUser(CPI)) {
12880b57cec5SDimitry Andric         // Get or create the virtual register to hold the pointer or code.  Mark
12890b57cec5SDimitry Andric         // the live in physreg and copy into the vreg.
12900b57cec5SDimitry Andric         MCPhysReg EHPhysReg = TLI->getExceptionPointerRegister(PersonalityFn);
12910b57cec5SDimitry Andric         assert(EHPhysReg && "target lacks exception pointer register");
12920b57cec5SDimitry Andric         MBB->addLiveIn(EHPhysReg);
12930b57cec5SDimitry Andric         unsigned VReg = FuncInfo->getCatchPadExceptionPointerVReg(CPI, PtrRC);
12940b57cec5SDimitry Andric         BuildMI(*MBB, FuncInfo->InsertPt, SDB->getCurDebugLoc(),
12950b57cec5SDimitry Andric                 TII->get(TargetOpcode::COPY), VReg)
12960b57cec5SDimitry Andric             .addReg(EHPhysReg, RegState::Kill);
12970b57cec5SDimitry Andric       }
12980b57cec5SDimitry Andric     }
12990b57cec5SDimitry Andric     return true;
13000b57cec5SDimitry Andric   }
13010b57cec5SDimitry Andric 
13020b57cec5SDimitry Andric   // Add a label to mark the beginning of the landing pad.  Deletion of the
13030b57cec5SDimitry Andric   // landing pad can thus be detected via the MachineModuleInfo.
13040b57cec5SDimitry Andric   MCSymbol *Label = MF->addLandingPad(MBB);
13050b57cec5SDimitry Andric 
13060b57cec5SDimitry Andric   const MCInstrDesc &II = TII->get(TargetOpcode::EH_LABEL);
13070b57cec5SDimitry Andric   BuildMI(*MBB, FuncInfo->InsertPt, SDB->getCurDebugLoc(), II)
13080b57cec5SDimitry Andric     .addSym(Label);
13090b57cec5SDimitry Andric 
1310e8d8bef9SDimitry Andric   // If the unwinder does not preserve all registers, ensure that the
1311e8d8bef9SDimitry Andric   // function marks the clobbered registers as used.
1312e8d8bef9SDimitry Andric   const TargetRegisterInfo &TRI = *MF->getSubtarget().getRegisterInfo();
1313e8d8bef9SDimitry Andric   if (auto *RegMask = TRI.getCustomEHPadPreservedMask(*MF))
1314e8d8bef9SDimitry Andric     MF->getRegInfo().addPhysRegsUsedFromRegMask(RegMask);
1315e8d8bef9SDimitry Andric 
13160b57cec5SDimitry Andric   if (Pers == EHPersonality::Wasm_CXX) {
13170b57cec5SDimitry Andric     if (const auto *CPI = dyn_cast<CatchPadInst>(LLVMBB->getFirstNonPHI()))
13180b57cec5SDimitry Andric       mapWasmLandingPadIndex(MBB, CPI);
13190b57cec5SDimitry Andric   } else {
13200b57cec5SDimitry Andric     // Assign the call site to the landing pad's begin label.
13210b57cec5SDimitry Andric     MF->setCallSiteLandingPad(Label, SDB->LPadToCallSiteMap[MBB]);
13220b57cec5SDimitry Andric     // Mark exception register as live in.
13230b57cec5SDimitry Andric     if (unsigned Reg = TLI->getExceptionPointerRegister(PersonalityFn))
13240b57cec5SDimitry Andric       FuncInfo->ExceptionPointerVirtReg = MBB->addLiveIn(Reg, PtrRC);
13250b57cec5SDimitry Andric     // Mark exception selector register as live in.
13260b57cec5SDimitry Andric     if (unsigned Reg = TLI->getExceptionSelectorRegister(PersonalityFn))
13270b57cec5SDimitry Andric       FuncInfo->ExceptionSelectorVirtReg = MBB->addLiveIn(Reg, PtrRC);
13280b57cec5SDimitry Andric   }
13290b57cec5SDimitry Andric 
13300b57cec5SDimitry Andric   return true;
13310b57cec5SDimitry Andric }
13320b57cec5SDimitry Andric 
133306c3fb27SDimitry Andric // Mark and Report IPToState for each Block under IsEHa
133406c3fb27SDimitry Andric void SelectionDAGISel::reportIPToStateForBlocks(MachineFunction *MF) {
133506c3fb27SDimitry Andric   MachineModuleInfo &MMI = MF->getMMI();
133606c3fb27SDimitry Andric   llvm::WinEHFuncInfo *EHInfo = MF->getWinEHFuncInfo();
133706c3fb27SDimitry Andric   if (!EHInfo)
133806c3fb27SDimitry Andric     return;
133906c3fb27SDimitry Andric   for (auto MBBI = MF->begin(), E = MF->end(); MBBI != E; ++MBBI) {
134006c3fb27SDimitry Andric     MachineBasicBlock *MBB = &*MBBI;
134106c3fb27SDimitry Andric     const BasicBlock *BB = MBB->getBasicBlock();
134206c3fb27SDimitry Andric     int State = EHInfo->BlockToStateMap[BB];
134306c3fb27SDimitry Andric     if (BB->getFirstMayFaultInst()) {
134406c3fb27SDimitry Andric       // Report IP range only for blocks with Faulty inst
134506c3fb27SDimitry Andric       auto MBBb = MBB->getFirstNonPHI();
134606c3fb27SDimitry Andric       MachineInstr *MIb = &*MBBb;
134706c3fb27SDimitry Andric       if (MIb->isTerminator())
134806c3fb27SDimitry Andric         continue;
134906c3fb27SDimitry Andric 
135006c3fb27SDimitry Andric       // Insert EH Labels
135106c3fb27SDimitry Andric       MCSymbol *BeginLabel = MMI.getContext().createTempSymbol();
135206c3fb27SDimitry Andric       MCSymbol *EndLabel = MMI.getContext().createTempSymbol();
135306c3fb27SDimitry Andric       EHInfo->addIPToStateRange(State, BeginLabel, EndLabel);
135406c3fb27SDimitry Andric       BuildMI(*MBB, MBBb, SDB->getCurDebugLoc(),
135506c3fb27SDimitry Andric               TII->get(TargetOpcode::EH_LABEL))
135606c3fb27SDimitry Andric           .addSym(BeginLabel);
135706c3fb27SDimitry Andric       auto MBBe = MBB->instr_end();
135806c3fb27SDimitry Andric       MachineInstr *MIe = &*(--MBBe);
135906c3fb27SDimitry Andric       // insert before (possible multiple) terminators
136006c3fb27SDimitry Andric       while (MIe->isTerminator())
136106c3fb27SDimitry Andric         MIe = &*(--MBBe);
136206c3fb27SDimitry Andric       ++MBBe;
136306c3fb27SDimitry Andric       BuildMI(*MBB, MBBe, SDB->getCurDebugLoc(),
136406c3fb27SDimitry Andric               TII->get(TargetOpcode::EH_LABEL))
136506c3fb27SDimitry Andric           .addSym(EndLabel);
136606c3fb27SDimitry Andric     }
136706c3fb27SDimitry Andric   }
136806c3fb27SDimitry Andric }
136906c3fb27SDimitry Andric 
13700b57cec5SDimitry Andric /// isFoldedOrDeadInstruction - Return true if the specified instruction is
13710b57cec5SDimitry Andric /// side-effect free and is either dead or folded into a generated instruction.
13720b57cec5SDimitry Andric /// Return false if it needs to be emitted.
13730b57cec5SDimitry Andric static bool isFoldedOrDeadInstruction(const Instruction *I,
1374480093f4SDimitry Andric                                       const FunctionLoweringInfo &FuncInfo) {
13750b57cec5SDimitry Andric   return !I->mayWriteToMemory() && // Side-effecting instructions aren't folded.
13760b57cec5SDimitry Andric          !I->isTerminator() &&     // Terminators aren't folded.
13770b57cec5SDimitry Andric          !isa<DbgInfoIntrinsic>(I) && // Debug instructions aren't folded.
13780b57cec5SDimitry Andric          !I->isEHPad() &&             // EH pad instructions aren't folded.
1379480093f4SDimitry Andric          !FuncInfo.isExportedInst(I); // Exported instrs must be computed.
13800b57cec5SDimitry Andric }
13810b57cec5SDimitry Andric 
138206c3fb27SDimitry Andric static bool processIfEntryValueDbgDeclare(FunctionLoweringInfo &FuncInfo,
138306c3fb27SDimitry Andric                                           const Value *Arg, DIExpression *Expr,
138406c3fb27SDimitry Andric                                           DILocalVariable *Var,
138506c3fb27SDimitry Andric                                           DebugLoc DbgLoc) {
138606c3fb27SDimitry Andric   if (!Expr->isEntryValue() || !isa<Argument>(Arg))
138706c3fb27SDimitry Andric     return false;
138806c3fb27SDimitry Andric 
138906c3fb27SDimitry Andric   auto ArgIt = FuncInfo.ValueMap.find(Arg);
139006c3fb27SDimitry Andric   if (ArgIt == FuncInfo.ValueMap.end())
139106c3fb27SDimitry Andric     return false;
139206c3fb27SDimitry Andric   Register ArgVReg = ArgIt->getSecond();
139306c3fb27SDimitry Andric 
139406c3fb27SDimitry Andric   // Find the corresponding livein physical register to this argument.
139506c3fb27SDimitry Andric   for (auto [PhysReg, VirtReg] : FuncInfo.RegInfo->liveins())
139606c3fb27SDimitry Andric     if (VirtReg == ArgVReg) {
1397*5f757f3fSDimitry Andric       // Append an op deref to account for the fact that this is a dbg_declare.
1398*5f757f3fSDimitry Andric       Expr = DIExpression::append(Expr, dwarf::DW_OP_deref);
139906c3fb27SDimitry Andric       FuncInfo.MF->setVariableDbgInfo(Var, Expr, PhysReg, DbgLoc);
140006c3fb27SDimitry Andric       LLVM_DEBUG(dbgs() << "processDbgDeclare: setVariableDbgInfo Var=" << *Var
140106c3fb27SDimitry Andric                         << ", Expr=" << *Expr << ",  MCRegister=" << PhysReg
140206c3fb27SDimitry Andric                         << ", DbgLoc=" << DbgLoc << "\n");
140306c3fb27SDimitry Andric       return true;
140406c3fb27SDimitry Andric     }
140506c3fb27SDimitry Andric   return false;
140606c3fb27SDimitry Andric }
140706c3fb27SDimitry Andric 
140806c3fb27SDimitry Andric static bool processDbgDeclare(FunctionLoweringInfo &FuncInfo,
1409bdd1243dSDimitry Andric                               const Value *Address, DIExpression *Expr,
1410bdd1243dSDimitry Andric                               DILocalVariable *Var, DebugLoc DbgLoc) {
141106c3fb27SDimitry Andric   if (!Address) {
141206c3fb27SDimitry Andric     LLVM_DEBUG(dbgs() << "processDbgDeclares skipping " << *Var
141306c3fb27SDimitry Andric                       << " (bad address)\n");
141406c3fb27SDimitry Andric     return false;
141506c3fb27SDimitry Andric   }
141606c3fb27SDimitry Andric 
141706c3fb27SDimitry Andric   if (processIfEntryValueDbgDeclare(FuncInfo, Address, Expr, Var, DbgLoc))
141806c3fb27SDimitry Andric     return true;
141906c3fb27SDimitry Andric 
1420480093f4SDimitry Andric   MachineFunction *MF = FuncInfo.MF;
14210b57cec5SDimitry Andric   const DataLayout &DL = MF->getDataLayout();
14220b57cec5SDimitry Andric 
1423bdd1243dSDimitry Andric   assert(Var && "Missing variable");
1424bdd1243dSDimitry Andric   assert(DbgLoc && "Missing location");
14250b57cec5SDimitry Andric 
14260b57cec5SDimitry Andric   // Look through casts and constant offset GEPs. These mostly come from
14270b57cec5SDimitry Andric   // inalloca.
14280b57cec5SDimitry Andric   APInt Offset(DL.getTypeSizeInBits(Address->getType()), 0);
14290b57cec5SDimitry Andric   Address = Address->stripAndAccumulateInBoundsConstantOffsets(DL, Offset);
14300b57cec5SDimitry Andric 
14310b57cec5SDimitry Andric   // Check if the variable is a static alloca or a byval or inalloca
14320b57cec5SDimitry Andric   // argument passed in memory. If it is not, then we will ignore this
14330b57cec5SDimitry Andric   // intrinsic and handle this during isel like dbg.value.
14340b57cec5SDimitry Andric   int FI = std::numeric_limits<int>::max();
14350b57cec5SDimitry Andric   if (const auto *AI = dyn_cast<AllocaInst>(Address)) {
1436480093f4SDimitry Andric     auto SI = FuncInfo.StaticAllocaMap.find(AI);
1437480093f4SDimitry Andric     if (SI != FuncInfo.StaticAllocaMap.end())
14380b57cec5SDimitry Andric       FI = SI->second;
14390b57cec5SDimitry Andric   } else if (const auto *Arg = dyn_cast<Argument>(Address))
1440480093f4SDimitry Andric     FI = FuncInfo.getArgumentFrameIndex(Arg);
14410b57cec5SDimitry Andric 
14420b57cec5SDimitry Andric   if (FI == std::numeric_limits<int>::max())
144306c3fb27SDimitry Andric     return false;
14440b57cec5SDimitry Andric 
14450b57cec5SDimitry Andric   if (Offset.getBoolValue())
14460b57cec5SDimitry Andric     Expr = DIExpression::prepend(Expr, DIExpression::ApplyOffset,
14470b57cec5SDimitry Andric                                  Offset.getZExtValue());
1448bdd1243dSDimitry Andric 
1449bdd1243dSDimitry Andric   LLVM_DEBUG(dbgs() << "processDbgDeclare: setVariableDbgInfo Var=" << *Var
1450bdd1243dSDimitry Andric                     << ", Expr=" << *Expr << ",  FI=" << FI
1451bdd1243dSDimitry Andric                     << ", DbgLoc=" << DbgLoc << "\n");
1452bdd1243dSDimitry Andric   MF->setVariableDbgInfo(Var, Expr, FI, DbgLoc);
145306c3fb27SDimitry Andric   return true;
1454bdd1243dSDimitry Andric }
1455bdd1243dSDimitry Andric 
1456bdd1243dSDimitry Andric /// Collect llvm.dbg.declare information. This is done after argument lowering
1457bdd1243dSDimitry Andric /// in case the declarations refer to arguments.
1458bdd1243dSDimitry Andric static void processDbgDeclares(FunctionLoweringInfo &FuncInfo) {
145906c3fb27SDimitry Andric   for (const auto &I : instructions(*FuncInfo.Fn)) {
146006c3fb27SDimitry Andric     const auto *DI = dyn_cast<DbgDeclareInst>(&I);
146106c3fb27SDimitry Andric     if (DI && processDbgDeclare(FuncInfo, DI->getAddress(), DI->getExpression(),
146206c3fb27SDimitry Andric                                 DI->getVariable(), DI->getDebugLoc()))
146306c3fb27SDimitry Andric       FuncInfo.PreprocessedDbgDeclares.insert(DI);
1464*5f757f3fSDimitry Andric 
1465*5f757f3fSDimitry Andric     for (const DPValue &DPV : I.getDbgValueRange()) {
1466*5f757f3fSDimitry Andric       if (DPV.getType() == DPValue::LocationType::Declare &&
1467*5f757f3fSDimitry Andric           processDbgDeclare(FuncInfo, DPV.getVariableLocationOp(0),
1468*5f757f3fSDimitry Andric                             DPV.getExpression(), DPV.getVariable(),
1469*5f757f3fSDimitry Andric                             DPV.getDebugLoc()))
1470*5f757f3fSDimitry Andric         FuncInfo.PreprocessedDPVDeclares.insert(&DPV);
1471*5f757f3fSDimitry Andric     }
14720b57cec5SDimitry Andric   }
1473bdd1243dSDimitry Andric }
1474bdd1243dSDimitry Andric 
1475bdd1243dSDimitry Andric /// Collect single location variable information generated with assignment
1476bdd1243dSDimitry Andric /// tracking. This is done after argument lowering in case the declarations
1477bdd1243dSDimitry Andric /// refer to arguments.
1478bdd1243dSDimitry Andric static void processSingleLocVars(FunctionLoweringInfo &FuncInfo,
1479bdd1243dSDimitry Andric                                  FunctionVarLocs const *FnVarLocs) {
1480bdd1243dSDimitry Andric   for (auto It = FnVarLocs->single_locs_begin(),
1481bdd1243dSDimitry Andric             End = FnVarLocs->single_locs_end();
148206c3fb27SDimitry Andric        It != End; ++It) {
148306c3fb27SDimitry Andric     assert(!It->Values.hasArgList() && "Single loc variadic ops not supported");
148406c3fb27SDimitry Andric     processDbgDeclare(FuncInfo, It->Values.getVariableLocationOp(0), It->Expr,
1485bdd1243dSDimitry Andric                       FnVarLocs->getDILocalVariable(It->VariableID), It->DL);
1486bdd1243dSDimitry Andric   }
148706c3fb27SDimitry Andric }
14880b57cec5SDimitry Andric 
14890b57cec5SDimitry Andric void SelectionDAGISel::SelectAllBasicBlocks(const Function &Fn) {
14900b57cec5SDimitry Andric   FastISelFailed = false;
14910b57cec5SDimitry Andric   // Initialize the Fast-ISel state, if needed.
14920b57cec5SDimitry Andric   FastISel *FastIS = nullptr;
14930b57cec5SDimitry Andric   if (TM.Options.EnableFastISel) {
14940b57cec5SDimitry Andric     LLVM_DEBUG(dbgs() << "Enabling fast-isel\n");
14950b57cec5SDimitry Andric     FastIS = TLI->createFastISel(*FuncInfo, LibInfo);
14960b57cec5SDimitry Andric   }
14970b57cec5SDimitry Andric 
14980b57cec5SDimitry Andric   ReversePostOrderTraversal<const Function*> RPOT(&Fn);
14990b57cec5SDimitry Andric 
15000b57cec5SDimitry Andric   // Lower arguments up front. An RPO iteration always visits the entry block
15010b57cec5SDimitry Andric   // first.
15020b57cec5SDimitry Andric   assert(*RPOT.begin() == &Fn.getEntryBlock());
15030b57cec5SDimitry Andric   ++NumEntryBlocks;
15040b57cec5SDimitry Andric 
15050b57cec5SDimitry Andric   // Set up FuncInfo for ISel. Entry blocks never have PHIs.
15060b57cec5SDimitry Andric   FuncInfo->MBB = FuncInfo->MBBMap[&Fn.getEntryBlock()];
15070b57cec5SDimitry Andric   FuncInfo->InsertPt = FuncInfo->MBB->begin();
15080b57cec5SDimitry Andric 
1509480093f4SDimitry Andric   CurDAG->setFunctionLoweringInfo(FuncInfo.get());
15100b57cec5SDimitry Andric 
15110b57cec5SDimitry Andric   if (!FastIS) {
15120b57cec5SDimitry Andric     LowerArguments(Fn);
15130b57cec5SDimitry Andric   } else {
15140b57cec5SDimitry Andric     // See if fast isel can lower the arguments.
15150b57cec5SDimitry Andric     FastIS->startNewBlock();
15160b57cec5SDimitry Andric     if (!FastIS->lowerArguments()) {
15170b57cec5SDimitry Andric       FastISelFailed = true;
15180b57cec5SDimitry Andric       // Fast isel failed to lower these arguments
15190b57cec5SDimitry Andric       ++NumFastIselFailLowerArguments;
15200b57cec5SDimitry Andric 
15210b57cec5SDimitry Andric       OptimizationRemarkMissed R("sdagisel", "FastISelFailure",
15220b57cec5SDimitry Andric                                  Fn.getSubprogram(),
15230b57cec5SDimitry Andric                                  &Fn.getEntryBlock());
15240b57cec5SDimitry Andric       R << "FastISel didn't lower all arguments: "
152506c3fb27SDimitry Andric         << ore::NV("Prototype", Fn.getFunctionType());
15260b57cec5SDimitry Andric       reportFastISelFailure(*MF, *ORE, R, EnableFastISelAbort > 1);
15270b57cec5SDimitry Andric 
15280b57cec5SDimitry Andric       // Use SelectionDAG argument lowering
15290b57cec5SDimitry Andric       LowerArguments(Fn);
15300b57cec5SDimitry Andric       CurDAG->setRoot(SDB->getControlRoot());
15310b57cec5SDimitry Andric       SDB->clear();
15320b57cec5SDimitry Andric       CodeGenAndEmitDAG();
15330b57cec5SDimitry Andric     }
15340b57cec5SDimitry Andric 
15350b57cec5SDimitry Andric     // If we inserted any instructions at the beginning, make a note of
15360b57cec5SDimitry Andric     // where they are, so we can be sure to emit subsequent instructions
15370b57cec5SDimitry Andric     // after them.
15380b57cec5SDimitry Andric     if (FuncInfo->InsertPt != FuncInfo->MBB->begin())
15390b57cec5SDimitry Andric       FastIS->setLastLocalValue(&*std::prev(FuncInfo->InsertPt));
15400b57cec5SDimitry Andric     else
15410b57cec5SDimitry Andric       FastIS->setLastLocalValue(nullptr);
15420b57cec5SDimitry Andric   }
15430b57cec5SDimitry Andric 
15440b57cec5SDimitry Andric   bool Inserted = SwiftError->createEntriesInEntryBlock(SDB->getCurDebugLoc());
15450b57cec5SDimitry Andric 
15460b57cec5SDimitry Andric   if (FastIS && Inserted)
15470b57cec5SDimitry Andric     FastIS->setLastLocalValue(&*std::prev(FuncInfo->InsertPt));
15480b57cec5SDimitry Andric 
1549bdd1243dSDimitry Andric   if (isAssignmentTrackingEnabled(*Fn.getParent())) {
1550bdd1243dSDimitry Andric     assert(CurDAG->getFunctionVarLocs() &&
1551bdd1243dSDimitry Andric            "expected AssignmentTrackingAnalysis pass results");
1552bdd1243dSDimitry Andric     processSingleLocVars(*FuncInfo, CurDAG->getFunctionVarLocs());
1553bdd1243dSDimitry Andric   } else {
1554480093f4SDimitry Andric     processDbgDeclares(*FuncInfo);
1555bdd1243dSDimitry Andric   }
15560b57cec5SDimitry Andric 
15570b57cec5SDimitry Andric   // Iterate over all basic blocks in the function.
15580b57cec5SDimitry Andric   StackProtector &SP = getAnalysis<StackProtector>();
15590b57cec5SDimitry Andric   for (const BasicBlock *LLVMBB : RPOT) {
1560*5f757f3fSDimitry Andric     if (OptLevel != CodeGenOptLevel::None) {
15610b57cec5SDimitry Andric       bool AllPredsVisited = true;
1562fe6060f1SDimitry Andric       for (const BasicBlock *Pred : predecessors(LLVMBB)) {
1563fe6060f1SDimitry Andric         if (!FuncInfo->VisitedBBs.count(Pred)) {
15640b57cec5SDimitry Andric           AllPredsVisited = false;
15650b57cec5SDimitry Andric           break;
15660b57cec5SDimitry Andric         }
15670b57cec5SDimitry Andric       }
15680b57cec5SDimitry Andric 
15690b57cec5SDimitry Andric       if (AllPredsVisited) {
15700b57cec5SDimitry Andric         for (const PHINode &PN : LLVMBB->phis())
15710b57cec5SDimitry Andric           FuncInfo->ComputePHILiveOutRegInfo(&PN);
15720b57cec5SDimitry Andric       } else {
15730b57cec5SDimitry Andric         for (const PHINode &PN : LLVMBB->phis())
15740b57cec5SDimitry Andric           FuncInfo->InvalidatePHILiveOutRegInfo(&PN);
15750b57cec5SDimitry Andric       }
15760b57cec5SDimitry Andric 
15770b57cec5SDimitry Andric       FuncInfo->VisitedBBs.insert(LLVMBB);
15780b57cec5SDimitry Andric     }
15790b57cec5SDimitry Andric 
15800b57cec5SDimitry Andric     BasicBlock::const_iterator const Begin =
15810b57cec5SDimitry Andric         LLVMBB->getFirstNonPHI()->getIterator();
15820b57cec5SDimitry Andric     BasicBlock::const_iterator const End = LLVMBB->end();
15830b57cec5SDimitry Andric     BasicBlock::const_iterator BI = End;
15840b57cec5SDimitry Andric 
15850b57cec5SDimitry Andric     FuncInfo->MBB = FuncInfo->MBBMap[LLVMBB];
15860b57cec5SDimitry Andric     if (!FuncInfo->MBB)
15870b57cec5SDimitry Andric       continue; // Some blocks like catchpads have no code or MBB.
15880b57cec5SDimitry Andric 
15890b57cec5SDimitry Andric     // Insert new instructions after any phi or argument setup code.
15900b57cec5SDimitry Andric     FuncInfo->InsertPt = FuncInfo->MBB->end();
15910b57cec5SDimitry Andric 
15920b57cec5SDimitry Andric     // Setup an EH landing-pad block.
15930b57cec5SDimitry Andric     FuncInfo->ExceptionPointerVirtReg = 0;
15940b57cec5SDimitry Andric     FuncInfo->ExceptionSelectorVirtReg = 0;
15950b57cec5SDimitry Andric     if (LLVMBB->isEHPad())
15960b57cec5SDimitry Andric       if (!PrepareEHLandingPad())
15970b57cec5SDimitry Andric         continue;
15980b57cec5SDimitry Andric 
15990b57cec5SDimitry Andric     // Before doing SelectionDAG ISel, see if FastISel has been requested.
16000b57cec5SDimitry Andric     if (FastIS) {
16010b57cec5SDimitry Andric       if (LLVMBB != &Fn.getEntryBlock())
16020b57cec5SDimitry Andric         FastIS->startNewBlock();
16030b57cec5SDimitry Andric 
16040b57cec5SDimitry Andric       unsigned NumFastIselRemaining = std::distance(Begin, End);
16050b57cec5SDimitry Andric 
16060b57cec5SDimitry Andric       // Pre-assign swifterror vregs.
16070b57cec5SDimitry Andric       SwiftError->preassignVRegs(FuncInfo->MBB, Begin, End);
16080b57cec5SDimitry Andric 
16090b57cec5SDimitry Andric       // Do FastISel on as many instructions as possible.
16100b57cec5SDimitry Andric       for (; BI != Begin; --BI) {
16110b57cec5SDimitry Andric         const Instruction *Inst = &*std::prev(BI);
16120b57cec5SDimitry Andric 
16130b57cec5SDimitry Andric         // If we no longer require this instruction, skip it.
1614480093f4SDimitry Andric         if (isFoldedOrDeadInstruction(Inst, *FuncInfo) ||
16150b57cec5SDimitry Andric             ElidedArgCopyInstrs.count(Inst)) {
16160b57cec5SDimitry Andric           --NumFastIselRemaining;
16170b57cec5SDimitry Andric           continue;
16180b57cec5SDimitry Andric         }
16190b57cec5SDimitry Andric 
16200b57cec5SDimitry Andric         // Bottom-up: reset the insert pos at the top, after any local-value
16210b57cec5SDimitry Andric         // instructions.
16220b57cec5SDimitry Andric         FastIS->recomputeInsertPt();
16230b57cec5SDimitry Andric 
16240b57cec5SDimitry Andric         // Try to select the instruction with FastISel.
16250b57cec5SDimitry Andric         if (FastIS->selectInstruction(Inst)) {
16260b57cec5SDimitry Andric           --NumFastIselRemaining;
16270b57cec5SDimitry Andric           ++NumFastIselSuccess;
16280b57cec5SDimitry Andric           // If fast isel succeeded, skip over all the folded instructions, and
16290b57cec5SDimitry Andric           // then see if there is a load right before the selected instructions.
16300b57cec5SDimitry Andric           // Try to fold the load if so.
16310b57cec5SDimitry Andric           const Instruction *BeforeInst = Inst;
16320b57cec5SDimitry Andric           while (BeforeInst != &*Begin) {
16330b57cec5SDimitry Andric             BeforeInst = &*std::prev(BasicBlock::const_iterator(BeforeInst));
1634480093f4SDimitry Andric             if (!isFoldedOrDeadInstruction(BeforeInst, *FuncInfo))
16350b57cec5SDimitry Andric               break;
16360b57cec5SDimitry Andric           }
16370b57cec5SDimitry Andric           if (BeforeInst != Inst && isa<LoadInst>(BeforeInst) &&
16380b57cec5SDimitry Andric               BeforeInst->hasOneUse() &&
16390b57cec5SDimitry Andric               FastIS->tryToFoldLoad(cast<LoadInst>(BeforeInst), Inst)) {
16400b57cec5SDimitry Andric             // If we succeeded, don't re-select the load.
164181ad6265SDimitry Andric             LLVM_DEBUG(dbgs()
164281ad6265SDimitry Andric                        << "FastISel folded load: " << *BeforeInst << "\n");
16430b57cec5SDimitry Andric             BI = std::next(BasicBlock::const_iterator(BeforeInst));
16440b57cec5SDimitry Andric             --NumFastIselRemaining;
16450b57cec5SDimitry Andric             ++NumFastIselSuccess;
16460b57cec5SDimitry Andric           }
16470b57cec5SDimitry Andric           continue;
16480b57cec5SDimitry Andric         }
16490b57cec5SDimitry Andric 
16500b57cec5SDimitry Andric         FastISelFailed = true;
16510b57cec5SDimitry Andric 
16520b57cec5SDimitry Andric         // Then handle certain instructions as single-LLVM-Instruction blocks.
16530b57cec5SDimitry Andric         // We cannot separate out GCrelocates to their own blocks since we need
16540b57cec5SDimitry Andric         // to keep track of gc-relocates for a particular gc-statepoint. This is
16550b57cec5SDimitry Andric         // done by SelectionDAGBuilder::LowerAsSTATEPOINT, called before
16560b57cec5SDimitry Andric         // visitGCRelocate.
16575ffd83dbSDimitry Andric         if (isa<CallInst>(Inst) && !isa<GCStatepointInst>(Inst) &&
16585ffd83dbSDimitry Andric             !isa<GCRelocateInst>(Inst) && !isa<GCResultInst>(Inst)) {
16590b57cec5SDimitry Andric           OptimizationRemarkMissed R("sdagisel", "FastISelFailure",
16600b57cec5SDimitry Andric                                      Inst->getDebugLoc(), LLVMBB);
16610b57cec5SDimitry Andric 
16620b57cec5SDimitry Andric           R << "FastISel missed call";
16630b57cec5SDimitry Andric 
16640b57cec5SDimitry Andric           if (R.isEnabled() || EnableFastISelAbort) {
16650b57cec5SDimitry Andric             std::string InstStrStorage;
16660b57cec5SDimitry Andric             raw_string_ostream InstStr(InstStrStorage);
16670b57cec5SDimitry Andric             InstStr << *Inst;
16680b57cec5SDimitry Andric 
16690b57cec5SDimitry Andric             R << ": " << InstStr.str();
16700b57cec5SDimitry Andric           }
16710b57cec5SDimitry Andric 
16720b57cec5SDimitry Andric           reportFastISelFailure(*MF, *ORE, R, EnableFastISelAbort > 2);
16730b57cec5SDimitry Andric 
16740b57cec5SDimitry Andric           if (!Inst->getType()->isVoidTy() && !Inst->getType()->isTokenTy() &&
16750b57cec5SDimitry Andric               !Inst->use_empty()) {
16765ffd83dbSDimitry Andric             Register &R = FuncInfo->ValueMap[Inst];
16770b57cec5SDimitry Andric             if (!R)
16780b57cec5SDimitry Andric               R = FuncInfo->CreateRegs(Inst);
16790b57cec5SDimitry Andric           }
16800b57cec5SDimitry Andric 
16810b57cec5SDimitry Andric           bool HadTailCall = false;
16820b57cec5SDimitry Andric           MachineBasicBlock::iterator SavedInsertPt = FuncInfo->InsertPt;
16830b57cec5SDimitry Andric           SelectBasicBlock(Inst->getIterator(), BI, HadTailCall);
16840b57cec5SDimitry Andric 
16850b57cec5SDimitry Andric           // If the call was emitted as a tail call, we're done with the block.
16860b57cec5SDimitry Andric           // We also need to delete any previously emitted instructions.
16870b57cec5SDimitry Andric           if (HadTailCall) {
16880b57cec5SDimitry Andric             FastIS->removeDeadCode(SavedInsertPt, FuncInfo->MBB->end());
16890b57cec5SDimitry Andric             --BI;
16900b57cec5SDimitry Andric             break;
16910b57cec5SDimitry Andric           }
16920b57cec5SDimitry Andric 
16930b57cec5SDimitry Andric           // Recompute NumFastIselRemaining as Selection DAG instruction
16940b57cec5SDimitry Andric           // selection may have handled the call, input args, etc.
16950b57cec5SDimitry Andric           unsigned RemainingNow = std::distance(Begin, BI);
16960b57cec5SDimitry Andric           NumFastIselFailures += NumFastIselRemaining - RemainingNow;
16970b57cec5SDimitry Andric           NumFastIselRemaining = RemainingNow;
16980b57cec5SDimitry Andric           continue;
16990b57cec5SDimitry Andric         }
17000b57cec5SDimitry Andric 
17010b57cec5SDimitry Andric         OptimizationRemarkMissed R("sdagisel", "FastISelFailure",
17020b57cec5SDimitry Andric                                    Inst->getDebugLoc(), LLVMBB);
17030b57cec5SDimitry Andric 
17040b57cec5SDimitry Andric         bool ShouldAbort = EnableFastISelAbort;
17050b57cec5SDimitry Andric         if (Inst->isTerminator()) {
17060b57cec5SDimitry Andric           // Use a different message for terminator misses.
17070b57cec5SDimitry Andric           R << "FastISel missed terminator";
17080b57cec5SDimitry Andric           // Don't abort for terminator unless the level is really high
17090b57cec5SDimitry Andric           ShouldAbort = (EnableFastISelAbort > 2);
17100b57cec5SDimitry Andric         } else {
17110b57cec5SDimitry Andric           R << "FastISel missed";
17120b57cec5SDimitry Andric         }
17130b57cec5SDimitry Andric 
17140b57cec5SDimitry Andric         if (R.isEnabled() || EnableFastISelAbort) {
17150b57cec5SDimitry Andric           std::string InstStrStorage;
17160b57cec5SDimitry Andric           raw_string_ostream InstStr(InstStrStorage);
17170b57cec5SDimitry Andric           InstStr << *Inst;
17180b57cec5SDimitry Andric           R << ": " << InstStr.str();
17190b57cec5SDimitry Andric         }
17200b57cec5SDimitry Andric 
17210b57cec5SDimitry Andric         reportFastISelFailure(*MF, *ORE, R, ShouldAbort);
17220b57cec5SDimitry Andric 
17230b57cec5SDimitry Andric         NumFastIselFailures += NumFastIselRemaining;
17240b57cec5SDimitry Andric         break;
17250b57cec5SDimitry Andric       }
17260b57cec5SDimitry Andric 
17270b57cec5SDimitry Andric       FastIS->recomputeInsertPt();
17280b57cec5SDimitry Andric     }
17290b57cec5SDimitry Andric 
17300b57cec5SDimitry Andric     if (SP.shouldEmitSDCheck(*LLVMBB)) {
17310b57cec5SDimitry Andric       bool FunctionBasedInstrumentation =
17320b57cec5SDimitry Andric           TLI->getSSPStackGuardCheck(*Fn.getParent());
17330b57cec5SDimitry Andric       SDB->SPDescriptor.initialize(LLVMBB, FuncInfo->MBBMap[LLVMBB],
17340b57cec5SDimitry Andric                                    FunctionBasedInstrumentation);
17350b57cec5SDimitry Andric     }
17360b57cec5SDimitry Andric 
17370b57cec5SDimitry Andric     if (Begin != BI)
17380b57cec5SDimitry Andric       ++NumDAGBlocks;
17390b57cec5SDimitry Andric     else
17400b57cec5SDimitry Andric       ++NumFastIselBlocks;
17410b57cec5SDimitry Andric 
17420b57cec5SDimitry Andric     if (Begin != BI) {
17430b57cec5SDimitry Andric       // Run SelectionDAG instruction selection on the remainder of the block
17440b57cec5SDimitry Andric       // not handled by FastISel. If FastISel is not run, this is the entire
17450b57cec5SDimitry Andric       // block.
17460b57cec5SDimitry Andric       bool HadTailCall;
17470b57cec5SDimitry Andric       SelectBasicBlock(Begin, BI, HadTailCall);
17480b57cec5SDimitry Andric 
17490b57cec5SDimitry Andric       // But if FastISel was run, we already selected some of the block.
17500b57cec5SDimitry Andric       // If we emitted a tail-call, we need to delete any previously emitted
17510b57cec5SDimitry Andric       // instruction that follows it.
1752480093f4SDimitry Andric       if (FastIS && HadTailCall && FuncInfo->InsertPt != FuncInfo->MBB->end())
17530b57cec5SDimitry Andric         FastIS->removeDeadCode(FuncInfo->InsertPt, FuncInfo->MBB->end());
17540b57cec5SDimitry Andric     }
17550b57cec5SDimitry Andric 
17560b57cec5SDimitry Andric     if (FastIS)
17570b57cec5SDimitry Andric       FastIS->finishBasicBlock();
17580b57cec5SDimitry Andric     FinishBasicBlock();
17590b57cec5SDimitry Andric     FuncInfo->PHINodesToUpdate.clear();
17600b57cec5SDimitry Andric     ElidedArgCopyInstrs.clear();
17610b57cec5SDimitry Andric   }
17620b57cec5SDimitry Andric 
176306c3fb27SDimitry Andric   // AsynchEH: Report Block State under -AsynchEH
176406c3fb27SDimitry Andric   if (Fn.getParent()->getModuleFlag("eh-asynch"))
176506c3fb27SDimitry Andric     reportIPToStateForBlocks(MF);
176606c3fb27SDimitry Andric 
17670b57cec5SDimitry Andric   SP.copyToMachineFrameInfo(MF->getFrameInfo());
17680b57cec5SDimitry Andric 
17690b57cec5SDimitry Andric   SwiftError->propagateVRegs();
17700b57cec5SDimitry Andric 
17710b57cec5SDimitry Andric   delete FastIS;
17720b57cec5SDimitry Andric   SDB->clearDanglingDebugInfo();
17730b57cec5SDimitry Andric   SDB->SPDescriptor.resetPerFunctionState();
17740b57cec5SDimitry Andric }
17750b57cec5SDimitry Andric 
17760b57cec5SDimitry Andric void
17770b57cec5SDimitry Andric SelectionDAGISel::FinishBasicBlock() {
17780b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "Total amount of phi nodes to update: "
17790b57cec5SDimitry Andric                     << FuncInfo->PHINodesToUpdate.size() << "\n";
17800b57cec5SDimitry Andric              for (unsigned i = 0, e = FuncInfo->PHINodesToUpdate.size(); i != e;
17810b57cec5SDimitry Andric                   ++i) dbgs()
17820b57cec5SDimitry Andric              << "Node " << i << " : (" << FuncInfo->PHINodesToUpdate[i].first
17830b57cec5SDimitry Andric              << ", " << FuncInfo->PHINodesToUpdate[i].second << ")\n");
17840b57cec5SDimitry Andric 
17850b57cec5SDimitry Andric   // Next, now that we know what the last MBB the LLVM BB expanded is, update
17860b57cec5SDimitry Andric   // PHI nodes in successors.
17870b57cec5SDimitry Andric   for (unsigned i = 0, e = FuncInfo->PHINodesToUpdate.size(); i != e; ++i) {
17880b57cec5SDimitry Andric     MachineInstrBuilder PHI(*MF, FuncInfo->PHINodesToUpdate[i].first);
17890b57cec5SDimitry Andric     assert(PHI->isPHI() &&
17900b57cec5SDimitry Andric            "This is not a machine PHI node that we are updating!");
17910b57cec5SDimitry Andric     if (!FuncInfo->MBB->isSuccessor(PHI->getParent()))
17920b57cec5SDimitry Andric       continue;
17930b57cec5SDimitry Andric     PHI.addReg(FuncInfo->PHINodesToUpdate[i].second).addMBB(FuncInfo->MBB);
17940b57cec5SDimitry Andric   }
17950b57cec5SDimitry Andric 
17960b57cec5SDimitry Andric   // Handle stack protector.
17970b57cec5SDimitry Andric   if (SDB->SPDescriptor.shouldEmitFunctionBasedCheckStackProtector()) {
17980b57cec5SDimitry Andric     // The target provides a guard check function. There is no need to
17990b57cec5SDimitry Andric     // generate error handling code or to split current basic block.
18000b57cec5SDimitry Andric     MachineBasicBlock *ParentMBB = SDB->SPDescriptor.getParentMBB();
18010b57cec5SDimitry Andric 
18020b57cec5SDimitry Andric     // Add load and check to the basicblock.
18030b57cec5SDimitry Andric     FuncInfo->MBB = ParentMBB;
18040b57cec5SDimitry Andric     FuncInfo->InsertPt =
1805349cc55cSDimitry Andric         findSplitPointForStackProtector(ParentMBB, *TII);
18060b57cec5SDimitry Andric     SDB->visitSPDescriptorParent(SDB->SPDescriptor, ParentMBB);
18070b57cec5SDimitry Andric     CurDAG->setRoot(SDB->getRoot());
18080b57cec5SDimitry Andric     SDB->clear();
18090b57cec5SDimitry Andric     CodeGenAndEmitDAG();
18100b57cec5SDimitry Andric 
18110b57cec5SDimitry Andric     // Clear the Per-BB State.
18120b57cec5SDimitry Andric     SDB->SPDescriptor.resetPerBBState();
18130b57cec5SDimitry Andric   } else if (SDB->SPDescriptor.shouldEmitStackProtector()) {
18140b57cec5SDimitry Andric     MachineBasicBlock *ParentMBB = SDB->SPDescriptor.getParentMBB();
18150b57cec5SDimitry Andric     MachineBasicBlock *SuccessMBB = SDB->SPDescriptor.getSuccessMBB();
18160b57cec5SDimitry Andric 
18170b57cec5SDimitry Andric     // Find the split point to split the parent mbb. At the same time copy all
18180b57cec5SDimitry Andric     // physical registers used in the tail of parent mbb into virtual registers
18190b57cec5SDimitry Andric     // before the split point and back into physical registers after the split
18200b57cec5SDimitry Andric     // point. This prevents us needing to deal with Live-ins and many other
18210b57cec5SDimitry Andric     // register allocation issues caused by us splitting the parent mbb. The
18220b57cec5SDimitry Andric     // register allocator will clean up said virtual copies later on.
18230b57cec5SDimitry Andric     MachineBasicBlock::iterator SplitPoint =
1824349cc55cSDimitry Andric         findSplitPointForStackProtector(ParentMBB, *TII);
18250b57cec5SDimitry Andric 
18260b57cec5SDimitry Andric     // Splice the terminator of ParentMBB into SuccessMBB.
18270b57cec5SDimitry Andric     SuccessMBB->splice(SuccessMBB->end(), ParentMBB,
18280b57cec5SDimitry Andric                        SplitPoint,
18290b57cec5SDimitry Andric                        ParentMBB->end());
18300b57cec5SDimitry Andric 
18310b57cec5SDimitry Andric     // Add compare/jump on neq/jump to the parent BB.
18320b57cec5SDimitry Andric     FuncInfo->MBB = ParentMBB;
18330b57cec5SDimitry Andric     FuncInfo->InsertPt = ParentMBB->end();
18340b57cec5SDimitry Andric     SDB->visitSPDescriptorParent(SDB->SPDescriptor, ParentMBB);
18350b57cec5SDimitry Andric     CurDAG->setRoot(SDB->getRoot());
18360b57cec5SDimitry Andric     SDB->clear();
18370b57cec5SDimitry Andric     CodeGenAndEmitDAG();
18380b57cec5SDimitry Andric 
18390b57cec5SDimitry Andric     // CodeGen Failure MBB if we have not codegened it yet.
18400b57cec5SDimitry Andric     MachineBasicBlock *FailureMBB = SDB->SPDescriptor.getFailureMBB();
18410b57cec5SDimitry Andric     if (FailureMBB->empty()) {
18420b57cec5SDimitry Andric       FuncInfo->MBB = FailureMBB;
18430b57cec5SDimitry Andric       FuncInfo->InsertPt = FailureMBB->end();
18440b57cec5SDimitry Andric       SDB->visitSPDescriptorFailure(SDB->SPDescriptor);
18450b57cec5SDimitry Andric       CurDAG->setRoot(SDB->getRoot());
18460b57cec5SDimitry Andric       SDB->clear();
18470b57cec5SDimitry Andric       CodeGenAndEmitDAG();
18480b57cec5SDimitry Andric     }
18490b57cec5SDimitry Andric 
18500b57cec5SDimitry Andric     // Clear the Per-BB State.
18510b57cec5SDimitry Andric     SDB->SPDescriptor.resetPerBBState();
18520b57cec5SDimitry Andric   }
18530b57cec5SDimitry Andric 
18540b57cec5SDimitry Andric   // Lower each BitTestBlock.
18550b57cec5SDimitry Andric   for (auto &BTB : SDB->SL->BitTestCases) {
18560b57cec5SDimitry Andric     // Lower header first, if it wasn't already lowered
18570b57cec5SDimitry Andric     if (!BTB.Emitted) {
18580b57cec5SDimitry Andric       // Set the current basic block to the mbb we wish to insert the code into
18590b57cec5SDimitry Andric       FuncInfo->MBB = BTB.Parent;
18600b57cec5SDimitry Andric       FuncInfo->InsertPt = FuncInfo->MBB->end();
18610b57cec5SDimitry Andric       // Emit the code
18620b57cec5SDimitry Andric       SDB->visitBitTestHeader(BTB, FuncInfo->MBB);
18630b57cec5SDimitry Andric       CurDAG->setRoot(SDB->getRoot());
18640b57cec5SDimitry Andric       SDB->clear();
18650b57cec5SDimitry Andric       CodeGenAndEmitDAG();
18660b57cec5SDimitry Andric     }
18670b57cec5SDimitry Andric 
18680b57cec5SDimitry Andric     BranchProbability UnhandledProb = BTB.Prob;
18690b57cec5SDimitry Andric     for (unsigned j = 0, ej = BTB.Cases.size(); j != ej; ++j) {
18700b57cec5SDimitry Andric       UnhandledProb -= BTB.Cases[j].ExtraProb;
18710b57cec5SDimitry Andric       // Set the current basic block to the mbb we wish to insert the code into
18720b57cec5SDimitry Andric       FuncInfo->MBB = BTB.Cases[j].ThisBB;
18730b57cec5SDimitry Andric       FuncInfo->InsertPt = FuncInfo->MBB->end();
18740b57cec5SDimitry Andric       // Emit the code
18750b57cec5SDimitry Andric 
18760b57cec5SDimitry Andric       // If all cases cover a contiguous range, it is not necessary to jump to
18770b57cec5SDimitry Andric       // the default block after the last bit test fails. This is because the
18780b57cec5SDimitry Andric       // range check during bit test header creation has guaranteed that every
18790b57cec5SDimitry Andric       // case here doesn't go outside the range. In this case, there is no need
18800b57cec5SDimitry Andric       // to perform the last bit test, as it will always be true. Instead, make
18810b57cec5SDimitry Andric       // the second-to-last bit-test fall through to the target of the last bit
18820b57cec5SDimitry Andric       // test, and delete the last bit test.
18830b57cec5SDimitry Andric 
18840b57cec5SDimitry Andric       MachineBasicBlock *NextMBB;
1885349cc55cSDimitry Andric       if ((BTB.ContiguousRange || BTB.FallthroughUnreachable) && j + 2 == ej) {
1886349cc55cSDimitry Andric         // Second-to-last bit-test with contiguous range or omitted range
1887349cc55cSDimitry Andric         // check: fall through to the target of the final bit test.
18880b57cec5SDimitry Andric         NextMBB = BTB.Cases[j + 1].TargetBB;
18890b57cec5SDimitry Andric       } else if (j + 1 == ej) {
18900b57cec5SDimitry Andric         // For the last bit test, fall through to Default.
18910b57cec5SDimitry Andric         NextMBB = BTB.Default;
18920b57cec5SDimitry Andric       } else {
18930b57cec5SDimitry Andric         // Otherwise, fall through to the next bit test.
18940b57cec5SDimitry Andric         NextMBB = BTB.Cases[j + 1].ThisBB;
18950b57cec5SDimitry Andric       }
18960b57cec5SDimitry Andric 
18970b57cec5SDimitry Andric       SDB->visitBitTestCase(BTB, NextMBB, UnhandledProb, BTB.Reg, BTB.Cases[j],
18980b57cec5SDimitry Andric                             FuncInfo->MBB);
18990b57cec5SDimitry Andric 
19000b57cec5SDimitry Andric       CurDAG->setRoot(SDB->getRoot());
19010b57cec5SDimitry Andric       SDB->clear();
19020b57cec5SDimitry Andric       CodeGenAndEmitDAG();
19030b57cec5SDimitry Andric 
1904349cc55cSDimitry Andric       if ((BTB.ContiguousRange || BTB.FallthroughUnreachable) && j + 2 == ej) {
19050b57cec5SDimitry Andric         // Since we're not going to use the final bit test, remove it.
19060b57cec5SDimitry Andric         BTB.Cases.pop_back();
19070b57cec5SDimitry Andric         break;
19080b57cec5SDimitry Andric       }
19090b57cec5SDimitry Andric     }
19100b57cec5SDimitry Andric 
19110b57cec5SDimitry Andric     // Update PHI Nodes
19120eae32dcSDimitry Andric     for (const std::pair<MachineInstr *, unsigned> &P :
19130eae32dcSDimitry Andric          FuncInfo->PHINodesToUpdate) {
19140eae32dcSDimitry Andric       MachineInstrBuilder PHI(*MF, P.first);
19150b57cec5SDimitry Andric       MachineBasicBlock *PHIBB = PHI->getParent();
19160b57cec5SDimitry Andric       assert(PHI->isPHI() &&
19170b57cec5SDimitry Andric              "This is not a machine PHI node that we are updating!");
19180b57cec5SDimitry Andric       // This is "default" BB. We have two jumps to it. From "header" BB and
19190b57cec5SDimitry Andric       // from last "case" BB, unless the latter was skipped.
19200b57cec5SDimitry Andric       if (PHIBB == BTB.Default) {
19210eae32dcSDimitry Andric         PHI.addReg(P.second).addMBB(BTB.Parent);
19220b57cec5SDimitry Andric         if (!BTB.ContiguousRange) {
19230eae32dcSDimitry Andric           PHI.addReg(P.second).addMBB(BTB.Cases.back().ThisBB);
19240b57cec5SDimitry Andric          }
19250b57cec5SDimitry Andric       }
19260b57cec5SDimitry Andric       // One of "cases" BB.
19270eae32dcSDimitry Andric       for (const SwitchCG::BitTestCase &BT : BTB.Cases) {
19280eae32dcSDimitry Andric         MachineBasicBlock* cBB = BT.ThisBB;
19290b57cec5SDimitry Andric         if (cBB->isSuccessor(PHIBB))
19300eae32dcSDimitry Andric           PHI.addReg(P.second).addMBB(cBB);
19310b57cec5SDimitry Andric       }
19320b57cec5SDimitry Andric     }
19330b57cec5SDimitry Andric   }
19340b57cec5SDimitry Andric   SDB->SL->BitTestCases.clear();
19350b57cec5SDimitry Andric 
19360b57cec5SDimitry Andric   // If the JumpTable record is filled in, then we need to emit a jump table.
19370b57cec5SDimitry Andric   // Updating the PHI nodes is tricky in this case, since we need to determine
19380b57cec5SDimitry Andric   // whether the PHI is a successor of the range check MBB or the jump table MBB
19390b57cec5SDimitry Andric   for (unsigned i = 0, e = SDB->SL->JTCases.size(); i != e; ++i) {
19400b57cec5SDimitry Andric     // Lower header first, if it wasn't already lowered
19410b57cec5SDimitry Andric     if (!SDB->SL->JTCases[i].first.Emitted) {
19420b57cec5SDimitry Andric       // Set the current basic block to the mbb we wish to insert the code into
19430b57cec5SDimitry Andric       FuncInfo->MBB = SDB->SL->JTCases[i].first.HeaderBB;
19440b57cec5SDimitry Andric       FuncInfo->InsertPt = FuncInfo->MBB->end();
19450b57cec5SDimitry Andric       // Emit the code
19460b57cec5SDimitry Andric       SDB->visitJumpTableHeader(SDB->SL->JTCases[i].second,
19470b57cec5SDimitry Andric                                 SDB->SL->JTCases[i].first, FuncInfo->MBB);
19480b57cec5SDimitry Andric       CurDAG->setRoot(SDB->getRoot());
19490b57cec5SDimitry Andric       SDB->clear();
19500b57cec5SDimitry Andric       CodeGenAndEmitDAG();
19510b57cec5SDimitry Andric     }
19520b57cec5SDimitry Andric 
19530b57cec5SDimitry Andric     // Set the current basic block to the mbb we wish to insert the code into
19540b57cec5SDimitry Andric     FuncInfo->MBB = SDB->SL->JTCases[i].second.MBB;
19550b57cec5SDimitry Andric     FuncInfo->InsertPt = FuncInfo->MBB->end();
19560b57cec5SDimitry Andric     // Emit the code
19570b57cec5SDimitry Andric     SDB->visitJumpTable(SDB->SL->JTCases[i].second);
19580b57cec5SDimitry Andric     CurDAG->setRoot(SDB->getRoot());
19590b57cec5SDimitry Andric     SDB->clear();
19600b57cec5SDimitry Andric     CodeGenAndEmitDAG();
19610b57cec5SDimitry Andric 
19620b57cec5SDimitry Andric     // Update PHI Nodes
19630b57cec5SDimitry Andric     for (unsigned pi = 0, pe = FuncInfo->PHINodesToUpdate.size();
19640b57cec5SDimitry Andric          pi != pe; ++pi) {
19650b57cec5SDimitry Andric       MachineInstrBuilder PHI(*MF, FuncInfo->PHINodesToUpdate[pi].first);
19660b57cec5SDimitry Andric       MachineBasicBlock *PHIBB = PHI->getParent();
19670b57cec5SDimitry Andric       assert(PHI->isPHI() &&
19680b57cec5SDimitry Andric              "This is not a machine PHI node that we are updating!");
19690b57cec5SDimitry Andric       // "default" BB. We can go there only from header BB.
19700b57cec5SDimitry Andric       if (PHIBB == SDB->SL->JTCases[i].second.Default)
19710b57cec5SDimitry Andric         PHI.addReg(FuncInfo->PHINodesToUpdate[pi].second)
19720b57cec5SDimitry Andric            .addMBB(SDB->SL->JTCases[i].first.HeaderBB);
19730b57cec5SDimitry Andric       // JT BB. Just iterate over successors here
19740b57cec5SDimitry Andric       if (FuncInfo->MBB->isSuccessor(PHIBB))
19750b57cec5SDimitry Andric         PHI.addReg(FuncInfo->PHINodesToUpdate[pi].second).addMBB(FuncInfo->MBB);
19760b57cec5SDimitry Andric     }
19770b57cec5SDimitry Andric   }
19780b57cec5SDimitry Andric   SDB->SL->JTCases.clear();
19790b57cec5SDimitry Andric 
19800b57cec5SDimitry Andric   // If we generated any switch lowering information, build and codegen any
19810b57cec5SDimitry Andric   // additional DAGs necessary.
19820b57cec5SDimitry Andric   for (unsigned i = 0, e = SDB->SL->SwitchCases.size(); i != e; ++i) {
19830b57cec5SDimitry Andric     // Set the current basic block to the mbb we wish to insert the code into
19840b57cec5SDimitry Andric     FuncInfo->MBB = SDB->SL->SwitchCases[i].ThisBB;
19850b57cec5SDimitry Andric     FuncInfo->InsertPt = FuncInfo->MBB->end();
19860b57cec5SDimitry Andric 
19870b57cec5SDimitry Andric     // Determine the unique successors.
19880b57cec5SDimitry Andric     SmallVector<MachineBasicBlock *, 2> Succs;
19890b57cec5SDimitry Andric     Succs.push_back(SDB->SL->SwitchCases[i].TrueBB);
19900b57cec5SDimitry Andric     if (SDB->SL->SwitchCases[i].TrueBB != SDB->SL->SwitchCases[i].FalseBB)
19910b57cec5SDimitry Andric       Succs.push_back(SDB->SL->SwitchCases[i].FalseBB);
19920b57cec5SDimitry Andric 
19930b57cec5SDimitry Andric     // Emit the code. Note that this could result in FuncInfo->MBB being split.
19940b57cec5SDimitry Andric     SDB->visitSwitchCase(SDB->SL->SwitchCases[i], FuncInfo->MBB);
19950b57cec5SDimitry Andric     CurDAG->setRoot(SDB->getRoot());
19960b57cec5SDimitry Andric     SDB->clear();
19970b57cec5SDimitry Andric     CodeGenAndEmitDAG();
19980b57cec5SDimitry Andric 
19990b57cec5SDimitry Andric     // Remember the last block, now that any splitting is done, for use in
20000b57cec5SDimitry Andric     // populating PHI nodes in successors.
20010b57cec5SDimitry Andric     MachineBasicBlock *ThisBB = FuncInfo->MBB;
20020b57cec5SDimitry Andric 
20030b57cec5SDimitry Andric     // Handle any PHI nodes in successors of this chunk, as if we were coming
20040b57cec5SDimitry Andric     // from the original BB before switch expansion.  Note that PHI nodes can
20050b57cec5SDimitry Andric     // occur multiple times in PHINodesToUpdate.  We have to be very careful to
20060b57cec5SDimitry Andric     // handle them the right number of times.
20070b57cec5SDimitry Andric     for (unsigned i = 0, e = Succs.size(); i != e; ++i) {
20080b57cec5SDimitry Andric       FuncInfo->MBB = Succs[i];
20090b57cec5SDimitry Andric       FuncInfo->InsertPt = FuncInfo->MBB->end();
20100b57cec5SDimitry Andric       // FuncInfo->MBB may have been removed from the CFG if a branch was
20110b57cec5SDimitry Andric       // constant folded.
20120b57cec5SDimitry Andric       if (ThisBB->isSuccessor(FuncInfo->MBB)) {
20130b57cec5SDimitry Andric         for (MachineBasicBlock::iterator
20140b57cec5SDimitry Andric              MBBI = FuncInfo->MBB->begin(), MBBE = FuncInfo->MBB->end();
20150b57cec5SDimitry Andric              MBBI != MBBE && MBBI->isPHI(); ++MBBI) {
20160b57cec5SDimitry Andric           MachineInstrBuilder PHI(*MF, MBBI);
20170b57cec5SDimitry Andric           // This value for this PHI node is recorded in PHINodesToUpdate.
20180b57cec5SDimitry Andric           for (unsigned pn = 0; ; ++pn) {
20190b57cec5SDimitry Andric             assert(pn != FuncInfo->PHINodesToUpdate.size() &&
20200b57cec5SDimitry Andric                    "Didn't find PHI entry!");
20210b57cec5SDimitry Andric             if (FuncInfo->PHINodesToUpdate[pn].first == PHI) {
20220b57cec5SDimitry Andric               PHI.addReg(FuncInfo->PHINodesToUpdate[pn].second).addMBB(ThisBB);
20230b57cec5SDimitry Andric               break;
20240b57cec5SDimitry Andric             }
20250b57cec5SDimitry Andric           }
20260b57cec5SDimitry Andric         }
20270b57cec5SDimitry Andric       }
20280b57cec5SDimitry Andric     }
20290b57cec5SDimitry Andric   }
20300b57cec5SDimitry Andric   SDB->SL->SwitchCases.clear();
20310b57cec5SDimitry Andric }
20320b57cec5SDimitry Andric 
20330b57cec5SDimitry Andric /// Create the scheduler. If a specific scheduler was specified
20340b57cec5SDimitry Andric /// via the SchedulerRegistry, use it, otherwise select the
20350b57cec5SDimitry Andric /// one preferred by the target.
20360b57cec5SDimitry Andric ///
20370b57cec5SDimitry Andric ScheduleDAGSDNodes *SelectionDAGISel::CreateScheduler() {
20380b57cec5SDimitry Andric   return ISHeuristic(this, OptLevel);
20390b57cec5SDimitry Andric }
20400b57cec5SDimitry Andric 
20410b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
20420b57cec5SDimitry Andric // Helper functions used by the generated instruction selector.
20430b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
20440b57cec5SDimitry Andric // Calls to these methods are generated by tblgen.
20450b57cec5SDimitry Andric 
20460b57cec5SDimitry Andric /// CheckAndMask - The isel is trying to match something like (and X, 255).  If
20470b57cec5SDimitry Andric /// the dag combiner simplified the 255, we still want to match.  RHS is the
20480b57cec5SDimitry Andric /// actual value in the DAG on the RHS of an AND, and DesiredMaskS is the value
20490b57cec5SDimitry Andric /// specified in the .td file (e.g. 255).
20500b57cec5SDimitry Andric bool SelectionDAGISel::CheckAndMask(SDValue LHS, ConstantSDNode *RHS,
20510b57cec5SDimitry Andric                                     int64_t DesiredMaskS) const {
20520b57cec5SDimitry Andric   const APInt &ActualMask = RHS->getAPIntValue();
20530b57cec5SDimitry Andric   const APInt &DesiredMask = APInt(LHS.getValueSizeInBits(), DesiredMaskS);
20540b57cec5SDimitry Andric 
20550b57cec5SDimitry Andric   // If the actual mask exactly matches, success!
20560b57cec5SDimitry Andric   if (ActualMask == DesiredMask)
20570b57cec5SDimitry Andric     return true;
20580b57cec5SDimitry Andric 
20590b57cec5SDimitry Andric   // If the actual AND mask is allowing unallowed bits, this doesn't match.
20600b57cec5SDimitry Andric   if (!ActualMask.isSubsetOf(DesiredMask))
20610b57cec5SDimitry Andric     return false;
20620b57cec5SDimitry Andric 
20630b57cec5SDimitry Andric   // Otherwise, the DAG Combiner may have proven that the value coming in is
20640b57cec5SDimitry Andric   // either already zero or is not demanded.  Check for known zero input bits.
20650b57cec5SDimitry Andric   APInt NeededMask = DesiredMask & ~ActualMask;
20660b57cec5SDimitry Andric   if (CurDAG->MaskedValueIsZero(LHS, NeededMask))
20670b57cec5SDimitry Andric     return true;
20680b57cec5SDimitry Andric 
20690b57cec5SDimitry Andric   // TODO: check to see if missing bits are just not demanded.
20700b57cec5SDimitry Andric 
20710b57cec5SDimitry Andric   // Otherwise, this pattern doesn't match.
20720b57cec5SDimitry Andric   return false;
20730b57cec5SDimitry Andric }
20740b57cec5SDimitry Andric 
20750b57cec5SDimitry Andric /// CheckOrMask - The isel is trying to match something like (or X, 255).  If
20760b57cec5SDimitry Andric /// the dag combiner simplified the 255, we still want to match.  RHS is the
20770b57cec5SDimitry Andric /// actual value in the DAG on the RHS of an OR, and DesiredMaskS is the value
20780b57cec5SDimitry Andric /// specified in the .td file (e.g. 255).
20790b57cec5SDimitry Andric bool SelectionDAGISel::CheckOrMask(SDValue LHS, ConstantSDNode *RHS,
20800b57cec5SDimitry Andric                                    int64_t DesiredMaskS) const {
20810b57cec5SDimitry Andric   const APInt &ActualMask = RHS->getAPIntValue();
20820b57cec5SDimitry Andric   const APInt &DesiredMask = APInt(LHS.getValueSizeInBits(), DesiredMaskS);
20830b57cec5SDimitry Andric 
20840b57cec5SDimitry Andric   // If the actual mask exactly matches, success!
20850b57cec5SDimitry Andric   if (ActualMask == DesiredMask)
20860b57cec5SDimitry Andric     return true;
20870b57cec5SDimitry Andric 
20880b57cec5SDimitry Andric   // If the actual AND mask is allowing unallowed bits, this doesn't match.
20890b57cec5SDimitry Andric   if (!ActualMask.isSubsetOf(DesiredMask))
20900b57cec5SDimitry Andric     return false;
20910b57cec5SDimitry Andric 
20920b57cec5SDimitry Andric   // Otherwise, the DAG Combiner may have proven that the value coming in is
20930b57cec5SDimitry Andric   // either already zero or is not demanded.  Check for known zero input bits.
20940b57cec5SDimitry Andric   APInt NeededMask = DesiredMask & ~ActualMask;
20950b57cec5SDimitry Andric   KnownBits Known = CurDAG->computeKnownBits(LHS);
20960b57cec5SDimitry Andric 
20970b57cec5SDimitry Andric   // If all the missing bits in the or are already known to be set, match!
20980b57cec5SDimitry Andric   if (NeededMask.isSubsetOf(Known.One))
20990b57cec5SDimitry Andric     return true;
21000b57cec5SDimitry Andric 
21010b57cec5SDimitry Andric   // TODO: check to see if missing bits are just not demanded.
21020b57cec5SDimitry Andric 
21030b57cec5SDimitry Andric   // Otherwise, this pattern doesn't match.
21040b57cec5SDimitry Andric   return false;
21050b57cec5SDimitry Andric }
21060b57cec5SDimitry Andric 
21070b57cec5SDimitry Andric /// SelectInlineAsmMemoryOperands - Calls to this are automatically generated
21080b57cec5SDimitry Andric /// by tblgen.  Others should not call it.
21090b57cec5SDimitry Andric void SelectionDAGISel::SelectInlineAsmMemoryOperands(std::vector<SDValue> &Ops,
21100b57cec5SDimitry Andric                                                      const SDLoc &DL) {
21110b57cec5SDimitry Andric   std::vector<SDValue> InOps;
21120b57cec5SDimitry Andric   std::swap(InOps, Ops);
21130b57cec5SDimitry Andric 
21140b57cec5SDimitry Andric   Ops.push_back(InOps[InlineAsm::Op_InputChain]); // 0
21150b57cec5SDimitry Andric   Ops.push_back(InOps[InlineAsm::Op_AsmString]);  // 1
21160b57cec5SDimitry Andric   Ops.push_back(InOps[InlineAsm::Op_MDNode]);     // 2, !srcloc
21170b57cec5SDimitry Andric   Ops.push_back(InOps[InlineAsm::Op_ExtraInfo]);  // 3 (SideEffect, AlignStack)
21180b57cec5SDimitry Andric 
21190b57cec5SDimitry Andric   unsigned i = InlineAsm::Op_FirstOperand, e = InOps.size();
21200b57cec5SDimitry Andric   if (InOps[e-1].getValueType() == MVT::Glue)
21210b57cec5SDimitry Andric     --e;  // Don't process a glue operand if it is here.
21220b57cec5SDimitry Andric 
21230b57cec5SDimitry Andric   while (i != e) {
2124*5f757f3fSDimitry Andric     InlineAsm::Flag Flags(cast<ConstantSDNode>(InOps[i])->getZExtValue());
2125*5f757f3fSDimitry Andric     if (!Flags.isMemKind() && !Flags.isFuncKind()) {
21260b57cec5SDimitry Andric       // Just skip over this operand, copying the operands verbatim.
21270b57cec5SDimitry Andric       Ops.insert(Ops.end(), InOps.begin() + i,
2128*5f757f3fSDimitry Andric                  InOps.begin() + i + Flags.getNumOperandRegisters() + 1);
2129*5f757f3fSDimitry Andric       i += Flags.getNumOperandRegisters() + 1;
21300b57cec5SDimitry Andric     } else {
2131*5f757f3fSDimitry Andric       assert(Flags.getNumOperandRegisters() == 1 &&
21320b57cec5SDimitry Andric              "Memory operand with multiple values?");
21330b57cec5SDimitry Andric 
21340b57cec5SDimitry Andric       unsigned TiedToOperand;
2135*5f757f3fSDimitry Andric       if (Flags.isUseOperandTiedToDef(TiedToOperand)) {
21360b57cec5SDimitry Andric         // We need the constraint ID from the operand this is tied to.
21370b57cec5SDimitry Andric         unsigned CurOp = InlineAsm::Op_FirstOperand;
2138*5f757f3fSDimitry Andric         Flags =
2139*5f757f3fSDimitry Andric             InlineAsm::Flag(cast<ConstantSDNode>(InOps[CurOp])->getZExtValue());
21400b57cec5SDimitry Andric         for (; TiedToOperand; --TiedToOperand) {
2141*5f757f3fSDimitry Andric           CurOp += Flags.getNumOperandRegisters() + 1;
2142*5f757f3fSDimitry Andric           Flags = InlineAsm::Flag(
2143*5f757f3fSDimitry Andric               cast<ConstantSDNode>(InOps[CurOp])->getZExtValue());
21440b57cec5SDimitry Andric         }
21450b57cec5SDimitry Andric       }
21460b57cec5SDimitry Andric 
21470b57cec5SDimitry Andric       // Otherwise, this is a memory operand.  Ask the target to select it.
21480b57cec5SDimitry Andric       std::vector<SDValue> SelOps;
2149*5f757f3fSDimitry Andric       const InlineAsm::ConstraintCode ConstraintID =
2150*5f757f3fSDimitry Andric           Flags.getMemoryConstraintID();
21510b57cec5SDimitry Andric       if (SelectInlineAsmMemoryOperand(InOps[i+1], ConstraintID, SelOps))
21520b57cec5SDimitry Andric         report_fatal_error("Could not match memory address.  Inline asm"
21530b57cec5SDimitry Andric                            " failure!");
21540b57cec5SDimitry Andric 
21550b57cec5SDimitry Andric       // Add this to the output node.
2156*5f757f3fSDimitry Andric       Flags = InlineAsm::Flag(Flags.isMemKind() ? InlineAsm::Kind::Mem
2157*5f757f3fSDimitry Andric                                                 : InlineAsm::Kind::Func,
2158*5f757f3fSDimitry Andric                               SelOps.size());
2159*5f757f3fSDimitry Andric       Flags.setMemConstraint(ConstraintID);
2160*5f757f3fSDimitry Andric       Ops.push_back(CurDAG->getTargetConstant(Flags, DL, MVT::i32));
2161e8d8bef9SDimitry Andric       llvm::append_range(Ops, SelOps);
21620b57cec5SDimitry Andric       i += 2;
21630b57cec5SDimitry Andric     }
21640b57cec5SDimitry Andric   }
21650b57cec5SDimitry Andric 
21660b57cec5SDimitry Andric   // Add the glue input back if present.
21670b57cec5SDimitry Andric   if (e != InOps.size())
21680b57cec5SDimitry Andric     Ops.push_back(InOps.back());
21690b57cec5SDimitry Andric }
21700b57cec5SDimitry Andric 
21710b57cec5SDimitry Andric /// findGlueUse - Return use of MVT::Glue value produced by the specified
21720b57cec5SDimitry Andric /// SDNode.
21730b57cec5SDimitry Andric ///
21740b57cec5SDimitry Andric static SDNode *findGlueUse(SDNode *N) {
21750b57cec5SDimitry Andric   unsigned FlagResNo = N->getNumValues()-1;
21760b57cec5SDimitry Andric   for (SDNode::use_iterator I = N->use_begin(), E = N->use_end(); I != E; ++I) {
21770b57cec5SDimitry Andric     SDUse &Use = I.getUse();
21780b57cec5SDimitry Andric     if (Use.getResNo() == FlagResNo)
21790b57cec5SDimitry Andric       return Use.getUser();
21800b57cec5SDimitry Andric   }
21810b57cec5SDimitry Andric   return nullptr;
21820b57cec5SDimitry Andric }
21830b57cec5SDimitry Andric 
21840b57cec5SDimitry Andric /// findNonImmUse - Return true if "Def" is a predecessor of "Root" via a path
21850b57cec5SDimitry Andric /// beyond "ImmedUse".  We may ignore chains as they are checked separately.
21860b57cec5SDimitry Andric static bool findNonImmUse(SDNode *Root, SDNode *Def, SDNode *ImmedUse,
21870b57cec5SDimitry Andric                           bool IgnoreChains) {
21880b57cec5SDimitry Andric   SmallPtrSet<const SDNode *, 16> Visited;
21890b57cec5SDimitry Andric   SmallVector<const SDNode *, 16> WorkList;
21900b57cec5SDimitry Andric   // Only check if we have non-immediate uses of Def.
21910b57cec5SDimitry Andric   if (ImmedUse->isOnlyUserOf(Def))
21920b57cec5SDimitry Andric     return false;
21930b57cec5SDimitry Andric 
21940b57cec5SDimitry Andric   // We don't care about paths to Def that go through ImmedUse so mark it
21950b57cec5SDimitry Andric   // visited and mark non-def operands as used.
21960b57cec5SDimitry Andric   Visited.insert(ImmedUse);
21970b57cec5SDimitry Andric   for (const SDValue &Op : ImmedUse->op_values()) {
21980b57cec5SDimitry Andric     SDNode *N = Op.getNode();
21990b57cec5SDimitry Andric     // Ignore chain deps (they are validated by
22000b57cec5SDimitry Andric     // HandleMergeInputChains) and immediate uses
22010b57cec5SDimitry Andric     if ((Op.getValueType() == MVT::Other && IgnoreChains) || N == Def)
22020b57cec5SDimitry Andric       continue;
22030b57cec5SDimitry Andric     if (!Visited.insert(N).second)
22040b57cec5SDimitry Andric       continue;
22050b57cec5SDimitry Andric     WorkList.push_back(N);
22060b57cec5SDimitry Andric   }
22070b57cec5SDimitry Andric 
22080b57cec5SDimitry Andric   // Initialize worklist to operands of Root.
22090b57cec5SDimitry Andric   if (Root != ImmedUse) {
22100b57cec5SDimitry Andric     for (const SDValue &Op : Root->op_values()) {
22110b57cec5SDimitry Andric       SDNode *N = Op.getNode();
22120b57cec5SDimitry Andric       // Ignore chains (they are validated by HandleMergeInputChains)
22130b57cec5SDimitry Andric       if ((Op.getValueType() == MVT::Other && IgnoreChains) || N == Def)
22140b57cec5SDimitry Andric         continue;
22150b57cec5SDimitry Andric       if (!Visited.insert(N).second)
22160b57cec5SDimitry Andric         continue;
22170b57cec5SDimitry Andric       WorkList.push_back(N);
22180b57cec5SDimitry Andric     }
22190b57cec5SDimitry Andric   }
22200b57cec5SDimitry Andric 
22210b57cec5SDimitry Andric   return SDNode::hasPredecessorHelper(Def, Visited, WorkList, 0, true);
22220b57cec5SDimitry Andric }
22230b57cec5SDimitry Andric 
22240b57cec5SDimitry Andric /// IsProfitableToFold - Returns true if it's profitable to fold the specific
22250b57cec5SDimitry Andric /// operand node N of U during instruction selection that starts at Root.
22260b57cec5SDimitry Andric bool SelectionDAGISel::IsProfitableToFold(SDValue N, SDNode *U,
22270b57cec5SDimitry Andric                                           SDNode *Root) const {
2228*5f757f3fSDimitry Andric   if (OptLevel == CodeGenOptLevel::None)
2229*5f757f3fSDimitry Andric     return false;
22300b57cec5SDimitry Andric   return N.hasOneUse();
22310b57cec5SDimitry Andric }
22320b57cec5SDimitry Andric 
22330b57cec5SDimitry Andric /// IsLegalToFold - Returns true if the specific operand node N of
22340b57cec5SDimitry Andric /// U can be folded during instruction selection that starts at Root.
22350b57cec5SDimitry Andric bool SelectionDAGISel::IsLegalToFold(SDValue N, SDNode *U, SDNode *Root,
2236*5f757f3fSDimitry Andric                                      CodeGenOptLevel OptLevel,
22370b57cec5SDimitry Andric                                      bool IgnoreChains) {
2238*5f757f3fSDimitry Andric   if (OptLevel == CodeGenOptLevel::None)
2239*5f757f3fSDimitry Andric     return false;
22400b57cec5SDimitry Andric 
2241*5f757f3fSDimitry Andric   // If Root use can somehow reach N through a path that doesn't contain
22420b57cec5SDimitry Andric   // U then folding N would create a cycle. e.g. In the following
22430b57cec5SDimitry Andric   // diagram, Root can reach N through X. If N is folded into Root, then
22440b57cec5SDimitry Andric   // X is both a predecessor and a successor of U.
22450b57cec5SDimitry Andric   //
22460b57cec5SDimitry Andric   //          [N*]           //
22470b57cec5SDimitry Andric   //         ^   ^           //
22480b57cec5SDimitry Andric   //        /     \          //
22490b57cec5SDimitry Andric   //      [U*]    [X]?       //
22500b57cec5SDimitry Andric   //        ^     ^          //
22510b57cec5SDimitry Andric   //         \   /           //
22520b57cec5SDimitry Andric   //          \ /            //
22530b57cec5SDimitry Andric   //         [Root*]         //
22540b57cec5SDimitry Andric   //
22550b57cec5SDimitry Andric   // * indicates nodes to be folded together.
22560b57cec5SDimitry Andric   //
22570b57cec5SDimitry Andric   // If Root produces glue, then it gets (even more) interesting. Since it
22580b57cec5SDimitry Andric   // will be "glued" together with its glue use in the scheduler, we need to
22590b57cec5SDimitry Andric   // check if it might reach N.
22600b57cec5SDimitry Andric   //
22610b57cec5SDimitry Andric   //          [N*]           //
22620b57cec5SDimitry Andric   //         ^   ^           //
22630b57cec5SDimitry Andric   //        /     \          //
22640b57cec5SDimitry Andric   //      [U*]    [X]?       //
22650b57cec5SDimitry Andric   //        ^       ^        //
22660b57cec5SDimitry Andric   //         \       \       //
22670b57cec5SDimitry Andric   //          \      |       //
22680b57cec5SDimitry Andric   //         [Root*] |       //
22690b57cec5SDimitry Andric   //          ^      |       //
22700b57cec5SDimitry Andric   //          f      |       //
22710b57cec5SDimitry Andric   //          |      /       //
22720b57cec5SDimitry Andric   //         [Y]    /        //
22730b57cec5SDimitry Andric   //           ^   /         //
22740b57cec5SDimitry Andric   //           f  /          //
22750b57cec5SDimitry Andric   //           | /           //
22760b57cec5SDimitry Andric   //          [GU]           //
22770b57cec5SDimitry Andric   //
22780b57cec5SDimitry Andric   // If GU (glue use) indirectly reaches N (the load), and Root folds N
22790b57cec5SDimitry Andric   // (call it Fold), then X is a predecessor of GU and a successor of
22800b57cec5SDimitry Andric   // Fold. But since Fold and GU are glued together, this will create
22810b57cec5SDimitry Andric   // a cycle in the scheduling graph.
22820b57cec5SDimitry Andric 
22830b57cec5SDimitry Andric   // If the node has glue, walk down the graph to the "lowest" node in the
22840b57cec5SDimitry Andric   // glueged set.
22850b57cec5SDimitry Andric   EVT VT = Root->getValueType(Root->getNumValues()-1);
22860b57cec5SDimitry Andric   while (VT == MVT::Glue) {
22870b57cec5SDimitry Andric     SDNode *GU = findGlueUse(Root);
22880b57cec5SDimitry Andric     if (!GU)
22890b57cec5SDimitry Andric       break;
22900b57cec5SDimitry Andric     Root = GU;
22910b57cec5SDimitry Andric     VT = Root->getValueType(Root->getNumValues()-1);
22920b57cec5SDimitry Andric 
22930b57cec5SDimitry Andric     // If our query node has a glue result with a use, we've walked up it.  If
22940b57cec5SDimitry Andric     // the user (which has already been selected) has a chain or indirectly uses
22950b57cec5SDimitry Andric     // the chain, HandleMergeInputChains will not consider it.  Because of
22960b57cec5SDimitry Andric     // this, we cannot ignore chains in this predicate.
22970b57cec5SDimitry Andric     IgnoreChains = false;
22980b57cec5SDimitry Andric   }
22990b57cec5SDimitry Andric 
23000b57cec5SDimitry Andric   return !findNonImmUse(Root, N.getNode(), U, IgnoreChains);
23010b57cec5SDimitry Andric }
23020b57cec5SDimitry Andric 
23035ffd83dbSDimitry Andric void SelectionDAGISel::Select_INLINEASM(SDNode *N) {
23040b57cec5SDimitry Andric   SDLoc DL(N);
23050b57cec5SDimitry Andric 
23060b57cec5SDimitry Andric   std::vector<SDValue> Ops(N->op_begin(), N->op_end());
23070b57cec5SDimitry Andric   SelectInlineAsmMemoryOperands(Ops, DL);
23080b57cec5SDimitry Andric 
23090b57cec5SDimitry Andric   const EVT VTs[] = {MVT::Other, MVT::Glue};
23105ffd83dbSDimitry Andric   SDValue New = CurDAG->getNode(N->getOpcode(), DL, VTs, Ops);
23110b57cec5SDimitry Andric   New->setNodeId(-1);
23120b57cec5SDimitry Andric   ReplaceUses(N, New.getNode());
23130b57cec5SDimitry Andric   CurDAG->RemoveDeadNode(N);
23140b57cec5SDimitry Andric }
23150b57cec5SDimitry Andric 
23160b57cec5SDimitry Andric void SelectionDAGISel::Select_READ_REGISTER(SDNode *Op) {
23170b57cec5SDimitry Andric   SDLoc dl(Op);
2318480093f4SDimitry Andric   MDNodeSDNode *MD = cast<MDNodeSDNode>(Op->getOperand(1));
2319480093f4SDimitry Andric   const MDString *RegStr = cast<MDString>(MD->getMD()->getOperand(0));
2320480093f4SDimitry Andric 
2321480093f4SDimitry Andric   EVT VT = Op->getValueType(0);
2322480093f4SDimitry Andric   LLT Ty = VT.isSimple() ? getLLTForMVT(VT.getSimpleVT()) : LLT();
23238bcb0991SDimitry Andric   Register Reg =
2324480093f4SDimitry Andric       TLI->getRegisterByName(RegStr->getString().data(), Ty,
23258bcb0991SDimitry Andric                              CurDAG->getMachineFunction());
23260b57cec5SDimitry Andric   SDValue New = CurDAG->getCopyFromReg(
23270b57cec5SDimitry Andric                         Op->getOperand(0), dl, Reg, Op->getValueType(0));
23280b57cec5SDimitry Andric   New->setNodeId(-1);
23290b57cec5SDimitry Andric   ReplaceUses(Op, New.getNode());
23300b57cec5SDimitry Andric   CurDAG->RemoveDeadNode(Op);
23310b57cec5SDimitry Andric }
23320b57cec5SDimitry Andric 
23330b57cec5SDimitry Andric void SelectionDAGISel::Select_WRITE_REGISTER(SDNode *Op) {
23340b57cec5SDimitry Andric   SDLoc dl(Op);
2335480093f4SDimitry Andric   MDNodeSDNode *MD = cast<MDNodeSDNode>(Op->getOperand(1));
2336480093f4SDimitry Andric   const MDString *RegStr = cast<MDString>(MD->getMD()->getOperand(0));
2337480093f4SDimitry Andric 
2338480093f4SDimitry Andric   EVT VT = Op->getOperand(2).getValueType();
2339480093f4SDimitry Andric   LLT Ty = VT.isSimple() ? getLLTForMVT(VT.getSimpleVT()) : LLT();
2340480093f4SDimitry Andric 
2341480093f4SDimitry Andric   Register Reg = TLI->getRegisterByName(RegStr->getString().data(), Ty,
23428bcb0991SDimitry Andric                                         CurDAG->getMachineFunction());
23430b57cec5SDimitry Andric   SDValue New = CurDAG->getCopyToReg(
23440b57cec5SDimitry Andric                         Op->getOperand(0), dl, Reg, Op->getOperand(2));
23450b57cec5SDimitry Andric   New->setNodeId(-1);
23460b57cec5SDimitry Andric   ReplaceUses(Op, New.getNode());
23470b57cec5SDimitry Andric   CurDAG->RemoveDeadNode(Op);
23480b57cec5SDimitry Andric }
23490b57cec5SDimitry Andric 
23500b57cec5SDimitry Andric void SelectionDAGISel::Select_UNDEF(SDNode *N) {
23510b57cec5SDimitry Andric   CurDAG->SelectNodeTo(N, TargetOpcode::IMPLICIT_DEF, N->getValueType(0));
23520b57cec5SDimitry Andric }
23530b57cec5SDimitry Andric 
23545ffd83dbSDimitry Andric void SelectionDAGISel::Select_FREEZE(SDNode *N) {
23555ffd83dbSDimitry Andric   // TODO: We don't have FREEZE pseudo-instruction in MachineInstr-level now.
23565ffd83dbSDimitry Andric   // If FREEZE instruction is added later, the code below must be changed as
23575ffd83dbSDimitry Andric   // well.
23585ffd83dbSDimitry Andric   CurDAG->SelectNodeTo(N, TargetOpcode::COPY, N->getValueType(0),
23595ffd83dbSDimitry Andric                        N->getOperand(0));
23605ffd83dbSDimitry Andric }
23615ffd83dbSDimitry Andric 
2362fe6060f1SDimitry Andric void SelectionDAGISel::Select_ARITH_FENCE(SDNode *N) {
2363fe6060f1SDimitry Andric   CurDAG->SelectNodeTo(N, TargetOpcode::ARITH_FENCE, N->getValueType(0),
2364fe6060f1SDimitry Andric                        N->getOperand(0));
2365fe6060f1SDimitry Andric }
2366fe6060f1SDimitry Andric 
2367bdd1243dSDimitry Andric void SelectionDAGISel::Select_MEMBARRIER(SDNode *N) {
2368bdd1243dSDimitry Andric   CurDAG->SelectNodeTo(N, TargetOpcode::MEMBARRIER, N->getValueType(0),
2369bdd1243dSDimitry Andric                        N->getOperand(0));
2370bdd1243dSDimitry Andric }
2371bdd1243dSDimitry Andric 
2372fcaf7f86SDimitry Andric void SelectionDAGISel::pushStackMapLiveVariable(SmallVectorImpl<SDValue> &Ops,
2373fcaf7f86SDimitry Andric                                                 SDValue OpVal, SDLoc DL) {
2374fcaf7f86SDimitry Andric   SDNode *OpNode = OpVal.getNode();
2375fcaf7f86SDimitry Andric 
2376fcaf7f86SDimitry Andric   // FrameIndex nodes should have been directly emitted to TargetFrameIndex
2377fcaf7f86SDimitry Andric   // nodes at DAG-construction time.
2378fcaf7f86SDimitry Andric   assert(OpNode->getOpcode() != ISD::FrameIndex);
2379fcaf7f86SDimitry Andric 
2380fcaf7f86SDimitry Andric   if (OpNode->getOpcode() == ISD::Constant) {
2381fcaf7f86SDimitry Andric     Ops.push_back(
2382fcaf7f86SDimitry Andric         CurDAG->getTargetConstant(StackMaps::ConstantOp, DL, MVT::i64));
2383fcaf7f86SDimitry Andric     Ops.push_back(
2384fcaf7f86SDimitry Andric         CurDAG->getTargetConstant(cast<ConstantSDNode>(OpNode)->getZExtValue(),
2385fcaf7f86SDimitry Andric                                   DL, OpVal.getValueType()));
2386fcaf7f86SDimitry Andric   } else {
2387fcaf7f86SDimitry Andric     Ops.push_back(OpVal);
2388fcaf7f86SDimitry Andric   }
2389fcaf7f86SDimitry Andric }
2390fcaf7f86SDimitry Andric 
2391753f127fSDimitry Andric void SelectionDAGISel::Select_STACKMAP(SDNode *N) {
2392fcaf7f86SDimitry Andric   SmallVector<SDValue, 32> Ops;
2393753f127fSDimitry Andric   auto *It = N->op_begin();
2394753f127fSDimitry Andric   SDLoc DL(N);
2395753f127fSDimitry Andric 
2396753f127fSDimitry Andric   // Stash the chain and glue operands so we can move them to the end.
2397753f127fSDimitry Andric   SDValue Chain = *It++;
239806c3fb27SDimitry Andric   SDValue InGlue = *It++;
2399753f127fSDimitry Andric 
2400753f127fSDimitry Andric   // <id> operand.
2401753f127fSDimitry Andric   SDValue ID = *It++;
2402753f127fSDimitry Andric   assert(ID.getValueType() == MVT::i64);
2403753f127fSDimitry Andric   Ops.push_back(ID);
2404753f127fSDimitry Andric 
2405753f127fSDimitry Andric   // <numShadowBytes> operand.
2406753f127fSDimitry Andric   SDValue Shad = *It++;
2407753f127fSDimitry Andric   assert(Shad.getValueType() == MVT::i32);
2408753f127fSDimitry Andric   Ops.push_back(Shad);
2409753f127fSDimitry Andric 
2410753f127fSDimitry Andric   // Live variable operands.
2411fcaf7f86SDimitry Andric   for (; It != N->op_end(); It++)
2412fcaf7f86SDimitry Andric     pushStackMapLiveVariable(Ops, *It, DL);
2413753f127fSDimitry Andric 
2414753f127fSDimitry Andric   Ops.push_back(Chain);
241506c3fb27SDimitry Andric   Ops.push_back(InGlue);
2416753f127fSDimitry Andric 
2417753f127fSDimitry Andric   SDVTList NodeTys = CurDAG->getVTList(MVT::Other, MVT::Glue);
2418753f127fSDimitry Andric   CurDAG->SelectNodeTo(N, TargetOpcode::STACKMAP, NodeTys, Ops);
2419753f127fSDimitry Andric }
2420753f127fSDimitry Andric 
2421fcaf7f86SDimitry Andric void SelectionDAGISel::Select_PATCHPOINT(SDNode *N) {
2422fcaf7f86SDimitry Andric   SmallVector<SDValue, 32> Ops;
2423fcaf7f86SDimitry Andric   auto *It = N->op_begin();
2424fcaf7f86SDimitry Andric   SDLoc DL(N);
2425fcaf7f86SDimitry Andric 
2426fcaf7f86SDimitry Andric   // Cache arguments that will be moved to the end in the target node.
2427fcaf7f86SDimitry Andric   SDValue Chain = *It++;
2428bdd1243dSDimitry Andric   std::optional<SDValue> Glue;
2429fcaf7f86SDimitry Andric   if (It->getValueType() == MVT::Glue)
2430fcaf7f86SDimitry Andric     Glue = *It++;
2431fcaf7f86SDimitry Andric   SDValue RegMask = *It++;
2432fcaf7f86SDimitry Andric 
2433fcaf7f86SDimitry Andric   // <id> operand.
2434fcaf7f86SDimitry Andric   SDValue ID = *It++;
2435fcaf7f86SDimitry Andric   assert(ID.getValueType() == MVT::i64);
2436fcaf7f86SDimitry Andric   Ops.push_back(ID);
2437fcaf7f86SDimitry Andric 
2438fcaf7f86SDimitry Andric   // <numShadowBytes> operand.
2439fcaf7f86SDimitry Andric   SDValue Shad = *It++;
2440fcaf7f86SDimitry Andric   assert(Shad.getValueType() == MVT::i32);
2441fcaf7f86SDimitry Andric   Ops.push_back(Shad);
2442fcaf7f86SDimitry Andric 
2443fcaf7f86SDimitry Andric   // Add the callee.
2444fcaf7f86SDimitry Andric   Ops.push_back(*It++);
2445fcaf7f86SDimitry Andric 
2446fcaf7f86SDimitry Andric   // Add <numArgs>.
2447fcaf7f86SDimitry Andric   SDValue NumArgs = *It++;
2448fcaf7f86SDimitry Andric   assert(NumArgs.getValueType() == MVT::i32);
2449fcaf7f86SDimitry Andric   Ops.push_back(NumArgs);
2450fcaf7f86SDimitry Andric 
2451fcaf7f86SDimitry Andric   // Calling convention.
2452fcaf7f86SDimitry Andric   Ops.push_back(*It++);
2453fcaf7f86SDimitry Andric 
2454fcaf7f86SDimitry Andric   // Push the args for the call.
2455fcaf7f86SDimitry Andric   for (uint64_t I = cast<ConstantSDNode>(NumArgs)->getZExtValue(); I != 0; I--)
2456fcaf7f86SDimitry Andric     Ops.push_back(*It++);
2457fcaf7f86SDimitry Andric 
2458fcaf7f86SDimitry Andric   // Now push the live variables.
2459fcaf7f86SDimitry Andric   for (; It != N->op_end(); It++)
2460fcaf7f86SDimitry Andric     pushStackMapLiveVariable(Ops, *It, DL);
2461fcaf7f86SDimitry Andric 
2462fcaf7f86SDimitry Andric   // Finally, the regmask, chain and (if present) glue are moved to the end.
2463fcaf7f86SDimitry Andric   Ops.push_back(RegMask);
2464fcaf7f86SDimitry Andric   Ops.push_back(Chain);
2465fcaf7f86SDimitry Andric   if (Glue.has_value())
2466bdd1243dSDimitry Andric     Ops.push_back(*Glue);
2467fcaf7f86SDimitry Andric 
2468fcaf7f86SDimitry Andric   SDVTList NodeTys = N->getVTList();
2469fcaf7f86SDimitry Andric   CurDAG->SelectNodeTo(N, TargetOpcode::PATCHPOINT, NodeTys, Ops);
2470fcaf7f86SDimitry Andric }
2471fcaf7f86SDimitry Andric 
24720b57cec5SDimitry Andric /// GetVBR - decode a vbr encoding whose top bit is set.
2473e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static uint64_t
24740b57cec5SDimitry Andric GetVBR(uint64_t Val, const unsigned char *MatcherTable, unsigned &Idx) {
24750b57cec5SDimitry Andric   assert(Val >= 128 && "Not a VBR");
24760b57cec5SDimitry Andric   Val &= 127;  // Remove first vbr bit.
24770b57cec5SDimitry Andric 
24780b57cec5SDimitry Andric   unsigned Shift = 7;
24790b57cec5SDimitry Andric   uint64_t NextBits;
24800b57cec5SDimitry Andric   do {
24810b57cec5SDimitry Andric     NextBits = MatcherTable[Idx++];
24820b57cec5SDimitry Andric     Val |= (NextBits&127) << Shift;
24830b57cec5SDimitry Andric     Shift += 7;
24840b57cec5SDimitry Andric   } while (NextBits & 128);
24850b57cec5SDimitry Andric 
24860b57cec5SDimitry Andric   return Val;
24870b57cec5SDimitry Andric }
24880b57cec5SDimitry Andric 
2489*5f757f3fSDimitry Andric void SelectionDAGISel::Select_JUMP_TABLE_DEBUG_INFO(SDNode *N) {
2490*5f757f3fSDimitry Andric   SDLoc dl(N);
2491*5f757f3fSDimitry Andric   CurDAG->SelectNodeTo(N, TargetOpcode::JUMP_TABLE_DEBUG_INFO, MVT::Glue,
2492*5f757f3fSDimitry Andric                        CurDAG->getTargetConstant(N->getConstantOperandVal(1),
2493*5f757f3fSDimitry Andric                                                  dl, MVT::i64, true));
2494*5f757f3fSDimitry Andric }
2495*5f757f3fSDimitry Andric 
24960b57cec5SDimitry Andric /// When a match is complete, this method updates uses of interior chain results
24970b57cec5SDimitry Andric /// to use the new results.
24980b57cec5SDimitry Andric void SelectionDAGISel::UpdateChains(
24990b57cec5SDimitry Andric     SDNode *NodeToMatch, SDValue InputChain,
25000b57cec5SDimitry Andric     SmallVectorImpl<SDNode *> &ChainNodesMatched, bool isMorphNodeTo) {
25010b57cec5SDimitry Andric   SmallVector<SDNode*, 4> NowDeadNodes;
25020b57cec5SDimitry Andric 
25030b57cec5SDimitry Andric   // Now that all the normal results are replaced, we replace the chain and
25040b57cec5SDimitry Andric   // glue results if present.
25050b57cec5SDimitry Andric   if (!ChainNodesMatched.empty()) {
25060b57cec5SDimitry Andric     assert(InputChain.getNode() &&
25070b57cec5SDimitry Andric            "Matched input chains but didn't produce a chain");
25080b57cec5SDimitry Andric     // Loop over all of the nodes we matched that produced a chain result.
25090b57cec5SDimitry Andric     // Replace all the chain results with the final chain we ended up with.
25100b57cec5SDimitry Andric     for (unsigned i = 0, e = ChainNodesMatched.size(); i != e; ++i) {
25110b57cec5SDimitry Andric       SDNode *ChainNode = ChainNodesMatched[i];
25120b57cec5SDimitry Andric       // If ChainNode is null, it's because we replaced it on a previous
25130b57cec5SDimitry Andric       // iteration and we cleared it out of the map. Just skip it.
25140b57cec5SDimitry Andric       if (!ChainNode)
25150b57cec5SDimitry Andric         continue;
25160b57cec5SDimitry Andric 
25170b57cec5SDimitry Andric       assert(ChainNode->getOpcode() != ISD::DELETED_NODE &&
25180b57cec5SDimitry Andric              "Deleted node left in chain");
25190b57cec5SDimitry Andric 
25200b57cec5SDimitry Andric       // Don't replace the results of the root node if we're doing a
25210b57cec5SDimitry Andric       // MorphNodeTo.
25220b57cec5SDimitry Andric       if (ChainNode == NodeToMatch && isMorphNodeTo)
25230b57cec5SDimitry Andric         continue;
25240b57cec5SDimitry Andric 
25250b57cec5SDimitry Andric       SDValue ChainVal = SDValue(ChainNode, ChainNode->getNumValues()-1);
25260b57cec5SDimitry Andric       if (ChainVal.getValueType() == MVT::Glue)
25270b57cec5SDimitry Andric         ChainVal = ChainVal.getValue(ChainVal->getNumValues()-2);
25280b57cec5SDimitry Andric       assert(ChainVal.getValueType() == MVT::Other && "Not a chain?");
25290b57cec5SDimitry Andric       SelectionDAG::DAGNodeDeletedListener NDL(
25300b57cec5SDimitry Andric           *CurDAG, [&](SDNode *N, SDNode *E) {
25310b57cec5SDimitry Andric             std::replace(ChainNodesMatched.begin(), ChainNodesMatched.end(), N,
25320b57cec5SDimitry Andric                          static_cast<SDNode *>(nullptr));
25330b57cec5SDimitry Andric           });
25340b57cec5SDimitry Andric       if (ChainNode->getOpcode() != ISD::TokenFactor)
25350b57cec5SDimitry Andric         ReplaceUses(ChainVal, InputChain);
25360b57cec5SDimitry Andric 
25370b57cec5SDimitry Andric       // If the node became dead and we haven't already seen it, delete it.
25380b57cec5SDimitry Andric       if (ChainNode != NodeToMatch && ChainNode->use_empty() &&
2539e8d8bef9SDimitry Andric           !llvm::is_contained(NowDeadNodes, ChainNode))
25400b57cec5SDimitry Andric         NowDeadNodes.push_back(ChainNode);
25410b57cec5SDimitry Andric     }
25420b57cec5SDimitry Andric   }
25430b57cec5SDimitry Andric 
25440b57cec5SDimitry Andric   if (!NowDeadNodes.empty())
25450b57cec5SDimitry Andric     CurDAG->RemoveDeadNodes(NowDeadNodes);
25460b57cec5SDimitry Andric 
25470b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "ISEL: Match complete!\n");
25480b57cec5SDimitry Andric }
25490b57cec5SDimitry Andric 
25500b57cec5SDimitry Andric /// HandleMergeInputChains - This implements the OPC_EmitMergeInputChains
25510b57cec5SDimitry Andric /// operation for when the pattern matched at least one node with a chains.  The
25520b57cec5SDimitry Andric /// input vector contains a list of all of the chained nodes that we match.  We
25530b57cec5SDimitry Andric /// must determine if this is a valid thing to cover (i.e. matching it won't
25540b57cec5SDimitry Andric /// induce cycles in the DAG) and if so, creating a TokenFactor node. that will
25550b57cec5SDimitry Andric /// be used as the input node chain for the generated nodes.
25560b57cec5SDimitry Andric static SDValue
25570b57cec5SDimitry Andric HandleMergeInputChains(SmallVectorImpl<SDNode*> &ChainNodesMatched,
25580b57cec5SDimitry Andric                        SelectionDAG *CurDAG) {
25590b57cec5SDimitry Andric 
25600b57cec5SDimitry Andric   SmallPtrSet<const SDNode *, 16> Visited;
25610b57cec5SDimitry Andric   SmallVector<const SDNode *, 8> Worklist;
25620b57cec5SDimitry Andric   SmallVector<SDValue, 3> InputChains;
25630b57cec5SDimitry Andric   unsigned int Max = 8192;
25640b57cec5SDimitry Andric 
25650b57cec5SDimitry Andric   // Quick exit on trivial merge.
25660b57cec5SDimitry Andric   if (ChainNodesMatched.size() == 1)
25670b57cec5SDimitry Andric     return ChainNodesMatched[0]->getOperand(0);
25680b57cec5SDimitry Andric 
25690b57cec5SDimitry Andric   // Add chains that aren't already added (internal). Peek through
25700b57cec5SDimitry Andric   // token factors.
25710b57cec5SDimitry Andric   std::function<void(const SDValue)> AddChains = [&](const SDValue V) {
25720b57cec5SDimitry Andric     if (V.getValueType() != MVT::Other)
25730b57cec5SDimitry Andric       return;
25740b57cec5SDimitry Andric     if (V->getOpcode() == ISD::EntryToken)
25750b57cec5SDimitry Andric       return;
25760b57cec5SDimitry Andric     if (!Visited.insert(V.getNode()).second)
25770b57cec5SDimitry Andric       return;
25780b57cec5SDimitry Andric     if (V->getOpcode() == ISD::TokenFactor) {
25790b57cec5SDimitry Andric       for (const SDValue &Op : V->op_values())
25800b57cec5SDimitry Andric         AddChains(Op);
25810b57cec5SDimitry Andric     } else
25820b57cec5SDimitry Andric       InputChains.push_back(V);
25830b57cec5SDimitry Andric   };
25840b57cec5SDimitry Andric 
25850b57cec5SDimitry Andric   for (auto *N : ChainNodesMatched) {
25860b57cec5SDimitry Andric     Worklist.push_back(N);
25870b57cec5SDimitry Andric     Visited.insert(N);
25880b57cec5SDimitry Andric   }
25890b57cec5SDimitry Andric 
25900b57cec5SDimitry Andric   while (!Worklist.empty())
25910b57cec5SDimitry Andric     AddChains(Worklist.pop_back_val()->getOperand(0));
25920b57cec5SDimitry Andric 
25930b57cec5SDimitry Andric   // Skip the search if there are no chain dependencies.
25940b57cec5SDimitry Andric   if (InputChains.size() == 0)
25950b57cec5SDimitry Andric     return CurDAG->getEntryNode();
25960b57cec5SDimitry Andric 
25970b57cec5SDimitry Andric   // If one of these chains is a successor of input, we must have a
25980b57cec5SDimitry Andric   // node that is both the predecessor and successor of the
25990b57cec5SDimitry Andric   // to-be-merged nodes. Fail.
26000b57cec5SDimitry Andric   Visited.clear();
26010b57cec5SDimitry Andric   for (SDValue V : InputChains)
26020b57cec5SDimitry Andric     Worklist.push_back(V.getNode());
26030b57cec5SDimitry Andric 
26040b57cec5SDimitry Andric   for (auto *N : ChainNodesMatched)
26050b57cec5SDimitry Andric     if (SDNode::hasPredecessorHelper(N, Visited, Worklist, Max, true))
26060b57cec5SDimitry Andric       return SDValue();
26070b57cec5SDimitry Andric 
26080b57cec5SDimitry Andric   // Return merged chain.
26090b57cec5SDimitry Andric   if (InputChains.size() == 1)
26100b57cec5SDimitry Andric     return InputChains[0];
26110b57cec5SDimitry Andric   return CurDAG->getNode(ISD::TokenFactor, SDLoc(ChainNodesMatched[0]),
26120b57cec5SDimitry Andric                          MVT::Other, InputChains);
26130b57cec5SDimitry Andric }
26140b57cec5SDimitry Andric 
26150b57cec5SDimitry Andric /// MorphNode - Handle morphing a node in place for the selector.
26160b57cec5SDimitry Andric SDNode *SelectionDAGISel::
26170b57cec5SDimitry Andric MorphNode(SDNode *Node, unsigned TargetOpc, SDVTList VTList,
26180b57cec5SDimitry Andric           ArrayRef<SDValue> Ops, unsigned EmitNodeInfo) {
26190b57cec5SDimitry Andric   // It is possible we're using MorphNodeTo to replace a node with no
26200b57cec5SDimitry Andric   // normal results with one that has a normal result (or we could be
26210b57cec5SDimitry Andric   // adding a chain) and the input could have glue and chains as well.
26220b57cec5SDimitry Andric   // In this case we need to shift the operands down.
26230b57cec5SDimitry Andric   // FIXME: This is a horrible hack and broken in obscure cases, no worse
26240b57cec5SDimitry Andric   // than the old isel though.
26250b57cec5SDimitry Andric   int OldGlueResultNo = -1, OldChainResultNo = -1;
26260b57cec5SDimitry Andric 
26270b57cec5SDimitry Andric   unsigned NTMNumResults = Node->getNumValues();
26280b57cec5SDimitry Andric   if (Node->getValueType(NTMNumResults-1) == MVT::Glue) {
26290b57cec5SDimitry Andric     OldGlueResultNo = NTMNumResults-1;
26300b57cec5SDimitry Andric     if (NTMNumResults != 1 &&
26310b57cec5SDimitry Andric         Node->getValueType(NTMNumResults-2) == MVT::Other)
26320b57cec5SDimitry Andric       OldChainResultNo = NTMNumResults-2;
26330b57cec5SDimitry Andric   } else if (Node->getValueType(NTMNumResults-1) == MVT::Other)
26340b57cec5SDimitry Andric     OldChainResultNo = NTMNumResults-1;
26350b57cec5SDimitry Andric 
26360b57cec5SDimitry Andric   // Call the underlying SelectionDAG routine to do the transmogrification. Note
26370b57cec5SDimitry Andric   // that this deletes operands of the old node that become dead.
26380b57cec5SDimitry Andric   SDNode *Res = CurDAG->MorphNodeTo(Node, ~TargetOpc, VTList, Ops);
26390b57cec5SDimitry Andric 
26400b57cec5SDimitry Andric   // MorphNodeTo can operate in two ways: if an existing node with the
26410b57cec5SDimitry Andric   // specified operands exists, it can just return it.  Otherwise, it
26420b57cec5SDimitry Andric   // updates the node in place to have the requested operands.
26430b57cec5SDimitry Andric   if (Res == Node) {
26440b57cec5SDimitry Andric     // If we updated the node in place, reset the node ID.  To the isel,
26450b57cec5SDimitry Andric     // this should be just like a newly allocated machine node.
26460b57cec5SDimitry Andric     Res->setNodeId(-1);
26470b57cec5SDimitry Andric   }
26480b57cec5SDimitry Andric 
26490b57cec5SDimitry Andric   unsigned ResNumResults = Res->getNumValues();
26500b57cec5SDimitry Andric   // Move the glue if needed.
26510b57cec5SDimitry Andric   if ((EmitNodeInfo & OPFL_GlueOutput) && OldGlueResultNo != -1 &&
2652*5f757f3fSDimitry Andric       static_cast<unsigned>(OldGlueResultNo) != ResNumResults - 1)
26530b57cec5SDimitry Andric     ReplaceUses(SDValue(Node, OldGlueResultNo),
26540b57cec5SDimitry Andric                 SDValue(Res, ResNumResults - 1));
26550b57cec5SDimitry Andric 
26560b57cec5SDimitry Andric   if ((EmitNodeInfo & OPFL_GlueOutput) != 0)
26570b57cec5SDimitry Andric     --ResNumResults;
26580b57cec5SDimitry Andric 
26590b57cec5SDimitry Andric   // Move the chain reference if needed.
26600b57cec5SDimitry Andric   if ((EmitNodeInfo & OPFL_Chain) && OldChainResultNo != -1 &&
2661*5f757f3fSDimitry Andric       static_cast<unsigned>(OldChainResultNo) != ResNumResults - 1)
26620b57cec5SDimitry Andric     ReplaceUses(SDValue(Node, OldChainResultNo),
26630b57cec5SDimitry Andric                 SDValue(Res, ResNumResults - 1));
26640b57cec5SDimitry Andric 
26650b57cec5SDimitry Andric   // Otherwise, no replacement happened because the node already exists. Replace
26660b57cec5SDimitry Andric   // Uses of the old node with the new one.
26670b57cec5SDimitry Andric   if (Res != Node) {
26680b57cec5SDimitry Andric     ReplaceNode(Node, Res);
26690b57cec5SDimitry Andric   } else {
26700b57cec5SDimitry Andric     EnforceNodeIdInvariant(Res);
26710b57cec5SDimitry Andric   }
26720b57cec5SDimitry Andric 
26730b57cec5SDimitry Andric   return Res;
26740b57cec5SDimitry Andric }
26750b57cec5SDimitry Andric 
26760b57cec5SDimitry Andric /// CheckSame - Implements OP_CheckSame.
2677e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
2678e8d8bef9SDimitry Andric CheckSame(const unsigned char *MatcherTable, unsigned &MatcherIndex, SDValue N,
26790b57cec5SDimitry Andric           const SmallVectorImpl<std::pair<SDValue, SDNode *>> &RecordedNodes) {
26800b57cec5SDimitry Andric   // Accept if it is exactly the same as a previously recorded node.
26810b57cec5SDimitry Andric   unsigned RecNo = MatcherTable[MatcherIndex++];
26820b57cec5SDimitry Andric   assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
26830b57cec5SDimitry Andric   return N == RecordedNodes[RecNo].first;
26840b57cec5SDimitry Andric }
26850b57cec5SDimitry Andric 
26860b57cec5SDimitry Andric /// CheckChildSame - Implements OP_CheckChildXSame.
2687e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool CheckChildSame(
2688e8d8bef9SDimitry Andric     const unsigned char *MatcherTable, unsigned &MatcherIndex, SDValue N,
26890b57cec5SDimitry Andric     const SmallVectorImpl<std::pair<SDValue, SDNode *>> &RecordedNodes,
26900b57cec5SDimitry Andric     unsigned ChildNo) {
26910b57cec5SDimitry Andric   if (ChildNo >= N.getNumOperands())
26920b57cec5SDimitry Andric     return false;  // Match fails if out of range child #.
26930b57cec5SDimitry Andric   return ::CheckSame(MatcherTable, MatcherIndex, N.getOperand(ChildNo),
26940b57cec5SDimitry Andric                      RecordedNodes);
26950b57cec5SDimitry Andric }
26960b57cec5SDimitry Andric 
26970b57cec5SDimitry Andric /// CheckPatternPredicate - Implements OP_CheckPatternPredicate.
2698e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
26990b57cec5SDimitry Andric CheckPatternPredicate(const unsigned char *MatcherTable, unsigned &MatcherIndex,
2700*5f757f3fSDimitry Andric                       const SelectionDAGISel &SDISel, bool TwoBytePredNo) {
2701*5f757f3fSDimitry Andric   unsigned PredNo = MatcherTable[MatcherIndex++];
2702*5f757f3fSDimitry Andric   if (TwoBytePredNo)
2703*5f757f3fSDimitry Andric     PredNo |= MatcherTable[MatcherIndex++] << 8;
2704*5f757f3fSDimitry Andric   return SDISel.CheckPatternPredicate(PredNo);
27050b57cec5SDimitry Andric }
27060b57cec5SDimitry Andric 
27070b57cec5SDimitry Andric /// CheckNodePredicate - Implements OP_CheckNodePredicate.
2708e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
27090b57cec5SDimitry Andric CheckNodePredicate(const unsigned char *MatcherTable, unsigned &MatcherIndex,
27100b57cec5SDimitry Andric                    const SelectionDAGISel &SDISel, SDNode *N) {
27110b57cec5SDimitry Andric   return SDISel.CheckNodePredicate(N, MatcherTable[MatcherIndex++]);
27120b57cec5SDimitry Andric }
27130b57cec5SDimitry Andric 
2714e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
27150b57cec5SDimitry Andric CheckOpcode(const unsigned char *MatcherTable, unsigned &MatcherIndex,
27160b57cec5SDimitry Andric             SDNode *N) {
27170b57cec5SDimitry Andric   uint16_t Opc = MatcherTable[MatcherIndex++];
2718*5f757f3fSDimitry Andric   Opc |= static_cast<uint16_t>(MatcherTable[MatcherIndex++]) << 8;
27190b57cec5SDimitry Andric   return N->getOpcode() == Opc;
27200b57cec5SDimitry Andric }
27210b57cec5SDimitry Andric 
2722*5f757f3fSDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool CheckType(MVT::SimpleValueType VT,
2723*5f757f3fSDimitry Andric                                                    SDValue N,
2724*5f757f3fSDimitry Andric                                                    const TargetLowering *TLI,
2725*5f757f3fSDimitry Andric                                                    const DataLayout &DL) {
2726*5f757f3fSDimitry Andric   if (N.getValueType() == VT)
2727*5f757f3fSDimitry Andric     return true;
27280b57cec5SDimitry Andric 
27290b57cec5SDimitry Andric   // Handle the case when VT is iPTR.
27300b57cec5SDimitry Andric   return VT == MVT::iPTR && N.getValueType() == TLI->getPointerTy(DL);
27310b57cec5SDimitry Andric }
27320b57cec5SDimitry Andric 
2733e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
2734*5f757f3fSDimitry Andric CheckChildType(MVT::SimpleValueType VT, SDValue N, const TargetLowering *TLI,
2735*5f757f3fSDimitry Andric                const DataLayout &DL, unsigned ChildNo) {
27360b57cec5SDimitry Andric   if (ChildNo >= N.getNumOperands())
27370b57cec5SDimitry Andric     return false; // Match fails if out of range child #.
2738*5f757f3fSDimitry Andric   return ::CheckType(VT, N.getOperand(ChildNo), TLI, DL);
27390b57cec5SDimitry Andric }
27400b57cec5SDimitry Andric 
2741e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
27420b57cec5SDimitry Andric CheckCondCode(const unsigned char *MatcherTable, unsigned &MatcherIndex,
27430b57cec5SDimitry Andric               SDValue N) {
27440b57cec5SDimitry Andric   return cast<CondCodeSDNode>(N)->get() ==
2745*5f757f3fSDimitry Andric          static_cast<ISD::CondCode>(MatcherTable[MatcherIndex++]);
27460b57cec5SDimitry Andric }
27470b57cec5SDimitry Andric 
2748e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
27490b57cec5SDimitry Andric CheckChild2CondCode(const unsigned char *MatcherTable, unsigned &MatcherIndex,
27500b57cec5SDimitry Andric                     SDValue N) {
27510b57cec5SDimitry Andric   if (2 >= N.getNumOperands())
27520b57cec5SDimitry Andric     return false;
27530b57cec5SDimitry Andric   return ::CheckCondCode(MatcherTable, MatcherIndex, N.getOperand(2));
27540b57cec5SDimitry Andric }
27550b57cec5SDimitry Andric 
2756e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
27570b57cec5SDimitry Andric CheckValueType(const unsigned char *MatcherTable, unsigned &MatcherIndex,
27580b57cec5SDimitry Andric                SDValue N, const TargetLowering *TLI, const DataLayout &DL) {
2759*5f757f3fSDimitry Andric   MVT::SimpleValueType VT =
2760*5f757f3fSDimitry Andric       static_cast<MVT::SimpleValueType>(MatcherTable[MatcherIndex++]);
27610b57cec5SDimitry Andric   if (cast<VTSDNode>(N)->getVT() == VT)
27620b57cec5SDimitry Andric     return true;
27630b57cec5SDimitry Andric 
27640b57cec5SDimitry Andric   // Handle the case when VT is iPTR.
27650b57cec5SDimitry Andric   return VT == MVT::iPTR && cast<VTSDNode>(N)->getVT() == TLI->getPointerTy(DL);
27660b57cec5SDimitry Andric }
27670b57cec5SDimitry Andric 
2768fe6060f1SDimitry Andric // Bit 0 stores the sign of the immediate. The upper bits contain the magnitude
2769fe6060f1SDimitry Andric // shifted left by 1.
2770fe6060f1SDimitry Andric static uint64_t decodeSignRotatedValue(uint64_t V) {
2771fe6060f1SDimitry Andric   if ((V & 1) == 0)
2772fe6060f1SDimitry Andric     return V >> 1;
2773fe6060f1SDimitry Andric   if (V != 1)
2774fe6060f1SDimitry Andric     return -(V >> 1);
2775fe6060f1SDimitry Andric   // There is no such thing as -0 with integers.  "-0" really means MININT.
2776fe6060f1SDimitry Andric   return 1ULL << 63;
2777fe6060f1SDimitry Andric }
2778fe6060f1SDimitry Andric 
2779e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
27800b57cec5SDimitry Andric CheckInteger(const unsigned char *MatcherTable, unsigned &MatcherIndex,
27810b57cec5SDimitry Andric              SDValue N) {
27820b57cec5SDimitry Andric   int64_t Val = MatcherTable[MatcherIndex++];
27830b57cec5SDimitry Andric   if (Val & 128)
27840b57cec5SDimitry Andric     Val = GetVBR(Val, MatcherTable, MatcherIndex);
27850b57cec5SDimitry Andric 
2786fe6060f1SDimitry Andric   Val = decodeSignRotatedValue(Val);
2787fe6060f1SDimitry Andric 
27880b57cec5SDimitry Andric   ConstantSDNode *C = dyn_cast<ConstantSDNode>(N);
27890b57cec5SDimitry Andric   return C && C->getSExtValue() == Val;
27900b57cec5SDimitry Andric }
27910b57cec5SDimitry Andric 
2792e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
27930b57cec5SDimitry Andric CheckChildInteger(const unsigned char *MatcherTable, unsigned &MatcherIndex,
27940b57cec5SDimitry Andric                   SDValue N, unsigned ChildNo) {
27950b57cec5SDimitry Andric   if (ChildNo >= N.getNumOperands())
27960b57cec5SDimitry Andric     return false;  // Match fails if out of range child #.
27970b57cec5SDimitry Andric   return ::CheckInteger(MatcherTable, MatcherIndex, N.getOperand(ChildNo));
27980b57cec5SDimitry Andric }
27990b57cec5SDimitry Andric 
2800e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
28010b57cec5SDimitry Andric CheckAndImm(const unsigned char *MatcherTable, unsigned &MatcherIndex,
28020b57cec5SDimitry Andric             SDValue N, const SelectionDAGISel &SDISel) {
28030b57cec5SDimitry Andric   int64_t Val = MatcherTable[MatcherIndex++];
28040b57cec5SDimitry Andric   if (Val & 128)
28050b57cec5SDimitry Andric     Val = GetVBR(Val, MatcherTable, MatcherIndex);
28060b57cec5SDimitry Andric 
28070b57cec5SDimitry Andric   if (N->getOpcode() != ISD::AND) return false;
28080b57cec5SDimitry Andric 
28090b57cec5SDimitry Andric   ConstantSDNode *C = dyn_cast<ConstantSDNode>(N->getOperand(1));
28100b57cec5SDimitry Andric   return C && SDISel.CheckAndMask(N.getOperand(0), C, Val);
28110b57cec5SDimitry Andric }
28120b57cec5SDimitry Andric 
2813e8d8bef9SDimitry Andric LLVM_ATTRIBUTE_ALWAYS_INLINE static bool
2814e8d8bef9SDimitry Andric CheckOrImm(const unsigned char *MatcherTable, unsigned &MatcherIndex, SDValue N,
2815e8d8bef9SDimitry Andric            const SelectionDAGISel &SDISel) {
28160b57cec5SDimitry Andric   int64_t Val = MatcherTable[MatcherIndex++];
28170b57cec5SDimitry Andric   if (Val & 128)
28180b57cec5SDimitry Andric     Val = GetVBR(Val, MatcherTable, MatcherIndex);
28190b57cec5SDimitry Andric 
28200b57cec5SDimitry Andric   if (N->getOpcode() != ISD::OR) return false;
28210b57cec5SDimitry Andric 
28220b57cec5SDimitry Andric   ConstantSDNode *C = dyn_cast<ConstantSDNode>(N->getOperand(1));
28230b57cec5SDimitry Andric   return C && SDISel.CheckOrMask(N.getOperand(0), C, Val);
28240b57cec5SDimitry Andric }
28250b57cec5SDimitry Andric 
28260b57cec5SDimitry Andric /// IsPredicateKnownToFail - If we know how and can do so without pushing a
28270b57cec5SDimitry Andric /// scope, evaluate the current node.  If the current predicate is known to
28280b57cec5SDimitry Andric /// fail, set Result=true and return anything.  If the current predicate is
28290b57cec5SDimitry Andric /// known to pass, set Result=false and return the MatcherIndex to continue
28300b57cec5SDimitry Andric /// with.  If the current predicate is unknown, set Result=false and return the
28310b57cec5SDimitry Andric /// MatcherIndex to continue with.
28320b57cec5SDimitry Andric static unsigned IsPredicateKnownToFail(const unsigned char *Table,
28330b57cec5SDimitry Andric                                        unsigned Index, SDValue N,
28340b57cec5SDimitry Andric                                        bool &Result,
28350b57cec5SDimitry Andric                                        const SelectionDAGISel &SDISel,
28360b57cec5SDimitry Andric                   SmallVectorImpl<std::pair<SDValue, SDNode*>> &RecordedNodes) {
2837*5f757f3fSDimitry Andric   unsigned Opcode = Table[Index++];
2838*5f757f3fSDimitry Andric   switch (Opcode) {
28390b57cec5SDimitry Andric   default:
28400b57cec5SDimitry Andric     Result = false;
28410b57cec5SDimitry Andric     return Index-1;  // Could not evaluate this predicate.
28420b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckSame:
28430b57cec5SDimitry Andric     Result = !::CheckSame(Table, Index, N, RecordedNodes);
28440b57cec5SDimitry Andric     return Index;
28450b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild0Same:
28460b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild1Same:
28470b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild2Same:
28480b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild3Same:
28490b57cec5SDimitry Andric     Result = !::CheckChildSame(Table, Index, N, RecordedNodes,
28500b57cec5SDimitry Andric                         Table[Index-1] - SelectionDAGISel::OPC_CheckChild0Same);
28510b57cec5SDimitry Andric     return Index;
28520b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckPatternPredicate:
2853*5f757f3fSDimitry Andric   case SelectionDAGISel::OPC_CheckPatternPredicate2:
2854*5f757f3fSDimitry Andric     Result = !::CheckPatternPredicate(
2855*5f757f3fSDimitry Andric         Table, Index, SDISel,
2856*5f757f3fSDimitry Andric         Table[Index - 1] == SelectionDAGISel::OPC_CheckPatternPredicate2);
28570b57cec5SDimitry Andric     return Index;
28580b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckPredicate:
28590b57cec5SDimitry Andric     Result = !::CheckNodePredicate(Table, Index, SDISel, N.getNode());
28600b57cec5SDimitry Andric     return Index;
28610b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckOpcode:
28620b57cec5SDimitry Andric     Result = !::CheckOpcode(Table, Index, N.getNode());
28630b57cec5SDimitry Andric     return Index;
28640b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckType:
2865*5f757f3fSDimitry Andric   case SelectionDAGISel::OPC_CheckTypeI32:
2866*5f757f3fSDimitry Andric   case SelectionDAGISel::OPC_CheckTypeI64: {
2867*5f757f3fSDimitry Andric     MVT::SimpleValueType VT;
2868*5f757f3fSDimitry Andric     switch (Opcode) {
2869*5f757f3fSDimitry Andric     case SelectionDAGISel::OPC_CheckTypeI32:
2870*5f757f3fSDimitry Andric       VT = MVT::i32;
2871*5f757f3fSDimitry Andric       break;
2872*5f757f3fSDimitry Andric     case SelectionDAGISel::OPC_CheckTypeI64:
2873*5f757f3fSDimitry Andric       VT = MVT::i64;
2874*5f757f3fSDimitry Andric       break;
2875*5f757f3fSDimitry Andric     default:
2876*5f757f3fSDimitry Andric       VT = static_cast<MVT::SimpleValueType>(Table[Index++]);
2877*5f757f3fSDimitry Andric       break;
2878*5f757f3fSDimitry Andric     }
2879*5f757f3fSDimitry Andric     Result = !::CheckType(VT, N, SDISel.TLI, SDISel.CurDAG->getDataLayout());
28800b57cec5SDimitry Andric     return Index;
2881*5f757f3fSDimitry Andric   }
28820b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckTypeRes: {
28830b57cec5SDimitry Andric     unsigned Res = Table[Index++];
2884*5f757f3fSDimitry Andric     Result = !::CheckType(static_cast<MVT::SimpleValueType>(Table[Index++]),
2885*5f757f3fSDimitry Andric                           N.getValue(Res), SDISel.TLI,
28860b57cec5SDimitry Andric                           SDISel.CurDAG->getDataLayout());
28870b57cec5SDimitry Andric     return Index;
28880b57cec5SDimitry Andric   }
28890b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild0Type:
28900b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild1Type:
28910b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild2Type:
28920b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild3Type:
28930b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild4Type:
28940b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild5Type:
28950b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild6Type:
28960b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild7Type:
2897*5f757f3fSDimitry Andric   case SelectionDAGISel::OPC_CheckChild0TypeI32:
2898*5f757f3fSDimitry Andric   case SelectionDAGISel::OPC_CheckChild1TypeI32:
2899*5f757f3fSDimitry Andric   case SelectionDAGISel::OPC_CheckChild2TypeI32:
2900*5f757f3fSDimitry Andric   case SelectionDAGISel::OPC_CheckChild3TypeI32:
2901*5f757f3fSDimitry Andric   case SelectionDAGISel::OPC_CheckChild4TypeI32:
2902*5f757f3fSDimitry Andric   case SelectionDAGISel::OPC_CheckChild5TypeI32:
2903*5f757f3fSDimitry Andric   case SelectionDAGISel::OPC_CheckChild6TypeI32:
2904*5f757f3fSDimitry Andric   case SelectionDAGISel::OPC_CheckChild7TypeI32:
2905*5f757f3fSDimitry Andric   case SelectionDAGISel::OPC_CheckChild0TypeI64:
2906*5f757f3fSDimitry Andric   case SelectionDAGISel::OPC_CheckChild1TypeI64:
2907*5f757f3fSDimitry Andric   case SelectionDAGISel::OPC_CheckChild2TypeI64:
2908*5f757f3fSDimitry Andric   case SelectionDAGISel::OPC_CheckChild3TypeI64:
2909*5f757f3fSDimitry Andric   case SelectionDAGISel::OPC_CheckChild4TypeI64:
2910*5f757f3fSDimitry Andric   case SelectionDAGISel::OPC_CheckChild5TypeI64:
2911*5f757f3fSDimitry Andric   case SelectionDAGISel::OPC_CheckChild6TypeI64:
2912*5f757f3fSDimitry Andric   case SelectionDAGISel::OPC_CheckChild7TypeI64: {
2913*5f757f3fSDimitry Andric     MVT::SimpleValueType VT;
2914*5f757f3fSDimitry Andric     unsigned ChildNo;
2915*5f757f3fSDimitry Andric     if (Opcode >= SelectionDAGISel::OPC_CheckChild0TypeI32 &&
2916*5f757f3fSDimitry Andric         Opcode <= SelectionDAGISel::OPC_CheckChild7TypeI32) {
2917*5f757f3fSDimitry Andric       VT = MVT::i32;
2918*5f757f3fSDimitry Andric       ChildNo = Opcode - SelectionDAGISel::OPC_CheckChild0TypeI32;
2919*5f757f3fSDimitry Andric     } else if (Opcode >= SelectionDAGISel::OPC_CheckChild0TypeI64 &&
2920*5f757f3fSDimitry Andric                Opcode <= SelectionDAGISel::OPC_CheckChild7TypeI64) {
2921*5f757f3fSDimitry Andric       VT = MVT::i64;
2922*5f757f3fSDimitry Andric       ChildNo = Opcode - SelectionDAGISel::OPC_CheckChild0TypeI64;
2923*5f757f3fSDimitry Andric     } else {
2924*5f757f3fSDimitry Andric       VT = static_cast<MVT::SimpleValueType>(Table[Index++]);
2925*5f757f3fSDimitry Andric       ChildNo = Opcode - SelectionDAGISel::OPC_CheckChild0Type;
2926*5f757f3fSDimitry Andric     }
2927*5f757f3fSDimitry Andric     Result = !::CheckChildType(VT, N, SDISel.TLI,
2928*5f757f3fSDimitry Andric                                SDISel.CurDAG->getDataLayout(), ChildNo);
29290b57cec5SDimitry Andric     return Index;
2930*5f757f3fSDimitry Andric   }
29310b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckCondCode:
29320b57cec5SDimitry Andric     Result = !::CheckCondCode(Table, Index, N);
29330b57cec5SDimitry Andric     return Index;
29340b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild2CondCode:
29350b57cec5SDimitry Andric     Result = !::CheckChild2CondCode(Table, Index, N);
29360b57cec5SDimitry Andric     return Index;
29370b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckValueType:
29380b57cec5SDimitry Andric     Result = !::CheckValueType(Table, Index, N, SDISel.TLI,
29390b57cec5SDimitry Andric                                SDISel.CurDAG->getDataLayout());
29400b57cec5SDimitry Andric     return Index;
29410b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckInteger:
29420b57cec5SDimitry Andric     Result = !::CheckInteger(Table, Index, N);
29430b57cec5SDimitry Andric     return Index;
29440b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild0Integer:
29450b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild1Integer:
29460b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild2Integer:
29470b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild3Integer:
29480b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckChild4Integer:
29490b57cec5SDimitry Andric     Result = !::CheckChildInteger(Table, Index, N,
29500b57cec5SDimitry Andric                      Table[Index-1] - SelectionDAGISel::OPC_CheckChild0Integer);
29510b57cec5SDimitry Andric     return Index;
29520b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckAndImm:
29530b57cec5SDimitry Andric     Result = !::CheckAndImm(Table, Index, N, SDISel);
29540b57cec5SDimitry Andric     return Index;
29550b57cec5SDimitry Andric   case SelectionDAGISel::OPC_CheckOrImm:
29560b57cec5SDimitry Andric     Result = !::CheckOrImm(Table, Index, N, SDISel);
29570b57cec5SDimitry Andric     return Index;
29580b57cec5SDimitry Andric   }
29590b57cec5SDimitry Andric }
29600b57cec5SDimitry Andric 
29610b57cec5SDimitry Andric namespace {
29620b57cec5SDimitry Andric 
29630b57cec5SDimitry Andric struct MatchScope {
29640b57cec5SDimitry Andric   /// FailIndex - If this match fails, this is the index to continue with.
29650b57cec5SDimitry Andric   unsigned FailIndex;
29660b57cec5SDimitry Andric 
29670b57cec5SDimitry Andric   /// NodeStack - The node stack when the scope was formed.
29680b57cec5SDimitry Andric   SmallVector<SDValue, 4> NodeStack;
29690b57cec5SDimitry Andric 
29700b57cec5SDimitry Andric   /// NumRecordedNodes - The number of recorded nodes when the scope was formed.
29710b57cec5SDimitry Andric   unsigned NumRecordedNodes;
29720b57cec5SDimitry Andric 
29730b57cec5SDimitry Andric   /// NumMatchedMemRefs - The number of matched memref entries.
29740b57cec5SDimitry Andric   unsigned NumMatchedMemRefs;
29750b57cec5SDimitry Andric 
29760b57cec5SDimitry Andric   /// InputChain/InputGlue - The current chain/glue
29770b57cec5SDimitry Andric   SDValue InputChain, InputGlue;
29780b57cec5SDimitry Andric 
29790b57cec5SDimitry Andric   /// HasChainNodesMatched - True if the ChainNodesMatched list is non-empty.
29800b57cec5SDimitry Andric   bool HasChainNodesMatched;
29810b57cec5SDimitry Andric };
29820b57cec5SDimitry Andric 
29830b57cec5SDimitry Andric /// \A DAG update listener to keep the matching state
29840b57cec5SDimitry Andric /// (i.e. RecordedNodes and MatchScope) uptodate if the target is allowed to
29850b57cec5SDimitry Andric /// change the DAG while matching.  X86 addressing mode matcher is an example
29860b57cec5SDimitry Andric /// for this.
29870b57cec5SDimitry Andric class MatchStateUpdater : public SelectionDAG::DAGUpdateListener
29880b57cec5SDimitry Andric {
29890b57cec5SDimitry Andric   SDNode **NodeToMatch;
29900b57cec5SDimitry Andric   SmallVectorImpl<std::pair<SDValue, SDNode *>> &RecordedNodes;
29910b57cec5SDimitry Andric   SmallVectorImpl<MatchScope> &MatchScopes;
29920b57cec5SDimitry Andric 
29930b57cec5SDimitry Andric public:
29940b57cec5SDimitry Andric   MatchStateUpdater(SelectionDAG &DAG, SDNode **NodeToMatch,
29950b57cec5SDimitry Andric                     SmallVectorImpl<std::pair<SDValue, SDNode *>> &RN,
29960b57cec5SDimitry Andric                     SmallVectorImpl<MatchScope> &MS)
29970b57cec5SDimitry Andric       : SelectionDAG::DAGUpdateListener(DAG), NodeToMatch(NodeToMatch),
29980b57cec5SDimitry Andric         RecordedNodes(RN), MatchScopes(MS) {}
29990b57cec5SDimitry Andric 
30000b57cec5SDimitry Andric   void NodeDeleted(SDNode *N, SDNode *E) override {
30010b57cec5SDimitry Andric     // Some early-returns here to avoid the search if we deleted the node or
30020b57cec5SDimitry Andric     // if the update comes from MorphNodeTo (MorphNodeTo is the last thing we
30030b57cec5SDimitry Andric     // do, so it's unnecessary to update matching state at that point).
30040b57cec5SDimitry Andric     // Neither of these can occur currently because we only install this
30050b57cec5SDimitry Andric     // update listener during matching a complex patterns.
30060b57cec5SDimitry Andric     if (!E || E->isMachineOpcode())
30070b57cec5SDimitry Andric       return;
30080b57cec5SDimitry Andric     // Check if NodeToMatch was updated.
30090b57cec5SDimitry Andric     if (N == *NodeToMatch)
30100b57cec5SDimitry Andric       *NodeToMatch = E;
30110b57cec5SDimitry Andric     // Performing linear search here does not matter because we almost never
30120b57cec5SDimitry Andric     // run this code.  You'd have to have a CSE during complex pattern
30130b57cec5SDimitry Andric     // matching.
30140b57cec5SDimitry Andric     for (auto &I : RecordedNodes)
30150b57cec5SDimitry Andric       if (I.first.getNode() == N)
30160b57cec5SDimitry Andric         I.first.setNode(E);
30170b57cec5SDimitry Andric 
30180b57cec5SDimitry Andric     for (auto &I : MatchScopes)
30190b57cec5SDimitry Andric       for (auto &J : I.NodeStack)
30200b57cec5SDimitry Andric         if (J.getNode() == N)
30210b57cec5SDimitry Andric           J.setNode(E);
30220b57cec5SDimitry Andric   }
30230b57cec5SDimitry Andric };
30240b57cec5SDimitry Andric 
30250b57cec5SDimitry Andric } // end anonymous namespace
30260b57cec5SDimitry Andric 
30270b57cec5SDimitry Andric void SelectionDAGISel::SelectCodeCommon(SDNode *NodeToMatch,
30280b57cec5SDimitry Andric                                         const unsigned char *MatcherTable,
30290b57cec5SDimitry Andric                                         unsigned TableSize) {
30300b57cec5SDimitry Andric   // FIXME: Should these even be selected?  Handle these cases in the caller?
30310b57cec5SDimitry Andric   switch (NodeToMatch->getOpcode()) {
30320b57cec5SDimitry Andric   default:
30330b57cec5SDimitry Andric     break;
30340b57cec5SDimitry Andric   case ISD::EntryToken:       // These nodes remain the same.
30350b57cec5SDimitry Andric   case ISD::BasicBlock:
30360b57cec5SDimitry Andric   case ISD::Register:
30370b57cec5SDimitry Andric   case ISD::RegisterMask:
30380b57cec5SDimitry Andric   case ISD::HANDLENODE:
30390b57cec5SDimitry Andric   case ISD::MDNODE_SDNODE:
30400b57cec5SDimitry Andric   case ISD::TargetConstant:
30410b57cec5SDimitry Andric   case ISD::TargetConstantFP:
30420b57cec5SDimitry Andric   case ISD::TargetConstantPool:
30430b57cec5SDimitry Andric   case ISD::TargetFrameIndex:
30440b57cec5SDimitry Andric   case ISD::TargetExternalSymbol:
30450b57cec5SDimitry Andric   case ISD::MCSymbol:
30460b57cec5SDimitry Andric   case ISD::TargetBlockAddress:
30470b57cec5SDimitry Andric   case ISD::TargetJumpTable:
30480b57cec5SDimitry Andric   case ISD::TargetGlobalTLSAddress:
30490b57cec5SDimitry Andric   case ISD::TargetGlobalAddress:
30500b57cec5SDimitry Andric   case ISD::TokenFactor:
30510b57cec5SDimitry Andric   case ISD::CopyFromReg:
30520b57cec5SDimitry Andric   case ISD::CopyToReg:
30530b57cec5SDimitry Andric   case ISD::EH_LABEL:
30540b57cec5SDimitry Andric   case ISD::ANNOTATION_LABEL:
30550b57cec5SDimitry Andric   case ISD::LIFETIME_START:
30560b57cec5SDimitry Andric   case ISD::LIFETIME_END:
3057e8d8bef9SDimitry Andric   case ISD::PSEUDO_PROBE:
30580b57cec5SDimitry Andric     NodeToMatch->setNodeId(-1); // Mark selected.
30590b57cec5SDimitry Andric     return;
30600b57cec5SDimitry Andric   case ISD::AssertSext:
30610b57cec5SDimitry Andric   case ISD::AssertZext:
30625ffd83dbSDimitry Andric   case ISD::AssertAlign:
30630b57cec5SDimitry Andric     ReplaceUses(SDValue(NodeToMatch, 0), NodeToMatch->getOperand(0));
30640b57cec5SDimitry Andric     CurDAG->RemoveDeadNode(NodeToMatch);
30650b57cec5SDimitry Andric     return;
30660b57cec5SDimitry Andric   case ISD::INLINEASM:
30670b57cec5SDimitry Andric   case ISD::INLINEASM_BR:
30685ffd83dbSDimitry Andric     Select_INLINEASM(NodeToMatch);
30690b57cec5SDimitry Andric     return;
30700b57cec5SDimitry Andric   case ISD::READ_REGISTER:
30710b57cec5SDimitry Andric     Select_READ_REGISTER(NodeToMatch);
30720b57cec5SDimitry Andric     return;
30730b57cec5SDimitry Andric   case ISD::WRITE_REGISTER:
30740b57cec5SDimitry Andric     Select_WRITE_REGISTER(NodeToMatch);
30750b57cec5SDimitry Andric     return;
30760b57cec5SDimitry Andric   case ISD::UNDEF:
30770b57cec5SDimitry Andric     Select_UNDEF(NodeToMatch);
30780b57cec5SDimitry Andric     return;
30795ffd83dbSDimitry Andric   case ISD::FREEZE:
30805ffd83dbSDimitry Andric     Select_FREEZE(NodeToMatch);
30815ffd83dbSDimitry Andric     return;
3082fe6060f1SDimitry Andric   case ISD::ARITH_FENCE:
3083fe6060f1SDimitry Andric     Select_ARITH_FENCE(NodeToMatch);
3084fe6060f1SDimitry Andric     return;
3085bdd1243dSDimitry Andric   case ISD::MEMBARRIER:
3086bdd1243dSDimitry Andric     Select_MEMBARRIER(NodeToMatch);
3087bdd1243dSDimitry Andric     return;
3088753f127fSDimitry Andric   case ISD::STACKMAP:
3089753f127fSDimitry Andric     Select_STACKMAP(NodeToMatch);
3090753f127fSDimitry Andric     return;
3091fcaf7f86SDimitry Andric   case ISD::PATCHPOINT:
3092fcaf7f86SDimitry Andric     Select_PATCHPOINT(NodeToMatch);
3093fcaf7f86SDimitry Andric     return;
3094*5f757f3fSDimitry Andric   case ISD::JUMP_TABLE_DEBUG_INFO:
3095*5f757f3fSDimitry Andric     Select_JUMP_TABLE_DEBUG_INFO(NodeToMatch);
3096*5f757f3fSDimitry Andric     return;
30970b57cec5SDimitry Andric   }
30980b57cec5SDimitry Andric 
30990b57cec5SDimitry Andric   assert(!NodeToMatch->isMachineOpcode() && "Node already selected!");
31000b57cec5SDimitry Andric 
31010b57cec5SDimitry Andric   // Set up the node stack with NodeToMatch as the only node on the stack.
31020b57cec5SDimitry Andric   SmallVector<SDValue, 8> NodeStack;
31030b57cec5SDimitry Andric   SDValue N = SDValue(NodeToMatch, 0);
31040b57cec5SDimitry Andric   NodeStack.push_back(N);
31050b57cec5SDimitry Andric 
31060b57cec5SDimitry Andric   // MatchScopes - Scopes used when matching, if a match failure happens, this
31070b57cec5SDimitry Andric   // indicates where to continue checking.
31080b57cec5SDimitry Andric   SmallVector<MatchScope, 8> MatchScopes;
31090b57cec5SDimitry Andric 
31100b57cec5SDimitry Andric   // RecordedNodes - This is the set of nodes that have been recorded by the
31110b57cec5SDimitry Andric   // state machine.  The second value is the parent of the node, or null if the
31120b57cec5SDimitry Andric   // root is recorded.
31130b57cec5SDimitry Andric   SmallVector<std::pair<SDValue, SDNode*>, 8> RecordedNodes;
31140b57cec5SDimitry Andric 
31150b57cec5SDimitry Andric   // MatchedMemRefs - This is the set of MemRef's we've seen in the input
31160b57cec5SDimitry Andric   // pattern.
31170b57cec5SDimitry Andric   SmallVector<MachineMemOperand*, 2> MatchedMemRefs;
31180b57cec5SDimitry Andric 
31190b57cec5SDimitry Andric   // These are the current input chain and glue for use when generating nodes.
31200b57cec5SDimitry Andric   // Various Emit operations change these.  For example, emitting a copytoreg
31210b57cec5SDimitry Andric   // uses and updates these.
31220b57cec5SDimitry Andric   SDValue InputChain, InputGlue;
31230b57cec5SDimitry Andric 
31240b57cec5SDimitry Andric   // ChainNodesMatched - If a pattern matches nodes that have input/output
31250b57cec5SDimitry Andric   // chains, the OPC_EmitMergeInputChains operation is emitted which indicates
31260b57cec5SDimitry Andric   // which ones they are.  The result is captured into this list so that we can
31270b57cec5SDimitry Andric   // update the chain results when the pattern is complete.
31280b57cec5SDimitry Andric   SmallVector<SDNode*, 3> ChainNodesMatched;
31290b57cec5SDimitry Andric 
31300b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "ISEL: Starting pattern match\n");
31310b57cec5SDimitry Andric 
31320b57cec5SDimitry Andric   // Determine where to start the interpreter.  Normally we start at opcode #0,
31330b57cec5SDimitry Andric   // but if the state machine starts with an OPC_SwitchOpcode, then we
31340b57cec5SDimitry Andric   // accelerate the first lookup (which is guaranteed to be hot) with the
31350b57cec5SDimitry Andric   // OpcodeOffset table.
31360b57cec5SDimitry Andric   unsigned MatcherIndex = 0;
31370b57cec5SDimitry Andric 
31380b57cec5SDimitry Andric   if (!OpcodeOffset.empty()) {
31390b57cec5SDimitry Andric     // Already computed the OpcodeOffset table, just index into it.
31400b57cec5SDimitry Andric     if (N.getOpcode() < OpcodeOffset.size())
31410b57cec5SDimitry Andric       MatcherIndex = OpcodeOffset[N.getOpcode()];
31420b57cec5SDimitry Andric     LLVM_DEBUG(dbgs() << "  Initial Opcode index to " << MatcherIndex << "\n");
31430b57cec5SDimitry Andric 
31440b57cec5SDimitry Andric   } else if (MatcherTable[0] == OPC_SwitchOpcode) {
31450b57cec5SDimitry Andric     // Otherwise, the table isn't computed, but the state machine does start
31460b57cec5SDimitry Andric     // with an OPC_SwitchOpcode instruction.  Populate the table now, since this
31470b57cec5SDimitry Andric     // is the first time we're selecting an instruction.
31480b57cec5SDimitry Andric     unsigned Idx = 1;
31490b57cec5SDimitry Andric     while (true) {
31500b57cec5SDimitry Andric       // Get the size of this case.
31510b57cec5SDimitry Andric       unsigned CaseSize = MatcherTable[Idx++];
31520b57cec5SDimitry Andric       if (CaseSize & 128)
31530b57cec5SDimitry Andric         CaseSize = GetVBR(CaseSize, MatcherTable, Idx);
31540b57cec5SDimitry Andric       if (CaseSize == 0) break;
31550b57cec5SDimitry Andric 
31560b57cec5SDimitry Andric       // Get the opcode, add the index to the table.
31570b57cec5SDimitry Andric       uint16_t Opc = MatcherTable[Idx++];
3158*5f757f3fSDimitry Andric       Opc |= static_cast<uint16_t>(MatcherTable[Idx++]) << 8;
31590b57cec5SDimitry Andric       if (Opc >= OpcodeOffset.size())
31600b57cec5SDimitry Andric         OpcodeOffset.resize((Opc+1)*2);
31610b57cec5SDimitry Andric       OpcodeOffset[Opc] = Idx;
31620b57cec5SDimitry Andric       Idx += CaseSize;
31630b57cec5SDimitry Andric     }
31640b57cec5SDimitry Andric 
31650b57cec5SDimitry Andric     // Okay, do the lookup for the first opcode.
31660b57cec5SDimitry Andric     if (N.getOpcode() < OpcodeOffset.size())
31670b57cec5SDimitry Andric       MatcherIndex = OpcodeOffset[N.getOpcode()];
31680b57cec5SDimitry Andric   }
31690b57cec5SDimitry Andric 
31700b57cec5SDimitry Andric   while (true) {
31710b57cec5SDimitry Andric     assert(MatcherIndex < TableSize && "Invalid index");
31720b57cec5SDimitry Andric #ifndef NDEBUG
31730b57cec5SDimitry Andric     unsigned CurrentOpcodeIndex = MatcherIndex;
31740b57cec5SDimitry Andric #endif
3175*5f757f3fSDimitry Andric     BuiltinOpcodes Opcode =
3176*5f757f3fSDimitry Andric         static_cast<BuiltinOpcodes>(MatcherTable[MatcherIndex++]);
31770b57cec5SDimitry Andric     switch (Opcode) {
31780b57cec5SDimitry Andric     case OPC_Scope: {
31790b57cec5SDimitry Andric       // Okay, the semantics of this operation are that we should push a scope
31800b57cec5SDimitry Andric       // then evaluate the first child.  However, pushing a scope only to have
31810b57cec5SDimitry Andric       // the first check fail (which then pops it) is inefficient.  If we can
31820b57cec5SDimitry Andric       // determine immediately that the first check (or first several) will
31830b57cec5SDimitry Andric       // immediately fail, don't even bother pushing a scope for them.
31840b57cec5SDimitry Andric       unsigned FailIndex;
31850b57cec5SDimitry Andric 
31860b57cec5SDimitry Andric       while (true) {
31870b57cec5SDimitry Andric         unsigned NumToSkip = MatcherTable[MatcherIndex++];
31880b57cec5SDimitry Andric         if (NumToSkip & 128)
31890b57cec5SDimitry Andric           NumToSkip = GetVBR(NumToSkip, MatcherTable, MatcherIndex);
31900b57cec5SDimitry Andric         // Found the end of the scope with no match.
31910b57cec5SDimitry Andric         if (NumToSkip == 0) {
31920b57cec5SDimitry Andric           FailIndex = 0;
31930b57cec5SDimitry Andric           break;
31940b57cec5SDimitry Andric         }
31950b57cec5SDimitry Andric 
31960b57cec5SDimitry Andric         FailIndex = MatcherIndex+NumToSkip;
31970b57cec5SDimitry Andric 
31980b57cec5SDimitry Andric         unsigned MatcherIndexOfPredicate = MatcherIndex;
31990b57cec5SDimitry Andric         (void)MatcherIndexOfPredicate; // silence warning.
32000b57cec5SDimitry Andric 
32010b57cec5SDimitry Andric         // If we can't evaluate this predicate without pushing a scope (e.g. if
32020b57cec5SDimitry Andric         // it is a 'MoveParent') or if the predicate succeeds on this node, we
32030b57cec5SDimitry Andric         // push the scope and evaluate the full predicate chain.
32040b57cec5SDimitry Andric         bool Result;
32050b57cec5SDimitry Andric         MatcherIndex = IsPredicateKnownToFail(MatcherTable, MatcherIndex, N,
32060b57cec5SDimitry Andric                                               Result, *this, RecordedNodes);
32070b57cec5SDimitry Andric         if (!Result)
32080b57cec5SDimitry Andric           break;
32090b57cec5SDimitry Andric 
32100b57cec5SDimitry Andric         LLVM_DEBUG(
32110b57cec5SDimitry Andric             dbgs() << "  Skipped scope entry (due to false predicate) at "
32120b57cec5SDimitry Andric                    << "index " << MatcherIndexOfPredicate << ", continuing at "
32130b57cec5SDimitry Andric                    << FailIndex << "\n");
32140b57cec5SDimitry Andric         ++NumDAGIselRetries;
32150b57cec5SDimitry Andric 
32160b57cec5SDimitry Andric         // Otherwise, we know that this case of the Scope is guaranteed to fail,
32170b57cec5SDimitry Andric         // move to the next case.
32180b57cec5SDimitry Andric         MatcherIndex = FailIndex;
32190b57cec5SDimitry Andric       }
32200b57cec5SDimitry Andric 
32210b57cec5SDimitry Andric       // If the whole scope failed to match, bail.
32220b57cec5SDimitry Andric       if (FailIndex == 0) break;
32230b57cec5SDimitry Andric 
32240b57cec5SDimitry Andric       // Push a MatchScope which indicates where to go if the first child fails
32250b57cec5SDimitry Andric       // to match.
32260b57cec5SDimitry Andric       MatchScope NewEntry;
32270b57cec5SDimitry Andric       NewEntry.FailIndex = FailIndex;
32280b57cec5SDimitry Andric       NewEntry.NodeStack.append(NodeStack.begin(), NodeStack.end());
32290b57cec5SDimitry Andric       NewEntry.NumRecordedNodes = RecordedNodes.size();
32300b57cec5SDimitry Andric       NewEntry.NumMatchedMemRefs = MatchedMemRefs.size();
32310b57cec5SDimitry Andric       NewEntry.InputChain = InputChain;
32320b57cec5SDimitry Andric       NewEntry.InputGlue = InputGlue;
32330b57cec5SDimitry Andric       NewEntry.HasChainNodesMatched = !ChainNodesMatched.empty();
32340b57cec5SDimitry Andric       MatchScopes.push_back(NewEntry);
32350b57cec5SDimitry Andric       continue;
32360b57cec5SDimitry Andric     }
32370b57cec5SDimitry Andric     case OPC_RecordNode: {
32380b57cec5SDimitry Andric       // Remember this node, it may end up being an operand in the pattern.
32390b57cec5SDimitry Andric       SDNode *Parent = nullptr;
32400b57cec5SDimitry Andric       if (NodeStack.size() > 1)
32410b57cec5SDimitry Andric         Parent = NodeStack[NodeStack.size()-2].getNode();
32420b57cec5SDimitry Andric       RecordedNodes.push_back(std::make_pair(N, Parent));
32430b57cec5SDimitry Andric       continue;
32440b57cec5SDimitry Andric     }
32450b57cec5SDimitry Andric 
32460b57cec5SDimitry Andric     case OPC_RecordChild0: case OPC_RecordChild1:
32470b57cec5SDimitry Andric     case OPC_RecordChild2: case OPC_RecordChild3:
32480b57cec5SDimitry Andric     case OPC_RecordChild4: case OPC_RecordChild5:
32490b57cec5SDimitry Andric     case OPC_RecordChild6: case OPC_RecordChild7: {
32500b57cec5SDimitry Andric       unsigned ChildNo = Opcode-OPC_RecordChild0;
32510b57cec5SDimitry Andric       if (ChildNo >= N.getNumOperands())
32520b57cec5SDimitry Andric         break;  // Match fails if out of range child #.
32530b57cec5SDimitry Andric 
32540b57cec5SDimitry Andric       RecordedNodes.push_back(std::make_pair(N->getOperand(ChildNo),
32550b57cec5SDimitry Andric                                              N.getNode()));
32560b57cec5SDimitry Andric       continue;
32570b57cec5SDimitry Andric     }
32580b57cec5SDimitry Andric     case OPC_RecordMemRef:
32590b57cec5SDimitry Andric       if (auto *MN = dyn_cast<MemSDNode>(N))
32600b57cec5SDimitry Andric         MatchedMemRefs.push_back(MN->getMemOperand());
32610b57cec5SDimitry Andric       else {
32620b57cec5SDimitry Andric         LLVM_DEBUG(dbgs() << "Expected MemSDNode "; N->dump(CurDAG);
32630b57cec5SDimitry Andric                    dbgs() << '\n');
32640b57cec5SDimitry Andric       }
32650b57cec5SDimitry Andric 
32660b57cec5SDimitry Andric       continue;
32670b57cec5SDimitry Andric 
32680b57cec5SDimitry Andric     case OPC_CaptureGlueInput:
32690b57cec5SDimitry Andric       // If the current node has an input glue, capture it in InputGlue.
32700b57cec5SDimitry Andric       if (N->getNumOperands() != 0 &&
32710b57cec5SDimitry Andric           N->getOperand(N->getNumOperands()-1).getValueType() == MVT::Glue)
32720b57cec5SDimitry Andric         InputGlue = N->getOperand(N->getNumOperands()-1);
32730b57cec5SDimitry Andric       continue;
32740b57cec5SDimitry Andric 
32750b57cec5SDimitry Andric     case OPC_MoveChild: {
32760b57cec5SDimitry Andric       unsigned ChildNo = MatcherTable[MatcherIndex++];
32770b57cec5SDimitry Andric       if (ChildNo >= N.getNumOperands())
32780b57cec5SDimitry Andric         break;  // Match fails if out of range child #.
32790b57cec5SDimitry Andric       N = N.getOperand(ChildNo);
32800b57cec5SDimitry Andric       NodeStack.push_back(N);
32810b57cec5SDimitry Andric       continue;
32820b57cec5SDimitry Andric     }
32830b57cec5SDimitry Andric 
32840b57cec5SDimitry Andric     case OPC_MoveChild0: case OPC_MoveChild1:
32850b57cec5SDimitry Andric     case OPC_MoveChild2: case OPC_MoveChild3:
32860b57cec5SDimitry Andric     case OPC_MoveChild4: case OPC_MoveChild5:
32870b57cec5SDimitry Andric     case OPC_MoveChild6: case OPC_MoveChild7: {
32880b57cec5SDimitry Andric       unsigned ChildNo = Opcode-OPC_MoveChild0;
32890b57cec5SDimitry Andric       if (ChildNo >= N.getNumOperands())
32900b57cec5SDimitry Andric         break;  // Match fails if out of range child #.
32910b57cec5SDimitry Andric       N = N.getOperand(ChildNo);
32920b57cec5SDimitry Andric       NodeStack.push_back(N);
32930b57cec5SDimitry Andric       continue;
32940b57cec5SDimitry Andric     }
32950b57cec5SDimitry Andric 
3296*5f757f3fSDimitry Andric     case OPC_MoveSibling:
3297*5f757f3fSDimitry Andric     case OPC_MoveSibling0:
3298*5f757f3fSDimitry Andric     case OPC_MoveSibling1:
3299*5f757f3fSDimitry Andric     case OPC_MoveSibling2:
3300*5f757f3fSDimitry Andric     case OPC_MoveSibling3:
3301*5f757f3fSDimitry Andric     case OPC_MoveSibling4:
3302*5f757f3fSDimitry Andric     case OPC_MoveSibling5:
3303*5f757f3fSDimitry Andric     case OPC_MoveSibling6:
3304*5f757f3fSDimitry Andric     case OPC_MoveSibling7: {
3305*5f757f3fSDimitry Andric       // Pop the current node off the NodeStack.
3306*5f757f3fSDimitry Andric       NodeStack.pop_back();
3307*5f757f3fSDimitry Andric       assert(!NodeStack.empty() && "Node stack imbalance!");
3308*5f757f3fSDimitry Andric       N = NodeStack.back();
3309*5f757f3fSDimitry Andric 
3310*5f757f3fSDimitry Andric       unsigned SiblingNo = Opcode == OPC_MoveSibling
3311*5f757f3fSDimitry Andric                                ? MatcherTable[MatcherIndex++]
3312*5f757f3fSDimitry Andric                                : Opcode - OPC_MoveSibling0;
3313*5f757f3fSDimitry Andric       if (SiblingNo >= N.getNumOperands())
3314*5f757f3fSDimitry Andric         break; // Match fails if out of range sibling #.
3315*5f757f3fSDimitry Andric       N = N.getOperand(SiblingNo);
3316*5f757f3fSDimitry Andric       NodeStack.push_back(N);
3317*5f757f3fSDimitry Andric       continue;
3318*5f757f3fSDimitry Andric     }
33190b57cec5SDimitry Andric     case OPC_MoveParent:
33200b57cec5SDimitry Andric       // Pop the current node off the NodeStack.
33210b57cec5SDimitry Andric       NodeStack.pop_back();
33220b57cec5SDimitry Andric       assert(!NodeStack.empty() && "Node stack imbalance!");
33230b57cec5SDimitry Andric       N = NodeStack.back();
33240b57cec5SDimitry Andric       continue;
33250b57cec5SDimitry Andric 
33260b57cec5SDimitry Andric     case OPC_CheckSame:
33270b57cec5SDimitry Andric       if (!::CheckSame(MatcherTable, MatcherIndex, N, RecordedNodes)) break;
33280b57cec5SDimitry Andric       continue;
33290b57cec5SDimitry Andric 
33300b57cec5SDimitry Andric     case OPC_CheckChild0Same: case OPC_CheckChild1Same:
33310b57cec5SDimitry Andric     case OPC_CheckChild2Same: case OPC_CheckChild3Same:
33320b57cec5SDimitry Andric       if (!::CheckChildSame(MatcherTable, MatcherIndex, N, RecordedNodes,
33330b57cec5SDimitry Andric                             Opcode-OPC_CheckChild0Same))
33340b57cec5SDimitry Andric         break;
33350b57cec5SDimitry Andric       continue;
33360b57cec5SDimitry Andric 
33370b57cec5SDimitry Andric     case OPC_CheckPatternPredicate:
3338*5f757f3fSDimitry Andric     case OPC_CheckPatternPredicate2:
3339*5f757f3fSDimitry Andric       if (!::CheckPatternPredicate(MatcherTable, MatcherIndex, *this,
3340*5f757f3fSDimitry Andric                                    Opcode == OPC_CheckPatternPredicate2))
3341*5f757f3fSDimitry Andric         break;
33420b57cec5SDimitry Andric       continue;
33430b57cec5SDimitry Andric     case OPC_CheckPredicate:
33440b57cec5SDimitry Andric       if (!::CheckNodePredicate(MatcherTable, MatcherIndex, *this,
33450b57cec5SDimitry Andric                                 N.getNode()))
33460b57cec5SDimitry Andric         break;
33470b57cec5SDimitry Andric       continue;
33480b57cec5SDimitry Andric     case OPC_CheckPredicateWithOperands: {
33490b57cec5SDimitry Andric       unsigned OpNum = MatcherTable[MatcherIndex++];
33500b57cec5SDimitry Andric       SmallVector<SDValue, 8> Operands;
33510b57cec5SDimitry Andric 
33520b57cec5SDimitry Andric       for (unsigned i = 0; i < OpNum; ++i)
33530b57cec5SDimitry Andric         Operands.push_back(RecordedNodes[MatcherTable[MatcherIndex++]].first);
33540b57cec5SDimitry Andric 
33550b57cec5SDimitry Andric       unsigned PredNo = MatcherTable[MatcherIndex++];
33560b57cec5SDimitry Andric       if (!CheckNodePredicateWithOperands(N.getNode(), PredNo, Operands))
33570b57cec5SDimitry Andric         break;
33580b57cec5SDimitry Andric       continue;
33590b57cec5SDimitry Andric     }
33600b57cec5SDimitry Andric     case OPC_CheckComplexPat: {
33610b57cec5SDimitry Andric       unsigned CPNum = MatcherTable[MatcherIndex++];
33620b57cec5SDimitry Andric       unsigned RecNo = MatcherTable[MatcherIndex++];
33630b57cec5SDimitry Andric       assert(RecNo < RecordedNodes.size() && "Invalid CheckComplexPat");
33640b57cec5SDimitry Andric 
33650b57cec5SDimitry Andric       // If target can modify DAG during matching, keep the matching state
33660b57cec5SDimitry Andric       // consistent.
33670b57cec5SDimitry Andric       std::unique_ptr<MatchStateUpdater> MSU;
33680b57cec5SDimitry Andric       if (ComplexPatternFuncMutatesDAG())
33690b57cec5SDimitry Andric         MSU.reset(new MatchStateUpdater(*CurDAG, &NodeToMatch, RecordedNodes,
33700b57cec5SDimitry Andric                                         MatchScopes));
33710b57cec5SDimitry Andric 
33720b57cec5SDimitry Andric       if (!CheckComplexPattern(NodeToMatch, RecordedNodes[RecNo].second,
33730b57cec5SDimitry Andric                                RecordedNodes[RecNo].first, CPNum,
33740b57cec5SDimitry Andric                                RecordedNodes))
33750b57cec5SDimitry Andric         break;
33760b57cec5SDimitry Andric       continue;
33770b57cec5SDimitry Andric     }
33780b57cec5SDimitry Andric     case OPC_CheckOpcode:
33790b57cec5SDimitry Andric       if (!::CheckOpcode(MatcherTable, MatcherIndex, N.getNode())) break;
33800b57cec5SDimitry Andric       continue;
33810b57cec5SDimitry Andric 
33820b57cec5SDimitry Andric     case OPC_CheckType:
3383*5f757f3fSDimitry Andric     case OPC_CheckTypeI32:
3384*5f757f3fSDimitry Andric     case OPC_CheckTypeI64:
3385*5f757f3fSDimitry Andric       MVT::SimpleValueType VT;
3386*5f757f3fSDimitry Andric       switch (Opcode) {
3387*5f757f3fSDimitry Andric       case OPC_CheckTypeI32:
3388*5f757f3fSDimitry Andric         VT = MVT::i32;
3389*5f757f3fSDimitry Andric         break;
3390*5f757f3fSDimitry Andric       case OPC_CheckTypeI64:
3391*5f757f3fSDimitry Andric         VT = MVT::i64;
3392*5f757f3fSDimitry Andric         break;
3393*5f757f3fSDimitry Andric       default:
3394*5f757f3fSDimitry Andric         VT = static_cast<MVT::SimpleValueType>(MatcherTable[MatcherIndex++]);
3395*5f757f3fSDimitry Andric         break;
3396*5f757f3fSDimitry Andric       }
3397*5f757f3fSDimitry Andric       if (!::CheckType(VT, N, TLI, CurDAG->getDataLayout()))
33980b57cec5SDimitry Andric         break;
33990b57cec5SDimitry Andric       continue;
34000b57cec5SDimitry Andric 
34010b57cec5SDimitry Andric     case OPC_CheckTypeRes: {
34020b57cec5SDimitry Andric       unsigned Res = MatcherTable[MatcherIndex++];
3403*5f757f3fSDimitry Andric       if (!::CheckType(
3404*5f757f3fSDimitry Andric               static_cast<MVT::SimpleValueType>(MatcherTable[MatcherIndex++]),
3405*5f757f3fSDimitry Andric               N.getValue(Res), TLI, CurDAG->getDataLayout()))
34060b57cec5SDimitry Andric         break;
34070b57cec5SDimitry Andric       continue;
34080b57cec5SDimitry Andric     }
34090b57cec5SDimitry Andric 
34100b57cec5SDimitry Andric     case OPC_SwitchOpcode: {
34110b57cec5SDimitry Andric       unsigned CurNodeOpcode = N.getOpcode();
34120b57cec5SDimitry Andric       unsigned SwitchStart = MatcherIndex-1; (void)SwitchStart;
34130b57cec5SDimitry Andric       unsigned CaseSize;
34140b57cec5SDimitry Andric       while (true) {
34150b57cec5SDimitry Andric         // Get the size of this case.
34160b57cec5SDimitry Andric         CaseSize = MatcherTable[MatcherIndex++];
34170b57cec5SDimitry Andric         if (CaseSize & 128)
34180b57cec5SDimitry Andric           CaseSize = GetVBR(CaseSize, MatcherTable, MatcherIndex);
34190b57cec5SDimitry Andric         if (CaseSize == 0) break;
34200b57cec5SDimitry Andric 
34210b57cec5SDimitry Andric         uint16_t Opc = MatcherTable[MatcherIndex++];
3422*5f757f3fSDimitry Andric         Opc |= static_cast<uint16_t>(MatcherTable[MatcherIndex++]) << 8;
34230b57cec5SDimitry Andric 
34240b57cec5SDimitry Andric         // If the opcode matches, then we will execute this case.
34250b57cec5SDimitry Andric         if (CurNodeOpcode == Opc)
34260b57cec5SDimitry Andric           break;
34270b57cec5SDimitry Andric 
34280b57cec5SDimitry Andric         // Otherwise, skip over this case.
34290b57cec5SDimitry Andric         MatcherIndex += CaseSize;
34300b57cec5SDimitry Andric       }
34310b57cec5SDimitry Andric 
34320b57cec5SDimitry Andric       // If no cases matched, bail out.
34330b57cec5SDimitry Andric       if (CaseSize == 0) break;
34340b57cec5SDimitry Andric 
34350b57cec5SDimitry Andric       // Otherwise, execute the case we found.
34360b57cec5SDimitry Andric       LLVM_DEBUG(dbgs() << "  OpcodeSwitch from " << SwitchStart << " to "
34370b57cec5SDimitry Andric                         << MatcherIndex << "\n");
34380b57cec5SDimitry Andric       continue;
34390b57cec5SDimitry Andric     }
34400b57cec5SDimitry Andric 
34410b57cec5SDimitry Andric     case OPC_SwitchType: {
34420b57cec5SDimitry Andric       MVT CurNodeVT = N.getSimpleValueType();
34430b57cec5SDimitry Andric       unsigned SwitchStart = MatcherIndex-1; (void)SwitchStart;
34440b57cec5SDimitry Andric       unsigned CaseSize;
34450b57cec5SDimitry Andric       while (true) {
34460b57cec5SDimitry Andric         // Get the size of this case.
34470b57cec5SDimitry Andric         CaseSize = MatcherTable[MatcherIndex++];
34480b57cec5SDimitry Andric         if (CaseSize & 128)
34490b57cec5SDimitry Andric           CaseSize = GetVBR(CaseSize, MatcherTable, MatcherIndex);
34500b57cec5SDimitry Andric         if (CaseSize == 0) break;
34510b57cec5SDimitry Andric 
3452*5f757f3fSDimitry Andric         MVT CaseVT =
3453*5f757f3fSDimitry Andric             static_cast<MVT::SimpleValueType>(MatcherTable[MatcherIndex++]);
34540b57cec5SDimitry Andric         if (CaseVT == MVT::iPTR)
34550b57cec5SDimitry Andric           CaseVT = TLI->getPointerTy(CurDAG->getDataLayout());
34560b57cec5SDimitry Andric 
34570b57cec5SDimitry Andric         // If the VT matches, then we will execute this case.
34580b57cec5SDimitry Andric         if (CurNodeVT == CaseVT)
34590b57cec5SDimitry Andric           break;
34600b57cec5SDimitry Andric 
34610b57cec5SDimitry Andric         // Otherwise, skip over this case.
34620b57cec5SDimitry Andric         MatcherIndex += CaseSize;
34630b57cec5SDimitry Andric       }
34640b57cec5SDimitry Andric 
34650b57cec5SDimitry Andric       // If no cases matched, bail out.
34660b57cec5SDimitry Andric       if (CaseSize == 0) break;
34670b57cec5SDimitry Andric 
34680b57cec5SDimitry Andric       // Otherwise, execute the case we found.
346906c3fb27SDimitry Andric       LLVM_DEBUG(dbgs() << "  TypeSwitch[" << CurNodeVT
34700b57cec5SDimitry Andric                         << "] from " << SwitchStart << " to " << MatcherIndex
34710b57cec5SDimitry Andric                         << '\n');
34720b57cec5SDimitry Andric       continue;
34730b57cec5SDimitry Andric     }
3474*5f757f3fSDimitry Andric     case OPC_CheckChild0Type:
3475*5f757f3fSDimitry Andric     case OPC_CheckChild1Type:
3476*5f757f3fSDimitry Andric     case OPC_CheckChild2Type:
3477*5f757f3fSDimitry Andric     case OPC_CheckChild3Type:
3478*5f757f3fSDimitry Andric     case OPC_CheckChild4Type:
3479*5f757f3fSDimitry Andric     case OPC_CheckChild5Type:
3480*5f757f3fSDimitry Andric     case OPC_CheckChild6Type:
3481*5f757f3fSDimitry Andric     case OPC_CheckChild7Type:
3482*5f757f3fSDimitry Andric     case OPC_CheckChild0TypeI32:
3483*5f757f3fSDimitry Andric     case OPC_CheckChild1TypeI32:
3484*5f757f3fSDimitry Andric     case OPC_CheckChild2TypeI32:
3485*5f757f3fSDimitry Andric     case OPC_CheckChild3TypeI32:
3486*5f757f3fSDimitry Andric     case OPC_CheckChild4TypeI32:
3487*5f757f3fSDimitry Andric     case OPC_CheckChild5TypeI32:
3488*5f757f3fSDimitry Andric     case OPC_CheckChild6TypeI32:
3489*5f757f3fSDimitry Andric     case OPC_CheckChild7TypeI32:
3490*5f757f3fSDimitry Andric     case OPC_CheckChild0TypeI64:
3491*5f757f3fSDimitry Andric     case OPC_CheckChild1TypeI64:
3492*5f757f3fSDimitry Andric     case OPC_CheckChild2TypeI64:
3493*5f757f3fSDimitry Andric     case OPC_CheckChild3TypeI64:
3494*5f757f3fSDimitry Andric     case OPC_CheckChild4TypeI64:
3495*5f757f3fSDimitry Andric     case OPC_CheckChild5TypeI64:
3496*5f757f3fSDimitry Andric     case OPC_CheckChild6TypeI64:
3497*5f757f3fSDimitry Andric     case OPC_CheckChild7TypeI64: {
3498*5f757f3fSDimitry Andric       MVT::SimpleValueType VT;
3499*5f757f3fSDimitry Andric       unsigned ChildNo;
3500*5f757f3fSDimitry Andric       if (Opcode >= SelectionDAGISel::OPC_CheckChild0TypeI32 &&
3501*5f757f3fSDimitry Andric           Opcode <= SelectionDAGISel::OPC_CheckChild7TypeI32) {
3502*5f757f3fSDimitry Andric         VT = MVT::i32;
3503*5f757f3fSDimitry Andric         ChildNo = Opcode - SelectionDAGISel::OPC_CheckChild0TypeI32;
3504*5f757f3fSDimitry Andric       } else if (Opcode >= SelectionDAGISel::OPC_CheckChild0TypeI64 &&
3505*5f757f3fSDimitry Andric                  Opcode <= SelectionDAGISel::OPC_CheckChild7TypeI64) {
3506*5f757f3fSDimitry Andric         VT = MVT::i64;
3507*5f757f3fSDimitry Andric         ChildNo = Opcode - SelectionDAGISel::OPC_CheckChild0TypeI64;
3508*5f757f3fSDimitry Andric       } else {
3509*5f757f3fSDimitry Andric         VT = static_cast<MVT::SimpleValueType>(MatcherTable[MatcherIndex++]);
3510*5f757f3fSDimitry Andric         ChildNo = Opcode - SelectionDAGISel::OPC_CheckChild0Type;
3511*5f757f3fSDimitry Andric       }
3512*5f757f3fSDimitry Andric       if (!::CheckChildType(VT, N, TLI, CurDAG->getDataLayout(), ChildNo))
35130b57cec5SDimitry Andric         break;
35140b57cec5SDimitry Andric       continue;
3515*5f757f3fSDimitry Andric     }
35160b57cec5SDimitry Andric     case OPC_CheckCondCode:
35170b57cec5SDimitry Andric       if (!::CheckCondCode(MatcherTable, MatcherIndex, N)) break;
35180b57cec5SDimitry Andric       continue;
35190b57cec5SDimitry Andric     case OPC_CheckChild2CondCode:
35200b57cec5SDimitry Andric       if (!::CheckChild2CondCode(MatcherTable, MatcherIndex, N)) break;
35210b57cec5SDimitry Andric       continue;
35220b57cec5SDimitry Andric     case OPC_CheckValueType:
35230b57cec5SDimitry Andric       if (!::CheckValueType(MatcherTable, MatcherIndex, N, TLI,
35240b57cec5SDimitry Andric                             CurDAG->getDataLayout()))
35250b57cec5SDimitry Andric         break;
35260b57cec5SDimitry Andric       continue;
35270b57cec5SDimitry Andric     case OPC_CheckInteger:
35280b57cec5SDimitry Andric       if (!::CheckInteger(MatcherTable, MatcherIndex, N)) break;
35290b57cec5SDimitry Andric       continue;
35300b57cec5SDimitry Andric     case OPC_CheckChild0Integer: case OPC_CheckChild1Integer:
35310b57cec5SDimitry Andric     case OPC_CheckChild2Integer: case OPC_CheckChild3Integer:
35320b57cec5SDimitry Andric     case OPC_CheckChild4Integer:
35330b57cec5SDimitry Andric       if (!::CheckChildInteger(MatcherTable, MatcherIndex, N,
35340b57cec5SDimitry Andric                                Opcode-OPC_CheckChild0Integer)) break;
35350b57cec5SDimitry Andric       continue;
35360b57cec5SDimitry Andric     case OPC_CheckAndImm:
35370b57cec5SDimitry Andric       if (!::CheckAndImm(MatcherTable, MatcherIndex, N, *this)) break;
35380b57cec5SDimitry Andric       continue;
35390b57cec5SDimitry Andric     case OPC_CheckOrImm:
35400b57cec5SDimitry Andric       if (!::CheckOrImm(MatcherTable, MatcherIndex, N, *this)) break;
35410b57cec5SDimitry Andric       continue;
35420b57cec5SDimitry Andric     case OPC_CheckImmAllOnesV:
3543e8d8bef9SDimitry Andric       if (!ISD::isConstantSplatVectorAllOnes(N.getNode()))
3544e8d8bef9SDimitry Andric         break;
35450b57cec5SDimitry Andric       continue;
35460b57cec5SDimitry Andric     case OPC_CheckImmAllZerosV:
3547e8d8bef9SDimitry Andric       if (!ISD::isConstantSplatVectorAllZeros(N.getNode()))
3548e8d8bef9SDimitry Andric         break;
35490b57cec5SDimitry Andric       continue;
35500b57cec5SDimitry Andric 
35510b57cec5SDimitry Andric     case OPC_CheckFoldableChainNode: {
35520b57cec5SDimitry Andric       assert(NodeStack.size() != 1 && "No parent node");
35530b57cec5SDimitry Andric       // Verify that all intermediate nodes between the root and this one have
3554480093f4SDimitry Andric       // a single use (ignoring chains, which are handled in UpdateChains).
35550b57cec5SDimitry Andric       bool HasMultipleUses = false;
3556480093f4SDimitry Andric       for (unsigned i = 1, e = NodeStack.size()-1; i != e; ++i) {
3557480093f4SDimitry Andric         unsigned NNonChainUses = 0;
3558480093f4SDimitry Andric         SDNode *NS = NodeStack[i].getNode();
3559480093f4SDimitry Andric         for (auto UI = NS->use_begin(), UE = NS->use_end(); UI != UE; ++UI)
3560480093f4SDimitry Andric           if (UI.getUse().getValueType() != MVT::Other)
3561480093f4SDimitry Andric             if (++NNonChainUses > 1) {
35620b57cec5SDimitry Andric               HasMultipleUses = true;
35630b57cec5SDimitry Andric               break;
35640b57cec5SDimitry Andric             }
35650b57cec5SDimitry Andric         if (HasMultipleUses) break;
3566480093f4SDimitry Andric       }
3567480093f4SDimitry Andric       if (HasMultipleUses) break;
35680b57cec5SDimitry Andric 
35690b57cec5SDimitry Andric       // Check to see that the target thinks this is profitable to fold and that
35700b57cec5SDimitry Andric       // we can fold it without inducing cycles in the graph.
35710b57cec5SDimitry Andric       if (!IsProfitableToFold(N, NodeStack[NodeStack.size()-2].getNode(),
35720b57cec5SDimitry Andric                               NodeToMatch) ||
35730b57cec5SDimitry Andric           !IsLegalToFold(N, NodeStack[NodeStack.size()-2].getNode(),
35740b57cec5SDimitry Andric                          NodeToMatch, OptLevel,
35750b57cec5SDimitry Andric                          true/*We validate our own chains*/))
35760b57cec5SDimitry Andric         break;
35770b57cec5SDimitry Andric 
35780b57cec5SDimitry Andric       continue;
35790b57cec5SDimitry Andric     }
3580fe6060f1SDimitry Andric     case OPC_EmitInteger:
3581*5f757f3fSDimitry Andric     case OPC_EmitInteger8:
3582*5f757f3fSDimitry Andric     case OPC_EmitInteger16:
3583*5f757f3fSDimitry Andric     case OPC_EmitInteger32:
3584*5f757f3fSDimitry Andric     case OPC_EmitInteger64:
3585*5f757f3fSDimitry Andric     case OPC_EmitStringInteger:
3586*5f757f3fSDimitry Andric     case OPC_EmitStringInteger32: {
3587*5f757f3fSDimitry Andric       MVT::SimpleValueType VT;
3588*5f757f3fSDimitry Andric       switch (Opcode) {
3589*5f757f3fSDimitry Andric       case OPC_EmitInteger8:
3590*5f757f3fSDimitry Andric         VT = MVT::i8;
3591*5f757f3fSDimitry Andric         break;
3592*5f757f3fSDimitry Andric       case OPC_EmitInteger16:
3593*5f757f3fSDimitry Andric         VT = MVT::i16;
3594*5f757f3fSDimitry Andric         break;
3595*5f757f3fSDimitry Andric       case OPC_EmitInteger32:
3596*5f757f3fSDimitry Andric       case OPC_EmitStringInteger32:
3597*5f757f3fSDimitry Andric         VT = MVT::i32;
3598*5f757f3fSDimitry Andric         break;
3599*5f757f3fSDimitry Andric       case OPC_EmitInteger64:
3600*5f757f3fSDimitry Andric         VT = MVT::i64;
3601*5f757f3fSDimitry Andric         break;
3602*5f757f3fSDimitry Andric       default:
3603*5f757f3fSDimitry Andric         VT = static_cast<MVT::SimpleValueType>(MatcherTable[MatcherIndex++]);
3604*5f757f3fSDimitry Andric         break;
3605*5f757f3fSDimitry Andric       }
36060b57cec5SDimitry Andric       int64_t Val = MatcherTable[MatcherIndex++];
36070b57cec5SDimitry Andric       if (Val & 128)
36080b57cec5SDimitry Andric         Val = GetVBR(Val, MatcherTable, MatcherIndex);
3609*5f757f3fSDimitry Andric       if (Opcode >= OPC_EmitInteger && Opcode <= OPC_EmitInteger64)
3610fe6060f1SDimitry Andric         Val = decodeSignRotatedValue(Val);
36110b57cec5SDimitry Andric       RecordedNodes.push_back(std::pair<SDValue, SDNode *>(
3612*5f757f3fSDimitry Andric           CurDAG->getTargetConstant(Val, SDLoc(NodeToMatch), VT), nullptr));
36130b57cec5SDimitry Andric       continue;
36140b57cec5SDimitry Andric     }
36150b57cec5SDimitry Andric     case OPC_EmitRegister: {
36160b57cec5SDimitry Andric       MVT::SimpleValueType VT =
3617*5f757f3fSDimitry Andric           static_cast<MVT::SimpleValueType>(MatcherTable[MatcherIndex++]);
36180b57cec5SDimitry Andric       unsigned RegNo = MatcherTable[MatcherIndex++];
36190b57cec5SDimitry Andric       RecordedNodes.push_back(std::pair<SDValue, SDNode*>(
36200b57cec5SDimitry Andric                               CurDAG->getRegister(RegNo, VT), nullptr));
36210b57cec5SDimitry Andric       continue;
36220b57cec5SDimitry Andric     }
36230b57cec5SDimitry Andric     case OPC_EmitRegister2: {
36240b57cec5SDimitry Andric       // For targets w/ more than 256 register names, the register enum
36250b57cec5SDimitry Andric       // values are stored in two bytes in the matcher table (just like
36260b57cec5SDimitry Andric       // opcodes).
36270b57cec5SDimitry Andric       MVT::SimpleValueType VT =
3628*5f757f3fSDimitry Andric           static_cast<MVT::SimpleValueType>(MatcherTable[MatcherIndex++]);
36290b57cec5SDimitry Andric       unsigned RegNo = MatcherTable[MatcherIndex++];
36300b57cec5SDimitry Andric       RegNo |= MatcherTable[MatcherIndex++] << 8;
36310b57cec5SDimitry Andric       RecordedNodes.push_back(std::pair<SDValue, SDNode*>(
36320b57cec5SDimitry Andric                               CurDAG->getRegister(RegNo, VT), nullptr));
36330b57cec5SDimitry Andric       continue;
36340b57cec5SDimitry Andric     }
36350b57cec5SDimitry Andric 
3636*5f757f3fSDimitry Andric     case OPC_EmitConvertToTarget:
3637*5f757f3fSDimitry Andric     case OPC_EmitConvertToTarget0:
3638*5f757f3fSDimitry Andric     case OPC_EmitConvertToTarget1:
3639*5f757f3fSDimitry Andric     case OPC_EmitConvertToTarget2:
3640*5f757f3fSDimitry Andric     case OPC_EmitConvertToTarget3:
3641*5f757f3fSDimitry Andric     case OPC_EmitConvertToTarget4:
3642*5f757f3fSDimitry Andric     case OPC_EmitConvertToTarget5:
3643*5f757f3fSDimitry Andric     case OPC_EmitConvertToTarget6:
3644*5f757f3fSDimitry Andric     case OPC_EmitConvertToTarget7: {
36450b57cec5SDimitry Andric       // Convert from IMM/FPIMM to target version.
3646*5f757f3fSDimitry Andric       unsigned RecNo = Opcode == OPC_EmitConvertToTarget
3647*5f757f3fSDimitry Andric                            ? MatcherTable[MatcherIndex++]
3648*5f757f3fSDimitry Andric                            : Opcode - OPC_EmitConvertToTarget0;
36490b57cec5SDimitry Andric       assert(RecNo < RecordedNodes.size() && "Invalid EmitConvertToTarget");
36500b57cec5SDimitry Andric       SDValue Imm = RecordedNodes[RecNo].first;
36510b57cec5SDimitry Andric 
36520b57cec5SDimitry Andric       if (Imm->getOpcode() == ISD::Constant) {
36530b57cec5SDimitry Andric         const ConstantInt *Val=cast<ConstantSDNode>(Imm)->getConstantIntValue();
36540b57cec5SDimitry Andric         Imm = CurDAG->getTargetConstant(*Val, SDLoc(NodeToMatch),
36550b57cec5SDimitry Andric                                         Imm.getValueType());
36560b57cec5SDimitry Andric       } else if (Imm->getOpcode() == ISD::ConstantFP) {
36570b57cec5SDimitry Andric         const ConstantFP *Val=cast<ConstantFPSDNode>(Imm)->getConstantFPValue();
36580b57cec5SDimitry Andric         Imm = CurDAG->getTargetConstantFP(*Val, SDLoc(NodeToMatch),
36590b57cec5SDimitry Andric                                           Imm.getValueType());
36600b57cec5SDimitry Andric       }
36610b57cec5SDimitry Andric 
36620b57cec5SDimitry Andric       RecordedNodes.push_back(std::make_pair(Imm, RecordedNodes[RecNo].second));
36630b57cec5SDimitry Andric       continue;
36640b57cec5SDimitry Andric     }
36650b57cec5SDimitry Andric 
36660b57cec5SDimitry Andric     case OPC_EmitMergeInputChains1_0:    // OPC_EmitMergeInputChains, 1, 0
36670b57cec5SDimitry Andric     case OPC_EmitMergeInputChains1_1:    // OPC_EmitMergeInputChains, 1, 1
36680b57cec5SDimitry Andric     case OPC_EmitMergeInputChains1_2: {  // OPC_EmitMergeInputChains, 1, 2
36690b57cec5SDimitry Andric       // These are space-optimized forms of OPC_EmitMergeInputChains.
36700b57cec5SDimitry Andric       assert(!InputChain.getNode() &&
36710b57cec5SDimitry Andric              "EmitMergeInputChains should be the first chain producing node");
36720b57cec5SDimitry Andric       assert(ChainNodesMatched.empty() &&
36730b57cec5SDimitry Andric              "Should only have one EmitMergeInputChains per match");
36740b57cec5SDimitry Andric 
36750b57cec5SDimitry Andric       // Read all of the chained nodes.
36760b57cec5SDimitry Andric       unsigned RecNo = Opcode - OPC_EmitMergeInputChains1_0;
36770b57cec5SDimitry Andric       assert(RecNo < RecordedNodes.size() && "Invalid EmitMergeInputChains");
36780b57cec5SDimitry Andric       ChainNodesMatched.push_back(RecordedNodes[RecNo].first.getNode());
36790b57cec5SDimitry Andric 
368081ad6265SDimitry Andric       // If the chained node is not the root, we can't fold it if it has
368181ad6265SDimitry Andric       // multiple uses.
36820b57cec5SDimitry Andric       // FIXME: What if other value results of the node have uses not matched
36830b57cec5SDimitry Andric       // by this pattern?
36840b57cec5SDimitry Andric       if (ChainNodesMatched.back() != NodeToMatch &&
36850b57cec5SDimitry Andric           !RecordedNodes[RecNo].first.hasOneUse()) {
36860b57cec5SDimitry Andric         ChainNodesMatched.clear();
36870b57cec5SDimitry Andric         break;
36880b57cec5SDimitry Andric       }
36890b57cec5SDimitry Andric 
36900b57cec5SDimitry Andric       // Merge the input chains if they are not intra-pattern references.
36910b57cec5SDimitry Andric       InputChain = HandleMergeInputChains(ChainNodesMatched, CurDAG);
36920b57cec5SDimitry Andric 
36930b57cec5SDimitry Andric       if (!InputChain.getNode())
36940b57cec5SDimitry Andric         break;  // Failed to merge.
36950b57cec5SDimitry Andric       continue;
36960b57cec5SDimitry Andric     }
36970b57cec5SDimitry Andric 
36980b57cec5SDimitry Andric     case OPC_EmitMergeInputChains: {
36990b57cec5SDimitry Andric       assert(!InputChain.getNode() &&
37000b57cec5SDimitry Andric              "EmitMergeInputChains should be the first chain producing node");
37010b57cec5SDimitry Andric       // This node gets a list of nodes we matched in the input that have
37020b57cec5SDimitry Andric       // chains.  We want to token factor all of the input chains to these nodes
37030b57cec5SDimitry Andric       // together.  However, if any of the input chains is actually one of the
37040b57cec5SDimitry Andric       // nodes matched in this pattern, then we have an intra-match reference.
37050b57cec5SDimitry Andric       // Ignore these because the newly token factored chain should not refer to
37060b57cec5SDimitry Andric       // the old nodes.
37070b57cec5SDimitry Andric       unsigned NumChains = MatcherTable[MatcherIndex++];
37080b57cec5SDimitry Andric       assert(NumChains != 0 && "Can't TF zero chains");
37090b57cec5SDimitry Andric 
37100b57cec5SDimitry Andric       assert(ChainNodesMatched.empty() &&
37110b57cec5SDimitry Andric              "Should only have one EmitMergeInputChains per match");
37120b57cec5SDimitry Andric 
37130b57cec5SDimitry Andric       // Read all of the chained nodes.
37140b57cec5SDimitry Andric       for (unsigned i = 0; i != NumChains; ++i) {
37150b57cec5SDimitry Andric         unsigned RecNo = MatcherTable[MatcherIndex++];
37160b57cec5SDimitry Andric         assert(RecNo < RecordedNodes.size() && "Invalid EmitMergeInputChains");
37170b57cec5SDimitry Andric         ChainNodesMatched.push_back(RecordedNodes[RecNo].first.getNode());
37180b57cec5SDimitry Andric 
371981ad6265SDimitry Andric         // If the chained node is not the root, we can't fold it if it has
372081ad6265SDimitry Andric         // multiple uses.
37210b57cec5SDimitry Andric         // FIXME: What if other value results of the node have uses not matched
37220b57cec5SDimitry Andric         // by this pattern?
37230b57cec5SDimitry Andric         if (ChainNodesMatched.back() != NodeToMatch &&
37240b57cec5SDimitry Andric             !RecordedNodes[RecNo].first.hasOneUse()) {
37250b57cec5SDimitry Andric           ChainNodesMatched.clear();
37260b57cec5SDimitry Andric           break;
37270b57cec5SDimitry Andric         }
37280b57cec5SDimitry Andric       }
37290b57cec5SDimitry Andric 
37300b57cec5SDimitry Andric       // If the inner loop broke out, the match fails.
37310b57cec5SDimitry Andric       if (ChainNodesMatched.empty())
37320b57cec5SDimitry Andric         break;
37330b57cec5SDimitry Andric 
37340b57cec5SDimitry Andric       // Merge the input chains if they are not intra-pattern references.
37350b57cec5SDimitry Andric       InputChain = HandleMergeInputChains(ChainNodesMatched, CurDAG);
37360b57cec5SDimitry Andric 
37370b57cec5SDimitry Andric       if (!InputChain.getNode())
37380b57cec5SDimitry Andric         break;  // Failed to merge.
37390b57cec5SDimitry Andric 
37400b57cec5SDimitry Andric       continue;
37410b57cec5SDimitry Andric     }
37420b57cec5SDimitry Andric 
37438bcb0991SDimitry Andric     case OPC_EmitCopyToReg:
3744*5f757f3fSDimitry Andric     case OPC_EmitCopyToReg0:
3745*5f757f3fSDimitry Andric     case OPC_EmitCopyToReg1:
3746*5f757f3fSDimitry Andric     case OPC_EmitCopyToReg2:
3747*5f757f3fSDimitry Andric     case OPC_EmitCopyToReg3:
3748*5f757f3fSDimitry Andric     case OPC_EmitCopyToReg4:
3749*5f757f3fSDimitry Andric     case OPC_EmitCopyToReg5:
3750*5f757f3fSDimitry Andric     case OPC_EmitCopyToReg6:
3751*5f757f3fSDimitry Andric     case OPC_EmitCopyToReg7:
3752*5f757f3fSDimitry Andric     case OPC_EmitCopyToRegTwoByte: {
3753*5f757f3fSDimitry Andric       unsigned RecNo =
3754*5f757f3fSDimitry Andric           Opcode >= OPC_EmitCopyToReg0 && Opcode <= OPC_EmitCopyToReg7
3755*5f757f3fSDimitry Andric               ? Opcode - OPC_EmitCopyToReg0
3756*5f757f3fSDimitry Andric               : MatcherTable[MatcherIndex++];
37570b57cec5SDimitry Andric       assert(RecNo < RecordedNodes.size() && "Invalid EmitCopyToReg");
37580b57cec5SDimitry Andric       unsigned DestPhysReg = MatcherTable[MatcherIndex++];
3759*5f757f3fSDimitry Andric       if (Opcode == OPC_EmitCopyToRegTwoByte)
37608bcb0991SDimitry Andric         DestPhysReg |= MatcherTable[MatcherIndex++] << 8;
37610b57cec5SDimitry Andric 
37620b57cec5SDimitry Andric       if (!InputChain.getNode())
37630b57cec5SDimitry Andric         InputChain = CurDAG->getEntryNode();
37640b57cec5SDimitry Andric 
37650b57cec5SDimitry Andric       InputChain = CurDAG->getCopyToReg(InputChain, SDLoc(NodeToMatch),
37660b57cec5SDimitry Andric                                         DestPhysReg, RecordedNodes[RecNo].first,
37670b57cec5SDimitry Andric                                         InputGlue);
37680b57cec5SDimitry Andric 
37690b57cec5SDimitry Andric       InputGlue = InputChain.getValue(1);
37700b57cec5SDimitry Andric       continue;
37710b57cec5SDimitry Andric     }
37720b57cec5SDimitry Andric 
37730b57cec5SDimitry Andric     case OPC_EmitNodeXForm: {
37740b57cec5SDimitry Andric       unsigned XFormNo = MatcherTable[MatcherIndex++];
37750b57cec5SDimitry Andric       unsigned RecNo = MatcherTable[MatcherIndex++];
37760b57cec5SDimitry Andric       assert(RecNo < RecordedNodes.size() && "Invalid EmitNodeXForm");
37770b57cec5SDimitry Andric       SDValue Res = RunSDNodeXForm(RecordedNodes[RecNo].first, XFormNo);
37780b57cec5SDimitry Andric       RecordedNodes.push_back(std::pair<SDValue,SDNode*>(Res, nullptr));
37790b57cec5SDimitry Andric       continue;
37800b57cec5SDimitry Andric     }
37810b57cec5SDimitry Andric     case OPC_Coverage: {
37820b57cec5SDimitry Andric       // This is emitted right before MorphNode/EmitNode.
37830b57cec5SDimitry Andric       // So it should be safe to assume that this node has been selected
37840b57cec5SDimitry Andric       unsigned index = MatcherTable[MatcherIndex++];
37850b57cec5SDimitry Andric       index |= (MatcherTable[MatcherIndex++] << 8);
37860b57cec5SDimitry Andric       dbgs() << "COVERED: " << getPatternForIndex(index) << "\n";
37870b57cec5SDimitry Andric       dbgs() << "INCLUDED: " << getIncludePathForIndex(index) << "\n";
37880b57cec5SDimitry Andric       continue;
37890b57cec5SDimitry Andric     }
37900b57cec5SDimitry Andric 
3791*5f757f3fSDimitry Andric     case OPC_EmitNode:
3792*5f757f3fSDimitry Andric     case OPC_EmitNode0:
3793*5f757f3fSDimitry Andric     case OPC_EmitNode1:
3794*5f757f3fSDimitry Andric     case OPC_EmitNode2:
3795*5f757f3fSDimitry Andric     case OPC_EmitNode0None:
3796*5f757f3fSDimitry Andric     case OPC_EmitNode1None:
3797*5f757f3fSDimitry Andric     case OPC_EmitNode2None:
3798*5f757f3fSDimitry Andric     case OPC_EmitNode0Chain:
3799*5f757f3fSDimitry Andric     case OPC_EmitNode1Chain:
3800*5f757f3fSDimitry Andric     case OPC_EmitNode2Chain:
3801*5f757f3fSDimitry Andric     case OPC_MorphNodeTo:
3802*5f757f3fSDimitry Andric     case OPC_MorphNodeTo0:
3803*5f757f3fSDimitry Andric     case OPC_MorphNodeTo1:
3804*5f757f3fSDimitry Andric     case OPC_MorphNodeTo2:
3805*5f757f3fSDimitry Andric     case OPC_MorphNodeTo0None:
3806*5f757f3fSDimitry Andric     case OPC_MorphNodeTo1None:
3807*5f757f3fSDimitry Andric     case OPC_MorphNodeTo2None:
3808*5f757f3fSDimitry Andric     case OPC_MorphNodeTo0Chain:
3809*5f757f3fSDimitry Andric     case OPC_MorphNodeTo1Chain:
3810*5f757f3fSDimitry Andric     case OPC_MorphNodeTo2Chain:
3811*5f757f3fSDimitry Andric     case OPC_MorphNodeTo0GlueInput:
3812*5f757f3fSDimitry Andric     case OPC_MorphNodeTo1GlueInput:
3813*5f757f3fSDimitry Andric     case OPC_MorphNodeTo2GlueInput:
3814*5f757f3fSDimitry Andric     case OPC_MorphNodeTo0GlueOutput:
3815*5f757f3fSDimitry Andric     case OPC_MorphNodeTo1GlueOutput:
3816*5f757f3fSDimitry Andric     case OPC_MorphNodeTo2GlueOutput: {
38170b57cec5SDimitry Andric       uint16_t TargetOpc = MatcherTable[MatcherIndex++];
3818*5f757f3fSDimitry Andric       TargetOpc |= static_cast<uint16_t>(MatcherTable[MatcherIndex++]) << 8;
3819*5f757f3fSDimitry Andric       unsigned EmitNodeInfo;
3820*5f757f3fSDimitry Andric       if (Opcode >= OPC_EmitNode0None && Opcode <= OPC_EmitNode2Chain) {
3821*5f757f3fSDimitry Andric         if (Opcode >= OPC_EmitNode0Chain && Opcode <= OPC_EmitNode2Chain)
3822*5f757f3fSDimitry Andric           EmitNodeInfo = OPFL_Chain;
3823*5f757f3fSDimitry Andric         else
3824*5f757f3fSDimitry Andric           EmitNodeInfo = OPFL_None;
3825*5f757f3fSDimitry Andric       } else if (Opcode >= OPC_MorphNodeTo0None &&
3826*5f757f3fSDimitry Andric                  Opcode <= OPC_MorphNodeTo2GlueOutput) {
3827*5f757f3fSDimitry Andric         if (Opcode >= OPC_MorphNodeTo0Chain && Opcode <= OPC_MorphNodeTo2Chain)
3828*5f757f3fSDimitry Andric           EmitNodeInfo = OPFL_Chain;
3829*5f757f3fSDimitry Andric         else if (Opcode >= OPC_MorphNodeTo0GlueInput &&
3830*5f757f3fSDimitry Andric                  Opcode <= OPC_MorphNodeTo2GlueInput)
3831*5f757f3fSDimitry Andric           EmitNodeInfo = OPFL_GlueInput;
3832*5f757f3fSDimitry Andric         else if (Opcode >= OPC_MorphNodeTo0GlueOutput &&
3833*5f757f3fSDimitry Andric                  Opcode <= OPC_MorphNodeTo2GlueOutput)
3834*5f757f3fSDimitry Andric           EmitNodeInfo = OPFL_GlueOutput;
3835*5f757f3fSDimitry Andric         else
3836*5f757f3fSDimitry Andric           EmitNodeInfo = OPFL_None;
3837*5f757f3fSDimitry Andric       } else
3838*5f757f3fSDimitry Andric         EmitNodeInfo = MatcherTable[MatcherIndex++];
38390b57cec5SDimitry Andric       // Get the result VT list.
38400b57cec5SDimitry Andric       unsigned NumVTs;
38410b57cec5SDimitry Andric       // If this is one of the compressed forms, get the number of VTs based
38420b57cec5SDimitry Andric       // on the Opcode. Otherwise read the next byte from the table.
38430b57cec5SDimitry Andric       if (Opcode >= OPC_MorphNodeTo0 && Opcode <= OPC_MorphNodeTo2)
38440b57cec5SDimitry Andric         NumVTs = Opcode - OPC_MorphNodeTo0;
3845*5f757f3fSDimitry Andric       else if (Opcode >= OPC_MorphNodeTo0None && Opcode <= OPC_MorphNodeTo2None)
3846*5f757f3fSDimitry Andric         NumVTs = Opcode - OPC_MorphNodeTo0None;
3847*5f757f3fSDimitry Andric       else if (Opcode >= OPC_MorphNodeTo0Chain &&
3848*5f757f3fSDimitry Andric                Opcode <= OPC_MorphNodeTo2Chain)
3849*5f757f3fSDimitry Andric         NumVTs = Opcode - OPC_MorphNodeTo0Chain;
3850*5f757f3fSDimitry Andric       else if (Opcode >= OPC_MorphNodeTo0GlueInput &&
3851*5f757f3fSDimitry Andric                Opcode <= OPC_MorphNodeTo2GlueInput)
3852*5f757f3fSDimitry Andric         NumVTs = Opcode - OPC_MorphNodeTo0GlueInput;
3853*5f757f3fSDimitry Andric       else if (Opcode >= OPC_MorphNodeTo0GlueOutput &&
3854*5f757f3fSDimitry Andric                Opcode <= OPC_MorphNodeTo2GlueOutput)
3855*5f757f3fSDimitry Andric         NumVTs = Opcode - OPC_MorphNodeTo0GlueOutput;
38560b57cec5SDimitry Andric       else if (Opcode >= OPC_EmitNode0 && Opcode <= OPC_EmitNode2)
38570b57cec5SDimitry Andric         NumVTs = Opcode - OPC_EmitNode0;
3858*5f757f3fSDimitry Andric       else if (Opcode >= OPC_EmitNode0None && Opcode <= OPC_EmitNode2None)
3859*5f757f3fSDimitry Andric         NumVTs = Opcode - OPC_EmitNode0None;
3860*5f757f3fSDimitry Andric       else if (Opcode >= OPC_EmitNode0Chain && Opcode <= OPC_EmitNode2Chain)
3861*5f757f3fSDimitry Andric         NumVTs = Opcode - OPC_EmitNode0Chain;
38620b57cec5SDimitry Andric       else
38630b57cec5SDimitry Andric         NumVTs = MatcherTable[MatcherIndex++];
38640b57cec5SDimitry Andric       SmallVector<EVT, 4> VTs;
38650b57cec5SDimitry Andric       for (unsigned i = 0; i != NumVTs; ++i) {
38660b57cec5SDimitry Andric         MVT::SimpleValueType VT =
3867*5f757f3fSDimitry Andric             static_cast<MVT::SimpleValueType>(MatcherTable[MatcherIndex++]);
38680b57cec5SDimitry Andric         if (VT == MVT::iPTR)
38690b57cec5SDimitry Andric           VT = TLI->getPointerTy(CurDAG->getDataLayout()).SimpleTy;
38700b57cec5SDimitry Andric         VTs.push_back(VT);
38710b57cec5SDimitry Andric       }
38720b57cec5SDimitry Andric 
38730b57cec5SDimitry Andric       if (EmitNodeInfo & OPFL_Chain)
38740b57cec5SDimitry Andric         VTs.push_back(MVT::Other);
38750b57cec5SDimitry Andric       if (EmitNodeInfo & OPFL_GlueOutput)
38760b57cec5SDimitry Andric         VTs.push_back(MVT::Glue);
38770b57cec5SDimitry Andric 
38780b57cec5SDimitry Andric       // This is hot code, so optimize the two most common cases of 1 and 2
38790b57cec5SDimitry Andric       // results.
38800b57cec5SDimitry Andric       SDVTList VTList;
38810b57cec5SDimitry Andric       if (VTs.size() == 1)
38820b57cec5SDimitry Andric         VTList = CurDAG->getVTList(VTs[0]);
38830b57cec5SDimitry Andric       else if (VTs.size() == 2)
38840b57cec5SDimitry Andric         VTList = CurDAG->getVTList(VTs[0], VTs[1]);
38850b57cec5SDimitry Andric       else
38860b57cec5SDimitry Andric         VTList = CurDAG->getVTList(VTs);
38870b57cec5SDimitry Andric 
38880b57cec5SDimitry Andric       // Get the operand list.
38890b57cec5SDimitry Andric       unsigned NumOps = MatcherTable[MatcherIndex++];
38900b57cec5SDimitry Andric       SmallVector<SDValue, 8> Ops;
38910b57cec5SDimitry Andric       for (unsigned i = 0; i != NumOps; ++i) {
38920b57cec5SDimitry Andric         unsigned RecNo = MatcherTable[MatcherIndex++];
38930b57cec5SDimitry Andric         if (RecNo & 128)
38940b57cec5SDimitry Andric           RecNo = GetVBR(RecNo, MatcherTable, MatcherIndex);
38950b57cec5SDimitry Andric 
38960b57cec5SDimitry Andric         assert(RecNo < RecordedNodes.size() && "Invalid EmitNode");
38970b57cec5SDimitry Andric         Ops.push_back(RecordedNodes[RecNo].first);
38980b57cec5SDimitry Andric       }
38990b57cec5SDimitry Andric 
39000b57cec5SDimitry Andric       // If there are variadic operands to add, handle them now.
39010b57cec5SDimitry Andric       if (EmitNodeInfo & OPFL_VariadicInfo) {
39020b57cec5SDimitry Andric         // Determine the start index to copy from.
39030b57cec5SDimitry Andric         unsigned FirstOpToCopy = getNumFixedFromVariadicInfo(EmitNodeInfo);
39040b57cec5SDimitry Andric         FirstOpToCopy += (EmitNodeInfo & OPFL_Chain) ? 1 : 0;
39050b57cec5SDimitry Andric         assert(NodeToMatch->getNumOperands() >= FirstOpToCopy &&
39060b57cec5SDimitry Andric                "Invalid variadic node");
39070b57cec5SDimitry Andric         // Copy all of the variadic operands, not including a potential glue
39080b57cec5SDimitry Andric         // input.
39090b57cec5SDimitry Andric         for (unsigned i = FirstOpToCopy, e = NodeToMatch->getNumOperands();
39100b57cec5SDimitry Andric              i != e; ++i) {
39110b57cec5SDimitry Andric           SDValue V = NodeToMatch->getOperand(i);
39120b57cec5SDimitry Andric           if (V.getValueType() == MVT::Glue) break;
39130b57cec5SDimitry Andric           Ops.push_back(V);
39140b57cec5SDimitry Andric         }
39150b57cec5SDimitry Andric       }
39160b57cec5SDimitry Andric 
39170b57cec5SDimitry Andric       // If this has chain/glue inputs, add them.
39180b57cec5SDimitry Andric       if (EmitNodeInfo & OPFL_Chain)
39190b57cec5SDimitry Andric         Ops.push_back(InputChain);
39200b57cec5SDimitry Andric       if ((EmitNodeInfo & OPFL_GlueInput) && InputGlue.getNode() != nullptr)
39210b57cec5SDimitry Andric         Ops.push_back(InputGlue);
39220b57cec5SDimitry Andric 
3923480093f4SDimitry Andric       // Check whether any matched node could raise an FP exception.  Since all
3924480093f4SDimitry Andric       // such nodes must have a chain, it suffices to check ChainNodesMatched.
3925480093f4SDimitry Andric       // We need to perform this check before potentially modifying one of the
3926480093f4SDimitry Andric       // nodes via MorphNode.
392781ad6265SDimitry Andric       bool MayRaiseFPException =
392881ad6265SDimitry Andric           llvm::any_of(ChainNodesMatched, [this](SDNode *N) {
392981ad6265SDimitry Andric             return mayRaiseFPException(N) && !N->getFlags().hasNoFPExcept();
393081ad6265SDimitry Andric           });
3931480093f4SDimitry Andric 
39320b57cec5SDimitry Andric       // Create the node.
39330b57cec5SDimitry Andric       MachineSDNode *Res = nullptr;
3934*5f757f3fSDimitry Andric       bool IsMorphNodeTo =
3935*5f757f3fSDimitry Andric           Opcode == OPC_MorphNodeTo ||
3936*5f757f3fSDimitry Andric           (Opcode >= OPC_MorphNodeTo0 && Opcode <= OPC_MorphNodeTo2GlueOutput);
39370b57cec5SDimitry Andric       if (!IsMorphNodeTo) {
39380b57cec5SDimitry Andric         // If this is a normal EmitNode command, just create the new node and
39390b57cec5SDimitry Andric         // add the results to the RecordedNodes list.
39400b57cec5SDimitry Andric         Res = CurDAG->getMachineNode(TargetOpc, SDLoc(NodeToMatch),
39410b57cec5SDimitry Andric                                      VTList, Ops);
39420b57cec5SDimitry Andric 
39430b57cec5SDimitry Andric         // Add all the non-glue/non-chain results to the RecordedNodes list.
39440b57cec5SDimitry Andric         for (unsigned i = 0, e = VTs.size(); i != e; ++i) {
39450b57cec5SDimitry Andric           if (VTs[i] == MVT::Other || VTs[i] == MVT::Glue) break;
39460b57cec5SDimitry Andric           RecordedNodes.push_back(std::pair<SDValue,SDNode*>(SDValue(Res, i),
39470b57cec5SDimitry Andric                                                              nullptr));
39480b57cec5SDimitry Andric         }
39490b57cec5SDimitry Andric       } else {
39500b57cec5SDimitry Andric         assert(NodeToMatch->getOpcode() != ISD::DELETED_NODE &&
39510b57cec5SDimitry Andric                "NodeToMatch was removed partway through selection");
39520b57cec5SDimitry Andric         SelectionDAG::DAGNodeDeletedListener NDL(*CurDAG, [&](SDNode *N,
39530b57cec5SDimitry Andric                                                               SDNode *E) {
39540b57cec5SDimitry Andric           CurDAG->salvageDebugInfo(*N);
39550b57cec5SDimitry Andric           auto &Chain = ChainNodesMatched;
39560b57cec5SDimitry Andric           assert((!E || !is_contained(Chain, N)) &&
39570b57cec5SDimitry Andric                  "Chain node replaced during MorphNode");
3958*5f757f3fSDimitry Andric           llvm::erase(Chain, N);
39590b57cec5SDimitry Andric         });
39600b57cec5SDimitry Andric         Res = cast<MachineSDNode>(MorphNode(NodeToMatch, TargetOpc, VTList,
39610b57cec5SDimitry Andric                                             Ops, EmitNodeInfo));
39620b57cec5SDimitry Andric       }
39630b57cec5SDimitry Andric 
3964480093f4SDimitry Andric       // Set the NoFPExcept flag when no original matched node could
3965480093f4SDimitry Andric       // raise an FP exception, but the new node potentially might.
3966480093f4SDimitry Andric       if (!MayRaiseFPException && mayRaiseFPException(Res)) {
3967480093f4SDimitry Andric         SDNodeFlags Flags = Res->getFlags();
3968480093f4SDimitry Andric         Flags.setNoFPExcept(true);
3969480093f4SDimitry Andric         Res->setFlags(Flags);
3970480093f4SDimitry Andric       }
3971480093f4SDimitry Andric 
39720b57cec5SDimitry Andric       // If the node had chain/glue results, update our notion of the current
39730b57cec5SDimitry Andric       // chain and glue.
39740b57cec5SDimitry Andric       if (EmitNodeInfo & OPFL_GlueOutput) {
39750b57cec5SDimitry Andric         InputGlue = SDValue(Res, VTs.size()-1);
39760b57cec5SDimitry Andric         if (EmitNodeInfo & OPFL_Chain)
39770b57cec5SDimitry Andric           InputChain = SDValue(Res, VTs.size()-2);
39780b57cec5SDimitry Andric       } else if (EmitNodeInfo & OPFL_Chain)
39790b57cec5SDimitry Andric         InputChain = SDValue(Res, VTs.size()-1);
39800b57cec5SDimitry Andric 
39810b57cec5SDimitry Andric       // If the OPFL_MemRefs glue is set on this node, slap all of the
39820b57cec5SDimitry Andric       // accumulated memrefs onto it.
39830b57cec5SDimitry Andric       //
39840b57cec5SDimitry Andric       // FIXME: This is vastly incorrect for patterns with multiple outputs
39850b57cec5SDimitry Andric       // instructions that access memory and for ComplexPatterns that match
39860b57cec5SDimitry Andric       // loads.
39870b57cec5SDimitry Andric       if (EmitNodeInfo & OPFL_MemRefs) {
39880b57cec5SDimitry Andric         // Only attach load or store memory operands if the generated
39890b57cec5SDimitry Andric         // instruction may load or store.
39900b57cec5SDimitry Andric         const MCInstrDesc &MCID = TII->get(TargetOpc);
39910b57cec5SDimitry Andric         bool mayLoad = MCID.mayLoad();
39920b57cec5SDimitry Andric         bool mayStore = MCID.mayStore();
39930b57cec5SDimitry Andric 
39940b57cec5SDimitry Andric         // We expect to have relatively few of these so just filter them into a
39950b57cec5SDimitry Andric         // temporary buffer so that we can easily add them to the instruction.
39960b57cec5SDimitry Andric         SmallVector<MachineMemOperand *, 4> FilteredMemRefs;
39970b57cec5SDimitry Andric         for (MachineMemOperand *MMO : MatchedMemRefs) {
39980b57cec5SDimitry Andric           if (MMO->isLoad()) {
39990b57cec5SDimitry Andric             if (mayLoad)
40000b57cec5SDimitry Andric               FilteredMemRefs.push_back(MMO);
40010b57cec5SDimitry Andric           } else if (MMO->isStore()) {
40020b57cec5SDimitry Andric             if (mayStore)
40030b57cec5SDimitry Andric               FilteredMemRefs.push_back(MMO);
40040b57cec5SDimitry Andric           } else {
40050b57cec5SDimitry Andric             FilteredMemRefs.push_back(MMO);
40060b57cec5SDimitry Andric           }
40070b57cec5SDimitry Andric         }
40080b57cec5SDimitry Andric 
40090b57cec5SDimitry Andric         CurDAG->setNodeMemRefs(Res, FilteredMemRefs);
40100b57cec5SDimitry Andric       }
40110b57cec5SDimitry Andric 
40120b57cec5SDimitry Andric       LLVM_DEBUG(if (!MatchedMemRefs.empty() && Res->memoperands_empty()) dbgs()
40130b57cec5SDimitry Andric                      << "  Dropping mem operands\n";
40140b57cec5SDimitry Andric                  dbgs() << "  " << (IsMorphNodeTo ? "Morphed" : "Created")
40150b57cec5SDimitry Andric                         << " node: ";
40160b57cec5SDimitry Andric                  Res->dump(CurDAG););
40170b57cec5SDimitry Andric 
40180b57cec5SDimitry Andric       // If this was a MorphNodeTo then we're completely done!
40190b57cec5SDimitry Andric       if (IsMorphNodeTo) {
40200b57cec5SDimitry Andric         // Update chain uses.
40210b57cec5SDimitry Andric         UpdateChains(Res, InputChain, ChainNodesMatched, true);
40220b57cec5SDimitry Andric         return;
40230b57cec5SDimitry Andric       }
40240b57cec5SDimitry Andric       continue;
40250b57cec5SDimitry Andric     }
40260b57cec5SDimitry Andric 
40270b57cec5SDimitry Andric     case OPC_CompleteMatch: {
40280b57cec5SDimitry Andric       // The match has been completed, and any new nodes (if any) have been
40290b57cec5SDimitry Andric       // created.  Patch up references to the matched dag to use the newly
40300b57cec5SDimitry Andric       // created nodes.
40310b57cec5SDimitry Andric       unsigned NumResults = MatcherTable[MatcherIndex++];
40320b57cec5SDimitry Andric 
40330b57cec5SDimitry Andric       for (unsigned i = 0; i != NumResults; ++i) {
40340b57cec5SDimitry Andric         unsigned ResSlot = MatcherTable[MatcherIndex++];
40350b57cec5SDimitry Andric         if (ResSlot & 128)
40360b57cec5SDimitry Andric           ResSlot = GetVBR(ResSlot, MatcherTable, MatcherIndex);
40370b57cec5SDimitry Andric 
40380b57cec5SDimitry Andric         assert(ResSlot < RecordedNodes.size() && "Invalid CompleteMatch");
40390b57cec5SDimitry Andric         SDValue Res = RecordedNodes[ResSlot].first;
40400b57cec5SDimitry Andric 
40410b57cec5SDimitry Andric         assert(i < NodeToMatch->getNumValues() &&
40420b57cec5SDimitry Andric                NodeToMatch->getValueType(i) != MVT::Other &&
40430b57cec5SDimitry Andric                NodeToMatch->getValueType(i) != MVT::Glue &&
40440b57cec5SDimitry Andric                "Invalid number of results to complete!");
40450b57cec5SDimitry Andric         assert((NodeToMatch->getValueType(i) == Res.getValueType() ||
40460b57cec5SDimitry Andric                 NodeToMatch->getValueType(i) == MVT::iPTR ||
40470b57cec5SDimitry Andric                 Res.getValueType() == MVT::iPTR ||
40480b57cec5SDimitry Andric                 NodeToMatch->getValueType(i).getSizeInBits() ==
40490b57cec5SDimitry Andric                     Res.getValueSizeInBits()) &&
40500b57cec5SDimitry Andric                "invalid replacement");
40510b57cec5SDimitry Andric         ReplaceUses(SDValue(NodeToMatch, i), Res);
40520b57cec5SDimitry Andric       }
40530b57cec5SDimitry Andric 
40540b57cec5SDimitry Andric       // Update chain uses.
40550b57cec5SDimitry Andric       UpdateChains(NodeToMatch, InputChain, ChainNodesMatched, false);
40560b57cec5SDimitry Andric 
40570b57cec5SDimitry Andric       // If the root node defines glue, we need to update it to the glue result.
40580b57cec5SDimitry Andric       // TODO: This never happens in our tests and I think it can be removed /
40590b57cec5SDimitry Andric       // replaced with an assert, but if we do it this the way the change is
40600b57cec5SDimitry Andric       // NFC.
40610b57cec5SDimitry Andric       if (NodeToMatch->getValueType(NodeToMatch->getNumValues() - 1) ==
40620b57cec5SDimitry Andric               MVT::Glue &&
40630b57cec5SDimitry Andric           InputGlue.getNode())
40640b57cec5SDimitry Andric         ReplaceUses(SDValue(NodeToMatch, NodeToMatch->getNumValues() - 1),
40650b57cec5SDimitry Andric                     InputGlue);
40660b57cec5SDimitry Andric 
40670b57cec5SDimitry Andric       assert(NodeToMatch->use_empty() &&
40680b57cec5SDimitry Andric              "Didn't replace all uses of the node?");
40690b57cec5SDimitry Andric       CurDAG->RemoveDeadNode(NodeToMatch);
40700b57cec5SDimitry Andric 
40710b57cec5SDimitry Andric       return;
40720b57cec5SDimitry Andric     }
40730b57cec5SDimitry Andric     }
40740b57cec5SDimitry Andric 
40750b57cec5SDimitry Andric     // If the code reached this point, then the match failed.  See if there is
40760b57cec5SDimitry Andric     // another child to try in the current 'Scope', otherwise pop it until we
40770b57cec5SDimitry Andric     // find a case to check.
40780b57cec5SDimitry Andric     LLVM_DEBUG(dbgs() << "  Match failed at index " << CurrentOpcodeIndex
40790b57cec5SDimitry Andric                       << "\n");
40800b57cec5SDimitry Andric     ++NumDAGIselRetries;
40810b57cec5SDimitry Andric     while (true) {
40820b57cec5SDimitry Andric       if (MatchScopes.empty()) {
40830b57cec5SDimitry Andric         CannotYetSelect(NodeToMatch);
40840b57cec5SDimitry Andric         return;
40850b57cec5SDimitry Andric       }
40860b57cec5SDimitry Andric 
40870b57cec5SDimitry Andric       // Restore the interpreter state back to the point where the scope was
40880b57cec5SDimitry Andric       // formed.
40890b57cec5SDimitry Andric       MatchScope &LastScope = MatchScopes.back();
40900b57cec5SDimitry Andric       RecordedNodes.resize(LastScope.NumRecordedNodes);
40910b57cec5SDimitry Andric       NodeStack.clear();
40920b57cec5SDimitry Andric       NodeStack.append(LastScope.NodeStack.begin(), LastScope.NodeStack.end());
40930b57cec5SDimitry Andric       N = NodeStack.back();
40940b57cec5SDimitry Andric 
40950b57cec5SDimitry Andric       if (LastScope.NumMatchedMemRefs != MatchedMemRefs.size())
40960b57cec5SDimitry Andric         MatchedMemRefs.resize(LastScope.NumMatchedMemRefs);
40970b57cec5SDimitry Andric       MatcherIndex = LastScope.FailIndex;
40980b57cec5SDimitry Andric 
40990b57cec5SDimitry Andric       LLVM_DEBUG(dbgs() << "  Continuing at " << MatcherIndex << "\n");
41000b57cec5SDimitry Andric 
41010b57cec5SDimitry Andric       InputChain = LastScope.InputChain;
41020b57cec5SDimitry Andric       InputGlue = LastScope.InputGlue;
41030b57cec5SDimitry Andric       if (!LastScope.HasChainNodesMatched)
41040b57cec5SDimitry Andric         ChainNodesMatched.clear();
41050b57cec5SDimitry Andric 
41060b57cec5SDimitry Andric       // Check to see what the offset is at the new MatcherIndex.  If it is zero
41070b57cec5SDimitry Andric       // we have reached the end of this scope, otherwise we have another child
41080b57cec5SDimitry Andric       // in the current scope to try.
41090b57cec5SDimitry Andric       unsigned NumToSkip = MatcherTable[MatcherIndex++];
41100b57cec5SDimitry Andric       if (NumToSkip & 128)
41110b57cec5SDimitry Andric         NumToSkip = GetVBR(NumToSkip, MatcherTable, MatcherIndex);
41120b57cec5SDimitry Andric 
41130b57cec5SDimitry Andric       // If we have another child in this scope to match, update FailIndex and
41140b57cec5SDimitry Andric       // try it.
41150b57cec5SDimitry Andric       if (NumToSkip != 0) {
41160b57cec5SDimitry Andric         LastScope.FailIndex = MatcherIndex+NumToSkip;
41170b57cec5SDimitry Andric         break;
41180b57cec5SDimitry Andric       }
41190b57cec5SDimitry Andric 
41200b57cec5SDimitry Andric       // End of this scope, pop it and try the next child in the containing
41210b57cec5SDimitry Andric       // scope.
41220b57cec5SDimitry Andric       MatchScopes.pop_back();
41230b57cec5SDimitry Andric     }
41240b57cec5SDimitry Andric   }
41250b57cec5SDimitry Andric }
41260b57cec5SDimitry Andric 
4127480093f4SDimitry Andric /// Return whether the node may raise an FP exception.
4128480093f4SDimitry Andric bool SelectionDAGISel::mayRaiseFPException(SDNode *N) const {
4129480093f4SDimitry Andric   // For machine opcodes, consult the MCID flag.
4130480093f4SDimitry Andric   if (N->isMachineOpcode()) {
4131480093f4SDimitry Andric     const MCInstrDesc &MCID = TII->get(N->getMachineOpcode());
4132480093f4SDimitry Andric     return MCID.mayRaiseFPException();
4133480093f4SDimitry Andric   }
4134480093f4SDimitry Andric 
4135480093f4SDimitry Andric   // For ISD opcodes, only StrictFP opcodes may raise an FP
4136480093f4SDimitry Andric   // exception.
4137480093f4SDimitry Andric   if (N->isTargetOpcode())
4138480093f4SDimitry Andric     return N->isTargetStrictFPOpcode();
4139480093f4SDimitry Andric   return N->isStrictFPOpcode();
4140480093f4SDimitry Andric }
4141480093f4SDimitry Andric 
41420b57cec5SDimitry Andric bool SelectionDAGISel::isOrEquivalentToAdd(const SDNode *N) const {
41430b57cec5SDimitry Andric   assert(N->getOpcode() == ISD::OR && "Unexpected opcode");
41440b57cec5SDimitry Andric   auto *C = dyn_cast<ConstantSDNode>(N->getOperand(1));
41450b57cec5SDimitry Andric   if (!C)
41460b57cec5SDimitry Andric     return false;
41470b57cec5SDimitry Andric 
41480b57cec5SDimitry Andric   // Detect when "or" is used to add an offset to a stack object.
41490b57cec5SDimitry Andric   if (auto *FN = dyn_cast<FrameIndexSDNode>(N->getOperand(0))) {
41500b57cec5SDimitry Andric     MachineFrameInfo &MFI = MF->getFrameInfo();
41515ffd83dbSDimitry Andric     Align A = MFI.getObjectAlign(FN->getIndex());
41520b57cec5SDimitry Andric     int32_t Off = C->getSExtValue();
41530b57cec5SDimitry Andric     // If the alleged offset fits in the zero bits guaranteed by
41540b57cec5SDimitry Andric     // the alignment, then this or is really an add.
41555ffd83dbSDimitry Andric     return (Off >= 0) && (((A.value() - 1) & Off) == unsigned(Off));
41560b57cec5SDimitry Andric   }
41570b57cec5SDimitry Andric   return false;
41580b57cec5SDimitry Andric }
41590b57cec5SDimitry Andric 
41600b57cec5SDimitry Andric void SelectionDAGISel::CannotYetSelect(SDNode *N) {
41610b57cec5SDimitry Andric   std::string msg;
41620b57cec5SDimitry Andric   raw_string_ostream Msg(msg);
41630b57cec5SDimitry Andric   Msg << "Cannot select: ";
41640b57cec5SDimitry Andric 
41650b57cec5SDimitry Andric   if (N->getOpcode() != ISD::INTRINSIC_W_CHAIN &&
41660b57cec5SDimitry Andric       N->getOpcode() != ISD::INTRINSIC_WO_CHAIN &&
41670b57cec5SDimitry Andric       N->getOpcode() != ISD::INTRINSIC_VOID) {
41680b57cec5SDimitry Andric     N->printrFull(Msg, CurDAG);
41690b57cec5SDimitry Andric     Msg << "\nIn function: " << MF->getName();
41700b57cec5SDimitry Andric   } else {
41710b57cec5SDimitry Andric     bool HasInputChain = N->getOperand(0).getValueType() == MVT::Other;
41720b57cec5SDimitry Andric     unsigned iid =
41730b57cec5SDimitry Andric       cast<ConstantSDNode>(N->getOperand(HasInputChain))->getZExtValue();
41740b57cec5SDimitry Andric     if (iid < Intrinsic::num_intrinsics)
4175fe6060f1SDimitry Andric       Msg << "intrinsic %" << Intrinsic::getBaseName((Intrinsic::ID)iid);
41760b57cec5SDimitry Andric     else if (const TargetIntrinsicInfo *TII = TM.getIntrinsicInfo())
41770b57cec5SDimitry Andric       Msg << "target intrinsic %" << TII->getName(iid);
41780b57cec5SDimitry Andric     else
41790b57cec5SDimitry Andric       Msg << "unknown intrinsic #" << iid;
41800b57cec5SDimitry Andric   }
4181349cc55cSDimitry Andric   report_fatal_error(Twine(Msg.str()));
41820b57cec5SDimitry Andric }
4183