blob: e62e5591b8d31411dd7deb64319a23cd03b88774 [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"
Jeff Cohen7e881032006-02-24 02:52:40 +000059#include <algorithm>
Chris Lattner1c08c712005-01-07 07:47:53 +000060using namespace llvm;
61
Chris Lattneread0d882008-06-17 06:09:18 +000062static cl::opt<bool>
Dan Gohman293d5f82008-09-09 22:06:46 +000063EnableFastISelVerbose("fast-isel-verbose", cl::Hidden,
Dan Gohmand659d502008-10-20 21:30:12 +000064 cl::desc("Enable verbose messages in the \"fast\" "
Dan Gohman293d5f82008-09-09 22:06:46 +000065 "instruction selector"));
66static cl::opt<bool>
Dan Gohman4344a5d2008-09-09 23:05:00 +000067EnableFastISelAbort("fast-isel-abort", cl::Hidden,
68 cl::desc("Enable abort calls when \"fast\" instruction fails"));
Dan Gohman8a110532008-09-05 22:59:21 +000069static cl::opt<bool>
Evan Chengdf8ed022009-11-09 06:49:37 +000070SchedLiveInCopies("schedule-livein-copies", cl::Hidden,
Dan Gohman8a110532008-09-05 22:59:21 +000071 cl::desc("Schedule copies of livein registers"),
72 cl::init(false));
Chris Lattneread0d882008-06-17 06:09:18 +000073
Chris Lattnerda8abb02005-09-01 18:44:10 +000074#ifndef NDEBUG
Chris Lattner7944d9d2005-01-12 03:41:21 +000075static cl::opt<bool>
Dan Gohman462dc7f2008-07-21 20:00:07 +000076ViewDAGCombine1("view-dag-combine1-dags", cl::Hidden,
77 cl::desc("Pop up a window to show dags before the first "
78 "dag combine pass"));
79static cl::opt<bool>
80ViewLegalizeTypesDAGs("view-legalize-types-dags", cl::Hidden,
81 cl::desc("Pop up a window to show dags before legalize types"));
82static cl::opt<bool>
83ViewLegalizeDAGs("view-legalize-dags", cl::Hidden,
84 cl::desc("Pop up a window to show dags before legalize"));
85static cl::opt<bool>
86ViewDAGCombine2("view-dag-combine2-dags", cl::Hidden,
87 cl::desc("Pop up a window to show dags before the second "
88 "dag combine pass"));
89static cl::opt<bool>
Duncan Sands25cf2272008-11-24 14:53:14 +000090ViewDAGCombineLT("view-dag-combine-lt-dags", cl::Hidden,
91 cl::desc("Pop up a window to show dags before the post legalize types"
92 " dag combine pass"));
93static cl::opt<bool>
Evan Chenga9c20912006-01-21 02:32:06 +000094ViewISelDAGs("view-isel-dags", cl::Hidden,
95 cl::desc("Pop up a window to show isel dags as they are selected"));
96static cl::opt<bool>
97ViewSchedDAGs("view-sched-dags", cl::Hidden,
98 cl::desc("Pop up a window to show sched dags as they are processed"));
Dan Gohman3e1a7ae2007-08-28 20:32:58 +000099static cl::opt<bool>
100ViewSUnitDAGs("view-sunit-dags", cl::Hidden,
Chris Lattner5bab7852008-01-25 17:24:52 +0000101 cl::desc("Pop up a window to show SUnit dags after they are processed"));
Chris Lattner7944d9d2005-01-12 03:41:21 +0000102#else
Dan Gohman462dc7f2008-07-21 20:00:07 +0000103static const bool ViewDAGCombine1 = false,
104 ViewLegalizeTypesDAGs = false, ViewLegalizeDAGs = false,
105 ViewDAGCombine2 = false,
Duncan Sands25cf2272008-11-24 14:53:14 +0000106 ViewDAGCombineLT = false,
Dan Gohman462dc7f2008-07-21 20:00:07 +0000107 ViewISelDAGs = false, ViewSchedDAGs = false,
108 ViewSUnitDAGs = false;
Chris Lattner7944d9d2005-01-12 03:41:21 +0000109#endif
110
Jim Laskeyeb577ba2006-08-02 12:30:23 +0000111//===---------------------------------------------------------------------===//
112///
113/// RegisterScheduler class - Track the registration of instruction schedulers.
114///
115//===---------------------------------------------------------------------===//
116MachinePassRegistry RegisterScheduler::Registry;
117
118//===---------------------------------------------------------------------===//
119///
120/// ISHeuristic command line option for instruction schedulers.
121///
122//===---------------------------------------------------------------------===//
Dan Gohman844731a2008-05-13 00:00:25 +0000123static cl::opt<RegisterScheduler::FunctionPassCtor, false,
124 RegisterPassParser<RegisterScheduler> >
125ISHeuristic("pre-RA-sched",
126 cl::init(&createDefaultScheduler),
127 cl::desc("Instruction schedulers available (before register"
128 " allocation):"));
Jim Laskey13ec7022006-08-01 14:21:23 +0000129
Dan Gohman844731a2008-05-13 00:00:25 +0000130static RegisterScheduler
Dan Gohmanb8cab922008-10-14 20:25:08 +0000131defaultListDAGScheduler("default", "Best scheduler for the target",
Dan Gohman844731a2008-05-13 00:00:25 +0000132 createDefaultScheduler);
Evan Cheng4ef10862006-01-23 07:01:07 +0000133
Chris Lattner1c08c712005-01-07 07:47:53 +0000134namespace llvm {
135 //===--------------------------------------------------------------------===//
Jim Laskey9373beb2006-08-01 19:14:14 +0000136 /// createDefaultScheduler - This creates an instruction scheduler appropriate
137 /// for the target.
Dan Gohman47ac0f02009-02-11 04:27:20 +0000138 ScheduleDAGSDNodes* createDefaultScheduler(SelectionDAGISel *IS,
Bill Wendling98a366d2009-04-29 23:29:43 +0000139 CodeGenOpt::Level OptLevel) {
Dan Gohmane9530ec2009-01-15 16:58:17 +0000140 const TargetLowering &TLI = IS->getTargetLowering();
141
Bill Wendling98a366d2009-04-29 23:29:43 +0000142 if (OptLevel == CodeGenOpt::None)
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000143 return createFastDAGScheduler(IS, OptLevel);
Dan Gohman9e76fea2008-11-20 03:11:19 +0000144 if (TLI.getSchedulingPreference() == TargetLowering::SchedulingForLatency)
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000145 return createTDListDAGScheduler(IS, OptLevel);
Dan Gohman9e76fea2008-11-20 03:11:19 +0000146 assert(TLI.getSchedulingPreference() ==
Bill Wendling187361b2010-01-23 10:26:57 +0000147 TargetLowering::SchedulingForRegPressure && "Unknown sched type!");
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000148 return createBURRListDAGScheduler(IS, OptLevel);
Jim Laskey9373beb2006-08-01 19:14:14 +0000149 }
Chris Lattner1c08c712005-01-07 07:47:53 +0000150}
151
Evan Chengff9b3732008-01-30 18:18:23 +0000152// EmitInstrWithCustomInserter - This method should be implemented by targets
Dan Gohman533297b2009-10-29 18:10:34 +0000153// that mark instructions with the 'usesCustomInserter' flag. These
Chris Lattner025c39b2005-08-26 20:54:47 +0000154// instructions are special in various ways, which require special support to
155// insert. The specified MachineInstr is created but not inserted into any
Dan Gohman533297b2009-10-29 18:10:34 +0000156// basic blocks, and this method is called to expand it into a sequence of
157// instructions, potentially also creating new basic blocks and control flow.
158// When new basic blocks are inserted and the edges from MBB to its successors
159// are modified, the method should insert pairs of <OldSucc, NewSucc> into the
160// DenseMap.
Evan Chengff9b3732008-01-30 18:18:23 +0000161MachineBasicBlock *TargetLowering::EmitInstrWithCustomInserter(MachineInstr *MI,
Evan Chengfb2e7522009-09-18 21:02:19 +0000162 MachineBasicBlock *MBB,
163 DenseMap<MachineBasicBlock*, MachineBasicBlock*> *EM) const {
Torok Edwinf3689232009-07-12 20:07:01 +0000164#ifndef NDEBUG
David Greene1a053232010-01-05 01:26:11 +0000165 dbgs() << "If a target marks an instruction with "
Dan Gohman533297b2009-10-29 18:10:34 +0000166 "'usesCustomInserter', it must implement "
Torok Edwinf3689232009-07-12 20:07:01 +0000167 "TargetLowering::EmitInstrWithCustomInserter!";
168#endif
Torok Edwinc23197a2009-07-14 16:55:14 +0000169 llvm_unreachable(0);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000170 return 0;
Chris Lattner025c39b2005-08-26 20:54:47 +0000171}
172
Dan Gohman8a110532008-09-05 22:59:21 +0000173/// EmitLiveInCopy - Emit a copy for a live in physical register. If the
174/// physical register has only a single copy use, then coalesced the copy
175/// if possible.
176static void EmitLiveInCopy(MachineBasicBlock *MBB,
177 MachineBasicBlock::iterator &InsertPos,
178 unsigned VirtReg, unsigned PhysReg,
179 const TargetRegisterClass *RC,
180 DenseMap<MachineInstr*, unsigned> &CopyRegMap,
181 const MachineRegisterInfo &MRI,
182 const TargetRegisterInfo &TRI,
183 const TargetInstrInfo &TII) {
184 unsigned NumUses = 0;
185 MachineInstr *UseMI = NULL;
186 for (MachineRegisterInfo::use_iterator UI = MRI.use_begin(VirtReg),
187 UE = MRI.use_end(); UI != UE; ++UI) {
188 UseMI = &*UI;
189 if (++NumUses > 1)
190 break;
191 }
192
193 // If the number of uses is not one, or the use is not a move instruction,
194 // don't coalesce. Also, only coalesce away a virtual register to virtual
195 // register copy.
196 bool Coalesced = false;
Evan Cheng04ee5a12009-01-20 19:12:24 +0000197 unsigned SrcReg, DstReg, SrcSubReg, DstSubReg;
Dan Gohman8a110532008-09-05 22:59:21 +0000198 if (NumUses == 1 &&
Evan Cheng04ee5a12009-01-20 19:12:24 +0000199 TII.isMoveInstr(*UseMI, SrcReg, DstReg, SrcSubReg, DstSubReg) &&
Dan Gohman8a110532008-09-05 22:59:21 +0000200 TargetRegisterInfo::isVirtualRegister(DstReg)) {
201 VirtReg = DstReg;
202 Coalesced = true;
203 }
204
205 // Now find an ideal location to insert the copy.
206 MachineBasicBlock::iterator Pos = InsertPos;
207 while (Pos != MBB->begin()) {
208 MachineInstr *PrevMI = prior(Pos);
209 DenseMap<MachineInstr*, unsigned>::iterator RI = CopyRegMap.find(PrevMI);
210 // copyRegToReg might emit multiple instructions to do a copy.
211 unsigned CopyDstReg = (RI == CopyRegMap.end()) ? 0 : RI->second;
212 if (CopyDstReg && !TRI.regsOverlap(CopyDstReg, PhysReg))
213 // This is what the BB looks like right now:
214 // r1024 = mov r0
215 // ...
216 // r1 = mov r1024
217 //
218 // We want to insert "r1025 = mov r1". Inserting this copy below the
219 // move to r1024 makes it impossible for that move to be coalesced.
220 //
221 // r1025 = mov r1
222 // r1024 = mov r0
223 // ...
224 // r1 = mov 1024
225 // r2 = mov 1025
226 break; // Woot! Found a good location.
227 --Pos;
228 }
229
David Goodwinf1daf7d2009-07-08 23:10:31 +0000230 bool Emitted = TII.copyRegToReg(*MBB, Pos, VirtReg, PhysReg, RC, RC);
231 assert(Emitted && "Unable to issue a live-in copy instruction!\n");
232 (void) Emitted;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000233
Zhongxing Xu931424a2009-10-16 05:42:28 +0000234 CopyRegMap.insert(std::make_pair(prior(Pos), VirtReg));
Dan Gohman8a110532008-09-05 22:59:21 +0000235 if (Coalesced) {
236 if (&*InsertPos == UseMI) ++InsertPos;
237 MBB->erase(UseMI);
238 }
239}
240
241/// EmitLiveInCopies - If this is the first basic block in the function,
242/// and if it has live ins that need to be copied into vregs, emit the
243/// copies into the block.
244static void EmitLiveInCopies(MachineBasicBlock *EntryMBB,
245 const MachineRegisterInfo &MRI,
246 const TargetRegisterInfo &TRI,
247 const TargetInstrInfo &TII) {
248 if (SchedLiveInCopies) {
249 // Emit the copies at a heuristically-determined location in the block.
250 DenseMap<MachineInstr*, unsigned> CopyRegMap;
251 MachineBasicBlock::iterator InsertPos = EntryMBB->begin();
252 for (MachineRegisterInfo::livein_iterator LI = MRI.livein_begin(),
253 E = MRI.livein_end(); LI != E; ++LI)
254 if (LI->second) {
255 const TargetRegisterClass *RC = MRI.getRegClass(LI->second);
256 EmitLiveInCopy(EntryMBB, InsertPos, LI->second, LI->first,
257 RC, CopyRegMap, MRI, TRI, TII);
258 }
259 } else {
260 // Emit the copies into the top of the block.
261 for (MachineRegisterInfo::livein_iterator LI = MRI.livein_begin(),
262 E = MRI.livein_end(); LI != E; ++LI)
263 if (LI->second) {
264 const TargetRegisterClass *RC = MRI.getRegClass(LI->second);
David Goodwinf1daf7d2009-07-08 23:10:31 +0000265 bool Emitted = TII.copyRegToReg(*EntryMBB, EntryMBB->begin(),
266 LI->second, LI->first, RC, RC);
267 assert(Emitted && "Unable to issue a live-in copy instruction!\n");
268 (void) Emitted;
Dan Gohman8a110532008-09-05 22:59:21 +0000269 }
270 }
271}
272
Chris Lattner7041ee32005-01-11 05:56:49 +0000273//===----------------------------------------------------------------------===//
274// SelectionDAGISel code
275//===----------------------------------------------------------------------===//
Chris Lattner1c08c712005-01-07 07:47:53 +0000276
Bill Wendling98a366d2009-04-29 23:29:43 +0000277SelectionDAGISel::SelectionDAGISel(TargetMachine &tm, CodeGenOpt::Level OL) :
Dan Gohmanad2afc22009-07-31 18:16:33 +0000278 MachineFunctionPass(&ID), TM(tm), TLI(*tm.getTargetLowering()),
Dan Gohman7c3234c2008-08-27 23:52:12 +0000279 FuncInfo(new FunctionLoweringInfo(TLI)),
280 CurDAG(new SelectionDAG(TLI, *FuncInfo)),
Dan Gohman2048b852009-11-23 18:04:58 +0000281 SDB(new SelectionDAGBuilder(*CurDAG, TLI, *FuncInfo, OL)),
Dan Gohman7c3234c2008-08-27 23:52:12 +0000282 GFI(),
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000283 OptLevel(OL),
Dan Gohman7c3234c2008-08-27 23:52:12 +0000284 DAGSize(0)
285{}
286
287SelectionDAGISel::~SelectionDAGISel() {
Dan Gohman2048b852009-11-23 18:04:58 +0000288 delete SDB;
Dan Gohman7c3234c2008-08-27 23:52:12 +0000289 delete CurDAG;
290 delete FuncInfo;
291}
292
Owen Andersone50ed302009-08-10 22:56:29 +0000293unsigned SelectionDAGISel::MakeReg(EVT VT) {
Chris Lattner84bc5422007-12-31 04:13:23 +0000294 return RegInfo->createVirtualRegister(TLI.getRegClassFor(VT));
Chris Lattner1c08c712005-01-07 07:47:53 +0000295}
296
Chris Lattner495a0b52005-08-17 06:37:43 +0000297void SelectionDAGISel::getAnalysisUsage(AnalysisUsage &AU) const {
Jim Laskeyc7c3f112006-10-16 20:52:31 +0000298 AU.addRequired<AliasAnalysis>();
Dan Gohmana3477fe2009-07-31 23:36:22 +0000299 AU.addPreserved<AliasAnalysis>();
Gordon Henriksen5eca0752008-08-17 18:44:35 +0000300 AU.addRequired<GCModuleInfo>();
Dan Gohmana3477fe2009-07-31 23:36:22 +0000301 AU.addPreserved<GCModuleInfo>();
Devang Patel6e7a1612009-01-09 19:11:50 +0000302 AU.addRequired<DwarfWriter>();
Dan Gohmana3477fe2009-07-31 23:36:22 +0000303 AU.addPreserved<DwarfWriter>();
Dan Gohmanad2afc22009-07-31 18:16:33 +0000304 MachineFunctionPass::getAnalysisUsage(AU);
Chris Lattner495a0b52005-08-17 06:37:43 +0000305}
Chris Lattner1c08c712005-01-07 07:47:53 +0000306
Dan Gohmanad2afc22009-07-31 18:16:33 +0000307bool SelectionDAGISel::runOnMachineFunction(MachineFunction &mf) {
308 Function &Fn = *mf.getFunction();
309
Dan Gohman4344a5d2008-09-09 23:05:00 +0000310 // Do some sanity-checking on the command-line options.
311 assert((!EnableFastISelVerbose || EnableFastISel) &&
312 "-fast-isel-verbose requires -fast-isel");
313 assert((!EnableFastISelAbort || EnableFastISel) &&
314 "-fast-isel-abort requires -fast-isel");
315
Dan Gohman5f43f922007-08-27 16:26:13 +0000316 // Get alias analysis for load/store combining.
317 AA = &getAnalysis<AliasAnalysis>();
318
Dan Gohmanad2afc22009-07-31 18:16:33 +0000319 MF = &mf;
Dan Gohman8a110532008-09-05 22:59:21 +0000320 const TargetInstrInfo &TII = *TM.getInstrInfo();
321 const TargetRegisterInfo &TRI = *TM.getRegisterInfo();
322
Dan Gohmanf7d6cd42009-08-01 03:51:09 +0000323 if (Fn.hasGC())
324 GFI = &getAnalysis<GCModuleInfo>().getFunctionInfo(Fn);
Gordon Henriksence224772008-01-07 01:30:38 +0000325 else
Gordon Henriksen5eca0752008-08-17 18:44:35 +0000326 GFI = 0;
Dan Gohman79ce2762009-01-15 19:20:50 +0000327 RegInfo = &MF->getRegInfo();
David Greene1a053232010-01-05 01:26:11 +0000328 DEBUG(dbgs() << "\n\n\n=== " << Fn.getName() << "\n");
Chris Lattner1c08c712005-01-07 07:47:53 +0000329
Duncan Sands1465d612009-01-28 13:14:17 +0000330 MachineModuleInfo *MMI = getAnalysisIfAvailable<MachineModuleInfo>();
331 DwarfWriter *DW = getAnalysisIfAvailable<DwarfWriter>();
Owen Anderson5dcaceb2009-07-09 18:44:09 +0000332 CurDAG->init(*MF, MMI, DW);
Dan Gohman6277eb22009-11-23 17:16:22 +0000333 FuncInfo->set(Fn, *MF, EnableFastISel);
Dan Gohman2048b852009-11-23 18:04:58 +0000334 SDB->init(GFI, *AA);
Chris Lattner1c08c712005-01-07 07:47:53 +0000335
Dale Johannesen1532f3d2008-04-02 00:25:04 +0000336 for (Function::iterator I = Fn.begin(), E = Fn.end(); I != E; ++I)
337 if (InvokeInst *Invoke = dyn_cast<InvokeInst>(I->getTerminator()))
338 // Mark landing pad.
Dan Gohman7c3234c2008-08-27 23:52:12 +0000339 FuncInfo->MBBMap[Invoke->getSuccessor(1)]->setIsLandingPad();
Duncan Sands9fac0b52007-06-06 10:05:18 +0000340
Dan Gohman79ce2762009-01-15 19:20:50 +0000341 SelectAllBasicBlocks(Fn, *MF, MMI, DW, TII);
Misha Brukmanedf128a2005-04-21 22:36:52 +0000342
Dan Gohman8a110532008-09-05 22:59:21 +0000343 // If the first basic block in the function has live ins that need to be
344 // copied into vregs, emit the copies into the top of the block before
345 // emitting the code for the block.
Dan Gohman79ce2762009-01-15 19:20:50 +0000346 EmitLiveInCopies(MF->begin(), *RegInfo, TRI, TII);
Dan Gohman8a110532008-09-05 22:59:21 +0000347
Evan Chengad2070c2007-02-10 02:43:39 +0000348 // Add function live-ins to entry block live-in set.
Dan Gohman8a110532008-09-05 22:59:21 +0000349 for (MachineRegisterInfo::livein_iterator I = RegInfo->livein_begin(),
350 E = RegInfo->livein_end(); I != E; ++I)
Dan Gohman79ce2762009-01-15 19:20:50 +0000351 MF->begin()->addLiveIn(I->first);
Evan Chengad2070c2007-02-10 02:43:39 +0000352
Duncan Sandsf4070822007-06-15 19:04:19 +0000353#ifndef NDEBUG
Dan Gohman7c3234c2008-08-27 23:52:12 +0000354 assert(FuncInfo->CatchInfoFound.size() == FuncInfo->CatchInfoLost.size() &&
Duncan Sandsf4070822007-06-15 19:04:19 +0000355 "Not all catch info was assigned to a landing pad!");
356#endif
357
Dan Gohman7c3234c2008-08-27 23:52:12 +0000358 FuncInfo->clear();
359
Chris Lattner1c08c712005-01-07 07:47:53 +0000360 return true;
361}
362
Dan Gohman07f111e2009-12-05 00:27:08 +0000363/// SetDebugLoc - Update MF's and SDB's DebugLocs if debug information is
364/// attached with this instruction.
Chris Lattner3990b122009-12-28 23:41:32 +0000365static void SetDebugLoc(unsigned MDDbgKind, Instruction *I,
366 SelectionDAGBuilder *SDB,
Chris Lattner0eb41982009-12-28 20:45:51 +0000367 FastISel *FastIS, MachineFunction *MF) {
368 if (isa<DbgInfoIntrinsic>(I)) return;
369
Chris Lattner3990b122009-12-28 23:41:32 +0000370 if (MDNode *Dbg = I->getMetadata(MDDbgKind)) {
Chris Lattner0eb41982009-12-28 20:45:51 +0000371 DILocation DILoc(Dbg);
372 DebugLoc Loc = ExtractDebugLocation(DILoc, MF->getDebugLocInfo());
Dan Gohman07f111e2009-12-05 00:27:08 +0000373
Chris Lattner0eb41982009-12-28 20:45:51 +0000374 SDB->setCurDebugLoc(Loc);
Dan Gohman07f111e2009-12-05 00:27:08 +0000375
Chris Lattner0eb41982009-12-28 20:45:51 +0000376 if (FastIS)
377 FastIS->setCurDebugLoc(Loc);
Dan Gohman07f111e2009-12-05 00:27:08 +0000378
Chris Lattner0eb41982009-12-28 20:45:51 +0000379 // If the function doesn't have a default debug location yet, set
380 // it. This is kind of a hack.
381 if (MF->getDefaultDebugLoc().isUnknown())
382 MF->setDefaultDebugLoc(Loc);
383 }
Dan Gohman07f111e2009-12-05 00:27:08 +0000384}
385
386/// ResetDebugLoc - Set MF's and SDB's DebugLocs to Unknown.
Chris Lattner3990b122009-12-28 23:41:32 +0000387static void ResetDebugLoc(SelectionDAGBuilder *SDB, FastISel *FastIS) {
Dan Gohman07f111e2009-12-05 00:27:08 +0000388 SDB->setCurDebugLoc(DebugLoc::getUnknownLoc());
389 if (FastIS)
Dan Gohman688fb802009-12-14 23:08:09 +0000390 FastIS->setCurDebugLoc(DebugLoc::getUnknownLoc());
Dan Gohman07f111e2009-12-05 00:27:08 +0000391}
392
Dan Gohmanf350b272008-08-23 02:25:05 +0000393void SelectionDAGISel::SelectBasicBlock(BasicBlock *LLVMBB,
394 BasicBlock::iterator Begin,
Dan Gohmanb4afb132009-11-20 02:51:26 +0000395 BasicBlock::iterator End,
396 bool &HadTailCall) {
Dan Gohman2048b852009-11-23 18:04:58 +0000397 SDB->setCurrentBasicBlock(BB);
Chris Lattner08113472009-12-29 09:01:33 +0000398 unsigned MDDbgKind = LLVMBB->getContext().getMDKindID("dbg");
Dan Gohmanf350b272008-08-23 02:25:05 +0000399
Dan Gohman98ca4f22009-08-05 01:29:28 +0000400 // Lower all of the non-terminator instructions. If a call is emitted
401 // as a tail call, cease emitting nodes for this block.
Dan Gohman2048b852009-11-23 18:04:58 +0000402 for (BasicBlock::iterator I = Begin; I != End && !SDB->HasTailCall; ++I) {
Chris Lattner3990b122009-12-28 23:41:32 +0000403 SetDebugLoc(MDDbgKind, I, SDB, 0, MF);
Dan Gohman07f111e2009-12-05 00:27:08 +0000404
405 if (!isa<TerminatorInst>(I)) {
Dan Gohman2048b852009-11-23 18:04:58 +0000406 SDB->visit(*I);
Dan Gohman07f111e2009-12-05 00:27:08 +0000407
408 // Set the current debug location back to "unknown" so that it doesn't
409 // spuriously apply to subsequent instructions.
410 ResetDebugLoc(SDB, 0);
411 }
Devang Patel123eaa72009-09-16 20:39:11 +0000412 }
Dan Gohmanf350b272008-08-23 02:25:05 +0000413
Dan Gohman2048b852009-11-23 18:04:58 +0000414 if (!SDB->HasTailCall) {
Dan Gohman98ca4f22009-08-05 01:29:28 +0000415 // Ensure that all instructions which are used outside of their defining
416 // blocks are available as virtual registers. Invoke is handled elsewhere.
417 for (BasicBlock::iterator I = Begin; I != End; ++I)
418 if (!isa<PHINode>(I) && !isa<InvokeInst>(I))
Dan Gohman2048b852009-11-23 18:04:58 +0000419 SDB->CopyToExportRegsIfNeeded(I);
Dan Gohmanf350b272008-08-23 02:25:05 +0000420
Dan Gohman98ca4f22009-08-05 01:29:28 +0000421 // Handle PHI nodes in successor blocks.
422 if (End == LLVMBB->end()) {
423 HandlePHINodesInSuccessorBlocks(LLVMBB);
Dan Gohman3df24e62008-09-03 23:12:08 +0000424
Dan Gohman98ca4f22009-08-05 01:29:28 +0000425 // Lower the terminator after the copies are emitted.
Chris Lattner3990b122009-12-28 23:41:32 +0000426 SetDebugLoc(MDDbgKind, LLVMBB->getTerminator(), SDB, 0, MF);
Dan Gohman2048b852009-11-23 18:04:58 +0000427 SDB->visit(*LLVMBB->getTerminator());
Dan Gohman07f111e2009-12-05 00:27:08 +0000428 ResetDebugLoc(SDB, 0);
Dan Gohman98ca4f22009-08-05 01:29:28 +0000429 }
Dan Gohman3df24e62008-09-03 23:12:08 +0000430 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000431
Chris Lattnera651cf62005-01-17 19:43:36 +0000432 // Make sure the root of the DAG is up-to-date.
Dan Gohman2048b852009-11-23 18:04:58 +0000433 CurDAG->setRoot(SDB->getControlRoot());
Arnold Schwaighoferc85e1712007-10-11 19:40:01 +0000434
Dan Gohmanf350b272008-08-23 02:25:05 +0000435 // Final step, emit the lowered DAG as machine code.
436 CodeGenAndEmitDAG();
Dan Gohman2048b852009-11-23 18:04:58 +0000437 HadTailCall = SDB->HasTailCall;
438 SDB->clear();
Chris Lattner1c08c712005-01-07 07:47:53 +0000439}
440
Evan Cheng54e146b2010-01-09 00:21:08 +0000441namespace {
442/// WorkListRemover - This class is a DAGUpdateListener that removes any deleted
443/// nodes from the worklist.
444class SDOPsWorkListRemover : public SelectionDAG::DAGUpdateListener {
445 SmallVector<SDNode*, 128> &Worklist;
446public:
447 SDOPsWorkListRemover(SmallVector<SDNode*, 128> &wl) : Worklist(wl) {}
448
449 virtual void NodeDeleted(SDNode *N, SDNode *E) {
450 Worklist.erase(std::remove(Worklist.begin(), Worklist.end(), N),
451 Worklist.end());
452 }
453
454 virtual void NodeUpdated(SDNode *N) {
455 // Ignore updates.
456 }
457};
458}
459
Evan Cheng046632f2010-02-10 02:17:34 +0000460/// TrivialTruncElim - Eliminate some trivial nops that can result from
461/// ShrinkDemandedOps: (trunc (ext n)) -> n.
462static bool TrivialTruncElim(SDValue Op,
463 TargetLowering::TargetLoweringOpt &TLO) {
464 SDValue N0 = Op.getOperand(0);
465 EVT VT = Op.getValueType();
466 if ((N0.getOpcode() == ISD::ZERO_EXTEND ||
467 N0.getOpcode() == ISD::SIGN_EXTEND ||
468 N0.getOpcode() == ISD::ANY_EXTEND) &&
469 N0.getOperand(0).getValueType() == VT) {
470 return TLO.CombineTo(Op, N0.getOperand(0));
471 }
472 return false;
473}
474
Evan Cheng54eb4c22010-01-06 19:43:21 +0000475/// ShrinkDemandedOps - A late transformation pass that shrink expressions
476/// using TargetLowering::TargetLoweringOpt::ShrinkDemandedOp. It converts
477/// x+y to (VT)((SmallVT)x+(SmallVT)y) if the casts are free.
Evan Chengd40d03e2010-01-06 19:38:29 +0000478void SelectionDAGISel::ShrinkDemandedOps() {
479 SmallVector<SDNode*, 128> Worklist;
480
481 // Add all the dag nodes to the worklist.
482 Worklist.reserve(CurDAG->allnodes_size());
483 for (SelectionDAG::allnodes_iterator I = CurDAG->allnodes_begin(),
484 E = CurDAG->allnodes_end(); I != E; ++I)
485 Worklist.push_back(I);
486
487 APInt Mask;
488 APInt KnownZero;
489 APInt KnownOne;
490
491 TargetLowering::TargetLoweringOpt TLO(*CurDAG, true);
492 while (!Worklist.empty()) {
Benjamin Kramer7b1e2a52010-01-07 17:27:56 +0000493 SDNode *N = Worklist.pop_back_val();
Evan Chengd40d03e2010-01-06 19:38:29 +0000494
495 if (N->use_empty() && N != CurDAG->getRoot().getNode()) {
Evan Cheng54e146b2010-01-09 00:21:08 +0000496 CurDAG->DeleteNode(N);
Evan Chengd40d03e2010-01-06 19:38:29 +0000497 continue;
498 }
499
500 // Run ShrinkDemandedOp on scalar binary operations.
501 if (N->getNumValues() == 1 &&
502 N->getValueType(0).isSimple() && N->getValueType(0).isInteger()) {
Evan Chengd40d03e2010-01-06 19:38:29 +0000503 unsigned BitWidth = N->getValueType(0).getScalarType().getSizeInBits();
504 APInt Demanded = APInt::getAllOnesValue(BitWidth);
505 APInt KnownZero, KnownOne;
506 if (TLI.SimplifyDemandedBits(SDValue(N, 0), Demanded,
Evan Cheng046632f2010-02-10 02:17:34 +0000507 KnownZero, KnownOne, TLO) ||
508 (N->getOpcode() == ISD::TRUNCATE &&
509 TrivialTruncElim(SDValue(N, 0), TLO))) {
Evan Chengd40d03e2010-01-06 19:38:29 +0000510 // Revisit the node.
511 Worklist.erase(std::remove(Worklist.begin(), Worklist.end(), N),
512 Worklist.end());
513 Worklist.push_back(N);
514
515 // Replace the old value with the new one.
516 DEBUG(errs() << "\nReplacing ";
517 TLO.Old.getNode()->dump(CurDAG);
518 errs() << "\nWith: ";
519 TLO.New.getNode()->dump(CurDAG);
520 errs() << '\n');
521
522 Worklist.push_back(TLO.New.getNode());
Evan Cheng54e146b2010-01-09 00:21:08 +0000523
524 SDOPsWorkListRemover DeadNodes(Worklist);
525 CurDAG->ReplaceAllUsesOfValueWith(TLO.Old, TLO.New, &DeadNodes);
Evan Chengd40d03e2010-01-06 19:38:29 +0000526
527 if (TLO.Old.getNode()->use_empty()) {
528 for (unsigned i = 0, e = TLO.Old.getNode()->getNumOperands();
529 i != e; ++i) {
530 SDNode *OpNode = TLO.Old.getNode()->getOperand(i).getNode();
531 if (OpNode->hasOneUse()) {
532 Worklist.erase(std::remove(Worklist.begin(), Worklist.end(),
Evan Cheng54e146b2010-01-09 00:21:08 +0000533 OpNode), Worklist.end());
534 Worklist.push_back(OpNode);
Evan Chengd40d03e2010-01-06 19:38:29 +0000535 }
536 }
537
538 Worklist.erase(std::remove(Worklist.begin(), Worklist.end(),
Evan Cheng54e146b2010-01-09 00:21:08 +0000539 TLO.Old.getNode()), Worklist.end());
Evan Chengd40d03e2010-01-06 19:38:29 +0000540 CurDAG->DeleteNode(TLO.Old.getNode());
541 }
542 }
543 }
544 }
545}
546
Dan Gohmanf350b272008-08-23 02:25:05 +0000547void SelectionDAGISel::ComputeLiveOutVRegInfo() {
Chris Lattneread0d882008-06-17 06:09:18 +0000548 SmallPtrSet<SDNode*, 128> VisitedNodes;
549 SmallVector<SDNode*, 128> Worklist;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000550
Gabor Greifba36cb52008-08-28 21:40:38 +0000551 Worklist.push_back(CurDAG->getRoot().getNode());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000552
Chris Lattneread0d882008-06-17 06:09:18 +0000553 APInt Mask;
554 APInt KnownZero;
555 APInt KnownOne;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000556
Benjamin Kramer7b1e2a52010-01-07 17:27:56 +0000557 do {
558 SDNode *N = Worklist.pop_back_val();
Daniel Dunbara279bc32009-09-20 02:20:51 +0000559
Chris Lattneread0d882008-06-17 06:09:18 +0000560 // If we've already seen this node, ignore it.
561 if (!VisitedNodes.insert(N))
562 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000563
Chris Lattneread0d882008-06-17 06:09:18 +0000564 // Otherwise, add all chain operands to the worklist.
565 for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i)
Owen Anderson825b72b2009-08-11 20:47:22 +0000566 if (N->getOperand(i).getValueType() == MVT::Other)
Gabor Greifba36cb52008-08-28 21:40:38 +0000567 Worklist.push_back(N->getOperand(i).getNode());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000568
Chris Lattneread0d882008-06-17 06:09:18 +0000569 // If this is a CopyToReg with a vreg dest, process it.
570 if (N->getOpcode() != ISD::CopyToReg)
571 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000572
Chris Lattneread0d882008-06-17 06:09:18 +0000573 unsigned DestReg = cast<RegisterSDNode>(N->getOperand(1))->getReg();
574 if (!TargetRegisterInfo::isVirtualRegister(DestReg))
575 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000576
Chris Lattneread0d882008-06-17 06:09:18 +0000577 // Ignore non-scalar or non-integer values.
Dan Gohman475871a2008-07-27 21:46:04 +0000578 SDValue Src = N->getOperand(2);
Owen Andersone50ed302009-08-10 22:56:29 +0000579 EVT SrcVT = Src.getValueType();
Chris Lattneread0d882008-06-17 06:09:18 +0000580 if (!SrcVT.isInteger() || SrcVT.isVector())
581 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000582
Dan Gohmanf350b272008-08-23 02:25:05 +0000583 unsigned NumSignBits = CurDAG->ComputeNumSignBits(Src);
Chris Lattneread0d882008-06-17 06:09:18 +0000584 Mask = APInt::getAllOnesValue(SrcVT.getSizeInBits());
Dan Gohmanf350b272008-08-23 02:25:05 +0000585 CurDAG->ComputeMaskedBits(Src, Mask, KnownZero, KnownOne);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000586
Chris Lattneread0d882008-06-17 06:09:18 +0000587 // Only install this information if it tells us something.
588 if (NumSignBits != 1 || KnownZero != 0 || KnownOne != 0) {
589 DestReg -= TargetRegisterInfo::FirstVirtualRegister;
Dan Gohmanf7d6cd42009-08-01 03:51:09 +0000590 if (DestReg >= FuncInfo->LiveOutRegInfo.size())
591 FuncInfo->LiveOutRegInfo.resize(DestReg+1);
592 FunctionLoweringInfo::LiveOutInfo &LOI =
593 FuncInfo->LiveOutRegInfo[DestReg];
Chris Lattneread0d882008-06-17 06:09:18 +0000594 LOI.NumSignBits = NumSignBits;
Dan Gohmana80efce2009-03-27 23:55:04 +0000595 LOI.KnownOne = KnownOne;
596 LOI.KnownZero = KnownZero;
Chris Lattneread0d882008-06-17 06:09:18 +0000597 }
Benjamin Kramer7b1e2a52010-01-07 17:27:56 +0000598 } while (!Worklist.empty());
Chris Lattneread0d882008-06-17 06:09:18 +0000599}
600
Dan Gohmanf350b272008-08-23 02:25:05 +0000601void SelectionDAGISel::CodeGenAndEmitDAG() {
Dan Gohman462dc7f2008-07-21 20:00:07 +0000602 std::string GroupName;
603 if (TimePassesIsEnabled)
604 GroupName = "Instruction Selection and Scheduling";
605 std::string BlockName;
606 if (ViewDAGCombine1 || ViewLegalizeTypesDAGs || ViewLegalizeDAGs ||
Duncan Sands25cf2272008-11-24 14:53:14 +0000607 ViewDAGCombine2 || ViewDAGCombineLT || ViewISelDAGs || ViewSchedDAGs ||
608 ViewSUnitDAGs)
Dan Gohmanf7d6cd42009-08-01 03:51:09 +0000609 BlockName = MF->getFunction()->getNameStr() + ":" +
Daniel Dunbarf6ccee52009-07-24 08:24:36 +0000610 BB->getBasicBlock()->getNameStr();
Dan Gohman462dc7f2008-07-21 20:00:07 +0000611
David Greene1a053232010-01-05 01:26:11 +0000612 DEBUG(dbgs() << "Initial selection DAG:\n");
Dan Gohmanf350b272008-08-23 02:25:05 +0000613 DEBUG(CurDAG->dump());
Dan Gohman462dc7f2008-07-21 20:00:07 +0000614
Dan Gohmanf350b272008-08-23 02:25:05 +0000615 if (ViewDAGCombine1) CurDAG->viewGraph("dag-combine1 input for " + BlockName);
Dan Gohman417e11b2007-10-08 15:12:17 +0000616
Chris Lattneraf21d552005-10-10 16:47:10 +0000617 // Run the DAG combiner in pre-legalize mode.
Evan Chengebffb662008-07-01 17:59:20 +0000618 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000619 NamedRegionTimer T("DAG Combining 1", GroupName);
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000620 CurDAG->Combine(Unrestricted, *AA, OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000621 } else {
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000622 CurDAG->Combine(Unrestricted, *AA, OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000623 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000624
David Greene1a053232010-01-05 01:26:11 +0000625 DEBUG(dbgs() << "Optimized lowered selection DAG:\n");
Dan Gohmanf350b272008-08-23 02:25:05 +0000626 DEBUG(CurDAG->dump());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000627
Chris Lattner1c08c712005-01-07 07:47:53 +0000628 // Second step, hack on the DAG until it only uses operations and types that
629 // the target supports.
Dan Gohman714efc62009-12-05 17:51:33 +0000630 if (ViewLegalizeTypesDAGs) CurDAG->viewGraph("legalize-types input for " +
631 BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000632
Dan Gohman714efc62009-12-05 17:51:33 +0000633 bool Changed;
634 if (TimePassesIsEnabled) {
635 NamedRegionTimer T("Type Legalization", GroupName);
636 Changed = CurDAG->LegalizeTypes();
637 } else {
638 Changed = CurDAG->LegalizeTypes();
639 }
640
David Greene1a053232010-01-05 01:26:11 +0000641 DEBUG(dbgs() << "Type-legalized selection DAG:\n");
Dan Gohman714efc62009-12-05 17:51:33 +0000642 DEBUG(CurDAG->dump());
643
644 if (Changed) {
645 if (ViewDAGCombineLT)
646 CurDAG->viewGraph("dag-combine-lt input for " + BlockName);
647
648 // Run the DAG combiner in post-type-legalize mode.
Dan Gohman462dc7f2008-07-21 20:00:07 +0000649 if (TimePassesIsEnabled) {
Dan Gohman714efc62009-12-05 17:51:33 +0000650 NamedRegionTimer T("DAG Combining after legalize types", GroupName);
651 CurDAG->Combine(NoIllegalTypes, *AA, OptLevel);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000652 } else {
Dan Gohman714efc62009-12-05 17:51:33 +0000653 CurDAG->Combine(NoIllegalTypes, *AA, OptLevel);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000654 }
655
David Greene1a053232010-01-05 01:26:11 +0000656 DEBUG(dbgs() << "Optimized type-legalized selection DAG:\n");
Dan Gohmanf350b272008-08-23 02:25:05 +0000657 DEBUG(CurDAG->dump());
Dan Gohman714efc62009-12-05 17:51:33 +0000658 }
Dan Gohman462dc7f2008-07-21 20:00:07 +0000659
Dan Gohman714efc62009-12-05 17:51:33 +0000660 if (TimePassesIsEnabled) {
661 NamedRegionTimer T("Vector Legalization", GroupName);
662 Changed = CurDAG->LegalizeVectors();
663 } else {
664 Changed = CurDAG->LegalizeVectors();
665 }
Duncan Sands25cf2272008-11-24 14:53:14 +0000666
Dan Gohman714efc62009-12-05 17:51:33 +0000667 if (Changed) {
Eli Friedman5c22c802009-05-23 12:35:30 +0000668 if (TimePassesIsEnabled) {
Dan Gohman714efc62009-12-05 17:51:33 +0000669 NamedRegionTimer T("Type Legalization 2", GroupName);
Bill Wendling98820072009-12-28 01:51:30 +0000670 CurDAG->LegalizeTypes();
Eli Friedman5c22c802009-05-23 12:35:30 +0000671 } else {
Bill Wendling98820072009-12-28 01:51:30 +0000672 CurDAG->LegalizeTypes();
Eli Friedman5c22c802009-05-23 12:35:30 +0000673 }
674
Dan Gohman714efc62009-12-05 17:51:33 +0000675 if (ViewDAGCombineLT)
676 CurDAG->viewGraph("dag-combine-lv input for " + BlockName);
Eli Friedman5c22c802009-05-23 12:35:30 +0000677
Dan Gohman714efc62009-12-05 17:51:33 +0000678 // Run the DAG combiner in post-type-legalize mode.
679 if (TimePassesIsEnabled) {
680 NamedRegionTimer T("DAG Combining after legalize vectors", GroupName);
681 CurDAG->Combine(NoIllegalOperations, *AA, OptLevel);
682 } else {
683 CurDAG->Combine(NoIllegalOperations, *AA, OptLevel);
Eli Friedman5c22c802009-05-23 12:35:30 +0000684 }
Dan Gohman714efc62009-12-05 17:51:33 +0000685
David Greene1a053232010-01-05 01:26:11 +0000686 DEBUG(dbgs() << "Optimized vector-legalized selection DAG:\n");
Dan Gohman714efc62009-12-05 17:51:33 +0000687 DEBUG(CurDAG->dump());
Chris Lattner70587ea2008-07-10 23:37:50 +0000688 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000689
Dan Gohmanf350b272008-08-23 02:25:05 +0000690 if (ViewLegalizeDAGs) CurDAG->viewGraph("legalize input for " + BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000691
Evan Chengebffb662008-07-01 17:59:20 +0000692 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000693 NamedRegionTimer T("DAG Legalization", GroupName);
Dan Gohman714efc62009-12-05 17:51:33 +0000694 CurDAG->Legalize(OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000695 } else {
Dan Gohman714efc62009-12-05 17:51:33 +0000696 CurDAG->Legalize(OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000697 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000698
David Greene1a053232010-01-05 01:26:11 +0000699 DEBUG(dbgs() << "Legalized selection DAG:\n");
Dan Gohmanf350b272008-08-23 02:25:05 +0000700 DEBUG(CurDAG->dump());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000701
Dan Gohmanf350b272008-08-23 02:25:05 +0000702 if (ViewDAGCombine2) CurDAG->viewGraph("dag-combine2 input for " + BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000703
Chris Lattneraf21d552005-10-10 16:47:10 +0000704 // Run the DAG combiner in post-legalize mode.
Evan Chengebffb662008-07-01 17:59:20 +0000705 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000706 NamedRegionTimer T("DAG Combining 2", GroupName);
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000707 CurDAG->Combine(NoIllegalOperations, *AA, OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000708 } else {
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000709 CurDAG->Combine(NoIllegalOperations, *AA, OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000710 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000711
David Greene1a053232010-01-05 01:26:11 +0000712 DEBUG(dbgs() << "Optimized legalized selection DAG:\n");
Dan Gohmanf350b272008-08-23 02:25:05 +0000713 DEBUG(CurDAG->dump());
Dan Gohman417e11b2007-10-08 15:12:17 +0000714
Dan Gohmanf350b272008-08-23 02:25:05 +0000715 if (ViewISelDAGs) CurDAG->viewGraph("isel input for " + BlockName);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000716
Evan Chengd40d03e2010-01-06 19:38:29 +0000717 if (OptLevel != CodeGenOpt::None) {
718 ShrinkDemandedOps();
Dan Gohmanf350b272008-08-23 02:25:05 +0000719 ComputeLiveOutVRegInfo();
Evan Chengd40d03e2010-01-06 19:38:29 +0000720 }
Evan Cheng552c4a82006-04-28 02:09:19 +0000721
Chris Lattnera33ef482005-03-30 01:10:47 +0000722 // Third, instruction select all of the operations to machine code, adding the
723 // code to the MachineBasicBlock.
Evan Chengebffb662008-07-01 17:59:20 +0000724 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000725 NamedRegionTimer T("Instruction Selection", GroupName);
Dan Gohmanf350b272008-08-23 02:25:05 +0000726 InstructionSelect();
Evan Chengebffb662008-07-01 17:59:20 +0000727 } else {
Dan Gohmanf350b272008-08-23 02:25:05 +0000728 InstructionSelect();
Evan Chengebffb662008-07-01 17:59:20 +0000729 }
Evan Chengdb8d56b2008-06-30 20:45:06 +0000730
David Greene1a053232010-01-05 01:26:11 +0000731 DEBUG(dbgs() << "Selected selection DAG:\n");
Dan Gohmanf350b272008-08-23 02:25:05 +0000732 DEBUG(CurDAG->dump());
Dan Gohman462dc7f2008-07-21 20:00:07 +0000733
Dan Gohmanf350b272008-08-23 02:25:05 +0000734 if (ViewSchedDAGs) CurDAG->viewGraph("scheduler input for " + BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000735
Dan Gohman5e843682008-07-14 18:19:29 +0000736 // Schedule machine code.
Dan Gohman47ac0f02009-02-11 04:27:20 +0000737 ScheduleDAGSDNodes *Scheduler = CreateScheduler();
Dan Gohman5e843682008-07-14 18:19:29 +0000738 if (TimePassesIsEnabled) {
739 NamedRegionTimer T("Instruction Scheduling", GroupName);
Dan Gohman47ac0f02009-02-11 04:27:20 +0000740 Scheduler->Run(CurDAG, BB, BB->end());
Dan Gohman5e843682008-07-14 18:19:29 +0000741 } else {
Dan Gohman47ac0f02009-02-11 04:27:20 +0000742 Scheduler->Run(CurDAG, BB, BB->end());
Dan Gohman5e843682008-07-14 18:19:29 +0000743 }
744
Dan Gohman462dc7f2008-07-21 20:00:07 +0000745 if (ViewSUnitDAGs) Scheduler->viewGraph();
746
Daniel Dunbara279bc32009-09-20 02:20:51 +0000747 // Emit machine code to BB. This can change 'BB' to the last block being
Evan Chengdb8d56b2008-06-30 20:45:06 +0000748 // inserted into.
Evan Chengebffb662008-07-01 17:59:20 +0000749 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000750 NamedRegionTimer T("Instruction Creation", GroupName);
Dan Gohman2048b852009-11-23 18:04:58 +0000751 BB = Scheduler->EmitSchedule(&SDB->EdgeMapping);
Evan Chengebffb662008-07-01 17:59:20 +0000752 } else {
Dan Gohman2048b852009-11-23 18:04:58 +0000753 BB = Scheduler->EmitSchedule(&SDB->EdgeMapping);
Dan Gohman5e843682008-07-14 18:19:29 +0000754 }
755
756 // Free the scheduler state.
757 if (TimePassesIsEnabled) {
758 NamedRegionTimer T("Instruction Scheduling Cleanup", GroupName);
759 delete Scheduler;
760 } else {
761 delete Scheduler;
Evan Chengebffb662008-07-01 17:59:20 +0000762 }
Evan Chengdb8d56b2008-06-30 20:45:06 +0000763
David Greene1a053232010-01-05 01:26:11 +0000764 DEBUG(dbgs() << "Selected machine code:\n");
Chris Lattner1c08c712005-01-07 07:47:53 +0000765 DEBUG(BB->dump());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000766}
Chris Lattner1c08c712005-01-07 07:47:53 +0000767
Dan Gohman79ce2762009-01-15 19:20:50 +0000768void SelectionDAGISel::SelectAllBasicBlocks(Function &Fn,
769 MachineFunction &MF,
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000770 MachineModuleInfo *MMI,
Devang Patel83489bb2009-01-13 00:35:13 +0000771 DwarfWriter *DW,
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000772 const TargetInstrInfo &TII) {
Dan Gohmana43abd12008-09-29 21:55:50 +0000773 // Initialize the Fast-ISel state, if needed.
774 FastISel *FastIS = 0;
775 if (EnableFastISel)
Dan Gohman79ce2762009-01-15 19:20:50 +0000776 FastIS = TLI.createFastISel(MF, MMI, DW,
Dan Gohmana43abd12008-09-29 21:55:50 +0000777 FuncInfo->ValueMap,
778 FuncInfo->MBBMap,
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000779 FuncInfo->StaticAllocaMap
780#ifndef NDEBUG
781 , FuncInfo->CatchInfoLost
782#endif
783 );
Dan Gohmana43abd12008-09-29 21:55:50 +0000784
Chris Lattner08113472009-12-29 09:01:33 +0000785 unsigned MDDbgKind = Fn.getContext().getMDKindID("dbg");
Devang Patel123eaa72009-09-16 20:39:11 +0000786
Dan Gohmana43abd12008-09-29 21:55:50 +0000787 // Iterate over all basic blocks in the function.
Evan Cheng39fd6e82008-08-07 00:43:25 +0000788 for (Function::iterator I = Fn.begin(), E = Fn.end(); I != E; ++I) {
789 BasicBlock *LLVMBB = &*I;
Dan Gohman7c3234c2008-08-27 23:52:12 +0000790 BB = FuncInfo->MBBMap[LLVMBB];
Dan Gohmanf350b272008-08-23 02:25:05 +0000791
Dan Gohman3df24e62008-09-03 23:12:08 +0000792 BasicBlock::iterator const Begin = LLVMBB->begin();
793 BasicBlock::iterator const End = LLVMBB->end();
Evan Cheng9f118502008-09-08 16:01:27 +0000794 BasicBlock::iterator BI = Begin;
Dan Gohman5edd3612008-08-28 20:28:56 +0000795
796 // Lower any arguments needed in this block if this is the entry block.
Dan Gohman33134c42008-09-25 17:05:24 +0000797 bool SuppressFastISel = false;
798 if (LLVMBB == &Fn.getEntryBlock()) {
Dan Gohman5edd3612008-08-28 20:28:56 +0000799 LowerArguments(LLVMBB);
Dan Gohmanf350b272008-08-23 02:25:05 +0000800
Dan Gohman33134c42008-09-25 17:05:24 +0000801 // If any of the arguments has the byval attribute, forgo
802 // fast-isel in the entry block.
Dan Gohmana43abd12008-09-29 21:55:50 +0000803 if (FastIS) {
Dan Gohman33134c42008-09-25 17:05:24 +0000804 unsigned j = 1;
805 for (Function::arg_iterator I = Fn.arg_begin(), E = Fn.arg_end();
806 I != E; ++I, ++j)
Devang Patel05988662008-09-25 21:00:45 +0000807 if (Fn.paramHasAttr(j, Attribute::ByVal)) {
Dan Gohman77ca41e2008-09-25 17:21:42 +0000808 if (EnableFastISelVerbose || EnableFastISelAbort)
David Greene1a053232010-01-05 01:26:11 +0000809 dbgs() << "FastISel skips entry block due to byval argument\n";
Dan Gohman33134c42008-09-25 17:05:24 +0000810 SuppressFastISel = true;
811 break;
812 }
813 }
814 }
815
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000816 if (MMI && BB->isLandingPad()) {
817 // Add a label to mark the beginning of the landing pad. Deletion of the
818 // landing pad can thus be detected via the MachineModuleInfo.
819 unsigned LabelID = MMI->addLandingPad(BB);
820
Chris Lattner518bb532010-02-09 19:54:29 +0000821 const TargetInstrDesc &II = TII.get(TargetOpcode::EH_LABEL);
Dan Gohman2048b852009-11-23 18:04:58 +0000822 BuildMI(BB, SDB->getCurDebugLoc(), II).addImm(LabelID);
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000823
824 // Mark exception register as live in.
825 unsigned Reg = TLI.getExceptionAddressRegister();
826 if (Reg) BB->addLiveIn(Reg);
827
828 // Mark exception selector register as live in.
829 Reg = TLI.getExceptionSelectorRegister();
830 if (Reg) BB->addLiveIn(Reg);
831
832 // FIXME: Hack around an exception handling flaw (PR1508): the personality
833 // function and list of typeids logically belong to the invoke (or, if you
834 // like, the basic block containing the invoke), and need to be associated
835 // with it in the dwarf exception handling tables. Currently however the
836 // information is provided by an intrinsic (eh.selector) that can be moved
837 // to unexpected places by the optimizers: if the unwind edge is critical,
838 // then breaking it can result in the intrinsics being in the successor of
Jim Grosbachf4549b02010-01-15 00:36:15 +0000839 // the landing pad, not the landing pad itself. This results
840 // in exceptions not being caught because no typeids are associated with
841 // the invoke. This may not be the only way things can go wrong, but it
842 // is the only way we try to work around for the moment.
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000843 BranchInst *Br = dyn_cast<BranchInst>(LLVMBB->getTerminator());
844
845 if (Br && Br->isUnconditional()) { // Critical edge?
846 BasicBlock::iterator I, E;
847 for (I = LLVMBB->begin(), E = --LLVMBB->end(); I != E; ++I)
848 if (isa<EHSelectorInst>(I))
849 break;
850
851 if (I == E)
852 // No catch info found - try to extract some from the successor.
Dan Gohman5fca8b12009-11-23 18:12:11 +0000853 CopyCatchInfo(Br->getSuccessor(0), LLVMBB, MMI, *FuncInfo);
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000854 }
855 }
856
Dan Gohmanf350b272008-08-23 02:25:05 +0000857 // Before doing SelectionDAG ISel, see if FastISel has been requested.
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000858 if (FastIS && !SuppressFastISel) {
Dan Gohmana43abd12008-09-29 21:55:50 +0000859 // Emit code for any incoming arguments. This must happen before
860 // beginning FastISel on the entry block.
861 if (LLVMBB == &Fn.getEntryBlock()) {
Dan Gohman2048b852009-11-23 18:04:58 +0000862 CurDAG->setRoot(SDB->getControlRoot());
Dan Gohmana43abd12008-09-29 21:55:50 +0000863 CodeGenAndEmitDAG();
Dan Gohman2048b852009-11-23 18:04:58 +0000864 SDB->clear();
Dan Gohmana43abd12008-09-29 21:55:50 +0000865 }
Dan Gohman241f4642008-10-04 00:56:36 +0000866 FastIS->startNewBlock(BB);
Dan Gohmana43abd12008-09-29 21:55:50 +0000867 // Do FastISel on as many instructions as possible.
868 for (; BI != End; ++BI) {
869 // Just before the terminator instruction, insert instructions to
870 // feed PHI nodes in successor blocks.
871 if (isa<TerminatorInst>(BI))
872 if (!HandlePHINodesInSuccessorBlocksFast(LLVMBB, FastIS)) {
Dan Gohman07f111e2009-12-05 00:27:08 +0000873 ResetDebugLoc(SDB, FastIS);
Dan Gohman4344a5d2008-09-09 23:05:00 +0000874 if (EnableFastISelVerbose || EnableFastISelAbort) {
David Greene1a053232010-01-05 01:26:11 +0000875 dbgs() << "FastISel miss: ";
Dan Gohman293d5f82008-09-09 22:06:46 +0000876 BI->dump();
877 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000878 assert(!EnableFastISelAbort &&
Torok Edwinf3689232009-07-12 20:07:01 +0000879 "FastISel didn't handle a PHI in a successor");
Dan Gohmana43abd12008-09-29 21:55:50 +0000880 break;
Dan Gohmanf350b272008-08-23 02:25:05 +0000881 }
Dan Gohmana43abd12008-09-29 21:55:50 +0000882
Chris Lattner3990b122009-12-28 23:41:32 +0000883 SetDebugLoc(MDDbgKind, BI, SDB, FastIS, &MF);
Dan Gohman381ca552009-12-05 01:29:04 +0000884
Dan Gohman21c14e32010-01-12 04:32:35 +0000885 // Try to select the instruction with FastISel.
Dan Gohman07f111e2009-12-05 00:27:08 +0000886 if (FastIS->SelectInstruction(BI)) {
887 ResetDebugLoc(SDB, FastIS);
Dan Gohmana43abd12008-09-29 21:55:50 +0000888 continue;
Dan Gohman07f111e2009-12-05 00:27:08 +0000889 }
Dan Gohmana43abd12008-09-29 21:55:50 +0000890
Dan Gohman07f111e2009-12-05 00:27:08 +0000891 // Clear out the debug location so that it doesn't carry over to
892 // unrelated instructions.
893 ResetDebugLoc(SDB, FastIS);
Dan Gohmana43abd12008-09-29 21:55:50 +0000894
895 // Then handle certain instructions as single-LLVM-Instruction blocks.
896 if (isa<CallInst>(BI)) {
897 if (EnableFastISelVerbose || EnableFastISelAbort) {
David Greene1a053232010-01-05 01:26:11 +0000898 dbgs() << "FastISel missed call: ";
Dan Gohmana43abd12008-09-29 21:55:50 +0000899 BI->dump();
900 }
901
Benjamin Kramerf0127052010-01-05 13:12:22 +0000902 if (!BI->getType()->isVoidTy()) {
Dan Gohmana43abd12008-09-29 21:55:50 +0000903 unsigned &R = FuncInfo->ValueMap[BI];
904 if (!R)
905 R = FuncInfo->CreateRegForValue(BI);
906 }
907
Dan Gohmanb4afb132009-11-20 02:51:26 +0000908 bool HadTailCall = false;
Chris Lattner7896c9f2009-12-03 00:50:42 +0000909 SelectBasicBlock(LLVMBB, BI, llvm::next(BI), HadTailCall);
Dan Gohmanb4afb132009-11-20 02:51:26 +0000910
911 // If the call was emitted as a tail call, we're done with the block.
912 if (HadTailCall) {
913 BI = End;
914 break;
915 }
916
Dan Gohman241f4642008-10-04 00:56:36 +0000917 // If the instruction was codegen'd with multiple blocks,
918 // inform the FastISel object where to resume inserting.
919 FastIS->setCurrentBlock(BB);
Dan Gohmana43abd12008-09-29 21:55:50 +0000920 continue;
Dan Gohmanf350b272008-08-23 02:25:05 +0000921 }
Dan Gohmana43abd12008-09-29 21:55:50 +0000922
923 // Otherwise, give up on FastISel for the rest of the block.
924 // For now, be a little lenient about non-branch terminators.
925 if (!isa<TerminatorInst>(BI) || isa<BranchInst>(BI)) {
926 if (EnableFastISelVerbose || EnableFastISelAbort) {
David Greene1a053232010-01-05 01:26:11 +0000927 dbgs() << "FastISel miss: ";
Dan Gohmana43abd12008-09-29 21:55:50 +0000928 BI->dump();
929 }
930 if (EnableFastISelAbort)
931 // The "fast" selector couldn't handle something and bailed.
932 // For the purpose of debugging, just abort.
Torok Edwinc23197a2009-07-14 16:55:14 +0000933 llvm_unreachable("FastISel didn't select the entire block");
Dan Gohmana43abd12008-09-29 21:55:50 +0000934 }
935 break;
Dan Gohmanf350b272008-08-23 02:25:05 +0000936 }
937 }
938
Dan Gohmand2ff6472008-09-02 20:17:56 +0000939 // Run SelectionDAG instruction selection on the remainder of the block
940 // not handled by FastISel. If FastISel is not run, this is the entire
Dan Gohman3df24e62008-09-03 23:12:08 +0000941 // block.
Devang Patel390f3ac2009-04-16 01:33:10 +0000942 if (BI != End) {
Dan Gohmanb4afb132009-11-20 02:51:26 +0000943 bool HadTailCall;
944 SelectBasicBlock(LLVMBB, BI, End, HadTailCall);
Devang Patel390f3ac2009-04-16 01:33:10 +0000945 }
Dan Gohmanf350b272008-08-23 02:25:05 +0000946
Dan Gohman7c3234c2008-08-27 23:52:12 +0000947 FinishBasicBlock();
Evan Cheng39fd6e82008-08-07 00:43:25 +0000948 }
Dan Gohmana43abd12008-09-29 21:55:50 +0000949
950 delete FastIS;
Dan Gohman0e5f1302008-07-07 23:02:41 +0000951}
952
Dan Gohmanfed90b62008-07-28 21:51:04 +0000953void
Dan Gohman7c3234c2008-08-27 23:52:12 +0000954SelectionDAGISel::FinishBasicBlock() {
Dan Gohmanf350b272008-08-23 02:25:05 +0000955
David Greene1a053232010-01-05 01:26:11 +0000956 DEBUG(dbgs() << "Target-post-processed machine code:\n");
Dan Gohmanf350b272008-08-23 02:25:05 +0000957 DEBUG(BB->dump());
Nate Begemanf15485a2006-03-27 01:32:24 +0000958
David Greene1a053232010-01-05 01:26:11 +0000959 DEBUG(dbgs() << "Total amount of phi nodes to update: "
Dan Gohman2048b852009-11-23 18:04:58 +0000960 << SDB->PHINodesToUpdate.size() << "\n");
961 DEBUG(for (unsigned i = 0, e = SDB->PHINodesToUpdate.size(); i != e; ++i)
David Greene1a053232010-01-05 01:26:11 +0000962 dbgs() << "Node " << i << " : ("
Dan Gohman2048b852009-11-23 18:04:58 +0000963 << SDB->PHINodesToUpdate[i].first
964 << ", " << SDB->PHINodesToUpdate[i].second << ")\n");
Daniel Dunbara279bc32009-09-20 02:20:51 +0000965
Chris Lattnera33ef482005-03-30 01:10:47 +0000966 // Next, now that we know what the last MBB the LLVM BB expanded is, update
Chris Lattner1c08c712005-01-07 07:47:53 +0000967 // PHI nodes in successors.
Dan Gohman2048b852009-11-23 18:04:58 +0000968 if (SDB->SwitchCases.empty() &&
969 SDB->JTCases.empty() &&
970 SDB->BitTestCases.empty()) {
971 for (unsigned i = 0, e = SDB->PHINodesToUpdate.size(); i != e; ++i) {
972 MachineInstr *PHI = SDB->PHINodesToUpdate[i].first;
Chris Lattner518bb532010-02-09 19:54:29 +0000973 assert(PHI->isPHI() &&
Nate Begemanf15485a2006-03-27 01:32:24 +0000974 "This is not a machine PHI node that we are updating!");
Dan Gohman2048b852009-11-23 18:04:58 +0000975 PHI->addOperand(MachineOperand::CreateReg(SDB->PHINodesToUpdate[i].second,
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000976 false));
977 PHI->addOperand(MachineOperand::CreateMBB(BB));
Nate Begemanf15485a2006-03-27 01:32:24 +0000978 }
Dan Gohman2048b852009-11-23 18:04:58 +0000979 SDB->PHINodesToUpdate.clear();
Nate Begemanf15485a2006-03-27 01:32:24 +0000980 return;
Chris Lattner1c08c712005-01-07 07:47:53 +0000981 }
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000982
Dan Gohman2048b852009-11-23 18:04:58 +0000983 for (unsigned i = 0, e = SDB->BitTestCases.size(); i != e; ++i) {
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000984 // Lower header first, if it wasn't already lowered
Dan Gohman2048b852009-11-23 18:04:58 +0000985 if (!SDB->BitTestCases[i].Emitted) {
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000986 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman2048b852009-11-23 18:04:58 +0000987 BB = SDB->BitTestCases[i].Parent;
988 SDB->setCurrentBasicBlock(BB);
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000989 // Emit the code
Dan Gohman2048b852009-11-23 18:04:58 +0000990 SDB->visitBitTestHeader(SDB->BitTestCases[i]);
991 CurDAG->setRoot(SDB->getRoot());
Dan Gohmanf350b272008-08-23 02:25:05 +0000992 CodeGenAndEmitDAG();
Dan Gohman2048b852009-11-23 18:04:58 +0000993 SDB->clear();
Daniel Dunbara279bc32009-09-20 02:20:51 +0000994 }
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000995
Dan Gohman2048b852009-11-23 18:04:58 +0000996 for (unsigned j = 0, ej = SDB->BitTestCases[i].Cases.size(); j != ej; ++j) {
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000997 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman2048b852009-11-23 18:04:58 +0000998 BB = SDB->BitTestCases[i].Cases[j].ThisBB;
999 SDB->setCurrentBasicBlock(BB);
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001000 // Emit the code
1001 if (j+1 != ej)
Dan Gohman2048b852009-11-23 18:04:58 +00001002 SDB->visitBitTestCase(SDB->BitTestCases[i].Cases[j+1].ThisBB,
1003 SDB->BitTestCases[i].Reg,
1004 SDB->BitTestCases[i].Cases[j]);
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001005 else
Dan Gohman2048b852009-11-23 18:04:58 +00001006 SDB->visitBitTestCase(SDB->BitTestCases[i].Default,
1007 SDB->BitTestCases[i].Reg,
1008 SDB->BitTestCases[i].Cases[j]);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001009
1010
Dan Gohman2048b852009-11-23 18:04:58 +00001011 CurDAG->setRoot(SDB->getRoot());
Dan Gohmanf350b272008-08-23 02:25:05 +00001012 CodeGenAndEmitDAG();
Dan Gohman2048b852009-11-23 18:04:58 +00001013 SDB->clear();
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001014 }
1015
1016 // Update PHI Nodes
Dan Gohman2048b852009-11-23 18:04:58 +00001017 for (unsigned pi = 0, pe = SDB->PHINodesToUpdate.size(); pi != pe; ++pi) {
1018 MachineInstr *PHI = SDB->PHINodesToUpdate[pi].first;
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001019 MachineBasicBlock *PHIBB = PHI->getParent();
Chris Lattner518bb532010-02-09 19:54:29 +00001020 assert(PHI->isPHI() &&
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001021 "This is not a machine PHI node that we are updating!");
1022 // This is "default" BB. We have two jumps to it. From "header" BB and
1023 // from last "case" BB.
Dan Gohman2048b852009-11-23 18:04:58 +00001024 if (PHIBB == SDB->BitTestCases[i].Default) {
Jim Grosbachf4549b02010-01-15 00:36:15 +00001025 PHI->addOperand(MachineOperand::
1026 CreateReg(SDB->PHINodesToUpdate[pi].second, false));
Dan Gohman2048b852009-11-23 18:04:58 +00001027 PHI->addOperand(MachineOperand::CreateMBB(SDB->BitTestCases[i].Parent));
Jim Grosbachf4549b02010-01-15 00:36:15 +00001028 PHI->addOperand(MachineOperand::
1029 CreateReg(SDB->PHINodesToUpdate[pi].second, false));
Dan Gohman2048b852009-11-23 18:04:58 +00001030 PHI->addOperand(MachineOperand::CreateMBB(SDB->BitTestCases[i].Cases.
Chris Lattner9ce2e9d2007-12-30 00:57:42 +00001031 back().ThisBB));
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001032 }
1033 // One of "cases" BB.
Dan Gohman2048b852009-11-23 18:04:58 +00001034 for (unsigned j = 0, ej = SDB->BitTestCases[i].Cases.size();
Dan Gohman7c3234c2008-08-27 23:52:12 +00001035 j != ej; ++j) {
Dan Gohman2048b852009-11-23 18:04:58 +00001036 MachineBasicBlock* cBB = SDB->BitTestCases[i].Cases[j].ThisBB;
Jakob Stoklund Olesendd437ba2010-01-11 21:02:33 +00001037 if (cBB->isSuccessor(PHIBB)) {
Jim Grosbachf4549b02010-01-15 00:36:15 +00001038 PHI->addOperand(MachineOperand::
1039 CreateReg(SDB->PHINodesToUpdate[pi].second, false));
Chris Lattner9ce2e9d2007-12-30 00:57:42 +00001040 PHI->addOperand(MachineOperand::CreateMBB(cBB));
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001041 }
1042 }
1043 }
1044 }
Dan Gohman2048b852009-11-23 18:04:58 +00001045 SDB->BitTestCases.clear();
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001046
Nate Begeman9453eea2006-04-23 06:26:20 +00001047 // If the JumpTable record is filled in, then we need to emit a jump table.
1048 // Updating the PHI nodes is tricky in this case, since we need to determine
1049 // whether the PHI is a successor of the range check MBB or the jump table MBB
Dan Gohman2048b852009-11-23 18:04:58 +00001050 for (unsigned i = 0, e = SDB->JTCases.size(); i != e; ++i) {
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +00001051 // Lower header first, if it wasn't already lowered
Dan Gohman2048b852009-11-23 18:04:58 +00001052 if (!SDB->JTCases[i].first.Emitted) {
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +00001053 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman2048b852009-11-23 18:04:58 +00001054 BB = SDB->JTCases[i].first.HeaderBB;
1055 SDB->setCurrentBasicBlock(BB);
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +00001056 // Emit the code
Dan Gohman2048b852009-11-23 18:04:58 +00001057 SDB->visitJumpTableHeader(SDB->JTCases[i].second, SDB->JTCases[i].first);
1058 CurDAG->setRoot(SDB->getRoot());
Dan Gohmanf350b272008-08-23 02:25:05 +00001059 CodeGenAndEmitDAG();
Dan Gohman2048b852009-11-23 18:04:58 +00001060 SDB->clear();
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001061 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001062
Nate Begeman37efe672006-04-22 18:53:45 +00001063 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman2048b852009-11-23 18:04:58 +00001064 BB = SDB->JTCases[i].second.MBB;
1065 SDB->setCurrentBasicBlock(BB);
Nate Begeman37efe672006-04-22 18:53:45 +00001066 // Emit the code
Dan Gohman2048b852009-11-23 18:04:58 +00001067 SDB->visitJumpTable(SDB->JTCases[i].second);
1068 CurDAG->setRoot(SDB->getRoot());
Dan Gohmanf350b272008-08-23 02:25:05 +00001069 CodeGenAndEmitDAG();
Dan Gohman2048b852009-11-23 18:04:58 +00001070 SDB->clear();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001071
Nate Begeman37efe672006-04-22 18:53:45 +00001072 // Update PHI Nodes
Dan Gohman2048b852009-11-23 18:04:58 +00001073 for (unsigned pi = 0, pe = SDB->PHINodesToUpdate.size(); pi != pe; ++pi) {
1074 MachineInstr *PHI = SDB->PHINodesToUpdate[pi].first;
Nate Begeman37efe672006-04-22 18:53:45 +00001075 MachineBasicBlock *PHIBB = PHI->getParent();
Chris Lattner518bb532010-02-09 19:54:29 +00001076 assert(PHI->isPHI() &&
Nate Begeman37efe672006-04-22 18:53:45 +00001077 "This is not a machine PHI node that we are updating!");
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001078 // "default" BB. We can go there only from header BB.
Dan Gohman2048b852009-11-23 18:04:58 +00001079 if (PHIBB == SDB->JTCases[i].second.Default) {
Evan Chengce319102009-09-19 09:51:03 +00001080 PHI->addOperand
Dan Gohman2048b852009-11-23 18:04:58 +00001081 (MachineOperand::CreateReg(SDB->PHINodesToUpdate[pi].second, false));
Evan Chengce319102009-09-19 09:51:03 +00001082 PHI->addOperand
Dan Gohman2048b852009-11-23 18:04:58 +00001083 (MachineOperand::CreateMBB(SDB->JTCases[i].first.HeaderBB));
Nate Begemanf4360a42006-05-03 03:48:02 +00001084 }
Anton Korobeynikov4198c582007-04-09 12:31:58 +00001085 // JT BB. Just iterate over successors here
Jakob Stoklund Olesendd437ba2010-01-11 21:02:33 +00001086 if (BB->isSuccessor(PHIBB)) {
Evan Chengce319102009-09-19 09:51:03 +00001087 PHI->addOperand
Dan Gohman2048b852009-11-23 18:04:58 +00001088 (MachineOperand::CreateReg(SDB->PHINodesToUpdate[pi].second, false));
Chris Lattner9ce2e9d2007-12-30 00:57:42 +00001089 PHI->addOperand(MachineOperand::CreateMBB(BB));
Nate Begeman37efe672006-04-22 18:53:45 +00001090 }
1091 }
Nate Begeman37efe672006-04-22 18:53:45 +00001092 }
Dan Gohman2048b852009-11-23 18:04:58 +00001093 SDB->JTCases.clear();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001094
Chris Lattnerb2e806e2006-10-22 23:00:53 +00001095 // If the switch block involved a branch to one of the actual successors, we
1096 // need to update PHI nodes in that block.
Dan Gohman2048b852009-11-23 18:04:58 +00001097 for (unsigned i = 0, e = SDB->PHINodesToUpdate.size(); i != e; ++i) {
1098 MachineInstr *PHI = SDB->PHINodesToUpdate[i].first;
Chris Lattner518bb532010-02-09 19:54:29 +00001099 assert(PHI->isPHI() &&
Chris Lattnerb2e806e2006-10-22 23:00:53 +00001100 "This is not a machine PHI node that we are updating!");
1101 if (BB->isSuccessor(PHI->getParent())) {
Dan Gohman2048b852009-11-23 18:04:58 +00001102 PHI->addOperand(MachineOperand::CreateReg(SDB->PHINodesToUpdate[i].second,
Chris Lattner9ce2e9d2007-12-30 00:57:42 +00001103 false));
1104 PHI->addOperand(MachineOperand::CreateMBB(BB));
Chris Lattnerb2e806e2006-10-22 23:00:53 +00001105 }
1106 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001107
Nate Begemanf15485a2006-03-27 01:32:24 +00001108 // If we generated any switch lowering information, build and codegen any
1109 // additional DAGs necessary.
Dan Gohman2048b852009-11-23 18:04:58 +00001110 for (unsigned i = 0, e = SDB->SwitchCases.size(); i != e; ++i) {
Nate Begemanf15485a2006-03-27 01:32:24 +00001111 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman2048b852009-11-23 18:04:58 +00001112 MachineBasicBlock *ThisBB = BB = SDB->SwitchCases[i].ThisBB;
1113 SDB->setCurrentBasicBlock(BB);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001114
Nate Begemanf15485a2006-03-27 01:32:24 +00001115 // Emit the code
Dan Gohman2048b852009-11-23 18:04:58 +00001116 SDB->visitSwitchCase(SDB->SwitchCases[i]);
1117 CurDAG->setRoot(SDB->getRoot());
Dan Gohmanf350b272008-08-23 02:25:05 +00001118 CodeGenAndEmitDAG();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001119
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001120 // Handle any PHI nodes in successors of this chunk, as if we were coming
1121 // from the original BB before switch expansion. Note that PHI nodes can
1122 // occur multiple times in PHINodesToUpdate. We have to be very careful to
1123 // handle them the right number of times.
Dan Gohman2048b852009-11-23 18:04:58 +00001124 while ((BB = SDB->SwitchCases[i].TrueBB)) { // Handle LHS and RHS.
Evan Chengfb2e7522009-09-18 21:02:19 +00001125 // If new BB's are created during scheduling, the edges may have been
Evan Chengce319102009-09-19 09:51:03 +00001126 // updated. That is, the edge from ThisBB to BB may have been split and
1127 // BB's predecessor is now another block.
Evan Chengfb2e7522009-09-18 21:02:19 +00001128 DenseMap<MachineBasicBlock*, MachineBasicBlock*>::iterator EI =
Dan Gohman2048b852009-11-23 18:04:58 +00001129 SDB->EdgeMapping.find(BB);
1130 if (EI != SDB->EdgeMapping.end())
Evan Chengfb2e7522009-09-18 21:02:19 +00001131 ThisBB = EI->second;
Jakob Stoklund Olesendd437ba2010-01-11 21:02:33 +00001132
1133 // BB may have been removed from the CFG if a branch was constant folded.
1134 if (ThisBB->isSuccessor(BB)) {
1135 for (MachineBasicBlock::iterator Phi = BB->begin();
Chris Lattner518bb532010-02-09 19:54:29 +00001136 Phi != BB->end() && Phi->isPHI();
Jakob Stoklund Olesendd437ba2010-01-11 21:02:33 +00001137 ++Phi) {
1138 // This value for this PHI node is recorded in PHINodesToUpdate.
1139 for (unsigned pn = 0; ; ++pn) {
1140 assert(pn != SDB->PHINodesToUpdate.size() &&
1141 "Didn't find PHI entry!");
1142 if (SDB->PHINodesToUpdate[pn].first == Phi) {
1143 Phi->addOperand(MachineOperand::
1144 CreateReg(SDB->PHINodesToUpdate[pn].second,
1145 false));
1146 Phi->addOperand(MachineOperand::CreateMBB(ThisBB));
1147 break;
1148 }
Evan Cheng8be58a12009-09-18 08:26:06 +00001149 }
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001150 }
Nate Begemanf15485a2006-03-27 01:32:24 +00001151 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001152
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001153 // Don't process RHS if same block as LHS.
Dan Gohman2048b852009-11-23 18:04:58 +00001154 if (BB == SDB->SwitchCases[i].FalseBB)
1155 SDB->SwitchCases[i].FalseBB = 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001156
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001157 // If we haven't handled the RHS, do so now. Otherwise, we're done.
Dan Gohman2048b852009-11-23 18:04:58 +00001158 SDB->SwitchCases[i].TrueBB = SDB->SwitchCases[i].FalseBB;
1159 SDB->SwitchCases[i].FalseBB = 0;
Nate Begemanf15485a2006-03-27 01:32:24 +00001160 }
Dan Gohman2048b852009-11-23 18:04:58 +00001161 assert(SDB->SwitchCases[i].TrueBB == 0 && SDB->SwitchCases[i].FalseBB == 0);
1162 SDB->clear();
Chris Lattnera33ef482005-03-30 01:10:47 +00001163 }
Dan Gohman2048b852009-11-23 18:04:58 +00001164 SDB->SwitchCases.clear();
Dan Gohman7c3234c2008-08-27 23:52:12 +00001165
Dan Gohman2048b852009-11-23 18:04:58 +00001166 SDB->PHINodesToUpdate.clear();
Chris Lattner1c08c712005-01-07 07:47:53 +00001167}
Evan Chenga9c20912006-01-21 02:32:06 +00001168
Jim Laskey13ec7022006-08-01 14:21:23 +00001169
Dan Gohman0a3776d2009-02-06 18:26:51 +00001170/// Create the scheduler. If a specific scheduler was specified
1171/// via the SchedulerRegistry, use it, otherwise select the
1172/// one preferred by the target.
Dan Gohman5e843682008-07-14 18:19:29 +00001173///
Dan Gohman47ac0f02009-02-11 04:27:20 +00001174ScheduleDAGSDNodes *SelectionDAGISel::CreateScheduler() {
Jim Laskeyeb577ba2006-08-02 12:30:23 +00001175 RegisterScheduler::FunctionPassCtor Ctor = RegisterScheduler::getDefault();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001176
Jim Laskey13ec7022006-08-01 14:21:23 +00001177 if (!Ctor) {
Jim Laskeyeb577ba2006-08-02 12:30:23 +00001178 Ctor = ISHeuristic;
Jim Laskey9373beb2006-08-01 19:14:14 +00001179 RegisterScheduler::setDefault(Ctor);
Evan Cheng4ef10862006-01-23 07:01:07 +00001180 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001181
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +00001182 return Ctor(this, OptLevel);
Evan Chenga9c20912006-01-21 02:32:06 +00001183}
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001184
Dan Gohmanfc54c552009-01-15 22:18:12 +00001185ScheduleHazardRecognizer *SelectionDAGISel::CreateTargetHazardRecognizer() {
1186 return new ScheduleHazardRecognizer();
Jim Laskey9ff542f2006-08-01 18:29:48 +00001187}
1188
Chris Lattner75548062006-10-11 03:58:02 +00001189//===----------------------------------------------------------------------===//
1190// Helper functions used by the generated instruction selector.
1191//===----------------------------------------------------------------------===//
1192// Calls to these methods are generated by tblgen.
1193
1194/// CheckAndMask - The isel is trying to match something like (and X, 255). If
1195/// the dag combiner simplified the 255, we still want to match. RHS is the
1196/// actual value in the DAG on the RHS of an AND, and DesiredMaskS is the value
1197/// specified in the .td file (e.g. 255).
Daniel Dunbara279bc32009-09-20 02:20:51 +00001198bool SelectionDAGISel::CheckAndMask(SDValue LHS, ConstantSDNode *RHS,
Dan Gohmandc9b3d02007-07-24 23:00:27 +00001199 int64_t DesiredMaskS) const {
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001200 const APInt &ActualMask = RHS->getAPIntValue();
1201 const APInt &DesiredMask = APInt(LHS.getValueSizeInBits(), DesiredMaskS);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001202
Chris Lattner75548062006-10-11 03:58:02 +00001203 // If the actual mask exactly matches, success!
1204 if (ActualMask == DesiredMask)
1205 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001206
Chris Lattner75548062006-10-11 03:58:02 +00001207 // If the actual AND mask is allowing unallowed bits, this doesn't match.
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001208 if (ActualMask.intersects(~DesiredMask))
Chris Lattner75548062006-10-11 03:58:02 +00001209 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001210
Chris Lattner75548062006-10-11 03:58:02 +00001211 // Otherwise, the DAG Combiner may have proven that the value coming in is
1212 // either already zero or is not demanded. Check for known zero input bits.
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001213 APInt NeededMask = DesiredMask & ~ActualMask;
Dan Gohmanea859be2007-06-22 14:59:07 +00001214 if (CurDAG->MaskedValueIsZero(LHS, NeededMask))
Chris Lattner75548062006-10-11 03:58:02 +00001215 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001216
Chris Lattner75548062006-10-11 03:58:02 +00001217 // TODO: check to see if missing bits are just not demanded.
1218
1219 // Otherwise, this pattern doesn't match.
1220 return false;
1221}
1222
1223/// CheckOrMask - The isel is trying to match something like (or X, 255). If
1224/// the dag combiner simplified the 255, we still want to match. RHS is the
1225/// actual value in the DAG on the RHS of an OR, and DesiredMaskS is the value
1226/// specified in the .td file (e.g. 255).
Daniel Dunbara279bc32009-09-20 02:20:51 +00001227bool SelectionDAGISel::CheckOrMask(SDValue LHS, ConstantSDNode *RHS,
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001228 int64_t DesiredMaskS) const {
1229 const APInt &ActualMask = RHS->getAPIntValue();
1230 const APInt &DesiredMask = APInt(LHS.getValueSizeInBits(), DesiredMaskS);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001231
Chris Lattner75548062006-10-11 03:58:02 +00001232 // If the actual mask exactly matches, success!
1233 if (ActualMask == DesiredMask)
1234 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001235
Chris Lattner75548062006-10-11 03:58:02 +00001236 // If the actual AND mask is allowing unallowed bits, this doesn't match.
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001237 if (ActualMask.intersects(~DesiredMask))
Chris Lattner75548062006-10-11 03:58:02 +00001238 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001239
Chris Lattner75548062006-10-11 03:58:02 +00001240 // Otherwise, the DAG Combiner may have proven that the value coming in is
1241 // either already zero or is not demanded. Check for known zero input bits.
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001242 APInt NeededMask = DesiredMask & ~ActualMask;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001243
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001244 APInt KnownZero, KnownOne;
Dan Gohmanea859be2007-06-22 14:59:07 +00001245 CurDAG->ComputeMaskedBits(LHS, NeededMask, KnownZero, KnownOne);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001246
Chris Lattner75548062006-10-11 03:58:02 +00001247 // If all the missing bits in the or are already known to be set, match!
1248 if ((NeededMask & KnownOne) == NeededMask)
1249 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001250
Chris Lattner75548062006-10-11 03:58:02 +00001251 // TODO: check to see if missing bits are just not demanded.
Daniel Dunbara279bc32009-09-20 02:20:51 +00001252
Chris Lattner75548062006-10-11 03:58:02 +00001253 // Otherwise, this pattern doesn't match.
1254 return false;
1255}
1256
Jim Laskey9ff542f2006-08-01 18:29:48 +00001257
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001258/// SelectInlineAsmMemoryOperands - Calls to this are automatically generated
1259/// by tblgen. Others should not call it.
1260void SelectionDAGISel::
Dan Gohmanf350b272008-08-23 02:25:05 +00001261SelectInlineAsmMemoryOperands(std::vector<SDValue> &Ops) {
Dan Gohman475871a2008-07-27 21:46:04 +00001262 std::vector<SDValue> InOps;
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001263 std::swap(InOps, Ops);
1264
1265 Ops.push_back(InOps[0]); // input chain.
1266 Ops.push_back(InOps[1]); // input asm string.
1267
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001268 unsigned i = 2, e = InOps.size();
Owen Anderson825b72b2009-08-11 20:47:22 +00001269 if (InOps[e-1].getValueType() == MVT::Flag)
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001270 --e; // Don't process a flag operand if it is here.
Daniel Dunbara279bc32009-09-20 02:20:51 +00001271
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001272 while (i != e) {
Dan Gohmanf5aeb1a2008-09-12 16:56:44 +00001273 unsigned Flags = cast<ConstantSDNode>(InOps[i])->getZExtValue();
Dale Johannesen86b49f82008-09-24 01:07:17 +00001274 if ((Flags & 7) != 4 /*MEM*/) {
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001275 // Just skip over this operand, copying the operands verbatim.
Evan Cheng697cbbf2009-03-20 18:03:34 +00001276 Ops.insert(Ops.end(), InOps.begin()+i,
1277 InOps.begin()+i+InlineAsm::getNumOperandRegisters(Flags) + 1);
1278 i += InlineAsm::getNumOperandRegisters(Flags) + 1;
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001279 } else {
Evan Cheng697cbbf2009-03-20 18:03:34 +00001280 assert(InlineAsm::getNumOperandRegisters(Flags) == 1 &&
1281 "Memory operand with multiple values?");
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001282 // Otherwise, this is a memory operand. Ask the target to select it.
Dan Gohman475871a2008-07-27 21:46:04 +00001283 std::vector<SDValue> SelOps;
Dan Gohmanf350b272008-08-23 02:25:05 +00001284 if (SelectInlineAsmMemoryOperand(InOps[i+1], 'm', SelOps)) {
Torok Edwin7d696d82009-07-11 13:10:19 +00001285 llvm_report_error("Could not match memory address. Inline asm"
1286 " failure!");
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001287 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001288
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001289 // Add this to the output node.
Dale Johannesen86b49f82008-09-24 01:07:17 +00001290 Ops.push_back(CurDAG->getTargetConstant(4/*MEM*/ | (SelOps.size()<< 3),
Dale Johannesen99499332009-12-23 07:32:51 +00001291 MVT::i32));
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001292 Ops.insert(Ops.end(), SelOps.begin(), SelOps.end());
1293 i += 2;
1294 }
1295 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001296
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001297 // Add the flag input back if present.
1298 if (e != InOps.size())
1299 Ops.push_back(InOps.back());
1300}
Devang Patel794fd752007-05-01 21:15:47 +00001301
Owen Andersone50ed302009-08-10 22:56:29 +00001302/// findFlagUse - Return use of EVT::Flag value produced by the specified
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001303/// SDNode.
1304///
1305static SDNode *findFlagUse(SDNode *N) {
1306 unsigned FlagResNo = N->getNumValues()-1;
1307 for (SDNode::use_iterator I = N->use_begin(), E = N->use_end(); I != E; ++I) {
1308 SDUse &Use = I.getUse();
1309 if (Use.getResNo() == FlagResNo)
1310 return Use.getUser();
1311 }
1312 return NULL;
1313}
1314
1315/// findNonImmUse - Return true if "Use" is a non-immediate use of "Def".
1316/// This function recursively traverses up the operand chain, ignoring
1317/// certain nodes.
1318static bool findNonImmUse(SDNode *Use, SDNode* Def, SDNode *ImmedUse,
1319 SDNode *Root,
1320 SmallPtrSet<SDNode*, 16> &Visited) {
Chris Lattnerda244a02010-02-23 19:32:27 +00001321 // The NodeID's are given uniques ID's where a node ID is guaranteed to be
1322 // greater than all of its (recursive) operands. If we scan to a point where
1323 // 'use' is smaller than the node we're scanning for, then we know we will
1324 // never find it.
1325 //
1326 // The Use may be -1 (unassigned) if it is a newly allocated node. This can
1327 // happen because we scan down to newly selected nodes in the case of flag
1328 // uses.
1329 if ((Use->getNodeId() < Def->getNodeId() && Use->getNodeId() != -1))
1330 return false;
1331
1332 // Don't revisit nodes if we already scanned it and didn't fail, we know we
1333 // won't fail if we scan it again.
1334 if (!Visited.insert(Use))
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001335 return false;
1336
1337 for (unsigned i = 0, e = Use->getNumOperands(); i != e; ++i) {
1338 SDNode *N = Use->getOperand(i).getNode();
1339 if (N == Def) {
1340 if (Use == ImmedUse || Use == Root)
1341 continue; // We are not looking for immediate use.
1342 assert(N != Root);
1343 return true;
1344 }
1345
1346 // Traverse up the operand chain.
1347 if (findNonImmUse(N, Def, ImmedUse, Root, Visited))
1348 return true;
1349 }
1350 return false;
1351}
1352
1353/// isNonImmUse - Start searching from Root up the DAG to check is Def can
1354/// be reached. Return true if that's the case. However, ignore direct uses
1355/// by ImmedUse (which would be U in the example illustrated in
Evan Cheng014bf212010-02-15 19:41:07 +00001356/// IsLegalToFold) and by Root (which can happen in the store case).
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001357/// FIXME: to be really generic, we should allow direct use by any node
1358/// that is being folded. But realisticly since we only fold loads which
1359/// have one non-chain use, we only need to watch out for load/op/store
1360/// and load/op/cmp case where the root (store / cmp) may reach the load via
1361/// its chain operand.
1362static inline bool isNonImmUse(SDNode *Root, SDNode *Def, SDNode *ImmedUse) {
1363 SmallPtrSet<SDNode*, 16> Visited;
1364 return findNonImmUse(Root, Def, ImmedUse, Root, Visited);
1365}
1366
Evan Cheng014bf212010-02-15 19:41:07 +00001367/// IsProfitableToFold - Returns true if it's profitable to fold the specific
1368/// operand node N of U during instruction selection that starts at Root.
1369bool SelectionDAGISel::IsProfitableToFold(SDValue N, SDNode *U,
1370 SDNode *Root) const {
1371 if (OptLevel == CodeGenOpt::None) return false;
1372 return N.hasOneUse();
1373}
1374
1375/// IsLegalToFold - Returns true if the specific operand node N of
1376/// U can be folded during instruction selection that starts at Root.
1377bool SelectionDAGISel::IsLegalToFold(SDValue N, SDNode *U, SDNode *Root) const {
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001378 if (OptLevel == CodeGenOpt::None) return false;
1379
1380 // If Root use can somehow reach N through a path that that doesn't contain
1381 // U then folding N would create a cycle. e.g. In the following
1382 // diagram, Root can reach N through X. If N is folded into into Root, then
1383 // X is both a predecessor and a successor of U.
1384 //
1385 // [N*] //
1386 // ^ ^ //
1387 // / \ //
1388 // [U*] [X]? //
1389 // ^ ^ //
1390 // \ / //
1391 // \ / //
1392 // [Root*] //
1393 //
1394 // * indicates nodes to be folded together.
1395 //
1396 // If Root produces a flag, then it gets (even more) interesting. Since it
1397 // will be "glued" together with its flag use in the scheduler, we need to
1398 // check if it might reach N.
1399 //
1400 // [N*] //
1401 // ^ ^ //
1402 // / \ //
1403 // [U*] [X]? //
1404 // ^ ^ //
1405 // \ \ //
1406 // \ | //
1407 // [Root*] | //
1408 // ^ | //
1409 // f | //
1410 // | / //
1411 // [Y] / //
1412 // ^ / //
1413 // f / //
1414 // | / //
1415 // [FU] //
1416 //
1417 // If FU (flag use) indirectly reaches N (the load), and Root folds N
1418 // (call it Fold), then X is a predecessor of FU and a successor of
1419 // Fold. But since Fold and FU are flagged together, this will create
1420 // a cycle in the scheduling graph.
1421
Owen Andersone50ed302009-08-10 22:56:29 +00001422 EVT VT = Root->getValueType(Root->getNumValues()-1);
Owen Anderson825b72b2009-08-11 20:47:22 +00001423 while (VT == MVT::Flag) {
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001424 SDNode *FU = findFlagUse(Root);
1425 if (FU == NULL)
1426 break;
1427 Root = FU;
1428 VT = Root->getValueType(Root->getNumValues()-1);
1429 }
1430
Evan Cheng014bf212010-02-15 19:41:07 +00001431 return !isNonImmUse(Root, N.getNode(), U);
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001432}
1433
Dan Gohmaneeb3a002010-01-05 01:24:18 +00001434SDNode *SelectionDAGISel::Select_INLINEASM(SDNode *N) {
1435 std::vector<SDValue> Ops(N->op_begin(), N->op_end());
Dan Gohmane1f188f2009-10-29 22:30:23 +00001436 SelectInlineAsmMemoryOperands(Ops);
1437
1438 std::vector<EVT> VTs;
1439 VTs.push_back(MVT::Other);
1440 VTs.push_back(MVT::Flag);
Dan Gohmaneeb3a002010-01-05 01:24:18 +00001441 SDValue New = CurDAG->getNode(ISD::INLINEASM, N->getDebugLoc(),
Dan Gohmane1f188f2009-10-29 22:30:23 +00001442 VTs, &Ops[0], Ops.size());
1443 return New.getNode();
1444}
1445
Dan Gohmaneeb3a002010-01-05 01:24:18 +00001446SDNode *SelectionDAGISel::Select_UNDEF(SDNode *N) {
Chris Lattner518bb532010-02-09 19:54:29 +00001447 return CurDAG->SelectNodeTo(N, TargetOpcode::IMPLICIT_DEF,N->getValueType(0));
Dan Gohmane1f188f2009-10-29 22:30:23 +00001448}
1449
Dan Gohmaneeb3a002010-01-05 01:24:18 +00001450SDNode *SelectionDAGISel::Select_EH_LABEL(SDNode *N) {
1451 SDValue Chain = N->getOperand(0);
Dan Gohmane1f188f2009-10-29 22:30:23 +00001452 unsigned C = cast<LabelSDNode>(N)->getLabelID();
1453 SDValue Tmp = CurDAG->getTargetConstant(C, MVT::i32);
Chris Lattner518bb532010-02-09 19:54:29 +00001454 return CurDAG->SelectNodeTo(N, TargetOpcode::EH_LABEL,
Dan Gohmane1f188f2009-10-29 22:30:23 +00001455 MVT::Other, Tmp, Chain);
1456}
1457
Chris Lattner2a49d572010-02-28 22:37:22 +00001458/// GetVBR - decode a vbr encoding whose top bit is set.
1459ALWAYS_INLINE static uint64_t
1460GetVBR(uint64_t Val, const unsigned char *MatcherTable, unsigned &Idx) {
1461 assert(Val >= 128 && "Not a VBR");
1462 Val &= 127; // Remove first vbr bit.
1463
1464 unsigned Shift = 7;
1465 uint64_t NextBits;
1466 do {
Chris Lattner14df8dc2010-02-28 22:38:43 +00001467 NextBits = MatcherTable[Idx++];
Chris Lattner2a49d572010-02-28 22:37:22 +00001468 Val |= (NextBits&127) << Shift;
1469 Shift += 7;
1470 } while (NextBits & 128);
1471
1472 return Val;
1473}
1474
Chris Lattner2a49d572010-02-28 22:37:22 +00001475
1476/// UpdateChainsAndFlags - When a match is complete, this method updates uses of
1477/// interior flag and chain results to use the new flag and chain results.
1478static void UpdateChainsAndFlags(SDNode *NodeToMatch, SDValue InputChain,
1479 const SmallVectorImpl<SDNode*> &ChainNodesMatched,
1480 SDValue InputFlag,
1481 const SmallVectorImpl<SDNode*> &FlagResultNodesMatched,
1482 bool isMorphNodeTo, SelectionDAG *CurDAG) {
1483 // Now that all the normal results are replaced, we replace the chain and
1484 // flag results if present.
1485 if (!ChainNodesMatched.empty()) {
1486 assert(InputChain.getNode() != 0 &&
1487 "Matched input chains but didn't produce a chain");
1488 // Loop over all of the nodes we matched that produced a chain result.
1489 // Replace all the chain results with the final chain we ended up with.
1490 for (unsigned i = 0, e = ChainNodesMatched.size(); i != e; ++i) {
1491 SDNode *ChainNode = ChainNodesMatched[i];
1492
1493 // Don't replace the results of the root node if we're doing a
1494 // MorphNodeTo.
1495 if (ChainNode == NodeToMatch && isMorphNodeTo)
1496 continue;
1497
1498 SDValue ChainVal = SDValue(ChainNode, ChainNode->getNumValues()-1);
1499 if (ChainVal.getValueType() == MVT::Flag)
1500 ChainVal = ChainVal.getValue(ChainVal->getNumValues()-2);
1501 assert(ChainVal.getValueType() == MVT::Other && "Not a chain?");
1502 CurDAG->ReplaceAllUsesOfValueWith(ChainVal, InputChain);
1503 }
1504 }
1505
1506 // If the result produces a flag, update any flag results in the matched
1507 // pattern with the flag result.
1508 if (InputFlag.getNode() != 0) {
1509 // Handle any interior nodes explicitly marked.
1510 for (unsigned i = 0, e = FlagResultNodesMatched.size(); i != e; ++i) {
1511 SDNode *FRN = FlagResultNodesMatched[i];
1512 assert(FRN->getValueType(FRN->getNumValues()-1) == MVT::Flag &&
1513 "Doesn't have a flag result");
1514 CurDAG->ReplaceAllUsesOfValueWith(SDValue(FRN, FRN->getNumValues()-1),
1515 InputFlag);
1516 }
1517 }
1518
1519 DEBUG(errs() << "ISEL: Match complete!\n");
1520}
1521
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001522enum ChainResult {
1523 CR_Simple,
1524 CR_InducesCycle,
1525 CR_LeadsToInteriorNode
1526};
1527
1528/// WalkChainUsers - Walk down the users of the specified chained node that is
1529/// part of the pattern we're matching, looking at all of the users we find.
1530/// This determines whether something is an interior node, whether we have a
1531/// non-pattern node in between two pattern nodes (which prevent folding because
1532/// it would induce a cycle) and whether we have a TokenFactor node sandwiched
1533/// between pattern nodes (in which case the TF becomes part of the pattern).
1534///
1535/// The walk we do here is guaranteed to be small because we quickly get down to
1536/// already selected nodes "below" us.
1537static ChainResult
1538WalkChainUsers(SDNode *ChainedNode,
1539 SmallVectorImpl<SDNode*> &ChainedNodesInPattern,
1540 SmallVectorImpl<SDNode*> &InteriorChainedNodes) {
1541 ChainResult Result = CR_Simple;
1542
1543 for (SDNode::use_iterator UI = ChainedNode->use_begin(),
1544 E = ChainedNode->use_end(); UI != E; ++UI) {
1545 // Make sure the use is of the chain, not some other value we produce.
1546 if (UI.getUse().getValueType() != MVT::Other) continue;
1547
1548 SDNode *User = *UI;
1549
1550 // If we see an already-selected machine node, then we've gone beyond the
1551 // pattern that we're selecting down into the already selected chunk of the
1552 // DAG.
1553 if (User->isMachineOpcode() ||
1554 User->getOpcode() == ISD::CopyToReg ||
1555 User->getOpcode() == ISD::CopyFromReg ||
1556 User->getOpcode() == ISD::INLINEASM ||
1557 User->getOpcode() == ISD::HANDLENODE) // Root of the graph.
1558 continue;
1559
1560 // If we have a TokenFactor, we handle it specially.
1561 if (User->getOpcode() != ISD::TokenFactor) {
1562 // If the node isn't a token factor and isn't part of our pattern, then it
1563 // must be a random chained node in between two nodes we're selecting.
1564 // This happens when we have something like:
1565 // x = load ptr
1566 // call
1567 // y = x+4
1568 // store y -> ptr
1569 // Because we structurally match the load/store as a read/modify/write,
1570 // but the call is chained between them. We cannot fold in this case
1571 // because it would induce a cycle in the graph.
1572 if (!std::count(ChainedNodesInPattern.begin(),
1573 ChainedNodesInPattern.end(), User))
1574 return CR_InducesCycle;
1575
1576 // Otherwise we found a node that is part of our pattern. For example in:
1577 // x = load ptr
1578 // y = x+4
1579 // store y -> ptr
1580 // This would happen when we're scanning down from the load and see the
1581 // store as a user. Record that there is a use of ChainedNode that is
1582 // part of the pattern and keep scanning uses.
1583 Result = CR_LeadsToInteriorNode;
1584 InteriorChainedNodes.push_back(User);
1585 continue;
1586 }
1587
1588 // If we found a TokenFactor, there are two cases to consider: first if the
1589 // TokenFactor is just hanging "below" the pattern we're matching (i.e. no
1590 // uses of the TF are in our pattern) we just want to ignore it. Second,
1591 // the TokenFactor can be sandwiched in between two chained nodes, like so:
1592 // [Load chain]
1593 // ^
1594 // |
1595 // [Load]
1596 // ^ ^
1597 // | \ DAG's like cheese
1598 // / \ do you?
1599 // / |
1600 // [TokenFactor] [Op]
1601 // ^ ^
1602 // | |
1603 // \ /
1604 // \ /
1605 // [Store]
1606 //
1607 // In this case, the TokenFactor becomes part of our match and we rewrite it
1608 // as a new TokenFactor.
1609 //
1610 // To distinguish these two cases, do a recursive walk down the uses.
1611 switch (WalkChainUsers(User, ChainedNodesInPattern, InteriorChainedNodes)) {
1612 case CR_Simple:
1613 // If the uses of the TokenFactor are just already-selected nodes, ignore
1614 // it, it is "below" our pattern.
1615 continue;
1616 case CR_InducesCycle:
1617 // If the uses of the TokenFactor lead to nodes that are not part of our
1618 // pattern that are not selected, folding would turn this into a cycle,
1619 // bail out now.
1620 return CR_InducesCycle;
1621 case CR_LeadsToInteriorNode:
1622 break; // Otherwise, keep processing.
1623 }
1624
1625 // Okay, we know we're in the interesting interior case. The TokenFactor
1626 // is now going to be considered part of the pattern so that we rewrite its
1627 // uses (it may have uses that are not part of the pattern) with the
1628 // ultimate chain result of the generated code. We will also add its chain
1629 // inputs as inputs to the ultimate TokenFactor we create.
1630 Result = CR_LeadsToInteriorNode;
1631 ChainedNodesInPattern.push_back(User);
1632 InteriorChainedNodes.push_back(User);
1633 continue;
1634 }
1635
1636 return Result;
1637}
1638
Chris Lattner6b307922010-03-02 00:00:03 +00001639/// HandleMergeInputChains - This implements the OPC_EmitMergeInputChains
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001640/// operation for when the pattern matched multiple nodes with chains. The
1641/// input vector contains a list of all of the chained nodes that we match. We
1642/// must determine if this is a valid thing to cover (i.e. matching it won't
1643/// induce cycles in the DAG) and if so, creating a TokenFactor node. that will
1644/// be used as the input node chain for the generated nodes.
Chris Lattner6b307922010-03-02 00:00:03 +00001645static SDValue
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001646HandleMergeInputChains(SmallVectorImpl<SDNode*> &ChainNodesMatched,
Chris Lattner6b307922010-03-02 00:00:03 +00001647 SelectionDAG *CurDAG) {
1648 assert(ChainNodesMatched.size() > 1 &&
1649 "Should only happen for multi chain node case");
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001650
1651 // Walk all of the chained nodes we've matched, recursively scanning down the
1652 // users of the chain result. This adds any TokenFactor nodes that are caught
1653 // in between chained nodes to the chained and interior nodes list.
1654 SmallVector<SDNode*, 3> InteriorChainedNodes;
1655 for (unsigned i = 0, e = ChainNodesMatched.size(); i != e; ++i) {
1656 if (WalkChainUsers(ChainNodesMatched[i], ChainNodesMatched,
1657 InteriorChainedNodes) == CR_InducesCycle)
1658 return SDValue(); // Would induce a cycle.
1659 }
Chris Lattner6b307922010-03-02 00:00:03 +00001660
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001661 // Okay, we have walked all the matched nodes and collected TokenFactor nodes
1662 // that we are interested in. Form our input TokenFactor node.
Chris Lattner6b307922010-03-02 00:00:03 +00001663 SmallVector<SDValue, 3> InputChains;
1664 for (unsigned i = 0, e = ChainNodesMatched.size(); i != e; ++i) {
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001665 // Add the input chain of this node to the InputChains list (which will be
1666 // the operands of the generated TokenFactor) if it's not an interior node.
1667 SDNode *N = ChainNodesMatched[i];
1668 if (N->getOpcode() != ISD::TokenFactor) {
1669 if (std::count(InteriorChainedNodes.begin(),InteriorChainedNodes.end(),N))
1670 continue;
1671
1672 // Otherwise, add the input chain.
1673 SDValue InChain = ChainNodesMatched[i]->getOperand(0);
1674 assert(InChain.getValueType() == MVT::Other && "Not a chain");
Chris Lattner6b307922010-03-02 00:00:03 +00001675 InputChains.push_back(InChain);
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001676 continue;
1677 }
1678
1679 // If we have a token factor, we want to add all inputs of the token factor
1680 // that are not part of the pattern we're matching.
1681 for (unsigned op = 0, e = N->getNumOperands(); op != e; ++op) {
1682 if (!std::count(ChainNodesMatched.begin(), ChainNodesMatched.end(),
1683 N->getOperand(i).getNode()))
1684 InputChains.push_back(N->getOperand(i));
1685 }
Chris Lattner6b307922010-03-02 00:00:03 +00001686 }
1687
1688 SDValue Res;
1689 if (InputChains.size() == 1)
1690 return InputChains[0];
1691 return CurDAG->getNode(ISD::TokenFactor, ChainNodesMatched[0]->getDebugLoc(),
1692 MVT::Other, &InputChains[0], InputChains.size());
1693}
Chris Lattner2a49d572010-02-28 22:37:22 +00001694
1695struct MatchScope {
1696 /// FailIndex - If this match fails, this is the index to continue with.
1697 unsigned FailIndex;
1698
1699 /// NodeStack - The node stack when the scope was formed.
1700 SmallVector<SDValue, 4> NodeStack;
1701
1702 /// NumRecordedNodes - The number of recorded nodes when the scope was formed.
1703 unsigned NumRecordedNodes;
1704
1705 /// NumMatchedMemRefs - The number of matched memref entries.
1706 unsigned NumMatchedMemRefs;
1707
1708 /// InputChain/InputFlag - The current chain/flag
1709 SDValue InputChain, InputFlag;
1710
1711 /// HasChainNodesMatched - True if the ChainNodesMatched list is non-empty.
1712 bool HasChainNodesMatched, HasFlagResultNodesMatched;
1713};
1714
1715SDNode *SelectionDAGISel::
1716SelectCodeCommon(SDNode *NodeToMatch, const unsigned char *MatcherTable,
1717 unsigned TableSize) {
1718 // FIXME: Should these even be selected? Handle these cases in the caller?
1719 switch (NodeToMatch->getOpcode()) {
1720 default:
1721 break;
1722 case ISD::EntryToken: // These nodes remain the same.
1723 case ISD::BasicBlock:
1724 case ISD::Register:
1725 case ISD::HANDLENODE:
1726 case ISD::TargetConstant:
1727 case ISD::TargetConstantFP:
1728 case ISD::TargetConstantPool:
1729 case ISD::TargetFrameIndex:
1730 case ISD::TargetExternalSymbol:
1731 case ISD::TargetBlockAddress:
1732 case ISD::TargetJumpTable:
1733 case ISD::TargetGlobalTLSAddress:
1734 case ISD::TargetGlobalAddress:
1735 case ISD::TokenFactor:
1736 case ISD::CopyFromReg:
1737 case ISD::CopyToReg:
1738 return 0;
1739 case ISD::AssertSext:
1740 case ISD::AssertZext:
1741 CurDAG->ReplaceAllUsesOfValueWith(SDValue(NodeToMatch, 0),
1742 NodeToMatch->getOperand(0));
1743 return 0;
1744 case ISD::INLINEASM: return Select_INLINEASM(NodeToMatch);
1745 case ISD::EH_LABEL: return Select_EH_LABEL(NodeToMatch);
1746 case ISD::UNDEF: return Select_UNDEF(NodeToMatch);
1747 }
1748
1749 assert(!NodeToMatch->isMachineOpcode() && "Node already selected!");
1750
1751 // Set up the node stack with NodeToMatch as the only node on the stack.
1752 SmallVector<SDValue, 8> NodeStack;
1753 SDValue N = SDValue(NodeToMatch, 0);
1754 NodeStack.push_back(N);
1755
1756 // MatchScopes - Scopes used when matching, if a match failure happens, this
1757 // indicates where to continue checking.
1758 SmallVector<MatchScope, 8> MatchScopes;
1759
1760 // RecordedNodes - This is the set of nodes that have been recorded by the
1761 // state machine.
1762 SmallVector<SDValue, 8> RecordedNodes;
1763
1764 // MatchedMemRefs - This is the set of MemRef's we've seen in the input
1765 // pattern.
1766 SmallVector<MachineMemOperand*, 2> MatchedMemRefs;
1767
1768 // These are the current input chain and flag for use when generating nodes.
1769 // Various Emit operations change these. For example, emitting a copytoreg
1770 // uses and updates these.
1771 SDValue InputChain, InputFlag;
1772
1773 // ChainNodesMatched - If a pattern matches nodes that have input/output
1774 // chains, the OPC_EmitMergeInputChains operation is emitted which indicates
1775 // which ones they are. The result is captured into this list so that we can
1776 // update the chain results when the pattern is complete.
1777 SmallVector<SDNode*, 3> ChainNodesMatched;
1778 SmallVector<SDNode*, 3> FlagResultNodesMatched;
1779
1780 DEBUG(errs() << "ISEL: Starting pattern match on root node: ";
1781 NodeToMatch->dump(CurDAG);
1782 errs() << '\n');
1783
Chris Lattner7390eeb2010-03-01 18:47:11 +00001784 // Determine where to start the interpreter. Normally we start at opcode #0,
1785 // but if the state machine starts with an OPC_SwitchOpcode, then we
1786 // accelerate the first lookup (which is guaranteed to be hot) with the
1787 // OpcodeOffset table.
Chris Lattner2a49d572010-02-28 22:37:22 +00001788 unsigned MatcherIndex = 0;
Chris Lattner7390eeb2010-03-01 18:47:11 +00001789
1790 if (!OpcodeOffset.empty()) {
1791 // Already computed the OpcodeOffset table, just index into it.
1792 if (N.getOpcode() < OpcodeOffset.size())
1793 MatcherIndex = OpcodeOffset[N.getOpcode()];
1794 DEBUG(errs() << " Initial Opcode index to " << MatcherIndex << "\n");
1795
1796 } else if (MatcherTable[0] == OPC_SwitchOpcode) {
1797 // Otherwise, the table isn't computed, but the state machine does start
1798 // with an OPC_SwitchOpcode instruction. Populate the table now, since this
1799 // is the first time we're selecting an instruction.
1800 unsigned Idx = 1;
1801 while (1) {
1802 // Get the size of this case.
1803 unsigned CaseSize = MatcherTable[Idx++];
1804 if (CaseSize & 128)
1805 CaseSize = GetVBR(CaseSize, MatcherTable, Idx);
1806 if (CaseSize == 0) break;
1807
1808 // Get the opcode, add the index to the table.
1809 unsigned Opc = MatcherTable[Idx++];
1810 if (Opc >= OpcodeOffset.size())
1811 OpcodeOffset.resize((Opc+1)*2);
1812 OpcodeOffset[Opc] = Idx;
1813 Idx += CaseSize;
1814 }
1815
1816 // Okay, do the lookup for the first opcode.
1817 if (N.getOpcode() < OpcodeOffset.size())
1818 MatcherIndex = OpcodeOffset[N.getOpcode()];
1819 }
1820
Chris Lattner2a49d572010-02-28 22:37:22 +00001821 while (1) {
1822 assert(MatcherIndex < TableSize && "Invalid index");
1823 BuiltinOpcodes Opcode = (BuiltinOpcodes)MatcherTable[MatcherIndex++];
1824 switch (Opcode) {
1825 case OPC_Scope: {
1826 unsigned NumToSkip = MatcherTable[MatcherIndex++];
1827 if (NumToSkip & 128)
1828 NumToSkip = GetVBR(NumToSkip, MatcherTable, MatcherIndex);
1829 assert(NumToSkip != 0 &&
1830 "First entry of OPC_Scope shouldn't be 0, scope has no children?");
1831
1832 // Push a MatchScope which indicates where to go if the first child fails
1833 // to match.
1834 MatchScope NewEntry;
1835 NewEntry.FailIndex = MatcherIndex+NumToSkip;
1836 NewEntry.NodeStack.append(NodeStack.begin(), NodeStack.end());
1837 NewEntry.NumRecordedNodes = RecordedNodes.size();
1838 NewEntry.NumMatchedMemRefs = MatchedMemRefs.size();
1839 NewEntry.InputChain = InputChain;
1840 NewEntry.InputFlag = InputFlag;
1841 NewEntry.HasChainNodesMatched = !ChainNodesMatched.empty();
1842 NewEntry.HasFlagResultNodesMatched = !FlagResultNodesMatched.empty();
1843 MatchScopes.push_back(NewEntry);
1844 continue;
1845 }
1846 case OPC_RecordNode:
1847 // Remember this node, it may end up being an operand in the pattern.
1848 RecordedNodes.push_back(N);
1849 continue;
1850
1851 case OPC_RecordChild0: case OPC_RecordChild1:
1852 case OPC_RecordChild2: case OPC_RecordChild3:
1853 case OPC_RecordChild4: case OPC_RecordChild5:
1854 case OPC_RecordChild6: case OPC_RecordChild7: {
1855 unsigned ChildNo = Opcode-OPC_RecordChild0;
1856 if (ChildNo >= N.getNumOperands())
1857 break; // Match fails if out of range child #.
1858
1859 RecordedNodes.push_back(N->getOperand(ChildNo));
1860 continue;
1861 }
1862 case OPC_RecordMemRef:
1863 MatchedMemRefs.push_back(cast<MemSDNode>(N)->getMemOperand());
1864 continue;
1865
1866 case OPC_CaptureFlagInput:
1867 // If the current node has an input flag, capture it in InputFlag.
1868 if (N->getNumOperands() != 0 &&
1869 N->getOperand(N->getNumOperands()-1).getValueType() == MVT::Flag)
1870 InputFlag = N->getOperand(N->getNumOperands()-1);
1871 continue;
1872
1873 case OPC_MoveChild: {
1874 unsigned ChildNo = MatcherTable[MatcherIndex++];
1875 if (ChildNo >= N.getNumOperands())
1876 break; // Match fails if out of range child #.
1877 N = N.getOperand(ChildNo);
1878 NodeStack.push_back(N);
1879 continue;
1880 }
1881
1882 case OPC_MoveParent:
1883 // Pop the current node off the NodeStack.
1884 NodeStack.pop_back();
1885 assert(!NodeStack.empty() && "Node stack imbalance!");
1886 N = NodeStack.back();
1887 continue;
1888
1889 case OPC_CheckSame: {
1890 // Accept if it is exactly the same as a previously recorded node.
1891 unsigned RecNo = MatcherTable[MatcherIndex++];
1892 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
1893 if (N != RecordedNodes[RecNo]) break;
1894 continue;
1895 }
1896 case OPC_CheckPatternPredicate:
1897 if (!CheckPatternPredicate(MatcherTable[MatcherIndex++])) break;
1898 continue;
1899 case OPC_CheckPredicate:
1900 if (!CheckNodePredicate(N.getNode(), MatcherTable[MatcherIndex++])) break;
1901 continue;
1902 case OPC_CheckComplexPat:
1903 if (!CheckComplexPattern(NodeToMatch, N,
1904 MatcherTable[MatcherIndex++], RecordedNodes))
1905 break;
1906 continue;
1907 case OPC_CheckOpcode:
1908 if (N->getOpcode() != MatcherTable[MatcherIndex++]) break;
1909 continue;
1910
Chris Lattnereb669212010-03-01 06:59:22 +00001911 case OPC_SwitchOpcode: {
1912 unsigned CurNodeOpcode = N.getOpcode();
1913
Chris Lattner7d892d62010-03-01 07:43:08 +00001914 unsigned SwitchStart = MatcherIndex-1; (void)SwitchStart;
Chris Lattnereb669212010-03-01 06:59:22 +00001915
1916 unsigned CaseSize;
1917 while (1) {
1918 // Get the size of this case.
1919 CaseSize = MatcherTable[MatcherIndex++];
1920 if (CaseSize & 128)
1921 CaseSize = GetVBR(CaseSize, MatcherTable, MatcherIndex);
1922 if (CaseSize == 0) break;
1923
1924 // If the opcode matches, then we will execute this case.
1925 if (CurNodeOpcode == MatcherTable[MatcherIndex++])
1926 break;
1927
1928 // Otherwise, skip over this case.
1929 MatcherIndex += CaseSize;
1930 }
1931
1932 // If we failed to match, bail out.
1933 if (CaseSize == 0) break;
1934
1935 // Otherwise, execute the case we found.
1936 DEBUG(errs() << " OpcodeSwitch from " << SwitchStart
1937 << " to " << MatcherIndex << "\n");
1938 continue;
1939 }
1940
Chris Lattner2a49d572010-02-28 22:37:22 +00001941 case OPC_CheckType: {
1942 MVT::SimpleValueType VT =
1943 (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
1944 if (N.getValueType() != VT) {
1945 // Handle the case when VT is iPTR.
1946 if (VT != MVT::iPTR || N.getValueType() != TLI.getPointerTy())
1947 break;
1948 }
1949 continue;
1950 }
1951 case OPC_CheckChild0Type: case OPC_CheckChild1Type:
1952 case OPC_CheckChild2Type: case OPC_CheckChild3Type:
1953 case OPC_CheckChild4Type: case OPC_CheckChild5Type:
1954 case OPC_CheckChild6Type: case OPC_CheckChild7Type: {
1955 unsigned ChildNo = Opcode-OPC_CheckChild0Type;
1956 if (ChildNo >= N.getNumOperands())
1957 break; // Match fails if out of range child #.
1958
1959 MVT::SimpleValueType VT =
1960 (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
1961 EVT ChildVT = N.getOperand(ChildNo).getValueType();
1962 if (ChildVT != VT) {
1963 // Handle the case when VT is iPTR.
1964 if (VT != MVT::iPTR || ChildVT != TLI.getPointerTy())
1965 break;
1966 }
1967 continue;
1968 }
1969 case OPC_CheckCondCode:
1970 if (cast<CondCodeSDNode>(N)->get() !=
1971 (ISD::CondCode)MatcherTable[MatcherIndex++]) break;
1972 continue;
1973 case OPC_CheckValueType: {
1974 MVT::SimpleValueType VT =
1975 (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
1976 if (cast<VTSDNode>(N)->getVT() != VT) {
1977 // Handle the case when VT is iPTR.
1978 if (VT != MVT::iPTR || cast<VTSDNode>(N)->getVT() != TLI.getPointerTy())
1979 break;
1980 }
1981 continue;
1982 }
1983 case OPC_CheckInteger: {
1984 int64_t Val = MatcherTable[MatcherIndex++];
1985 if (Val & 128)
1986 Val = GetVBR(Val, MatcherTable, MatcherIndex);
1987
1988 ConstantSDNode *C = dyn_cast<ConstantSDNode>(N);
1989 if (C == 0 || C->getSExtValue() != Val)
1990 break;
1991 continue;
1992 }
1993 case OPC_CheckAndImm: {
1994 int64_t Val = MatcherTable[MatcherIndex++];
1995 if (Val & 128)
1996 Val = GetVBR(Val, MatcherTable, MatcherIndex);
1997
1998 if (N->getOpcode() != ISD::AND) break;
1999 ConstantSDNode *C = dyn_cast<ConstantSDNode>(N->getOperand(1));
2000 if (C == 0 || !CheckAndMask(N.getOperand(0), C, Val))
2001 break;
2002 continue;
2003 }
2004 case OPC_CheckOrImm: {
2005 int64_t Val = MatcherTable[MatcherIndex++];
2006 if (Val & 128)
2007 Val = GetVBR(Val, MatcherTable, MatcherIndex);
2008
2009 if (N->getOpcode() != ISD::OR) break;
2010
2011 ConstantSDNode *C = dyn_cast<ConstantSDNode>(N->getOperand(1));
2012 if (C == 0 || !CheckOrMask(N.getOperand(0), C, Val))
2013 break;
2014 continue;
2015 }
2016
2017 case OPC_CheckFoldableChainNode: {
2018 assert(NodeStack.size() != 1 && "No parent node");
2019 // Verify that all intermediate nodes between the root and this one have
2020 // a single use.
2021 bool HasMultipleUses = false;
2022 for (unsigned i = 1, e = NodeStack.size()-1; i != e; ++i)
2023 if (!NodeStack[i].hasOneUse()) {
2024 HasMultipleUses = true;
2025 break;
2026 }
2027 if (HasMultipleUses) break;
2028
2029 // Check to see that the target thinks this is profitable to fold and that
2030 // we can fold it without inducing cycles in the graph.
2031 if (!IsProfitableToFold(N, NodeStack[NodeStack.size()-2].getNode(),
2032 NodeToMatch) ||
2033 !IsLegalToFold(N, NodeStack[NodeStack.size()-2].getNode(),
2034 NodeToMatch))
2035 break;
2036
2037 continue;
2038 }
Chris Lattner2a49d572010-02-28 22:37:22 +00002039 case OPC_EmitInteger: {
2040 MVT::SimpleValueType VT =
2041 (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
2042 int64_t Val = MatcherTable[MatcherIndex++];
2043 if (Val & 128)
2044 Val = GetVBR(Val, MatcherTable, MatcherIndex);
2045 RecordedNodes.push_back(CurDAG->getTargetConstant(Val, VT));
2046 continue;
2047 }
2048 case OPC_EmitRegister: {
2049 MVT::SimpleValueType VT =
2050 (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
2051 unsigned RegNo = MatcherTable[MatcherIndex++];
2052 RecordedNodes.push_back(CurDAG->getRegister(RegNo, VT));
2053 continue;
2054 }
2055
2056 case OPC_EmitConvertToTarget: {
2057 // Convert from IMM/FPIMM to target version.
2058 unsigned RecNo = MatcherTable[MatcherIndex++];
2059 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2060 SDValue Imm = RecordedNodes[RecNo];
2061
2062 if (Imm->getOpcode() == ISD::Constant) {
2063 int64_t Val = cast<ConstantSDNode>(Imm)->getZExtValue();
2064 Imm = CurDAG->getTargetConstant(Val, Imm.getValueType());
2065 } else if (Imm->getOpcode() == ISD::ConstantFP) {
2066 const ConstantFP *Val=cast<ConstantFPSDNode>(Imm)->getConstantFPValue();
2067 Imm = CurDAG->getTargetConstantFP(*Val, Imm.getValueType());
2068 }
2069
2070 RecordedNodes.push_back(Imm);
2071 continue;
2072 }
2073
2074 case OPC_EmitMergeInputChains: {
2075 assert(InputChain.getNode() == 0 &&
2076 "EmitMergeInputChains should be the first chain producing node");
2077 // This node gets a list of nodes we matched in the input that have
2078 // chains. We want to token factor all of the input chains to these nodes
2079 // together. However, if any of the input chains is actually one of the
2080 // nodes matched in this pattern, then we have an intra-match reference.
2081 // Ignore these because the newly token factored chain should not refer to
2082 // the old nodes.
2083 unsigned NumChains = MatcherTable[MatcherIndex++];
2084 assert(NumChains != 0 && "Can't TF zero chains");
2085
2086 assert(ChainNodesMatched.empty() &&
2087 "Should only have one EmitMergeInputChains per match");
2088
2089 // Handle the first chain.
2090 unsigned RecNo = MatcherTable[MatcherIndex++];
2091 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2092 ChainNodesMatched.push_back(RecordedNodes[RecNo].getNode());
2093
2094 // If the chained node is not the root, we can't fold it if it has
2095 // multiple uses.
2096 // FIXME: What if other value results of the node have uses not matched by
2097 // this pattern?
2098 if (ChainNodesMatched.back() != NodeToMatch &&
2099 !RecordedNodes[RecNo].hasOneUse()) {
2100 ChainNodesMatched.clear();
2101 break;
2102 }
2103
2104 // The common case here is that we have exactly one chain, which is really
2105 // cheap to handle, just do it.
2106 if (NumChains == 1) {
2107 InputChain = RecordedNodes[RecNo].getOperand(0);
2108 assert(InputChain.getValueType() == MVT::Other && "Not a chain");
2109 continue;
2110 }
2111
2112 // Read all of the chained nodes.
2113 for (unsigned i = 1; i != NumChains; ++i) {
2114 RecNo = MatcherTable[MatcherIndex++];
2115 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2116 ChainNodesMatched.push_back(RecordedNodes[RecNo].getNode());
2117
2118 // FIXME: What if other value results of the node have uses not matched
2119 // by this pattern?
2120 if (ChainNodesMatched.back() != NodeToMatch &&
2121 !RecordedNodes[RecNo].hasOneUse()) {
2122 ChainNodesMatched.clear();
2123 break;
2124 }
2125 }
Chris Lattner6b307922010-03-02 00:00:03 +00002126
2127 // If the inner loop broke out, the match fails.
2128 if (ChainNodesMatched.empty())
2129 break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002130
Chris Lattner6b307922010-03-02 00:00:03 +00002131 // Merge the input chains if they are not intra-pattern references.
2132 InputChain = HandleMergeInputChains(ChainNodesMatched, CurDAG);
2133
2134 if (InputChain.getNode() == 0)
2135 break; // Failed to merge.
Chris Lattner2a49d572010-02-28 22:37:22 +00002136
Chris Lattner2a49d572010-02-28 22:37:22 +00002137 continue;
2138 }
2139
2140 case OPC_EmitCopyToReg: {
2141 unsigned RecNo = MatcherTable[MatcherIndex++];
2142 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2143 unsigned DestPhysReg = MatcherTable[MatcherIndex++];
2144
2145 if (InputChain.getNode() == 0)
2146 InputChain = CurDAG->getEntryNode();
2147
2148 InputChain = CurDAG->getCopyToReg(InputChain, NodeToMatch->getDebugLoc(),
2149 DestPhysReg, RecordedNodes[RecNo],
2150 InputFlag);
2151
2152 InputFlag = InputChain.getValue(1);
2153 continue;
2154 }
2155
2156 case OPC_EmitNodeXForm: {
2157 unsigned XFormNo = MatcherTable[MatcherIndex++];
2158 unsigned RecNo = MatcherTable[MatcherIndex++];
2159 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2160 RecordedNodes.push_back(RunSDNodeXForm(RecordedNodes[RecNo], XFormNo));
2161 continue;
2162 }
2163
2164 case OPC_EmitNode:
2165 case OPC_MorphNodeTo: {
Chris Lattner14df8dc2010-02-28 22:38:43 +00002166 uint16_t TargetOpc = MatcherTable[MatcherIndex++];
2167 TargetOpc |= (unsigned short)MatcherTable[MatcherIndex++] << 8;
Chris Lattner2a49d572010-02-28 22:37:22 +00002168 unsigned EmitNodeInfo = MatcherTable[MatcherIndex++];
2169 // Get the result VT list.
2170 unsigned NumVTs = MatcherTable[MatcherIndex++];
2171 SmallVector<EVT, 4> VTs;
2172 for (unsigned i = 0; i != NumVTs; ++i) {
2173 MVT::SimpleValueType VT =
2174 (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
2175 if (VT == MVT::iPTR) VT = TLI.getPointerTy().SimpleTy;
2176 VTs.push_back(VT);
2177 }
2178
2179 if (EmitNodeInfo & OPFL_Chain)
2180 VTs.push_back(MVT::Other);
2181 if (EmitNodeInfo & OPFL_FlagOutput)
2182 VTs.push_back(MVT::Flag);
2183
Chris Lattner7d892d62010-03-01 07:43:08 +00002184 // This is hot code, so optimize the two most common cases of 1 and 2
2185 // results.
2186 SDVTList VTList;
2187 if (VTs.size() == 1)
2188 VTList = CurDAG->getVTList(VTs[0]);
2189 else if (VTs.size() == 2)
2190 VTList = CurDAG->getVTList(VTs[0], VTs[1]);
2191 else
2192 VTList = CurDAG->getVTList(VTs.data(), VTs.size());
Chris Lattner2a49d572010-02-28 22:37:22 +00002193
2194 // Get the operand list.
2195 unsigned NumOps = MatcherTable[MatcherIndex++];
2196 SmallVector<SDValue, 8> Ops;
2197 for (unsigned i = 0; i != NumOps; ++i) {
2198 unsigned RecNo = MatcherTable[MatcherIndex++];
2199 if (RecNo & 128)
2200 RecNo = GetVBR(RecNo, MatcherTable, MatcherIndex);
2201
2202 assert(RecNo < RecordedNodes.size() && "Invalid EmitNode");
2203 Ops.push_back(RecordedNodes[RecNo]);
2204 }
2205
2206 // If there are variadic operands to add, handle them now.
2207 if (EmitNodeInfo & OPFL_VariadicInfo) {
2208 // Determine the start index to copy from.
2209 unsigned FirstOpToCopy = getNumFixedFromVariadicInfo(EmitNodeInfo);
2210 FirstOpToCopy += (EmitNodeInfo & OPFL_Chain) ? 1 : 0;
2211 assert(NodeToMatch->getNumOperands() >= FirstOpToCopy &&
2212 "Invalid variadic node");
2213 // Copy all of the variadic operands, not including a potential flag
2214 // input.
2215 for (unsigned i = FirstOpToCopy, e = NodeToMatch->getNumOperands();
2216 i != e; ++i) {
2217 SDValue V = NodeToMatch->getOperand(i);
2218 if (V.getValueType() == MVT::Flag) break;
2219 Ops.push_back(V);
2220 }
2221 }
2222
2223 // If this has chain/flag inputs, add them.
2224 if (EmitNodeInfo & OPFL_Chain)
2225 Ops.push_back(InputChain);
2226 if ((EmitNodeInfo & OPFL_FlagInput) && InputFlag.getNode() != 0)
2227 Ops.push_back(InputFlag);
2228
2229 // Create the node.
2230 SDNode *Res = 0;
2231 if (Opcode != OPC_MorphNodeTo) {
2232 // If this is a normal EmitNode command, just create the new node and
2233 // add the results to the RecordedNodes list.
2234 Res = CurDAG->getMachineNode(TargetOpc, NodeToMatch->getDebugLoc(),
2235 VTList, Ops.data(), Ops.size());
2236
2237 // Add all the non-flag/non-chain results to the RecordedNodes list.
2238 for (unsigned i = 0, e = VTs.size(); i != e; ++i) {
2239 if (VTs[i] == MVT::Other || VTs[i] == MVT::Flag) break;
2240 RecordedNodes.push_back(SDValue(Res, i));
2241 }
2242
2243 } else {
2244 // It is possible we're using MorphNodeTo to replace a node with no
2245 // normal results with one that has a normal result (or we could be
2246 // adding a chain) and the input could have flags and chains as well.
2247 // In this case we need to shifting the operands down.
2248 // FIXME: This is a horrible hack and broken in obscure cases, no worse
2249 // than the old isel though. We should sink this into MorphNodeTo.
2250 int OldFlagResultNo = -1, OldChainResultNo = -1;
2251
2252 unsigned NTMNumResults = NodeToMatch->getNumValues();
2253 if (NodeToMatch->getValueType(NTMNumResults-1) == MVT::Flag) {
2254 OldFlagResultNo = NTMNumResults-1;
2255 if (NTMNumResults != 1 &&
2256 NodeToMatch->getValueType(NTMNumResults-2) == MVT::Other)
2257 OldChainResultNo = NTMNumResults-2;
2258 } else if (NodeToMatch->getValueType(NTMNumResults-1) == MVT::Other)
2259 OldChainResultNo = NTMNumResults-1;
2260
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00002261 // FIXME: If this matches multiple nodes it will just leave them here
2262 // dead with noone to love them. These dead nodes can block future
2263 // matches (!).
Chris Lattner2a49d572010-02-28 22:37:22 +00002264 Res = CurDAG->MorphNodeTo(NodeToMatch, ~TargetOpc, VTList,
2265 Ops.data(), Ops.size());
2266
2267 // MorphNodeTo can operate in two ways: if an existing node with the
2268 // specified operands exists, it can just return it. Otherwise, it
2269 // updates the node in place to have the requested operands.
2270 if (Res == NodeToMatch) {
2271 // If we updated the node in place, reset the node ID. To the isel,
2272 // this should be just like a newly allocated machine node.
2273 Res->setNodeId(-1);
2274 }
2275
2276 unsigned ResNumResults = Res->getNumValues();
2277 // Move the flag if needed.
2278 if ((EmitNodeInfo & OPFL_FlagOutput) && OldFlagResultNo != -1 &&
2279 (unsigned)OldFlagResultNo != ResNumResults-1)
2280 CurDAG->ReplaceAllUsesOfValueWith(SDValue(NodeToMatch,
2281 OldFlagResultNo),
2282 SDValue(Res, ResNumResults-1));
2283
2284 if ((EmitNodeInfo & OPFL_FlagOutput) != 0)
2285 --ResNumResults;
2286
2287 // Move the chain reference if needed.
2288 if ((EmitNodeInfo & OPFL_Chain) && OldChainResultNo != -1 &&
2289 (unsigned)OldChainResultNo != ResNumResults-1)
2290 CurDAG->ReplaceAllUsesOfValueWith(SDValue(NodeToMatch,
2291 OldChainResultNo),
2292 SDValue(Res, ResNumResults-1));
2293
2294 if (Res != NodeToMatch) {
2295 // Otherwise, no replacement happened because the node already exists.
2296 CurDAG->ReplaceAllUsesWith(NodeToMatch, Res);
2297 }
2298 }
2299
2300 // If the node had chain/flag results, update our notion of the current
2301 // chain and flag.
2302 if (VTs.back() == MVT::Flag) {
2303 InputFlag = SDValue(Res, VTs.size()-1);
2304 if (EmitNodeInfo & OPFL_Chain)
2305 InputChain = SDValue(Res, VTs.size()-2);
2306 } else if (EmitNodeInfo & OPFL_Chain)
2307 InputChain = SDValue(Res, VTs.size()-1);
2308
2309 // If the OPFL_MemRefs flag is set on this node, slap all of the
2310 // accumulated memrefs onto it.
2311 //
2312 // FIXME: This is vastly incorrect for patterns with multiple outputs
2313 // instructions that access memory and for ComplexPatterns that match
2314 // loads.
2315 if (EmitNodeInfo & OPFL_MemRefs) {
2316 MachineSDNode::mmo_iterator MemRefs =
2317 MF->allocateMemRefsArray(MatchedMemRefs.size());
2318 std::copy(MatchedMemRefs.begin(), MatchedMemRefs.end(), MemRefs);
2319 cast<MachineSDNode>(Res)
2320 ->setMemRefs(MemRefs, MemRefs + MatchedMemRefs.size());
2321 }
2322
2323 DEBUG(errs() << " "
2324 << (Opcode == OPC_MorphNodeTo ? "Morphed" : "Created")
2325 << " node: "; Res->dump(CurDAG); errs() << "\n");
2326
2327 // If this was a MorphNodeTo then we're completely done!
2328 if (Opcode == OPC_MorphNodeTo) {
2329 // Update chain and flag uses.
2330 UpdateChainsAndFlags(NodeToMatch, InputChain, ChainNodesMatched,
2331 InputFlag, FlagResultNodesMatched, true, CurDAG);
2332 return 0;
2333 }
2334
2335 continue;
2336 }
2337
2338 case OPC_MarkFlagResults: {
2339 unsigned NumNodes = MatcherTable[MatcherIndex++];
2340
2341 // Read and remember all the flag-result nodes.
2342 for (unsigned i = 0; i != NumNodes; ++i) {
2343 unsigned RecNo = MatcherTable[MatcherIndex++];
2344 if (RecNo & 128)
2345 RecNo = GetVBR(RecNo, MatcherTable, MatcherIndex);
2346
2347 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2348 FlagResultNodesMatched.push_back(RecordedNodes[RecNo].getNode());
2349 }
2350 continue;
2351 }
2352
2353 case OPC_CompleteMatch: {
2354 // The match has been completed, and any new nodes (if any) have been
2355 // created. Patch up references to the matched dag to use the newly
2356 // created nodes.
2357 unsigned NumResults = MatcherTable[MatcherIndex++];
2358
2359 for (unsigned i = 0; i != NumResults; ++i) {
2360 unsigned ResSlot = MatcherTable[MatcherIndex++];
2361 if (ResSlot & 128)
2362 ResSlot = GetVBR(ResSlot, MatcherTable, MatcherIndex);
2363
2364 assert(ResSlot < RecordedNodes.size() && "Invalid CheckSame");
2365 SDValue Res = RecordedNodes[ResSlot];
2366
2367 // FIXME2: Eliminate this horrible hack by fixing the 'Gen' program
2368 // after (parallel) on input patterns are removed. This would also
2369 // allow us to stop encoding #results in OPC_CompleteMatch's table
2370 // entry.
2371 if (NodeToMatch->getNumValues() <= i ||
2372 NodeToMatch->getValueType(i) == MVT::Other ||
2373 NodeToMatch->getValueType(i) == MVT::Flag)
2374 break;
2375 assert((NodeToMatch->getValueType(i) == Res.getValueType() ||
2376 NodeToMatch->getValueType(i) == MVT::iPTR ||
2377 Res.getValueType() == MVT::iPTR ||
2378 NodeToMatch->getValueType(i).getSizeInBits() ==
2379 Res.getValueType().getSizeInBits()) &&
2380 "invalid replacement");
2381 CurDAG->ReplaceAllUsesOfValueWith(SDValue(NodeToMatch, i), Res);
2382 }
2383
2384 // If the root node defines a flag, add it to the flag nodes to update
2385 // list.
2386 if (NodeToMatch->getValueType(NodeToMatch->getNumValues()-1) == MVT::Flag)
2387 FlagResultNodesMatched.push_back(NodeToMatch);
2388
2389 // Update chain and flag uses.
2390 UpdateChainsAndFlags(NodeToMatch, InputChain, ChainNodesMatched,
2391 InputFlag, FlagResultNodesMatched, false, CurDAG);
2392
2393 assert(NodeToMatch->use_empty() &&
2394 "Didn't replace all uses of the node?");
2395
2396 // FIXME: We just return here, which interacts correctly with SelectRoot
2397 // above. We should fix this to not return an SDNode* anymore.
2398 return 0;
2399 }
2400 }
2401
2402 // If the code reached this point, then the match failed. See if there is
2403 // another child to try in the current 'Scope', otherwise pop it until we
2404 // find a case to check.
2405 while (1) {
2406 if (MatchScopes.empty()) {
2407 CannotYetSelect(NodeToMatch);
2408 return 0;
2409 }
2410
2411 // Restore the interpreter state back to the point where the scope was
2412 // formed.
2413 MatchScope &LastScope = MatchScopes.back();
2414 RecordedNodes.resize(LastScope.NumRecordedNodes);
2415 NodeStack.clear();
2416 NodeStack.append(LastScope.NodeStack.begin(), LastScope.NodeStack.end());
2417 N = NodeStack.back();
2418
2419 DEBUG(errs() << " Match failed at index " << MatcherIndex
2420 << " continuing at " << LastScope.FailIndex << "\n");
2421
2422 if (LastScope.NumMatchedMemRefs != MatchedMemRefs.size())
2423 MatchedMemRefs.resize(LastScope.NumMatchedMemRefs);
2424 MatcherIndex = LastScope.FailIndex;
2425
2426 InputChain = LastScope.InputChain;
2427 InputFlag = LastScope.InputFlag;
2428 if (!LastScope.HasChainNodesMatched)
2429 ChainNodesMatched.clear();
2430 if (!LastScope.HasFlagResultNodesMatched)
2431 FlagResultNodesMatched.clear();
2432
2433 // Check to see what the offset is at the new MatcherIndex. If it is zero
2434 // we have reached the end of this scope, otherwise we have another child
2435 // in the current scope to try.
2436 unsigned NumToSkip = MatcherTable[MatcherIndex++];
2437 if (NumToSkip & 128)
2438 NumToSkip = GetVBR(NumToSkip, MatcherTable, MatcherIndex);
2439
2440 // If we have another child in this scope to match, update FailIndex and
2441 // try it.
2442 if (NumToSkip != 0) {
2443 LastScope.FailIndex = MatcherIndex+NumToSkip;
2444 break;
2445 }
2446
2447 // End of this scope, pop it and try the next child in the containing
2448 // scope.
2449 MatchScopes.pop_back();
2450 }
2451 }
2452}
2453
2454
2455
Dan Gohmaneeb3a002010-01-05 01:24:18 +00002456void SelectionDAGISel::CannotYetSelect(SDNode *N) {
Chris Lattner409ac582010-02-17 06:28:22 +00002457 if (N->getOpcode() == ISD::INTRINSIC_W_CHAIN ||
2458 N->getOpcode() == ISD::INTRINSIC_WO_CHAIN ||
2459 N->getOpcode() == ISD::INTRINSIC_VOID)
2460 return CannotYetSelectIntrinsic(N);
2461
Dan Gohmane1f188f2009-10-29 22:30:23 +00002462 std::string msg;
2463 raw_string_ostream Msg(msg);
2464 Msg << "Cannot yet select: ";
David Greenece6715f2010-01-19 20:37:34 +00002465 N->printrFull(Msg, CurDAG);
Dan Gohmane1f188f2009-10-29 22:30:23 +00002466 llvm_report_error(Msg.str());
2467}
2468
Dan Gohmaneeb3a002010-01-05 01:24:18 +00002469void SelectionDAGISel::CannotYetSelectIntrinsic(SDNode *N) {
David Greene1a053232010-01-05 01:26:11 +00002470 dbgs() << "Cannot yet select: ";
Dan Gohmane1f188f2009-10-29 22:30:23 +00002471 unsigned iid =
Jim Grosbachf4549b02010-01-15 00:36:15 +00002472 cast<ConstantSDNode>(N->getOperand(N->getOperand(0).getValueType() ==
2473 MVT::Other))->getZExtValue();
Dan Gohmane1f188f2009-10-29 22:30:23 +00002474 if (iid < Intrinsic::num_intrinsics)
Jim Grosbachf4549b02010-01-15 00:36:15 +00002475 llvm_report_error("Cannot yet select: intrinsic %" +
2476 Intrinsic::getName((Intrinsic::ID)iid));
Dan Gohmane1f188f2009-10-29 22:30:23 +00002477 else if (const TargetIntrinsicInfo *tii = TM.getIntrinsicInfo())
2478 llvm_report_error(Twine("Cannot yet select: target intrinsic %") +
2479 tii->getName(iid));
2480}
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00002481
Devang Patel19974732007-05-03 01:11:54 +00002482char SelectionDAGISel::ID = 0;