blob: a63c40a52fd5de5bc11e615034db26664822dd25 [file] [log] [blame]
Chris Lattner1c08c712005-01-07 07:47:53 +00001//===-- SelectionDAGISel.cpp - Implement the SelectionDAGISel class -------===//
Misha Brukmanedf128a2005-04-21 22:36:52 +00002//
Chris Lattner1c08c712005-01-07 07:47:53 +00003// The LLVM Compiler Infrastructure
4//
Chris Lattner4ee451d2007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Misha Brukmanedf128a2005-04-21 22:36:52 +00007//
Chris Lattner1c08c712005-01-07 07:47:53 +00008//===----------------------------------------------------------------------===//
9//
10// This implements the SelectionDAGISel class.
11//
12//===----------------------------------------------------------------------===//
13
14#define DEBUG_TYPE "isel"
Dan Gohman84fbac52009-02-06 17:22:58 +000015#include "ScheduleDAGSDNodes.h"
Dan Gohman2048b852009-11-23 18:04:58 +000016#include "SelectionDAGBuilder.h"
Dan Gohman6277eb22009-11-23 17:16:22 +000017#include "FunctionLoweringInfo.h"
Dan Gohman84fbac52009-02-06 17:22:58 +000018#include "llvm/CodeGen/SelectionDAGISel.h"
Jim Laskeyc7c3f112006-10-16 20:52:31 +000019#include "llvm/Analysis/AliasAnalysis.h"
Devang Patel713f0432009-09-16 21:09:07 +000020#include "llvm/Analysis/DebugInfo.h"
Anton Korobeynikov5502bf62007-04-04 21:14:49 +000021#include "llvm/Constants.h"
Chris 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"
Dan Gohman78eca172008-08-19 22:33:34 +000028#include "llvm/CodeGen/FastISel.h"
Gordon Henriksen5a29c9e2008-08-17 12:56:54 +000029#include "llvm/CodeGen/GCStrategy.h"
Gordon Henriksen5eca0752008-08-17 18:44:35 +000030#include "llvm/CodeGen/GCMetadata.h"
Chris Lattner1c08c712005-01-07 07:47:53 +000031#include "llvm/CodeGen/MachineFunction.h"
Chris Lattner1c08c712005-01-07 07:47:53 +000032#include "llvm/CodeGen/MachineInstrBuilder.h"
Chris Lattner84bc5422007-12-31 04:13:23 +000033#include "llvm/CodeGen/MachineModuleInfo.h"
34#include "llvm/CodeGen/MachineRegisterInfo.h"
Dan Gohmanfc54c552009-01-15 22:18:12 +000035#include "llvm/CodeGen/ScheduleHazardRecognizer.h"
Jim Laskeyeb577ba2006-08-02 12:30:23 +000036#include "llvm/CodeGen/SchedulerRegistry.h"
Chris Lattner1c08c712005-01-07 07:47:53 +000037#include "llvm/CodeGen/SelectionDAG.h"
Dan Gohman6f0d0242008-02-10 18:45:23 +000038#include "llvm/Target/TargetRegisterInfo.h"
Dan Gohmane1f188f2009-10-29 22:30:23 +000039#include "llvm/Target/TargetIntrinsicInfo.h"
Chris Lattner1c08c712005-01-07 07:47:53 +000040#include "llvm/Target/TargetInstrInfo.h"
41#include "llvm/Target/TargetLowering.h"
42#include "llvm/Target/TargetMachine.h"
Vladimir Prus12472912006-05-23 13:43:15 +000043#include "llvm/Target/TargetOptions.h"
Chris Lattnera4f0b3a2006-08-27 12:54:02 +000044#include "llvm/Support/Compiler.h"
Evan Chengdb8d56b2008-06-30 20:45:06 +000045#include "llvm/Support/Debug.h"
Torok Edwin7d696d82009-07-11 13:10:19 +000046#include "llvm/Support/ErrorHandling.h"
Evan Chengdb8d56b2008-06-30 20:45:06 +000047#include "llvm/Support/Timer.h"
Daniel Dunbarce63ffb2009-07-25 00:23:56 +000048#include "llvm/Support/raw_ostream.h"
Chris Lattnerc19ae9d2010-03-04 19:46:56 +000049#include "llvm/ADT/Statistic.h"
Jeff Cohen7e881032006-02-24 02:52:40 +000050#include <algorithm>
Chris Lattner1c08c712005-01-07 07:47:53 +000051using namespace llvm;
52
Chris Lattnerc19ae9d2010-03-04 19:46:56 +000053STATISTIC(NumFastIselFailures, "Number of instructions fast isel failed on");
Chris Lattnerbed993d2010-03-28 19:46:56 +000054STATISTIC(NumDAGIselRetries,"Number of times dag isel has to try another path");
Chris Lattnerc19ae9d2010-03-04 19:46:56 +000055
Chris Lattneread0d882008-06-17 06:09:18 +000056static cl::opt<bool>
Dan Gohman293d5f82008-09-09 22:06:46 +000057EnableFastISelVerbose("fast-isel-verbose", cl::Hidden,
Dan Gohmand659d502008-10-20 21:30:12 +000058 cl::desc("Enable verbose messages in the \"fast\" "
Dan Gohman293d5f82008-09-09 22:06:46 +000059 "instruction selector"));
60static cl::opt<bool>
Dan Gohman4344a5d2008-09-09 23:05:00 +000061EnableFastISelAbort("fast-isel-abort", cl::Hidden,
62 cl::desc("Enable abort calls when \"fast\" instruction fails"));
Chris Lattneread0d882008-06-17 06:09:18 +000063
Chris Lattnerda8abb02005-09-01 18:44:10 +000064#ifndef NDEBUG
Chris Lattner7944d9d2005-01-12 03:41:21 +000065static cl::opt<bool>
Dan Gohman462dc7f2008-07-21 20:00:07 +000066ViewDAGCombine1("view-dag-combine1-dags", cl::Hidden,
67 cl::desc("Pop up a window to show dags before the first "
68 "dag combine pass"));
69static cl::opt<bool>
70ViewLegalizeTypesDAGs("view-legalize-types-dags", cl::Hidden,
71 cl::desc("Pop up a window to show dags before legalize types"));
72static cl::opt<bool>
73ViewLegalizeDAGs("view-legalize-dags", cl::Hidden,
74 cl::desc("Pop up a window to show dags before legalize"));
75static cl::opt<bool>
76ViewDAGCombine2("view-dag-combine2-dags", cl::Hidden,
77 cl::desc("Pop up a window to show dags before the second "
78 "dag combine pass"));
79static cl::opt<bool>
Duncan Sands25cf2272008-11-24 14:53:14 +000080ViewDAGCombineLT("view-dag-combine-lt-dags", cl::Hidden,
81 cl::desc("Pop up a window to show dags before the post legalize types"
82 " dag combine pass"));
83static cl::opt<bool>
Evan Chenga9c20912006-01-21 02:32:06 +000084ViewISelDAGs("view-isel-dags", cl::Hidden,
85 cl::desc("Pop up a window to show isel dags as they are selected"));
86static cl::opt<bool>
87ViewSchedDAGs("view-sched-dags", cl::Hidden,
88 cl::desc("Pop up a window to show sched dags as they are processed"));
Dan Gohman3e1a7ae2007-08-28 20:32:58 +000089static cl::opt<bool>
90ViewSUnitDAGs("view-sunit-dags", cl::Hidden,
Chris Lattner5bab7852008-01-25 17:24:52 +000091 cl::desc("Pop up a window to show SUnit dags after they are processed"));
Chris Lattner7944d9d2005-01-12 03:41:21 +000092#else
Dan Gohman462dc7f2008-07-21 20:00:07 +000093static const bool ViewDAGCombine1 = false,
94 ViewLegalizeTypesDAGs = false, ViewLegalizeDAGs = false,
95 ViewDAGCombine2 = false,
Duncan Sands25cf2272008-11-24 14:53:14 +000096 ViewDAGCombineLT = false,
Dan Gohman462dc7f2008-07-21 20:00:07 +000097 ViewISelDAGs = false, ViewSchedDAGs = false,
98 ViewSUnitDAGs = false;
Chris Lattner7944d9d2005-01-12 03:41:21 +000099#endif
100
Jim Laskeyeb577ba2006-08-02 12:30:23 +0000101//===---------------------------------------------------------------------===//
102///
103/// RegisterScheduler class - Track the registration of instruction schedulers.
104///
105//===---------------------------------------------------------------------===//
106MachinePassRegistry RegisterScheduler::Registry;
107
108//===---------------------------------------------------------------------===//
109///
110/// ISHeuristic command line option for instruction schedulers.
111///
112//===---------------------------------------------------------------------===//
Dan Gohman844731a2008-05-13 00:00:25 +0000113static cl::opt<RegisterScheduler::FunctionPassCtor, false,
114 RegisterPassParser<RegisterScheduler> >
115ISHeuristic("pre-RA-sched",
116 cl::init(&createDefaultScheduler),
117 cl::desc("Instruction schedulers available (before register"
118 " allocation):"));
Jim Laskey13ec7022006-08-01 14:21:23 +0000119
Dan Gohman844731a2008-05-13 00:00:25 +0000120static RegisterScheduler
Dan Gohmanb8cab922008-10-14 20:25:08 +0000121defaultListDAGScheduler("default", "Best scheduler for the target",
Dan Gohman844731a2008-05-13 00:00:25 +0000122 createDefaultScheduler);
Evan Cheng4ef10862006-01-23 07:01:07 +0000123
Chris Lattner1c08c712005-01-07 07:47:53 +0000124namespace llvm {
125 //===--------------------------------------------------------------------===//
Jim Laskey9373beb2006-08-01 19:14:14 +0000126 /// createDefaultScheduler - This creates an instruction scheduler appropriate
127 /// for the target.
Dan Gohman47ac0f02009-02-11 04:27:20 +0000128 ScheduleDAGSDNodes* createDefaultScheduler(SelectionDAGISel *IS,
Bill Wendling98a366d2009-04-29 23:29:43 +0000129 CodeGenOpt::Level OptLevel) {
Dan Gohmane9530ec2009-01-15 16:58:17 +0000130 const TargetLowering &TLI = IS->getTargetLowering();
131
Bill Wendling98a366d2009-04-29 23:29:43 +0000132 if (OptLevel == CodeGenOpt::None)
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000133 return createFastDAGScheduler(IS, OptLevel);
Dan Gohman9e76fea2008-11-20 03:11:19 +0000134 if (TLI.getSchedulingPreference() == TargetLowering::SchedulingForLatency)
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000135 return createTDListDAGScheduler(IS, OptLevel);
Dan Gohman9e76fea2008-11-20 03:11:19 +0000136 assert(TLI.getSchedulingPreference() ==
Bill Wendling187361b2010-01-23 10:26:57 +0000137 TargetLowering::SchedulingForRegPressure && "Unknown sched type!");
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000138 return createBURRListDAGScheduler(IS, OptLevel);
Jim Laskey9373beb2006-08-01 19:14:14 +0000139 }
Chris Lattner1c08c712005-01-07 07:47:53 +0000140}
141
Evan Chengff9b3732008-01-30 18:18:23 +0000142// EmitInstrWithCustomInserter - This method should be implemented by targets
Dan Gohman533297b2009-10-29 18:10:34 +0000143// that mark instructions with the 'usesCustomInserter' flag. These
Chris Lattner025c39b2005-08-26 20:54:47 +0000144// instructions are special in various ways, which require special support to
145// insert. The specified MachineInstr is created but not inserted into any
Dan Gohman533297b2009-10-29 18:10:34 +0000146// basic blocks, and this method is called to expand it into a sequence of
147// instructions, potentially also creating new basic blocks and control flow.
148// When new basic blocks are inserted and the edges from MBB to its successors
149// are modified, the method should insert pairs of <OldSucc, NewSucc> into the
150// DenseMap.
Dan Gohmanaf1d8ca2010-05-01 00:01:06 +0000151MachineBasicBlock *
152TargetLowering::EmitInstrWithCustomInserter(MachineInstr *MI,
153 MachineBasicBlock *MBB) const {
Torok Edwinf3689232009-07-12 20:07:01 +0000154#ifndef NDEBUG
David Greene1a053232010-01-05 01:26:11 +0000155 dbgs() << "If a target marks an instruction with "
Dan Gohman533297b2009-10-29 18:10:34 +0000156 "'usesCustomInserter', it must implement "
Torok Edwinf3689232009-07-12 20:07:01 +0000157 "TargetLowering::EmitInstrWithCustomInserter!";
158#endif
Torok Edwinc23197a2009-07-14 16:55:14 +0000159 llvm_unreachable(0);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000160 return 0;
Chris Lattner025c39b2005-08-26 20:54:47 +0000161}
162
Chris Lattner7041ee32005-01-11 05:56:49 +0000163//===----------------------------------------------------------------------===//
164// SelectionDAGISel code
165//===----------------------------------------------------------------------===//
Chris Lattner1c08c712005-01-07 07:47:53 +0000166
Dan Gohmanf0757b02010-04-21 01:34:56 +0000167SelectionDAGISel::SelectionDAGISel(const TargetMachine &tm, CodeGenOpt::Level OL) :
Dan Gohmanad2afc22009-07-31 18:16:33 +0000168 MachineFunctionPass(&ID), TM(tm), TLI(*tm.getTargetLowering()),
Dan Gohman7c3234c2008-08-27 23:52:12 +0000169 FuncInfo(new FunctionLoweringInfo(TLI)),
Dan Gohman50d2b1a2010-04-19 19:22:07 +0000170 CurDAG(new SelectionDAG(tm, *FuncInfo)),
Dan Gohman55e59c12010-04-19 19:05:59 +0000171 SDB(new SelectionDAGBuilder(*CurDAG, *FuncInfo, OL)),
Dan Gohman7c3234c2008-08-27 23:52:12 +0000172 GFI(),
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000173 OptLevel(OL),
Dan Gohman7c3234c2008-08-27 23:52:12 +0000174 DAGSize(0)
175{}
176
177SelectionDAGISel::~SelectionDAGISel() {
Dan Gohman2048b852009-11-23 18:04:58 +0000178 delete SDB;
Dan Gohman7c3234c2008-08-27 23:52:12 +0000179 delete CurDAG;
180 delete FuncInfo;
181}
182
Chris Lattner495a0b52005-08-17 06:37:43 +0000183void SelectionDAGISel::getAnalysisUsage(AnalysisUsage &AU) const {
Jim Laskeyc7c3f112006-10-16 20:52:31 +0000184 AU.addRequired<AliasAnalysis>();
Dan Gohmana3477fe2009-07-31 23:36:22 +0000185 AU.addPreserved<AliasAnalysis>();
Gordon Henriksen5eca0752008-08-17 18:44:35 +0000186 AU.addRequired<GCModuleInfo>();
Dan Gohmana3477fe2009-07-31 23:36:22 +0000187 AU.addPreserved<GCModuleInfo>();
Dan Gohmanad2afc22009-07-31 18:16:33 +0000188 MachineFunctionPass::getAnalysisUsage(AU);
Chris Lattner495a0b52005-08-17 06:37:43 +0000189}
Chris Lattner1c08c712005-01-07 07:47:53 +0000190
Dan Gohmanad2afc22009-07-31 18:16:33 +0000191bool SelectionDAGISel::runOnMachineFunction(MachineFunction &mf) {
Dan Gohman4344a5d2008-09-09 23:05:00 +0000192 // Do some sanity-checking on the command-line options.
193 assert((!EnableFastISelVerbose || EnableFastISel) &&
194 "-fast-isel-verbose requires -fast-isel");
195 assert((!EnableFastISelAbort || EnableFastISel) &&
196 "-fast-isel-abort requires -fast-isel");
197
Dan Gohmanae541aa2010-04-15 04:33:49 +0000198 const Function &Fn = *mf.getFunction();
Dan Gohman8a110532008-09-05 22:59:21 +0000199 const TargetInstrInfo &TII = *TM.getInstrInfo();
200 const TargetRegisterInfo &TRI = *TM.getRegisterInfo();
201
Dan Gohmanc0fb65d2010-04-14 17:02:07 +0000202 MF = &mf;
Dan Gohman79ce2762009-01-15 19:20:50 +0000203 RegInfo = &MF->getRegInfo();
Dan Gohmanc0fb65d2010-04-14 17:02:07 +0000204 AA = &getAnalysis<AliasAnalysis>();
205 GFI = Fn.hasGC() ? &getAnalysis<GCModuleInfo>().getFunctionInfo(Fn) : 0;
206
David Greene1a053232010-01-05 01:26:11 +0000207 DEBUG(dbgs() << "\n\n\n=== " << Fn.getName() << "\n");
Chris Lattner1c08c712005-01-07 07:47:53 +0000208
Chris Lattnerde6e7832010-04-05 06:10:13 +0000209 CurDAG->init(*MF);
Dan Gohman6277eb22009-11-23 17:16:22 +0000210 FuncInfo->set(Fn, *MF, EnableFastISel);
Dan Gohman2048b852009-11-23 18:04:58 +0000211 SDB->init(GFI, *AA);
Chris Lattner1c08c712005-01-07 07:47:53 +0000212
Dan Gohman6a732b52010-04-14 20:05:00 +0000213 SelectAllBasicBlocks(Fn);
Misha Brukmanedf128a2005-04-21 22:36:52 +0000214
Evan Chenga36acad2010-04-29 06:33:38 +0000215 // If the first basic block in the function has live ins that need to be
Dan Gohman8a110532008-09-05 22:59:21 +0000216 // copied into vregs, emit the copies into the top of the block before
217 // emitting the code for the block.
Evan Cheng2ad0fcf2010-04-28 23:08:54 +0000218 MachineBasicBlock *EntryMBB = MF->begin();
219 RegInfo->EmitLiveInCopies(EntryMBB, TRI, TII);
220
221 // Insert DBG_VALUE instructions for function arguments to the entry block.
222 for (unsigned i = 0, e = FuncInfo->ArgDbgValues.size(); i != e; ++i) {
223 MachineInstr *MI = FuncInfo->ArgDbgValues[e-i-1];
224 unsigned Reg = MI->getOperand(0).getReg();
225 if (TargetRegisterInfo::isPhysicalRegister(Reg))
226 EntryMBB->insert(EntryMBB->begin(), MI);
227 else {
228 MachineInstr *Def = RegInfo->getVRegDef(Reg);
229 MachineBasicBlock::iterator InsertPos = Def;
230 EntryMBB->insert(llvm::next(InsertPos), MI);
231 }
232 }
233
234 // Release function-specific state. SDB and CurDAG are already cleared
235 // at this point.
236 FuncInfo->clear();
Dan Gohman8a110532008-09-05 22:59:21 +0000237
Chris Lattner1c08c712005-01-07 07:47:53 +0000238 return true;
239}
240
Dan Gohmana9a33212010-04-20 00:29:35 +0000241MachineBasicBlock *
242SelectionDAGISel::SelectBasicBlock(MachineBasicBlock *BB,
243 const BasicBlock *LLVMBB,
244 BasicBlock::const_iterator Begin,
245 BasicBlock::const_iterator End,
246 bool &HadTailCall) {
Dan Gohman98ca4f22009-08-05 01:29:28 +0000247 // Lower all of the non-terminator instructions. If a call is emitted
Dan Gohmanf89d1dc2010-04-16 05:06:56 +0000248 // as a tail call, cease emitting nodes for this block. Terminators
249 // are handled below.
Dan Gohmanc105a2b2010-04-22 20:55:53 +0000250 for (BasicBlock::const_iterator I = Begin; I != End && !SDB->HasTailCall; ++I)
Dan Gohmanf89d1dc2010-04-16 05:06:56 +0000251 SDB->visit(*I);
Dan Gohmanf350b272008-08-23 02:25:05 +0000252
Chris Lattnera651cf62005-01-17 19:43:36 +0000253 // Make sure the root of the DAG is up-to-date.
Dan Gohman2048b852009-11-23 18:04:58 +0000254 CurDAG->setRoot(SDB->getControlRoot());
Arnold Schwaighoferc85e1712007-10-11 19:40:01 +0000255
Dan Gohmanf350b272008-08-23 02:25:05 +0000256 // Final step, emit the lowered DAG as machine code.
Dan Gohmana9a33212010-04-20 00:29:35 +0000257 BB = CodeGenAndEmitDAG(BB);
Dan Gohman2048b852009-11-23 18:04:58 +0000258 HadTailCall = SDB->HasTailCall;
259 SDB->clear();
Dan Gohmana9a33212010-04-20 00:29:35 +0000260 return BB;
Chris Lattner1c08c712005-01-07 07:47:53 +0000261}
262
Evan Cheng54e146b2010-01-09 00:21:08 +0000263namespace {
264/// WorkListRemover - This class is a DAGUpdateListener that removes any deleted
265/// nodes from the worklist.
266class SDOPsWorkListRemover : public SelectionDAG::DAGUpdateListener {
267 SmallVector<SDNode*, 128> &Worklist;
Chris Lattner25e0ab92010-03-14 19:43:04 +0000268 SmallPtrSet<SDNode*, 128> &InWorklist;
Evan Cheng54e146b2010-01-09 00:21:08 +0000269public:
Chris Lattner25e0ab92010-03-14 19:43:04 +0000270 SDOPsWorkListRemover(SmallVector<SDNode*, 128> &wl,
271 SmallPtrSet<SDNode*, 128> &inwl)
272 : Worklist(wl), InWorklist(inwl) {}
Evan Cheng54e146b2010-01-09 00:21:08 +0000273
Chris Lattner25e0ab92010-03-14 19:43:04 +0000274 void RemoveFromWorklist(SDNode *N) {
275 if (!InWorklist.erase(N)) return;
276
277 SmallVector<SDNode*, 128>::iterator I =
278 std::find(Worklist.begin(), Worklist.end(), N);
279 assert(I != Worklist.end() && "Not in worklist");
280
281 *I = Worklist.back();
282 Worklist.pop_back();
283 }
284
Evan Cheng54e146b2010-01-09 00:21:08 +0000285 virtual void NodeDeleted(SDNode *N, SDNode *E) {
Chris Lattner25e0ab92010-03-14 19:43:04 +0000286 RemoveFromWorklist(N);
Evan Cheng54e146b2010-01-09 00:21:08 +0000287 }
288
289 virtual void NodeUpdated(SDNode *N) {
290 // Ignore updates.
291 }
292};
293}
294
Evan Cheng046632f2010-02-10 02:17:34 +0000295/// TrivialTruncElim - Eliminate some trivial nops that can result from
296/// ShrinkDemandedOps: (trunc (ext n)) -> n.
297static bool TrivialTruncElim(SDValue Op,
298 TargetLowering::TargetLoweringOpt &TLO) {
299 SDValue N0 = Op.getOperand(0);
300 EVT VT = Op.getValueType();
301 if ((N0.getOpcode() == ISD::ZERO_EXTEND ||
302 N0.getOpcode() == ISD::SIGN_EXTEND ||
303 N0.getOpcode() == ISD::ANY_EXTEND) &&
304 N0.getOperand(0).getValueType() == VT) {
305 return TLO.CombineTo(Op, N0.getOperand(0));
306 }
307 return false;
308}
309
Evan Cheng54eb4c22010-01-06 19:43:21 +0000310/// ShrinkDemandedOps - A late transformation pass that shrink expressions
311/// using TargetLowering::TargetLoweringOpt::ShrinkDemandedOp. It converts
312/// x+y to (VT)((SmallVT)x+(SmallVT)y) if the casts are free.
Evan Chengd40d03e2010-01-06 19:38:29 +0000313void SelectionDAGISel::ShrinkDemandedOps() {
314 SmallVector<SDNode*, 128> Worklist;
Chris Lattner25e0ab92010-03-14 19:43:04 +0000315 SmallPtrSet<SDNode*, 128> InWorklist;
Evan Chengd40d03e2010-01-06 19:38:29 +0000316
317 // Add all the dag nodes to the worklist.
318 Worklist.reserve(CurDAG->allnodes_size());
319 for (SelectionDAG::allnodes_iterator I = CurDAG->allnodes_begin(),
Chris Lattner25e0ab92010-03-14 19:43:04 +0000320 E = CurDAG->allnodes_end(); I != E; ++I) {
Evan Chengd40d03e2010-01-06 19:38:29 +0000321 Worklist.push_back(I);
Chris Lattner25e0ab92010-03-14 19:43:04 +0000322 InWorklist.insert(I);
323 }
Evan Chengd40d03e2010-01-06 19:38:29 +0000324
Evan Chenge5b51ac2010-04-17 06:13:15 +0000325 TargetLowering::TargetLoweringOpt TLO(*CurDAG, true, true, true);
Evan Chengd40d03e2010-01-06 19:38:29 +0000326 while (!Worklist.empty()) {
Benjamin Kramer7b1e2a52010-01-07 17:27:56 +0000327 SDNode *N = Worklist.pop_back_val();
Chris Lattner25e0ab92010-03-14 19:43:04 +0000328 InWorklist.erase(N);
Evan Chengd40d03e2010-01-06 19:38:29 +0000329
330 if (N->use_empty() && N != CurDAG->getRoot().getNode()) {
Chris Lattnerc4a3f232010-03-14 19:46:02 +0000331 // Deleting this node may make its operands dead, add them to the worklist
332 // if they aren't already there.
333 for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i)
334 if (InWorklist.insert(N->getOperand(i).getNode()))
335 Worklist.push_back(N->getOperand(i).getNode());
336
Evan Cheng54e146b2010-01-09 00:21:08 +0000337 CurDAG->DeleteNode(N);
Evan Chengd40d03e2010-01-06 19:38:29 +0000338 continue;
339 }
340
341 // Run ShrinkDemandedOp on scalar binary operations.
Chris Lattner25e0ab92010-03-14 19:43:04 +0000342 if (N->getNumValues() != 1 ||
343 !N->getValueType(0).isSimple() || !N->getValueType(0).isInteger())
344 continue;
345
346 unsigned BitWidth = N->getValueType(0).getScalarType().getSizeInBits();
347 APInt Demanded = APInt::getAllOnesValue(BitWidth);
348 APInt KnownZero, KnownOne;
349 if (!TLI.SimplifyDemandedBits(SDValue(N, 0), Demanded,
350 KnownZero, KnownOne, TLO) &&
351 (N->getOpcode() != ISD::TRUNCATE ||
352 !TrivialTruncElim(SDValue(N, 0), TLO)))
353 continue;
354
355 // Revisit the node.
356 assert(!InWorklist.count(N) && "Already in worklist");
357 Worklist.push_back(N);
358 InWorklist.insert(N);
Evan Chengd40d03e2010-01-06 19:38:29 +0000359
Chris Lattner25e0ab92010-03-14 19:43:04 +0000360 // Replace the old value with the new one.
361 DEBUG(errs() << "\nShrinkDemandedOps replacing ";
362 TLO.Old.getNode()->dump(CurDAG);
363 errs() << "\nWith: ";
364 TLO.New.getNode()->dump(CurDAG);
365 errs() << '\n');
Evan Chengd40d03e2010-01-06 19:38:29 +0000366
Chris Lattner25e0ab92010-03-14 19:43:04 +0000367 if (InWorklist.insert(TLO.New.getNode()))
368 Worklist.push_back(TLO.New.getNode());
Evan Cheng54e146b2010-01-09 00:21:08 +0000369
Chris Lattner25e0ab92010-03-14 19:43:04 +0000370 SDOPsWorkListRemover DeadNodes(Worklist, InWorklist);
371 CurDAG->ReplaceAllUsesOfValueWith(TLO.Old, TLO.New, &DeadNodes);
Evan Chengd40d03e2010-01-06 19:38:29 +0000372
Chris Lattner25e0ab92010-03-14 19:43:04 +0000373 if (!TLO.Old.getNode()->use_empty()) continue;
374
375 for (unsigned i = 0, e = TLO.Old.getNode()->getNumOperands();
376 i != e; ++i) {
377 SDNode *OpNode = TLO.Old.getNode()->getOperand(i).getNode();
378 if (OpNode->hasOneUse()) {
379 // Add OpNode to the end of the list to revisit.
380 DeadNodes.RemoveFromWorklist(OpNode);
381 Worklist.push_back(OpNode);
382 InWorklist.insert(OpNode);
Evan Chengd40d03e2010-01-06 19:38:29 +0000383 }
384 }
Chris Lattner25e0ab92010-03-14 19:43:04 +0000385
386 DeadNodes.RemoveFromWorklist(TLO.Old.getNode());
387 CurDAG->DeleteNode(TLO.Old.getNode());
Evan Chengd40d03e2010-01-06 19:38:29 +0000388 }
389}
390
Dan Gohmanf350b272008-08-23 02:25:05 +0000391void SelectionDAGISel::ComputeLiveOutVRegInfo() {
Chris Lattneread0d882008-06-17 06:09:18 +0000392 SmallPtrSet<SDNode*, 128> VisitedNodes;
393 SmallVector<SDNode*, 128> Worklist;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000394
Gabor Greifba36cb52008-08-28 21:40:38 +0000395 Worklist.push_back(CurDAG->getRoot().getNode());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000396
Chris Lattneread0d882008-06-17 06:09:18 +0000397 APInt Mask;
398 APInt KnownZero;
399 APInt KnownOne;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000400
Benjamin Kramer7b1e2a52010-01-07 17:27:56 +0000401 do {
402 SDNode *N = Worklist.pop_back_val();
Daniel Dunbara279bc32009-09-20 02:20:51 +0000403
Chris Lattneread0d882008-06-17 06:09:18 +0000404 // If we've already seen this node, ignore it.
405 if (!VisitedNodes.insert(N))
406 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000407
Chris Lattneread0d882008-06-17 06:09:18 +0000408 // Otherwise, add all chain operands to the worklist.
409 for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i)
Owen Anderson825b72b2009-08-11 20:47:22 +0000410 if (N->getOperand(i).getValueType() == MVT::Other)
Gabor Greifba36cb52008-08-28 21:40:38 +0000411 Worklist.push_back(N->getOperand(i).getNode());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000412
Chris Lattneread0d882008-06-17 06:09:18 +0000413 // If this is a CopyToReg with a vreg dest, process it.
414 if (N->getOpcode() != ISD::CopyToReg)
415 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000416
Chris Lattneread0d882008-06-17 06:09:18 +0000417 unsigned DestReg = cast<RegisterSDNode>(N->getOperand(1))->getReg();
418 if (!TargetRegisterInfo::isVirtualRegister(DestReg))
419 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000420
Chris Lattneread0d882008-06-17 06:09:18 +0000421 // Ignore non-scalar or non-integer values.
Dan Gohman475871a2008-07-27 21:46:04 +0000422 SDValue Src = N->getOperand(2);
Owen Andersone50ed302009-08-10 22:56:29 +0000423 EVT SrcVT = Src.getValueType();
Chris Lattneread0d882008-06-17 06:09:18 +0000424 if (!SrcVT.isInteger() || SrcVT.isVector())
425 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000426
Dan Gohmanf350b272008-08-23 02:25:05 +0000427 unsigned NumSignBits = CurDAG->ComputeNumSignBits(Src);
Chris Lattneread0d882008-06-17 06:09:18 +0000428 Mask = APInt::getAllOnesValue(SrcVT.getSizeInBits());
Dan Gohmanf350b272008-08-23 02:25:05 +0000429 CurDAG->ComputeMaskedBits(Src, Mask, KnownZero, KnownOne);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000430
Chris Lattneread0d882008-06-17 06:09:18 +0000431 // Only install this information if it tells us something.
432 if (NumSignBits != 1 || KnownZero != 0 || KnownOne != 0) {
433 DestReg -= TargetRegisterInfo::FirstVirtualRegister;
Dan Gohmanf7d6cd42009-08-01 03:51:09 +0000434 if (DestReg >= FuncInfo->LiveOutRegInfo.size())
435 FuncInfo->LiveOutRegInfo.resize(DestReg+1);
436 FunctionLoweringInfo::LiveOutInfo &LOI =
437 FuncInfo->LiveOutRegInfo[DestReg];
Chris Lattneread0d882008-06-17 06:09:18 +0000438 LOI.NumSignBits = NumSignBits;
Dan Gohmana80efce2009-03-27 23:55:04 +0000439 LOI.KnownOne = KnownOne;
440 LOI.KnownZero = KnownZero;
Chris Lattneread0d882008-06-17 06:09:18 +0000441 }
Benjamin Kramer7b1e2a52010-01-07 17:27:56 +0000442 } while (!Worklist.empty());
Chris Lattneread0d882008-06-17 06:09:18 +0000443}
444
Dan Gohmana9a33212010-04-20 00:29:35 +0000445MachineBasicBlock *SelectionDAGISel::CodeGenAndEmitDAG(MachineBasicBlock *BB) {
Dan Gohman462dc7f2008-07-21 20:00:07 +0000446 std::string GroupName;
447 if (TimePassesIsEnabled)
448 GroupName = "Instruction Selection and Scheduling";
449 std::string BlockName;
450 if (ViewDAGCombine1 || ViewLegalizeTypesDAGs || ViewLegalizeDAGs ||
Duncan Sands25cf2272008-11-24 14:53:14 +0000451 ViewDAGCombine2 || ViewDAGCombineLT || ViewISelDAGs || ViewSchedDAGs ||
452 ViewSUnitDAGs)
Dan Gohmanf7d6cd42009-08-01 03:51:09 +0000453 BlockName = MF->getFunction()->getNameStr() + ":" +
Daniel Dunbarf6ccee52009-07-24 08:24:36 +0000454 BB->getBasicBlock()->getNameStr();
Dan Gohman462dc7f2008-07-21 20:00:07 +0000455
David Greene1a053232010-01-05 01:26:11 +0000456 DEBUG(dbgs() << "Initial selection DAG:\n");
Dan Gohmanf350b272008-08-23 02:25:05 +0000457 DEBUG(CurDAG->dump());
Dan Gohman462dc7f2008-07-21 20:00:07 +0000458
Dan Gohmanf350b272008-08-23 02:25:05 +0000459 if (ViewDAGCombine1) CurDAG->viewGraph("dag-combine1 input for " + BlockName);
Dan Gohman417e11b2007-10-08 15:12:17 +0000460
Chris Lattneraf21d552005-10-10 16:47:10 +0000461 // Run the DAG combiner in pre-legalize mode.
Evan Chengebffb662008-07-01 17:59:20 +0000462 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000463 NamedRegionTimer T("DAG Combining 1", GroupName);
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000464 CurDAG->Combine(Unrestricted, *AA, OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000465 } else {
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000466 CurDAG->Combine(Unrestricted, *AA, OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000467 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000468
David Greene1a053232010-01-05 01:26:11 +0000469 DEBUG(dbgs() << "Optimized lowered selection DAG:\n");
Dan Gohmanf350b272008-08-23 02:25:05 +0000470 DEBUG(CurDAG->dump());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000471
Chris Lattner1c08c712005-01-07 07:47:53 +0000472 // Second step, hack on the DAG until it only uses operations and types that
473 // the target supports.
Dan Gohman714efc62009-12-05 17:51:33 +0000474 if (ViewLegalizeTypesDAGs) CurDAG->viewGraph("legalize-types input for " +
475 BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000476
Dan Gohman714efc62009-12-05 17:51:33 +0000477 bool Changed;
478 if (TimePassesIsEnabled) {
479 NamedRegionTimer T("Type Legalization", GroupName);
480 Changed = CurDAG->LegalizeTypes();
481 } else {
482 Changed = CurDAG->LegalizeTypes();
483 }
484
David Greene1a053232010-01-05 01:26:11 +0000485 DEBUG(dbgs() << "Type-legalized selection DAG:\n");
Dan Gohman714efc62009-12-05 17:51:33 +0000486 DEBUG(CurDAG->dump());
487
488 if (Changed) {
489 if (ViewDAGCombineLT)
490 CurDAG->viewGraph("dag-combine-lt input for " + BlockName);
491
492 // Run the DAG combiner in post-type-legalize mode.
Dan Gohman462dc7f2008-07-21 20:00:07 +0000493 if (TimePassesIsEnabled) {
Dan Gohman714efc62009-12-05 17:51:33 +0000494 NamedRegionTimer T("DAG Combining after legalize types", GroupName);
495 CurDAG->Combine(NoIllegalTypes, *AA, OptLevel);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000496 } else {
Dan Gohman714efc62009-12-05 17:51:33 +0000497 CurDAG->Combine(NoIllegalTypes, *AA, OptLevel);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000498 }
499
David Greene1a053232010-01-05 01:26:11 +0000500 DEBUG(dbgs() << "Optimized type-legalized selection DAG:\n");
Dan Gohmanf350b272008-08-23 02:25:05 +0000501 DEBUG(CurDAG->dump());
Dan Gohman714efc62009-12-05 17:51:33 +0000502 }
Dan Gohman462dc7f2008-07-21 20:00:07 +0000503
Dan Gohman714efc62009-12-05 17:51:33 +0000504 if (TimePassesIsEnabled) {
505 NamedRegionTimer T("Vector Legalization", GroupName);
506 Changed = CurDAG->LegalizeVectors();
507 } else {
508 Changed = CurDAG->LegalizeVectors();
509 }
Duncan Sands25cf2272008-11-24 14:53:14 +0000510
Dan Gohman714efc62009-12-05 17:51:33 +0000511 if (Changed) {
Eli Friedman5c22c802009-05-23 12:35:30 +0000512 if (TimePassesIsEnabled) {
Dan Gohman714efc62009-12-05 17:51:33 +0000513 NamedRegionTimer T("Type Legalization 2", GroupName);
Bill Wendling98820072009-12-28 01:51:30 +0000514 CurDAG->LegalizeTypes();
Eli Friedman5c22c802009-05-23 12:35:30 +0000515 } else {
Bill Wendling98820072009-12-28 01:51:30 +0000516 CurDAG->LegalizeTypes();
Eli Friedman5c22c802009-05-23 12:35:30 +0000517 }
518
Dan Gohman714efc62009-12-05 17:51:33 +0000519 if (ViewDAGCombineLT)
520 CurDAG->viewGraph("dag-combine-lv input for " + BlockName);
Eli Friedman5c22c802009-05-23 12:35:30 +0000521
Dan Gohman714efc62009-12-05 17:51:33 +0000522 // Run the DAG combiner in post-type-legalize mode.
523 if (TimePassesIsEnabled) {
524 NamedRegionTimer T("DAG Combining after legalize vectors", GroupName);
525 CurDAG->Combine(NoIllegalOperations, *AA, OptLevel);
526 } else {
527 CurDAG->Combine(NoIllegalOperations, *AA, OptLevel);
Eli Friedman5c22c802009-05-23 12:35:30 +0000528 }
Dan Gohman714efc62009-12-05 17:51:33 +0000529
David Greene1a053232010-01-05 01:26:11 +0000530 DEBUG(dbgs() << "Optimized vector-legalized selection DAG:\n");
Dan Gohman714efc62009-12-05 17:51:33 +0000531 DEBUG(CurDAG->dump());
Chris Lattner70587ea2008-07-10 23:37:50 +0000532 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000533
Dan Gohmanf350b272008-08-23 02:25:05 +0000534 if (ViewLegalizeDAGs) CurDAG->viewGraph("legalize input for " + BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000535
Evan Chengebffb662008-07-01 17:59:20 +0000536 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000537 NamedRegionTimer T("DAG Legalization", GroupName);
Dan Gohman714efc62009-12-05 17:51:33 +0000538 CurDAG->Legalize(OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000539 } else {
Dan Gohman714efc62009-12-05 17:51:33 +0000540 CurDAG->Legalize(OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000541 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000542
David Greene1a053232010-01-05 01:26:11 +0000543 DEBUG(dbgs() << "Legalized selection DAG:\n");
Dan Gohmanf350b272008-08-23 02:25:05 +0000544 DEBUG(CurDAG->dump());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000545
Dan Gohmanf350b272008-08-23 02:25:05 +0000546 if (ViewDAGCombine2) CurDAG->viewGraph("dag-combine2 input for " + BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000547
Chris Lattneraf21d552005-10-10 16:47:10 +0000548 // Run the DAG combiner in post-legalize mode.
Evan Chengebffb662008-07-01 17:59:20 +0000549 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000550 NamedRegionTimer T("DAG Combining 2", GroupName);
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000551 CurDAG->Combine(NoIllegalOperations, *AA, OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000552 } else {
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000553 CurDAG->Combine(NoIllegalOperations, *AA, OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000554 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000555
David Greene1a053232010-01-05 01:26:11 +0000556 DEBUG(dbgs() << "Optimized legalized selection DAG:\n");
Dan Gohmanf350b272008-08-23 02:25:05 +0000557 DEBUG(CurDAG->dump());
Dan Gohman417e11b2007-10-08 15:12:17 +0000558
Evan Chengd40d03e2010-01-06 19:38:29 +0000559 if (OptLevel != CodeGenOpt::None) {
560 ShrinkDemandedOps();
Dan Gohmanf350b272008-08-23 02:25:05 +0000561 ComputeLiveOutVRegInfo();
Evan Chengd40d03e2010-01-06 19:38:29 +0000562 }
Evan Cheng552c4a82006-04-28 02:09:19 +0000563
Chris Lattner552186d2010-03-14 19:27:55 +0000564 if (ViewISelDAGs) CurDAG->viewGraph("isel input for " + BlockName);
565
Chris Lattnera33ef482005-03-30 01:10:47 +0000566 // Third, instruction select all of the operations to machine code, adding the
567 // code to the MachineBasicBlock.
Evan Chengebffb662008-07-01 17:59:20 +0000568 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000569 NamedRegionTimer T("Instruction Selection", GroupName);
Chris Lattner7c306da2010-03-02 06:34:30 +0000570 DoInstructionSelection();
Evan Chengebffb662008-07-01 17:59:20 +0000571 } else {
Chris Lattner7c306da2010-03-02 06:34:30 +0000572 DoInstructionSelection();
Evan Chengebffb662008-07-01 17:59:20 +0000573 }
Evan Chengdb8d56b2008-06-30 20:45:06 +0000574
David Greene1a053232010-01-05 01:26:11 +0000575 DEBUG(dbgs() << "Selected selection DAG:\n");
Dan Gohmanf350b272008-08-23 02:25:05 +0000576 DEBUG(CurDAG->dump());
Dan Gohman462dc7f2008-07-21 20:00:07 +0000577
Dan Gohmanf350b272008-08-23 02:25:05 +0000578 if (ViewSchedDAGs) CurDAG->viewGraph("scheduler input for " + BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000579
Dan Gohman5e843682008-07-14 18:19:29 +0000580 // Schedule machine code.
Dan Gohman47ac0f02009-02-11 04:27:20 +0000581 ScheduleDAGSDNodes *Scheduler = CreateScheduler();
Dan Gohman5e843682008-07-14 18:19:29 +0000582 if (TimePassesIsEnabled) {
583 NamedRegionTimer T("Instruction Scheduling", GroupName);
Dan Gohman47ac0f02009-02-11 04:27:20 +0000584 Scheduler->Run(CurDAG, BB, BB->end());
Dan Gohman5e843682008-07-14 18:19:29 +0000585 } else {
Dan Gohman47ac0f02009-02-11 04:27:20 +0000586 Scheduler->Run(CurDAG, BB, BB->end());
Dan Gohman5e843682008-07-14 18:19:29 +0000587 }
588
Dan Gohman462dc7f2008-07-21 20:00:07 +0000589 if (ViewSUnitDAGs) Scheduler->viewGraph();
590
Daniel Dunbara279bc32009-09-20 02:20:51 +0000591 // Emit machine code to BB. This can change 'BB' to the last block being
Evan Chengdb8d56b2008-06-30 20:45:06 +0000592 // inserted into.
Evan Chengebffb662008-07-01 17:59:20 +0000593 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000594 NamedRegionTimer T("Instruction Creation", GroupName);
Dan Gohmanaf1d8ca2010-05-01 00:01:06 +0000595 BB = Scheduler->EmitSchedule();
Evan Chengebffb662008-07-01 17:59:20 +0000596 } else {
Dan Gohmanaf1d8ca2010-05-01 00:01:06 +0000597 BB = Scheduler->EmitSchedule();
Dan Gohman5e843682008-07-14 18:19:29 +0000598 }
599
600 // Free the scheduler state.
601 if (TimePassesIsEnabled) {
602 NamedRegionTimer T("Instruction Scheduling Cleanup", GroupName);
603 delete Scheduler;
604 } else {
605 delete Scheduler;
Evan Chengebffb662008-07-01 17:59:20 +0000606 }
Evan Chengdb8d56b2008-06-30 20:45:06 +0000607
Dan Gohmana9a33212010-04-20 00:29:35 +0000608 return BB;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000609}
Chris Lattner1c08c712005-01-07 07:47:53 +0000610
Chris Lattner7c306da2010-03-02 06:34:30 +0000611void SelectionDAGISel::DoInstructionSelection() {
612 DEBUG(errs() << "===== Instruction selection begins:\n");
613
614 PreprocessISelDAG();
615
616 // Select target instructions for the DAG.
617 {
618 // Number all nodes with a topological order and set DAGSize.
619 DAGSize = CurDAG->AssignTopologicalOrder();
620
621 // Create a dummy node (which is not added to allnodes), that adds
622 // a reference to the root node, preventing it from being deleted,
623 // and tracking any changes of the root.
624 HandleSDNode Dummy(CurDAG->getRoot());
625 ISelPosition = SelectionDAG::allnodes_iterator(CurDAG->getRoot().getNode());
626 ++ISelPosition;
627
628 // The AllNodes list is now topological-sorted. Visit the
629 // nodes by starting at the end of the list (the root of the
630 // graph) and preceding back toward the beginning (the entry
631 // node).
632 while (ISelPosition != CurDAG->allnodes_begin()) {
633 SDNode *Node = --ISelPosition;
634 // Skip dead nodes. DAGCombiner is expected to eliminate all dead nodes,
635 // but there are currently some corner cases that it misses. Also, this
636 // makes it theoretically possible to disable the DAGCombiner.
637 if (Node->use_empty())
638 continue;
639
640 SDNode *ResNode = Select(Node);
641
Chris Lattner82dd3d32010-03-02 07:50:03 +0000642 // FIXME: This is pretty gross. 'Select' should be changed to not return
643 // anything at all and this code should be nuked with a tactical strike.
644
Chris Lattner7c306da2010-03-02 06:34:30 +0000645 // If node should not be replaced, continue with the next one.
Chris Lattner82dd3d32010-03-02 07:50:03 +0000646 if (ResNode == Node || Node->getOpcode() == ISD::DELETED_NODE)
Chris Lattner7c306da2010-03-02 06:34:30 +0000647 continue;
648 // Replace node.
649 if (ResNode)
650 ReplaceUses(Node, ResNode);
651
652 // If after the replacement this node is not used any more,
653 // remove this dead node.
654 if (Node->use_empty()) { // Don't delete EntryToken, etc.
655 ISelUpdater ISU(ISelPosition);
656 CurDAG->RemoveDeadNode(Node, &ISU);
657 }
658 }
659
660 CurDAG->setRoot(Dummy.getValue());
661 }
662 DEBUG(errs() << "===== Instruction selection ends:\n");
663
664 PostprocessISelDAG();
Chris Lattner7c306da2010-03-02 06:34:30 +0000665}
666
Dan Gohman25208642010-04-14 19:53:31 +0000667/// PrepareEHLandingPad - Emit an EH_LABEL, set up live-in registers, and
668/// do other setup for EH landing-pad blocks.
669void SelectionDAGISel::PrepareEHLandingPad(MachineBasicBlock *BB) {
670 // Add a label to mark the beginning of the landing pad. Deletion of the
671 // landing pad can thus be detected via the MachineModuleInfo.
672 MCSymbol *Label = MF->getMMI().addLandingPad(BB);
673
Dan Gohman55e59c12010-04-19 19:05:59 +0000674 const TargetInstrDesc &II = TM.getInstrInfo()->get(TargetOpcode::EH_LABEL);
Dan Gohman25208642010-04-14 19:53:31 +0000675 BuildMI(BB, SDB->getCurDebugLoc(), II).addSym(Label);
676
677 // Mark exception register as live in.
678 unsigned Reg = TLI.getExceptionAddressRegister();
679 if (Reg) BB->addLiveIn(Reg);
680
681 // Mark exception selector register as live in.
682 Reg = TLI.getExceptionSelectorRegister();
683 if (Reg) BB->addLiveIn(Reg);
684
685 // FIXME: Hack around an exception handling flaw (PR1508): the personality
686 // function and list of typeids logically belong to the invoke (or, if you
687 // like, the basic block containing the invoke), and need to be associated
688 // with it in the dwarf exception handling tables. Currently however the
689 // information is provided by an intrinsic (eh.selector) that can be moved
690 // to unexpected places by the optimizers: if the unwind edge is critical,
691 // then breaking it can result in the intrinsics being in the successor of
692 // the landing pad, not the landing pad itself. This results
693 // in exceptions not being caught because no typeids are associated with
694 // the invoke. This may not be the only way things can go wrong, but it
695 // is the only way we try to work around for the moment.
696 const BasicBlock *LLVMBB = BB->getBasicBlock();
697 const BranchInst *Br = dyn_cast<BranchInst>(LLVMBB->getTerminator());
698
699 if (Br && Br->isUnconditional()) { // Critical edge?
700 BasicBlock::const_iterator I, E;
701 for (I = LLVMBB->begin(), E = --LLVMBB->end(); I != E; ++I)
702 if (isa<EHSelectorInst>(I))
703 break;
704
705 if (I == E)
706 // No catch info found - try to extract some from the successor.
707 CopyCatchInfo(Br->getSuccessor(0), LLVMBB, &MF->getMMI(), *FuncInfo);
708 }
709}
Chris Lattner7c306da2010-03-02 06:34:30 +0000710
Dan Gohman46510a72010-04-15 01:51:59 +0000711void SelectionDAGISel::SelectAllBasicBlocks(const Function &Fn) {
Dan Gohmana43abd12008-09-29 21:55:50 +0000712 // Initialize the Fast-ISel state, if needed.
713 FastISel *FastIS = 0;
714 if (EnableFastISel)
Dan Gohman6a732b52010-04-14 20:05:00 +0000715 FastIS = TLI.createFastISel(*MF, FuncInfo->ValueMap, FuncInfo->MBBMap,
Dan Gohmanf81eca02010-04-22 20:46:50 +0000716 FuncInfo->StaticAllocaMap,
717 FuncInfo->PHINodesToUpdate
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000718#ifndef NDEBUG
719 , FuncInfo->CatchInfoLost
720#endif
721 );
Dan Gohmana43abd12008-09-29 21:55:50 +0000722
723 // Iterate over all basic blocks in the function.
Dan Gohman46510a72010-04-15 01:51:59 +0000724 for (Function::const_iterator I = Fn.begin(), E = Fn.end(); I != E; ++I) {
725 const BasicBlock *LLVMBB = &*I;
Dan Gohmana9a33212010-04-20 00:29:35 +0000726 MachineBasicBlock *BB = FuncInfo->MBBMap[LLVMBB];
Dan Gohmanf350b272008-08-23 02:25:05 +0000727
Dan Gohmanba5be5c2010-04-20 15:00:41 +0000728 BasicBlock::const_iterator const Begin = LLVMBB->getFirstNonPHI();
Dan Gohman46510a72010-04-15 01:51:59 +0000729 BasicBlock::const_iterator const End = LLVMBB->end();
730 BasicBlock::const_iterator BI = Begin;
Dan Gohman5edd3612008-08-28 20:28:56 +0000731
732 // Lower any arguments needed in this block if this is the entry block.
Dan Gohman33134c42008-09-25 17:05:24 +0000733 bool SuppressFastISel = false;
734 if (LLVMBB == &Fn.getEntryBlock()) {
Dan Gohman5edd3612008-08-28 20:28:56 +0000735 LowerArguments(LLVMBB);
Dan Gohmanf350b272008-08-23 02:25:05 +0000736
Dan Gohman33134c42008-09-25 17:05:24 +0000737 // If any of the arguments has the byval attribute, forgo
738 // fast-isel in the entry block.
Dan Gohmana43abd12008-09-29 21:55:50 +0000739 if (FastIS) {
Dan Gohman33134c42008-09-25 17:05:24 +0000740 unsigned j = 1;
Dan Gohman46510a72010-04-15 01:51:59 +0000741 for (Function::const_arg_iterator I = Fn.arg_begin(), E = Fn.arg_end();
Dan Gohman33134c42008-09-25 17:05:24 +0000742 I != E; ++I, ++j)
Devang Patel05988662008-09-25 21:00:45 +0000743 if (Fn.paramHasAttr(j, Attribute::ByVal)) {
Dan Gohman77ca41e2008-09-25 17:21:42 +0000744 if (EnableFastISelVerbose || EnableFastISelAbort)
David Greene1a053232010-01-05 01:26:11 +0000745 dbgs() << "FastISel skips entry block due to byval argument\n";
Dan Gohman33134c42008-09-25 17:05:24 +0000746 SuppressFastISel = true;
747 break;
748 }
749 }
750 }
751
Dan Gohman25208642010-04-14 19:53:31 +0000752 // Setup an EH landing-pad block.
753 if (BB->isLandingPad())
754 PrepareEHLandingPad(BB);
755
Dan Gohmanf350b272008-08-23 02:25:05 +0000756 // Before doing SelectionDAG ISel, see if FastISel has been requested.
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000757 if (FastIS && !SuppressFastISel) {
Dan Gohmana43abd12008-09-29 21:55:50 +0000758 // Emit code for any incoming arguments. This must happen before
759 // beginning FastISel on the entry block.
760 if (LLVMBB == &Fn.getEntryBlock()) {
Dan Gohman2048b852009-11-23 18:04:58 +0000761 CurDAG->setRoot(SDB->getControlRoot());
Dan Gohmana9a33212010-04-20 00:29:35 +0000762 BB = CodeGenAndEmitDAG(BB);
Dan Gohman2048b852009-11-23 18:04:58 +0000763 SDB->clear();
Dan Gohmana43abd12008-09-29 21:55:50 +0000764 }
Dan Gohman241f4642008-10-04 00:56:36 +0000765 FastIS->startNewBlock(BB);
Dan Gohmana43abd12008-09-29 21:55:50 +0000766 // Do FastISel on as many instructions as possible.
767 for (; BI != End; ++BI) {
Dan Gohman21c14e32010-01-12 04:32:35 +0000768 // Try to select the instruction with FastISel.
Dan Gohman8ba3aa72010-04-20 00:48:35 +0000769 if (FastIS->SelectInstruction(BI))
Dan Gohmana43abd12008-09-29 21:55:50 +0000770 continue;
Dan Gohmana43abd12008-09-29 21:55:50 +0000771
772 // Then handle certain instructions as single-LLVM-Instruction blocks.
773 if (isa<CallInst>(BI)) {
Chris Lattnerc19ae9d2010-03-04 19:46:56 +0000774 ++NumFastIselFailures;
Dan Gohmana43abd12008-09-29 21:55:50 +0000775 if (EnableFastISelVerbose || EnableFastISelAbort) {
David Greene1a053232010-01-05 01:26:11 +0000776 dbgs() << "FastISel missed call: ";
Dan Gohmana43abd12008-09-29 21:55:50 +0000777 BI->dump();
778 }
779
Dan Gohman33b7a292010-04-16 17:15:02 +0000780 if (!BI->getType()->isVoidTy() && !BI->use_empty()) {
Dan Gohmana43abd12008-09-29 21:55:50 +0000781 unsigned &R = FuncInfo->ValueMap[BI];
782 if (!R)
783 R = FuncInfo->CreateRegForValue(BI);
784 }
785
Dan Gohmanb4afb132009-11-20 02:51:26 +0000786 bool HadTailCall = false;
Dan Gohmana9a33212010-04-20 00:29:35 +0000787 BB = SelectBasicBlock(BB, LLVMBB, BI, llvm::next(BI), HadTailCall);
Dan Gohmanb4afb132009-11-20 02:51:26 +0000788
789 // If the call was emitted as a tail call, we're done with the block.
790 if (HadTailCall) {
791 BI = End;
792 break;
793 }
794
Dan Gohman241f4642008-10-04 00:56:36 +0000795 // If the instruction was codegen'd with multiple blocks,
796 // inform the FastISel object where to resume inserting.
797 FastIS->setCurrentBlock(BB);
Dan Gohmana43abd12008-09-29 21:55:50 +0000798 continue;
Dan Gohmanf350b272008-08-23 02:25:05 +0000799 }
Dan Gohmana43abd12008-09-29 21:55:50 +0000800
801 // Otherwise, give up on FastISel for the rest of the block.
802 // For now, be a little lenient about non-branch terminators.
803 if (!isa<TerminatorInst>(BI) || isa<BranchInst>(BI)) {
Chris Lattnerc19ae9d2010-03-04 19:46:56 +0000804 ++NumFastIselFailures;
Dan Gohmana43abd12008-09-29 21:55:50 +0000805 if (EnableFastISelVerbose || EnableFastISelAbort) {
David Greene1a053232010-01-05 01:26:11 +0000806 dbgs() << "FastISel miss: ";
Dan Gohmana43abd12008-09-29 21:55:50 +0000807 BI->dump();
808 }
809 if (EnableFastISelAbort)
810 // The "fast" selector couldn't handle something and bailed.
811 // For the purpose of debugging, just abort.
Torok Edwinc23197a2009-07-14 16:55:14 +0000812 llvm_unreachable("FastISel didn't select the entire block");
Dan Gohmana43abd12008-09-29 21:55:50 +0000813 }
814 break;
Dan Gohmanf350b272008-08-23 02:25:05 +0000815 }
816 }
817
Dan Gohmand2ff6472008-09-02 20:17:56 +0000818 // Run SelectionDAG instruction selection on the remainder of the block
819 // not handled by FastISel. If FastISel is not run, this is the entire
Dan Gohman3df24e62008-09-03 23:12:08 +0000820 // block.
Devang Patel390f3ac2009-04-16 01:33:10 +0000821 if (BI != End) {
Dan Gohmanb4afb132009-11-20 02:51:26 +0000822 bool HadTailCall;
Dan Gohmana9a33212010-04-20 00:29:35 +0000823 BB = SelectBasicBlock(BB, LLVMBB, BI, End, HadTailCall);
Devang Patel390f3ac2009-04-16 01:33:10 +0000824 }
Dan Gohmanf350b272008-08-23 02:25:05 +0000825
Dan Gohmana9a33212010-04-20 00:29:35 +0000826 FinishBasicBlock(BB);
Dan Gohman620427d2010-04-22 19:55:20 +0000827 FuncInfo->PHINodesToUpdate.clear();
Evan Cheng39fd6e82008-08-07 00:43:25 +0000828 }
Dan Gohmana43abd12008-09-29 21:55:50 +0000829
830 delete FastIS;
Dan Gohman0e5f1302008-07-07 23:02:41 +0000831}
832
Dan Gohmanfed90b62008-07-28 21:51:04 +0000833void
Dan Gohmana9a33212010-04-20 00:29:35 +0000834SelectionDAGISel::FinishBasicBlock(MachineBasicBlock *BB) {
Dan Gohmanf350b272008-08-23 02:25:05 +0000835
David Greene1a053232010-01-05 01:26:11 +0000836 DEBUG(dbgs() << "Total amount of phi nodes to update: "
Dan Gohman620427d2010-04-22 19:55:20 +0000837 << FuncInfo->PHINodesToUpdate.size() << "\n");
838 DEBUG(for (unsigned i = 0, e = FuncInfo->PHINodesToUpdate.size(); i != e; ++i)
David Greene1a053232010-01-05 01:26:11 +0000839 dbgs() << "Node " << i << " : ("
Dan Gohman620427d2010-04-22 19:55:20 +0000840 << FuncInfo->PHINodesToUpdate[i].first
841 << ", " << FuncInfo->PHINodesToUpdate[i].second << ")\n");
Daniel Dunbara279bc32009-09-20 02:20:51 +0000842
Chris Lattnera33ef482005-03-30 01:10:47 +0000843 // Next, now that we know what the last MBB the LLVM BB expanded is, update
Chris Lattner1c08c712005-01-07 07:47:53 +0000844 // PHI nodes in successors.
Dan Gohman2048b852009-11-23 18:04:58 +0000845 if (SDB->SwitchCases.empty() &&
846 SDB->JTCases.empty() &&
847 SDB->BitTestCases.empty()) {
Dan Gohman620427d2010-04-22 19:55:20 +0000848 for (unsigned i = 0, e = FuncInfo->PHINodesToUpdate.size(); i != e; ++i) {
849 MachineInstr *PHI = FuncInfo->PHINodesToUpdate[i].first;
Chris Lattner518bb532010-02-09 19:54:29 +0000850 assert(PHI->isPHI() &&
Nate Begemanf15485a2006-03-27 01:32:24 +0000851 "This is not a machine PHI node that we are updating!");
Jakob Stoklund Olesen580bba22010-03-05 21:49:10 +0000852 if (!BB->isSuccessor(PHI->getParent()))
853 continue;
Dan Gohman620427d2010-04-22 19:55:20 +0000854 PHI->addOperand(
855 MachineOperand::CreateReg(FuncInfo->PHINodesToUpdate[i].second, false));
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000856 PHI->addOperand(MachineOperand::CreateMBB(BB));
Nate Begemanf15485a2006-03-27 01:32:24 +0000857 }
858 return;
Chris Lattner1c08c712005-01-07 07:47:53 +0000859 }
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000860
Dan Gohman2048b852009-11-23 18:04:58 +0000861 for (unsigned i = 0, e = SDB->BitTestCases.size(); i != e; ++i) {
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000862 // Lower header first, if it wasn't already lowered
Dan Gohman2048b852009-11-23 18:04:58 +0000863 if (!SDB->BitTestCases[i].Emitted) {
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000864 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman2048b852009-11-23 18:04:58 +0000865 BB = SDB->BitTestCases[i].Parent;
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000866 // Emit the code
Dan Gohman99be8ae2010-04-19 22:41:47 +0000867 SDB->visitBitTestHeader(SDB->BitTestCases[i], BB);
Dan Gohman2048b852009-11-23 18:04:58 +0000868 CurDAG->setRoot(SDB->getRoot());
Dan Gohmana9a33212010-04-20 00:29:35 +0000869 BB = CodeGenAndEmitDAG(BB);
Dan Gohman2048b852009-11-23 18:04:58 +0000870 SDB->clear();
Daniel Dunbara279bc32009-09-20 02:20:51 +0000871 }
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000872
Dan Gohman2048b852009-11-23 18:04:58 +0000873 for (unsigned j = 0, ej = SDB->BitTestCases[i].Cases.size(); j != ej; ++j) {
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000874 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman2048b852009-11-23 18:04:58 +0000875 BB = SDB->BitTestCases[i].Cases[j].ThisBB;
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000876 // Emit the code
877 if (j+1 != ej)
Dan Gohman2048b852009-11-23 18:04:58 +0000878 SDB->visitBitTestCase(SDB->BitTestCases[i].Cases[j+1].ThisBB,
879 SDB->BitTestCases[i].Reg,
Dan Gohman99be8ae2010-04-19 22:41:47 +0000880 SDB->BitTestCases[i].Cases[j],
881 BB);
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000882 else
Dan Gohman2048b852009-11-23 18:04:58 +0000883 SDB->visitBitTestCase(SDB->BitTestCases[i].Default,
884 SDB->BitTestCases[i].Reg,
Dan Gohman99be8ae2010-04-19 22:41:47 +0000885 SDB->BitTestCases[i].Cases[j],
886 BB);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000887
888
Dan Gohman2048b852009-11-23 18:04:58 +0000889 CurDAG->setRoot(SDB->getRoot());
Dan Gohmana9a33212010-04-20 00:29:35 +0000890 BB = CodeGenAndEmitDAG(BB);
Dan Gohman2048b852009-11-23 18:04:58 +0000891 SDB->clear();
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000892 }
893
894 // Update PHI Nodes
Dan Gohman620427d2010-04-22 19:55:20 +0000895 for (unsigned pi = 0, pe = FuncInfo->PHINodesToUpdate.size();
896 pi != pe; ++pi) {
897 MachineInstr *PHI = FuncInfo->PHINodesToUpdate[pi].first;
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000898 MachineBasicBlock *PHIBB = PHI->getParent();
Chris Lattner518bb532010-02-09 19:54:29 +0000899 assert(PHI->isPHI() &&
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000900 "This is not a machine PHI node that we are updating!");
901 // This is "default" BB. We have two jumps to it. From "header" BB and
902 // from last "case" BB.
Dan Gohman2048b852009-11-23 18:04:58 +0000903 if (PHIBB == SDB->BitTestCases[i].Default) {
Jim Grosbachf4549b02010-01-15 00:36:15 +0000904 PHI->addOperand(MachineOperand::
Dan Gohman620427d2010-04-22 19:55:20 +0000905 CreateReg(FuncInfo->PHINodesToUpdate[pi].second,
906 false));
Dan Gohman2048b852009-11-23 18:04:58 +0000907 PHI->addOperand(MachineOperand::CreateMBB(SDB->BitTestCases[i].Parent));
Jim Grosbachf4549b02010-01-15 00:36:15 +0000908 PHI->addOperand(MachineOperand::
Dan Gohman620427d2010-04-22 19:55:20 +0000909 CreateReg(FuncInfo->PHINodesToUpdate[pi].second,
910 false));
Dan Gohman2048b852009-11-23 18:04:58 +0000911 PHI->addOperand(MachineOperand::CreateMBB(SDB->BitTestCases[i].Cases.
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000912 back().ThisBB));
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000913 }
914 // One of "cases" BB.
Dan Gohman2048b852009-11-23 18:04:58 +0000915 for (unsigned j = 0, ej = SDB->BitTestCases[i].Cases.size();
Dan Gohman7c3234c2008-08-27 23:52:12 +0000916 j != ej; ++j) {
Dan Gohman2048b852009-11-23 18:04:58 +0000917 MachineBasicBlock* cBB = SDB->BitTestCases[i].Cases[j].ThisBB;
Jakob Stoklund Olesendd437ba2010-01-11 21:02:33 +0000918 if (cBB->isSuccessor(PHIBB)) {
Jim Grosbachf4549b02010-01-15 00:36:15 +0000919 PHI->addOperand(MachineOperand::
Dan Gohman620427d2010-04-22 19:55:20 +0000920 CreateReg(FuncInfo->PHINodesToUpdate[pi].second,
921 false));
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000922 PHI->addOperand(MachineOperand::CreateMBB(cBB));
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000923 }
924 }
925 }
926 }
Dan Gohman2048b852009-11-23 18:04:58 +0000927 SDB->BitTestCases.clear();
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000928
Nate Begeman9453eea2006-04-23 06:26:20 +0000929 // If the JumpTable record is filled in, then we need to emit a jump table.
930 // Updating the PHI nodes is tricky in this case, since we need to determine
931 // whether the PHI is a successor of the range check MBB or the jump table MBB
Dan Gohman2048b852009-11-23 18:04:58 +0000932 for (unsigned i = 0, e = SDB->JTCases.size(); i != e; ++i) {
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +0000933 // Lower header first, if it wasn't already lowered
Dan Gohman2048b852009-11-23 18:04:58 +0000934 if (!SDB->JTCases[i].first.Emitted) {
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +0000935 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman2048b852009-11-23 18:04:58 +0000936 BB = SDB->JTCases[i].first.HeaderBB;
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +0000937 // Emit the code
Dan Gohman99be8ae2010-04-19 22:41:47 +0000938 SDB->visitJumpTableHeader(SDB->JTCases[i].second, SDB->JTCases[i].first,
939 BB);
Dan Gohman2048b852009-11-23 18:04:58 +0000940 CurDAG->setRoot(SDB->getRoot());
Dan Gohmana9a33212010-04-20 00:29:35 +0000941 BB = CodeGenAndEmitDAG(BB);
Dan Gohman2048b852009-11-23 18:04:58 +0000942 SDB->clear();
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000943 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000944
Nate Begeman37efe672006-04-22 18:53:45 +0000945 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman2048b852009-11-23 18:04:58 +0000946 BB = SDB->JTCases[i].second.MBB;
Nate Begeman37efe672006-04-22 18:53:45 +0000947 // Emit the code
Dan Gohman2048b852009-11-23 18:04:58 +0000948 SDB->visitJumpTable(SDB->JTCases[i].second);
949 CurDAG->setRoot(SDB->getRoot());
Dan Gohmana9a33212010-04-20 00:29:35 +0000950 BB = CodeGenAndEmitDAG(BB);
Dan Gohman2048b852009-11-23 18:04:58 +0000951 SDB->clear();
Daniel Dunbara279bc32009-09-20 02:20:51 +0000952
Nate Begeman37efe672006-04-22 18:53:45 +0000953 // Update PHI Nodes
Dan Gohman620427d2010-04-22 19:55:20 +0000954 for (unsigned pi = 0, pe = FuncInfo->PHINodesToUpdate.size();
955 pi != pe; ++pi) {
956 MachineInstr *PHI = FuncInfo->PHINodesToUpdate[pi].first;
Nate Begeman37efe672006-04-22 18:53:45 +0000957 MachineBasicBlock *PHIBB = PHI->getParent();
Chris Lattner518bb532010-02-09 19:54:29 +0000958 assert(PHI->isPHI() &&
Nate Begeman37efe672006-04-22 18:53:45 +0000959 "This is not a machine PHI node that we are updating!");
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000960 // "default" BB. We can go there only from header BB.
Dan Gohman2048b852009-11-23 18:04:58 +0000961 if (PHIBB == SDB->JTCases[i].second.Default) {
Evan Chengce319102009-09-19 09:51:03 +0000962 PHI->addOperand
Dan Gohman620427d2010-04-22 19:55:20 +0000963 (MachineOperand::CreateReg(FuncInfo->PHINodesToUpdate[pi].second,
964 false));
Evan Chengce319102009-09-19 09:51:03 +0000965 PHI->addOperand
Dan Gohman2048b852009-11-23 18:04:58 +0000966 (MachineOperand::CreateMBB(SDB->JTCases[i].first.HeaderBB));
Nate Begemanf4360a42006-05-03 03:48:02 +0000967 }
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000968 // JT BB. Just iterate over successors here
Jakob Stoklund Olesendd437ba2010-01-11 21:02:33 +0000969 if (BB->isSuccessor(PHIBB)) {
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));
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000973 PHI->addOperand(MachineOperand::CreateMBB(BB));
Nate Begeman37efe672006-04-22 18:53:45 +0000974 }
975 }
Nate Begeman37efe672006-04-22 18:53:45 +0000976 }
Dan Gohman2048b852009-11-23 18:04:58 +0000977 SDB->JTCases.clear();
Daniel Dunbara279bc32009-09-20 02:20:51 +0000978
Chris Lattnerb2e806e2006-10-22 23:00:53 +0000979 // If the switch block involved a branch to one of the actual successors, we
980 // need to update PHI nodes in that block.
Dan Gohman620427d2010-04-22 19:55:20 +0000981 for (unsigned i = 0, e = FuncInfo->PHINodesToUpdate.size(); i != e; ++i) {
982 MachineInstr *PHI = FuncInfo->PHINodesToUpdate[i].first;
Chris Lattner518bb532010-02-09 19:54:29 +0000983 assert(PHI->isPHI() &&
Chris Lattnerb2e806e2006-10-22 23:00:53 +0000984 "This is not a machine PHI node that we are updating!");
985 if (BB->isSuccessor(PHI->getParent())) {
Dan Gohman620427d2010-04-22 19:55:20 +0000986 PHI->addOperand(
987 MachineOperand::CreateReg(FuncInfo->PHINodesToUpdate[i].second, false));
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000988 PHI->addOperand(MachineOperand::CreateMBB(BB));
Chris Lattnerb2e806e2006-10-22 23:00:53 +0000989 }
990 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000991
Nate Begemanf15485a2006-03-27 01:32:24 +0000992 // If we generated any switch lowering information, build and codegen any
993 // additional DAGs necessary.
Dan Gohman2048b852009-11-23 18:04:58 +0000994 for (unsigned i = 0, e = SDB->SwitchCases.size(); i != e; ++i) {
Nate Begemanf15485a2006-03-27 01:32:24 +0000995 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman2048b852009-11-23 18:04:58 +0000996 MachineBasicBlock *ThisBB = BB = SDB->SwitchCases[i].ThisBB;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000997
Nate Begemanf15485a2006-03-27 01:32:24 +0000998 // Emit the code
Dan Gohman99be8ae2010-04-19 22:41:47 +0000999 SDB->visitSwitchCase(SDB->SwitchCases[i], BB);
Dan Gohman2048b852009-11-23 18:04:58 +00001000 CurDAG->setRoot(SDB->getRoot());
Dan Gohmanaf1d8ca2010-05-01 00:01:06 +00001001 ThisBB = CodeGenAndEmitDAG(BB);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001002
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001003 // Handle any PHI nodes in successors of this chunk, as if we were coming
1004 // from the original BB before switch expansion. Note that PHI nodes can
1005 // occur multiple times in PHINodesToUpdate. We have to be very careful to
1006 // handle them the right number of times.
Dan Gohman2048b852009-11-23 18:04:58 +00001007 while ((BB = SDB->SwitchCases[i].TrueBB)) { // Handle LHS and RHS.
Jakob Stoklund Olesendd437ba2010-01-11 21:02:33 +00001008 // BB may have been removed from the CFG if a branch was constant folded.
1009 if (ThisBB->isSuccessor(BB)) {
1010 for (MachineBasicBlock::iterator Phi = BB->begin();
Chris Lattner518bb532010-02-09 19:54:29 +00001011 Phi != BB->end() && Phi->isPHI();
Jakob Stoklund Olesendd437ba2010-01-11 21:02:33 +00001012 ++Phi) {
1013 // This value for this PHI node is recorded in PHINodesToUpdate.
1014 for (unsigned pn = 0; ; ++pn) {
Dan Gohman620427d2010-04-22 19:55:20 +00001015 assert(pn != FuncInfo->PHINodesToUpdate.size() &&
Jakob Stoklund Olesendd437ba2010-01-11 21:02:33 +00001016 "Didn't find PHI entry!");
Dan Gohman620427d2010-04-22 19:55:20 +00001017 if (FuncInfo->PHINodesToUpdate[pn].first == Phi) {
Jakob Stoklund Olesendd437ba2010-01-11 21:02:33 +00001018 Phi->addOperand(MachineOperand::
Dan Gohman620427d2010-04-22 19:55:20 +00001019 CreateReg(FuncInfo->PHINodesToUpdate[pn].second,
Jakob Stoklund Olesendd437ba2010-01-11 21:02:33 +00001020 false));
1021 Phi->addOperand(MachineOperand::CreateMBB(ThisBB));
1022 break;
1023 }
Evan Cheng8be58a12009-09-18 08:26:06 +00001024 }
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001025 }
Nate Begemanf15485a2006-03-27 01:32:24 +00001026 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001027
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001028 // Don't process RHS if same block as LHS.
Dan Gohman2048b852009-11-23 18:04:58 +00001029 if (BB == SDB->SwitchCases[i].FalseBB)
1030 SDB->SwitchCases[i].FalseBB = 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001031
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001032 // If we haven't handled the RHS, do so now. Otherwise, we're done.
Dan Gohman2048b852009-11-23 18:04:58 +00001033 SDB->SwitchCases[i].TrueBB = SDB->SwitchCases[i].FalseBB;
1034 SDB->SwitchCases[i].FalseBB = 0;
Nate Begemanf15485a2006-03-27 01:32:24 +00001035 }
Dan Gohman2048b852009-11-23 18:04:58 +00001036 assert(SDB->SwitchCases[i].TrueBB == 0 && SDB->SwitchCases[i].FalseBB == 0);
1037 SDB->clear();
Chris Lattnera33ef482005-03-30 01:10:47 +00001038 }
Dan Gohman2048b852009-11-23 18:04:58 +00001039 SDB->SwitchCases.clear();
Chris Lattner1c08c712005-01-07 07:47:53 +00001040}
Evan Chenga9c20912006-01-21 02:32:06 +00001041
Jim Laskey13ec7022006-08-01 14:21:23 +00001042
Dan Gohman0a3776d2009-02-06 18:26:51 +00001043/// Create the scheduler. If a specific scheduler was specified
1044/// via the SchedulerRegistry, use it, otherwise select the
1045/// one preferred by the target.
Dan Gohman5e843682008-07-14 18:19:29 +00001046///
Dan Gohman47ac0f02009-02-11 04:27:20 +00001047ScheduleDAGSDNodes *SelectionDAGISel::CreateScheduler() {
Jim Laskeyeb577ba2006-08-02 12:30:23 +00001048 RegisterScheduler::FunctionPassCtor Ctor = RegisterScheduler::getDefault();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001049
Jim Laskey13ec7022006-08-01 14:21:23 +00001050 if (!Ctor) {
Jim Laskeyeb577ba2006-08-02 12:30:23 +00001051 Ctor = ISHeuristic;
Jim Laskey9373beb2006-08-01 19:14:14 +00001052 RegisterScheduler::setDefault(Ctor);
Evan Cheng4ef10862006-01-23 07:01:07 +00001053 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001054
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +00001055 return Ctor(this, OptLevel);
Evan Chenga9c20912006-01-21 02:32:06 +00001056}
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001057
Dan Gohmanfc54c552009-01-15 22:18:12 +00001058ScheduleHazardRecognizer *SelectionDAGISel::CreateTargetHazardRecognizer() {
1059 return new ScheduleHazardRecognizer();
Jim Laskey9ff542f2006-08-01 18:29:48 +00001060}
1061
Chris Lattner75548062006-10-11 03:58:02 +00001062//===----------------------------------------------------------------------===//
1063// Helper functions used by the generated instruction selector.
1064//===----------------------------------------------------------------------===//
1065// Calls to these methods are generated by tblgen.
1066
1067/// CheckAndMask - The isel is trying to match something like (and X, 255). If
1068/// the dag combiner simplified the 255, we still want to match. RHS is the
1069/// actual value in the DAG on the RHS of an AND, and DesiredMaskS is the value
1070/// specified in the .td file (e.g. 255).
Daniel Dunbara279bc32009-09-20 02:20:51 +00001071bool SelectionDAGISel::CheckAndMask(SDValue LHS, ConstantSDNode *RHS,
Dan Gohmandc9b3d02007-07-24 23:00:27 +00001072 int64_t DesiredMaskS) const {
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001073 const APInt &ActualMask = RHS->getAPIntValue();
1074 const APInt &DesiredMask = APInt(LHS.getValueSizeInBits(), DesiredMaskS);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001075
Chris Lattner75548062006-10-11 03:58:02 +00001076 // If the actual mask exactly matches, success!
1077 if (ActualMask == DesiredMask)
1078 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001079
Chris Lattner75548062006-10-11 03:58:02 +00001080 // If the actual AND mask is allowing unallowed bits, this doesn't match.
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001081 if (ActualMask.intersects(~DesiredMask))
Chris Lattner75548062006-10-11 03:58:02 +00001082 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001083
Chris Lattner75548062006-10-11 03:58:02 +00001084 // Otherwise, the DAG Combiner may have proven that the value coming in is
1085 // either already zero or is not demanded. Check for known zero input bits.
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001086 APInt NeededMask = DesiredMask & ~ActualMask;
Dan Gohmanea859be2007-06-22 14:59:07 +00001087 if (CurDAG->MaskedValueIsZero(LHS, NeededMask))
Chris Lattner75548062006-10-11 03:58:02 +00001088 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001089
Chris Lattner75548062006-10-11 03:58:02 +00001090 // TODO: check to see if missing bits are just not demanded.
1091
1092 // Otherwise, this pattern doesn't match.
1093 return false;
1094}
1095
1096/// CheckOrMask - The isel is trying to match something like (or X, 255). If
1097/// the dag combiner simplified the 255, we still want to match. RHS is the
1098/// actual value in the DAG on the RHS of an OR, and DesiredMaskS is the value
1099/// specified in the .td file (e.g. 255).
Daniel Dunbara279bc32009-09-20 02:20:51 +00001100bool SelectionDAGISel::CheckOrMask(SDValue LHS, ConstantSDNode *RHS,
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001101 int64_t DesiredMaskS) const {
1102 const APInt &ActualMask = RHS->getAPIntValue();
1103 const APInt &DesiredMask = APInt(LHS.getValueSizeInBits(), DesiredMaskS);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001104
Chris Lattner75548062006-10-11 03:58:02 +00001105 // If the actual mask exactly matches, success!
1106 if (ActualMask == DesiredMask)
1107 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001108
Chris Lattner75548062006-10-11 03:58:02 +00001109 // If the actual AND mask is allowing unallowed bits, this doesn't match.
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001110 if (ActualMask.intersects(~DesiredMask))
Chris Lattner75548062006-10-11 03:58:02 +00001111 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001112
Chris Lattner75548062006-10-11 03:58:02 +00001113 // Otherwise, the DAG Combiner may have proven that the value coming in is
1114 // either already zero or is not demanded. Check for known zero input bits.
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001115 APInt NeededMask = DesiredMask & ~ActualMask;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001116
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001117 APInt KnownZero, KnownOne;
Dan Gohmanea859be2007-06-22 14:59:07 +00001118 CurDAG->ComputeMaskedBits(LHS, NeededMask, KnownZero, KnownOne);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001119
Chris Lattner75548062006-10-11 03:58:02 +00001120 // If all the missing bits in the or are already known to be set, match!
1121 if ((NeededMask & KnownOne) == NeededMask)
1122 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001123
Chris Lattner75548062006-10-11 03:58:02 +00001124 // TODO: check to see if missing bits are just not demanded.
Daniel Dunbara279bc32009-09-20 02:20:51 +00001125
Chris Lattner75548062006-10-11 03:58:02 +00001126 // Otherwise, this pattern doesn't match.
1127 return false;
1128}
1129
Jim Laskey9ff542f2006-08-01 18:29:48 +00001130
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001131/// SelectInlineAsmMemoryOperands - Calls to this are automatically generated
1132/// by tblgen. Others should not call it.
1133void SelectionDAGISel::
Dan Gohmanf350b272008-08-23 02:25:05 +00001134SelectInlineAsmMemoryOperands(std::vector<SDValue> &Ops) {
Dan Gohman475871a2008-07-27 21:46:04 +00001135 std::vector<SDValue> InOps;
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001136 std::swap(InOps, Ops);
1137
Chris Lattnerdecc2672010-04-07 05:20:54 +00001138 Ops.push_back(InOps[InlineAsm::Op_InputChain]); // 0
1139 Ops.push_back(InOps[InlineAsm::Op_AsmString]); // 1
1140 Ops.push_back(InOps[InlineAsm::Op_MDNode]); // 2, !srcloc
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001141
Chris Lattnerdecc2672010-04-07 05:20:54 +00001142 unsigned i = InlineAsm::Op_FirstOperand, e = InOps.size();
Owen Anderson825b72b2009-08-11 20:47:22 +00001143 if (InOps[e-1].getValueType() == MVT::Flag)
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001144 --e; // Don't process a flag operand if it is here.
Daniel Dunbara279bc32009-09-20 02:20:51 +00001145
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001146 while (i != e) {
Dan Gohmanf5aeb1a2008-09-12 16:56:44 +00001147 unsigned Flags = cast<ConstantSDNode>(InOps[i])->getZExtValue();
Chris Lattnerdecc2672010-04-07 05:20:54 +00001148 if (!InlineAsm::isMemKind(Flags)) {
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001149 // Just skip over this operand, copying the operands verbatim.
Evan Cheng697cbbf2009-03-20 18:03:34 +00001150 Ops.insert(Ops.end(), InOps.begin()+i,
1151 InOps.begin()+i+InlineAsm::getNumOperandRegisters(Flags) + 1);
1152 i += InlineAsm::getNumOperandRegisters(Flags) + 1;
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001153 } else {
Evan Cheng697cbbf2009-03-20 18:03:34 +00001154 assert(InlineAsm::getNumOperandRegisters(Flags) == 1 &&
1155 "Memory operand with multiple values?");
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001156 // Otherwise, this is a memory operand. Ask the target to select it.
Dan Gohman475871a2008-07-27 21:46:04 +00001157 std::vector<SDValue> SelOps;
Chris Lattnerdecc2672010-04-07 05:20:54 +00001158 if (SelectInlineAsmMemoryOperand(InOps[i+1], 'm', SelOps))
Chris Lattner75361b62010-04-07 22:58:41 +00001159 report_fatal_error("Could not match memory address. Inline asm"
Chris Lattner87d677c2010-04-07 23:50:38 +00001160 " failure!");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001161
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001162 // Add this to the output node.
Chris Lattnerdecc2672010-04-07 05:20:54 +00001163 unsigned NewFlags =
1164 InlineAsm::getFlagWord(InlineAsm::Kind_Mem, SelOps.size());
1165 Ops.push_back(CurDAG->getTargetConstant(NewFlags, MVT::i32));
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001166 Ops.insert(Ops.end(), SelOps.begin(), SelOps.end());
1167 i += 2;
1168 }
1169 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001170
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001171 // Add the flag input back if present.
1172 if (e != InOps.size())
1173 Ops.push_back(InOps.back());
1174}
Devang Patel794fd752007-05-01 21:15:47 +00001175
Owen Andersone50ed302009-08-10 22:56:29 +00001176/// findFlagUse - Return use of EVT::Flag value produced by the specified
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001177/// SDNode.
1178///
1179static SDNode *findFlagUse(SDNode *N) {
1180 unsigned FlagResNo = N->getNumValues()-1;
1181 for (SDNode::use_iterator I = N->use_begin(), E = N->use_end(); I != E; ++I) {
1182 SDUse &Use = I.getUse();
1183 if (Use.getResNo() == FlagResNo)
1184 return Use.getUser();
1185 }
1186 return NULL;
1187}
1188
1189/// findNonImmUse - Return true if "Use" is a non-immediate use of "Def".
1190/// This function recursively traverses up the operand chain, ignoring
1191/// certain nodes.
1192static bool findNonImmUse(SDNode *Use, SDNode* Def, SDNode *ImmedUse,
Chris Lattnerd1b73822010-03-02 22:20:06 +00001193 SDNode *Root, SmallPtrSet<SDNode*, 16> &Visited,
1194 bool IgnoreChains) {
Chris Lattnerda244a02010-02-23 19:32:27 +00001195 // The NodeID's are given uniques ID's where a node ID is guaranteed to be
1196 // greater than all of its (recursive) operands. If we scan to a point where
1197 // 'use' is smaller than the node we're scanning for, then we know we will
1198 // never find it.
1199 //
1200 // The Use may be -1 (unassigned) if it is a newly allocated node. This can
1201 // happen because we scan down to newly selected nodes in the case of flag
1202 // uses.
1203 if ((Use->getNodeId() < Def->getNodeId() && Use->getNodeId() != -1))
1204 return false;
1205
1206 // Don't revisit nodes if we already scanned it and didn't fail, we know we
1207 // won't fail if we scan it again.
1208 if (!Visited.insert(Use))
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001209 return false;
1210
1211 for (unsigned i = 0, e = Use->getNumOperands(); i != e; ++i) {
Chris Lattnerd1b73822010-03-02 22:20:06 +00001212 // Ignore chain uses, they are validated by HandleMergeInputChains.
1213 if (Use->getOperand(i).getValueType() == MVT::Other && IgnoreChains)
1214 continue;
1215
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001216 SDNode *N = Use->getOperand(i).getNode();
1217 if (N == Def) {
1218 if (Use == ImmedUse || Use == Root)
1219 continue; // We are not looking for immediate use.
1220 assert(N != Root);
1221 return true;
1222 }
1223
1224 // Traverse up the operand chain.
Chris Lattnerd1b73822010-03-02 22:20:06 +00001225 if (findNonImmUse(N, Def, ImmedUse, Root, Visited, IgnoreChains))
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001226 return true;
1227 }
1228 return false;
1229}
1230
Evan Cheng014bf212010-02-15 19:41:07 +00001231/// IsProfitableToFold - Returns true if it's profitable to fold the specific
1232/// operand node N of U during instruction selection that starts at Root.
1233bool SelectionDAGISel::IsProfitableToFold(SDValue N, SDNode *U,
1234 SDNode *Root) const {
1235 if (OptLevel == CodeGenOpt::None) return false;
1236 return N.hasOneUse();
1237}
1238
1239/// IsLegalToFold - Returns true if the specific operand node N of
1240/// U can be folded during instruction selection that starts at Root.
Chris Lattnerd1b73822010-03-02 22:20:06 +00001241bool SelectionDAGISel::IsLegalToFold(SDValue N, SDNode *U, SDNode *Root,
Dan Gohmand858e902010-04-17 15:26:15 +00001242 CodeGenOpt::Level OptLevel,
1243 bool IgnoreChains) {
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001244 if (OptLevel == CodeGenOpt::None) return false;
1245
1246 // If Root use can somehow reach N through a path that that doesn't contain
1247 // U then folding N would create a cycle. e.g. In the following
1248 // diagram, Root can reach N through X. If N is folded into into Root, then
1249 // X is both a predecessor and a successor of U.
1250 //
1251 // [N*] //
1252 // ^ ^ //
1253 // / \ //
1254 // [U*] [X]? //
1255 // ^ ^ //
1256 // \ / //
1257 // \ / //
1258 // [Root*] //
1259 //
1260 // * indicates nodes to be folded together.
1261 //
1262 // If Root produces a flag, then it gets (even more) interesting. Since it
1263 // will be "glued" together with its flag use in the scheduler, we need to
1264 // check if it might reach N.
1265 //
1266 // [N*] //
1267 // ^ ^ //
1268 // / \ //
1269 // [U*] [X]? //
1270 // ^ ^ //
1271 // \ \ //
1272 // \ | //
1273 // [Root*] | //
1274 // ^ | //
1275 // f | //
1276 // | / //
1277 // [Y] / //
1278 // ^ / //
1279 // f / //
1280 // | / //
1281 // [FU] //
1282 //
1283 // If FU (flag use) indirectly reaches N (the load), and Root folds N
1284 // (call it Fold), then X is a predecessor of FU and a successor of
1285 // Fold. But since Fold and FU are flagged together, this will create
1286 // a cycle in the scheduling graph.
1287
Chris Lattner9bbcd5e2010-03-05 05:49:45 +00001288 // If the node has flags, walk down the graph to the "lowest" node in the
1289 // flagged set.
Owen Andersone50ed302009-08-10 22:56:29 +00001290 EVT VT = Root->getValueType(Root->getNumValues()-1);
Owen Anderson825b72b2009-08-11 20:47:22 +00001291 while (VT == MVT::Flag) {
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001292 SDNode *FU = findFlagUse(Root);
1293 if (FU == NULL)
1294 break;
1295 Root = FU;
1296 VT = Root->getValueType(Root->getNumValues()-1);
Chris Lattner18fdaba2010-03-05 06:19:13 +00001297
1298 // If our query node has a flag result with a use, we've walked up it. If
1299 // the user (which has already been selected) has a chain or indirectly uses
1300 // the chain, our WalkChainUsers predicate will not consider it. Because of
1301 // this, we cannot ignore chains in this predicate.
1302 IgnoreChains = false;
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001303 }
Chris Lattner18fdaba2010-03-05 06:19:13 +00001304
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001305
Chris Lattner9bbcd5e2010-03-05 05:49:45 +00001306 SmallPtrSet<SDNode*, 16> Visited;
1307 return !findNonImmUse(Root, N.getNode(), U, Root, Visited, IgnoreChains);
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001308}
1309
Dan Gohmaneeb3a002010-01-05 01:24:18 +00001310SDNode *SelectionDAGISel::Select_INLINEASM(SDNode *N) {
1311 std::vector<SDValue> Ops(N->op_begin(), N->op_end());
Dan Gohmane1f188f2009-10-29 22:30:23 +00001312 SelectInlineAsmMemoryOperands(Ops);
1313
1314 std::vector<EVT> VTs;
1315 VTs.push_back(MVT::Other);
1316 VTs.push_back(MVT::Flag);
Dan Gohmaneeb3a002010-01-05 01:24:18 +00001317 SDValue New = CurDAG->getNode(ISD::INLINEASM, N->getDebugLoc(),
Dan Gohmane1f188f2009-10-29 22:30:23 +00001318 VTs, &Ops[0], Ops.size());
Chris Lattnerd1b73822010-03-02 22:20:06 +00001319 New->setNodeId(-1);
Dan Gohmane1f188f2009-10-29 22:30:23 +00001320 return New.getNode();
1321}
1322
Dan Gohmaneeb3a002010-01-05 01:24:18 +00001323SDNode *SelectionDAGISel::Select_UNDEF(SDNode *N) {
Chris Lattner518bb532010-02-09 19:54:29 +00001324 return CurDAG->SelectNodeTo(N, TargetOpcode::IMPLICIT_DEF,N->getValueType(0));
Dan Gohmane1f188f2009-10-29 22:30:23 +00001325}
1326
Chris Lattner2a49d572010-02-28 22:37:22 +00001327/// GetVBR - decode a vbr encoding whose top bit is set.
1328ALWAYS_INLINE static uint64_t
1329GetVBR(uint64_t Val, const unsigned char *MatcherTable, unsigned &Idx) {
1330 assert(Val >= 128 && "Not a VBR");
1331 Val &= 127; // Remove first vbr bit.
1332
1333 unsigned Shift = 7;
1334 uint64_t NextBits;
1335 do {
Chris Lattner14df8dc2010-02-28 22:38:43 +00001336 NextBits = MatcherTable[Idx++];
Chris Lattner2a49d572010-02-28 22:37:22 +00001337 Val |= (NextBits&127) << Shift;
1338 Shift += 7;
1339 } while (NextBits & 128);
1340
1341 return Val;
1342}
1343
Chris Lattner2a49d572010-02-28 22:37:22 +00001344
1345/// UpdateChainsAndFlags - When a match is complete, this method updates uses of
1346/// interior flag and chain results to use the new flag and chain results.
Chris Lattner82dd3d32010-03-02 07:50:03 +00001347void SelectionDAGISel::
1348UpdateChainsAndFlags(SDNode *NodeToMatch, SDValue InputChain,
1349 const SmallVectorImpl<SDNode*> &ChainNodesMatched,
1350 SDValue InputFlag,
1351 const SmallVectorImpl<SDNode*> &FlagResultNodesMatched,
1352 bool isMorphNodeTo) {
1353 SmallVector<SDNode*, 4> NowDeadNodes;
1354
1355 ISelUpdater ISU(ISelPosition);
1356
Chris Lattner2a49d572010-02-28 22:37:22 +00001357 // Now that all the normal results are replaced, we replace the chain and
1358 // flag results if present.
1359 if (!ChainNodesMatched.empty()) {
1360 assert(InputChain.getNode() != 0 &&
1361 "Matched input chains but didn't produce a chain");
1362 // Loop over all of the nodes we matched that produced a chain result.
1363 // Replace all the chain results with the final chain we ended up with.
1364 for (unsigned i = 0, e = ChainNodesMatched.size(); i != e; ++i) {
1365 SDNode *ChainNode = ChainNodesMatched[i];
1366
Chris Lattner82dd3d32010-03-02 07:50:03 +00001367 // If this node was already deleted, don't look at it.
1368 if (ChainNode->getOpcode() == ISD::DELETED_NODE)
1369 continue;
1370
Chris Lattner2a49d572010-02-28 22:37:22 +00001371 // Don't replace the results of the root node if we're doing a
1372 // MorphNodeTo.
1373 if (ChainNode == NodeToMatch && isMorphNodeTo)
1374 continue;
1375
1376 SDValue ChainVal = SDValue(ChainNode, ChainNode->getNumValues()-1);
1377 if (ChainVal.getValueType() == MVT::Flag)
1378 ChainVal = ChainVal.getValue(ChainVal->getNumValues()-2);
1379 assert(ChainVal.getValueType() == MVT::Other && "Not a chain?");
Chris Lattner82dd3d32010-03-02 07:50:03 +00001380 CurDAG->ReplaceAllUsesOfValueWith(ChainVal, InputChain, &ISU);
1381
Chris Lattner856fb392010-03-28 05:28:31 +00001382 // If the node became dead and we haven't already seen it, delete it.
1383 if (ChainNode->use_empty() &&
1384 !std::count(NowDeadNodes.begin(), NowDeadNodes.end(), ChainNode))
Chris Lattner82dd3d32010-03-02 07:50:03 +00001385 NowDeadNodes.push_back(ChainNode);
Chris Lattner2a49d572010-02-28 22:37:22 +00001386 }
1387 }
1388
1389 // If the result produces a flag, update any flag results in the matched
1390 // pattern with the flag result.
1391 if (InputFlag.getNode() != 0) {
1392 // Handle any interior nodes explicitly marked.
1393 for (unsigned i = 0, e = FlagResultNodesMatched.size(); i != e; ++i) {
1394 SDNode *FRN = FlagResultNodesMatched[i];
Chris Lattner82dd3d32010-03-02 07:50:03 +00001395
1396 // If this node was already deleted, don't look at it.
1397 if (FRN->getOpcode() == ISD::DELETED_NODE)
1398 continue;
1399
Chris Lattner2a49d572010-02-28 22:37:22 +00001400 assert(FRN->getValueType(FRN->getNumValues()-1) == MVT::Flag &&
1401 "Doesn't have a flag result");
1402 CurDAG->ReplaceAllUsesOfValueWith(SDValue(FRN, FRN->getNumValues()-1),
Chris Lattner82dd3d32010-03-02 07:50:03 +00001403 InputFlag, &ISU);
1404
Chris Lattner19e37cb2010-03-28 04:54:33 +00001405 // If the node became dead and we haven't already seen it, delete it.
1406 if (FRN->use_empty() &&
1407 !std::count(NowDeadNodes.begin(), NowDeadNodes.end(), FRN))
Chris Lattner82dd3d32010-03-02 07:50:03 +00001408 NowDeadNodes.push_back(FRN);
Chris Lattner2a49d572010-02-28 22:37:22 +00001409 }
1410 }
1411
Chris Lattner82dd3d32010-03-02 07:50:03 +00001412 if (!NowDeadNodes.empty())
1413 CurDAG->RemoveDeadNodes(NowDeadNodes, &ISU);
1414
Chris Lattner2a49d572010-02-28 22:37:22 +00001415 DEBUG(errs() << "ISEL: Match complete!\n");
1416}
1417
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001418enum ChainResult {
1419 CR_Simple,
1420 CR_InducesCycle,
1421 CR_LeadsToInteriorNode
1422};
1423
1424/// WalkChainUsers - Walk down the users of the specified chained node that is
1425/// part of the pattern we're matching, looking at all of the users we find.
1426/// This determines whether something is an interior node, whether we have a
1427/// non-pattern node in between two pattern nodes (which prevent folding because
1428/// it would induce a cycle) and whether we have a TokenFactor node sandwiched
1429/// between pattern nodes (in which case the TF becomes part of the pattern).
1430///
1431/// The walk we do here is guaranteed to be small because we quickly get down to
1432/// already selected nodes "below" us.
1433static ChainResult
1434WalkChainUsers(SDNode *ChainedNode,
1435 SmallVectorImpl<SDNode*> &ChainedNodesInPattern,
1436 SmallVectorImpl<SDNode*> &InteriorChainedNodes) {
1437 ChainResult Result = CR_Simple;
1438
1439 for (SDNode::use_iterator UI = ChainedNode->use_begin(),
1440 E = ChainedNode->use_end(); UI != E; ++UI) {
1441 // Make sure the use is of the chain, not some other value we produce.
1442 if (UI.getUse().getValueType() != MVT::Other) continue;
1443
1444 SDNode *User = *UI;
1445
1446 // If we see an already-selected machine node, then we've gone beyond the
1447 // pattern that we're selecting down into the already selected chunk of the
1448 // DAG.
1449 if (User->isMachineOpcode() ||
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001450 User->getOpcode() == ISD::HANDLENODE) // Root of the graph.
1451 continue;
Chris Lattnerd1b73822010-03-02 22:20:06 +00001452
1453 if (User->getOpcode() == ISD::CopyToReg ||
1454 User->getOpcode() == ISD::CopyFromReg ||
Chris Lattner7561d482010-03-14 02:33:54 +00001455 User->getOpcode() == ISD::INLINEASM ||
1456 User->getOpcode() == ISD::EH_LABEL) {
Chris Lattnerd1b73822010-03-02 22:20:06 +00001457 // If their node ID got reset to -1 then they've already been selected.
1458 // Treat them like a MachineOpcode.
1459 if (User->getNodeId() == -1)
1460 continue;
1461 }
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001462
1463 // If we have a TokenFactor, we handle it specially.
1464 if (User->getOpcode() != ISD::TokenFactor) {
1465 // If the node isn't a token factor and isn't part of our pattern, then it
1466 // must be a random chained node in between two nodes we're selecting.
1467 // This happens when we have something like:
1468 // x = load ptr
1469 // call
1470 // y = x+4
1471 // store y -> ptr
1472 // Because we structurally match the load/store as a read/modify/write,
1473 // but the call is chained between them. We cannot fold in this case
1474 // because it would induce a cycle in the graph.
1475 if (!std::count(ChainedNodesInPattern.begin(),
1476 ChainedNodesInPattern.end(), User))
1477 return CR_InducesCycle;
1478
1479 // Otherwise we found a node that is part of our pattern. For example in:
1480 // x = load ptr
1481 // y = x+4
1482 // store y -> ptr
1483 // This would happen when we're scanning down from the load and see the
1484 // store as a user. Record that there is a use of ChainedNode that is
1485 // part of the pattern and keep scanning uses.
1486 Result = CR_LeadsToInteriorNode;
1487 InteriorChainedNodes.push_back(User);
1488 continue;
1489 }
1490
1491 // If we found a TokenFactor, there are two cases to consider: first if the
1492 // TokenFactor is just hanging "below" the pattern we're matching (i.e. no
1493 // uses of the TF are in our pattern) we just want to ignore it. Second,
1494 // the TokenFactor can be sandwiched in between two chained nodes, like so:
1495 // [Load chain]
1496 // ^
1497 // |
1498 // [Load]
1499 // ^ ^
1500 // | \ DAG's like cheese
1501 // / \ do you?
1502 // / |
1503 // [TokenFactor] [Op]
1504 // ^ ^
1505 // | |
1506 // \ /
1507 // \ /
1508 // [Store]
1509 //
1510 // In this case, the TokenFactor becomes part of our match and we rewrite it
1511 // as a new TokenFactor.
1512 //
1513 // To distinguish these two cases, do a recursive walk down the uses.
1514 switch (WalkChainUsers(User, ChainedNodesInPattern, InteriorChainedNodes)) {
1515 case CR_Simple:
1516 // If the uses of the TokenFactor are just already-selected nodes, ignore
1517 // it, it is "below" our pattern.
1518 continue;
1519 case CR_InducesCycle:
1520 // If the uses of the TokenFactor lead to nodes that are not part of our
1521 // pattern that are not selected, folding would turn this into a cycle,
1522 // bail out now.
1523 return CR_InducesCycle;
1524 case CR_LeadsToInteriorNode:
1525 break; // Otherwise, keep processing.
1526 }
1527
1528 // Okay, we know we're in the interesting interior case. The TokenFactor
1529 // is now going to be considered part of the pattern so that we rewrite its
1530 // uses (it may have uses that are not part of the pattern) with the
1531 // ultimate chain result of the generated code. We will also add its chain
1532 // inputs as inputs to the ultimate TokenFactor we create.
1533 Result = CR_LeadsToInteriorNode;
1534 ChainedNodesInPattern.push_back(User);
1535 InteriorChainedNodes.push_back(User);
1536 continue;
1537 }
1538
1539 return Result;
1540}
1541
Chris Lattner6b307922010-03-02 00:00:03 +00001542/// HandleMergeInputChains - This implements the OPC_EmitMergeInputChains
Chris Lattner00592ec2010-03-02 19:34:59 +00001543/// operation for when the pattern matched at least one node with a chains. The
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001544/// input vector contains a list of all of the chained nodes that we match. We
1545/// must determine if this is a valid thing to cover (i.e. matching it won't
1546/// induce cycles in the DAG) and if so, creating a TokenFactor node. that will
1547/// be used as the input node chain for the generated nodes.
Chris Lattner6b307922010-03-02 00:00:03 +00001548static SDValue
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001549HandleMergeInputChains(SmallVectorImpl<SDNode*> &ChainNodesMatched,
Chris Lattner6b307922010-03-02 00:00:03 +00001550 SelectionDAG *CurDAG) {
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001551 // Walk all of the chained nodes we've matched, recursively scanning down the
1552 // users of the chain result. This adds any TokenFactor nodes that are caught
1553 // in between chained nodes to the chained and interior nodes list.
1554 SmallVector<SDNode*, 3> InteriorChainedNodes;
1555 for (unsigned i = 0, e = ChainNodesMatched.size(); i != e; ++i) {
1556 if (WalkChainUsers(ChainNodesMatched[i], ChainNodesMatched,
1557 InteriorChainedNodes) == CR_InducesCycle)
1558 return SDValue(); // Would induce a cycle.
1559 }
Chris Lattner6b307922010-03-02 00:00:03 +00001560
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001561 // Okay, we have walked all the matched nodes and collected TokenFactor nodes
1562 // that we are interested in. Form our input TokenFactor node.
Chris Lattner6b307922010-03-02 00:00:03 +00001563 SmallVector<SDValue, 3> InputChains;
1564 for (unsigned i = 0, e = ChainNodesMatched.size(); i != e; ++i) {
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001565 // Add the input chain of this node to the InputChains list (which will be
1566 // the operands of the generated TokenFactor) if it's not an interior node.
1567 SDNode *N = ChainNodesMatched[i];
1568 if (N->getOpcode() != ISD::TokenFactor) {
1569 if (std::count(InteriorChainedNodes.begin(),InteriorChainedNodes.end(),N))
1570 continue;
1571
1572 // Otherwise, add the input chain.
1573 SDValue InChain = ChainNodesMatched[i]->getOperand(0);
1574 assert(InChain.getValueType() == MVT::Other && "Not a chain");
Chris Lattner6b307922010-03-02 00:00:03 +00001575 InputChains.push_back(InChain);
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001576 continue;
1577 }
1578
1579 // If we have a token factor, we want to add all inputs of the token factor
1580 // that are not part of the pattern we're matching.
1581 for (unsigned op = 0, e = N->getNumOperands(); op != e; ++op) {
1582 if (!std::count(ChainNodesMatched.begin(), ChainNodesMatched.end(),
Chris Lattner6183fbd2010-03-02 02:37:23 +00001583 N->getOperand(op).getNode()))
1584 InputChains.push_back(N->getOperand(op));
Chris Lattnerc6d7ad32010-03-02 02:22:10 +00001585 }
Chris Lattner6b307922010-03-02 00:00:03 +00001586 }
1587
1588 SDValue Res;
1589 if (InputChains.size() == 1)
1590 return InputChains[0];
1591 return CurDAG->getNode(ISD::TokenFactor, ChainNodesMatched[0]->getDebugLoc(),
1592 MVT::Other, &InputChains[0], InputChains.size());
1593}
Chris Lattner2a49d572010-02-28 22:37:22 +00001594
Chris Lattner3ff1e4d2010-03-02 06:55:04 +00001595/// MorphNode - Handle morphing a node in place for the selector.
1596SDNode *SelectionDAGISel::
1597MorphNode(SDNode *Node, unsigned TargetOpc, SDVTList VTList,
1598 const SDValue *Ops, unsigned NumOps, unsigned EmitNodeInfo) {
1599 // It is possible we're using MorphNodeTo to replace a node with no
1600 // normal results with one that has a normal result (or we could be
1601 // adding a chain) and the input could have flags and chains as well.
Chris Lattnercaa88702010-03-28 08:43:23 +00001602 // In this case we need to shift the operands down.
Chris Lattner3ff1e4d2010-03-02 06:55:04 +00001603 // FIXME: This is a horrible hack and broken in obscure cases, no worse
Chris Lattnera6f86932010-03-27 18:54:50 +00001604 // than the old isel though.
Chris Lattner3ff1e4d2010-03-02 06:55:04 +00001605 int OldFlagResultNo = -1, OldChainResultNo = -1;
1606
1607 unsigned NTMNumResults = Node->getNumValues();
1608 if (Node->getValueType(NTMNumResults-1) == MVT::Flag) {
1609 OldFlagResultNo = NTMNumResults-1;
1610 if (NTMNumResults != 1 &&
1611 Node->getValueType(NTMNumResults-2) == MVT::Other)
1612 OldChainResultNo = NTMNumResults-2;
1613 } else if (Node->getValueType(NTMNumResults-1) == MVT::Other)
1614 OldChainResultNo = NTMNumResults-1;
1615
Chris Lattner61c97f62010-03-02 07:14:49 +00001616 // Call the underlying SelectionDAG routine to do the transmogrification. Note
1617 // that this deletes operands of the old node that become dead.
Chris Lattner3ff1e4d2010-03-02 06:55:04 +00001618 SDNode *Res = CurDAG->MorphNodeTo(Node, ~TargetOpc, VTList, Ops, NumOps);
1619
1620 // MorphNodeTo can operate in two ways: if an existing node with the
1621 // specified operands exists, it can just return it. Otherwise, it
1622 // updates the node in place to have the requested operands.
1623 if (Res == Node) {
1624 // If we updated the node in place, reset the node ID. To the isel,
1625 // this should be just like a newly allocated machine node.
1626 Res->setNodeId(-1);
1627 }
1628
1629 unsigned ResNumResults = Res->getNumValues();
1630 // Move the flag if needed.
1631 if ((EmitNodeInfo & OPFL_FlagOutput) && OldFlagResultNo != -1 &&
1632 (unsigned)OldFlagResultNo != ResNumResults-1)
1633 CurDAG->ReplaceAllUsesOfValueWith(SDValue(Node, OldFlagResultNo),
1634 SDValue(Res, ResNumResults-1));
1635
1636 if ((EmitNodeInfo & OPFL_FlagOutput) != 0)
1637 --ResNumResults;
1638
1639 // Move the chain reference if needed.
1640 if ((EmitNodeInfo & OPFL_Chain) && OldChainResultNo != -1 &&
1641 (unsigned)OldChainResultNo != ResNumResults-1)
1642 CurDAG->ReplaceAllUsesOfValueWith(SDValue(Node, OldChainResultNo),
1643 SDValue(Res, ResNumResults-1));
1644
1645 // Otherwise, no replacement happened because the node already exists. Replace
1646 // Uses of the old node with the new one.
1647 if (Res != Node)
1648 CurDAG->ReplaceAllUsesWith(Node, Res);
1649
1650 return Res;
1651}
1652
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001653/// CheckPatternPredicate - Implements OP_CheckPatternPredicate.
1654ALWAYS_INLINE static bool
Chris Lattnerda828e32010-03-03 07:46:25 +00001655CheckSame(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1656 SDValue N, const SmallVectorImpl<SDValue> &RecordedNodes) {
1657 // Accept if it is exactly the same as a previously recorded node.
1658 unsigned RecNo = MatcherTable[MatcherIndex++];
1659 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
1660 return N == RecordedNodes[RecNo];
1661}
1662
1663/// CheckPatternPredicate - Implements OP_CheckPatternPredicate.
1664ALWAYS_INLINE static bool
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001665CheckPatternPredicate(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1666 SelectionDAGISel &SDISel) {
1667 return SDISel.CheckPatternPredicate(MatcherTable[MatcherIndex++]);
1668}
1669
1670/// CheckNodePredicate - Implements OP_CheckNodePredicate.
1671ALWAYS_INLINE static bool
1672CheckNodePredicate(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1673 SelectionDAGISel &SDISel, SDNode *N) {
1674 return SDISel.CheckNodePredicate(N, MatcherTable[MatcherIndex++]);
1675}
1676
1677ALWAYS_INLINE static bool
1678CheckOpcode(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1679 SDNode *N) {
Chris Lattnerd5d5a3d2010-03-25 06:33:05 +00001680 uint16_t Opc = MatcherTable[MatcherIndex++];
1681 Opc |= (unsigned short)MatcherTable[MatcherIndex++] << 8;
1682 return N->getOpcode() == Opc;
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001683}
1684
1685ALWAYS_INLINE static bool
1686CheckType(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1687 SDValue N, const TargetLowering &TLI) {
1688 MVT::SimpleValueType VT = (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
1689 if (N.getValueType() == VT) return true;
1690
1691 // Handle the case when VT is iPTR.
1692 return VT == MVT::iPTR && N.getValueType() == TLI.getPointerTy();
1693}
1694
1695ALWAYS_INLINE static bool
Chris Lattnerda828e32010-03-03 07:46:25 +00001696CheckChildType(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1697 SDValue N, const TargetLowering &TLI,
1698 unsigned ChildNo) {
1699 if (ChildNo >= N.getNumOperands())
1700 return false; // Match fails if out of range child #.
1701 return ::CheckType(MatcherTable, MatcherIndex, N.getOperand(ChildNo), TLI);
1702}
1703
1704
1705ALWAYS_INLINE static bool
1706CheckCondCode(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1707 SDValue N) {
1708 return cast<CondCodeSDNode>(N)->get() ==
1709 (ISD::CondCode)MatcherTable[MatcherIndex++];
1710}
1711
1712ALWAYS_INLINE static bool
1713CheckValueType(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1714 SDValue N, const TargetLowering &TLI) {
1715 MVT::SimpleValueType VT = (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
1716 if (cast<VTSDNode>(N)->getVT() == VT)
1717 return true;
1718
1719 // Handle the case when VT is iPTR.
1720 return VT == MVT::iPTR && cast<VTSDNode>(N)->getVT() == TLI.getPointerTy();
1721}
1722
1723ALWAYS_INLINE static bool
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001724CheckInteger(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1725 SDValue N) {
1726 int64_t Val = MatcherTable[MatcherIndex++];
1727 if (Val & 128)
1728 Val = GetVBR(Val, MatcherTable, MatcherIndex);
1729
1730 ConstantSDNode *C = dyn_cast<ConstantSDNode>(N);
1731 return C != 0 && C->getSExtValue() == Val;
1732}
1733
Chris Lattnerda828e32010-03-03 07:46:25 +00001734ALWAYS_INLINE static bool
1735CheckAndImm(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1736 SDValue N, SelectionDAGISel &SDISel) {
1737 int64_t Val = MatcherTable[MatcherIndex++];
1738 if (Val & 128)
1739 Val = GetVBR(Val, MatcherTable, MatcherIndex);
1740
1741 if (N->getOpcode() != ISD::AND) return false;
1742
1743 ConstantSDNode *C = dyn_cast<ConstantSDNode>(N->getOperand(1));
1744 return C != 0 && SDISel.CheckAndMask(N.getOperand(0), C, Val);
1745}
1746
1747ALWAYS_INLINE static bool
1748CheckOrImm(const unsigned char *MatcherTable, unsigned &MatcherIndex,
1749 SDValue N, SelectionDAGISel &SDISel) {
1750 int64_t Val = MatcherTable[MatcherIndex++];
1751 if (Val & 128)
1752 Val = GetVBR(Val, MatcherTable, MatcherIndex);
1753
1754 if (N->getOpcode() != ISD::OR) return false;
1755
1756 ConstantSDNode *C = dyn_cast<ConstantSDNode>(N->getOperand(1));
1757 return C != 0 && SDISel.CheckOrMask(N.getOperand(0), C, Val);
1758}
1759
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001760/// IsPredicateKnownToFail - If we know how and can do so without pushing a
1761/// scope, evaluate the current node. If the current predicate is known to
1762/// fail, set Result=true and return anything. If the current predicate is
1763/// known to pass, set Result=false and return the MatcherIndex to continue
1764/// with. If the current predicate is unknown, set Result=false and return the
1765/// MatcherIndex to continue with.
1766static unsigned IsPredicateKnownToFail(const unsigned char *Table,
1767 unsigned Index, SDValue N,
Chris Lattnerda828e32010-03-03 07:46:25 +00001768 bool &Result, SelectionDAGISel &SDISel,
1769 SmallVectorImpl<SDValue> &RecordedNodes){
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001770 switch (Table[Index++]) {
1771 default:
1772 Result = false;
1773 return Index-1; // Could not evaluate this predicate.
Chris Lattnerda828e32010-03-03 07:46:25 +00001774 case SelectionDAGISel::OPC_CheckSame:
1775 Result = !::CheckSame(Table, Index, N, RecordedNodes);
1776 return Index;
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001777 case SelectionDAGISel::OPC_CheckPatternPredicate:
Chris Lattnerda828e32010-03-03 07:46:25 +00001778 Result = !::CheckPatternPredicate(Table, Index, SDISel);
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001779 return Index;
1780 case SelectionDAGISel::OPC_CheckPredicate:
Chris Lattnerda828e32010-03-03 07:46:25 +00001781 Result = !::CheckNodePredicate(Table, Index, SDISel, N.getNode());
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001782 return Index;
1783 case SelectionDAGISel::OPC_CheckOpcode:
1784 Result = !::CheckOpcode(Table, Index, N.getNode());
1785 return Index;
1786 case SelectionDAGISel::OPC_CheckType:
1787 Result = !::CheckType(Table, Index, N, SDISel.TLI);
1788 return Index;
1789 case SelectionDAGISel::OPC_CheckChild0Type:
1790 case SelectionDAGISel::OPC_CheckChild1Type:
1791 case SelectionDAGISel::OPC_CheckChild2Type:
1792 case SelectionDAGISel::OPC_CheckChild3Type:
1793 case SelectionDAGISel::OPC_CheckChild4Type:
1794 case SelectionDAGISel::OPC_CheckChild5Type:
1795 case SelectionDAGISel::OPC_CheckChild6Type:
Chris Lattnerda828e32010-03-03 07:46:25 +00001796 case SelectionDAGISel::OPC_CheckChild7Type:
1797 Result = !::CheckChildType(Table, Index, N, SDISel.TLI,
1798 Table[Index-1] - SelectionDAGISel::OPC_CheckChild0Type);
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001799 return Index;
Chris Lattnerda828e32010-03-03 07:46:25 +00001800 case SelectionDAGISel::OPC_CheckCondCode:
1801 Result = !::CheckCondCode(Table, Index, N);
1802 return Index;
1803 case SelectionDAGISel::OPC_CheckValueType:
1804 Result = !::CheckValueType(Table, Index, N, SDISel.TLI);
1805 return Index;
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001806 case SelectionDAGISel::OPC_CheckInteger:
1807 Result = !::CheckInteger(Table, Index, N);
1808 return Index;
Chris Lattnerda828e32010-03-03 07:46:25 +00001809 case SelectionDAGISel::OPC_CheckAndImm:
1810 Result = !::CheckAndImm(Table, Index, N, SDISel);
1811 return Index;
1812 case SelectionDAGISel::OPC_CheckOrImm:
1813 Result = !::CheckOrImm(Table, Index, N, SDISel);
1814 return Index;
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001815 }
1816}
1817
Dan Gohmanb3579832010-04-15 17:08:50 +00001818namespace {
Chris Lattner3ff1e4d2010-03-02 06:55:04 +00001819
Chris Lattner2a49d572010-02-28 22:37:22 +00001820struct MatchScope {
1821 /// FailIndex - If this match fails, this is the index to continue with.
1822 unsigned FailIndex;
1823
1824 /// NodeStack - The node stack when the scope was formed.
1825 SmallVector<SDValue, 4> NodeStack;
1826
1827 /// NumRecordedNodes - The number of recorded nodes when the scope was formed.
1828 unsigned NumRecordedNodes;
1829
1830 /// NumMatchedMemRefs - The number of matched memref entries.
1831 unsigned NumMatchedMemRefs;
1832
1833 /// InputChain/InputFlag - The current chain/flag
1834 SDValue InputChain, InputFlag;
1835
1836 /// HasChainNodesMatched - True if the ChainNodesMatched list is non-empty.
1837 bool HasChainNodesMatched, HasFlagResultNodesMatched;
1838};
1839
Dan Gohmanb3579832010-04-15 17:08:50 +00001840}
1841
Chris Lattner2a49d572010-02-28 22:37:22 +00001842SDNode *SelectionDAGISel::
1843SelectCodeCommon(SDNode *NodeToMatch, const unsigned char *MatcherTable,
1844 unsigned TableSize) {
1845 // FIXME: Should these even be selected? Handle these cases in the caller?
1846 switch (NodeToMatch->getOpcode()) {
1847 default:
1848 break;
1849 case ISD::EntryToken: // These nodes remain the same.
1850 case ISD::BasicBlock:
1851 case ISD::Register:
Chris Lattner3b9d6212010-03-14 17:53:23 +00001852 //case ISD::VALUETYPE:
1853 //case ISD::CONDCODE:
Chris Lattner2a49d572010-02-28 22:37:22 +00001854 case ISD::HANDLENODE:
Chris Lattnerdecc2672010-04-07 05:20:54 +00001855 case ISD::MDNODE_SDNODE:
Chris Lattner2a49d572010-02-28 22:37:22 +00001856 case ISD::TargetConstant:
1857 case ISD::TargetConstantFP:
1858 case ISD::TargetConstantPool:
1859 case ISD::TargetFrameIndex:
1860 case ISD::TargetExternalSymbol:
1861 case ISD::TargetBlockAddress:
1862 case ISD::TargetJumpTable:
1863 case ISD::TargetGlobalTLSAddress:
1864 case ISD::TargetGlobalAddress:
1865 case ISD::TokenFactor:
1866 case ISD::CopyFromReg:
1867 case ISD::CopyToReg:
Chris Lattner7561d482010-03-14 02:33:54 +00001868 case ISD::EH_LABEL:
Chris Lattnerd1b73822010-03-02 22:20:06 +00001869 NodeToMatch->setNodeId(-1); // Mark selected.
Chris Lattner2a49d572010-02-28 22:37:22 +00001870 return 0;
1871 case ISD::AssertSext:
1872 case ISD::AssertZext:
1873 CurDAG->ReplaceAllUsesOfValueWith(SDValue(NodeToMatch, 0),
1874 NodeToMatch->getOperand(0));
1875 return 0;
1876 case ISD::INLINEASM: return Select_INLINEASM(NodeToMatch);
Chris Lattner2a49d572010-02-28 22:37:22 +00001877 case ISD::UNDEF: return Select_UNDEF(NodeToMatch);
1878 }
1879
1880 assert(!NodeToMatch->isMachineOpcode() && "Node already selected!");
1881
1882 // Set up the node stack with NodeToMatch as the only node on the stack.
1883 SmallVector<SDValue, 8> NodeStack;
1884 SDValue N = SDValue(NodeToMatch, 0);
1885 NodeStack.push_back(N);
1886
1887 // MatchScopes - Scopes used when matching, if a match failure happens, this
1888 // indicates where to continue checking.
1889 SmallVector<MatchScope, 8> MatchScopes;
1890
1891 // RecordedNodes - This is the set of nodes that have been recorded by the
1892 // state machine.
1893 SmallVector<SDValue, 8> RecordedNodes;
1894
1895 // MatchedMemRefs - This is the set of MemRef's we've seen in the input
1896 // pattern.
1897 SmallVector<MachineMemOperand*, 2> MatchedMemRefs;
1898
1899 // These are the current input chain and flag for use when generating nodes.
1900 // Various Emit operations change these. For example, emitting a copytoreg
1901 // uses and updates these.
1902 SDValue InputChain, InputFlag;
1903
1904 // ChainNodesMatched - If a pattern matches nodes that have input/output
1905 // chains, the OPC_EmitMergeInputChains operation is emitted which indicates
1906 // which ones they are. The result is captured into this list so that we can
1907 // update the chain results when the pattern is complete.
1908 SmallVector<SDNode*, 3> ChainNodesMatched;
1909 SmallVector<SDNode*, 3> FlagResultNodesMatched;
1910
1911 DEBUG(errs() << "ISEL: Starting pattern match on root node: ";
1912 NodeToMatch->dump(CurDAG);
1913 errs() << '\n');
1914
Chris Lattner7390eeb2010-03-01 18:47:11 +00001915 // Determine where to start the interpreter. Normally we start at opcode #0,
1916 // but if the state machine starts with an OPC_SwitchOpcode, then we
1917 // accelerate the first lookup (which is guaranteed to be hot) with the
1918 // OpcodeOffset table.
Chris Lattner2a49d572010-02-28 22:37:22 +00001919 unsigned MatcherIndex = 0;
Chris Lattner7390eeb2010-03-01 18:47:11 +00001920
1921 if (!OpcodeOffset.empty()) {
1922 // Already computed the OpcodeOffset table, just index into it.
1923 if (N.getOpcode() < OpcodeOffset.size())
1924 MatcherIndex = OpcodeOffset[N.getOpcode()];
1925 DEBUG(errs() << " Initial Opcode index to " << MatcherIndex << "\n");
1926
1927 } else if (MatcherTable[0] == OPC_SwitchOpcode) {
1928 // Otherwise, the table isn't computed, but the state machine does start
1929 // with an OPC_SwitchOpcode instruction. Populate the table now, since this
1930 // is the first time we're selecting an instruction.
1931 unsigned Idx = 1;
1932 while (1) {
1933 // Get the size of this case.
1934 unsigned CaseSize = MatcherTable[Idx++];
1935 if (CaseSize & 128)
1936 CaseSize = GetVBR(CaseSize, MatcherTable, Idx);
1937 if (CaseSize == 0) break;
1938
1939 // Get the opcode, add the index to the table.
Chris Lattnerd5d5a3d2010-03-25 06:33:05 +00001940 uint16_t Opc = MatcherTable[Idx++];
1941 Opc |= (unsigned short)MatcherTable[Idx++] << 8;
Chris Lattner7390eeb2010-03-01 18:47:11 +00001942 if (Opc >= OpcodeOffset.size())
1943 OpcodeOffset.resize((Opc+1)*2);
1944 OpcodeOffset[Opc] = Idx;
1945 Idx += CaseSize;
1946 }
1947
1948 // Okay, do the lookup for the first opcode.
1949 if (N.getOpcode() < OpcodeOffset.size())
1950 MatcherIndex = OpcodeOffset[N.getOpcode()];
1951 }
1952
Chris Lattner2a49d572010-02-28 22:37:22 +00001953 while (1) {
1954 assert(MatcherIndex < TableSize && "Invalid index");
Dan Gohman19b38262010-03-09 02:15:05 +00001955#ifndef NDEBUG
1956 unsigned CurrentOpcodeIndex = MatcherIndex;
1957#endif
Chris Lattner2a49d572010-02-28 22:37:22 +00001958 BuiltinOpcodes Opcode = (BuiltinOpcodes)MatcherTable[MatcherIndex++];
1959 switch (Opcode) {
1960 case OPC_Scope: {
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001961 // Okay, the semantics of this operation are that we should push a scope
1962 // then evaluate the first child. However, pushing a scope only to have
1963 // the first check fail (which then pops it) is inefficient. If we can
1964 // determine immediately that the first check (or first several) will
1965 // immediately fail, don't even bother pushing a scope for them.
1966 unsigned FailIndex;
1967
1968 while (1) {
1969 unsigned NumToSkip = MatcherTable[MatcherIndex++];
1970 if (NumToSkip & 128)
1971 NumToSkip = GetVBR(NumToSkip, MatcherTable, MatcherIndex);
1972 // Found the end of the scope with no match.
1973 if (NumToSkip == 0) {
1974 FailIndex = 0;
1975 break;
1976 }
1977
1978 FailIndex = MatcherIndex+NumToSkip;
1979
Chris Lattnera6f86932010-03-27 18:54:50 +00001980 unsigned MatcherIndexOfPredicate = MatcherIndex;
1981 (void)MatcherIndexOfPredicate; // silence warning.
1982
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001983 // If we can't evaluate this predicate without pushing a scope (e.g. if
1984 // it is a 'MoveParent') or if the predicate succeeds on this node, we
1985 // push the scope and evaluate the full predicate chain.
1986 bool Result;
1987 MatcherIndex = IsPredicateKnownToFail(MatcherTable, MatcherIndex, N,
Chris Lattnerda828e32010-03-03 07:46:25 +00001988 Result, *this, RecordedNodes);
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001989 if (!Result)
1990 break;
1991
Chris Lattnera6f86932010-03-27 18:54:50 +00001992 DEBUG(errs() << " Skipped scope entry (due to false predicate) at "
1993 << "index " << MatcherIndexOfPredicate
1994 << ", continuing at " << FailIndex << "\n");
Chris Lattnerbed993d2010-03-28 19:46:56 +00001995 ++NumDAGIselRetries;
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00001996
1997 // Otherwise, we know that this case of the Scope is guaranteed to fail,
1998 // move to the next case.
1999 MatcherIndex = FailIndex;
2000 }
2001
2002 // If the whole scope failed to match, bail.
2003 if (FailIndex == 0) break;
2004
Chris Lattner2a49d572010-02-28 22:37:22 +00002005 // Push a MatchScope which indicates where to go if the first child fails
2006 // to match.
2007 MatchScope NewEntry;
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002008 NewEntry.FailIndex = FailIndex;
Chris Lattner2a49d572010-02-28 22:37:22 +00002009 NewEntry.NodeStack.append(NodeStack.begin(), NodeStack.end());
2010 NewEntry.NumRecordedNodes = RecordedNodes.size();
2011 NewEntry.NumMatchedMemRefs = MatchedMemRefs.size();
2012 NewEntry.InputChain = InputChain;
2013 NewEntry.InputFlag = InputFlag;
2014 NewEntry.HasChainNodesMatched = !ChainNodesMatched.empty();
2015 NewEntry.HasFlagResultNodesMatched = !FlagResultNodesMatched.empty();
2016 MatchScopes.push_back(NewEntry);
2017 continue;
2018 }
2019 case OPC_RecordNode:
2020 // Remember this node, it may end up being an operand in the pattern.
2021 RecordedNodes.push_back(N);
2022 continue;
2023
2024 case OPC_RecordChild0: case OPC_RecordChild1:
2025 case OPC_RecordChild2: case OPC_RecordChild3:
2026 case OPC_RecordChild4: case OPC_RecordChild5:
2027 case OPC_RecordChild6: case OPC_RecordChild7: {
2028 unsigned ChildNo = Opcode-OPC_RecordChild0;
2029 if (ChildNo >= N.getNumOperands())
2030 break; // Match fails if out of range child #.
2031
2032 RecordedNodes.push_back(N->getOperand(ChildNo));
2033 continue;
2034 }
2035 case OPC_RecordMemRef:
2036 MatchedMemRefs.push_back(cast<MemSDNode>(N)->getMemOperand());
2037 continue;
2038
2039 case OPC_CaptureFlagInput:
2040 // If the current node has an input flag, capture it in InputFlag.
2041 if (N->getNumOperands() != 0 &&
2042 N->getOperand(N->getNumOperands()-1).getValueType() == MVT::Flag)
2043 InputFlag = N->getOperand(N->getNumOperands()-1);
2044 continue;
2045
2046 case OPC_MoveChild: {
2047 unsigned ChildNo = MatcherTable[MatcherIndex++];
2048 if (ChildNo >= N.getNumOperands())
2049 break; // Match fails if out of range child #.
2050 N = N.getOperand(ChildNo);
2051 NodeStack.push_back(N);
2052 continue;
2053 }
2054
2055 case OPC_MoveParent:
2056 // Pop the current node off the NodeStack.
2057 NodeStack.pop_back();
2058 assert(!NodeStack.empty() && "Node stack imbalance!");
2059 N = NodeStack.back();
2060 continue;
2061
Chris Lattnerda828e32010-03-03 07:46:25 +00002062 case OPC_CheckSame:
2063 if (!::CheckSame(MatcherTable, MatcherIndex, N, RecordedNodes)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002064 continue;
Chris Lattner2a49d572010-02-28 22:37:22 +00002065 case OPC_CheckPatternPredicate:
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002066 if (!::CheckPatternPredicate(MatcherTable, MatcherIndex, *this)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002067 continue;
2068 case OPC_CheckPredicate:
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002069 if (!::CheckNodePredicate(MatcherTable, MatcherIndex, *this,
2070 N.getNode()))
2071 break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002072 continue;
Chris Lattner57bf8a42010-03-04 01:23:08 +00002073 case OPC_CheckComplexPat: {
2074 unsigned CPNum = MatcherTable[MatcherIndex++];
2075 unsigned RecNo = MatcherTable[MatcherIndex++];
2076 assert(RecNo < RecordedNodes.size() && "Invalid CheckComplexPat");
2077 if (!CheckComplexPattern(NodeToMatch, RecordedNodes[RecNo], CPNum,
2078 RecordedNodes))
Chris Lattner2a49d572010-02-28 22:37:22 +00002079 break;
2080 continue;
Chris Lattner57bf8a42010-03-04 01:23:08 +00002081 }
Chris Lattner2a49d572010-02-28 22:37:22 +00002082 case OPC_CheckOpcode:
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002083 if (!::CheckOpcode(MatcherTable, MatcherIndex, N.getNode())) break;
2084 continue;
2085
2086 case OPC_CheckType:
2087 if (!::CheckType(MatcherTable, MatcherIndex, N, TLI)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002088 continue;
2089
Chris Lattnereb669212010-03-01 06:59:22 +00002090 case OPC_SwitchOpcode: {
2091 unsigned CurNodeOpcode = N.getOpcode();
Chris Lattner7d892d62010-03-01 07:43:08 +00002092 unsigned SwitchStart = MatcherIndex-1; (void)SwitchStart;
Chris Lattnereb669212010-03-01 06:59:22 +00002093 unsigned CaseSize;
2094 while (1) {
2095 // Get the size of this case.
2096 CaseSize = MatcherTable[MatcherIndex++];
2097 if (CaseSize & 128)
2098 CaseSize = GetVBR(CaseSize, MatcherTable, MatcherIndex);
2099 if (CaseSize == 0) break;
2100
Chris Lattnerd5d5a3d2010-03-25 06:33:05 +00002101 uint16_t Opc = MatcherTable[MatcherIndex++];
2102 Opc |= (unsigned short)MatcherTable[MatcherIndex++] << 8;
2103
Chris Lattnereb669212010-03-01 06:59:22 +00002104 // If the opcode matches, then we will execute this case.
Chris Lattnerd5d5a3d2010-03-25 06:33:05 +00002105 if (CurNodeOpcode == Opc)
Chris Lattnereb669212010-03-01 06:59:22 +00002106 break;
2107
2108 // Otherwise, skip over this case.
2109 MatcherIndex += CaseSize;
2110 }
2111
Chris Lattnercfe2eab2010-03-03 06:28:15 +00002112 // If no cases matched, bail out.
Chris Lattnereb669212010-03-01 06:59:22 +00002113 if (CaseSize == 0) break;
2114
2115 // Otherwise, execute the case we found.
2116 DEBUG(errs() << " OpcodeSwitch from " << SwitchStart
2117 << " to " << MatcherIndex << "\n");
2118 continue;
2119 }
2120
Chris Lattnercfe2eab2010-03-03 06:28:15 +00002121 case OPC_SwitchType: {
2122 MVT::SimpleValueType CurNodeVT = N.getValueType().getSimpleVT().SimpleTy;
2123 unsigned SwitchStart = MatcherIndex-1; (void)SwitchStart;
2124 unsigned CaseSize;
2125 while (1) {
2126 // Get the size of this case.
2127 CaseSize = MatcherTable[MatcherIndex++];
2128 if (CaseSize & 128)
2129 CaseSize = GetVBR(CaseSize, MatcherTable, MatcherIndex);
2130 if (CaseSize == 0) break;
2131
2132 MVT::SimpleValueType CaseVT =
2133 (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
2134 if (CaseVT == MVT::iPTR)
2135 CaseVT = TLI.getPointerTy().SimpleTy;
2136
2137 // If the VT matches, then we will execute this case.
2138 if (CurNodeVT == CaseVT)
2139 break;
2140
2141 // Otherwise, skip over this case.
2142 MatcherIndex += CaseSize;
2143 }
2144
2145 // If no cases matched, bail out.
2146 if (CaseSize == 0) break;
2147
2148 // Otherwise, execute the case we found.
2149 DEBUG(errs() << " TypeSwitch[" << EVT(CurNodeVT).getEVTString()
2150 << "] from " << SwitchStart << " to " << MatcherIndex<<'\n');
2151 continue;
2152 }
Chris Lattner2a49d572010-02-28 22:37:22 +00002153 case OPC_CheckChild0Type: case OPC_CheckChild1Type:
2154 case OPC_CheckChild2Type: case OPC_CheckChild3Type:
2155 case OPC_CheckChild4Type: case OPC_CheckChild5Type:
Chris Lattnerda828e32010-03-03 07:46:25 +00002156 case OPC_CheckChild6Type: case OPC_CheckChild7Type:
2157 if (!::CheckChildType(MatcherTable, MatcherIndex, N, TLI,
2158 Opcode-OPC_CheckChild0Type))
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002159 break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002160 continue;
Chris Lattner2a49d572010-02-28 22:37:22 +00002161 case OPC_CheckCondCode:
Chris Lattnerda828e32010-03-03 07:46:25 +00002162 if (!::CheckCondCode(MatcherTable, MatcherIndex, N)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002163 continue;
Chris Lattnerda828e32010-03-03 07:46:25 +00002164 case OPC_CheckValueType:
2165 if (!::CheckValueType(MatcherTable, MatcherIndex, N, TLI)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002166 continue;
Chris Lattnerf1b7c7d2010-03-03 07:31:15 +00002167 case OPC_CheckInteger:
2168 if (!::CheckInteger(MatcherTable, MatcherIndex, N)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002169 continue;
Chris Lattnerda828e32010-03-03 07:46:25 +00002170 case OPC_CheckAndImm:
2171 if (!::CheckAndImm(MatcherTable, MatcherIndex, N, *this)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002172 continue;
Chris Lattnerda828e32010-03-03 07:46:25 +00002173 case OPC_CheckOrImm:
2174 if (!::CheckOrImm(MatcherTable, MatcherIndex, N, *this)) break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002175 continue;
Chris Lattner2a49d572010-02-28 22:37:22 +00002176
2177 case OPC_CheckFoldableChainNode: {
2178 assert(NodeStack.size() != 1 && "No parent node");
2179 // Verify that all intermediate nodes between the root and this one have
2180 // a single use.
2181 bool HasMultipleUses = false;
2182 for (unsigned i = 1, e = NodeStack.size()-1; i != e; ++i)
2183 if (!NodeStack[i].hasOneUse()) {
2184 HasMultipleUses = true;
2185 break;
2186 }
2187 if (HasMultipleUses) break;
2188
2189 // Check to see that the target thinks this is profitable to fold and that
2190 // we can fold it without inducing cycles in the graph.
2191 if (!IsProfitableToFold(N, NodeStack[NodeStack.size()-2].getNode(),
2192 NodeToMatch) ||
2193 !IsLegalToFold(N, NodeStack[NodeStack.size()-2].getNode(),
Dan Gohmand858e902010-04-17 15:26:15 +00002194 NodeToMatch, OptLevel,
2195 true/*We validate our own chains*/))
Chris Lattner2a49d572010-02-28 22:37:22 +00002196 break;
2197
2198 continue;
2199 }
Chris Lattner2a49d572010-02-28 22:37:22 +00002200 case OPC_EmitInteger: {
2201 MVT::SimpleValueType VT =
2202 (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
2203 int64_t Val = MatcherTable[MatcherIndex++];
2204 if (Val & 128)
2205 Val = GetVBR(Val, MatcherTable, MatcherIndex);
2206 RecordedNodes.push_back(CurDAG->getTargetConstant(Val, VT));
2207 continue;
2208 }
2209 case OPC_EmitRegister: {
2210 MVT::SimpleValueType VT =
2211 (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
2212 unsigned RegNo = MatcherTable[MatcherIndex++];
2213 RecordedNodes.push_back(CurDAG->getRegister(RegNo, VT));
2214 continue;
2215 }
2216
2217 case OPC_EmitConvertToTarget: {
2218 // Convert from IMM/FPIMM to target version.
2219 unsigned RecNo = MatcherTable[MatcherIndex++];
2220 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2221 SDValue Imm = RecordedNodes[RecNo];
2222
2223 if (Imm->getOpcode() == ISD::Constant) {
2224 int64_t Val = cast<ConstantSDNode>(Imm)->getZExtValue();
2225 Imm = CurDAG->getTargetConstant(Val, Imm.getValueType());
2226 } else if (Imm->getOpcode() == ISD::ConstantFP) {
2227 const ConstantFP *Val=cast<ConstantFPSDNode>(Imm)->getConstantFPValue();
2228 Imm = CurDAG->getTargetConstantFP(*Val, Imm.getValueType());
2229 }
2230
2231 RecordedNodes.push_back(Imm);
2232 continue;
2233 }
2234
Chris Lattneraa4e3392010-03-28 05:50:16 +00002235 case OPC_EmitMergeInputChains1_0: // OPC_EmitMergeInputChains, 1, 0
2236 case OPC_EmitMergeInputChains1_1: { // OPC_EmitMergeInputChains, 1, 1
2237 // These are space-optimized forms of OPC_EmitMergeInputChains.
2238 assert(InputChain.getNode() == 0 &&
2239 "EmitMergeInputChains should be the first chain producing node");
2240 assert(ChainNodesMatched.empty() &&
2241 "Should only have one EmitMergeInputChains per match");
2242
2243 // Read all of the chained nodes.
2244 unsigned RecNo = Opcode == OPC_EmitMergeInputChains1_1;
2245 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2246 ChainNodesMatched.push_back(RecordedNodes[RecNo].getNode());
2247
2248 // FIXME: What if other value results of the node have uses not matched
2249 // by this pattern?
2250 if (ChainNodesMatched.back() != NodeToMatch &&
2251 !RecordedNodes[RecNo].hasOneUse()) {
2252 ChainNodesMatched.clear();
2253 break;
2254 }
2255
2256 // Merge the input chains if they are not intra-pattern references.
2257 InputChain = HandleMergeInputChains(ChainNodesMatched, CurDAG);
2258
2259 if (InputChain.getNode() == 0)
2260 break; // Failed to merge.
2261 continue;
2262 }
2263
Chris Lattner2a49d572010-02-28 22:37:22 +00002264 case OPC_EmitMergeInputChains: {
2265 assert(InputChain.getNode() == 0 &&
2266 "EmitMergeInputChains should be the first chain producing node");
2267 // This node gets a list of nodes we matched in the input that have
2268 // chains. We want to token factor all of the input chains to these nodes
2269 // together. However, if any of the input chains is actually one of the
2270 // nodes matched in this pattern, then we have an intra-match reference.
2271 // Ignore these because the newly token factored chain should not refer to
2272 // the old nodes.
2273 unsigned NumChains = MatcherTable[MatcherIndex++];
2274 assert(NumChains != 0 && "Can't TF zero chains");
2275
2276 assert(ChainNodesMatched.empty() &&
2277 "Should only have one EmitMergeInputChains per match");
2278
Chris Lattner2a49d572010-02-28 22:37:22 +00002279 // Read all of the chained nodes.
Chris Lattner00592ec2010-03-02 19:34:59 +00002280 for (unsigned i = 0; i != NumChains; ++i) {
2281 unsigned RecNo = MatcherTable[MatcherIndex++];
Chris Lattner2a49d572010-02-28 22:37:22 +00002282 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2283 ChainNodesMatched.push_back(RecordedNodes[RecNo].getNode());
2284
2285 // FIXME: What if other value results of the node have uses not matched
2286 // by this pattern?
2287 if (ChainNodesMatched.back() != NodeToMatch &&
2288 !RecordedNodes[RecNo].hasOneUse()) {
2289 ChainNodesMatched.clear();
2290 break;
2291 }
2292 }
Chris Lattner6b307922010-03-02 00:00:03 +00002293
2294 // If the inner loop broke out, the match fails.
2295 if (ChainNodesMatched.empty())
2296 break;
Chris Lattner2a49d572010-02-28 22:37:22 +00002297
Chris Lattner6b307922010-03-02 00:00:03 +00002298 // Merge the input chains if they are not intra-pattern references.
2299 InputChain = HandleMergeInputChains(ChainNodesMatched, CurDAG);
2300
2301 if (InputChain.getNode() == 0)
2302 break; // Failed to merge.
Chris Lattner2a49d572010-02-28 22:37:22 +00002303
Chris Lattner2a49d572010-02-28 22:37:22 +00002304 continue;
2305 }
2306
2307 case OPC_EmitCopyToReg: {
2308 unsigned RecNo = MatcherTable[MatcherIndex++];
2309 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2310 unsigned DestPhysReg = MatcherTable[MatcherIndex++];
2311
2312 if (InputChain.getNode() == 0)
2313 InputChain = CurDAG->getEntryNode();
2314
2315 InputChain = CurDAG->getCopyToReg(InputChain, NodeToMatch->getDebugLoc(),
2316 DestPhysReg, RecordedNodes[RecNo],
2317 InputFlag);
2318
2319 InputFlag = InputChain.getValue(1);
2320 continue;
2321 }
2322
2323 case OPC_EmitNodeXForm: {
2324 unsigned XFormNo = MatcherTable[MatcherIndex++];
2325 unsigned RecNo = MatcherTable[MatcherIndex++];
2326 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2327 RecordedNodes.push_back(RunSDNodeXForm(RecordedNodes[RecNo], XFormNo));
2328 continue;
2329 }
2330
2331 case OPC_EmitNode:
2332 case OPC_MorphNodeTo: {
Chris Lattner14df8dc2010-02-28 22:38:43 +00002333 uint16_t TargetOpc = MatcherTable[MatcherIndex++];
2334 TargetOpc |= (unsigned short)MatcherTable[MatcherIndex++] << 8;
Chris Lattner2a49d572010-02-28 22:37:22 +00002335 unsigned EmitNodeInfo = MatcherTable[MatcherIndex++];
2336 // Get the result VT list.
2337 unsigned NumVTs = MatcherTable[MatcherIndex++];
2338 SmallVector<EVT, 4> VTs;
2339 for (unsigned i = 0; i != NumVTs; ++i) {
2340 MVT::SimpleValueType VT =
2341 (MVT::SimpleValueType)MatcherTable[MatcherIndex++];
2342 if (VT == MVT::iPTR) VT = TLI.getPointerTy().SimpleTy;
2343 VTs.push_back(VT);
2344 }
2345
2346 if (EmitNodeInfo & OPFL_Chain)
2347 VTs.push_back(MVT::Other);
2348 if (EmitNodeInfo & OPFL_FlagOutput)
2349 VTs.push_back(MVT::Flag);
2350
Chris Lattner7d892d62010-03-01 07:43:08 +00002351 // This is hot code, so optimize the two most common cases of 1 and 2
2352 // results.
2353 SDVTList VTList;
2354 if (VTs.size() == 1)
2355 VTList = CurDAG->getVTList(VTs[0]);
2356 else if (VTs.size() == 2)
2357 VTList = CurDAG->getVTList(VTs[0], VTs[1]);
2358 else
2359 VTList = CurDAG->getVTList(VTs.data(), VTs.size());
Chris Lattner2a49d572010-02-28 22:37:22 +00002360
2361 // Get the operand list.
2362 unsigned NumOps = MatcherTable[MatcherIndex++];
2363 SmallVector<SDValue, 8> Ops;
2364 for (unsigned i = 0; i != NumOps; ++i) {
2365 unsigned RecNo = MatcherTable[MatcherIndex++];
2366 if (RecNo & 128)
2367 RecNo = GetVBR(RecNo, MatcherTable, MatcherIndex);
2368
2369 assert(RecNo < RecordedNodes.size() && "Invalid EmitNode");
2370 Ops.push_back(RecordedNodes[RecNo]);
2371 }
2372
2373 // If there are variadic operands to add, handle them now.
2374 if (EmitNodeInfo & OPFL_VariadicInfo) {
2375 // Determine the start index to copy from.
2376 unsigned FirstOpToCopy = getNumFixedFromVariadicInfo(EmitNodeInfo);
2377 FirstOpToCopy += (EmitNodeInfo & OPFL_Chain) ? 1 : 0;
2378 assert(NodeToMatch->getNumOperands() >= FirstOpToCopy &&
2379 "Invalid variadic node");
2380 // Copy all of the variadic operands, not including a potential flag
2381 // input.
2382 for (unsigned i = FirstOpToCopy, e = NodeToMatch->getNumOperands();
2383 i != e; ++i) {
2384 SDValue V = NodeToMatch->getOperand(i);
2385 if (V.getValueType() == MVT::Flag) break;
2386 Ops.push_back(V);
2387 }
2388 }
2389
2390 // If this has chain/flag inputs, add them.
2391 if (EmitNodeInfo & OPFL_Chain)
2392 Ops.push_back(InputChain);
2393 if ((EmitNodeInfo & OPFL_FlagInput) && InputFlag.getNode() != 0)
2394 Ops.push_back(InputFlag);
2395
2396 // Create the node.
2397 SDNode *Res = 0;
2398 if (Opcode != OPC_MorphNodeTo) {
2399 // If this is a normal EmitNode command, just create the new node and
2400 // add the results to the RecordedNodes list.
2401 Res = CurDAG->getMachineNode(TargetOpc, NodeToMatch->getDebugLoc(),
2402 VTList, Ops.data(), Ops.size());
2403
2404 // Add all the non-flag/non-chain results to the RecordedNodes list.
2405 for (unsigned i = 0, e = VTs.size(); i != e; ++i) {
2406 if (VTs[i] == MVT::Other || VTs[i] == MVT::Flag) break;
2407 RecordedNodes.push_back(SDValue(Res, i));
2408 }
2409
2410 } else {
Chris Lattner3ff1e4d2010-03-02 06:55:04 +00002411 Res = MorphNode(NodeToMatch, TargetOpc, VTList, Ops.data(), Ops.size(),
2412 EmitNodeInfo);
Chris Lattner2a49d572010-02-28 22:37:22 +00002413 }
2414
2415 // If the node had chain/flag results, update our notion of the current
2416 // chain and flag.
Chris Lattner82dd3d32010-03-02 07:50:03 +00002417 if (EmitNodeInfo & OPFL_FlagOutput) {
Chris Lattner2a49d572010-02-28 22:37:22 +00002418 InputFlag = SDValue(Res, VTs.size()-1);
2419 if (EmitNodeInfo & OPFL_Chain)
2420 InputChain = SDValue(Res, VTs.size()-2);
2421 } else if (EmitNodeInfo & OPFL_Chain)
2422 InputChain = SDValue(Res, VTs.size()-1);
2423
2424 // If the OPFL_MemRefs flag is set on this node, slap all of the
2425 // accumulated memrefs onto it.
2426 //
2427 // FIXME: This is vastly incorrect for patterns with multiple outputs
2428 // instructions that access memory and for ComplexPatterns that match
2429 // loads.
2430 if (EmitNodeInfo & OPFL_MemRefs) {
2431 MachineSDNode::mmo_iterator MemRefs =
2432 MF->allocateMemRefsArray(MatchedMemRefs.size());
2433 std::copy(MatchedMemRefs.begin(), MatchedMemRefs.end(), MemRefs);
2434 cast<MachineSDNode>(Res)
2435 ->setMemRefs(MemRefs, MemRefs + MatchedMemRefs.size());
2436 }
2437
2438 DEBUG(errs() << " "
2439 << (Opcode == OPC_MorphNodeTo ? "Morphed" : "Created")
2440 << " node: "; Res->dump(CurDAG); errs() << "\n");
2441
2442 // If this was a MorphNodeTo then we're completely done!
2443 if (Opcode == OPC_MorphNodeTo) {
2444 // Update chain and flag uses.
2445 UpdateChainsAndFlags(NodeToMatch, InputChain, ChainNodesMatched,
Chris Lattner82dd3d32010-03-02 07:50:03 +00002446 InputFlag, FlagResultNodesMatched, true);
2447 return Res;
Chris Lattner2a49d572010-02-28 22:37:22 +00002448 }
2449
2450 continue;
2451 }
2452
2453 case OPC_MarkFlagResults: {
2454 unsigned NumNodes = MatcherTable[MatcherIndex++];
2455
2456 // Read and remember all the flag-result nodes.
2457 for (unsigned i = 0; i != NumNodes; ++i) {
2458 unsigned RecNo = MatcherTable[MatcherIndex++];
2459 if (RecNo & 128)
2460 RecNo = GetVBR(RecNo, MatcherTable, MatcherIndex);
2461
2462 assert(RecNo < RecordedNodes.size() && "Invalid CheckSame");
2463 FlagResultNodesMatched.push_back(RecordedNodes[RecNo].getNode());
2464 }
2465 continue;
2466 }
2467
2468 case OPC_CompleteMatch: {
2469 // The match has been completed, and any new nodes (if any) have been
2470 // created. Patch up references to the matched dag to use the newly
2471 // created nodes.
2472 unsigned NumResults = MatcherTable[MatcherIndex++];
2473
2474 for (unsigned i = 0; i != NumResults; ++i) {
2475 unsigned ResSlot = MatcherTable[MatcherIndex++];
2476 if (ResSlot & 128)
2477 ResSlot = GetVBR(ResSlot, MatcherTable, MatcherIndex);
2478
2479 assert(ResSlot < RecordedNodes.size() && "Invalid CheckSame");
2480 SDValue Res = RecordedNodes[ResSlot];
2481
Chris Lattnerb0e483e2010-03-28 05:54:03 +00002482 assert(i < NodeToMatch->getNumValues() &&
2483 NodeToMatch->getValueType(i) != MVT::Other &&
2484 NodeToMatch->getValueType(i) != MVT::Flag &&
2485 "Invalid number of results to complete!");
Chris Lattner2a49d572010-02-28 22:37:22 +00002486 assert((NodeToMatch->getValueType(i) == Res.getValueType() ||
2487 NodeToMatch->getValueType(i) == MVT::iPTR ||
2488 Res.getValueType() == MVT::iPTR ||
2489 NodeToMatch->getValueType(i).getSizeInBits() ==
2490 Res.getValueType().getSizeInBits()) &&
2491 "invalid replacement");
2492 CurDAG->ReplaceAllUsesOfValueWith(SDValue(NodeToMatch, i), Res);
2493 }
2494
2495 // If the root node defines a flag, add it to the flag nodes to update
2496 // list.
2497 if (NodeToMatch->getValueType(NodeToMatch->getNumValues()-1) == MVT::Flag)
2498 FlagResultNodesMatched.push_back(NodeToMatch);
2499
2500 // Update chain and flag uses.
2501 UpdateChainsAndFlags(NodeToMatch, InputChain, ChainNodesMatched,
Chris Lattner82dd3d32010-03-02 07:50:03 +00002502 InputFlag, FlagResultNodesMatched, false);
Chris Lattner2a49d572010-02-28 22:37:22 +00002503
2504 assert(NodeToMatch->use_empty() &&
2505 "Didn't replace all uses of the node?");
2506
2507 // FIXME: We just return here, which interacts correctly with SelectRoot
2508 // above. We should fix this to not return an SDNode* anymore.
2509 return 0;
2510 }
2511 }
2512
2513 // If the code reached this point, then the match failed. See if there is
2514 // another child to try in the current 'Scope', otherwise pop it until we
2515 // find a case to check.
Dan Gohman19b38262010-03-09 02:15:05 +00002516 DEBUG(errs() << " Match failed at index " << CurrentOpcodeIndex << "\n");
Chris Lattnerbed993d2010-03-28 19:46:56 +00002517 ++NumDAGIselRetries;
Chris Lattner2a49d572010-02-28 22:37:22 +00002518 while (1) {
2519 if (MatchScopes.empty()) {
2520 CannotYetSelect(NodeToMatch);
2521 return 0;
2522 }
2523
2524 // Restore the interpreter state back to the point where the scope was
2525 // formed.
2526 MatchScope &LastScope = MatchScopes.back();
2527 RecordedNodes.resize(LastScope.NumRecordedNodes);
2528 NodeStack.clear();
2529 NodeStack.append(LastScope.NodeStack.begin(), LastScope.NodeStack.end());
2530 N = NodeStack.back();
2531
Chris Lattner2a49d572010-02-28 22:37:22 +00002532 if (LastScope.NumMatchedMemRefs != MatchedMemRefs.size())
2533 MatchedMemRefs.resize(LastScope.NumMatchedMemRefs);
2534 MatcherIndex = LastScope.FailIndex;
2535
Dan Gohman19b38262010-03-09 02:15:05 +00002536 DEBUG(errs() << " Continuing at " << MatcherIndex << "\n");
2537
Chris Lattner2a49d572010-02-28 22:37:22 +00002538 InputChain = LastScope.InputChain;
2539 InputFlag = LastScope.InputFlag;
2540 if (!LastScope.HasChainNodesMatched)
2541 ChainNodesMatched.clear();
2542 if (!LastScope.HasFlagResultNodesMatched)
2543 FlagResultNodesMatched.clear();
2544
2545 // Check to see what the offset is at the new MatcherIndex. If it is zero
2546 // we have reached the end of this scope, otherwise we have another child
2547 // in the current scope to try.
2548 unsigned NumToSkip = MatcherTable[MatcherIndex++];
2549 if (NumToSkip & 128)
2550 NumToSkip = GetVBR(NumToSkip, MatcherTable, MatcherIndex);
2551
2552 // If we have another child in this scope to match, update FailIndex and
2553 // try it.
2554 if (NumToSkip != 0) {
2555 LastScope.FailIndex = MatcherIndex+NumToSkip;
2556 break;
2557 }
2558
2559 // End of this scope, pop it and try the next child in the containing
2560 // scope.
2561 MatchScopes.pop_back();
2562 }
2563 }
2564}
2565
2566
2567
Dan Gohmaneeb3a002010-01-05 01:24:18 +00002568void SelectionDAGISel::CannotYetSelect(SDNode *N) {
Dan Gohmane1f188f2009-10-29 22:30:23 +00002569 std::string msg;
2570 raw_string_ostream Msg(msg);
2571 Msg << "Cannot yet select: ";
Chris Lattner2c4afd12010-03-04 00:21:16 +00002572
2573 if (N->getOpcode() != ISD::INTRINSIC_W_CHAIN &&
2574 N->getOpcode() != ISD::INTRINSIC_WO_CHAIN &&
2575 N->getOpcode() != ISD::INTRINSIC_VOID) {
2576 N->printrFull(Msg, CurDAG);
2577 } else {
2578 bool HasInputChain = N->getOperand(0).getValueType() == MVT::Other;
2579 unsigned iid =
2580 cast<ConstantSDNode>(N->getOperand(HasInputChain))->getZExtValue();
2581 if (iid < Intrinsic::num_intrinsics)
2582 Msg << "intrinsic %" << Intrinsic::getName((Intrinsic::ID)iid);
2583 else if (const TargetIntrinsicInfo *TII = TM.getIntrinsicInfo())
2584 Msg << "target intrinsic %" << TII->getName(iid);
2585 else
2586 Msg << "unknown intrinsic #" << iid;
2587 }
Chris Lattner75361b62010-04-07 22:58:41 +00002588 report_fatal_error(Msg.str());
Dan Gohmane1f188f2009-10-29 22:30:23 +00002589}
2590
Devang Patel19974732007-05-03 01:11:54 +00002591char SelectionDAGISel::ID = 0;