blob: e43b32535c3ddbecaf58f446844c405129fd36de [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"
Evan Cheng381cb072008-08-08 07:27:28 +000015#include "llvm/CodeGen/SelectionDAGISel.h"
Dan Gohmanf0cbcd42008-09-03 16:12:24 +000016#include "SelectionDAGBuild.h"
Anton Korobeynikov5502bf62007-04-04 21:14:49 +000017#include "llvm/ADT/BitVector.h"
Jim Laskeyc7c3f112006-10-16 20:52:31 +000018#include "llvm/Analysis/AliasAnalysis.h"
Anton Korobeynikov5502bf62007-04-04 21:14:49 +000019#include "llvm/Constants.h"
Chris Lattneradf6a962005-05-13 18:50:42 +000020#include "llvm/CallingConv.h"
Chris Lattner1c08c712005-01-07 07:47:53 +000021#include "llvm/DerivedTypes.h"
22#include "llvm/Function.h"
Chris Lattner36ce6912005-11-29 06:21:05 +000023#include "llvm/GlobalVariable.h"
Chris Lattnerce7518c2006-01-26 22:24:51 +000024#include "llvm/InlineAsm.h"
Chris Lattner1c08c712005-01-07 07:47:53 +000025#include "llvm/Instructions.h"
26#include "llvm/Intrinsics.h"
Jim Laskey43970fe2006-03-23 18:06:46 +000027#include "llvm/IntrinsicInst.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"
32#include "llvm/CodeGen/MachineFrameInfo.h"
33#include "llvm/CodeGen/MachineInstrBuilder.h"
Chris Lattner84bc5422007-12-31 04:13:23 +000034#include "llvm/CodeGen/MachineJumpTableInfo.h"
35#include "llvm/CodeGen/MachineModuleInfo.h"
36#include "llvm/CodeGen/MachineRegisterInfo.h"
Dan Gohman343f0c02008-11-19 23:18:57 +000037#include "llvm/CodeGen/ScheduleDAGSDNodes.h"
Jim Laskeyeb577ba2006-08-02 12:30:23 +000038#include "llvm/CodeGen/SchedulerRegistry.h"
Chris Lattner1c08c712005-01-07 07:47:53 +000039#include "llvm/CodeGen/SelectionDAG.h"
Dan Gohman6f0d0242008-02-10 18:45:23 +000040#include "llvm/Target/TargetRegisterInfo.h"
Chris Lattner1c08c712005-01-07 07:47:53 +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"
Vladimir Prus12472912006-05-23 13:43:15 +000046#include "llvm/Target/TargetOptions.h"
Chris Lattnera4f0b3a2006-08-27 12:54:02 +000047#include "llvm/Support/Compiler.h"
Evan Chengdb8d56b2008-06-30 20:45:06 +000048#include "llvm/Support/Debug.h"
49#include "llvm/Support/MathExtras.h"
50#include "llvm/Support/Timer.h"
Jeff Cohen7e881032006-02-24 02:52:40 +000051#include <algorithm>
Chris Lattner1c08c712005-01-07 07:47:53 +000052using namespace llvm;
53
Chris Lattneread0d882008-06-17 06:09:18 +000054static cl::opt<bool>
Chris Lattner70587ea2008-07-10 23:37:50 +000055EnableValueProp("enable-value-prop", cl::Hidden);
56static cl::opt<bool>
Duncan Sands7cb07872008-10-27 08:42:46 +000057DisableLegalizeTypes("disable-legalize-types", cl::Hidden);
Dan Gohman727809a2008-10-28 19:08:46 +000058#ifndef NDEBUG
Dan Gohman78eca172008-08-19 22:33:34 +000059static cl::opt<bool>
Dan Gohman293d5f82008-09-09 22:06:46 +000060EnableFastISelVerbose("fast-isel-verbose", cl::Hidden,
Dan Gohmand659d502008-10-20 21:30:12 +000061 cl::desc("Enable verbose messages in the \"fast\" "
Dan Gohman293d5f82008-09-09 22:06:46 +000062 "instruction selector"));
63static cl::opt<bool>
Dan Gohman4344a5d2008-09-09 23:05:00 +000064EnableFastISelAbort("fast-isel-abort", cl::Hidden,
65 cl::desc("Enable abort calls when \"fast\" instruction fails"));
Dan Gohman22751052008-10-28 20:35:31 +000066#else
67static const bool EnableFastISelVerbose = false,
68 EnableFastISelAbort = false;
Dan Gohman727809a2008-10-28 19:08:46 +000069#endif
Dan Gohman8a110532008-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 Lattneread0d882008-06-17 06:09:18 +000074
Chris Lattnerda8abb02005-09-01 18:44:10 +000075#ifndef NDEBUG
Chris Lattner7944d9d2005-01-12 03:41:21 +000076static cl::opt<bool>
Dan Gohman462dc7f2008-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>
Evan Chenga9c20912006-01-21 02:32:06 +000091ViewISelDAGs("view-isel-dags", cl::Hidden,
92 cl::desc("Pop up a window to show isel dags as they are selected"));
93static cl::opt<bool>
94ViewSchedDAGs("view-sched-dags", cl::Hidden,
95 cl::desc("Pop up a window to show sched dags as they are processed"));
Dan Gohman3e1a7ae2007-08-28 20:32:58 +000096static cl::opt<bool>
97ViewSUnitDAGs("view-sunit-dags", cl::Hidden,
Chris Lattner5bab7852008-01-25 17:24:52 +000098 cl::desc("Pop up a window to show SUnit dags after they are processed"));
Chris Lattner7944d9d2005-01-12 03:41:21 +000099#else
Dan Gohman462dc7f2008-07-21 20:00:07 +0000100static const bool ViewDAGCombine1 = false,
101 ViewLegalizeTypesDAGs = false, ViewLegalizeDAGs = false,
102 ViewDAGCombine2 = false,
103 ViewISelDAGs = false, ViewSchedDAGs = false,
104 ViewSUnitDAGs = false;
Chris Lattner7944d9d2005-01-12 03:41:21 +0000105#endif
106
Jim Laskeyeb577ba2006-08-02 12:30:23 +0000107//===---------------------------------------------------------------------===//
108///
109/// RegisterScheduler class - Track the registration of instruction schedulers.
110///
111//===---------------------------------------------------------------------===//
112MachinePassRegistry RegisterScheduler::Registry;
113
114//===---------------------------------------------------------------------===//
115///
116/// ISHeuristic command line option for instruction schedulers.
117///
118//===---------------------------------------------------------------------===//
Dan Gohman844731a2008-05-13 00:00:25 +0000119static cl::opt<RegisterScheduler::FunctionPassCtor, false,
120 RegisterPassParser<RegisterScheduler> >
121ISHeuristic("pre-RA-sched",
122 cl::init(&createDefaultScheduler),
123 cl::desc("Instruction schedulers available (before register"
124 " allocation):"));
Jim Laskey13ec7022006-08-01 14:21:23 +0000125
Dan Gohman844731a2008-05-13 00:00:25 +0000126static RegisterScheduler
Dan Gohmanb8cab922008-10-14 20:25:08 +0000127defaultListDAGScheduler("default", "Best scheduler for the target",
Dan Gohman844731a2008-05-13 00:00:25 +0000128 createDefaultScheduler);
Evan Cheng4ef10862006-01-23 07:01:07 +0000129
Chris Lattner1c08c712005-01-07 07:47:53 +0000130namespace llvm {
131 //===--------------------------------------------------------------------===//
Jim Laskey9373beb2006-08-01 19:14:14 +0000132 /// createDefaultScheduler - This creates an instruction scheduler appropriate
133 /// for the target.
134 ScheduleDAG* createDefaultScheduler(SelectionDAGISel *IS,
135 SelectionDAG *DAG,
Dan Gohman9b75b372008-11-11 17:50:47 +0000136 const TargetMachine *TM,
Evan Cheng4576f6d2008-07-01 18:05:03 +0000137 MachineBasicBlock *BB,
138 bool Fast) {
Jim Laskey9373beb2006-08-01 19:14:14 +0000139 TargetLowering &TLI = IS->getTargetLowering();
140
141 if (TLI.getSchedulingPreference() == TargetLowering::SchedulingForLatency) {
Dan Gohman9b75b372008-11-11 17:50:47 +0000142 return createTDListDAGScheduler(IS, DAG, TM, BB, Fast);
Jim Laskey9373beb2006-08-01 19:14:14 +0000143 } else {
144 assert(TLI.getSchedulingPreference() ==
145 TargetLowering::SchedulingForRegPressure && "Unknown sched type!");
Dan Gohman9b75b372008-11-11 17:50:47 +0000146 return createBURRListDAGScheduler(IS, DAG, TM, BB, Fast);
Jim Laskey9373beb2006-08-01 19:14:14 +0000147 }
148 }
Chris Lattner1c08c712005-01-07 07:47:53 +0000149}
150
Evan Chengff9b3732008-01-30 18:18:23 +0000151// EmitInstrWithCustomInserter - This method should be implemented by targets
152// that mark instructions with the 'usesCustomDAGSchedInserter' flag. These
Chris Lattner025c39b2005-08-26 20:54:47 +0000153// instructions are special in various ways, which require special support to
154// insert. The specified MachineInstr is created but not inserted into any
155// basic blocks, and the scheduler passes ownership of it to this method.
Evan Chengff9b3732008-01-30 18:18:23 +0000156MachineBasicBlock *TargetLowering::EmitInstrWithCustomInserter(MachineInstr *MI,
Chris Lattner025c39b2005-08-26 20:54:47 +0000157 MachineBasicBlock *MBB) {
Bill Wendling832171c2006-12-07 20:04:42 +0000158 cerr << "If a target marks an instruction with "
159 << "'usesCustomDAGSchedInserter', it must implement "
Evan Chengff9b3732008-01-30 18:18:23 +0000160 << "TargetLowering::EmitInstrWithCustomInserter!\n";
Chris Lattner025c39b2005-08-26 20:54:47 +0000161 abort();
162 return 0;
163}
164
Dan Gohman8a110532008-09-05 22:59:21 +0000165/// EmitLiveInCopy - Emit a copy for a live in physical register. If the
166/// physical register has only a single copy use, then coalesced the copy
167/// if possible.
168static void EmitLiveInCopy(MachineBasicBlock *MBB,
169 MachineBasicBlock::iterator &InsertPos,
170 unsigned VirtReg, unsigned PhysReg,
171 const TargetRegisterClass *RC,
172 DenseMap<MachineInstr*, unsigned> &CopyRegMap,
173 const MachineRegisterInfo &MRI,
174 const TargetRegisterInfo &TRI,
175 const TargetInstrInfo &TII) {
176 unsigned NumUses = 0;
177 MachineInstr *UseMI = NULL;
178 for (MachineRegisterInfo::use_iterator UI = MRI.use_begin(VirtReg),
179 UE = MRI.use_end(); UI != UE; ++UI) {
180 UseMI = &*UI;
181 if (++NumUses > 1)
182 break;
183 }
184
185 // If the number of uses is not one, or the use is not a move instruction,
186 // don't coalesce. Also, only coalesce away a virtual register to virtual
187 // register copy.
188 bool Coalesced = false;
189 unsigned SrcReg, DstReg;
190 if (NumUses == 1 &&
191 TII.isMoveInstr(*UseMI, SrcReg, DstReg) &&
192 TargetRegisterInfo::isVirtualRegister(DstReg)) {
193 VirtReg = DstReg;
194 Coalesced = true;
195 }
196
197 // Now find an ideal location to insert the copy.
198 MachineBasicBlock::iterator Pos = InsertPos;
199 while (Pos != MBB->begin()) {
200 MachineInstr *PrevMI = prior(Pos);
201 DenseMap<MachineInstr*, unsigned>::iterator RI = CopyRegMap.find(PrevMI);
202 // copyRegToReg might emit multiple instructions to do a copy.
203 unsigned CopyDstReg = (RI == CopyRegMap.end()) ? 0 : RI->second;
204 if (CopyDstReg && !TRI.regsOverlap(CopyDstReg, PhysReg))
205 // This is what the BB looks like right now:
206 // r1024 = mov r0
207 // ...
208 // r1 = mov r1024
209 //
210 // We want to insert "r1025 = mov r1". Inserting this copy below the
211 // move to r1024 makes it impossible for that move to be coalesced.
212 //
213 // r1025 = mov r1
214 // r1024 = mov r0
215 // ...
216 // r1 = mov 1024
217 // r2 = mov 1025
218 break; // Woot! Found a good location.
219 --Pos;
220 }
221
222 TII.copyRegToReg(*MBB, Pos, VirtReg, PhysReg, RC, RC);
223 CopyRegMap.insert(std::make_pair(prior(Pos), VirtReg));
224 if (Coalesced) {
225 if (&*InsertPos == UseMI) ++InsertPos;
226 MBB->erase(UseMI);
227 }
228}
229
230/// EmitLiveInCopies - If this is the first basic block in the function,
231/// and if it has live ins that need to be copied into vregs, emit the
232/// copies into the block.
233static void EmitLiveInCopies(MachineBasicBlock *EntryMBB,
234 const MachineRegisterInfo &MRI,
235 const TargetRegisterInfo &TRI,
236 const TargetInstrInfo &TII) {
237 if (SchedLiveInCopies) {
238 // Emit the copies at a heuristically-determined location in the block.
239 DenseMap<MachineInstr*, unsigned> CopyRegMap;
240 MachineBasicBlock::iterator InsertPos = EntryMBB->begin();
241 for (MachineRegisterInfo::livein_iterator LI = MRI.livein_begin(),
242 E = MRI.livein_end(); LI != E; ++LI)
243 if (LI->second) {
244 const TargetRegisterClass *RC = MRI.getRegClass(LI->second);
245 EmitLiveInCopy(EntryMBB, InsertPos, LI->second, LI->first,
246 RC, CopyRegMap, MRI, TRI, TII);
247 }
248 } else {
249 // Emit the copies into the top of the block.
250 for (MachineRegisterInfo::livein_iterator LI = MRI.livein_begin(),
251 E = MRI.livein_end(); LI != E; ++LI)
252 if (LI->second) {
253 const TargetRegisterClass *RC = MRI.getRegClass(LI->second);
254 TII.copyRegToReg(*EntryMBB, EntryMBB->begin(),
255 LI->second, LI->first, RC, RC);
256 }
257 }
258}
259
Chris Lattner7041ee32005-01-11 05:56:49 +0000260//===----------------------------------------------------------------------===//
261// SelectionDAGISel code
262//===----------------------------------------------------------------------===//
Chris Lattner1c08c712005-01-07 07:47:53 +0000263
Dan Gohman7c3234c2008-08-27 23:52:12 +0000264SelectionDAGISel::SelectionDAGISel(TargetLowering &tli, bool fast) :
Dan Gohmanae73dc12008-09-04 17:05:41 +0000265 FunctionPass(&ID), TLI(tli),
Dan Gohman7c3234c2008-08-27 23:52:12 +0000266 FuncInfo(new FunctionLoweringInfo(TLI)),
267 CurDAG(new SelectionDAG(TLI, *FuncInfo)),
268 SDL(new SelectionDAGLowering(*CurDAG, TLI, *FuncInfo)),
269 GFI(),
270 Fast(fast),
271 DAGSize(0)
272{}
273
274SelectionDAGISel::~SelectionDAGISel() {
275 delete SDL;
276 delete CurDAG;
277 delete FuncInfo;
278}
279
Duncan Sands83ec4b62008-06-06 12:08:01 +0000280unsigned SelectionDAGISel::MakeReg(MVT VT) {
Chris Lattner84bc5422007-12-31 04:13:23 +0000281 return RegInfo->createVirtualRegister(TLI.getRegClassFor(VT));
Chris Lattner1c08c712005-01-07 07:47:53 +0000282}
283
Chris Lattner495a0b52005-08-17 06:37:43 +0000284void SelectionDAGISel::getAnalysisUsage(AnalysisUsage &AU) const {
Jim Laskeyc7c3f112006-10-16 20:52:31 +0000285 AU.addRequired<AliasAnalysis>();
Gordon Henriksen5eca0752008-08-17 18:44:35 +0000286 AU.addRequired<GCModuleInfo>();
Chris Lattnerc8d288f2007-03-31 04:18:03 +0000287 AU.setPreservesAll();
Chris Lattner495a0b52005-08-17 06:37:43 +0000288}
Chris Lattner1c08c712005-01-07 07:47:53 +0000289
Chris Lattner1c08c712005-01-07 07:47:53 +0000290bool SelectionDAGISel::runOnFunction(Function &Fn) {
Dan Gohman4344a5d2008-09-09 23:05:00 +0000291 // Do some sanity-checking on the command-line options.
292 assert((!EnableFastISelVerbose || EnableFastISel) &&
293 "-fast-isel-verbose requires -fast-isel");
294 assert((!EnableFastISelAbort || EnableFastISel) &&
295 "-fast-isel-abort requires -fast-isel");
296
Dan Gohman5f43f922007-08-27 16:26:13 +0000297 // Get alias analysis for load/store combining.
298 AA = &getAnalysis<AliasAnalysis>();
299
Dan Gohman8a110532008-09-05 22:59:21 +0000300 TargetMachine &TM = TLI.getTargetMachine();
301 MachineFunction &MF = MachineFunction::construct(&Fn, TM);
302 const MachineRegisterInfo &MRI = MF.getRegInfo();
303 const TargetInstrInfo &TII = *TM.getInstrInfo();
304 const TargetRegisterInfo &TRI = *TM.getRegisterInfo();
305
Gordon Henriksen5eca0752008-08-17 18:44:35 +0000306 if (MF.getFunction()->hasGC())
307 GFI = &getAnalysis<GCModuleInfo>().getFunctionInfo(*MF.getFunction());
Gordon Henriksence224772008-01-07 01:30:38 +0000308 else
Gordon Henriksen5eca0752008-08-17 18:44:35 +0000309 GFI = 0;
Chris Lattner84bc5422007-12-31 04:13:23 +0000310 RegInfo = &MF.getRegInfo();
Bill Wendling832171c2006-12-07 20:04:42 +0000311 DOUT << "\n\n\n=== " << Fn.getName() << "\n";
Chris Lattner1c08c712005-01-07 07:47:53 +0000312
Dan Gohmanf0cbcd42008-09-03 16:12:24 +0000313 FuncInfo->set(Fn, MF, EnableFastISel);
Dan Gohmand57dd5f2008-09-23 21:53:34 +0000314 MachineModuleInfo *MMI = getAnalysisToUpdate<MachineModuleInfo>();
315 CurDAG->init(MF, MMI);
Dan Gohman7c3234c2008-08-27 23:52:12 +0000316 SDL->init(GFI, *AA);
Chris Lattner1c08c712005-01-07 07:47:53 +0000317
Dale Johannesen1532f3d2008-04-02 00:25:04 +0000318 for (Function::iterator I = Fn.begin(), E = Fn.end(); I != E; ++I)
319 if (InvokeInst *Invoke = dyn_cast<InvokeInst>(I->getTerminator()))
320 // Mark landing pad.
Dan Gohman7c3234c2008-08-27 23:52:12 +0000321 FuncInfo->MBBMap[Invoke->getSuccessor(1)]->setIsLandingPad();
Duncan Sands9fac0b52007-06-06 10:05:18 +0000322
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000323 SelectAllBasicBlocks(Fn, MF, MMI, TII);
Misha Brukmanedf128a2005-04-21 22:36:52 +0000324
Dan Gohman8a110532008-09-05 22:59:21 +0000325 // If the first basic block in the function has live ins that need to be
326 // copied into vregs, emit the copies into the top of the block before
327 // emitting the code for the block.
328 EmitLiveInCopies(MF.begin(), MRI, TRI, TII);
329
Evan Chengad2070c2007-02-10 02:43:39 +0000330 // Add function live-ins to entry block live-in set.
Dan Gohman8a110532008-09-05 22:59:21 +0000331 for (MachineRegisterInfo::livein_iterator I = RegInfo->livein_begin(),
332 E = RegInfo->livein_end(); I != E; ++I)
333 MF.begin()->addLiveIn(I->first);
Evan Chengad2070c2007-02-10 02:43:39 +0000334
Duncan Sandsf4070822007-06-15 19:04:19 +0000335#ifndef NDEBUG
Dan Gohman7c3234c2008-08-27 23:52:12 +0000336 assert(FuncInfo->CatchInfoFound.size() == FuncInfo->CatchInfoLost.size() &&
Duncan Sandsf4070822007-06-15 19:04:19 +0000337 "Not all catch info was assigned to a landing pad!");
338#endif
339
Dan Gohman7c3234c2008-08-27 23:52:12 +0000340 FuncInfo->clear();
341
Chris Lattner1c08c712005-01-07 07:47:53 +0000342 return true;
343}
344
Duncan Sandsf4070822007-06-15 19:04:19 +0000345static void copyCatchInfo(BasicBlock *SrcBB, BasicBlock *DestBB,
346 MachineModuleInfo *MMI, FunctionLoweringInfo &FLI) {
Duncan Sandsf4070822007-06-15 19:04:19 +0000347 for (BasicBlock::iterator I = SrcBB->begin(), E = --SrcBB->end(); I != E; ++I)
Dan Gohmanf0cbcd42008-09-03 16:12:24 +0000348 if (EHSelectorInst *EHSel = dyn_cast<EHSelectorInst>(I)) {
Duncan Sandsf4070822007-06-15 19:04:19 +0000349 // Apply the catch info to DestBB.
Dan Gohmanf0cbcd42008-09-03 16:12:24 +0000350 AddCatchInfo(*EHSel, MMI, FLI.MBBMap[DestBB]);
Duncan Sandsf4070822007-06-15 19:04:19 +0000351#ifndef NDEBUG
Duncan Sands560a7372007-11-15 09:54:37 +0000352 if (!FLI.MBBMap[SrcBB]->isLandingPad())
Dan Gohmanf0cbcd42008-09-03 16:12:24 +0000353 FLI.CatchInfoFound.insert(EHSel);
Duncan Sandsf4070822007-06-15 19:04:19 +0000354#endif
355 }
356}
357
Arnold Schwaighofer30e62c02008-04-30 09:16:33 +0000358/// IsFixedFrameObjectWithPosOffset - Check if object is a fixed frame object and
359/// whether object offset >= 0.
360static bool
Dan Gohman475871a2008-07-27 21:46:04 +0000361IsFixedFrameObjectWithPosOffset(MachineFrameInfo * MFI, SDValue Op) {
Arnold Schwaighofer30e62c02008-04-30 09:16:33 +0000362 if (!isa<FrameIndexSDNode>(Op)) return false;
363
364 FrameIndexSDNode * FrameIdxNode = dyn_cast<FrameIndexSDNode>(Op);
365 int FrameIdx = FrameIdxNode->getIndex();
366 return MFI->isFixedObjectIndex(FrameIdx) &&
367 MFI->getObjectOffset(FrameIdx) >= 0;
368}
369
370/// IsPossiblyOverwrittenArgumentOfTailCall - Check if the operand could
371/// possibly be overwritten when lowering the outgoing arguments in a tail
372/// call. Currently the implementation of this call is very conservative and
373/// assumes all arguments sourcing from FORMAL_ARGUMENTS or a CopyFromReg with
374/// virtual registers would be overwritten by direct lowering.
Dan Gohman475871a2008-07-27 21:46:04 +0000375static bool IsPossiblyOverwrittenArgumentOfTailCall(SDValue Op,
Arnold Schwaighofer30e62c02008-04-30 09:16:33 +0000376 MachineFrameInfo * MFI) {
377 RegisterSDNode * OpReg = NULL;
378 if (Op.getOpcode() == ISD::FORMAL_ARGUMENTS ||
379 (Op.getOpcode()== ISD::CopyFromReg &&
380 (OpReg = dyn_cast<RegisterSDNode>(Op.getOperand(1))) &&
381 (OpReg->getReg() >= TargetRegisterInfo::FirstVirtualRegister)) ||
382 (Op.getOpcode() == ISD::LOAD &&
383 IsFixedFrameObjectWithPosOffset(MFI, Op.getOperand(1))) ||
384 (Op.getOpcode() == ISD::MERGE_VALUES &&
Gabor Greif99a6cb92008-08-26 22:36:50 +0000385 Op.getOperand(Op.getResNo()).getOpcode() == ISD::LOAD &&
386 IsFixedFrameObjectWithPosOffset(MFI, Op.getOperand(Op.getResNo()).
Arnold Schwaighofer30e62c02008-04-30 09:16:33 +0000387 getOperand(1))))
388 return true;
389 return false;
390}
391
Arnold Schwaighoferc85e1712007-10-11 19:40:01 +0000392/// CheckDAGForTailCallsAndFixThem - This Function looks for CALL nodes in the
Arnold Schwaighofer48abc5c2007-10-12 21:30:57 +0000393/// DAG and fixes their tailcall attribute operand.
Arnold Schwaighoferc85e1712007-10-11 19:40:01 +0000394static void CheckDAGForTailCallsAndFixThem(SelectionDAG &DAG,
395 TargetLowering& TLI) {
396 SDNode * Ret = NULL;
Dan Gohman475871a2008-07-27 21:46:04 +0000397 SDValue Terminator = DAG.getRoot();
Arnold Schwaighoferc85e1712007-10-11 19:40:01 +0000398
399 // Find RET node.
400 if (Terminator.getOpcode() == ISD::RET) {
Gabor Greifba36cb52008-08-28 21:40:38 +0000401 Ret = Terminator.getNode();
Arnold Schwaighoferc85e1712007-10-11 19:40:01 +0000402 }
403
404 // Fix tail call attribute of CALL nodes.
405 for (SelectionDAG::allnodes_iterator BE = DAG.allnodes_begin(),
Dan Gohman0e5f1302008-07-07 23:02:41 +0000406 BI = DAG.allnodes_end(); BI != BE; ) {
407 --BI;
Dan Gohman095cc292008-09-13 01:54:27 +0000408 if (CallSDNode *TheCall = dyn_cast<CallSDNode>(BI)) {
Dan Gohman475871a2008-07-27 21:46:04 +0000409 SDValue OpRet(Ret, 0);
410 SDValue OpCall(BI, 0);
Dan Gohman095cc292008-09-13 01:54:27 +0000411 bool isMarkedTailCall = TheCall->isTailCall();
Arnold Schwaighoferc85e1712007-10-11 19:40:01 +0000412 // If CALL node has tail call attribute set to true and the call is not
413 // eligible (no RET or the target rejects) the attribute is fixed to
Arnold Schwaighofer48abc5c2007-10-12 21:30:57 +0000414 // false. The TargetLowering::IsEligibleForTailCallOptimization function
Arnold Schwaighoferc85e1712007-10-11 19:40:01 +0000415 // must correctly identify tail call optimizable calls.
Arnold Schwaighofer30e62c02008-04-30 09:16:33 +0000416 if (!isMarkedTailCall) continue;
417 if (Ret==NULL ||
Dan Gohman095cc292008-09-13 01:54:27 +0000418 !TLI.IsEligibleForTailCallOptimization(TheCall, OpRet, DAG)) {
419 // Not eligible. Mark CALL node as non tail call. Note that we
420 // can modify the call node in place since calls are not CSE'd.
421 TheCall->setNotTailCall();
Arnold Schwaighofer30e62c02008-04-30 09:16:33 +0000422 } else {
423 // Look for tail call clobbered arguments. Emit a series of
424 // copyto/copyfrom virtual register nodes to protect them.
Dan Gohman475871a2008-07-27 21:46:04 +0000425 SmallVector<SDValue, 32> Ops;
Dan Gohman095cc292008-09-13 01:54:27 +0000426 SDValue Chain = TheCall->getChain(), InFlag;
427 Ops.push_back(Chain);
428 Ops.push_back(TheCall->getCallee());
429 for (unsigned i = 0, e = TheCall->getNumArgs(); i != e; ++i) {
430 SDValue Arg = TheCall->getArg(i);
431 bool isByVal = TheCall->getArgFlags(i).isByVal();
432 MachineFunction &MF = DAG.getMachineFunction();
433 MachineFrameInfo *MFI = MF.getFrameInfo();
434 if (!isByVal &&
435 IsPossiblyOverwrittenArgumentOfTailCall(Arg, MFI)) {
436 MVT VT = Arg.getValueType();
437 unsigned VReg = MF.getRegInfo().
438 createVirtualRegister(TLI.getRegClassFor(VT));
439 Chain = DAG.getCopyToReg(Chain, VReg, Arg, InFlag);
440 InFlag = Chain.getValue(1);
441 Arg = DAG.getCopyFromReg(Chain, VReg, VT, InFlag);
442 Chain = Arg.getValue(1);
443 InFlag = Arg.getValue(2);
Arnold Schwaighofer30e62c02008-04-30 09:16:33 +0000444 }
445 Ops.push_back(Arg);
Dan Gohman095cc292008-09-13 01:54:27 +0000446 Ops.push_back(TheCall->getArgFlagsVal(i));
Arnold Schwaighofer30e62c02008-04-30 09:16:33 +0000447 }
448 // Link in chain of CopyTo/CopyFromReg.
449 Ops[0] = Chain;
450 DAG.UpdateNodeOperands(OpCall, Ops.begin(), Ops.size());
Arnold Schwaighoferc85e1712007-10-11 19:40:01 +0000451 }
452 }
453 }
454}
455
Dan Gohmanf350b272008-08-23 02:25:05 +0000456void SelectionDAGISel::SelectBasicBlock(BasicBlock *LLVMBB,
457 BasicBlock::iterator Begin,
Dan Gohman5edd3612008-08-28 20:28:56 +0000458 BasicBlock::iterator End) {
Dan Gohman7c3234c2008-08-27 23:52:12 +0000459 SDL->setCurrentBasicBlock(BB);
Dan Gohmanf350b272008-08-23 02:25:05 +0000460
Dan Gohmanf350b272008-08-23 02:25:05 +0000461 // Lower all of the non-terminator instructions.
462 for (BasicBlock::iterator I = Begin; I != End; ++I)
463 if (!isa<TerminatorInst>(I))
Dan Gohman7c3234c2008-08-27 23:52:12 +0000464 SDL->visit(*I);
Dan Gohmanf350b272008-08-23 02:25:05 +0000465
466 // Ensure that all instructions which are used outside of their defining
467 // blocks are available as virtual registers. Invoke is handled elsewhere.
468 for (BasicBlock::iterator I = Begin; I != End; ++I)
469 if (!I->use_empty() && !isa<PHINode>(I) && !isa<InvokeInst>(I)) {
Dan Gohman7c3234c2008-08-27 23:52:12 +0000470 DenseMap<const Value*,unsigned>::iterator VMI =FuncInfo->ValueMap.find(I);
471 if (VMI != FuncInfo->ValueMap.end())
472 SDL->CopyValueToVirtualRegister(I, VMI->second);
Dan Gohmanf350b272008-08-23 02:25:05 +0000473 }
474
475 // Handle PHI nodes in successor blocks.
Dan Gohman3df24e62008-09-03 23:12:08 +0000476 if (End == LLVMBB->end()) {
Dan Gohman7c3234c2008-08-27 23:52:12 +0000477 HandlePHINodesInSuccessorBlocks(LLVMBB);
Dan Gohman3df24e62008-09-03 23:12:08 +0000478
479 // Lower the terminator after the copies are emitted.
480 SDL->visit(*LLVMBB->getTerminator());
481 }
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000482
Chris Lattnera651cf62005-01-17 19:43:36 +0000483 // Make sure the root of the DAG is up-to-date.
Dan Gohman7c3234c2008-08-27 23:52:12 +0000484 CurDAG->setRoot(SDL->getControlRoot());
Arnold Schwaighoferc85e1712007-10-11 19:40:01 +0000485
486 // Check whether calls in this block are real tail calls. Fix up CALL nodes
487 // with correct tailcall attribute so that the target can rely on the tailcall
488 // attribute indicating whether the call is really eligible for tail call
489 // optimization.
Dan Gohman1937e2f2008-09-16 01:42:28 +0000490 if (PerformTailCallOpt)
491 CheckDAGForTailCallsAndFixThem(*CurDAG, TLI);
Dan Gohmanf350b272008-08-23 02:25:05 +0000492
493 // Final step, emit the lowered DAG as machine code.
494 CodeGenAndEmitDAG();
Dan Gohman7c3234c2008-08-27 23:52:12 +0000495 SDL->clear();
Chris Lattner1c08c712005-01-07 07:47:53 +0000496}
497
Dan Gohmanf350b272008-08-23 02:25:05 +0000498void SelectionDAGISel::ComputeLiveOutVRegInfo() {
Chris Lattneread0d882008-06-17 06:09:18 +0000499 SmallPtrSet<SDNode*, 128> VisitedNodes;
500 SmallVector<SDNode*, 128> Worklist;
501
Gabor Greifba36cb52008-08-28 21:40:38 +0000502 Worklist.push_back(CurDAG->getRoot().getNode());
Chris Lattneread0d882008-06-17 06:09:18 +0000503
504 APInt Mask;
505 APInt KnownZero;
506 APInt KnownOne;
507
508 while (!Worklist.empty()) {
509 SDNode *N = Worklist.back();
510 Worklist.pop_back();
511
512 // If we've already seen this node, ignore it.
513 if (!VisitedNodes.insert(N))
514 continue;
515
516 // Otherwise, add all chain operands to the worklist.
517 for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i)
518 if (N->getOperand(i).getValueType() == MVT::Other)
Gabor Greifba36cb52008-08-28 21:40:38 +0000519 Worklist.push_back(N->getOperand(i).getNode());
Chris Lattneread0d882008-06-17 06:09:18 +0000520
521 // If this is a CopyToReg with a vreg dest, process it.
522 if (N->getOpcode() != ISD::CopyToReg)
523 continue;
524
525 unsigned DestReg = cast<RegisterSDNode>(N->getOperand(1))->getReg();
526 if (!TargetRegisterInfo::isVirtualRegister(DestReg))
527 continue;
528
529 // Ignore non-scalar or non-integer values.
Dan Gohman475871a2008-07-27 21:46:04 +0000530 SDValue Src = N->getOperand(2);
Chris Lattneread0d882008-06-17 06:09:18 +0000531 MVT SrcVT = Src.getValueType();
532 if (!SrcVT.isInteger() || SrcVT.isVector())
533 continue;
534
Dan Gohmanf350b272008-08-23 02:25:05 +0000535 unsigned NumSignBits = CurDAG->ComputeNumSignBits(Src);
Chris Lattneread0d882008-06-17 06:09:18 +0000536 Mask = APInt::getAllOnesValue(SrcVT.getSizeInBits());
Dan Gohmanf350b272008-08-23 02:25:05 +0000537 CurDAG->ComputeMaskedBits(Src, Mask, KnownZero, KnownOne);
Chris Lattneread0d882008-06-17 06:09:18 +0000538
539 // Only install this information if it tells us something.
540 if (NumSignBits != 1 || KnownZero != 0 || KnownOne != 0) {
541 DestReg -= TargetRegisterInfo::FirstVirtualRegister;
Dan Gohmanf350b272008-08-23 02:25:05 +0000542 FunctionLoweringInfo &FLI = CurDAG->getFunctionLoweringInfo();
Chris Lattneread0d882008-06-17 06:09:18 +0000543 if (DestReg >= FLI.LiveOutRegInfo.size())
544 FLI.LiveOutRegInfo.resize(DestReg+1);
545 FunctionLoweringInfo::LiveOutInfo &LOI = FLI.LiveOutRegInfo[DestReg];
546 LOI.NumSignBits = NumSignBits;
547 LOI.KnownOne = NumSignBits;
548 LOI.KnownZero = NumSignBits;
549 }
550 }
551}
552
Dan Gohmanf350b272008-08-23 02:25:05 +0000553void SelectionDAGISel::CodeGenAndEmitDAG() {
Dan Gohman462dc7f2008-07-21 20:00:07 +0000554 std::string GroupName;
555 if (TimePassesIsEnabled)
556 GroupName = "Instruction Selection and Scheduling";
557 std::string BlockName;
558 if (ViewDAGCombine1 || ViewLegalizeTypesDAGs || ViewLegalizeDAGs ||
559 ViewDAGCombine2 || ViewISelDAGs || ViewSchedDAGs || ViewSUnitDAGs)
Dan Gohmanf350b272008-08-23 02:25:05 +0000560 BlockName = CurDAG->getMachineFunction().getFunction()->getName() + ':' +
Dan Gohman462dc7f2008-07-21 20:00:07 +0000561 BB->getBasicBlock()->getName();
562
563 DOUT << "Initial selection DAG:\n";
Dan Gohmanf350b272008-08-23 02:25:05 +0000564 DEBUG(CurDAG->dump());
Dan Gohman462dc7f2008-07-21 20:00:07 +0000565
Dan Gohmanf350b272008-08-23 02:25:05 +0000566 if (ViewDAGCombine1) CurDAG->viewGraph("dag-combine1 input for " + BlockName);
Dan Gohman417e11b2007-10-08 15:12:17 +0000567
Chris Lattneraf21d552005-10-10 16:47:10 +0000568 // Run the DAG combiner in pre-legalize mode.
Evan Chengebffb662008-07-01 17:59:20 +0000569 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000570 NamedRegionTimer T("DAG Combining 1", GroupName);
Dan Gohmanf350b272008-08-23 02:25:05 +0000571 CurDAG->Combine(false, *AA, Fast);
Evan Chengebffb662008-07-01 17:59:20 +0000572 } else {
Dan Gohmanf350b272008-08-23 02:25:05 +0000573 CurDAG->Combine(false, *AA, Fast);
Evan Chengebffb662008-07-01 17:59:20 +0000574 }
Nate Begeman2300f552005-09-07 00:15:36 +0000575
Dan Gohman417e11b2007-10-08 15:12:17 +0000576 DOUT << "Optimized lowered selection DAG:\n";
Dan Gohmanf350b272008-08-23 02:25:05 +0000577 DEBUG(CurDAG->dump());
Duncan Sandsf00e74f2008-07-17 17:06:03 +0000578
Chris Lattner1c08c712005-01-07 07:47:53 +0000579 // Second step, hack on the DAG until it only uses operations and types that
580 // the target supports.
Duncan Sands7cb07872008-10-27 08:42:46 +0000581 if (!DisableLegalizeTypes) {
Dan Gohmanf350b272008-08-23 02:25:05 +0000582 if (ViewLegalizeTypesDAGs) CurDAG->viewGraph("legalize-types input for " +
583 BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000584
585 if (TimePassesIsEnabled) {
586 NamedRegionTimer T("Type Legalization", GroupName);
Dan Gohmanf350b272008-08-23 02:25:05 +0000587 CurDAG->LegalizeTypes();
Dan Gohman462dc7f2008-07-21 20:00:07 +0000588 } else {
Dan Gohmanf350b272008-08-23 02:25:05 +0000589 CurDAG->LegalizeTypes();
Dan Gohman462dc7f2008-07-21 20:00:07 +0000590 }
591
592 DOUT << "Type-legalized selection DAG:\n";
Dan Gohmanf350b272008-08-23 02:25:05 +0000593 DEBUG(CurDAG->dump());
Dan Gohman462dc7f2008-07-21 20:00:07 +0000594
Chris Lattner70587ea2008-07-10 23:37:50 +0000595 // TODO: enable a dag combine pass here.
596 }
Duncan Sandsf00e74f2008-07-17 17:06:03 +0000597
Dan Gohmanf350b272008-08-23 02:25:05 +0000598 if (ViewLegalizeDAGs) CurDAG->viewGraph("legalize input for " + BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000599
Evan Chengebffb662008-07-01 17:59:20 +0000600 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000601 NamedRegionTimer T("DAG Legalization", GroupName);
Dan Gohmanf350b272008-08-23 02:25:05 +0000602 CurDAG->Legalize();
Evan Chengebffb662008-07-01 17:59:20 +0000603 } else {
Dan Gohmanf350b272008-08-23 02:25:05 +0000604 CurDAG->Legalize();
Evan Chengebffb662008-07-01 17:59:20 +0000605 }
Nate Begemanf15485a2006-03-27 01:32:24 +0000606
Bill Wendling832171c2006-12-07 20:04:42 +0000607 DOUT << "Legalized selection DAG:\n";
Dan Gohmanf350b272008-08-23 02:25:05 +0000608 DEBUG(CurDAG->dump());
Nate Begemanf15485a2006-03-27 01:32:24 +0000609
Dan Gohmanf350b272008-08-23 02:25:05 +0000610 if (ViewDAGCombine2) CurDAG->viewGraph("dag-combine2 input for " + BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000611
Chris Lattneraf21d552005-10-10 16:47:10 +0000612 // Run the DAG combiner in post-legalize mode.
Evan Chengebffb662008-07-01 17:59:20 +0000613 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000614 NamedRegionTimer T("DAG Combining 2", GroupName);
Dan Gohmanf350b272008-08-23 02:25:05 +0000615 CurDAG->Combine(true, *AA, Fast);
Evan Chengebffb662008-07-01 17:59:20 +0000616 } else {
Dan Gohmanf350b272008-08-23 02:25:05 +0000617 CurDAG->Combine(true, *AA, Fast);
Evan Chengebffb662008-07-01 17:59:20 +0000618 }
Nate Begeman2300f552005-09-07 00:15:36 +0000619
Dan Gohman417e11b2007-10-08 15:12:17 +0000620 DOUT << "Optimized legalized selection DAG:\n";
Dan Gohmanf350b272008-08-23 02:25:05 +0000621 DEBUG(CurDAG->dump());
Dan Gohman417e11b2007-10-08 15:12:17 +0000622
Dan Gohmanf350b272008-08-23 02:25:05 +0000623 if (ViewISelDAGs) CurDAG->viewGraph("isel input for " + BlockName);
Chris Lattneread0d882008-06-17 06:09:18 +0000624
Dan Gohman925a7e82008-08-13 19:47:40 +0000625 if (!Fast && EnableValueProp)
Dan Gohmanf350b272008-08-23 02:25:05 +0000626 ComputeLiveOutVRegInfo();
Evan Cheng552c4a82006-04-28 02:09:19 +0000627
Chris Lattnera33ef482005-03-30 01:10:47 +0000628 // Third, instruction select all of the operations to machine code, adding the
629 // code to the MachineBasicBlock.
Evan Chengebffb662008-07-01 17:59:20 +0000630 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000631 NamedRegionTimer T("Instruction Selection", GroupName);
Dan Gohmanf350b272008-08-23 02:25:05 +0000632 InstructionSelect();
Evan Chengebffb662008-07-01 17:59:20 +0000633 } else {
Dan Gohmanf350b272008-08-23 02:25:05 +0000634 InstructionSelect();
Evan Chengebffb662008-07-01 17:59:20 +0000635 }
Evan Chengdb8d56b2008-06-30 20:45:06 +0000636
Dan Gohman462dc7f2008-07-21 20:00:07 +0000637 DOUT << "Selected selection DAG:\n";
Dan Gohmanf350b272008-08-23 02:25:05 +0000638 DEBUG(CurDAG->dump());
Dan Gohman462dc7f2008-07-21 20:00:07 +0000639
Dan Gohmanf350b272008-08-23 02:25:05 +0000640 if (ViewSchedDAGs) CurDAG->viewGraph("scheduler input for " + BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000641
Dan Gohman5e843682008-07-14 18:19:29 +0000642 // Schedule machine code.
643 ScheduleDAG *Scheduler;
644 if (TimePassesIsEnabled) {
645 NamedRegionTimer T("Instruction Scheduling", GroupName);
Dan Gohmanf350b272008-08-23 02:25:05 +0000646 Scheduler = Schedule();
Dan Gohman5e843682008-07-14 18:19:29 +0000647 } else {
Dan Gohmanf350b272008-08-23 02:25:05 +0000648 Scheduler = Schedule();
Dan Gohman5e843682008-07-14 18:19:29 +0000649 }
650
Dan Gohman462dc7f2008-07-21 20:00:07 +0000651 if (ViewSUnitDAGs) Scheduler->viewGraph();
652
Evan Chengdb8d56b2008-06-30 20:45:06 +0000653 // Emit machine code to BB. This can change 'BB' to the last block being
654 // inserted into.
Evan Chengebffb662008-07-01 17:59:20 +0000655 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000656 NamedRegionTimer T("Instruction Creation", GroupName);
657 BB = Scheduler->EmitSchedule();
Evan Chengebffb662008-07-01 17:59:20 +0000658 } else {
Dan Gohman5e843682008-07-14 18:19:29 +0000659 BB = Scheduler->EmitSchedule();
660 }
661
662 // Free the scheduler state.
663 if (TimePassesIsEnabled) {
664 NamedRegionTimer T("Instruction Scheduling Cleanup", GroupName);
665 delete Scheduler;
666 } else {
667 delete Scheduler;
Evan Chengebffb662008-07-01 17:59:20 +0000668 }
Evan Chengdb8d56b2008-06-30 20:45:06 +0000669
Bill Wendling832171c2006-12-07 20:04:42 +0000670 DOUT << "Selected machine code:\n";
Chris Lattner1c08c712005-01-07 07:47:53 +0000671 DEBUG(BB->dump());
Nate Begemanf15485a2006-03-27 01:32:24 +0000672}
Chris Lattner1c08c712005-01-07 07:47:53 +0000673
Dan Gohmand57dd5f2008-09-23 21:53:34 +0000674void SelectionDAGISel::SelectAllBasicBlocks(Function &Fn, MachineFunction &MF,
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000675 MachineModuleInfo *MMI,
676 const TargetInstrInfo &TII) {
Dan Gohmana43abd12008-09-29 21:55:50 +0000677 // Initialize the Fast-ISel state, if needed.
678 FastISel *FastIS = 0;
679 if (EnableFastISel)
680 FastIS = TLI.createFastISel(*FuncInfo->MF, MMI,
681 FuncInfo->ValueMap,
682 FuncInfo->MBBMap,
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000683 FuncInfo->StaticAllocaMap
684#ifndef NDEBUG
685 , FuncInfo->CatchInfoLost
686#endif
687 );
Dan Gohmana43abd12008-09-29 21:55:50 +0000688
689 // Iterate over all basic blocks in the function.
Evan Cheng39fd6e82008-08-07 00:43:25 +0000690 for (Function::iterator I = Fn.begin(), E = Fn.end(); I != E; ++I) {
691 BasicBlock *LLVMBB = &*I;
Dan Gohman7c3234c2008-08-27 23:52:12 +0000692 BB = FuncInfo->MBBMap[LLVMBB];
Dan Gohmanf350b272008-08-23 02:25:05 +0000693
Dan Gohman3df24e62008-09-03 23:12:08 +0000694 BasicBlock::iterator const Begin = LLVMBB->begin();
695 BasicBlock::iterator const End = LLVMBB->end();
Evan Cheng9f118502008-09-08 16:01:27 +0000696 BasicBlock::iterator BI = Begin;
Dan Gohman5edd3612008-08-28 20:28:56 +0000697
698 // Lower any arguments needed in this block if this is the entry block.
Dan Gohman33134c42008-09-25 17:05:24 +0000699 bool SuppressFastISel = false;
700 if (LLVMBB == &Fn.getEntryBlock()) {
Dan Gohman5edd3612008-08-28 20:28:56 +0000701 LowerArguments(LLVMBB);
Dan Gohmanf350b272008-08-23 02:25:05 +0000702
Dan Gohman33134c42008-09-25 17:05:24 +0000703 // If any of the arguments has the byval attribute, forgo
704 // fast-isel in the entry block.
Dan Gohmana43abd12008-09-29 21:55:50 +0000705 if (FastIS) {
Dan Gohman33134c42008-09-25 17:05:24 +0000706 unsigned j = 1;
707 for (Function::arg_iterator I = Fn.arg_begin(), E = Fn.arg_end();
708 I != E; ++I, ++j)
Devang Patel05988662008-09-25 21:00:45 +0000709 if (Fn.paramHasAttr(j, Attribute::ByVal)) {
Dan Gohman77ca41e2008-09-25 17:21:42 +0000710 if (EnableFastISelVerbose || EnableFastISelAbort)
711 cerr << "FastISel skips entry block due to byval argument\n";
Dan Gohman33134c42008-09-25 17:05:24 +0000712 SuppressFastISel = true;
713 break;
714 }
715 }
716 }
717
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000718 if (MMI && BB->isLandingPad()) {
719 // Add a label to mark the beginning of the landing pad. Deletion of the
720 // landing pad can thus be detected via the MachineModuleInfo.
721 unsigned LabelID = MMI->addLandingPad(BB);
722
723 const TargetInstrDesc &II = TII.get(TargetInstrInfo::EH_LABEL);
724 BuildMI(BB, II).addImm(LabelID);
725
726 // Mark exception register as live in.
727 unsigned Reg = TLI.getExceptionAddressRegister();
728 if (Reg) BB->addLiveIn(Reg);
729
730 // Mark exception selector register as live in.
731 Reg = TLI.getExceptionSelectorRegister();
732 if (Reg) BB->addLiveIn(Reg);
733
734 // FIXME: Hack around an exception handling flaw (PR1508): the personality
735 // function and list of typeids logically belong to the invoke (or, if you
736 // like, the basic block containing the invoke), and need to be associated
737 // with it in the dwarf exception handling tables. Currently however the
738 // information is provided by an intrinsic (eh.selector) that can be moved
739 // to unexpected places by the optimizers: if the unwind edge is critical,
740 // then breaking it can result in the intrinsics being in the successor of
741 // the landing pad, not the landing pad itself. This results in exceptions
742 // not being caught because no typeids are associated with the invoke.
743 // This may not be the only way things can go wrong, but it is the only way
744 // we try to work around for the moment.
745 BranchInst *Br = dyn_cast<BranchInst>(LLVMBB->getTerminator());
746
747 if (Br && Br->isUnconditional()) { // Critical edge?
748 BasicBlock::iterator I, E;
749 for (I = LLVMBB->begin(), E = --LLVMBB->end(); I != E; ++I)
750 if (isa<EHSelectorInst>(I))
751 break;
752
753 if (I == E)
754 // No catch info found - try to extract some from the successor.
755 copyCatchInfo(Br->getSuccessor(0), LLVMBB, MMI, *FuncInfo);
756 }
757 }
758
Dan Gohmanf350b272008-08-23 02:25:05 +0000759 // Before doing SelectionDAG ISel, see if FastISel has been requested.
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000760 if (FastIS && !SuppressFastISel) {
Dan Gohmana43abd12008-09-29 21:55:50 +0000761 // Emit code for any incoming arguments. This must happen before
762 // beginning FastISel on the entry block.
763 if (LLVMBB == &Fn.getEntryBlock()) {
764 CurDAG->setRoot(SDL->getControlRoot());
765 CodeGenAndEmitDAG();
766 SDL->clear();
767 }
Dan Gohman241f4642008-10-04 00:56:36 +0000768 FastIS->startNewBlock(BB);
Dan Gohmana43abd12008-09-29 21:55:50 +0000769 // Do FastISel on as many instructions as possible.
770 for (; BI != End; ++BI) {
771 // Just before the terminator instruction, insert instructions to
772 // feed PHI nodes in successor blocks.
773 if (isa<TerminatorInst>(BI))
774 if (!HandlePHINodesInSuccessorBlocksFast(LLVMBB, FastIS)) {
Dan Gohman4344a5d2008-09-09 23:05:00 +0000775 if (EnableFastISelVerbose || EnableFastISelAbort) {
Dan Gohman293d5f82008-09-09 22:06:46 +0000776 cerr << "FastISel miss: ";
777 BI->dump();
778 }
Dan Gohman4344a5d2008-09-09 23:05:00 +0000779 if (EnableFastISelAbort)
Dan Gohmana43abd12008-09-29 21:55:50 +0000780 assert(0 && "FastISel didn't handle a PHI in a successor");
781 break;
Dan Gohmanf350b272008-08-23 02:25:05 +0000782 }
Dan Gohmana43abd12008-09-29 21:55:50 +0000783
784 // First try normal tablegen-generated "fast" selection.
785 if (FastIS->SelectInstruction(BI))
786 continue;
787
788 // Next, try calling the target to attempt to handle the instruction.
789 if (FastIS->TargetSelectInstruction(BI))
790 continue;
791
792 // Then handle certain instructions as single-LLVM-Instruction blocks.
793 if (isa<CallInst>(BI)) {
794 if (EnableFastISelVerbose || EnableFastISelAbort) {
795 cerr << "FastISel missed call: ";
796 BI->dump();
797 }
798
799 if (BI->getType() != Type::VoidTy) {
800 unsigned &R = FuncInfo->ValueMap[BI];
801 if (!R)
802 R = FuncInfo->CreateRegForValue(BI);
803 }
804
805 SelectBasicBlock(LLVMBB, BI, next(BI));
Dan Gohman241f4642008-10-04 00:56:36 +0000806 // If the instruction was codegen'd with multiple blocks,
807 // inform the FastISel object where to resume inserting.
808 FastIS->setCurrentBlock(BB);
Dan Gohmana43abd12008-09-29 21:55:50 +0000809 continue;
Dan Gohmanf350b272008-08-23 02:25:05 +0000810 }
Dan Gohmana43abd12008-09-29 21:55:50 +0000811
812 // Otherwise, give up on FastISel for the rest of the block.
813 // For now, be a little lenient about non-branch terminators.
814 if (!isa<TerminatorInst>(BI) || isa<BranchInst>(BI)) {
815 if (EnableFastISelVerbose || EnableFastISelAbort) {
816 cerr << "FastISel miss: ";
817 BI->dump();
818 }
819 if (EnableFastISelAbort)
820 // The "fast" selector couldn't handle something and bailed.
821 // For the purpose of debugging, just abort.
822 assert(0 && "FastISel didn't select the entire block");
823 }
824 break;
Dan Gohmanf350b272008-08-23 02:25:05 +0000825 }
826 }
827
Dan Gohmand2ff6472008-09-02 20:17:56 +0000828 // Run SelectionDAG instruction selection on the remainder of the block
829 // not handled by FastISel. If FastISel is not run, this is the entire
Dan Gohman3df24e62008-09-03 23:12:08 +0000830 // block.
Evan Cheng9f118502008-09-08 16:01:27 +0000831 if (BI != End)
832 SelectBasicBlock(LLVMBB, BI, End);
Dan Gohmanf350b272008-08-23 02:25:05 +0000833
Dan Gohman7c3234c2008-08-27 23:52:12 +0000834 FinishBasicBlock();
Evan Cheng39fd6e82008-08-07 00:43:25 +0000835 }
Dan Gohmana43abd12008-09-29 21:55:50 +0000836
837 delete FastIS;
Dan Gohman0e5f1302008-07-07 23:02:41 +0000838}
839
Dan Gohmanfed90b62008-07-28 21:51:04 +0000840void
Dan Gohman7c3234c2008-08-27 23:52:12 +0000841SelectionDAGISel::FinishBasicBlock() {
Dan Gohmanf350b272008-08-23 02:25:05 +0000842
Dan Gohmanf350b272008-08-23 02:25:05 +0000843 DOUT << "Target-post-processed machine code:\n";
844 DEBUG(BB->dump());
Nate Begemanf15485a2006-03-27 01:32:24 +0000845
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000846 DOUT << "Total amount of phi nodes to update: "
Dan Gohman7c3234c2008-08-27 23:52:12 +0000847 << SDL->PHINodesToUpdate.size() << "\n";
848 DEBUG(for (unsigned i = 0, e = SDL->PHINodesToUpdate.size(); i != e; ++i)
849 DOUT << "Node " << i << " : (" << SDL->PHINodesToUpdate[i].first
850 << ", " << SDL->PHINodesToUpdate[i].second << ")\n";);
Nate Begemanf15485a2006-03-27 01:32:24 +0000851
Chris Lattnera33ef482005-03-30 01:10:47 +0000852 // Next, now that we know what the last MBB the LLVM BB expanded is, update
Chris Lattner1c08c712005-01-07 07:47:53 +0000853 // PHI nodes in successors.
Dan Gohman7c3234c2008-08-27 23:52:12 +0000854 if (SDL->SwitchCases.empty() &&
855 SDL->JTCases.empty() &&
856 SDL->BitTestCases.empty()) {
857 for (unsigned i = 0, e = SDL->PHINodesToUpdate.size(); i != e; ++i) {
858 MachineInstr *PHI = SDL->PHINodesToUpdate[i].first;
Nate Begemanf15485a2006-03-27 01:32:24 +0000859 assert(PHI->getOpcode() == TargetInstrInfo::PHI &&
860 "This is not a machine PHI node that we are updating!");
Dan Gohman7c3234c2008-08-27 23:52:12 +0000861 PHI->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[i].second,
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000862 false));
863 PHI->addOperand(MachineOperand::CreateMBB(BB));
Nate Begemanf15485a2006-03-27 01:32:24 +0000864 }
Dan Gohman7c3234c2008-08-27 23:52:12 +0000865 SDL->PHINodesToUpdate.clear();
Nate Begemanf15485a2006-03-27 01:32:24 +0000866 return;
Chris Lattner1c08c712005-01-07 07:47:53 +0000867 }
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000868
Dan Gohman7c3234c2008-08-27 23:52:12 +0000869 for (unsigned i = 0, e = SDL->BitTestCases.size(); i != e; ++i) {
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000870 // Lower header first, if it wasn't already lowered
Dan Gohman7c3234c2008-08-27 23:52:12 +0000871 if (!SDL->BitTestCases[i].Emitted) {
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000872 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman7c3234c2008-08-27 23:52:12 +0000873 BB = SDL->BitTestCases[i].Parent;
874 SDL->setCurrentBasicBlock(BB);
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000875 // Emit the code
Dan Gohman7c3234c2008-08-27 23:52:12 +0000876 SDL->visitBitTestHeader(SDL->BitTestCases[i]);
877 CurDAG->setRoot(SDL->getRoot());
Dan Gohmanf350b272008-08-23 02:25:05 +0000878 CodeGenAndEmitDAG();
Dan Gohman7c3234c2008-08-27 23:52:12 +0000879 SDL->clear();
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000880 }
881
Dan Gohman7c3234c2008-08-27 23:52:12 +0000882 for (unsigned j = 0, ej = SDL->BitTestCases[i].Cases.size(); j != ej; ++j) {
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000883 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman7c3234c2008-08-27 23:52:12 +0000884 BB = SDL->BitTestCases[i].Cases[j].ThisBB;
885 SDL->setCurrentBasicBlock(BB);
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000886 // Emit the code
887 if (j+1 != ej)
Dan Gohman7c3234c2008-08-27 23:52:12 +0000888 SDL->visitBitTestCase(SDL->BitTestCases[i].Cases[j+1].ThisBB,
889 SDL->BitTestCases[i].Reg,
890 SDL->BitTestCases[i].Cases[j]);
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000891 else
Dan Gohman7c3234c2008-08-27 23:52:12 +0000892 SDL->visitBitTestCase(SDL->BitTestCases[i].Default,
893 SDL->BitTestCases[i].Reg,
894 SDL->BitTestCases[i].Cases[j]);
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000895
896
Dan Gohman7c3234c2008-08-27 23:52:12 +0000897 CurDAG->setRoot(SDL->getRoot());
Dan Gohmanf350b272008-08-23 02:25:05 +0000898 CodeGenAndEmitDAG();
Dan Gohman7c3234c2008-08-27 23:52:12 +0000899 SDL->clear();
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000900 }
901
902 // Update PHI Nodes
Dan Gohman7c3234c2008-08-27 23:52:12 +0000903 for (unsigned pi = 0, pe = SDL->PHINodesToUpdate.size(); pi != pe; ++pi) {
904 MachineInstr *PHI = SDL->PHINodesToUpdate[pi].first;
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000905 MachineBasicBlock *PHIBB = PHI->getParent();
906 assert(PHI->getOpcode() == TargetInstrInfo::PHI &&
907 "This is not a machine PHI node that we are updating!");
908 // This is "default" BB. We have two jumps to it. From "header" BB and
909 // from last "case" BB.
Dan Gohman7c3234c2008-08-27 23:52:12 +0000910 if (PHIBB == SDL->BitTestCases[i].Default) {
911 PHI->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[pi].second,
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000912 false));
Dan Gohman7c3234c2008-08-27 23:52:12 +0000913 PHI->addOperand(MachineOperand::CreateMBB(SDL->BitTestCases[i].Parent));
914 PHI->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[pi].second,
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000915 false));
Dan Gohman7c3234c2008-08-27 23:52:12 +0000916 PHI->addOperand(MachineOperand::CreateMBB(SDL->BitTestCases[i].Cases.
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000917 back().ThisBB));
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000918 }
919 // One of "cases" BB.
Dan Gohman7c3234c2008-08-27 23:52:12 +0000920 for (unsigned j = 0, ej = SDL->BitTestCases[i].Cases.size();
921 j != ej; ++j) {
922 MachineBasicBlock* cBB = SDL->BitTestCases[i].Cases[j].ThisBB;
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000923 if (cBB->succ_end() !=
924 std::find(cBB->succ_begin(),cBB->succ_end(), PHIBB)) {
Dan Gohman7c3234c2008-08-27 23:52:12 +0000925 PHI->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[pi].second,
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000926 false));
927 PHI->addOperand(MachineOperand::CreateMBB(cBB));
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000928 }
929 }
930 }
931 }
Dan Gohman7c3234c2008-08-27 23:52:12 +0000932 SDL->BitTestCases.clear();
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000933
Nate Begeman9453eea2006-04-23 06:26:20 +0000934 // If the JumpTable record is filled in, then we need to emit a jump table.
935 // Updating the PHI nodes is tricky in this case, since we need to determine
936 // whether the PHI is a successor of the range check MBB or the jump table MBB
Dan Gohman7c3234c2008-08-27 23:52:12 +0000937 for (unsigned i = 0, e = SDL->JTCases.size(); i != e; ++i) {
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +0000938 // Lower header first, if it wasn't already lowered
Dan Gohman7c3234c2008-08-27 23:52:12 +0000939 if (!SDL->JTCases[i].first.Emitted) {
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +0000940 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman7c3234c2008-08-27 23:52:12 +0000941 BB = SDL->JTCases[i].first.HeaderBB;
942 SDL->setCurrentBasicBlock(BB);
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +0000943 // Emit the code
Dan Gohman7c3234c2008-08-27 23:52:12 +0000944 SDL->visitJumpTableHeader(SDL->JTCases[i].second, SDL->JTCases[i].first);
945 CurDAG->setRoot(SDL->getRoot());
Dan Gohmanf350b272008-08-23 02:25:05 +0000946 CodeGenAndEmitDAG();
Dan Gohman7c3234c2008-08-27 23:52:12 +0000947 SDL->clear();
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000948 }
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +0000949
Nate Begeman37efe672006-04-22 18:53:45 +0000950 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman7c3234c2008-08-27 23:52:12 +0000951 BB = SDL->JTCases[i].second.MBB;
952 SDL->setCurrentBasicBlock(BB);
Nate Begeman37efe672006-04-22 18:53:45 +0000953 // Emit the code
Dan Gohman7c3234c2008-08-27 23:52:12 +0000954 SDL->visitJumpTable(SDL->JTCases[i].second);
955 CurDAG->setRoot(SDL->getRoot());
Dan Gohmanf350b272008-08-23 02:25:05 +0000956 CodeGenAndEmitDAG();
Dan Gohman7c3234c2008-08-27 23:52:12 +0000957 SDL->clear();
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +0000958
Nate Begeman37efe672006-04-22 18:53:45 +0000959 // Update PHI Nodes
Dan Gohman7c3234c2008-08-27 23:52:12 +0000960 for (unsigned pi = 0, pe = SDL->PHINodesToUpdate.size(); pi != pe; ++pi) {
961 MachineInstr *PHI = SDL->PHINodesToUpdate[pi].first;
Nate Begeman37efe672006-04-22 18:53:45 +0000962 MachineBasicBlock *PHIBB = PHI->getParent();
963 assert(PHI->getOpcode() == TargetInstrInfo::PHI &&
964 "This is not a machine PHI node that we are updating!");
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000965 // "default" BB. We can go there only from header BB.
Dan Gohman7c3234c2008-08-27 23:52:12 +0000966 if (PHIBB == SDL->JTCases[i].second.Default) {
967 PHI->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[pi].second,
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000968 false));
Dan Gohman7c3234c2008-08-27 23:52:12 +0000969 PHI->addOperand(MachineOperand::CreateMBB(SDL->JTCases[i].first.HeaderBB));
Nate Begemanf4360a42006-05-03 03:48:02 +0000970 }
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000971 // JT BB. Just iterate over successors here
Nate Begemanf4360a42006-05-03 03:48:02 +0000972 if (BB->succ_end() != std::find(BB->succ_begin(),BB->succ_end(), PHIBB)) {
Dan Gohman7c3234c2008-08-27 23:52:12 +0000973 PHI->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[pi].second,
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000974 false));
975 PHI->addOperand(MachineOperand::CreateMBB(BB));
Nate Begeman37efe672006-04-22 18:53:45 +0000976 }
977 }
Nate Begeman37efe672006-04-22 18:53:45 +0000978 }
Dan Gohman7c3234c2008-08-27 23:52:12 +0000979 SDL->JTCases.clear();
Nate Begeman37efe672006-04-22 18:53:45 +0000980
Chris Lattnerb2e806e2006-10-22 23:00:53 +0000981 // If the switch block involved a branch to one of the actual successors, we
982 // need to update PHI nodes in that block.
Dan Gohman7c3234c2008-08-27 23:52:12 +0000983 for (unsigned i = 0, e = SDL->PHINodesToUpdate.size(); i != e; ++i) {
984 MachineInstr *PHI = SDL->PHINodesToUpdate[i].first;
Chris Lattnerb2e806e2006-10-22 23:00:53 +0000985 assert(PHI->getOpcode() == TargetInstrInfo::PHI &&
986 "This is not a machine PHI node that we are updating!");
987 if (BB->isSuccessor(PHI->getParent())) {
Dan Gohman7c3234c2008-08-27 23:52:12 +0000988 PHI->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[i].second,
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000989 false));
990 PHI->addOperand(MachineOperand::CreateMBB(BB));
Chris Lattnerb2e806e2006-10-22 23:00:53 +0000991 }
992 }
993
Nate Begemanf15485a2006-03-27 01:32:24 +0000994 // If we generated any switch lowering information, build and codegen any
995 // additional DAGs necessary.
Dan Gohman7c3234c2008-08-27 23:52:12 +0000996 for (unsigned i = 0, e = SDL->SwitchCases.size(); i != e; ++i) {
Nate Begemanf15485a2006-03-27 01:32:24 +0000997 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman7c3234c2008-08-27 23:52:12 +0000998 BB = SDL->SwitchCases[i].ThisBB;
999 SDL->setCurrentBasicBlock(BB);
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001000
Nate Begemanf15485a2006-03-27 01:32:24 +00001001 // Emit the code
Dan Gohman7c3234c2008-08-27 23:52:12 +00001002 SDL->visitSwitchCase(SDL->SwitchCases[i]);
1003 CurDAG->setRoot(SDL->getRoot());
Dan Gohmanf350b272008-08-23 02:25:05 +00001004 CodeGenAndEmitDAG();
Dan Gohman7c3234c2008-08-27 23:52:12 +00001005 SDL->clear();
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001006
1007 // Handle any PHI nodes in successors of this chunk, as if we were coming
1008 // from the original BB before switch expansion. Note that PHI nodes can
1009 // occur multiple times in PHINodesToUpdate. We have to be very careful to
1010 // handle them the right number of times.
Dan Gohman7c3234c2008-08-27 23:52:12 +00001011 while ((BB = SDL->SwitchCases[i].TrueBB)) { // Handle LHS and RHS.
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001012 for (MachineBasicBlock::iterator Phi = BB->begin();
1013 Phi != BB->end() && Phi->getOpcode() == TargetInstrInfo::PHI; ++Phi){
1014 // This value for this PHI node is recorded in PHINodesToUpdate, get it.
1015 for (unsigned pn = 0; ; ++pn) {
Dan Gohman7c3234c2008-08-27 23:52:12 +00001016 assert(pn != SDL->PHINodesToUpdate.size() &&
1017 "Didn't find PHI entry!");
1018 if (SDL->PHINodesToUpdate[pn].first == Phi) {
1019 Phi->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[pn].
Chris Lattner9ce2e9d2007-12-30 00:57:42 +00001020 second, false));
Dan Gohman7c3234c2008-08-27 23:52:12 +00001021 Phi->addOperand(MachineOperand::CreateMBB(SDL->SwitchCases[i].ThisBB));
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001022 break;
1023 }
1024 }
Nate Begemanf15485a2006-03-27 01:32:24 +00001025 }
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001026
1027 // Don't process RHS if same block as LHS.
Dan Gohman7c3234c2008-08-27 23:52:12 +00001028 if (BB == SDL->SwitchCases[i].FalseBB)
1029 SDL->SwitchCases[i].FalseBB = 0;
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001030
1031 // If we haven't handled the RHS, do so now. Otherwise, we're done.
Dan Gohman7c3234c2008-08-27 23:52:12 +00001032 SDL->SwitchCases[i].TrueBB = SDL->SwitchCases[i].FalseBB;
1033 SDL->SwitchCases[i].FalseBB = 0;
Nate Begemanf15485a2006-03-27 01:32:24 +00001034 }
Dan Gohman7c3234c2008-08-27 23:52:12 +00001035 assert(SDL->SwitchCases[i].TrueBB == 0 && SDL->SwitchCases[i].FalseBB == 0);
Chris Lattnera33ef482005-03-30 01:10:47 +00001036 }
Dan Gohman7c3234c2008-08-27 23:52:12 +00001037 SDL->SwitchCases.clear();
1038
1039 SDL->PHINodesToUpdate.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 Gohman5e843682008-07-14 18:19:29 +00001043/// Schedule - Pick a safe ordering for instructions for each
Evan Chenga9c20912006-01-21 02:32:06 +00001044/// target node in the graph.
Dan Gohman5e843682008-07-14 18:19:29 +00001045///
Dan Gohmanf350b272008-08-23 02:25:05 +00001046ScheduleDAG *SelectionDAGISel::Schedule() {
Jim Laskeyeb577ba2006-08-02 12:30:23 +00001047 RegisterScheduler::FunctionPassCtor Ctor = RegisterScheduler::getDefault();
Jim Laskey13ec7022006-08-01 14:21:23 +00001048
1049 if (!Ctor) {
Jim Laskeyeb577ba2006-08-02 12:30:23 +00001050 Ctor = ISHeuristic;
Jim Laskey9373beb2006-08-01 19:14:14 +00001051 RegisterScheduler::setDefault(Ctor);
Evan Cheng4ef10862006-01-23 07:01:07 +00001052 }
Jim Laskey13ec7022006-08-01 14:21:23 +00001053
Dan Gohman9b75b372008-11-11 17:50:47 +00001054 TargetMachine &TM = getTargetLowering().getTargetMachine();
1055 ScheduleDAG *Scheduler = Ctor(this, CurDAG, &TM, BB, Fast);
Dan Gohman5e843682008-07-14 18:19:29 +00001056 Scheduler->Run();
Dan Gohman3e1a7ae2007-08-28 20:32:58 +00001057
Dan Gohman5e843682008-07-14 18:19:29 +00001058 return Scheduler;
Evan Chenga9c20912006-01-21 02:32:06 +00001059}
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001060
Chris Lattner03fc53c2006-03-06 00:22:00 +00001061
Jim Laskey9ff542f2006-08-01 18:29:48 +00001062HazardRecognizer *SelectionDAGISel::CreateTargetHazardRecognizer() {
1063 return new HazardRecognizer();
1064}
1065
Chris Lattner75548062006-10-11 03:58:02 +00001066//===----------------------------------------------------------------------===//
1067// Helper functions used by the generated instruction selector.
1068//===----------------------------------------------------------------------===//
1069// Calls to these methods are generated by tblgen.
1070
1071/// CheckAndMask - The isel is trying to match something like (and X, 255). If
1072/// the dag combiner simplified the 255, we still want to match. RHS is the
1073/// actual value in the DAG on the RHS of an AND, and DesiredMaskS is the value
1074/// specified in the .td file (e.g. 255).
Dan Gohman475871a2008-07-27 21:46:04 +00001075bool SelectionDAGISel::CheckAndMask(SDValue LHS, ConstantSDNode *RHS,
Dan Gohmandc9b3d02007-07-24 23:00:27 +00001076 int64_t DesiredMaskS) const {
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001077 const APInt &ActualMask = RHS->getAPIntValue();
1078 const APInt &DesiredMask = APInt(LHS.getValueSizeInBits(), DesiredMaskS);
Chris Lattner75548062006-10-11 03:58:02 +00001079
1080 // If the actual mask exactly matches, success!
1081 if (ActualMask == DesiredMask)
1082 return true;
1083
1084 // If the actual AND mask is allowing unallowed bits, this doesn't match.
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001085 if (ActualMask.intersects(~DesiredMask))
Chris Lattner75548062006-10-11 03:58:02 +00001086 return false;
1087
1088 // Otherwise, the DAG Combiner may have proven that the value coming in is
1089 // either already zero or is not demanded. Check for known zero input bits.
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001090 APInt NeededMask = DesiredMask & ~ActualMask;
Dan Gohmanea859be2007-06-22 14:59:07 +00001091 if (CurDAG->MaskedValueIsZero(LHS, NeededMask))
Chris Lattner75548062006-10-11 03:58:02 +00001092 return true;
1093
1094 // TODO: check to see if missing bits are just not demanded.
1095
1096 // Otherwise, this pattern doesn't match.
1097 return false;
1098}
1099
1100/// CheckOrMask - The isel is trying to match something like (or X, 255). If
1101/// the dag combiner simplified the 255, we still want to match. RHS is the
1102/// actual value in the DAG on the RHS of an OR, and DesiredMaskS is the value
1103/// specified in the .td file (e.g. 255).
Dan Gohman475871a2008-07-27 21:46:04 +00001104bool SelectionDAGISel::CheckOrMask(SDValue LHS, ConstantSDNode *RHS,
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001105 int64_t DesiredMaskS) const {
1106 const APInt &ActualMask = RHS->getAPIntValue();
1107 const APInt &DesiredMask = APInt(LHS.getValueSizeInBits(), DesiredMaskS);
Chris Lattner75548062006-10-11 03:58:02 +00001108
1109 // If the actual mask exactly matches, success!
1110 if (ActualMask == DesiredMask)
1111 return true;
1112
1113 // If the actual AND mask is allowing unallowed bits, this doesn't match.
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001114 if (ActualMask.intersects(~DesiredMask))
Chris Lattner75548062006-10-11 03:58:02 +00001115 return false;
1116
1117 // Otherwise, the DAG Combiner may have proven that the value coming in is
1118 // either already zero or is not demanded. Check for known zero input bits.
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001119 APInt NeededMask = DesiredMask & ~ActualMask;
Chris Lattner75548062006-10-11 03:58:02 +00001120
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001121 APInt KnownZero, KnownOne;
Dan Gohmanea859be2007-06-22 14:59:07 +00001122 CurDAG->ComputeMaskedBits(LHS, NeededMask, KnownZero, KnownOne);
Chris Lattner75548062006-10-11 03:58:02 +00001123
1124 // If all the missing bits in the or are already known to be set, match!
1125 if ((NeededMask & KnownOne) == NeededMask)
1126 return true;
1127
1128 // TODO: check to see if missing bits are just not demanded.
1129
1130 // Otherwise, this pattern doesn't match.
1131 return false;
1132}
1133
Jim Laskey9ff542f2006-08-01 18:29:48 +00001134
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001135/// SelectInlineAsmMemoryOperands - Calls to this are automatically generated
1136/// by tblgen. Others should not call it.
1137void SelectionDAGISel::
Dan Gohmanf350b272008-08-23 02:25:05 +00001138SelectInlineAsmMemoryOperands(std::vector<SDValue> &Ops) {
Dan Gohman475871a2008-07-27 21:46:04 +00001139 std::vector<SDValue> InOps;
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001140 std::swap(InOps, Ops);
1141
1142 Ops.push_back(InOps[0]); // input chain.
1143 Ops.push_back(InOps[1]); // input asm string.
1144
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001145 unsigned i = 2, e = InOps.size();
1146 if (InOps[e-1].getValueType() == MVT::Flag)
1147 --e; // Don't process a flag operand if it is here.
1148
1149 while (i != e) {
Dan Gohmanf5aeb1a2008-09-12 16:56:44 +00001150 unsigned Flags = cast<ConstantSDNode>(InOps[i])->getZExtValue();
Dale Johannesen86b49f82008-09-24 01:07:17 +00001151 if ((Flags & 7) != 4 /*MEM*/) {
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001152 // Just skip over this operand, copying the operands verbatim.
1153 Ops.insert(Ops.end(), InOps.begin()+i, InOps.begin()+i+(Flags >> 3) + 1);
1154 i += (Flags >> 3) + 1;
1155 } else {
1156 assert((Flags >> 3) == 1 && "Memory operand with multiple values?");
1157 // Otherwise, this is a memory operand. Ask the target to select it.
Dan Gohman475871a2008-07-27 21:46:04 +00001158 std::vector<SDValue> SelOps;
Dan Gohmanf350b272008-08-23 02:25:05 +00001159 if (SelectInlineAsmMemoryOperand(InOps[i+1], 'm', SelOps)) {
Bill Wendling832171c2006-12-07 20:04:42 +00001160 cerr << "Could not match memory address. Inline asm failure!\n";
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001161 exit(1);
1162 }
1163
1164 // Add this to the output node.
Dan Gohmanf350b272008-08-23 02:25:05 +00001165 MVT IntPtrTy = CurDAG->getTargetLoweringInfo().getPointerTy();
Dale Johannesen86b49f82008-09-24 01:07:17 +00001166 Ops.push_back(CurDAG->getTargetConstant(4/*MEM*/ | (SelOps.size()<< 3),
Dan Gohmanf350b272008-08-23 02:25:05 +00001167 IntPtrTy));
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001168 Ops.insert(Ops.end(), SelOps.begin(), SelOps.end());
1169 i += 2;
1170 }
1171 }
1172
1173 // Add the flag input back if present.
1174 if (e != InOps.size())
1175 Ops.push_back(InOps.back());
1176}
Devang Patel794fd752007-05-01 21:15:47 +00001177
Devang Patel19974732007-05-03 01:11:54 +00001178char SelectionDAGISel::ID = 0;