blob: 37fb129e57086bd9c4db3e8a0268dd1c6098885c [file] [log] [blame]
Chris Lattner1c08c712005-01-07 07:47:53 +00001//===-- SelectionDAGISel.cpp - Implement the SelectionDAGISel class -------===//
Misha Brukmanedf128a2005-04-21 22:36:52 +00002//
Chris Lattner1c08c712005-01-07 07:47:53 +00003// The LLVM Compiler Infrastructure
4//
Chris Lattner4ee451d2007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Misha Brukmanedf128a2005-04-21 22:36:52 +00007//
Chris Lattner1c08c712005-01-07 07:47:53 +00008//===----------------------------------------------------------------------===//
9//
10// This implements the SelectionDAGISel class.
11//
12//===----------------------------------------------------------------------===//
13
14#define DEBUG_TYPE "isel"
Dan Gohman84fbac52009-02-06 17:22:58 +000015#include "ScheduleDAGSDNodes.h"
Dan Gohman2048b852009-11-23 18:04:58 +000016#include "SelectionDAGBuilder.h"
Dan Gohman6277eb22009-11-23 17:16:22 +000017#include "FunctionLoweringInfo.h"
Dan Gohman84fbac52009-02-06 17:22:58 +000018#include "llvm/CodeGen/SelectionDAGISel.h"
Jim Laskeyc7c3f112006-10-16 20:52:31 +000019#include "llvm/Analysis/AliasAnalysis.h"
Devang Patel713f0432009-09-16 21:09:07 +000020#include "llvm/Analysis/DebugInfo.h"
Anton Korobeynikov5502bf62007-04-04 21:14:49 +000021#include "llvm/Constants.h"
Chris Lattneradf6a962005-05-13 18:50:42 +000022#include "llvm/CallingConv.h"
Chris Lattner1c08c712005-01-07 07:47:53 +000023#include "llvm/DerivedTypes.h"
24#include "llvm/Function.h"
Chris Lattner36ce6912005-11-29 06:21:05 +000025#include "llvm/GlobalVariable.h"
Chris Lattnerce7518c2006-01-26 22:24:51 +000026#include "llvm/InlineAsm.h"
Chris Lattner1c08c712005-01-07 07:47:53 +000027#include "llvm/Instructions.h"
28#include "llvm/Intrinsics.h"
Jim Laskey43970fe2006-03-23 18:06:46 +000029#include "llvm/IntrinsicInst.h"
Chris Lattner75c478a2009-10-27 17:02:08 +000030#include "llvm/LLVMContext.h"
Dan Gohman78eca172008-08-19 22:33:34 +000031#include "llvm/CodeGen/FastISel.h"
Gordon Henriksen5a29c9e2008-08-17 12:56:54 +000032#include "llvm/CodeGen/GCStrategy.h"
Gordon Henriksen5eca0752008-08-17 18:44:35 +000033#include "llvm/CodeGen/GCMetadata.h"
Chris Lattner1c08c712005-01-07 07:47:53 +000034#include "llvm/CodeGen/MachineFunction.h"
Dan Gohmanad2afc22009-07-31 18:16:33 +000035#include "llvm/CodeGen/MachineFunctionAnalysis.h"
Chris Lattner1c08c712005-01-07 07:47:53 +000036#include "llvm/CodeGen/MachineFrameInfo.h"
37#include "llvm/CodeGen/MachineInstrBuilder.h"
Chris Lattner84bc5422007-12-31 04:13:23 +000038#include "llvm/CodeGen/MachineJumpTableInfo.h"
39#include "llvm/CodeGen/MachineModuleInfo.h"
40#include "llvm/CodeGen/MachineRegisterInfo.h"
Dan Gohmanfc54c552009-01-15 22:18:12 +000041#include "llvm/CodeGen/ScheduleHazardRecognizer.h"
Jim Laskeyeb577ba2006-08-02 12:30:23 +000042#include "llvm/CodeGen/SchedulerRegistry.h"
Chris Lattner1c08c712005-01-07 07:47:53 +000043#include "llvm/CodeGen/SelectionDAG.h"
Devang Patel6e7a1612009-01-09 19:11:50 +000044#include "llvm/CodeGen/DwarfWriter.h"
Dan Gohman6f0d0242008-02-10 18:45:23 +000045#include "llvm/Target/TargetRegisterInfo.h"
Chris Lattner1c08c712005-01-07 07:47:53 +000046#include "llvm/Target/TargetData.h"
47#include "llvm/Target/TargetFrameInfo.h"
Dan Gohmane1f188f2009-10-29 22:30:23 +000048#include "llvm/Target/TargetIntrinsicInfo.h"
Chris Lattner1c08c712005-01-07 07:47:53 +000049#include "llvm/Target/TargetInstrInfo.h"
50#include "llvm/Target/TargetLowering.h"
51#include "llvm/Target/TargetMachine.h"
Vladimir Prus12472912006-05-23 13:43:15 +000052#include "llvm/Target/TargetOptions.h"
Chris Lattnera4f0b3a2006-08-27 12:54:02 +000053#include "llvm/Support/Compiler.h"
Evan Chengdb8d56b2008-06-30 20:45:06 +000054#include "llvm/Support/Debug.h"
Torok Edwin7d696d82009-07-11 13:10:19 +000055#include "llvm/Support/ErrorHandling.h"
Evan Chengdb8d56b2008-06-30 20:45:06 +000056#include "llvm/Support/MathExtras.h"
57#include "llvm/Support/Timer.h"
Daniel Dunbarce63ffb2009-07-25 00:23:56 +000058#include "llvm/Support/raw_ostream.h"
Chris Lattnerc19ae9d2010-03-04 19:46:56 +000059#include "llvm/ADT/Statistic.h"
Jeff Cohen7e881032006-02-24 02:52:40 +000060#include <algorithm>
Chris Lattner1c08c712005-01-07 07:47:53 +000061using namespace llvm;
62
Chris Lattnerc19ae9d2010-03-04 19:46:56 +000063STATISTIC(NumFastIselFailures, "Number of instructions fast isel failed on");
Chris Lattnerbed993d2010-03-28 19:46:56 +000064STATISTIC(NumDAGIselRetries,"Number of times dag isel has to try another path");
Chris Lattnerc19ae9d2010-03-04 19:46:56 +000065
Chris Lattneread0d882008-06-17 06:09:18 +000066static cl::opt<bool>
Dan Gohman293d5f82008-09-09 22:06:46 +000067EnableFastISelVerbose("fast-isel-verbose", cl::Hidden,
Dan Gohmand659d502008-10-20 21:30:12 +000068 cl::desc("Enable verbose messages in the \"fast\" "
Dan Gohman293d5f82008-09-09 22:06:46 +000069 "instruction selector"));
70static cl::opt<bool>
Dan Gohman4344a5d2008-09-09 23:05:00 +000071EnableFastISelAbort("fast-isel-abort", cl::Hidden,
72 cl::desc("Enable abort calls when \"fast\" instruction fails"));
Dan Gohman8a110532008-09-05 22:59:21 +000073static cl::opt<bool>
Evan Chengdf8ed022009-11-09 06:49:37 +000074SchedLiveInCopies("schedule-livein-copies", cl::Hidden,
Dan Gohman8a110532008-09-05 22:59:21 +000075 cl::desc("Schedule copies of livein registers"),
76 cl::init(false));
Chris Lattneread0d882008-06-17 06:09:18 +000077
Chris Lattnerda8abb02005-09-01 18:44:10 +000078#ifndef NDEBUG
Chris Lattner7944d9d2005-01-12 03:41:21 +000079static cl::opt<bool>
Dan Gohman462dc7f2008-07-21 20:00:07 +000080ViewDAGCombine1("view-dag-combine1-dags", cl::Hidden,
81 cl::desc("Pop up a window to show dags before the first "
82 "dag combine pass"));
83static cl::opt<bool>
84ViewLegalizeTypesDAGs("view-legalize-types-dags", cl::Hidden,
85 cl::desc("Pop up a window to show dags before legalize types"));
86static cl::opt<bool>
87ViewLegalizeDAGs("view-legalize-dags", cl::Hidden,
88 cl::desc("Pop up a window to show dags before legalize"));
89static cl::opt<bool>
90ViewDAGCombine2("view-dag-combine2-dags", cl::Hidden,
91 cl::desc("Pop up a window to show dags before the second "
92 "dag combine pass"));
93static cl::opt<bool>
Duncan Sands25cf2272008-11-24 14:53:14 +000094ViewDAGCombineLT("view-dag-combine-lt-dags", cl::Hidden,
95 cl::desc("Pop up a window to show dags before the post legalize types"
96 " dag combine pass"));
97static cl::opt<bool>
Evan Chenga9c20912006-01-21 02:32:06 +000098ViewISelDAGs("view-isel-dags", cl::Hidden,
99 cl::desc("Pop up a window to show isel dags as they are selected"));
100static cl::opt<bool>
101ViewSchedDAGs("view-sched-dags", cl::Hidden,
102 cl::desc("Pop up a window to show sched dags as they are processed"));
Dan Gohman3e1a7ae2007-08-28 20:32:58 +0000103static cl::opt<bool>
104ViewSUnitDAGs("view-sunit-dags", cl::Hidden,
Chris Lattner5bab7852008-01-25 17:24:52 +0000105 cl::desc("Pop up a window to show SUnit dags after they are processed"));
Chris Lattner7944d9d2005-01-12 03:41:21 +0000106#else
Dan Gohman462dc7f2008-07-21 20:00:07 +0000107static const bool ViewDAGCombine1 = false,
108 ViewLegalizeTypesDAGs = false, ViewLegalizeDAGs = false,
109 ViewDAGCombine2 = false,
Duncan Sands25cf2272008-11-24 14:53:14 +0000110 ViewDAGCombineLT = false,
Dan Gohman462dc7f2008-07-21 20:00:07 +0000111 ViewISelDAGs = false, ViewSchedDAGs = false,
112 ViewSUnitDAGs = false;
Chris Lattner7944d9d2005-01-12 03:41:21 +0000113#endif
114
Jim Laskeyeb577ba2006-08-02 12:30:23 +0000115//===---------------------------------------------------------------------===//
116///
117/// RegisterScheduler class - Track the registration of instruction schedulers.
118///
119//===---------------------------------------------------------------------===//
120MachinePassRegistry RegisterScheduler::Registry;
121
122//===---------------------------------------------------------------------===//
123///
124/// ISHeuristic command line option for instruction schedulers.
125///
126//===---------------------------------------------------------------------===//
Dan Gohman844731a2008-05-13 00:00:25 +0000127static cl::opt<RegisterScheduler::FunctionPassCtor, false,
128 RegisterPassParser<RegisterScheduler> >
129ISHeuristic("pre-RA-sched",
130 cl::init(&createDefaultScheduler),
131 cl::desc("Instruction schedulers available (before register"
132 " allocation):"));
Jim Laskey13ec7022006-08-01 14:21:23 +0000133
Dan Gohman844731a2008-05-13 00:00:25 +0000134static RegisterScheduler
Dan Gohmanb8cab922008-10-14 20:25:08 +0000135defaultListDAGScheduler("default", "Best scheduler for the target",
Dan Gohman844731a2008-05-13 00:00:25 +0000136 createDefaultScheduler);
Evan Cheng4ef10862006-01-23 07:01:07 +0000137
Chris Lattner1c08c712005-01-07 07:47:53 +0000138namespace llvm {
139 //===--------------------------------------------------------------------===//
Jim Laskey9373beb2006-08-01 19:14:14 +0000140 /// createDefaultScheduler - This creates an instruction scheduler appropriate
141 /// for the target.
Dan Gohman47ac0f02009-02-11 04:27:20 +0000142 ScheduleDAGSDNodes* createDefaultScheduler(SelectionDAGISel *IS,
Bill Wendling98a366d2009-04-29 23:29:43 +0000143 CodeGenOpt::Level OptLevel) {
Dan Gohmane9530ec2009-01-15 16:58:17 +0000144 const TargetLowering &TLI = IS->getTargetLowering();
145
Bill Wendling98a366d2009-04-29 23:29:43 +0000146 if (OptLevel == CodeGenOpt::None)
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000147 return createFastDAGScheduler(IS, OptLevel);
Dan Gohman9e76fea2008-11-20 03:11:19 +0000148 if (TLI.getSchedulingPreference() == TargetLowering::SchedulingForLatency)
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000149 return createTDListDAGScheduler(IS, OptLevel);
Dan Gohman9e76fea2008-11-20 03:11:19 +0000150 assert(TLI.getSchedulingPreference() ==
Bill Wendling187361b2010-01-23 10:26:57 +0000151 TargetLowering::SchedulingForRegPressure && "Unknown sched type!");
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000152 return createBURRListDAGScheduler(IS, OptLevel);
Jim Laskey9373beb2006-08-01 19:14:14 +0000153 }
Chris Lattner1c08c712005-01-07 07:47:53 +0000154}
155
Evan Chengff9b3732008-01-30 18:18:23 +0000156// EmitInstrWithCustomInserter - This method should be implemented by targets
Dan Gohman533297b2009-10-29 18:10:34 +0000157// that mark instructions with the 'usesCustomInserter' flag. These
Chris Lattner025c39b2005-08-26 20:54:47 +0000158// instructions are special in various ways, which require special support to
159// insert. The specified MachineInstr is created but not inserted into any
Dan Gohman533297b2009-10-29 18:10:34 +0000160// basic blocks, and this method is called to expand it into a sequence of
161// instructions, potentially also creating new basic blocks and control flow.
162// When new basic blocks are inserted and the edges from MBB to its successors
163// are modified, the method should insert pairs of <OldSucc, NewSucc> into the
164// DenseMap.
Evan Chengff9b3732008-01-30 18:18:23 +0000165MachineBasicBlock *TargetLowering::EmitInstrWithCustomInserter(MachineInstr *MI,
Evan Chengfb2e7522009-09-18 21:02:19 +0000166 MachineBasicBlock *MBB,
167 DenseMap<MachineBasicBlock*, MachineBasicBlock*> *EM) const {
Torok Edwinf3689232009-07-12 20:07:01 +0000168#ifndef NDEBUG
David Greene1a053232010-01-05 01:26:11 +0000169 dbgs() << "If a target marks an instruction with "
Dan Gohman533297b2009-10-29 18:10:34 +0000170 "'usesCustomInserter', it must implement "
Torok Edwinf3689232009-07-12 20:07:01 +0000171 "TargetLowering::EmitInstrWithCustomInserter!";
172#endif
Torok Edwinc23197a2009-07-14 16:55:14 +0000173 llvm_unreachable(0);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000174 return 0;
Chris Lattner025c39b2005-08-26 20:54:47 +0000175}
176
Dan Gohman8a110532008-09-05 22:59:21 +0000177/// EmitLiveInCopy - Emit a copy for a live in physical register. If the
178/// physical register has only a single copy use, then coalesced the copy
179/// if possible.
180static void EmitLiveInCopy(MachineBasicBlock *MBB,
181 MachineBasicBlock::iterator &InsertPos,
182 unsigned VirtReg, unsigned PhysReg,
183 const TargetRegisterClass *RC,
184 DenseMap<MachineInstr*, unsigned> &CopyRegMap,
185 const MachineRegisterInfo &MRI,
186 const TargetRegisterInfo &TRI,
187 const TargetInstrInfo &TII) {
188 unsigned NumUses = 0;
189 MachineInstr *UseMI = NULL;
190 for (MachineRegisterInfo::use_iterator UI = MRI.use_begin(VirtReg),
191 UE = MRI.use_end(); UI != UE; ++UI) {
192 UseMI = &*UI;
193 if (++NumUses > 1)
194 break;
195 }
196
197 // If the number of uses is not one, or the use is not a move instruction,
198 // don't coalesce. Also, only coalesce away a virtual register to virtual
199 // register copy.
200 bool Coalesced = false;
Evan Cheng04ee5a12009-01-20 19:12:24 +0000201 unsigned SrcReg, DstReg, SrcSubReg, DstSubReg;
Dan Gohman8a110532008-09-05 22:59:21 +0000202 if (NumUses == 1 &&
Evan Cheng04ee5a12009-01-20 19:12:24 +0000203 TII.isMoveInstr(*UseMI, SrcReg, DstReg, SrcSubReg, DstSubReg) &&
Dan Gohman8a110532008-09-05 22:59:21 +0000204 TargetRegisterInfo::isVirtualRegister(DstReg)) {
205 VirtReg = DstReg;
206 Coalesced = true;
207 }
208
209 // Now find an ideal location to insert the copy.
210 MachineBasicBlock::iterator Pos = InsertPos;
211 while (Pos != MBB->begin()) {
212 MachineInstr *PrevMI = prior(Pos);
213 DenseMap<MachineInstr*, unsigned>::iterator RI = CopyRegMap.find(PrevMI);
214 // copyRegToReg might emit multiple instructions to do a copy.
215 unsigned CopyDstReg = (RI == CopyRegMap.end()) ? 0 : RI->second;
216 if (CopyDstReg && !TRI.regsOverlap(CopyDstReg, PhysReg))
217 // This is what the BB looks like right now:
218 // r1024 = mov r0
219 // ...
220 // r1 = mov r1024
221 //
222 // We want to insert "r1025 = mov r1". Inserting this copy below the
223 // move to r1024 makes it impossible for that move to be coalesced.
224 //
225 // r1025 = mov r1
226 // r1024 = mov r0
227 // ...
228 // r1 = mov 1024
229 // r2 = mov 1025
230 break; // Woot! Found a good location.
231 --Pos;
232 }
233
David Goodwinf1daf7d2009-07-08 23:10:31 +0000234 bool Emitted = TII.copyRegToReg(*MBB, Pos, VirtReg, PhysReg, RC, RC);
235 assert(Emitted && "Unable to issue a live-in copy instruction!\n");
236 (void) Emitted;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000237
Zhongxing Xu931424a2009-10-16 05:42:28 +0000238 CopyRegMap.insert(std::make_pair(prior(Pos), VirtReg));
Dan Gohman8a110532008-09-05 22:59:21 +0000239 if (Coalesced) {
240 if (&*InsertPos == UseMI) ++InsertPos;
241 MBB->erase(UseMI);
242 }
243}
244
245/// EmitLiveInCopies - If this is the first basic block in the function,
246/// and if it has live ins that need to be copied into vregs, emit the
247/// copies into the block.
248static void EmitLiveInCopies(MachineBasicBlock *EntryMBB,
249 const MachineRegisterInfo &MRI,
250 const TargetRegisterInfo &TRI,
251 const TargetInstrInfo &TII) {
252 if (SchedLiveInCopies) {
253 // Emit the copies at a heuristically-determined location in the block.
254 DenseMap<MachineInstr*, unsigned> CopyRegMap;
255 MachineBasicBlock::iterator InsertPos = EntryMBB->begin();
256 for (MachineRegisterInfo::livein_iterator LI = MRI.livein_begin(),
257 E = MRI.livein_end(); LI != E; ++LI)
258 if (LI->second) {
259 const TargetRegisterClass *RC = MRI.getRegClass(LI->second);
260 EmitLiveInCopy(EntryMBB, InsertPos, LI->second, LI->first,
261 RC, CopyRegMap, MRI, TRI, TII);
262 }
263 } else {
264 // Emit the copies into the top of the block.
265 for (MachineRegisterInfo::livein_iterator LI = MRI.livein_begin(),
266 E = MRI.livein_end(); LI != E; ++LI)
267 if (LI->second) {
268 const TargetRegisterClass *RC = MRI.getRegClass(LI->second);
David Goodwinf1daf7d2009-07-08 23:10:31 +0000269 bool Emitted = TII.copyRegToReg(*EntryMBB, EntryMBB->begin(),
270 LI->second, LI->first, RC, RC);
271 assert(Emitted && "Unable to issue a live-in copy instruction!\n");
272 (void) Emitted;
Dan Gohman8a110532008-09-05 22:59:21 +0000273 }
274 }
275}
276
Chris Lattner7041ee32005-01-11 05:56:49 +0000277//===----------------------------------------------------------------------===//
278// SelectionDAGISel code
279//===----------------------------------------------------------------------===//
Chris Lattner1c08c712005-01-07 07:47:53 +0000280
Bill Wendling98a366d2009-04-29 23:29:43 +0000281SelectionDAGISel::SelectionDAGISel(TargetMachine &tm, CodeGenOpt::Level OL) :
Dan Gohmanad2afc22009-07-31 18:16:33 +0000282 MachineFunctionPass(&ID), TM(tm), TLI(*tm.getTargetLowering()),
Dan Gohman7c3234c2008-08-27 23:52:12 +0000283 FuncInfo(new FunctionLoweringInfo(TLI)),
284 CurDAG(new SelectionDAG(TLI, *FuncInfo)),
Dan Gohman2048b852009-11-23 18:04:58 +0000285 SDB(new SelectionDAGBuilder(*CurDAG, TLI, *FuncInfo, OL)),
Dan Gohman7c3234c2008-08-27 23:52:12 +0000286 GFI(),
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000287 OptLevel(OL),
Dan Gohman7c3234c2008-08-27 23:52:12 +0000288 DAGSize(0)
289{}
290
291SelectionDAGISel::~SelectionDAGISel() {
Dan Gohman2048b852009-11-23 18:04:58 +0000292 delete SDB;
Dan Gohman7c3234c2008-08-27 23:52:12 +0000293 delete CurDAG;
294 delete FuncInfo;
295}
296
Owen Andersone50ed302009-08-10 22:56:29 +0000297unsigned SelectionDAGISel::MakeReg(EVT VT) {
Chris Lattner84bc5422007-12-31 04:13:23 +0000298 return RegInfo->createVirtualRegister(TLI.getRegClassFor(VT));
Chris Lattner1c08c712005-01-07 07:47:53 +0000299}
300
Chris Lattner495a0b52005-08-17 06:37:43 +0000301void SelectionDAGISel::getAnalysisUsage(AnalysisUsage &AU) const {
Jim Laskeyc7c3f112006-10-16 20:52:31 +0000302 AU.addRequired<AliasAnalysis>();
Dan Gohmana3477fe2009-07-31 23:36:22 +0000303 AU.addPreserved<AliasAnalysis>();
Gordon Henriksen5eca0752008-08-17 18:44:35 +0000304 AU.addRequired<GCModuleInfo>();
Dan Gohmana3477fe2009-07-31 23:36:22 +0000305 AU.addPreserved<GCModuleInfo>();
Devang Patel6e7a1612009-01-09 19:11:50 +0000306 AU.addRequired<DwarfWriter>();
Dan Gohmana3477fe2009-07-31 23:36:22 +0000307 AU.addPreserved<DwarfWriter>();
Dan Gohmanad2afc22009-07-31 18:16:33 +0000308 MachineFunctionPass::getAnalysisUsage(AU);
Chris Lattner495a0b52005-08-17 06:37:43 +0000309}
Chris Lattner1c08c712005-01-07 07:47:53 +0000310
Dan Gohmanad2afc22009-07-31 18:16:33 +0000311bool SelectionDAGISel::runOnMachineFunction(MachineFunction &mf) {
312 Function &Fn = *mf.getFunction();
313
Dan Gohman4344a5d2008-09-09 23:05:00 +0000314 // Do some sanity-checking on the command-line options.
315 assert((!EnableFastISelVerbose || EnableFastISel) &&
316 "-fast-isel-verbose requires -fast-isel");
317 assert((!EnableFastISelAbort || EnableFastISel) &&
318 "-fast-isel-abort requires -fast-isel");
319
Dan Gohman5f43f922007-08-27 16:26:13 +0000320 // Get alias analysis for load/store combining.
321 AA = &getAnalysis<AliasAnalysis>();
322
Dan Gohmanad2afc22009-07-31 18:16:33 +0000323 MF = &mf;
Dan Gohman8a110532008-09-05 22:59:21 +0000324 const TargetInstrInfo &TII = *TM.getInstrInfo();
325 const TargetRegisterInfo &TRI = *TM.getRegisterInfo();
326
Dan Gohmanf7d6cd42009-08-01 03:51:09 +0000327 if (Fn.hasGC())
328 GFI = &getAnalysis<GCModuleInfo>().getFunctionInfo(Fn);
Gordon Henriksence224772008-01-07 01:30:38 +0000329 else
Gordon Henriksen5eca0752008-08-17 18:44:35 +0000330 GFI = 0;
Dan Gohman79ce2762009-01-15 19:20:50 +0000331 RegInfo = &MF->getRegInfo();
David Greene1a053232010-01-05 01:26:11 +0000332 DEBUG(dbgs() << "\n\n\n=== " << Fn.getName() << "\n");
Chris Lattner1c08c712005-01-07 07:47:53 +0000333
Duncan Sands1465d612009-01-28 13:14:17 +0000334 MachineModuleInfo *MMI = getAnalysisIfAvailable<MachineModuleInfo>();
335 DwarfWriter *DW = getAnalysisIfAvailable<DwarfWriter>();
Owen Anderson5dcaceb2009-07-09 18:44:09 +0000336 CurDAG->init(*MF, MMI, DW);
Dan Gohman6277eb22009-11-23 17:16:22 +0000337 FuncInfo->set(Fn, *MF, EnableFastISel);
Dan Gohman2048b852009-11-23 18:04:58 +0000338 SDB->init(GFI, *AA);
Chris Lattner1c08c712005-01-07 07:47:53 +0000339
Dale Johannesen1532f3d2008-04-02 00:25:04 +0000340 for (Function::iterator I = Fn.begin(), E = Fn.end(); I != E; ++I)
341 if (InvokeInst *Invoke = dyn_cast<InvokeInst>(I->getTerminator()))
342 // Mark landing pad.
Dan Gohman7c3234c2008-08-27 23:52:12 +0000343 FuncInfo->MBBMap[Invoke->getSuccessor(1)]->setIsLandingPad();
Duncan Sands9fac0b52007-06-06 10:05:18 +0000344
Dan Gohman79ce2762009-01-15 19:20:50 +0000345 SelectAllBasicBlocks(Fn, *MF, MMI, DW, TII);
Misha Brukmanedf128a2005-04-21 22:36:52 +0000346
Dan Gohman8a110532008-09-05 22:59:21 +0000347 // If the first basic block in the function has live ins that need to be
348 // copied into vregs, emit the copies into the top of the block before
349 // emitting the code for the block.
Dan Gohman79ce2762009-01-15 19:20:50 +0000350 EmitLiveInCopies(MF->begin(), *RegInfo, TRI, TII);
Dan Gohman8a110532008-09-05 22:59:21 +0000351
Evan Chengad2070c2007-02-10 02:43:39 +0000352 // Add function live-ins to entry block live-in set.
Dan Gohman8a110532008-09-05 22:59:21 +0000353 for (MachineRegisterInfo::livein_iterator I = RegInfo->livein_begin(),
354 E = RegInfo->livein_end(); I != E; ++I)
Dan Gohman79ce2762009-01-15 19:20:50 +0000355 MF->begin()->addLiveIn(I->first);
Evan Chengad2070c2007-02-10 02:43:39 +0000356
Duncan Sandsf4070822007-06-15 19:04:19 +0000357#ifndef NDEBUG
Dan Gohman7c3234c2008-08-27 23:52:12 +0000358 assert(FuncInfo->CatchInfoFound.size() == FuncInfo->CatchInfoLost.size() &&
Duncan Sandsf4070822007-06-15 19:04:19 +0000359 "Not all catch info was assigned to a landing pad!");
360#endif
361
Dan Gohman7c3234c2008-08-27 23:52:12 +0000362 FuncInfo->clear();
363
Chris Lattner1c08c712005-01-07 07:47:53 +0000364 return true;
365}
366
Dan Gohman07f111e2009-12-05 00:27:08 +0000367/// SetDebugLoc - Update MF's and SDB's DebugLocs if debug information is
368/// attached with this instruction.
Chris Lattner3990b122009-12-28 23:41:32 +0000369static void SetDebugLoc(unsigned MDDbgKind, Instruction *I,
370 SelectionDAGBuilder *SDB,
Chris Lattner0eb41982009-12-28 20:45:51 +0000371 FastISel *FastIS, MachineFunction *MF) {
Chris Lattner3990b122009-12-28 23:41:32 +0000372 if (MDNode *Dbg = I->getMetadata(MDDbgKind)) {
Chris Lattner0eb41982009-12-28 20:45:51 +0000373 DILocation DILoc(Dbg);
374 DebugLoc Loc = ExtractDebugLocation(DILoc, MF->getDebugLocInfo());
Dan Gohman07f111e2009-12-05 00:27:08 +0000375
Chris Lattner0eb41982009-12-28 20:45:51 +0000376 SDB->setCurDebugLoc(Loc);
Dan Gohman07f111e2009-12-05 00:27:08 +0000377
Chris Lattner0eb41982009-12-28 20:45:51 +0000378 if (FastIS)
379 FastIS->setCurDebugLoc(Loc);
Dan Gohman07f111e2009-12-05 00:27:08 +0000380
Chris Lattner0eb41982009-12-28 20:45:51 +0000381 // If the function doesn't have a default debug location yet, set
382 // it. This is kind of a hack.
383 if (MF->getDefaultDebugLoc().isUnknown())
384 MF->setDefaultDebugLoc(Loc);
385 }
Dan Gohman07f111e2009-12-05 00:27:08 +0000386}
387
388/// ResetDebugLoc - Set MF's and SDB's DebugLocs to Unknown.
Chris Lattner3990b122009-12-28 23:41:32 +0000389static void ResetDebugLoc(SelectionDAGBuilder *SDB, FastISel *FastIS) {
Dan Gohman07f111e2009-12-05 00:27:08 +0000390 SDB->setCurDebugLoc(DebugLoc::getUnknownLoc());
391 if (FastIS)
Dan Gohman688fb802009-12-14 23:08:09 +0000392 FastIS->setCurDebugLoc(DebugLoc::getUnknownLoc());
Dan Gohman07f111e2009-12-05 00:27:08 +0000393}
394
Dan Gohmanf350b272008-08-23 02:25:05 +0000395void SelectionDAGISel::SelectBasicBlock(BasicBlock *LLVMBB,
396 BasicBlock::iterator Begin,
Dan Gohmanb4afb132009-11-20 02:51:26 +0000397 BasicBlock::iterator End,
398 bool &HadTailCall) {
Dan Gohman2048b852009-11-23 18:04:58 +0000399 SDB->setCurrentBasicBlock(BB);
Chris Lattner08113472009-12-29 09:01:33 +0000400 unsigned MDDbgKind = LLVMBB->getContext().getMDKindID("dbg");
Dan Gohmanf350b272008-08-23 02:25:05 +0000401
Dan Gohman98ca4f22009-08-05 01:29:28 +0000402 // Lower all of the non-terminator instructions. If a call is emitted
403 // as a tail call, cease emitting nodes for this block.
Dan Gohman2048b852009-11-23 18:04:58 +0000404 for (BasicBlock::iterator I = Begin; I != End && !SDB->HasTailCall; ++I) {
Chris Lattner3990b122009-12-28 23:41:32 +0000405 SetDebugLoc(MDDbgKind, I, SDB, 0, MF);
Dan Gohman07f111e2009-12-05 00:27:08 +0000406
407 if (!isa<TerminatorInst>(I)) {
Dan Gohman2048b852009-11-23 18:04:58 +0000408 SDB->visit(*I);
Dan Gohman07f111e2009-12-05 00:27:08 +0000409
410 // Set the current debug location back to "unknown" so that it doesn't
411 // spuriously apply to subsequent instructions.
412 ResetDebugLoc(SDB, 0);
413 }
Devang Patel123eaa72009-09-16 20:39:11 +0000414 }
Dan Gohmanf350b272008-08-23 02:25:05 +0000415
Dan Gohman2048b852009-11-23 18:04:58 +0000416 if (!SDB->HasTailCall) {
Dan Gohman98ca4f22009-08-05 01:29:28 +0000417 // Ensure that all instructions which are used outside of their defining
418 // blocks are available as virtual registers. Invoke is handled elsewhere.
419 for (BasicBlock::iterator I = Begin; I != End; ++I)
420 if (!isa<PHINode>(I) && !isa<InvokeInst>(I))
Dan Gohman2048b852009-11-23 18:04:58 +0000421 SDB->CopyToExportRegsIfNeeded(I);
Dan Gohmanf350b272008-08-23 02:25:05 +0000422
Dan Gohman98ca4f22009-08-05 01:29:28 +0000423 // Handle PHI nodes in successor blocks.
424 if (End == LLVMBB->end()) {
425 HandlePHINodesInSuccessorBlocks(LLVMBB);
Dan Gohman3df24e62008-09-03 23:12:08 +0000426
Dan Gohman98ca4f22009-08-05 01:29:28 +0000427 // Lower the terminator after the copies are emitted.
Chris Lattner3990b122009-12-28 23:41:32 +0000428 SetDebugLoc(MDDbgKind, LLVMBB->getTerminator(), SDB, 0, MF);
Dan Gohman2048b852009-11-23 18:04:58 +0000429 SDB->visit(*LLVMBB->getTerminator());
Dan Gohman07f111e2009-12-05 00:27:08 +0000430 ResetDebugLoc(SDB, 0);
Dan Gohman98ca4f22009-08-05 01:29:28 +0000431 }
Dan Gohman3df24e62008-09-03 23:12:08 +0000432 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000433
Chris Lattnera651cf62005-01-17 19:43:36 +0000434 // Make sure the root of the DAG is up-to-date.
Dan Gohman2048b852009-11-23 18:04:58 +0000435 CurDAG->setRoot(SDB->getControlRoot());
Arnold Schwaighoferc85e1712007-10-11 19:40:01 +0000436
Dan Gohmanf350b272008-08-23 02:25:05 +0000437 // Final step, emit the lowered DAG as machine code.
438 CodeGenAndEmitDAG();
Dan Gohman2048b852009-11-23 18:04:58 +0000439 HadTailCall = SDB->HasTailCall;
440 SDB->clear();
Chris Lattner1c08c712005-01-07 07:47:53 +0000441}
442
Evan Cheng54e146b2010-01-09 00:21:08 +0000443namespace {
444/// WorkListRemover - This class is a DAGUpdateListener that removes any deleted
445/// nodes from the worklist.
446class SDOPsWorkListRemover : public SelectionDAG::DAGUpdateListener {
447 SmallVector<SDNode*, 128> &Worklist;
Chris Lattner25e0ab92010-03-14 19:43:04 +0000448 SmallPtrSet<SDNode*, 128> &InWorklist;
Evan Cheng54e146b2010-01-09 00:21:08 +0000449public:
Chris Lattner25e0ab92010-03-14 19:43:04 +0000450 SDOPsWorkListRemover(SmallVector<SDNode*, 128> &wl,
451 SmallPtrSet<SDNode*, 128> &inwl)
452 : Worklist(wl), InWorklist(inwl) {}
Evan Cheng54e146b2010-01-09 00:21:08 +0000453
Chris Lattner25e0ab92010-03-14 19:43:04 +0000454 void RemoveFromWorklist(SDNode *N) {
455 if (!InWorklist.erase(N)) return;
456
457 SmallVector<SDNode*, 128>::iterator I =
458 std::find(Worklist.begin(), Worklist.end(), N);
459 assert(I != Worklist.end() && "Not in worklist");
460
461 *I = Worklist.back();
462 Worklist.pop_back();
463 }
464
Evan Cheng54e146b2010-01-09 00:21:08 +0000465 virtual void NodeDeleted(SDNode *N, SDNode *E) {
Chris Lattner25e0ab92010-03-14 19:43:04 +0000466 RemoveFromWorklist(N);
Evan Cheng54e146b2010-01-09 00:21:08 +0000467 }
468
469 virtual void NodeUpdated(SDNode *N) {
470 // Ignore updates.
471 }
472};
473}
474
Evan Cheng046632f2010-02-10 02:17:34 +0000475/// TrivialTruncElim - Eliminate some trivial nops that can result from
476/// ShrinkDemandedOps: (trunc (ext n)) -> n.
477static bool TrivialTruncElim(SDValue Op,
478 TargetLowering::TargetLoweringOpt &TLO) {
479 SDValue N0 = Op.getOperand(0);
480 EVT VT = Op.getValueType();
481 if ((N0.getOpcode() == ISD::ZERO_EXTEND ||
482 N0.getOpcode() == ISD::SIGN_EXTEND ||
483 N0.getOpcode() == ISD::ANY_EXTEND) &&
484 N0.getOperand(0).getValueType() == VT) {
485 return TLO.CombineTo(Op, N0.getOperand(0));
486 }
487 return false;
488}
489
Evan Cheng54eb4c22010-01-06 19:43:21 +0000490/// ShrinkDemandedOps - A late transformation pass that shrink expressions
491/// using TargetLowering::TargetLoweringOpt::ShrinkDemandedOp. It converts
492/// x+y to (VT)((SmallVT)x+(SmallVT)y) if the casts are free.
Evan Chengd40d03e2010-01-06 19:38:29 +0000493void SelectionDAGISel::ShrinkDemandedOps() {
494 SmallVector<SDNode*, 128> Worklist;
Chris Lattner25e0ab92010-03-14 19:43:04 +0000495 SmallPtrSet<SDNode*, 128> InWorklist;
Evan Chengd40d03e2010-01-06 19:38:29 +0000496
497 // Add all the dag nodes to the worklist.
498 Worklist.reserve(CurDAG->allnodes_size());
499 for (SelectionDAG::allnodes_iterator I = CurDAG->allnodes_begin(),
Chris Lattner25e0ab92010-03-14 19:43:04 +0000500 E = CurDAG->allnodes_end(); I != E; ++I) {
Evan Chengd40d03e2010-01-06 19:38:29 +0000501 Worklist.push_back(I);
Chris Lattner25e0ab92010-03-14 19:43:04 +0000502 InWorklist.insert(I);
503 }
Evan Chengd40d03e2010-01-06 19:38:29 +0000504
505 TargetLowering::TargetLoweringOpt TLO(*CurDAG, true);
506 while (!Worklist.empty()) {
Benjamin Kramer7b1e2a52010-01-07 17:27:56 +0000507 SDNode *N = Worklist.pop_back_val();
Chris Lattner25e0ab92010-03-14 19:43:04 +0000508 InWorklist.erase(N);
Evan Chengd40d03e2010-01-06 19:38:29 +0000509
510 if (N->use_empty() && N != CurDAG->getRoot().getNode()) {
Chris Lattnerc4a3f232010-03-14 19:46:02 +0000511 // Deleting this node may make its operands dead, add them to the worklist
512 // if they aren't already there.
513 for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i)
514 if (InWorklist.insert(N->getOperand(i).getNode()))
515 Worklist.push_back(N->getOperand(i).getNode());
516
Evan Cheng54e146b2010-01-09 00:21:08 +0000517 CurDAG->DeleteNode(N);
Evan Chengd40d03e2010-01-06 19:38:29 +0000518 continue;
519 }
520
521 // Run ShrinkDemandedOp on scalar binary operations.
Chris Lattner25e0ab92010-03-14 19:43:04 +0000522 if (N->getNumValues() != 1 ||
523 !N->getValueType(0).isSimple() || !N->getValueType(0).isInteger())
524 continue;
525
526 unsigned BitWidth = N->getValueType(0).getScalarType().getSizeInBits();
527 APInt Demanded = APInt::getAllOnesValue(BitWidth);
528 APInt KnownZero, KnownOne;
529 if (!TLI.SimplifyDemandedBits(SDValue(N, 0), Demanded,
530 KnownZero, KnownOne, TLO) &&
531 (N->getOpcode() != ISD::TRUNCATE ||
532 !TrivialTruncElim(SDValue(N, 0), TLO)))
533 continue;
534
535 // Revisit the node.
536 assert(!InWorklist.count(N) && "Already in worklist");
537 Worklist.push_back(N);
538 InWorklist.insert(N);
Evan Chengd40d03e2010-01-06 19:38:29 +0000539
Chris Lattner25e0ab92010-03-14 19:43:04 +0000540 // Replace the old value with the new one.
541 DEBUG(errs() << "\nShrinkDemandedOps replacing ";
542 TLO.Old.getNode()->dump(CurDAG);
543 errs() << "\nWith: ";
544 TLO.New.getNode()->dump(CurDAG);
545 errs() << '\n');
Evan Chengd40d03e2010-01-06 19:38:29 +0000546
Chris Lattner25e0ab92010-03-14 19:43:04 +0000547 if (InWorklist.insert(TLO.New.getNode()))
548 Worklist.push_back(TLO.New.getNode());
Evan Cheng54e146b2010-01-09 00:21:08 +0000549
Chris Lattner25e0ab92010-03-14 19:43:04 +0000550 SDOPsWorkListRemover DeadNodes(Worklist, InWorklist);
551 CurDAG->ReplaceAllUsesOfValueWith(TLO.Old, TLO.New, &DeadNodes);
Evan Chengd40d03e2010-01-06 19:38:29 +0000552
Chris Lattner25e0ab92010-03-14 19:43:04 +0000553 if (!TLO.Old.getNode()->use_empty()) continue;
554
555 for (unsigned i = 0, e = TLO.Old.getNode()->getNumOperands();
556 i != e; ++i) {
557 SDNode *OpNode = TLO.Old.getNode()->getOperand(i).getNode();
558 if (OpNode->hasOneUse()) {
559 // Add OpNode to the end of the list to revisit.
560 DeadNodes.RemoveFromWorklist(OpNode);
561 Worklist.push_back(OpNode);
562 InWorklist.insert(OpNode);
Evan Chengd40d03e2010-01-06 19:38:29 +0000563 }
564 }
Chris Lattner25e0ab92010-03-14 19:43:04 +0000565
566 DeadNodes.RemoveFromWorklist(TLO.Old.getNode());
567 CurDAG->DeleteNode(TLO.Old.getNode());
Evan Chengd40d03e2010-01-06 19:38:29 +0000568 }
569}
570
Dan Gohmanf350b272008-08-23 02:25:05 +0000571void SelectionDAGISel::ComputeLiveOutVRegInfo() {
Chris Lattneread0d882008-06-17 06:09:18 +0000572 SmallPtrSet<SDNode*, 128> VisitedNodes;
573 SmallVector<SDNode*, 128> Worklist;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000574
Gabor Greifba36cb52008-08-28 21:40:38 +0000575 Worklist.push_back(CurDAG->getRoot().getNode());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000576
Chris Lattneread0d882008-06-17 06:09:18 +0000577 APInt Mask;
578 APInt KnownZero;
579 APInt KnownOne;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000580
Benjamin Kramer7b1e2a52010-01-07 17:27:56 +0000581 do {
582 SDNode *N = Worklist.pop_back_val();
Daniel Dunbara279bc32009-09-20 02:20:51 +0000583
Chris Lattneread0d882008-06-17 06:09:18 +0000584 // If we've already seen this node, ignore it.
585 if (!VisitedNodes.insert(N))
586 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000587
Chris Lattneread0d882008-06-17 06:09:18 +0000588 // Otherwise, add all chain operands to the worklist.
589 for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i)
Owen Anderson825b72b2009-08-11 20:47:22 +0000590 if (N->getOperand(i).getValueType() == MVT::Other)
Gabor Greifba36cb52008-08-28 21:40:38 +0000591 Worklist.push_back(N->getOperand(i).getNode());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000592
Chris Lattneread0d882008-06-17 06:09:18 +0000593 // If this is a CopyToReg with a vreg dest, process it.
594 if (N->getOpcode() != ISD::CopyToReg)
595 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000596
Chris Lattneread0d882008-06-17 06:09:18 +0000597 unsigned DestReg = cast<RegisterSDNode>(N->getOperand(1))->getReg();
598 if (!TargetRegisterInfo::isVirtualRegister(DestReg))
599 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000600
Chris Lattneread0d882008-06-17 06:09:18 +0000601 // Ignore non-scalar or non-integer values.
Dan Gohman475871a2008-07-27 21:46:04 +0000602 SDValue Src = N->getOperand(2);
Owen Andersone50ed302009-08-10 22:56:29 +0000603 EVT SrcVT = Src.getValueType();
Chris Lattneread0d882008-06-17 06:09:18 +0000604 if (!SrcVT.isInteger() || SrcVT.isVector())
605 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000606
Dan Gohmanf350b272008-08-23 02:25:05 +0000607 unsigned NumSignBits = CurDAG->ComputeNumSignBits(Src);
Chris Lattneread0d882008-06-17 06:09:18 +0000608 Mask = APInt::getAllOnesValue(SrcVT.getSizeInBits());
Dan Gohmanf350b272008-08-23 02:25:05 +0000609 CurDAG->ComputeMaskedBits(Src, Mask, KnownZero, KnownOne);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000610
Chris Lattneread0d882008-06-17 06:09:18 +0000611 // Only install this information if it tells us something.
612 if (NumSignBits != 1 || KnownZero != 0 || KnownOne != 0) {
613 DestReg -= TargetRegisterInfo::FirstVirtualRegister;
Dan Gohmanf7d6cd42009-08-01 03:51:09 +0000614 if (DestReg >= FuncInfo->LiveOutRegInfo.size())
615 FuncInfo->LiveOutRegInfo.resize(DestReg+1);
616 FunctionLoweringInfo::LiveOutInfo &LOI =
617 FuncInfo->LiveOutRegInfo[DestReg];
Chris Lattneread0d882008-06-17 06:09:18 +0000618 LOI.NumSignBits = NumSignBits;
Dan Gohmana80efce2009-03-27 23:55:04 +0000619 LOI.KnownOne = KnownOne;
620 LOI.KnownZero = KnownZero;
Chris Lattneread0d882008-06-17 06:09:18 +0000621 }
Benjamin Kramer7b1e2a52010-01-07 17:27:56 +0000622 } while (!Worklist.empty());
Chris Lattneread0d882008-06-17 06:09:18 +0000623}
624
Dan Gohmanf350b272008-08-23 02:25:05 +0000625void SelectionDAGISel::CodeGenAndEmitDAG() {
Dan Gohman462dc7f2008-07-21 20:00:07 +0000626 std::string GroupName;
627 if (TimePassesIsEnabled)
628 GroupName = "Instruction Selection and Scheduling";
629 std::string BlockName;
630 if (ViewDAGCombine1 || ViewLegalizeTypesDAGs || ViewLegalizeDAGs ||
Duncan Sands25cf2272008-11-24 14:53:14 +0000631 ViewDAGCombine2 || ViewDAGCombineLT || ViewISelDAGs || ViewSchedDAGs ||
632 ViewSUnitDAGs)
Dan Gohmanf7d6cd42009-08-01 03:51:09 +0000633 BlockName = MF->getFunction()->getNameStr() + ":" +
Daniel Dunbarf6ccee52009-07-24 08:24:36 +0000634 BB->getBasicBlock()->getNameStr();
Dan Gohman462dc7f2008-07-21 20:00:07 +0000635
David Greene1a053232010-01-05 01:26:11 +0000636 DEBUG(dbgs() << "Initial selection DAG:\n");
Dan Gohmanf350b272008-08-23 02:25:05 +0000637 DEBUG(CurDAG->dump());
Dan Gohman462dc7f2008-07-21 20:00:07 +0000638
Dan Gohmanf350b272008-08-23 02:25:05 +0000639 if (ViewDAGCombine1) CurDAG->viewGraph("dag-combine1 input for " + BlockName);
Dan Gohman417e11b2007-10-08 15:12:17 +0000640
Chris Lattneraf21d552005-10-10 16:47:10 +0000641 // Run the DAG combiner in pre-legalize mode.
Evan Chengebffb662008-07-01 17:59:20 +0000642 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000643 NamedRegionTimer T("DAG Combining 1", GroupName);
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000644 CurDAG->Combine(Unrestricted, *AA, OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000645 } else {
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000646 CurDAG->Combine(Unrestricted, *AA, OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000647 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000648
David Greene1a053232010-01-05 01:26:11 +0000649 DEBUG(dbgs() << "Optimized lowered selection DAG:\n");
Dan Gohmanf350b272008-08-23 02:25:05 +0000650 DEBUG(CurDAG->dump());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000651
Chris Lattner1c08c712005-01-07 07:47:53 +0000652 // Second step, hack on the DAG until it only uses operations and types that
653 // the target supports.
Dan Gohman714efc62009-12-05 17:51:33 +0000654 if (ViewLegalizeTypesDAGs) CurDAG->viewGraph("legalize-types input for " +
655 BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000656
Dan Gohman714efc62009-12-05 17:51:33 +0000657 bool Changed;
658 if (TimePassesIsEnabled) {
659 NamedRegionTimer T("Type Legalization", GroupName);
660 Changed = CurDAG->LegalizeTypes();
661 } else {
662 Changed = CurDAG->LegalizeTypes();
663 }
664
David Greene1a053232010-01-05 01:26:11 +0000665 DEBUG(dbgs() << "Type-legalized selection DAG:\n");
Dan Gohman714efc62009-12-05 17:51:33 +0000666 DEBUG(CurDAG->dump());
667
668 if (Changed) {
669 if (ViewDAGCombineLT)
670 CurDAG->viewGraph("dag-combine-lt input for " + BlockName);
671
672 // Run the DAG combiner in post-type-legalize mode.
Dan Gohman462dc7f2008-07-21 20:00:07 +0000673 if (TimePassesIsEnabled) {
Dan Gohman714efc62009-12-05 17:51:33 +0000674 NamedRegionTimer T("DAG Combining after legalize types", GroupName);
675 CurDAG->Combine(NoIllegalTypes, *AA, OptLevel);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000676 } else {
Dan Gohman714efc62009-12-05 17:51:33 +0000677 CurDAG->Combine(NoIllegalTypes, *AA, OptLevel);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000678 }
679
David Greene1a053232010-01-05 01:26:11 +0000680 DEBUG(dbgs() << "Optimized type-legalized selection DAG:\n");
Dan Gohmanf350b272008-08-23 02:25:05 +0000681 DEBUG(CurDAG->dump());
Dan Gohman714efc62009-12-05 17:51:33 +0000682 }
Dan Gohman462dc7f2008-07-21 20:00:07 +0000683
Dan Gohman714efc62009-12-05 17:51:33 +0000684 if (TimePassesIsEnabled) {
685 NamedRegionTimer T("Vector Legalization", GroupName);
686 Changed = CurDAG->LegalizeVectors();
687 } else {
688 Changed = CurDAG->LegalizeVectors();
689 }
Duncan Sands25cf2272008-11-24 14:53:14 +0000690
Dan Gohman714efc62009-12-05 17:51:33 +0000691 if (Changed) {
Eli Friedman5c22c802009-05-23 12:35:30 +0000692 if (TimePassesIsEnabled) {
Dan Gohman714efc62009-12-05 17:51:33 +0000693 NamedRegionTimer T("Type Legalization 2", GroupName);
Bill Wendling98820072009-12-28 01:51:30 +0000694 CurDAG->LegalizeTypes();
Eli Friedman5c22c802009-05-23 12:35:30 +0000695 } else {
Bill Wendling98820072009-12-28 01:51:30 +0000696 CurDAG->LegalizeTypes();
Eli Friedman5c22c802009-05-23 12:35:30 +0000697 }
698
Dan Gohman714efc62009-12-05 17:51:33 +0000699 if (ViewDAGCombineLT)
700 CurDAG->viewGraph("dag-combine-lv input for " + BlockName);
Eli Friedman5c22c802009-05-23 12:35:30 +0000701
Dan Gohman714efc62009-12-05 17:51:33 +0000702 // Run the DAG combiner in post-type-legalize mode.
703 if (TimePassesIsEnabled) {
704 NamedRegionTimer T("DAG Combining after legalize vectors", GroupName);
705 CurDAG->Combine(NoIllegalOperations, *AA, OptLevel);
706 } else {
707 CurDAG->Combine(NoIllegalOperations, *AA, OptLevel);
Eli Friedman5c22c802009-05-23 12:35:30 +0000708 }
Dan Gohman714efc62009-12-05 17:51:33 +0000709
David Greene1a053232010-01-05 01:26:11 +0000710 DEBUG(dbgs() << "Optimized vector-legalized selection DAG:\n");
Dan Gohman714efc62009-12-05 17:51:33 +0000711 DEBUG(CurDAG->dump());
Chris Lattner70587ea2008-07-10 23:37:50 +0000712 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000713
Dan Gohmanf350b272008-08-23 02:25:05 +0000714 if (ViewLegalizeDAGs) CurDAG->viewGraph("legalize input for " + BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000715
Evan Chengebffb662008-07-01 17:59:20 +0000716 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000717 NamedRegionTimer T("DAG Legalization", GroupName);
Dan Gohman714efc62009-12-05 17:51:33 +0000718 CurDAG->Legalize(OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000719 } else {
Dan Gohman714efc62009-12-05 17:51:33 +0000720 CurDAG->Legalize(OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000721 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000722
David Greene1a053232010-01-05 01:26:11 +0000723 DEBUG(dbgs() << "Legalized selection DAG:\n");
Dan Gohmanf350b272008-08-23 02:25:05 +0000724 DEBUG(CurDAG->dump());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000725
Dan Gohmanf350b272008-08-23 02:25:05 +0000726 if (ViewDAGCombine2) CurDAG->viewGraph("dag-combine2 input for " + BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000727
Chris Lattneraf21d552005-10-10 16:47:10 +0000728 // Run the DAG combiner in post-legalize mode.
Evan Chengebffb662008-07-01 17:59:20 +0000729 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000730 NamedRegionTimer T("DAG Combining 2", GroupName);
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000731 CurDAG->Combine(NoIllegalOperations, *AA, OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000732 } else {
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000733 CurDAG->Combine(NoIllegalOperations, *AA, OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000734 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000735
David Greene1a053232010-01-05 01:26:11 +0000736 DEBUG(dbgs() << "Optimized legalized selection DAG:\n");
Dan Gohmanf350b272008-08-23 02:25:05 +0000737 DEBUG(CurDAG->dump());
Dan Gohman417e11b2007-10-08 15:12:17 +0000738
Evan Chengd40d03e2010-01-06 19:38:29 +0000739 if (OptLevel != CodeGenOpt::None) {
740 ShrinkDemandedOps();
Dan Gohmanf350b272008-08-23 02:25:05 +0000741 ComputeLiveOutVRegInfo();
Evan Chengd40d03e2010-01-06 19:38:29 +0000742 }
Evan Cheng552c4a82006-04-28 02:09:19 +0000743
Chris Lattner552186d2010-03-14 19:27:55 +0000744 if (ViewISelDAGs) CurDAG->viewGraph("isel input for " + BlockName);
745
Chris Lattnera33ef482005-03-30 01:10:47 +0000746 // Third, instruction select all of the operations to machine code, adding the
747 // code to the MachineBasicBlock.
Evan Chengebffb662008-07-01 17:59:20 +0000748 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000749 NamedRegionTimer T("Instruction Selection", GroupName);
Chris Lattner7c306da2010-03-02 06:34:30 +0000750 DoInstructionSelection();
Evan Chengebffb662008-07-01 17:59:20 +0000751 } else {
Chris Lattner7c306da2010-03-02 06:34:30 +0000752 DoInstructionSelection();
Evan Chengebffb662008-07-01 17:59:20 +0000753 }
Evan Chengdb8d56b2008-06-30 20:45:06 +0000754
David Greene1a053232010-01-05 01:26:11 +0000755 DEBUG(dbgs() << "Selected selection DAG:\n");
Dan Gohmanf350b272008-08-23 02:25:05 +0000756 DEBUG(CurDAG->dump());
Dan Gohman462dc7f2008-07-21 20:00:07 +0000757
Dan Gohmanf350b272008-08-23 02:25:05 +0000758 if (ViewSchedDAGs) CurDAG->viewGraph("scheduler input for " + BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000759
Dan Gohman5e843682008-07-14 18:19:29 +0000760 // Schedule machine code.
Dan Gohman47ac0f02009-02-11 04:27:20 +0000761 ScheduleDAGSDNodes *Scheduler = CreateScheduler();
Dan Gohman5e843682008-07-14 18:19:29 +0000762 if (TimePassesIsEnabled) {
763 NamedRegionTimer T("Instruction Scheduling", GroupName);
Dan Gohman47ac0f02009-02-11 04:27:20 +0000764 Scheduler->Run(CurDAG, BB, BB->end());
Dan Gohman5e843682008-07-14 18:19:29 +0000765 } else {
Dan Gohman47ac0f02009-02-11 04:27:20 +0000766 Scheduler->Run(CurDAG, BB, BB->end());
Dan Gohman5e843682008-07-14 18:19:29 +0000767 }
768
Dan Gohman462dc7f2008-07-21 20:00:07 +0000769 if (ViewSUnitDAGs) Scheduler->viewGraph();
770
Daniel Dunbara279bc32009-09-20 02:20:51 +0000771 // Emit machine code to BB. This can change 'BB' to the last block being
Evan Chengdb8d56b2008-06-30 20:45:06 +0000772 // inserted into.
Evan Chengebffb662008-07-01 17:59:20 +0000773 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000774 NamedRegionTimer T("Instruction Creation", GroupName);
Dan Gohman2048b852009-11-23 18:04:58 +0000775 BB = Scheduler->EmitSchedule(&SDB->EdgeMapping);
Evan Chengebffb662008-07-01 17:59:20 +0000776 } else {
Dan Gohman2048b852009-11-23 18:04:58 +0000777 BB = Scheduler->EmitSchedule(&SDB->EdgeMapping);
Dan Gohman5e843682008-07-14 18:19:29 +0000778 }
779
780 // Free the scheduler state.
781 if (TimePassesIsEnabled) {
782 NamedRegionTimer T("Instruction Scheduling Cleanup", GroupName);
783 delete Scheduler;
784 } else {
785 delete Scheduler;
Evan Chengebffb662008-07-01 17:59:20 +0000786 }
Evan Chengdb8d56b2008-06-30 20:45:06 +0000787
David Greene1a053232010-01-05 01:26:11 +0000788 DEBUG(dbgs() << "Selected machine code:\n");
Chris Lattner1c08c712005-01-07 07:47:53 +0000789 DEBUG(BB->dump());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000790}
Chris Lattner1c08c712005-01-07 07:47:53 +0000791
Chris Lattner7c306da2010-03-02 06:34:30 +0000792void SelectionDAGISel::DoInstructionSelection() {
793 DEBUG(errs() << "===== Instruction selection begins:\n");
794
795 PreprocessISelDAG();
796
797 // Select target instructions for the DAG.
798 {
799 // Number all nodes with a topological order and set DAGSize.
800 DAGSize = CurDAG->AssignTopologicalOrder();
801
802 // Create a dummy node (which is not added to allnodes), that adds
803 // a reference to the root node, preventing it from being deleted,
804 // and tracking any changes of the root.
805 HandleSDNode Dummy(CurDAG->getRoot());
806 ISelPosition = SelectionDAG::allnodes_iterator(CurDAG->getRoot().getNode());
807 ++ISelPosition;
808
809 // The AllNodes list is now topological-sorted. Visit the
810 // nodes by starting at the end of the list (the root of the
811 // graph) and preceding back toward the beginning (the entry
812 // node).
813 while (ISelPosition != CurDAG->allnodes_begin()) {
814 SDNode *Node = --ISelPosition;
815 // Skip dead nodes. DAGCombiner is expected to eliminate all dead nodes,
816 // but there are currently some corner cases that it misses. Also, this
817 // makes it theoretically possible to disable the DAGCombiner.
818 if (Node->use_empty())
819 continue;
820
821 SDNode *ResNode = Select(Node);
822
Chris Lattner82dd3d32010-03-02 07:50:03 +0000823 // FIXME: This is pretty gross. 'Select' should be changed to not return
824 // anything at all and this code should be nuked with a tactical strike.
825
Chris Lattner7c306da2010-03-02 06:34:30 +0000826 // If node should not be replaced, continue with the next one.
Chris Lattner82dd3d32010-03-02 07:50:03 +0000827 if (ResNode == Node || Node->getOpcode() == ISD::DELETED_NODE)
Chris Lattner7c306da2010-03-02 06:34:30 +0000828 continue;
829 // Replace node.
830 if (ResNode)
831 ReplaceUses(Node, ResNode);
832
833 // If after the replacement this node is not used any more,
834 // remove this dead node.
835 if (Node->use_empty()) { // Don't delete EntryToken, etc.
836 ISelUpdater ISU(ISelPosition);
837 CurDAG->RemoveDeadNode(Node, &ISU);
838 }
839 }
840
841 CurDAG->setRoot(Dummy.getValue());
842 }
843 DEBUG(errs() << "===== Instruction selection ends:\n");
844
845 PostprocessISelDAG();
Chris Lattner7c306da2010-03-02 06:34:30 +0000846}
847
848
Dan Gohman79ce2762009-01-15 19:20:50 +0000849void SelectionDAGISel::SelectAllBasicBlocks(Function &Fn,
850 MachineFunction &MF,
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000851 MachineModuleInfo *MMI,
Devang Patel83489bb2009-01-13 00:35:13 +0000852 DwarfWriter *DW,
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000853 const TargetInstrInfo &TII) {
Dan Gohmana43abd12008-09-29 21:55:50 +0000854 // Initialize the Fast-ISel state, if needed.
855 FastISel *FastIS = 0;
856 if (EnableFastISel)
Dan Gohman79ce2762009-01-15 19:20:50 +0000857 FastIS = TLI.createFastISel(MF, MMI, DW,
Dan Gohmana43abd12008-09-29 21:55:50 +0000858 FuncInfo->ValueMap,
859 FuncInfo->MBBMap,
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000860 FuncInfo->StaticAllocaMap
861#ifndef NDEBUG
862 , FuncInfo->CatchInfoLost
863#endif
864 );
Dan Gohmana43abd12008-09-29 21:55:50 +0000865
Chris Lattner08113472009-12-29 09:01:33 +0000866 unsigned MDDbgKind = Fn.getContext().getMDKindID("dbg");
Devang Patel123eaa72009-09-16 20:39:11 +0000867
Dan Gohmana43abd12008-09-29 21:55:50 +0000868 // Iterate over all basic blocks in the function.
Evan Cheng39fd6e82008-08-07 00:43:25 +0000869 for (Function::iterator I = Fn.begin(), E = Fn.end(); I != E; ++I) {
870 BasicBlock *LLVMBB = &*I;
Dan Gohman7c3234c2008-08-27 23:52:12 +0000871 BB = FuncInfo->MBBMap[LLVMBB];
Dan Gohmanf350b272008-08-23 02:25:05 +0000872
Dan Gohman3df24e62008-09-03 23:12:08 +0000873 BasicBlock::iterator const Begin = LLVMBB->begin();
874 BasicBlock::iterator const End = LLVMBB->end();
Evan Cheng9f118502008-09-08 16:01:27 +0000875 BasicBlock::iterator BI = Begin;
Dan Gohman5edd3612008-08-28 20:28:56 +0000876
877 // Lower any arguments needed in this block if this is the entry block.
Dan Gohman33134c42008-09-25 17:05:24 +0000878 bool SuppressFastISel = false;
879 if (LLVMBB == &Fn.getEntryBlock()) {
Dan Gohman5edd3612008-08-28 20:28:56 +0000880 LowerArguments(LLVMBB);
Dan Gohmanf350b272008-08-23 02:25:05 +0000881
Dan Gohman33134c42008-09-25 17:05:24 +0000882 // If any of the arguments has the byval attribute, forgo
883 // fast-isel in the entry block.
Dan Gohmana43abd12008-09-29 21:55:50 +0000884 if (FastIS) {
Dan Gohman33134c42008-09-25 17:05:24 +0000885 unsigned j = 1;
886 for (Function::arg_iterator I = Fn.arg_begin(), E = Fn.arg_end();
887 I != E; ++I, ++j)
Devang Patel05988662008-09-25 21:00:45 +0000888 if (Fn.paramHasAttr(j, Attribute::ByVal)) {
Dan Gohman77ca41e2008-09-25 17:21:42 +0000889 if (EnableFastISelVerbose || EnableFastISelAbort)
David Greene1a053232010-01-05 01:26:11 +0000890 dbgs() << "FastISel skips entry block due to byval argument\n";
Dan Gohman33134c42008-09-25 17:05:24 +0000891 SuppressFastISel = true;
892 break;
893 }
894 }
895 }
896
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000897 if (MMI && BB->isLandingPad()) {
898 // Add a label to mark the beginning of the landing pad. Deletion of the
899 // landing pad can thus be detected via the MachineModuleInfo.
Chris Lattner7561d482010-03-14 02:33:54 +0000900 MCSymbol *Label = MMI->addLandingPad(BB);
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000901
Chris Lattner518bb532010-02-09 19:54:29 +0000902 const TargetInstrDesc &II = TII.get(TargetOpcode::EH_LABEL);
Chris Lattner7561d482010-03-14 02:33:54 +0000903 BuildMI(BB, SDB->getCurDebugLoc(), II).addSym(Label);
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000904
905 // Mark exception register as live in.
906 unsigned Reg = TLI.getExceptionAddressRegister();
907 if (Reg) BB->addLiveIn(Reg);
908
909 // Mark exception selector register as live in.
910 Reg = TLI.getExceptionSelectorRegister();
911 if (Reg) BB->addLiveIn(Reg);
912
913 // FIXME: Hack around an exception handling flaw (PR1508): the personality
914 // function and list of typeids logically belong to the invoke (or, if you
915 // like, the basic block containing the invoke), and need to be associated
916 // with it in the dwarf exception handling tables. Currently however the
917 // information is provided by an intrinsic (eh.selector) that can be moved
918 // to unexpected places by the optimizers: if the unwind edge is critical,
919 // then breaking it can result in the intrinsics being in the successor of
Jim Grosbachf4549b02010-01-15 00:36:15 +0000920 // the landing pad, not the landing pad itself. This results
921 // in exceptions not being caught because no typeids are associated with
922 // the invoke. This may not be the only way things can go wrong, but it
923 // is the only way we try to work around for the moment.
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000924 BranchInst *Br = dyn_cast<BranchInst>(LLVMBB->getTerminator());
925
926 if (Br && Br->isUnconditional()) { // Critical edge?
927 BasicBlock::iterator I, E;
928 for (I = LLVMBB->begin(), E = --LLVMBB->end(); I != E; ++I)
929 if (isa<EHSelectorInst>(I))
930 break;
931
932 if (I == E)
933 // No catch info found - try to extract some from the successor.
Dan Gohman5fca8b12009-11-23 18:12:11 +0000934 CopyCatchInfo(Br->getSuccessor(0), LLVMBB, MMI, *FuncInfo);
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000935 }
936 }
937
Dan Gohmanf350b272008-08-23 02:25:05 +0000938 // Before doing SelectionDAG ISel, see if FastISel has been requested.
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000939 if (FastIS && !SuppressFastISel) {
Dan Gohmana43abd12008-09-29 21:55:50 +0000940 // Emit code for any incoming arguments. This must happen before
941 // beginning FastISel on the entry block.
942 if (LLVMBB == &Fn.getEntryBlock()) {
Dan Gohman2048b852009-11-23 18:04:58 +0000943 CurDAG->setRoot(SDB->getControlRoot());
Dan Gohmana43abd12008-09-29 21:55:50 +0000944 CodeGenAndEmitDAG();
Dan Gohman2048b852009-11-23 18:04:58 +0000945 SDB->clear();
Dan Gohmana43abd12008-09-29 21:55:50 +0000946 }
Dan Gohman241f4642008-10-04 00:56:36 +0000947 FastIS->startNewBlock(BB);
Dan Gohmana43abd12008-09-29 21:55:50 +0000948 // Do FastISel on as many instructions as possible.
949 for (; BI != End; ++BI) {
950 // Just before the terminator instruction, insert instructions to
951 // feed PHI nodes in successor blocks.
952 if (isa<TerminatorInst>(BI))
953 if (!HandlePHINodesInSuccessorBlocksFast(LLVMBB, FastIS)) {
Chris Lattnerc19ae9d2010-03-04 19:46:56 +0000954 ++NumFastIselFailures;
Dan Gohman07f111e2009-12-05 00:27:08 +0000955 ResetDebugLoc(SDB, FastIS);
Dan Gohman4344a5d2008-09-09 23:05:00 +0000956 if (EnableFastISelVerbose || EnableFastISelAbort) {
David Greene1a053232010-01-05 01:26:11 +0000957 dbgs() << "FastISel miss: ";
Dan Gohman293d5f82008-09-09 22:06:46 +0000958 BI->dump();
959 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000960 assert(!EnableFastISelAbort &&
Torok Edwinf3689232009-07-12 20:07:01 +0000961 "FastISel didn't handle a PHI in a successor");
Dan Gohmana43abd12008-09-29 21:55:50 +0000962 break;
Dan Gohmanf350b272008-08-23 02:25:05 +0000963 }
Dan Gohmana43abd12008-09-29 21:55:50 +0000964
Chris Lattner3990b122009-12-28 23:41:32 +0000965 SetDebugLoc(MDDbgKind, BI, SDB, FastIS, &MF);
Dan Gohman381ca552009-12-05 01:29:04 +0000966
Dan Gohman21c14e32010-01-12 04:32:35 +0000967 // Try to select the instruction with FastISel.
Dan Gohman07f111e2009-12-05 00:27:08 +0000968 if (FastIS->SelectInstruction(BI)) {
969 ResetDebugLoc(SDB, FastIS);
Dan Gohmana43abd12008-09-29 21:55:50 +0000970 continue;
Dan Gohman07f111e2009-12-05 00:27:08 +0000971 }
Dan Gohmana43abd12008-09-29 21:55:50 +0000972
Dan Gohman07f111e2009-12-05 00:27:08 +0000973 // Clear out the debug location so that it doesn't carry over to
974 // unrelated instructions.
975 ResetDebugLoc(SDB, FastIS);
Dan Gohmana43abd12008-09-29 21:55:50 +0000976
977 // Then handle certain instructions as single-LLVM-Instruction blocks.
978 if (isa<CallInst>(BI)) {
Chris Lattnerc19ae9d2010-03-04 19:46:56 +0000979 ++NumFastIselFailures;
Dan Gohmana43abd12008-09-29 21:55:50 +0000980 if (EnableFastISelVerbose || EnableFastISelAbort) {
David Greene1a053232010-01-05 01:26:11 +0000981 dbgs() << "FastISel missed call: ";
Dan Gohmana43abd12008-09-29 21:55:50 +0000982 BI->dump();
983 }
984
Benjamin Kramerf0127052010-01-05 13:12:22 +0000985 if (!BI->getType()->isVoidTy()) {
Dan Gohmana43abd12008-09-29 21:55:50 +0000986 unsigned &R = FuncInfo->ValueMap[BI];
987 if (!R)
988 R = FuncInfo->CreateRegForValue(BI);
989 }
990
Dan Gohmanb4afb132009-11-20 02:51:26 +0000991 bool HadTailCall = false;
Chris Lattner7896c9f2009-12-03 00:50:42 +0000992 SelectBasicBlock(LLVMBB, BI, llvm::next(BI), HadTailCall);
Dan Gohmanb4afb132009-11-20 02:51:26 +0000993
994 // If the call was emitted as a tail call, we're done with the block.
995 if (HadTailCall) {
996 BI = End;
997 break;
998 }
999
Dan Gohman241f4642008-10-04 00:56:36 +00001000 // If the instruction was codegen'd with multiple blocks,
1001 // inform the FastISel object where to resume inserting.
1002 FastIS->setCurrentBlock(BB);
Dan Gohmana43abd12008-09-29 21:55:50 +00001003 continue;
Dan Gohmanf350b272008-08-23 02:25:05 +00001004 }
Dan Gohmana43abd12008-09-29 21:55:50 +00001005
1006 // Otherwise, give up on FastISel for the rest of the block.
1007 // For now, be a little lenient about non-branch terminators.
1008 if (!isa<TerminatorInst>(BI) || isa<BranchInst>(BI)) {
Chris Lattnerc19ae9d2010-03-04 19:46:56 +00001009 ++NumFastIselFailures;
Dan Gohmana43abd12008-09-29 21:55:50 +00001010 if (EnableFastISelVerbose || EnableFastISelAbort) {
David Greene1a053232010-01-05 01:26:11 +00001011 dbgs() << "FastISel miss: ";
Dan Gohmana43abd12008-09-29 21:55:50 +00001012 BI->dump();
1013 }
1014 if (EnableFastISelAbort)
1015 // The "fast" selector couldn't handle something and bailed.
1016 // For the purpose of debugging, just abort.
Torok Edwinc23197a2009-07-14 16:55:14 +00001017 llvm_unreachable("FastISel didn't select the entire block");
Dan Gohmana43abd12008-09-29 21:55:50 +00001018 }
1019 break;
Dan Gohmanf350b272008-08-23 02:25:05 +00001020 }
1021 }
1022
Dan Gohmand2ff6472008-09-02 20:17:56 +00001023 // Run SelectionDAG instruction selection on the remainder of the block
1024 // not handled by FastISel. If FastISel is not run, this is the entire
Dan Gohman3df24e62008-09-03 23:12:08 +00001025 // block.
Devang Patel390f3ac2009-04-16 01:33:10 +00001026 if (BI != End) {
Dan Gohmanb4afb132009-11-20 02:51:26 +00001027 bool HadTailCall;
1028 SelectBasicBlock(LLVMBB, BI, End, HadTailCall);
Devang Patel390f3ac2009-04-16 01:33:10 +00001029 }
Dan Gohmanf350b272008-08-23 02:25:05 +00001030
Dan Gohman7c3234c2008-08-27 23:52:12 +00001031 FinishBasicBlock();
Evan Cheng39fd6e82008-08-07 00:43:25 +00001032 }
Dan Gohmana43abd12008-09-29 21:55:50 +00001033
1034 delete FastIS;
Dan Gohman0e5f1302008-07-07 23:02:41 +00001035}
1036
Dan Gohmanfed90b62008-07-28 21:51:04 +00001037void
Dan Gohman7c3234c2008-08-27 23:52:12 +00001038SelectionDAGISel::FinishBasicBlock() {
Dan Gohmanf350b272008-08-23 02:25:05 +00001039
David Greene1a053232010-01-05 01:26:11 +00001040 DEBUG(dbgs() << "Target-post-processed machine code:\n");
Dan Gohmanf350b272008-08-23 02:25:05 +00001041 DEBUG(BB->dump());
Nate Begemanf15485a2006-03-27 01:32:24 +00001042
David Greene1a053232010-01-05 01:26:11 +00001043 DEBUG(dbgs() << "Total amount of phi nodes to update: "
Dan Gohman2048b852009-11-23 18:04:58 +00001044 << SDB->PHINodesToUpdate.size() << "\n");
1045 DEBUG(for (unsigned i = 0, e = SDB->PHINodesToUpdate.size(); i != e; ++i)
David Greene1a053232010-01-05 01:26:11 +00001046 dbgs() << "Node " << i << " : ("
Dan Gohman2048b852009-11-23 18:04:58 +00001047 << SDB->PHINodesToUpdate[i].first
1048 << ", " << SDB->PHINodesToUpdate[i].second << ")\n");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001049
Chris Lattnera33ef482005-03-30 01:10:47 +00001050 // Next, now that we know what the last MBB the LLVM BB expanded is, update
Chris Lattner1c08c712005-01-07 07:47:53 +00001051 // PHI nodes in successors.
Dan Gohman2048b852009-11-23 18:04:58 +00001052 if (SDB->SwitchCases.empty() &&
1053 SDB->JTCases.empty() &&
1054 SDB->BitTestCases.empty()) {
1055 for (unsigned i = 0, e = SDB->PHINodesToUpdate.size(); i != e; ++i) {
1056 MachineInstr *PHI = SDB->PHINodesToUpdate[i].first;
Chris Lattner518bb532010-02-09 19:54:29 +00001057 assert(PHI->isPHI() &&
Nate Begemanf15485a2006-03-27 01:32:24 +00001058 "This is not a machine PHI node that we are updating!");
Jakob Stoklund Olesen580bba22010-03-05 21:49:10 +00001059 if (!BB->isSuccessor(PHI->getParent()))
1060 continue;
Dan Gohman2048b852009-11-23 18:04:58 +00001061 PHI->addOperand(MachineOperand::CreateReg(SDB->PHINodesToUpdate[i].second,
Chris Lattner9ce2e9d2007-12-30 00:57:42 +00001062 false));
1063 PHI->addOperand(MachineOperand::CreateMBB(BB));
Nate Begemanf15485a2006-03-27 01:32:24 +00001064 }
Dan Gohman2048b852009-11-23 18:04:58 +00001065 SDB->PHINodesToUpdate.clear();
Nate Begemanf15485a2006-03-27 01:32:24 +00001066 return;
Chris Lattner1c08c712005-01-07 07:47:53 +00001067 }
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001068
Dan Gohman2048b852009-11-23 18:04:58 +00001069 for (unsigned i = 0, e = SDB->BitTestCases.size(); i != e; ++i) {
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001070 // Lower header first, if it wasn't already lowered
Dan Gohman2048b852009-11-23 18:04:58 +00001071 if (!SDB->BitTestCases[i].Emitted) {
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001072 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman2048b852009-11-23 18:04:58 +00001073 BB = SDB->BitTestCases[i].Parent;
1074 SDB->setCurrentBasicBlock(BB);
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001075 // Emit the code
Dan Gohman2048b852009-11-23 18:04:58 +00001076 SDB->visitBitTestHeader(SDB->BitTestCases[i]);
1077 CurDAG->setRoot(SDB->getRoot());
Dan Gohmanf350b272008-08-23 02:25:05 +00001078 CodeGenAndEmitDAG();
Dan Gohman2048b852009-11-23 18:04:58 +00001079 SDB->clear();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001080 }
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001081
Dan Gohman2048b852009-11-23 18:04:58 +00001082 for (unsigned j = 0, ej = SDB->BitTestCases[i].Cases.size(); j != ej; ++j) {
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001083 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman2048b852009-11-23 18:04:58 +00001084 BB = SDB->BitTestCases[i].Cases[j].ThisBB;
1085 SDB->setCurrentBasicBlock(BB);
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001086 // Emit the code
1087 if (j+1 != ej)
Dan Gohman2048b852009-11-23 18:04:58 +00001088 SDB->visitBitTestCase(SDB->BitTestCases[i].Cases[j+1].ThisBB,
1089 SDB->BitTestCases[i].Reg,
1090 SDB->BitTestCases[i].Cases[j]);
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001091 else
Dan Gohman2048b852009-11-23 18:04:58 +00001092 SDB->visitBitTestCase(SDB->BitTestCases[i].Default,
1093 SDB->BitTestCases[i].Reg,
1094 SDB->BitTestCases[i].Cases[j]);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001095
1096
Dan Gohman2048b852009-11-23 18:04:58 +00001097 CurDAG->setRoot(SDB->getRoot());
Dan Gohmanf350b272008-08-23 02:25:05 +00001098 CodeGenAndEmitDAG();
Dan Gohman2048b852009-11-23 18:04:58 +00001099 SDB->clear();
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001100 }
1101
1102 // Update PHI Nodes
Dan Gohman2048b852009-11-23 18:04:58 +00001103 for (unsigned pi = 0, pe = SDB->PHINodesToUpdate.size(); pi != pe; ++pi) {
1104 MachineInstr *PHI = SDB->PHINodesToUpdate[pi].first;
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001105 MachineBasicBlock *PHIBB = PHI->getParent();
Chris Lattner518bb532010-02-09 19:54:29 +00001106 assert(PHI->isPHI() &&
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001107 "This is not a machine PHI node that we are updating!");
1108 // This is "default" BB. We have two jumps to it. From "header" BB and
1109 // from last "case" BB.
Dan Gohman2048b852009-11-23 18:04:58 +00001110 if (PHIBB == SDB->BitTestCases[i].Default) {
Jim Grosbachf4549b02010-01-15 00:36:15 +00001111 PHI->addOperand(MachineOperand::
1112 CreateReg(SDB->PHINodesToUpdate[pi].second, false));
Dan Gohman2048b852009-11-23 18:04:58 +00001113 PHI->addOperand(MachineOperand::CreateMBB(SDB->BitTestCases[i].Parent));
Jim Grosbachf4549b02010-01-15 00:36:15 +00001114 PHI->addOperand(MachineOperand::
1115 CreateReg(SDB->PHINodesToUpdate[pi].second, false));
Dan Gohman2048b852009-11-23 18:04:58 +00001116 PHI->addOperand(MachineOperand::CreateMBB(SDB->BitTestCases[i].Cases.
Chris Lattner9ce2e9d2007-12-30 00:57:42 +00001117 back().ThisBB));
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001118 }
1119 // One of "cases" BB.
Dan Gohman2048b852009-11-23 18:04:58 +00001120 for (unsigned j = 0, ej = SDB->BitTestCases[i].Cases.size();
Dan Gohman7c3234c2008-08-27 23:52:12 +00001121 j != ej; ++j) {
Dan Gohman2048b852009-11-23 18:04:58 +00001122 MachineBasicBlock* cBB = SDB->BitTestCases[i].Cases[j].ThisBB;
Jakob Stoklund Olesendd437ba2010-01-11 21:02:33 +00001123 if (cBB->isSuccessor(PHIBB)) {
Jim Grosbachf4549b02010-01-15 00:36:15 +00001124 PHI->addOperand(MachineOperand::
1125 CreateReg(SDB->PHINodesToUpdate[pi].second, false));
Chris Lattner9ce2e9d2007-12-30 00:57:42 +00001126 PHI->addOperand(MachineOperand::CreateMBB(cBB));
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001127 }
1128 }
1129 }
1130 }
Dan Gohman2048b852009-11-23 18:04:58 +00001131 SDB->BitTestCases.clear();
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001132
Nate Begeman9453eea2006-04-23 06:26:20 +00001133 // If the JumpTable record is filled in, then we need to emit a jump table.
1134 // Updating the PHI nodes is tricky in this case, since we need to determine
1135 // whether the PHI is a successor of the range check MBB or the jump table MBB
Dan Gohman2048b852009-11-23 18:04:58 +00001136 for (unsigned i = 0, e = SDB->JTCases.size(); i != e; ++i) {
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +00001137 // Lower header first, if it wasn't already lowered
Dan Gohman2048b852009-11-23 18:04:58 +00001138 if (!SDB->JTCases[i].first.Emitted) {
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +00001139 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman2048b852009-11-23 18:04:58 +00001140 BB = SDB->JTCases[i].first.HeaderBB;
1141 SDB->setCurrentBasicBlock(BB);
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +00001142 // Emit the code
Dan Gohman2048b852009-11-23 18:04:58 +00001143 SDB->visitJumpTableHeader(SDB->JTCases[i].second, SDB->JTCases[i].first);
1144 CurDAG->setRoot(SDB->getRoot());
Dan Gohmanf350b272008-08-23 02:25:05 +00001145 CodeGenAndEmitDAG();
Dan Gohman2048b852009-11-23 18:04:58 +00001146 SDB->clear();
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001147 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001148
Nate Begeman37efe672006-04-22 18:53:45 +00001149 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman2048b852009-11-23 18:04:58 +00001150 BB = SDB->JTCases[i].second.MBB;
1151 SDB->setCurrentBasicBlock(BB);
Nate Begeman37efe672006-04-22 18:53:45 +00001152 // Emit the code
Dan Gohman2048b852009-11-23 18:04:58 +00001153 SDB->visitJumpTable(SDB->JTCases[i].second);
1154 CurDAG->setRoot(SDB->getRoot());
Dan Gohmanf350b272008-08-23 02:25:05 +00001155 CodeGenAndEmitDAG();
Dan Gohman2048b852009-11-23 18:04:58 +00001156 SDB->clear();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001157
Nate Begeman37efe672006-04-22 18:53:45 +00001158 // Update PHI Nodes
Dan Gohman2048b852009-11-23 18:04:58 +00001159 for (unsigned pi = 0, pe = SDB->PHINodesToUpdate.size(); pi != pe; ++pi) {
1160 MachineInstr *PHI = SDB->PHINodesToUpdate[pi].first;
Nate Begeman37efe672006-04-22 18:53:45 +00001161 MachineBasicBlock *PHIBB = PHI->getParent();
Chris Lattner518bb532010-02-09 19:54:29 +00001162 assert(PHI->isPHI() &&
Nate Begeman37efe672006-04-22 18:53:45 +00001163 "This is not a machine PHI node that we are updating!");
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001164 // "default" BB. We can go there only from header BB.
Dan Gohman2048b852009-11-23 18:04:58 +00001165 if (PHIBB == SDB->JTCases[i].second.Default) {
Evan Chengce319102009-09-19 09:51:03 +00001166 PHI->addOperand
Dan Gohman2048b852009-11-23 18:04:58 +00001167 (MachineOperand::CreateReg(SDB->PHINodesToUpdate[pi].second, false));
Evan Chengce319102009-09-19 09:51:03 +00001168 PHI->addOperand
Dan Gohman2048b852009-11-23 18:04:58 +00001169 (MachineOperand::CreateMBB(SDB->JTCases[i].first.HeaderBB));
Nate Begemanf4360a42006-05-03 03:48:02 +00001170 }
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001171 // JT BB. Just iterate over successors here
Jakob Stoklund Olesendd437ba2010-01-11 21:02:33 +00001172 if (BB->isSuccessor(PHIBB)) {
Evan Chengce319102009-09-19 09:51:03 +00001173 PHI->addOperand
Dan Gohman2048b852009-11-23 18:04:58 +00001174 (MachineOperand::CreateReg(SDB->PHINodesToUpdate[pi].second, false));
Chris Lattner9ce2e9d2007-12-30 00:57:42 +00001175 PHI->addOperand(MachineOperand::CreateMBB(BB));
Nate Begeman37efe672006-04-22 18:53:45 +00001176 }
1177 }
Nate Begeman37efe672006-04-22 18:53:45 +00001178 }
Dan Gohman2048b852009-11-23 18:04:58 +00001179 SDB->JTCases.clear();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001180
Chris Lattnerb2e806e2006-10-22 23:00:53 +00001181 // If the switch block involved a branch to one of the actual successors, we
1182 // need to update PHI nodes in that block.
Dan Gohman2048b852009-11-23 18:04:58 +00001183 for (unsigned i = 0, e = SDB->PHINodesToUpdate.size(); i != e; ++i) {
1184 MachineInstr *PHI = SDB->PHINodesToUpdate[i].first;
Chris Lattner518bb532010-02-09 19:54:29 +00001185 assert(PHI->isPHI() &&
Chris Lattnerb2e806e2006-10-22 23:00:53 +00001186 "This is not a machine PHI node that we are updating!");
1187 if (BB->isSuccessor(PHI->getParent())) {
Dan Gohman2048b852009-11-23 18:04:58 +00001188 PHI->addOperand(MachineOperand::CreateReg(SDB->PHINodesToUpdate[i].second,
Chris Lattner9ce2e9d2007-12-30 00:57:42 +00001189 false));
1190 PHI->addOperand(MachineOperand::CreateMBB(BB));
Chris Lattnerb2e806e2006-10-22 23:00:53 +00001191 }
1192 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001193
Nate Begemanf15485a2006-03-27 01:32:24 +00001194 // If we generated any switch lowering information, build and codegen any
1195 // additional DAGs necessary.
Dan Gohman2048b852009-11-23 18:04:58 +00001196 for (unsigned i = 0, e = SDB->SwitchCases.size(); i != e; ++i) {
Nate Begemanf15485a2006-03-27 01:32:24 +00001197 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman2048b852009-11-23 18:04:58 +00001198 MachineBasicBlock *ThisBB = BB = SDB->SwitchCases[i].ThisBB;
1199 SDB->setCurrentBasicBlock(BB);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001200
Nate Begemanf15485a2006-03-27 01:32:24 +00001201 // Emit the code
Dan Gohman2048b852009-11-23 18:04:58 +00001202 SDB->visitSwitchCase(SDB->SwitchCases[i]);
1203 CurDAG->setRoot(SDB->getRoot());
Dan Gohmanf350b272008-08-23 02:25:05 +00001204 CodeGenAndEmitDAG();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001205
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001206 // Handle any PHI nodes in successors of this chunk, as if we were coming
1207 // from the original BB before switch expansion. Note that PHI nodes can
1208 // occur multiple times in PHINodesToUpdate. We have to be very careful to
1209 // handle them the right number of times.
Dan Gohman2048b852009-11-23 18:04:58 +00001210 while ((BB = SDB->SwitchCases[i].TrueBB)) { // Handle LHS and RHS.
Evan Chengfb2e7522009-09-18 21:02:19 +00001211 // If new BB's are created during scheduling, the edges may have been
Evan Chengce319102009-09-19 09:51:03 +00001212 // updated. That is, the edge from ThisBB to BB may have been split and
1213 // BB's predecessor is now another block.
Evan Chengfb2e7522009-09-18 21:02:19 +00001214 DenseMap<MachineBasicBlock*, MachineBasicBlock*>::iterator EI =
Dan Gohman2048b852009-11-23 18:04:58 +00001215 SDB->EdgeMapping.find(BB);
1216 if (EI != SDB->EdgeMapping.end())
Evan Chengfb2e7522009-09-18 21:02:19 +00001217 ThisBB = EI->second;
Jakob Stoklund Olesendd437ba2010-01-11 21:02:33 +00001218
1219 // BB may have been removed from the CFG if a branch was constant folded.
1220 if (ThisBB->isSuccessor(BB)) {
1221 for (MachineBasicBlock::iterator Phi = BB->begin();
Chris Lattner518bb532010-02-09 19:54:29 +00001222 Phi != BB->end() && Phi->isPHI();
Jakob Stoklund Olesendd437ba2010-01-11 21:02:33 +00001223 ++Phi) {
1224 // This value for this PHI node is recorded in PHINodesToUpdate.
1225 for (unsigned pn = 0; ; ++pn) {
1226 assert(pn != SDB->PHINodesToUpdate.size() &&
1227 "Didn't find PHI entry!");
1228 if (SDB->PHINodesToUpdate[pn].first == Phi) {
1229 Phi->addOperand(MachineOperand::
1230 CreateReg(SDB->PHINodesToUpdate[pn].second,
1231 false));
1232 Phi->addOperand(MachineOperand::CreateMBB(ThisBB));
1233 break;
1234 }
Evan Cheng8be58a12009-09-18 08:26:06 +00001235 }
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001236 }
Nate Begemanf15485a2006-03-27 01:32:24 +00001237 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001238
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001239 // Don't process RHS if same block as LHS.
Dan Gohman2048b852009-11-23 18:04:58 +00001240 if (BB == SDB->SwitchCases[i].FalseBB)
1241 SDB->SwitchCases[i].FalseBB = 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001242
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001243 // If we haven't handled the RHS, do so now. Otherwise, we're done.
Dan Gohman2048b852009-11-23 18:04:58 +00001244 SDB->SwitchCases[i].TrueBB = SDB->SwitchCases[i].FalseBB;
1245 SDB->SwitchCases[i].FalseBB = 0;
Nate Begemanf15485a2006-03-27 01:32:24 +00001246 }
Dan Gohman2048b852009-11-23 18:04:58 +00001247 assert(SDB->SwitchCases[i].TrueBB == 0 && SDB->SwitchCases[i].FalseBB == 0);
1248 SDB->clear();
Chris Lattnera33ef482005-03-30 01:10:47 +00001249 }
Dan Gohman2048b852009-11-23 18:04:58 +00001250 SDB->SwitchCases.clear();
Dan Gohman7c3234c2008-08-27 23:52:12 +00001251
Dan Gohman2048b852009-11-23 18:04:58 +00001252 SDB->PHINodesToUpdate.clear();
Chris Lattner1c08c712005-01-07 07:47:53 +00001253}
Evan Chenga9c20912006-01-21 02:32:06 +00001254
Jim Laskey13ec7022006-08-01 14:21:23 +00001255
Dan Gohman0a3776d2009-02-06 18:26:51 +00001256/// Create the scheduler. If a specific scheduler was specified
1257/// via the SchedulerRegistry, use it, otherwise select the
1258/// one preferred by the target.
Dan Gohman5e843682008-07-14 18:19:29 +00001259///
Dan Gohman47ac0f02009-02-11 04:27:20 +00001260ScheduleDAGSDNodes *SelectionDAGISel::CreateScheduler() {
Jim Laskeyeb577ba2006-08-02 12:30:23 +00001261 RegisterScheduler::FunctionPassCtor Ctor = RegisterScheduler::getDefault();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001262
Jim Laskey13ec7022006-08-01 14:21:23 +00001263 if (!Ctor) {
Jim Laskeyeb577ba2006-08-02 12:30:23 +00001264 Ctor = ISHeuristic;
Jim Laskey9373beb2006-08-01 19:14:14 +00001265 RegisterScheduler::setDefault(Ctor);
Evan Cheng4ef10862006-01-23 07:01:07 +00001266 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001267
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +00001268 return Ctor(this, OptLevel);
Evan Chenga9c20912006-01-21 02:32:06 +00001269}
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001270
Dan Gohmanfc54c552009-01-15 22:18:12 +00001271ScheduleHazardRecognizer *SelectionDAGISel::CreateTargetHazardRecognizer() {
1272 return new ScheduleHazardRecognizer();
Jim Laskey9ff542f2006-08-01 18:29:48 +00001273}
1274
Chris Lattner75548062006-10-11 03:58:02 +00001275//===----------------------------------------------------------------------===//
1276// Helper functions used by the generated instruction selector.
1277//===----------------------------------------------------------------------===//
1278// Calls to these methods are generated by tblgen.
1279
1280/// CheckAndMask - The isel is trying to match something like (and X, 255). If
1281/// the dag combiner simplified the 255, we still want to match. RHS is the
1282/// actual value in the DAG on the RHS of an AND, and DesiredMaskS is the value
1283/// specified in the .td file (e.g. 255).
Daniel Dunbara279bc32009-09-20 02:20:51 +00001284bool SelectionDAGISel::CheckAndMask(SDValue LHS, ConstantSDNode *RHS,
Dan Gohmandc9b3d02007-07-24 23:00:27 +00001285 int64_t DesiredMaskS) const {
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001286 const APInt &ActualMask = RHS->getAPIntValue();
1287 const APInt &DesiredMask = APInt(LHS.getValueSizeInBits(), DesiredMaskS);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001288
Chris Lattner75548062006-10-11 03:58:02 +00001289 // If the actual mask exactly matches, success!
1290 if (ActualMask == DesiredMask)
1291 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001292
Chris Lattner75548062006-10-11 03:58:02 +00001293 // If the actual AND mask is allowing unallowed bits, this doesn't match.
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001294 if (ActualMask.intersects(~DesiredMask))
Chris Lattner75548062006-10-11 03:58:02 +00001295 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001296
Chris Lattner75548062006-10-11 03:58:02 +00001297 // Otherwise, the DAG Combiner may have proven that the value coming in is
1298 // either already zero or is not demanded. Check for known zero input bits.
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001299 APInt NeededMask = DesiredMask & ~ActualMask;
Dan Gohmanea859be2007-06-22 14:59:07 +00001300 if (CurDAG->MaskedValueIsZero(LHS, NeededMask))
Chris Lattner75548062006-10-11 03:58:02 +00001301 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001302
Chris Lattner75548062006-10-11 03:58:02 +00001303 // TODO: check to see if missing bits are just not demanded.
1304
1305 // Otherwise, this pattern doesn't match.
1306 return false;
1307}
1308
1309/// CheckOrMask - The isel is trying to match something like (or X, 255). If
1310/// the dag combiner simplified the 255, we still want to match. RHS is the
1311/// actual value in the DAG on the RHS of an OR, and DesiredMaskS is the value
1312/// specified in the .td file (e.g. 255).
Daniel Dunbara279bc32009-09-20 02:20:51 +00001313bool SelectionDAGISel::CheckOrMask(SDValue LHS, ConstantSDNode *RHS,
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001314 int64_t DesiredMaskS) const {
1315 const APInt &ActualMask = RHS->getAPIntValue();
1316 const APInt &DesiredMask = APInt(LHS.getValueSizeInBits(), DesiredMaskS);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001317
Chris Lattner75548062006-10-11 03:58:02 +00001318 // If the actual mask exactly matches, success!
1319 if (ActualMask == DesiredMask)
1320 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001321
Chris Lattner75548062006-10-11 03:58:02 +00001322 // If the actual AND mask is allowing unallowed bits, this doesn't match.
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001323 if (ActualMask.intersects(~DesiredMask))
Chris Lattner75548062006-10-11 03:58:02 +00001324 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001325
Chris Lattner75548062006-10-11 03:58:02 +00001326 // Otherwise, the DAG Combiner may have proven that the value coming in is
1327 // either already zero or is not demanded. Check for known zero input bits.
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001328 APInt NeededMask = DesiredMask & ~ActualMask;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001329
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001330 APInt KnownZero, KnownOne;
Dan Gohmanea859be2007-06-22 14:59:07 +00001331 CurDAG->ComputeMaskedBits(LHS, NeededMask, KnownZero, KnownOne);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001332
Chris Lattner75548062006-10-11 03:58:02 +00001333 // If all the missing bits in the or are already known to be set, match!
1334 if ((NeededMask & KnownOne) == NeededMask)
1335 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001336
Chris Lattner75548062006-10-11 03:58:02 +00001337 // TODO: check to see if missing bits are just not demanded.
Daniel Dunbara279bc32009-09-20 02:20:51 +00001338
Chris Lattner75548062006-10-11 03:58:02 +00001339 // Otherwise, this pattern doesn't match.
1340 return false;
1341}
1342
Jim Laskey9ff542f2006-08-01 18:29:48 +00001343
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001344/// SelectInlineAsmMemoryOperands - Calls to this are automatically generated
1345/// by tblgen. Others should not call it.
1346void SelectionDAGISel::
Dan Gohmanf350b272008-08-23 02:25:05 +00001347SelectInlineAsmMemoryOperands(std::vector<SDValue> &Ops) {
Dan Gohman475871a2008-07-27 21:46:04 +00001348 std::vector<SDValue> InOps;
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001349 std::swap(InOps, Ops);
1350
1351 Ops.push_back(InOps[0]); // input chain.
1352 Ops.push_back(InOps[1]); // input asm string.
1353
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001354 unsigned i = 2, e = InOps.size();
Owen Anderson825b72b2009-08-11 20:47:22 +00001355 if (InOps[e-1].getValueType() == MVT::Flag)
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001356 --e; // Don't process a flag operand if it is here.
Daniel Dunbara279bc32009-09-20 02:20:51 +00001357
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001358 while (i != e) {
Dan Gohmanf5aeb1a2008-09-12 16:56:44 +00001359 unsigned Flags = cast<ConstantSDNode>(InOps[i])->getZExtValue();
Dale Johannesen86b49f82008-09-24 01:07:17 +00001360 if ((Flags & 7) != 4 /*MEM*/) {
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001361 // Just skip over this operand, copying the operands verbatim.
Evan Cheng697cbbf2009-03-20 18:03:34 +00001362 Ops.insert(Ops.end(), InOps.begin()+i,
1363 InOps.begin()+i+InlineAsm::getNumOperandRegisters(Flags) + 1);
1364 i += InlineAsm::getNumOperandRegisters(Flags) + 1;
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001365 } else {
Evan Cheng697cbbf2009-03-20 18:03:34 +00001366 assert(InlineAsm::getNumOperandRegisters(Flags) == 1 &&
1367 "Memory operand with multiple values?");
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001368 // Otherwise, this is a memory operand. Ask the target to select it.
Dan Gohman475871a2008-07-27 21:46:04 +00001369 std::vector<SDValue> SelOps;
Dan Gohmanf350b272008-08-23 02:25:05 +00001370 if (SelectInlineAsmMemoryOperand(InOps[i+1], 'm', SelOps)) {
Torok Edwin7d696d82009-07-11 13:10:19 +00001371 llvm_report_error("Could not match memory address. Inline asm"
1372 " failure!");
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001373 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001374
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001375 // Add this to the output node.
Dale Johannesen86b49f82008-09-24 01:07:17 +00001376 Ops.push_back(CurDAG->getTargetConstant(4/*MEM*/ | (SelOps.size()<< 3),
Dale Johannesen99499332009-12-23 07:32:51 +00001377 MVT::i32));
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001378 Ops.insert(Ops.end(), SelOps.begin(), SelOps.end());
1379 i += 2;
1380 }
1381 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001382
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001383 // Add the flag input back if present.
1384 if (e != InOps.size())
1385 Ops.push_back(InOps.back());
1386}
Devang Patel794fd752007-05-01 21:15:47 +00001387
Owen Andersone50ed302009-08-10 22:56:29 +00001388/// findFlagUse - Return use of EVT::Flag value produced by the specified
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001389/// SDNode.
1390///
1391static SDNode *findFlagUse(SDNode *N) {
1392 unsigned FlagResNo = N->getNumValues()-1;
1393 for (SDNode::use_iterator I = N->use_begin(), E = N->use_end(); I != E; ++I) {
1394 SDUse &Use = I.getUse();
1395 if (Use.getResNo() == FlagResNo)
1396 return Use.getUser();
1397 }
1398 return NULL;
1399}
1400
1401/// findNonImmUse - Return true if "Use" is a non-immediate use of "Def".
1402/// This function recursively traverses up the operand chain, ignoring
1403/// certain nodes.
1404static bool findNonImmUse(SDNode *Use, SDNode* Def, SDNode *ImmedUse,
Chris Lattnerd1b73822010-03-02 22:20:06 +00001405 SDNode *Root, SmallPtrSet<SDNode*, 16> &Visited,
1406 bool IgnoreChains) {
Chris Lattnerda244a02010-02-23 19:32:27 +00001407 // The NodeID's are given uniques ID's where a node ID is guaranteed to be
1408 // greater than all of its (recursive) operands. If we scan to a point where
1409 // 'use' is smaller than the node we're scanning for, then we know we will
1410 // never find it.
1411 //
1412 // The Use may be -1 (unassigned) if it is a newly allocated node. This can
1413 // happen because we scan down to newly selected nodes in the case of flag
1414 // uses.
1415 if ((Use->getNodeId() < Def->getNodeId() && Use->getNodeId() != -1))
1416 return false;
1417
1418 // Don't revisit nodes if we already scanned it and didn't fail, we know we
1419 // won't fail if we scan it again.
1420 if (!Visited.insert(Use))
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001421 return false;
1422
1423 for (unsigned i = 0, e = Use->getNumOperands(); i != e; ++i) {
Chris Lattnerd1b73822010-03-02 22:20:06 +00001424 // Ignore chain uses, they are validated by HandleMergeInputChains.
1425 if (Use->getOperand(i).getValueType() == MVT::Other && IgnoreChains)
1426 continue;
1427
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001428 SDNode *N = Use->getOperand(i).getNode();
1429 if (N == Def) {
1430 if (Use == ImmedUse || Use == Root)
1431 continue; // We are not looking for immediate use.
1432 assert(N != Root);
1433 return true;
1434 }
1435
1436 // Traverse up the operand chain.
Chris Lattnerd1b73822010-03-02 22:20:06 +00001437 if (findNonImmUse(N, Def, ImmedUse, Root, Visited, IgnoreChains))
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001438 return true;
1439 }
1440 return false;
1441}
1442
Evan Cheng014bf212010-02-15 19:41:07 +00001443/// IsProfitableToFold - Returns true if it's profitable to fold the specific
1444/// operand node N of U during instruction selection that starts at Root.
1445bool SelectionDAGISel::IsProfitableToFold(SDValue N, SDNode *U,
1446 SDNode *Root) const {
1447 if (OptLevel == CodeGenOpt::None) return false;
1448 return N.hasOneUse();
1449}
1450
1451/// IsLegalToFold - Returns true if the specific operand node N of
1452/// U can be folded during instruction selection that starts at Root.
Chris Lattnerd1b73822010-03-02 22:20:06 +00001453bool SelectionDAGISel::IsLegalToFold(SDValue N, SDNode *U, SDNode *Root,
1454 bool IgnoreChains) const {
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001455 if (OptLevel == CodeGenOpt::None) return false;
1456
1457 // If Root use can somehow reach N through a path that that doesn't contain
1458 // U then folding N would create a cycle. e.g. In the following
1459 // diagram, Root can reach N through X. If N is folded into into Root, then
1460 // X is both a predecessor and a successor of U.
1461 //
1462 // [N*] //
1463 // ^ ^ //
1464 // / \ //
1465 // [U*] [X]? //
1466 // ^ ^ //
1467 // \ / //
1468 // \ / //
1469 // [Root*] //
1470 //
1471 // * indicates nodes to be folded together.
1472 //
1473 // If Root produces a flag, then it gets (even more) interesting. Since it
1474 // will be "glued" together with its flag use in the scheduler, we need to
1475 // check if it might reach N.
1476 //
1477 // [N*] //
1478 // ^ ^ //
1479 // / \ //
1480 // [U*] [X]? //
1481 // ^ ^ //
1482 // \ \ //
1483 // \ | //
1484 // [Root*] | //
1485 // ^ | //
1486 // f | //
1487 // | / //
1488 // [Y] / //
1489 // ^ / //
1490 // f / //
1491 // | / //
1492 // [FU] //
1493 //
1494 // If FU (flag use) indirectly reaches N (the load), and Root folds N
1495 // (call it Fold), then X is a predecessor of FU and a successor of
1496 // Fold. But since Fold and FU are flagged together, this will create
1497 // a cycle in the scheduling graph.
1498
Chris Lattner9bbcd5e2010-03-05 05:49:45 +00001499 // If the node has flags, walk down the graph to the "lowest" node in the
1500 // flagged set.
Owen Andersone50ed302009-08-10 22:56:29 +00001501 EVT VT = Root->getValueType(Root->getNumValues()-1);
Owen Anderson825b72b2009-08-11 20:47:22 +00001502 while (VT == MVT::Flag) {
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001503 SDNode *FU = findFlagUse(Root);
1504 if (FU == NULL)
1505 break;
1506 Root = FU;
1507 VT = Root->getValueType(Root->getNumValues()-1);
Chris Lattner18fdaba2010-03-05 06:19:13 +00001508
1509 // If our query node has a flag result with a use, we've walked up it. If
1510 // the user (which has already been selected) has a chain or indirectly uses
1511 // the chain, our WalkChainUsers predicate will not consider it. Because of
1512 // this, we cannot ignore chains in this predicate.
1513 IgnoreChains = false;
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001514 }
Chris Lattner18fdaba2010-03-05 06:19:13 +00001515
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001516
Chris Lattner9bbcd5e2010-03-05 05:49:45 +00001517 SmallPtrSet<SDNode*, 16> Visited;
1518 return !findNonImmUse(Root, N.getNode(), U, Root, Visited, IgnoreChains);
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001519}
1520
Dan Gohmaneeb3a002010-01-05 01:24:18 +00001521SDNode *SelectionDAGISel::Select_INLINEASM(SDNode *N) {
1522 std::vector<SDValue> Ops(N->op_begin(), N->op_end());
Dan Gohmane1f188f2009-10-29 22:30:23 +00001523 SelectInlineAsmMemoryOperands(Ops);
1524
1525 std::vector<EVT> VTs;
1526 VTs.push_back(MVT::Other);
1527 VTs.push_back(MVT::Flag);
Dan Gohmaneeb3a002010-01-05 01:24:18 +00001528 SDValue New = CurDAG->getNode(ISD::INLINEASM, N->getDebugLoc(),
Dan Gohmane1f188f2009-10-29 22:30:23 +00001529 VTs, &Ops[0], Ops.size());
Chris Lattnerd1b73822010-03-02 22:20:06 +00001530 New->setNodeId(-1);
Dan Gohmane1f188f2009-10-29 22:30:23 +00001531 return New.getNode();
1532}
1533
Dan Gohmaneeb3a002010-01-05 01:24:18 +00001534SDNode *SelectionDAGISel::Select_UNDEF(SDNode *N) {
Chris Lattner518bb532010-02-09 19:54:29 +00001535 return CurDAG->SelectNodeTo(N, TargetOpcode::IMPLICIT_DEF,N->getValueType(0));
Dan Gohmane1f188f2009-10-29 22:30:23 +00001536}
1537
Chris Lattner2a49d572010-02-28 22:37:22 +00001538/// GetVBR - decode a vbr encoding whose top bit is set.
1539ALWAYS_INLINE static uint64_t
1540GetVBR(uint64_t Val, const unsigned char *MatcherTable, unsigned &Idx) {
1541 assert(Val >= 128 && "Not a VBR");
1542 Val &= 127; // Remove first vbr bit.
1543
1544 unsigned Shift = 7;
1545 uint64_t NextBits;
1546 do {
Chris Lattner14df8dc2010-02-28 22:38:43 +00001547 NextBits = MatcherTable[Idx++];
Chris Lattner2a49d572010-02-28 22:37:22 +00001548 Val |= (NextBits&127) << Shift;
1549 Shift += 7;
1550 } while (NextBits & 128);
1551
1552 return Val;
1553}
1554
Chris Lattner2a49d572010-02-28 22:37:22 +00001555
1556/// UpdateChainsAndFlags - When a match is complete, this method updates uses of
1557/// interior flag and chain results to use the new flag and chain results.
Chris Lattner82dd3d32010-03-02 07:50:03 +00001558void SelectionDAGISel::
1559UpdateChainsAndFlags(SDNode *NodeToMatch, SDValue InputChain,
1560 const SmallVectorImpl<SDNode*> &ChainNodesMatched,
1561 SDValue InputFlag,
1562 const SmallVectorImpl<SDNode*> &FlagResultNodesMatched,
1563 bool isMorphNodeTo) {
1564 SmallVector<SDNode*, 4> NowDeadNodes;
1565
1566 ISelUpdater ISU(ISelPosition);
1567
Chris Lattner2a49d572010-02-28 22:37:22 +00001568 // Now that all the normal results are replaced, we replace the chain and
1569 // flag results if present.
1570 if (!ChainNodesMatched.empty()) {
1571 assert(InputChain.getNode() != 0 &&
1572 "Matched input chains but didn't produce a chain");
1573 // Loop over all of the nodes we matched that produced a chain result.
1574 // Replace all the chain results with the final chain we ended up with.
1575 for (unsigned i = 0, e = ChainNodesMatched.size(); i != e; ++i) {
1576 SDNode *ChainNode = ChainNodesMatched[i];
1577
Chris Lattner82dd3d32010-03-02 07:50:03 +00001578 // If this node was already deleted, don't look at it.
1579 if (ChainNode->getOpcode() == ISD::DELETED_NODE)
1580 continue;
1581
Chris Lattner2a49d572010-02-28 22:37:22 +00001582 // Don't replace the results of the root node if we're doing a
1583 // MorphNodeTo.
1584 if (ChainNode == NodeToMatch && isMorphNodeTo)
1585 continue;
1586
1587 SDValue ChainVal = SDValue(ChainNode, ChainNode->getNumValues()-1);
1588 if (ChainVal.getValueType() == MVT::Flag)
1589 ChainVal = ChainVal.getValue(ChainVal->getNumValues()-2);
1590 assert(ChainVal.getValueType() == MVT::Other && "Not a chain?");
Chris Lattner82dd3d32010-03-02 07:50:03 +00001591 CurDAG->ReplaceAllUsesOfValueWith(ChainVal, InputChain, &ISU);
1592
Chris Lattner856fb392010-03-28 05:28:31 +00001593 // If the node became dead and we haven't already seen it, delete it.
1594 if (ChainNode->use_empty() &&
1595 !std::count(NowDeadNodes.begin(), NowDeadNodes.end(), ChainNode))
Chris Lattner82dd3d32010-03-02 07:50:03 +00001596 NowDeadNodes.push_back(ChainNode);
Chris Lattner2a49d572010-02-28 22:37:22 +00001597 }
1598 }
1599
1600 // If the result produces a flag, update any flag results in the matched
1601 // pattern with the flag result.
1602 if (InputFlag.getNode() != 0) {
1603 // Handle any interior nodes explicitly marked.
1604 for (unsigned i = 0, e = FlagResultNodesMatched.size(); i != e; ++i) {
1605 SDNode *FRN = FlagResultNodesMatched[i];
Chris Lattner82dd3d32010-03-02 07:50:03 +00001606
1607 // If this node was already deleted, don't look at it.
1608 if (FRN->getOpcode() == ISD::DELETED_NODE)
1609 continue;
1610
Chris Lattner2a49d572010-02-28 22:37:22 +00001611 assert(FRN->getValueType(FRN->getNumValues()-1) == MVT::Flag &&
1612 "Doesn't have a flag result");
1613 CurDAG->ReplaceAllUsesOfValueWith(SDValue(FRN, FRN->getNumValues()-1),
Chris Lattner82dd3d32010-03-02 07:50:03 +00001614 InputFlag, &ISU);
1615
Chris Lattner19e37cb2010-03-28 04:54:33 +00001616 // If the node became dead and we haven't already seen it, delete it.
1617 if (FRN->use_empty() &&
1618 !std::count(NowDeadNodes.begin(), NowDeadNodes.end(), FRN))
Chris Lattner82dd3d32010-03-02 07:50:03 +00001619 NowDeadNodes.push_back(FRN);
Chris Lattner2a49d572010-02-28 22:37:22 +00001620 }
1621 }
1622
Chris Lattner82dd3d32010-03-02 07:50:03 +00001623 if (!NowDeadNodes.empty())
1624 CurDAG->RemoveDeadNodes(NowDeadNodes, &ISU);
1625
Chris Lattner2a49d572010-02-28 22:37:22 +00001626 DEBUG(errs() << "ISEL: Match complete!\n");
1627}
1628
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001629enum ChainResult {
1630 CR_Simple,
1631 CR_InducesCycle,
1632 CR_LeadsToInteriorNode
1633};
1634
1635/// WalkChainUsers - Walk down the users of the specified chained node that is
1636/// part of the pattern we're matching, looking at all of the users we find.
1637/// This determines whether something is an interior node, whether we have a
1638/// non-pattern node in between two pattern nodes (which prevent folding because
1639/// it would induce a cycle) and whether we have a TokenFactor node sandwiched
1640/// between pattern nodes (in which case the TF becomes part of the pattern).
1641///
1642/// The walk we do here is guaranteed to be small because we quickly get down to
1643/// already selected nodes "below" us.
1644static ChainResult
1645WalkChainUsers(SDNode *ChainedNode,
1646 SmallVectorImpl<SDNode*> &ChainedNodesInPattern,
1647 SmallVectorImpl<SDNode*> &InteriorChainedNodes) {
1648 ChainResult Result = CR_Simple;
1649
1650 for (SDNode::use_iterator UI = ChainedNode->use_begin(),
1651 E = ChainedNode->use_end(); UI != E; ++UI) {
1652 // Make sure the use is of the chain, not some other value we produce.
1653 if (UI.getUse().getValueType() != MVT::Other) continue;
1654
1655 SDNode *User = *UI;
1656
1657 // If we see an already-selected machine node, then we've gone beyond the
1658 // pattern that we're selecting down into the already selected chunk of the
1659 // DAG.
1660 if (User->isMachineOpcode() ||
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001661 User->getOpcode() == ISD::HANDLENODE) // Root of the graph.
1662 continue;
Chris Lattnerd1b73822010-03-02 22:20:06 +00001663
1664 if (User->getOpcode() == ISD::CopyToReg ||
1665 User->getOpcode() == ISD::CopyFromReg ||
Chris Lattner7561d482010-03-14 02:33:54 +00001666 User->getOpcode() == ISD::INLINEASM ||
1667 User->getOpcode() == ISD::EH_LABEL) {
Chris Lattnerd1b73822010-03-02 22:20:06 +00001668 // If their node ID got reset to -1 then they've already been selected.
1669 // Treat them like a MachineOpcode.
1670 if (User->getNodeId() == -1)
1671 continue;
1672 }
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001673
1674 // If we have a TokenFactor, we handle it specially.
1675 if (User->getOpcode() != ISD::TokenFactor) {
1676 // If the node isn't a token factor and isn't part of our pattern, then it
1677 // must be a random chained node in between two nodes we're selecting.
1678 // This happens when we have something like:
1679 // x = load ptr
1680 // call
1681 // y = x+4
1682 // store y -> ptr
1683 // Because we structurally match the load/store as a read/modify/write,
1684 // but the call is chained between them. We cannot fold in this case
1685 // because it would induce a cycle in the graph.
1686 if (!std::count(ChainedNodesInPattern.begin(),
1687 ChainedNodesInPattern.end(), User))
1688 return CR_InducesCycle;
1689
1690 // Otherwise we found a node that is part of our pattern. For example in:
1691 // x = load ptr
1692 // y = x+4
1693 // store y -> ptr
1694 // This would happen when we're scanning down from the load and see the
1695 // store as a user. Record that there is a use of ChainedNode that is
1696 // part of the pattern and keep scanning uses.
1697 Result = CR_LeadsToInteriorNode;
1698 InteriorChainedNodes.push_back(User);
1699 continue;
1700 }
1701
1702 // If we found a TokenFactor, there are two cases to consider: first if the
1703 // TokenFactor is just hanging "below" the pattern we're matching (i.e. no
1704 // uses of the TF are in our pattern) we just want to ignore it. Second,
1705 // the TokenFactor can be sandwiched in between two chained nodes, like so:
1706 // [Load chain]
1707 // ^
1708 // |
1709 // [Load]
1710 // ^ ^
1711 // | \ DAG's like cheese
1712 // / \ do you?
1713 // / |
1714 // [TokenFactor] [Op]
1715 // ^ ^
1716 // | |
1717 // \ /
1718 // \ /
1719 // [Store]
1720 //
1721 // In this case, the TokenFactor becomes part of our match and we rewrite it
1722 // as a new TokenFactor.
1723 //
1724 // To distinguish these two cases, do a recursive walk down the uses.
1725 switch (WalkChainUsers(User, ChainedNodesInPattern, InteriorChainedNodes)) {
1726 case CR_Simple:
1727 // If the uses of the TokenFactor are just already-selected nodes, ignore
1728 // it, it is "below" our pattern.
1729 continue;
1730 case CR_InducesCycle:
1731 // If the uses of the TokenFactor lead to nodes that are not part of our
1732 // pattern that are not selected, folding would turn this into a cycle,
1733 // bail out now.
1734 return CR_InducesCycle;
1735 case CR_LeadsToInteriorNode:
1736 break; // Otherwise, keep processing.
1737 }
1738
1739 // Okay, we know we're in the interesting interior case. The TokenFactor
1740 // is now going to be considered part of the pattern so that we rewrite its
1741 // uses (it may have uses that are not part of the pattern) with the
1742 // ultimate chain result of the generated code. We will also add its chain
1743 // inputs as inputs to the ultimate TokenFactor we create.
1744 Result = CR_LeadsToInteriorNode;
1745 ChainedNodesInPattern.push_back(User);
1746 InteriorChainedNodes.push_back(User);
1747 continue;
1748 }
1749
1750 return Result;
1751}
1752
Chris Lattner6b307922010-03-02 00:00:03 +00001753/// HandleMergeInputChains - This implements the OPC_EmitMergeInputChains
Chris Lattner00592ec2010-03-02 19:34:59 +00001754/// operation for when the pattern matched at least one node with a chains. The
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001755/// input vector contains a list of all of the chained nodes that we match. We
1756/// must determine if this is a valid thing to cover (i.e. matching it won't
1757/// induce cycles in the DAG) and if so, creating a TokenFactor node. that will
1758/// be used as the input node chain for the generated nodes.
Chris Lattner6b307922010-03-02 00:00:03 +00001759static SDValue
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001760HandleMergeInputChains(SmallVectorImpl<SDNode*> &ChainNodesMatched,
Chris Lattner6b307922010-03-02 00:00:03 +00001761 SelectionDAG *CurDAG) {
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001762 // Walk all of the chained nodes we've matched, recursively scanning down the
1763 // users of the chain result. This adds any TokenFactor nodes that are caught
1764 // in between chained nodes to the chained and interior nodes list.
1765 SmallVector<SDNode*, 3> InteriorChainedNodes;
1766 for (unsigned i = 0, e = ChainNodesMatched.size(); i != e; ++i) {
1767 if (WalkChainUsers(ChainNodesMatched[i], ChainNodesMatched,
1768 InteriorChainedNodes) == CR_InducesCycle)
1769 return SDValue(); // Would induce a cycle.
1770 }
Chris Lattner6b307922010-03-02 00:00:03 +00001771
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001772 // Okay, we have walked all the matched nodes and collected TokenFactor nodes
1773 // that we are interested in. Form our input TokenFactor node.
Chris Lattner6b307922010-03-02 00:00:03 +00001774 SmallVector<SDValue, 3> InputChains;
1775 for (unsigned i = 0, e = ChainNodesMatched.size(); i != e; ++i) {
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001776 // Add the input chain of this node to the InputChains list (which will be
1777 // the operands of the generated TokenFactor) if it's not an interior node.
1778 SDNode *N = ChainNodesMatched[i];
1779 if (N->getOpcode() != ISD::TokenFactor) {
1780 if (std::count(InteriorChainedNodes.begin(),InteriorChainedNodes.end(),N))
1781 continue;
1782
1783 // Otherwise, add the input chain.
1784 SDValue InChain = ChainNodesMatched[i]->getOperand(0);
1785 assert(InChain.getValueType() == MVT::Other && "Not a chain");
Chris Lattner6b307922010-03-02 00:00:03 +00001786 InputChains.push_back(InChain);
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001787 continue;
1788 }
1789
1790 // If we have a token factor, we want to add all inputs of the token factor
1791 // that are not part of the pattern we're matching.
1792 for (unsigned op = 0, e = N->getNumOperands(); op != e; ++op) {
1793 if (!std::count(ChainNodesMatched.begin(), ChainNodesMatched.end(),
Chris Lattner6183fbd2010-03-02 02:37:23 +00001794 N->getOperand(op).getNode()))
1795 InputChains.push_back(N->getOperand(op));
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001796 }
Chris Lattner6b307922010-03-02 00:00:03 +00001797 }
1798
1799 SDValue Res;
1800 if (InputChains.size() == 1)
1801 return InputChains[0];
1802 return CurDAG->getNode(ISD::TokenFactor, ChainNodesMatched[0]->getDebugLoc(),
1803 MVT::Other, &InputChains[0], InputChains.size());
1804}
Chris Lattner2a49d572010-02-28 22:37:22 +00001805
Chris Lattner3ff1e4d2010-03-02 06:55:04 +00001806/// MorphNode - Handle morphing a node in place for the selector.
1807SDNode *SelectionDAGISel::
1808MorphNode(SDNode *Node, unsigned TargetOpc, SDVTList VTList,
1809 const SDValue *Ops, unsigned NumOps, unsigned EmitNodeInfo) {
1810 // It is possible we're using MorphNodeTo to replace a node with no
1811 // normal results with one that has a normal result (or we could be
1812 // adding a chain) and the input could have flags and chains as well.
Chris Lattnercaa88702010-03-28 08:43:23 +00001813 // In this case we need to shift the operands down.
Chris Lattner3ff1e4d2010-03-02 06:55:04 +00001814 // FIXME: This is a horrible hack and broken in obscure cases, no worse
Chris Lattnera6f86932010-03-27 18:54:50 +00001815 // than the old isel though.
Chris Lattner3ff1e4d2010-03-02 06:55:04 +00001816 int OldFlagResultNo = -1, OldChainResultNo = -1;
1817
1818 unsigned NTMNumResults = Node->getNumValues();
1819 if (Node->getValueType(NTMNumResults-1) == MVT::Flag) {
1820 OldFlagResultNo = NTMNumResults-1;
1821 if (NTMNumResults != 1 &&
1822 Node->getValueType(NTMNumResults-2) == MVT::Other)
1823 OldChainResultNo = NTMNumResults-2;
1824 } else if (Node->getValueType(NTMNumResults-1) == MVT::Other)
1825 OldChainResultNo = NTMNumResults-1;
1826
Chris Lattner61c97f62010-03-02 07:14:49 +00001827 // Call the underlying SelectionDAG routine to do the transmogrification. Note
1828 // that this deletes operands of the old node that become dead.
Chris Lattner3ff1e4d2010-03-02 06:55:04 +00001829 SDNode *Res = CurDAG->MorphNodeTo(Node, ~TargetOpc, VTList, Ops, NumOps);
1830
1831 // MorphNodeTo can operate in two ways: if an existing node with the
1832 // specified operands exists, it can just return it. Otherwise, it
1833 // updates the node in place to have the requested operands.
1834 if (Res == Node) {
1835 // If we updated the node in place, reset the node ID. To the isel,
1836 // this should be just like a newly allocated machine node.
1837 Res->setNodeId(-1);
1838 }
1839
1840 unsigned ResNumResults = Res->getNumValues();
1841 // Move the flag if needed.
1842 if ((EmitNodeInfo & OPFL_FlagOutput) && OldFlagResultNo != -1 &&
1843 (unsigned)OldFlagResultNo != ResNumResults-1)
1844 CurDAG->ReplaceAllUsesOfValueWith(SDValue(Node, OldFlagResultNo),
1845 SDValue(Res, ResNumResults-1));
1846
1847 if ((EmitNodeInfo & OPFL_FlagOutput) != 0)
1848 --ResNumResults;
1849
1850 // Move the chain reference if needed.
1851 if ((EmitNodeInfo & OPFL_Chain) && OldChainResultNo != -1 &&
1852 (unsigned)OldChainResultNo != ResNumResults-1)
1853 CurDAG->ReplaceAllUsesOfValueWith(SDValue(Node, OldChainResultNo),
1854 SDValue(Res, ResNumResults-1));
1855
1856 // Otherwise, no replacement happened because the node already exists. Replace
1857 // Uses of the old node with the new one.
1858 if (Res != Node)
1859 CurDAG->ReplaceAllUsesWith(Node, Res);
1860
1861 return Res;
1862}
1863
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001864/// CheckPatternPredicate - Implements OP_CheckPatternPredicate.
1865ALWAYS_INLINE static bool
Chris Lattnerda828e32010-03-03 07:46:25 +00001866CheckSame(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1867 SDValue N, const SmallVectorImpl<SDValue> &RecordedNodes) {
1868 // Accept if it is exactly the same as a previously recorded node.
1869 unsigned RecNo = MatcherTable[MatcherIndex++];
1870 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
1871 return N == RecordedNodes[RecNo];
1872}
1873
1874/// CheckPatternPredicate - Implements OP_CheckPatternPredicate.
1875ALWAYS_INLINE static bool
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001876CheckPatternPredicate(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1877 SelectionDAGISel &SDISel) {
1878 return SDISel.CheckPatternPredicate(MatcherTable[MatcherIndex++]);
1879}
1880
1881/// CheckNodePredicate - Implements OP_CheckNodePredicate.
1882ALWAYS_INLINE static bool
1883CheckNodePredicate(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1884 SelectionDAGISel &SDISel, SDNode *N) {
1885 return SDISel.CheckNodePredicate(N, MatcherTable[MatcherIndex++]);
1886}
1887
1888ALWAYS_INLINE static bool
1889CheckOpcode(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1890 SDNode *N) {
Chris Lattnerd5d5a3d2010-03-25 06:33:05 +00001891 uint16_t Opc = MatcherTable[MatcherIndex++];
1892 Opc |= (unsigned short)MatcherTable[MatcherIndex++] << 8;
1893 return N->getOpcode() == Opc;
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001894}
1895
1896ALWAYS_INLINE static bool
1897CheckType(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1898 SDValue N, const TargetLowering &TLI) {
1899 MVT::SimpleValueType VT = (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
1900 if (N.getValueType() == VT) return true;
1901
1902 // Handle the case when VT is iPTR.
1903 return VT == MVT::iPTR && N.getValueType() == TLI.getPointerTy();
1904}
1905
1906ALWAYS_INLINE static bool
Chris Lattnerda828e32010-03-03 07:46:25 +00001907CheckChildType(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1908 SDValue N, const TargetLowering &TLI,
1909 unsigned ChildNo) {
1910 if (ChildNo >= N.getNumOperands())
1911 return false; // Match fails if out of range child #.
1912 return ::CheckType(MatcherTable, MatcherIndex, N.getOperand(ChildNo), TLI);
1913}
1914
1915
1916ALWAYS_INLINE static bool
1917CheckCondCode(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1918 SDValue N) {
1919 return cast<CondCodeSDNode>(N)->get() ==
1920 (ISD::CondCode)MatcherTable[MatcherIndex++];
1921}
1922
1923ALWAYS_INLINE static bool
1924CheckValueType(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1925 SDValue N, const TargetLowering &TLI) {
1926 MVT::SimpleValueType VT = (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
1927 if (cast<VTSDNode>(N)->getVT() == VT)
1928 return true;
1929
1930 // Handle the case when VT is iPTR.
1931 return VT == MVT::iPTR && cast<VTSDNode>(N)->getVT() == TLI.getPointerTy();
1932}
1933
1934ALWAYS_INLINE static bool
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001935CheckInteger(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1936 SDValue N) {
1937 int64_t Val = MatcherTable[MatcherIndex++];
1938 if (Val & 128)
1939 Val = GetVBR(Val, MatcherTable, MatcherIndex);
1940
1941 ConstantSDNode *C = dyn_cast<ConstantSDNode>(N);
1942 return C != 0 && C->getSExtValue() == Val;
1943}
1944
Chris Lattnerda828e32010-03-03 07:46:25 +00001945ALWAYS_INLINE static bool
1946CheckAndImm(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1947 SDValue N, SelectionDAGISel &SDISel) {
1948 int64_t Val = MatcherTable[MatcherIndex++];
1949 if (Val & 128)
1950 Val = GetVBR(Val, MatcherTable, MatcherIndex);
1951
1952 if (N->getOpcode() != ISD::AND) return false;
1953
1954 ConstantSDNode *C = dyn_cast<ConstantSDNode>(N->getOperand(1));
1955 return C != 0 && SDISel.CheckAndMask(N.getOperand(0), C, Val);
1956}
1957
1958ALWAYS_INLINE static bool
1959CheckOrImm(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1960 SDValue N, SelectionDAGISel &SDISel) {
1961 int64_t Val = MatcherTable[MatcherIndex++];
1962 if (Val & 128)
1963 Val = GetVBR(Val, MatcherTable, MatcherIndex);
1964
1965 if (N->getOpcode() != ISD::OR) return false;
1966
1967 ConstantSDNode *C = dyn_cast<ConstantSDNode>(N->getOperand(1));
1968 return C != 0 && SDISel.CheckOrMask(N.getOperand(0), C, Val);
1969}
1970
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001971/// IsPredicateKnownToFail - If we know how and can do so without pushing a
1972/// scope, evaluate the current node. If the current predicate is known to
1973/// fail, set Result=true and return anything. If the current predicate is
1974/// known to pass, set Result=false and return the MatcherIndex to continue
1975/// with. If the current predicate is unknown, set Result=false and return the
1976/// MatcherIndex to continue with.
1977static unsigned IsPredicateKnownToFail(const unsigned char *Table,
1978 unsigned Index, SDValue N,
Chris Lattnerda828e32010-03-03 07:46:25 +00001979 bool &Result, SelectionDAGISel &SDISel,
1980 SmallVectorImpl<SDValue> &RecordedNodes){
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001981 switch (Table[Index++]) {
1982 default:
1983 Result = false;
1984 return Index-1; // Could not evaluate this predicate.
Chris Lattnerda828e32010-03-03 07:46:25 +00001985 case SelectionDAGISel::OPC_CheckSame:
1986 Result = !::CheckSame(Table, Index, N, RecordedNodes);
1987 return Index;
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001988 case SelectionDAGISel::OPC_CheckPatternPredicate:
Chris Lattnerda828e32010-03-03 07:46:25 +00001989 Result = !::CheckPatternPredicate(Table, Index, SDISel);
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001990 return Index;
1991 case SelectionDAGISel::OPC_CheckPredicate:
Chris Lattnerda828e32010-03-03 07:46:25 +00001992 Result = !::CheckNodePredicate(Table, Index, SDISel, N.getNode());
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001993 return Index;
1994 case SelectionDAGISel::OPC_CheckOpcode:
1995 Result = !::CheckOpcode(Table, Index, N.getNode());
1996 return Index;
1997 case SelectionDAGISel::OPC_CheckType:
1998 Result = !::CheckType(Table, Index, N, SDISel.TLI);
1999 return Index;
2000 case SelectionDAGISel::OPC_CheckChild0Type:
2001 case SelectionDAGISel::OPC_CheckChild1Type:
2002 case SelectionDAGISel::OPC_CheckChild2Type:
2003 case SelectionDAGISel::OPC_CheckChild3Type:
2004 case SelectionDAGISel::OPC_CheckChild4Type:
2005 case SelectionDAGISel::OPC_CheckChild5Type:
2006 case SelectionDAGISel::OPC_CheckChild6Type:
Chris Lattnerda828e32010-03-03 07:46:25 +00002007 case SelectionDAGISel::OPC_CheckChild7Type:
2008 Result = !::CheckChildType(Table, Index, N, SDISel.TLI,
2009 Table[Index-1] - SelectionDAGISel::OPC_CheckChild0Type);
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002010 return Index;
Chris Lattnerda828e32010-03-03 07:46:25 +00002011 case SelectionDAGISel::OPC_CheckCondCode:
2012 Result = !::CheckCondCode(Table, Index, N);
2013 return Index;
2014 case SelectionDAGISel::OPC_CheckValueType:
2015 Result = !::CheckValueType(Table, Index, N, SDISel.TLI);
2016 return Index;
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002017 case SelectionDAGISel::OPC_CheckInteger:
2018 Result = !::CheckInteger(Table, Index, N);
2019 return Index;
Chris Lattnerda828e32010-03-03 07:46:25 +00002020 case SelectionDAGISel::OPC_CheckAndImm:
2021 Result = !::CheckAndImm(Table, Index, N, SDISel);
2022 return Index;
2023 case SelectionDAGISel::OPC_CheckOrImm:
2024 Result = !::CheckOrImm(Table, Index, N, SDISel);
2025 return Index;
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002026 }
2027}
2028
Chris Lattner3ff1e4d2010-03-02 06:55:04 +00002029
Chris Lattner2a49d572010-02-28 22:37:22 +00002030struct MatchScope {
2031 /// FailIndex - If this match fails, this is the index to continue with.
2032 unsigned FailIndex;
2033
2034 /// NodeStack - The node stack when the scope was formed.
2035 SmallVector<SDValue, 4> NodeStack;
2036
2037 /// NumRecordedNodes - The number of recorded nodes when the scope was formed.
2038 unsigned NumRecordedNodes;
2039
2040 /// NumMatchedMemRefs - The number of matched memref entries.
2041 unsigned NumMatchedMemRefs;
2042
2043 /// InputChain/InputFlag - The current chain/flag
2044 SDValue InputChain, InputFlag;
2045
2046 /// HasChainNodesMatched - True if the ChainNodesMatched list is non-empty.
2047 bool HasChainNodesMatched, HasFlagResultNodesMatched;
2048};
2049
2050SDNode *SelectionDAGISel::
2051SelectCodeCommon(SDNode *NodeToMatch, const unsigned char *MatcherTable,
2052 unsigned TableSize) {
2053 // FIXME: Should these even be selected? Handle these cases in the caller?
2054 switch (NodeToMatch->getOpcode()) {
2055 default:
2056 break;
2057 case ISD::EntryToken: // These nodes remain the same.
2058 case ISD::BasicBlock:
2059 case ISD::Register:
Chris Lattner3b9d6212010-03-14 17:53:23 +00002060 //case ISD::VALUETYPE:
2061 //case ISD::CONDCODE:
Chris Lattner2a49d572010-02-28 22:37:22 +00002062 case ISD::HANDLENODE:
2063 case ISD::TargetConstant:
2064 case ISD::TargetConstantFP:
2065 case ISD::TargetConstantPool:
2066 case ISD::TargetFrameIndex:
2067 case ISD::TargetExternalSymbol:
2068 case ISD::TargetBlockAddress:
2069 case ISD::TargetJumpTable:
2070 case ISD::TargetGlobalTLSAddress:
2071 case ISD::TargetGlobalAddress:
2072 case ISD::TokenFactor:
2073 case ISD::CopyFromReg:
2074 case ISD::CopyToReg:
Chris Lattner7561d482010-03-14 02:33:54 +00002075 case ISD::EH_LABEL:
Chris Lattnerd1b73822010-03-02 22:20:06 +00002076 NodeToMatch->setNodeId(-1); // Mark selected.
Chris Lattner2a49d572010-02-28 22:37:22 +00002077 return 0;
2078 case ISD::AssertSext:
2079 case ISD::AssertZext:
2080 CurDAG->ReplaceAllUsesOfValueWith(SDValue(NodeToMatch, 0),
2081 NodeToMatch->getOperand(0));
2082 return 0;
2083 case ISD::INLINEASM: return Select_INLINEASM(NodeToMatch);
Chris Lattner2a49d572010-02-28 22:37:22 +00002084 case ISD::UNDEF: return Select_UNDEF(NodeToMatch);
2085 }
2086
2087 assert(!NodeToMatch->isMachineOpcode() && "Node already selected!");
2088
2089 // Set up the node stack with NodeToMatch as the only node on the stack.
2090 SmallVector<SDValue, 8> NodeStack;
2091 SDValue N = SDValue(NodeToMatch, 0);
2092 NodeStack.push_back(N);
2093
2094 // MatchScopes - Scopes used when matching, if a match failure happens, this
2095 // indicates where to continue checking.
2096 SmallVector<MatchScope, 8> MatchScopes;
2097
2098 // RecordedNodes - This is the set of nodes that have been recorded by the
2099 // state machine.
2100 SmallVector<SDValue, 8> RecordedNodes;
2101
2102 // MatchedMemRefs - This is the set of MemRef's we've seen in the input
2103 // pattern.
2104 SmallVector<MachineMemOperand*, 2> MatchedMemRefs;
2105
2106 // These are the current input chain and flag for use when generating nodes.
2107 // Various Emit operations change these. For example, emitting a copytoreg
2108 // uses and updates these.
2109 SDValue InputChain, InputFlag;
2110
2111 // ChainNodesMatched - If a pattern matches nodes that have input/output
2112 // chains, the OPC_EmitMergeInputChains operation is emitted which indicates
2113 // which ones they are. The result is captured into this list so that we can
2114 // update the chain results when the pattern is complete.
2115 SmallVector<SDNode*, 3> ChainNodesMatched;
2116 SmallVector<SDNode*, 3> FlagResultNodesMatched;
2117
2118 DEBUG(errs() << "ISEL: Starting pattern match on root node: ";
2119 NodeToMatch->dump(CurDAG);
2120 errs() << '\n');
2121
Chris Lattner7390eeb2010-03-01 18:47:11 +00002122 // Determine where to start the interpreter. Normally we start at opcode #0,
2123 // but if the state machine starts with an OPC_SwitchOpcode, then we
2124 // accelerate the first lookup (which is guaranteed to be hot) with the
2125 // OpcodeOffset table.
Chris Lattner2a49d572010-02-28 22:37:22 +00002126 unsigned MatcherIndex = 0;
Chris Lattner7390eeb2010-03-01 18:47:11 +00002127
2128 if (!OpcodeOffset.empty()) {
2129 // Already computed the OpcodeOffset table, just index into it.
2130 if (N.getOpcode() < OpcodeOffset.size())
2131 MatcherIndex = OpcodeOffset[N.getOpcode()];
2132 DEBUG(errs() << " Initial Opcode index to " << MatcherIndex << "\n");
2133
2134 } else if (MatcherTable[0] == OPC_SwitchOpcode) {
2135 // Otherwise, the table isn't computed, but the state machine does start
2136 // with an OPC_SwitchOpcode instruction. Populate the table now, since this
2137 // is the first time we're selecting an instruction.
2138 unsigned Idx = 1;
2139 while (1) {
2140 // Get the size of this case.
2141 unsigned CaseSize = MatcherTable[Idx++];
2142 if (CaseSize & 128)
2143 CaseSize = GetVBR(CaseSize, MatcherTable, Idx);
2144 if (CaseSize == 0) break;
2145
2146 // Get the opcode, add the index to the table.
Chris Lattnerd5d5a3d2010-03-25 06:33:05 +00002147 uint16_t Opc = MatcherTable[Idx++];
2148 Opc |= (unsigned short)MatcherTable[Idx++] << 8;
Chris Lattner7390eeb2010-03-01 18:47:11 +00002149 if (Opc >= OpcodeOffset.size())
2150 OpcodeOffset.resize((Opc+1)*2);
2151 OpcodeOffset[Opc] = Idx;
2152 Idx += CaseSize;
2153 }
2154
2155 // Okay, do the lookup for the first opcode.
2156 if (N.getOpcode() < OpcodeOffset.size())
2157 MatcherIndex = OpcodeOffset[N.getOpcode()];
2158 }
2159
Chris Lattner2a49d572010-02-28 22:37:22 +00002160 while (1) {
2161 assert(MatcherIndex < TableSize && "Invalid index");
Dan Gohman19b38262010-03-09 02:15:05 +00002162#ifndef NDEBUG
2163 unsigned CurrentOpcodeIndex = MatcherIndex;
2164#endif
Chris Lattner2a49d572010-02-28 22:37:22 +00002165 BuiltinOpcodes Opcode = (BuiltinOpcodes)MatcherTable[MatcherIndex++];
2166 switch (Opcode) {
2167 case OPC_Scope: {
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002168 // Okay, the semantics of this operation are that we should push a scope
2169 // then evaluate the first child. However, pushing a scope only to have
2170 // the first check fail (which then pops it) is inefficient. If we can
2171 // determine immediately that the first check (or first several) will
2172 // immediately fail, don't even bother pushing a scope for them.
2173 unsigned FailIndex;
2174
2175 while (1) {
2176 unsigned NumToSkip = MatcherTable[MatcherIndex++];
2177 if (NumToSkip & 128)
2178 NumToSkip = GetVBR(NumToSkip, MatcherTable, MatcherIndex);
2179 // Found the end of the scope with no match.
2180 if (NumToSkip == 0) {
2181 FailIndex = 0;
2182 break;
2183 }
2184
2185 FailIndex = MatcherIndex+NumToSkip;
2186
Chris Lattnera6f86932010-03-27 18:54:50 +00002187 unsigned MatcherIndexOfPredicate = MatcherIndex;
2188 (void)MatcherIndexOfPredicate; // silence warning.
2189
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002190 // If we can't evaluate this predicate without pushing a scope (e.g. if
2191 // it is a 'MoveParent') or if the predicate succeeds on this node, we
2192 // push the scope and evaluate the full predicate chain.
2193 bool Result;
2194 MatcherIndex = IsPredicateKnownToFail(MatcherTable, MatcherIndex, N,
Chris Lattnerda828e32010-03-03 07:46:25 +00002195 Result, *this, RecordedNodes);
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002196 if (!Result)
2197 break;
2198
Chris Lattnera6f86932010-03-27 18:54:50 +00002199 DEBUG(errs() << " Skipped scope entry (due to false predicate) at "
2200 << "index " << MatcherIndexOfPredicate
2201 << ", continuing at " << FailIndex << "\n");
Chris Lattnerbed993d2010-03-28 19:46:56 +00002202 ++NumDAGIselRetries;
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002203
2204 // Otherwise, we know that this case of the Scope is guaranteed to fail,
2205 // move to the next case.
2206 MatcherIndex = FailIndex;
2207 }
2208
2209 // If the whole scope failed to match, bail.
2210 if (FailIndex == 0) break;
2211
Chris Lattner2a49d572010-02-28 22:37:22 +00002212 // Push a MatchScope which indicates where to go if the first child fails
2213 // to match.
2214 MatchScope NewEntry;
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002215 NewEntry.FailIndex = FailIndex;
Chris Lattner2a49d572010-02-28 22:37:22 +00002216 NewEntry.NodeStack.append(NodeStack.begin(), NodeStack.end());
2217 NewEntry.NumRecordedNodes = RecordedNodes.size();
2218 NewEntry.NumMatchedMemRefs = MatchedMemRefs.size();
2219 NewEntry.InputChain = InputChain;
2220 NewEntry.InputFlag = InputFlag;
2221 NewEntry.HasChainNodesMatched = !ChainNodesMatched.empty();
2222 NewEntry.HasFlagResultNodesMatched = !FlagResultNodesMatched.empty();
2223 MatchScopes.push_back(NewEntry);
2224 continue;
2225 }
2226 case OPC_RecordNode:
2227 // Remember this node, it may end up being an operand in the pattern.
2228 RecordedNodes.push_back(N);
2229 continue;
2230
2231 case OPC_RecordChild0: case OPC_RecordChild1:
2232 case OPC_RecordChild2: case OPC_RecordChild3:
2233 case OPC_RecordChild4: case OPC_RecordChild5:
2234 case OPC_RecordChild6: case OPC_RecordChild7: {
2235 unsigned ChildNo = Opcode-OPC_RecordChild0;
2236 if (ChildNo >= N.getNumOperands())
2237 break; // Match fails if out of range child #.
2238
2239 RecordedNodes.push_back(N->getOperand(ChildNo));
2240 continue;
2241 }
2242 case OPC_RecordMemRef:
2243 MatchedMemRefs.push_back(cast<MemSDNode>(N)->getMemOperand());
2244 continue;
2245
2246 case OPC_CaptureFlagInput:
2247 // If the current node has an input flag, capture it in InputFlag.
2248 if (N->getNumOperands() != 0 &&
2249 N->getOperand(N->getNumOperands()-1).getValueType() == MVT::Flag)
2250 InputFlag = N->getOperand(N->getNumOperands()-1);
2251 continue;
2252
2253 case OPC_MoveChild: {
2254 unsigned ChildNo = MatcherTable[MatcherIndex++];
2255 if (ChildNo >= N.getNumOperands())
2256 break; // Match fails if out of range child #.
2257 N = N.getOperand(ChildNo);
2258 NodeStack.push_back(N);
2259 continue;
2260 }
2261
2262 case OPC_MoveParent:
2263 // Pop the current node off the NodeStack.
2264 NodeStack.pop_back();
2265 assert(!NodeStack.empty() && "Node stack imbalance!");
2266 N = NodeStack.back();
2267 continue;
2268
Chris Lattnerda828e32010-03-03 07:46:25 +00002269 case OPC_CheckSame:
2270 if (!::CheckSame(MatcherTable, MatcherIndex, N, RecordedNodes)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002271 continue;
Chris Lattner2a49d572010-02-28 22:37:22 +00002272 case OPC_CheckPatternPredicate:
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002273 if (!::CheckPatternPredicate(MatcherTable, MatcherIndex, *this)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002274 continue;
2275 case OPC_CheckPredicate:
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002276 if (!::CheckNodePredicate(MatcherTable, MatcherIndex, *this,
2277 N.getNode()))
2278 break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002279 continue;
Chris Lattner57bf8a42010-03-04 01:23:08 +00002280 case OPC_CheckComplexPat: {
2281 unsigned CPNum = MatcherTable[MatcherIndex++];
2282 unsigned RecNo = MatcherTable[MatcherIndex++];
2283 assert(RecNo < RecordedNodes.size() && "Invalid CheckComplexPat");
2284 if (!CheckComplexPattern(NodeToMatch, RecordedNodes[RecNo], CPNum,
2285 RecordedNodes))
Chris Lattner2a49d572010-02-28 22:37:22 +00002286 break;
2287 continue;
Chris Lattner57bf8a42010-03-04 01:23:08 +00002288 }
Chris Lattner2a49d572010-02-28 22:37:22 +00002289 case OPC_CheckOpcode:
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002290 if (!::CheckOpcode(MatcherTable, MatcherIndex, N.getNode())) break;
2291 continue;
2292
2293 case OPC_CheckType:
2294 if (!::CheckType(MatcherTable, MatcherIndex, N, TLI)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002295 continue;
2296
Chris Lattnereb669212010-03-01 06:59:22 +00002297 case OPC_SwitchOpcode: {
2298 unsigned CurNodeOpcode = N.getOpcode();
Chris Lattner7d892d62010-03-01 07:43:08 +00002299 unsigned SwitchStart = MatcherIndex-1; (void)SwitchStart;
Chris Lattnereb669212010-03-01 06:59:22 +00002300 unsigned CaseSize;
2301 while (1) {
2302 // Get the size of this case.
2303 CaseSize = MatcherTable[MatcherIndex++];
2304 if (CaseSize & 128)
2305 CaseSize = GetVBR(CaseSize, MatcherTable, MatcherIndex);
2306 if (CaseSize == 0) break;
2307
Chris Lattnerd5d5a3d2010-03-25 06:33:05 +00002308 uint16_t Opc = MatcherTable[MatcherIndex++];
2309 Opc |= (unsigned short)MatcherTable[MatcherIndex++] << 8;
2310
Chris Lattnereb669212010-03-01 06:59:22 +00002311 // If the opcode matches, then we will execute this case.
Chris Lattnerd5d5a3d2010-03-25 06:33:05 +00002312 if (CurNodeOpcode == Opc)
Chris Lattnereb669212010-03-01 06:59:22 +00002313 break;
2314
2315 // Otherwise, skip over this case.
2316 MatcherIndex += CaseSize;
2317 }
2318
Chris Lattnercfe2eab2010-03-03 06:28:15 +00002319 // If no cases matched, bail out.
Chris Lattnereb669212010-03-01 06:59:22 +00002320 if (CaseSize == 0) break;
2321
2322 // Otherwise, execute the case we found.
2323 DEBUG(errs() << " OpcodeSwitch from " << SwitchStart
2324 << " to " << MatcherIndex << "\n");
2325 continue;
2326 }
2327
Chris Lattnercfe2eab2010-03-03 06:28:15 +00002328 case OPC_SwitchType: {
2329 MVT::SimpleValueType CurNodeVT = N.getValueType().getSimpleVT().SimpleTy;
2330 unsigned SwitchStart = MatcherIndex-1; (void)SwitchStart;
2331 unsigned CaseSize;
2332 while (1) {
2333 // Get the size of this case.
2334 CaseSize = MatcherTable[MatcherIndex++];
2335 if (CaseSize & 128)
2336 CaseSize = GetVBR(CaseSize, MatcherTable, MatcherIndex);
2337 if (CaseSize == 0) break;
2338
2339 MVT::SimpleValueType CaseVT =
2340 (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
2341 if (CaseVT == MVT::iPTR)
2342 CaseVT = TLI.getPointerTy().SimpleTy;
2343
2344 // If the VT matches, then we will execute this case.
2345 if (CurNodeVT == CaseVT)
2346 break;
2347
2348 // Otherwise, skip over this case.
2349 MatcherIndex += CaseSize;
2350 }
2351
2352 // If no cases matched, bail out.
2353 if (CaseSize == 0) break;
2354
2355 // Otherwise, execute the case we found.
2356 DEBUG(errs() << " TypeSwitch[" << EVT(CurNodeVT).getEVTString()
2357 << "] from " << SwitchStart << " to " << MatcherIndex<<'\n');
2358 continue;
2359 }
Chris Lattner2a49d572010-02-28 22:37:22 +00002360 case OPC_CheckChild0Type: case OPC_CheckChild1Type:
2361 case OPC_CheckChild2Type: case OPC_CheckChild3Type:
2362 case OPC_CheckChild4Type: case OPC_CheckChild5Type:
Chris Lattnerda828e32010-03-03 07:46:25 +00002363 case OPC_CheckChild6Type: case OPC_CheckChild7Type:
2364 if (!::CheckChildType(MatcherTable, MatcherIndex, N, TLI,
2365 Opcode-OPC_CheckChild0Type))
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002366 break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002367 continue;
Chris Lattner2a49d572010-02-28 22:37:22 +00002368 case OPC_CheckCondCode:
Chris Lattnerda828e32010-03-03 07:46:25 +00002369 if (!::CheckCondCode(MatcherTable, MatcherIndex, N)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002370 continue;
Chris Lattnerda828e32010-03-03 07:46:25 +00002371 case OPC_CheckValueType:
2372 if (!::CheckValueType(MatcherTable, MatcherIndex, N, TLI)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002373 continue;
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002374 case OPC_CheckInteger:
2375 if (!::CheckInteger(MatcherTable, MatcherIndex, N)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002376 continue;
Chris Lattnerda828e32010-03-03 07:46:25 +00002377 case OPC_CheckAndImm:
2378 if (!::CheckAndImm(MatcherTable, MatcherIndex, N, *this)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002379 continue;
Chris Lattnerda828e32010-03-03 07:46:25 +00002380 case OPC_CheckOrImm:
2381 if (!::CheckOrImm(MatcherTable, MatcherIndex, N, *this)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002382 continue;
Chris Lattner2a49d572010-02-28 22:37:22 +00002383
2384 case OPC_CheckFoldableChainNode: {
2385 assert(NodeStack.size() != 1 && "No parent node");
2386 // Verify that all intermediate nodes between the root and this one have
2387 // a single use.
2388 bool HasMultipleUses = false;
2389 for (unsigned i = 1, e = NodeStack.size()-1; i != e; ++i)
2390 if (!NodeStack[i].hasOneUse()) {
2391 HasMultipleUses = true;
2392 break;
2393 }
2394 if (HasMultipleUses) break;
2395
2396 // Check to see that the target thinks this is profitable to fold and that
2397 // we can fold it without inducing cycles in the graph.
2398 if (!IsProfitableToFold(N, NodeStack[NodeStack.size()-2].getNode(),
2399 NodeToMatch) ||
2400 !IsLegalToFold(N, NodeStack[NodeStack.size()-2].getNode(),
Chris Lattnerd1b73822010-03-02 22:20:06 +00002401 NodeToMatch, true/*We validate our own chains*/))
Chris Lattner2a49d572010-02-28 22:37:22 +00002402 break;
2403
2404 continue;
2405 }
Chris Lattner2a49d572010-02-28 22:37:22 +00002406 case OPC_EmitInteger: {
2407 MVT::SimpleValueType VT =
2408 (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
2409 int64_t Val = MatcherTable[MatcherIndex++];
2410 if (Val & 128)
2411 Val = GetVBR(Val, MatcherTable, MatcherIndex);
2412 RecordedNodes.push_back(CurDAG->getTargetConstant(Val, VT));
2413 continue;
2414 }
2415 case OPC_EmitRegister: {
2416 MVT::SimpleValueType VT =
2417 (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
2418 unsigned RegNo = MatcherTable[MatcherIndex++];
2419 RecordedNodes.push_back(CurDAG->getRegister(RegNo, VT));
2420 continue;
2421 }
2422
2423 case OPC_EmitConvertToTarget: {
2424 // Convert from IMM/FPIMM to target version.
2425 unsigned RecNo = MatcherTable[MatcherIndex++];
2426 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2427 SDValue Imm = RecordedNodes[RecNo];
2428
2429 if (Imm->getOpcode() == ISD::Constant) {
2430 int64_t Val = cast<ConstantSDNode>(Imm)->getZExtValue();
2431 Imm = CurDAG->getTargetConstant(Val, Imm.getValueType());
2432 } else if (Imm->getOpcode() == ISD::ConstantFP) {
2433 const ConstantFP *Val=cast<ConstantFPSDNode>(Imm)->getConstantFPValue();
2434 Imm = CurDAG->getTargetConstantFP(*Val, Imm.getValueType());
2435 }
2436
2437 RecordedNodes.push_back(Imm);
2438 continue;
2439 }
2440
Chris Lattneraa4e3392010-03-28 05:50:16 +00002441 case OPC_EmitMergeInputChains1_0: // OPC_EmitMergeInputChains, 1, 0
2442 case OPC_EmitMergeInputChains1_1: { // OPC_EmitMergeInputChains, 1, 1
2443 // These are space-optimized forms of OPC_EmitMergeInputChains.
2444 assert(InputChain.getNode() == 0 &&
2445 "EmitMergeInputChains should be the first chain producing node");
2446 assert(ChainNodesMatched.empty() &&
2447 "Should only have one EmitMergeInputChains per match");
2448
2449 // Read all of the chained nodes.
2450 unsigned RecNo = Opcode == OPC_EmitMergeInputChains1_1;
2451 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2452 ChainNodesMatched.push_back(RecordedNodes[RecNo].getNode());
2453
2454 // FIXME: What if other value results of the node have uses not matched
2455 // by this pattern?
2456 if (ChainNodesMatched.back() != NodeToMatch &&
2457 !RecordedNodes[RecNo].hasOneUse()) {
2458 ChainNodesMatched.clear();
2459 break;
2460 }
2461
2462 // Merge the input chains if they are not intra-pattern references.
2463 InputChain = HandleMergeInputChains(ChainNodesMatched, CurDAG);
2464
2465 if (InputChain.getNode() == 0)
2466 break; // Failed to merge.
2467 continue;
2468 }
2469
Chris Lattner2a49d572010-02-28 22:37:22 +00002470 case OPC_EmitMergeInputChains: {
2471 assert(InputChain.getNode() == 0 &&
2472 "EmitMergeInputChains should be the first chain producing node");
2473 // This node gets a list of nodes we matched in the input that have
2474 // chains. We want to token factor all of the input chains to these nodes
2475 // together. However, if any of the input chains is actually one of the
2476 // nodes matched in this pattern, then we have an intra-match reference.
2477 // Ignore these because the newly token factored chain should not refer to
2478 // the old nodes.
2479 unsigned NumChains = MatcherTable[MatcherIndex++];
2480 assert(NumChains != 0 && "Can't TF zero chains");
2481
2482 assert(ChainNodesMatched.empty() &&
2483 "Should only have one EmitMergeInputChains per match");
2484
Chris Lattner2a49d572010-02-28 22:37:22 +00002485 // Read all of the chained nodes.
Chris Lattner00592ec2010-03-02 19:34:59 +00002486 for (unsigned i = 0; i != NumChains; ++i) {
2487 unsigned RecNo = MatcherTable[MatcherIndex++];
Chris Lattner2a49d572010-02-28 22:37:22 +00002488 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2489 ChainNodesMatched.push_back(RecordedNodes[RecNo].getNode());
2490
2491 // FIXME: What if other value results of the node have uses not matched
2492 // by this pattern?
2493 if (ChainNodesMatched.back() != NodeToMatch &&
2494 !RecordedNodes[RecNo].hasOneUse()) {
2495 ChainNodesMatched.clear();
2496 break;
2497 }
2498 }
Chris Lattner6b307922010-03-02 00:00:03 +00002499
2500 // If the inner loop broke out, the match fails.
2501 if (ChainNodesMatched.empty())
2502 break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002503
Chris Lattner6b307922010-03-02 00:00:03 +00002504 // Merge the input chains if they are not intra-pattern references.
2505 InputChain = HandleMergeInputChains(ChainNodesMatched, CurDAG);
2506
2507 if (InputChain.getNode() == 0)
2508 break; // Failed to merge.
Chris Lattner2a49d572010-02-28 22:37:22 +00002509
Chris Lattner2a49d572010-02-28 22:37:22 +00002510 continue;
2511 }
2512
2513 case OPC_EmitCopyToReg: {
2514 unsigned RecNo = MatcherTable[MatcherIndex++];
2515 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2516 unsigned DestPhysReg = MatcherTable[MatcherIndex++];
2517
2518 if (InputChain.getNode() == 0)
2519 InputChain = CurDAG->getEntryNode();
2520
2521 InputChain = CurDAG->getCopyToReg(InputChain, NodeToMatch->getDebugLoc(),
2522 DestPhysReg, RecordedNodes[RecNo],
2523 InputFlag);
2524
2525 InputFlag = InputChain.getValue(1);
2526 continue;
2527 }
2528
2529 case OPC_EmitNodeXForm: {
2530 unsigned XFormNo = MatcherTable[MatcherIndex++];
2531 unsigned RecNo = MatcherTable[MatcherIndex++];
2532 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2533 RecordedNodes.push_back(RunSDNodeXForm(RecordedNodes[RecNo], XFormNo));
2534 continue;
2535 }
2536
2537 case OPC_EmitNode:
2538 case OPC_MorphNodeTo: {
Chris Lattner14df8dc2010-02-28 22:38:43 +00002539 uint16_t TargetOpc = MatcherTable[MatcherIndex++];
2540 TargetOpc |= (unsigned short)MatcherTable[MatcherIndex++] << 8;
Chris Lattner2a49d572010-02-28 22:37:22 +00002541 unsigned EmitNodeInfo = MatcherTable[MatcherIndex++];
2542 // Get the result VT list.
2543 unsigned NumVTs = MatcherTable[MatcherIndex++];
2544 SmallVector<EVT, 4> VTs;
2545 for (unsigned i = 0; i != NumVTs; ++i) {
2546 MVT::SimpleValueType VT =
2547 (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
2548 if (VT == MVT::iPTR) VT = TLI.getPointerTy().SimpleTy;
2549 VTs.push_back(VT);
2550 }
2551
2552 if (EmitNodeInfo & OPFL_Chain)
2553 VTs.push_back(MVT::Other);
2554 if (EmitNodeInfo & OPFL_FlagOutput)
2555 VTs.push_back(MVT::Flag);
2556
Chris Lattner7d892d62010-03-01 07:43:08 +00002557 // This is hot code, so optimize the two most common cases of 1 and 2
2558 // results.
2559 SDVTList VTList;
2560 if (VTs.size() == 1)
2561 VTList = CurDAG->getVTList(VTs[0]);
2562 else if (VTs.size() == 2)
2563 VTList = CurDAG->getVTList(VTs[0], VTs[1]);
2564 else
2565 VTList = CurDAG->getVTList(VTs.data(), VTs.size());
Chris Lattner2a49d572010-02-28 22:37:22 +00002566
2567 // Get the operand list.
2568 unsigned NumOps = MatcherTable[MatcherIndex++];
2569 SmallVector<SDValue, 8> Ops;
2570 for (unsigned i = 0; i != NumOps; ++i) {
2571 unsigned RecNo = MatcherTable[MatcherIndex++];
2572 if (RecNo & 128)
2573 RecNo = GetVBR(RecNo, MatcherTable, MatcherIndex);
2574
2575 assert(RecNo < RecordedNodes.size() && "Invalid EmitNode");
2576 Ops.push_back(RecordedNodes[RecNo]);
2577 }
2578
2579 // If there are variadic operands to add, handle them now.
2580 if (EmitNodeInfo & OPFL_VariadicInfo) {
2581 // Determine the start index to copy from.
2582 unsigned FirstOpToCopy = getNumFixedFromVariadicInfo(EmitNodeInfo);
2583 FirstOpToCopy += (EmitNodeInfo & OPFL_Chain) ? 1 : 0;
2584 assert(NodeToMatch->getNumOperands() >= FirstOpToCopy &&
2585 "Invalid variadic node");
2586 // Copy all of the variadic operands, not including a potential flag
2587 // input.
2588 for (unsigned i = FirstOpToCopy, e = NodeToMatch->getNumOperands();
2589 i != e; ++i) {
2590 SDValue V = NodeToMatch->getOperand(i);
2591 if (V.getValueType() == MVT::Flag) break;
2592 Ops.push_back(V);
2593 }
2594 }
2595
2596 // If this has chain/flag inputs, add them.
2597 if (EmitNodeInfo & OPFL_Chain)
2598 Ops.push_back(InputChain);
2599 if ((EmitNodeInfo & OPFL_FlagInput) && InputFlag.getNode() != 0)
2600 Ops.push_back(InputFlag);
2601
2602 // Create the node.
2603 SDNode *Res = 0;
2604 if (Opcode != OPC_MorphNodeTo) {
2605 // If this is a normal EmitNode command, just create the new node and
2606 // add the results to the RecordedNodes list.
2607 Res = CurDAG->getMachineNode(TargetOpc, NodeToMatch->getDebugLoc(),
2608 VTList, Ops.data(), Ops.size());
2609
2610 // Add all the non-flag/non-chain results to the RecordedNodes list.
2611 for (unsigned i = 0, e = VTs.size(); i != e; ++i) {
2612 if (VTs[i] == MVT::Other || VTs[i] == MVT::Flag) break;
2613 RecordedNodes.push_back(SDValue(Res, i));
2614 }
2615
2616 } else {
Chris Lattner3ff1e4d2010-03-02 06:55:04 +00002617 Res = MorphNode(NodeToMatch, TargetOpc, VTList, Ops.data(), Ops.size(),
2618 EmitNodeInfo);
Chris Lattner2a49d572010-02-28 22:37:22 +00002619 }
2620
2621 // If the node had chain/flag results, update our notion of the current
2622 // chain and flag.
Chris Lattner82dd3d32010-03-02 07:50:03 +00002623 if (EmitNodeInfo & OPFL_FlagOutput) {
Chris Lattner2a49d572010-02-28 22:37:22 +00002624 InputFlag = SDValue(Res, VTs.size()-1);
2625 if (EmitNodeInfo & OPFL_Chain)
2626 InputChain = SDValue(Res, VTs.size()-2);
2627 } else if (EmitNodeInfo & OPFL_Chain)
2628 InputChain = SDValue(Res, VTs.size()-1);
2629
2630 // If the OPFL_MemRefs flag is set on this node, slap all of the
2631 // accumulated memrefs onto it.
2632 //
2633 // FIXME: This is vastly incorrect for patterns with multiple outputs
2634 // instructions that access memory and for ComplexPatterns that match
2635 // loads.
2636 if (EmitNodeInfo & OPFL_MemRefs) {
2637 MachineSDNode::mmo_iterator MemRefs =
2638 MF->allocateMemRefsArray(MatchedMemRefs.size());
2639 std::copy(MatchedMemRefs.begin(), MatchedMemRefs.end(), MemRefs);
2640 cast<MachineSDNode>(Res)
2641 ->setMemRefs(MemRefs, MemRefs + MatchedMemRefs.size());
2642 }
2643
2644 DEBUG(errs() << " "
2645 << (Opcode == OPC_MorphNodeTo ? "Morphed" : "Created")
2646 << " node: "; Res->dump(CurDAG); errs() << "\n");
2647
2648 // If this was a MorphNodeTo then we're completely done!
2649 if (Opcode == OPC_MorphNodeTo) {
2650 // Update chain and flag uses.
2651 UpdateChainsAndFlags(NodeToMatch, InputChain, ChainNodesMatched,
Chris Lattner82dd3d32010-03-02 07:50:03 +00002652 InputFlag, FlagResultNodesMatched, true);
2653 return Res;
Chris Lattner2a49d572010-02-28 22:37:22 +00002654 }
2655
2656 continue;
2657 }
2658
2659 case OPC_MarkFlagResults: {
2660 unsigned NumNodes = MatcherTable[MatcherIndex++];
2661
2662 // Read and remember all the flag-result nodes.
2663 for (unsigned i = 0; i != NumNodes; ++i) {
2664 unsigned RecNo = MatcherTable[MatcherIndex++];
2665 if (RecNo & 128)
2666 RecNo = GetVBR(RecNo, MatcherTable, MatcherIndex);
2667
2668 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2669 FlagResultNodesMatched.push_back(RecordedNodes[RecNo].getNode());
2670 }
2671 continue;
2672 }
2673
2674 case OPC_CompleteMatch: {
2675 // The match has been completed, and any new nodes (if any) have been
2676 // created. Patch up references to the matched dag to use the newly
2677 // created nodes.
2678 unsigned NumResults = MatcherTable[MatcherIndex++];
2679
2680 for (unsigned i = 0; i != NumResults; ++i) {
2681 unsigned ResSlot = MatcherTable[MatcherIndex++];
2682 if (ResSlot & 128)
2683 ResSlot = GetVBR(ResSlot, MatcherTable, MatcherIndex);
2684
2685 assert(ResSlot < RecordedNodes.size() && "Invalid CheckSame");
2686 SDValue Res = RecordedNodes[ResSlot];
2687
Chris Lattnerb0e483e2010-03-28 05:54:03 +00002688 assert(i < NodeToMatch->getNumValues() &&
2689 NodeToMatch->getValueType(i) != MVT::Other &&
2690 NodeToMatch->getValueType(i) != MVT::Flag &&
2691 "Invalid number of results to complete!");
Chris Lattner2a49d572010-02-28 22:37:22 +00002692 assert((NodeToMatch->getValueType(i) == Res.getValueType() ||
2693 NodeToMatch->getValueType(i) == MVT::iPTR ||
2694 Res.getValueType() == MVT::iPTR ||
2695 NodeToMatch->getValueType(i).getSizeInBits() ==
2696 Res.getValueType().getSizeInBits()) &&
2697 "invalid replacement");
2698 CurDAG->ReplaceAllUsesOfValueWith(SDValue(NodeToMatch, i), Res);
2699 }
2700
2701 // If the root node defines a flag, add it to the flag nodes to update
2702 // list.
2703 if (NodeToMatch->getValueType(NodeToMatch->getNumValues()-1) == MVT::Flag)
2704 FlagResultNodesMatched.push_back(NodeToMatch);
2705
2706 // Update chain and flag uses.
2707 UpdateChainsAndFlags(NodeToMatch, InputChain, ChainNodesMatched,
Chris Lattner82dd3d32010-03-02 07:50:03 +00002708 InputFlag, FlagResultNodesMatched, false);
Chris Lattner2a49d572010-02-28 22:37:22 +00002709
2710 assert(NodeToMatch->use_empty() &&
2711 "Didn't replace all uses of the node?");
2712
2713 // FIXME: We just return here, which interacts correctly with SelectRoot
2714 // above. We should fix this to not return an SDNode* anymore.
2715 return 0;
2716 }
2717 }
2718
2719 // If the code reached this point, then the match failed. See if there is
2720 // another child to try in the current 'Scope', otherwise pop it until we
2721 // find a case to check.
Dan Gohman19b38262010-03-09 02:15:05 +00002722 DEBUG(errs() << " Match failed at index " << CurrentOpcodeIndex << "\n");
Chris Lattnerbed993d2010-03-28 19:46:56 +00002723 ++NumDAGIselRetries;
Chris Lattner2a49d572010-02-28 22:37:22 +00002724 while (1) {
2725 if (MatchScopes.empty()) {
2726 CannotYetSelect(NodeToMatch);
2727 return 0;
2728 }
2729
2730 // Restore the interpreter state back to the point where the scope was
2731 // formed.
2732 MatchScope &LastScope = MatchScopes.back();
2733 RecordedNodes.resize(LastScope.NumRecordedNodes);
2734 NodeStack.clear();
2735 NodeStack.append(LastScope.NodeStack.begin(), LastScope.NodeStack.end());
2736 N = NodeStack.back();
2737
Chris Lattner2a49d572010-02-28 22:37:22 +00002738 if (LastScope.NumMatchedMemRefs != MatchedMemRefs.size())
2739 MatchedMemRefs.resize(LastScope.NumMatchedMemRefs);
2740 MatcherIndex = LastScope.FailIndex;
2741
Dan Gohman19b38262010-03-09 02:15:05 +00002742 DEBUG(errs() << " Continuing at " << MatcherIndex << "\n");
2743
Chris Lattner2a49d572010-02-28 22:37:22 +00002744 InputChain = LastScope.InputChain;
2745 InputFlag = LastScope.InputFlag;
2746 if (!LastScope.HasChainNodesMatched)
2747 ChainNodesMatched.clear();
2748 if (!LastScope.HasFlagResultNodesMatched)
2749 FlagResultNodesMatched.clear();
2750
2751 // Check to see what the offset is at the new MatcherIndex. If it is zero
2752 // we have reached the end of this scope, otherwise we have another child
2753 // in the current scope to try.
2754 unsigned NumToSkip = MatcherTable[MatcherIndex++];
2755 if (NumToSkip & 128)
2756 NumToSkip = GetVBR(NumToSkip, MatcherTable, MatcherIndex);
2757
2758 // If we have another child in this scope to match, update FailIndex and
2759 // try it.
2760 if (NumToSkip != 0) {
2761 LastScope.FailIndex = MatcherIndex+NumToSkip;
2762 break;
2763 }
2764
2765 // End of this scope, pop it and try the next child in the containing
2766 // scope.
2767 MatchScopes.pop_back();
2768 }
2769 }
2770}
2771
2772
2773
Dan Gohmaneeb3a002010-01-05 01:24:18 +00002774void SelectionDAGISel::CannotYetSelect(SDNode *N) {
Dan Gohmane1f188f2009-10-29 22:30:23 +00002775 std::string msg;
2776 raw_string_ostream Msg(msg);
2777 Msg << "Cannot yet select: ";
Chris Lattner2c4afd12010-03-04 00:21:16 +00002778
2779 if (N->getOpcode() != ISD::INTRINSIC_W_CHAIN &&
2780 N->getOpcode() != ISD::INTRINSIC_WO_CHAIN &&
2781 N->getOpcode() != ISD::INTRINSIC_VOID) {
2782 N->printrFull(Msg, CurDAG);
2783 } else {
2784 bool HasInputChain = N->getOperand(0).getValueType() == MVT::Other;
2785 unsigned iid =
2786 cast<ConstantSDNode>(N->getOperand(HasInputChain))->getZExtValue();
2787 if (iid < Intrinsic::num_intrinsics)
2788 Msg << "intrinsic %" << Intrinsic::getName((Intrinsic::ID)iid);
2789 else if (const TargetIntrinsicInfo *TII = TM.getIntrinsicInfo())
2790 Msg << "target intrinsic %" << TII->getName(iid);
2791 else
2792 Msg << "unknown intrinsic #" << iid;
2793 }
Dan Gohmane1f188f2009-10-29 22:30:23 +00002794 llvm_report_error(Msg.str());
2795}
2796
Devang Patel19974732007-05-03 01:11:54 +00002797char SelectionDAGISel::ID = 0;