Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 1 | //===-- SelectionDAGISel.cpp - Implement the SelectionDAGISel class -------===// |
Misha Brukman | edf128a | 2005-04-21 22:36:52 +0000 | [diff] [blame] | 2 | // |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
Chris Lattner | 4ee451d | 2007-12-29 20:36:04 +0000 | [diff] [blame] | 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
Misha Brukman | edf128a | 2005-04-21 22:36:52 +0000 | [diff] [blame] | 7 | // |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
| 10 | // This implements the SelectionDAGISel class. |
| 11 | // |
| 12 | //===----------------------------------------------------------------------===// |
| 13 | |
| 14 | #define DEBUG_TYPE "isel" |
Dan Gohman | 84fbac5 | 2009-02-06 17:22:58 +0000 | [diff] [blame] | 15 | #include "ScheduleDAGSDNodes.h" |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 16 | #include "SelectionDAGBuilder.h" |
Dan Gohman | 6277eb2 | 2009-11-23 17:16:22 +0000 | [diff] [blame] | 17 | #include "FunctionLoweringInfo.h" |
Dan Gohman | 84fbac5 | 2009-02-06 17:22:58 +0000 | [diff] [blame] | 18 | #include "llvm/CodeGen/SelectionDAGISel.h" |
Jim Laskey | c7c3f11 | 2006-10-16 20:52:31 +0000 | [diff] [blame] | 19 | #include "llvm/Analysis/AliasAnalysis.h" |
Devang Patel | 713f043 | 2009-09-16 21:09:07 +0000 | [diff] [blame] | 20 | #include "llvm/Analysis/DebugInfo.h" |
Anton Korobeynikov | 5502bf6 | 2007-04-04 21:14:49 +0000 | [diff] [blame] | 21 | #include "llvm/Constants.h" |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 22 | #include "llvm/Function.h" |
Chris Lattner | ce7518c | 2006-01-26 22:24:51 +0000 | [diff] [blame] | 23 | #include "llvm/InlineAsm.h" |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 24 | #include "llvm/Instructions.h" |
| 25 | #include "llvm/Intrinsics.h" |
Jim Laskey | 43970fe | 2006-03-23 18:06:46 +0000 | [diff] [blame] | 26 | #include "llvm/IntrinsicInst.h" |
Chris Lattner | 75c478a | 2009-10-27 17:02:08 +0000 | [diff] [blame] | 27 | #include "llvm/LLVMContext.h" |
Dan Gohman | 78eca17 | 2008-08-19 22:33:34 +0000 | [diff] [blame] | 28 | #include "llvm/CodeGen/FastISel.h" |
Gordon Henriksen | 5a29c9e | 2008-08-17 12:56:54 +0000 | [diff] [blame] | 29 | #include "llvm/CodeGen/GCStrategy.h" |
Gordon Henriksen | 5eca075 | 2008-08-17 18:44:35 +0000 | [diff] [blame] | 30 | #include "llvm/CodeGen/GCMetadata.h" |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 31 | #include "llvm/CodeGen/MachineFunction.h" |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 32 | #include "llvm/CodeGen/MachineInstrBuilder.h" |
Chris Lattner | 84bc542 | 2007-12-31 04:13:23 +0000 | [diff] [blame] | 33 | #include "llvm/CodeGen/MachineModuleInfo.h" |
| 34 | #include "llvm/CodeGen/MachineRegisterInfo.h" |
Dan Gohman | fc54c55 | 2009-01-15 22:18:12 +0000 | [diff] [blame] | 35 | #include "llvm/CodeGen/ScheduleHazardRecognizer.h" |
Jim Laskey | eb577ba | 2006-08-02 12:30:23 +0000 | [diff] [blame] | 36 | #include "llvm/CodeGen/SchedulerRegistry.h" |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 37 | #include "llvm/CodeGen/SelectionDAG.h" |
Dan Gohman | 6f0d024 | 2008-02-10 18:45:23 +0000 | [diff] [blame] | 38 | #include "llvm/Target/TargetRegisterInfo.h" |
Dan Gohman | e1f188f | 2009-10-29 22:30:23 +0000 | [diff] [blame] | 39 | #include "llvm/Target/TargetIntrinsicInfo.h" |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 40 | #include "llvm/Target/TargetInstrInfo.h" |
| 41 | #include "llvm/Target/TargetLowering.h" |
| 42 | #include "llvm/Target/TargetMachine.h" |
Vladimir Prus | 1247291 | 2006-05-23 13:43:15 +0000 | [diff] [blame] | 43 | #include "llvm/Target/TargetOptions.h" |
Chris Lattner | a4f0b3a | 2006-08-27 12:54:02 +0000 | [diff] [blame] | 44 | #include "llvm/Support/Compiler.h" |
Evan Cheng | db8d56b | 2008-06-30 20:45:06 +0000 | [diff] [blame] | 45 | #include "llvm/Support/Debug.h" |
Torok Edwin | 7d696d8 | 2009-07-11 13:10:19 +0000 | [diff] [blame] | 46 | #include "llvm/Support/ErrorHandling.h" |
Evan Cheng | db8d56b | 2008-06-30 20:45:06 +0000 | [diff] [blame] | 47 | #include "llvm/Support/Timer.h" |
Daniel Dunbar | ce63ffb | 2009-07-25 00:23:56 +0000 | [diff] [blame] | 48 | #include "llvm/Support/raw_ostream.h" |
Chris Lattner | c19ae9d | 2010-03-04 19:46:56 +0000 | [diff] [blame] | 49 | #include "llvm/ADT/Statistic.h" |
Jeff Cohen | 7e88103 | 2006-02-24 02:52:40 +0000 | [diff] [blame] | 50 | #include <algorithm> |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 51 | using namespace llvm; |
| 52 | |
Chris Lattner | c19ae9d | 2010-03-04 19:46:56 +0000 | [diff] [blame] | 53 | STATISTIC(NumFastIselFailures, "Number of instructions fast isel failed on"); |
Chris Lattner | bed993d | 2010-03-28 19:46:56 +0000 | [diff] [blame] | 54 | STATISTIC(NumDAGIselRetries,"Number of times dag isel has to try another path"); |
Chris Lattner | c19ae9d | 2010-03-04 19:46:56 +0000 | [diff] [blame] | 55 | |
Chris Lattner | ead0d88 | 2008-06-17 06:09:18 +0000 | [diff] [blame] | 56 | static cl::opt<bool> |
Dan Gohman | 293d5f8 | 2008-09-09 22:06:46 +0000 | [diff] [blame] | 57 | EnableFastISelVerbose("fast-isel-verbose", cl::Hidden, |
Dan Gohman | d659d50 | 2008-10-20 21:30:12 +0000 | [diff] [blame] | 58 | cl::desc("Enable verbose messages in the \"fast\" " |
Dan Gohman | 293d5f8 | 2008-09-09 22:06:46 +0000 | [diff] [blame] | 59 | "instruction selector")); |
| 60 | static cl::opt<bool> |
Dan Gohman | 4344a5d | 2008-09-09 23:05:00 +0000 | [diff] [blame] | 61 | EnableFastISelAbort("fast-isel-abort", cl::Hidden, |
| 62 | cl::desc("Enable abort calls when \"fast\" instruction fails")); |
Chris Lattner | ead0d88 | 2008-06-17 06:09:18 +0000 | [diff] [blame] | 63 | |
Chris Lattner | da8abb0 | 2005-09-01 18:44:10 +0000 | [diff] [blame] | 64 | #ifndef NDEBUG |
Chris Lattner | 7944d9d | 2005-01-12 03:41:21 +0000 | [diff] [blame] | 65 | static cl::opt<bool> |
Dan Gohman | 462dc7f | 2008-07-21 20:00:07 +0000 | [diff] [blame] | 66 | ViewDAGCombine1("view-dag-combine1-dags", cl::Hidden, |
| 67 | cl::desc("Pop up a window to show dags before the first " |
| 68 | "dag combine pass")); |
| 69 | static cl::opt<bool> |
| 70 | ViewLegalizeTypesDAGs("view-legalize-types-dags", cl::Hidden, |
| 71 | cl::desc("Pop up a window to show dags before legalize types")); |
| 72 | static cl::opt<bool> |
| 73 | ViewLegalizeDAGs("view-legalize-dags", cl::Hidden, |
| 74 | cl::desc("Pop up a window to show dags before legalize")); |
| 75 | static cl::opt<bool> |
| 76 | ViewDAGCombine2("view-dag-combine2-dags", cl::Hidden, |
| 77 | cl::desc("Pop up a window to show dags before the second " |
| 78 | "dag combine pass")); |
| 79 | static cl::opt<bool> |
Duncan Sands | 25cf227 | 2008-11-24 14:53:14 +0000 | [diff] [blame] | 80 | ViewDAGCombineLT("view-dag-combine-lt-dags", cl::Hidden, |
| 81 | cl::desc("Pop up a window to show dags before the post legalize types" |
| 82 | " dag combine pass")); |
| 83 | static cl::opt<bool> |
Evan Cheng | a9c2091 | 2006-01-21 02:32:06 +0000 | [diff] [blame] | 84 | ViewISelDAGs("view-isel-dags", cl::Hidden, |
| 85 | cl::desc("Pop up a window to show isel dags as they are selected")); |
| 86 | static cl::opt<bool> |
| 87 | ViewSchedDAGs("view-sched-dags", cl::Hidden, |
| 88 | cl::desc("Pop up a window to show sched dags as they are processed")); |
Dan Gohman | 3e1a7ae | 2007-08-28 20:32:58 +0000 | [diff] [blame] | 89 | static cl::opt<bool> |
| 90 | ViewSUnitDAGs("view-sunit-dags", cl::Hidden, |
Chris Lattner | 5bab785 | 2008-01-25 17:24:52 +0000 | [diff] [blame] | 91 | cl::desc("Pop up a window to show SUnit dags after they are processed")); |
Chris Lattner | 7944d9d | 2005-01-12 03:41:21 +0000 | [diff] [blame] | 92 | #else |
Dan Gohman | 462dc7f | 2008-07-21 20:00:07 +0000 | [diff] [blame] | 93 | static const bool ViewDAGCombine1 = false, |
| 94 | ViewLegalizeTypesDAGs = false, ViewLegalizeDAGs = false, |
| 95 | ViewDAGCombine2 = false, |
Duncan Sands | 25cf227 | 2008-11-24 14:53:14 +0000 | [diff] [blame] | 96 | ViewDAGCombineLT = false, |
Dan Gohman | 462dc7f | 2008-07-21 20:00:07 +0000 | [diff] [blame] | 97 | ViewISelDAGs = false, ViewSchedDAGs = false, |
| 98 | ViewSUnitDAGs = false; |
Chris Lattner | 7944d9d | 2005-01-12 03:41:21 +0000 | [diff] [blame] | 99 | #endif |
| 100 | |
Jim Laskey | eb577ba | 2006-08-02 12:30:23 +0000 | [diff] [blame] | 101 | //===---------------------------------------------------------------------===// |
| 102 | /// |
| 103 | /// RegisterScheduler class - Track the registration of instruction schedulers. |
| 104 | /// |
| 105 | //===---------------------------------------------------------------------===// |
| 106 | MachinePassRegistry RegisterScheduler::Registry; |
| 107 | |
| 108 | //===---------------------------------------------------------------------===// |
| 109 | /// |
| 110 | /// ISHeuristic command line option for instruction schedulers. |
| 111 | /// |
| 112 | //===---------------------------------------------------------------------===// |
Dan Gohman | 844731a | 2008-05-13 00:00:25 +0000 | [diff] [blame] | 113 | static cl::opt<RegisterScheduler::FunctionPassCtor, false, |
| 114 | RegisterPassParser<RegisterScheduler> > |
| 115 | ISHeuristic("pre-RA-sched", |
| 116 | cl::init(&createDefaultScheduler), |
| 117 | cl::desc("Instruction schedulers available (before register" |
| 118 | " allocation):")); |
Jim Laskey | 13ec702 | 2006-08-01 14:21:23 +0000 | [diff] [blame] | 119 | |
Dan Gohman | 844731a | 2008-05-13 00:00:25 +0000 | [diff] [blame] | 120 | static RegisterScheduler |
Dan Gohman | b8cab92 | 2008-10-14 20:25:08 +0000 | [diff] [blame] | 121 | defaultListDAGScheduler("default", "Best scheduler for the target", |
Dan Gohman | 844731a | 2008-05-13 00:00:25 +0000 | [diff] [blame] | 122 | createDefaultScheduler); |
Evan Cheng | 4ef1086 | 2006-01-23 07:01:07 +0000 | [diff] [blame] | 123 | |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 124 | namespace llvm { |
| 125 | //===--------------------------------------------------------------------===// |
Jim Laskey | 9373beb | 2006-08-01 19:14:14 +0000 | [diff] [blame] | 126 | /// createDefaultScheduler - This creates an instruction scheduler appropriate |
| 127 | /// for the target. |
Dan Gohman | 47ac0f0 | 2009-02-11 04:27:20 +0000 | [diff] [blame] | 128 | ScheduleDAGSDNodes* createDefaultScheduler(SelectionDAGISel *IS, |
Bill Wendling | 98a366d | 2009-04-29 23:29:43 +0000 | [diff] [blame] | 129 | CodeGenOpt::Level OptLevel) { |
Dan Gohman | e9530ec | 2009-01-15 16:58:17 +0000 | [diff] [blame] | 130 | const TargetLowering &TLI = IS->getTargetLowering(); |
| 131 | |
Bill Wendling | 98a366d | 2009-04-29 23:29:43 +0000 | [diff] [blame] | 132 | if (OptLevel == CodeGenOpt::None) |
Bill Wendling | be8cc2a | 2009-04-29 00:15:41 +0000 | [diff] [blame] | 133 | return createFastDAGScheduler(IS, OptLevel); |
Dan Gohman | 9e76fea | 2008-11-20 03:11:19 +0000 | [diff] [blame] | 134 | if (TLI.getSchedulingPreference() == TargetLowering::SchedulingForLatency) |
Bill Wendling | be8cc2a | 2009-04-29 00:15:41 +0000 | [diff] [blame] | 135 | return createTDListDAGScheduler(IS, OptLevel); |
Dan Gohman | 9e76fea | 2008-11-20 03:11:19 +0000 | [diff] [blame] | 136 | assert(TLI.getSchedulingPreference() == |
Bill Wendling | 187361b | 2010-01-23 10:26:57 +0000 | [diff] [blame] | 137 | TargetLowering::SchedulingForRegPressure && "Unknown sched type!"); |
Bill Wendling | be8cc2a | 2009-04-29 00:15:41 +0000 | [diff] [blame] | 138 | return createBURRListDAGScheduler(IS, OptLevel); |
Jim Laskey | 9373beb | 2006-08-01 19:14:14 +0000 | [diff] [blame] | 139 | } |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 140 | } |
| 141 | |
Evan Cheng | ff9b373 | 2008-01-30 18:18:23 +0000 | [diff] [blame] | 142 | // EmitInstrWithCustomInserter - This method should be implemented by targets |
Dan Gohman | 533297b | 2009-10-29 18:10:34 +0000 | [diff] [blame] | 143 | // that mark instructions with the 'usesCustomInserter' flag. These |
Chris Lattner | 025c39b | 2005-08-26 20:54:47 +0000 | [diff] [blame] | 144 | // instructions are special in various ways, which require special support to |
| 145 | // insert. The specified MachineInstr is created but not inserted into any |
Dan Gohman | 533297b | 2009-10-29 18:10:34 +0000 | [diff] [blame] | 146 | // basic blocks, and this method is called to expand it into a sequence of |
| 147 | // instructions, potentially also creating new basic blocks and control flow. |
| 148 | // When new basic blocks are inserted and the edges from MBB to its successors |
| 149 | // are modified, the method should insert pairs of <OldSucc, NewSucc> into the |
| 150 | // DenseMap. |
Evan Cheng | ff9b373 | 2008-01-30 18:18:23 +0000 | [diff] [blame] | 151 | MachineBasicBlock *TargetLowering::EmitInstrWithCustomInserter(MachineInstr *MI, |
Evan Cheng | fb2e752 | 2009-09-18 21:02:19 +0000 | [diff] [blame] | 152 | MachineBasicBlock *MBB, |
| 153 | DenseMap<MachineBasicBlock*, MachineBasicBlock*> *EM) const { |
Torok Edwin | f368923 | 2009-07-12 20:07:01 +0000 | [diff] [blame] | 154 | #ifndef NDEBUG |
David Greene | 1a05323 | 2010-01-05 01:26:11 +0000 | [diff] [blame] | 155 | dbgs() << "If a target marks an instruction with " |
Dan Gohman | 533297b | 2009-10-29 18:10:34 +0000 | [diff] [blame] | 156 | "'usesCustomInserter', it must implement " |
Torok Edwin | f368923 | 2009-07-12 20:07:01 +0000 | [diff] [blame] | 157 | "TargetLowering::EmitInstrWithCustomInserter!"; |
| 158 | #endif |
Torok Edwin | c23197a | 2009-07-14 16:55:14 +0000 | [diff] [blame] | 159 | llvm_unreachable(0); |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 160 | return 0; |
Chris Lattner | 025c39b | 2005-08-26 20:54:47 +0000 | [diff] [blame] | 161 | } |
| 162 | |
Chris Lattner | 7041ee3 | 2005-01-11 05:56:49 +0000 | [diff] [blame] | 163 | //===----------------------------------------------------------------------===// |
| 164 | // SelectionDAGISel code |
| 165 | //===----------------------------------------------------------------------===// |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 166 | |
Bill Wendling | 98a366d | 2009-04-29 23:29:43 +0000 | [diff] [blame] | 167 | SelectionDAGISel::SelectionDAGISel(TargetMachine &tm, CodeGenOpt::Level OL) : |
Dan Gohman | ad2afc2 | 2009-07-31 18:16:33 +0000 | [diff] [blame] | 168 | MachineFunctionPass(&ID), TM(tm), TLI(*tm.getTargetLowering()), |
Dan Gohman | 7c3234c | 2008-08-27 23:52:12 +0000 | [diff] [blame] | 169 | FuncInfo(new FunctionLoweringInfo(TLI)), |
| 170 | CurDAG(new SelectionDAG(TLI, *FuncInfo)), |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 171 | SDB(new SelectionDAGBuilder(*CurDAG, TLI, *FuncInfo, OL)), |
Dan Gohman | 7c3234c | 2008-08-27 23:52:12 +0000 | [diff] [blame] | 172 | GFI(), |
Bill Wendling | be8cc2a | 2009-04-29 00:15:41 +0000 | [diff] [blame] | 173 | OptLevel(OL), |
Dan Gohman | 7c3234c | 2008-08-27 23:52:12 +0000 | [diff] [blame] | 174 | DAGSize(0) |
| 175 | {} |
| 176 | |
| 177 | SelectionDAGISel::~SelectionDAGISel() { |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 178 | delete SDB; |
Dan Gohman | 7c3234c | 2008-08-27 23:52:12 +0000 | [diff] [blame] | 179 | delete CurDAG; |
| 180 | delete FuncInfo; |
| 181 | } |
| 182 | |
Owen Anderson | e50ed30 | 2009-08-10 22:56:29 +0000 | [diff] [blame] | 183 | unsigned SelectionDAGISel::MakeReg(EVT VT) { |
Chris Lattner | 84bc542 | 2007-12-31 04:13:23 +0000 | [diff] [blame] | 184 | return RegInfo->createVirtualRegister(TLI.getRegClassFor(VT)); |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 185 | } |
| 186 | |
Chris Lattner | 495a0b5 | 2005-08-17 06:37:43 +0000 | [diff] [blame] | 187 | void SelectionDAGISel::getAnalysisUsage(AnalysisUsage &AU) const { |
Jim Laskey | c7c3f11 | 2006-10-16 20:52:31 +0000 | [diff] [blame] | 188 | AU.addRequired<AliasAnalysis>(); |
Dan Gohman | a3477fe | 2009-07-31 23:36:22 +0000 | [diff] [blame] | 189 | AU.addPreserved<AliasAnalysis>(); |
Gordon Henriksen | 5eca075 | 2008-08-17 18:44:35 +0000 | [diff] [blame] | 190 | AU.addRequired<GCModuleInfo>(); |
Dan Gohman | a3477fe | 2009-07-31 23:36:22 +0000 | [diff] [blame] | 191 | AU.addPreserved<GCModuleInfo>(); |
Dan Gohman | ad2afc2 | 2009-07-31 18:16:33 +0000 | [diff] [blame] | 192 | MachineFunctionPass::getAnalysisUsage(AU); |
Chris Lattner | 495a0b5 | 2005-08-17 06:37:43 +0000 | [diff] [blame] | 193 | } |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 194 | |
Dan Gohman | ad2afc2 | 2009-07-31 18:16:33 +0000 | [diff] [blame] | 195 | bool SelectionDAGISel::runOnMachineFunction(MachineFunction &mf) { |
| 196 | Function &Fn = *mf.getFunction(); |
| 197 | |
Dan Gohman | 4344a5d | 2008-09-09 23:05:00 +0000 | [diff] [blame] | 198 | // Do some sanity-checking on the command-line options. |
| 199 | assert((!EnableFastISelVerbose || EnableFastISel) && |
| 200 | "-fast-isel-verbose requires -fast-isel"); |
| 201 | assert((!EnableFastISelAbort || EnableFastISel) && |
| 202 | "-fast-isel-abort requires -fast-isel"); |
| 203 | |
Dan Gohman | 5f43f92 | 2007-08-27 16:26:13 +0000 | [diff] [blame] | 204 | // Get alias analysis for load/store combining. |
| 205 | AA = &getAnalysis<AliasAnalysis>(); |
| 206 | |
Dan Gohman | ad2afc2 | 2009-07-31 18:16:33 +0000 | [diff] [blame] | 207 | MF = &mf; |
Dan Gohman | 8a11053 | 2008-09-05 22:59:21 +0000 | [diff] [blame] | 208 | const TargetInstrInfo &TII = *TM.getInstrInfo(); |
| 209 | const TargetRegisterInfo &TRI = *TM.getRegisterInfo(); |
| 210 | |
Dan Gohman | f7d6cd4 | 2009-08-01 03:51:09 +0000 | [diff] [blame] | 211 | if (Fn.hasGC()) |
| 212 | GFI = &getAnalysis<GCModuleInfo>().getFunctionInfo(Fn); |
Gordon Henriksen | ce22477 | 2008-01-07 01:30:38 +0000 | [diff] [blame] | 213 | else |
Gordon Henriksen | 5eca075 | 2008-08-17 18:44:35 +0000 | [diff] [blame] | 214 | GFI = 0; |
Dan Gohman | 79ce276 | 2009-01-15 19:20:50 +0000 | [diff] [blame] | 215 | RegInfo = &MF->getRegInfo(); |
David Greene | 1a05323 | 2010-01-05 01:26:11 +0000 | [diff] [blame] | 216 | DEBUG(dbgs() << "\n\n\n=== " << Fn.getName() << "\n"); |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 217 | |
Chris Lattner | de6e783 | 2010-04-05 06:10:13 +0000 | [diff] [blame] | 218 | CurDAG->init(*MF); |
Dan Gohman | 6277eb2 | 2009-11-23 17:16:22 +0000 | [diff] [blame] | 219 | FuncInfo->set(Fn, *MF, EnableFastISel); |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 220 | SDB->init(GFI, *AA); |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 221 | |
Chris Lattner | de6e783 | 2010-04-05 06:10:13 +0000 | [diff] [blame] | 222 | SelectAllBasicBlocks(Fn, *MF, TII); |
Misha Brukman | edf128a | 2005-04-21 22:36:52 +0000 | [diff] [blame] | 223 | |
Dan Gohman | 8a11053 | 2008-09-05 22:59:21 +0000 | [diff] [blame] | 224 | // If the first basic block in the function has live ins that need to be |
| 225 | // copied into vregs, emit the copies into the top of the block before |
| 226 | // emitting the code for the block. |
Dan Gohman | 9870826 | 2010-04-14 16:51:49 +0000 | [diff] [blame] | 227 | RegInfo->EmitLiveInCopies(MF->begin(), TRI, TII); |
Dan Gohman | 8a11053 | 2008-09-05 22:59:21 +0000 | [diff] [blame] | 228 | |
Evan Cheng | ad2070c | 2007-02-10 02:43:39 +0000 | [diff] [blame] | 229 | // Add function live-ins to entry block live-in set. |
Dan Gohman | 8a11053 | 2008-09-05 22:59:21 +0000 | [diff] [blame] | 230 | for (MachineRegisterInfo::livein_iterator I = RegInfo->livein_begin(), |
| 231 | E = RegInfo->livein_end(); I != E; ++I) |
Dan Gohman | 79ce276 | 2009-01-15 19:20:50 +0000 | [diff] [blame] | 232 | MF->begin()->addLiveIn(I->first); |
Evan Cheng | ad2070c | 2007-02-10 02:43:39 +0000 | [diff] [blame] | 233 | |
Duncan Sands | f407082 | 2007-06-15 19:04:19 +0000 | [diff] [blame] | 234 | #ifndef NDEBUG |
Dan Gohman | 7c3234c | 2008-08-27 23:52:12 +0000 | [diff] [blame] | 235 | assert(FuncInfo->CatchInfoFound.size() == FuncInfo->CatchInfoLost.size() && |
Duncan Sands | f407082 | 2007-06-15 19:04:19 +0000 | [diff] [blame] | 236 | "Not all catch info was assigned to a landing pad!"); |
| 237 | #endif |
| 238 | |
Dan Gohman | 7c3234c | 2008-08-27 23:52:12 +0000 | [diff] [blame] | 239 | FuncInfo->clear(); |
| 240 | |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 241 | return true; |
| 242 | } |
| 243 | |
Dan Gohman | 07f111e | 2009-12-05 00:27:08 +0000 | [diff] [blame] | 244 | /// SetDebugLoc - Update MF's and SDB's DebugLocs if debug information is |
| 245 | /// attached with this instruction. |
Chris Lattner | 23dbf66 | 2010-03-31 04:24:50 +0000 | [diff] [blame] | 246 | static void SetDebugLoc(Instruction *I, SelectionDAGBuilder *SDB, |
Chris Lattner | 0eb4198 | 2009-12-28 20:45:51 +0000 | [diff] [blame] | 247 | FastISel *FastIS, MachineFunction *MF) { |
Chris Lattner | de4845c | 2010-04-02 19:42:39 +0000 | [diff] [blame] | 248 | DebugLoc DL = I->getDebugLoc(); |
| 249 | if (DL.isUnknown()) return; |
Chris Lattner | 23dbf66 | 2010-03-31 04:24:50 +0000 | [diff] [blame] | 250 | |
Chris Lattner | de4845c | 2010-04-02 19:42:39 +0000 | [diff] [blame] | 251 | SDB->setCurDebugLoc(DL); |
Dan Gohman | 07f111e | 2009-12-05 00:27:08 +0000 | [diff] [blame] | 252 | |
Chris Lattner | 23dbf66 | 2010-03-31 04:24:50 +0000 | [diff] [blame] | 253 | if (FastIS) |
Chris Lattner | de4845c | 2010-04-02 19:42:39 +0000 | [diff] [blame] | 254 | FastIS->setCurDebugLoc(DL); |
Dan Gohman | 07f111e | 2009-12-05 00:27:08 +0000 | [diff] [blame] | 255 | |
Chris Lattner | 23dbf66 | 2010-03-31 04:24:50 +0000 | [diff] [blame] | 256 | // If the function doesn't have a default debug location yet, set |
| 257 | // it. This is kind of a hack. |
| 258 | if (MF->getDefaultDebugLoc().isUnknown()) |
Chris Lattner | de4845c | 2010-04-02 19:42:39 +0000 | [diff] [blame] | 259 | MF->setDefaultDebugLoc(DL); |
Dan Gohman | 07f111e | 2009-12-05 00:27:08 +0000 | [diff] [blame] | 260 | } |
| 261 | |
| 262 | /// ResetDebugLoc - Set MF's and SDB's DebugLocs to Unknown. |
Chris Lattner | 3990b12 | 2009-12-28 23:41:32 +0000 | [diff] [blame] | 263 | static void ResetDebugLoc(SelectionDAGBuilder *SDB, FastISel *FastIS) { |
Chris Lattner | a4f2bb0 | 2010-04-02 20:17:23 +0000 | [diff] [blame] | 264 | SDB->setCurDebugLoc(DebugLoc()); |
Dan Gohman | 07f111e | 2009-12-05 00:27:08 +0000 | [diff] [blame] | 265 | if (FastIS) |
Chris Lattner | a4f2bb0 | 2010-04-02 20:17:23 +0000 | [diff] [blame] | 266 | FastIS->setCurDebugLoc(DebugLoc()); |
Dan Gohman | 07f111e | 2009-12-05 00:27:08 +0000 | [diff] [blame] | 267 | } |
| 268 | |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 269 | void SelectionDAGISel::SelectBasicBlock(BasicBlock *LLVMBB, |
| 270 | BasicBlock::iterator Begin, |
Dan Gohman | b4afb13 | 2009-11-20 02:51:26 +0000 | [diff] [blame] | 271 | BasicBlock::iterator End, |
| 272 | bool &HadTailCall) { |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 273 | SDB->setCurrentBasicBlock(BB); |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 274 | |
Dan Gohman | 98ca4f2 | 2009-08-05 01:29:28 +0000 | [diff] [blame] | 275 | // Lower all of the non-terminator instructions. If a call is emitted |
| 276 | // as a tail call, cease emitting nodes for this block. |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 277 | for (BasicBlock::iterator I = Begin; I != End && !SDB->HasTailCall; ++I) { |
Chris Lattner | 23dbf66 | 2010-03-31 04:24:50 +0000 | [diff] [blame] | 278 | SetDebugLoc(I, SDB, 0, MF); |
Dan Gohman | 07f111e | 2009-12-05 00:27:08 +0000 | [diff] [blame] | 279 | |
| 280 | if (!isa<TerminatorInst>(I)) { |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 281 | SDB->visit(*I); |
Dan Gohman | 07f111e | 2009-12-05 00:27:08 +0000 | [diff] [blame] | 282 | |
| 283 | // Set the current debug location back to "unknown" so that it doesn't |
| 284 | // spuriously apply to subsequent instructions. |
| 285 | ResetDebugLoc(SDB, 0); |
| 286 | } |
Devang Patel | 123eaa7 | 2009-09-16 20:39:11 +0000 | [diff] [blame] | 287 | } |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 288 | |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 289 | if (!SDB->HasTailCall) { |
Dan Gohman | 98ca4f2 | 2009-08-05 01:29:28 +0000 | [diff] [blame] | 290 | // Ensure that all instructions which are used outside of their defining |
| 291 | // blocks are available as virtual registers. Invoke is handled elsewhere. |
| 292 | for (BasicBlock::iterator I = Begin; I != End; ++I) |
| 293 | if (!isa<PHINode>(I) && !isa<InvokeInst>(I)) |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 294 | SDB->CopyToExportRegsIfNeeded(I); |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 295 | |
Dan Gohman | 98ca4f2 | 2009-08-05 01:29:28 +0000 | [diff] [blame] | 296 | // Handle PHI nodes in successor blocks. |
| 297 | if (End == LLVMBB->end()) { |
| 298 | HandlePHINodesInSuccessorBlocks(LLVMBB); |
Dan Gohman | 3df24e6 | 2008-09-03 23:12:08 +0000 | [diff] [blame] | 299 | |
Dan Gohman | 98ca4f2 | 2009-08-05 01:29:28 +0000 | [diff] [blame] | 300 | // Lower the terminator after the copies are emitted. |
Chris Lattner | 23dbf66 | 2010-03-31 04:24:50 +0000 | [diff] [blame] | 301 | SetDebugLoc(LLVMBB->getTerminator(), SDB, 0, MF); |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 302 | SDB->visit(*LLVMBB->getTerminator()); |
Dan Gohman | 07f111e | 2009-12-05 00:27:08 +0000 | [diff] [blame] | 303 | ResetDebugLoc(SDB, 0); |
Dan Gohman | 98ca4f2 | 2009-08-05 01:29:28 +0000 | [diff] [blame] | 304 | } |
Dan Gohman | 3df24e6 | 2008-09-03 23:12:08 +0000 | [diff] [blame] | 305 | } |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 306 | |
Chris Lattner | a651cf6 | 2005-01-17 19:43:36 +0000 | [diff] [blame] | 307 | // Make sure the root of the DAG is up-to-date. |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 308 | CurDAG->setRoot(SDB->getControlRoot()); |
Arnold Schwaighofer | c85e171 | 2007-10-11 19:40:01 +0000 | [diff] [blame] | 309 | |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 310 | // Final step, emit the lowered DAG as machine code. |
| 311 | CodeGenAndEmitDAG(); |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 312 | HadTailCall = SDB->HasTailCall; |
| 313 | SDB->clear(); |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 314 | } |
| 315 | |
Evan Cheng | 54e146b | 2010-01-09 00:21:08 +0000 | [diff] [blame] | 316 | namespace { |
| 317 | /// WorkListRemover - This class is a DAGUpdateListener that removes any deleted |
| 318 | /// nodes from the worklist. |
| 319 | class SDOPsWorkListRemover : public SelectionDAG::DAGUpdateListener { |
| 320 | SmallVector<SDNode*, 128> &Worklist; |
Chris Lattner | 25e0ab9 | 2010-03-14 19:43:04 +0000 | [diff] [blame] | 321 | SmallPtrSet<SDNode*, 128> &InWorklist; |
Evan Cheng | 54e146b | 2010-01-09 00:21:08 +0000 | [diff] [blame] | 322 | public: |
Chris Lattner | 25e0ab9 | 2010-03-14 19:43:04 +0000 | [diff] [blame] | 323 | SDOPsWorkListRemover(SmallVector<SDNode*, 128> &wl, |
| 324 | SmallPtrSet<SDNode*, 128> &inwl) |
| 325 | : Worklist(wl), InWorklist(inwl) {} |
Evan Cheng | 54e146b | 2010-01-09 00:21:08 +0000 | [diff] [blame] | 326 | |
Chris Lattner | 25e0ab9 | 2010-03-14 19:43:04 +0000 | [diff] [blame] | 327 | void RemoveFromWorklist(SDNode *N) { |
| 328 | if (!InWorklist.erase(N)) return; |
| 329 | |
| 330 | SmallVector<SDNode*, 128>::iterator I = |
| 331 | std::find(Worklist.begin(), Worklist.end(), N); |
| 332 | assert(I != Worklist.end() && "Not in worklist"); |
| 333 | |
| 334 | *I = Worklist.back(); |
| 335 | Worklist.pop_back(); |
| 336 | } |
| 337 | |
Evan Cheng | 54e146b | 2010-01-09 00:21:08 +0000 | [diff] [blame] | 338 | virtual void NodeDeleted(SDNode *N, SDNode *E) { |
Chris Lattner | 25e0ab9 | 2010-03-14 19:43:04 +0000 | [diff] [blame] | 339 | RemoveFromWorklist(N); |
Evan Cheng | 54e146b | 2010-01-09 00:21:08 +0000 | [diff] [blame] | 340 | } |
| 341 | |
| 342 | virtual void NodeUpdated(SDNode *N) { |
| 343 | // Ignore updates. |
| 344 | } |
| 345 | }; |
| 346 | } |
| 347 | |
Evan Cheng | 046632f | 2010-02-10 02:17:34 +0000 | [diff] [blame] | 348 | /// TrivialTruncElim - Eliminate some trivial nops that can result from |
| 349 | /// ShrinkDemandedOps: (trunc (ext n)) -> n. |
| 350 | static bool TrivialTruncElim(SDValue Op, |
| 351 | TargetLowering::TargetLoweringOpt &TLO) { |
| 352 | SDValue N0 = Op.getOperand(0); |
| 353 | EVT VT = Op.getValueType(); |
| 354 | if ((N0.getOpcode() == ISD::ZERO_EXTEND || |
| 355 | N0.getOpcode() == ISD::SIGN_EXTEND || |
| 356 | N0.getOpcode() == ISD::ANY_EXTEND) && |
| 357 | N0.getOperand(0).getValueType() == VT) { |
| 358 | return TLO.CombineTo(Op, N0.getOperand(0)); |
| 359 | } |
| 360 | return false; |
| 361 | } |
| 362 | |
Evan Cheng | 54eb4c2 | 2010-01-06 19:43:21 +0000 | [diff] [blame] | 363 | /// ShrinkDemandedOps - A late transformation pass that shrink expressions |
| 364 | /// using TargetLowering::TargetLoweringOpt::ShrinkDemandedOp. It converts |
| 365 | /// x+y to (VT)((SmallVT)x+(SmallVT)y) if the casts are free. |
Evan Cheng | d40d03e | 2010-01-06 19:38:29 +0000 | [diff] [blame] | 366 | void SelectionDAGISel::ShrinkDemandedOps() { |
| 367 | SmallVector<SDNode*, 128> Worklist; |
Chris Lattner | 25e0ab9 | 2010-03-14 19:43:04 +0000 | [diff] [blame] | 368 | SmallPtrSet<SDNode*, 128> InWorklist; |
Evan Cheng | d40d03e | 2010-01-06 19:38:29 +0000 | [diff] [blame] | 369 | |
| 370 | // Add all the dag nodes to the worklist. |
| 371 | Worklist.reserve(CurDAG->allnodes_size()); |
| 372 | for (SelectionDAG::allnodes_iterator I = CurDAG->allnodes_begin(), |
Chris Lattner | 25e0ab9 | 2010-03-14 19:43:04 +0000 | [diff] [blame] | 373 | E = CurDAG->allnodes_end(); I != E; ++I) { |
Evan Cheng | d40d03e | 2010-01-06 19:38:29 +0000 | [diff] [blame] | 374 | Worklist.push_back(I); |
Chris Lattner | 25e0ab9 | 2010-03-14 19:43:04 +0000 | [diff] [blame] | 375 | InWorklist.insert(I); |
| 376 | } |
Evan Cheng | d40d03e | 2010-01-06 19:38:29 +0000 | [diff] [blame] | 377 | |
| 378 | TargetLowering::TargetLoweringOpt TLO(*CurDAG, true); |
| 379 | while (!Worklist.empty()) { |
Benjamin Kramer | 7b1e2a5 | 2010-01-07 17:27:56 +0000 | [diff] [blame] | 380 | SDNode *N = Worklist.pop_back_val(); |
Chris Lattner | 25e0ab9 | 2010-03-14 19:43:04 +0000 | [diff] [blame] | 381 | InWorklist.erase(N); |
Evan Cheng | d40d03e | 2010-01-06 19:38:29 +0000 | [diff] [blame] | 382 | |
| 383 | if (N->use_empty() && N != CurDAG->getRoot().getNode()) { |
Chris Lattner | c4a3f23 | 2010-03-14 19:46:02 +0000 | [diff] [blame] | 384 | // Deleting this node may make its operands dead, add them to the worklist |
| 385 | // if they aren't already there. |
| 386 | for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i) |
| 387 | if (InWorklist.insert(N->getOperand(i).getNode())) |
| 388 | Worklist.push_back(N->getOperand(i).getNode()); |
| 389 | |
Evan Cheng | 54e146b | 2010-01-09 00:21:08 +0000 | [diff] [blame] | 390 | CurDAG->DeleteNode(N); |
Evan Cheng | d40d03e | 2010-01-06 19:38:29 +0000 | [diff] [blame] | 391 | continue; |
| 392 | } |
| 393 | |
| 394 | // Run ShrinkDemandedOp on scalar binary operations. |
Chris Lattner | 25e0ab9 | 2010-03-14 19:43:04 +0000 | [diff] [blame] | 395 | if (N->getNumValues() != 1 || |
| 396 | !N->getValueType(0).isSimple() || !N->getValueType(0).isInteger()) |
| 397 | continue; |
| 398 | |
| 399 | unsigned BitWidth = N->getValueType(0).getScalarType().getSizeInBits(); |
| 400 | APInt Demanded = APInt::getAllOnesValue(BitWidth); |
| 401 | APInt KnownZero, KnownOne; |
| 402 | if (!TLI.SimplifyDemandedBits(SDValue(N, 0), Demanded, |
| 403 | KnownZero, KnownOne, TLO) && |
| 404 | (N->getOpcode() != ISD::TRUNCATE || |
| 405 | !TrivialTruncElim(SDValue(N, 0), TLO))) |
| 406 | continue; |
| 407 | |
| 408 | // Revisit the node. |
| 409 | assert(!InWorklist.count(N) && "Already in worklist"); |
| 410 | Worklist.push_back(N); |
| 411 | InWorklist.insert(N); |
Evan Cheng | d40d03e | 2010-01-06 19:38:29 +0000 | [diff] [blame] | 412 | |
Chris Lattner | 25e0ab9 | 2010-03-14 19:43:04 +0000 | [diff] [blame] | 413 | // Replace the old value with the new one. |
| 414 | DEBUG(errs() << "\nShrinkDemandedOps replacing "; |
| 415 | TLO.Old.getNode()->dump(CurDAG); |
| 416 | errs() << "\nWith: "; |
| 417 | TLO.New.getNode()->dump(CurDAG); |
| 418 | errs() << '\n'); |
Evan Cheng | d40d03e | 2010-01-06 19:38:29 +0000 | [diff] [blame] | 419 | |
Chris Lattner | 25e0ab9 | 2010-03-14 19:43:04 +0000 | [diff] [blame] | 420 | if (InWorklist.insert(TLO.New.getNode())) |
| 421 | Worklist.push_back(TLO.New.getNode()); |
Evan Cheng | 54e146b | 2010-01-09 00:21:08 +0000 | [diff] [blame] | 422 | |
Chris Lattner | 25e0ab9 | 2010-03-14 19:43:04 +0000 | [diff] [blame] | 423 | SDOPsWorkListRemover DeadNodes(Worklist, InWorklist); |
| 424 | CurDAG->ReplaceAllUsesOfValueWith(TLO.Old, TLO.New, &DeadNodes); |
Evan Cheng | d40d03e | 2010-01-06 19:38:29 +0000 | [diff] [blame] | 425 | |
Chris Lattner | 25e0ab9 | 2010-03-14 19:43:04 +0000 | [diff] [blame] | 426 | if (!TLO.Old.getNode()->use_empty()) continue; |
| 427 | |
| 428 | for (unsigned i = 0, e = TLO.Old.getNode()->getNumOperands(); |
| 429 | i != e; ++i) { |
| 430 | SDNode *OpNode = TLO.Old.getNode()->getOperand(i).getNode(); |
| 431 | if (OpNode->hasOneUse()) { |
| 432 | // Add OpNode to the end of the list to revisit. |
| 433 | DeadNodes.RemoveFromWorklist(OpNode); |
| 434 | Worklist.push_back(OpNode); |
| 435 | InWorklist.insert(OpNode); |
Evan Cheng | d40d03e | 2010-01-06 19:38:29 +0000 | [diff] [blame] | 436 | } |
| 437 | } |
Chris Lattner | 25e0ab9 | 2010-03-14 19:43:04 +0000 | [diff] [blame] | 438 | |
| 439 | DeadNodes.RemoveFromWorklist(TLO.Old.getNode()); |
| 440 | CurDAG->DeleteNode(TLO.Old.getNode()); |
Evan Cheng | d40d03e | 2010-01-06 19:38:29 +0000 | [diff] [blame] | 441 | } |
| 442 | } |
| 443 | |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 444 | void SelectionDAGISel::ComputeLiveOutVRegInfo() { |
Chris Lattner | ead0d88 | 2008-06-17 06:09:18 +0000 | [diff] [blame] | 445 | SmallPtrSet<SDNode*, 128> VisitedNodes; |
| 446 | SmallVector<SDNode*, 128> Worklist; |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 447 | |
Gabor Greif | ba36cb5 | 2008-08-28 21:40:38 +0000 | [diff] [blame] | 448 | Worklist.push_back(CurDAG->getRoot().getNode()); |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 449 | |
Chris Lattner | ead0d88 | 2008-06-17 06:09:18 +0000 | [diff] [blame] | 450 | APInt Mask; |
| 451 | APInt KnownZero; |
| 452 | APInt KnownOne; |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 453 | |
Benjamin Kramer | 7b1e2a5 | 2010-01-07 17:27:56 +0000 | [diff] [blame] | 454 | do { |
| 455 | SDNode *N = Worklist.pop_back_val(); |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 456 | |
Chris Lattner | ead0d88 | 2008-06-17 06:09:18 +0000 | [diff] [blame] | 457 | // If we've already seen this node, ignore it. |
| 458 | if (!VisitedNodes.insert(N)) |
| 459 | continue; |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 460 | |
Chris Lattner | ead0d88 | 2008-06-17 06:09:18 +0000 | [diff] [blame] | 461 | // Otherwise, add all chain operands to the worklist. |
| 462 | for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i) |
Owen Anderson | 825b72b | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 463 | if (N->getOperand(i).getValueType() == MVT::Other) |
Gabor Greif | ba36cb5 | 2008-08-28 21:40:38 +0000 | [diff] [blame] | 464 | Worklist.push_back(N->getOperand(i).getNode()); |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 465 | |
Chris Lattner | ead0d88 | 2008-06-17 06:09:18 +0000 | [diff] [blame] | 466 | // If this is a CopyToReg with a vreg dest, process it. |
| 467 | if (N->getOpcode() != ISD::CopyToReg) |
| 468 | continue; |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 469 | |
Chris Lattner | ead0d88 | 2008-06-17 06:09:18 +0000 | [diff] [blame] | 470 | unsigned DestReg = cast<RegisterSDNode>(N->getOperand(1))->getReg(); |
| 471 | if (!TargetRegisterInfo::isVirtualRegister(DestReg)) |
| 472 | continue; |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 473 | |
Chris Lattner | ead0d88 | 2008-06-17 06:09:18 +0000 | [diff] [blame] | 474 | // Ignore non-scalar or non-integer values. |
Dan Gohman | 475871a | 2008-07-27 21:46:04 +0000 | [diff] [blame] | 475 | SDValue Src = N->getOperand(2); |
Owen Anderson | e50ed30 | 2009-08-10 22:56:29 +0000 | [diff] [blame] | 476 | EVT SrcVT = Src.getValueType(); |
Chris Lattner | ead0d88 | 2008-06-17 06:09:18 +0000 | [diff] [blame] | 477 | if (!SrcVT.isInteger() || SrcVT.isVector()) |
| 478 | continue; |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 479 | |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 480 | unsigned NumSignBits = CurDAG->ComputeNumSignBits(Src); |
Chris Lattner | ead0d88 | 2008-06-17 06:09:18 +0000 | [diff] [blame] | 481 | Mask = APInt::getAllOnesValue(SrcVT.getSizeInBits()); |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 482 | CurDAG->ComputeMaskedBits(Src, Mask, KnownZero, KnownOne); |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 483 | |
Chris Lattner | ead0d88 | 2008-06-17 06:09:18 +0000 | [diff] [blame] | 484 | // Only install this information if it tells us something. |
| 485 | if (NumSignBits != 1 || KnownZero != 0 || KnownOne != 0) { |
| 486 | DestReg -= TargetRegisterInfo::FirstVirtualRegister; |
Dan Gohman | f7d6cd4 | 2009-08-01 03:51:09 +0000 | [diff] [blame] | 487 | if (DestReg >= FuncInfo->LiveOutRegInfo.size()) |
| 488 | FuncInfo->LiveOutRegInfo.resize(DestReg+1); |
| 489 | FunctionLoweringInfo::LiveOutInfo &LOI = |
| 490 | FuncInfo->LiveOutRegInfo[DestReg]; |
Chris Lattner | ead0d88 | 2008-06-17 06:09:18 +0000 | [diff] [blame] | 491 | LOI.NumSignBits = NumSignBits; |
Dan Gohman | a80efce | 2009-03-27 23:55:04 +0000 | [diff] [blame] | 492 | LOI.KnownOne = KnownOne; |
| 493 | LOI.KnownZero = KnownZero; |
Chris Lattner | ead0d88 | 2008-06-17 06:09:18 +0000 | [diff] [blame] | 494 | } |
Benjamin Kramer | 7b1e2a5 | 2010-01-07 17:27:56 +0000 | [diff] [blame] | 495 | } while (!Worklist.empty()); |
Chris Lattner | ead0d88 | 2008-06-17 06:09:18 +0000 | [diff] [blame] | 496 | } |
| 497 | |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 498 | void SelectionDAGISel::CodeGenAndEmitDAG() { |
Dan Gohman | 462dc7f | 2008-07-21 20:00:07 +0000 | [diff] [blame] | 499 | std::string GroupName; |
| 500 | if (TimePassesIsEnabled) |
| 501 | GroupName = "Instruction Selection and Scheduling"; |
| 502 | std::string BlockName; |
| 503 | if (ViewDAGCombine1 || ViewLegalizeTypesDAGs || ViewLegalizeDAGs || |
Duncan Sands | 25cf227 | 2008-11-24 14:53:14 +0000 | [diff] [blame] | 504 | ViewDAGCombine2 || ViewDAGCombineLT || ViewISelDAGs || ViewSchedDAGs || |
| 505 | ViewSUnitDAGs) |
Dan Gohman | f7d6cd4 | 2009-08-01 03:51:09 +0000 | [diff] [blame] | 506 | BlockName = MF->getFunction()->getNameStr() + ":" + |
Daniel Dunbar | f6ccee5 | 2009-07-24 08:24:36 +0000 | [diff] [blame] | 507 | BB->getBasicBlock()->getNameStr(); |
Dan Gohman | 462dc7f | 2008-07-21 20:00:07 +0000 | [diff] [blame] | 508 | |
David Greene | 1a05323 | 2010-01-05 01:26:11 +0000 | [diff] [blame] | 509 | DEBUG(dbgs() << "Initial selection DAG:\n"); |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 510 | DEBUG(CurDAG->dump()); |
Dan Gohman | 462dc7f | 2008-07-21 20:00:07 +0000 | [diff] [blame] | 511 | |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 512 | if (ViewDAGCombine1) CurDAG->viewGraph("dag-combine1 input for " + BlockName); |
Dan Gohman | 417e11b | 2007-10-08 15:12:17 +0000 | [diff] [blame] | 513 | |
Chris Lattner | af21d55 | 2005-10-10 16:47:10 +0000 | [diff] [blame] | 514 | // Run the DAG combiner in pre-legalize mode. |
Evan Cheng | ebffb66 | 2008-07-01 17:59:20 +0000 | [diff] [blame] | 515 | if (TimePassesIsEnabled) { |
Dan Gohman | 5e84368 | 2008-07-14 18:19:29 +0000 | [diff] [blame] | 516 | NamedRegionTimer T("DAG Combining 1", GroupName); |
Bill Wendling | be8cc2a | 2009-04-29 00:15:41 +0000 | [diff] [blame] | 517 | CurDAG->Combine(Unrestricted, *AA, OptLevel); |
Evan Cheng | ebffb66 | 2008-07-01 17:59:20 +0000 | [diff] [blame] | 518 | } else { |
Bill Wendling | be8cc2a | 2009-04-29 00:15:41 +0000 | [diff] [blame] | 519 | CurDAG->Combine(Unrestricted, *AA, OptLevel); |
Evan Cheng | ebffb66 | 2008-07-01 17:59:20 +0000 | [diff] [blame] | 520 | } |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 521 | |
David Greene | 1a05323 | 2010-01-05 01:26:11 +0000 | [diff] [blame] | 522 | DEBUG(dbgs() << "Optimized lowered selection DAG:\n"); |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 523 | DEBUG(CurDAG->dump()); |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 524 | |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 525 | // Second step, hack on the DAG until it only uses operations and types that |
| 526 | // the target supports. |
Dan Gohman | 714efc6 | 2009-12-05 17:51:33 +0000 | [diff] [blame] | 527 | if (ViewLegalizeTypesDAGs) CurDAG->viewGraph("legalize-types input for " + |
| 528 | BlockName); |
Dan Gohman | 462dc7f | 2008-07-21 20:00:07 +0000 | [diff] [blame] | 529 | |
Dan Gohman | 714efc6 | 2009-12-05 17:51:33 +0000 | [diff] [blame] | 530 | bool Changed; |
| 531 | if (TimePassesIsEnabled) { |
| 532 | NamedRegionTimer T("Type Legalization", GroupName); |
| 533 | Changed = CurDAG->LegalizeTypes(); |
| 534 | } else { |
| 535 | Changed = CurDAG->LegalizeTypes(); |
| 536 | } |
| 537 | |
David Greene | 1a05323 | 2010-01-05 01:26:11 +0000 | [diff] [blame] | 538 | DEBUG(dbgs() << "Type-legalized selection DAG:\n"); |
Dan Gohman | 714efc6 | 2009-12-05 17:51:33 +0000 | [diff] [blame] | 539 | DEBUG(CurDAG->dump()); |
| 540 | |
| 541 | if (Changed) { |
| 542 | if (ViewDAGCombineLT) |
| 543 | CurDAG->viewGraph("dag-combine-lt input for " + BlockName); |
| 544 | |
| 545 | // Run the DAG combiner in post-type-legalize mode. |
Dan Gohman | 462dc7f | 2008-07-21 20:00:07 +0000 | [diff] [blame] | 546 | if (TimePassesIsEnabled) { |
Dan Gohman | 714efc6 | 2009-12-05 17:51:33 +0000 | [diff] [blame] | 547 | NamedRegionTimer T("DAG Combining after legalize types", GroupName); |
| 548 | CurDAG->Combine(NoIllegalTypes, *AA, OptLevel); |
Dan Gohman | 462dc7f | 2008-07-21 20:00:07 +0000 | [diff] [blame] | 549 | } else { |
Dan Gohman | 714efc6 | 2009-12-05 17:51:33 +0000 | [diff] [blame] | 550 | CurDAG->Combine(NoIllegalTypes, *AA, OptLevel); |
Dan Gohman | 462dc7f | 2008-07-21 20:00:07 +0000 | [diff] [blame] | 551 | } |
| 552 | |
David Greene | 1a05323 | 2010-01-05 01:26:11 +0000 | [diff] [blame] | 553 | DEBUG(dbgs() << "Optimized type-legalized selection DAG:\n"); |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 554 | DEBUG(CurDAG->dump()); |
Dan Gohman | 714efc6 | 2009-12-05 17:51:33 +0000 | [diff] [blame] | 555 | } |
Dan Gohman | 462dc7f | 2008-07-21 20:00:07 +0000 | [diff] [blame] | 556 | |
Dan Gohman | 714efc6 | 2009-12-05 17:51:33 +0000 | [diff] [blame] | 557 | if (TimePassesIsEnabled) { |
| 558 | NamedRegionTimer T("Vector Legalization", GroupName); |
| 559 | Changed = CurDAG->LegalizeVectors(); |
| 560 | } else { |
| 561 | Changed = CurDAG->LegalizeVectors(); |
| 562 | } |
Duncan Sands | 25cf227 | 2008-11-24 14:53:14 +0000 | [diff] [blame] | 563 | |
Dan Gohman | 714efc6 | 2009-12-05 17:51:33 +0000 | [diff] [blame] | 564 | if (Changed) { |
Eli Friedman | 5c22c80 | 2009-05-23 12:35:30 +0000 | [diff] [blame] | 565 | if (TimePassesIsEnabled) { |
Dan Gohman | 714efc6 | 2009-12-05 17:51:33 +0000 | [diff] [blame] | 566 | NamedRegionTimer T("Type Legalization 2", GroupName); |
Bill Wendling | 9882007 | 2009-12-28 01:51:30 +0000 | [diff] [blame] | 567 | CurDAG->LegalizeTypes(); |
Eli Friedman | 5c22c80 | 2009-05-23 12:35:30 +0000 | [diff] [blame] | 568 | } else { |
Bill Wendling | 9882007 | 2009-12-28 01:51:30 +0000 | [diff] [blame] | 569 | CurDAG->LegalizeTypes(); |
Eli Friedman | 5c22c80 | 2009-05-23 12:35:30 +0000 | [diff] [blame] | 570 | } |
| 571 | |
Dan Gohman | 714efc6 | 2009-12-05 17:51:33 +0000 | [diff] [blame] | 572 | if (ViewDAGCombineLT) |
| 573 | CurDAG->viewGraph("dag-combine-lv input for " + BlockName); |
Eli Friedman | 5c22c80 | 2009-05-23 12:35:30 +0000 | [diff] [blame] | 574 | |
Dan Gohman | 714efc6 | 2009-12-05 17:51:33 +0000 | [diff] [blame] | 575 | // Run the DAG combiner in post-type-legalize mode. |
| 576 | if (TimePassesIsEnabled) { |
| 577 | NamedRegionTimer T("DAG Combining after legalize vectors", GroupName); |
| 578 | CurDAG->Combine(NoIllegalOperations, *AA, OptLevel); |
| 579 | } else { |
| 580 | CurDAG->Combine(NoIllegalOperations, *AA, OptLevel); |
Eli Friedman | 5c22c80 | 2009-05-23 12:35:30 +0000 | [diff] [blame] | 581 | } |
Dan Gohman | 714efc6 | 2009-12-05 17:51:33 +0000 | [diff] [blame] | 582 | |
David Greene | 1a05323 | 2010-01-05 01:26:11 +0000 | [diff] [blame] | 583 | DEBUG(dbgs() << "Optimized vector-legalized selection DAG:\n"); |
Dan Gohman | 714efc6 | 2009-12-05 17:51:33 +0000 | [diff] [blame] | 584 | DEBUG(CurDAG->dump()); |
Chris Lattner | 70587ea | 2008-07-10 23:37:50 +0000 | [diff] [blame] | 585 | } |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 586 | |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 587 | if (ViewLegalizeDAGs) CurDAG->viewGraph("legalize input for " + BlockName); |
Dan Gohman | 462dc7f | 2008-07-21 20:00:07 +0000 | [diff] [blame] | 588 | |
Evan Cheng | ebffb66 | 2008-07-01 17:59:20 +0000 | [diff] [blame] | 589 | if (TimePassesIsEnabled) { |
Dan Gohman | 5e84368 | 2008-07-14 18:19:29 +0000 | [diff] [blame] | 590 | NamedRegionTimer T("DAG Legalization", GroupName); |
Dan Gohman | 714efc6 | 2009-12-05 17:51:33 +0000 | [diff] [blame] | 591 | CurDAG->Legalize(OptLevel); |
Evan Cheng | ebffb66 | 2008-07-01 17:59:20 +0000 | [diff] [blame] | 592 | } else { |
Dan Gohman | 714efc6 | 2009-12-05 17:51:33 +0000 | [diff] [blame] | 593 | CurDAG->Legalize(OptLevel); |
Evan Cheng | ebffb66 | 2008-07-01 17:59:20 +0000 | [diff] [blame] | 594 | } |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 595 | |
David Greene | 1a05323 | 2010-01-05 01:26:11 +0000 | [diff] [blame] | 596 | DEBUG(dbgs() << "Legalized selection DAG:\n"); |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 597 | DEBUG(CurDAG->dump()); |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 598 | |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 599 | if (ViewDAGCombine2) CurDAG->viewGraph("dag-combine2 input for " + BlockName); |
Dan Gohman | 462dc7f | 2008-07-21 20:00:07 +0000 | [diff] [blame] | 600 | |
Chris Lattner | af21d55 | 2005-10-10 16:47:10 +0000 | [diff] [blame] | 601 | // Run the DAG combiner in post-legalize mode. |
Evan Cheng | ebffb66 | 2008-07-01 17:59:20 +0000 | [diff] [blame] | 602 | if (TimePassesIsEnabled) { |
Dan Gohman | 5e84368 | 2008-07-14 18:19:29 +0000 | [diff] [blame] | 603 | NamedRegionTimer T("DAG Combining 2", GroupName); |
Bill Wendling | be8cc2a | 2009-04-29 00:15:41 +0000 | [diff] [blame] | 604 | CurDAG->Combine(NoIllegalOperations, *AA, OptLevel); |
Evan Cheng | ebffb66 | 2008-07-01 17:59:20 +0000 | [diff] [blame] | 605 | } else { |
Bill Wendling | be8cc2a | 2009-04-29 00:15:41 +0000 | [diff] [blame] | 606 | CurDAG->Combine(NoIllegalOperations, *AA, OptLevel); |
Evan Cheng | ebffb66 | 2008-07-01 17:59:20 +0000 | [diff] [blame] | 607 | } |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 608 | |
David Greene | 1a05323 | 2010-01-05 01:26:11 +0000 | [diff] [blame] | 609 | DEBUG(dbgs() << "Optimized legalized selection DAG:\n"); |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 610 | DEBUG(CurDAG->dump()); |
Dan Gohman | 417e11b | 2007-10-08 15:12:17 +0000 | [diff] [blame] | 611 | |
Evan Cheng | d40d03e | 2010-01-06 19:38:29 +0000 | [diff] [blame] | 612 | if (OptLevel != CodeGenOpt::None) { |
| 613 | ShrinkDemandedOps(); |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 614 | ComputeLiveOutVRegInfo(); |
Evan Cheng | d40d03e | 2010-01-06 19:38:29 +0000 | [diff] [blame] | 615 | } |
Evan Cheng | 552c4a8 | 2006-04-28 02:09:19 +0000 | [diff] [blame] | 616 | |
Chris Lattner | 552186d | 2010-03-14 19:27:55 +0000 | [diff] [blame] | 617 | if (ViewISelDAGs) CurDAG->viewGraph("isel input for " + BlockName); |
| 618 | |
Chris Lattner | a33ef48 | 2005-03-30 01:10:47 +0000 | [diff] [blame] | 619 | // Third, instruction select all of the operations to machine code, adding the |
| 620 | // code to the MachineBasicBlock. |
Evan Cheng | ebffb66 | 2008-07-01 17:59:20 +0000 | [diff] [blame] | 621 | if (TimePassesIsEnabled) { |
Dan Gohman | 5e84368 | 2008-07-14 18:19:29 +0000 | [diff] [blame] | 622 | NamedRegionTimer T("Instruction Selection", GroupName); |
Chris Lattner | 7c306da | 2010-03-02 06:34:30 +0000 | [diff] [blame] | 623 | DoInstructionSelection(); |
Evan Cheng | ebffb66 | 2008-07-01 17:59:20 +0000 | [diff] [blame] | 624 | } else { |
Chris Lattner | 7c306da | 2010-03-02 06:34:30 +0000 | [diff] [blame] | 625 | DoInstructionSelection(); |
Evan Cheng | ebffb66 | 2008-07-01 17:59:20 +0000 | [diff] [blame] | 626 | } |
Evan Cheng | db8d56b | 2008-06-30 20:45:06 +0000 | [diff] [blame] | 627 | |
David Greene | 1a05323 | 2010-01-05 01:26:11 +0000 | [diff] [blame] | 628 | DEBUG(dbgs() << "Selected selection DAG:\n"); |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 629 | DEBUG(CurDAG->dump()); |
Dan Gohman | 462dc7f | 2008-07-21 20:00:07 +0000 | [diff] [blame] | 630 | |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 631 | if (ViewSchedDAGs) CurDAG->viewGraph("scheduler input for " + BlockName); |
Dan Gohman | 462dc7f | 2008-07-21 20:00:07 +0000 | [diff] [blame] | 632 | |
Dan Gohman | 5e84368 | 2008-07-14 18:19:29 +0000 | [diff] [blame] | 633 | // Schedule machine code. |
Dan Gohman | 47ac0f0 | 2009-02-11 04:27:20 +0000 | [diff] [blame] | 634 | ScheduleDAGSDNodes *Scheduler = CreateScheduler(); |
Dan Gohman | 5e84368 | 2008-07-14 18:19:29 +0000 | [diff] [blame] | 635 | if (TimePassesIsEnabled) { |
| 636 | NamedRegionTimer T("Instruction Scheduling", GroupName); |
Dan Gohman | 47ac0f0 | 2009-02-11 04:27:20 +0000 | [diff] [blame] | 637 | Scheduler->Run(CurDAG, BB, BB->end()); |
Dan Gohman | 5e84368 | 2008-07-14 18:19:29 +0000 | [diff] [blame] | 638 | } else { |
Dan Gohman | 47ac0f0 | 2009-02-11 04:27:20 +0000 | [diff] [blame] | 639 | Scheduler->Run(CurDAG, BB, BB->end()); |
Dan Gohman | 5e84368 | 2008-07-14 18:19:29 +0000 | [diff] [blame] | 640 | } |
| 641 | |
Dan Gohman | 462dc7f | 2008-07-21 20:00:07 +0000 | [diff] [blame] | 642 | if (ViewSUnitDAGs) Scheduler->viewGraph(); |
| 643 | |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 644 | // Emit machine code to BB. This can change 'BB' to the last block being |
Evan Cheng | db8d56b | 2008-06-30 20:45:06 +0000 | [diff] [blame] | 645 | // inserted into. |
Evan Cheng | ebffb66 | 2008-07-01 17:59:20 +0000 | [diff] [blame] | 646 | if (TimePassesIsEnabled) { |
Dan Gohman | 5e84368 | 2008-07-14 18:19:29 +0000 | [diff] [blame] | 647 | NamedRegionTimer T("Instruction Creation", GroupName); |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 648 | BB = Scheduler->EmitSchedule(&SDB->EdgeMapping); |
Evan Cheng | ebffb66 | 2008-07-01 17:59:20 +0000 | [diff] [blame] | 649 | } else { |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 650 | BB = Scheduler->EmitSchedule(&SDB->EdgeMapping); |
Dan Gohman | 5e84368 | 2008-07-14 18:19:29 +0000 | [diff] [blame] | 651 | } |
| 652 | |
| 653 | // Free the scheduler state. |
| 654 | if (TimePassesIsEnabled) { |
| 655 | NamedRegionTimer T("Instruction Scheduling Cleanup", GroupName); |
| 656 | delete Scheduler; |
| 657 | } else { |
| 658 | delete Scheduler; |
Evan Cheng | ebffb66 | 2008-07-01 17:59:20 +0000 | [diff] [blame] | 659 | } |
Evan Cheng | db8d56b | 2008-06-30 20:45:06 +0000 | [diff] [blame] | 660 | |
David Greene | 1a05323 | 2010-01-05 01:26:11 +0000 | [diff] [blame] | 661 | DEBUG(dbgs() << "Selected machine code:\n"); |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 662 | DEBUG(BB->dump()); |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 663 | } |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 664 | |
Chris Lattner | 7c306da | 2010-03-02 06:34:30 +0000 | [diff] [blame] | 665 | void SelectionDAGISel::DoInstructionSelection() { |
| 666 | DEBUG(errs() << "===== Instruction selection begins:\n"); |
| 667 | |
| 668 | PreprocessISelDAG(); |
| 669 | |
| 670 | // Select target instructions for the DAG. |
| 671 | { |
| 672 | // Number all nodes with a topological order and set DAGSize. |
| 673 | DAGSize = CurDAG->AssignTopologicalOrder(); |
| 674 | |
| 675 | // Create a dummy node (which is not added to allnodes), that adds |
| 676 | // a reference to the root node, preventing it from being deleted, |
| 677 | // and tracking any changes of the root. |
| 678 | HandleSDNode Dummy(CurDAG->getRoot()); |
| 679 | ISelPosition = SelectionDAG::allnodes_iterator(CurDAG->getRoot().getNode()); |
| 680 | ++ISelPosition; |
| 681 | |
| 682 | // The AllNodes list is now topological-sorted. Visit the |
| 683 | // nodes by starting at the end of the list (the root of the |
| 684 | // graph) and preceding back toward the beginning (the entry |
| 685 | // node). |
| 686 | while (ISelPosition != CurDAG->allnodes_begin()) { |
| 687 | SDNode *Node = --ISelPosition; |
| 688 | // Skip dead nodes. DAGCombiner is expected to eliminate all dead nodes, |
| 689 | // but there are currently some corner cases that it misses. Also, this |
| 690 | // makes it theoretically possible to disable the DAGCombiner. |
| 691 | if (Node->use_empty()) |
| 692 | continue; |
| 693 | |
| 694 | SDNode *ResNode = Select(Node); |
| 695 | |
Chris Lattner | 82dd3d3 | 2010-03-02 07:50:03 +0000 | [diff] [blame] | 696 | // FIXME: This is pretty gross. 'Select' should be changed to not return |
| 697 | // anything at all and this code should be nuked with a tactical strike. |
| 698 | |
Chris Lattner | 7c306da | 2010-03-02 06:34:30 +0000 | [diff] [blame] | 699 | // If node should not be replaced, continue with the next one. |
Chris Lattner | 82dd3d3 | 2010-03-02 07:50:03 +0000 | [diff] [blame] | 700 | if (ResNode == Node || Node->getOpcode() == ISD::DELETED_NODE) |
Chris Lattner | 7c306da | 2010-03-02 06:34:30 +0000 | [diff] [blame] | 701 | continue; |
| 702 | // Replace node. |
| 703 | if (ResNode) |
| 704 | ReplaceUses(Node, ResNode); |
| 705 | |
| 706 | // If after the replacement this node is not used any more, |
| 707 | // remove this dead node. |
| 708 | if (Node->use_empty()) { // Don't delete EntryToken, etc. |
| 709 | ISelUpdater ISU(ISelPosition); |
| 710 | CurDAG->RemoveDeadNode(Node, &ISU); |
| 711 | } |
| 712 | } |
| 713 | |
| 714 | CurDAG->setRoot(Dummy.getValue()); |
| 715 | } |
| 716 | DEBUG(errs() << "===== Instruction selection ends:\n"); |
| 717 | |
| 718 | PostprocessISelDAG(); |
Chris Lattner | 7c306da | 2010-03-02 06:34:30 +0000 | [diff] [blame] | 719 | } |
| 720 | |
| 721 | |
Dan Gohman | 79ce276 | 2009-01-15 19:20:50 +0000 | [diff] [blame] | 722 | void SelectionDAGISel::SelectAllBasicBlocks(Function &Fn, |
| 723 | MachineFunction &MF, |
Dan Gohman | dd5b58a | 2008-10-14 23:54:11 +0000 | [diff] [blame] | 724 | const TargetInstrInfo &TII) { |
Dan Gohman | a43abd1 | 2008-09-29 21:55:50 +0000 | [diff] [blame] | 725 | // Initialize the Fast-ISel state, if needed. |
| 726 | FastISel *FastIS = 0; |
| 727 | if (EnableFastISel) |
Chris Lattner | ed3a806 | 2010-04-05 06:05:26 +0000 | [diff] [blame] | 728 | FastIS = TLI.createFastISel(MF, FuncInfo->ValueMap, FuncInfo->MBBMap, |
Dan Gohman | dd5b58a | 2008-10-14 23:54:11 +0000 | [diff] [blame] | 729 | FuncInfo->StaticAllocaMap |
| 730 | #ifndef NDEBUG |
| 731 | , FuncInfo->CatchInfoLost |
| 732 | #endif |
| 733 | ); |
Dan Gohman | a43abd1 | 2008-09-29 21:55:50 +0000 | [diff] [blame] | 734 | |
| 735 | // Iterate over all basic blocks in the function. |
Evan Cheng | 39fd6e8 | 2008-08-07 00:43:25 +0000 | [diff] [blame] | 736 | for (Function::iterator I = Fn.begin(), E = Fn.end(); I != E; ++I) { |
| 737 | BasicBlock *LLVMBB = &*I; |
Dan Gohman | 7c3234c | 2008-08-27 23:52:12 +0000 | [diff] [blame] | 738 | BB = FuncInfo->MBBMap[LLVMBB]; |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 739 | |
Dan Gohman | 3df24e6 | 2008-09-03 23:12:08 +0000 | [diff] [blame] | 740 | BasicBlock::iterator const Begin = LLVMBB->begin(); |
| 741 | BasicBlock::iterator const End = LLVMBB->end(); |
Evan Cheng | 9f11850 | 2008-09-08 16:01:27 +0000 | [diff] [blame] | 742 | BasicBlock::iterator BI = Begin; |
Dan Gohman | 5edd361 | 2008-08-28 20:28:56 +0000 | [diff] [blame] | 743 | |
| 744 | // Lower any arguments needed in this block if this is the entry block. |
Dan Gohman | 33134c4 | 2008-09-25 17:05:24 +0000 | [diff] [blame] | 745 | bool SuppressFastISel = false; |
| 746 | if (LLVMBB == &Fn.getEntryBlock()) { |
Dan Gohman | 5edd361 | 2008-08-28 20:28:56 +0000 | [diff] [blame] | 747 | LowerArguments(LLVMBB); |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 748 | |
Dan Gohman | 33134c4 | 2008-09-25 17:05:24 +0000 | [diff] [blame] | 749 | // If any of the arguments has the byval attribute, forgo |
| 750 | // fast-isel in the entry block. |
Dan Gohman | a43abd1 | 2008-09-29 21:55:50 +0000 | [diff] [blame] | 751 | if (FastIS) { |
Dan Gohman | 33134c4 | 2008-09-25 17:05:24 +0000 | [diff] [blame] | 752 | unsigned j = 1; |
| 753 | for (Function::arg_iterator I = Fn.arg_begin(), E = Fn.arg_end(); |
| 754 | I != E; ++I, ++j) |
Devang Patel | 0598866 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 755 | if (Fn.paramHasAttr(j, Attribute::ByVal)) { |
Dan Gohman | 77ca41e | 2008-09-25 17:21:42 +0000 | [diff] [blame] | 756 | if (EnableFastISelVerbose || EnableFastISelAbort) |
David Greene | 1a05323 | 2010-01-05 01:26:11 +0000 | [diff] [blame] | 757 | dbgs() << "FastISel skips entry block due to byval argument\n"; |
Dan Gohman | 33134c4 | 2008-09-25 17:05:24 +0000 | [diff] [blame] | 758 | SuppressFastISel = true; |
| 759 | break; |
| 760 | } |
| 761 | } |
| 762 | } |
| 763 | |
Chris Lattner | de6e783 | 2010-04-05 06:10:13 +0000 | [diff] [blame] | 764 | if (BB->isLandingPad()) { |
Dan Gohman | dd5b58a | 2008-10-14 23:54:11 +0000 | [diff] [blame] | 765 | // Add a label to mark the beginning of the landing pad. Deletion of the |
| 766 | // landing pad can thus be detected via the MachineModuleInfo. |
Chris Lattner | de6e783 | 2010-04-05 06:10:13 +0000 | [diff] [blame] | 767 | MCSymbol *Label = MF.getMMI().addLandingPad(BB); |
Dan Gohman | dd5b58a | 2008-10-14 23:54:11 +0000 | [diff] [blame] | 768 | |
Chris Lattner | 518bb53 | 2010-02-09 19:54:29 +0000 | [diff] [blame] | 769 | const TargetInstrDesc &II = TII.get(TargetOpcode::EH_LABEL); |
Chris Lattner | 7561d48 | 2010-03-14 02:33:54 +0000 | [diff] [blame] | 770 | BuildMI(BB, SDB->getCurDebugLoc(), II).addSym(Label); |
Dan Gohman | dd5b58a | 2008-10-14 23:54:11 +0000 | [diff] [blame] | 771 | |
| 772 | // Mark exception register as live in. |
| 773 | unsigned Reg = TLI.getExceptionAddressRegister(); |
| 774 | if (Reg) BB->addLiveIn(Reg); |
| 775 | |
| 776 | // Mark exception selector register as live in. |
| 777 | Reg = TLI.getExceptionSelectorRegister(); |
| 778 | if (Reg) BB->addLiveIn(Reg); |
| 779 | |
| 780 | // FIXME: Hack around an exception handling flaw (PR1508): the personality |
| 781 | // function and list of typeids logically belong to the invoke (or, if you |
| 782 | // like, the basic block containing the invoke), and need to be associated |
| 783 | // with it in the dwarf exception handling tables. Currently however the |
| 784 | // information is provided by an intrinsic (eh.selector) that can be moved |
| 785 | // to unexpected places by the optimizers: if the unwind edge is critical, |
| 786 | // then breaking it can result in the intrinsics being in the successor of |
Jim Grosbach | f4549b0 | 2010-01-15 00:36:15 +0000 | [diff] [blame] | 787 | // the landing pad, not the landing pad itself. This results |
| 788 | // in exceptions not being caught because no typeids are associated with |
| 789 | // the invoke. This may not be the only way things can go wrong, but it |
| 790 | // is the only way we try to work around for the moment. |
Dan Gohman | dd5b58a | 2008-10-14 23:54:11 +0000 | [diff] [blame] | 791 | BranchInst *Br = dyn_cast<BranchInst>(LLVMBB->getTerminator()); |
| 792 | |
| 793 | if (Br && Br->isUnconditional()) { // Critical edge? |
| 794 | BasicBlock::iterator I, E; |
| 795 | for (I = LLVMBB->begin(), E = --LLVMBB->end(); I != E; ++I) |
| 796 | if (isa<EHSelectorInst>(I)) |
| 797 | break; |
| 798 | |
| 799 | if (I == E) |
| 800 | // No catch info found - try to extract some from the successor. |
Chris Lattner | de6e783 | 2010-04-05 06:10:13 +0000 | [diff] [blame] | 801 | CopyCatchInfo(Br->getSuccessor(0), LLVMBB, &MF.getMMI(), *FuncInfo); |
Dan Gohman | dd5b58a | 2008-10-14 23:54:11 +0000 | [diff] [blame] | 802 | } |
| 803 | } |
| 804 | |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 805 | // Before doing SelectionDAG ISel, see if FastISel has been requested. |
Dan Gohman | dd5b58a | 2008-10-14 23:54:11 +0000 | [diff] [blame] | 806 | if (FastIS && !SuppressFastISel) { |
Dan Gohman | a43abd1 | 2008-09-29 21:55:50 +0000 | [diff] [blame] | 807 | // Emit code for any incoming arguments. This must happen before |
| 808 | // beginning FastISel on the entry block. |
| 809 | if (LLVMBB == &Fn.getEntryBlock()) { |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 810 | CurDAG->setRoot(SDB->getControlRoot()); |
Dan Gohman | a43abd1 | 2008-09-29 21:55:50 +0000 | [diff] [blame] | 811 | CodeGenAndEmitDAG(); |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 812 | SDB->clear(); |
Dan Gohman | a43abd1 | 2008-09-29 21:55:50 +0000 | [diff] [blame] | 813 | } |
Dan Gohman | 241f464 | 2008-10-04 00:56:36 +0000 | [diff] [blame] | 814 | FastIS->startNewBlock(BB); |
Dan Gohman | a43abd1 | 2008-09-29 21:55:50 +0000 | [diff] [blame] | 815 | // Do FastISel on as many instructions as possible. |
| 816 | for (; BI != End; ++BI) { |
| 817 | // Just before the terminator instruction, insert instructions to |
| 818 | // feed PHI nodes in successor blocks. |
| 819 | if (isa<TerminatorInst>(BI)) |
| 820 | if (!HandlePHINodesInSuccessorBlocksFast(LLVMBB, FastIS)) { |
Chris Lattner | c19ae9d | 2010-03-04 19:46:56 +0000 | [diff] [blame] | 821 | ++NumFastIselFailures; |
Dan Gohman | 07f111e | 2009-12-05 00:27:08 +0000 | [diff] [blame] | 822 | ResetDebugLoc(SDB, FastIS); |
Dan Gohman | 4344a5d | 2008-09-09 23:05:00 +0000 | [diff] [blame] | 823 | if (EnableFastISelVerbose || EnableFastISelAbort) { |
David Greene | 1a05323 | 2010-01-05 01:26:11 +0000 | [diff] [blame] | 824 | dbgs() << "FastISel miss: "; |
Dan Gohman | 293d5f8 | 2008-09-09 22:06:46 +0000 | [diff] [blame] | 825 | BI->dump(); |
| 826 | } |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 827 | assert(!EnableFastISelAbort && |
Torok Edwin | f368923 | 2009-07-12 20:07:01 +0000 | [diff] [blame] | 828 | "FastISel didn't handle a PHI in a successor"); |
Dan Gohman | a43abd1 | 2008-09-29 21:55:50 +0000 | [diff] [blame] | 829 | break; |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 830 | } |
Dan Gohman | a43abd1 | 2008-09-29 21:55:50 +0000 | [diff] [blame] | 831 | |
Chris Lattner | 23dbf66 | 2010-03-31 04:24:50 +0000 | [diff] [blame] | 832 | SetDebugLoc(BI, SDB, FastIS, &MF); |
Dan Gohman | 381ca55 | 2009-12-05 01:29:04 +0000 | [diff] [blame] | 833 | |
Dan Gohman | 21c14e3 | 2010-01-12 04:32:35 +0000 | [diff] [blame] | 834 | // Try to select the instruction with FastISel. |
Dan Gohman | 07f111e | 2009-12-05 00:27:08 +0000 | [diff] [blame] | 835 | if (FastIS->SelectInstruction(BI)) { |
| 836 | ResetDebugLoc(SDB, FastIS); |
Dan Gohman | a43abd1 | 2008-09-29 21:55:50 +0000 | [diff] [blame] | 837 | continue; |
Dan Gohman | 07f111e | 2009-12-05 00:27:08 +0000 | [diff] [blame] | 838 | } |
Dan Gohman | a43abd1 | 2008-09-29 21:55:50 +0000 | [diff] [blame] | 839 | |
Dan Gohman | 07f111e | 2009-12-05 00:27:08 +0000 | [diff] [blame] | 840 | // Clear out the debug location so that it doesn't carry over to |
| 841 | // unrelated instructions. |
| 842 | ResetDebugLoc(SDB, FastIS); |
Dan Gohman | a43abd1 | 2008-09-29 21:55:50 +0000 | [diff] [blame] | 843 | |
| 844 | // Then handle certain instructions as single-LLVM-Instruction blocks. |
| 845 | if (isa<CallInst>(BI)) { |
Chris Lattner | c19ae9d | 2010-03-04 19:46:56 +0000 | [diff] [blame] | 846 | ++NumFastIselFailures; |
Dan Gohman | a43abd1 | 2008-09-29 21:55:50 +0000 | [diff] [blame] | 847 | if (EnableFastISelVerbose || EnableFastISelAbort) { |
David Greene | 1a05323 | 2010-01-05 01:26:11 +0000 | [diff] [blame] | 848 | dbgs() << "FastISel missed call: "; |
Dan Gohman | a43abd1 | 2008-09-29 21:55:50 +0000 | [diff] [blame] | 849 | BI->dump(); |
| 850 | } |
| 851 | |
Benjamin Kramer | f012705 | 2010-01-05 13:12:22 +0000 | [diff] [blame] | 852 | if (!BI->getType()->isVoidTy()) { |
Dan Gohman | a43abd1 | 2008-09-29 21:55:50 +0000 | [diff] [blame] | 853 | unsigned &R = FuncInfo->ValueMap[BI]; |
| 854 | if (!R) |
| 855 | R = FuncInfo->CreateRegForValue(BI); |
| 856 | } |
| 857 | |
Dan Gohman | b4afb13 | 2009-11-20 02:51:26 +0000 | [diff] [blame] | 858 | bool HadTailCall = false; |
Chris Lattner | 7896c9f | 2009-12-03 00:50:42 +0000 | [diff] [blame] | 859 | SelectBasicBlock(LLVMBB, BI, llvm::next(BI), HadTailCall); |
Dan Gohman | b4afb13 | 2009-11-20 02:51:26 +0000 | [diff] [blame] | 860 | |
| 861 | // If the call was emitted as a tail call, we're done with the block. |
| 862 | if (HadTailCall) { |
| 863 | BI = End; |
| 864 | break; |
| 865 | } |
| 866 | |
Dan Gohman | 241f464 | 2008-10-04 00:56:36 +0000 | [diff] [blame] | 867 | // If the instruction was codegen'd with multiple blocks, |
| 868 | // inform the FastISel object where to resume inserting. |
| 869 | FastIS->setCurrentBlock(BB); |
Dan Gohman | a43abd1 | 2008-09-29 21:55:50 +0000 | [diff] [blame] | 870 | continue; |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 871 | } |
Dan Gohman | a43abd1 | 2008-09-29 21:55:50 +0000 | [diff] [blame] | 872 | |
| 873 | // Otherwise, give up on FastISel for the rest of the block. |
| 874 | // For now, be a little lenient about non-branch terminators. |
| 875 | if (!isa<TerminatorInst>(BI) || isa<BranchInst>(BI)) { |
Chris Lattner | c19ae9d | 2010-03-04 19:46:56 +0000 | [diff] [blame] | 876 | ++NumFastIselFailures; |
Dan Gohman | a43abd1 | 2008-09-29 21:55:50 +0000 | [diff] [blame] | 877 | if (EnableFastISelVerbose || EnableFastISelAbort) { |
David Greene | 1a05323 | 2010-01-05 01:26:11 +0000 | [diff] [blame] | 878 | dbgs() << "FastISel miss: "; |
Dan Gohman | a43abd1 | 2008-09-29 21:55:50 +0000 | [diff] [blame] | 879 | BI->dump(); |
| 880 | } |
| 881 | if (EnableFastISelAbort) |
| 882 | // The "fast" selector couldn't handle something and bailed. |
| 883 | // For the purpose of debugging, just abort. |
Torok Edwin | c23197a | 2009-07-14 16:55:14 +0000 | [diff] [blame] | 884 | llvm_unreachable("FastISel didn't select the entire block"); |
Dan Gohman | a43abd1 | 2008-09-29 21:55:50 +0000 | [diff] [blame] | 885 | } |
| 886 | break; |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 887 | } |
| 888 | } |
| 889 | |
Dan Gohman | d2ff647 | 2008-09-02 20:17:56 +0000 | [diff] [blame] | 890 | // Run SelectionDAG instruction selection on the remainder of the block |
| 891 | // not handled by FastISel. If FastISel is not run, this is the entire |
Dan Gohman | 3df24e6 | 2008-09-03 23:12:08 +0000 | [diff] [blame] | 892 | // block. |
Devang Patel | 390f3ac | 2009-04-16 01:33:10 +0000 | [diff] [blame] | 893 | if (BI != End) { |
Dan Gohman | b4afb13 | 2009-11-20 02:51:26 +0000 | [diff] [blame] | 894 | bool HadTailCall; |
| 895 | SelectBasicBlock(LLVMBB, BI, End, HadTailCall); |
Devang Patel | 390f3ac | 2009-04-16 01:33:10 +0000 | [diff] [blame] | 896 | } |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 897 | |
Dan Gohman | 7c3234c | 2008-08-27 23:52:12 +0000 | [diff] [blame] | 898 | FinishBasicBlock(); |
Evan Cheng | 39fd6e8 | 2008-08-07 00:43:25 +0000 | [diff] [blame] | 899 | } |
Dan Gohman | a43abd1 | 2008-09-29 21:55:50 +0000 | [diff] [blame] | 900 | |
| 901 | delete FastIS; |
Dan Gohman | 0e5f130 | 2008-07-07 23:02:41 +0000 | [diff] [blame] | 902 | } |
| 903 | |
Dan Gohman | fed90b6 | 2008-07-28 21:51:04 +0000 | [diff] [blame] | 904 | void |
Dan Gohman | 7c3234c | 2008-08-27 23:52:12 +0000 | [diff] [blame] | 905 | SelectionDAGISel::FinishBasicBlock() { |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 906 | |
David Greene | 1a05323 | 2010-01-05 01:26:11 +0000 | [diff] [blame] | 907 | DEBUG(dbgs() << "Target-post-processed machine code:\n"); |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 908 | DEBUG(BB->dump()); |
Nate Begeman | f15485a | 2006-03-27 01:32:24 +0000 | [diff] [blame] | 909 | |
David Greene | 1a05323 | 2010-01-05 01:26:11 +0000 | [diff] [blame] | 910 | DEBUG(dbgs() << "Total amount of phi nodes to update: " |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 911 | << SDB->PHINodesToUpdate.size() << "\n"); |
| 912 | DEBUG(for (unsigned i = 0, e = SDB->PHINodesToUpdate.size(); i != e; ++i) |
David Greene | 1a05323 | 2010-01-05 01:26:11 +0000 | [diff] [blame] | 913 | dbgs() << "Node " << i << " : (" |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 914 | << SDB->PHINodesToUpdate[i].first |
| 915 | << ", " << SDB->PHINodesToUpdate[i].second << ")\n"); |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 916 | |
Chris Lattner | a33ef48 | 2005-03-30 01:10:47 +0000 | [diff] [blame] | 917 | // Next, now that we know what the last MBB the LLVM BB expanded is, update |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 918 | // PHI nodes in successors. |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 919 | if (SDB->SwitchCases.empty() && |
| 920 | SDB->JTCases.empty() && |
| 921 | SDB->BitTestCases.empty()) { |
| 922 | for (unsigned i = 0, e = SDB->PHINodesToUpdate.size(); i != e; ++i) { |
| 923 | MachineInstr *PHI = SDB->PHINodesToUpdate[i].first; |
Chris Lattner | 518bb53 | 2010-02-09 19:54:29 +0000 | [diff] [blame] | 924 | assert(PHI->isPHI() && |
Nate Begeman | f15485a | 2006-03-27 01:32:24 +0000 | [diff] [blame] | 925 | "This is not a machine PHI node that we are updating!"); |
Jakob Stoklund Olesen | 580bba2 | 2010-03-05 21:49:10 +0000 | [diff] [blame] | 926 | if (!BB->isSuccessor(PHI->getParent())) |
| 927 | continue; |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 928 | PHI->addOperand(MachineOperand::CreateReg(SDB->PHINodesToUpdate[i].second, |
Chris Lattner | 9ce2e9d | 2007-12-30 00:57:42 +0000 | [diff] [blame] | 929 | false)); |
| 930 | PHI->addOperand(MachineOperand::CreateMBB(BB)); |
Nate Begeman | f15485a | 2006-03-27 01:32:24 +0000 | [diff] [blame] | 931 | } |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 932 | SDB->PHINodesToUpdate.clear(); |
Nate Begeman | f15485a | 2006-03-27 01:32:24 +0000 | [diff] [blame] | 933 | return; |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 934 | } |
Anton Korobeynikov | 4198c58 | 2007-04-09 12:31:58 +0000 | [diff] [blame] | 935 | |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 936 | for (unsigned i = 0, e = SDB->BitTestCases.size(); i != e; ++i) { |
Anton Korobeynikov | 4198c58 | 2007-04-09 12:31:58 +0000 | [diff] [blame] | 937 | // Lower header first, if it wasn't already lowered |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 938 | if (!SDB->BitTestCases[i].Emitted) { |
Anton Korobeynikov | 4198c58 | 2007-04-09 12:31:58 +0000 | [diff] [blame] | 939 | // Set the current basic block to the mbb we wish to insert the code into |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 940 | BB = SDB->BitTestCases[i].Parent; |
| 941 | SDB->setCurrentBasicBlock(BB); |
Anton Korobeynikov | 4198c58 | 2007-04-09 12:31:58 +0000 | [diff] [blame] | 942 | // Emit the code |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 943 | SDB->visitBitTestHeader(SDB->BitTestCases[i]); |
| 944 | CurDAG->setRoot(SDB->getRoot()); |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 945 | CodeGenAndEmitDAG(); |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 946 | SDB->clear(); |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 947 | } |
Anton Korobeynikov | 4198c58 | 2007-04-09 12:31:58 +0000 | [diff] [blame] | 948 | |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 949 | for (unsigned j = 0, ej = SDB->BitTestCases[i].Cases.size(); j != ej; ++j) { |
Anton Korobeynikov | 4198c58 | 2007-04-09 12:31:58 +0000 | [diff] [blame] | 950 | // Set the current basic block to the mbb we wish to insert the code into |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 951 | BB = SDB->BitTestCases[i].Cases[j].ThisBB; |
| 952 | SDB->setCurrentBasicBlock(BB); |
Anton Korobeynikov | 4198c58 | 2007-04-09 12:31:58 +0000 | [diff] [blame] | 953 | // Emit the code |
| 954 | if (j+1 != ej) |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 955 | SDB->visitBitTestCase(SDB->BitTestCases[i].Cases[j+1].ThisBB, |
| 956 | SDB->BitTestCases[i].Reg, |
| 957 | SDB->BitTestCases[i].Cases[j]); |
Anton Korobeynikov | 4198c58 | 2007-04-09 12:31:58 +0000 | [diff] [blame] | 958 | else |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 959 | SDB->visitBitTestCase(SDB->BitTestCases[i].Default, |
| 960 | SDB->BitTestCases[i].Reg, |
| 961 | SDB->BitTestCases[i].Cases[j]); |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 962 | |
| 963 | |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 964 | CurDAG->setRoot(SDB->getRoot()); |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 965 | CodeGenAndEmitDAG(); |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 966 | SDB->clear(); |
Anton Korobeynikov | 4198c58 | 2007-04-09 12:31:58 +0000 | [diff] [blame] | 967 | } |
| 968 | |
| 969 | // Update PHI Nodes |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 970 | for (unsigned pi = 0, pe = SDB->PHINodesToUpdate.size(); pi != pe; ++pi) { |
| 971 | MachineInstr *PHI = SDB->PHINodesToUpdate[pi].first; |
Anton Korobeynikov | 4198c58 | 2007-04-09 12:31:58 +0000 | [diff] [blame] | 972 | MachineBasicBlock *PHIBB = PHI->getParent(); |
Chris Lattner | 518bb53 | 2010-02-09 19:54:29 +0000 | [diff] [blame] | 973 | assert(PHI->isPHI() && |
Anton Korobeynikov | 4198c58 | 2007-04-09 12:31:58 +0000 | [diff] [blame] | 974 | "This is not a machine PHI node that we are updating!"); |
| 975 | // This is "default" BB. We have two jumps to it. From "header" BB and |
| 976 | // from last "case" BB. |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 977 | if (PHIBB == SDB->BitTestCases[i].Default) { |
Jim Grosbach | f4549b0 | 2010-01-15 00:36:15 +0000 | [diff] [blame] | 978 | PHI->addOperand(MachineOperand:: |
| 979 | CreateReg(SDB->PHINodesToUpdate[pi].second, false)); |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 980 | PHI->addOperand(MachineOperand::CreateMBB(SDB->BitTestCases[i].Parent)); |
Jim Grosbach | f4549b0 | 2010-01-15 00:36:15 +0000 | [diff] [blame] | 981 | PHI->addOperand(MachineOperand:: |
| 982 | CreateReg(SDB->PHINodesToUpdate[pi].second, false)); |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 983 | PHI->addOperand(MachineOperand::CreateMBB(SDB->BitTestCases[i].Cases. |
Chris Lattner | 9ce2e9d | 2007-12-30 00:57:42 +0000 | [diff] [blame] | 984 | back().ThisBB)); |
Anton Korobeynikov | 4198c58 | 2007-04-09 12:31:58 +0000 | [diff] [blame] | 985 | } |
| 986 | // One of "cases" BB. |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 987 | for (unsigned j = 0, ej = SDB->BitTestCases[i].Cases.size(); |
Dan Gohman | 7c3234c | 2008-08-27 23:52:12 +0000 | [diff] [blame] | 988 | j != ej; ++j) { |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 989 | MachineBasicBlock* cBB = SDB->BitTestCases[i].Cases[j].ThisBB; |
Jakob Stoklund Olesen | dd437ba | 2010-01-11 21:02:33 +0000 | [diff] [blame] | 990 | if (cBB->isSuccessor(PHIBB)) { |
Jim Grosbach | f4549b0 | 2010-01-15 00:36:15 +0000 | [diff] [blame] | 991 | PHI->addOperand(MachineOperand:: |
| 992 | CreateReg(SDB->PHINodesToUpdate[pi].second, false)); |
Chris Lattner | 9ce2e9d | 2007-12-30 00:57:42 +0000 | [diff] [blame] | 993 | PHI->addOperand(MachineOperand::CreateMBB(cBB)); |
Anton Korobeynikov | 4198c58 | 2007-04-09 12:31:58 +0000 | [diff] [blame] | 994 | } |
| 995 | } |
| 996 | } |
| 997 | } |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 998 | SDB->BitTestCases.clear(); |
Anton Korobeynikov | 4198c58 | 2007-04-09 12:31:58 +0000 | [diff] [blame] | 999 | |
Nate Begeman | 9453eea | 2006-04-23 06:26:20 +0000 | [diff] [blame] | 1000 | // If the JumpTable record is filled in, then we need to emit a jump table. |
| 1001 | // Updating the PHI nodes is tricky in this case, since we need to determine |
| 1002 | // whether the PHI is a successor of the range check MBB or the jump table MBB |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1003 | for (unsigned i = 0, e = SDB->JTCases.size(); i != e; ++i) { |
Anton Korobeynikov | 3a84b9b | 2007-03-25 15:07:15 +0000 | [diff] [blame] | 1004 | // Lower header first, if it wasn't already lowered |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1005 | if (!SDB->JTCases[i].first.Emitted) { |
Anton Korobeynikov | 3a84b9b | 2007-03-25 15:07:15 +0000 | [diff] [blame] | 1006 | // Set the current basic block to the mbb we wish to insert the code into |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1007 | BB = SDB->JTCases[i].first.HeaderBB; |
| 1008 | SDB->setCurrentBasicBlock(BB); |
Anton Korobeynikov | 3a84b9b | 2007-03-25 15:07:15 +0000 | [diff] [blame] | 1009 | // Emit the code |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1010 | SDB->visitJumpTableHeader(SDB->JTCases[i].second, SDB->JTCases[i].first); |
| 1011 | CurDAG->setRoot(SDB->getRoot()); |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 1012 | CodeGenAndEmitDAG(); |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1013 | SDB->clear(); |
Anton Korobeynikov | 4198c58 | 2007-04-09 12:31:58 +0000 | [diff] [blame] | 1014 | } |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1015 | |
Nate Begeman | 37efe67 | 2006-04-22 18:53:45 +0000 | [diff] [blame] | 1016 | // Set the current basic block to the mbb we wish to insert the code into |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1017 | BB = SDB->JTCases[i].second.MBB; |
| 1018 | SDB->setCurrentBasicBlock(BB); |
Nate Begeman | 37efe67 | 2006-04-22 18:53:45 +0000 | [diff] [blame] | 1019 | // Emit the code |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1020 | SDB->visitJumpTable(SDB->JTCases[i].second); |
| 1021 | CurDAG->setRoot(SDB->getRoot()); |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 1022 | CodeGenAndEmitDAG(); |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1023 | SDB->clear(); |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1024 | |
Nate Begeman | 37efe67 | 2006-04-22 18:53:45 +0000 | [diff] [blame] | 1025 | // Update PHI Nodes |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1026 | for (unsigned pi = 0, pe = SDB->PHINodesToUpdate.size(); pi != pe; ++pi) { |
| 1027 | MachineInstr *PHI = SDB->PHINodesToUpdate[pi].first; |
Nate Begeman | 37efe67 | 2006-04-22 18:53:45 +0000 | [diff] [blame] | 1028 | MachineBasicBlock *PHIBB = PHI->getParent(); |
Chris Lattner | 518bb53 | 2010-02-09 19:54:29 +0000 | [diff] [blame] | 1029 | assert(PHI->isPHI() && |
Nate Begeman | 37efe67 | 2006-04-22 18:53:45 +0000 | [diff] [blame] | 1030 | "This is not a machine PHI node that we are updating!"); |
Anton Korobeynikov | 4198c58 | 2007-04-09 12:31:58 +0000 | [diff] [blame] | 1031 | // "default" BB. We can go there only from header BB. |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1032 | if (PHIBB == SDB->JTCases[i].second.Default) { |
Evan Cheng | ce31910 | 2009-09-19 09:51:03 +0000 | [diff] [blame] | 1033 | PHI->addOperand |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1034 | (MachineOperand::CreateReg(SDB->PHINodesToUpdate[pi].second, false)); |
Evan Cheng | ce31910 | 2009-09-19 09:51:03 +0000 | [diff] [blame] | 1035 | PHI->addOperand |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1036 | (MachineOperand::CreateMBB(SDB->JTCases[i].first.HeaderBB)); |
Nate Begeman | f4360a4 | 2006-05-03 03:48:02 +0000 | [diff] [blame] | 1037 | } |
Anton Korobeynikov | 4198c58 | 2007-04-09 12:31:58 +0000 | [diff] [blame] | 1038 | // JT BB. Just iterate over successors here |
Jakob Stoklund Olesen | dd437ba | 2010-01-11 21:02:33 +0000 | [diff] [blame] | 1039 | if (BB->isSuccessor(PHIBB)) { |
Evan Cheng | ce31910 | 2009-09-19 09:51:03 +0000 | [diff] [blame] | 1040 | PHI->addOperand |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1041 | (MachineOperand::CreateReg(SDB->PHINodesToUpdate[pi].second, false)); |
Chris Lattner | 9ce2e9d | 2007-12-30 00:57:42 +0000 | [diff] [blame] | 1042 | PHI->addOperand(MachineOperand::CreateMBB(BB)); |
Nate Begeman | 37efe67 | 2006-04-22 18:53:45 +0000 | [diff] [blame] | 1043 | } |
| 1044 | } |
Nate Begeman | 37efe67 | 2006-04-22 18:53:45 +0000 | [diff] [blame] | 1045 | } |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1046 | SDB->JTCases.clear(); |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1047 | |
Chris Lattner | b2e806e | 2006-10-22 23:00:53 +0000 | [diff] [blame] | 1048 | // If the switch block involved a branch to one of the actual successors, we |
| 1049 | // need to update PHI nodes in that block. |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1050 | for (unsigned i = 0, e = SDB->PHINodesToUpdate.size(); i != e; ++i) { |
| 1051 | MachineInstr *PHI = SDB->PHINodesToUpdate[i].first; |
Chris Lattner | 518bb53 | 2010-02-09 19:54:29 +0000 | [diff] [blame] | 1052 | assert(PHI->isPHI() && |
Chris Lattner | b2e806e | 2006-10-22 23:00:53 +0000 | [diff] [blame] | 1053 | "This is not a machine PHI node that we are updating!"); |
| 1054 | if (BB->isSuccessor(PHI->getParent())) { |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1055 | PHI->addOperand(MachineOperand::CreateReg(SDB->PHINodesToUpdate[i].second, |
Chris Lattner | 9ce2e9d | 2007-12-30 00:57:42 +0000 | [diff] [blame] | 1056 | false)); |
| 1057 | PHI->addOperand(MachineOperand::CreateMBB(BB)); |
Chris Lattner | b2e806e | 2006-10-22 23:00:53 +0000 | [diff] [blame] | 1058 | } |
| 1059 | } |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1060 | |
Nate Begeman | f15485a | 2006-03-27 01:32:24 +0000 | [diff] [blame] | 1061 | // If we generated any switch lowering information, build and codegen any |
| 1062 | // additional DAGs necessary. |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1063 | for (unsigned i = 0, e = SDB->SwitchCases.size(); i != e; ++i) { |
Nate Begeman | f15485a | 2006-03-27 01:32:24 +0000 | [diff] [blame] | 1064 | // Set the current basic block to the mbb we wish to insert the code into |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1065 | MachineBasicBlock *ThisBB = BB = SDB->SwitchCases[i].ThisBB; |
| 1066 | SDB->setCurrentBasicBlock(BB); |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1067 | |
Nate Begeman | f15485a | 2006-03-27 01:32:24 +0000 | [diff] [blame] | 1068 | // Emit the code |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1069 | SDB->visitSwitchCase(SDB->SwitchCases[i]); |
| 1070 | CurDAG->setRoot(SDB->getRoot()); |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 1071 | CodeGenAndEmitDAG(); |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1072 | |
Chris Lattner | d5e93c0 | 2006-09-07 01:59:34 +0000 | [diff] [blame] | 1073 | // Handle any PHI nodes in successors of this chunk, as if we were coming |
| 1074 | // from the original BB before switch expansion. Note that PHI nodes can |
| 1075 | // occur multiple times in PHINodesToUpdate. We have to be very careful to |
| 1076 | // handle them the right number of times. |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1077 | while ((BB = SDB->SwitchCases[i].TrueBB)) { // Handle LHS and RHS. |
Evan Cheng | fb2e752 | 2009-09-18 21:02:19 +0000 | [diff] [blame] | 1078 | // If new BB's are created during scheduling, the edges may have been |
Evan Cheng | ce31910 | 2009-09-19 09:51:03 +0000 | [diff] [blame] | 1079 | // updated. That is, the edge from ThisBB to BB may have been split and |
| 1080 | // BB's predecessor is now another block. |
Evan Cheng | fb2e752 | 2009-09-18 21:02:19 +0000 | [diff] [blame] | 1081 | DenseMap<MachineBasicBlock*, MachineBasicBlock*>::iterator EI = |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1082 | SDB->EdgeMapping.find(BB); |
| 1083 | if (EI != SDB->EdgeMapping.end()) |
Evan Cheng | fb2e752 | 2009-09-18 21:02:19 +0000 | [diff] [blame] | 1084 | ThisBB = EI->second; |
Jakob Stoklund Olesen | dd437ba | 2010-01-11 21:02:33 +0000 | [diff] [blame] | 1085 | |
| 1086 | // BB may have been removed from the CFG if a branch was constant folded. |
| 1087 | if (ThisBB->isSuccessor(BB)) { |
| 1088 | for (MachineBasicBlock::iterator Phi = BB->begin(); |
Chris Lattner | 518bb53 | 2010-02-09 19:54:29 +0000 | [diff] [blame] | 1089 | Phi != BB->end() && Phi->isPHI(); |
Jakob Stoklund Olesen | dd437ba | 2010-01-11 21:02:33 +0000 | [diff] [blame] | 1090 | ++Phi) { |
| 1091 | // This value for this PHI node is recorded in PHINodesToUpdate. |
| 1092 | for (unsigned pn = 0; ; ++pn) { |
| 1093 | assert(pn != SDB->PHINodesToUpdate.size() && |
| 1094 | "Didn't find PHI entry!"); |
| 1095 | if (SDB->PHINodesToUpdate[pn].first == Phi) { |
| 1096 | Phi->addOperand(MachineOperand:: |
| 1097 | CreateReg(SDB->PHINodesToUpdate[pn].second, |
| 1098 | false)); |
| 1099 | Phi->addOperand(MachineOperand::CreateMBB(ThisBB)); |
| 1100 | break; |
| 1101 | } |
Evan Cheng | 8be58a1 | 2009-09-18 08:26:06 +0000 | [diff] [blame] | 1102 | } |
Chris Lattner | d5e93c0 | 2006-09-07 01:59:34 +0000 | [diff] [blame] | 1103 | } |
Nate Begeman | f15485a | 2006-03-27 01:32:24 +0000 | [diff] [blame] | 1104 | } |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1105 | |
Chris Lattner | d5e93c0 | 2006-09-07 01:59:34 +0000 | [diff] [blame] | 1106 | // Don't process RHS if same block as LHS. |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1107 | if (BB == SDB->SwitchCases[i].FalseBB) |
| 1108 | SDB->SwitchCases[i].FalseBB = 0; |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1109 | |
Chris Lattner | d5e93c0 | 2006-09-07 01:59:34 +0000 | [diff] [blame] | 1110 | // If we haven't handled the RHS, do so now. Otherwise, we're done. |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1111 | SDB->SwitchCases[i].TrueBB = SDB->SwitchCases[i].FalseBB; |
| 1112 | SDB->SwitchCases[i].FalseBB = 0; |
Nate Begeman | f15485a | 2006-03-27 01:32:24 +0000 | [diff] [blame] | 1113 | } |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1114 | assert(SDB->SwitchCases[i].TrueBB == 0 && SDB->SwitchCases[i].FalseBB == 0); |
| 1115 | SDB->clear(); |
Chris Lattner | a33ef48 | 2005-03-30 01:10:47 +0000 | [diff] [blame] | 1116 | } |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1117 | SDB->SwitchCases.clear(); |
Dan Gohman | 7c3234c | 2008-08-27 23:52:12 +0000 | [diff] [blame] | 1118 | |
Dan Gohman | 2048b85 | 2009-11-23 18:04:58 +0000 | [diff] [blame] | 1119 | SDB->PHINodesToUpdate.clear(); |
Chris Lattner | 1c08c71 | 2005-01-07 07:47:53 +0000 | [diff] [blame] | 1120 | } |
Evan Cheng | a9c2091 | 2006-01-21 02:32:06 +0000 | [diff] [blame] | 1121 | |
Jim Laskey | 13ec702 | 2006-08-01 14:21:23 +0000 | [diff] [blame] | 1122 | |
Dan Gohman | 0a3776d | 2009-02-06 18:26:51 +0000 | [diff] [blame] | 1123 | /// Create the scheduler. If a specific scheduler was specified |
| 1124 | /// via the SchedulerRegistry, use it, otherwise select the |
| 1125 | /// one preferred by the target. |
Dan Gohman | 5e84368 | 2008-07-14 18:19:29 +0000 | [diff] [blame] | 1126 | /// |
Dan Gohman | 47ac0f0 | 2009-02-11 04:27:20 +0000 | [diff] [blame] | 1127 | ScheduleDAGSDNodes *SelectionDAGISel::CreateScheduler() { |
Jim Laskey | eb577ba | 2006-08-02 12:30:23 +0000 | [diff] [blame] | 1128 | RegisterScheduler::FunctionPassCtor Ctor = RegisterScheduler::getDefault(); |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1129 | |
Jim Laskey | 13ec702 | 2006-08-01 14:21:23 +0000 | [diff] [blame] | 1130 | if (!Ctor) { |
Jim Laskey | eb577ba | 2006-08-02 12:30:23 +0000 | [diff] [blame] | 1131 | Ctor = ISHeuristic; |
Jim Laskey | 9373beb | 2006-08-01 19:14:14 +0000 | [diff] [blame] | 1132 | RegisterScheduler::setDefault(Ctor); |
Evan Cheng | 4ef1086 | 2006-01-23 07:01:07 +0000 | [diff] [blame] | 1133 | } |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1134 | |
Bill Wendling | be8cc2a | 2009-04-29 00:15:41 +0000 | [diff] [blame] | 1135 | return Ctor(this, OptLevel); |
Evan Cheng | a9c2091 | 2006-01-21 02:32:06 +0000 | [diff] [blame] | 1136 | } |
Chris Lattner | 0e43f2b | 2006-02-24 02:13:54 +0000 | [diff] [blame] | 1137 | |
Dan Gohman | fc54c55 | 2009-01-15 22:18:12 +0000 | [diff] [blame] | 1138 | ScheduleHazardRecognizer *SelectionDAGISel::CreateTargetHazardRecognizer() { |
| 1139 | return new ScheduleHazardRecognizer(); |
Jim Laskey | 9ff542f | 2006-08-01 18:29:48 +0000 | [diff] [blame] | 1140 | } |
| 1141 | |
Chris Lattner | 7554806 | 2006-10-11 03:58:02 +0000 | [diff] [blame] | 1142 | //===----------------------------------------------------------------------===// |
| 1143 | // Helper functions used by the generated instruction selector. |
| 1144 | //===----------------------------------------------------------------------===// |
| 1145 | // Calls to these methods are generated by tblgen. |
| 1146 | |
| 1147 | /// CheckAndMask - The isel is trying to match something like (and X, 255). If |
| 1148 | /// the dag combiner simplified the 255, we still want to match. RHS is the |
| 1149 | /// actual value in the DAG on the RHS of an AND, and DesiredMaskS is the value |
| 1150 | /// specified in the .td file (e.g. 255). |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1151 | bool SelectionDAGISel::CheckAndMask(SDValue LHS, ConstantSDNode *RHS, |
Dan Gohman | dc9b3d0 | 2007-07-24 23:00:27 +0000 | [diff] [blame] | 1152 | int64_t DesiredMaskS) const { |
Dan Gohman | 2e68b6f | 2008-02-25 21:11:39 +0000 | [diff] [blame] | 1153 | const APInt &ActualMask = RHS->getAPIntValue(); |
| 1154 | const APInt &DesiredMask = APInt(LHS.getValueSizeInBits(), DesiredMaskS); |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1155 | |
Chris Lattner | 7554806 | 2006-10-11 03:58:02 +0000 | [diff] [blame] | 1156 | // If the actual mask exactly matches, success! |
| 1157 | if (ActualMask == DesiredMask) |
| 1158 | return true; |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1159 | |
Chris Lattner | 7554806 | 2006-10-11 03:58:02 +0000 | [diff] [blame] | 1160 | // If the actual AND mask is allowing unallowed bits, this doesn't match. |
Dan Gohman | 2e68b6f | 2008-02-25 21:11:39 +0000 | [diff] [blame] | 1161 | if (ActualMask.intersects(~DesiredMask)) |
Chris Lattner | 7554806 | 2006-10-11 03:58:02 +0000 | [diff] [blame] | 1162 | return false; |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1163 | |
Chris Lattner | 7554806 | 2006-10-11 03:58:02 +0000 | [diff] [blame] | 1164 | // Otherwise, the DAG Combiner may have proven that the value coming in is |
| 1165 | // either already zero or is not demanded. Check for known zero input bits. |
Dan Gohman | 2e68b6f | 2008-02-25 21:11:39 +0000 | [diff] [blame] | 1166 | APInt NeededMask = DesiredMask & ~ActualMask; |
Dan Gohman | ea859be | 2007-06-22 14:59:07 +0000 | [diff] [blame] | 1167 | if (CurDAG->MaskedValueIsZero(LHS, NeededMask)) |
Chris Lattner | 7554806 | 2006-10-11 03:58:02 +0000 | [diff] [blame] | 1168 | return true; |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1169 | |
Chris Lattner | 7554806 | 2006-10-11 03:58:02 +0000 | [diff] [blame] | 1170 | // TODO: check to see if missing bits are just not demanded. |
| 1171 | |
| 1172 | // Otherwise, this pattern doesn't match. |
| 1173 | return false; |
| 1174 | } |
| 1175 | |
| 1176 | /// CheckOrMask - The isel is trying to match something like (or X, 255). If |
| 1177 | /// the dag combiner simplified the 255, we still want to match. RHS is the |
| 1178 | /// actual value in the DAG on the RHS of an OR, and DesiredMaskS is the value |
| 1179 | /// specified in the .td file (e.g. 255). |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1180 | bool SelectionDAGISel::CheckOrMask(SDValue LHS, ConstantSDNode *RHS, |
Dan Gohman | 2e68b6f | 2008-02-25 21:11:39 +0000 | [diff] [blame] | 1181 | int64_t DesiredMaskS) const { |
| 1182 | const APInt &ActualMask = RHS->getAPIntValue(); |
| 1183 | const APInt &DesiredMask = APInt(LHS.getValueSizeInBits(), DesiredMaskS); |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1184 | |
Chris Lattner | 7554806 | 2006-10-11 03:58:02 +0000 | [diff] [blame] | 1185 | // If the actual mask exactly matches, success! |
| 1186 | if (ActualMask == DesiredMask) |
| 1187 | return true; |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1188 | |
Chris Lattner | 7554806 | 2006-10-11 03:58:02 +0000 | [diff] [blame] | 1189 | // If the actual AND mask is allowing unallowed bits, this doesn't match. |
Dan Gohman | 2e68b6f | 2008-02-25 21:11:39 +0000 | [diff] [blame] | 1190 | if (ActualMask.intersects(~DesiredMask)) |
Chris Lattner | 7554806 | 2006-10-11 03:58:02 +0000 | [diff] [blame] | 1191 | return false; |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1192 | |
Chris Lattner | 7554806 | 2006-10-11 03:58:02 +0000 | [diff] [blame] | 1193 | // Otherwise, the DAG Combiner may have proven that the value coming in is |
| 1194 | // either already zero or is not demanded. Check for known zero input bits. |
Dan Gohman | 2e68b6f | 2008-02-25 21:11:39 +0000 | [diff] [blame] | 1195 | APInt NeededMask = DesiredMask & ~ActualMask; |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1196 | |
Dan Gohman | 2e68b6f | 2008-02-25 21:11:39 +0000 | [diff] [blame] | 1197 | APInt KnownZero, KnownOne; |
Dan Gohman | ea859be | 2007-06-22 14:59:07 +0000 | [diff] [blame] | 1198 | CurDAG->ComputeMaskedBits(LHS, NeededMask, KnownZero, KnownOne); |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1199 | |
Chris Lattner | 7554806 | 2006-10-11 03:58:02 +0000 | [diff] [blame] | 1200 | // If all the missing bits in the or are already known to be set, match! |
| 1201 | if ((NeededMask & KnownOne) == NeededMask) |
| 1202 | return true; |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1203 | |
Chris Lattner | 7554806 | 2006-10-11 03:58:02 +0000 | [diff] [blame] | 1204 | // TODO: check to see if missing bits are just not demanded. |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1205 | |
Chris Lattner | 7554806 | 2006-10-11 03:58:02 +0000 | [diff] [blame] | 1206 | // Otherwise, this pattern doesn't match. |
| 1207 | return false; |
| 1208 | } |
| 1209 | |
Jim Laskey | 9ff542f | 2006-08-01 18:29:48 +0000 | [diff] [blame] | 1210 | |
Chris Lattner | 0e43f2b | 2006-02-24 02:13:54 +0000 | [diff] [blame] | 1211 | /// SelectInlineAsmMemoryOperands - Calls to this are automatically generated |
| 1212 | /// by tblgen. Others should not call it. |
| 1213 | void SelectionDAGISel:: |
Dan Gohman | f350b27 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 1214 | SelectInlineAsmMemoryOperands(std::vector<SDValue> &Ops) { |
Dan Gohman | 475871a | 2008-07-27 21:46:04 +0000 | [diff] [blame] | 1215 | std::vector<SDValue> InOps; |
Chris Lattner | 0e43f2b | 2006-02-24 02:13:54 +0000 | [diff] [blame] | 1216 | std::swap(InOps, Ops); |
| 1217 | |
Chris Lattner | decc267 | 2010-04-07 05:20:54 +0000 | [diff] [blame] | 1218 | Ops.push_back(InOps[InlineAsm::Op_InputChain]); // 0 |
| 1219 | Ops.push_back(InOps[InlineAsm::Op_AsmString]); // 1 |
| 1220 | Ops.push_back(InOps[InlineAsm::Op_MDNode]); // 2, !srcloc |
Chris Lattner | 0e43f2b | 2006-02-24 02:13:54 +0000 | [diff] [blame] | 1221 | |
Chris Lattner | decc267 | 2010-04-07 05:20:54 +0000 | [diff] [blame] | 1222 | unsigned i = InlineAsm::Op_FirstOperand, e = InOps.size(); |
Owen Anderson | 825b72b | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 1223 | if (InOps[e-1].getValueType() == MVT::Flag) |
Chris Lattner | 0e43f2b | 2006-02-24 02:13:54 +0000 | [diff] [blame] | 1224 | --e; // Don't process a flag operand if it is here. |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1225 | |
Chris Lattner | 0e43f2b | 2006-02-24 02:13:54 +0000 | [diff] [blame] | 1226 | while (i != e) { |
Dan Gohman | f5aeb1a | 2008-09-12 16:56:44 +0000 | [diff] [blame] | 1227 | unsigned Flags = cast<ConstantSDNode>(InOps[i])->getZExtValue(); |
Chris Lattner | decc267 | 2010-04-07 05:20:54 +0000 | [diff] [blame] | 1228 | if (!InlineAsm::isMemKind(Flags)) { |
Chris Lattner | 0e43f2b | 2006-02-24 02:13:54 +0000 | [diff] [blame] | 1229 | // Just skip over this operand, copying the operands verbatim. |
Evan Cheng | 697cbbf | 2009-03-20 18:03:34 +0000 | [diff] [blame] | 1230 | Ops.insert(Ops.end(), InOps.begin()+i, |
| 1231 | InOps.begin()+i+InlineAsm::getNumOperandRegisters(Flags) + 1); |
| 1232 | i += InlineAsm::getNumOperandRegisters(Flags) + 1; |
Chris Lattner | 0e43f2b | 2006-02-24 02:13:54 +0000 | [diff] [blame] | 1233 | } else { |
Evan Cheng | 697cbbf | 2009-03-20 18:03:34 +0000 | [diff] [blame] | 1234 | assert(InlineAsm::getNumOperandRegisters(Flags) == 1 && |
| 1235 | "Memory operand with multiple values?"); |
Chris Lattner | 0e43f2b | 2006-02-24 02:13:54 +0000 | [diff] [blame] | 1236 | // Otherwise, this is a memory operand. Ask the target to select it. |
Dan Gohman | 475871a | 2008-07-27 21:46:04 +0000 | [diff] [blame] | 1237 | std::vector<SDValue> SelOps; |
Chris Lattner | decc267 | 2010-04-07 05:20:54 +0000 | [diff] [blame] | 1238 | if (SelectInlineAsmMemoryOperand(InOps[i+1], 'm', SelOps)) |
Chris Lattner | 75361b6 | 2010-04-07 22:58:41 +0000 | [diff] [blame] | 1239 | report_fatal_error("Could not match memory address. Inline asm" |
Chris Lattner | 87d677c | 2010-04-07 23:50:38 +0000 | [diff] [blame] | 1240 | " failure!"); |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1241 | |
Chris Lattner | 0e43f2b | 2006-02-24 02:13:54 +0000 | [diff] [blame] | 1242 | // Add this to the output node. |
Chris Lattner | decc267 | 2010-04-07 05:20:54 +0000 | [diff] [blame] | 1243 | unsigned NewFlags = |
| 1244 | InlineAsm::getFlagWord(InlineAsm::Kind_Mem, SelOps.size()); |
| 1245 | Ops.push_back(CurDAG->getTargetConstant(NewFlags, MVT::i32)); |
Chris Lattner | 0e43f2b | 2006-02-24 02:13:54 +0000 | [diff] [blame] | 1246 | Ops.insert(Ops.end(), SelOps.begin(), SelOps.end()); |
| 1247 | i += 2; |
| 1248 | } |
| 1249 | } |
Daniel Dunbar | a279bc3 | 2009-09-20 02:20:51 +0000 | [diff] [blame] | 1250 | |
Chris Lattner | 0e43f2b | 2006-02-24 02:13:54 +0000 | [diff] [blame] | 1251 | // Add the flag input back if present. |
| 1252 | if (e != InOps.size()) |
| 1253 | Ops.push_back(InOps.back()); |
| 1254 | } |
Devang Patel | 794fd75 | 2007-05-01 21:15:47 +0000 | [diff] [blame] | 1255 | |
Owen Anderson | e50ed30 | 2009-08-10 22:56:29 +0000 | [diff] [blame] | 1256 | /// findFlagUse - Return use of EVT::Flag value produced by the specified |
Anton Korobeynikov | c1c6ef8 | 2009-05-08 18:51:58 +0000 | [diff] [blame] | 1257 | /// SDNode. |
| 1258 | /// |
| 1259 | static SDNode *findFlagUse(SDNode *N) { |
| 1260 | unsigned FlagResNo = N->getNumValues()-1; |
| 1261 | for (SDNode::use_iterator I = N->use_begin(), E = N->use_end(); I != E; ++I) { |
| 1262 | SDUse &Use = I.getUse(); |
| 1263 | if (Use.getResNo() == FlagResNo) |
| 1264 | return Use.getUser(); |
| 1265 | } |
| 1266 | return NULL; |
| 1267 | } |
| 1268 | |
| 1269 | /// findNonImmUse - Return true if "Use" is a non-immediate use of "Def". |
| 1270 | /// This function recursively traverses up the operand chain, ignoring |
| 1271 | /// certain nodes. |
| 1272 | static bool findNonImmUse(SDNode *Use, SDNode* Def, SDNode *ImmedUse, |
Chris Lattner | d1b7382 | 2010-03-02 22:20:06 +0000 | [diff] [blame] | 1273 | SDNode *Root, SmallPtrSet<SDNode*, 16> &Visited, |
| 1274 | bool IgnoreChains) { |
Chris Lattner | da244a0 | 2010-02-23 19:32:27 +0000 | [diff] [blame] | 1275 | // The NodeID's are given uniques ID's where a node ID is guaranteed to be |
| 1276 | // greater than all of its (recursive) operands. If we scan to a point where |
| 1277 | // 'use' is smaller than the node we're scanning for, then we know we will |
| 1278 | // never find it. |
| 1279 | // |
| 1280 | // The Use may be -1 (unassigned) if it is a newly allocated node. This can |
| 1281 | // happen because we scan down to newly selected nodes in the case of flag |
| 1282 | // uses. |
| 1283 | if ((Use->getNodeId() < Def->getNodeId() && Use->getNodeId() != -1)) |
| 1284 | return false; |
| 1285 | |
| 1286 | // Don't revisit nodes if we already scanned it and didn't fail, we know we |
| 1287 | // won't fail if we scan it again. |
| 1288 | if (!Visited.insert(Use)) |
Anton Korobeynikov | c1c6ef8 | 2009-05-08 18:51:58 +0000 | [diff] [blame] | 1289 | return false; |
| 1290 | |
| 1291 | for (unsigned i = 0, e = Use->getNumOperands(); i != e; ++i) { |
Chris Lattner | d1b7382 | 2010-03-02 22:20:06 +0000 | [diff] [blame] | 1292 | // Ignore chain uses, they are validated by HandleMergeInputChains. |
| 1293 | if (Use->getOperand(i).getValueType() == MVT::Other && IgnoreChains) |
| 1294 | continue; |
| 1295 | |
Anton Korobeynikov | c1c6ef8 | 2009-05-08 18:51:58 +0000 | [diff] [blame] | 1296 | SDNode *N = Use->getOperand(i).getNode(); |
| 1297 | if (N == Def) { |
| 1298 | if (Use == ImmedUse || Use == Root) |
| 1299 | continue; // We are not looking for immediate use. |
| 1300 | assert(N != Root); |
| 1301 | return true; |
| 1302 | } |
| 1303 | |
| 1304 | // Traverse up the operand chain. |
Chris Lattner | d1b7382 | 2010-03-02 22:20:06 +0000 | [diff] [blame] | 1305 | if (findNonImmUse(N, Def, ImmedUse, Root, Visited, IgnoreChains)) |
Anton Korobeynikov | c1c6ef8 | 2009-05-08 18:51:58 +0000 | [diff] [blame] | 1306 | return true; |
| 1307 | } |
| 1308 | return false; |
| 1309 | } |
| 1310 | |
Evan Cheng | 014bf21 | 2010-02-15 19:41:07 +0000 | [diff] [blame] | 1311 | /// IsProfitableToFold - Returns true if it's profitable to fold the specific |
| 1312 | /// operand node N of U during instruction selection that starts at Root. |
| 1313 | bool SelectionDAGISel::IsProfitableToFold(SDValue N, SDNode *U, |
| 1314 | SDNode *Root) const { |
| 1315 | if (OptLevel == CodeGenOpt::None) return false; |
| 1316 | return N.hasOneUse(); |
| 1317 | } |
| 1318 | |
| 1319 | /// IsLegalToFold - Returns true if the specific operand node N of |
| 1320 | /// U can be folded during instruction selection that starts at Root. |
Chris Lattner | d1b7382 | 2010-03-02 22:20:06 +0000 | [diff] [blame] | 1321 | bool SelectionDAGISel::IsLegalToFold(SDValue N, SDNode *U, SDNode *Root, |
| 1322 | bool IgnoreChains) const { |
Anton Korobeynikov | c1c6ef8 | 2009-05-08 18:51:58 +0000 | [diff] [blame] | 1323 | if (OptLevel == CodeGenOpt::None) return false; |
| 1324 | |
| 1325 | // If Root use can somehow reach N through a path that that doesn't contain |
| 1326 | // U then folding N would create a cycle. e.g. In the following |
| 1327 | // diagram, Root can reach N through X. If N is folded into into Root, then |
| 1328 | // X is both a predecessor and a successor of U. |
| 1329 | // |
| 1330 | // [N*] // |
| 1331 | // ^ ^ // |
| 1332 | // / \ // |
| 1333 | // [U*] [X]? // |
| 1334 | // ^ ^ // |
| 1335 | // \ / // |
| 1336 | // \ / // |
| 1337 | // [Root*] // |
| 1338 | // |
| 1339 | // * indicates nodes to be folded together. |
| 1340 | // |
| 1341 | // If Root produces a flag, then it gets (even more) interesting. Since it |
| 1342 | // will be "glued" together with its flag use in the scheduler, we need to |
| 1343 | // check if it might reach N. |
| 1344 | // |
| 1345 | // [N*] // |
| 1346 | // ^ ^ // |
| 1347 | // / \ // |
| 1348 | // [U*] [X]? // |
| 1349 | // ^ ^ // |
| 1350 | // \ \ // |
| 1351 | // \ | // |
| 1352 | // [Root*] | // |
| 1353 | // ^ | // |
| 1354 | // f | // |
| 1355 | // | / // |
| 1356 | // [Y] / // |
| 1357 | // ^ / // |
| 1358 | // f / // |
| 1359 | // | / // |
| 1360 | // [FU] // |
| 1361 | // |
| 1362 | // If FU (flag use) indirectly reaches N (the load), and Root folds N |
| 1363 | // (call it Fold), then X is a predecessor of FU and a successor of |
| 1364 | // Fold. But since Fold and FU are flagged together, this will create |
| 1365 | // a cycle in the scheduling graph. |
| 1366 | |
Chris Lattner | 9bbcd5e | 2010-03-05 05:49:45 +0000 | [diff] [blame] | 1367 | // If the node has flags, walk down the graph to the "lowest" node in the |
| 1368 | // flagged set. |
Owen Anderson | e50ed30 | 2009-08-10 22:56:29 +0000 | [diff] [blame] | 1369 | EVT VT = Root->getValueType(Root->getNumValues()-1); |
Owen Anderson | 825b72b | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 1370 | while (VT == MVT::Flag) { |
Anton Korobeynikov | c1c6ef8 | 2009-05-08 18:51:58 +0000 | [diff] [blame] | 1371 | SDNode *FU = findFlagUse(Root); |
| 1372 | if (FU == NULL) |
| 1373 | break; |
| 1374 | Root = FU; |
| 1375 | VT = Root->getValueType(Root->getNumValues()-1); |
Chris Lattner | 18fdaba | 2010-03-05 06:19:13 +0000 | [diff] [blame] | 1376 | |
| 1377 | // If our query node has a flag result with a use, we've walked up it. If |
| 1378 | // the user (which has already been selected) has a chain or indirectly uses |
| 1379 | // the chain, our WalkChainUsers predicate will not consider it. Because of |
| 1380 | // this, we cannot ignore chains in this predicate. |
| 1381 | IgnoreChains = false; |
Anton Korobeynikov | c1c6ef8 | 2009-05-08 18:51:58 +0000 | [diff] [blame] | 1382 | } |
Chris Lattner | 18fdaba | 2010-03-05 06:19:13 +0000 | [diff] [blame] | 1383 | |
Anton Korobeynikov | c1c6ef8 | 2009-05-08 18:51:58 +0000 | [diff] [blame] | 1384 | |
Chris Lattner | 9bbcd5e | 2010-03-05 05:49:45 +0000 | [diff] [blame] | 1385 | SmallPtrSet<SDNode*, 16> Visited; |
| 1386 | return !findNonImmUse(Root, N.getNode(), U, Root, Visited, IgnoreChains); |
Anton Korobeynikov | c1c6ef8 | 2009-05-08 18:51:58 +0000 | [diff] [blame] | 1387 | } |
| 1388 | |
Dan Gohman | eeb3a00 | 2010-01-05 01:24:18 +0000 | [diff] [blame] | 1389 | SDNode *SelectionDAGISel::Select_INLINEASM(SDNode *N) { |
| 1390 | std::vector<SDValue> Ops(N->op_begin(), N->op_end()); |
Dan Gohman | e1f188f | 2009-10-29 22:30:23 +0000 | [diff] [blame] | 1391 | SelectInlineAsmMemoryOperands(Ops); |
| 1392 | |
| 1393 | std::vector<EVT> VTs; |
| 1394 | VTs.push_back(MVT::Other); |
| 1395 | VTs.push_back(MVT::Flag); |
Dan Gohman | eeb3a00 | 2010-01-05 01:24:18 +0000 | [diff] [blame] | 1396 | SDValue New = CurDAG->getNode(ISD::INLINEASM, N->getDebugLoc(), |
Dan Gohman | e1f188f | 2009-10-29 22:30:23 +0000 | [diff] [blame] | 1397 | VTs, &Ops[0], Ops.size()); |
Chris Lattner | d1b7382 | 2010-03-02 22:20:06 +0000 | [diff] [blame] | 1398 | New->setNodeId(-1); |
Dan Gohman | e1f188f | 2009-10-29 22:30:23 +0000 | [diff] [blame] | 1399 | return New.getNode(); |
| 1400 | } |
| 1401 | |
Dan Gohman | eeb3a00 | 2010-01-05 01:24:18 +0000 | [diff] [blame] | 1402 | SDNode *SelectionDAGISel::Select_UNDEF(SDNode *N) { |
Chris Lattner | 518bb53 | 2010-02-09 19:54:29 +0000 | [diff] [blame] | 1403 | return CurDAG->SelectNodeTo(N, TargetOpcode::IMPLICIT_DEF,N->getValueType(0)); |
Dan Gohman | e1f188f | 2009-10-29 22:30:23 +0000 | [diff] [blame] | 1404 | } |
| 1405 | |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 1406 | /// GetVBR - decode a vbr encoding whose top bit is set. |
| 1407 | ALWAYS_INLINE static uint64_t |
| 1408 | GetVBR(uint64_t Val, const unsigned char *MatcherTable, unsigned &Idx) { |
| 1409 | assert(Val >= 128 && "Not a VBR"); |
| 1410 | Val &= 127; // Remove first vbr bit. |
| 1411 | |
| 1412 | unsigned Shift = 7; |
| 1413 | uint64_t NextBits; |
| 1414 | do { |
Chris Lattner | 14df8dc | 2010-02-28 22:38:43 +0000 | [diff] [blame] | 1415 | NextBits = MatcherTable[Idx++]; |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 1416 | Val |= (NextBits&127) << Shift; |
| 1417 | Shift += 7; |
| 1418 | } while (NextBits & 128); |
| 1419 | |
| 1420 | return Val; |
| 1421 | } |
| 1422 | |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 1423 | |
| 1424 | /// UpdateChainsAndFlags - When a match is complete, this method updates uses of |
| 1425 | /// interior flag and chain results to use the new flag and chain results. |
Chris Lattner | 82dd3d3 | 2010-03-02 07:50:03 +0000 | [diff] [blame] | 1426 | void SelectionDAGISel:: |
| 1427 | UpdateChainsAndFlags(SDNode *NodeToMatch, SDValue InputChain, |
| 1428 | const SmallVectorImpl<SDNode*> &ChainNodesMatched, |
| 1429 | SDValue InputFlag, |
| 1430 | const SmallVectorImpl<SDNode*> &FlagResultNodesMatched, |
| 1431 | bool isMorphNodeTo) { |
| 1432 | SmallVector<SDNode*, 4> NowDeadNodes; |
| 1433 | |
| 1434 | ISelUpdater ISU(ISelPosition); |
| 1435 | |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 1436 | // Now that all the normal results are replaced, we replace the chain and |
| 1437 | // flag results if present. |
| 1438 | if (!ChainNodesMatched.empty()) { |
| 1439 | assert(InputChain.getNode() != 0 && |
| 1440 | "Matched input chains but didn't produce a chain"); |
| 1441 | // Loop over all of the nodes we matched that produced a chain result. |
| 1442 | // Replace all the chain results with the final chain we ended up with. |
| 1443 | for (unsigned i = 0, e = ChainNodesMatched.size(); i != e; ++i) { |
| 1444 | SDNode *ChainNode = ChainNodesMatched[i]; |
| 1445 | |
Chris Lattner | 82dd3d3 | 2010-03-02 07:50:03 +0000 | [diff] [blame] | 1446 | // If this node was already deleted, don't look at it. |
| 1447 | if (ChainNode->getOpcode() == ISD::DELETED_NODE) |
| 1448 | continue; |
| 1449 | |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 1450 | // Don't replace the results of the root node if we're doing a |
| 1451 | // MorphNodeTo. |
| 1452 | if (ChainNode == NodeToMatch && isMorphNodeTo) |
| 1453 | continue; |
| 1454 | |
| 1455 | SDValue ChainVal = SDValue(ChainNode, ChainNode->getNumValues()-1); |
| 1456 | if (ChainVal.getValueType() == MVT::Flag) |
| 1457 | ChainVal = ChainVal.getValue(ChainVal->getNumValues()-2); |
| 1458 | assert(ChainVal.getValueType() == MVT::Other && "Not a chain?"); |
Chris Lattner | 82dd3d3 | 2010-03-02 07:50:03 +0000 | [diff] [blame] | 1459 | CurDAG->ReplaceAllUsesOfValueWith(ChainVal, InputChain, &ISU); |
| 1460 | |
Chris Lattner | 856fb39 | 2010-03-28 05:28:31 +0000 | [diff] [blame] | 1461 | // If the node became dead and we haven't already seen it, delete it. |
| 1462 | if (ChainNode->use_empty() && |
| 1463 | !std::count(NowDeadNodes.begin(), NowDeadNodes.end(), ChainNode)) |
Chris Lattner | 82dd3d3 | 2010-03-02 07:50:03 +0000 | [diff] [blame] | 1464 | NowDeadNodes.push_back(ChainNode); |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 1465 | } |
| 1466 | } |
| 1467 | |
| 1468 | // If the result produces a flag, update any flag results in the matched |
| 1469 | // pattern with the flag result. |
| 1470 | if (InputFlag.getNode() != 0) { |
| 1471 | // Handle any interior nodes explicitly marked. |
| 1472 | for (unsigned i = 0, e = FlagResultNodesMatched.size(); i != e; ++i) { |
| 1473 | SDNode *FRN = FlagResultNodesMatched[i]; |
Chris Lattner | 82dd3d3 | 2010-03-02 07:50:03 +0000 | [diff] [blame] | 1474 | |
| 1475 | // If this node was already deleted, don't look at it. |
| 1476 | if (FRN->getOpcode() == ISD::DELETED_NODE) |
| 1477 | continue; |
| 1478 | |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 1479 | assert(FRN->getValueType(FRN->getNumValues()-1) == MVT::Flag && |
| 1480 | "Doesn't have a flag result"); |
| 1481 | CurDAG->ReplaceAllUsesOfValueWith(SDValue(FRN, FRN->getNumValues()-1), |
Chris Lattner | 82dd3d3 | 2010-03-02 07:50:03 +0000 | [diff] [blame] | 1482 | InputFlag, &ISU); |
| 1483 | |
Chris Lattner | 19e37cb | 2010-03-28 04:54:33 +0000 | [diff] [blame] | 1484 | // If the node became dead and we haven't already seen it, delete it. |
| 1485 | if (FRN->use_empty() && |
| 1486 | !std::count(NowDeadNodes.begin(), NowDeadNodes.end(), FRN)) |
Chris Lattner | 82dd3d3 | 2010-03-02 07:50:03 +0000 | [diff] [blame] | 1487 | NowDeadNodes.push_back(FRN); |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 1488 | } |
| 1489 | } |
| 1490 | |
Chris Lattner | 82dd3d3 | 2010-03-02 07:50:03 +0000 | [diff] [blame] | 1491 | if (!NowDeadNodes.empty()) |
| 1492 | CurDAG->RemoveDeadNodes(NowDeadNodes, &ISU); |
| 1493 | |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 1494 | DEBUG(errs() << "ISEL: Match complete!\n"); |
| 1495 | } |
| 1496 | |
Chris Lattner | c6d7ad3 | 2010-03-02 02:22:10 +0000 | [diff] [blame] | 1497 | enum ChainResult { |
| 1498 | CR_Simple, |
| 1499 | CR_InducesCycle, |
| 1500 | CR_LeadsToInteriorNode |
| 1501 | }; |
| 1502 | |
| 1503 | /// WalkChainUsers - Walk down the users of the specified chained node that is |
| 1504 | /// part of the pattern we're matching, looking at all of the users we find. |
| 1505 | /// This determines whether something is an interior node, whether we have a |
| 1506 | /// non-pattern node in between two pattern nodes (which prevent folding because |
| 1507 | /// it would induce a cycle) and whether we have a TokenFactor node sandwiched |
| 1508 | /// between pattern nodes (in which case the TF becomes part of the pattern). |
| 1509 | /// |
| 1510 | /// The walk we do here is guaranteed to be small because we quickly get down to |
| 1511 | /// already selected nodes "below" us. |
| 1512 | static ChainResult |
| 1513 | WalkChainUsers(SDNode *ChainedNode, |
| 1514 | SmallVectorImpl<SDNode*> &ChainedNodesInPattern, |
| 1515 | SmallVectorImpl<SDNode*> &InteriorChainedNodes) { |
| 1516 | ChainResult Result = CR_Simple; |
| 1517 | |
| 1518 | for (SDNode::use_iterator UI = ChainedNode->use_begin(), |
| 1519 | E = ChainedNode->use_end(); UI != E; ++UI) { |
| 1520 | // Make sure the use is of the chain, not some other value we produce. |
| 1521 | if (UI.getUse().getValueType() != MVT::Other) continue; |
| 1522 | |
| 1523 | SDNode *User = *UI; |
| 1524 | |
| 1525 | // If we see an already-selected machine node, then we've gone beyond the |
| 1526 | // pattern that we're selecting down into the already selected chunk of the |
| 1527 | // DAG. |
| 1528 | if (User->isMachineOpcode() || |
Chris Lattner | c6d7ad3 | 2010-03-02 02:22:10 +0000 | [diff] [blame] | 1529 | User->getOpcode() == ISD::HANDLENODE) // Root of the graph. |
| 1530 | continue; |
Chris Lattner | d1b7382 | 2010-03-02 22:20:06 +0000 | [diff] [blame] | 1531 | |
| 1532 | if (User->getOpcode() == ISD::CopyToReg || |
| 1533 | User->getOpcode() == ISD::CopyFromReg || |
Chris Lattner | 7561d48 | 2010-03-14 02:33:54 +0000 | [diff] [blame] | 1534 | User->getOpcode() == ISD::INLINEASM || |
| 1535 | User->getOpcode() == ISD::EH_LABEL) { |
Chris Lattner | d1b7382 | 2010-03-02 22:20:06 +0000 | [diff] [blame] | 1536 | // If their node ID got reset to -1 then they've already been selected. |
| 1537 | // Treat them like a MachineOpcode. |
| 1538 | if (User->getNodeId() == -1) |
| 1539 | continue; |
| 1540 | } |
Chris Lattner | c6d7ad3 | 2010-03-02 02:22:10 +0000 | [diff] [blame] | 1541 | |
| 1542 | // If we have a TokenFactor, we handle it specially. |
| 1543 | if (User->getOpcode() != ISD::TokenFactor) { |
| 1544 | // If the node isn't a token factor and isn't part of our pattern, then it |
| 1545 | // must be a random chained node in between two nodes we're selecting. |
| 1546 | // This happens when we have something like: |
| 1547 | // x = load ptr |
| 1548 | // call |
| 1549 | // y = x+4 |
| 1550 | // store y -> ptr |
| 1551 | // Because we structurally match the load/store as a read/modify/write, |
| 1552 | // but the call is chained between them. We cannot fold in this case |
| 1553 | // because it would induce a cycle in the graph. |
| 1554 | if (!std::count(ChainedNodesInPattern.begin(), |
| 1555 | ChainedNodesInPattern.end(), User)) |
| 1556 | return CR_InducesCycle; |
| 1557 | |
| 1558 | // Otherwise we found a node that is part of our pattern. For example in: |
| 1559 | // x = load ptr |
| 1560 | // y = x+4 |
| 1561 | // store y -> ptr |
| 1562 | // This would happen when we're scanning down from the load and see the |
| 1563 | // store as a user. Record that there is a use of ChainedNode that is |
| 1564 | // part of the pattern and keep scanning uses. |
| 1565 | Result = CR_LeadsToInteriorNode; |
| 1566 | InteriorChainedNodes.push_back(User); |
| 1567 | continue; |
| 1568 | } |
| 1569 | |
| 1570 | // If we found a TokenFactor, there are two cases to consider: first if the |
| 1571 | // TokenFactor is just hanging "below" the pattern we're matching (i.e. no |
| 1572 | // uses of the TF are in our pattern) we just want to ignore it. Second, |
| 1573 | // the TokenFactor can be sandwiched in between two chained nodes, like so: |
| 1574 | // [Load chain] |
| 1575 | // ^ |
| 1576 | // | |
| 1577 | // [Load] |
| 1578 | // ^ ^ |
| 1579 | // | \ DAG's like cheese |
| 1580 | // / \ do you? |
| 1581 | // / | |
| 1582 | // [TokenFactor] [Op] |
| 1583 | // ^ ^ |
| 1584 | // | | |
| 1585 | // \ / |
| 1586 | // \ / |
| 1587 | // [Store] |
| 1588 | // |
| 1589 | // In this case, the TokenFactor becomes part of our match and we rewrite it |
| 1590 | // as a new TokenFactor. |
| 1591 | // |
| 1592 | // To distinguish these two cases, do a recursive walk down the uses. |
| 1593 | switch (WalkChainUsers(User, ChainedNodesInPattern, InteriorChainedNodes)) { |
| 1594 | case CR_Simple: |
| 1595 | // If the uses of the TokenFactor are just already-selected nodes, ignore |
| 1596 | // it, it is "below" our pattern. |
| 1597 | continue; |
| 1598 | case CR_InducesCycle: |
| 1599 | // If the uses of the TokenFactor lead to nodes that are not part of our |
| 1600 | // pattern that are not selected, folding would turn this into a cycle, |
| 1601 | // bail out now. |
| 1602 | return CR_InducesCycle; |
| 1603 | case CR_LeadsToInteriorNode: |
| 1604 | break; // Otherwise, keep processing. |
| 1605 | } |
| 1606 | |
| 1607 | // Okay, we know we're in the interesting interior case. The TokenFactor |
| 1608 | // is now going to be considered part of the pattern so that we rewrite its |
| 1609 | // uses (it may have uses that are not part of the pattern) with the |
| 1610 | // ultimate chain result of the generated code. We will also add its chain |
| 1611 | // inputs as inputs to the ultimate TokenFactor we create. |
| 1612 | Result = CR_LeadsToInteriorNode; |
| 1613 | ChainedNodesInPattern.push_back(User); |
| 1614 | InteriorChainedNodes.push_back(User); |
| 1615 | continue; |
| 1616 | } |
| 1617 | |
| 1618 | return Result; |
| 1619 | } |
| 1620 | |
Chris Lattner | 6b30792 | 2010-03-02 00:00:03 +0000 | [diff] [blame] | 1621 | /// HandleMergeInputChains - This implements the OPC_EmitMergeInputChains |
Chris Lattner | 00592ec | 2010-03-02 19:34:59 +0000 | [diff] [blame] | 1622 | /// operation for when the pattern matched at least one node with a chains. The |
Chris Lattner | c6d7ad3 | 2010-03-02 02:22:10 +0000 | [diff] [blame] | 1623 | /// input vector contains a list of all of the chained nodes that we match. We |
| 1624 | /// must determine if this is a valid thing to cover (i.e. matching it won't |
| 1625 | /// induce cycles in the DAG) and if so, creating a TokenFactor node. that will |
| 1626 | /// be used as the input node chain for the generated nodes. |
Chris Lattner | 6b30792 | 2010-03-02 00:00:03 +0000 | [diff] [blame] | 1627 | static SDValue |
Chris Lattner | c6d7ad3 | 2010-03-02 02:22:10 +0000 | [diff] [blame] | 1628 | HandleMergeInputChains(SmallVectorImpl<SDNode*> &ChainNodesMatched, |
Chris Lattner | 6b30792 | 2010-03-02 00:00:03 +0000 | [diff] [blame] | 1629 | SelectionDAG *CurDAG) { |
Chris Lattner | c6d7ad3 | 2010-03-02 02:22:10 +0000 | [diff] [blame] | 1630 | // Walk all of the chained nodes we've matched, recursively scanning down the |
| 1631 | // users of the chain result. This adds any TokenFactor nodes that are caught |
| 1632 | // in between chained nodes to the chained and interior nodes list. |
| 1633 | SmallVector<SDNode*, 3> InteriorChainedNodes; |
| 1634 | for (unsigned i = 0, e = ChainNodesMatched.size(); i != e; ++i) { |
| 1635 | if (WalkChainUsers(ChainNodesMatched[i], ChainNodesMatched, |
| 1636 | InteriorChainedNodes) == CR_InducesCycle) |
| 1637 | return SDValue(); // Would induce a cycle. |
| 1638 | } |
Chris Lattner | 6b30792 | 2010-03-02 00:00:03 +0000 | [diff] [blame] | 1639 | |
Chris Lattner | c6d7ad3 | 2010-03-02 02:22:10 +0000 | [diff] [blame] | 1640 | // Okay, we have walked all the matched nodes and collected TokenFactor nodes |
| 1641 | // that we are interested in. Form our input TokenFactor node. |
Chris Lattner | 6b30792 | 2010-03-02 00:00:03 +0000 | [diff] [blame] | 1642 | SmallVector<SDValue, 3> InputChains; |
| 1643 | for (unsigned i = 0, e = ChainNodesMatched.size(); i != e; ++i) { |
Chris Lattner | c6d7ad3 | 2010-03-02 02:22:10 +0000 | [diff] [blame] | 1644 | // Add the input chain of this node to the InputChains list (which will be |
| 1645 | // the operands of the generated TokenFactor) if it's not an interior node. |
| 1646 | SDNode *N = ChainNodesMatched[i]; |
| 1647 | if (N->getOpcode() != ISD::TokenFactor) { |
| 1648 | if (std::count(InteriorChainedNodes.begin(),InteriorChainedNodes.end(),N)) |
| 1649 | continue; |
| 1650 | |
| 1651 | // Otherwise, add the input chain. |
| 1652 | SDValue InChain = ChainNodesMatched[i]->getOperand(0); |
| 1653 | assert(InChain.getValueType() == MVT::Other && "Not a chain"); |
Chris Lattner | 6b30792 | 2010-03-02 00:00:03 +0000 | [diff] [blame] | 1654 | InputChains.push_back(InChain); |
Chris Lattner | c6d7ad3 | 2010-03-02 02:22:10 +0000 | [diff] [blame] | 1655 | continue; |
| 1656 | } |
| 1657 | |
| 1658 | // If we have a token factor, we want to add all inputs of the token factor |
| 1659 | // that are not part of the pattern we're matching. |
| 1660 | for (unsigned op = 0, e = N->getNumOperands(); op != e; ++op) { |
| 1661 | if (!std::count(ChainNodesMatched.begin(), ChainNodesMatched.end(), |
Chris Lattner | 6183fbd | 2010-03-02 02:37:23 +0000 | [diff] [blame] | 1662 | N->getOperand(op).getNode())) |
| 1663 | InputChains.push_back(N->getOperand(op)); |
Chris Lattner | c6d7ad3 | 2010-03-02 02:22:10 +0000 | [diff] [blame] | 1664 | } |
Chris Lattner | 6b30792 | 2010-03-02 00:00:03 +0000 | [diff] [blame] | 1665 | } |
| 1666 | |
| 1667 | SDValue Res; |
| 1668 | if (InputChains.size() == 1) |
| 1669 | return InputChains[0]; |
| 1670 | return CurDAG->getNode(ISD::TokenFactor, ChainNodesMatched[0]->getDebugLoc(), |
| 1671 | MVT::Other, &InputChains[0], InputChains.size()); |
| 1672 | } |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 1673 | |
Chris Lattner | 3ff1e4d | 2010-03-02 06:55:04 +0000 | [diff] [blame] | 1674 | /// MorphNode - Handle morphing a node in place for the selector. |
| 1675 | SDNode *SelectionDAGISel:: |
| 1676 | MorphNode(SDNode *Node, unsigned TargetOpc, SDVTList VTList, |
| 1677 | const SDValue *Ops, unsigned NumOps, unsigned EmitNodeInfo) { |
| 1678 | // It is possible we're using MorphNodeTo to replace a node with no |
| 1679 | // normal results with one that has a normal result (or we could be |
| 1680 | // adding a chain) and the input could have flags and chains as well. |
Chris Lattner | caa8870 | 2010-03-28 08:43:23 +0000 | [diff] [blame] | 1681 | // In this case we need to shift the operands down. |
Chris Lattner | 3ff1e4d | 2010-03-02 06:55:04 +0000 | [diff] [blame] | 1682 | // FIXME: This is a horrible hack and broken in obscure cases, no worse |
Chris Lattner | a6f8693 | 2010-03-27 18:54:50 +0000 | [diff] [blame] | 1683 | // than the old isel though. |
Chris Lattner | 3ff1e4d | 2010-03-02 06:55:04 +0000 | [diff] [blame] | 1684 | int OldFlagResultNo = -1, OldChainResultNo = -1; |
| 1685 | |
| 1686 | unsigned NTMNumResults = Node->getNumValues(); |
| 1687 | if (Node->getValueType(NTMNumResults-1) == MVT::Flag) { |
| 1688 | OldFlagResultNo = NTMNumResults-1; |
| 1689 | if (NTMNumResults != 1 && |
| 1690 | Node->getValueType(NTMNumResults-2) == MVT::Other) |
| 1691 | OldChainResultNo = NTMNumResults-2; |
| 1692 | } else if (Node->getValueType(NTMNumResults-1) == MVT::Other) |
| 1693 | OldChainResultNo = NTMNumResults-1; |
| 1694 | |
Chris Lattner | 61c97f6 | 2010-03-02 07:14:49 +0000 | [diff] [blame] | 1695 | // Call the underlying SelectionDAG routine to do the transmogrification. Note |
| 1696 | // that this deletes operands of the old node that become dead. |
Chris Lattner | 3ff1e4d | 2010-03-02 06:55:04 +0000 | [diff] [blame] | 1697 | SDNode *Res = CurDAG->MorphNodeTo(Node, ~TargetOpc, VTList, Ops, NumOps); |
| 1698 | |
| 1699 | // MorphNodeTo can operate in two ways: if an existing node with the |
| 1700 | // specified operands exists, it can just return it. Otherwise, it |
| 1701 | // updates the node in place to have the requested operands. |
| 1702 | if (Res == Node) { |
| 1703 | // If we updated the node in place, reset the node ID. To the isel, |
| 1704 | // this should be just like a newly allocated machine node. |
| 1705 | Res->setNodeId(-1); |
| 1706 | } |
| 1707 | |
| 1708 | unsigned ResNumResults = Res->getNumValues(); |
| 1709 | // Move the flag if needed. |
| 1710 | if ((EmitNodeInfo & OPFL_FlagOutput) && OldFlagResultNo != -1 && |
| 1711 | (unsigned)OldFlagResultNo != ResNumResults-1) |
| 1712 | CurDAG->ReplaceAllUsesOfValueWith(SDValue(Node, OldFlagResultNo), |
| 1713 | SDValue(Res, ResNumResults-1)); |
| 1714 | |
| 1715 | if ((EmitNodeInfo & OPFL_FlagOutput) != 0) |
| 1716 | --ResNumResults; |
| 1717 | |
| 1718 | // Move the chain reference if needed. |
| 1719 | if ((EmitNodeInfo & OPFL_Chain) && OldChainResultNo != -1 && |
| 1720 | (unsigned)OldChainResultNo != ResNumResults-1) |
| 1721 | CurDAG->ReplaceAllUsesOfValueWith(SDValue(Node, OldChainResultNo), |
| 1722 | SDValue(Res, ResNumResults-1)); |
| 1723 | |
| 1724 | // Otherwise, no replacement happened because the node already exists. Replace |
| 1725 | // Uses of the old node with the new one. |
| 1726 | if (Res != Node) |
| 1727 | CurDAG->ReplaceAllUsesWith(Node, Res); |
| 1728 | |
| 1729 | return Res; |
| 1730 | } |
| 1731 | |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 1732 | /// CheckPatternPredicate - Implements OP_CheckPatternPredicate. |
| 1733 | ALWAYS_INLINE static bool |
Chris Lattner | da828e3 | 2010-03-03 07:46:25 +0000 | [diff] [blame] | 1734 | CheckSame(const unsigned char *MatcherTable, unsigned &MatcherIndex, |
| 1735 | SDValue N, const SmallVectorImpl<SDValue> &RecordedNodes) { |
| 1736 | // Accept if it is exactly the same as a previously recorded node. |
| 1737 | unsigned RecNo = MatcherTable[MatcherIndex++]; |
| 1738 | assert(RecNo < RecordedNodes.size() && "Invalid CheckSame"); |
| 1739 | return N == RecordedNodes[RecNo]; |
| 1740 | } |
| 1741 | |
| 1742 | /// CheckPatternPredicate - Implements OP_CheckPatternPredicate. |
| 1743 | ALWAYS_INLINE static bool |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 1744 | CheckPatternPredicate(const unsigned char *MatcherTable, unsigned &MatcherIndex, |
| 1745 | SelectionDAGISel &SDISel) { |
| 1746 | return SDISel.CheckPatternPredicate(MatcherTable[MatcherIndex++]); |
| 1747 | } |
| 1748 | |
| 1749 | /// CheckNodePredicate - Implements OP_CheckNodePredicate. |
| 1750 | ALWAYS_INLINE static bool |
| 1751 | CheckNodePredicate(const unsigned char *MatcherTable, unsigned &MatcherIndex, |
| 1752 | SelectionDAGISel &SDISel, SDNode *N) { |
| 1753 | return SDISel.CheckNodePredicate(N, MatcherTable[MatcherIndex++]); |
| 1754 | } |
| 1755 | |
| 1756 | ALWAYS_INLINE static bool |
| 1757 | CheckOpcode(const unsigned char *MatcherTable, unsigned &MatcherIndex, |
| 1758 | SDNode *N) { |
Chris Lattner | d5d5a3d | 2010-03-25 06:33:05 +0000 | [diff] [blame] | 1759 | uint16_t Opc = MatcherTable[MatcherIndex++]; |
| 1760 | Opc |= (unsigned short)MatcherTable[MatcherIndex++] << 8; |
| 1761 | return N->getOpcode() == Opc; |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 1762 | } |
| 1763 | |
| 1764 | ALWAYS_INLINE static bool |
| 1765 | CheckType(const unsigned char *MatcherTable, unsigned &MatcherIndex, |
| 1766 | SDValue N, const TargetLowering &TLI) { |
| 1767 | MVT::SimpleValueType VT = (MVT::SimpleValueType)MatcherTable[MatcherIndex++]; |
| 1768 | if (N.getValueType() == VT) return true; |
| 1769 | |
| 1770 | // Handle the case when VT is iPTR. |
| 1771 | return VT == MVT::iPTR && N.getValueType() == TLI.getPointerTy(); |
| 1772 | } |
| 1773 | |
| 1774 | ALWAYS_INLINE static bool |
Chris Lattner | da828e3 | 2010-03-03 07:46:25 +0000 | [diff] [blame] | 1775 | CheckChildType(const unsigned char *MatcherTable, unsigned &MatcherIndex, |
| 1776 | SDValue N, const TargetLowering &TLI, |
| 1777 | unsigned ChildNo) { |
| 1778 | if (ChildNo >= N.getNumOperands()) |
| 1779 | return false; // Match fails if out of range child #. |
| 1780 | return ::CheckType(MatcherTable, MatcherIndex, N.getOperand(ChildNo), TLI); |
| 1781 | } |
| 1782 | |
| 1783 | |
| 1784 | ALWAYS_INLINE static bool |
| 1785 | CheckCondCode(const unsigned char *MatcherTable, unsigned &MatcherIndex, |
| 1786 | SDValue N) { |
| 1787 | return cast<CondCodeSDNode>(N)->get() == |
| 1788 | (ISD::CondCode)MatcherTable[MatcherIndex++]; |
| 1789 | } |
| 1790 | |
| 1791 | ALWAYS_INLINE static bool |
| 1792 | CheckValueType(const unsigned char *MatcherTable, unsigned &MatcherIndex, |
| 1793 | SDValue N, const TargetLowering &TLI) { |
| 1794 | MVT::SimpleValueType VT = (MVT::SimpleValueType)MatcherTable[MatcherIndex++]; |
| 1795 | if (cast<VTSDNode>(N)->getVT() == VT) |
| 1796 | return true; |
| 1797 | |
| 1798 | // Handle the case when VT is iPTR. |
| 1799 | return VT == MVT::iPTR && cast<VTSDNode>(N)->getVT() == TLI.getPointerTy(); |
| 1800 | } |
| 1801 | |
| 1802 | ALWAYS_INLINE static bool |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 1803 | CheckInteger(const unsigned char *MatcherTable, unsigned &MatcherIndex, |
| 1804 | SDValue N) { |
| 1805 | int64_t Val = MatcherTable[MatcherIndex++]; |
| 1806 | if (Val & 128) |
| 1807 | Val = GetVBR(Val, MatcherTable, MatcherIndex); |
| 1808 | |
| 1809 | ConstantSDNode *C = dyn_cast<ConstantSDNode>(N); |
| 1810 | return C != 0 && C->getSExtValue() == Val; |
| 1811 | } |
| 1812 | |
Chris Lattner | da828e3 | 2010-03-03 07:46:25 +0000 | [diff] [blame] | 1813 | ALWAYS_INLINE static bool |
| 1814 | CheckAndImm(const unsigned char *MatcherTable, unsigned &MatcherIndex, |
| 1815 | SDValue N, SelectionDAGISel &SDISel) { |
| 1816 | int64_t Val = MatcherTable[MatcherIndex++]; |
| 1817 | if (Val & 128) |
| 1818 | Val = GetVBR(Val, MatcherTable, MatcherIndex); |
| 1819 | |
| 1820 | if (N->getOpcode() != ISD::AND) return false; |
| 1821 | |
| 1822 | ConstantSDNode *C = dyn_cast<ConstantSDNode>(N->getOperand(1)); |
| 1823 | return C != 0 && SDISel.CheckAndMask(N.getOperand(0), C, Val); |
| 1824 | } |
| 1825 | |
| 1826 | ALWAYS_INLINE static bool |
| 1827 | CheckOrImm(const unsigned char *MatcherTable, unsigned &MatcherIndex, |
| 1828 | SDValue N, SelectionDAGISel &SDISel) { |
| 1829 | int64_t Val = MatcherTable[MatcherIndex++]; |
| 1830 | if (Val & 128) |
| 1831 | Val = GetVBR(Val, MatcherTable, MatcherIndex); |
| 1832 | |
| 1833 | if (N->getOpcode() != ISD::OR) return false; |
| 1834 | |
| 1835 | ConstantSDNode *C = dyn_cast<ConstantSDNode>(N->getOperand(1)); |
| 1836 | return C != 0 && SDISel.CheckOrMask(N.getOperand(0), C, Val); |
| 1837 | } |
| 1838 | |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 1839 | /// IsPredicateKnownToFail - If we know how and can do so without pushing a |
| 1840 | /// scope, evaluate the current node. If the current predicate is known to |
| 1841 | /// fail, set Result=true and return anything. If the current predicate is |
| 1842 | /// known to pass, set Result=false and return the MatcherIndex to continue |
| 1843 | /// with. If the current predicate is unknown, set Result=false and return the |
| 1844 | /// MatcherIndex to continue with. |
| 1845 | static unsigned IsPredicateKnownToFail(const unsigned char *Table, |
| 1846 | unsigned Index, SDValue N, |
Chris Lattner | da828e3 | 2010-03-03 07:46:25 +0000 | [diff] [blame] | 1847 | bool &Result, SelectionDAGISel &SDISel, |
| 1848 | SmallVectorImpl<SDValue> &RecordedNodes){ |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 1849 | switch (Table[Index++]) { |
| 1850 | default: |
| 1851 | Result = false; |
| 1852 | return Index-1; // Could not evaluate this predicate. |
Chris Lattner | da828e3 | 2010-03-03 07:46:25 +0000 | [diff] [blame] | 1853 | case SelectionDAGISel::OPC_CheckSame: |
| 1854 | Result = !::CheckSame(Table, Index, N, RecordedNodes); |
| 1855 | return Index; |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 1856 | case SelectionDAGISel::OPC_CheckPatternPredicate: |
Chris Lattner | da828e3 | 2010-03-03 07:46:25 +0000 | [diff] [blame] | 1857 | Result = !::CheckPatternPredicate(Table, Index, SDISel); |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 1858 | return Index; |
| 1859 | case SelectionDAGISel::OPC_CheckPredicate: |
Chris Lattner | da828e3 | 2010-03-03 07:46:25 +0000 | [diff] [blame] | 1860 | Result = !::CheckNodePredicate(Table, Index, SDISel, N.getNode()); |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 1861 | return Index; |
| 1862 | case SelectionDAGISel::OPC_CheckOpcode: |
| 1863 | Result = !::CheckOpcode(Table, Index, N.getNode()); |
| 1864 | return Index; |
| 1865 | case SelectionDAGISel::OPC_CheckType: |
| 1866 | Result = !::CheckType(Table, Index, N, SDISel.TLI); |
| 1867 | return Index; |
| 1868 | case SelectionDAGISel::OPC_CheckChild0Type: |
| 1869 | case SelectionDAGISel::OPC_CheckChild1Type: |
| 1870 | case SelectionDAGISel::OPC_CheckChild2Type: |
| 1871 | case SelectionDAGISel::OPC_CheckChild3Type: |
| 1872 | case SelectionDAGISel::OPC_CheckChild4Type: |
| 1873 | case SelectionDAGISel::OPC_CheckChild5Type: |
| 1874 | case SelectionDAGISel::OPC_CheckChild6Type: |
Chris Lattner | da828e3 | 2010-03-03 07:46:25 +0000 | [diff] [blame] | 1875 | case SelectionDAGISel::OPC_CheckChild7Type: |
| 1876 | Result = !::CheckChildType(Table, Index, N, SDISel.TLI, |
| 1877 | Table[Index-1] - SelectionDAGISel::OPC_CheckChild0Type); |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 1878 | return Index; |
Chris Lattner | da828e3 | 2010-03-03 07:46:25 +0000 | [diff] [blame] | 1879 | case SelectionDAGISel::OPC_CheckCondCode: |
| 1880 | Result = !::CheckCondCode(Table, Index, N); |
| 1881 | return Index; |
| 1882 | case SelectionDAGISel::OPC_CheckValueType: |
| 1883 | Result = !::CheckValueType(Table, Index, N, SDISel.TLI); |
| 1884 | return Index; |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 1885 | case SelectionDAGISel::OPC_CheckInteger: |
| 1886 | Result = !::CheckInteger(Table, Index, N); |
| 1887 | return Index; |
Chris Lattner | da828e3 | 2010-03-03 07:46:25 +0000 | [diff] [blame] | 1888 | case SelectionDAGISel::OPC_CheckAndImm: |
| 1889 | Result = !::CheckAndImm(Table, Index, N, SDISel); |
| 1890 | return Index; |
| 1891 | case SelectionDAGISel::OPC_CheckOrImm: |
| 1892 | Result = !::CheckOrImm(Table, Index, N, SDISel); |
| 1893 | return Index; |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 1894 | } |
| 1895 | } |
| 1896 | |
Chris Lattner | 3ff1e4d | 2010-03-02 06:55:04 +0000 | [diff] [blame] | 1897 | |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 1898 | struct MatchScope { |
| 1899 | /// FailIndex - If this match fails, this is the index to continue with. |
| 1900 | unsigned FailIndex; |
| 1901 | |
| 1902 | /// NodeStack - The node stack when the scope was formed. |
| 1903 | SmallVector<SDValue, 4> NodeStack; |
| 1904 | |
| 1905 | /// NumRecordedNodes - The number of recorded nodes when the scope was formed. |
| 1906 | unsigned NumRecordedNodes; |
| 1907 | |
| 1908 | /// NumMatchedMemRefs - The number of matched memref entries. |
| 1909 | unsigned NumMatchedMemRefs; |
| 1910 | |
| 1911 | /// InputChain/InputFlag - The current chain/flag |
| 1912 | SDValue InputChain, InputFlag; |
| 1913 | |
| 1914 | /// HasChainNodesMatched - True if the ChainNodesMatched list is non-empty. |
| 1915 | bool HasChainNodesMatched, HasFlagResultNodesMatched; |
| 1916 | }; |
| 1917 | |
| 1918 | SDNode *SelectionDAGISel:: |
| 1919 | SelectCodeCommon(SDNode *NodeToMatch, const unsigned char *MatcherTable, |
| 1920 | unsigned TableSize) { |
| 1921 | // FIXME: Should these even be selected? Handle these cases in the caller? |
| 1922 | switch (NodeToMatch->getOpcode()) { |
| 1923 | default: |
| 1924 | break; |
| 1925 | case ISD::EntryToken: // These nodes remain the same. |
| 1926 | case ISD::BasicBlock: |
| 1927 | case ISD::Register: |
Chris Lattner | 3b9d621 | 2010-03-14 17:53:23 +0000 | [diff] [blame] | 1928 | //case ISD::VALUETYPE: |
| 1929 | //case ISD::CONDCODE: |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 1930 | case ISD::HANDLENODE: |
Chris Lattner | decc267 | 2010-04-07 05:20:54 +0000 | [diff] [blame] | 1931 | case ISD::MDNODE_SDNODE: |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 1932 | case ISD::TargetConstant: |
| 1933 | case ISD::TargetConstantFP: |
| 1934 | case ISD::TargetConstantPool: |
| 1935 | case ISD::TargetFrameIndex: |
| 1936 | case ISD::TargetExternalSymbol: |
| 1937 | case ISD::TargetBlockAddress: |
| 1938 | case ISD::TargetJumpTable: |
| 1939 | case ISD::TargetGlobalTLSAddress: |
| 1940 | case ISD::TargetGlobalAddress: |
| 1941 | case ISD::TokenFactor: |
| 1942 | case ISD::CopyFromReg: |
| 1943 | case ISD::CopyToReg: |
Chris Lattner | 7561d48 | 2010-03-14 02:33:54 +0000 | [diff] [blame] | 1944 | case ISD::EH_LABEL: |
Chris Lattner | d1b7382 | 2010-03-02 22:20:06 +0000 | [diff] [blame] | 1945 | NodeToMatch->setNodeId(-1); // Mark selected. |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 1946 | return 0; |
| 1947 | case ISD::AssertSext: |
| 1948 | case ISD::AssertZext: |
| 1949 | CurDAG->ReplaceAllUsesOfValueWith(SDValue(NodeToMatch, 0), |
| 1950 | NodeToMatch->getOperand(0)); |
| 1951 | return 0; |
| 1952 | case ISD::INLINEASM: return Select_INLINEASM(NodeToMatch); |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 1953 | case ISD::UNDEF: return Select_UNDEF(NodeToMatch); |
| 1954 | } |
| 1955 | |
| 1956 | assert(!NodeToMatch->isMachineOpcode() && "Node already selected!"); |
| 1957 | |
| 1958 | // Set up the node stack with NodeToMatch as the only node on the stack. |
| 1959 | SmallVector<SDValue, 8> NodeStack; |
| 1960 | SDValue N = SDValue(NodeToMatch, 0); |
| 1961 | NodeStack.push_back(N); |
| 1962 | |
| 1963 | // MatchScopes - Scopes used when matching, if a match failure happens, this |
| 1964 | // indicates where to continue checking. |
| 1965 | SmallVector<MatchScope, 8> MatchScopes; |
| 1966 | |
| 1967 | // RecordedNodes - This is the set of nodes that have been recorded by the |
| 1968 | // state machine. |
| 1969 | SmallVector<SDValue, 8> RecordedNodes; |
| 1970 | |
| 1971 | // MatchedMemRefs - This is the set of MemRef's we've seen in the input |
| 1972 | // pattern. |
| 1973 | SmallVector<MachineMemOperand*, 2> MatchedMemRefs; |
| 1974 | |
| 1975 | // These are the current input chain and flag for use when generating nodes. |
| 1976 | // Various Emit operations change these. For example, emitting a copytoreg |
| 1977 | // uses and updates these. |
| 1978 | SDValue InputChain, InputFlag; |
| 1979 | |
| 1980 | // ChainNodesMatched - If a pattern matches nodes that have input/output |
| 1981 | // chains, the OPC_EmitMergeInputChains operation is emitted which indicates |
| 1982 | // which ones they are. The result is captured into this list so that we can |
| 1983 | // update the chain results when the pattern is complete. |
| 1984 | SmallVector<SDNode*, 3> ChainNodesMatched; |
| 1985 | SmallVector<SDNode*, 3> FlagResultNodesMatched; |
| 1986 | |
| 1987 | DEBUG(errs() << "ISEL: Starting pattern match on root node: "; |
| 1988 | NodeToMatch->dump(CurDAG); |
| 1989 | errs() << '\n'); |
| 1990 | |
Chris Lattner | 7390eeb | 2010-03-01 18:47:11 +0000 | [diff] [blame] | 1991 | // Determine where to start the interpreter. Normally we start at opcode #0, |
| 1992 | // but if the state machine starts with an OPC_SwitchOpcode, then we |
| 1993 | // accelerate the first lookup (which is guaranteed to be hot) with the |
| 1994 | // OpcodeOffset table. |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 1995 | unsigned MatcherIndex = 0; |
Chris Lattner | 7390eeb | 2010-03-01 18:47:11 +0000 | [diff] [blame] | 1996 | |
| 1997 | if (!OpcodeOffset.empty()) { |
| 1998 | // Already computed the OpcodeOffset table, just index into it. |
| 1999 | if (N.getOpcode() < OpcodeOffset.size()) |
| 2000 | MatcherIndex = OpcodeOffset[N.getOpcode()]; |
| 2001 | DEBUG(errs() << " Initial Opcode index to " << MatcherIndex << "\n"); |
| 2002 | |
| 2003 | } else if (MatcherTable[0] == OPC_SwitchOpcode) { |
| 2004 | // Otherwise, the table isn't computed, but the state machine does start |
| 2005 | // with an OPC_SwitchOpcode instruction. Populate the table now, since this |
| 2006 | // is the first time we're selecting an instruction. |
| 2007 | unsigned Idx = 1; |
| 2008 | while (1) { |
| 2009 | // Get the size of this case. |
| 2010 | unsigned CaseSize = MatcherTable[Idx++]; |
| 2011 | if (CaseSize & 128) |
| 2012 | CaseSize = GetVBR(CaseSize, MatcherTable, Idx); |
| 2013 | if (CaseSize == 0) break; |
| 2014 | |
| 2015 | // Get the opcode, add the index to the table. |
Chris Lattner | d5d5a3d | 2010-03-25 06:33:05 +0000 | [diff] [blame] | 2016 | uint16_t Opc = MatcherTable[Idx++]; |
| 2017 | Opc |= (unsigned short)MatcherTable[Idx++] << 8; |
Chris Lattner | 7390eeb | 2010-03-01 18:47:11 +0000 | [diff] [blame] | 2018 | if (Opc >= OpcodeOffset.size()) |
| 2019 | OpcodeOffset.resize((Opc+1)*2); |
| 2020 | OpcodeOffset[Opc] = Idx; |
| 2021 | Idx += CaseSize; |
| 2022 | } |
| 2023 | |
| 2024 | // Okay, do the lookup for the first opcode. |
| 2025 | if (N.getOpcode() < OpcodeOffset.size()) |
| 2026 | MatcherIndex = OpcodeOffset[N.getOpcode()]; |
| 2027 | } |
| 2028 | |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2029 | while (1) { |
| 2030 | assert(MatcherIndex < TableSize && "Invalid index"); |
Dan Gohman | 19b3826 | 2010-03-09 02:15:05 +0000 | [diff] [blame] | 2031 | #ifndef NDEBUG |
| 2032 | unsigned CurrentOpcodeIndex = MatcherIndex; |
| 2033 | #endif |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2034 | BuiltinOpcodes Opcode = (BuiltinOpcodes)MatcherTable[MatcherIndex++]; |
| 2035 | switch (Opcode) { |
| 2036 | case OPC_Scope: { |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 2037 | // Okay, the semantics of this operation are that we should push a scope |
| 2038 | // then evaluate the first child. However, pushing a scope only to have |
| 2039 | // the first check fail (which then pops it) is inefficient. If we can |
| 2040 | // determine immediately that the first check (or first several) will |
| 2041 | // immediately fail, don't even bother pushing a scope for them. |
| 2042 | unsigned FailIndex; |
| 2043 | |
| 2044 | while (1) { |
| 2045 | unsigned NumToSkip = MatcherTable[MatcherIndex++]; |
| 2046 | if (NumToSkip & 128) |
| 2047 | NumToSkip = GetVBR(NumToSkip, MatcherTable, MatcherIndex); |
| 2048 | // Found the end of the scope with no match. |
| 2049 | if (NumToSkip == 0) { |
| 2050 | FailIndex = 0; |
| 2051 | break; |
| 2052 | } |
| 2053 | |
| 2054 | FailIndex = MatcherIndex+NumToSkip; |
| 2055 | |
Chris Lattner | a6f8693 | 2010-03-27 18:54:50 +0000 | [diff] [blame] | 2056 | unsigned MatcherIndexOfPredicate = MatcherIndex; |
| 2057 | (void)MatcherIndexOfPredicate; // silence warning. |
| 2058 | |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 2059 | // If we can't evaluate this predicate without pushing a scope (e.g. if |
| 2060 | // it is a 'MoveParent') or if the predicate succeeds on this node, we |
| 2061 | // push the scope and evaluate the full predicate chain. |
| 2062 | bool Result; |
| 2063 | MatcherIndex = IsPredicateKnownToFail(MatcherTable, MatcherIndex, N, |
Chris Lattner | da828e3 | 2010-03-03 07:46:25 +0000 | [diff] [blame] | 2064 | Result, *this, RecordedNodes); |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 2065 | if (!Result) |
| 2066 | break; |
| 2067 | |
Chris Lattner | a6f8693 | 2010-03-27 18:54:50 +0000 | [diff] [blame] | 2068 | DEBUG(errs() << " Skipped scope entry (due to false predicate) at " |
| 2069 | << "index " << MatcherIndexOfPredicate |
| 2070 | << ", continuing at " << FailIndex << "\n"); |
Chris Lattner | bed993d | 2010-03-28 19:46:56 +0000 | [diff] [blame] | 2071 | ++NumDAGIselRetries; |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 2072 | |
| 2073 | // Otherwise, we know that this case of the Scope is guaranteed to fail, |
| 2074 | // move to the next case. |
| 2075 | MatcherIndex = FailIndex; |
| 2076 | } |
| 2077 | |
| 2078 | // If the whole scope failed to match, bail. |
| 2079 | if (FailIndex == 0) break; |
| 2080 | |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2081 | // Push a MatchScope which indicates where to go if the first child fails |
| 2082 | // to match. |
| 2083 | MatchScope NewEntry; |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 2084 | NewEntry.FailIndex = FailIndex; |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2085 | NewEntry.NodeStack.append(NodeStack.begin(), NodeStack.end()); |
| 2086 | NewEntry.NumRecordedNodes = RecordedNodes.size(); |
| 2087 | NewEntry.NumMatchedMemRefs = MatchedMemRefs.size(); |
| 2088 | NewEntry.InputChain = InputChain; |
| 2089 | NewEntry.InputFlag = InputFlag; |
| 2090 | NewEntry.HasChainNodesMatched = !ChainNodesMatched.empty(); |
| 2091 | NewEntry.HasFlagResultNodesMatched = !FlagResultNodesMatched.empty(); |
| 2092 | MatchScopes.push_back(NewEntry); |
| 2093 | continue; |
| 2094 | } |
| 2095 | case OPC_RecordNode: |
| 2096 | // Remember this node, it may end up being an operand in the pattern. |
| 2097 | RecordedNodes.push_back(N); |
| 2098 | continue; |
| 2099 | |
| 2100 | case OPC_RecordChild0: case OPC_RecordChild1: |
| 2101 | case OPC_RecordChild2: case OPC_RecordChild3: |
| 2102 | case OPC_RecordChild4: case OPC_RecordChild5: |
| 2103 | case OPC_RecordChild6: case OPC_RecordChild7: { |
| 2104 | unsigned ChildNo = Opcode-OPC_RecordChild0; |
| 2105 | if (ChildNo >= N.getNumOperands()) |
| 2106 | break; // Match fails if out of range child #. |
| 2107 | |
| 2108 | RecordedNodes.push_back(N->getOperand(ChildNo)); |
| 2109 | continue; |
| 2110 | } |
| 2111 | case OPC_RecordMemRef: |
| 2112 | MatchedMemRefs.push_back(cast<MemSDNode>(N)->getMemOperand()); |
| 2113 | continue; |
| 2114 | |
| 2115 | case OPC_CaptureFlagInput: |
| 2116 | // If the current node has an input flag, capture it in InputFlag. |
| 2117 | if (N->getNumOperands() != 0 && |
| 2118 | N->getOperand(N->getNumOperands()-1).getValueType() == MVT::Flag) |
| 2119 | InputFlag = N->getOperand(N->getNumOperands()-1); |
| 2120 | continue; |
| 2121 | |
| 2122 | case OPC_MoveChild: { |
| 2123 | unsigned ChildNo = MatcherTable[MatcherIndex++]; |
| 2124 | if (ChildNo >= N.getNumOperands()) |
| 2125 | break; // Match fails if out of range child #. |
| 2126 | N = N.getOperand(ChildNo); |
| 2127 | NodeStack.push_back(N); |
| 2128 | continue; |
| 2129 | } |
| 2130 | |
| 2131 | case OPC_MoveParent: |
| 2132 | // Pop the current node off the NodeStack. |
| 2133 | NodeStack.pop_back(); |
| 2134 | assert(!NodeStack.empty() && "Node stack imbalance!"); |
| 2135 | N = NodeStack.back(); |
| 2136 | continue; |
| 2137 | |
Chris Lattner | da828e3 | 2010-03-03 07:46:25 +0000 | [diff] [blame] | 2138 | case OPC_CheckSame: |
| 2139 | if (!::CheckSame(MatcherTable, MatcherIndex, N, RecordedNodes)) break; |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2140 | continue; |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2141 | case OPC_CheckPatternPredicate: |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 2142 | if (!::CheckPatternPredicate(MatcherTable, MatcherIndex, *this)) break; |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2143 | continue; |
| 2144 | case OPC_CheckPredicate: |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 2145 | if (!::CheckNodePredicate(MatcherTable, MatcherIndex, *this, |
| 2146 | N.getNode())) |
| 2147 | break; |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2148 | continue; |
Chris Lattner | 57bf8a4 | 2010-03-04 01:23:08 +0000 | [diff] [blame] | 2149 | case OPC_CheckComplexPat: { |
| 2150 | unsigned CPNum = MatcherTable[MatcherIndex++]; |
| 2151 | unsigned RecNo = MatcherTable[MatcherIndex++]; |
| 2152 | assert(RecNo < RecordedNodes.size() && "Invalid CheckComplexPat"); |
| 2153 | if (!CheckComplexPattern(NodeToMatch, RecordedNodes[RecNo], CPNum, |
| 2154 | RecordedNodes)) |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2155 | break; |
| 2156 | continue; |
Chris Lattner | 57bf8a4 | 2010-03-04 01:23:08 +0000 | [diff] [blame] | 2157 | } |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2158 | case OPC_CheckOpcode: |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 2159 | if (!::CheckOpcode(MatcherTable, MatcherIndex, N.getNode())) break; |
| 2160 | continue; |
| 2161 | |
| 2162 | case OPC_CheckType: |
| 2163 | if (!::CheckType(MatcherTable, MatcherIndex, N, TLI)) break; |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2164 | continue; |
| 2165 | |
Chris Lattner | eb66921 | 2010-03-01 06:59:22 +0000 | [diff] [blame] | 2166 | case OPC_SwitchOpcode: { |
| 2167 | unsigned CurNodeOpcode = N.getOpcode(); |
Chris Lattner | 7d892d6 | 2010-03-01 07:43:08 +0000 | [diff] [blame] | 2168 | unsigned SwitchStart = MatcherIndex-1; (void)SwitchStart; |
Chris Lattner | eb66921 | 2010-03-01 06:59:22 +0000 | [diff] [blame] | 2169 | unsigned CaseSize; |
| 2170 | while (1) { |
| 2171 | // Get the size of this case. |
| 2172 | CaseSize = MatcherTable[MatcherIndex++]; |
| 2173 | if (CaseSize & 128) |
| 2174 | CaseSize = GetVBR(CaseSize, MatcherTable, MatcherIndex); |
| 2175 | if (CaseSize == 0) break; |
| 2176 | |
Chris Lattner | d5d5a3d | 2010-03-25 06:33:05 +0000 | [diff] [blame] | 2177 | uint16_t Opc = MatcherTable[MatcherIndex++]; |
| 2178 | Opc |= (unsigned short)MatcherTable[MatcherIndex++] << 8; |
| 2179 | |
Chris Lattner | eb66921 | 2010-03-01 06:59:22 +0000 | [diff] [blame] | 2180 | // If the opcode matches, then we will execute this case. |
Chris Lattner | d5d5a3d | 2010-03-25 06:33:05 +0000 | [diff] [blame] | 2181 | if (CurNodeOpcode == Opc) |
Chris Lattner | eb66921 | 2010-03-01 06:59:22 +0000 | [diff] [blame] | 2182 | break; |
| 2183 | |
| 2184 | // Otherwise, skip over this case. |
| 2185 | MatcherIndex += CaseSize; |
| 2186 | } |
| 2187 | |
Chris Lattner | cfe2eab | 2010-03-03 06:28:15 +0000 | [diff] [blame] | 2188 | // If no cases matched, bail out. |
Chris Lattner | eb66921 | 2010-03-01 06:59:22 +0000 | [diff] [blame] | 2189 | if (CaseSize == 0) break; |
| 2190 | |
| 2191 | // Otherwise, execute the case we found. |
| 2192 | DEBUG(errs() << " OpcodeSwitch from " << SwitchStart |
| 2193 | << " to " << MatcherIndex << "\n"); |
| 2194 | continue; |
| 2195 | } |
| 2196 | |
Chris Lattner | cfe2eab | 2010-03-03 06:28:15 +0000 | [diff] [blame] | 2197 | case OPC_SwitchType: { |
| 2198 | MVT::SimpleValueType CurNodeVT = N.getValueType().getSimpleVT().SimpleTy; |
| 2199 | unsigned SwitchStart = MatcherIndex-1; (void)SwitchStart; |
| 2200 | unsigned CaseSize; |
| 2201 | while (1) { |
| 2202 | // Get the size of this case. |
| 2203 | CaseSize = MatcherTable[MatcherIndex++]; |
| 2204 | if (CaseSize & 128) |
| 2205 | CaseSize = GetVBR(CaseSize, MatcherTable, MatcherIndex); |
| 2206 | if (CaseSize == 0) break; |
| 2207 | |
| 2208 | MVT::SimpleValueType CaseVT = |
| 2209 | (MVT::SimpleValueType)MatcherTable[MatcherIndex++]; |
| 2210 | if (CaseVT == MVT::iPTR) |
| 2211 | CaseVT = TLI.getPointerTy().SimpleTy; |
| 2212 | |
| 2213 | // If the VT matches, then we will execute this case. |
| 2214 | if (CurNodeVT == CaseVT) |
| 2215 | break; |
| 2216 | |
| 2217 | // Otherwise, skip over this case. |
| 2218 | MatcherIndex += CaseSize; |
| 2219 | } |
| 2220 | |
| 2221 | // If no cases matched, bail out. |
| 2222 | if (CaseSize == 0) break; |
| 2223 | |
| 2224 | // Otherwise, execute the case we found. |
| 2225 | DEBUG(errs() << " TypeSwitch[" << EVT(CurNodeVT).getEVTString() |
| 2226 | << "] from " << SwitchStart << " to " << MatcherIndex<<'\n'); |
| 2227 | continue; |
| 2228 | } |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2229 | case OPC_CheckChild0Type: case OPC_CheckChild1Type: |
| 2230 | case OPC_CheckChild2Type: case OPC_CheckChild3Type: |
| 2231 | case OPC_CheckChild4Type: case OPC_CheckChild5Type: |
Chris Lattner | da828e3 | 2010-03-03 07:46:25 +0000 | [diff] [blame] | 2232 | case OPC_CheckChild6Type: case OPC_CheckChild7Type: |
| 2233 | if (!::CheckChildType(MatcherTable, MatcherIndex, N, TLI, |
| 2234 | Opcode-OPC_CheckChild0Type)) |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 2235 | break; |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2236 | continue; |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2237 | case OPC_CheckCondCode: |
Chris Lattner | da828e3 | 2010-03-03 07:46:25 +0000 | [diff] [blame] | 2238 | if (!::CheckCondCode(MatcherTable, MatcherIndex, N)) break; |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2239 | continue; |
Chris Lattner | da828e3 | 2010-03-03 07:46:25 +0000 | [diff] [blame] | 2240 | case OPC_CheckValueType: |
| 2241 | if (!::CheckValueType(MatcherTable, MatcherIndex, N, TLI)) break; |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2242 | continue; |
Chris Lattner | f1b7c7d | 2010-03-03 07:31:15 +0000 | [diff] [blame] | 2243 | case OPC_CheckInteger: |
| 2244 | if (!::CheckInteger(MatcherTable, MatcherIndex, N)) break; |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2245 | continue; |
Chris Lattner | da828e3 | 2010-03-03 07:46:25 +0000 | [diff] [blame] | 2246 | case OPC_CheckAndImm: |
| 2247 | if (!::CheckAndImm(MatcherTable, MatcherIndex, N, *this)) break; |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2248 | continue; |
Chris Lattner | da828e3 | 2010-03-03 07:46:25 +0000 | [diff] [blame] | 2249 | case OPC_CheckOrImm: |
| 2250 | if (!::CheckOrImm(MatcherTable, MatcherIndex, N, *this)) break; |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2251 | continue; |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2252 | |
| 2253 | case OPC_CheckFoldableChainNode: { |
| 2254 | assert(NodeStack.size() != 1 && "No parent node"); |
| 2255 | // Verify that all intermediate nodes between the root and this one have |
| 2256 | // a single use. |
| 2257 | bool HasMultipleUses = false; |
| 2258 | for (unsigned i = 1, e = NodeStack.size()-1; i != e; ++i) |
| 2259 | if (!NodeStack[i].hasOneUse()) { |
| 2260 | HasMultipleUses = true; |
| 2261 | break; |
| 2262 | } |
| 2263 | if (HasMultipleUses) break; |
| 2264 | |
| 2265 | // Check to see that the target thinks this is profitable to fold and that |
| 2266 | // we can fold it without inducing cycles in the graph. |
| 2267 | if (!IsProfitableToFold(N, NodeStack[NodeStack.size()-2].getNode(), |
| 2268 | NodeToMatch) || |
| 2269 | !IsLegalToFold(N, NodeStack[NodeStack.size()-2].getNode(), |
Chris Lattner | d1b7382 | 2010-03-02 22:20:06 +0000 | [diff] [blame] | 2270 | NodeToMatch, true/*We validate our own chains*/)) |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2271 | break; |
| 2272 | |
| 2273 | continue; |
| 2274 | } |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2275 | case OPC_EmitInteger: { |
| 2276 | MVT::SimpleValueType VT = |
| 2277 | (MVT::SimpleValueType)MatcherTable[MatcherIndex++]; |
| 2278 | int64_t Val = MatcherTable[MatcherIndex++]; |
| 2279 | if (Val & 128) |
| 2280 | Val = GetVBR(Val, MatcherTable, MatcherIndex); |
| 2281 | RecordedNodes.push_back(CurDAG->getTargetConstant(Val, VT)); |
| 2282 | continue; |
| 2283 | } |
| 2284 | case OPC_EmitRegister: { |
| 2285 | MVT::SimpleValueType VT = |
| 2286 | (MVT::SimpleValueType)MatcherTable[MatcherIndex++]; |
| 2287 | unsigned RegNo = MatcherTable[MatcherIndex++]; |
| 2288 | RecordedNodes.push_back(CurDAG->getRegister(RegNo, VT)); |
| 2289 | continue; |
| 2290 | } |
| 2291 | |
| 2292 | case OPC_EmitConvertToTarget: { |
| 2293 | // Convert from IMM/FPIMM to target version. |
| 2294 | unsigned RecNo = MatcherTable[MatcherIndex++]; |
| 2295 | assert(RecNo < RecordedNodes.size() && "Invalid CheckSame"); |
| 2296 | SDValue Imm = RecordedNodes[RecNo]; |
| 2297 | |
| 2298 | if (Imm->getOpcode() == ISD::Constant) { |
| 2299 | int64_t Val = cast<ConstantSDNode>(Imm)->getZExtValue(); |
| 2300 | Imm = CurDAG->getTargetConstant(Val, Imm.getValueType()); |
| 2301 | } else if (Imm->getOpcode() == ISD::ConstantFP) { |
| 2302 | const ConstantFP *Val=cast<ConstantFPSDNode>(Imm)->getConstantFPValue(); |
| 2303 | Imm = CurDAG->getTargetConstantFP(*Val, Imm.getValueType()); |
| 2304 | } |
| 2305 | |
| 2306 | RecordedNodes.push_back(Imm); |
| 2307 | continue; |
| 2308 | } |
| 2309 | |
Chris Lattner | aa4e339 | 2010-03-28 05:50:16 +0000 | [diff] [blame] | 2310 | case OPC_EmitMergeInputChains1_0: // OPC_EmitMergeInputChains, 1, 0 |
| 2311 | case OPC_EmitMergeInputChains1_1: { // OPC_EmitMergeInputChains, 1, 1 |
| 2312 | // These are space-optimized forms of OPC_EmitMergeInputChains. |
| 2313 | assert(InputChain.getNode() == 0 && |
| 2314 | "EmitMergeInputChains should be the first chain producing node"); |
| 2315 | assert(ChainNodesMatched.empty() && |
| 2316 | "Should only have one EmitMergeInputChains per match"); |
| 2317 | |
| 2318 | // Read all of the chained nodes. |
| 2319 | unsigned RecNo = Opcode == OPC_EmitMergeInputChains1_1; |
| 2320 | assert(RecNo < RecordedNodes.size() && "Invalid CheckSame"); |
| 2321 | ChainNodesMatched.push_back(RecordedNodes[RecNo].getNode()); |
| 2322 | |
| 2323 | // FIXME: What if other value results of the node have uses not matched |
| 2324 | // by this pattern? |
| 2325 | if (ChainNodesMatched.back() != NodeToMatch && |
| 2326 | !RecordedNodes[RecNo].hasOneUse()) { |
| 2327 | ChainNodesMatched.clear(); |
| 2328 | break; |
| 2329 | } |
| 2330 | |
| 2331 | // Merge the input chains if they are not intra-pattern references. |
| 2332 | InputChain = HandleMergeInputChains(ChainNodesMatched, CurDAG); |
| 2333 | |
| 2334 | if (InputChain.getNode() == 0) |
| 2335 | break; // Failed to merge. |
| 2336 | continue; |
| 2337 | } |
| 2338 | |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2339 | case OPC_EmitMergeInputChains: { |
| 2340 | assert(InputChain.getNode() == 0 && |
| 2341 | "EmitMergeInputChains should be the first chain producing node"); |
| 2342 | // This node gets a list of nodes we matched in the input that have |
| 2343 | // chains. We want to token factor all of the input chains to these nodes |
| 2344 | // together. However, if any of the input chains is actually one of the |
| 2345 | // nodes matched in this pattern, then we have an intra-match reference. |
| 2346 | // Ignore these because the newly token factored chain should not refer to |
| 2347 | // the old nodes. |
| 2348 | unsigned NumChains = MatcherTable[MatcherIndex++]; |
| 2349 | assert(NumChains != 0 && "Can't TF zero chains"); |
| 2350 | |
| 2351 | assert(ChainNodesMatched.empty() && |
| 2352 | "Should only have one EmitMergeInputChains per match"); |
| 2353 | |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2354 | // Read all of the chained nodes. |
Chris Lattner | 00592ec | 2010-03-02 19:34:59 +0000 | [diff] [blame] | 2355 | for (unsigned i = 0; i != NumChains; ++i) { |
| 2356 | unsigned RecNo = MatcherTable[MatcherIndex++]; |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2357 | assert(RecNo < RecordedNodes.size() && "Invalid CheckSame"); |
| 2358 | ChainNodesMatched.push_back(RecordedNodes[RecNo].getNode()); |
| 2359 | |
| 2360 | // FIXME: What if other value results of the node have uses not matched |
| 2361 | // by this pattern? |
| 2362 | if (ChainNodesMatched.back() != NodeToMatch && |
| 2363 | !RecordedNodes[RecNo].hasOneUse()) { |
| 2364 | ChainNodesMatched.clear(); |
| 2365 | break; |
| 2366 | } |
| 2367 | } |
Chris Lattner | 6b30792 | 2010-03-02 00:00:03 +0000 | [diff] [blame] | 2368 | |
| 2369 | // If the inner loop broke out, the match fails. |
| 2370 | if (ChainNodesMatched.empty()) |
| 2371 | break; |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2372 | |
Chris Lattner | 6b30792 | 2010-03-02 00:00:03 +0000 | [diff] [blame] | 2373 | // Merge the input chains if they are not intra-pattern references. |
| 2374 | InputChain = HandleMergeInputChains(ChainNodesMatched, CurDAG); |
| 2375 | |
| 2376 | if (InputChain.getNode() == 0) |
| 2377 | break; // Failed to merge. |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2378 | |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2379 | continue; |
| 2380 | } |
| 2381 | |
| 2382 | case OPC_EmitCopyToReg: { |
| 2383 | unsigned RecNo = MatcherTable[MatcherIndex++]; |
| 2384 | assert(RecNo < RecordedNodes.size() && "Invalid CheckSame"); |
| 2385 | unsigned DestPhysReg = MatcherTable[MatcherIndex++]; |
| 2386 | |
| 2387 | if (InputChain.getNode() == 0) |
| 2388 | InputChain = CurDAG->getEntryNode(); |
| 2389 | |
| 2390 | InputChain = CurDAG->getCopyToReg(InputChain, NodeToMatch->getDebugLoc(), |
| 2391 | DestPhysReg, RecordedNodes[RecNo], |
| 2392 | InputFlag); |
| 2393 | |
| 2394 | InputFlag = InputChain.getValue(1); |
| 2395 | continue; |
| 2396 | } |
| 2397 | |
| 2398 | case OPC_EmitNodeXForm: { |
| 2399 | unsigned XFormNo = MatcherTable[MatcherIndex++]; |
| 2400 | unsigned RecNo = MatcherTable[MatcherIndex++]; |
| 2401 | assert(RecNo < RecordedNodes.size() && "Invalid CheckSame"); |
| 2402 | RecordedNodes.push_back(RunSDNodeXForm(RecordedNodes[RecNo], XFormNo)); |
| 2403 | continue; |
| 2404 | } |
| 2405 | |
| 2406 | case OPC_EmitNode: |
| 2407 | case OPC_MorphNodeTo: { |
Chris Lattner | 14df8dc | 2010-02-28 22:38:43 +0000 | [diff] [blame] | 2408 | uint16_t TargetOpc = MatcherTable[MatcherIndex++]; |
| 2409 | TargetOpc |= (unsigned short)MatcherTable[MatcherIndex++] << 8; |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2410 | unsigned EmitNodeInfo = MatcherTable[MatcherIndex++]; |
| 2411 | // Get the result VT list. |
| 2412 | unsigned NumVTs = MatcherTable[MatcherIndex++]; |
| 2413 | SmallVector<EVT, 4> VTs; |
| 2414 | for (unsigned i = 0; i != NumVTs; ++i) { |
| 2415 | MVT::SimpleValueType VT = |
| 2416 | (MVT::SimpleValueType)MatcherTable[MatcherIndex++]; |
| 2417 | if (VT == MVT::iPTR) VT = TLI.getPointerTy().SimpleTy; |
| 2418 | VTs.push_back(VT); |
| 2419 | } |
| 2420 | |
| 2421 | if (EmitNodeInfo & OPFL_Chain) |
| 2422 | VTs.push_back(MVT::Other); |
| 2423 | if (EmitNodeInfo & OPFL_FlagOutput) |
| 2424 | VTs.push_back(MVT::Flag); |
| 2425 | |
Chris Lattner | 7d892d6 | 2010-03-01 07:43:08 +0000 | [diff] [blame] | 2426 | // This is hot code, so optimize the two most common cases of 1 and 2 |
| 2427 | // results. |
| 2428 | SDVTList VTList; |
| 2429 | if (VTs.size() == 1) |
| 2430 | VTList = CurDAG->getVTList(VTs[0]); |
| 2431 | else if (VTs.size() == 2) |
| 2432 | VTList = CurDAG->getVTList(VTs[0], VTs[1]); |
| 2433 | else |
| 2434 | VTList = CurDAG->getVTList(VTs.data(), VTs.size()); |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2435 | |
| 2436 | // Get the operand list. |
| 2437 | unsigned NumOps = MatcherTable[MatcherIndex++]; |
| 2438 | SmallVector<SDValue, 8> Ops; |
| 2439 | for (unsigned i = 0; i != NumOps; ++i) { |
| 2440 | unsigned RecNo = MatcherTable[MatcherIndex++]; |
| 2441 | if (RecNo & 128) |
| 2442 | RecNo = GetVBR(RecNo, MatcherTable, MatcherIndex); |
| 2443 | |
| 2444 | assert(RecNo < RecordedNodes.size() && "Invalid EmitNode"); |
| 2445 | Ops.push_back(RecordedNodes[RecNo]); |
| 2446 | } |
| 2447 | |
| 2448 | // If there are variadic operands to add, handle them now. |
| 2449 | if (EmitNodeInfo & OPFL_VariadicInfo) { |
| 2450 | // Determine the start index to copy from. |
| 2451 | unsigned FirstOpToCopy = getNumFixedFromVariadicInfo(EmitNodeInfo); |
| 2452 | FirstOpToCopy += (EmitNodeInfo & OPFL_Chain) ? 1 : 0; |
| 2453 | assert(NodeToMatch->getNumOperands() >= FirstOpToCopy && |
| 2454 | "Invalid variadic node"); |
| 2455 | // Copy all of the variadic operands, not including a potential flag |
| 2456 | // input. |
| 2457 | for (unsigned i = FirstOpToCopy, e = NodeToMatch->getNumOperands(); |
| 2458 | i != e; ++i) { |
| 2459 | SDValue V = NodeToMatch->getOperand(i); |
| 2460 | if (V.getValueType() == MVT::Flag) break; |
| 2461 | Ops.push_back(V); |
| 2462 | } |
| 2463 | } |
| 2464 | |
| 2465 | // If this has chain/flag inputs, add them. |
| 2466 | if (EmitNodeInfo & OPFL_Chain) |
| 2467 | Ops.push_back(InputChain); |
| 2468 | if ((EmitNodeInfo & OPFL_FlagInput) && InputFlag.getNode() != 0) |
| 2469 | Ops.push_back(InputFlag); |
| 2470 | |
| 2471 | // Create the node. |
| 2472 | SDNode *Res = 0; |
| 2473 | if (Opcode != OPC_MorphNodeTo) { |
| 2474 | // If this is a normal EmitNode command, just create the new node and |
| 2475 | // add the results to the RecordedNodes list. |
| 2476 | Res = CurDAG->getMachineNode(TargetOpc, NodeToMatch->getDebugLoc(), |
| 2477 | VTList, Ops.data(), Ops.size()); |
| 2478 | |
| 2479 | // Add all the non-flag/non-chain results to the RecordedNodes list. |
| 2480 | for (unsigned i = 0, e = VTs.size(); i != e; ++i) { |
| 2481 | if (VTs[i] == MVT::Other || VTs[i] == MVT::Flag) break; |
| 2482 | RecordedNodes.push_back(SDValue(Res, i)); |
| 2483 | } |
| 2484 | |
| 2485 | } else { |
Chris Lattner | 3ff1e4d | 2010-03-02 06:55:04 +0000 | [diff] [blame] | 2486 | Res = MorphNode(NodeToMatch, TargetOpc, VTList, Ops.data(), Ops.size(), |
| 2487 | EmitNodeInfo); |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2488 | } |
| 2489 | |
| 2490 | // If the node had chain/flag results, update our notion of the current |
| 2491 | // chain and flag. |
Chris Lattner | 82dd3d3 | 2010-03-02 07:50:03 +0000 | [diff] [blame] | 2492 | if (EmitNodeInfo & OPFL_FlagOutput) { |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2493 | InputFlag = SDValue(Res, VTs.size()-1); |
| 2494 | if (EmitNodeInfo & OPFL_Chain) |
| 2495 | InputChain = SDValue(Res, VTs.size()-2); |
| 2496 | } else if (EmitNodeInfo & OPFL_Chain) |
| 2497 | InputChain = SDValue(Res, VTs.size()-1); |
| 2498 | |
| 2499 | // If the OPFL_MemRefs flag is set on this node, slap all of the |
| 2500 | // accumulated memrefs onto it. |
| 2501 | // |
| 2502 | // FIXME: This is vastly incorrect for patterns with multiple outputs |
| 2503 | // instructions that access memory and for ComplexPatterns that match |
| 2504 | // loads. |
| 2505 | if (EmitNodeInfo & OPFL_MemRefs) { |
| 2506 | MachineSDNode::mmo_iterator MemRefs = |
| 2507 | MF->allocateMemRefsArray(MatchedMemRefs.size()); |
| 2508 | std::copy(MatchedMemRefs.begin(), MatchedMemRefs.end(), MemRefs); |
| 2509 | cast<MachineSDNode>(Res) |
| 2510 | ->setMemRefs(MemRefs, MemRefs + MatchedMemRefs.size()); |
| 2511 | } |
| 2512 | |
| 2513 | DEBUG(errs() << " " |
| 2514 | << (Opcode == OPC_MorphNodeTo ? "Morphed" : "Created") |
| 2515 | << " node: "; Res->dump(CurDAG); errs() << "\n"); |
| 2516 | |
| 2517 | // If this was a MorphNodeTo then we're completely done! |
| 2518 | if (Opcode == OPC_MorphNodeTo) { |
| 2519 | // Update chain and flag uses. |
| 2520 | UpdateChainsAndFlags(NodeToMatch, InputChain, ChainNodesMatched, |
Chris Lattner | 82dd3d3 | 2010-03-02 07:50:03 +0000 | [diff] [blame] | 2521 | InputFlag, FlagResultNodesMatched, true); |
| 2522 | return Res; |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2523 | } |
| 2524 | |
| 2525 | continue; |
| 2526 | } |
| 2527 | |
| 2528 | case OPC_MarkFlagResults: { |
| 2529 | unsigned NumNodes = MatcherTable[MatcherIndex++]; |
| 2530 | |
| 2531 | // Read and remember all the flag-result nodes. |
| 2532 | for (unsigned i = 0; i != NumNodes; ++i) { |
| 2533 | unsigned RecNo = MatcherTable[MatcherIndex++]; |
| 2534 | if (RecNo & 128) |
| 2535 | RecNo = GetVBR(RecNo, MatcherTable, MatcherIndex); |
| 2536 | |
| 2537 | assert(RecNo < RecordedNodes.size() && "Invalid CheckSame"); |
| 2538 | FlagResultNodesMatched.push_back(RecordedNodes[RecNo].getNode()); |
| 2539 | } |
| 2540 | continue; |
| 2541 | } |
| 2542 | |
| 2543 | case OPC_CompleteMatch: { |
| 2544 | // The match has been completed, and any new nodes (if any) have been |
| 2545 | // created. Patch up references to the matched dag to use the newly |
| 2546 | // created nodes. |
| 2547 | unsigned NumResults = MatcherTable[MatcherIndex++]; |
| 2548 | |
| 2549 | for (unsigned i = 0; i != NumResults; ++i) { |
| 2550 | unsigned ResSlot = MatcherTable[MatcherIndex++]; |
| 2551 | if (ResSlot & 128) |
| 2552 | ResSlot = GetVBR(ResSlot, MatcherTable, MatcherIndex); |
| 2553 | |
| 2554 | assert(ResSlot < RecordedNodes.size() && "Invalid CheckSame"); |
| 2555 | SDValue Res = RecordedNodes[ResSlot]; |
| 2556 | |
Chris Lattner | b0e483e | 2010-03-28 05:54:03 +0000 | [diff] [blame] | 2557 | assert(i < NodeToMatch->getNumValues() && |
| 2558 | NodeToMatch->getValueType(i) != MVT::Other && |
| 2559 | NodeToMatch->getValueType(i) != MVT::Flag && |
| 2560 | "Invalid number of results to complete!"); |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2561 | assert((NodeToMatch->getValueType(i) == Res.getValueType() || |
| 2562 | NodeToMatch->getValueType(i) == MVT::iPTR || |
| 2563 | Res.getValueType() == MVT::iPTR || |
| 2564 | NodeToMatch->getValueType(i).getSizeInBits() == |
| 2565 | Res.getValueType().getSizeInBits()) && |
| 2566 | "invalid replacement"); |
| 2567 | CurDAG->ReplaceAllUsesOfValueWith(SDValue(NodeToMatch, i), Res); |
| 2568 | } |
| 2569 | |
| 2570 | // If the root node defines a flag, add it to the flag nodes to update |
| 2571 | // list. |
| 2572 | if (NodeToMatch->getValueType(NodeToMatch->getNumValues()-1) == MVT::Flag) |
| 2573 | FlagResultNodesMatched.push_back(NodeToMatch); |
| 2574 | |
| 2575 | // Update chain and flag uses. |
| 2576 | UpdateChainsAndFlags(NodeToMatch, InputChain, ChainNodesMatched, |
Chris Lattner | 82dd3d3 | 2010-03-02 07:50:03 +0000 | [diff] [blame] | 2577 | InputFlag, FlagResultNodesMatched, false); |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2578 | |
| 2579 | assert(NodeToMatch->use_empty() && |
| 2580 | "Didn't replace all uses of the node?"); |
| 2581 | |
| 2582 | // FIXME: We just return here, which interacts correctly with SelectRoot |
| 2583 | // above. We should fix this to not return an SDNode* anymore. |
| 2584 | return 0; |
| 2585 | } |
| 2586 | } |
| 2587 | |
| 2588 | // If the code reached this point, then the match failed. See if there is |
| 2589 | // another child to try in the current 'Scope', otherwise pop it until we |
| 2590 | // find a case to check. |
Dan Gohman | 19b3826 | 2010-03-09 02:15:05 +0000 | [diff] [blame] | 2591 | DEBUG(errs() << " Match failed at index " << CurrentOpcodeIndex << "\n"); |
Chris Lattner | bed993d | 2010-03-28 19:46:56 +0000 | [diff] [blame] | 2592 | ++NumDAGIselRetries; |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2593 | while (1) { |
| 2594 | if (MatchScopes.empty()) { |
| 2595 | CannotYetSelect(NodeToMatch); |
| 2596 | return 0; |
| 2597 | } |
| 2598 | |
| 2599 | // Restore the interpreter state back to the point where the scope was |
| 2600 | // formed. |
| 2601 | MatchScope &LastScope = MatchScopes.back(); |
| 2602 | RecordedNodes.resize(LastScope.NumRecordedNodes); |
| 2603 | NodeStack.clear(); |
| 2604 | NodeStack.append(LastScope.NodeStack.begin(), LastScope.NodeStack.end()); |
| 2605 | N = NodeStack.back(); |
| 2606 | |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2607 | if (LastScope.NumMatchedMemRefs != MatchedMemRefs.size()) |
| 2608 | MatchedMemRefs.resize(LastScope.NumMatchedMemRefs); |
| 2609 | MatcherIndex = LastScope.FailIndex; |
| 2610 | |
Dan Gohman | 19b3826 | 2010-03-09 02:15:05 +0000 | [diff] [blame] | 2611 | DEBUG(errs() << " Continuing at " << MatcherIndex << "\n"); |
| 2612 | |
Chris Lattner | 2a49d57 | 2010-02-28 22:37:22 +0000 | [diff] [blame] | 2613 | InputChain = LastScope.InputChain; |
| 2614 | InputFlag = LastScope.InputFlag; |
| 2615 | if (!LastScope.HasChainNodesMatched) |
| 2616 | ChainNodesMatched.clear(); |
| 2617 | if (!LastScope.HasFlagResultNodesMatched) |
| 2618 | FlagResultNodesMatched.clear(); |
| 2619 | |
| 2620 | // Check to see what the offset is at the new MatcherIndex. If it is zero |
| 2621 | // we have reached the end of this scope, otherwise we have another child |
| 2622 | // in the current scope to try. |
| 2623 | unsigned NumToSkip = MatcherTable[MatcherIndex++]; |
| 2624 | if (NumToSkip & 128) |
| 2625 | NumToSkip = GetVBR(NumToSkip, MatcherTable, MatcherIndex); |
| 2626 | |
| 2627 | // If we have another child in this scope to match, update FailIndex and |
| 2628 | // try it. |
| 2629 | if (NumToSkip != 0) { |
| 2630 | LastScope.FailIndex = MatcherIndex+NumToSkip; |
| 2631 | break; |
| 2632 | } |
| 2633 | |
| 2634 | // End of this scope, pop it and try the next child in the containing |
| 2635 | // scope. |
| 2636 | MatchScopes.pop_back(); |
| 2637 | } |
| 2638 | } |
| 2639 | } |
| 2640 | |
| 2641 | |
| 2642 | |
Dan Gohman | eeb3a00 | 2010-01-05 01:24:18 +0000 | [diff] [blame] | 2643 | void SelectionDAGISel::CannotYetSelect(SDNode *N) { |
Dan Gohman | e1f188f | 2009-10-29 22:30:23 +0000 | [diff] [blame] | 2644 | std::string msg; |
| 2645 | raw_string_ostream Msg(msg); |
| 2646 | Msg << "Cannot yet select: "; |
Chris Lattner | 2c4afd1 | 2010-03-04 00:21:16 +0000 | [diff] [blame] | 2647 | |
| 2648 | if (N->getOpcode() != ISD::INTRINSIC_W_CHAIN && |
| 2649 | N->getOpcode() != ISD::INTRINSIC_WO_CHAIN && |
| 2650 | N->getOpcode() != ISD::INTRINSIC_VOID) { |
| 2651 | N->printrFull(Msg, CurDAG); |
| 2652 | } else { |
| 2653 | bool HasInputChain = N->getOperand(0).getValueType() == MVT::Other; |
| 2654 | unsigned iid = |
| 2655 | cast<ConstantSDNode>(N->getOperand(HasInputChain))->getZExtValue(); |
| 2656 | if (iid < Intrinsic::num_intrinsics) |
| 2657 | Msg << "intrinsic %" << Intrinsic::getName((Intrinsic::ID)iid); |
| 2658 | else if (const TargetIntrinsicInfo *TII = TM.getIntrinsicInfo()) |
| 2659 | Msg << "target intrinsic %" << TII->getName(iid); |
| 2660 | else |
| 2661 | Msg << "unknown intrinsic #" << iid; |
| 2662 | } |
Chris Lattner | 75361b6 | 2010-04-07 22:58:41 +0000 | [diff] [blame] | 2663 | report_fatal_error(Msg.str()); |
Dan Gohman | e1f188f | 2009-10-29 22:30:23 +0000 | [diff] [blame] | 2664 | } |
| 2665 | |
Devang Patel | 1997473 | 2007-05-03 01:11:54 +0000 | [diff] [blame] | 2666 | char SelectionDAGISel::ID = 0; |