blob: 7b2898dcbfb77c5592d85b6e28f8b70f5b35d5fa [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 Gohman4c3fd9f2010-07-07 16:01:37 +000017#include "llvm/CodeGen/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 Lattner1c08c712005-01-07 07:47:53 +000022#include "llvm/Function.h"
Chris Lattnerce7518c2006-01-26 22:24:51 +000023#include "llvm/InlineAsm.h"
Chris Lattner1c08c712005-01-07 07:47:53 +000024#include "llvm/Instructions.h"
25#include "llvm/Intrinsics.h"
Jim Laskey43970fe2006-03-23 18:06:46 +000026#include "llvm/IntrinsicInst.h"
Chris Lattner75c478a2009-10-27 17:02:08 +000027#include "llvm/LLVMContext.h"
Bill Wendling9af7e9a2010-05-26 19:46:12 +000028#include "llvm/Module.h"
Dan Gohman78eca172008-08-19 22:33:34 +000029#include "llvm/CodeGen/FastISel.h"
Gordon Henriksen5a29c9e2008-08-17 12:56:54 +000030#include "llvm/CodeGen/GCStrategy.h"
Gordon Henriksen5eca0752008-08-17 18:44:35 +000031#include "llvm/CodeGen/GCMetadata.h"
Bill Wendlingb92187a2010-05-14 21:14:32 +000032#include "llvm/CodeGen/MachineFrameInfo.h"
Chris Lattner1c08c712005-01-07 07:47:53 +000033#include "llvm/CodeGen/MachineFunction.h"
Chris Lattner1c08c712005-01-07 07:47:53 +000034#include "llvm/CodeGen/MachineInstrBuilder.h"
Chris Lattner84bc5422007-12-31 04:13:23 +000035#include "llvm/CodeGen/MachineModuleInfo.h"
36#include "llvm/CodeGen/MachineRegisterInfo.h"
Dan Gohmanfc54c552009-01-15 22:18:12 +000037#include "llvm/CodeGen/ScheduleHazardRecognizer.h"
Jim Laskeyeb577ba2006-08-02 12:30:23 +000038#include "llvm/CodeGen/SchedulerRegistry.h"
Chris Lattner1c08c712005-01-07 07:47:53 +000039#include "llvm/CodeGen/SelectionDAG.h"
Dan Gohman6f0d0242008-02-10 18:45:23 +000040#include "llvm/Target/TargetRegisterInfo.h"
Dan Gohmane1f188f2009-10-29 22:30:23 +000041#include "llvm/Target/TargetIntrinsicInfo.h"
Chris Lattner1c08c712005-01-07 07:47:53 +000042#include "llvm/Target/TargetInstrInfo.h"
43#include "llvm/Target/TargetLowering.h"
44#include "llvm/Target/TargetMachine.h"
Vladimir Prus12472912006-05-23 13:43:15 +000045#include "llvm/Target/TargetOptions.h"
Chris Lattnera4f0b3a2006-08-27 12:54:02 +000046#include "llvm/Support/Compiler.h"
Evan Chengdb8d56b2008-06-30 20:45:06 +000047#include "llvm/Support/Debug.h"
Torok Edwin7d696d82009-07-11 13:10:19 +000048#include "llvm/Support/ErrorHandling.h"
Evan Chengdb8d56b2008-06-30 20:45:06 +000049#include "llvm/Support/Timer.h"
Daniel Dunbarce63ffb2009-07-25 00:23:56 +000050#include "llvm/Support/raw_ostream.h"
Chris Lattnerc19ae9d2010-03-04 19:46:56 +000051#include "llvm/ADT/Statistic.h"
Jeff Cohen7e881032006-02-24 02:52:40 +000052#include <algorithm>
Chris Lattner1c08c712005-01-07 07:47:53 +000053using namespace llvm;
54
Chris Lattnerc19ae9d2010-03-04 19:46:56 +000055STATISTIC(NumFastIselFailures, "Number of instructions fast isel failed on");
Chris Lattnerbed993d2010-03-28 19:46:56 +000056STATISTIC(NumDAGIselRetries,"Number of times dag isel has to try another path");
Chris Lattnerc19ae9d2010-03-04 19:46:56 +000057
Chris Lattneread0d882008-06-17 06:09:18 +000058static cl::opt<bool>
Dan Gohman293d5f82008-09-09 22:06:46 +000059EnableFastISelVerbose("fast-isel-verbose", cl::Hidden,
Dan Gohmand659d502008-10-20 21:30:12 +000060 cl::desc("Enable verbose messages in the \"fast\" "
Dan Gohman293d5f82008-09-09 22:06:46 +000061 "instruction selector"));
62static cl::opt<bool>
Dan Gohman4344a5d2008-09-09 23:05:00 +000063EnableFastISelAbort("fast-isel-abort", cl::Hidden,
64 cl::desc("Enable abort calls when \"fast\" instruction fails"));
Chris Lattneread0d882008-06-17 06:09:18 +000065
Chris Lattnerda8abb02005-09-01 18:44:10 +000066#ifndef NDEBUG
Chris Lattner7944d9d2005-01-12 03:41:21 +000067static cl::opt<bool>
Dan Gohman462dc7f2008-07-21 20:00:07 +000068ViewDAGCombine1("view-dag-combine1-dags", cl::Hidden,
69 cl::desc("Pop up a window to show dags before the first "
70 "dag combine pass"));
71static cl::opt<bool>
72ViewLegalizeTypesDAGs("view-legalize-types-dags", cl::Hidden,
73 cl::desc("Pop up a window to show dags before legalize types"));
74static cl::opt<bool>
75ViewLegalizeDAGs("view-legalize-dags", cl::Hidden,
76 cl::desc("Pop up a window to show dags before legalize"));
77static cl::opt<bool>
78ViewDAGCombine2("view-dag-combine2-dags", cl::Hidden,
79 cl::desc("Pop up a window to show dags before the second "
80 "dag combine pass"));
81static cl::opt<bool>
Duncan Sands25cf2272008-11-24 14:53:14 +000082ViewDAGCombineLT("view-dag-combine-lt-dags", cl::Hidden,
83 cl::desc("Pop up a window to show dags before the post legalize types"
84 " dag combine pass"));
85static cl::opt<bool>
Evan Chenga9c20912006-01-21 02:32:06 +000086ViewISelDAGs("view-isel-dags", cl::Hidden,
87 cl::desc("Pop up a window to show isel dags as they are selected"));
88static cl::opt<bool>
89ViewSchedDAGs("view-sched-dags", cl::Hidden,
90 cl::desc("Pop up a window to show sched dags as they are processed"));
Dan Gohman3e1a7ae2007-08-28 20:32:58 +000091static cl::opt<bool>
92ViewSUnitDAGs("view-sunit-dags", cl::Hidden,
Chris Lattner5bab7852008-01-25 17:24:52 +000093 cl::desc("Pop up a window to show SUnit dags after they are processed"));
Chris Lattner7944d9d2005-01-12 03:41:21 +000094#else
Dan Gohman462dc7f2008-07-21 20:00:07 +000095static const bool ViewDAGCombine1 = false,
96 ViewLegalizeTypesDAGs = false, ViewLegalizeDAGs = false,
97 ViewDAGCombine2 = false,
Duncan Sands25cf2272008-11-24 14:53:14 +000098 ViewDAGCombineLT = false,
Dan Gohman462dc7f2008-07-21 20:00:07 +000099 ViewISelDAGs = false, ViewSchedDAGs = false,
100 ViewSUnitDAGs = false;
Chris Lattner7944d9d2005-01-12 03:41:21 +0000101#endif
102
Jim Laskeyeb577ba2006-08-02 12:30:23 +0000103//===---------------------------------------------------------------------===//
104///
105/// RegisterScheduler class - Track the registration of instruction schedulers.
106///
107//===---------------------------------------------------------------------===//
108MachinePassRegistry RegisterScheduler::Registry;
109
110//===---------------------------------------------------------------------===//
111///
112/// ISHeuristic command line option for instruction schedulers.
113///
114//===---------------------------------------------------------------------===//
Dan Gohman844731a2008-05-13 00:00:25 +0000115static cl::opt<RegisterScheduler::FunctionPassCtor, false,
116 RegisterPassParser<RegisterScheduler> >
117ISHeuristic("pre-RA-sched",
118 cl::init(&createDefaultScheduler),
119 cl::desc("Instruction schedulers available (before register"
120 " allocation):"));
Jim Laskey13ec7022006-08-01 14:21:23 +0000121
Dan Gohman844731a2008-05-13 00:00:25 +0000122static RegisterScheduler
Dan Gohmanb8cab922008-10-14 20:25:08 +0000123defaultListDAGScheduler("default", "Best scheduler for the target",
Dan Gohman844731a2008-05-13 00:00:25 +0000124 createDefaultScheduler);
Evan Cheng4ef10862006-01-23 07:01:07 +0000125
Chris Lattner1c08c712005-01-07 07:47:53 +0000126namespace llvm {
127 //===--------------------------------------------------------------------===//
Jim Laskey9373beb2006-08-01 19:14:14 +0000128 /// createDefaultScheduler - This creates an instruction scheduler appropriate
129 /// for the target.
Dan Gohman47ac0f02009-02-11 04:27:20 +0000130 ScheduleDAGSDNodes* createDefaultScheduler(SelectionDAGISel *IS,
Bill Wendling98a366d2009-04-29 23:29:43 +0000131 CodeGenOpt::Level OptLevel) {
Dan Gohmane9530ec2009-01-15 16:58:17 +0000132 const TargetLowering &TLI = IS->getTargetLowering();
133
Bill Wendling98a366d2009-04-29 23:29:43 +0000134 if (OptLevel == CodeGenOpt::None)
Dan Gohmane667e012010-07-16 02:01:19 +0000135 return createSourceListDAGScheduler(IS, OptLevel);
Evan Cheng211ffa12010-05-19 20:19:50 +0000136 if (TLI.getSchedulingPreference() == Sched::Latency)
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000137 return createTDListDAGScheduler(IS, OptLevel);
Evan Cheng15a16de2010-05-20 06:13:19 +0000138 if (TLI.getSchedulingPreference() == Sched::RegPressure)
139 return createBURRListDAGScheduler(IS, OptLevel);
Evan Cheng70017e42010-07-24 00:39:05 +0000140 if (TLI.getSchedulingPreference() == Sched::Hybrid)
141 return createHybridListDAGScheduler(IS, OptLevel);
142 assert(TLI.getSchedulingPreference() == Sched::ILP &&
Evan Cheng211ffa12010-05-19 20:19:50 +0000143 "Unknown sched type!");
Evan Cheng70017e42010-07-24 00:39:05 +0000144 return createILPListDAGScheduler(IS, OptLevel);
Jim Laskey9373beb2006-08-01 19:14:14 +0000145 }
Chris Lattner1c08c712005-01-07 07:47:53 +0000146}
147
Evan Chengff9b3732008-01-30 18:18:23 +0000148// EmitInstrWithCustomInserter - This method should be implemented by targets
Dan Gohman533297b2009-10-29 18:10:34 +0000149// that mark instructions with the 'usesCustomInserter' flag. These
Chris Lattner025c39b2005-08-26 20:54:47 +0000150// instructions are special in various ways, which require special support to
151// insert. The specified MachineInstr is created but not inserted into any
Dan Gohman533297b2009-10-29 18:10:34 +0000152// basic blocks, and this method is called to expand it into a sequence of
153// instructions, potentially also creating new basic blocks and control flow.
154// When new basic blocks are inserted and the edges from MBB to its successors
155// are modified, the method should insert pairs of <OldSucc, NewSucc> into the
156// DenseMap.
Dan Gohmanaf1d8ca2010-05-01 00:01:06 +0000157MachineBasicBlock *
158TargetLowering::EmitInstrWithCustomInserter(MachineInstr *MI,
159 MachineBasicBlock *MBB) const {
Torok Edwinf3689232009-07-12 20:07:01 +0000160#ifndef NDEBUG
David Greene1a053232010-01-05 01:26:11 +0000161 dbgs() << "If a target marks an instruction with "
Dan Gohman533297b2009-10-29 18:10:34 +0000162 "'usesCustomInserter', it must implement "
Torok Edwinf3689232009-07-12 20:07:01 +0000163 "TargetLowering::EmitInstrWithCustomInserter!";
164#endif
Torok Edwinc23197a2009-07-14 16:55:14 +0000165 llvm_unreachable(0);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000166 return 0;
Chris Lattner025c39b2005-08-26 20:54:47 +0000167}
168
Chris Lattner7041ee32005-01-11 05:56:49 +0000169//===----------------------------------------------------------------------===//
170// SelectionDAGISel code
171//===----------------------------------------------------------------------===//
Chris Lattner1c08c712005-01-07 07:47:53 +0000172
Dan Gohmanf0757b02010-04-21 01:34:56 +0000173SelectionDAGISel::SelectionDAGISel(const TargetMachine &tm, CodeGenOpt::Level OL) :
Owen Anderson90c579d2010-08-06 18:33:48 +0000174 MachineFunctionPass(ID), TM(tm), TLI(*tm.getTargetLowering()),
Dan Gohman7c3234c2008-08-27 23:52:12 +0000175 FuncInfo(new FunctionLoweringInfo(TLI)),
Dan Gohman7451d3e2010-05-29 17:03:36 +0000176 CurDAG(new SelectionDAG(tm)),
Dan Gohman55e59c12010-04-19 19:05:59 +0000177 SDB(new SelectionDAGBuilder(*CurDAG, *FuncInfo, OL)),
Dan Gohman7c3234c2008-08-27 23:52:12 +0000178 GFI(),
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000179 OptLevel(OL),
Dan Gohman7c3234c2008-08-27 23:52:12 +0000180 DAGSize(0)
181{}
182
183SelectionDAGISel::~SelectionDAGISel() {
Dan Gohman2048b852009-11-23 18:04:58 +0000184 delete SDB;
Dan Gohman7c3234c2008-08-27 23:52:12 +0000185 delete CurDAG;
186 delete FuncInfo;
187}
188
Chris Lattner495a0b52005-08-17 06:37:43 +0000189void SelectionDAGISel::getAnalysisUsage(AnalysisUsage &AU) const {
Jim Laskeyc7c3f112006-10-16 20:52:31 +0000190 AU.addRequired<AliasAnalysis>();
Dan Gohmana3477fe2009-07-31 23:36:22 +0000191 AU.addPreserved<AliasAnalysis>();
Gordon Henriksen5eca0752008-08-17 18:44:35 +0000192 AU.addRequired<GCModuleInfo>();
Dan Gohmana3477fe2009-07-31 23:36:22 +0000193 AU.addPreserved<GCModuleInfo>();
Dan Gohmanad2afc22009-07-31 18:16:33 +0000194 MachineFunctionPass::getAnalysisUsage(AU);
Chris Lattner495a0b52005-08-17 06:37:43 +0000195}
Chris Lattner1c08c712005-01-07 07:47:53 +0000196
Bill Wendling9af7e9a2010-05-26 19:46:12 +0000197/// FunctionCallsSetJmp - Return true if the function has a call to setjmp or
Bill Wendling8d717c72010-05-26 20:39:00 +0000198/// other function that gcc recognizes as "returning twice". This is used to
199/// limit code-gen optimizations on the machine function.
Bill Wendling804a2312010-05-26 21:53:50 +0000200///
201/// FIXME: Remove after <rdar://problem/8031714> is fixed.
Bill Wendling9af7e9a2010-05-26 19:46:12 +0000202static bool FunctionCallsSetJmp(const Function *F) {
203 const Module *M = F->getParent();
Bill Wendling8d717c72010-05-26 20:39:00 +0000204 static const char *ReturnsTwiceFns[] = {
205 "setjmp",
206 "sigsetjmp",
207 "setjmp_syscall",
208 "savectx",
209 "qsetjmp",
210 "vfork",
211 "getcontext"
212 };
213#define NUM_RETURNS_TWICE_FNS sizeof(ReturnsTwiceFns) / sizeof(const char *)
Bill Wendling9af7e9a2010-05-26 19:46:12 +0000214
Bill Wendling8d717c72010-05-26 20:39:00 +0000215 for (unsigned I = 0; I < NUM_RETURNS_TWICE_FNS; ++I)
216 if (const Function *Callee = M->getFunction(ReturnsTwiceFns[I])) {
217 if (!Callee->use_empty())
218 for (Value::const_use_iterator
219 I = Callee->use_begin(), E = Callee->use_end();
220 I != E; ++I)
Gabor Greif96f1d8e2010-07-22 13:36:47 +0000221 if (const CallInst *CI = dyn_cast<CallInst>(*I))
Bill Wendling8d717c72010-05-26 20:39:00 +0000222 if (CI->getParent()->getParent() == F)
223 return true;
224 }
Bill Wendling9af7e9a2010-05-26 19:46:12 +0000225
226 return false;
Bill Wendling8d717c72010-05-26 20:39:00 +0000227#undef NUM_RETURNS_TWICE_FNS
Bill Wendling9af7e9a2010-05-26 19:46:12 +0000228}
229
Dan Gohmanad2afc22009-07-31 18:16:33 +0000230bool SelectionDAGISel::runOnMachineFunction(MachineFunction &mf) {
Dan Gohman4344a5d2008-09-09 23:05:00 +0000231 // Do some sanity-checking on the command-line options.
232 assert((!EnableFastISelVerbose || EnableFastISel) &&
233 "-fast-isel-verbose requires -fast-isel");
234 assert((!EnableFastISelAbort || EnableFastISel) &&
235 "-fast-isel-abort requires -fast-isel");
236
Dan Gohmanae541aa2010-04-15 04:33:49 +0000237 const Function &Fn = *mf.getFunction();
Dan Gohman8a110532008-09-05 22:59:21 +0000238 const TargetInstrInfo &TII = *TM.getInstrInfo();
239 const TargetRegisterInfo &TRI = *TM.getRegisterInfo();
240
Dan Gohmanc0fb65d2010-04-14 17:02:07 +0000241 MF = &mf;
Dan Gohman79ce2762009-01-15 19:20:50 +0000242 RegInfo = &MF->getRegInfo();
Dan Gohmanc0fb65d2010-04-14 17:02:07 +0000243 AA = &getAnalysis<AliasAnalysis>();
244 GFI = Fn.hasGC() ? &getAnalysis<GCModuleInfo>().getFunctionInfo(Fn) : 0;
245
David Greene1a053232010-01-05 01:26:11 +0000246 DEBUG(dbgs() << "\n\n\n=== " << Fn.getName() << "\n");
Chris Lattner1c08c712005-01-07 07:47:53 +0000247
Chris Lattnerde6e7832010-04-05 06:10:13 +0000248 CurDAG->init(*MF);
Dan Gohman7451d3e2010-05-29 17:03:36 +0000249 FuncInfo->set(Fn, *MF);
Dan Gohman2048b852009-11-23 18:04:58 +0000250 SDB->init(GFI, *AA);
Chris Lattner1c08c712005-01-07 07:47:53 +0000251
Dan Gohman6a732b52010-04-14 20:05:00 +0000252 SelectAllBasicBlocks(Fn);
Misha Brukmanedf128a2005-04-21 22:36:52 +0000253
Dan Gohman1b79a2f2010-05-01 00:33:28 +0000254 // If the first basic block in the function has live ins that need to be
Dan Gohman8a110532008-09-05 22:59:21 +0000255 // copied into vregs, emit the copies into the top of the block before
256 // emitting the code for the block.
Evan Cheng2ad0fcf2010-04-28 23:08:54 +0000257 MachineBasicBlock *EntryMBB = MF->begin();
258 RegInfo->EmitLiveInCopies(EntryMBB, TRI, TII);
259
Devang Patel394427b2010-05-26 20:18:50 +0000260 DenseMap<unsigned, unsigned> LiveInMap;
261 if (!FuncInfo->ArgDbgValues.empty())
262 for (MachineRegisterInfo::livein_iterator LI = RegInfo->livein_begin(),
263 E = RegInfo->livein_end(); LI != E; ++LI)
264 if (LI->second)
265 LiveInMap.insert(std::make_pair(LI->first, LI->second));
266
Evan Cheng2ad0fcf2010-04-28 23:08:54 +0000267 // Insert DBG_VALUE instructions for function arguments to the entry block.
268 for (unsigned i = 0, e = FuncInfo->ArgDbgValues.size(); i != e; ++i) {
269 MachineInstr *MI = FuncInfo->ArgDbgValues[e-i-1];
270 unsigned Reg = MI->getOperand(0).getReg();
271 if (TargetRegisterInfo::isPhysicalRegister(Reg))
272 EntryMBB->insert(EntryMBB->begin(), MI);
273 else {
274 MachineInstr *Def = RegInfo->getVRegDef(Reg);
275 MachineBasicBlock::iterator InsertPos = Def;
Evan Chengf1ced252010-05-04 00:58:39 +0000276 // FIXME: VR def may not be in entry block.
277 Def->getParent()->insert(llvm::next(InsertPos), MI);
Evan Cheng2ad0fcf2010-04-28 23:08:54 +0000278 }
Devang Patel394427b2010-05-26 20:18:50 +0000279
280 // If Reg is live-in then update debug info to track its copy in a vreg.
281 DenseMap<unsigned, unsigned>::iterator LDI = LiveInMap.find(Reg);
282 if (LDI != LiveInMap.end()) {
283 MachineInstr *Def = RegInfo->getVRegDef(LDI->second);
284 MachineBasicBlock::iterator InsertPos = Def;
285 const MDNode *Variable =
286 MI->getOperand(MI->getNumOperands()-1).getMetadata();
287 unsigned Offset = MI->getOperand(1).getImm();
288 // Def is never a terminator here, so it is ok to increment InsertPos.
289 BuildMI(*EntryMBB, ++InsertPos, MI->getDebugLoc(),
290 TII.get(TargetOpcode::DBG_VALUE))
291 .addReg(LDI->second, RegState::Debug)
292 .addImm(Offset).addMetadata(Variable);
293 }
Evan Cheng2ad0fcf2010-04-28 23:08:54 +0000294 }
295
Bill Wendling53f76022010-05-17 23:09:50 +0000296 // Determine if there are any calls in this machine function.
297 MachineFrameInfo *MFI = MF->getFrameInfo();
298 if (!MFI->hasCalls()) {
299 for (MachineFunction::const_iterator
300 I = MF->begin(), E = MF->end(); I != E; ++I) {
301 const MachineBasicBlock *MBB = I;
302 for (MachineBasicBlock::const_iterator
303 II = MBB->begin(), IE = MBB->end(); II != IE; ++II) {
304 const TargetInstrDesc &TID = TM.getInstrInfo()->get(II->getOpcode());
Bill Wendling8886c492010-07-09 19:44:12 +0000305
306 // Operand 1 of an inline asm instruction indicates whether the asm
307 // needs stack or not.
Bill Wendlingc930cbc2010-07-08 22:38:02 +0000308 if ((II->isInlineAsm() && II->getOperand(1).getImm()) ||
309 (TID.isCall() && !TID.isReturn())) {
Bill Wendling53f76022010-05-17 23:09:50 +0000310 MFI->setHasCalls(true);
311 goto done;
312 }
313 }
314 }
315 done:;
316 }
317
Bill Wendling9af7e9a2010-05-26 19:46:12 +0000318 // Determine if there is a call to setjmp in the machine function.
319 MF->setCallsSetJmp(FunctionCallsSetJmp(&Fn));
320
Dan Gohman84023e02010-07-10 09:00:22 +0000321 // Replace forward-declared registers with the registers containing
322 // the desired value.
323 MachineRegisterInfo &MRI = MF->getRegInfo();
324 for (DenseMap<unsigned, unsigned>::iterator
325 I = FuncInfo->RegFixups.begin(), E = FuncInfo->RegFixups.end();
326 I != E; ++I) {
327 unsigned From = I->first;
328 unsigned To = I->second;
329 // If To is also scheduled to be replaced, find what its ultimate
330 // replacement is.
331 for (;;) {
332 DenseMap<unsigned, unsigned>::iterator J =
333 FuncInfo->RegFixups.find(To);
334 if (J == E) break;
335 To = J->second;
336 }
337 // Replace it.
338 MRI.replaceRegWith(From, To);
339 }
340
Evan Cheng2ad0fcf2010-04-28 23:08:54 +0000341 // Release function-specific state. SDB and CurDAG are already cleared
342 // at this point.
343 FuncInfo->clear();
Dan Gohman8a110532008-09-05 22:59:21 +0000344
Chris Lattner1c08c712005-01-07 07:47:53 +0000345 return true;
346}
347
Dan Gohman84023e02010-07-10 09:00:22 +0000348void
349SelectionDAGISel::SelectBasicBlock(BasicBlock::const_iterator Begin,
Dan Gohmana9a33212010-04-20 00:29:35 +0000350 BasicBlock::const_iterator End,
351 bool &HadTailCall) {
Dan Gohman98ca4f22009-08-05 01:29:28 +0000352 // Lower all of the non-terminator instructions. If a call is emitted
Dan Gohmanf89d1dc2010-04-16 05:06:56 +0000353 // as a tail call, cease emitting nodes for this block. Terminators
354 // are handled below.
Dan Gohmanc105a2b2010-04-22 20:55:53 +0000355 for (BasicBlock::const_iterator I = Begin; I != End && !SDB->HasTailCall; ++I)
Dan Gohmanf89d1dc2010-04-16 05:06:56 +0000356 SDB->visit(*I);
Dan Gohmanf350b272008-08-23 02:25:05 +0000357
Chris Lattnera651cf62005-01-17 19:43:36 +0000358 // Make sure the root of the DAG is up-to-date.
Dan Gohman2048b852009-11-23 18:04:58 +0000359 CurDAG->setRoot(SDB->getControlRoot());
Dan Gohman2048b852009-11-23 18:04:58 +0000360 HadTailCall = SDB->HasTailCall;
361 SDB->clear();
Dan Gohman95140a42010-05-01 00:25:44 +0000362
363 // Final step, emit the lowered DAG as machine code.
Dan Gohman84023e02010-07-10 09:00:22 +0000364 CodeGenAndEmitDAG();
Chris Lattner1c08c712005-01-07 07:47:53 +0000365}
366
Dan Gohmanf350b272008-08-23 02:25:05 +0000367void SelectionDAGISel::ComputeLiveOutVRegInfo() {
Chris Lattneread0d882008-06-17 06:09:18 +0000368 SmallPtrSet<SDNode*, 128> VisitedNodes;
369 SmallVector<SDNode*, 128> Worklist;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000370
Gabor Greifba36cb52008-08-28 21:40:38 +0000371 Worklist.push_back(CurDAG->getRoot().getNode());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000372
Chris Lattneread0d882008-06-17 06:09:18 +0000373 APInt Mask;
374 APInt KnownZero;
375 APInt KnownOne;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000376
Benjamin Kramer7b1e2a52010-01-07 17:27:56 +0000377 do {
378 SDNode *N = Worklist.pop_back_val();
Daniel Dunbara279bc32009-09-20 02:20:51 +0000379
Chris Lattneread0d882008-06-17 06:09:18 +0000380 // If we've already seen this node, ignore it.
381 if (!VisitedNodes.insert(N))
382 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000383
Chris Lattneread0d882008-06-17 06:09:18 +0000384 // Otherwise, add all chain operands to the worklist.
385 for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i)
Owen Anderson825b72b2009-08-11 20:47:22 +0000386 if (N->getOperand(i).getValueType() == MVT::Other)
Gabor Greifba36cb52008-08-28 21:40:38 +0000387 Worklist.push_back(N->getOperand(i).getNode());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000388
Chris Lattneread0d882008-06-17 06:09:18 +0000389 // If this is a CopyToReg with a vreg dest, process it.
390 if (N->getOpcode() != ISD::CopyToReg)
391 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000392
Chris Lattneread0d882008-06-17 06:09:18 +0000393 unsigned DestReg = cast<RegisterSDNode>(N->getOperand(1))->getReg();
394 if (!TargetRegisterInfo::isVirtualRegister(DestReg))
395 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000396
Chris Lattneread0d882008-06-17 06:09:18 +0000397 // Ignore non-scalar or non-integer values.
Dan Gohman475871a2008-07-27 21:46:04 +0000398 SDValue Src = N->getOperand(2);
Owen Andersone50ed302009-08-10 22:56:29 +0000399 EVT SrcVT = Src.getValueType();
Chris Lattneread0d882008-06-17 06:09:18 +0000400 if (!SrcVT.isInteger() || SrcVT.isVector())
401 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000402
Dan Gohmanf350b272008-08-23 02:25:05 +0000403 unsigned NumSignBits = CurDAG->ComputeNumSignBits(Src);
Chris Lattneread0d882008-06-17 06:09:18 +0000404 Mask = APInt::getAllOnesValue(SrcVT.getSizeInBits());
Dan Gohmanf350b272008-08-23 02:25:05 +0000405 CurDAG->ComputeMaskedBits(Src, Mask, KnownZero, KnownOne);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000406
Chris Lattneread0d882008-06-17 06:09:18 +0000407 // Only install this information if it tells us something.
408 if (NumSignBits != 1 || KnownZero != 0 || KnownOne != 0) {
409 DestReg -= TargetRegisterInfo::FirstVirtualRegister;
Dan Gohmanf7d6cd42009-08-01 03:51:09 +0000410 if (DestReg >= FuncInfo->LiveOutRegInfo.size())
411 FuncInfo->LiveOutRegInfo.resize(DestReg+1);
412 FunctionLoweringInfo::LiveOutInfo &LOI =
413 FuncInfo->LiveOutRegInfo[DestReg];
Chris Lattneread0d882008-06-17 06:09:18 +0000414 LOI.NumSignBits = NumSignBits;
Dan Gohmana80efce2009-03-27 23:55:04 +0000415 LOI.KnownOne = KnownOne;
416 LOI.KnownZero = KnownZero;
Chris Lattneread0d882008-06-17 06:09:18 +0000417 }
Benjamin Kramer7b1e2a52010-01-07 17:27:56 +0000418 } while (!Worklist.empty());
Chris Lattneread0d882008-06-17 06:09:18 +0000419}
420
Dan Gohman84023e02010-07-10 09:00:22 +0000421void SelectionDAGISel::CodeGenAndEmitDAG() {
Dan Gohman462dc7f2008-07-21 20:00:07 +0000422 std::string GroupName;
423 if (TimePassesIsEnabled)
424 GroupName = "Instruction Selection and Scheduling";
425 std::string BlockName;
426 if (ViewDAGCombine1 || ViewLegalizeTypesDAGs || ViewLegalizeDAGs ||
Duncan Sands25cf2272008-11-24 14:53:14 +0000427 ViewDAGCombine2 || ViewDAGCombineLT || ViewISelDAGs || ViewSchedDAGs ||
428 ViewSUnitDAGs)
Dan Gohmanf7d6cd42009-08-01 03:51:09 +0000429 BlockName = MF->getFunction()->getNameStr() + ":" +
Dan Gohman84023e02010-07-10 09:00:22 +0000430 FuncInfo->MBB->getBasicBlock()->getNameStr();
Dan Gohman462dc7f2008-07-21 20:00:07 +0000431
Dan Gohman927f8662010-06-18 16:00:29 +0000432 DEBUG(dbgs() << "Initial selection DAG:\n"; CurDAG->dump());
Dan Gohman462dc7f2008-07-21 20:00:07 +0000433
Dan Gohmanf350b272008-08-23 02:25:05 +0000434 if (ViewDAGCombine1) CurDAG->viewGraph("dag-combine1 input for " + BlockName);
Dan Gohman417e11b2007-10-08 15:12:17 +0000435
Chris Lattneraf21d552005-10-10 16:47:10 +0000436 // Run the DAG combiner in pre-legalize mode.
Dan Gohman03c3dc72010-06-18 15:56:31 +0000437 {
438 NamedRegionTimer T("DAG Combining 1", GroupName, TimePassesIsEnabled);
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000439 CurDAG->Combine(Unrestricted, *AA, OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000440 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000441
Dan Gohman927f8662010-06-18 16:00:29 +0000442 DEBUG(dbgs() << "Optimized lowered selection DAG:\n"; CurDAG->dump());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000443
Chris Lattner1c08c712005-01-07 07:47:53 +0000444 // Second step, hack on the DAG until it only uses operations and types that
445 // the target supports.
Dan Gohman714efc62009-12-05 17:51:33 +0000446 if (ViewLegalizeTypesDAGs) CurDAG->viewGraph("legalize-types input for " +
447 BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000448
Dan Gohman714efc62009-12-05 17:51:33 +0000449 bool Changed;
Dan Gohman03c3dc72010-06-18 15:56:31 +0000450 {
451 NamedRegionTimer T("Type Legalization", GroupName, TimePassesIsEnabled);
Dan Gohman714efc62009-12-05 17:51:33 +0000452 Changed = CurDAG->LegalizeTypes();
453 }
454
Dan Gohman927f8662010-06-18 16:00:29 +0000455 DEBUG(dbgs() << "Type-legalized selection DAG:\n"; CurDAG->dump());
Dan Gohman714efc62009-12-05 17:51:33 +0000456
457 if (Changed) {
458 if (ViewDAGCombineLT)
459 CurDAG->viewGraph("dag-combine-lt input for " + BlockName);
460
461 // Run the DAG combiner in post-type-legalize mode.
Dan Gohman03c3dc72010-06-18 15:56:31 +0000462 {
463 NamedRegionTimer T("DAG Combining after legalize types", GroupName,
464 TimePassesIsEnabled);
Dan Gohman714efc62009-12-05 17:51:33 +0000465 CurDAG->Combine(NoIllegalTypes, *AA, OptLevel);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000466 }
467
Dan Gohman927f8662010-06-18 16:00:29 +0000468 DEBUG(dbgs() << "Optimized type-legalized selection DAG:\n";
469 CurDAG->dump());
Dan Gohman714efc62009-12-05 17:51:33 +0000470 }
Dan Gohman462dc7f2008-07-21 20:00:07 +0000471
Dan Gohman03c3dc72010-06-18 15:56:31 +0000472 {
473 NamedRegionTimer T("Vector Legalization", GroupName, TimePassesIsEnabled);
Dan Gohman714efc62009-12-05 17:51:33 +0000474 Changed = CurDAG->LegalizeVectors();
475 }
Duncan Sands25cf2272008-11-24 14:53:14 +0000476
Dan Gohman714efc62009-12-05 17:51:33 +0000477 if (Changed) {
Dan Gohman03c3dc72010-06-18 15:56:31 +0000478 {
479 NamedRegionTimer T("Type Legalization 2", GroupName, TimePassesIsEnabled);
Bill Wendling98820072009-12-28 01:51:30 +0000480 CurDAG->LegalizeTypes();
Eli Friedman5c22c802009-05-23 12:35:30 +0000481 }
482
Dan Gohman714efc62009-12-05 17:51:33 +0000483 if (ViewDAGCombineLT)
484 CurDAG->viewGraph("dag-combine-lv input for " + BlockName);
Eli Friedman5c22c802009-05-23 12:35:30 +0000485
Dan Gohman714efc62009-12-05 17:51:33 +0000486 // Run the DAG combiner in post-type-legalize mode.
Dan Gohman03c3dc72010-06-18 15:56:31 +0000487 {
488 NamedRegionTimer T("DAG Combining after legalize vectors", GroupName,
489 TimePassesIsEnabled);
Dan Gohman714efc62009-12-05 17:51:33 +0000490 CurDAG->Combine(NoIllegalOperations, *AA, OptLevel);
Eli Friedman5c22c802009-05-23 12:35:30 +0000491 }
Dan Gohman714efc62009-12-05 17:51:33 +0000492
Dan Gohman927f8662010-06-18 16:00:29 +0000493 DEBUG(dbgs() << "Optimized vector-legalized selection DAG:\n";
494 CurDAG->dump());
Chris Lattner70587ea2008-07-10 23:37:50 +0000495 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000496
Dan Gohmanf350b272008-08-23 02:25:05 +0000497 if (ViewLegalizeDAGs) CurDAG->viewGraph("legalize input for " + BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000498
Dan Gohman03c3dc72010-06-18 15:56:31 +0000499 {
500 NamedRegionTimer T("DAG Legalization", GroupName, TimePassesIsEnabled);
Dan Gohman714efc62009-12-05 17:51:33 +0000501 CurDAG->Legalize(OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000502 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000503
Dan Gohman927f8662010-06-18 16:00:29 +0000504 DEBUG(dbgs() << "Legalized selection DAG:\n"; CurDAG->dump());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000505
Dan Gohmanf350b272008-08-23 02:25:05 +0000506 if (ViewDAGCombine2) CurDAG->viewGraph("dag-combine2 input for " + BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000507
Chris Lattneraf21d552005-10-10 16:47:10 +0000508 // Run the DAG combiner in post-legalize mode.
Dan Gohman03c3dc72010-06-18 15:56:31 +0000509 {
510 NamedRegionTimer T("DAG Combining 2", GroupName, TimePassesIsEnabled);
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000511 CurDAG->Combine(NoIllegalOperations, *AA, OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000512 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000513
Dan Gohman927f8662010-06-18 16:00:29 +0000514 DEBUG(dbgs() << "Optimized legalized selection DAG:\n"; CurDAG->dump());
Dan Gohman417e11b2007-10-08 15:12:17 +0000515
Dan Gohman4e39e9d2010-06-24 14:30:44 +0000516 if (OptLevel != CodeGenOpt::None)
Dan Gohmanf350b272008-08-23 02:25:05 +0000517 ComputeLiveOutVRegInfo();
Evan Cheng552c4a82006-04-28 02:09:19 +0000518
Chris Lattner552186d2010-03-14 19:27:55 +0000519 if (ViewISelDAGs) CurDAG->viewGraph("isel input for " + BlockName);
520
Chris Lattnera33ef482005-03-30 01:10:47 +0000521 // Third, instruction select all of the operations to machine code, adding the
522 // code to the MachineBasicBlock.
Dan Gohman03c3dc72010-06-18 15:56:31 +0000523 {
524 NamedRegionTimer T("Instruction Selection", GroupName, TimePassesIsEnabled);
Chris Lattner7c306da2010-03-02 06:34:30 +0000525 DoInstructionSelection();
Evan Chengebffb662008-07-01 17:59:20 +0000526 }
Evan Chengdb8d56b2008-06-30 20:45:06 +0000527
Dan Gohman927f8662010-06-18 16:00:29 +0000528 DEBUG(dbgs() << "Selected selection DAG:\n"; CurDAG->dump());
Dan Gohman462dc7f2008-07-21 20:00:07 +0000529
Dan Gohmanf350b272008-08-23 02:25:05 +0000530 if (ViewSchedDAGs) CurDAG->viewGraph("scheduler input for " + BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000531
Dan Gohman5e843682008-07-14 18:19:29 +0000532 // Schedule machine code.
Dan Gohman47ac0f02009-02-11 04:27:20 +0000533 ScheduleDAGSDNodes *Scheduler = CreateScheduler();
Dan Gohman03c3dc72010-06-18 15:56:31 +0000534 {
535 NamedRegionTimer T("Instruction Scheduling", GroupName,
536 TimePassesIsEnabled);
Dan Gohman84023e02010-07-10 09:00:22 +0000537 Scheduler->Run(CurDAG, FuncInfo->MBB, FuncInfo->InsertPt);
Dan Gohman5e843682008-07-14 18:19:29 +0000538 }
539
Dan Gohman462dc7f2008-07-21 20:00:07 +0000540 if (ViewSUnitDAGs) Scheduler->viewGraph();
541
Daniel Dunbara279bc32009-09-20 02:20:51 +0000542 // Emit machine code to BB. This can change 'BB' to the last block being
Evan Chengdb8d56b2008-06-30 20:45:06 +0000543 // inserted into.
Dan Gohman03c3dc72010-06-18 15:56:31 +0000544 {
545 NamedRegionTimer T("Instruction Creation", GroupName, TimePassesIsEnabled);
Dan Gohman84023e02010-07-10 09:00:22 +0000546
547 FuncInfo->MBB = Scheduler->EmitSchedule();
548 FuncInfo->InsertPt = Scheduler->InsertPos;
Dan Gohman5e843682008-07-14 18:19:29 +0000549 }
550
551 // Free the scheduler state.
Dan Gohman03c3dc72010-06-18 15:56:31 +0000552 {
553 NamedRegionTimer T("Instruction Scheduling Cleanup", GroupName,
554 TimePassesIsEnabled);
Dan Gohman5e843682008-07-14 18:19:29 +0000555 delete Scheduler;
Evan Chengebffb662008-07-01 17:59:20 +0000556 }
Evan Chengdb8d56b2008-06-30 20:45:06 +0000557
Dan Gohman95140a42010-05-01 00:25:44 +0000558 // Free the SelectionDAG state, now that we're finished with it.
559 CurDAG->clear();
Daniel Dunbara279bc32009-09-20 02:20:51 +0000560}
Chris Lattner1c08c712005-01-07 07:47:53 +0000561
Chris Lattner7c306da2010-03-02 06:34:30 +0000562void SelectionDAGISel::DoInstructionSelection() {
563 DEBUG(errs() << "===== Instruction selection begins:\n");
564
565 PreprocessISelDAG();
566
567 // Select target instructions for the DAG.
568 {
569 // Number all nodes with a topological order and set DAGSize.
570 DAGSize = CurDAG->AssignTopologicalOrder();
571
572 // Create a dummy node (which is not added to allnodes), that adds
573 // a reference to the root node, preventing it from being deleted,
574 // and tracking any changes of the root.
575 HandleSDNode Dummy(CurDAG->getRoot());
576 ISelPosition = SelectionDAG::allnodes_iterator(CurDAG->getRoot().getNode());
577 ++ISelPosition;
578
579 // The AllNodes list is now topological-sorted. Visit the
580 // nodes by starting at the end of the list (the root of the
581 // graph) and preceding back toward the beginning (the entry
582 // node).
583 while (ISelPosition != CurDAG->allnodes_begin()) {
584 SDNode *Node = --ISelPosition;
585 // Skip dead nodes. DAGCombiner is expected to eliminate all dead nodes,
586 // but there are currently some corner cases that it misses. Also, this
587 // makes it theoretically possible to disable the DAGCombiner.
588 if (Node->use_empty())
589 continue;
590
591 SDNode *ResNode = Select(Node);
592
Chris Lattner82dd3d32010-03-02 07:50:03 +0000593 // FIXME: This is pretty gross. 'Select' should be changed to not return
594 // anything at all and this code should be nuked with a tactical strike.
595
Chris Lattner7c306da2010-03-02 06:34:30 +0000596 // If node should not be replaced, continue with the next one.
Chris Lattner82dd3d32010-03-02 07:50:03 +0000597 if (ResNode == Node || Node->getOpcode() == ISD::DELETED_NODE)
Chris Lattner7c306da2010-03-02 06:34:30 +0000598 continue;
599 // Replace node.
600 if (ResNode)
601 ReplaceUses(Node, ResNode);
602
603 // If after the replacement this node is not used any more,
604 // remove this dead node.
605 if (Node->use_empty()) { // Don't delete EntryToken, etc.
606 ISelUpdater ISU(ISelPosition);
607 CurDAG->RemoveDeadNode(Node, &ISU);
608 }
609 }
610
611 CurDAG->setRoot(Dummy.getValue());
612 }
Bill Wendling53f76022010-05-17 23:09:50 +0000613
Chris Lattner7c306da2010-03-02 06:34:30 +0000614 DEBUG(errs() << "===== Instruction selection ends:\n");
615
616 PostprocessISelDAG();
Chris Lattner7c306da2010-03-02 06:34:30 +0000617}
618
Dan Gohman25208642010-04-14 19:53:31 +0000619/// PrepareEHLandingPad - Emit an EH_LABEL, set up live-in registers, and
620/// do other setup for EH landing-pad blocks.
Dan Gohman84023e02010-07-10 09:00:22 +0000621void SelectionDAGISel::PrepareEHLandingPad() {
Dan Gohman25208642010-04-14 19:53:31 +0000622 // Add a label to mark the beginning of the landing pad. Deletion of the
623 // landing pad can thus be detected via the MachineModuleInfo.
Dan Gohman84023e02010-07-10 09:00:22 +0000624 MCSymbol *Label = MF->getMMI().addLandingPad(FuncInfo->MBB);
Dan Gohman25208642010-04-14 19:53:31 +0000625
Dan Gohman55e59c12010-04-19 19:05:59 +0000626 const TargetInstrDesc &II = TM.getInstrInfo()->get(TargetOpcode::EH_LABEL);
Dan Gohman84023e02010-07-10 09:00:22 +0000627 BuildMI(*FuncInfo->MBB, FuncInfo->InsertPt, SDB->getCurDebugLoc(), II)
628 .addSym(Label);
Dan Gohman25208642010-04-14 19:53:31 +0000629
630 // Mark exception register as live in.
631 unsigned Reg = TLI.getExceptionAddressRegister();
Dan Gohman84023e02010-07-10 09:00:22 +0000632 if (Reg) FuncInfo->MBB->addLiveIn(Reg);
Dan Gohman25208642010-04-14 19:53:31 +0000633
634 // Mark exception selector register as live in.
635 Reg = TLI.getExceptionSelectorRegister();
Dan Gohman84023e02010-07-10 09:00:22 +0000636 if (Reg) FuncInfo->MBB->addLiveIn(Reg);
Dan Gohman25208642010-04-14 19:53:31 +0000637
638 // FIXME: Hack around an exception handling flaw (PR1508): the personality
639 // function and list of typeids logically belong to the invoke (or, if you
640 // like, the basic block containing the invoke), and need to be associated
641 // with it in the dwarf exception handling tables. Currently however the
642 // information is provided by an intrinsic (eh.selector) that can be moved
643 // to unexpected places by the optimizers: if the unwind edge is critical,
644 // then breaking it can result in the intrinsics being in the successor of
645 // the landing pad, not the landing pad itself. This results
646 // in exceptions not being caught because no typeids are associated with
647 // the invoke. This may not be the only way things can go wrong, but it
648 // is the only way we try to work around for the moment.
Dan Gohman84023e02010-07-10 09:00:22 +0000649 const BasicBlock *LLVMBB = FuncInfo->MBB->getBasicBlock();
Dan Gohman25208642010-04-14 19:53:31 +0000650 const BranchInst *Br = dyn_cast<BranchInst>(LLVMBB->getTerminator());
651
652 if (Br && Br->isUnconditional()) { // Critical edge?
653 BasicBlock::const_iterator I, E;
654 for (I = LLVMBB->begin(), E = --LLVMBB->end(); I != E; ++I)
655 if (isa<EHSelectorInst>(I))
656 break;
657
658 if (I == E)
659 // No catch info found - try to extract some from the successor.
660 CopyCatchInfo(Br->getSuccessor(0), LLVMBB, &MF->getMMI(), *FuncInfo);
661 }
662}
Chris Lattner7c306da2010-03-02 06:34:30 +0000663
Chris Lattnerbeac75d2010-09-05 02:18:34 +0000664
665
666
667bool SelectionDAGISel::TryToFoldFastISelLoad(const LoadInst *LI,
668 FastISel *FastIS) {
669 // Don't try to fold volatile loads. Target has to deal with alignment
670 // constraints.
671 if (LI->isVolatile()) return false;
672
673 // Figure out which vreg this is going into.
674 unsigned LoadReg = FastIS->getRegForValue(LI);
675 assert(LoadReg && "Load isn't already assigned a vreg? ");
676
677 // Check to see what the uses of this vreg are. If it has no uses, or more
678 // than one use (at the machine instr level) then we can't fold it.
679 MachineRegisterInfo::reg_iterator RI = RegInfo->reg_begin(LoadReg);
680 if (RI == RegInfo->reg_end())
681 return false;
682
683 // See if there is exactly one use of the vreg. If there are multiple uses,
684 // then the instruction got lowered to multiple machine instructions or the
685 // use of the loaded value ended up being multiple operands of the result, in
686 // either case, we can't fold this.
687 MachineRegisterInfo::reg_iterator PostRI = RI; ++PostRI;
688 if (PostRI != RegInfo->reg_end())
689 return false;
690
691 assert(RI.getOperand().isUse() &&
692 "The only use of the vreg must be a use, we haven't emitted the def!");
693
694 // Ask the target to try folding the load.
695 return FastIS->TryToFoldLoad(&*RI, RI.getOperandNo(), LI);
696}
697
698
699
700
Dan Gohman46510a72010-04-15 01:51:59 +0000701void SelectionDAGISel::SelectAllBasicBlocks(const Function &Fn) {
Dan Gohmana43abd12008-09-29 21:55:50 +0000702 // Initialize the Fast-ISel state, if needed.
703 FastISel *FastIS = 0;
704 if (EnableFastISel)
Dan Gohmana4160c32010-07-07 16:29:44 +0000705 FastIS = TLI.createFastISel(*FuncInfo);
Dan Gohmana43abd12008-09-29 21:55:50 +0000706
707 // Iterate over all basic blocks in the function.
Dan Gohman46510a72010-04-15 01:51:59 +0000708 for (Function::const_iterator I = Fn.begin(), E = Fn.end(); I != E; ++I) {
709 const BasicBlock *LLVMBB = &*I;
Dan Gohman84023e02010-07-10 09:00:22 +0000710 FuncInfo->MBB = FuncInfo->MBBMap[LLVMBB];
711 FuncInfo->InsertPt = FuncInfo->MBB->getFirstNonPHI();
Dan Gohmanf350b272008-08-23 02:25:05 +0000712
Dan Gohmanba5be5c2010-04-20 15:00:41 +0000713 BasicBlock::const_iterator const Begin = LLVMBB->getFirstNonPHI();
Dan Gohman46510a72010-04-15 01:51:59 +0000714 BasicBlock::const_iterator const End = LLVMBB->end();
Dan Gohman84023e02010-07-10 09:00:22 +0000715 BasicBlock::const_iterator BI = End;
Dan Gohman5edd3612008-08-28 20:28:56 +0000716
Dan Gohman84023e02010-07-10 09:00:22 +0000717 FuncInfo->InsertPt = FuncInfo->MBB->getFirstNonPHI();
718
719 // Setup an EH landing-pad block.
720 if (FuncInfo->MBB->isLandingPad())
721 PrepareEHLandingPad();
722
Dan Gohmanf5951412010-07-08 01:00:56 +0000723 // Lower any arguments needed in this block if this is the entry block.
724 if (LLVMBB == &Fn.getEntryBlock())
725 LowerArguments(LLVMBB);
726
Dan Gohmanf350b272008-08-23 02:25:05 +0000727 // Before doing SelectionDAG ISel, see if FastISel has been requested.
Dan Gohmand725f042010-05-01 02:44:23 +0000728 if (FastIS) {
Dan Gohman84023e02010-07-10 09:00:22 +0000729 FastIS->startNewBlock();
730
Dan Gohmanf5951412010-07-08 01:00:56 +0000731 // Emit code for any incoming arguments. This must happen before
732 // beginning FastISel on the entry block.
733 if (LLVMBB == &Fn.getEntryBlock()) {
734 CurDAG->setRoot(SDB->getControlRoot());
735 SDB->clear();
Dan Gohman84023e02010-07-10 09:00:22 +0000736 CodeGenAndEmitDAG();
737
738 // If we inserted any instructions at the beginning, make a note of
739 // where they are, so we can be sure to emit subsequent instructions
740 // after them.
741 if (FuncInfo->InsertPt != FuncInfo->MBB->begin())
742 FastIS->setLastLocalValue(llvm::prior(FuncInfo->InsertPt));
743 else
744 FastIS->setLastLocalValue(0);
Dan Gohmanf5951412010-07-08 01:00:56 +0000745 }
Dan Gohman84023e02010-07-10 09:00:22 +0000746
Dan Gohmana43abd12008-09-29 21:55:50 +0000747 // Do FastISel on as many instructions as possible.
Dan Gohman84023e02010-07-10 09:00:22 +0000748 for (; BI != Begin; --BI) {
749 const Instruction *Inst = llvm::prior(BI);
750
751 // If we no longer require this instruction, skip it.
752 if (!Inst->mayWriteToMemory() &&
753 !isa<TerminatorInst>(Inst) &&
754 !isa<DbgInfoIntrinsic>(Inst) &&
755 !FuncInfo->isExportedInst(Inst))
Dan Gohman20d4be12010-07-01 02:58:57 +0000756 continue;
Dan Gohman84023e02010-07-10 09:00:22 +0000757
758 // Bottom-up: reset the insert pos at the top, after any local-value
759 // instructions.
760 FastIS->recomputeInsertPt();
Dan Gohman20d4be12010-07-01 02:58:57 +0000761
Dan Gohman21c14e32010-01-12 04:32:35 +0000762 // Try to select the instruction with FastISel.
Chris Lattnerbeac75d2010-09-05 02:18:34 +0000763 if (FastIS->SelectInstruction(Inst)) {
764 // If fast isel succeeded, check to see if there is a single-use
765 // non-volatile load right before the selected instruction, and see if
766 // the load is used by the instruction. If so, try to fold it.
767 const Instruction *BeforeInst = 0;
768 if (Inst != Begin)
769 BeforeInst = llvm::prior(llvm::prior(BI));
770 if (BeforeInst && isa<LoadInst>(BeforeInst) &&
771 BeforeInst->hasOneUse() && *BeforeInst->use_begin() == Inst &&
772 TryToFoldFastISelLoad(cast<LoadInst>(BeforeInst), FastIS)) {
773 // If we succeeded, don't re-select the load.
774 --BI;
775 }
Dan Gohmana43abd12008-09-29 21:55:50 +0000776 continue;
Chris Lattnerbeac75d2010-09-05 02:18:34 +0000777 }
Dan Gohmana43abd12008-09-29 21:55:50 +0000778
779 // Then handle certain instructions as single-LLVM-Instruction blocks.
Dan Gohman84023e02010-07-10 09:00:22 +0000780 if (isa<CallInst>(Inst)) {
Chris Lattnerc19ae9d2010-03-04 19:46:56 +0000781 ++NumFastIselFailures;
Dan Gohmana43abd12008-09-29 21:55:50 +0000782 if (EnableFastISelVerbose || EnableFastISelAbort) {
David Greene1a053232010-01-05 01:26:11 +0000783 dbgs() << "FastISel missed call: ";
Dan Gohman84023e02010-07-10 09:00:22 +0000784 Inst->dump();
Dan Gohmana43abd12008-09-29 21:55:50 +0000785 }
786
Dan Gohman84023e02010-07-10 09:00:22 +0000787 if (!Inst->getType()->isVoidTy() && !Inst->use_empty()) {
788 unsigned &R = FuncInfo->ValueMap[Inst];
Dan Gohmana43abd12008-09-29 21:55:50 +0000789 if (!R)
Dan Gohman84023e02010-07-10 09:00:22 +0000790 R = FuncInfo->CreateRegs(Inst->getType());
Dan Gohmana43abd12008-09-29 21:55:50 +0000791 }
792
Dan Gohmanb4afb132009-11-20 02:51:26 +0000793 bool HadTailCall = false;
Dan Gohman84023e02010-07-10 09:00:22 +0000794 SelectBasicBlock(Inst, BI, HadTailCall);
Dan Gohmanb4afb132009-11-20 02:51:26 +0000795
796 // If the call was emitted as a tail call, we're done with the block.
797 if (HadTailCall) {
Dan Gohman84023e02010-07-10 09:00:22 +0000798 --BI;
Dan Gohmanb4afb132009-11-20 02:51:26 +0000799 break;
800 }
801
Dan Gohmana43abd12008-09-29 21:55:50 +0000802 continue;
Dan Gohmanf350b272008-08-23 02:25:05 +0000803 }
Dan Gohmana43abd12008-09-29 21:55:50 +0000804
805 // Otherwise, give up on FastISel for the rest of the block.
806 // For now, be a little lenient about non-branch terminators.
Dan Gohman84023e02010-07-10 09:00:22 +0000807 if (!isa<TerminatorInst>(Inst) || isa<BranchInst>(Inst)) {
Chris Lattnerc19ae9d2010-03-04 19:46:56 +0000808 ++NumFastIselFailures;
Dan Gohmana43abd12008-09-29 21:55:50 +0000809 if (EnableFastISelVerbose || EnableFastISelAbort) {
David Greene1a053232010-01-05 01:26:11 +0000810 dbgs() << "FastISel miss: ";
Dan Gohman84023e02010-07-10 09:00:22 +0000811 Inst->dump();
Dan Gohmana43abd12008-09-29 21:55:50 +0000812 }
813 if (EnableFastISelAbort)
814 // The "fast" selector couldn't handle something and bailed.
815 // For the purpose of debugging, just abort.
Torok Edwinc23197a2009-07-14 16:55:14 +0000816 llvm_unreachable("FastISel didn't select the entire block");
Dan Gohmana43abd12008-09-29 21:55:50 +0000817 }
818 break;
Dan Gohmanf350b272008-08-23 02:25:05 +0000819 }
Dan Gohman84023e02010-07-10 09:00:22 +0000820
821 FastIS->recomputeInsertPt();
Dan Gohmanf350b272008-08-23 02:25:05 +0000822 }
823
Dan Gohmand2ff6472008-09-02 20:17:56 +0000824 // Run SelectionDAG instruction selection on the remainder of the block
825 // not handled by FastISel. If FastISel is not run, this is the entire
Dan Gohman3df24e62008-09-03 23:12:08 +0000826 // block.
Dan Gohman84023e02010-07-10 09:00:22 +0000827 bool HadTailCall;
828 SelectBasicBlock(Begin, BI, HadTailCall);
Dan Gohmanf350b272008-08-23 02:25:05 +0000829
Dan Gohman84023e02010-07-10 09:00:22 +0000830 FinishBasicBlock();
Dan Gohman620427d2010-04-22 19:55:20 +0000831 FuncInfo->PHINodesToUpdate.clear();
Evan Cheng39fd6e82008-08-07 00:43:25 +0000832 }
Dan Gohmana43abd12008-09-29 21:55:50 +0000833
834 delete FastIS;
Dan Gohman0e5f1302008-07-07 23:02:41 +0000835}
836
Dan Gohmanfed90b62008-07-28 21:51:04 +0000837void
Dan Gohman84023e02010-07-10 09:00:22 +0000838SelectionDAGISel::FinishBasicBlock() {
Dan Gohmanf350b272008-08-23 02:25:05 +0000839
David Greene1a053232010-01-05 01:26:11 +0000840 DEBUG(dbgs() << "Total amount of phi nodes to update: "
Dan Gohman927f8662010-06-18 16:00:29 +0000841 << FuncInfo->PHINodesToUpdate.size() << "\n";
842 for (unsigned i = 0, e = FuncInfo->PHINodesToUpdate.size(); i != e; ++i)
David Greene1a053232010-01-05 01:26:11 +0000843 dbgs() << "Node " << i << " : ("
Dan Gohman620427d2010-04-22 19:55:20 +0000844 << FuncInfo->PHINodesToUpdate[i].first
845 << ", " << FuncInfo->PHINodesToUpdate[i].second << ")\n");
Daniel Dunbara279bc32009-09-20 02:20:51 +0000846
Chris Lattnera33ef482005-03-30 01:10:47 +0000847 // Next, now that we know what the last MBB the LLVM BB expanded is, update
Chris Lattner1c08c712005-01-07 07:47:53 +0000848 // PHI nodes in successors.
Dan Gohman2048b852009-11-23 18:04:58 +0000849 if (SDB->SwitchCases.empty() &&
850 SDB->JTCases.empty() &&
851 SDB->BitTestCases.empty()) {
Dan Gohman620427d2010-04-22 19:55:20 +0000852 for (unsigned i = 0, e = FuncInfo->PHINodesToUpdate.size(); i != e; ++i) {
853 MachineInstr *PHI = FuncInfo->PHINodesToUpdate[i].first;
Chris Lattner518bb532010-02-09 19:54:29 +0000854 assert(PHI->isPHI() &&
Nate Begemanf15485a2006-03-27 01:32:24 +0000855 "This is not a machine PHI node that we are updating!");
Dan Gohman84023e02010-07-10 09:00:22 +0000856 if (!FuncInfo->MBB->isSuccessor(PHI->getParent()))
Jakob Stoklund Olesen580bba22010-03-05 21:49:10 +0000857 continue;
Dan Gohman620427d2010-04-22 19:55:20 +0000858 PHI->addOperand(
859 MachineOperand::CreateReg(FuncInfo->PHINodesToUpdate[i].second, false));
Dan Gohman84023e02010-07-10 09:00:22 +0000860 PHI->addOperand(MachineOperand::CreateMBB(FuncInfo->MBB));
Nate Begemanf15485a2006-03-27 01:32:24 +0000861 }
862 return;
Chris Lattner1c08c712005-01-07 07:47:53 +0000863 }
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000864
Dan Gohman2048b852009-11-23 18:04:58 +0000865 for (unsigned i = 0, e = SDB->BitTestCases.size(); i != e; ++i) {
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000866 // Lower header first, if it wasn't already lowered
Dan Gohman2048b852009-11-23 18:04:58 +0000867 if (!SDB->BitTestCases[i].Emitted) {
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000868 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman84023e02010-07-10 09:00:22 +0000869 FuncInfo->MBB = SDB->BitTestCases[i].Parent;
870 FuncInfo->InsertPt = FuncInfo->MBB->end();
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000871 // Emit the code
Dan Gohman84023e02010-07-10 09:00:22 +0000872 SDB->visitBitTestHeader(SDB->BitTestCases[i], FuncInfo->MBB);
Dan Gohman2048b852009-11-23 18:04:58 +0000873 CurDAG->setRoot(SDB->getRoot());
Dan Gohman2048b852009-11-23 18:04:58 +0000874 SDB->clear();
Dan Gohman84023e02010-07-10 09:00:22 +0000875 CodeGenAndEmitDAG();
Daniel Dunbara279bc32009-09-20 02:20:51 +0000876 }
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000877
Dan Gohman2048b852009-11-23 18:04:58 +0000878 for (unsigned j = 0, ej = SDB->BitTestCases[i].Cases.size(); j != ej; ++j) {
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000879 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman84023e02010-07-10 09:00:22 +0000880 FuncInfo->MBB = SDB->BitTestCases[i].Cases[j].ThisBB;
881 FuncInfo->InsertPt = FuncInfo->MBB->end();
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000882 // Emit the code
883 if (j+1 != ej)
Dan Gohman2048b852009-11-23 18:04:58 +0000884 SDB->visitBitTestCase(SDB->BitTestCases[i].Cases[j+1].ThisBB,
885 SDB->BitTestCases[i].Reg,
Dan Gohman99be8ae2010-04-19 22:41:47 +0000886 SDB->BitTestCases[i].Cases[j],
Dan Gohman84023e02010-07-10 09:00:22 +0000887 FuncInfo->MBB);
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000888 else
Dan Gohman2048b852009-11-23 18:04:58 +0000889 SDB->visitBitTestCase(SDB->BitTestCases[i].Default,
890 SDB->BitTestCases[i].Reg,
Dan Gohman99be8ae2010-04-19 22:41:47 +0000891 SDB->BitTestCases[i].Cases[j],
Dan Gohman84023e02010-07-10 09:00:22 +0000892 FuncInfo->MBB);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000893
894
Dan Gohman2048b852009-11-23 18:04:58 +0000895 CurDAG->setRoot(SDB->getRoot());
Dan Gohman2048b852009-11-23 18:04:58 +0000896 SDB->clear();
Dan Gohman84023e02010-07-10 09:00:22 +0000897 CodeGenAndEmitDAG();
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000898 }
899
900 // Update PHI Nodes
Dan Gohman620427d2010-04-22 19:55:20 +0000901 for (unsigned pi = 0, pe = FuncInfo->PHINodesToUpdate.size();
902 pi != pe; ++pi) {
903 MachineInstr *PHI = FuncInfo->PHINodesToUpdate[pi].first;
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000904 MachineBasicBlock *PHIBB = PHI->getParent();
Chris Lattner518bb532010-02-09 19:54:29 +0000905 assert(PHI->isPHI() &&
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000906 "This is not a machine PHI node that we are updating!");
907 // This is "default" BB. We have two jumps to it. From "header" BB and
908 // from last "case" BB.
Dan Gohman2048b852009-11-23 18:04:58 +0000909 if (PHIBB == SDB->BitTestCases[i].Default) {
Jim Grosbachf4549b02010-01-15 00:36:15 +0000910 PHI->addOperand(MachineOperand::
Dan Gohman620427d2010-04-22 19:55:20 +0000911 CreateReg(FuncInfo->PHINodesToUpdate[pi].second,
912 false));
Dan Gohman2048b852009-11-23 18:04:58 +0000913 PHI->addOperand(MachineOperand::CreateMBB(SDB->BitTestCases[i].Parent));
Jim Grosbachf4549b02010-01-15 00:36:15 +0000914 PHI->addOperand(MachineOperand::
Dan Gohman620427d2010-04-22 19:55:20 +0000915 CreateReg(FuncInfo->PHINodesToUpdate[pi].second,
916 false));
Dan Gohman2048b852009-11-23 18:04:58 +0000917 PHI->addOperand(MachineOperand::CreateMBB(SDB->BitTestCases[i].Cases.
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000918 back().ThisBB));
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000919 }
920 // One of "cases" BB.
Dan Gohman2048b852009-11-23 18:04:58 +0000921 for (unsigned j = 0, ej = SDB->BitTestCases[i].Cases.size();
Dan Gohman7c3234c2008-08-27 23:52:12 +0000922 j != ej; ++j) {
Dan Gohman2048b852009-11-23 18:04:58 +0000923 MachineBasicBlock* cBB = SDB->BitTestCases[i].Cases[j].ThisBB;
Jakob Stoklund Olesendd437ba2010-01-11 21:02:33 +0000924 if (cBB->isSuccessor(PHIBB)) {
Jim Grosbachf4549b02010-01-15 00:36:15 +0000925 PHI->addOperand(MachineOperand::
Dan Gohman620427d2010-04-22 19:55:20 +0000926 CreateReg(FuncInfo->PHINodesToUpdate[pi].second,
927 false));
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000928 PHI->addOperand(MachineOperand::CreateMBB(cBB));
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000929 }
930 }
931 }
932 }
Dan Gohman2048b852009-11-23 18:04:58 +0000933 SDB->BitTestCases.clear();
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000934
Nate Begeman9453eea2006-04-23 06:26:20 +0000935 // If the JumpTable record is filled in, then we need to emit a jump table.
936 // Updating the PHI nodes is tricky in this case, since we need to determine
937 // whether the PHI is a successor of the range check MBB or the jump table MBB
Dan Gohman2048b852009-11-23 18:04:58 +0000938 for (unsigned i = 0, e = SDB->JTCases.size(); i != e; ++i) {
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +0000939 // Lower header first, if it wasn't already lowered
Dan Gohman2048b852009-11-23 18:04:58 +0000940 if (!SDB->JTCases[i].first.Emitted) {
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +0000941 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman84023e02010-07-10 09:00:22 +0000942 FuncInfo->MBB = SDB->JTCases[i].first.HeaderBB;
943 FuncInfo->InsertPt = FuncInfo->MBB->end();
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +0000944 // Emit the code
Dan Gohman99be8ae2010-04-19 22:41:47 +0000945 SDB->visitJumpTableHeader(SDB->JTCases[i].second, SDB->JTCases[i].first,
Dan Gohman84023e02010-07-10 09:00:22 +0000946 FuncInfo->MBB);
Dan Gohman2048b852009-11-23 18:04:58 +0000947 CurDAG->setRoot(SDB->getRoot());
Dan Gohman2048b852009-11-23 18:04:58 +0000948 SDB->clear();
Dan Gohman84023e02010-07-10 09:00:22 +0000949 CodeGenAndEmitDAG();
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000950 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000951
Nate Begeman37efe672006-04-22 18:53:45 +0000952 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman84023e02010-07-10 09:00:22 +0000953 FuncInfo->MBB = SDB->JTCases[i].second.MBB;
954 FuncInfo->InsertPt = FuncInfo->MBB->end();
Nate Begeman37efe672006-04-22 18:53:45 +0000955 // Emit the code
Dan Gohman2048b852009-11-23 18:04:58 +0000956 SDB->visitJumpTable(SDB->JTCases[i].second);
957 CurDAG->setRoot(SDB->getRoot());
Dan Gohman2048b852009-11-23 18:04:58 +0000958 SDB->clear();
Dan Gohman84023e02010-07-10 09:00:22 +0000959 CodeGenAndEmitDAG();
Daniel Dunbara279bc32009-09-20 02:20:51 +0000960
Nate Begeman37efe672006-04-22 18:53:45 +0000961 // Update PHI Nodes
Dan Gohman620427d2010-04-22 19:55:20 +0000962 for (unsigned pi = 0, pe = FuncInfo->PHINodesToUpdate.size();
963 pi != pe; ++pi) {
964 MachineInstr *PHI = FuncInfo->PHINodesToUpdate[pi].first;
Nate Begeman37efe672006-04-22 18:53:45 +0000965 MachineBasicBlock *PHIBB = PHI->getParent();
Chris Lattner518bb532010-02-09 19:54:29 +0000966 assert(PHI->isPHI() &&
Nate Begeman37efe672006-04-22 18:53:45 +0000967 "This is not a machine PHI node that we are updating!");
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000968 // "default" BB. We can go there only from header BB.
Dan Gohman2048b852009-11-23 18:04:58 +0000969 if (PHIBB == SDB->JTCases[i].second.Default) {
Evan Chengce319102009-09-19 09:51:03 +0000970 PHI->addOperand
Dan Gohman620427d2010-04-22 19:55:20 +0000971 (MachineOperand::CreateReg(FuncInfo->PHINodesToUpdate[pi].second,
972 false));
Evan Chengce319102009-09-19 09:51:03 +0000973 PHI->addOperand
Dan Gohman2048b852009-11-23 18:04:58 +0000974 (MachineOperand::CreateMBB(SDB->JTCases[i].first.HeaderBB));
Nate Begemanf4360a42006-05-03 03:48:02 +0000975 }
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000976 // JT BB. Just iterate over successors here
Dan Gohman84023e02010-07-10 09:00:22 +0000977 if (FuncInfo->MBB->isSuccessor(PHIBB)) {
Evan Chengce319102009-09-19 09:51:03 +0000978 PHI->addOperand
Dan Gohman620427d2010-04-22 19:55:20 +0000979 (MachineOperand::CreateReg(FuncInfo->PHINodesToUpdate[pi].second,
980 false));
Dan Gohman84023e02010-07-10 09:00:22 +0000981 PHI->addOperand(MachineOperand::CreateMBB(FuncInfo->MBB));
Nate Begeman37efe672006-04-22 18:53:45 +0000982 }
983 }
Nate Begeman37efe672006-04-22 18:53:45 +0000984 }
Dan Gohman2048b852009-11-23 18:04:58 +0000985 SDB->JTCases.clear();
Daniel Dunbara279bc32009-09-20 02:20:51 +0000986
Chris Lattnerb2e806e2006-10-22 23:00:53 +0000987 // If the switch block involved a branch to one of the actual successors, we
988 // need to update PHI nodes in that block.
Dan Gohman620427d2010-04-22 19:55:20 +0000989 for (unsigned i = 0, e = FuncInfo->PHINodesToUpdate.size(); i != e; ++i) {
990 MachineInstr *PHI = FuncInfo->PHINodesToUpdate[i].first;
Chris Lattner518bb532010-02-09 19:54:29 +0000991 assert(PHI->isPHI() &&
Chris Lattnerb2e806e2006-10-22 23:00:53 +0000992 "This is not a machine PHI node that we are updating!");
Dan Gohman84023e02010-07-10 09:00:22 +0000993 if (FuncInfo->MBB->isSuccessor(PHI->getParent())) {
Dan Gohman620427d2010-04-22 19:55:20 +0000994 PHI->addOperand(
995 MachineOperand::CreateReg(FuncInfo->PHINodesToUpdate[i].second, false));
Dan Gohman84023e02010-07-10 09:00:22 +0000996 PHI->addOperand(MachineOperand::CreateMBB(FuncInfo->MBB));
Chris Lattnerb2e806e2006-10-22 23:00:53 +0000997 }
998 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000999
Nate Begemanf15485a2006-03-27 01:32:24 +00001000 // If we generated any switch lowering information, build and codegen any
1001 // additional DAGs necessary.
Dan Gohman2048b852009-11-23 18:04:58 +00001002 for (unsigned i = 0, e = SDB->SwitchCases.size(); i != e; ++i) {
Nate Begemanf15485a2006-03-27 01:32:24 +00001003 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman84023e02010-07-10 09:00:22 +00001004 MachineBasicBlock *ThisBB = FuncInfo->MBB = SDB->SwitchCases[i].ThisBB;
1005 FuncInfo->InsertPt = FuncInfo->MBB->end();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001006
Dan Gohman95140a42010-05-01 00:25:44 +00001007 // Determine the unique successors.
1008 SmallVector<MachineBasicBlock *, 2> Succs;
1009 Succs.push_back(SDB->SwitchCases[i].TrueBB);
1010 if (SDB->SwitchCases[i].TrueBB != SDB->SwitchCases[i].FalseBB)
1011 Succs.push_back(SDB->SwitchCases[i].FalseBB);
1012
1013 // Emit the code. Note that this could result in ThisBB being split, so
1014 // we need to check for updates.
Dan Gohman84023e02010-07-10 09:00:22 +00001015 SDB->visitSwitchCase(SDB->SwitchCases[i], FuncInfo->MBB);
Dan Gohman2048b852009-11-23 18:04:58 +00001016 CurDAG->setRoot(SDB->getRoot());
Dan Gohman95140a42010-05-01 00:25:44 +00001017 SDB->clear();
Dan Gohman84023e02010-07-10 09:00:22 +00001018 CodeGenAndEmitDAG();
1019 ThisBB = FuncInfo->MBB;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001020
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001021 // Handle any PHI nodes in successors of this chunk, as if we were coming
1022 // from the original BB before switch expansion. Note that PHI nodes can
1023 // occur multiple times in PHINodesToUpdate. We have to be very careful to
1024 // handle them the right number of times.
Dan Gohman95140a42010-05-01 00:25:44 +00001025 for (unsigned i = 0, e = Succs.size(); i != e; ++i) {
Dan Gohman84023e02010-07-10 09:00:22 +00001026 FuncInfo->MBB = Succs[i];
1027 FuncInfo->InsertPt = FuncInfo->MBB->end();
1028 // FuncInfo->MBB may have been removed from the CFG if a branch was
1029 // constant folded.
1030 if (ThisBB->isSuccessor(FuncInfo->MBB)) {
1031 for (MachineBasicBlock::iterator Phi = FuncInfo->MBB->begin();
1032 Phi != FuncInfo->MBB->end() && Phi->isPHI();
Jakob Stoklund Olesendd437ba2010-01-11 21:02:33 +00001033 ++Phi) {
1034 // This value for this PHI node is recorded in PHINodesToUpdate.
1035 for (unsigned pn = 0; ; ++pn) {
Dan Gohman620427d2010-04-22 19:55:20 +00001036 assert(pn != FuncInfo->PHINodesToUpdate.size() &&
Jakob Stoklund Olesendd437ba2010-01-11 21:02:33 +00001037 "Didn't find PHI entry!");
Dan Gohman620427d2010-04-22 19:55:20 +00001038 if (FuncInfo->PHINodesToUpdate[pn].first == Phi) {
Jakob Stoklund Olesendd437ba2010-01-11 21:02:33 +00001039 Phi->addOperand(MachineOperand::
Dan Gohman620427d2010-04-22 19:55:20 +00001040 CreateReg(FuncInfo->PHINodesToUpdate[pn].second,
Jakob Stoklund Olesendd437ba2010-01-11 21:02:33 +00001041 false));
1042 Phi->addOperand(MachineOperand::CreateMBB(ThisBB));
1043 break;
1044 }
Evan Cheng8be58a12009-09-18 08:26:06 +00001045 }
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001046 }
Nate Begemanf15485a2006-03-27 01:32:24 +00001047 }
1048 }
Chris Lattnera33ef482005-03-30 01:10:47 +00001049 }
Dan Gohman2048b852009-11-23 18:04:58 +00001050 SDB->SwitchCases.clear();
Chris Lattner1c08c712005-01-07 07:47:53 +00001051}
Evan Chenga9c20912006-01-21 02:32:06 +00001052
Jim Laskey13ec7022006-08-01 14:21:23 +00001053
Dan Gohman0a3776d2009-02-06 18:26:51 +00001054/// Create the scheduler. If a specific scheduler was specified
1055/// via the SchedulerRegistry, use it, otherwise select the
1056/// one preferred by the target.
Dan Gohman5e843682008-07-14 18:19:29 +00001057///
Dan Gohman47ac0f02009-02-11 04:27:20 +00001058ScheduleDAGSDNodes *SelectionDAGISel::CreateScheduler() {
Jim Laskeyeb577ba2006-08-02 12:30:23 +00001059 RegisterScheduler::FunctionPassCtor Ctor = RegisterScheduler::getDefault();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001060
Jim Laskey13ec7022006-08-01 14:21:23 +00001061 if (!Ctor) {
Jim Laskeyeb577ba2006-08-02 12:30:23 +00001062 Ctor = ISHeuristic;
Jim Laskey9373beb2006-08-01 19:14:14 +00001063 RegisterScheduler::setDefault(Ctor);
Evan Cheng4ef10862006-01-23 07:01:07 +00001064 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001065
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +00001066 return Ctor(this, OptLevel);
Evan Chenga9c20912006-01-21 02:32:06 +00001067}
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001068
Dan Gohmanfc54c552009-01-15 22:18:12 +00001069ScheduleHazardRecognizer *SelectionDAGISel::CreateTargetHazardRecognizer() {
1070 return new ScheduleHazardRecognizer();
Jim Laskey9ff542f2006-08-01 18:29:48 +00001071}
1072
Chris Lattner75548062006-10-11 03:58:02 +00001073//===----------------------------------------------------------------------===//
1074// Helper functions used by the generated instruction selector.
1075//===----------------------------------------------------------------------===//
1076// Calls to these methods are generated by tblgen.
1077
1078/// CheckAndMask - The isel is trying to match something like (and X, 255). If
1079/// the dag combiner simplified the 255, we still want to match. RHS is the
1080/// actual value in the DAG on the RHS of an AND, and DesiredMaskS is the value
1081/// specified in the .td file (e.g. 255).
Daniel Dunbara279bc32009-09-20 02:20:51 +00001082bool SelectionDAGISel::CheckAndMask(SDValue LHS, ConstantSDNode *RHS,
Dan Gohmandc9b3d02007-07-24 23:00:27 +00001083 int64_t DesiredMaskS) const {
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001084 const APInt &ActualMask = RHS->getAPIntValue();
1085 const APInt &DesiredMask = APInt(LHS.getValueSizeInBits(), DesiredMaskS);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001086
Chris Lattner75548062006-10-11 03:58:02 +00001087 // If the actual mask exactly matches, success!
1088 if (ActualMask == DesiredMask)
1089 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001090
Chris Lattner75548062006-10-11 03:58:02 +00001091 // If the actual AND mask is allowing unallowed bits, this doesn't match.
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001092 if (ActualMask.intersects(~DesiredMask))
Chris Lattner75548062006-10-11 03:58:02 +00001093 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001094
Chris Lattner75548062006-10-11 03:58:02 +00001095 // Otherwise, the DAG Combiner may have proven that the value coming in is
1096 // either already zero or is not demanded. Check for known zero input bits.
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001097 APInt NeededMask = DesiredMask & ~ActualMask;
Dan Gohmanea859be2007-06-22 14:59:07 +00001098 if (CurDAG->MaskedValueIsZero(LHS, NeededMask))
Chris Lattner75548062006-10-11 03:58:02 +00001099 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001100
Chris Lattner75548062006-10-11 03:58:02 +00001101 // TODO: check to see if missing bits are just not demanded.
1102
1103 // Otherwise, this pattern doesn't match.
1104 return false;
1105}
1106
1107/// CheckOrMask - The isel is trying to match something like (or X, 255). If
1108/// the dag combiner simplified the 255, we still want to match. RHS is the
1109/// actual value in the DAG on the RHS of an OR, and DesiredMaskS is the value
1110/// specified in the .td file (e.g. 255).
Daniel Dunbara279bc32009-09-20 02:20:51 +00001111bool SelectionDAGISel::CheckOrMask(SDValue LHS, ConstantSDNode *RHS,
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001112 int64_t DesiredMaskS) const {
1113 const APInt &ActualMask = RHS->getAPIntValue();
1114 const APInt &DesiredMask = APInt(LHS.getValueSizeInBits(), DesiredMaskS);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001115
Chris Lattner75548062006-10-11 03:58:02 +00001116 // If the actual mask exactly matches, success!
1117 if (ActualMask == DesiredMask)
1118 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001119
Chris Lattner75548062006-10-11 03:58:02 +00001120 // If the actual AND mask is allowing unallowed bits, this doesn't match.
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001121 if (ActualMask.intersects(~DesiredMask))
Chris Lattner75548062006-10-11 03:58:02 +00001122 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001123
Chris Lattner75548062006-10-11 03:58:02 +00001124 // Otherwise, the DAG Combiner may have proven that the value coming in is
1125 // either already zero or is not demanded. Check for known zero input bits.
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001126 APInt NeededMask = DesiredMask & ~ActualMask;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001127
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001128 APInt KnownZero, KnownOne;
Dan Gohmanea859be2007-06-22 14:59:07 +00001129 CurDAG->ComputeMaskedBits(LHS, NeededMask, KnownZero, KnownOne);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001130
Chris Lattner75548062006-10-11 03:58:02 +00001131 // If all the missing bits in the or are already known to be set, match!
1132 if ((NeededMask & KnownOne) == NeededMask)
1133 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001134
Chris Lattner75548062006-10-11 03:58:02 +00001135 // TODO: check to see if missing bits are just not demanded.
Daniel Dunbara279bc32009-09-20 02:20:51 +00001136
Chris Lattner75548062006-10-11 03:58:02 +00001137 // Otherwise, this pattern doesn't match.
1138 return false;
1139}
1140
Jim Laskey9ff542f2006-08-01 18:29:48 +00001141
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001142/// SelectInlineAsmMemoryOperands - Calls to this are automatically generated
1143/// by tblgen. Others should not call it.
1144void SelectionDAGISel::
Dan Gohmanf350b272008-08-23 02:25:05 +00001145SelectInlineAsmMemoryOperands(std::vector<SDValue> &Ops) {
Dan Gohman475871a2008-07-27 21:46:04 +00001146 std::vector<SDValue> InOps;
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001147 std::swap(InOps, Ops);
1148
Chris Lattnerdecc2672010-04-07 05:20:54 +00001149 Ops.push_back(InOps[InlineAsm::Op_InputChain]); // 0
1150 Ops.push_back(InOps[InlineAsm::Op_AsmString]); // 1
1151 Ops.push_back(InOps[InlineAsm::Op_MDNode]); // 2, !srcloc
Dale Johannesenf1e309e2010-07-02 20:16:09 +00001152 Ops.push_back(InOps[InlineAsm::Op_IsAlignStack]); // 3
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001153
Chris Lattnerdecc2672010-04-07 05:20:54 +00001154 unsigned i = InlineAsm::Op_FirstOperand, e = InOps.size();
Owen Anderson825b72b2009-08-11 20:47:22 +00001155 if (InOps[e-1].getValueType() == MVT::Flag)
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001156 --e; // Don't process a flag operand if it is here.
Daniel Dunbara279bc32009-09-20 02:20:51 +00001157
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001158 while (i != e) {
Dan Gohmanf5aeb1a2008-09-12 16:56:44 +00001159 unsigned Flags = cast<ConstantSDNode>(InOps[i])->getZExtValue();
Chris Lattnerdecc2672010-04-07 05:20:54 +00001160 if (!InlineAsm::isMemKind(Flags)) {
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001161 // Just skip over this operand, copying the operands verbatim.
Evan Cheng697cbbf2009-03-20 18:03:34 +00001162 Ops.insert(Ops.end(), InOps.begin()+i,
1163 InOps.begin()+i+InlineAsm::getNumOperandRegisters(Flags) + 1);
1164 i += InlineAsm::getNumOperandRegisters(Flags) + 1;
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001165 } else {
Evan Cheng697cbbf2009-03-20 18:03:34 +00001166 assert(InlineAsm::getNumOperandRegisters(Flags) == 1 &&
1167 "Memory operand with multiple values?");
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001168 // Otherwise, this is a memory operand. Ask the target to select it.
Dan Gohman475871a2008-07-27 21:46:04 +00001169 std::vector<SDValue> SelOps;
Chris Lattnerdecc2672010-04-07 05:20:54 +00001170 if (SelectInlineAsmMemoryOperand(InOps[i+1], 'm', SelOps))
Chris Lattner75361b62010-04-07 22:58:41 +00001171 report_fatal_error("Could not match memory address. Inline asm"
Chris Lattner87d677c2010-04-07 23:50:38 +00001172 " failure!");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001173
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001174 // Add this to the output node.
Chris Lattnerdecc2672010-04-07 05:20:54 +00001175 unsigned NewFlags =
1176 InlineAsm::getFlagWord(InlineAsm::Kind_Mem, SelOps.size());
1177 Ops.push_back(CurDAG->getTargetConstant(NewFlags, MVT::i32));
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001178 Ops.insert(Ops.end(), SelOps.begin(), SelOps.end());
1179 i += 2;
1180 }
1181 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001182
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001183 // Add the flag input back if present.
1184 if (e != InOps.size())
1185 Ops.push_back(InOps.back());
1186}
Devang Patel794fd752007-05-01 21:15:47 +00001187
Owen Andersone50ed302009-08-10 22:56:29 +00001188/// findFlagUse - Return use of EVT::Flag value produced by the specified
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001189/// SDNode.
1190///
1191static SDNode *findFlagUse(SDNode *N) {
1192 unsigned FlagResNo = N->getNumValues()-1;
1193 for (SDNode::use_iterator I = N->use_begin(), E = N->use_end(); I != E; ++I) {
1194 SDUse &Use = I.getUse();
1195 if (Use.getResNo() == FlagResNo)
1196 return Use.getUser();
1197 }
1198 return NULL;
1199}
1200
1201/// findNonImmUse - Return true if "Use" is a non-immediate use of "Def".
1202/// This function recursively traverses up the operand chain, ignoring
1203/// certain nodes.
1204static bool findNonImmUse(SDNode *Use, SDNode* Def, SDNode *ImmedUse,
Chris Lattnerd1b73822010-03-02 22:20:06 +00001205 SDNode *Root, SmallPtrSet<SDNode*, 16> &Visited,
1206 bool IgnoreChains) {
Chris Lattnerda244a02010-02-23 19:32:27 +00001207 // The NodeID's are given uniques ID's where a node ID is guaranteed to be
1208 // greater than all of its (recursive) operands. If we scan to a point where
1209 // 'use' is smaller than the node we're scanning for, then we know we will
1210 // never find it.
1211 //
1212 // The Use may be -1 (unassigned) if it is a newly allocated node. This can
1213 // happen because we scan down to newly selected nodes in the case of flag
1214 // uses.
1215 if ((Use->getNodeId() < Def->getNodeId() && Use->getNodeId() != -1))
1216 return false;
1217
1218 // Don't revisit nodes if we already scanned it and didn't fail, we know we
1219 // won't fail if we scan it again.
1220 if (!Visited.insert(Use))
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001221 return false;
1222
1223 for (unsigned i = 0, e = Use->getNumOperands(); i != e; ++i) {
Chris Lattnerd1b73822010-03-02 22:20:06 +00001224 // Ignore chain uses, they are validated by HandleMergeInputChains.
1225 if (Use->getOperand(i).getValueType() == MVT::Other && IgnoreChains)
1226 continue;
1227
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001228 SDNode *N = Use->getOperand(i).getNode();
1229 if (N == Def) {
1230 if (Use == ImmedUse || Use == Root)
1231 continue; // We are not looking for immediate use.
1232 assert(N != Root);
1233 return true;
1234 }
1235
1236 // Traverse up the operand chain.
Chris Lattnerd1b73822010-03-02 22:20:06 +00001237 if (findNonImmUse(N, Def, ImmedUse, Root, Visited, IgnoreChains))
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001238 return true;
1239 }
1240 return false;
1241}
1242
Evan Cheng014bf212010-02-15 19:41:07 +00001243/// IsProfitableToFold - Returns true if it's profitable to fold the specific
1244/// operand node N of U during instruction selection that starts at Root.
1245bool SelectionDAGISel::IsProfitableToFold(SDValue N, SDNode *U,
1246 SDNode *Root) const {
1247 if (OptLevel == CodeGenOpt::None) return false;
1248 return N.hasOneUse();
1249}
1250
1251/// IsLegalToFold - Returns true if the specific operand node N of
1252/// U can be folded during instruction selection that starts at Root.
Chris Lattnerd1b73822010-03-02 22:20:06 +00001253bool SelectionDAGISel::IsLegalToFold(SDValue N, SDNode *U, SDNode *Root,
Dan Gohmand858e902010-04-17 15:26:15 +00001254 CodeGenOpt::Level OptLevel,
1255 bool IgnoreChains) {
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001256 if (OptLevel == CodeGenOpt::None) return false;
1257
1258 // If Root use can somehow reach N through a path that that doesn't contain
1259 // U then folding N would create a cycle. e.g. In the following
1260 // diagram, Root can reach N through X. If N is folded into into Root, then
1261 // X is both a predecessor and a successor of U.
1262 //
1263 // [N*] //
1264 // ^ ^ //
1265 // / \ //
1266 // [U*] [X]? //
1267 // ^ ^ //
1268 // \ / //
1269 // \ / //
1270 // [Root*] //
1271 //
1272 // * indicates nodes to be folded together.
1273 //
1274 // If Root produces a flag, then it gets (even more) interesting. Since it
1275 // will be "glued" together with its flag use in the scheduler, we need to
1276 // check if it might reach N.
1277 //
1278 // [N*] //
1279 // ^ ^ //
1280 // / \ //
1281 // [U*] [X]? //
1282 // ^ ^ //
1283 // \ \ //
1284 // \ | //
1285 // [Root*] | //
1286 // ^ | //
1287 // f | //
1288 // | / //
1289 // [Y] / //
1290 // ^ / //
1291 // f / //
1292 // | / //
1293 // [FU] //
1294 //
1295 // If FU (flag use) indirectly reaches N (the load), and Root folds N
1296 // (call it Fold), then X is a predecessor of FU and a successor of
1297 // Fold. But since Fold and FU are flagged together, this will create
1298 // a cycle in the scheduling graph.
1299
Chris Lattner9bbcd5e2010-03-05 05:49:45 +00001300 // If the node has flags, walk down the graph to the "lowest" node in the
1301 // flagged set.
Owen Andersone50ed302009-08-10 22:56:29 +00001302 EVT VT = Root->getValueType(Root->getNumValues()-1);
Owen Anderson825b72b2009-08-11 20:47:22 +00001303 while (VT == MVT::Flag) {
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001304 SDNode *FU = findFlagUse(Root);
1305 if (FU == NULL)
1306 break;
1307 Root = FU;
1308 VT = Root->getValueType(Root->getNumValues()-1);
Chris Lattner18fdaba2010-03-05 06:19:13 +00001309
1310 // If our query node has a flag result with a use, we've walked up it. If
1311 // the user (which has already been selected) has a chain or indirectly uses
1312 // the chain, our WalkChainUsers predicate will not consider it. Because of
1313 // this, we cannot ignore chains in this predicate.
1314 IgnoreChains = false;
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001315 }
Chris Lattner18fdaba2010-03-05 06:19:13 +00001316
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001317
Chris Lattner9bbcd5e2010-03-05 05:49:45 +00001318 SmallPtrSet<SDNode*, 16> Visited;
1319 return !findNonImmUse(Root, N.getNode(), U, Root, Visited, IgnoreChains);
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001320}
1321
Dan Gohmaneeb3a002010-01-05 01:24:18 +00001322SDNode *SelectionDAGISel::Select_INLINEASM(SDNode *N) {
1323 std::vector<SDValue> Ops(N->op_begin(), N->op_end());
Dan Gohmane1f188f2009-10-29 22:30:23 +00001324 SelectInlineAsmMemoryOperands(Ops);
1325
1326 std::vector<EVT> VTs;
1327 VTs.push_back(MVT::Other);
1328 VTs.push_back(MVT::Flag);
Dan Gohmaneeb3a002010-01-05 01:24:18 +00001329 SDValue New = CurDAG->getNode(ISD::INLINEASM, N->getDebugLoc(),
Dan Gohmane1f188f2009-10-29 22:30:23 +00001330 VTs, &Ops[0], Ops.size());
Chris Lattnerd1b73822010-03-02 22:20:06 +00001331 New->setNodeId(-1);
Dan Gohmane1f188f2009-10-29 22:30:23 +00001332 return New.getNode();
1333}
1334
Dan Gohmaneeb3a002010-01-05 01:24:18 +00001335SDNode *SelectionDAGISel::Select_UNDEF(SDNode *N) {
Chris Lattner518bb532010-02-09 19:54:29 +00001336 return CurDAG->SelectNodeTo(N, TargetOpcode::IMPLICIT_DEF,N->getValueType(0));
Dan Gohmane1f188f2009-10-29 22:30:23 +00001337}
1338
Chris Lattner2a49d572010-02-28 22:37:22 +00001339/// GetVBR - decode a vbr encoding whose top bit is set.
1340ALWAYS_INLINE static uint64_t
1341GetVBR(uint64_t Val, const unsigned char *MatcherTable, unsigned &Idx) {
1342 assert(Val >= 128 && "Not a VBR");
1343 Val &= 127; // Remove first vbr bit.
1344
1345 unsigned Shift = 7;
1346 uint64_t NextBits;
1347 do {
Chris Lattner14df8dc2010-02-28 22:38:43 +00001348 NextBits = MatcherTable[Idx++];
Chris Lattner2a49d572010-02-28 22:37:22 +00001349 Val |= (NextBits&127) << Shift;
1350 Shift += 7;
1351 } while (NextBits & 128);
1352
1353 return Val;
1354}
1355
Chris Lattner2a49d572010-02-28 22:37:22 +00001356
1357/// UpdateChainsAndFlags - When a match is complete, this method updates uses of
1358/// interior flag and chain results to use the new flag and chain results.
Chris Lattner82dd3d32010-03-02 07:50:03 +00001359void SelectionDAGISel::
1360UpdateChainsAndFlags(SDNode *NodeToMatch, SDValue InputChain,
1361 const SmallVectorImpl<SDNode*> &ChainNodesMatched,
1362 SDValue InputFlag,
1363 const SmallVectorImpl<SDNode*> &FlagResultNodesMatched,
1364 bool isMorphNodeTo) {
1365 SmallVector<SDNode*, 4> NowDeadNodes;
1366
1367 ISelUpdater ISU(ISelPosition);
1368
Chris Lattner2a49d572010-02-28 22:37:22 +00001369 // Now that all the normal results are replaced, we replace the chain and
1370 // flag results if present.
1371 if (!ChainNodesMatched.empty()) {
1372 assert(InputChain.getNode() != 0 &&
1373 "Matched input chains but didn't produce a chain");
1374 // Loop over all of the nodes we matched that produced a chain result.
1375 // Replace all the chain results with the final chain we ended up with.
1376 for (unsigned i = 0, e = ChainNodesMatched.size(); i != e; ++i) {
1377 SDNode *ChainNode = ChainNodesMatched[i];
1378
Chris Lattner82dd3d32010-03-02 07:50:03 +00001379 // If this node was already deleted, don't look at it.
1380 if (ChainNode->getOpcode() == ISD::DELETED_NODE)
1381 continue;
1382
Chris Lattner2a49d572010-02-28 22:37:22 +00001383 // Don't replace the results of the root node if we're doing a
1384 // MorphNodeTo.
1385 if (ChainNode == NodeToMatch && isMorphNodeTo)
1386 continue;
1387
1388 SDValue ChainVal = SDValue(ChainNode, ChainNode->getNumValues()-1);
1389 if (ChainVal.getValueType() == MVT::Flag)
1390 ChainVal = ChainVal.getValue(ChainVal->getNumValues()-2);
1391 assert(ChainVal.getValueType() == MVT::Other && "Not a chain?");
Chris Lattner82dd3d32010-03-02 07:50:03 +00001392 CurDAG->ReplaceAllUsesOfValueWith(ChainVal, InputChain, &ISU);
1393
Chris Lattner856fb392010-03-28 05:28:31 +00001394 // If the node became dead and we haven't already seen it, delete it.
1395 if (ChainNode->use_empty() &&
1396 !std::count(NowDeadNodes.begin(), NowDeadNodes.end(), ChainNode))
Chris Lattner82dd3d32010-03-02 07:50:03 +00001397 NowDeadNodes.push_back(ChainNode);
Chris Lattner2a49d572010-02-28 22:37:22 +00001398 }
1399 }
1400
1401 // If the result produces a flag, update any flag results in the matched
1402 // pattern with the flag result.
1403 if (InputFlag.getNode() != 0) {
1404 // Handle any interior nodes explicitly marked.
1405 for (unsigned i = 0, e = FlagResultNodesMatched.size(); i != e; ++i) {
1406 SDNode *FRN = FlagResultNodesMatched[i];
Chris Lattner82dd3d32010-03-02 07:50:03 +00001407
1408 // If this node was already deleted, don't look at it.
1409 if (FRN->getOpcode() == ISD::DELETED_NODE)
1410 continue;
1411
Chris Lattner2a49d572010-02-28 22:37:22 +00001412 assert(FRN->getValueType(FRN->getNumValues()-1) == MVT::Flag &&
1413 "Doesn't have a flag result");
1414 CurDAG->ReplaceAllUsesOfValueWith(SDValue(FRN, FRN->getNumValues()-1),
Chris Lattner82dd3d32010-03-02 07:50:03 +00001415 InputFlag, &ISU);
1416
Chris Lattner19e37cb2010-03-28 04:54:33 +00001417 // If the node became dead and we haven't already seen it, delete it.
1418 if (FRN->use_empty() &&
1419 !std::count(NowDeadNodes.begin(), NowDeadNodes.end(), FRN))
Chris Lattner82dd3d32010-03-02 07:50:03 +00001420 NowDeadNodes.push_back(FRN);
Chris Lattner2a49d572010-02-28 22:37:22 +00001421 }
1422 }
1423
Chris Lattner82dd3d32010-03-02 07:50:03 +00001424 if (!NowDeadNodes.empty())
1425 CurDAG->RemoveDeadNodes(NowDeadNodes, &ISU);
1426
Chris Lattner2a49d572010-02-28 22:37:22 +00001427 DEBUG(errs() << "ISEL: Match complete!\n");
1428}
1429
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001430enum ChainResult {
1431 CR_Simple,
1432 CR_InducesCycle,
1433 CR_LeadsToInteriorNode
1434};
1435
1436/// WalkChainUsers - Walk down the users of the specified chained node that is
1437/// part of the pattern we're matching, looking at all of the users we find.
1438/// This determines whether something is an interior node, whether we have a
1439/// non-pattern node in between two pattern nodes (which prevent folding because
1440/// it would induce a cycle) and whether we have a TokenFactor node sandwiched
1441/// between pattern nodes (in which case the TF becomes part of the pattern).
1442///
1443/// The walk we do here is guaranteed to be small because we quickly get down to
1444/// already selected nodes "below" us.
1445static ChainResult
1446WalkChainUsers(SDNode *ChainedNode,
1447 SmallVectorImpl<SDNode*> &ChainedNodesInPattern,
1448 SmallVectorImpl<SDNode*> &InteriorChainedNodes) {
1449 ChainResult Result = CR_Simple;
1450
1451 for (SDNode::use_iterator UI = ChainedNode->use_begin(),
1452 E = ChainedNode->use_end(); UI != E; ++UI) {
1453 // Make sure the use is of the chain, not some other value we produce.
1454 if (UI.getUse().getValueType() != MVT::Other) continue;
1455
1456 SDNode *User = *UI;
1457
1458 // If we see an already-selected machine node, then we've gone beyond the
1459 // pattern that we're selecting down into the already selected chunk of the
1460 // DAG.
1461 if (User->isMachineOpcode() ||
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001462 User->getOpcode() == ISD::HANDLENODE) // Root of the graph.
1463 continue;
Chris Lattnerd1b73822010-03-02 22:20:06 +00001464
1465 if (User->getOpcode() == ISD::CopyToReg ||
1466 User->getOpcode() == ISD::CopyFromReg ||
Chris Lattner7561d482010-03-14 02:33:54 +00001467 User->getOpcode() == ISD::INLINEASM ||
1468 User->getOpcode() == ISD::EH_LABEL) {
Chris Lattnerd1b73822010-03-02 22:20:06 +00001469 // If their node ID got reset to -1 then they've already been selected.
1470 // Treat them like a MachineOpcode.
1471 if (User->getNodeId() == -1)
1472 continue;
1473 }
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001474
1475 // If we have a TokenFactor, we handle it specially.
1476 if (User->getOpcode() != ISD::TokenFactor) {
1477 // If the node isn't a token factor and isn't part of our pattern, then it
1478 // must be a random chained node in between two nodes we're selecting.
1479 // This happens when we have something like:
1480 // x = load ptr
1481 // call
1482 // y = x+4
1483 // store y -> ptr
1484 // Because we structurally match the load/store as a read/modify/write,
1485 // but the call is chained between them. We cannot fold in this case
1486 // because it would induce a cycle in the graph.
1487 if (!std::count(ChainedNodesInPattern.begin(),
1488 ChainedNodesInPattern.end(), User))
1489 return CR_InducesCycle;
1490
1491 // Otherwise we found a node that is part of our pattern. For example in:
1492 // x = load ptr
1493 // y = x+4
1494 // store y -> ptr
1495 // This would happen when we're scanning down from the load and see the
1496 // store as a user. Record that there is a use of ChainedNode that is
1497 // part of the pattern and keep scanning uses.
1498 Result = CR_LeadsToInteriorNode;
1499 InteriorChainedNodes.push_back(User);
1500 continue;
1501 }
1502
1503 // If we found a TokenFactor, there are two cases to consider: first if the
1504 // TokenFactor is just hanging "below" the pattern we're matching (i.e. no
1505 // uses of the TF are in our pattern) we just want to ignore it. Second,
1506 // the TokenFactor can be sandwiched in between two chained nodes, like so:
1507 // [Load chain]
1508 // ^
1509 // |
1510 // [Load]
1511 // ^ ^
1512 // | \ DAG's like cheese
1513 // / \ do you?
1514 // / |
1515 // [TokenFactor] [Op]
1516 // ^ ^
1517 // | |
1518 // \ /
1519 // \ /
1520 // [Store]
1521 //
1522 // In this case, the TokenFactor becomes part of our match and we rewrite it
1523 // as a new TokenFactor.
1524 //
1525 // To distinguish these two cases, do a recursive walk down the uses.
1526 switch (WalkChainUsers(User, ChainedNodesInPattern, InteriorChainedNodes)) {
1527 case CR_Simple:
1528 // If the uses of the TokenFactor are just already-selected nodes, ignore
1529 // it, it is "below" our pattern.
1530 continue;
1531 case CR_InducesCycle:
1532 // If the uses of the TokenFactor lead to nodes that are not part of our
1533 // pattern that are not selected, folding would turn this into a cycle,
1534 // bail out now.
1535 return CR_InducesCycle;
1536 case CR_LeadsToInteriorNode:
1537 break; // Otherwise, keep processing.
1538 }
1539
1540 // Okay, we know we're in the interesting interior case. The TokenFactor
1541 // is now going to be considered part of the pattern so that we rewrite its
1542 // uses (it may have uses that are not part of the pattern) with the
1543 // ultimate chain result of the generated code. We will also add its chain
1544 // inputs as inputs to the ultimate TokenFactor we create.
1545 Result = CR_LeadsToInteriorNode;
1546 ChainedNodesInPattern.push_back(User);
1547 InteriorChainedNodes.push_back(User);
1548 continue;
1549 }
1550
1551 return Result;
1552}
1553
Chris Lattner6b307922010-03-02 00:00:03 +00001554/// HandleMergeInputChains - This implements the OPC_EmitMergeInputChains
Chris Lattner00592ec2010-03-02 19:34:59 +00001555/// operation for when the pattern matched at least one node with a chains. The
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001556/// input vector contains a list of all of the chained nodes that we match. We
1557/// must determine if this is a valid thing to cover (i.e. matching it won't
1558/// induce cycles in the DAG) and if so, creating a TokenFactor node. that will
1559/// be used as the input node chain for the generated nodes.
Chris Lattner6b307922010-03-02 00:00:03 +00001560static SDValue
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001561HandleMergeInputChains(SmallVectorImpl<SDNode*> &ChainNodesMatched,
Chris Lattner6b307922010-03-02 00:00:03 +00001562 SelectionDAG *CurDAG) {
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001563 // Walk all of the chained nodes we've matched, recursively scanning down the
1564 // users of the chain result. This adds any TokenFactor nodes that are caught
1565 // in between chained nodes to the chained and interior nodes list.
1566 SmallVector<SDNode*, 3> InteriorChainedNodes;
1567 for (unsigned i = 0, e = ChainNodesMatched.size(); i != e; ++i) {
1568 if (WalkChainUsers(ChainNodesMatched[i], ChainNodesMatched,
1569 InteriorChainedNodes) == CR_InducesCycle)
1570 return SDValue(); // Would induce a cycle.
1571 }
Chris Lattner6b307922010-03-02 00:00:03 +00001572
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001573 // Okay, we have walked all the matched nodes and collected TokenFactor nodes
1574 // that we are interested in. Form our input TokenFactor node.
Chris Lattner6b307922010-03-02 00:00:03 +00001575 SmallVector<SDValue, 3> InputChains;
1576 for (unsigned i = 0, e = ChainNodesMatched.size(); i != e; ++i) {
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001577 // Add the input chain of this node to the InputChains list (which will be
1578 // the operands of the generated TokenFactor) if it's not an interior node.
1579 SDNode *N = ChainNodesMatched[i];
1580 if (N->getOpcode() != ISD::TokenFactor) {
1581 if (std::count(InteriorChainedNodes.begin(),InteriorChainedNodes.end(),N))
1582 continue;
1583
1584 // Otherwise, add the input chain.
1585 SDValue InChain = ChainNodesMatched[i]->getOperand(0);
1586 assert(InChain.getValueType() == MVT::Other && "Not a chain");
Chris Lattner6b307922010-03-02 00:00:03 +00001587 InputChains.push_back(InChain);
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001588 continue;
1589 }
1590
1591 // If we have a token factor, we want to add all inputs of the token factor
1592 // that are not part of the pattern we're matching.
1593 for (unsigned op = 0, e = N->getNumOperands(); op != e; ++op) {
1594 if (!std::count(ChainNodesMatched.begin(), ChainNodesMatched.end(),
Chris Lattner6183fbd2010-03-02 02:37:23 +00001595 N->getOperand(op).getNode()))
1596 InputChains.push_back(N->getOperand(op));
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001597 }
Chris Lattner6b307922010-03-02 00:00:03 +00001598 }
1599
1600 SDValue Res;
1601 if (InputChains.size() == 1)
1602 return InputChains[0];
1603 return CurDAG->getNode(ISD::TokenFactor, ChainNodesMatched[0]->getDebugLoc(),
1604 MVT::Other, &InputChains[0], InputChains.size());
1605}
Chris Lattner2a49d572010-02-28 22:37:22 +00001606
Chris Lattner3ff1e4d2010-03-02 06:55:04 +00001607/// MorphNode - Handle morphing a node in place for the selector.
1608SDNode *SelectionDAGISel::
1609MorphNode(SDNode *Node, unsigned TargetOpc, SDVTList VTList,
1610 const SDValue *Ops, unsigned NumOps, unsigned EmitNodeInfo) {
1611 // It is possible we're using MorphNodeTo to replace a node with no
1612 // normal results with one that has a normal result (or we could be
1613 // adding a chain) and the input could have flags and chains as well.
Chris Lattnercaa88702010-03-28 08:43:23 +00001614 // In this case we need to shift the operands down.
Chris Lattner3ff1e4d2010-03-02 06:55:04 +00001615 // FIXME: This is a horrible hack and broken in obscure cases, no worse
Chris Lattnera6f86932010-03-27 18:54:50 +00001616 // than the old isel though.
Chris Lattner3ff1e4d2010-03-02 06:55:04 +00001617 int OldFlagResultNo = -1, OldChainResultNo = -1;
1618
1619 unsigned NTMNumResults = Node->getNumValues();
1620 if (Node->getValueType(NTMNumResults-1) == MVT::Flag) {
1621 OldFlagResultNo = NTMNumResults-1;
1622 if (NTMNumResults != 1 &&
1623 Node->getValueType(NTMNumResults-2) == MVT::Other)
1624 OldChainResultNo = NTMNumResults-2;
1625 } else if (Node->getValueType(NTMNumResults-1) == MVT::Other)
1626 OldChainResultNo = NTMNumResults-1;
1627
Chris Lattner61c97f62010-03-02 07:14:49 +00001628 // Call the underlying SelectionDAG routine to do the transmogrification. Note
1629 // that this deletes operands of the old node that become dead.
Chris Lattner3ff1e4d2010-03-02 06:55:04 +00001630 SDNode *Res = CurDAG->MorphNodeTo(Node, ~TargetOpc, VTList, Ops, NumOps);
1631
1632 // MorphNodeTo can operate in two ways: if an existing node with the
1633 // specified operands exists, it can just return it. Otherwise, it
1634 // updates the node in place to have the requested operands.
1635 if (Res == Node) {
1636 // If we updated the node in place, reset the node ID. To the isel,
1637 // this should be just like a newly allocated machine node.
1638 Res->setNodeId(-1);
1639 }
1640
1641 unsigned ResNumResults = Res->getNumValues();
1642 // Move the flag if needed.
1643 if ((EmitNodeInfo & OPFL_FlagOutput) && OldFlagResultNo != -1 &&
1644 (unsigned)OldFlagResultNo != ResNumResults-1)
1645 CurDAG->ReplaceAllUsesOfValueWith(SDValue(Node, OldFlagResultNo),
1646 SDValue(Res, ResNumResults-1));
1647
1648 if ((EmitNodeInfo & OPFL_FlagOutput) != 0)
Bill Wendlingd9cb7ca2010-07-04 08:58:43 +00001649 --ResNumResults;
Chris Lattner3ff1e4d2010-03-02 06:55:04 +00001650
1651 // Move the chain reference if needed.
1652 if ((EmitNodeInfo & OPFL_Chain) && OldChainResultNo != -1 &&
1653 (unsigned)OldChainResultNo != ResNumResults-1)
1654 CurDAG->ReplaceAllUsesOfValueWith(SDValue(Node, OldChainResultNo),
1655 SDValue(Res, ResNumResults-1));
1656
1657 // Otherwise, no replacement happened because the node already exists. Replace
1658 // Uses of the old node with the new one.
1659 if (Res != Node)
1660 CurDAG->ReplaceAllUsesWith(Node, Res);
1661
1662 return Res;
1663}
1664
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001665/// CheckPatternPredicate - Implements OP_CheckPatternPredicate.
1666ALWAYS_INLINE static bool
Chris Lattnerda828e32010-03-03 07:46:25 +00001667CheckSame(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1668 SDValue N, const SmallVectorImpl<SDValue> &RecordedNodes) {
1669 // Accept if it is exactly the same as a previously recorded node.
1670 unsigned RecNo = MatcherTable[MatcherIndex++];
1671 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
1672 return N == RecordedNodes[RecNo];
1673}
1674
1675/// CheckPatternPredicate - Implements OP_CheckPatternPredicate.
1676ALWAYS_INLINE static bool
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001677CheckPatternPredicate(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1678 SelectionDAGISel &SDISel) {
1679 return SDISel.CheckPatternPredicate(MatcherTable[MatcherIndex++]);
1680}
1681
1682/// CheckNodePredicate - Implements OP_CheckNodePredicate.
1683ALWAYS_INLINE static bool
1684CheckNodePredicate(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1685 SelectionDAGISel &SDISel, SDNode *N) {
1686 return SDISel.CheckNodePredicate(N, MatcherTable[MatcherIndex++]);
1687}
1688
1689ALWAYS_INLINE static bool
1690CheckOpcode(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1691 SDNode *N) {
Chris Lattnerd5d5a3d2010-03-25 06:33:05 +00001692 uint16_t Opc = MatcherTable[MatcherIndex++];
1693 Opc |= (unsigned short)MatcherTable[MatcherIndex++] << 8;
1694 return N->getOpcode() == Opc;
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001695}
1696
1697ALWAYS_INLINE static bool
1698CheckType(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1699 SDValue N, const TargetLowering &TLI) {
1700 MVT::SimpleValueType VT = (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
1701 if (N.getValueType() == VT) return true;
1702
1703 // Handle the case when VT is iPTR.
1704 return VT == MVT::iPTR && N.getValueType() == TLI.getPointerTy();
1705}
1706
1707ALWAYS_INLINE static bool
Chris Lattnerda828e32010-03-03 07:46:25 +00001708CheckChildType(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1709 SDValue N, const TargetLowering &TLI,
1710 unsigned ChildNo) {
1711 if (ChildNo >= N.getNumOperands())
1712 return false; // Match fails if out of range child #.
1713 return ::CheckType(MatcherTable, MatcherIndex, N.getOperand(ChildNo), TLI);
1714}
1715
1716
1717ALWAYS_INLINE static bool
1718CheckCondCode(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1719 SDValue N) {
1720 return cast<CondCodeSDNode>(N)->get() ==
1721 (ISD::CondCode)MatcherTable[MatcherIndex++];
1722}
1723
1724ALWAYS_INLINE static bool
1725CheckValueType(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1726 SDValue N, const TargetLowering &TLI) {
1727 MVT::SimpleValueType VT = (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
1728 if (cast<VTSDNode>(N)->getVT() == VT)
1729 return true;
1730
1731 // Handle the case when VT is iPTR.
1732 return VT == MVT::iPTR && cast<VTSDNode>(N)->getVT() == TLI.getPointerTy();
1733}
1734
1735ALWAYS_INLINE static bool
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001736CheckInteger(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1737 SDValue N) {
1738 int64_t Val = MatcherTable[MatcherIndex++];
1739 if (Val & 128)
1740 Val = GetVBR(Val, MatcherTable, MatcherIndex);
1741
1742 ConstantSDNode *C = dyn_cast<ConstantSDNode>(N);
1743 return C != 0 && C->getSExtValue() == Val;
1744}
1745
Chris Lattnerda828e32010-03-03 07:46:25 +00001746ALWAYS_INLINE static bool
1747CheckAndImm(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1748 SDValue N, SelectionDAGISel &SDISel) {
1749 int64_t Val = MatcherTable[MatcherIndex++];
1750 if (Val & 128)
1751 Val = GetVBR(Val, MatcherTable, MatcherIndex);
1752
1753 if (N->getOpcode() != ISD::AND) return false;
1754
1755 ConstantSDNode *C = dyn_cast<ConstantSDNode>(N->getOperand(1));
1756 return C != 0 && SDISel.CheckAndMask(N.getOperand(0), C, Val);
1757}
1758
1759ALWAYS_INLINE static bool
1760CheckOrImm(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1761 SDValue N, SelectionDAGISel &SDISel) {
1762 int64_t Val = MatcherTable[MatcherIndex++];
1763 if (Val & 128)
1764 Val = GetVBR(Val, MatcherTable, MatcherIndex);
1765
1766 if (N->getOpcode() != ISD::OR) return false;
1767
1768 ConstantSDNode *C = dyn_cast<ConstantSDNode>(N->getOperand(1));
1769 return C != 0 && SDISel.CheckOrMask(N.getOperand(0), C, Val);
1770}
1771
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001772/// IsPredicateKnownToFail - If we know how and can do so without pushing a
1773/// scope, evaluate the current node. If the current predicate is known to
1774/// fail, set Result=true and return anything. If the current predicate is
1775/// known to pass, set Result=false and return the MatcherIndex to continue
1776/// with. If the current predicate is unknown, set Result=false and return the
1777/// MatcherIndex to continue with.
1778static unsigned IsPredicateKnownToFail(const unsigned char *Table,
1779 unsigned Index, SDValue N,
Chris Lattnerda828e32010-03-03 07:46:25 +00001780 bool &Result, SelectionDAGISel &SDISel,
1781 SmallVectorImpl<SDValue> &RecordedNodes){
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001782 switch (Table[Index++]) {
1783 default:
1784 Result = false;
1785 return Index-1; // Could not evaluate this predicate.
Chris Lattnerda828e32010-03-03 07:46:25 +00001786 case SelectionDAGISel::OPC_CheckSame:
1787 Result = !::CheckSame(Table, Index, N, RecordedNodes);
1788 return Index;
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001789 case SelectionDAGISel::OPC_CheckPatternPredicate:
Chris Lattnerda828e32010-03-03 07:46:25 +00001790 Result = !::CheckPatternPredicate(Table, Index, SDISel);
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001791 return Index;
1792 case SelectionDAGISel::OPC_CheckPredicate:
Chris Lattnerda828e32010-03-03 07:46:25 +00001793 Result = !::CheckNodePredicate(Table, Index, SDISel, N.getNode());
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001794 return Index;
1795 case SelectionDAGISel::OPC_CheckOpcode:
1796 Result = !::CheckOpcode(Table, Index, N.getNode());
1797 return Index;
1798 case SelectionDAGISel::OPC_CheckType:
1799 Result = !::CheckType(Table, Index, N, SDISel.TLI);
1800 return Index;
1801 case SelectionDAGISel::OPC_CheckChild0Type:
1802 case SelectionDAGISel::OPC_CheckChild1Type:
1803 case SelectionDAGISel::OPC_CheckChild2Type:
1804 case SelectionDAGISel::OPC_CheckChild3Type:
1805 case SelectionDAGISel::OPC_CheckChild4Type:
1806 case SelectionDAGISel::OPC_CheckChild5Type:
1807 case SelectionDAGISel::OPC_CheckChild6Type:
Chris Lattnerda828e32010-03-03 07:46:25 +00001808 case SelectionDAGISel::OPC_CheckChild7Type:
1809 Result = !::CheckChildType(Table, Index, N, SDISel.TLI,
1810 Table[Index-1] - SelectionDAGISel::OPC_CheckChild0Type);
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001811 return Index;
Chris Lattnerda828e32010-03-03 07:46:25 +00001812 case SelectionDAGISel::OPC_CheckCondCode:
1813 Result = !::CheckCondCode(Table, Index, N);
1814 return Index;
1815 case SelectionDAGISel::OPC_CheckValueType:
1816 Result = !::CheckValueType(Table, Index, N, SDISel.TLI);
1817 return Index;
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001818 case SelectionDAGISel::OPC_CheckInteger:
1819 Result = !::CheckInteger(Table, Index, N);
1820 return Index;
Chris Lattnerda828e32010-03-03 07:46:25 +00001821 case SelectionDAGISel::OPC_CheckAndImm:
1822 Result = !::CheckAndImm(Table, Index, N, SDISel);
1823 return Index;
1824 case SelectionDAGISel::OPC_CheckOrImm:
1825 Result = !::CheckOrImm(Table, Index, N, SDISel);
1826 return Index;
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001827 }
1828}
1829
Dan Gohmanb3579832010-04-15 17:08:50 +00001830namespace {
Chris Lattner3ff1e4d2010-03-02 06:55:04 +00001831
Chris Lattner2a49d572010-02-28 22:37:22 +00001832struct MatchScope {
1833 /// FailIndex - If this match fails, this is the index to continue with.
1834 unsigned FailIndex;
1835
1836 /// NodeStack - The node stack when the scope was formed.
1837 SmallVector<SDValue, 4> NodeStack;
1838
1839 /// NumRecordedNodes - The number of recorded nodes when the scope was formed.
1840 unsigned NumRecordedNodes;
1841
1842 /// NumMatchedMemRefs - The number of matched memref entries.
1843 unsigned NumMatchedMemRefs;
1844
1845 /// InputChain/InputFlag - The current chain/flag
1846 SDValue InputChain, InputFlag;
1847
1848 /// HasChainNodesMatched - True if the ChainNodesMatched list is non-empty.
1849 bool HasChainNodesMatched, HasFlagResultNodesMatched;
1850};
1851
Dan Gohmanb3579832010-04-15 17:08:50 +00001852}
1853
Chris Lattner2a49d572010-02-28 22:37:22 +00001854SDNode *SelectionDAGISel::
1855SelectCodeCommon(SDNode *NodeToMatch, const unsigned char *MatcherTable,
1856 unsigned TableSize) {
1857 // FIXME: Should these even be selected? Handle these cases in the caller?
1858 switch (NodeToMatch->getOpcode()) {
1859 default:
1860 break;
1861 case ISD::EntryToken: // These nodes remain the same.
1862 case ISD::BasicBlock:
1863 case ISD::Register:
Chris Lattner3b9d6212010-03-14 17:53:23 +00001864 //case ISD::VALUETYPE:
1865 //case ISD::CONDCODE:
Chris Lattner2a49d572010-02-28 22:37:22 +00001866 case ISD::HANDLENODE:
Chris Lattnerdecc2672010-04-07 05:20:54 +00001867 case ISD::MDNODE_SDNODE:
Chris Lattner2a49d572010-02-28 22:37:22 +00001868 case ISD::TargetConstant:
1869 case ISD::TargetConstantFP:
1870 case ISD::TargetConstantPool:
1871 case ISD::TargetFrameIndex:
1872 case ISD::TargetExternalSymbol:
1873 case ISD::TargetBlockAddress:
1874 case ISD::TargetJumpTable:
1875 case ISD::TargetGlobalTLSAddress:
1876 case ISD::TargetGlobalAddress:
1877 case ISD::TokenFactor:
1878 case ISD::CopyFromReg:
1879 case ISD::CopyToReg:
Chris Lattner7561d482010-03-14 02:33:54 +00001880 case ISD::EH_LABEL:
Chris Lattnerd1b73822010-03-02 22:20:06 +00001881 NodeToMatch->setNodeId(-1); // Mark selected.
Chris Lattner2a49d572010-02-28 22:37:22 +00001882 return 0;
1883 case ISD::AssertSext:
1884 case ISD::AssertZext:
1885 CurDAG->ReplaceAllUsesOfValueWith(SDValue(NodeToMatch, 0),
1886 NodeToMatch->getOperand(0));
1887 return 0;
1888 case ISD::INLINEASM: return Select_INLINEASM(NodeToMatch);
Chris Lattner2a49d572010-02-28 22:37:22 +00001889 case ISD::UNDEF: return Select_UNDEF(NodeToMatch);
1890 }
1891
1892 assert(!NodeToMatch->isMachineOpcode() && "Node already selected!");
1893
1894 // Set up the node stack with NodeToMatch as the only node on the stack.
1895 SmallVector<SDValue, 8> NodeStack;
1896 SDValue N = SDValue(NodeToMatch, 0);
1897 NodeStack.push_back(N);
1898
1899 // MatchScopes - Scopes used when matching, if a match failure happens, this
1900 // indicates where to continue checking.
1901 SmallVector<MatchScope, 8> MatchScopes;
1902
1903 // RecordedNodes - This is the set of nodes that have been recorded by the
1904 // state machine.
1905 SmallVector<SDValue, 8> RecordedNodes;
1906
1907 // MatchedMemRefs - This is the set of MemRef's we've seen in the input
1908 // pattern.
1909 SmallVector<MachineMemOperand*, 2> MatchedMemRefs;
1910
1911 // These are the current input chain and flag for use when generating nodes.
1912 // Various Emit operations change these. For example, emitting a copytoreg
1913 // uses and updates these.
1914 SDValue InputChain, InputFlag;
1915
1916 // ChainNodesMatched - If a pattern matches nodes that have input/output
1917 // chains, the OPC_EmitMergeInputChains operation is emitted which indicates
1918 // which ones they are. The result is captured into this list so that we can
1919 // update the chain results when the pattern is complete.
1920 SmallVector<SDNode*, 3> ChainNodesMatched;
1921 SmallVector<SDNode*, 3> FlagResultNodesMatched;
1922
1923 DEBUG(errs() << "ISEL: Starting pattern match on root node: ";
1924 NodeToMatch->dump(CurDAG);
1925 errs() << '\n');
1926
Chris Lattner7390eeb2010-03-01 18:47:11 +00001927 // Determine where to start the interpreter. Normally we start at opcode #0,
1928 // but if the state machine starts with an OPC_SwitchOpcode, then we
1929 // accelerate the first lookup (which is guaranteed to be hot) with the
1930 // OpcodeOffset table.
Chris Lattner2a49d572010-02-28 22:37:22 +00001931 unsigned MatcherIndex = 0;
Chris Lattner7390eeb2010-03-01 18:47:11 +00001932
1933 if (!OpcodeOffset.empty()) {
1934 // Already computed the OpcodeOffset table, just index into it.
1935 if (N.getOpcode() < OpcodeOffset.size())
1936 MatcherIndex = OpcodeOffset[N.getOpcode()];
1937 DEBUG(errs() << " Initial Opcode index to " << MatcherIndex << "\n");
1938
1939 } else if (MatcherTable[0] == OPC_SwitchOpcode) {
1940 // Otherwise, the table isn't computed, but the state machine does start
1941 // with an OPC_SwitchOpcode instruction. Populate the table now, since this
1942 // is the first time we're selecting an instruction.
1943 unsigned Idx = 1;
1944 while (1) {
1945 // Get the size of this case.
1946 unsigned CaseSize = MatcherTable[Idx++];
1947 if (CaseSize & 128)
1948 CaseSize = GetVBR(CaseSize, MatcherTable, Idx);
1949 if (CaseSize == 0) break;
1950
1951 // Get the opcode, add the index to the table.
Chris Lattnerd5d5a3d2010-03-25 06:33:05 +00001952 uint16_t Opc = MatcherTable[Idx++];
1953 Opc |= (unsigned short)MatcherTable[Idx++] << 8;
Chris Lattner7390eeb2010-03-01 18:47:11 +00001954 if (Opc >= OpcodeOffset.size())
1955 OpcodeOffset.resize((Opc+1)*2);
1956 OpcodeOffset[Opc] = Idx;
1957 Idx += CaseSize;
1958 }
1959
1960 // Okay, do the lookup for the first opcode.
1961 if (N.getOpcode() < OpcodeOffset.size())
1962 MatcherIndex = OpcodeOffset[N.getOpcode()];
1963 }
1964
Chris Lattner2a49d572010-02-28 22:37:22 +00001965 while (1) {
1966 assert(MatcherIndex < TableSize && "Invalid index");
Dan Gohman19b38262010-03-09 02:15:05 +00001967#ifndef NDEBUG
1968 unsigned CurrentOpcodeIndex = MatcherIndex;
1969#endif
Chris Lattner2a49d572010-02-28 22:37:22 +00001970 BuiltinOpcodes Opcode = (BuiltinOpcodes)MatcherTable[MatcherIndex++];
1971 switch (Opcode) {
1972 case OPC_Scope: {
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001973 // Okay, the semantics of this operation are that we should push a scope
1974 // then evaluate the first child. However, pushing a scope only to have
1975 // the first check fail (which then pops it) is inefficient. If we can
1976 // determine immediately that the first check (or first several) will
1977 // immediately fail, don't even bother pushing a scope for them.
1978 unsigned FailIndex;
1979
1980 while (1) {
1981 unsigned NumToSkip = MatcherTable[MatcherIndex++];
1982 if (NumToSkip & 128)
1983 NumToSkip = GetVBR(NumToSkip, MatcherTable, MatcherIndex);
1984 // Found the end of the scope with no match.
1985 if (NumToSkip == 0) {
1986 FailIndex = 0;
1987 break;
1988 }
1989
1990 FailIndex = MatcherIndex+NumToSkip;
1991
Chris Lattnera6f86932010-03-27 18:54:50 +00001992 unsigned MatcherIndexOfPredicate = MatcherIndex;
1993 (void)MatcherIndexOfPredicate; // silence warning.
1994
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001995 // If we can't evaluate this predicate without pushing a scope (e.g. if
1996 // it is a 'MoveParent') or if the predicate succeeds on this node, we
1997 // push the scope and evaluate the full predicate chain.
1998 bool Result;
1999 MatcherIndex = IsPredicateKnownToFail(MatcherTable, MatcherIndex, N,
Chris Lattnerda828e32010-03-03 07:46:25 +00002000 Result, *this, RecordedNodes);
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002001 if (!Result)
2002 break;
2003
Chris Lattnera6f86932010-03-27 18:54:50 +00002004 DEBUG(errs() << " Skipped scope entry (due to false predicate) at "
2005 << "index " << MatcherIndexOfPredicate
2006 << ", continuing at " << FailIndex << "\n");
Chris Lattnerbed993d2010-03-28 19:46:56 +00002007 ++NumDAGIselRetries;
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002008
2009 // Otherwise, we know that this case of the Scope is guaranteed to fail,
2010 // move to the next case.
2011 MatcherIndex = FailIndex;
2012 }
2013
2014 // If the whole scope failed to match, bail.
2015 if (FailIndex == 0) break;
2016
Chris Lattner2a49d572010-02-28 22:37:22 +00002017 // Push a MatchScope which indicates where to go if the first child fails
2018 // to match.
2019 MatchScope NewEntry;
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002020 NewEntry.FailIndex = FailIndex;
Chris Lattner2a49d572010-02-28 22:37:22 +00002021 NewEntry.NodeStack.append(NodeStack.begin(), NodeStack.end());
2022 NewEntry.NumRecordedNodes = RecordedNodes.size();
2023 NewEntry.NumMatchedMemRefs = MatchedMemRefs.size();
2024 NewEntry.InputChain = InputChain;
2025 NewEntry.InputFlag = InputFlag;
2026 NewEntry.HasChainNodesMatched = !ChainNodesMatched.empty();
2027 NewEntry.HasFlagResultNodesMatched = !FlagResultNodesMatched.empty();
2028 MatchScopes.push_back(NewEntry);
2029 continue;
2030 }
2031 case OPC_RecordNode:
2032 // Remember this node, it may end up being an operand in the pattern.
2033 RecordedNodes.push_back(N);
2034 continue;
2035
2036 case OPC_RecordChild0: case OPC_RecordChild1:
2037 case OPC_RecordChild2: case OPC_RecordChild3:
2038 case OPC_RecordChild4: case OPC_RecordChild5:
2039 case OPC_RecordChild6: case OPC_RecordChild7: {
2040 unsigned ChildNo = Opcode-OPC_RecordChild0;
2041 if (ChildNo >= N.getNumOperands())
2042 break; // Match fails if out of range child #.
2043
2044 RecordedNodes.push_back(N->getOperand(ChildNo));
2045 continue;
2046 }
2047 case OPC_RecordMemRef:
2048 MatchedMemRefs.push_back(cast<MemSDNode>(N)->getMemOperand());
2049 continue;
2050
2051 case OPC_CaptureFlagInput:
2052 // If the current node has an input flag, capture it in InputFlag.
2053 if (N->getNumOperands() != 0 &&
2054 N->getOperand(N->getNumOperands()-1).getValueType() == MVT::Flag)
2055 InputFlag = N->getOperand(N->getNumOperands()-1);
2056 continue;
2057
2058 case OPC_MoveChild: {
2059 unsigned ChildNo = MatcherTable[MatcherIndex++];
2060 if (ChildNo >= N.getNumOperands())
2061 break; // Match fails if out of range child #.
2062 N = N.getOperand(ChildNo);
2063 NodeStack.push_back(N);
2064 continue;
2065 }
2066
2067 case OPC_MoveParent:
2068 // Pop the current node off the NodeStack.
2069 NodeStack.pop_back();
2070 assert(!NodeStack.empty() && "Node stack imbalance!");
2071 N = NodeStack.back();
2072 continue;
2073
Chris Lattnerda828e32010-03-03 07:46:25 +00002074 case OPC_CheckSame:
2075 if (!::CheckSame(MatcherTable, MatcherIndex, N, RecordedNodes)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002076 continue;
Chris Lattner2a49d572010-02-28 22:37:22 +00002077 case OPC_CheckPatternPredicate:
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002078 if (!::CheckPatternPredicate(MatcherTable, MatcherIndex, *this)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002079 continue;
2080 case OPC_CheckPredicate:
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002081 if (!::CheckNodePredicate(MatcherTable, MatcherIndex, *this,
2082 N.getNode()))
2083 break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002084 continue;
Chris Lattner57bf8a42010-03-04 01:23:08 +00002085 case OPC_CheckComplexPat: {
2086 unsigned CPNum = MatcherTable[MatcherIndex++];
2087 unsigned RecNo = MatcherTable[MatcherIndex++];
2088 assert(RecNo < RecordedNodes.size() && "Invalid CheckComplexPat");
Chris Lattner78d3af42010-09-21 20:37:12 +00002089 SDNode *Parent = 0;
2090 if (NodeStack.size() > 1)
2091 Parent = NodeStack[NodeStack.size()-2].getNode();
2092
2093 if (!CheckComplexPattern(NodeToMatch, Parent, RecordedNodes[RecNo], CPNum,
Chris Lattner57bf8a42010-03-04 01:23:08 +00002094 RecordedNodes))
Chris Lattner2a49d572010-02-28 22:37:22 +00002095 break;
2096 continue;
Chris Lattner57bf8a42010-03-04 01:23:08 +00002097 }
Chris Lattner2a49d572010-02-28 22:37:22 +00002098 case OPC_CheckOpcode:
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002099 if (!::CheckOpcode(MatcherTable, MatcherIndex, N.getNode())) break;
2100 continue;
2101
2102 case OPC_CheckType:
2103 if (!::CheckType(MatcherTable, MatcherIndex, N, TLI)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002104 continue;
2105
Chris Lattnereb669212010-03-01 06:59:22 +00002106 case OPC_SwitchOpcode: {
2107 unsigned CurNodeOpcode = N.getOpcode();
Chris Lattner7d892d62010-03-01 07:43:08 +00002108 unsigned SwitchStart = MatcherIndex-1; (void)SwitchStart;
Chris Lattnereb669212010-03-01 06:59:22 +00002109 unsigned CaseSize;
2110 while (1) {
2111 // Get the size of this case.
2112 CaseSize = MatcherTable[MatcherIndex++];
2113 if (CaseSize & 128)
2114 CaseSize = GetVBR(CaseSize, MatcherTable, MatcherIndex);
2115 if (CaseSize == 0) break;
2116
Chris Lattnerd5d5a3d2010-03-25 06:33:05 +00002117 uint16_t Opc = MatcherTable[MatcherIndex++];
2118 Opc |= (unsigned short)MatcherTable[MatcherIndex++] << 8;
2119
Chris Lattnereb669212010-03-01 06:59:22 +00002120 // If the opcode matches, then we will execute this case.
Chris Lattnerd5d5a3d2010-03-25 06:33:05 +00002121 if (CurNodeOpcode == Opc)
Chris Lattnereb669212010-03-01 06:59:22 +00002122 break;
2123
2124 // Otherwise, skip over this case.
2125 MatcherIndex += CaseSize;
2126 }
2127
Chris Lattnercfe2eab2010-03-03 06:28:15 +00002128 // If no cases matched, bail out.
Chris Lattnereb669212010-03-01 06:59:22 +00002129 if (CaseSize == 0) break;
2130
2131 // Otherwise, execute the case we found.
2132 DEBUG(errs() << " OpcodeSwitch from " << SwitchStart
2133 << " to " << MatcherIndex << "\n");
2134 continue;
2135 }
2136
Chris Lattnercfe2eab2010-03-03 06:28:15 +00002137 case OPC_SwitchType: {
2138 MVT::SimpleValueType CurNodeVT = N.getValueType().getSimpleVT().SimpleTy;
2139 unsigned SwitchStart = MatcherIndex-1; (void)SwitchStart;
2140 unsigned CaseSize;
2141 while (1) {
2142 // Get the size of this case.
2143 CaseSize = MatcherTable[MatcherIndex++];
2144 if (CaseSize & 128)
2145 CaseSize = GetVBR(CaseSize, MatcherTable, MatcherIndex);
2146 if (CaseSize == 0) break;
2147
2148 MVT::SimpleValueType CaseVT =
2149 (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
2150 if (CaseVT == MVT::iPTR)
2151 CaseVT = TLI.getPointerTy().SimpleTy;
2152
2153 // If the VT matches, then we will execute this case.
2154 if (CurNodeVT == CaseVT)
2155 break;
2156
2157 // Otherwise, skip over this case.
2158 MatcherIndex += CaseSize;
2159 }
2160
2161 // If no cases matched, bail out.
2162 if (CaseSize == 0) break;
2163
2164 // Otherwise, execute the case we found.
2165 DEBUG(errs() << " TypeSwitch[" << EVT(CurNodeVT).getEVTString()
2166 << "] from " << SwitchStart << " to " << MatcherIndex<<'\n');
2167 continue;
2168 }
Chris Lattner2a49d572010-02-28 22:37:22 +00002169 case OPC_CheckChild0Type: case OPC_CheckChild1Type:
2170 case OPC_CheckChild2Type: case OPC_CheckChild3Type:
2171 case OPC_CheckChild4Type: case OPC_CheckChild5Type:
Chris Lattnerda828e32010-03-03 07:46:25 +00002172 case OPC_CheckChild6Type: case OPC_CheckChild7Type:
2173 if (!::CheckChildType(MatcherTable, MatcherIndex, N, TLI,
2174 Opcode-OPC_CheckChild0Type))
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002175 break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002176 continue;
Chris Lattner2a49d572010-02-28 22:37:22 +00002177 case OPC_CheckCondCode:
Chris Lattnerda828e32010-03-03 07:46:25 +00002178 if (!::CheckCondCode(MatcherTable, MatcherIndex, N)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002179 continue;
Chris Lattnerda828e32010-03-03 07:46:25 +00002180 case OPC_CheckValueType:
2181 if (!::CheckValueType(MatcherTable, MatcherIndex, N, TLI)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002182 continue;
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002183 case OPC_CheckInteger:
2184 if (!::CheckInteger(MatcherTable, MatcherIndex, N)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002185 continue;
Chris Lattnerda828e32010-03-03 07:46:25 +00002186 case OPC_CheckAndImm:
2187 if (!::CheckAndImm(MatcherTable, MatcherIndex, N, *this)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002188 continue;
Chris Lattnerda828e32010-03-03 07:46:25 +00002189 case OPC_CheckOrImm:
2190 if (!::CheckOrImm(MatcherTable, MatcherIndex, N, *this)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002191 continue;
Chris Lattner2a49d572010-02-28 22:37:22 +00002192
2193 case OPC_CheckFoldableChainNode: {
2194 assert(NodeStack.size() != 1 && "No parent node");
2195 // Verify that all intermediate nodes between the root and this one have
2196 // a single use.
2197 bool HasMultipleUses = false;
2198 for (unsigned i = 1, e = NodeStack.size()-1; i != e; ++i)
2199 if (!NodeStack[i].hasOneUse()) {
2200 HasMultipleUses = true;
2201 break;
2202 }
2203 if (HasMultipleUses) break;
2204
2205 // Check to see that the target thinks this is profitable to fold and that
2206 // we can fold it without inducing cycles in the graph.
2207 if (!IsProfitableToFold(N, NodeStack[NodeStack.size()-2].getNode(),
2208 NodeToMatch) ||
2209 !IsLegalToFold(N, NodeStack[NodeStack.size()-2].getNode(),
Dan Gohmand858e902010-04-17 15:26:15 +00002210 NodeToMatch, OptLevel,
2211 true/*We validate our own chains*/))
Chris Lattner2a49d572010-02-28 22:37:22 +00002212 break;
2213
2214 continue;
2215 }
Chris Lattner2a49d572010-02-28 22:37:22 +00002216 case OPC_EmitInteger: {
2217 MVT::SimpleValueType VT =
2218 (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
2219 int64_t Val = MatcherTable[MatcherIndex++];
2220 if (Val & 128)
2221 Val = GetVBR(Val, MatcherTable, MatcherIndex);
2222 RecordedNodes.push_back(CurDAG->getTargetConstant(Val, VT));
2223 continue;
2224 }
2225 case OPC_EmitRegister: {
2226 MVT::SimpleValueType VT =
2227 (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
2228 unsigned RegNo = MatcherTable[MatcherIndex++];
2229 RecordedNodes.push_back(CurDAG->getRegister(RegNo, VT));
2230 continue;
2231 }
2232
2233 case OPC_EmitConvertToTarget: {
2234 // Convert from IMM/FPIMM to target version.
2235 unsigned RecNo = MatcherTable[MatcherIndex++];
2236 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2237 SDValue Imm = RecordedNodes[RecNo];
2238
2239 if (Imm->getOpcode() == ISD::Constant) {
2240 int64_t Val = cast<ConstantSDNode>(Imm)->getZExtValue();
2241 Imm = CurDAG->getTargetConstant(Val, Imm.getValueType());
2242 } else if (Imm->getOpcode() == ISD::ConstantFP) {
2243 const ConstantFP *Val=cast<ConstantFPSDNode>(Imm)->getConstantFPValue();
2244 Imm = CurDAG->getTargetConstantFP(*Val, Imm.getValueType());
2245 }
2246
2247 RecordedNodes.push_back(Imm);
2248 continue;
2249 }
2250
Chris Lattneraa4e3392010-03-28 05:50:16 +00002251 case OPC_EmitMergeInputChains1_0: // OPC_EmitMergeInputChains, 1, 0
2252 case OPC_EmitMergeInputChains1_1: { // OPC_EmitMergeInputChains, 1, 1
2253 // These are space-optimized forms of OPC_EmitMergeInputChains.
2254 assert(InputChain.getNode() == 0 &&
2255 "EmitMergeInputChains should be the first chain producing node");
2256 assert(ChainNodesMatched.empty() &&
2257 "Should only have one EmitMergeInputChains per match");
2258
2259 // Read all of the chained nodes.
2260 unsigned RecNo = Opcode == OPC_EmitMergeInputChains1_1;
2261 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2262 ChainNodesMatched.push_back(RecordedNodes[RecNo].getNode());
2263
2264 // FIXME: What if other value results of the node have uses not matched
2265 // by this pattern?
2266 if (ChainNodesMatched.back() != NodeToMatch &&
2267 !RecordedNodes[RecNo].hasOneUse()) {
2268 ChainNodesMatched.clear();
2269 break;
2270 }
2271
2272 // Merge the input chains if they are not intra-pattern references.
2273 InputChain = HandleMergeInputChains(ChainNodesMatched, CurDAG);
2274
2275 if (InputChain.getNode() == 0)
2276 break; // Failed to merge.
2277 continue;
2278 }
2279
Chris Lattner2a49d572010-02-28 22:37:22 +00002280 case OPC_EmitMergeInputChains: {
2281 assert(InputChain.getNode() == 0 &&
2282 "EmitMergeInputChains should be the first chain producing node");
2283 // This node gets a list of nodes we matched in the input that have
2284 // chains. We want to token factor all of the input chains to these nodes
2285 // together. However, if any of the input chains is actually one of the
2286 // nodes matched in this pattern, then we have an intra-match reference.
2287 // Ignore these because the newly token factored chain should not refer to
2288 // the old nodes.
2289 unsigned NumChains = MatcherTable[MatcherIndex++];
2290 assert(NumChains != 0 && "Can't TF zero chains");
2291
2292 assert(ChainNodesMatched.empty() &&
2293 "Should only have one EmitMergeInputChains per match");
2294
Chris Lattner2a49d572010-02-28 22:37:22 +00002295 // Read all of the chained nodes.
Chris Lattner00592ec2010-03-02 19:34:59 +00002296 for (unsigned i = 0; i != NumChains; ++i) {
2297 unsigned RecNo = MatcherTable[MatcherIndex++];
Chris Lattner2a49d572010-02-28 22:37:22 +00002298 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2299 ChainNodesMatched.push_back(RecordedNodes[RecNo].getNode());
2300
2301 // FIXME: What if other value results of the node have uses not matched
2302 // by this pattern?
2303 if (ChainNodesMatched.back() != NodeToMatch &&
2304 !RecordedNodes[RecNo].hasOneUse()) {
2305 ChainNodesMatched.clear();
2306 break;
2307 }
2308 }
Chris Lattner6b307922010-03-02 00:00:03 +00002309
2310 // If the inner loop broke out, the match fails.
2311 if (ChainNodesMatched.empty())
2312 break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002313
Chris Lattner6b307922010-03-02 00:00:03 +00002314 // Merge the input chains if they are not intra-pattern references.
2315 InputChain = HandleMergeInputChains(ChainNodesMatched, CurDAG);
2316
2317 if (InputChain.getNode() == 0)
2318 break; // Failed to merge.
Chris Lattner2a49d572010-02-28 22:37:22 +00002319
Chris Lattner2a49d572010-02-28 22:37:22 +00002320 continue;
2321 }
2322
2323 case OPC_EmitCopyToReg: {
2324 unsigned RecNo = MatcherTable[MatcherIndex++];
2325 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2326 unsigned DestPhysReg = MatcherTable[MatcherIndex++];
2327
2328 if (InputChain.getNode() == 0)
2329 InputChain = CurDAG->getEntryNode();
2330
2331 InputChain = CurDAG->getCopyToReg(InputChain, NodeToMatch->getDebugLoc(),
2332 DestPhysReg, RecordedNodes[RecNo],
2333 InputFlag);
2334
2335 InputFlag = InputChain.getValue(1);
2336 continue;
2337 }
2338
2339 case OPC_EmitNodeXForm: {
2340 unsigned XFormNo = MatcherTable[MatcherIndex++];
2341 unsigned RecNo = MatcherTable[MatcherIndex++];
2342 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2343 RecordedNodes.push_back(RunSDNodeXForm(RecordedNodes[RecNo], XFormNo));
2344 continue;
2345 }
2346
2347 case OPC_EmitNode:
2348 case OPC_MorphNodeTo: {
Chris Lattner14df8dc2010-02-28 22:38:43 +00002349 uint16_t TargetOpc = MatcherTable[MatcherIndex++];
2350 TargetOpc |= (unsigned short)MatcherTable[MatcherIndex++] << 8;
Chris Lattner2a49d572010-02-28 22:37:22 +00002351 unsigned EmitNodeInfo = MatcherTable[MatcherIndex++];
2352 // Get the result VT list.
2353 unsigned NumVTs = MatcherTable[MatcherIndex++];
2354 SmallVector<EVT, 4> VTs;
2355 for (unsigned i = 0; i != NumVTs; ++i) {
2356 MVT::SimpleValueType VT =
2357 (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
2358 if (VT == MVT::iPTR) VT = TLI.getPointerTy().SimpleTy;
2359 VTs.push_back(VT);
2360 }
2361
2362 if (EmitNodeInfo & OPFL_Chain)
2363 VTs.push_back(MVT::Other);
2364 if (EmitNodeInfo & OPFL_FlagOutput)
2365 VTs.push_back(MVT::Flag);
2366
Chris Lattner7d892d62010-03-01 07:43:08 +00002367 // This is hot code, so optimize the two most common cases of 1 and 2
2368 // results.
2369 SDVTList VTList;
2370 if (VTs.size() == 1)
2371 VTList = CurDAG->getVTList(VTs[0]);
2372 else if (VTs.size() == 2)
2373 VTList = CurDAG->getVTList(VTs[0], VTs[1]);
2374 else
2375 VTList = CurDAG->getVTList(VTs.data(), VTs.size());
Chris Lattner2a49d572010-02-28 22:37:22 +00002376
2377 // Get the operand list.
2378 unsigned NumOps = MatcherTable[MatcherIndex++];
2379 SmallVector<SDValue, 8> Ops;
2380 for (unsigned i = 0; i != NumOps; ++i) {
2381 unsigned RecNo = MatcherTable[MatcherIndex++];
2382 if (RecNo & 128)
2383 RecNo = GetVBR(RecNo, MatcherTable, MatcherIndex);
2384
2385 assert(RecNo < RecordedNodes.size() && "Invalid EmitNode");
2386 Ops.push_back(RecordedNodes[RecNo]);
2387 }
2388
2389 // If there are variadic operands to add, handle them now.
2390 if (EmitNodeInfo & OPFL_VariadicInfo) {
2391 // Determine the start index to copy from.
2392 unsigned FirstOpToCopy = getNumFixedFromVariadicInfo(EmitNodeInfo);
2393 FirstOpToCopy += (EmitNodeInfo & OPFL_Chain) ? 1 : 0;
2394 assert(NodeToMatch->getNumOperands() >= FirstOpToCopy &&
2395 "Invalid variadic node");
2396 // Copy all of the variadic operands, not including a potential flag
2397 // input.
2398 for (unsigned i = FirstOpToCopy, e = NodeToMatch->getNumOperands();
2399 i != e; ++i) {
2400 SDValue V = NodeToMatch->getOperand(i);
2401 if (V.getValueType() == MVT::Flag) break;
2402 Ops.push_back(V);
2403 }
2404 }
2405
2406 // If this has chain/flag inputs, add them.
2407 if (EmitNodeInfo & OPFL_Chain)
2408 Ops.push_back(InputChain);
2409 if ((EmitNodeInfo & OPFL_FlagInput) && InputFlag.getNode() != 0)
2410 Ops.push_back(InputFlag);
2411
2412 // Create the node.
2413 SDNode *Res = 0;
2414 if (Opcode != OPC_MorphNodeTo) {
2415 // If this is a normal EmitNode command, just create the new node and
2416 // add the results to the RecordedNodes list.
2417 Res = CurDAG->getMachineNode(TargetOpc, NodeToMatch->getDebugLoc(),
2418 VTList, Ops.data(), Ops.size());
2419
2420 // Add all the non-flag/non-chain results to the RecordedNodes list.
2421 for (unsigned i = 0, e = VTs.size(); i != e; ++i) {
2422 if (VTs[i] == MVT::Other || VTs[i] == MVT::Flag) break;
2423 RecordedNodes.push_back(SDValue(Res, i));
2424 }
2425
2426 } else {
Chris Lattner3ff1e4d2010-03-02 06:55:04 +00002427 Res = MorphNode(NodeToMatch, TargetOpc, VTList, Ops.data(), Ops.size(),
2428 EmitNodeInfo);
Chris Lattner2a49d572010-02-28 22:37:22 +00002429 }
2430
2431 // If the node had chain/flag results, update our notion of the current
2432 // chain and flag.
Chris Lattner82dd3d32010-03-02 07:50:03 +00002433 if (EmitNodeInfo & OPFL_FlagOutput) {
Chris Lattner2a49d572010-02-28 22:37:22 +00002434 InputFlag = SDValue(Res, VTs.size()-1);
2435 if (EmitNodeInfo & OPFL_Chain)
2436 InputChain = SDValue(Res, VTs.size()-2);
2437 } else if (EmitNodeInfo & OPFL_Chain)
2438 InputChain = SDValue(Res, VTs.size()-1);
2439
2440 // If the OPFL_MemRefs flag is set on this node, slap all of the
2441 // accumulated memrefs onto it.
2442 //
2443 // FIXME: This is vastly incorrect for patterns with multiple outputs
2444 // instructions that access memory and for ComplexPatterns that match
2445 // loads.
2446 if (EmitNodeInfo & OPFL_MemRefs) {
2447 MachineSDNode::mmo_iterator MemRefs =
2448 MF->allocateMemRefsArray(MatchedMemRefs.size());
2449 std::copy(MatchedMemRefs.begin(), MatchedMemRefs.end(), MemRefs);
2450 cast<MachineSDNode>(Res)
2451 ->setMemRefs(MemRefs, MemRefs + MatchedMemRefs.size());
2452 }
2453
2454 DEBUG(errs() << " "
2455 << (Opcode == OPC_MorphNodeTo ? "Morphed" : "Created")
2456 << " node: "; Res->dump(CurDAG); errs() << "\n");
2457
2458 // If this was a MorphNodeTo then we're completely done!
2459 if (Opcode == OPC_MorphNodeTo) {
2460 // Update chain and flag uses.
2461 UpdateChainsAndFlags(NodeToMatch, InputChain, ChainNodesMatched,
Chris Lattner82dd3d32010-03-02 07:50:03 +00002462 InputFlag, FlagResultNodesMatched, true);
2463 return Res;
Chris Lattner2a49d572010-02-28 22:37:22 +00002464 }
2465
2466 continue;
2467 }
2468
2469 case OPC_MarkFlagResults: {
2470 unsigned NumNodes = MatcherTable[MatcherIndex++];
2471
2472 // Read and remember all the flag-result nodes.
2473 for (unsigned i = 0; i != NumNodes; ++i) {
2474 unsigned RecNo = MatcherTable[MatcherIndex++];
2475 if (RecNo & 128)
2476 RecNo = GetVBR(RecNo, MatcherTable, MatcherIndex);
2477
2478 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2479 FlagResultNodesMatched.push_back(RecordedNodes[RecNo].getNode());
2480 }
2481 continue;
2482 }
2483
2484 case OPC_CompleteMatch: {
2485 // The match has been completed, and any new nodes (if any) have been
2486 // created. Patch up references to the matched dag to use the newly
2487 // created nodes.
2488 unsigned NumResults = MatcherTable[MatcherIndex++];
2489
2490 for (unsigned i = 0; i != NumResults; ++i) {
2491 unsigned ResSlot = MatcherTable[MatcherIndex++];
2492 if (ResSlot & 128)
2493 ResSlot = GetVBR(ResSlot, MatcherTable, MatcherIndex);
2494
2495 assert(ResSlot < RecordedNodes.size() && "Invalid CheckSame");
2496 SDValue Res = RecordedNodes[ResSlot];
2497
Chris Lattnerb0e483e2010-03-28 05:54:03 +00002498 assert(i < NodeToMatch->getNumValues() &&
2499 NodeToMatch->getValueType(i) != MVT::Other &&
2500 NodeToMatch->getValueType(i) != MVT::Flag &&
2501 "Invalid number of results to complete!");
Chris Lattner2a49d572010-02-28 22:37:22 +00002502 assert((NodeToMatch->getValueType(i) == Res.getValueType() ||
2503 NodeToMatch->getValueType(i) == MVT::iPTR ||
2504 Res.getValueType() == MVT::iPTR ||
2505 NodeToMatch->getValueType(i).getSizeInBits() ==
2506 Res.getValueType().getSizeInBits()) &&
2507 "invalid replacement");
2508 CurDAG->ReplaceAllUsesOfValueWith(SDValue(NodeToMatch, i), Res);
2509 }
2510
2511 // If the root node defines a flag, add it to the flag nodes to update
2512 // list.
2513 if (NodeToMatch->getValueType(NodeToMatch->getNumValues()-1) == MVT::Flag)
2514 FlagResultNodesMatched.push_back(NodeToMatch);
2515
2516 // Update chain and flag uses.
2517 UpdateChainsAndFlags(NodeToMatch, InputChain, ChainNodesMatched,
Chris Lattner82dd3d32010-03-02 07:50:03 +00002518 InputFlag, FlagResultNodesMatched, false);
Chris Lattner2a49d572010-02-28 22:37:22 +00002519
2520 assert(NodeToMatch->use_empty() &&
2521 "Didn't replace all uses of the node?");
2522
2523 // FIXME: We just return here, which interacts correctly with SelectRoot
2524 // above. We should fix this to not return an SDNode* anymore.
2525 return 0;
2526 }
2527 }
2528
2529 // If the code reached this point, then the match failed. See if there is
2530 // another child to try in the current 'Scope', otherwise pop it until we
2531 // find a case to check.
Dan Gohman19b38262010-03-09 02:15:05 +00002532 DEBUG(errs() << " Match failed at index " << CurrentOpcodeIndex << "\n");
Chris Lattnerbed993d2010-03-28 19:46:56 +00002533 ++NumDAGIselRetries;
Chris Lattner2a49d572010-02-28 22:37:22 +00002534 while (1) {
2535 if (MatchScopes.empty()) {
2536 CannotYetSelect(NodeToMatch);
2537 return 0;
2538 }
2539
2540 // Restore the interpreter state back to the point where the scope was
2541 // formed.
2542 MatchScope &LastScope = MatchScopes.back();
2543 RecordedNodes.resize(LastScope.NumRecordedNodes);
2544 NodeStack.clear();
2545 NodeStack.append(LastScope.NodeStack.begin(), LastScope.NodeStack.end());
2546 N = NodeStack.back();
2547
Chris Lattner2a49d572010-02-28 22:37:22 +00002548 if (LastScope.NumMatchedMemRefs != MatchedMemRefs.size())
2549 MatchedMemRefs.resize(LastScope.NumMatchedMemRefs);
2550 MatcherIndex = LastScope.FailIndex;
2551
Dan Gohman19b38262010-03-09 02:15:05 +00002552 DEBUG(errs() << " Continuing at " << MatcherIndex << "\n");
2553
Chris Lattner2a49d572010-02-28 22:37:22 +00002554 InputChain = LastScope.InputChain;
2555 InputFlag = LastScope.InputFlag;
2556 if (!LastScope.HasChainNodesMatched)
2557 ChainNodesMatched.clear();
2558 if (!LastScope.HasFlagResultNodesMatched)
2559 FlagResultNodesMatched.clear();
2560
2561 // Check to see what the offset is at the new MatcherIndex. If it is zero
2562 // we have reached the end of this scope, otherwise we have another child
2563 // in the current scope to try.
2564 unsigned NumToSkip = MatcherTable[MatcherIndex++];
2565 if (NumToSkip & 128)
2566 NumToSkip = GetVBR(NumToSkip, MatcherTable, MatcherIndex);
2567
2568 // If we have another child in this scope to match, update FailIndex and
2569 // try it.
2570 if (NumToSkip != 0) {
2571 LastScope.FailIndex = MatcherIndex+NumToSkip;
2572 break;
2573 }
2574
2575 // End of this scope, pop it and try the next child in the containing
2576 // scope.
2577 MatchScopes.pop_back();
2578 }
2579 }
2580}
2581
2582
2583
Dan Gohmaneeb3a002010-01-05 01:24:18 +00002584void SelectionDAGISel::CannotYetSelect(SDNode *N) {
Dan Gohmane1f188f2009-10-29 22:30:23 +00002585 std::string msg;
2586 raw_string_ostream Msg(msg);
2587 Msg << "Cannot yet select: ";
Chris Lattner2c4afd12010-03-04 00:21:16 +00002588
2589 if (N->getOpcode() != ISD::INTRINSIC_W_CHAIN &&
2590 N->getOpcode() != ISD::INTRINSIC_WO_CHAIN &&
2591 N->getOpcode() != ISD::INTRINSIC_VOID) {
2592 N->printrFull(Msg, CurDAG);
2593 } else {
2594 bool HasInputChain = N->getOperand(0).getValueType() == MVT::Other;
2595 unsigned iid =
2596 cast<ConstantSDNode>(N->getOperand(HasInputChain))->getZExtValue();
2597 if (iid < Intrinsic::num_intrinsics)
2598 Msg << "intrinsic %" << Intrinsic::getName((Intrinsic::ID)iid);
2599 else if (const TargetIntrinsicInfo *TII = TM.getIntrinsicInfo())
2600 Msg << "target intrinsic %" << TII->getName(iid);
2601 else
2602 Msg << "unknown intrinsic #" << iid;
2603 }
Chris Lattner75361b62010-04-07 22:58:41 +00002604 report_fatal_error(Msg.str());
Dan Gohmane1f188f2009-10-29 22:30:23 +00002605}
2606
Devang Patel19974732007-05-03 01:11:54 +00002607char SelectionDAGISel::ID = 0;