blob: 3d2fb3462691f3df3114e53d0471235f56d3226b [file] [log] [blame]
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001//===-- SelectionDAGISel.cpp - Implement the SelectionDAGISel class -------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner081ce942007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Dan Gohmanf17a25c2007-07-18 16:29:46 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This implements the SelectionDAGISel class.
11//
12//===----------------------------------------------------------------------===//
13
14#define DEBUG_TYPE "isel"
Evan Chengb11ac882008-08-08 07:27:28 +000015#include "llvm/CodeGen/SelectionDAGISel.h"
Dan Gohman13aeef92008-09-03 16:12:24 +000016#include "SelectionDAGBuild.h"
Dan Gohmanf17a25c2007-07-18 16:29:46 +000017#include "llvm/ADT/BitVector.h"
18#include "llvm/Analysis/AliasAnalysis.h"
Dan Gohmanf17a25c2007-07-18 16:29:46 +000019#include "llvm/Constants.h"
20#include "llvm/CallingConv.h"
21#include "llvm/DerivedTypes.h"
22#include "llvm/Function.h"
23#include "llvm/GlobalVariable.h"
24#include "llvm/InlineAsm.h"
25#include "llvm/Instructions.h"
26#include "llvm/Intrinsics.h"
27#include "llvm/IntrinsicInst.h"
Dan Gohman91ad3122008-08-19 22:33:34 +000028#include "llvm/CodeGen/FastISel.h"
Gordon Henriksenf194af22008-08-17 12:56:54 +000029#include "llvm/CodeGen/GCStrategy.h"
Gordon Henriksen1aed5992008-08-17 18:44:35 +000030#include "llvm/CodeGen/GCMetadata.h"
Dan Gohmanf17a25c2007-07-18 16:29:46 +000031#include "llvm/CodeGen/MachineFunction.h"
32#include "llvm/CodeGen/MachineFrameInfo.h"
Dan Gohmanf17a25c2007-07-18 16:29:46 +000033#include "llvm/CodeGen/MachineInstrBuilder.h"
Chris Lattner1b989192007-12-31 04:13:23 +000034#include "llvm/CodeGen/MachineJumpTableInfo.h"
35#include "llvm/CodeGen/MachineModuleInfo.h"
36#include "llvm/CodeGen/MachineRegisterInfo.h"
Dan Gohmand27a0e02008-11-19 23:18:57 +000037#include "llvm/CodeGen/ScheduleDAGSDNodes.h"
Dan Gohmanf17a25c2007-07-18 16:29:46 +000038#include "llvm/CodeGen/SchedulerRegistry.h"
39#include "llvm/CodeGen/SelectionDAG.h"
Dan Gohman1e57df32008-02-10 18:45:23 +000040#include "llvm/Target/TargetRegisterInfo.h"
Dan Gohmanf17a25c2007-07-18 16:29:46 +000041#include "llvm/Target/TargetData.h"
42#include "llvm/Target/TargetFrameInfo.h"
43#include "llvm/Target/TargetInstrInfo.h"
44#include "llvm/Target/TargetLowering.h"
45#include "llvm/Target/TargetMachine.h"
46#include "llvm/Target/TargetOptions.h"
Dan Gohmanf17a25c2007-07-18 16:29:46 +000047#include "llvm/Support/Compiler.h"
Evan Cheng34fd4f32008-06-30 20:45:06 +000048#include "llvm/Support/Debug.h"
49#include "llvm/Support/MathExtras.h"
50#include "llvm/Support/Timer.h"
Dan Gohmanf17a25c2007-07-18 16:29:46 +000051#include <algorithm>
52using namespace llvm;
53
Chris Lattner68068cc2008-06-17 06:09:18 +000054static cl::opt<bool>
Chris Lattnerb29a6a42008-07-10 23:37:50 +000055EnableValueProp("enable-value-prop", cl::Hidden);
56static cl::opt<bool>
Duncan Sandsb3ac3882008-10-27 08:42:46 +000057DisableLegalizeTypes("disable-legalize-types", cl::Hidden);
Dan Gohman71d2e682008-10-28 19:08:46 +000058#ifndef NDEBUG
Dan Gohman91ad3122008-08-19 22:33:34 +000059static cl::opt<bool>
Dan Gohman69740ec2008-09-09 22:06:46 +000060EnableFastISelVerbose("fast-isel-verbose", cl::Hidden,
Dan Gohmaneec72ed2008-10-20 21:30:12 +000061 cl::desc("Enable verbose messages in the \"fast\" "
Dan Gohman69740ec2008-09-09 22:06:46 +000062 "instruction selector"));
63static cl::opt<bool>
Dan Gohmanee70e1d2008-09-09 23:05:00 +000064EnableFastISelAbort("fast-isel-abort", cl::Hidden,
65 cl::desc("Enable abort calls when \"fast\" instruction fails"));
Dan Gohman8c71fc12008-10-28 20:35:31 +000066#else
67static const bool EnableFastISelVerbose = false,
68 EnableFastISelAbort = false;
Dan Gohman71d2e682008-10-28 19:08:46 +000069#endif
Dan Gohman283e4992008-09-05 22:59:21 +000070static cl::opt<bool>
71SchedLiveInCopies("schedule-livein-copies",
72 cl::desc("Schedule copies of livein registers"),
73 cl::init(false));
Chris Lattner68068cc2008-06-17 06:09:18 +000074
Dan Gohmanf17a25c2007-07-18 16:29:46 +000075#ifndef NDEBUG
76static cl::opt<bool>
Dan Gohmanb552df72008-07-21 20:00:07 +000077ViewDAGCombine1("view-dag-combine1-dags", cl::Hidden,
78 cl::desc("Pop up a window to show dags before the first "
79 "dag combine pass"));
80static cl::opt<bool>
81ViewLegalizeTypesDAGs("view-legalize-types-dags", cl::Hidden,
82 cl::desc("Pop up a window to show dags before legalize types"));
83static cl::opt<bool>
84ViewLegalizeDAGs("view-legalize-dags", cl::Hidden,
85 cl::desc("Pop up a window to show dags before legalize"));
86static cl::opt<bool>
87ViewDAGCombine2("view-dag-combine2-dags", cl::Hidden,
88 cl::desc("Pop up a window to show dags before the second "
89 "dag combine pass"));
90static cl::opt<bool>
Duncan Sandsa3e2cd02008-11-24 14:53:14 +000091ViewDAGCombineLT("view-dag-combine-lt-dags", cl::Hidden,
92 cl::desc("Pop up a window to show dags before the post legalize types"
93 " dag combine pass"));
94static cl::opt<bool>
Dan Gohmanf17a25c2007-07-18 16:29:46 +000095ViewISelDAGs("view-isel-dags", cl::Hidden,
96 cl::desc("Pop up a window to show isel dags as they are selected"));
97static cl::opt<bool>
98ViewSchedDAGs("view-sched-dags", cl::Hidden,
99 cl::desc("Pop up a window to show sched dags as they are processed"));
Dan Gohman134c5b62007-08-28 20:32:58 +0000100static cl::opt<bool>
101ViewSUnitDAGs("view-sunit-dags", cl::Hidden,
Chris Lattner2f69f132008-01-25 17:24:52 +0000102 cl::desc("Pop up a window to show SUnit dags after they are processed"));
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000103#else
Dan Gohmanb552df72008-07-21 20:00:07 +0000104static const bool ViewDAGCombine1 = false,
105 ViewLegalizeTypesDAGs = false, ViewLegalizeDAGs = false,
106 ViewDAGCombine2 = false,
Duncan Sandsa3e2cd02008-11-24 14:53:14 +0000107 ViewDAGCombineLT = false,
Dan Gohmanb552df72008-07-21 20:00:07 +0000108 ViewISelDAGs = false, ViewSchedDAGs = false,
109 ViewSUnitDAGs = false;
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000110#endif
111
112//===---------------------------------------------------------------------===//
113///
114/// RegisterScheduler class - Track the registration of instruction schedulers.
115///
116//===---------------------------------------------------------------------===//
117MachinePassRegistry RegisterScheduler::Registry;
118
119//===---------------------------------------------------------------------===//
120///
121/// ISHeuristic command line option for instruction schedulers.
122///
123//===---------------------------------------------------------------------===//
Dan Gohman089efff2008-05-13 00:00:25 +0000124static cl::opt<RegisterScheduler::FunctionPassCtor, false,
125 RegisterPassParser<RegisterScheduler> >
126ISHeuristic("pre-RA-sched",
127 cl::init(&createDefaultScheduler),
128 cl::desc("Instruction schedulers available (before register"
129 " allocation):"));
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000130
Dan Gohman089efff2008-05-13 00:00:25 +0000131static RegisterScheduler
Dan Gohman669b9bf2008-10-14 20:25:08 +0000132defaultListDAGScheduler("default", "Best scheduler for the target",
Dan Gohman089efff2008-05-13 00:00:25 +0000133 createDefaultScheduler);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000134
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000135namespace llvm {
136 //===--------------------------------------------------------------------===//
137 /// createDefaultScheduler - This creates an instruction scheduler appropriate
138 /// for the target.
139 ScheduleDAG* createDefaultScheduler(SelectionDAGISel *IS,
140 SelectionDAG *DAG,
Dan Gohman3ea46f72008-11-11 17:50:47 +0000141 const TargetMachine *TM,
Evan Cheng9b77cae2008-07-01 18:05:03 +0000142 MachineBasicBlock *BB,
143 bool Fast) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000144 TargetLowering &TLI = IS->getTargetLowering();
145
Dan Gohman09ab1d72008-11-20 03:11:19 +0000146 if (Fast)
147 return createFastDAGScheduler(IS, DAG, TM, BB, Fast);
148 if (TLI.getSchedulingPreference() == TargetLowering::SchedulingForLatency)
Dan Gohman3ea46f72008-11-11 17:50:47 +0000149 return createTDListDAGScheduler(IS, DAG, TM, BB, Fast);
Dan Gohman09ab1d72008-11-20 03:11:19 +0000150 assert(TLI.getSchedulingPreference() ==
151 TargetLowering::SchedulingForRegPressure && "Unknown sched type!");
152 return createBURRListDAGScheduler(IS, DAG, TM, BB, Fast);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000153 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000154}
155
Evan Chenge637db12008-01-30 18:18:23 +0000156// EmitInstrWithCustomInserter - This method should be implemented by targets
157// that mark instructions with the 'usesCustomDAGSchedInserter' flag. These
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000158// instructions are special in various ways, which require special support to
159// insert. The specified MachineInstr is created but not inserted into any
160// basic blocks, and the scheduler passes ownership of it to this method.
Evan Chenge637db12008-01-30 18:18:23 +0000161MachineBasicBlock *TargetLowering::EmitInstrWithCustomInserter(MachineInstr *MI,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000162 MachineBasicBlock *MBB) {
163 cerr << "If a target marks an instruction with "
164 << "'usesCustomDAGSchedInserter', it must implement "
Evan Chenge637db12008-01-30 18:18:23 +0000165 << "TargetLowering::EmitInstrWithCustomInserter!\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000166 abort();
167 return 0;
168}
169
Dan Gohman283e4992008-09-05 22:59:21 +0000170/// EmitLiveInCopy - Emit a copy for a live in physical register. If the
171/// physical register has only a single copy use, then coalesced the copy
172/// if possible.
173static void EmitLiveInCopy(MachineBasicBlock *MBB,
174 MachineBasicBlock::iterator &InsertPos,
175 unsigned VirtReg, unsigned PhysReg,
176 const TargetRegisterClass *RC,
177 DenseMap<MachineInstr*, unsigned> &CopyRegMap,
178 const MachineRegisterInfo &MRI,
179 const TargetRegisterInfo &TRI,
180 const TargetInstrInfo &TII) {
181 unsigned NumUses = 0;
182 MachineInstr *UseMI = NULL;
183 for (MachineRegisterInfo::use_iterator UI = MRI.use_begin(VirtReg),
184 UE = MRI.use_end(); UI != UE; ++UI) {
185 UseMI = &*UI;
186 if (++NumUses > 1)
187 break;
188 }
189
190 // If the number of uses is not one, or the use is not a move instruction,
191 // don't coalesce. Also, only coalesce away a virtual register to virtual
192 // register copy.
193 bool Coalesced = false;
194 unsigned SrcReg, DstReg;
195 if (NumUses == 1 &&
196 TII.isMoveInstr(*UseMI, SrcReg, DstReg) &&
197 TargetRegisterInfo::isVirtualRegister(DstReg)) {
198 VirtReg = DstReg;
199 Coalesced = true;
200 }
201
202 // Now find an ideal location to insert the copy.
203 MachineBasicBlock::iterator Pos = InsertPos;
204 while (Pos != MBB->begin()) {
205 MachineInstr *PrevMI = prior(Pos);
206 DenseMap<MachineInstr*, unsigned>::iterator RI = CopyRegMap.find(PrevMI);
207 // copyRegToReg might emit multiple instructions to do a copy.
208 unsigned CopyDstReg = (RI == CopyRegMap.end()) ? 0 : RI->second;
209 if (CopyDstReg && !TRI.regsOverlap(CopyDstReg, PhysReg))
210 // This is what the BB looks like right now:
211 // r1024 = mov r0
212 // ...
213 // r1 = mov r1024
214 //
215 // We want to insert "r1025 = mov r1". Inserting this copy below the
216 // move to r1024 makes it impossible for that move to be coalesced.
217 //
218 // r1025 = mov r1
219 // r1024 = mov r0
220 // ...
221 // r1 = mov 1024
222 // r2 = mov 1025
223 break; // Woot! Found a good location.
224 --Pos;
225 }
226
227 TII.copyRegToReg(*MBB, Pos, VirtReg, PhysReg, RC, RC);
228 CopyRegMap.insert(std::make_pair(prior(Pos), VirtReg));
229 if (Coalesced) {
230 if (&*InsertPos == UseMI) ++InsertPos;
231 MBB->erase(UseMI);
232 }
233}
234
235/// EmitLiveInCopies - If this is the first basic block in the function,
236/// and if it has live ins that need to be copied into vregs, emit the
237/// copies into the block.
238static void EmitLiveInCopies(MachineBasicBlock *EntryMBB,
239 const MachineRegisterInfo &MRI,
240 const TargetRegisterInfo &TRI,
241 const TargetInstrInfo &TII) {
242 if (SchedLiveInCopies) {
243 // Emit the copies at a heuristically-determined location in the block.
244 DenseMap<MachineInstr*, unsigned> CopyRegMap;
245 MachineBasicBlock::iterator InsertPos = EntryMBB->begin();
246 for (MachineRegisterInfo::livein_iterator LI = MRI.livein_begin(),
247 E = MRI.livein_end(); LI != E; ++LI)
248 if (LI->second) {
249 const TargetRegisterClass *RC = MRI.getRegClass(LI->second);
250 EmitLiveInCopy(EntryMBB, InsertPos, LI->second, LI->first,
251 RC, CopyRegMap, MRI, TRI, TII);
252 }
253 } else {
254 // Emit the copies into the top of the block.
255 for (MachineRegisterInfo::livein_iterator LI = MRI.livein_begin(),
256 E = MRI.livein_end(); LI != E; ++LI)
257 if (LI->second) {
258 const TargetRegisterClass *RC = MRI.getRegClass(LI->second);
259 TII.copyRegToReg(*EntryMBB, EntryMBB->begin(),
260 LI->second, LI->first, RC, RC);
261 }
262 }
263}
264
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000265//===----------------------------------------------------------------------===//
266// SelectionDAGISel code
267//===----------------------------------------------------------------------===//
268
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000269SelectionDAGISel::SelectionDAGISel(TargetLowering &tli, bool fast) :
Dan Gohman26f8c272008-09-04 17:05:41 +0000270 FunctionPass(&ID), TLI(tli),
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000271 FuncInfo(new FunctionLoweringInfo(TLI)),
272 CurDAG(new SelectionDAG(TLI, *FuncInfo)),
273 SDL(new SelectionDAGLowering(*CurDAG, TLI, *FuncInfo)),
274 GFI(),
275 Fast(fast),
276 DAGSize(0)
277{}
278
279SelectionDAGISel::~SelectionDAGISel() {
280 delete SDL;
281 delete CurDAG;
282 delete FuncInfo;
283}
284
Duncan Sands92c43912008-06-06 12:08:01 +0000285unsigned SelectionDAGISel::MakeReg(MVT VT) {
Chris Lattner1b989192007-12-31 04:13:23 +0000286 return RegInfo->createVirtualRegister(TLI.getRegClassFor(VT));
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000287}
288
289void SelectionDAGISel::getAnalysisUsage(AnalysisUsage &AU) const {
290 AU.addRequired<AliasAnalysis>();
Gordon Henriksen1aed5992008-08-17 18:44:35 +0000291 AU.addRequired<GCModuleInfo>();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000292 AU.setPreservesAll();
293}
294
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000295bool SelectionDAGISel::runOnFunction(Function &Fn) {
Dan Gohmanee70e1d2008-09-09 23:05:00 +0000296 // Do some sanity-checking on the command-line options.
297 assert((!EnableFastISelVerbose || EnableFastISel) &&
298 "-fast-isel-verbose requires -fast-isel");
299 assert((!EnableFastISelAbort || EnableFastISel) &&
300 "-fast-isel-abort requires -fast-isel");
301
Dan Gohmancc863aa2007-08-27 16:26:13 +0000302 // Get alias analysis for load/store combining.
303 AA = &getAnalysis<AliasAnalysis>();
304
Dan Gohman283e4992008-09-05 22:59:21 +0000305 TargetMachine &TM = TLI.getTargetMachine();
306 MachineFunction &MF = MachineFunction::construct(&Fn, TM);
307 const MachineRegisterInfo &MRI = MF.getRegInfo();
308 const TargetInstrInfo &TII = *TM.getInstrInfo();
309 const TargetRegisterInfo &TRI = *TM.getRegisterInfo();
310
Gordon Henriksen1aed5992008-08-17 18:44:35 +0000311 if (MF.getFunction()->hasGC())
312 GFI = &getAnalysis<GCModuleInfo>().getFunctionInfo(*MF.getFunction());
Gordon Henriksendf87fdc2008-01-07 01:30:38 +0000313 else
Gordon Henriksen1aed5992008-08-17 18:44:35 +0000314 GFI = 0;
Chris Lattner1b989192007-12-31 04:13:23 +0000315 RegInfo = &MF.getRegInfo();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000316 DOUT << "\n\n\n=== " << Fn.getName() << "\n";
317
Dan Gohman13aeef92008-09-03 16:12:24 +0000318 FuncInfo->set(Fn, MF, EnableFastISel);
Dan Gohman76dd96e2008-09-23 21:53:34 +0000319 MachineModuleInfo *MMI = getAnalysisToUpdate<MachineModuleInfo>();
320 CurDAG->init(MF, MMI);
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000321 SDL->init(GFI, *AA);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000322
Dale Johannesen85535762008-04-02 00:25:04 +0000323 for (Function::iterator I = Fn.begin(), E = Fn.end(); I != E; ++I)
324 if (InvokeInst *Invoke = dyn_cast<InvokeInst>(I->getTerminator()))
325 // Mark landing pad.
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000326 FuncInfo->MBBMap[Invoke->getSuccessor(1)]->setIsLandingPad();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000327
Dan Gohman9dd43582008-10-14 23:54:11 +0000328 SelectAllBasicBlocks(Fn, MF, MMI, TII);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000329
Dan Gohman283e4992008-09-05 22:59:21 +0000330 // If the first basic block in the function has live ins that need to be
331 // copied into vregs, emit the copies into the top of the block before
332 // emitting the code for the block.
333 EmitLiveInCopies(MF.begin(), MRI, TRI, TII);
334
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000335 // Add function live-ins to entry block live-in set.
Dan Gohman283e4992008-09-05 22:59:21 +0000336 for (MachineRegisterInfo::livein_iterator I = RegInfo->livein_begin(),
337 E = RegInfo->livein_end(); I != E; ++I)
338 MF.begin()->addLiveIn(I->first);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000339
340#ifndef NDEBUG
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000341 assert(FuncInfo->CatchInfoFound.size() == FuncInfo->CatchInfoLost.size() &&
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000342 "Not all catch info was assigned to a landing pad!");
343#endif
344
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000345 FuncInfo->clear();
346
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000347 return true;
348}
349
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000350static void copyCatchInfo(BasicBlock *SrcBB, BasicBlock *DestBB,
351 MachineModuleInfo *MMI, FunctionLoweringInfo &FLI) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000352 for (BasicBlock::iterator I = SrcBB->begin(), E = --SrcBB->end(); I != E; ++I)
Dan Gohman13aeef92008-09-03 16:12:24 +0000353 if (EHSelectorInst *EHSel = dyn_cast<EHSelectorInst>(I)) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000354 // Apply the catch info to DestBB.
Dan Gohman13aeef92008-09-03 16:12:24 +0000355 AddCatchInfo(*EHSel, MMI, FLI.MBBMap[DestBB]);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000356#ifndef NDEBUG
Duncan Sands9b7e1482007-11-15 09:54:37 +0000357 if (!FLI.MBBMap[SrcBB]->isLandingPad())
Dan Gohman13aeef92008-09-03 16:12:24 +0000358 FLI.CatchInfoFound.insert(EHSel);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000359#endif
360 }
361}
362
Arnold Schwaighofera0032722008-04-30 09:16:33 +0000363/// IsFixedFrameObjectWithPosOffset - Check if object is a fixed frame object and
364/// whether object offset >= 0.
365static bool
Dan Gohman8181bd12008-07-27 21:46:04 +0000366IsFixedFrameObjectWithPosOffset(MachineFrameInfo * MFI, SDValue Op) {
Arnold Schwaighofera0032722008-04-30 09:16:33 +0000367 if (!isa<FrameIndexSDNode>(Op)) return false;
368
369 FrameIndexSDNode * FrameIdxNode = dyn_cast<FrameIndexSDNode>(Op);
370 int FrameIdx = FrameIdxNode->getIndex();
371 return MFI->isFixedObjectIndex(FrameIdx) &&
372 MFI->getObjectOffset(FrameIdx) >= 0;
373}
374
375/// IsPossiblyOverwrittenArgumentOfTailCall - Check if the operand could
376/// possibly be overwritten when lowering the outgoing arguments in a tail
377/// call. Currently the implementation of this call is very conservative and
378/// assumes all arguments sourcing from FORMAL_ARGUMENTS or a CopyFromReg with
379/// virtual registers would be overwritten by direct lowering.
Dan Gohman8181bd12008-07-27 21:46:04 +0000380static bool IsPossiblyOverwrittenArgumentOfTailCall(SDValue Op,
Arnold Schwaighofera0032722008-04-30 09:16:33 +0000381 MachineFrameInfo * MFI) {
382 RegisterSDNode * OpReg = NULL;
383 if (Op.getOpcode() == ISD::FORMAL_ARGUMENTS ||
384 (Op.getOpcode()== ISD::CopyFromReg &&
385 (OpReg = dyn_cast<RegisterSDNode>(Op.getOperand(1))) &&
386 (OpReg->getReg() >= TargetRegisterInfo::FirstVirtualRegister)) ||
387 (Op.getOpcode() == ISD::LOAD &&
388 IsFixedFrameObjectWithPosOffset(MFI, Op.getOperand(1))) ||
389 (Op.getOpcode() == ISD::MERGE_VALUES &&
Gabor Greif46bf5472008-08-26 22:36:50 +0000390 Op.getOperand(Op.getResNo()).getOpcode() == ISD::LOAD &&
391 IsFixedFrameObjectWithPosOffset(MFI, Op.getOperand(Op.getResNo()).
Arnold Schwaighofera0032722008-04-30 09:16:33 +0000392 getOperand(1))))
393 return true;
394 return false;
395}
396
Arnold Schwaighofere2d6bbb2007-10-11 19:40:01 +0000397/// CheckDAGForTailCallsAndFixThem - This Function looks for CALL nodes in the
Arnold Schwaighofer373e8652007-10-12 21:30:57 +0000398/// DAG and fixes their tailcall attribute operand.
Arnold Schwaighofere2d6bbb2007-10-11 19:40:01 +0000399static void CheckDAGForTailCallsAndFixThem(SelectionDAG &DAG,
400 TargetLowering& TLI) {
401 SDNode * Ret = NULL;
Dan Gohman8181bd12008-07-27 21:46:04 +0000402 SDValue Terminator = DAG.getRoot();
Arnold Schwaighofere2d6bbb2007-10-11 19:40:01 +0000403
404 // Find RET node.
405 if (Terminator.getOpcode() == ISD::RET) {
Gabor Greif1c80d112008-08-28 21:40:38 +0000406 Ret = Terminator.getNode();
Arnold Schwaighofere2d6bbb2007-10-11 19:40:01 +0000407 }
408
409 // Fix tail call attribute of CALL nodes.
410 for (SelectionDAG::allnodes_iterator BE = DAG.allnodes_begin(),
Dan Gohmaned825d12008-07-07 23:02:41 +0000411 BI = DAG.allnodes_end(); BI != BE; ) {
412 --BI;
Dan Gohman705e3f72008-09-13 01:54:27 +0000413 if (CallSDNode *TheCall = dyn_cast<CallSDNode>(BI)) {
Dan Gohman8181bd12008-07-27 21:46:04 +0000414 SDValue OpRet(Ret, 0);
415 SDValue OpCall(BI, 0);
Dan Gohman705e3f72008-09-13 01:54:27 +0000416 bool isMarkedTailCall = TheCall->isTailCall();
Arnold Schwaighofere2d6bbb2007-10-11 19:40:01 +0000417 // If CALL node has tail call attribute set to true and the call is not
418 // eligible (no RET or the target rejects) the attribute is fixed to
Arnold Schwaighofer373e8652007-10-12 21:30:57 +0000419 // false. The TargetLowering::IsEligibleForTailCallOptimization function
Arnold Schwaighofere2d6bbb2007-10-11 19:40:01 +0000420 // must correctly identify tail call optimizable calls.
Arnold Schwaighofera0032722008-04-30 09:16:33 +0000421 if (!isMarkedTailCall) continue;
422 if (Ret==NULL ||
Dan Gohman705e3f72008-09-13 01:54:27 +0000423 !TLI.IsEligibleForTailCallOptimization(TheCall, OpRet, DAG)) {
424 // Not eligible. Mark CALL node as non tail call. Note that we
425 // can modify the call node in place since calls are not CSE'd.
426 TheCall->setNotTailCall();
Arnold Schwaighofera0032722008-04-30 09:16:33 +0000427 } else {
428 // Look for tail call clobbered arguments. Emit a series of
429 // copyto/copyfrom virtual register nodes to protect them.
Dan Gohman8181bd12008-07-27 21:46:04 +0000430 SmallVector<SDValue, 32> Ops;
Dan Gohman705e3f72008-09-13 01:54:27 +0000431 SDValue Chain = TheCall->getChain(), InFlag;
432 Ops.push_back(Chain);
433 Ops.push_back(TheCall->getCallee());
434 for (unsigned i = 0, e = TheCall->getNumArgs(); i != e; ++i) {
435 SDValue Arg = TheCall->getArg(i);
436 bool isByVal = TheCall->getArgFlags(i).isByVal();
437 MachineFunction &MF = DAG.getMachineFunction();
438 MachineFrameInfo *MFI = MF.getFrameInfo();
439 if (!isByVal &&
440 IsPossiblyOverwrittenArgumentOfTailCall(Arg, MFI)) {
441 MVT VT = Arg.getValueType();
442 unsigned VReg = MF.getRegInfo().
443 createVirtualRegister(TLI.getRegClassFor(VT));
444 Chain = DAG.getCopyToReg(Chain, VReg, Arg, InFlag);
445 InFlag = Chain.getValue(1);
446 Arg = DAG.getCopyFromReg(Chain, VReg, VT, InFlag);
447 Chain = Arg.getValue(1);
448 InFlag = Arg.getValue(2);
Arnold Schwaighofera0032722008-04-30 09:16:33 +0000449 }
450 Ops.push_back(Arg);
Dan Gohman705e3f72008-09-13 01:54:27 +0000451 Ops.push_back(TheCall->getArgFlagsVal(i));
Arnold Schwaighofera0032722008-04-30 09:16:33 +0000452 }
453 // Link in chain of CopyTo/CopyFromReg.
454 Ops[0] = Chain;
455 DAG.UpdateNodeOperands(OpCall, Ops.begin(), Ops.size());
Arnold Schwaighofere2d6bbb2007-10-11 19:40:01 +0000456 }
457 }
458 }
459}
460
Dan Gohman14a66442008-08-23 02:25:05 +0000461void SelectionDAGISel::SelectBasicBlock(BasicBlock *LLVMBB,
462 BasicBlock::iterator Begin,
Dan Gohman2c44e9a2008-08-28 20:28:56 +0000463 BasicBlock::iterator End) {
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000464 SDL->setCurrentBasicBlock(BB);
Dan Gohman14a66442008-08-23 02:25:05 +0000465
Dan Gohman14a66442008-08-23 02:25:05 +0000466 // Lower all of the non-terminator instructions.
467 for (BasicBlock::iterator I = Begin; I != End; ++I)
468 if (!isa<TerminatorInst>(I))
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000469 SDL->visit(*I);
Dan Gohman14a66442008-08-23 02:25:05 +0000470
471 // Ensure that all instructions which are used outside of their defining
472 // blocks are available as virtual registers. Invoke is handled elsewhere.
473 for (BasicBlock::iterator I = Begin; I != End; ++I)
474 if (!I->use_empty() && !isa<PHINode>(I) && !isa<InvokeInst>(I)) {
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000475 DenseMap<const Value*,unsigned>::iterator VMI =FuncInfo->ValueMap.find(I);
476 if (VMI != FuncInfo->ValueMap.end())
477 SDL->CopyValueToVirtualRegister(I, VMI->second);
Dan Gohman14a66442008-08-23 02:25:05 +0000478 }
479
480 // Handle PHI nodes in successor blocks.
Dan Gohmanca4857a2008-09-03 23:12:08 +0000481 if (End == LLVMBB->end()) {
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000482 HandlePHINodesInSuccessorBlocks(LLVMBB);
Dan Gohmanca4857a2008-09-03 23:12:08 +0000483
484 // Lower the terminator after the copies are emitted.
485 SDL->visit(*LLVMBB->getTerminator());
486 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000487
488 // Make sure the root of the DAG is up-to-date.
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000489 CurDAG->setRoot(SDL->getControlRoot());
Arnold Schwaighofere2d6bbb2007-10-11 19:40:01 +0000490
491 // Check whether calls in this block are real tail calls. Fix up CALL nodes
492 // with correct tailcall attribute so that the target can rely on the tailcall
493 // attribute indicating whether the call is really eligible for tail call
494 // optimization.
Dan Gohman5ea89a02008-09-16 01:42:28 +0000495 if (PerformTailCallOpt)
496 CheckDAGForTailCallsAndFixThem(*CurDAG, TLI);
Dan Gohman14a66442008-08-23 02:25:05 +0000497
498 // Final step, emit the lowered DAG as machine code.
499 CodeGenAndEmitDAG();
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000500 SDL->clear();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000501}
502
Dan Gohman14a66442008-08-23 02:25:05 +0000503void SelectionDAGISel::ComputeLiveOutVRegInfo() {
Chris Lattner68068cc2008-06-17 06:09:18 +0000504 SmallPtrSet<SDNode*, 128> VisitedNodes;
505 SmallVector<SDNode*, 128> Worklist;
506
Gabor Greif1c80d112008-08-28 21:40:38 +0000507 Worklist.push_back(CurDAG->getRoot().getNode());
Chris Lattner68068cc2008-06-17 06:09:18 +0000508
509 APInt Mask;
510 APInt KnownZero;
511 APInt KnownOne;
512
513 while (!Worklist.empty()) {
514 SDNode *N = Worklist.back();
515 Worklist.pop_back();
516
517 // If we've already seen this node, ignore it.
518 if (!VisitedNodes.insert(N))
519 continue;
520
521 // Otherwise, add all chain operands to the worklist.
522 for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i)
523 if (N->getOperand(i).getValueType() == MVT::Other)
Gabor Greif1c80d112008-08-28 21:40:38 +0000524 Worklist.push_back(N->getOperand(i).getNode());
Chris Lattner68068cc2008-06-17 06:09:18 +0000525
526 // If this is a CopyToReg with a vreg dest, process it.
527 if (N->getOpcode() != ISD::CopyToReg)
528 continue;
529
530 unsigned DestReg = cast<RegisterSDNode>(N->getOperand(1))->getReg();
531 if (!TargetRegisterInfo::isVirtualRegister(DestReg))
532 continue;
533
534 // Ignore non-scalar or non-integer values.
Dan Gohman8181bd12008-07-27 21:46:04 +0000535 SDValue Src = N->getOperand(2);
Chris Lattner68068cc2008-06-17 06:09:18 +0000536 MVT SrcVT = Src.getValueType();
537 if (!SrcVT.isInteger() || SrcVT.isVector())
538 continue;
539
Dan Gohman14a66442008-08-23 02:25:05 +0000540 unsigned NumSignBits = CurDAG->ComputeNumSignBits(Src);
Chris Lattner68068cc2008-06-17 06:09:18 +0000541 Mask = APInt::getAllOnesValue(SrcVT.getSizeInBits());
Dan Gohman14a66442008-08-23 02:25:05 +0000542 CurDAG->ComputeMaskedBits(Src, Mask, KnownZero, KnownOne);
Chris Lattner68068cc2008-06-17 06:09:18 +0000543
544 // Only install this information if it tells us something.
545 if (NumSignBits != 1 || KnownZero != 0 || KnownOne != 0) {
546 DestReg -= TargetRegisterInfo::FirstVirtualRegister;
Dan Gohman14a66442008-08-23 02:25:05 +0000547 FunctionLoweringInfo &FLI = CurDAG->getFunctionLoweringInfo();
Chris Lattner68068cc2008-06-17 06:09:18 +0000548 if (DestReg >= FLI.LiveOutRegInfo.size())
549 FLI.LiveOutRegInfo.resize(DestReg+1);
550 FunctionLoweringInfo::LiveOutInfo &LOI = FLI.LiveOutRegInfo[DestReg];
551 LOI.NumSignBits = NumSignBits;
552 LOI.KnownOne = NumSignBits;
553 LOI.KnownZero = NumSignBits;
554 }
555 }
556}
557
Dan Gohman14a66442008-08-23 02:25:05 +0000558void SelectionDAGISel::CodeGenAndEmitDAG() {
Dan Gohmanb552df72008-07-21 20:00:07 +0000559 std::string GroupName;
560 if (TimePassesIsEnabled)
561 GroupName = "Instruction Selection and Scheduling";
562 std::string BlockName;
563 if (ViewDAGCombine1 || ViewLegalizeTypesDAGs || ViewLegalizeDAGs ||
Duncan Sandsa3e2cd02008-11-24 14:53:14 +0000564 ViewDAGCombine2 || ViewDAGCombineLT || ViewISelDAGs || ViewSchedDAGs ||
565 ViewSUnitDAGs)
Dan Gohman14a66442008-08-23 02:25:05 +0000566 BlockName = CurDAG->getMachineFunction().getFunction()->getName() + ':' +
Dan Gohmanb552df72008-07-21 20:00:07 +0000567 BB->getBasicBlock()->getName();
568
569 DOUT << "Initial selection DAG:\n";
Dan Gohman14a66442008-08-23 02:25:05 +0000570 DEBUG(CurDAG->dump());
Dan Gohmanb552df72008-07-21 20:00:07 +0000571
Dan Gohman14a66442008-08-23 02:25:05 +0000572 if (ViewDAGCombine1) CurDAG->viewGraph("dag-combine1 input for " + BlockName);
Dan Gohmaneebf44e2007-10-08 15:12:17 +0000573
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000574 // Run the DAG combiner in pre-legalize mode.
Evan Cheng19733c42008-07-01 17:59:20 +0000575 if (TimePassesIsEnabled) {
Dan Gohman368a08b2008-07-14 18:19:29 +0000576 NamedRegionTimer T("DAG Combining 1", GroupName);
Duncan Sandsa3e2cd02008-11-24 14:53:14 +0000577 CurDAG->Combine(Unrestricted, *AA, Fast);
Evan Cheng19733c42008-07-01 17:59:20 +0000578 } else {
Duncan Sandsa3e2cd02008-11-24 14:53:14 +0000579 CurDAG->Combine(Unrestricted, *AA, Fast);
Evan Cheng19733c42008-07-01 17:59:20 +0000580 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000581
Dan Gohmaneebf44e2007-10-08 15:12:17 +0000582 DOUT << "Optimized lowered selection DAG:\n";
Dan Gohman14a66442008-08-23 02:25:05 +0000583 DEBUG(CurDAG->dump());
Duncan Sands31ddf4c2008-07-17 17:06:03 +0000584
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000585 // Second step, hack on the DAG until it only uses operations and types that
586 // the target supports.
Duncan Sandsb3ac3882008-10-27 08:42:46 +0000587 if (!DisableLegalizeTypes) {
Dan Gohman14a66442008-08-23 02:25:05 +0000588 if (ViewLegalizeTypesDAGs) CurDAG->viewGraph("legalize-types input for " +
589 BlockName);
Dan Gohmanb552df72008-07-21 20:00:07 +0000590
Duncan Sandsa3e2cd02008-11-24 14:53:14 +0000591 bool Changed;
Dan Gohmanb552df72008-07-21 20:00:07 +0000592 if (TimePassesIsEnabled) {
593 NamedRegionTimer T("Type Legalization", GroupName);
Duncan Sandsa3e2cd02008-11-24 14:53:14 +0000594 Changed = CurDAG->LegalizeTypes();
Dan Gohmanb552df72008-07-21 20:00:07 +0000595 } else {
Duncan Sandsa3e2cd02008-11-24 14:53:14 +0000596 Changed = CurDAG->LegalizeTypes();
Dan Gohmanb552df72008-07-21 20:00:07 +0000597 }
598
599 DOUT << "Type-legalized selection DAG:\n";
Dan Gohman14a66442008-08-23 02:25:05 +0000600 DEBUG(CurDAG->dump());
Dan Gohmanb552df72008-07-21 20:00:07 +0000601
Duncan Sandsa3e2cd02008-11-24 14:53:14 +0000602 if (Changed) {
603 if (ViewDAGCombineLT)
604 CurDAG->viewGraph("dag-combine-lt input for " + BlockName);
605
606 // Run the DAG combiner in post-type-legalize mode.
607 if (TimePassesIsEnabled) {
608 NamedRegionTimer T("DAG Combining after legalize types", GroupName);
609 CurDAG->Combine(NoIllegalTypes, *AA, Fast);
610 } else {
611 CurDAG->Combine(NoIllegalTypes, *AA, Fast);
612 }
613
614 DOUT << "Optimized type-legalized selection DAG:\n";
615 DEBUG(CurDAG->dump());
616 }
Chris Lattnerb29a6a42008-07-10 23:37:50 +0000617 }
Duncan Sands31ddf4c2008-07-17 17:06:03 +0000618
Dan Gohman14a66442008-08-23 02:25:05 +0000619 if (ViewLegalizeDAGs) CurDAG->viewGraph("legalize input for " + BlockName);
Dan Gohmanb552df72008-07-21 20:00:07 +0000620
Evan Cheng19733c42008-07-01 17:59:20 +0000621 if (TimePassesIsEnabled) {
Dan Gohman368a08b2008-07-14 18:19:29 +0000622 NamedRegionTimer T("DAG Legalization", GroupName);
Dan Gohman14a66442008-08-23 02:25:05 +0000623 CurDAG->Legalize();
Evan Cheng19733c42008-07-01 17:59:20 +0000624 } else {
Dan Gohman14a66442008-08-23 02:25:05 +0000625 CurDAG->Legalize();
Evan Cheng19733c42008-07-01 17:59:20 +0000626 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000627
628 DOUT << "Legalized selection DAG:\n";
Dan Gohman14a66442008-08-23 02:25:05 +0000629 DEBUG(CurDAG->dump());
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000630
Dan Gohman14a66442008-08-23 02:25:05 +0000631 if (ViewDAGCombine2) CurDAG->viewGraph("dag-combine2 input for " + BlockName);
Dan Gohmanb552df72008-07-21 20:00:07 +0000632
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000633 // Run the DAG combiner in post-legalize mode.
Evan Cheng19733c42008-07-01 17:59:20 +0000634 if (TimePassesIsEnabled) {
Dan Gohman368a08b2008-07-14 18:19:29 +0000635 NamedRegionTimer T("DAG Combining 2", GroupName);
Duncan Sandsa3e2cd02008-11-24 14:53:14 +0000636 CurDAG->Combine(NoIllegalOperations, *AA, Fast);
Evan Cheng19733c42008-07-01 17:59:20 +0000637 } else {
Duncan Sandsa3e2cd02008-11-24 14:53:14 +0000638 CurDAG->Combine(NoIllegalOperations, *AA, Fast);
Evan Cheng19733c42008-07-01 17:59:20 +0000639 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000640
Dan Gohmaneebf44e2007-10-08 15:12:17 +0000641 DOUT << "Optimized legalized selection DAG:\n";
Dan Gohman14a66442008-08-23 02:25:05 +0000642 DEBUG(CurDAG->dump());
Dan Gohmaneebf44e2007-10-08 15:12:17 +0000643
Dan Gohman14a66442008-08-23 02:25:05 +0000644 if (ViewISelDAGs) CurDAG->viewGraph("isel input for " + BlockName);
Chris Lattner68068cc2008-06-17 06:09:18 +0000645
Dan Gohman04dcf8d2008-08-13 19:47:40 +0000646 if (!Fast && EnableValueProp)
Dan Gohman14a66442008-08-23 02:25:05 +0000647 ComputeLiveOutVRegInfo();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000648
649 // Third, instruction select all of the operations to machine code, adding the
650 // code to the MachineBasicBlock.
Evan Cheng19733c42008-07-01 17:59:20 +0000651 if (TimePassesIsEnabled) {
Dan Gohman368a08b2008-07-14 18:19:29 +0000652 NamedRegionTimer T("Instruction Selection", GroupName);
Dan Gohman14a66442008-08-23 02:25:05 +0000653 InstructionSelect();
Evan Cheng19733c42008-07-01 17:59:20 +0000654 } else {
Dan Gohman14a66442008-08-23 02:25:05 +0000655 InstructionSelect();
Evan Cheng19733c42008-07-01 17:59:20 +0000656 }
Evan Cheng34fd4f32008-06-30 20:45:06 +0000657
Dan Gohmanb552df72008-07-21 20:00:07 +0000658 DOUT << "Selected selection DAG:\n";
Dan Gohman14a66442008-08-23 02:25:05 +0000659 DEBUG(CurDAG->dump());
Dan Gohmanb552df72008-07-21 20:00:07 +0000660
Dan Gohman14a66442008-08-23 02:25:05 +0000661 if (ViewSchedDAGs) CurDAG->viewGraph("scheduler input for " + BlockName);
Dan Gohmanb552df72008-07-21 20:00:07 +0000662
Dan Gohman368a08b2008-07-14 18:19:29 +0000663 // Schedule machine code.
664 ScheduleDAG *Scheduler;
665 if (TimePassesIsEnabled) {
666 NamedRegionTimer T("Instruction Scheduling", GroupName);
Dan Gohman14a66442008-08-23 02:25:05 +0000667 Scheduler = Schedule();
Dan Gohman368a08b2008-07-14 18:19:29 +0000668 } else {
Dan Gohman14a66442008-08-23 02:25:05 +0000669 Scheduler = Schedule();
Dan Gohman368a08b2008-07-14 18:19:29 +0000670 }
671
Dan Gohmanb552df72008-07-21 20:00:07 +0000672 if (ViewSUnitDAGs) Scheduler->viewGraph();
673
Evan Cheng34fd4f32008-06-30 20:45:06 +0000674 // Emit machine code to BB. This can change 'BB' to the last block being
675 // inserted into.
Evan Cheng19733c42008-07-01 17:59:20 +0000676 if (TimePassesIsEnabled) {
Dan Gohman368a08b2008-07-14 18:19:29 +0000677 NamedRegionTimer T("Instruction Creation", GroupName);
678 BB = Scheduler->EmitSchedule();
Evan Cheng19733c42008-07-01 17:59:20 +0000679 } else {
Dan Gohman368a08b2008-07-14 18:19:29 +0000680 BB = Scheduler->EmitSchedule();
681 }
682
683 // Free the scheduler state.
684 if (TimePassesIsEnabled) {
685 NamedRegionTimer T("Instruction Scheduling Cleanup", GroupName);
686 delete Scheduler;
687 } else {
688 delete Scheduler;
Evan Cheng19733c42008-07-01 17:59:20 +0000689 }
Evan Cheng34fd4f32008-06-30 20:45:06 +0000690
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000691 DOUT << "Selected machine code:\n";
692 DEBUG(BB->dump());
693}
694
Dan Gohman76dd96e2008-09-23 21:53:34 +0000695void SelectionDAGISel::SelectAllBasicBlocks(Function &Fn, MachineFunction &MF,
Dan Gohman9dd43582008-10-14 23:54:11 +0000696 MachineModuleInfo *MMI,
697 const TargetInstrInfo &TII) {
Dan Gohman9d7bc422008-09-29 21:55:50 +0000698 // Initialize the Fast-ISel state, if needed.
699 FastISel *FastIS = 0;
700 if (EnableFastISel)
701 FastIS = TLI.createFastISel(*FuncInfo->MF, MMI,
702 FuncInfo->ValueMap,
703 FuncInfo->MBBMap,
Dan Gohman9dd43582008-10-14 23:54:11 +0000704 FuncInfo->StaticAllocaMap
705#ifndef NDEBUG
706 , FuncInfo->CatchInfoLost
707#endif
708 );
Dan Gohman9d7bc422008-09-29 21:55:50 +0000709
710 // Iterate over all basic blocks in the function.
Evan Cheng61828a82008-08-07 00:43:25 +0000711 for (Function::iterator I = Fn.begin(), E = Fn.end(); I != E; ++I) {
712 BasicBlock *LLVMBB = &*I;
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000713 BB = FuncInfo->MBBMap[LLVMBB];
Dan Gohman14a66442008-08-23 02:25:05 +0000714
Dan Gohmanca4857a2008-09-03 23:12:08 +0000715 BasicBlock::iterator const Begin = LLVMBB->begin();
716 BasicBlock::iterator const End = LLVMBB->end();
Evan Chenge9d9a162008-09-08 16:01:27 +0000717 BasicBlock::iterator BI = Begin;
Dan Gohman2c44e9a2008-08-28 20:28:56 +0000718
719 // Lower any arguments needed in this block if this is the entry block.
Dan Gohman78ae76d2008-09-25 17:05:24 +0000720 bool SuppressFastISel = false;
721 if (LLVMBB == &Fn.getEntryBlock()) {
Dan Gohman2c44e9a2008-08-28 20:28:56 +0000722 LowerArguments(LLVMBB);
Dan Gohman14a66442008-08-23 02:25:05 +0000723
Dan Gohman78ae76d2008-09-25 17:05:24 +0000724 // If any of the arguments has the byval attribute, forgo
725 // fast-isel in the entry block.
Dan Gohman9d7bc422008-09-29 21:55:50 +0000726 if (FastIS) {
Dan Gohman78ae76d2008-09-25 17:05:24 +0000727 unsigned j = 1;
728 for (Function::arg_iterator I = Fn.arg_begin(), E = Fn.arg_end();
729 I != E; ++I, ++j)
Devang Pateld222f862008-09-25 21:00:45 +0000730 if (Fn.paramHasAttr(j, Attribute::ByVal)) {
Dan Gohman92902e82008-09-25 17:21:42 +0000731 if (EnableFastISelVerbose || EnableFastISelAbort)
732 cerr << "FastISel skips entry block due to byval argument\n";
Dan Gohman78ae76d2008-09-25 17:05:24 +0000733 SuppressFastISel = true;
734 break;
735 }
736 }
737 }
738
Dan Gohman9dd43582008-10-14 23:54:11 +0000739 if (MMI && BB->isLandingPad()) {
740 // Add a label to mark the beginning of the landing pad. Deletion of the
741 // landing pad can thus be detected via the MachineModuleInfo.
742 unsigned LabelID = MMI->addLandingPad(BB);
743
744 const TargetInstrDesc &II = TII.get(TargetInstrInfo::EH_LABEL);
745 BuildMI(BB, II).addImm(LabelID);
746
747 // Mark exception register as live in.
748 unsigned Reg = TLI.getExceptionAddressRegister();
749 if (Reg) BB->addLiveIn(Reg);
750
751 // Mark exception selector register as live in.
752 Reg = TLI.getExceptionSelectorRegister();
753 if (Reg) BB->addLiveIn(Reg);
754
755 // FIXME: Hack around an exception handling flaw (PR1508): the personality
756 // function and list of typeids logically belong to the invoke (or, if you
757 // like, the basic block containing the invoke), and need to be associated
758 // with it in the dwarf exception handling tables. Currently however the
759 // information is provided by an intrinsic (eh.selector) that can be moved
760 // to unexpected places by the optimizers: if the unwind edge is critical,
761 // then breaking it can result in the intrinsics being in the successor of
762 // the landing pad, not the landing pad itself. This results in exceptions
763 // not being caught because no typeids are associated with the invoke.
764 // This may not be the only way things can go wrong, but it is the only way
765 // we try to work around for the moment.
766 BranchInst *Br = dyn_cast<BranchInst>(LLVMBB->getTerminator());
767
768 if (Br && Br->isUnconditional()) { // Critical edge?
769 BasicBlock::iterator I, E;
770 for (I = LLVMBB->begin(), E = --LLVMBB->end(); I != E; ++I)
771 if (isa<EHSelectorInst>(I))
772 break;
773
774 if (I == E)
775 // No catch info found - try to extract some from the successor.
776 copyCatchInfo(Br->getSuccessor(0), LLVMBB, MMI, *FuncInfo);
777 }
778 }
779
Dan Gohman14a66442008-08-23 02:25:05 +0000780 // Before doing SelectionDAG ISel, see if FastISel has been requested.
Dan Gohman9dd43582008-10-14 23:54:11 +0000781 if (FastIS && !SuppressFastISel) {
Dan Gohman9d7bc422008-09-29 21:55:50 +0000782 // Emit code for any incoming arguments. This must happen before
783 // beginning FastISel on the entry block.
784 if (LLVMBB == &Fn.getEntryBlock()) {
785 CurDAG->setRoot(SDL->getControlRoot());
786 CodeGenAndEmitDAG();
787 SDL->clear();
788 }
Dan Gohman2764e172008-10-04 00:56:36 +0000789 FastIS->startNewBlock(BB);
Dan Gohman9d7bc422008-09-29 21:55:50 +0000790 // Do FastISel on as many instructions as possible.
791 for (; BI != End; ++BI) {
792 // Just before the terminator instruction, insert instructions to
793 // feed PHI nodes in successor blocks.
794 if (isa<TerminatorInst>(BI))
795 if (!HandlePHINodesInSuccessorBlocksFast(LLVMBB, FastIS)) {
Dan Gohmanee70e1d2008-09-09 23:05:00 +0000796 if (EnableFastISelVerbose || EnableFastISelAbort) {
Dan Gohman69740ec2008-09-09 22:06:46 +0000797 cerr << "FastISel miss: ";
798 BI->dump();
799 }
Dan Gohmanee70e1d2008-09-09 23:05:00 +0000800 if (EnableFastISelAbort)
Dan Gohman9d7bc422008-09-29 21:55:50 +0000801 assert(0 && "FastISel didn't handle a PHI in a successor");
802 break;
Dan Gohman14a66442008-08-23 02:25:05 +0000803 }
Dan Gohman9d7bc422008-09-29 21:55:50 +0000804
805 // First try normal tablegen-generated "fast" selection.
806 if (FastIS->SelectInstruction(BI))
807 continue;
808
809 // Next, try calling the target to attempt to handle the instruction.
810 if (FastIS->TargetSelectInstruction(BI))
811 continue;
812
813 // Then handle certain instructions as single-LLVM-Instruction blocks.
814 if (isa<CallInst>(BI)) {
815 if (EnableFastISelVerbose || EnableFastISelAbort) {
816 cerr << "FastISel missed call: ";
817 BI->dump();
818 }
819
820 if (BI->getType() != Type::VoidTy) {
821 unsigned &R = FuncInfo->ValueMap[BI];
822 if (!R)
823 R = FuncInfo->CreateRegForValue(BI);
824 }
825
826 SelectBasicBlock(LLVMBB, BI, next(BI));
Dan Gohman2764e172008-10-04 00:56:36 +0000827 // If the instruction was codegen'd with multiple blocks,
828 // inform the FastISel object where to resume inserting.
829 FastIS->setCurrentBlock(BB);
Dan Gohman9d7bc422008-09-29 21:55:50 +0000830 continue;
Dan Gohman14a66442008-08-23 02:25:05 +0000831 }
Dan Gohman9d7bc422008-09-29 21:55:50 +0000832
833 // Otherwise, give up on FastISel for the rest of the block.
834 // For now, be a little lenient about non-branch terminators.
835 if (!isa<TerminatorInst>(BI) || isa<BranchInst>(BI)) {
836 if (EnableFastISelVerbose || EnableFastISelAbort) {
837 cerr << "FastISel miss: ";
838 BI->dump();
839 }
840 if (EnableFastISelAbort)
841 // The "fast" selector couldn't handle something and bailed.
842 // For the purpose of debugging, just abort.
843 assert(0 && "FastISel didn't select the entire block");
844 }
845 break;
Dan Gohman14a66442008-08-23 02:25:05 +0000846 }
847 }
848
Dan Gohman61a1f1c2008-09-02 20:17:56 +0000849 // Run SelectionDAG instruction selection on the remainder of the block
850 // not handled by FastISel. If FastISel is not run, this is the entire
Dan Gohmanca4857a2008-09-03 23:12:08 +0000851 // block.
Evan Chenge9d9a162008-09-08 16:01:27 +0000852 if (BI != End)
853 SelectBasicBlock(LLVMBB, BI, End);
Dan Gohman14a66442008-08-23 02:25:05 +0000854
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000855 FinishBasicBlock();
Evan Cheng61828a82008-08-07 00:43:25 +0000856 }
Dan Gohman9d7bc422008-09-29 21:55:50 +0000857
858 delete FastIS;
Dan Gohmaned825d12008-07-07 23:02:41 +0000859}
860
Dan Gohman2fcbc7e2008-07-28 21:51:04 +0000861void
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000862SelectionDAGISel::FinishBasicBlock() {
Dan Gohman14a66442008-08-23 02:25:05 +0000863
Dan Gohman14a66442008-08-23 02:25:05 +0000864 DOUT << "Target-post-processed machine code:\n";
865 DEBUG(BB->dump());
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000866
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000867 DOUT << "Total amount of phi nodes to update: "
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000868 << SDL->PHINodesToUpdate.size() << "\n";
869 DEBUG(for (unsigned i = 0, e = SDL->PHINodesToUpdate.size(); i != e; ++i)
870 DOUT << "Node " << i << " : (" << SDL->PHINodesToUpdate[i].first
871 << ", " << SDL->PHINodesToUpdate[i].second << ")\n";);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000872
873 // Next, now that we know what the last MBB the LLVM BB expanded is, update
874 // PHI nodes in successors.
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000875 if (SDL->SwitchCases.empty() &&
876 SDL->JTCases.empty() &&
877 SDL->BitTestCases.empty()) {
878 for (unsigned i = 0, e = SDL->PHINodesToUpdate.size(); i != e; ++i) {
879 MachineInstr *PHI = SDL->PHINodesToUpdate[i].first;
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000880 assert(PHI->getOpcode() == TargetInstrInfo::PHI &&
881 "This is not a machine PHI node that we are updating!");
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000882 PHI->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[i].second,
Chris Lattnere44906f2007-12-30 00:57:42 +0000883 false));
884 PHI->addOperand(MachineOperand::CreateMBB(BB));
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000885 }
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000886 SDL->PHINodesToUpdate.clear();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000887 return;
888 }
889
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000890 for (unsigned i = 0, e = SDL->BitTestCases.size(); i != e; ++i) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000891 // Lower header first, if it wasn't already lowered
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000892 if (!SDL->BitTestCases[i].Emitted) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000893 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000894 BB = SDL->BitTestCases[i].Parent;
895 SDL->setCurrentBasicBlock(BB);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000896 // Emit the code
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000897 SDL->visitBitTestHeader(SDL->BitTestCases[i]);
898 CurDAG->setRoot(SDL->getRoot());
Dan Gohman14a66442008-08-23 02:25:05 +0000899 CodeGenAndEmitDAG();
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000900 SDL->clear();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000901 }
902
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000903 for (unsigned j = 0, ej = SDL->BitTestCases[i].Cases.size(); j != ej; ++j) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000904 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000905 BB = SDL->BitTestCases[i].Cases[j].ThisBB;
906 SDL->setCurrentBasicBlock(BB);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000907 // Emit the code
908 if (j+1 != ej)
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000909 SDL->visitBitTestCase(SDL->BitTestCases[i].Cases[j+1].ThisBB,
910 SDL->BitTestCases[i].Reg,
911 SDL->BitTestCases[i].Cases[j]);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000912 else
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000913 SDL->visitBitTestCase(SDL->BitTestCases[i].Default,
914 SDL->BitTestCases[i].Reg,
915 SDL->BitTestCases[i].Cases[j]);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000916
917
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000918 CurDAG->setRoot(SDL->getRoot());
Dan Gohman14a66442008-08-23 02:25:05 +0000919 CodeGenAndEmitDAG();
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000920 SDL->clear();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000921 }
922
923 // Update PHI Nodes
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000924 for (unsigned pi = 0, pe = SDL->PHINodesToUpdate.size(); pi != pe; ++pi) {
925 MachineInstr *PHI = SDL->PHINodesToUpdate[pi].first;
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000926 MachineBasicBlock *PHIBB = PHI->getParent();
927 assert(PHI->getOpcode() == TargetInstrInfo::PHI &&
928 "This is not a machine PHI node that we are updating!");
929 // This is "default" BB. We have two jumps to it. From "header" BB and
930 // from last "case" BB.
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000931 if (PHIBB == SDL->BitTestCases[i].Default) {
932 PHI->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[pi].second,
Chris Lattnere44906f2007-12-30 00:57:42 +0000933 false));
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000934 PHI->addOperand(MachineOperand::CreateMBB(SDL->BitTestCases[i].Parent));
935 PHI->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[pi].second,
Chris Lattnere44906f2007-12-30 00:57:42 +0000936 false));
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000937 PHI->addOperand(MachineOperand::CreateMBB(SDL->BitTestCases[i].Cases.
Chris Lattnere44906f2007-12-30 00:57:42 +0000938 back().ThisBB));
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000939 }
940 // One of "cases" BB.
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000941 for (unsigned j = 0, ej = SDL->BitTestCases[i].Cases.size();
942 j != ej; ++j) {
943 MachineBasicBlock* cBB = SDL->BitTestCases[i].Cases[j].ThisBB;
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000944 if (cBB->succ_end() !=
945 std::find(cBB->succ_begin(),cBB->succ_end(), PHIBB)) {
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000946 PHI->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[pi].second,
Chris Lattnere44906f2007-12-30 00:57:42 +0000947 false));
948 PHI->addOperand(MachineOperand::CreateMBB(cBB));
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000949 }
950 }
951 }
952 }
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000953 SDL->BitTestCases.clear();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000954
955 // If the JumpTable record is filled in, then we need to emit a jump table.
956 // Updating the PHI nodes is tricky in this case, since we need to determine
957 // whether the PHI is a successor of the range check MBB or the jump table MBB
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000958 for (unsigned i = 0, e = SDL->JTCases.size(); i != e; ++i) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000959 // Lower header first, if it wasn't already lowered
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000960 if (!SDL->JTCases[i].first.Emitted) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000961 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000962 BB = SDL->JTCases[i].first.HeaderBB;
963 SDL->setCurrentBasicBlock(BB);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000964 // Emit the code
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000965 SDL->visitJumpTableHeader(SDL->JTCases[i].second, SDL->JTCases[i].first);
966 CurDAG->setRoot(SDL->getRoot());
Dan Gohman14a66442008-08-23 02:25:05 +0000967 CodeGenAndEmitDAG();
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000968 SDL->clear();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000969 }
970
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000971 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000972 BB = SDL->JTCases[i].second.MBB;
973 SDL->setCurrentBasicBlock(BB);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000974 // Emit the code
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000975 SDL->visitJumpTable(SDL->JTCases[i].second);
976 CurDAG->setRoot(SDL->getRoot());
Dan Gohman14a66442008-08-23 02:25:05 +0000977 CodeGenAndEmitDAG();
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000978 SDL->clear();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000979
980 // Update PHI Nodes
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000981 for (unsigned pi = 0, pe = SDL->PHINodesToUpdate.size(); pi != pe; ++pi) {
982 MachineInstr *PHI = SDL->PHINodesToUpdate[pi].first;
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000983 MachineBasicBlock *PHIBB = PHI->getParent();
984 assert(PHI->getOpcode() == TargetInstrInfo::PHI &&
985 "This is not a machine PHI node that we are updating!");
986 // "default" BB. We can go there only from header BB.
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000987 if (PHIBB == SDL->JTCases[i].second.Default) {
988 PHI->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[pi].second,
Chris Lattnere44906f2007-12-30 00:57:42 +0000989 false));
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000990 PHI->addOperand(MachineOperand::CreateMBB(SDL->JTCases[i].first.HeaderBB));
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000991 }
992 // JT BB. Just iterate over successors here
993 if (BB->succ_end() != std::find(BB->succ_begin(),BB->succ_end(), PHIBB)) {
Dan Gohman0f2d71d2008-08-27 23:52:12 +0000994 PHI->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[pi].second,
Chris Lattnere44906f2007-12-30 00:57:42 +0000995 false));
996 PHI->addOperand(MachineOperand::CreateMBB(BB));
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000997 }
998 }
999 }
Dan Gohman0f2d71d2008-08-27 23:52:12 +00001000 SDL->JTCases.clear();
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001001
1002 // If the switch block involved a branch to one of the actual successors, we
1003 // need to update PHI nodes in that block.
Dan Gohman0f2d71d2008-08-27 23:52:12 +00001004 for (unsigned i = 0, e = SDL->PHINodesToUpdate.size(); i != e; ++i) {
1005 MachineInstr *PHI = SDL->PHINodesToUpdate[i].first;
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001006 assert(PHI->getOpcode() == TargetInstrInfo::PHI &&
1007 "This is not a machine PHI node that we are updating!");
1008 if (BB->isSuccessor(PHI->getParent())) {
Dan Gohman0f2d71d2008-08-27 23:52:12 +00001009 PHI->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[i].second,
Chris Lattnere44906f2007-12-30 00:57:42 +00001010 false));
1011 PHI->addOperand(MachineOperand::CreateMBB(BB));
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001012 }
1013 }
1014
1015 // If we generated any switch lowering information, build and codegen any
1016 // additional DAGs necessary.
Dan Gohman0f2d71d2008-08-27 23:52:12 +00001017 for (unsigned i = 0, e = SDL->SwitchCases.size(); i != e; ++i) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001018 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman0f2d71d2008-08-27 23:52:12 +00001019 BB = SDL->SwitchCases[i].ThisBB;
1020 SDL->setCurrentBasicBlock(BB);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001021
1022 // Emit the code
Dan Gohman0f2d71d2008-08-27 23:52:12 +00001023 SDL->visitSwitchCase(SDL->SwitchCases[i]);
1024 CurDAG->setRoot(SDL->getRoot());
Dan Gohman14a66442008-08-23 02:25:05 +00001025 CodeGenAndEmitDAG();
Dan Gohman0f2d71d2008-08-27 23:52:12 +00001026 SDL->clear();
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001027
1028 // Handle any PHI nodes in successors of this chunk, as if we were coming
1029 // from the original BB before switch expansion. Note that PHI nodes can
1030 // occur multiple times in PHINodesToUpdate. We have to be very careful to
1031 // handle them the right number of times.
Dan Gohman0f2d71d2008-08-27 23:52:12 +00001032 while ((BB = SDL->SwitchCases[i].TrueBB)) { // Handle LHS and RHS.
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001033 for (MachineBasicBlock::iterator Phi = BB->begin();
1034 Phi != BB->end() && Phi->getOpcode() == TargetInstrInfo::PHI; ++Phi){
1035 // This value for this PHI node is recorded in PHINodesToUpdate, get it.
1036 for (unsigned pn = 0; ; ++pn) {
Dan Gohman0f2d71d2008-08-27 23:52:12 +00001037 assert(pn != SDL->PHINodesToUpdate.size() &&
1038 "Didn't find PHI entry!");
1039 if (SDL->PHINodesToUpdate[pn].first == Phi) {
1040 Phi->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[pn].
Chris Lattnere44906f2007-12-30 00:57:42 +00001041 second, false));
Dan Gohman0f2d71d2008-08-27 23:52:12 +00001042 Phi->addOperand(MachineOperand::CreateMBB(SDL->SwitchCases[i].ThisBB));
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001043 break;
1044 }
1045 }
1046 }
1047
1048 // Don't process RHS if same block as LHS.
Dan Gohman0f2d71d2008-08-27 23:52:12 +00001049 if (BB == SDL->SwitchCases[i].FalseBB)
1050 SDL->SwitchCases[i].FalseBB = 0;
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001051
1052 // If we haven't handled the RHS, do so now. Otherwise, we're done.
Dan Gohman0f2d71d2008-08-27 23:52:12 +00001053 SDL->SwitchCases[i].TrueBB = SDL->SwitchCases[i].FalseBB;
1054 SDL->SwitchCases[i].FalseBB = 0;
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001055 }
Dan Gohman0f2d71d2008-08-27 23:52:12 +00001056 assert(SDL->SwitchCases[i].TrueBB == 0 && SDL->SwitchCases[i].FalseBB == 0);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001057 }
Dan Gohman0f2d71d2008-08-27 23:52:12 +00001058 SDL->SwitchCases.clear();
1059
1060 SDL->PHINodesToUpdate.clear();
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001061}
1062
1063
Dan Gohman368a08b2008-07-14 18:19:29 +00001064/// Schedule - Pick a safe ordering for instructions for each
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001065/// target node in the graph.
Dan Gohman368a08b2008-07-14 18:19:29 +00001066///
Dan Gohman14a66442008-08-23 02:25:05 +00001067ScheduleDAG *SelectionDAGISel::Schedule() {
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001068 RegisterScheduler::FunctionPassCtor Ctor = RegisterScheduler::getDefault();
1069
1070 if (!Ctor) {
1071 Ctor = ISHeuristic;
1072 RegisterScheduler::setDefault(Ctor);
1073 }
1074
Dan Gohman3ea46f72008-11-11 17:50:47 +00001075 TargetMachine &TM = getTargetLowering().getTargetMachine();
1076 ScheduleDAG *Scheduler = Ctor(this, CurDAG, &TM, BB, Fast);
Dan Gohman368a08b2008-07-14 18:19:29 +00001077 Scheduler->Run();
Dan Gohman134c5b62007-08-28 20:32:58 +00001078
Dan Gohman368a08b2008-07-14 18:19:29 +00001079 return Scheduler;
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001080}
1081
1082
1083HazardRecognizer *SelectionDAGISel::CreateTargetHazardRecognizer() {
1084 return new HazardRecognizer();
1085}
1086
1087//===----------------------------------------------------------------------===//
1088// Helper functions used by the generated instruction selector.
1089//===----------------------------------------------------------------------===//
1090// Calls to these methods are generated by tblgen.
1091
1092/// CheckAndMask - The isel is trying to match something like (and X, 255). If
1093/// the dag combiner simplified the 255, we still want to match. RHS is the
1094/// actual value in the DAG on the RHS of an AND, and DesiredMaskS is the value
1095/// specified in the .td file (e.g. 255).
Dan Gohman8181bd12008-07-27 21:46:04 +00001096bool SelectionDAGISel::CheckAndMask(SDValue LHS, ConstantSDNode *RHS,
Dan Gohmand6098272007-07-24 23:00:27 +00001097 int64_t DesiredMaskS) const {
Dan Gohman07961cd2008-02-25 21:11:39 +00001098 const APInt &ActualMask = RHS->getAPIntValue();
1099 const APInt &DesiredMask = APInt(LHS.getValueSizeInBits(), DesiredMaskS);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001100
1101 // If the actual mask exactly matches, success!
1102 if (ActualMask == DesiredMask)
1103 return true;
1104
1105 // If the actual AND mask is allowing unallowed bits, this doesn't match.
Dan Gohman07961cd2008-02-25 21:11:39 +00001106 if (ActualMask.intersects(~DesiredMask))
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001107 return false;
1108
1109 // Otherwise, the DAG Combiner may have proven that the value coming in is
1110 // either already zero or is not demanded. Check for known zero input bits.
Dan Gohman07961cd2008-02-25 21:11:39 +00001111 APInt NeededMask = DesiredMask & ~ActualMask;
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001112 if (CurDAG->MaskedValueIsZero(LHS, NeededMask))
1113 return true;
1114
1115 // TODO: check to see if missing bits are just not demanded.
1116
1117 // Otherwise, this pattern doesn't match.
1118 return false;
1119}
1120
1121/// CheckOrMask - The isel is trying to match something like (or X, 255). If
1122/// the dag combiner simplified the 255, we still want to match. RHS is the
1123/// actual value in the DAG on the RHS of an OR, and DesiredMaskS is the value
1124/// specified in the .td file (e.g. 255).
Dan Gohman8181bd12008-07-27 21:46:04 +00001125bool SelectionDAGISel::CheckOrMask(SDValue LHS, ConstantSDNode *RHS,
Dan Gohman07961cd2008-02-25 21:11:39 +00001126 int64_t DesiredMaskS) const {
1127 const APInt &ActualMask = RHS->getAPIntValue();
1128 const APInt &DesiredMask = APInt(LHS.getValueSizeInBits(), DesiredMaskS);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001129
1130 // If the actual mask exactly matches, success!
1131 if (ActualMask == DesiredMask)
1132 return true;
1133
1134 // If the actual AND mask is allowing unallowed bits, this doesn't match.
Dan Gohman07961cd2008-02-25 21:11:39 +00001135 if (ActualMask.intersects(~DesiredMask))
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001136 return false;
1137
1138 // Otherwise, the DAG Combiner may have proven that the value coming in is
1139 // either already zero or is not demanded. Check for known zero input bits.
Dan Gohman07961cd2008-02-25 21:11:39 +00001140 APInt NeededMask = DesiredMask & ~ActualMask;
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001141
Dan Gohman07961cd2008-02-25 21:11:39 +00001142 APInt KnownZero, KnownOne;
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001143 CurDAG->ComputeMaskedBits(LHS, NeededMask, KnownZero, KnownOne);
1144
1145 // If all the missing bits in the or are already known to be set, match!
1146 if ((NeededMask & KnownOne) == NeededMask)
1147 return true;
1148
1149 // TODO: check to see if missing bits are just not demanded.
1150
1151 // Otherwise, this pattern doesn't match.
1152 return false;
1153}
1154
1155
1156/// SelectInlineAsmMemoryOperands - Calls to this are automatically generated
1157/// by tblgen. Others should not call it.
1158void SelectionDAGISel::
Dan Gohman14a66442008-08-23 02:25:05 +00001159SelectInlineAsmMemoryOperands(std::vector<SDValue> &Ops) {
Dan Gohman8181bd12008-07-27 21:46:04 +00001160 std::vector<SDValue> InOps;
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001161 std::swap(InOps, Ops);
1162
1163 Ops.push_back(InOps[0]); // input chain.
1164 Ops.push_back(InOps[1]); // input asm string.
1165
1166 unsigned i = 2, e = InOps.size();
1167 if (InOps[e-1].getValueType() == MVT::Flag)
1168 --e; // Don't process a flag operand if it is here.
1169
1170 while (i != e) {
Dan Gohmanfaeb4a32008-09-12 16:56:44 +00001171 unsigned Flags = cast<ConstantSDNode>(InOps[i])->getZExtValue();
Dale Johannesen94464072008-09-24 01:07:17 +00001172 if ((Flags & 7) != 4 /*MEM*/) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001173 // Just skip over this operand, copying the operands verbatim.
1174 Ops.insert(Ops.end(), InOps.begin()+i, InOps.begin()+i+(Flags >> 3) + 1);
1175 i += (Flags >> 3) + 1;
1176 } else {
1177 assert((Flags >> 3) == 1 && "Memory operand with multiple values?");
1178 // Otherwise, this is a memory operand. Ask the target to select it.
Dan Gohman8181bd12008-07-27 21:46:04 +00001179 std::vector<SDValue> SelOps;
Dan Gohman14a66442008-08-23 02:25:05 +00001180 if (SelectInlineAsmMemoryOperand(InOps[i+1], 'm', SelOps)) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001181 cerr << "Could not match memory address. Inline asm failure!\n";
1182 exit(1);
1183 }
1184
1185 // Add this to the output node.
Dan Gohman14a66442008-08-23 02:25:05 +00001186 MVT IntPtrTy = CurDAG->getTargetLoweringInfo().getPointerTy();
Dale Johannesen94464072008-09-24 01:07:17 +00001187 Ops.push_back(CurDAG->getTargetConstant(4/*MEM*/ | (SelOps.size()<< 3),
Dan Gohman14a66442008-08-23 02:25:05 +00001188 IntPtrTy));
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001189 Ops.insert(Ops.end(), SelOps.begin(), SelOps.end());
1190 i += 2;
1191 }
1192 }
1193
1194 // Add the flag input back if present.
1195 if (e != InOps.size())
1196 Ops.push_back(InOps.back());
1197}
1198
1199char SelectionDAGISel::ID = 0;