blob: ba61c882df2fabf629053eaaeca8d085fffcd395 [file] [log] [blame]
Chris Lattner1c08c712005-01-07 07:47:53 +00001//===-- SelectionDAGISel.cpp - Implement the SelectionDAGISel class -------===//
Misha Brukmanedf128a2005-04-21 22:36:52 +00002//
Chris Lattner1c08c712005-01-07 07:47:53 +00003// The LLVM Compiler Infrastructure
4//
Chris Lattner4ee451d2007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Misha Brukmanedf128a2005-04-21 22:36:52 +00007//
Chris Lattner1c08c712005-01-07 07:47:53 +00008//===----------------------------------------------------------------------===//
9//
10// This implements the SelectionDAGISel class.
11//
12//===----------------------------------------------------------------------===//
13
14#define DEBUG_TYPE "isel"
Dan Gohman84fbac52009-02-06 17:22:58 +000015#include "ScheduleDAGSDNodes.h"
Dan Gohmanf0cbcd42008-09-03 16:12:24 +000016#include "SelectionDAGBuild.h"
Dan Gohman84fbac52009-02-06 17:22:58 +000017#include "llvm/CodeGen/SelectionDAGISel.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 Gohmanfc54c552009-01-15 22:18:12 +000037#include "llvm/CodeGen/ScheduleHazardRecognizer.h"
Jim Laskeyeb577ba2006-08-02 12:30:23 +000038#include "llvm/CodeGen/SchedulerRegistry.h"
Chris Lattner1c08c712005-01-07 07:47:53 +000039#include "llvm/CodeGen/SelectionDAG.h"
Devang Patel6e7a1612009-01-09 19:11:50 +000040#include "llvm/CodeGen/DwarfWriter.h"
Dan Gohman6f0d0242008-02-10 18:45:23 +000041#include "llvm/Target/TargetRegisterInfo.h"
Chris Lattner1c08c712005-01-07 07:47:53 +000042#include "llvm/Target/TargetData.h"
43#include "llvm/Target/TargetFrameInfo.h"
44#include "llvm/Target/TargetInstrInfo.h"
45#include "llvm/Target/TargetLowering.h"
46#include "llvm/Target/TargetMachine.h"
Vladimir Prus12472912006-05-23 13:43:15 +000047#include "llvm/Target/TargetOptions.h"
Chris Lattnera4f0b3a2006-08-27 12:54:02 +000048#include "llvm/Support/Compiler.h"
Evan Chengdb8d56b2008-06-30 20:45:06 +000049#include "llvm/Support/Debug.h"
50#include "llvm/Support/MathExtras.h"
51#include "llvm/Support/Timer.h"
Jeff Cohen7e881032006-02-24 02:52:40 +000052#include <algorithm>
Chris Lattner1c08c712005-01-07 07:47:53 +000053using namespace llvm;
54
Chris Lattneread0d882008-06-17 06:09:18 +000055static cl::opt<bool>
Duncan Sands7cb07872008-10-27 08:42:46 +000056DisableLegalizeTypes("disable-legalize-types", cl::Hidden);
Dan Gohman78eca172008-08-19 22:33:34 +000057static cl::opt<bool>
Dan Gohman293d5f82008-09-09 22:06:46 +000058EnableFastISelVerbose("fast-isel-verbose", cl::Hidden,
Dan Gohmand659d502008-10-20 21:30:12 +000059 cl::desc("Enable verbose messages in the \"fast\" "
Dan Gohman293d5f82008-09-09 22:06:46 +000060 "instruction selector"));
61static cl::opt<bool>
Dan Gohman4344a5d2008-09-09 23:05:00 +000062EnableFastISelAbort("fast-isel-abort", cl::Hidden,
63 cl::desc("Enable abort calls when \"fast\" instruction fails"));
Dan Gohman8a110532008-09-05 22:59:21 +000064static cl::opt<bool>
65SchedLiveInCopies("schedule-livein-copies",
66 cl::desc("Schedule copies of livein registers"),
67 cl::init(false));
Chris Lattneread0d882008-06-17 06:09:18 +000068
Chris Lattnerda8abb02005-09-01 18:44:10 +000069#ifndef NDEBUG
Chris Lattner7944d9d2005-01-12 03:41:21 +000070static cl::opt<bool>
Dan Gohman462dc7f2008-07-21 20:00:07 +000071ViewDAGCombine1("view-dag-combine1-dags", cl::Hidden,
72 cl::desc("Pop up a window to show dags before the first "
73 "dag combine pass"));
74static cl::opt<bool>
75ViewLegalizeTypesDAGs("view-legalize-types-dags", cl::Hidden,
76 cl::desc("Pop up a window to show dags before legalize types"));
77static cl::opt<bool>
78ViewLegalizeDAGs("view-legalize-dags", cl::Hidden,
79 cl::desc("Pop up a window to show dags before legalize"));
80static cl::opt<bool>
81ViewDAGCombine2("view-dag-combine2-dags", cl::Hidden,
82 cl::desc("Pop up a window to show dags before the second "
83 "dag combine pass"));
84static cl::opt<bool>
Duncan Sands25cf2272008-11-24 14:53:14 +000085ViewDAGCombineLT("view-dag-combine-lt-dags", cl::Hidden,
86 cl::desc("Pop up a window to show dags before the post legalize types"
87 " dag combine pass"));
88static cl::opt<bool>
Evan Chenga9c20912006-01-21 02:32:06 +000089ViewISelDAGs("view-isel-dags", cl::Hidden,
90 cl::desc("Pop up a window to show isel dags as they are selected"));
91static cl::opt<bool>
92ViewSchedDAGs("view-sched-dags", cl::Hidden,
93 cl::desc("Pop up a window to show sched dags as they are processed"));
Dan Gohman3e1a7ae2007-08-28 20:32:58 +000094static cl::opt<bool>
95ViewSUnitDAGs("view-sunit-dags", cl::Hidden,
Chris Lattner5bab7852008-01-25 17:24:52 +000096 cl::desc("Pop up a window to show SUnit dags after they are processed"));
Chris Lattner7944d9d2005-01-12 03:41:21 +000097#else
Dan Gohman462dc7f2008-07-21 20:00:07 +000098static const bool ViewDAGCombine1 = false,
99 ViewLegalizeTypesDAGs = false, ViewLegalizeDAGs = false,
100 ViewDAGCombine2 = false,
Duncan Sands25cf2272008-11-24 14:53:14 +0000101 ViewDAGCombineLT = false,
Dan Gohman462dc7f2008-07-21 20:00:07 +0000102 ViewISelDAGs = false, ViewSchedDAGs = false,
103 ViewSUnitDAGs = false;
Chris Lattner7944d9d2005-01-12 03:41:21 +0000104#endif
105
Jim Laskeyeb577ba2006-08-02 12:30:23 +0000106//===---------------------------------------------------------------------===//
107///
108/// RegisterScheduler class - Track the registration of instruction schedulers.
109///
110//===---------------------------------------------------------------------===//
111MachinePassRegistry RegisterScheduler::Registry;
112
113//===---------------------------------------------------------------------===//
114///
115/// ISHeuristic command line option for instruction schedulers.
116///
117//===---------------------------------------------------------------------===//
Dan Gohman844731a2008-05-13 00:00:25 +0000118static cl::opt<RegisterScheduler::FunctionPassCtor, false,
119 RegisterPassParser<RegisterScheduler> >
120ISHeuristic("pre-RA-sched",
121 cl::init(&createDefaultScheduler),
122 cl::desc("Instruction schedulers available (before register"
123 " allocation):"));
Jim Laskey13ec7022006-08-01 14:21:23 +0000124
Dan Gohman844731a2008-05-13 00:00:25 +0000125static RegisterScheduler
Dan Gohmanb8cab922008-10-14 20:25:08 +0000126defaultListDAGScheduler("default", "Best scheduler for the target",
Dan Gohman844731a2008-05-13 00:00:25 +0000127 createDefaultScheduler);
Evan Cheng4ef10862006-01-23 07:01:07 +0000128
Chris Lattner1c08c712005-01-07 07:47:53 +0000129namespace llvm {
130 //===--------------------------------------------------------------------===//
Jim Laskey9373beb2006-08-01 19:14:14 +0000131 /// createDefaultScheduler - This creates an instruction scheduler appropriate
132 /// for the target.
Dan Gohman47ac0f02009-02-11 04:27:20 +0000133 ScheduleDAGSDNodes* createDefaultScheduler(SelectionDAGISel *IS,
Bill Wendling98a366d2009-04-29 23:29:43 +0000134 CodeGenOpt::Level OptLevel) {
Dan Gohmane9530ec2009-01-15 16:58:17 +0000135 const TargetLowering &TLI = IS->getTargetLowering();
136
Bill Wendling98a366d2009-04-29 23:29:43 +0000137 if (OptLevel == CodeGenOpt::None)
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000138 return createFastDAGScheduler(IS, OptLevel);
Dan Gohman9e76fea2008-11-20 03:11:19 +0000139 if (TLI.getSchedulingPreference() == TargetLowering::SchedulingForLatency)
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000140 return createTDListDAGScheduler(IS, OptLevel);
Dan Gohman9e76fea2008-11-20 03:11:19 +0000141 assert(TLI.getSchedulingPreference() ==
142 TargetLowering::SchedulingForRegPressure && "Unknown sched type!");
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000143 return createBURRListDAGScheduler(IS, OptLevel);
Jim Laskey9373beb2006-08-01 19:14:14 +0000144 }
Chris Lattner1c08c712005-01-07 07:47:53 +0000145}
146
Evan Chengff9b3732008-01-30 18:18:23 +0000147// EmitInstrWithCustomInserter - This method should be implemented by targets
148// that mark instructions with the 'usesCustomDAGSchedInserter' flag. These
Chris Lattner025c39b2005-08-26 20:54:47 +0000149// instructions are special in various ways, which require special support to
150// insert. The specified MachineInstr is created but not inserted into any
151// basic blocks, and the scheduler passes ownership of it to this method.
Evan Chengff9b3732008-01-30 18:18:23 +0000152MachineBasicBlock *TargetLowering::EmitInstrWithCustomInserter(MachineInstr *MI,
Dan Gohman1fdbc1d2009-02-07 16:15:20 +0000153 MachineBasicBlock *MBB) const {
Bill Wendling832171c2006-12-07 20:04:42 +0000154 cerr << "If a target marks an instruction with "
155 << "'usesCustomDAGSchedInserter', it must implement "
Evan Chengff9b3732008-01-30 18:18:23 +0000156 << "TargetLowering::EmitInstrWithCustomInserter!\n";
Chris Lattner025c39b2005-08-26 20:54:47 +0000157 abort();
158 return 0;
159}
160
Dan Gohman8a110532008-09-05 22:59:21 +0000161/// EmitLiveInCopy - Emit a copy for a live in physical register. If the
162/// physical register has only a single copy use, then coalesced the copy
163/// if possible.
164static void EmitLiveInCopy(MachineBasicBlock *MBB,
165 MachineBasicBlock::iterator &InsertPos,
166 unsigned VirtReg, unsigned PhysReg,
167 const TargetRegisterClass *RC,
168 DenseMap<MachineInstr*, unsigned> &CopyRegMap,
169 const MachineRegisterInfo &MRI,
170 const TargetRegisterInfo &TRI,
171 const TargetInstrInfo &TII) {
172 unsigned NumUses = 0;
173 MachineInstr *UseMI = NULL;
174 for (MachineRegisterInfo::use_iterator UI = MRI.use_begin(VirtReg),
175 UE = MRI.use_end(); UI != UE; ++UI) {
176 UseMI = &*UI;
177 if (++NumUses > 1)
178 break;
179 }
180
181 // If the number of uses is not one, or the use is not a move instruction,
182 // don't coalesce. Also, only coalesce away a virtual register to virtual
183 // register copy.
184 bool Coalesced = false;
Evan Cheng04ee5a12009-01-20 19:12:24 +0000185 unsigned SrcReg, DstReg, SrcSubReg, DstSubReg;
Dan Gohman8a110532008-09-05 22:59:21 +0000186 if (NumUses == 1 &&
Evan Cheng04ee5a12009-01-20 19:12:24 +0000187 TII.isMoveInstr(*UseMI, SrcReg, DstReg, SrcSubReg, DstSubReg) &&
Dan Gohman8a110532008-09-05 22:59:21 +0000188 TargetRegisterInfo::isVirtualRegister(DstReg)) {
189 VirtReg = DstReg;
190 Coalesced = true;
191 }
192
193 // Now find an ideal location to insert the copy.
194 MachineBasicBlock::iterator Pos = InsertPos;
195 while (Pos != MBB->begin()) {
196 MachineInstr *PrevMI = prior(Pos);
197 DenseMap<MachineInstr*, unsigned>::iterator RI = CopyRegMap.find(PrevMI);
198 // copyRegToReg might emit multiple instructions to do a copy.
199 unsigned CopyDstReg = (RI == CopyRegMap.end()) ? 0 : RI->second;
200 if (CopyDstReg && !TRI.regsOverlap(CopyDstReg, PhysReg))
201 // This is what the BB looks like right now:
202 // r1024 = mov r0
203 // ...
204 // r1 = mov r1024
205 //
206 // We want to insert "r1025 = mov r1". Inserting this copy below the
207 // move to r1024 makes it impossible for that move to be coalesced.
208 //
209 // r1025 = mov r1
210 // r1024 = mov r0
211 // ...
212 // r1 = mov 1024
213 // r2 = mov 1025
214 break; // Woot! Found a good location.
215 --Pos;
216 }
217
218 TII.copyRegToReg(*MBB, Pos, VirtReg, PhysReg, RC, RC);
219 CopyRegMap.insert(std::make_pair(prior(Pos), VirtReg));
220 if (Coalesced) {
221 if (&*InsertPos == UseMI) ++InsertPos;
222 MBB->erase(UseMI);
223 }
224}
225
226/// EmitLiveInCopies - If this is the first basic block in the function,
227/// and if it has live ins that need to be copied into vregs, emit the
228/// copies into the block.
229static void EmitLiveInCopies(MachineBasicBlock *EntryMBB,
230 const MachineRegisterInfo &MRI,
231 const TargetRegisterInfo &TRI,
232 const TargetInstrInfo &TII) {
233 if (SchedLiveInCopies) {
234 // Emit the copies at a heuristically-determined location in the block.
235 DenseMap<MachineInstr*, unsigned> CopyRegMap;
236 MachineBasicBlock::iterator InsertPos = EntryMBB->begin();
237 for (MachineRegisterInfo::livein_iterator LI = MRI.livein_begin(),
238 E = MRI.livein_end(); LI != E; ++LI)
239 if (LI->second) {
240 const TargetRegisterClass *RC = MRI.getRegClass(LI->second);
241 EmitLiveInCopy(EntryMBB, InsertPos, LI->second, LI->first,
242 RC, CopyRegMap, MRI, TRI, TII);
243 }
244 } else {
245 // Emit the copies into the top of the block.
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 TII.copyRegToReg(*EntryMBB, EntryMBB->begin(),
251 LI->second, LI->first, RC, RC);
252 }
253 }
254}
255
Chris Lattner7041ee32005-01-11 05:56:49 +0000256//===----------------------------------------------------------------------===//
257// SelectionDAGISel code
258//===----------------------------------------------------------------------===//
Chris Lattner1c08c712005-01-07 07:47:53 +0000259
Bill Wendling98a366d2009-04-29 23:29:43 +0000260SelectionDAGISel::SelectionDAGISel(TargetMachine &tm, CodeGenOpt::Level OL) :
Dan Gohman79ce2762009-01-15 19:20:50 +0000261 FunctionPass(&ID), TM(tm), TLI(*tm.getTargetLowering()),
Dan Gohman7c3234c2008-08-27 23:52:12 +0000262 FuncInfo(new FunctionLoweringInfo(TLI)),
263 CurDAG(new SelectionDAG(TLI, *FuncInfo)),
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000264 SDL(new SelectionDAGLowering(*CurDAG, TLI, *FuncInfo, OL)),
Dan Gohman7c3234c2008-08-27 23:52:12 +0000265 GFI(),
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000266 OptLevel(OL),
Dan Gohman7c3234c2008-08-27 23:52:12 +0000267 DAGSize(0)
268{}
269
270SelectionDAGISel::~SelectionDAGISel() {
271 delete SDL;
272 delete CurDAG;
273 delete FuncInfo;
274}
275
Duncan Sands83ec4b62008-06-06 12:08:01 +0000276unsigned SelectionDAGISel::MakeReg(MVT VT) {
Chris Lattner84bc5422007-12-31 04:13:23 +0000277 return RegInfo->createVirtualRegister(TLI.getRegClassFor(VT));
Chris Lattner1c08c712005-01-07 07:47:53 +0000278}
279
Chris Lattner495a0b52005-08-17 06:37:43 +0000280void SelectionDAGISel::getAnalysisUsage(AnalysisUsage &AU) const {
Jim Laskeyc7c3f112006-10-16 20:52:31 +0000281 AU.addRequired<AliasAnalysis>();
Gordon Henriksen5eca0752008-08-17 18:44:35 +0000282 AU.addRequired<GCModuleInfo>();
Devang Patel6e7a1612009-01-09 19:11:50 +0000283 AU.addRequired<DwarfWriter>();
Chris Lattnerc8d288f2007-03-31 04:18:03 +0000284 AU.setPreservesAll();
Chris Lattner495a0b52005-08-17 06:37:43 +0000285}
Chris Lattner1c08c712005-01-07 07:47:53 +0000286
Chris Lattner1c08c712005-01-07 07:47:53 +0000287bool SelectionDAGISel::runOnFunction(Function &Fn) {
Dan Gohman4344a5d2008-09-09 23:05:00 +0000288 // Do some sanity-checking on the command-line options.
289 assert((!EnableFastISelVerbose || EnableFastISel) &&
290 "-fast-isel-verbose requires -fast-isel");
291 assert((!EnableFastISelAbort || EnableFastISel) &&
292 "-fast-isel-abort requires -fast-isel");
293
Devang Patel16f2ffd2009-04-16 02:33:41 +0000294 // Do not codegen any 'available_externally' functions at all, they have
295 // definitions outside the translation unit.
296 if (Fn.hasAvailableExternallyLinkage())
297 return false;
298
299
Dan Gohman5f43f922007-08-27 16:26:13 +0000300 // Get alias analysis for load/store combining.
301 AA = &getAnalysis<AliasAnalysis>();
302
Dan Gohman8a110532008-09-05 22:59:21 +0000303 TargetMachine &TM = TLI.getTargetMachine();
Dan Gohman79ce2762009-01-15 19:20:50 +0000304 MF = &MachineFunction::construct(&Fn, TM);
Dan Gohman8a110532008-09-05 22:59:21 +0000305 const TargetInstrInfo &TII = *TM.getInstrInfo();
306 const TargetRegisterInfo &TRI = *TM.getRegisterInfo();
307
Dan Gohman79ce2762009-01-15 19:20:50 +0000308 if (MF->getFunction()->hasGC())
309 GFI = &getAnalysis<GCModuleInfo>().getFunctionInfo(*MF->getFunction());
Gordon Henriksence224772008-01-07 01:30:38 +0000310 else
Gordon Henriksen5eca0752008-08-17 18:44:35 +0000311 GFI = 0;
Dan Gohman79ce2762009-01-15 19:20:50 +0000312 RegInfo = &MF->getRegInfo();
Bill Wendling832171c2006-12-07 20:04:42 +0000313 DOUT << "\n\n\n=== " << Fn.getName() << "\n";
Chris Lattner1c08c712005-01-07 07:47:53 +0000314
Duncan Sands1465d612009-01-28 13:14:17 +0000315 MachineModuleInfo *MMI = getAnalysisIfAvailable<MachineModuleInfo>();
316 DwarfWriter *DW = getAnalysisIfAvailable<DwarfWriter>();
Dan Gohman79ce2762009-01-15 19:20:50 +0000317 CurDAG->init(*MF, MMI, DW);
Devang Patelb51d40c2009-02-03 18:46:32 +0000318 FuncInfo->set(Fn, *MF, *CurDAG, EnableFastISel);
Dan Gohman7c3234c2008-08-27 23:52:12 +0000319 SDL->init(GFI, *AA);
Chris Lattner1c08c712005-01-07 07:47:53 +0000320
Dale Johannesen1532f3d2008-04-02 00:25:04 +0000321 for (Function::iterator I = Fn.begin(), E = Fn.end(); I != E; ++I)
322 if (InvokeInst *Invoke = dyn_cast<InvokeInst>(I->getTerminator()))
323 // Mark landing pad.
Dan Gohman7c3234c2008-08-27 23:52:12 +0000324 FuncInfo->MBBMap[Invoke->getSuccessor(1)]->setIsLandingPad();
Duncan Sands9fac0b52007-06-06 10:05:18 +0000325
Dan Gohman79ce2762009-01-15 19:20:50 +0000326 SelectAllBasicBlocks(Fn, *MF, MMI, DW, TII);
Misha Brukmanedf128a2005-04-21 22:36:52 +0000327
Dan Gohman8a110532008-09-05 22:59:21 +0000328 // If the first basic block in the function has live ins that need to be
329 // copied into vregs, emit the copies into the top of the block before
330 // emitting the code for the block.
Dan Gohman79ce2762009-01-15 19:20:50 +0000331 EmitLiveInCopies(MF->begin(), *RegInfo, TRI, TII);
Dan Gohman8a110532008-09-05 22:59:21 +0000332
Evan Chengad2070c2007-02-10 02:43:39 +0000333 // Add function live-ins to entry block live-in set.
Dan Gohman8a110532008-09-05 22:59:21 +0000334 for (MachineRegisterInfo::livein_iterator I = RegInfo->livein_begin(),
335 E = RegInfo->livein_end(); I != E; ++I)
Dan Gohman79ce2762009-01-15 19:20:50 +0000336 MF->begin()->addLiveIn(I->first);
Evan Chengad2070c2007-02-10 02:43:39 +0000337
Duncan Sandsf4070822007-06-15 19:04:19 +0000338#ifndef NDEBUG
Dan Gohman7c3234c2008-08-27 23:52:12 +0000339 assert(FuncInfo->CatchInfoFound.size() == FuncInfo->CatchInfoLost.size() &&
Duncan Sandsf4070822007-06-15 19:04:19 +0000340 "Not all catch info was assigned to a landing pad!");
341#endif
342
Dan Gohman7c3234c2008-08-27 23:52:12 +0000343 FuncInfo->clear();
344
Chris Lattner1c08c712005-01-07 07:47:53 +0000345 return true;
346}
347
Duncan Sandsf4070822007-06-15 19:04:19 +0000348static void copyCatchInfo(BasicBlock *SrcBB, BasicBlock *DestBB,
349 MachineModuleInfo *MMI, FunctionLoweringInfo &FLI) {
Duncan Sandsf4070822007-06-15 19:04:19 +0000350 for (BasicBlock::iterator I = SrcBB->begin(), E = --SrcBB->end(); I != E; ++I)
Dan Gohmanf0cbcd42008-09-03 16:12:24 +0000351 if (EHSelectorInst *EHSel = dyn_cast<EHSelectorInst>(I)) {
Duncan Sandsf4070822007-06-15 19:04:19 +0000352 // Apply the catch info to DestBB.
Dan Gohmanf0cbcd42008-09-03 16:12:24 +0000353 AddCatchInfo(*EHSel, MMI, FLI.MBBMap[DestBB]);
Duncan Sandsf4070822007-06-15 19:04:19 +0000354#ifndef NDEBUG
Duncan Sands560a7372007-11-15 09:54:37 +0000355 if (!FLI.MBBMap[SrcBB]->isLandingPad())
Dan Gohmanf0cbcd42008-09-03 16:12:24 +0000356 FLI.CatchInfoFound.insert(EHSel);
Duncan Sandsf4070822007-06-15 19:04:19 +0000357#endif
358 }
359}
360
Arnold Schwaighofer30e62c02008-04-30 09:16:33 +0000361/// IsFixedFrameObjectWithPosOffset - Check if object is a fixed frame object and
362/// whether object offset >= 0.
363static bool
Dan Gohman79ce2762009-01-15 19:20:50 +0000364IsFixedFrameObjectWithPosOffset(MachineFrameInfo *MFI, SDValue Op) {
Arnold Schwaighofer30e62c02008-04-30 09:16:33 +0000365 if (!isa<FrameIndexSDNode>(Op)) return false;
366
367 FrameIndexSDNode * FrameIdxNode = dyn_cast<FrameIndexSDNode>(Op);
368 int FrameIdx = FrameIdxNode->getIndex();
369 return MFI->isFixedObjectIndex(FrameIdx) &&
370 MFI->getObjectOffset(FrameIdx) >= 0;
371}
372
373/// IsPossiblyOverwrittenArgumentOfTailCall - Check if the operand could
374/// possibly be overwritten when lowering the outgoing arguments in a tail
375/// call. Currently the implementation of this call is very conservative and
376/// assumes all arguments sourcing from FORMAL_ARGUMENTS or a CopyFromReg with
377/// virtual registers would be overwritten by direct lowering.
Dan Gohman475871a2008-07-27 21:46:04 +0000378static bool IsPossiblyOverwrittenArgumentOfTailCall(SDValue Op,
Dan Gohman79ce2762009-01-15 19:20:50 +0000379 MachineFrameInfo *MFI) {
Arnold Schwaighofer30e62c02008-04-30 09:16:33 +0000380 RegisterSDNode * OpReg = NULL;
381 if (Op.getOpcode() == ISD::FORMAL_ARGUMENTS ||
382 (Op.getOpcode()== ISD::CopyFromReg &&
383 (OpReg = dyn_cast<RegisterSDNode>(Op.getOperand(1))) &&
384 (OpReg->getReg() >= TargetRegisterInfo::FirstVirtualRegister)) ||
385 (Op.getOpcode() == ISD::LOAD &&
386 IsFixedFrameObjectWithPosOffset(MFI, Op.getOperand(1))) ||
387 (Op.getOpcode() == ISD::MERGE_VALUES &&
Gabor Greif99a6cb92008-08-26 22:36:50 +0000388 Op.getOperand(Op.getResNo()).getOpcode() == ISD::LOAD &&
389 IsFixedFrameObjectWithPosOffset(MFI, Op.getOperand(Op.getResNo()).
Arnold Schwaighofer30e62c02008-04-30 09:16:33 +0000390 getOperand(1))))
391 return true;
392 return false;
393}
394
Arnold Schwaighoferc85e1712007-10-11 19:40:01 +0000395/// CheckDAGForTailCallsAndFixThem - This Function looks for CALL nodes in the
Arnold Schwaighofer48abc5c2007-10-12 21:30:57 +0000396/// DAG and fixes their tailcall attribute operand.
Arnold Schwaighoferc85e1712007-10-11 19:40:01 +0000397static void CheckDAGForTailCallsAndFixThem(SelectionDAG &DAG,
Dan Gohmane9530ec2009-01-15 16:58:17 +0000398 const TargetLowering& TLI) {
Arnold Schwaighoferc85e1712007-10-11 19:40:01 +0000399 SDNode * Ret = NULL;
Dan Gohman475871a2008-07-27 21:46:04 +0000400 SDValue Terminator = DAG.getRoot();
Arnold Schwaighoferc85e1712007-10-11 19:40:01 +0000401
402 // Find RET node.
403 if (Terminator.getOpcode() == ISD::RET) {
Gabor Greifba36cb52008-08-28 21:40:38 +0000404 Ret = Terminator.getNode();
Arnold Schwaighoferc85e1712007-10-11 19:40:01 +0000405 }
406
407 // Fix tail call attribute of CALL nodes.
408 for (SelectionDAG::allnodes_iterator BE = DAG.allnodes_begin(),
Dan Gohman0e5f1302008-07-07 23:02:41 +0000409 BI = DAG.allnodes_end(); BI != BE; ) {
410 --BI;
Dan Gohman095cc292008-09-13 01:54:27 +0000411 if (CallSDNode *TheCall = dyn_cast<CallSDNode>(BI)) {
Dan Gohman475871a2008-07-27 21:46:04 +0000412 SDValue OpRet(Ret, 0);
413 SDValue OpCall(BI, 0);
Dan Gohman095cc292008-09-13 01:54:27 +0000414 bool isMarkedTailCall = TheCall->isTailCall();
Arnold Schwaighoferc85e1712007-10-11 19:40:01 +0000415 // If CALL node has tail call attribute set to true and the call is not
416 // eligible (no RET or the target rejects) the attribute is fixed to
Arnold Schwaighofer48abc5c2007-10-12 21:30:57 +0000417 // false. The TargetLowering::IsEligibleForTailCallOptimization function
Arnold Schwaighoferc85e1712007-10-11 19:40:01 +0000418 // must correctly identify tail call optimizable calls.
Arnold Schwaighofer30e62c02008-04-30 09:16:33 +0000419 if (!isMarkedTailCall) continue;
420 if (Ret==NULL ||
Dan Gohman095cc292008-09-13 01:54:27 +0000421 !TLI.IsEligibleForTailCallOptimization(TheCall, OpRet, DAG)) {
422 // Not eligible. Mark CALL node as non tail call. Note that we
423 // can modify the call node in place since calls are not CSE'd.
424 TheCall->setNotTailCall();
Arnold Schwaighofer30e62c02008-04-30 09:16:33 +0000425 } else {
426 // Look for tail call clobbered arguments. Emit a series of
427 // copyto/copyfrom virtual register nodes to protect them.
Dan Gohman475871a2008-07-27 21:46:04 +0000428 SmallVector<SDValue, 32> Ops;
Dan Gohman095cc292008-09-13 01:54:27 +0000429 SDValue Chain = TheCall->getChain(), InFlag;
430 Ops.push_back(Chain);
431 Ops.push_back(TheCall->getCallee());
432 for (unsigned i = 0, e = TheCall->getNumArgs(); i != e; ++i) {
433 SDValue Arg = TheCall->getArg(i);
434 bool isByVal = TheCall->getArgFlags(i).isByVal();
435 MachineFunction &MF = DAG.getMachineFunction();
436 MachineFrameInfo *MFI = MF.getFrameInfo();
437 if (!isByVal &&
438 IsPossiblyOverwrittenArgumentOfTailCall(Arg, MFI)) {
439 MVT VT = Arg.getValueType();
440 unsigned VReg = MF.getRegInfo().
441 createVirtualRegister(TLI.getRegClassFor(VT));
Dale Johannesen6f38cb62009-02-07 19:59:05 +0000442 Chain = DAG.getCopyToReg(Chain, Arg.getDebugLoc(),
Dale Johannesenc460ae92009-02-04 00:13:36 +0000443 VReg, Arg, InFlag);
Dan Gohman095cc292008-09-13 01:54:27 +0000444 InFlag = Chain.getValue(1);
Dale Johannesen6f38cb62009-02-07 19:59:05 +0000445 Arg = DAG.getCopyFromReg(Chain, Arg.getDebugLoc(),
Dale Johannesenc460ae92009-02-04 00:13:36 +0000446 VReg, VT, InFlag);
Dan Gohman095cc292008-09-13 01:54:27 +0000447 Chain = Arg.getValue(1);
448 InFlag = Arg.getValue(2);
Arnold Schwaighofer30e62c02008-04-30 09:16:33 +0000449 }
450 Ops.push_back(Arg);
Dan Gohman095cc292008-09-13 01:54:27 +0000451 Ops.push_back(TheCall->getArgFlagsVal(i));
Arnold Schwaighofer30e62c02008-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 Schwaighoferc85e1712007-10-11 19:40:01 +0000456 }
457 }
458 }
459}
460
Dan Gohmanf350b272008-08-23 02:25:05 +0000461void SelectionDAGISel::SelectBasicBlock(BasicBlock *LLVMBB,
462 BasicBlock::iterator Begin,
Dan Gohman5edd3612008-08-28 20:28:56 +0000463 BasicBlock::iterator End) {
Dan Gohman7c3234c2008-08-27 23:52:12 +0000464 SDL->setCurrentBasicBlock(BB);
Dan Gohmanf350b272008-08-23 02:25:05 +0000465
Dan Gohmanf350b272008-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 Gohman7c3234c2008-08-27 23:52:12 +0000469 SDL->visit(*I);
Dan Gohmanf350b272008-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)
Dan Gohmanad62f532009-04-23 23:13:24 +0000474 if (!isa<PHINode>(I) && !isa<InvokeInst>(I))
475 SDL->CopyToExportRegsIfNeeded(I);
Dan Gohmanf350b272008-08-23 02:25:05 +0000476
477 // Handle PHI nodes in successor blocks.
Dan Gohman3df24e62008-09-03 23:12:08 +0000478 if (End == LLVMBB->end()) {
Dan Gohman7c3234c2008-08-27 23:52:12 +0000479 HandlePHINodesInSuccessorBlocks(LLVMBB);
Dan Gohman3df24e62008-09-03 23:12:08 +0000480
481 // Lower the terminator after the copies are emitted.
482 SDL->visit(*LLVMBB->getTerminator());
483 }
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000484
Chris Lattnera651cf62005-01-17 19:43:36 +0000485 // Make sure the root of the DAG is up-to-date.
Dan Gohman7c3234c2008-08-27 23:52:12 +0000486 CurDAG->setRoot(SDL->getControlRoot());
Arnold Schwaighoferc85e1712007-10-11 19:40:01 +0000487
488 // Check whether calls in this block are real tail calls. Fix up CALL nodes
489 // with correct tailcall attribute so that the target can rely on the tailcall
490 // attribute indicating whether the call is really eligible for tail call
491 // optimization.
Dan Gohman1937e2f2008-09-16 01:42:28 +0000492 if (PerformTailCallOpt)
493 CheckDAGForTailCallsAndFixThem(*CurDAG, TLI);
Dan Gohmanf350b272008-08-23 02:25:05 +0000494
495 // Final step, emit the lowered DAG as machine code.
496 CodeGenAndEmitDAG();
Dan Gohman7c3234c2008-08-27 23:52:12 +0000497 SDL->clear();
Chris Lattner1c08c712005-01-07 07:47:53 +0000498}
499
Dan Gohmanf350b272008-08-23 02:25:05 +0000500void SelectionDAGISel::ComputeLiveOutVRegInfo() {
Chris Lattneread0d882008-06-17 06:09:18 +0000501 SmallPtrSet<SDNode*, 128> VisitedNodes;
502 SmallVector<SDNode*, 128> Worklist;
503
Gabor Greifba36cb52008-08-28 21:40:38 +0000504 Worklist.push_back(CurDAG->getRoot().getNode());
Chris Lattneread0d882008-06-17 06:09:18 +0000505
506 APInt Mask;
507 APInt KnownZero;
508 APInt KnownOne;
509
510 while (!Worklist.empty()) {
511 SDNode *N = Worklist.back();
512 Worklist.pop_back();
513
514 // If we've already seen this node, ignore it.
515 if (!VisitedNodes.insert(N))
516 continue;
517
518 // Otherwise, add all chain operands to the worklist.
519 for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i)
520 if (N->getOperand(i).getValueType() == MVT::Other)
Gabor Greifba36cb52008-08-28 21:40:38 +0000521 Worklist.push_back(N->getOperand(i).getNode());
Chris Lattneread0d882008-06-17 06:09:18 +0000522
523 // If this is a CopyToReg with a vreg dest, process it.
524 if (N->getOpcode() != ISD::CopyToReg)
525 continue;
526
527 unsigned DestReg = cast<RegisterSDNode>(N->getOperand(1))->getReg();
528 if (!TargetRegisterInfo::isVirtualRegister(DestReg))
529 continue;
530
531 // Ignore non-scalar or non-integer values.
Dan Gohman475871a2008-07-27 21:46:04 +0000532 SDValue Src = N->getOperand(2);
Chris Lattneread0d882008-06-17 06:09:18 +0000533 MVT SrcVT = Src.getValueType();
534 if (!SrcVT.isInteger() || SrcVT.isVector())
535 continue;
536
Dan Gohmanf350b272008-08-23 02:25:05 +0000537 unsigned NumSignBits = CurDAG->ComputeNumSignBits(Src);
Chris Lattneread0d882008-06-17 06:09:18 +0000538 Mask = APInt::getAllOnesValue(SrcVT.getSizeInBits());
Dan Gohmanf350b272008-08-23 02:25:05 +0000539 CurDAG->ComputeMaskedBits(Src, Mask, KnownZero, KnownOne);
Chris Lattneread0d882008-06-17 06:09:18 +0000540
541 // Only install this information if it tells us something.
542 if (NumSignBits != 1 || KnownZero != 0 || KnownOne != 0) {
543 DestReg -= TargetRegisterInfo::FirstVirtualRegister;
Dan Gohmanf350b272008-08-23 02:25:05 +0000544 FunctionLoweringInfo &FLI = CurDAG->getFunctionLoweringInfo();
Chris Lattneread0d882008-06-17 06:09:18 +0000545 if (DestReg >= FLI.LiveOutRegInfo.size())
546 FLI.LiveOutRegInfo.resize(DestReg+1);
547 FunctionLoweringInfo::LiveOutInfo &LOI = FLI.LiveOutRegInfo[DestReg];
548 LOI.NumSignBits = NumSignBits;
Dan Gohmana80efce2009-03-27 23:55:04 +0000549 LOI.KnownOne = KnownOne;
550 LOI.KnownZero = KnownZero;
Chris Lattneread0d882008-06-17 06:09:18 +0000551 }
552 }
553}
554
Dan Gohmanf350b272008-08-23 02:25:05 +0000555void SelectionDAGISel::CodeGenAndEmitDAG() {
Dan Gohman462dc7f2008-07-21 20:00:07 +0000556 std::string GroupName;
557 if (TimePassesIsEnabled)
558 GroupName = "Instruction Selection and Scheduling";
559 std::string BlockName;
560 if (ViewDAGCombine1 || ViewLegalizeTypesDAGs || ViewLegalizeDAGs ||
Duncan Sands25cf2272008-11-24 14:53:14 +0000561 ViewDAGCombine2 || ViewDAGCombineLT || ViewISelDAGs || ViewSchedDAGs ||
562 ViewSUnitDAGs)
Dan Gohmanf350b272008-08-23 02:25:05 +0000563 BlockName = CurDAG->getMachineFunction().getFunction()->getName() + ':' +
Dan Gohman462dc7f2008-07-21 20:00:07 +0000564 BB->getBasicBlock()->getName();
565
566 DOUT << "Initial selection DAG:\n";
Dan Gohmanf350b272008-08-23 02:25:05 +0000567 DEBUG(CurDAG->dump());
Dan Gohman462dc7f2008-07-21 20:00:07 +0000568
Dan Gohmanf350b272008-08-23 02:25:05 +0000569 if (ViewDAGCombine1) CurDAG->viewGraph("dag-combine1 input for " + BlockName);
Dan Gohman417e11b2007-10-08 15:12:17 +0000570
Chris Lattneraf21d552005-10-10 16:47:10 +0000571 // Run the DAG combiner in pre-legalize mode.
Evan Chengebffb662008-07-01 17:59:20 +0000572 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000573 NamedRegionTimer T("DAG Combining 1", GroupName);
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000574 CurDAG->Combine(Unrestricted, *AA, OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000575 } else {
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000576 CurDAG->Combine(Unrestricted, *AA, OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000577 }
Nate Begeman2300f552005-09-07 00:15:36 +0000578
Dan Gohman417e11b2007-10-08 15:12:17 +0000579 DOUT << "Optimized lowered selection DAG:\n";
Dan Gohmanf350b272008-08-23 02:25:05 +0000580 DEBUG(CurDAG->dump());
Duncan Sandsf00e74f2008-07-17 17:06:03 +0000581
Chris Lattner1c08c712005-01-07 07:47:53 +0000582 // Second step, hack on the DAG until it only uses operations and types that
583 // the target supports.
Duncan Sands7cb07872008-10-27 08:42:46 +0000584 if (!DisableLegalizeTypes) {
Dan Gohmanf350b272008-08-23 02:25:05 +0000585 if (ViewLegalizeTypesDAGs) CurDAG->viewGraph("legalize-types input for " +
586 BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000587
Duncan Sands25cf2272008-11-24 14:53:14 +0000588 bool Changed;
Dan Gohman462dc7f2008-07-21 20:00:07 +0000589 if (TimePassesIsEnabled) {
590 NamedRegionTimer T("Type Legalization", GroupName);
Duncan Sands25cf2272008-11-24 14:53:14 +0000591 Changed = CurDAG->LegalizeTypes();
Dan Gohman462dc7f2008-07-21 20:00:07 +0000592 } else {
Duncan Sands25cf2272008-11-24 14:53:14 +0000593 Changed = CurDAG->LegalizeTypes();
Dan Gohman462dc7f2008-07-21 20:00:07 +0000594 }
595
596 DOUT << "Type-legalized selection DAG:\n";
Dan Gohmanf350b272008-08-23 02:25:05 +0000597 DEBUG(CurDAG->dump());
Dan Gohman462dc7f2008-07-21 20:00:07 +0000598
Duncan Sands25cf2272008-11-24 14:53:14 +0000599 if (Changed) {
600 if (ViewDAGCombineLT)
601 CurDAG->viewGraph("dag-combine-lt input for " + BlockName);
602
603 // Run the DAG combiner in post-type-legalize mode.
604 if (TimePassesIsEnabled) {
605 NamedRegionTimer T("DAG Combining after legalize types", GroupName);
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000606 CurDAG->Combine(NoIllegalTypes, *AA, OptLevel);
Duncan Sands25cf2272008-11-24 14:53:14 +0000607 } else {
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000608 CurDAG->Combine(NoIllegalTypes, *AA, OptLevel);
Duncan Sands25cf2272008-11-24 14:53:14 +0000609 }
610
611 DOUT << "Optimized type-legalized selection DAG:\n";
612 DEBUG(CurDAG->dump());
613 }
Chris Lattner70587ea2008-07-10 23:37:50 +0000614 }
Duncan Sandsf00e74f2008-07-17 17:06:03 +0000615
Dan Gohmanf350b272008-08-23 02:25:05 +0000616 if (ViewLegalizeDAGs) CurDAG->viewGraph("legalize input for " + BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000617
Evan Chengebffb662008-07-01 17:59:20 +0000618 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000619 NamedRegionTimer T("DAG Legalization", GroupName);
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000620 CurDAG->Legalize(DisableLegalizeTypes, OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000621 } else {
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000622 CurDAG->Legalize(DisableLegalizeTypes, OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000623 }
Nate Begemanf15485a2006-03-27 01:32:24 +0000624
Bill Wendling832171c2006-12-07 20:04:42 +0000625 DOUT << "Legalized selection DAG:\n";
Dan Gohmanf350b272008-08-23 02:25:05 +0000626 DEBUG(CurDAG->dump());
Nate Begemanf15485a2006-03-27 01:32:24 +0000627
Dan Gohmanf350b272008-08-23 02:25:05 +0000628 if (ViewDAGCombine2) CurDAG->viewGraph("dag-combine2 input for " + BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000629
Chris Lattneraf21d552005-10-10 16:47:10 +0000630 // Run the DAG combiner in post-legalize mode.
Evan Chengebffb662008-07-01 17:59:20 +0000631 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000632 NamedRegionTimer T("DAG Combining 2", GroupName);
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000633 CurDAG->Combine(NoIllegalOperations, *AA, OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000634 } else {
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000635 CurDAG->Combine(NoIllegalOperations, *AA, OptLevel);
Evan Chengebffb662008-07-01 17:59:20 +0000636 }
Nate Begeman2300f552005-09-07 00:15:36 +0000637
Dan Gohman417e11b2007-10-08 15:12:17 +0000638 DOUT << "Optimized legalized selection DAG:\n";
Dan Gohmanf350b272008-08-23 02:25:05 +0000639 DEBUG(CurDAG->dump());
Dan Gohman417e11b2007-10-08 15:12:17 +0000640
Dan Gohmanf350b272008-08-23 02:25:05 +0000641 if (ViewISelDAGs) CurDAG->viewGraph("isel input for " + BlockName);
Chris Lattneread0d882008-06-17 06:09:18 +0000642
Bill Wendling98a366d2009-04-29 23:29:43 +0000643 if (OptLevel != CodeGenOpt::None)
Dan Gohmanf350b272008-08-23 02:25:05 +0000644 ComputeLiveOutVRegInfo();
Evan Cheng552c4a82006-04-28 02:09:19 +0000645
Chris Lattnera33ef482005-03-30 01:10:47 +0000646 // Third, instruction select all of the operations to machine code, adding the
647 // code to the MachineBasicBlock.
Evan Chengebffb662008-07-01 17:59:20 +0000648 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000649 NamedRegionTimer T("Instruction Selection", GroupName);
Dan Gohmanf350b272008-08-23 02:25:05 +0000650 InstructionSelect();
Evan Chengebffb662008-07-01 17:59:20 +0000651 } else {
Dan Gohmanf350b272008-08-23 02:25:05 +0000652 InstructionSelect();
Evan Chengebffb662008-07-01 17:59:20 +0000653 }
Evan Chengdb8d56b2008-06-30 20:45:06 +0000654
Dan Gohman462dc7f2008-07-21 20:00:07 +0000655 DOUT << "Selected selection DAG:\n";
Dan Gohmanf350b272008-08-23 02:25:05 +0000656 DEBUG(CurDAG->dump());
Dan Gohman462dc7f2008-07-21 20:00:07 +0000657
Dan Gohmanf350b272008-08-23 02:25:05 +0000658 if (ViewSchedDAGs) CurDAG->viewGraph("scheduler input for " + BlockName);
Dan Gohman462dc7f2008-07-21 20:00:07 +0000659
Dan Gohman5e843682008-07-14 18:19:29 +0000660 // Schedule machine code.
Dan Gohman47ac0f02009-02-11 04:27:20 +0000661 ScheduleDAGSDNodes *Scheduler = CreateScheduler();
Dan Gohman5e843682008-07-14 18:19:29 +0000662 if (TimePassesIsEnabled) {
663 NamedRegionTimer T("Instruction Scheduling", GroupName);
Dan Gohman47ac0f02009-02-11 04:27:20 +0000664 Scheduler->Run(CurDAG, BB, BB->end());
Dan Gohman5e843682008-07-14 18:19:29 +0000665 } else {
Dan Gohman47ac0f02009-02-11 04:27:20 +0000666 Scheduler->Run(CurDAG, BB, BB->end());
Dan Gohman5e843682008-07-14 18:19:29 +0000667 }
668
Dan Gohman462dc7f2008-07-21 20:00:07 +0000669 if (ViewSUnitDAGs) Scheduler->viewGraph();
670
Evan Chengdb8d56b2008-06-30 20:45:06 +0000671 // Emit machine code to BB. This can change 'BB' to the last block being
672 // inserted into.
Evan Chengebffb662008-07-01 17:59:20 +0000673 if (TimePassesIsEnabled) {
Dan Gohman5e843682008-07-14 18:19:29 +0000674 NamedRegionTimer T("Instruction Creation", GroupName);
675 BB = Scheduler->EmitSchedule();
Evan Chengebffb662008-07-01 17:59:20 +0000676 } else {
Dan Gohman5e843682008-07-14 18:19:29 +0000677 BB = Scheduler->EmitSchedule();
678 }
679
680 // Free the scheduler state.
681 if (TimePassesIsEnabled) {
682 NamedRegionTimer T("Instruction Scheduling Cleanup", GroupName);
683 delete Scheduler;
684 } else {
685 delete Scheduler;
Evan Chengebffb662008-07-01 17:59:20 +0000686 }
Evan Chengdb8d56b2008-06-30 20:45:06 +0000687
Bill Wendling832171c2006-12-07 20:04:42 +0000688 DOUT << "Selected machine code:\n";
Chris Lattner1c08c712005-01-07 07:47:53 +0000689 DEBUG(BB->dump());
Nate Begemanf15485a2006-03-27 01:32:24 +0000690}
Chris Lattner1c08c712005-01-07 07:47:53 +0000691
Dan Gohman79ce2762009-01-15 19:20:50 +0000692void SelectionDAGISel::SelectAllBasicBlocks(Function &Fn,
693 MachineFunction &MF,
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000694 MachineModuleInfo *MMI,
Devang Patel83489bb2009-01-13 00:35:13 +0000695 DwarfWriter *DW,
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000696 const TargetInstrInfo &TII) {
Dan Gohmana43abd12008-09-29 21:55:50 +0000697 // Initialize the Fast-ISel state, if needed.
698 FastISel *FastIS = 0;
699 if (EnableFastISel)
Dan Gohman79ce2762009-01-15 19:20:50 +0000700 FastIS = TLI.createFastISel(MF, MMI, DW,
Dan Gohmana43abd12008-09-29 21:55:50 +0000701 FuncInfo->ValueMap,
702 FuncInfo->MBBMap,
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000703 FuncInfo->StaticAllocaMap
704#ifndef NDEBUG
705 , FuncInfo->CatchInfoLost
706#endif
707 );
Dan Gohmana43abd12008-09-29 21:55:50 +0000708
709 // Iterate over all basic blocks in the function.
Evan Cheng39fd6e82008-08-07 00:43:25 +0000710 for (Function::iterator I = Fn.begin(), E = Fn.end(); I != E; ++I) {
711 BasicBlock *LLVMBB = &*I;
Dan Gohman7c3234c2008-08-27 23:52:12 +0000712 BB = FuncInfo->MBBMap[LLVMBB];
Dan Gohmanf350b272008-08-23 02:25:05 +0000713
Dan Gohman3df24e62008-09-03 23:12:08 +0000714 BasicBlock::iterator const Begin = LLVMBB->begin();
715 BasicBlock::iterator const End = LLVMBB->end();
Evan Cheng9f118502008-09-08 16:01:27 +0000716 BasicBlock::iterator BI = Begin;
Dan Gohman5edd3612008-08-28 20:28:56 +0000717
718 // Lower any arguments needed in this block if this is the entry block.
Dan Gohman33134c42008-09-25 17:05:24 +0000719 bool SuppressFastISel = false;
720 if (LLVMBB == &Fn.getEntryBlock()) {
Dan Gohman5edd3612008-08-28 20:28:56 +0000721 LowerArguments(LLVMBB);
Dan Gohmanf350b272008-08-23 02:25:05 +0000722
Dan Gohman33134c42008-09-25 17:05:24 +0000723 // If any of the arguments has the byval attribute, forgo
724 // fast-isel in the entry block.
Dan Gohmana43abd12008-09-29 21:55:50 +0000725 if (FastIS) {
Dan Gohman33134c42008-09-25 17:05:24 +0000726 unsigned j = 1;
727 for (Function::arg_iterator I = Fn.arg_begin(), E = Fn.arg_end();
728 I != E; ++I, ++j)
Devang Patel05988662008-09-25 21:00:45 +0000729 if (Fn.paramHasAttr(j, Attribute::ByVal)) {
Dan Gohman77ca41e2008-09-25 17:21:42 +0000730 if (EnableFastISelVerbose || EnableFastISelAbort)
731 cerr << "FastISel skips entry block due to byval argument\n";
Dan Gohman33134c42008-09-25 17:05:24 +0000732 SuppressFastISel = true;
733 break;
734 }
735 }
736 }
737
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000738 if (MMI && BB->isLandingPad()) {
739 // Add a label to mark the beginning of the landing pad. Deletion of the
740 // landing pad can thus be detected via the MachineModuleInfo.
741 unsigned LabelID = MMI->addLandingPad(BB);
742
743 const TargetInstrDesc &II = TII.get(TargetInstrInfo::EH_LABEL);
Bill Wendlingb2884872009-02-03 01:55:42 +0000744 BuildMI(BB, SDL->getCurDebugLoc(), II).addImm(LabelID);
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000745
746 // Mark exception register as live in.
747 unsigned Reg = TLI.getExceptionAddressRegister();
748 if (Reg) BB->addLiveIn(Reg);
749
750 // Mark exception selector register as live in.
751 Reg = TLI.getExceptionSelectorRegister();
752 if (Reg) BB->addLiveIn(Reg);
753
754 // FIXME: Hack around an exception handling flaw (PR1508): the personality
755 // function and list of typeids logically belong to the invoke (or, if you
756 // like, the basic block containing the invoke), and need to be associated
757 // with it in the dwarf exception handling tables. Currently however the
758 // information is provided by an intrinsic (eh.selector) that can be moved
759 // to unexpected places by the optimizers: if the unwind edge is critical,
760 // then breaking it can result in the intrinsics being in the successor of
761 // the landing pad, not the landing pad itself. This results in exceptions
762 // not being caught because no typeids are associated with the invoke.
763 // This may not be the only way things can go wrong, but it is the only way
764 // we try to work around for the moment.
765 BranchInst *Br = dyn_cast<BranchInst>(LLVMBB->getTerminator());
766
767 if (Br && Br->isUnconditional()) { // Critical edge?
768 BasicBlock::iterator I, E;
769 for (I = LLVMBB->begin(), E = --LLVMBB->end(); I != E; ++I)
770 if (isa<EHSelectorInst>(I))
771 break;
772
773 if (I == E)
774 // No catch info found - try to extract some from the successor.
775 copyCatchInfo(Br->getSuccessor(0), LLVMBB, MMI, *FuncInfo);
776 }
777 }
778
Dan Gohmanf350b272008-08-23 02:25:05 +0000779 // Before doing SelectionDAG ISel, see if FastISel has been requested.
Dan Gohmandd5b58a2008-10-14 23:54:11 +0000780 if (FastIS && !SuppressFastISel) {
Dan Gohmana43abd12008-09-29 21:55:50 +0000781 // Emit code for any incoming arguments. This must happen before
782 // beginning FastISel on the entry block.
783 if (LLVMBB == &Fn.getEntryBlock()) {
784 CurDAG->setRoot(SDL->getControlRoot());
785 CodeGenAndEmitDAG();
786 SDL->clear();
787 }
Dan Gohman241f4642008-10-04 00:56:36 +0000788 FastIS->startNewBlock(BB);
Dan Gohmana43abd12008-09-29 21:55:50 +0000789 // Do FastISel on as many instructions as possible.
790 for (; BI != End; ++BI) {
791 // Just before the terminator instruction, insert instructions to
792 // feed PHI nodes in successor blocks.
793 if (isa<TerminatorInst>(BI))
794 if (!HandlePHINodesInSuccessorBlocksFast(LLVMBB, FastIS)) {
Dan Gohman4344a5d2008-09-09 23:05:00 +0000795 if (EnableFastISelVerbose || EnableFastISelAbort) {
Dan Gohman293d5f82008-09-09 22:06:46 +0000796 cerr << "FastISel miss: ";
797 BI->dump();
798 }
Dan Gohman4344a5d2008-09-09 23:05:00 +0000799 if (EnableFastISelAbort)
Dan Gohmana43abd12008-09-29 21:55:50 +0000800 assert(0 && "FastISel didn't handle a PHI in a successor");
801 break;
Dan Gohmanf350b272008-08-23 02:25:05 +0000802 }
Dan Gohmana43abd12008-09-29 21:55:50 +0000803
804 // First try normal tablegen-generated "fast" selection.
805 if (FastIS->SelectInstruction(BI))
806 continue;
807
808 // Next, try calling the target to attempt to handle the instruction.
809 if (FastIS->TargetSelectInstruction(BI))
810 continue;
811
812 // Then handle certain instructions as single-LLVM-Instruction blocks.
813 if (isa<CallInst>(BI)) {
814 if (EnableFastISelVerbose || EnableFastISelAbort) {
815 cerr << "FastISel missed call: ";
816 BI->dump();
817 }
818
819 if (BI->getType() != Type::VoidTy) {
820 unsigned &R = FuncInfo->ValueMap[BI];
821 if (!R)
822 R = FuncInfo->CreateRegForValue(BI);
823 }
824
Devang Patel390f3ac2009-04-16 01:33:10 +0000825 SDL->setCurDebugLoc(FastIS->getCurDebugLoc());
Dan Gohmana43abd12008-09-29 21:55:50 +0000826 SelectBasicBlock(LLVMBB, BI, next(BI));
Dan Gohman241f4642008-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 Gohmana43abd12008-09-29 21:55:50 +0000830 continue;
Dan Gohmanf350b272008-08-23 02:25:05 +0000831 }
Dan Gohmana43abd12008-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 Gohmanf350b272008-08-23 02:25:05 +0000846 }
847 }
848
Dan Gohmand2ff6472008-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 Gohman3df24e62008-09-03 23:12:08 +0000851 // block.
Devang Patel390f3ac2009-04-16 01:33:10 +0000852 if (BI != End) {
853 // If FastISel is run and it has known DebugLoc then use it.
854 if (FastIS && !FastIS->getCurDebugLoc().isUnknown())
855 SDL->setCurDebugLoc(FastIS->getCurDebugLoc());
Evan Cheng9f118502008-09-08 16:01:27 +0000856 SelectBasicBlock(LLVMBB, BI, End);
Devang Patel390f3ac2009-04-16 01:33:10 +0000857 }
Dan Gohmanf350b272008-08-23 02:25:05 +0000858
Dan Gohman7c3234c2008-08-27 23:52:12 +0000859 FinishBasicBlock();
Evan Cheng39fd6e82008-08-07 00:43:25 +0000860 }
Dan Gohmana43abd12008-09-29 21:55:50 +0000861
862 delete FastIS;
Dan Gohman0e5f1302008-07-07 23:02:41 +0000863}
864
Dan Gohmanfed90b62008-07-28 21:51:04 +0000865void
Dan Gohman7c3234c2008-08-27 23:52:12 +0000866SelectionDAGISel::FinishBasicBlock() {
Dan Gohmanf350b272008-08-23 02:25:05 +0000867
Dan Gohmanf350b272008-08-23 02:25:05 +0000868 DOUT << "Target-post-processed machine code:\n";
869 DEBUG(BB->dump());
Nate Begemanf15485a2006-03-27 01:32:24 +0000870
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000871 DOUT << "Total amount of phi nodes to update: "
Dan Gohman7c3234c2008-08-27 23:52:12 +0000872 << SDL->PHINodesToUpdate.size() << "\n";
873 DEBUG(for (unsigned i = 0, e = SDL->PHINodesToUpdate.size(); i != e; ++i)
874 DOUT << "Node " << i << " : (" << SDL->PHINodesToUpdate[i].first
875 << ", " << SDL->PHINodesToUpdate[i].second << ")\n";);
Nate Begemanf15485a2006-03-27 01:32:24 +0000876
Chris Lattnera33ef482005-03-30 01:10:47 +0000877 // Next, now that we know what the last MBB the LLVM BB expanded is, update
Chris Lattner1c08c712005-01-07 07:47:53 +0000878 // PHI nodes in successors.
Dan Gohman7c3234c2008-08-27 23:52:12 +0000879 if (SDL->SwitchCases.empty() &&
880 SDL->JTCases.empty() &&
881 SDL->BitTestCases.empty()) {
882 for (unsigned i = 0, e = SDL->PHINodesToUpdate.size(); i != e; ++i) {
883 MachineInstr *PHI = SDL->PHINodesToUpdate[i].first;
Nate Begemanf15485a2006-03-27 01:32:24 +0000884 assert(PHI->getOpcode() == TargetInstrInfo::PHI &&
885 "This is not a machine PHI node that we are updating!");
Dan Gohman7c3234c2008-08-27 23:52:12 +0000886 PHI->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[i].second,
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000887 false));
888 PHI->addOperand(MachineOperand::CreateMBB(BB));
Nate Begemanf15485a2006-03-27 01:32:24 +0000889 }
Dan Gohman7c3234c2008-08-27 23:52:12 +0000890 SDL->PHINodesToUpdate.clear();
Nate Begemanf15485a2006-03-27 01:32:24 +0000891 return;
Chris Lattner1c08c712005-01-07 07:47:53 +0000892 }
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000893
Dan Gohman7c3234c2008-08-27 23:52:12 +0000894 for (unsigned i = 0, e = SDL->BitTestCases.size(); i != e; ++i) {
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000895 // Lower header first, if it wasn't already lowered
Dan Gohman7c3234c2008-08-27 23:52:12 +0000896 if (!SDL->BitTestCases[i].Emitted) {
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000897 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman7c3234c2008-08-27 23:52:12 +0000898 BB = SDL->BitTestCases[i].Parent;
899 SDL->setCurrentBasicBlock(BB);
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000900 // Emit the code
Dan Gohman7c3234c2008-08-27 23:52:12 +0000901 SDL->visitBitTestHeader(SDL->BitTestCases[i]);
902 CurDAG->setRoot(SDL->getRoot());
Dan Gohmanf350b272008-08-23 02:25:05 +0000903 CodeGenAndEmitDAG();
Dan Gohman7c3234c2008-08-27 23:52:12 +0000904 SDL->clear();
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000905 }
906
Dan Gohman7c3234c2008-08-27 23:52:12 +0000907 for (unsigned j = 0, ej = SDL->BitTestCases[i].Cases.size(); j != ej; ++j) {
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000908 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman7c3234c2008-08-27 23:52:12 +0000909 BB = SDL->BitTestCases[i].Cases[j].ThisBB;
910 SDL->setCurrentBasicBlock(BB);
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000911 // Emit the code
912 if (j+1 != ej)
Dan Gohman7c3234c2008-08-27 23:52:12 +0000913 SDL->visitBitTestCase(SDL->BitTestCases[i].Cases[j+1].ThisBB,
914 SDL->BitTestCases[i].Reg,
915 SDL->BitTestCases[i].Cases[j]);
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000916 else
Dan Gohman7c3234c2008-08-27 23:52:12 +0000917 SDL->visitBitTestCase(SDL->BitTestCases[i].Default,
918 SDL->BitTestCases[i].Reg,
919 SDL->BitTestCases[i].Cases[j]);
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000920
921
Dan Gohman7c3234c2008-08-27 23:52:12 +0000922 CurDAG->setRoot(SDL->getRoot());
Dan Gohmanf350b272008-08-23 02:25:05 +0000923 CodeGenAndEmitDAG();
Dan Gohman7c3234c2008-08-27 23:52:12 +0000924 SDL->clear();
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000925 }
926
927 // Update PHI Nodes
Dan Gohman7c3234c2008-08-27 23:52:12 +0000928 for (unsigned pi = 0, pe = SDL->PHINodesToUpdate.size(); pi != pe; ++pi) {
929 MachineInstr *PHI = SDL->PHINodesToUpdate[pi].first;
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000930 MachineBasicBlock *PHIBB = PHI->getParent();
931 assert(PHI->getOpcode() == TargetInstrInfo::PHI &&
932 "This is not a machine PHI node that we are updating!");
933 // This is "default" BB. We have two jumps to it. From "header" BB and
934 // from last "case" BB.
Dan Gohman7c3234c2008-08-27 23:52:12 +0000935 if (PHIBB == SDL->BitTestCases[i].Default) {
936 PHI->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[pi].second,
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000937 false));
Dan Gohman7c3234c2008-08-27 23:52:12 +0000938 PHI->addOperand(MachineOperand::CreateMBB(SDL->BitTestCases[i].Parent));
939 PHI->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[pi].second,
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000940 false));
Dan Gohman7c3234c2008-08-27 23:52:12 +0000941 PHI->addOperand(MachineOperand::CreateMBB(SDL->BitTestCases[i].Cases.
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000942 back().ThisBB));
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000943 }
944 // One of "cases" BB.
Dan Gohman7c3234c2008-08-27 23:52:12 +0000945 for (unsigned j = 0, ej = SDL->BitTestCases[i].Cases.size();
946 j != ej; ++j) {
947 MachineBasicBlock* cBB = SDL->BitTestCases[i].Cases[j].ThisBB;
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000948 if (cBB->succ_end() !=
949 std::find(cBB->succ_begin(),cBB->succ_end(), PHIBB)) {
Dan Gohman7c3234c2008-08-27 23:52:12 +0000950 PHI->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[pi].second,
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000951 false));
952 PHI->addOperand(MachineOperand::CreateMBB(cBB));
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000953 }
954 }
955 }
956 }
Dan Gohman7c3234c2008-08-27 23:52:12 +0000957 SDL->BitTestCases.clear();
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000958
Nate Begeman9453eea2006-04-23 06:26:20 +0000959 // If the JumpTable record is filled in, then we need to emit a jump table.
960 // Updating the PHI nodes is tricky in this case, since we need to determine
961 // whether the PHI is a successor of the range check MBB or the jump table MBB
Dan Gohman7c3234c2008-08-27 23:52:12 +0000962 for (unsigned i = 0, e = SDL->JTCases.size(); i != e; ++i) {
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +0000963 // Lower header first, if it wasn't already lowered
Dan Gohman7c3234c2008-08-27 23:52:12 +0000964 if (!SDL->JTCases[i].first.Emitted) {
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +0000965 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman7c3234c2008-08-27 23:52:12 +0000966 BB = SDL->JTCases[i].first.HeaderBB;
967 SDL->setCurrentBasicBlock(BB);
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +0000968 // Emit the code
Dan Gohman7c3234c2008-08-27 23:52:12 +0000969 SDL->visitJumpTableHeader(SDL->JTCases[i].second, SDL->JTCases[i].first);
970 CurDAG->setRoot(SDL->getRoot());
Dan Gohmanf350b272008-08-23 02:25:05 +0000971 CodeGenAndEmitDAG();
Dan Gohman7c3234c2008-08-27 23:52:12 +0000972 SDL->clear();
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000973 }
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +0000974
Nate Begeman37efe672006-04-22 18:53:45 +0000975 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman7c3234c2008-08-27 23:52:12 +0000976 BB = SDL->JTCases[i].second.MBB;
977 SDL->setCurrentBasicBlock(BB);
Nate Begeman37efe672006-04-22 18:53:45 +0000978 // Emit the code
Dan Gohman7c3234c2008-08-27 23:52:12 +0000979 SDL->visitJumpTable(SDL->JTCases[i].second);
980 CurDAG->setRoot(SDL->getRoot());
Dan Gohmanf350b272008-08-23 02:25:05 +0000981 CodeGenAndEmitDAG();
Dan Gohman7c3234c2008-08-27 23:52:12 +0000982 SDL->clear();
Anton Korobeynikov3a84b9b2007-03-25 15:07:15 +0000983
Nate Begeman37efe672006-04-22 18:53:45 +0000984 // Update PHI Nodes
Dan Gohman7c3234c2008-08-27 23:52:12 +0000985 for (unsigned pi = 0, pe = SDL->PHINodesToUpdate.size(); pi != pe; ++pi) {
986 MachineInstr *PHI = SDL->PHINodesToUpdate[pi].first;
Nate Begeman37efe672006-04-22 18:53:45 +0000987 MachineBasicBlock *PHIBB = PHI->getParent();
988 assert(PHI->getOpcode() == TargetInstrInfo::PHI &&
989 "This is not a machine PHI node that we are updating!");
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000990 // "default" BB. We can go there only from header BB.
Dan Gohman7c3234c2008-08-27 23:52:12 +0000991 if (PHIBB == SDL->JTCases[i].second.Default) {
992 PHI->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[pi].second,
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000993 false));
Dan Gohman7c3234c2008-08-27 23:52:12 +0000994 PHI->addOperand(MachineOperand::CreateMBB(SDL->JTCases[i].first.HeaderBB));
Nate Begemanf4360a42006-05-03 03:48:02 +0000995 }
Anton Korobeynikov4198c582007-04-09 12:31:58 +0000996 // JT BB. Just iterate over successors here
Nate Begemanf4360a42006-05-03 03:48:02 +0000997 if (BB->succ_end() != std::find(BB->succ_begin(),BB->succ_end(), PHIBB)) {
Dan Gohman7c3234c2008-08-27 23:52:12 +0000998 PHI->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[pi].second,
Chris Lattner9ce2e9d2007-12-30 00:57:42 +0000999 false));
1000 PHI->addOperand(MachineOperand::CreateMBB(BB));
Nate Begeman37efe672006-04-22 18:53:45 +00001001 }
1002 }
Nate Begeman37efe672006-04-22 18:53:45 +00001003 }
Dan Gohman7c3234c2008-08-27 23:52:12 +00001004 SDL->JTCases.clear();
Nate Begeman37efe672006-04-22 18:53:45 +00001005
Chris Lattnerb2e806e2006-10-22 23:00:53 +00001006 // If the switch block involved a branch to one of the actual successors, we
1007 // need to update PHI nodes in that block.
Dan Gohman7c3234c2008-08-27 23:52:12 +00001008 for (unsigned i = 0, e = SDL->PHINodesToUpdate.size(); i != e; ++i) {
1009 MachineInstr *PHI = SDL->PHINodesToUpdate[i].first;
Chris Lattnerb2e806e2006-10-22 23:00:53 +00001010 assert(PHI->getOpcode() == TargetInstrInfo::PHI &&
1011 "This is not a machine PHI node that we are updating!");
1012 if (BB->isSuccessor(PHI->getParent())) {
Dan Gohman7c3234c2008-08-27 23:52:12 +00001013 PHI->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[i].second,
Chris Lattner9ce2e9d2007-12-30 00:57:42 +00001014 false));
1015 PHI->addOperand(MachineOperand::CreateMBB(BB));
Chris Lattnerb2e806e2006-10-22 23:00:53 +00001016 }
1017 }
1018
Nate Begemanf15485a2006-03-27 01:32:24 +00001019 // If we generated any switch lowering information, build and codegen any
1020 // additional DAGs necessary.
Dan Gohman7c3234c2008-08-27 23:52:12 +00001021 for (unsigned i = 0, e = SDL->SwitchCases.size(); i != e; ++i) {
Nate Begemanf15485a2006-03-27 01:32:24 +00001022 // Set the current basic block to the mbb we wish to insert the code into
Dan Gohman7c3234c2008-08-27 23:52:12 +00001023 BB = SDL->SwitchCases[i].ThisBB;
1024 SDL->setCurrentBasicBlock(BB);
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001025
Nate Begemanf15485a2006-03-27 01:32:24 +00001026 // Emit the code
Dan Gohman7c3234c2008-08-27 23:52:12 +00001027 SDL->visitSwitchCase(SDL->SwitchCases[i]);
1028 CurDAG->setRoot(SDL->getRoot());
Dan Gohmanf350b272008-08-23 02:25:05 +00001029 CodeGenAndEmitDAG();
Dan Gohman7c3234c2008-08-27 23:52:12 +00001030 SDL->clear();
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001031
1032 // Handle any PHI nodes in successors of this chunk, as if we were coming
1033 // from the original BB before switch expansion. Note that PHI nodes can
1034 // occur multiple times in PHINodesToUpdate. We have to be very careful to
1035 // handle them the right number of times.
Dan Gohman7c3234c2008-08-27 23:52:12 +00001036 while ((BB = SDL->SwitchCases[i].TrueBB)) { // Handle LHS and RHS.
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001037 for (MachineBasicBlock::iterator Phi = BB->begin();
1038 Phi != BB->end() && Phi->getOpcode() == TargetInstrInfo::PHI; ++Phi){
1039 // This value for this PHI node is recorded in PHINodesToUpdate, get it.
1040 for (unsigned pn = 0; ; ++pn) {
Dan Gohman7c3234c2008-08-27 23:52:12 +00001041 assert(pn != SDL->PHINodesToUpdate.size() &&
1042 "Didn't find PHI entry!");
1043 if (SDL->PHINodesToUpdate[pn].first == Phi) {
1044 Phi->addOperand(MachineOperand::CreateReg(SDL->PHINodesToUpdate[pn].
Chris Lattner9ce2e9d2007-12-30 00:57:42 +00001045 second, false));
Dan Gohman7c3234c2008-08-27 23:52:12 +00001046 Phi->addOperand(MachineOperand::CreateMBB(SDL->SwitchCases[i].ThisBB));
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001047 break;
1048 }
1049 }
Nate Begemanf15485a2006-03-27 01:32:24 +00001050 }
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001051
1052 // Don't process RHS if same block as LHS.
Dan Gohman7c3234c2008-08-27 23:52:12 +00001053 if (BB == SDL->SwitchCases[i].FalseBB)
1054 SDL->SwitchCases[i].FalseBB = 0;
Chris Lattnerd5e93c02006-09-07 01:59:34 +00001055
1056 // If we haven't handled the RHS, do so now. Otherwise, we're done.
Dan Gohman7c3234c2008-08-27 23:52:12 +00001057 SDL->SwitchCases[i].TrueBB = SDL->SwitchCases[i].FalseBB;
1058 SDL->SwitchCases[i].FalseBB = 0;
Nate Begemanf15485a2006-03-27 01:32:24 +00001059 }
Dan Gohman7c3234c2008-08-27 23:52:12 +00001060 assert(SDL->SwitchCases[i].TrueBB == 0 && SDL->SwitchCases[i].FalseBB == 0);
Chris Lattnera33ef482005-03-30 01:10:47 +00001061 }
Dan Gohman7c3234c2008-08-27 23:52:12 +00001062 SDL->SwitchCases.clear();
1063
1064 SDL->PHINodesToUpdate.clear();
Chris Lattner1c08c712005-01-07 07:47:53 +00001065}
Evan Chenga9c20912006-01-21 02:32:06 +00001066
Jim Laskey13ec7022006-08-01 14:21:23 +00001067
Dan Gohman0a3776d2009-02-06 18:26:51 +00001068/// Create the scheduler. If a specific scheduler was specified
1069/// via the SchedulerRegistry, use it, otherwise select the
1070/// one preferred by the target.
Dan Gohman5e843682008-07-14 18:19:29 +00001071///
Dan Gohman47ac0f02009-02-11 04:27:20 +00001072ScheduleDAGSDNodes *SelectionDAGISel::CreateScheduler() {
Jim Laskeyeb577ba2006-08-02 12:30:23 +00001073 RegisterScheduler::FunctionPassCtor Ctor = RegisterScheduler::getDefault();
Jim Laskey13ec7022006-08-01 14:21:23 +00001074
1075 if (!Ctor) {
Jim Laskeyeb577ba2006-08-02 12:30:23 +00001076 Ctor = ISHeuristic;
Jim Laskey9373beb2006-08-01 19:14:14 +00001077 RegisterScheduler::setDefault(Ctor);
Evan Cheng4ef10862006-01-23 07:01:07 +00001078 }
Jim Laskey13ec7022006-08-01 14:21:23 +00001079
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +00001080 return Ctor(this, OptLevel);
Evan Chenga9c20912006-01-21 02:32:06 +00001081}
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001082
Dan Gohmanfc54c552009-01-15 22:18:12 +00001083ScheduleHazardRecognizer *SelectionDAGISel::CreateTargetHazardRecognizer() {
1084 return new ScheduleHazardRecognizer();
Jim Laskey9ff542f2006-08-01 18:29:48 +00001085}
1086
Chris Lattner75548062006-10-11 03:58:02 +00001087//===----------------------------------------------------------------------===//
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 Gohman475871a2008-07-27 21:46:04 +00001096bool SelectionDAGISel::CheckAndMask(SDValue LHS, ConstantSDNode *RHS,
Dan Gohmandc9b3d02007-07-24 23:00:27 +00001097 int64_t DesiredMaskS) const {
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001098 const APInt &ActualMask = RHS->getAPIntValue();
1099 const APInt &DesiredMask = APInt(LHS.getValueSizeInBits(), DesiredMaskS);
Chris Lattner75548062006-10-11 03:58:02 +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 Gohman2e68b6f2008-02-25 21:11:39 +00001106 if (ActualMask.intersects(~DesiredMask))
Chris Lattner75548062006-10-11 03:58:02 +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 Gohman2e68b6f2008-02-25 21:11:39 +00001111 APInt NeededMask = DesiredMask & ~ActualMask;
Dan Gohmanea859be2007-06-22 14:59:07 +00001112 if (CurDAG->MaskedValueIsZero(LHS, NeededMask))
Chris Lattner75548062006-10-11 03:58:02 +00001113 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 Gohman475871a2008-07-27 21:46:04 +00001125bool SelectionDAGISel::CheckOrMask(SDValue LHS, ConstantSDNode *RHS,
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001126 int64_t DesiredMaskS) const {
1127 const APInt &ActualMask = RHS->getAPIntValue();
1128 const APInt &DesiredMask = APInt(LHS.getValueSizeInBits(), DesiredMaskS);
Chris Lattner75548062006-10-11 03:58:02 +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 Gohman2e68b6f2008-02-25 21:11:39 +00001135 if (ActualMask.intersects(~DesiredMask))
Chris Lattner75548062006-10-11 03:58:02 +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 Gohman2e68b6f2008-02-25 21:11:39 +00001140 APInt NeededMask = DesiredMask & ~ActualMask;
Chris Lattner75548062006-10-11 03:58:02 +00001141
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001142 APInt KnownZero, KnownOne;
Dan Gohmanea859be2007-06-22 14:59:07 +00001143 CurDAG->ComputeMaskedBits(LHS, NeededMask, KnownZero, KnownOne);
Chris Lattner75548062006-10-11 03:58:02 +00001144
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
Jim Laskey9ff542f2006-08-01 18:29:48 +00001155
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001156/// SelectInlineAsmMemoryOperands - Calls to this are automatically generated
1157/// by tblgen. Others should not call it.
1158void SelectionDAGISel::
Dan Gohmanf350b272008-08-23 02:25:05 +00001159SelectInlineAsmMemoryOperands(std::vector<SDValue> &Ops) {
Dan Gohman475871a2008-07-27 21:46:04 +00001160 std::vector<SDValue> InOps;
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001161 std::swap(InOps, Ops);
1162
1163 Ops.push_back(InOps[0]); // input chain.
1164 Ops.push_back(InOps[1]); // input asm string.
1165
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001166 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 Gohmanf5aeb1a2008-09-12 16:56:44 +00001171 unsigned Flags = cast<ConstantSDNode>(InOps[i])->getZExtValue();
Dale Johannesen86b49f82008-09-24 01:07:17 +00001172 if ((Flags & 7) != 4 /*MEM*/) {
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001173 // Just skip over this operand, copying the operands verbatim.
Evan Cheng697cbbf2009-03-20 18:03:34 +00001174 Ops.insert(Ops.end(), InOps.begin()+i,
1175 InOps.begin()+i+InlineAsm::getNumOperandRegisters(Flags) + 1);
1176 i += InlineAsm::getNumOperandRegisters(Flags) + 1;
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001177 } else {
Evan Cheng697cbbf2009-03-20 18:03:34 +00001178 assert(InlineAsm::getNumOperandRegisters(Flags) == 1 &&
1179 "Memory operand with multiple values?");
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001180 // Otherwise, this is a memory operand. Ask the target to select it.
Dan Gohman475871a2008-07-27 21:46:04 +00001181 std::vector<SDValue> SelOps;
Dan Gohmanf350b272008-08-23 02:25:05 +00001182 if (SelectInlineAsmMemoryOperand(InOps[i+1], 'm', SelOps)) {
Bill Wendling832171c2006-12-07 20:04:42 +00001183 cerr << "Could not match memory address. Inline asm failure!\n";
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001184 exit(1);
1185 }
1186
1187 // Add this to the output node.
Dan Gohmanf350b272008-08-23 02:25:05 +00001188 MVT IntPtrTy = CurDAG->getTargetLoweringInfo().getPointerTy();
Dale Johannesen86b49f82008-09-24 01:07:17 +00001189 Ops.push_back(CurDAG->getTargetConstant(4/*MEM*/ | (SelOps.size()<< 3),
Dan Gohmanf350b272008-08-23 02:25:05 +00001190 IntPtrTy));
Chris Lattner0e43f2b2006-02-24 02:13:54 +00001191 Ops.insert(Ops.end(), SelOps.begin(), SelOps.end());
1192 i += 2;
1193 }
1194 }
1195
1196 // Add the flag input back if present.
1197 if (e != InOps.size())
1198 Ops.push_back(InOps.back());
1199}
Devang Patel794fd752007-05-01 21:15:47 +00001200
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +00001201/// findFlagUse - Return use of MVT::Flag value produced by the specified
1202/// SDNode.
1203///
1204static SDNode *findFlagUse(SDNode *N) {
1205 unsigned FlagResNo = N->getNumValues()-1;
1206 for (SDNode::use_iterator I = N->use_begin(), E = N->use_end(); I != E; ++I) {
1207 SDUse &Use = I.getUse();
1208 if (Use.getResNo() == FlagResNo)
1209 return Use.getUser();
1210 }
1211 return NULL;
1212}
1213
1214/// findNonImmUse - Return true if "Use" is a non-immediate use of "Def".
1215/// This function recursively traverses up the operand chain, ignoring
1216/// certain nodes.
1217static bool findNonImmUse(SDNode *Use, SDNode* Def, SDNode *ImmedUse,
1218 SDNode *Root,
1219 SmallPtrSet<SDNode*, 16> &Visited) {
1220 if (Use->getNodeId() < Def->getNodeId() ||
1221 !Visited.insert(Use))
1222 return false;
1223
1224 for (unsigned i = 0, e = Use->getNumOperands(); i != e; ++i) {
1225 SDNode *N = Use->getOperand(i).getNode();
1226 if (N == Def) {
1227 if (Use == ImmedUse || Use == Root)
1228 continue; // We are not looking for immediate use.
1229 assert(N != Root);
1230 return true;
1231 }
1232
1233 // Traverse up the operand chain.
1234 if (findNonImmUse(N, Def, ImmedUse, Root, Visited))
1235 return true;
1236 }
1237 return false;
1238}
1239
1240/// isNonImmUse - Start searching from Root up the DAG to check is Def can
1241/// be reached. Return true if that's the case. However, ignore direct uses
1242/// by ImmedUse (which would be U in the example illustrated in
1243/// IsLegalAndProfitableToFold) and by Root (which can happen in the store
1244/// case).
1245/// FIXME: to be really generic, we should allow direct use by any node
1246/// that is being folded. But realisticly since we only fold loads which
1247/// have one non-chain use, we only need to watch out for load/op/store
1248/// and load/op/cmp case where the root (store / cmp) may reach the load via
1249/// its chain operand.
1250static inline bool isNonImmUse(SDNode *Root, SDNode *Def, SDNode *ImmedUse) {
1251 SmallPtrSet<SDNode*, 16> Visited;
1252 return findNonImmUse(Root, Def, ImmedUse, Root, Visited);
1253}
1254
1255/// IsLegalAndProfitableToFold - Returns true if the specific operand node N of
1256/// U can be folded during instruction selection that starts at Root and
1257/// folding N is profitable.
1258bool SelectionDAGISel::IsLegalAndProfitableToFold(SDNode *N, SDNode *U,
1259 SDNode *Root) const {
1260 if (OptLevel == CodeGenOpt::None) return false;
1261
1262 // If Root use can somehow reach N through a path that that doesn't contain
1263 // U then folding N would create a cycle. e.g. In the following
1264 // diagram, Root can reach N through X. If N is folded into into Root, then
1265 // X is both a predecessor and a successor of U.
1266 //
1267 // [N*] //
1268 // ^ ^ //
1269 // / \ //
1270 // [U*] [X]? //
1271 // ^ ^ //
1272 // \ / //
1273 // \ / //
1274 // [Root*] //
1275 //
1276 // * indicates nodes to be folded together.
1277 //
1278 // If Root produces a flag, then it gets (even more) interesting. Since it
1279 // will be "glued" together with its flag use in the scheduler, we need to
1280 // check if it might reach N.
1281 //
1282 // [N*] //
1283 // ^ ^ //
1284 // / \ //
1285 // [U*] [X]? //
1286 // ^ ^ //
1287 // \ \ //
1288 // \ | //
1289 // [Root*] | //
1290 // ^ | //
1291 // f | //
1292 // | / //
1293 // [Y] / //
1294 // ^ / //
1295 // f / //
1296 // | / //
1297 // [FU] //
1298 //
1299 // If FU (flag use) indirectly reaches N (the load), and Root folds N
1300 // (call it Fold), then X is a predecessor of FU and a successor of
1301 // Fold. But since Fold and FU are flagged together, this will create
1302 // a cycle in the scheduling graph.
1303
1304 MVT VT = Root->getValueType(Root->getNumValues()-1);
1305 while (VT == MVT::Flag) {
1306 SDNode *FU = findFlagUse(Root);
1307 if (FU == NULL)
1308 break;
1309 Root = FU;
1310 VT = Root->getValueType(Root->getNumValues()-1);
1311 }
1312
1313 return !isNonImmUse(Root, N, U);
1314}
1315
1316
Devang Patel19974732007-05-03 01:11:54 +00001317char SelectionDAGISel::ID = 0;