blob: a827187e357e524d9d2d154cd5a47ffdf70f3709 [file] [log] [blame]
Dan Gohman23785a12008-08-12 17:42:33 +00001//===----- ScheduleDAGRRList.cpp - Reg pressure reduction list scheduler --===//
Evan Chengd38c22b2006-05-11 23:55:42 +00002//
3// The LLVM Compiler Infrastructure
4//
Chris Lattnerf3ebc3f2007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Evan Chengd38c22b2006-05-11 23:55:42 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This implements bottom-up and top-down register pressure reduction list
11// schedulers, using standard algorithms. The basic approach uses a priority
12// queue of available nodes to schedule. One at a time, nodes are taken from
13// the priority queue (thus in priority order), checked for legality to
14// schedule, and emitted if legal.
15//
16//===----------------------------------------------------------------------===//
17
Dale Johannesen2182f062007-07-13 17:13:54 +000018#define DEBUG_TYPE "pre-RA-sched"
Dan Gohman483377c2009-02-06 17:22:58 +000019#include "ScheduleDAGSDNodes.h"
Chris Lattner3b9f02a2010-04-07 05:20:54 +000020#include "llvm/InlineAsm.h"
Jim Laskey29e635d2006-08-02 12:30:23 +000021#include "llvm/CodeGen/SchedulerRegistry.h"
Dan Gohman619ef482009-01-15 19:20:50 +000022#include "llvm/CodeGen/SelectionDAGISel.h"
Andrew Trick10ffc2b2010-12-24 05:03:26 +000023#include "llvm/CodeGen/ScheduleHazardRecognizer.h"
Dan Gohman3a4be0f2008-02-10 18:45:23 +000024#include "llvm/Target/TargetRegisterInfo.h"
Owen Anderson8c2c1e92006-05-12 06:33:49 +000025#include "llvm/Target/TargetData.h"
Evan Chengd38c22b2006-05-11 23:55:42 +000026#include "llvm/Target/TargetMachine.h"
27#include "llvm/Target/TargetInstrInfo.h"
Evan Chenga77f3d32010-07-21 06:09:07 +000028#include "llvm/Target/TargetLowering.h"
Evan Cheng5924bf72007-09-25 01:54:36 +000029#include "llvm/ADT/SmallSet.h"
Evan Chengd38c22b2006-05-11 23:55:42 +000030#include "llvm/ADT/Statistic.h"
Roman Levenstein6b371142008-04-29 09:07:59 +000031#include "llvm/ADT/STLExtras.h"
Chris Lattner3b9f02a2010-04-07 05:20:54 +000032#include "llvm/Support/Debug.h"
33#include "llvm/Support/ErrorHandling.h"
Chris Lattner4dc3edd2009-08-23 06:35:02 +000034#include "llvm/Support/raw_ostream.h"
Evan Chengd38c22b2006-05-11 23:55:42 +000035#include <climits>
Evan Chengd38c22b2006-05-11 23:55:42 +000036using namespace llvm;
37
Dan Gohmanfd227e92008-03-25 17:10:29 +000038STATISTIC(NumBacktracks, "Number of times scheduler backtracked");
Evan Cheng79e97132007-10-05 01:39:18 +000039STATISTIC(NumUnfolds, "Number of nodes unfolded");
Evan Cheng1ec79b42007-09-27 07:09:03 +000040STATISTIC(NumDups, "Number of duplicated nodes");
Evan Chengb2c42c62009-01-12 03:19:55 +000041STATISTIC(NumPRCopies, "Number of physical register copies");
Evan Cheng1ec79b42007-09-27 07:09:03 +000042
Jim Laskey95eda5b2006-08-01 14:21:23 +000043static RegisterScheduler
44 burrListDAGScheduler("list-burr",
Dan Gohman9c4b7d52008-10-14 20:25:08 +000045 "Bottom-up register reduction list scheduling",
Jim Laskey95eda5b2006-08-01 14:21:23 +000046 createBURRListDAGScheduler);
47static RegisterScheduler
48 tdrListrDAGScheduler("list-tdrr",
Dan Gohman9c4b7d52008-10-14 20:25:08 +000049 "Top-down register reduction list scheduling",
Jim Laskey95eda5b2006-08-01 14:21:23 +000050 createTDRRListDAGScheduler);
Bill Wendling8cbc25d2010-01-23 10:26:57 +000051static RegisterScheduler
52 sourceListDAGScheduler("source",
53 "Similar to list-burr but schedules in source "
54 "order when possible",
55 createSourceListDAGScheduler);
Jim Laskey95eda5b2006-08-01 14:21:23 +000056
Evan Chengbdd062d2010-05-20 06:13:19 +000057static RegisterScheduler
Evan Cheng725211e2010-05-21 00:42:32 +000058 hybridListDAGScheduler("list-hybrid",
Evan Cheng37b740c2010-07-24 00:39:05 +000059 "Bottom-up register pressure aware list scheduling "
60 "which tries to balance latency and register pressure",
Evan Chengbdd062d2010-05-20 06:13:19 +000061 createHybridListDAGScheduler);
62
Evan Cheng37b740c2010-07-24 00:39:05 +000063static RegisterScheduler
64 ILPListDAGScheduler("list-ilp",
65 "Bottom-up register pressure aware list scheduling "
66 "which tries to balance ILP and register pressure",
67 createILPListDAGScheduler);
68
Andrew Trick47ff14b2011-01-21 05:51:33 +000069static cl::opt<bool> DisableSchedCycles(
Andrew Trickbd428ec2011-01-21 06:19:05 +000070 "disable-sched-cycles", cl::Hidden, cl::init(false),
Andrew Trick47ff14b2011-01-21 05:51:33 +000071 cl::desc("Disable cycle-level precision during preRA scheduling"));
Andrew Trick10ffc2b2010-12-24 05:03:26 +000072
Andrew Trick641e2d42011-03-05 08:00:22 +000073// Temporary sched=list-ilp flags until the heuristics are robust.
Andrew Trickbfbd9722011-04-14 05:15:06 +000074// Some options are also available under sched=list-hybrid.
Andrew Trick641e2d42011-03-05 08:00:22 +000075static cl::opt<bool> DisableSchedRegPressure(
76 "disable-sched-reg-pressure", cl::Hidden, cl::init(false),
77 cl::desc("Disable regpressure priority in sched=list-ilp"));
78static cl::opt<bool> DisableSchedLiveUses(
Andrew Trickdd017322011-03-06 00:03:32 +000079 "disable-sched-live-uses", cl::Hidden, cl::init(true),
Andrew Trick641e2d42011-03-05 08:00:22 +000080 cl::desc("Disable live use priority in sched=list-ilp"));
Andrew Trick2ad0b372011-04-07 19:54:57 +000081static cl::opt<bool> DisableSchedVRegCycle(
82 "disable-sched-vrcycle", cl::Hidden, cl::init(false),
83 cl::desc("Disable virtual register cycle interference checks"));
Andrew Trickbfbd9722011-04-14 05:15:06 +000084static cl::opt<bool> DisableSchedPhysRegJoin(
85 "disable-sched-physreg-join", cl::Hidden, cl::init(false),
86 cl::desc("Disable physreg def-use affinity"));
Andrew Trick641e2d42011-03-05 08:00:22 +000087static cl::opt<bool> DisableSchedStalls(
Andrew Trickdd017322011-03-06 00:03:32 +000088 "disable-sched-stalls", cl::Hidden, cl::init(true),
Andrew Trick641e2d42011-03-05 08:00:22 +000089 cl::desc("Disable no-stall priority in sched=list-ilp"));
90static cl::opt<bool> DisableSchedCriticalPath(
91 "disable-sched-critical-path", cl::Hidden, cl::init(false),
92 cl::desc("Disable critical path priority in sched=list-ilp"));
93static cl::opt<bool> DisableSchedHeight(
94 "disable-sched-height", cl::Hidden, cl::init(false),
95 cl::desc("Disable scheduled-height priority in sched=list-ilp"));
96
97static cl::opt<int> MaxReorderWindow(
98 "max-sched-reorder", cl::Hidden, cl::init(6),
99 cl::desc("Number of instructions to allow ahead of the critical path "
100 "in sched=list-ilp"));
101
102static cl::opt<unsigned> AvgIPC(
103 "sched-avg-ipc", cl::Hidden, cl::init(1),
104 cl::desc("Average inst/cycle whan no target itinerary exists."));
105
106#ifndef NDEBUG
107namespace {
108 // For sched=list-ilp, Count the number of times each factor comes into play.
Andrew Trickb53a00d2011-04-13 00:38:32 +0000109 enum { FactPressureDiff, FactRegUses, FactStall, FactHeight, FactDepth,
110 FactStatic, FactOther, NumFactors };
Andrew Trick641e2d42011-03-05 08:00:22 +0000111}
112static const char *FactorName[NumFactors] =
Andrew Trickb53a00d2011-04-13 00:38:32 +0000113{"PressureDiff", "RegUses", "Stall", "Height", "Depth","Static", "Other"};
Andrew Trick641e2d42011-03-05 08:00:22 +0000114static int FactorCount[NumFactors];
115#endif //!NDEBUG
116
Evan Chengd38c22b2006-05-11 23:55:42 +0000117namespace {
Evan Chengd38c22b2006-05-11 23:55:42 +0000118//===----------------------------------------------------------------------===//
119/// ScheduleDAGRRList - The actual register reduction list scheduler
120/// implementation. This supports both top-down and bottom-up scheduling.
121///
Nick Lewycky02d5f772009-10-25 06:33:48 +0000122class ScheduleDAGRRList : public ScheduleDAGSDNodes {
Evan Chengd38c22b2006-05-11 23:55:42 +0000123private:
124 /// isBottomUp - This is true if the scheduling problem is bottom-up, false if
125 /// it is top-down.
126 bool isBottomUp;
Evan Cheng2c977312008-07-01 18:05:03 +0000127
Evan Chengbdd062d2010-05-20 06:13:19 +0000128 /// NeedLatency - True if the scheduler will make use of latency information.
129 ///
130 bool NeedLatency;
131
Evan Chengd38c22b2006-05-11 23:55:42 +0000132 /// AvailableQueue - The priority queue to use for the available SUnits.
Evan Chengd38c22b2006-05-11 23:55:42 +0000133 SchedulingPriorityQueue *AvailableQueue;
134
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000135 /// PendingQueue - This contains all of the instructions whose operands have
136 /// been issued, but their results are not ready yet (due to the latency of
137 /// the operation). Once the operands becomes available, the instruction is
138 /// added to the AvailableQueue.
139 std::vector<SUnit*> PendingQueue;
140
141 /// HazardRec - The hazard recognizer to use.
142 ScheduleHazardRecognizer *HazardRec;
143
Andrew Trick528fad92010-12-23 05:42:20 +0000144 /// CurCycle - The current scheduler state corresponds to this cycle.
145 unsigned CurCycle;
146
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000147 /// MinAvailableCycle - Cycle of the soonest available instruction.
148 unsigned MinAvailableCycle;
149
Andrew Trick641e2d42011-03-05 08:00:22 +0000150 /// IssueCount - Count instructions issued in this cycle
151 /// Currently valid only for bottom-up scheduling.
152 unsigned IssueCount;
153
Dan Gohmanc07f6862008-09-23 18:50:48 +0000154 /// LiveRegDefs - A set of physical registers and their definition
Evan Cheng5924bf72007-09-25 01:54:36 +0000155 /// that are "live". These nodes must be scheduled before any other nodes that
156 /// modifies the registers can be scheduled.
Dan Gohmanc07f6862008-09-23 18:50:48 +0000157 unsigned NumLiveRegs;
Evan Cheng5924bf72007-09-25 01:54:36 +0000158 std::vector<SUnit*> LiveRegDefs;
Andrew Tricka52f3252010-12-23 04:16:14 +0000159 std::vector<SUnit*> LiveRegGens;
Evan Cheng5924bf72007-09-25 01:54:36 +0000160
Dan Gohmanad2134d2008-11-25 00:52:40 +0000161 /// Topo - A topological ordering for SUnits which permits fast IsReachable
162 /// and similar queries.
163 ScheduleDAGTopologicalSort Topo;
164
Evan Chengd38c22b2006-05-11 23:55:42 +0000165public:
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000166 ScheduleDAGRRList(MachineFunction &mf, bool needlatency,
167 SchedulingPriorityQueue *availqueue,
168 CodeGenOpt::Level OptLevel)
169 : ScheduleDAGSDNodes(mf), isBottomUp(availqueue->isBottomUp()),
170 NeedLatency(needlatency), AvailableQueue(availqueue), CurCycle(0),
171 Topo(SUnits) {
172
173 const TargetMachine &tm = mf.getTarget();
Andrew Trick47ff14b2011-01-21 05:51:33 +0000174 if (DisableSchedCycles || !NeedLatency)
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000175 HazardRec = new ScheduleHazardRecognizer();
Andrew Trick47ff14b2011-01-21 05:51:33 +0000176 else
177 HazardRec = tm.getInstrInfo()->CreateTargetHazardRecognizer(&tm, this);
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000178 }
Evan Chengd38c22b2006-05-11 23:55:42 +0000179
180 ~ScheduleDAGRRList() {
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000181 delete HazardRec;
Evan Chengd38c22b2006-05-11 23:55:42 +0000182 delete AvailableQueue;
183 }
184
185 void Schedule();
186
Andrew Trick9ccce772011-01-14 21:11:41 +0000187 ScheduleHazardRecognizer *getHazardRec() { return HazardRec; }
188
Roman Levenstein733a4d62008-03-26 11:23:38 +0000189 /// IsReachable - Checks if SU is reachable from TargetSU.
Dan Gohmanad2134d2008-11-25 00:52:40 +0000190 bool IsReachable(const SUnit *SU, const SUnit *TargetSU) {
191 return Topo.IsReachable(SU, TargetSU);
192 }
Roman Levenstein7e71b4b2008-03-26 09:18:09 +0000193
Dan Gohman60d68442009-01-29 19:49:27 +0000194 /// WillCreateCycle - Returns true if adding an edge from SU to TargetSU will
Roman Levenstein7e71b4b2008-03-26 09:18:09 +0000195 /// create a cycle.
Dan Gohmanad2134d2008-11-25 00:52:40 +0000196 bool WillCreateCycle(SUnit *SU, SUnit *TargetSU) {
197 return Topo.WillCreateCycle(SU, TargetSU);
198 }
Roman Levenstein7e71b4b2008-03-26 09:18:09 +0000199
Dan Gohman2d170892008-12-09 22:54:47 +0000200 /// AddPred - adds a predecessor edge to SUnit SU.
Roman Levenstein733a4d62008-03-26 11:23:38 +0000201 /// This returns true if this is a new predecessor.
202 /// Updates the topological ordering if required.
Dan Gohman17214e62008-12-16 01:00:55 +0000203 void AddPred(SUnit *SU, const SDep &D) {
Dan Gohman2d170892008-12-09 22:54:47 +0000204 Topo.AddPred(SU, D.getSUnit());
Dan Gohman17214e62008-12-16 01:00:55 +0000205 SU->addPred(D);
Dan Gohmanad2134d2008-11-25 00:52:40 +0000206 }
Roman Levenstein7e71b4b2008-03-26 09:18:09 +0000207
Dan Gohman2d170892008-12-09 22:54:47 +0000208 /// RemovePred - removes a predecessor edge from SUnit SU.
209 /// This returns true if an edge was removed.
210 /// Updates the topological ordering if required.
Dan Gohman17214e62008-12-16 01:00:55 +0000211 void RemovePred(SUnit *SU, const SDep &D) {
Dan Gohman2d170892008-12-09 22:54:47 +0000212 Topo.RemovePred(SU, D.getSUnit());
Dan Gohman17214e62008-12-16 01:00:55 +0000213 SU->removePred(D);
Dan Gohmanad2134d2008-11-25 00:52:40 +0000214 }
Roman Levenstein7e71b4b2008-03-26 09:18:09 +0000215
Evan Chengd38c22b2006-05-11 23:55:42 +0000216private:
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000217 bool isReady(SUnit *SU) {
Andrew Trick47ff14b2011-01-21 05:51:33 +0000218 return DisableSchedCycles || !AvailableQueue->hasReadyFilter() ||
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000219 AvailableQueue->isReady(SU);
220 }
221
Dan Gohman60d68442009-01-29 19:49:27 +0000222 void ReleasePred(SUnit *SU, const SDep *PredEdge);
Andrew Tricka52f3252010-12-23 04:16:14 +0000223 void ReleasePredecessors(SUnit *SU);
Dan Gohman60d68442009-01-29 19:49:27 +0000224 void ReleaseSucc(SUnit *SU, const SDep *SuccEdge);
Dan Gohmanb9543432009-02-10 23:27:53 +0000225 void ReleaseSuccessors(SUnit *SU);
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000226 void ReleasePending();
227 void AdvanceToCycle(unsigned NextCycle);
228 void AdvancePastStalls(SUnit *SU);
229 void EmitNode(SUnit *SU);
Andrew Trick528fad92010-12-23 05:42:20 +0000230 void ScheduleNodeBottomUp(SUnit*);
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000231 void CapturePred(SDep *PredEdge);
Evan Cheng8e136a92007-09-26 21:36:17 +0000232 void UnscheduleNodeBottomUp(SUnit*);
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000233 void RestoreHazardCheckerBottomUp();
234 void BacktrackBottomUp(SUnit*, SUnit*);
Evan Cheng8e136a92007-09-26 21:36:17 +0000235 SUnit *CopyAndMoveSuccessors(SUnit*);
Evan Chengb2c42c62009-01-12 03:19:55 +0000236 void InsertCopiesAndMoveSuccs(SUnit*, unsigned,
237 const TargetRegisterClass*,
238 const TargetRegisterClass*,
239 SmallVector<SUnit*, 2>&);
Evan Cheng1ec79b42007-09-27 07:09:03 +0000240 bool DelayForLiveRegsBottomUp(SUnit*, SmallVector<unsigned, 4>&);
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000241
Andrew Trick528fad92010-12-23 05:42:20 +0000242 SUnit *PickNodeToScheduleBottomUp();
Evan Chengd38c22b2006-05-11 23:55:42 +0000243 void ListScheduleBottomUp();
Roman Levenstein7e71b4b2008-03-26 09:18:09 +0000244
Andrew Trick528fad92010-12-23 05:42:20 +0000245 void ScheduleNodeTopDown(SUnit*);
246 void ListScheduleTopDown();
247
Roman Levenstein7e71b4b2008-03-26 09:18:09 +0000248
249 /// CreateNewSUnit - Creates a new SUnit and returns a pointer to it.
Roman Levenstein733a4d62008-03-26 11:23:38 +0000250 /// Updates the topological ordering if required.
Roman Levenstein7e71b4b2008-03-26 09:18:09 +0000251 SUnit *CreateNewSUnit(SDNode *N) {
Dan Gohmanad2134d2008-11-25 00:52:40 +0000252 unsigned NumSUnits = SUnits.size();
Roman Levenstein7e71b4b2008-03-26 09:18:09 +0000253 SUnit *NewNode = NewSUnit(N);
Roman Levenstein733a4d62008-03-26 11:23:38 +0000254 // Update the topological ordering.
Dan Gohmanad2134d2008-11-25 00:52:40 +0000255 if (NewNode->NodeNum >= NumSUnits)
256 Topo.InitDAGTopologicalSorting();
Roman Levenstein7e71b4b2008-03-26 09:18:09 +0000257 return NewNode;
258 }
259
Roman Levenstein733a4d62008-03-26 11:23:38 +0000260 /// CreateClone - Creates a new SUnit from an existing one.
261 /// Updates the topological ordering if required.
Roman Levenstein7e71b4b2008-03-26 09:18:09 +0000262 SUnit *CreateClone(SUnit *N) {
Dan Gohmanad2134d2008-11-25 00:52:40 +0000263 unsigned NumSUnits = SUnits.size();
Roman Levenstein7e71b4b2008-03-26 09:18:09 +0000264 SUnit *NewNode = Clone(N);
Roman Levenstein733a4d62008-03-26 11:23:38 +0000265 // Update the topological ordering.
Dan Gohmanad2134d2008-11-25 00:52:40 +0000266 if (NewNode->NodeNum >= NumSUnits)
267 Topo.InitDAGTopologicalSorting();
Roman Levenstein7e71b4b2008-03-26 09:18:09 +0000268 return NewNode;
269 }
Dan Gohmandddc1ac2008-12-16 03:25:46 +0000270
Evan Chengbdd062d2010-05-20 06:13:19 +0000271 /// ForceUnitLatencies - Register-pressure-reducing scheduling doesn't
272 /// need actual latency information but the hybrid scheduler does.
273 bool ForceUnitLatencies() const {
274 return !NeedLatency;
275 }
Evan Chengd38c22b2006-05-11 23:55:42 +0000276};
277} // end anonymous namespace
278
Owen Anderson96adc4a2011-06-15 23:35:18 +0000279/// GetCostForDef - Looks up the register class and cost for a given definition.
280/// Typically this just means looking up the representative register class,
281/// but for untyped values (MVT::untyped) it means inspecting the node's
282/// opcode to determine what register class is being generated.
283static void GetCostForDef(const ScheduleDAGSDNodes::RegDefIter &RegDefPos,
284 const TargetLowering *TLI,
285 const TargetInstrInfo *TII,
286 const TargetRegisterInfo *TRI,
287 unsigned &RegClass, unsigned &Cost) {
288 EVT VT = RegDefPos.GetValue();
289
290 // Special handling for untyped values. These values can only come from
291 // the expansion of custom DAG-to-DAG patterns.
292 if (VT == MVT::untyped) {
Owen Andersond1955e72011-06-21 22:54:23 +0000293 const SDNode *Node = RegDefPos.GetNode();
294 unsigned Opcode = Node->getMachineOpcode();
295
296 if (Opcode == TargetOpcode::REG_SEQUENCE) {
297 unsigned DstRCIdx = cast<ConstantSDNode>(Node->getOperand(0))->getZExtValue();
298 const TargetRegisterClass *RC = TRI->getRegClass(DstRCIdx);
299 RegClass = RC->getID();
300 Cost = 1;
301 return;
302 }
303
Owen Anderson96adc4a2011-06-15 23:35:18 +0000304 unsigned Idx = RegDefPos.GetIdx();
305 const TargetInstrDesc Desc = TII->get(Opcode);
306 const TargetRegisterClass *RC = Desc.getRegClass(Idx, TRI);
307 RegClass = RC->getID();
308 // FIXME: Cost arbitrarily set to 1 because there doesn't seem to be a
309 // better way to determine it.
310 Cost = 1;
311 } else {
312 RegClass = TLI->getRepRegClassFor(VT)->getID();
313 Cost = TLI->getRepRegClassCostFor(VT);
314 }
315}
Evan Chengd38c22b2006-05-11 23:55:42 +0000316
317/// Schedule - Schedule the DAG using list scheduling.
318void ScheduleDAGRRList::Schedule() {
Evan Chenga77f3d32010-07-21 06:09:07 +0000319 DEBUG(dbgs()
320 << "********** List Scheduling BB#" << BB->getNumber()
Evan Cheng6c1414f2010-10-29 18:09:28 +0000321 << " '" << BB->getName() << "' **********\n");
Andrew Trick641e2d42011-03-05 08:00:22 +0000322#ifndef NDEBUG
323 for (int i = 0; i < NumFactors; ++i) {
324 FactorCount[i] = 0;
325 }
326#endif //!NDEBUG
Evan Cheng5924bf72007-09-25 01:54:36 +0000327
Andrew Trick528fad92010-12-23 05:42:20 +0000328 CurCycle = 0;
Andrew Trick641e2d42011-03-05 08:00:22 +0000329 IssueCount = 0;
Andrew Trick47ff14b2011-01-21 05:51:33 +0000330 MinAvailableCycle = DisableSchedCycles ? 0 : UINT_MAX;
Dan Gohmanc07f6862008-09-23 18:50:48 +0000331 NumLiveRegs = 0;
Andrew Trick2085a962010-12-21 22:25:04 +0000332 LiveRegDefs.resize(TRI->getNumRegs(), NULL);
Andrew Tricka52f3252010-12-23 04:16:14 +0000333 LiveRegGens.resize(TRI->getNumRegs(), NULL);
Evan Cheng5924bf72007-09-25 01:54:36 +0000334
Dan Gohman04543e72008-12-23 18:36:58 +0000335 // Build the scheduling graph.
Dan Gohman918ec532009-10-09 23:33:48 +0000336 BuildSchedGraph(NULL);
Evan Chengd38c22b2006-05-11 23:55:42 +0000337
Evan Chengd38c22b2006-05-11 23:55:42 +0000338 DEBUG(for (unsigned su = 0, e = SUnits.size(); su != e; ++su)
Dan Gohman22d07b12008-11-18 02:06:40 +0000339 SUnits[su].dumpAll(this));
Dan Gohmanad2134d2008-11-25 00:52:40 +0000340 Topo.InitDAGTopologicalSorting();
Evan Chengd38c22b2006-05-11 23:55:42 +0000341
Dan Gohman46520a22008-06-21 19:18:17 +0000342 AvailableQueue->initNodes(SUnits);
Andrew Trick2085a962010-12-21 22:25:04 +0000343
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000344 HazardRec->Reset();
345
Evan Chengd38c22b2006-05-11 23:55:42 +0000346 // Execute the actual scheduling loop Top-Down or Bottom-Up as appropriate.
347 if (isBottomUp)
348 ListScheduleBottomUp();
349 else
350 ListScheduleTopDown();
Andrew Trick2085a962010-12-21 22:25:04 +0000351
Andrew Trick641e2d42011-03-05 08:00:22 +0000352#ifndef NDEBUG
353 for (int i = 0; i < NumFactors; ++i) {
354 DEBUG(dbgs() << FactorName[i] << "\t" << FactorCount[i] << "\n");
355 }
356#endif // !NDEBUG
Evan Chengd38c22b2006-05-11 23:55:42 +0000357 AvailableQueue->releaseState();
Evan Chengafed73e2006-05-12 01:58:24 +0000358}
Evan Chengd38c22b2006-05-11 23:55:42 +0000359
360//===----------------------------------------------------------------------===//
361// Bottom-Up Scheduling
362//===----------------------------------------------------------------------===//
363
Evan Chengd38c22b2006-05-11 23:55:42 +0000364/// ReleasePred - Decrement the NumSuccsLeft count of a predecessor. Add it to
Dan Gohman54a187e2007-08-20 19:28:38 +0000365/// the AvailableQueue if the count reaches zero. Also update its cycle bound.
Dan Gohman60d68442009-01-29 19:49:27 +0000366void ScheduleDAGRRList::ReleasePred(SUnit *SU, const SDep *PredEdge) {
Dan Gohman2d170892008-12-09 22:54:47 +0000367 SUnit *PredSU = PredEdge->getSUnit();
Reid Klecknercea8dab2009-09-30 20:43:07 +0000368
Evan Chengd38c22b2006-05-11 23:55:42 +0000369#ifndef NDEBUG
Reid Klecknercea8dab2009-09-30 20:43:07 +0000370 if (PredSU->NumSuccsLeft == 0) {
David Greenef34d7ac2010-01-05 01:24:54 +0000371 dbgs() << "*** Scheduling failed! ***\n";
Dan Gohman22d07b12008-11-18 02:06:40 +0000372 PredSU->dump(this);
David Greenef34d7ac2010-01-05 01:24:54 +0000373 dbgs() << " has been released too many times!\n";
Torok Edwinfbcc6632009-07-14 16:55:14 +0000374 llvm_unreachable(0);
Evan Chengd38c22b2006-05-11 23:55:42 +0000375 }
376#endif
Reid Klecknercea8dab2009-09-30 20:43:07 +0000377 --PredSU->NumSuccsLeft;
378
Evan Chengbdd062d2010-05-20 06:13:19 +0000379 if (!ForceUnitLatencies()) {
380 // Updating predecessor's height. This is now the cycle when the
381 // predecessor can be scheduled without causing a pipeline stall.
382 PredSU->setHeightToAtLeast(SU->getHeight() + PredEdge->getLatency());
383 }
384
Dan Gohmanb9543432009-02-10 23:27:53 +0000385 // If all the node's successors are scheduled, this node is ready
386 // to be scheduled. Ignore the special EntrySU node.
387 if (PredSU->NumSuccsLeft == 0 && PredSU != &EntrySU) {
Dan Gohman4370f262008-04-15 01:22:18 +0000388 PredSU->isAvailable = true;
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000389
390 unsigned Height = PredSU->getHeight();
391 if (Height < MinAvailableCycle)
392 MinAvailableCycle = Height;
393
Andrew Trickc88b7ec2011-03-04 02:03:45 +0000394 if (isReady(PredSU)) {
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000395 AvailableQueue->push(PredSU);
396 }
397 // CapturePred and others may have left the node in the pending queue, avoid
398 // adding it twice.
399 else if (!PredSU->isPending) {
400 PredSU->isPending = true;
401 PendingQueue.push_back(PredSU);
402 }
Evan Chengd38c22b2006-05-11 23:55:42 +0000403 }
404}
405
Andrew Trick033efdf2010-12-23 03:15:51 +0000406/// Call ReleasePred for each predecessor, then update register live def/gen.
407/// Always update LiveRegDefs for a register dependence even if the current SU
408/// also defines the register. This effectively create one large live range
409/// across a sequence of two-address node. This is important because the
410/// entire chain must be scheduled together. Example:
411///
412/// flags = (3) add
413/// flags = (2) addc flags
414/// flags = (1) addc flags
415///
416/// results in
417///
418/// LiveRegDefs[flags] = 3
Andrew Tricka52f3252010-12-23 04:16:14 +0000419/// LiveRegGens[flags] = 1
Andrew Trick033efdf2010-12-23 03:15:51 +0000420///
421/// If (2) addc is unscheduled, then (1) addc must also be unscheduled to avoid
422/// interference on flags.
Andrew Tricka52f3252010-12-23 04:16:14 +0000423void ScheduleDAGRRList::ReleasePredecessors(SUnit *SU) {
Evan Chengd38c22b2006-05-11 23:55:42 +0000424 // Bottom up: release predecessors
Chris Lattnerd86418a2006-08-17 00:09:56 +0000425 for (SUnit::pred_iterator I = SU->Preds.begin(), E = SU->Preds.end();
Evan Cheng5924bf72007-09-25 01:54:36 +0000426 I != E; ++I) {
Dan Gohman2d170892008-12-09 22:54:47 +0000427 ReleasePred(SU, &*I);
428 if (I->isAssignedRegDep()) {
Evan Cheng5924bf72007-09-25 01:54:36 +0000429 // This is a physical register dependency and it's impossible or
Andrew Trick2085a962010-12-21 22:25:04 +0000430 // expensive to copy the register. Make sure nothing that can
Evan Cheng5924bf72007-09-25 01:54:36 +0000431 // clobber the register is scheduled between the predecessor and
432 // this node.
Andrew Tricka52f3252010-12-23 04:16:14 +0000433 SUnit *RegDef = LiveRegDefs[I->getReg()]; (void)RegDef;
Andrew Trick033efdf2010-12-23 03:15:51 +0000434 assert((!RegDef || RegDef == SU || RegDef == I->getSUnit()) &&
435 "interference on register dependence");
Andrew Tricka52f3252010-12-23 04:16:14 +0000436 LiveRegDefs[I->getReg()] = I->getSUnit();
437 if (!LiveRegGens[I->getReg()]) {
Dan Gohmanc07f6862008-09-23 18:50:48 +0000438 ++NumLiveRegs;
Andrew Tricka52f3252010-12-23 04:16:14 +0000439 LiveRegGens[I->getReg()] = SU;
Evan Cheng5924bf72007-09-25 01:54:36 +0000440 }
441 }
442 }
Dan Gohmanb9543432009-02-10 23:27:53 +0000443}
444
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000445/// Check to see if any of the pending instructions are ready to issue. If
446/// so, add them to the available queue.
447void ScheduleDAGRRList::ReleasePending() {
Andrew Trick47ff14b2011-01-21 05:51:33 +0000448 if (DisableSchedCycles) {
Andrew Trick5ce945c2010-12-24 07:10:19 +0000449 assert(PendingQueue.empty() && "pending instrs not allowed in this mode");
450 return;
451 }
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000452
453 // If the available queue is empty, it is safe to reset MinAvailableCycle.
454 if (AvailableQueue->empty())
455 MinAvailableCycle = UINT_MAX;
456
457 // Check to see if any of the pending instructions are ready to issue. If
458 // so, add them to the available queue.
459 for (unsigned i = 0, e = PendingQueue.size(); i != e; ++i) {
460 unsigned ReadyCycle =
461 isBottomUp ? PendingQueue[i]->getHeight() : PendingQueue[i]->getDepth();
462 if (ReadyCycle < MinAvailableCycle)
463 MinAvailableCycle = ReadyCycle;
464
465 if (PendingQueue[i]->isAvailable) {
466 if (!isReady(PendingQueue[i]))
467 continue;
468 AvailableQueue->push(PendingQueue[i]);
469 }
470 PendingQueue[i]->isPending = false;
471 PendingQueue[i] = PendingQueue.back();
472 PendingQueue.pop_back();
473 --i; --e;
474 }
475}
476
477/// Move the scheduler state forward by the specified number of Cycles.
478void ScheduleDAGRRList::AdvanceToCycle(unsigned NextCycle) {
479 if (NextCycle <= CurCycle)
480 return;
481
Andrew Trick641e2d42011-03-05 08:00:22 +0000482 IssueCount = 0;
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000483 AvailableQueue->setCurCycle(NextCycle);
Andrew Trick47ff14b2011-01-21 05:51:33 +0000484 if (!HazardRec->isEnabled()) {
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000485 // Bypass lots of virtual calls in case of long latency.
486 CurCycle = NextCycle;
487 }
488 else {
489 for (; CurCycle != NextCycle; ++CurCycle) {
490 if (isBottomUp)
491 HazardRec->RecedeCycle();
492 else
493 HazardRec->AdvanceCycle();
494 }
495 }
496 // FIXME: Instead of visiting the pending Q each time, set a dirty flag on the
497 // available Q to release pending nodes at least once before popping.
498 ReleasePending();
499}
500
501/// Move the scheduler state forward until the specified node's dependents are
502/// ready and can be scheduled with no resource conflicts.
503void ScheduleDAGRRList::AdvancePastStalls(SUnit *SU) {
Andrew Trick47ff14b2011-01-21 05:51:33 +0000504 if (DisableSchedCycles)
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000505 return;
506
Andrew Trickb53a00d2011-04-13 00:38:32 +0000507 // FIXME: Nodes such as CopyFromReg probably should not advance the current
508 // cycle. Otherwise, we can wrongly mask real stalls. If the non-machine node
509 // has predecessors the cycle will be advanced when they are scheduled.
510 // But given the crude nature of modeling latency though such nodes, we
511 // currently need to treat these nodes like real instructions.
512 // if (!SU->getNode() || !SU->getNode()->isMachineOpcode()) return;
513
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000514 unsigned ReadyCycle = isBottomUp ? SU->getHeight() : SU->getDepth();
515
516 // Bump CurCycle to account for latency. We assume the latency of other
517 // available instructions may be hidden by the stall (not a full pipe stall).
518 // This updates the hazard recognizer's cycle before reserving resources for
519 // this instruction.
520 AdvanceToCycle(ReadyCycle);
521
522 // Calls are scheduled in their preceding cycle, so don't conflict with
523 // hazards from instructions after the call. EmitNode will reset the
524 // scoreboard state before emitting the call.
525 if (isBottomUp && SU->isCall)
526 return;
527
528 // FIXME: For resource conflicts in very long non-pipelined stages, we
529 // should probably skip ahead here to avoid useless scoreboard checks.
530 int Stalls = 0;
531 while (true) {
532 ScheduleHazardRecognizer::HazardType HT =
533 HazardRec->getHazardType(SU, isBottomUp ? -Stalls : Stalls);
534
535 if (HT == ScheduleHazardRecognizer::NoHazard)
536 break;
537
538 ++Stalls;
539 }
540 AdvanceToCycle(CurCycle + Stalls);
541}
542
543/// Record this SUnit in the HazardRecognizer.
544/// Does not update CurCycle.
545void ScheduleDAGRRList::EmitNode(SUnit *SU) {
Andrew Trick47ff14b2011-01-21 05:51:33 +0000546 if (!HazardRec->isEnabled())
Andrew Trickc9405662010-12-24 06:46:50 +0000547 return;
548
549 // Check for phys reg copy.
550 if (!SU->getNode())
551 return;
552
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000553 switch (SU->getNode()->getOpcode()) {
554 default:
555 assert(SU->getNode()->isMachineOpcode() &&
556 "This target-independent node should not be scheduled.");
557 break;
558 case ISD::MERGE_VALUES:
559 case ISD::TokenFactor:
560 case ISD::CopyToReg:
561 case ISD::CopyFromReg:
562 case ISD::EH_LABEL:
563 // Noops don't affect the scoreboard state. Copies are likely to be
564 // removed.
565 return;
566 case ISD::INLINEASM:
567 // For inline asm, clear the pipeline state.
568 HazardRec->Reset();
569 return;
570 }
571 if (isBottomUp && SU->isCall) {
572 // Calls are scheduled with their preceding instructions. For bottom-up
573 // scheduling, clear the pipeline state before emitting.
574 HazardRec->Reset();
575 }
576
577 HazardRec->EmitInstruction(SU);
578
579 if (!isBottomUp && SU->isCall) {
580 HazardRec->Reset();
581 }
582}
583
Andrew Trickb53a00d2011-04-13 00:38:32 +0000584static void resetVRegCycle(SUnit *SU);
585
Dan Gohmanb9543432009-02-10 23:27:53 +0000586/// ScheduleNodeBottomUp - Add the node to the schedule. Decrement the pending
587/// count of its predecessors. If a predecessor pending count is zero, add it to
588/// the Available queue.
Andrew Trick528fad92010-12-23 05:42:20 +0000589void ScheduleDAGRRList::ScheduleNodeBottomUp(SUnit *SU) {
Andrew Trick1b60ad62011-04-12 20:14:07 +0000590 DEBUG(dbgs() << "\n*** Scheduling [" << CurCycle << "]: ");
Dan Gohmanb9543432009-02-10 23:27:53 +0000591 DEBUG(SU->dump(this));
592
Evan Chengbdd062d2010-05-20 06:13:19 +0000593#ifndef NDEBUG
594 if (CurCycle < SU->getHeight())
Andrew Trickb53a00d2011-04-13 00:38:32 +0000595 DEBUG(dbgs() << " Height [" << SU->getHeight()
596 << "] pipeline stall!\n");
Evan Chengbdd062d2010-05-20 06:13:19 +0000597#endif
598
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000599 // FIXME: Do not modify node height. It may interfere with
600 // backtracking. Instead add a "ready cycle" to SUnit. Before scheduling the
Eric Christopher1b4b1e52011-03-21 18:06:21 +0000601 // node its ready cycle can aid heuristics, and after scheduling it can
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000602 // indicate the scheduled cycle.
Dan Gohmanb9543432009-02-10 23:27:53 +0000603 SU->setHeightToAtLeast(CurCycle);
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000604
605 // Reserve resources for the scheduled intruction.
606 EmitNode(SU);
607
Dan Gohmanb9543432009-02-10 23:27:53 +0000608 Sequence.push_back(SU);
609
Evan Cheng28590382010-07-21 23:53:58 +0000610 AvailableQueue->ScheduledNode(SU);
Chris Lattner981afd22010-12-20 00:55:43 +0000611
Andrew Trick641e2d42011-03-05 08:00:22 +0000612 // If HazardRec is disabled, and each inst counts as one cycle, then
Andrew Trickb53a00d2011-04-13 00:38:32 +0000613 // advance CurCycle before ReleasePredecessors to avoid useless pushes to
Andrew Trickc88b7ec2011-03-04 02:03:45 +0000614 // PendingQueue for schedulers that implement HasReadyFilter.
Andrew Trick641e2d42011-03-05 08:00:22 +0000615 if (!HazardRec->isEnabled() && AvgIPC < 2)
Andrew Trickc88b7ec2011-03-04 02:03:45 +0000616 AdvanceToCycle(CurCycle + 1);
617
Andrew Trick033efdf2010-12-23 03:15:51 +0000618 // Update liveness of predecessors before successors to avoid treating a
619 // two-address node as a live range def.
Andrew Tricka52f3252010-12-23 04:16:14 +0000620 ReleasePredecessors(SU);
Evan Cheng5924bf72007-09-25 01:54:36 +0000621
622 // Release all the implicit physical register defs that are live.
623 for (SUnit::succ_iterator I = SU->Succs.begin(), E = SU->Succs.end();
624 I != E; ++I) {
Andrew Trick033efdf2010-12-23 03:15:51 +0000625 // LiveRegDegs[I->getReg()] != SU when SU is a two-address node.
626 if (I->isAssignedRegDep() && LiveRegDefs[I->getReg()] == SU) {
627 assert(NumLiveRegs > 0 && "NumLiveRegs is already zero!");
628 --NumLiveRegs;
629 LiveRegDefs[I->getReg()] = NULL;
Andrew Tricka52f3252010-12-23 04:16:14 +0000630 LiveRegGens[I->getReg()] = NULL;
Evan Cheng5924bf72007-09-25 01:54:36 +0000631 }
632 }
633
Andrew Trickb53a00d2011-04-13 00:38:32 +0000634 resetVRegCycle(SU);
635
Evan Chengd38c22b2006-05-11 23:55:42 +0000636 SU->isScheduled = true;
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000637
638 // Conditions under which the scheduler should eagerly advance the cycle:
639 // (1) No available instructions
640 // (2) All pipelines full, so available instructions must have hazards.
641 //
Andrew Trickb53a00d2011-04-13 00:38:32 +0000642 // If HazardRec is disabled, the cycle was pre-advanced before calling
643 // ReleasePredecessors. In that case, IssueCount should remain 0.
Andrew Trickc88b7ec2011-03-04 02:03:45 +0000644 //
645 // Check AvailableQueue after ReleasePredecessors in case of zero latency.
Andrew Trickb53a00d2011-04-13 00:38:32 +0000646 if (HazardRec->isEnabled() || AvgIPC > 1) {
647 if (SU->getNode() && SU->getNode()->isMachineOpcode())
648 ++IssueCount;
649 if ((HazardRec->isEnabled() && HazardRec->atIssueLimit())
650 || (!HazardRec->isEnabled() && IssueCount == AvgIPC))
651 AdvanceToCycle(CurCycle + 1);
652 }
Evan Chengd38c22b2006-05-11 23:55:42 +0000653}
654
Evan Cheng5924bf72007-09-25 01:54:36 +0000655/// CapturePred - This does the opposite of ReleasePred. Since SU is being
656/// unscheduled, incrcease the succ left count of its predecessors. Remove
657/// them from AvailableQueue if necessary.
Andrew Trick2085a962010-12-21 22:25:04 +0000658void ScheduleDAGRRList::CapturePred(SDep *PredEdge) {
Dan Gohman2d170892008-12-09 22:54:47 +0000659 SUnit *PredSU = PredEdge->getSUnit();
Evan Cheng5924bf72007-09-25 01:54:36 +0000660 if (PredSU->isAvailable) {
661 PredSU->isAvailable = false;
662 if (!PredSU->isPending)
663 AvailableQueue->remove(PredSU);
664 }
665
Reid Kleckner8ff5c192009-09-30 20:15:38 +0000666 assert(PredSU->NumSuccsLeft < UINT_MAX && "NumSuccsLeft will overflow!");
Evan Cheng038dcc52007-09-28 19:24:24 +0000667 ++PredSU->NumSuccsLeft;
Evan Cheng5924bf72007-09-25 01:54:36 +0000668}
669
670/// UnscheduleNodeBottomUp - Remove the node from the schedule, update its and
671/// its predecessor states to reflect the change.
672void ScheduleDAGRRList::UnscheduleNodeBottomUp(SUnit *SU) {
David Greenef34d7ac2010-01-05 01:24:54 +0000673 DEBUG(dbgs() << "*** Unscheduling [" << SU->getHeight() << "]: ");
Dan Gohman22d07b12008-11-18 02:06:40 +0000674 DEBUG(SU->dump(this));
Evan Cheng5924bf72007-09-25 01:54:36 +0000675
Evan Cheng5924bf72007-09-25 01:54:36 +0000676 for (SUnit::pred_iterator I = SU->Preds.begin(), E = SU->Preds.end();
677 I != E; ++I) {
Dan Gohman2d170892008-12-09 22:54:47 +0000678 CapturePred(&*I);
Andrew Tricka52f3252010-12-23 04:16:14 +0000679 if (I->isAssignedRegDep() && SU == LiveRegGens[I->getReg()]){
Dan Gohmanc07f6862008-09-23 18:50:48 +0000680 assert(NumLiveRegs > 0 && "NumLiveRegs is already zero!");
Dan Gohman2d170892008-12-09 22:54:47 +0000681 assert(LiveRegDefs[I->getReg()] == I->getSUnit() &&
Evan Cheng5924bf72007-09-25 01:54:36 +0000682 "Physical register dependency violated?");
Dan Gohmanc07f6862008-09-23 18:50:48 +0000683 --NumLiveRegs;
Dan Gohman2d170892008-12-09 22:54:47 +0000684 LiveRegDefs[I->getReg()] = NULL;
Andrew Tricka52f3252010-12-23 04:16:14 +0000685 LiveRegGens[I->getReg()] = NULL;
Evan Cheng5924bf72007-09-25 01:54:36 +0000686 }
687 }
688
689 for (SUnit::succ_iterator I = SU->Succs.begin(), E = SU->Succs.end();
690 I != E; ++I) {
Dan Gohman2d170892008-12-09 22:54:47 +0000691 if (I->isAssignedRegDep()) {
Andrew Trick033efdf2010-12-23 03:15:51 +0000692 // This becomes the nearest def. Note that an earlier def may still be
693 // pending if this is a two-address node.
694 LiveRegDefs[I->getReg()] = SU;
Dan Gohman2d170892008-12-09 22:54:47 +0000695 if (!LiveRegDefs[I->getReg()]) {
Dan Gohmanc07f6862008-09-23 18:50:48 +0000696 ++NumLiveRegs;
Evan Cheng5924bf72007-09-25 01:54:36 +0000697 }
Andrew Tricka52f3252010-12-23 04:16:14 +0000698 if (LiveRegGens[I->getReg()] == NULL ||
699 I->getSUnit()->getHeight() < LiveRegGens[I->getReg()]->getHeight())
700 LiveRegGens[I->getReg()] = I->getSUnit();
Evan Cheng5924bf72007-09-25 01:54:36 +0000701 }
702 }
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000703 if (SU->getHeight() < MinAvailableCycle)
704 MinAvailableCycle = SU->getHeight();
Evan Cheng5924bf72007-09-25 01:54:36 +0000705
Dan Gohmandddc1ac2008-12-16 03:25:46 +0000706 SU->setHeightDirty();
Evan Cheng5924bf72007-09-25 01:54:36 +0000707 SU->isScheduled = false;
708 SU->isAvailable = true;
Andrew Trick47ff14b2011-01-21 05:51:33 +0000709 if (!DisableSchedCycles && AvailableQueue->hasReadyFilter()) {
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000710 // Don't make available until backtracking is complete.
711 SU->isPending = true;
712 PendingQueue.push_back(SU);
713 }
714 else {
715 AvailableQueue->push(SU);
716 }
Evan Cheng28590382010-07-21 23:53:58 +0000717 AvailableQueue->UnscheduledNode(SU);
Evan Cheng5924bf72007-09-25 01:54:36 +0000718}
719
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000720/// After backtracking, the hazard checker needs to be restored to a state
721/// corresponding the the current cycle.
722void ScheduleDAGRRList::RestoreHazardCheckerBottomUp() {
723 HazardRec->Reset();
724
725 unsigned LookAhead = std::min((unsigned)Sequence.size(),
726 HazardRec->getMaxLookAhead());
727 if (LookAhead == 0)
728 return;
729
730 std::vector<SUnit*>::const_iterator I = (Sequence.end() - LookAhead);
731 unsigned HazardCycle = (*I)->getHeight();
732 for (std::vector<SUnit*>::const_iterator E = Sequence.end(); I != E; ++I) {
733 SUnit *SU = *I;
734 for (; SU->getHeight() > HazardCycle; ++HazardCycle) {
735 HazardRec->RecedeCycle();
736 }
737 EmitNode(SU);
738 }
739}
740
Evan Cheng8e136a92007-09-26 21:36:17 +0000741/// BacktrackBottomUp - Backtrack scheduling to a previous cycle specified in
Dan Gohman60d68442009-01-29 19:49:27 +0000742/// BTCycle in order to schedule a specific node.
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000743void ScheduleDAGRRList::BacktrackBottomUp(SUnit *SU, SUnit *BtSU) {
744 SUnit *OldSU = Sequence.back();
745 while (true) {
Evan Cheng5924bf72007-09-25 01:54:36 +0000746 Sequence.pop_back();
747 if (SU->isSucc(OldSU))
Evan Cheng8e136a92007-09-26 21:36:17 +0000748 // Don't try to remove SU from AvailableQueue.
749 SU->isAvailable = false;
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000750 // FIXME: use ready cycle instead of height
751 CurCycle = OldSU->getHeight();
Evan Cheng5924bf72007-09-25 01:54:36 +0000752 UnscheduleNodeBottomUp(OldSU);
Evan Chengbdd062d2010-05-20 06:13:19 +0000753 AvailableQueue->setCurCycle(CurCycle);
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000754 if (OldSU == BtSU)
755 break;
756 OldSU = Sequence.back();
Evan Cheng5924bf72007-09-25 01:54:36 +0000757 }
758
Dan Gohman60d68442009-01-29 19:49:27 +0000759 assert(!SU->isSucc(OldSU) && "Something is wrong!");
Evan Cheng1ec79b42007-09-27 07:09:03 +0000760
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000761 RestoreHazardCheckerBottomUp();
762
Andrew Trick5ce945c2010-12-24 07:10:19 +0000763 ReleasePending();
Andrew Trick10ffc2b2010-12-24 05:03:26 +0000764
Evan Cheng1ec79b42007-09-27 07:09:03 +0000765 ++NumBacktracks;
Evan Cheng5924bf72007-09-25 01:54:36 +0000766}
767
Evan Cheng3b245872010-02-05 01:27:11 +0000768static bool isOperandOf(const SUnit *SU, SDNode *N) {
769 for (const SDNode *SUNode = SU->getNode(); SUNode;
Chris Lattner11a33812010-12-23 17:24:32 +0000770 SUNode = SUNode->getGluedNode()) {
Evan Cheng3b245872010-02-05 01:27:11 +0000771 if (SUNode->isOperandOf(N))
772 return true;
773 }
774 return false;
775}
776
Evan Cheng5924bf72007-09-25 01:54:36 +0000777/// CopyAndMoveSuccessors - Clone the specified node and move its scheduled
778/// successors to the newly created node.
779SUnit *ScheduleDAGRRList::CopyAndMoveSuccessors(SUnit *SU) {
Dan Gohman1ddfcba2008-11-13 21:36:12 +0000780 SDNode *N = SU->getNode();
Evan Cheng79e97132007-10-05 01:39:18 +0000781 if (!N)
782 return NULL;
783
Andrew Trickc9405662010-12-24 06:46:50 +0000784 if (SU->getNode()->getGluedNode())
785 return NULL;
786
Evan Cheng79e97132007-10-05 01:39:18 +0000787 SUnit *NewSU;
Evan Cheng79e97132007-10-05 01:39:18 +0000788 bool TryUnfold = false;
Evan Cheng84d0ebc2007-10-05 01:42:35 +0000789 for (unsigned i = 0, e = N->getNumValues(); i != e; ++i) {
Owen Anderson53aa7a92009-08-10 22:56:29 +0000790 EVT VT = N->getValueType(i);
Chris Lattner3e5fbd72010-12-21 02:38:05 +0000791 if (VT == MVT::Glue)
Evan Cheng84d0ebc2007-10-05 01:42:35 +0000792 return NULL;
Owen Anderson9f944592009-08-11 20:47:22 +0000793 else if (VT == MVT::Other)
Evan Cheng84d0ebc2007-10-05 01:42:35 +0000794 TryUnfold = true;
795 }
Evan Cheng79e97132007-10-05 01:39:18 +0000796 for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i) {
Dan Gohman2ce6f2a2008-07-27 21:46:04 +0000797 const SDValue &Op = N->getOperand(i);
Owen Anderson53aa7a92009-08-10 22:56:29 +0000798 EVT VT = Op.getNode()->getValueType(Op.getResNo());
Chris Lattner3e5fbd72010-12-21 02:38:05 +0000799 if (VT == MVT::Glue)
Evan Cheng79e97132007-10-05 01:39:18 +0000800 return NULL;
Evan Cheng79e97132007-10-05 01:39:18 +0000801 }
802
803 if (TryUnfold) {
Dan Gohmane6e13482008-06-21 15:52:51 +0000804 SmallVector<SDNode*, 2> NewNodes;
Dan Gohman5a390b92008-11-13 21:21:28 +0000805 if (!TII->unfoldMemoryOperand(*DAG, N, NewNodes))
Evan Cheng79e97132007-10-05 01:39:18 +0000806 return NULL;
807
Evan Chengbdd062d2010-05-20 06:13:19 +0000808 DEBUG(dbgs() << "Unfolding SU #" << SU->NodeNum << "\n");
Evan Cheng79e97132007-10-05 01:39:18 +0000809 assert(NewNodes.size() == 2 && "Expected a load folding node!");
810
811 N = NewNodes[1];
812 SDNode *LoadNode = NewNodes[0];
Evan Cheng79e97132007-10-05 01:39:18 +0000813 unsigned NumVals = N->getNumValues();
Dan Gohman1ddfcba2008-11-13 21:36:12 +0000814 unsigned OldNumVals = SU->getNode()->getNumValues();
Evan Cheng79e97132007-10-05 01:39:18 +0000815 for (unsigned i = 0; i != NumVals; ++i)
Dan Gohman1ddfcba2008-11-13 21:36:12 +0000816 DAG->ReplaceAllUsesOfValueWith(SDValue(SU->getNode(), i), SDValue(N, i));
817 DAG->ReplaceAllUsesOfValueWith(SDValue(SU->getNode(), OldNumVals-1),
Dan Gohman5a390b92008-11-13 21:21:28 +0000818 SDValue(LoadNode, 1));
Evan Cheng79e97132007-10-05 01:39:18 +0000819
Dan Gohmane52e0892008-11-11 21:34:44 +0000820 // LoadNode may already exist. This can happen when there is another
821 // load from the same location and producing the same type of value
822 // but it has different alignment or volatileness.
823 bool isNewLoad = true;
824 SUnit *LoadSU;
825 if (LoadNode->getNodeId() != -1) {
826 LoadSU = &SUnits[LoadNode->getNodeId()];
827 isNewLoad = false;
828 } else {
829 LoadSU = CreateNewSUnit(LoadNode);
830 LoadNode->setNodeId(LoadSU->NodeNum);
Andrew Trickd0548ae2011-02-04 03:18:17 +0000831
832 InitNumRegDefsLeft(LoadSU);
Dan Gohmane52e0892008-11-11 21:34:44 +0000833 ComputeLatency(LoadSU);
834 }
835
Roman Levenstein7e71b4b2008-03-26 09:18:09 +0000836 SUnit *NewSU = CreateNewSUnit(N);
Dan Gohman46520a22008-06-21 19:18:17 +0000837 assert(N->getNodeId() == -1 && "Node already inserted!");
838 N->setNodeId(NewSU->NodeNum);
Andrew Trick2085a962010-12-21 22:25:04 +0000839
Dan Gohman17059682008-07-17 19:10:17 +0000840 const TargetInstrDesc &TID = TII->get(N->getMachineOpcode());
Dan Gohman856c0122008-02-16 00:25:40 +0000841 for (unsigned i = 0; i != TID.getNumOperands(); ++i) {
Chris Lattnerfd2e3382008-01-07 06:47:00 +0000842 if (TID.getOperandConstraint(i, TOI::TIED_TO) != -1) {
Evan Cheng79e97132007-10-05 01:39:18 +0000843 NewSU->isTwoAddress = true;
844 break;
845 }
846 }
Chris Lattnerfd2e3382008-01-07 06:47:00 +0000847 if (TID.isCommutable())
Evan Cheng79e97132007-10-05 01:39:18 +0000848 NewSU->isCommutable = true;
Andrew Trickd0548ae2011-02-04 03:18:17 +0000849
850 InitNumRegDefsLeft(NewSU);
Evan Cheng79e97132007-10-05 01:39:18 +0000851 ComputeLatency(NewSU);
852
Dan Gohmaned0e8d42009-03-23 20:20:43 +0000853 // Record all the edges to and from the old SU, by category.
Dan Gohman15af5522009-03-06 02:23:01 +0000854 SmallVector<SDep, 4> ChainPreds;
Evan Cheng79e97132007-10-05 01:39:18 +0000855 SmallVector<SDep, 4> ChainSuccs;
856 SmallVector<SDep, 4> LoadPreds;
857 SmallVector<SDep, 4> NodePreds;
858 SmallVector<SDep, 4> NodeSuccs;
859 for (SUnit::pred_iterator I = SU->Preds.begin(), E = SU->Preds.end();
860 I != E; ++I) {
Dan Gohman2d170892008-12-09 22:54:47 +0000861 if (I->isCtrl())
Dan Gohman15af5522009-03-06 02:23:01 +0000862 ChainPreds.push_back(*I);
Evan Cheng3b245872010-02-05 01:27:11 +0000863 else if (isOperandOf(I->getSUnit(), LoadNode))
Dan Gohman2d170892008-12-09 22:54:47 +0000864 LoadPreds.push_back(*I);
Evan Cheng79e97132007-10-05 01:39:18 +0000865 else
Dan Gohman2d170892008-12-09 22:54:47 +0000866 NodePreds.push_back(*I);
Evan Cheng79e97132007-10-05 01:39:18 +0000867 }
868 for (SUnit::succ_iterator I = SU->Succs.begin(), E = SU->Succs.end();
869 I != E; ++I) {
Dan Gohman2d170892008-12-09 22:54:47 +0000870 if (I->isCtrl())
871 ChainSuccs.push_back(*I);
Evan Cheng79e97132007-10-05 01:39:18 +0000872 else
Dan Gohman2d170892008-12-09 22:54:47 +0000873 NodeSuccs.push_back(*I);
Evan Cheng79e97132007-10-05 01:39:18 +0000874 }
875
Dan Gohmaned0e8d42009-03-23 20:20:43 +0000876 // Now assign edges to the newly-created nodes.
Dan Gohman15af5522009-03-06 02:23:01 +0000877 for (unsigned i = 0, e = ChainPreds.size(); i != e; ++i) {
878 const SDep &Pred = ChainPreds[i];
879 RemovePred(SU, Pred);
Dan Gohman4370f262008-04-15 01:22:18 +0000880 if (isNewLoad)
Dan Gohman15af5522009-03-06 02:23:01 +0000881 AddPred(LoadSU, Pred);
Roman Levenstein7e71b4b2008-03-26 09:18:09 +0000882 }
Evan Cheng79e97132007-10-05 01:39:18 +0000883 for (unsigned i = 0, e = LoadPreds.size(); i != e; ++i) {
Dan Gohman2d170892008-12-09 22:54:47 +0000884 const SDep &Pred = LoadPreds[i];
885 RemovePred(SU, Pred);
Dan Gohman15af5522009-03-06 02:23:01 +0000886 if (isNewLoad)
Dan Gohman2d170892008-12-09 22:54:47 +0000887 AddPred(LoadSU, Pred);
Evan Cheng79e97132007-10-05 01:39:18 +0000888 }
889 for (unsigned i = 0, e = NodePreds.size(); i != e; ++i) {
Dan Gohman2d170892008-12-09 22:54:47 +0000890 const SDep &Pred = NodePreds[i];
891 RemovePred(SU, Pred);
892 AddPred(NewSU, Pred);
Evan Cheng79e97132007-10-05 01:39:18 +0000893 }
894 for (unsigned i = 0, e = NodeSuccs.size(); i != e; ++i) {
Dan Gohman2d170892008-12-09 22:54:47 +0000895 SDep D = NodeSuccs[i];
896 SUnit *SuccDep = D.getSUnit();
897 D.setSUnit(SU);
898 RemovePred(SuccDep, D);
899 D.setSUnit(NewSU);
900 AddPred(SuccDep, D);
Andrew Trickd0548ae2011-02-04 03:18:17 +0000901 // Balance register pressure.
902 if (AvailableQueue->tracksRegPressure() && SuccDep->isScheduled
903 && !D.isCtrl() && NewSU->NumRegDefsLeft > 0)
904 --NewSU->NumRegDefsLeft;
Evan Cheng79e97132007-10-05 01:39:18 +0000905 }
906 for (unsigned i = 0, e = ChainSuccs.size(); i != e; ++i) {
Dan Gohman2d170892008-12-09 22:54:47 +0000907 SDep D = ChainSuccs[i];
908 SUnit *SuccDep = D.getSUnit();
909 D.setSUnit(SU);
910 RemovePred(SuccDep, D);
Roman Levenstein7e71b4b2008-03-26 09:18:09 +0000911 if (isNewLoad) {
Dan Gohman2d170892008-12-09 22:54:47 +0000912 D.setSUnit(LoadSU);
913 AddPred(SuccDep, D);
Roman Levenstein7e71b4b2008-03-26 09:18:09 +0000914 }
Andrew Trick2085a962010-12-21 22:25:04 +0000915 }
Dan Gohmaned0e8d42009-03-23 20:20:43 +0000916
917 // Add a data dependency to reflect that NewSU reads the value defined
918 // by LoadSU.
919 AddPred(NewSU, SDep(LoadSU, SDep::Data, LoadSU->Latency));
Evan Cheng79e97132007-10-05 01:39:18 +0000920
Evan Cheng91e0fc92007-12-18 08:42:10 +0000921 if (isNewLoad)
922 AvailableQueue->addNode(LoadSU);
Evan Cheng79e97132007-10-05 01:39:18 +0000923 AvailableQueue->addNode(NewSU);
924
925 ++NumUnfolds;
926
927 if (NewSU->NumSuccsLeft == 0) {
928 NewSU->isAvailable = true;
929 return NewSU;
Evan Cheng91e0fc92007-12-18 08:42:10 +0000930 }
931 SU = NewSU;
Evan Cheng79e97132007-10-05 01:39:18 +0000932 }
933
Evan Chengbdd062d2010-05-20 06:13:19 +0000934 DEBUG(dbgs() << " Duplicating SU #" << SU->NodeNum << "\n");
Roman Levenstein7e71b4b2008-03-26 09:18:09 +0000935 NewSU = CreateClone(SU);
Evan Cheng5924bf72007-09-25 01:54:36 +0000936
937 // New SUnit has the exact same predecessors.
938 for (SUnit::pred_iterator I = SU->Preds.begin(), E = SU->Preds.end();
939 I != E; ++I)
Dan Gohmandddc1ac2008-12-16 03:25:46 +0000940 if (!I->isArtificial())
Dan Gohman2d170892008-12-09 22:54:47 +0000941 AddPred(NewSU, *I);
Evan Cheng5924bf72007-09-25 01:54:36 +0000942
943 // Only copy scheduled successors. Cut them from old node's successor
944 // list and move them over.
Dan Gohman2d170892008-12-09 22:54:47 +0000945 SmallVector<std::pair<SUnit *, SDep>, 4> DelDeps;
Evan Cheng5924bf72007-09-25 01:54:36 +0000946 for (SUnit::succ_iterator I = SU->Succs.begin(), E = SU->Succs.end();
947 I != E; ++I) {
Dan Gohman2d170892008-12-09 22:54:47 +0000948 if (I->isArtificial())
Evan Cheng5924bf72007-09-25 01:54:36 +0000949 continue;
Dan Gohman2d170892008-12-09 22:54:47 +0000950 SUnit *SuccSU = I->getSUnit();
951 if (SuccSU->isScheduled) {
Dan Gohman2d170892008-12-09 22:54:47 +0000952 SDep D = *I;
953 D.setSUnit(NewSU);
954 AddPred(SuccSU, D);
955 D.setSUnit(SU);
956 DelDeps.push_back(std::make_pair(SuccSU, D));
Evan Cheng5924bf72007-09-25 01:54:36 +0000957 }
958 }
Dan Gohmandddc1ac2008-12-16 03:25:46 +0000959 for (unsigned i = 0, e = DelDeps.size(); i != e; ++i)
Dan Gohman2d170892008-12-09 22:54:47 +0000960 RemovePred(DelDeps[i].first, DelDeps[i].second);
Evan Cheng5924bf72007-09-25 01:54:36 +0000961
962 AvailableQueue->updateNode(SU);
963 AvailableQueue->addNode(NewSU);
964
Evan Cheng1ec79b42007-09-27 07:09:03 +0000965 ++NumDups;
Evan Cheng5924bf72007-09-25 01:54:36 +0000966 return NewSU;
967}
968
Evan Chengb2c42c62009-01-12 03:19:55 +0000969/// InsertCopiesAndMoveSuccs - Insert register copies and move all
970/// scheduled successors of the given SUnit to the last copy.
971void ScheduleDAGRRList::InsertCopiesAndMoveSuccs(SUnit *SU, unsigned Reg,
972 const TargetRegisterClass *DestRC,
973 const TargetRegisterClass *SrcRC,
Evan Cheng1ec79b42007-09-27 07:09:03 +0000974 SmallVector<SUnit*, 2> &Copies) {
Roman Levenstein7e71b4b2008-03-26 09:18:09 +0000975 SUnit *CopyFromSU = CreateNewSUnit(NULL);
Evan Cheng8e136a92007-09-26 21:36:17 +0000976 CopyFromSU->CopySrcRC = SrcRC;
977 CopyFromSU->CopyDstRC = DestRC;
Evan Cheng8e136a92007-09-26 21:36:17 +0000978
Roman Levenstein7e71b4b2008-03-26 09:18:09 +0000979 SUnit *CopyToSU = CreateNewSUnit(NULL);
Evan Cheng8e136a92007-09-26 21:36:17 +0000980 CopyToSU->CopySrcRC = DestRC;
981 CopyToSU->CopyDstRC = SrcRC;
982
983 // Only copy scheduled successors. Cut them from old node's successor
984 // list and move them over.
Dan Gohman2d170892008-12-09 22:54:47 +0000985 SmallVector<std::pair<SUnit *, SDep>, 4> DelDeps;
Evan Cheng8e136a92007-09-26 21:36:17 +0000986 for (SUnit::succ_iterator I = SU->Succs.begin(), E = SU->Succs.end();
987 I != E; ++I) {
Dan Gohman2d170892008-12-09 22:54:47 +0000988 if (I->isArtificial())
Evan Cheng8e136a92007-09-26 21:36:17 +0000989 continue;
Dan Gohman2d170892008-12-09 22:54:47 +0000990 SUnit *SuccSU = I->getSUnit();
991 if (SuccSU->isScheduled) {
992 SDep D = *I;
993 D.setSUnit(CopyToSU);
994 AddPred(SuccSU, D);
995 DelDeps.push_back(std::make_pair(SuccSU, *I));
Evan Cheng8e136a92007-09-26 21:36:17 +0000996 }
Andrew Trick13acae02011-03-23 20:42:39 +0000997 else {
998 // Avoid scheduling the def-side copy before other successors. Otherwise
999 // we could introduce another physreg interference on the copy and
1000 // continue inserting copies indefinitely.
1001 SDep D(CopyFromSU, SDep::Order, /*Latency=*/0,
1002 /*Reg=*/0, /*isNormalMemory=*/false,
1003 /*isMustAlias=*/false, /*isArtificial=*/true);
1004 AddPred(SuccSU, D);
1005 }
Evan Cheng8e136a92007-09-26 21:36:17 +00001006 }
Evan Chengb2c42c62009-01-12 03:19:55 +00001007 for (unsigned i = 0, e = DelDeps.size(); i != e; ++i)
Dan Gohman2d170892008-12-09 22:54:47 +00001008 RemovePred(DelDeps[i].first, DelDeps[i].second);
Evan Cheng8e136a92007-09-26 21:36:17 +00001009
Dan Gohman2d170892008-12-09 22:54:47 +00001010 AddPred(CopyFromSU, SDep(SU, SDep::Data, SU->Latency, Reg));
1011 AddPred(CopyToSU, SDep(CopyFromSU, SDep::Data, CopyFromSU->Latency, 0));
Evan Cheng8e136a92007-09-26 21:36:17 +00001012
1013 AvailableQueue->updateNode(SU);
1014 AvailableQueue->addNode(CopyFromSU);
1015 AvailableQueue->addNode(CopyToSU);
Evan Cheng1ec79b42007-09-27 07:09:03 +00001016 Copies.push_back(CopyFromSU);
1017 Copies.push_back(CopyToSU);
Evan Cheng8e136a92007-09-26 21:36:17 +00001018
Evan Chengb2c42c62009-01-12 03:19:55 +00001019 ++NumPRCopies;
Evan Cheng8e136a92007-09-26 21:36:17 +00001020}
1021
1022/// getPhysicalRegisterVT - Returns the ValueType of the physical register
1023/// definition of the specified node.
1024/// FIXME: Move to SelectionDAG?
Owen Anderson53aa7a92009-08-10 22:56:29 +00001025static EVT getPhysicalRegisterVT(SDNode *N, unsigned Reg,
Duncan Sands13237ac2008-06-06 12:08:01 +00001026 const TargetInstrInfo *TII) {
Dan Gohman17059682008-07-17 19:10:17 +00001027 const TargetInstrDesc &TID = TII->get(N->getMachineOpcode());
Evan Cheng8e136a92007-09-26 21:36:17 +00001028 assert(TID.ImplicitDefs && "Physical reg def must be in implicit def list!");
Chris Lattnerb0d06b42008-01-07 03:13:06 +00001029 unsigned NumRes = TID.getNumDefs();
1030 for (const unsigned *ImpDef = TID.getImplicitDefs(); *ImpDef; ++ImpDef) {
Evan Cheng8e136a92007-09-26 21:36:17 +00001031 if (Reg == *ImpDef)
1032 break;
1033 ++NumRes;
1034 }
1035 return N->getValueType(NumRes);
1036}
1037
Evan Chengb8905c42009-03-04 01:41:49 +00001038/// CheckForLiveRegDef - Return true and update live register vector if the
1039/// specified register def of the specified SUnit clobbers any "live" registers.
Chris Lattner0cfe8842010-12-20 00:51:56 +00001040static void CheckForLiveRegDef(SUnit *SU, unsigned Reg,
Evan Chengb8905c42009-03-04 01:41:49 +00001041 std::vector<SUnit*> &LiveRegDefs,
1042 SmallSet<unsigned, 4> &RegAdded,
1043 SmallVector<unsigned, 4> &LRegs,
1044 const TargetRegisterInfo *TRI) {
Andrew Trick12acde112010-12-23 03:43:21 +00001045 for (const unsigned *AliasI = TRI->getOverlaps(Reg); *AliasI; ++AliasI) {
1046
1047 // Check if Ref is live.
Andrew Trick0af2e472011-06-07 00:38:12 +00001048 if (!LiveRegDefs[*AliasI]) continue;
Andrew Trick12acde112010-12-23 03:43:21 +00001049
1050 // Allow multiple uses of the same def.
Andrew Trick0af2e472011-06-07 00:38:12 +00001051 if (LiveRegDefs[*AliasI] == SU) continue;
Andrew Trick12acde112010-12-23 03:43:21 +00001052
1053 // Add Reg to the set of interfering live regs.
Andrew Trick0af2e472011-06-07 00:38:12 +00001054 if (RegAdded.insert(*AliasI)) {
Andrew Trick0af2e472011-06-07 00:38:12 +00001055 LRegs.push_back(*AliasI);
1056 }
Evan Chengb8905c42009-03-04 01:41:49 +00001057 }
Evan Chengb8905c42009-03-04 01:41:49 +00001058}
1059
Evan Cheng5924bf72007-09-25 01:54:36 +00001060/// DelayForLiveRegsBottomUp - Returns true if it is necessary to delay
1061/// scheduling of the given node to satisfy live physical register dependencies.
1062/// If the specific node is the last one that's available to schedule, do
1063/// whatever is necessary (i.e. backtracking or cloning) to make it possible.
Chris Lattner0cfe8842010-12-20 00:51:56 +00001064bool ScheduleDAGRRList::
1065DelayForLiveRegsBottomUp(SUnit *SU, SmallVector<unsigned, 4> &LRegs) {
Dan Gohmanc07f6862008-09-23 18:50:48 +00001066 if (NumLiveRegs == 0)
Evan Cheng5924bf72007-09-25 01:54:36 +00001067 return false;
1068
Evan Chenge6f92252007-09-27 18:46:06 +00001069 SmallSet<unsigned, 4> RegAdded;
Evan Cheng5924bf72007-09-25 01:54:36 +00001070 // If this node would clobber any "live" register, then it's not ready.
Andrew Trickfbb3ed82010-12-21 22:27:44 +00001071 //
1072 // If SU is the currently live definition of the same register that it uses,
1073 // then we are free to schedule it.
Evan Cheng5924bf72007-09-25 01:54:36 +00001074 for (SUnit::pred_iterator I = SU->Preds.begin(), E = SU->Preds.end();
1075 I != E; ++I) {
Andrew Trickfbb3ed82010-12-21 22:27:44 +00001076 if (I->isAssignedRegDep() && LiveRegDefs[I->getReg()] != SU)
Evan Chengb8905c42009-03-04 01:41:49 +00001077 CheckForLiveRegDef(I->getSUnit(), I->getReg(), LiveRegDefs,
1078 RegAdded, LRegs, TRI);
Evan Cheng5924bf72007-09-25 01:54:36 +00001079 }
1080
Chris Lattner11a33812010-12-23 17:24:32 +00001081 for (SDNode *Node = SU->getNode(); Node; Node = Node->getGluedNode()) {
Evan Chengb8905c42009-03-04 01:41:49 +00001082 if (Node->getOpcode() == ISD::INLINEASM) {
1083 // Inline asm can clobber physical defs.
1084 unsigned NumOps = Node->getNumOperands();
Chris Lattner3e5fbd72010-12-21 02:38:05 +00001085 if (Node->getOperand(NumOps-1).getValueType() == MVT::Glue)
Chris Lattner11a33812010-12-23 17:24:32 +00001086 --NumOps; // Ignore the glue operand.
Evan Chengb8905c42009-03-04 01:41:49 +00001087
Chris Lattner3b9f02a2010-04-07 05:20:54 +00001088 for (unsigned i = InlineAsm::Op_FirstOperand; i != NumOps;) {
Evan Chengb8905c42009-03-04 01:41:49 +00001089 unsigned Flags =
1090 cast<ConstantSDNode>(Node->getOperand(i))->getZExtValue();
Chris Lattner3b9f02a2010-04-07 05:20:54 +00001091 unsigned NumVals = InlineAsm::getNumOperandRegisters(Flags);
Evan Chengb8905c42009-03-04 01:41:49 +00001092
1093 ++i; // Skip the ID value.
Chris Lattner3b9f02a2010-04-07 05:20:54 +00001094 if (InlineAsm::isRegDefKind(Flags) ||
1095 InlineAsm::isRegDefEarlyClobberKind(Flags)) {
Evan Chengb8905c42009-03-04 01:41:49 +00001096 // Check for def of register or earlyclobber register.
1097 for (; NumVals; --NumVals, ++i) {
1098 unsigned Reg = cast<RegisterSDNode>(Node->getOperand(i))->getReg();
1099 if (TargetRegisterInfo::isPhysicalRegister(Reg))
1100 CheckForLiveRegDef(SU, Reg, LiveRegDefs, RegAdded, LRegs, TRI);
1101 }
1102 } else
1103 i += NumVals;
1104 }
1105 continue;
1106 }
1107
Dan Gohman072734e2008-11-13 23:24:17 +00001108 if (!Node->isMachineOpcode())
Evan Cheng5924bf72007-09-25 01:54:36 +00001109 continue;
Dan Gohman17059682008-07-17 19:10:17 +00001110 const TargetInstrDesc &TID = TII->get(Node->getMachineOpcode());
Evan Cheng5924bf72007-09-25 01:54:36 +00001111 if (!TID.ImplicitDefs)
1112 continue;
Evan Chengb8905c42009-03-04 01:41:49 +00001113 for (const unsigned *Reg = TID.ImplicitDefs; *Reg; ++Reg)
1114 CheckForLiveRegDef(SU, *Reg, LiveRegDefs, RegAdded, LRegs, TRI);
Evan Cheng5924bf72007-09-25 01:54:36 +00001115 }
Andrew Trick2085a962010-12-21 22:25:04 +00001116
Evan Cheng5924bf72007-09-25 01:54:36 +00001117 return !LRegs.empty();
Evan Chengd38c22b2006-05-11 23:55:42 +00001118}
1119
Andrew Trick528fad92010-12-23 05:42:20 +00001120/// Return a node that can be scheduled in this cycle. Requirements:
1121/// (1) Ready: latency has been satisfied
Andrew Trick10ffc2b2010-12-24 05:03:26 +00001122/// (2) No Hazards: resources are available
Andrew Trick528fad92010-12-23 05:42:20 +00001123/// (3) No Interferences: may unschedule to break register interferences.
1124SUnit *ScheduleDAGRRList::PickNodeToScheduleBottomUp() {
1125 SmallVector<SUnit*, 4> Interferences;
1126 DenseMap<SUnit*, SmallVector<unsigned, 4> > LRegsMap;
1127
1128 SUnit *CurSU = AvailableQueue->pop();
1129 while (CurSU) {
1130 SmallVector<unsigned, 4> LRegs;
1131 if (!DelayForLiveRegsBottomUp(CurSU, LRegs))
1132 break;
1133 LRegsMap.insert(std::make_pair(CurSU, LRegs));
1134
1135 CurSU->isPending = true; // This SU is not in AvailableQueue right now.
1136 Interferences.push_back(CurSU);
1137 CurSU = AvailableQueue->pop();
1138 }
1139 if (CurSU) {
1140 // Add the nodes that aren't ready back onto the available list.
1141 for (unsigned i = 0, e = Interferences.size(); i != e; ++i) {
1142 Interferences[i]->isPending = false;
1143 assert(Interferences[i]->isAvailable && "must still be available");
1144 AvailableQueue->push(Interferences[i]);
1145 }
1146 return CurSU;
1147 }
1148
1149 // All candidates are delayed due to live physical reg dependencies.
1150 // Try backtracking, code duplication, or inserting cross class copies
1151 // to resolve it.
1152 for (unsigned i = 0, e = Interferences.size(); i != e; ++i) {
1153 SUnit *TrySU = Interferences[i];
1154 SmallVector<unsigned, 4> &LRegs = LRegsMap[TrySU];
1155
1156 // Try unscheduling up to the point where it's safe to schedule
1157 // this node.
Andrew Trick10ffc2b2010-12-24 05:03:26 +00001158 SUnit *BtSU = NULL;
1159 unsigned LiveCycle = UINT_MAX;
Andrew Trick528fad92010-12-23 05:42:20 +00001160 for (unsigned j = 0, ee = LRegs.size(); j != ee; ++j) {
1161 unsigned Reg = LRegs[j];
Andrew Trick10ffc2b2010-12-24 05:03:26 +00001162 if (LiveRegGens[Reg]->getHeight() < LiveCycle) {
1163 BtSU = LiveRegGens[Reg];
1164 LiveCycle = BtSU->getHeight();
1165 }
Andrew Trick528fad92010-12-23 05:42:20 +00001166 }
Andrew Trick10ffc2b2010-12-24 05:03:26 +00001167 if (!WillCreateCycle(TrySU, BtSU)) {
1168 BacktrackBottomUp(TrySU, BtSU);
Andrew Trick528fad92010-12-23 05:42:20 +00001169
1170 // Force the current node to be scheduled before the node that
1171 // requires the physical reg dep.
Andrew Trick10ffc2b2010-12-24 05:03:26 +00001172 if (BtSU->isAvailable) {
1173 BtSU->isAvailable = false;
1174 if (!BtSU->isPending)
1175 AvailableQueue->remove(BtSU);
Andrew Trick528fad92010-12-23 05:42:20 +00001176 }
Andrew Trick10ffc2b2010-12-24 05:03:26 +00001177 AddPred(TrySU, SDep(BtSU, SDep::Order, /*Latency=*/1,
Andrew Trick528fad92010-12-23 05:42:20 +00001178 /*Reg=*/0, /*isNormalMemory=*/false,
1179 /*isMustAlias=*/false, /*isArtificial=*/true));
1180
1181 // If one or more successors has been unscheduled, then the current
1182 // node is no longer avaialable. Schedule a successor that's now
1183 // available instead.
1184 if (!TrySU->isAvailable) {
1185 CurSU = AvailableQueue->pop();
1186 }
1187 else {
1188 CurSU = TrySU;
1189 TrySU->isPending = false;
1190 Interferences.erase(Interferences.begin()+i);
1191 }
1192 break;
1193 }
1194 }
1195
1196 if (!CurSU) {
1197 // Can't backtrack. If it's too expensive to copy the value, then try
1198 // duplicate the nodes that produces these "too expensive to copy"
1199 // values to break the dependency. In case even that doesn't work,
1200 // insert cross class copies.
1201 // If it's not too expensive, i.e. cost != -1, issue copies.
1202 SUnit *TrySU = Interferences[0];
1203 SmallVector<unsigned, 4> &LRegs = LRegsMap[TrySU];
1204 assert(LRegs.size() == 1 && "Can't handle this yet!");
1205 unsigned Reg = LRegs[0];
1206 SUnit *LRDef = LiveRegDefs[Reg];
1207 EVT VT = getPhysicalRegisterVT(LRDef->getNode(), Reg, TII);
1208 const TargetRegisterClass *RC =
1209 TRI->getMinimalPhysRegClass(Reg, VT);
1210 const TargetRegisterClass *DestRC = TRI->getCrossCopyRegClass(RC);
1211
Evan Chengb4c6a342011-03-10 00:16:32 +00001212 // If cross copy register class is the same as RC, then it must be possible
1213 // copy the value directly. Do not try duplicate the def.
1214 // If cross copy register class is not the same as RC, then it's possible to
1215 // copy the value but it require cross register class copies and it is
1216 // expensive.
1217 // If cross copy register class is null, then it's not possible to copy
1218 // the value at all.
Andrew Trick528fad92010-12-23 05:42:20 +00001219 SUnit *NewDef = 0;
Evan Chengb4c6a342011-03-10 00:16:32 +00001220 if (DestRC != RC) {
Andrew Trick528fad92010-12-23 05:42:20 +00001221 NewDef = CopyAndMoveSuccessors(LRDef);
Evan Chengb4c6a342011-03-10 00:16:32 +00001222 if (!DestRC && !NewDef)
1223 report_fatal_error("Can't handle live physical register dependency!");
1224 }
Andrew Trick528fad92010-12-23 05:42:20 +00001225 if (!NewDef) {
1226 // Issue copies, these can be expensive cross register class copies.
1227 SmallVector<SUnit*, 2> Copies;
1228 InsertCopiesAndMoveSuccs(LRDef, Reg, DestRC, RC, Copies);
1229 DEBUG(dbgs() << " Adding an edge from SU #" << TrySU->NodeNum
1230 << " to SU #" << Copies.front()->NodeNum << "\n");
1231 AddPred(TrySU, SDep(Copies.front(), SDep::Order, /*Latency=*/1,
1232 /*Reg=*/0, /*isNormalMemory=*/false,
1233 /*isMustAlias=*/false,
1234 /*isArtificial=*/true));
1235 NewDef = Copies.back();
1236 }
1237
1238 DEBUG(dbgs() << " Adding an edge from SU #" << NewDef->NodeNum
1239 << " to SU #" << TrySU->NodeNum << "\n");
1240 LiveRegDefs[Reg] = NewDef;
1241 AddPred(NewDef, SDep(TrySU, SDep::Order, /*Latency=*/1,
1242 /*Reg=*/0, /*isNormalMemory=*/false,
1243 /*isMustAlias=*/false,
1244 /*isArtificial=*/true));
1245 TrySU->isAvailable = false;
1246 CurSU = NewDef;
1247 }
1248
1249 assert(CurSU && "Unable to resolve live physical register dependencies!");
1250
1251 // Add the nodes that aren't ready back onto the available list.
1252 for (unsigned i = 0, e = Interferences.size(); i != e; ++i) {
1253 Interferences[i]->isPending = false;
1254 // May no longer be available due to backtracking.
1255 if (Interferences[i]->isAvailable) {
1256 AvailableQueue->push(Interferences[i]);
1257 }
1258 }
1259 return CurSU;
1260}
Evan Cheng1ec79b42007-09-27 07:09:03 +00001261
Evan Chengd38c22b2006-05-11 23:55:42 +00001262/// ListScheduleBottomUp - The main loop of list scheduling for bottom-up
1263/// schedulers.
1264void ScheduleDAGRRList::ListScheduleBottomUp() {
Dan Gohmanb9543432009-02-10 23:27:53 +00001265 // Release any predecessors of the special Exit node.
Andrew Tricka52f3252010-12-23 04:16:14 +00001266 ReleasePredecessors(&ExitSU);
Dan Gohmanb9543432009-02-10 23:27:53 +00001267
Evan Chengd38c22b2006-05-11 23:55:42 +00001268 // Add root to Available queue.
Dan Gohman4370f262008-04-15 01:22:18 +00001269 if (!SUnits.empty()) {
Dan Gohman5a390b92008-11-13 21:21:28 +00001270 SUnit *RootSU = &SUnits[DAG->getRoot().getNode()->getNodeId()];
Dan Gohman4370f262008-04-15 01:22:18 +00001271 assert(RootSU->Succs.empty() && "Graph root shouldn't have successors!");
1272 RootSU->isAvailable = true;
1273 AvailableQueue->push(RootSU);
1274 }
Evan Chengd38c22b2006-05-11 23:55:42 +00001275
1276 // While Available queue is not empty, grab the node with the highest
Dan Gohman54a187e2007-08-20 19:28:38 +00001277 // priority. If it is not ready put it back. Schedule the node.
Dan Gohmane6e13482008-06-21 15:52:51 +00001278 Sequence.reserve(SUnits.size());
Evan Chengd38c22b2006-05-11 23:55:42 +00001279 while (!AvailableQueue->empty()) {
Andrew Trickb53a00d2011-04-13 00:38:32 +00001280 DEBUG(dbgs() << "\nExamining Available:\n";
Andrew Trick10ffc2b2010-12-24 05:03:26 +00001281 AvailableQueue->dump(this));
1282
Andrew Trick528fad92010-12-23 05:42:20 +00001283 // Pick the best node to schedule taking all constraints into
1284 // consideration.
1285 SUnit *SU = PickNodeToScheduleBottomUp();
Evan Cheng1ec79b42007-09-27 07:09:03 +00001286
Andrew Trick10ffc2b2010-12-24 05:03:26 +00001287 AdvancePastStalls(SU);
Evan Cheng1ec79b42007-09-27 07:09:03 +00001288
Andrew Trick10ffc2b2010-12-24 05:03:26 +00001289 ScheduleNodeBottomUp(SU);
1290
1291 while (AvailableQueue->empty() && !PendingQueue.empty()) {
1292 // Advance the cycle to free resources. Skip ahead to the next ready SU.
1293 assert(MinAvailableCycle < UINT_MAX && "MinAvailableCycle uninitialized");
1294 AdvanceToCycle(std::max(CurCycle + 1, MinAvailableCycle));
1295 }
Evan Chengd38c22b2006-05-11 23:55:42 +00001296 }
1297
Evan Chengd38c22b2006-05-11 23:55:42 +00001298 // Reverse the order if it is bottom up.
1299 std::reverse(Sequence.begin(), Sequence.end());
Andrew Trick2085a962010-12-21 22:25:04 +00001300
Evan Chengd38c22b2006-05-11 23:55:42 +00001301#ifndef NDEBUG
Dan Gohman4ce15e12008-11-20 01:26:25 +00001302 VerifySchedule(isBottomUp);
Evan Chengd38c22b2006-05-11 23:55:42 +00001303#endif
1304}
1305
1306//===----------------------------------------------------------------------===//
1307// Top-Down Scheduling
1308//===----------------------------------------------------------------------===//
1309
1310/// ReleaseSucc - Decrement the NumPredsLeft count of a successor. Add it to
Dan Gohman54a187e2007-08-20 19:28:38 +00001311/// the AvailableQueue if the count reaches zero. Also update its cycle bound.
Dan Gohman60d68442009-01-29 19:49:27 +00001312void ScheduleDAGRRList::ReleaseSucc(SUnit *SU, const SDep *SuccEdge) {
Dan Gohman2d170892008-12-09 22:54:47 +00001313 SUnit *SuccSU = SuccEdge->getSUnit();
Reid Kleckner8ff5c192009-09-30 20:15:38 +00001314
Evan Chengd38c22b2006-05-11 23:55:42 +00001315#ifndef NDEBUG
Reid Kleckner8ff5c192009-09-30 20:15:38 +00001316 if (SuccSU->NumPredsLeft == 0) {
David Greenef34d7ac2010-01-05 01:24:54 +00001317 dbgs() << "*** Scheduling failed! ***\n";
Dan Gohman22d07b12008-11-18 02:06:40 +00001318 SuccSU->dump(this);
David Greenef34d7ac2010-01-05 01:24:54 +00001319 dbgs() << " has been released too many times!\n";
Torok Edwinfbcc6632009-07-14 16:55:14 +00001320 llvm_unreachable(0);
Evan Chengd38c22b2006-05-11 23:55:42 +00001321 }
1322#endif
Reid Kleckner8ff5c192009-09-30 20:15:38 +00001323 --SuccSU->NumPredsLeft;
1324
Dan Gohmanb9543432009-02-10 23:27:53 +00001325 // If all the node's predecessors are scheduled, this node is ready
1326 // to be scheduled. Ignore the special ExitSU node.
1327 if (SuccSU->NumPredsLeft == 0 && SuccSU != &ExitSU) {
Evan Chengd38c22b2006-05-11 23:55:42 +00001328 SuccSU->isAvailable = true;
1329 AvailableQueue->push(SuccSU);
1330 }
1331}
1332
Dan Gohmanb9543432009-02-10 23:27:53 +00001333void ScheduleDAGRRList::ReleaseSuccessors(SUnit *SU) {
1334 // Top down: release successors
1335 for (SUnit::succ_iterator I = SU->Succs.begin(), E = SU->Succs.end();
1336 I != E; ++I) {
1337 assert(!I->isAssignedRegDep() &&
1338 "The list-tdrr scheduler doesn't yet support physreg dependencies!");
1339
1340 ReleaseSucc(SU, &*I);
1341 }
1342}
1343
Evan Chengd38c22b2006-05-11 23:55:42 +00001344/// ScheduleNodeTopDown - Add the node to the schedule. Decrement the pending
1345/// count of its successors. If a successor pending count is zero, add it to
1346/// the Available queue.
Andrew Trick528fad92010-12-23 05:42:20 +00001347void ScheduleDAGRRList::ScheduleNodeTopDown(SUnit *SU) {
David Greenef34d7ac2010-01-05 01:24:54 +00001348 DEBUG(dbgs() << "*** Scheduling [" << CurCycle << "]: ");
Dan Gohman22d07b12008-11-18 02:06:40 +00001349 DEBUG(SU->dump(this));
Evan Chengd38c22b2006-05-11 23:55:42 +00001350
Dan Gohmandddc1ac2008-12-16 03:25:46 +00001351 assert(CurCycle >= SU->getDepth() && "Node scheduled above its depth!");
1352 SU->setDepthToAtLeast(CurCycle);
Dan Gohman92a36d72008-11-17 21:31:02 +00001353 Sequence.push_back(SU);
Evan Chengd38c22b2006-05-11 23:55:42 +00001354
Dan Gohmanb9543432009-02-10 23:27:53 +00001355 ReleaseSuccessors(SU);
Evan Chengd38c22b2006-05-11 23:55:42 +00001356 SU->isScheduled = true;
Dan Gohman92a36d72008-11-17 21:31:02 +00001357 AvailableQueue->ScheduledNode(SU);
Evan Chengd38c22b2006-05-11 23:55:42 +00001358}
1359
Dan Gohman54a187e2007-08-20 19:28:38 +00001360/// ListScheduleTopDown - The main loop of list scheduling for top-down
1361/// schedulers.
Evan Chengd38c22b2006-05-11 23:55:42 +00001362void ScheduleDAGRRList::ListScheduleTopDown() {
Evan Chengbdd062d2010-05-20 06:13:19 +00001363 AvailableQueue->setCurCycle(CurCycle);
Evan Chengd38c22b2006-05-11 23:55:42 +00001364
Dan Gohmanb9543432009-02-10 23:27:53 +00001365 // Release any successors of the special Entry node.
1366 ReleaseSuccessors(&EntrySU);
1367
Evan Chengd38c22b2006-05-11 23:55:42 +00001368 // All leaves to Available queue.
1369 for (unsigned i = 0, e = SUnits.size(); i != e; ++i) {
1370 // It is available if it has no predecessors.
Dan Gohman4370f262008-04-15 01:22:18 +00001371 if (SUnits[i].Preds.empty()) {
Evan Chengd38c22b2006-05-11 23:55:42 +00001372 AvailableQueue->push(&SUnits[i]);
1373 SUnits[i].isAvailable = true;
1374 }
1375 }
Andrew Trick2085a962010-12-21 22:25:04 +00001376
Evan Chengd38c22b2006-05-11 23:55:42 +00001377 // While Available queue is not empty, grab the node with the highest
Dan Gohman54a187e2007-08-20 19:28:38 +00001378 // priority. If it is not ready put it back. Schedule the node.
Dan Gohmane6e13482008-06-21 15:52:51 +00001379 Sequence.reserve(SUnits.size());
Evan Chengd38c22b2006-05-11 23:55:42 +00001380 while (!AvailableQueue->empty()) {
Evan Cheng5924bf72007-09-25 01:54:36 +00001381 SUnit *CurSU = AvailableQueue->pop();
Andrew Trick2085a962010-12-21 22:25:04 +00001382
Dan Gohmanc602dd42008-11-21 00:10:42 +00001383 if (CurSU)
Andrew Trick528fad92010-12-23 05:42:20 +00001384 ScheduleNodeTopDown(CurSU);
Dan Gohman4370f262008-04-15 01:22:18 +00001385 ++CurCycle;
Evan Chengbdd062d2010-05-20 06:13:19 +00001386 AvailableQueue->setCurCycle(CurCycle);
Evan Chengd38c22b2006-05-11 23:55:42 +00001387 }
Andrew Trick2085a962010-12-21 22:25:04 +00001388
Evan Chengd38c22b2006-05-11 23:55:42 +00001389#ifndef NDEBUG
Dan Gohman4ce15e12008-11-20 01:26:25 +00001390 VerifySchedule(isBottomUp);
Evan Chengd38c22b2006-05-11 23:55:42 +00001391#endif
1392}
1393
1394
Evan Chengd38c22b2006-05-11 23:55:42 +00001395//===----------------------------------------------------------------------===//
Andrew Trick9ccce772011-01-14 21:11:41 +00001396// RegReductionPriorityQueue Definition
Evan Chengd38c22b2006-05-11 23:55:42 +00001397//===----------------------------------------------------------------------===//
1398//
1399// This is a SchedulingPriorityQueue that schedules using Sethi Ullman numbers
1400// to reduce register pressure.
Andrew Trick2085a962010-12-21 22:25:04 +00001401//
Evan Chengd38c22b2006-05-11 23:55:42 +00001402namespace {
Andrew Trick9ccce772011-01-14 21:11:41 +00001403class RegReductionPQBase;
Andrew Trick2085a962010-12-21 22:25:04 +00001404
Andrew Trick9ccce772011-01-14 21:11:41 +00001405struct queue_sort : public std::binary_function<SUnit*, SUnit*, bool> {
1406 bool isReady(SUnit* SU, unsigned CurCycle) const { return true; }
1407};
1408
Andrew Trick3013b6a2011-06-15 17:16:12 +00001409#ifndef NDEBUG
1410template<class SF>
1411struct reverse_sort : public queue_sort {
1412 SF &SortFunc;
1413 reverse_sort(SF &sf) : SortFunc(sf) {}
1414 reverse_sort(const reverse_sort &RHS) : SortFunc(RHS.SortFunc) {}
1415
1416 bool operator()(SUnit* left, SUnit* right) const {
1417 // reverse left/right rather than simply !SortFunc(left, right)
1418 // to expose different paths in the comparison logic.
1419 return SortFunc(right, left);
1420 }
1421};
1422#endif // NDEBUG
1423
Andrew Trick9ccce772011-01-14 21:11:41 +00001424/// bu_ls_rr_sort - Priority function for bottom up register pressure
1425// reduction scheduler.
1426struct bu_ls_rr_sort : public queue_sort {
1427 enum {
1428 IsBottomUp = true,
1429 HasReadyFilter = false
Andrew Trick10ffc2b2010-12-24 05:03:26 +00001430 };
1431
Andrew Trick9ccce772011-01-14 21:11:41 +00001432 RegReductionPQBase *SPQ;
1433 bu_ls_rr_sort(RegReductionPQBase *spq) : SPQ(spq) {}
1434 bu_ls_rr_sort(const bu_ls_rr_sort &RHS) : SPQ(RHS.SPQ) {}
Andrew Trick10ffc2b2010-12-24 05:03:26 +00001435
Andrew Trick9ccce772011-01-14 21:11:41 +00001436 bool operator()(SUnit* left, SUnit* right) const;
1437};
Andrew Trick2085a962010-12-21 22:25:04 +00001438
Andrew Trick9ccce772011-01-14 21:11:41 +00001439// td_ls_rr_sort - Priority function for top down register pressure reduction
1440// scheduler.
1441struct td_ls_rr_sort : public queue_sort {
1442 enum {
1443 IsBottomUp = false,
1444 HasReadyFilter = false
Evan Chengd38c22b2006-05-11 23:55:42 +00001445 };
1446
Andrew Trick9ccce772011-01-14 21:11:41 +00001447 RegReductionPQBase *SPQ;
1448 td_ls_rr_sort(RegReductionPQBase *spq) : SPQ(spq) {}
1449 td_ls_rr_sort(const td_ls_rr_sort &RHS) : SPQ(RHS.SPQ) {}
Andrew Trick10ffc2b2010-12-24 05:03:26 +00001450
Andrew Trick9ccce772011-01-14 21:11:41 +00001451 bool operator()(const SUnit* left, const SUnit* right) const;
1452};
Andrew Trick2085a962010-12-21 22:25:04 +00001453
Andrew Trick9ccce772011-01-14 21:11:41 +00001454// src_ls_rr_sort - Priority function for source order scheduler.
1455struct src_ls_rr_sort : public queue_sort {
1456 enum {
1457 IsBottomUp = true,
1458 HasReadyFilter = false
Evan Chengd38c22b2006-05-11 23:55:42 +00001459 };
Bill Wendling8cbc25d2010-01-23 10:26:57 +00001460
Andrew Trick9ccce772011-01-14 21:11:41 +00001461 RegReductionPQBase *SPQ;
1462 src_ls_rr_sort(RegReductionPQBase *spq)
1463 : SPQ(spq) {}
1464 src_ls_rr_sort(const src_ls_rr_sort &RHS)
1465 : SPQ(RHS.SPQ) {}
Andrew Trick10ffc2b2010-12-24 05:03:26 +00001466
Andrew Trick9ccce772011-01-14 21:11:41 +00001467 bool operator()(SUnit* left, SUnit* right) const;
1468};
Andrew Trick2085a962010-12-21 22:25:04 +00001469
Andrew Trick9ccce772011-01-14 21:11:41 +00001470// hybrid_ls_rr_sort - Priority function for hybrid scheduler.
1471struct hybrid_ls_rr_sort : public queue_sort {
1472 enum {
1473 IsBottomUp = true,
Andrew Trickc88b7ec2011-03-04 02:03:45 +00001474 HasReadyFilter = false
Bill Wendling8cbc25d2010-01-23 10:26:57 +00001475 };
Evan Chengbdd062d2010-05-20 06:13:19 +00001476
Andrew Trick9ccce772011-01-14 21:11:41 +00001477 RegReductionPQBase *SPQ;
1478 hybrid_ls_rr_sort(RegReductionPQBase *spq)
1479 : SPQ(spq) {}
1480 hybrid_ls_rr_sort(const hybrid_ls_rr_sort &RHS)
1481 : SPQ(RHS.SPQ) {}
Andrew Trick10ffc2b2010-12-24 05:03:26 +00001482
Andrew Trick9ccce772011-01-14 21:11:41 +00001483 bool isReady(SUnit *SU, unsigned CurCycle) const;
Evan Chenga77f3d32010-07-21 06:09:07 +00001484
Andrew Trick9ccce772011-01-14 21:11:41 +00001485 bool operator()(SUnit* left, SUnit* right) const;
1486};
1487
1488// ilp_ls_rr_sort - Priority function for ILP (instruction level parallelism)
1489// scheduler.
1490struct ilp_ls_rr_sort : public queue_sort {
1491 enum {
1492 IsBottomUp = true,
Andrew Trickc88b7ec2011-03-04 02:03:45 +00001493 HasReadyFilter = false
Evan Chengbdd062d2010-05-20 06:13:19 +00001494 };
Evan Cheng37b740c2010-07-24 00:39:05 +00001495
Andrew Trick9ccce772011-01-14 21:11:41 +00001496 RegReductionPQBase *SPQ;
1497 ilp_ls_rr_sort(RegReductionPQBase *spq)
1498 : SPQ(spq) {}
1499 ilp_ls_rr_sort(const ilp_ls_rr_sort &RHS)
1500 : SPQ(RHS.SPQ) {}
Andrew Trick10ffc2b2010-12-24 05:03:26 +00001501
Andrew Trick9ccce772011-01-14 21:11:41 +00001502 bool isReady(SUnit *SU, unsigned CurCycle) const;
Evan Cheng37b740c2010-07-24 00:39:05 +00001503
Andrew Trick9ccce772011-01-14 21:11:41 +00001504 bool operator()(SUnit* left, SUnit* right) const;
1505};
Andrew Trick10ffc2b2010-12-24 05:03:26 +00001506
Andrew Trick9ccce772011-01-14 21:11:41 +00001507class RegReductionPQBase : public SchedulingPriorityQueue {
1508protected:
1509 std::vector<SUnit*> Queue;
1510 unsigned CurQueueId;
1511 bool TracksRegPressure;
1512
1513 // SUnits - The SUnits for the current graph.
1514 std::vector<SUnit> *SUnits;
1515
1516 MachineFunction &MF;
1517 const TargetInstrInfo *TII;
1518 const TargetRegisterInfo *TRI;
1519 const TargetLowering *TLI;
1520 ScheduleDAGRRList *scheduleDAG;
1521
1522 // SethiUllmanNumbers - The SethiUllman number for each node.
1523 std::vector<unsigned> SethiUllmanNumbers;
1524
1525 /// RegPressure - Tracking current reg pressure per register class.
1526 ///
1527 std::vector<unsigned> RegPressure;
1528
1529 /// RegLimit - Tracking the number of allocatable registers per register
1530 /// class.
1531 std::vector<unsigned> RegLimit;
1532
1533public:
1534 RegReductionPQBase(MachineFunction &mf,
1535 bool hasReadyFilter,
1536 bool tracksrp,
1537 const TargetInstrInfo *tii,
1538 const TargetRegisterInfo *tri,
1539 const TargetLowering *tli)
1540 : SchedulingPriorityQueue(hasReadyFilter),
1541 CurQueueId(0), TracksRegPressure(tracksrp),
1542 MF(mf), TII(tii), TRI(tri), TLI(tli), scheduleDAG(NULL) {
1543 if (TracksRegPressure) {
1544 unsigned NumRC = TRI->getNumRegClasses();
1545 RegLimit.resize(NumRC);
1546 RegPressure.resize(NumRC);
1547 std::fill(RegLimit.begin(), RegLimit.end(), 0);
1548 std::fill(RegPressure.begin(), RegPressure.end(), 0);
1549 for (TargetRegisterInfo::regclass_iterator I = TRI->regclass_begin(),
1550 E = TRI->regclass_end(); I != E; ++I)
Cameron Zwarichdf616942011-03-07 21:56:36 +00001551 RegLimit[(*I)->getID()] = tri->getRegPressureLimit(*I, MF);
Andrew Trick9ccce772011-01-14 21:11:41 +00001552 }
1553 }
1554
1555 void setScheduleDAG(ScheduleDAGRRList *scheduleDag) {
1556 scheduleDAG = scheduleDag;
1557 }
1558
1559 ScheduleHazardRecognizer* getHazardRec() {
1560 return scheduleDAG->getHazardRec();
1561 }
1562
1563 void initNodes(std::vector<SUnit> &sunits);
1564
1565 void addNode(const SUnit *SU);
1566
1567 void updateNode(const SUnit *SU);
1568
1569 void releaseState() {
1570 SUnits = 0;
1571 SethiUllmanNumbers.clear();
1572 std::fill(RegPressure.begin(), RegPressure.end(), 0);
1573 }
1574
1575 unsigned getNodePriority(const SUnit *SU) const;
1576
1577 unsigned getNodeOrdering(const SUnit *SU) const {
Andrew Trick3bd8b7a2011-03-25 06:40:55 +00001578 if (!SU->getNode()) return 0;
1579
Andrew Trick9ccce772011-01-14 21:11:41 +00001580 return scheduleDAG->DAG->GetOrdering(SU->getNode());
1581 }
1582
1583 bool empty() const { return Queue.empty(); }
1584
1585 void push(SUnit *U) {
1586 assert(!U->NodeQueueId && "Node in the queue already");
1587 U->NodeQueueId = ++CurQueueId;
1588 Queue.push_back(U);
1589 }
1590
1591 void remove(SUnit *SU) {
1592 assert(!Queue.empty() && "Queue is empty!");
1593 assert(SU->NodeQueueId != 0 && "Not in queue!");
1594 std::vector<SUnit *>::iterator I = std::find(Queue.begin(), Queue.end(),
1595 SU);
1596 if (I != prior(Queue.end()))
1597 std::swap(*I, Queue.back());
1598 Queue.pop_back();
1599 SU->NodeQueueId = 0;
1600 }
1601
Andrew Trickd0548ae2011-02-04 03:18:17 +00001602 bool tracksRegPressure() const { return TracksRegPressure; }
1603
Andrew Trick9ccce772011-01-14 21:11:41 +00001604 void dumpRegPressure() const;
1605
1606 bool HighRegPressure(const SUnit *SU) const;
1607
Andrew Trick641e2d42011-03-05 08:00:22 +00001608 bool MayReduceRegPressure(SUnit *SU) const;
1609
1610 int RegPressureDiff(SUnit *SU, unsigned &LiveUses) const;
Andrew Trick9ccce772011-01-14 21:11:41 +00001611
1612 void ScheduledNode(SUnit *SU);
1613
1614 void UnscheduledNode(SUnit *SU);
1615
1616protected:
1617 bool canClobber(const SUnit *SU, const SUnit *Op);
1618 void AddPseudoTwoAddrDeps();
1619 void PrescheduleNodesWithMultipleUses();
1620 void CalculateSethiUllmanNumbers();
1621};
1622
1623template<class SF>
Andrew Trick3013b6a2011-06-15 17:16:12 +00001624static SUnit *popFromQueueImpl(std::vector<SUnit*> &Q, SF &Picker) {
1625 std::vector<SUnit *>::iterator Best = Q.begin();
1626 for (std::vector<SUnit *>::iterator I = llvm::next(Q.begin()),
1627 E = Q.end(); I != E; ++I)
1628 if (Picker(*Best, *I))
1629 Best = I;
1630 SUnit *V = *Best;
1631 if (Best != prior(Q.end()))
1632 std::swap(*Best, Q.back());
1633 Q.pop_back();
1634 return V;
1635}
Andrew Trick9ccce772011-01-14 21:11:41 +00001636
Andrew Trick3013b6a2011-06-15 17:16:12 +00001637template<class SF>
1638SUnit *popFromQueue(std::vector<SUnit*> &Q, SF &Picker, ScheduleDAG *DAG) {
1639#ifndef NDEBUG
1640 if (DAG->StressSched) {
1641 reverse_sort<SF> RPicker(Picker);
1642 return popFromQueueImpl(Q, RPicker);
1643 }
1644#endif
1645 (void)DAG;
1646 return popFromQueueImpl(Q, Picker);
1647}
1648
1649template<class SF>
1650class RegReductionPriorityQueue : public RegReductionPQBase {
Andrew Trick9ccce772011-01-14 21:11:41 +00001651 SF Picker;
1652
1653public:
1654 RegReductionPriorityQueue(MachineFunction &mf,
1655 bool tracksrp,
1656 const TargetInstrInfo *tii,
1657 const TargetRegisterInfo *tri,
1658 const TargetLowering *tli)
1659 : RegReductionPQBase(mf, SF::HasReadyFilter, tracksrp, tii, tri, tli),
1660 Picker(this) {}
1661
1662 bool isBottomUp() const { return SF::IsBottomUp; }
1663
1664 bool isReady(SUnit *U) const {
1665 return Picker.HasReadyFilter && Picker.isReady(U, getCurCycle());
1666 }
1667
1668 SUnit *pop() {
1669 if (Queue.empty()) return NULL;
1670
Andrew Trick3013b6a2011-06-15 17:16:12 +00001671 SUnit *V = popFromQueue(Queue, Picker, scheduleDAG);
Andrew Trick9ccce772011-01-14 21:11:41 +00001672 V->NodeQueueId = 0;
1673 return V;
1674 }
1675
1676 void dump(ScheduleDAG *DAG) const {
1677 // Emulate pop() without clobbering NodeQueueIds.
1678 std::vector<SUnit*> DumpQueue = Queue;
1679 SF DumpPicker = Picker;
1680 while (!DumpQueue.empty()) {
Andrew Trick3013b6a2011-06-15 17:16:12 +00001681 SUnit *SU = popFromQueue(DumpQueue, DumpPicker, scheduleDAG);
Andrew Trick9ccce772011-01-14 21:11:41 +00001682 if (isBottomUp())
1683 dbgs() << "Height " << SU->getHeight() << ": ";
1684 else
1685 dbgs() << "Depth " << SU->getDepth() << ": ";
1686 SU->dump(DAG);
1687 }
1688 }
1689};
1690
1691typedef RegReductionPriorityQueue<bu_ls_rr_sort>
1692BURegReductionPriorityQueue;
1693
1694typedef RegReductionPriorityQueue<td_ls_rr_sort>
1695TDRegReductionPriorityQueue;
1696
1697typedef RegReductionPriorityQueue<src_ls_rr_sort>
1698SrcRegReductionPriorityQueue;
1699
1700typedef RegReductionPriorityQueue<hybrid_ls_rr_sort>
1701HybridBURRPriorityQueue;
1702
1703typedef RegReductionPriorityQueue<ilp_ls_rr_sort>
1704ILPBURRPriorityQueue;
1705} // end anonymous namespace
1706
1707//===----------------------------------------------------------------------===//
1708// Static Node Priority for Register Pressure Reduction
1709//===----------------------------------------------------------------------===//
Evan Chengd38c22b2006-05-11 23:55:42 +00001710
Andrew Trickbfbd9722011-04-14 05:15:06 +00001711// Check for special nodes that bypass scheduling heuristics.
1712// Currently this pushes TokenFactor nodes down, but may be used for other
1713// pseudo-ops as well.
1714//
1715// Return -1 to schedule right above left, 1 for left above right.
1716// Return 0 if no bias exists.
1717static int checkSpecialNodes(const SUnit *left, const SUnit *right) {
1718 bool LSchedLow = left->isScheduleLow;
1719 bool RSchedLow = right->isScheduleLow;
1720 if (LSchedLow != RSchedLow)
1721 return LSchedLow < RSchedLow ? 1 : -1;
1722 return 0;
1723}
1724
Dan Gohman186f65d2008-11-20 03:30:37 +00001725/// CalcNodeSethiUllmanNumber - Compute Sethi Ullman number.
1726/// Smaller number is the higher priority.
Evan Cheng7e4abde2008-07-02 09:23:51 +00001727static unsigned
Dan Gohman186f65d2008-11-20 03:30:37 +00001728CalcNodeSethiUllmanNumber(const SUnit *SU, std::vector<unsigned> &SUNumbers) {
Evan Cheng7e4abde2008-07-02 09:23:51 +00001729 unsigned &SethiUllmanNumber = SUNumbers[SU->NodeNum];
1730 if (SethiUllmanNumber != 0)
1731 return SethiUllmanNumber;
1732
1733 unsigned Extra = 0;
1734 for (SUnit::const_pred_iterator I = SU->Preds.begin(), E = SU->Preds.end();
1735 I != E; ++I) {
Dan Gohman2d170892008-12-09 22:54:47 +00001736 if (I->isCtrl()) continue; // ignore chain preds
1737 SUnit *PredSU = I->getSUnit();
Dan Gohman186f65d2008-11-20 03:30:37 +00001738 unsigned PredSethiUllman = CalcNodeSethiUllmanNumber(PredSU, SUNumbers);
Evan Cheng7e4abde2008-07-02 09:23:51 +00001739 if (PredSethiUllman > SethiUllmanNumber) {
1740 SethiUllmanNumber = PredSethiUllman;
1741 Extra = 0;
Evan Cheng3a14efa2009-02-12 08:59:45 +00001742 } else if (PredSethiUllman == SethiUllmanNumber)
Evan Cheng7e4abde2008-07-02 09:23:51 +00001743 ++Extra;
1744 }
1745
1746 SethiUllmanNumber += Extra;
1747
1748 if (SethiUllmanNumber == 0)
1749 SethiUllmanNumber = 1;
Andrew Trick2085a962010-12-21 22:25:04 +00001750
Evan Cheng7e4abde2008-07-02 09:23:51 +00001751 return SethiUllmanNumber;
1752}
1753
Andrew Trick9ccce772011-01-14 21:11:41 +00001754/// CalculateSethiUllmanNumbers - Calculate Sethi-Ullman numbers of all
1755/// scheduling units.
1756void RegReductionPQBase::CalculateSethiUllmanNumbers() {
1757 SethiUllmanNumbers.assign(SUnits->size(), 0);
Andrew Trick10ffc2b2010-12-24 05:03:26 +00001758
Andrew Trick9ccce772011-01-14 21:11:41 +00001759 for (unsigned i = 0, e = SUnits->size(); i != e; ++i)
1760 CalcNodeSethiUllmanNumber(&(*SUnits)[i], SethiUllmanNumbers);
Evan Chengd38c22b2006-05-11 23:55:42 +00001761}
1762
Andrew Trick9ccce772011-01-14 21:11:41 +00001763void RegReductionPQBase::addNode(const SUnit *SU) {
1764 unsigned SUSize = SethiUllmanNumbers.size();
1765 if (SUnits->size() > SUSize)
1766 SethiUllmanNumbers.resize(SUSize*2, 0);
1767 CalcNodeSethiUllmanNumber(SU, SethiUllmanNumbers);
1768}
1769
1770void RegReductionPQBase::updateNode(const SUnit *SU) {
1771 SethiUllmanNumbers[SU->NodeNum] = 0;
1772 CalcNodeSethiUllmanNumber(SU, SethiUllmanNumbers);
1773}
1774
Andrew Trick2cd1f0b2011-01-20 06:21:59 +00001775// Lower priority means schedule further down. For bottom-up scheduling, lower
1776// priority SUs are scheduled before higher priority SUs.
Andrew Trick9ccce772011-01-14 21:11:41 +00001777unsigned RegReductionPQBase::getNodePriority(const SUnit *SU) const {
1778 assert(SU->NodeNum < SethiUllmanNumbers.size());
1779 unsigned Opc = SU->getNode() ? SU->getNode()->getOpcode() : 0;
1780 if (Opc == ISD::TokenFactor || Opc == ISD::CopyToReg)
1781 // CopyToReg should be close to its uses to facilitate coalescing and
1782 // avoid spilling.
1783 return 0;
1784 if (Opc == TargetOpcode::EXTRACT_SUBREG ||
1785 Opc == TargetOpcode::SUBREG_TO_REG ||
1786 Opc == TargetOpcode::INSERT_SUBREG)
1787 // EXTRACT_SUBREG, INSERT_SUBREG, and SUBREG_TO_REG nodes should be
1788 // close to their uses to facilitate coalescing.
1789 return 0;
1790 if (SU->NumSuccs == 0 && SU->NumPreds != 0)
1791 // If SU does not have a register use, i.e. it doesn't produce a value
1792 // that would be consumed (e.g. store), then it terminates a chain of
1793 // computation. Give it a large SethiUllman number so it will be
1794 // scheduled right before its predecessors that it doesn't lengthen
1795 // their live ranges.
1796 return 0xffff;
1797 if (SU->NumPreds == 0 && SU->NumSuccs != 0)
1798 // If SU does not have a register def, schedule it close to its uses
1799 // because it does not lengthen any live ranges.
1800 return 0;
Evan Cheng1355bbd2011-04-26 21:31:35 +00001801#if 1
Andrew Trick9ccce772011-01-14 21:11:41 +00001802 return SethiUllmanNumbers[SU->NodeNum];
Evan Cheng1355bbd2011-04-26 21:31:35 +00001803#else
1804 unsigned Priority = SethiUllmanNumbers[SU->NodeNum];
1805 if (SU->isCallOp) {
1806 // FIXME: This assumes all of the defs are used as call operands.
1807 int NP = (int)Priority - SU->getNode()->getNumValues();
1808 return (NP > 0) ? NP : 0;
1809 }
1810 return Priority;
1811#endif
Andrew Trick9ccce772011-01-14 21:11:41 +00001812}
1813
1814//===----------------------------------------------------------------------===//
1815// Register Pressure Tracking
1816//===----------------------------------------------------------------------===//
1817
1818void RegReductionPQBase::dumpRegPressure() const {
1819 for (TargetRegisterInfo::regclass_iterator I = TRI->regclass_begin(),
1820 E = TRI->regclass_end(); I != E; ++I) {
1821 const TargetRegisterClass *RC = *I;
1822 unsigned Id = RC->getID();
1823 unsigned RP = RegPressure[Id];
1824 if (!RP) continue;
1825 DEBUG(dbgs() << RC->getName() << ": " << RP << " / " << RegLimit[Id]
1826 << '\n');
1827 }
1828}
1829
1830bool RegReductionPQBase::HighRegPressure(const SUnit *SU) const {
1831 if (!TLI)
1832 return false;
1833
1834 for (SUnit::const_pred_iterator I = SU->Preds.begin(),E = SU->Preds.end();
1835 I != E; ++I) {
1836 if (I->isCtrl())
1837 continue;
1838 SUnit *PredSU = I->getSUnit();
Andrew Trickd0548ae2011-02-04 03:18:17 +00001839 // NumRegDefsLeft is zero when enough uses of this node have been scheduled
1840 // to cover the number of registers defined (they are all live).
1841 if (PredSU->NumRegDefsLeft == 0) {
Andrew Trick2cd1f0b2011-01-20 06:21:59 +00001842 continue;
1843 }
Andrew Trickd0548ae2011-02-04 03:18:17 +00001844 for (ScheduleDAGSDNodes::RegDefIter RegDefPos(PredSU, scheduleDAG);
1845 RegDefPos.IsValid(); RegDefPos.Advance()) {
Owen Anderson96adc4a2011-06-15 23:35:18 +00001846 unsigned RCId, Cost;
1847 GetCostForDef(RegDefPos, TLI, TII, TRI, RCId, Cost);
1848
Andrew Trick9ccce772011-01-14 21:11:41 +00001849 if ((RegPressure[RCId] + Cost) >= RegLimit[RCId])
1850 return true;
1851 }
1852 }
Andrew Trick9ccce772011-01-14 21:11:41 +00001853 return false;
1854}
1855
Andrew Trick641e2d42011-03-05 08:00:22 +00001856bool RegReductionPQBase::MayReduceRegPressure(SUnit *SU) const {
Andrew Trick9ccce772011-01-14 21:11:41 +00001857 const SDNode *N = SU->getNode();
1858
1859 if (!N->isMachineOpcode() || !SU->NumSuccs)
1860 return false;
1861
1862 unsigned NumDefs = TII->get(N->getMachineOpcode()).getNumDefs();
1863 for (unsigned i = 0; i != NumDefs; ++i) {
1864 EVT VT = N->getValueType(i);
1865 if (!N->hasAnyUseOfValue(i))
1866 continue;
1867 unsigned RCId = TLI->getRepRegClassFor(VT)->getID();
1868 if (RegPressure[RCId] >= RegLimit[RCId])
1869 return true;
1870 }
1871 return false;
1872}
1873
Andrew Trick641e2d42011-03-05 08:00:22 +00001874// Compute the register pressure contribution by this instruction by count up
1875// for uses that are not live and down for defs. Only count register classes
1876// that are already under high pressure. As a side effect, compute the number of
1877// uses of registers that are already live.
1878//
1879// FIXME: This encompasses the logic in HighRegPressure and MayReduceRegPressure
1880// so could probably be factored.
1881int RegReductionPQBase::RegPressureDiff(SUnit *SU, unsigned &LiveUses) const {
1882 LiveUses = 0;
1883 int PDiff = 0;
1884 for (SUnit::const_pred_iterator I = SU->Preds.begin(),E = SU->Preds.end();
1885 I != E; ++I) {
1886 if (I->isCtrl())
1887 continue;
1888 SUnit *PredSU = I->getSUnit();
1889 // NumRegDefsLeft is zero when enough uses of this node have been scheduled
1890 // to cover the number of registers defined (they are all live).
1891 if (PredSU->NumRegDefsLeft == 0) {
1892 if (PredSU->getNode()->isMachineOpcode())
1893 ++LiveUses;
1894 continue;
1895 }
1896 for (ScheduleDAGSDNodes::RegDefIter RegDefPos(PredSU, scheduleDAG);
1897 RegDefPos.IsValid(); RegDefPos.Advance()) {
1898 EVT VT = RegDefPos.GetValue();
1899 unsigned RCId = TLI->getRepRegClassFor(VT)->getID();
1900 if (RegPressure[RCId] >= RegLimit[RCId])
1901 ++PDiff;
1902 }
1903 }
1904 const SDNode *N = SU->getNode();
1905
Eric Christopher7238cba2011-03-08 19:35:47 +00001906 if (!N || !N->isMachineOpcode() || !SU->NumSuccs)
Andrew Trick641e2d42011-03-05 08:00:22 +00001907 return PDiff;
1908
1909 unsigned NumDefs = TII->get(N->getMachineOpcode()).getNumDefs();
1910 for (unsigned i = 0; i != NumDefs; ++i) {
1911 EVT VT = N->getValueType(i);
1912 if (!N->hasAnyUseOfValue(i))
1913 continue;
1914 unsigned RCId = TLI->getRepRegClassFor(VT)->getID();
1915 if (RegPressure[RCId] >= RegLimit[RCId])
1916 --PDiff;
1917 }
1918 return PDiff;
1919}
1920
Andrew Trick9ccce772011-01-14 21:11:41 +00001921void RegReductionPQBase::ScheduledNode(SUnit *SU) {
1922 if (!TracksRegPressure)
1923 return;
1924
Eric Christopher7238cba2011-03-08 19:35:47 +00001925 if (!SU->getNode())
1926 return;
Andrew Tricka8846e02011-03-23 20:40:18 +00001927
Andrew Trick9ccce772011-01-14 21:11:41 +00001928 for (SUnit::pred_iterator I = SU->Preds.begin(), E = SU->Preds.end();
1929 I != E; ++I) {
1930 if (I->isCtrl())
1931 continue;
1932 SUnit *PredSU = I->getSUnit();
Andrew Trickd0548ae2011-02-04 03:18:17 +00001933 // NumRegDefsLeft is zero when enough uses of this node have been scheduled
1934 // to cover the number of registers defined (they are all live).
1935 if (PredSU->NumRegDefsLeft == 0) {
Andrew Trick9ccce772011-01-14 21:11:41 +00001936 continue;
1937 }
Andrew Trickd0548ae2011-02-04 03:18:17 +00001938 // FIXME: The ScheduleDAG currently loses information about which of a
1939 // node's values is consumed by each dependence. Consequently, if the node
1940 // defines multiple register classes, we don't know which to pressurize
1941 // here. Instead the following loop consumes the register defs in an
1942 // arbitrary order. At least it handles the common case of clustered loads
1943 // to the same class. For precise liveness, each SDep needs to indicate the
1944 // result number. But that tightly couples the ScheduleDAG with the
1945 // SelectionDAG making updates tricky. A simpler hack would be to attach a
1946 // value type or register class to SDep.
1947 //
1948 // The most important aspect of register tracking is balancing the increase
1949 // here with the reduction further below. Note that this SU may use multiple
1950 // defs in PredSU. The can't be determined here, but we've already
1951 // compensated by reducing NumRegDefsLeft in PredSU during
1952 // ScheduleDAGSDNodes::AddSchedEdges.
1953 --PredSU->NumRegDefsLeft;
1954 unsigned SkipRegDefs = PredSU->NumRegDefsLeft;
1955 for (ScheduleDAGSDNodes::RegDefIter RegDefPos(PredSU, scheduleDAG);
1956 RegDefPos.IsValid(); RegDefPos.Advance(), --SkipRegDefs) {
1957 if (SkipRegDefs)
Andrew Trick9ccce772011-01-14 21:11:41 +00001958 continue;
Owen Anderson96adc4a2011-06-15 23:35:18 +00001959
1960 unsigned RCId, Cost;
1961 GetCostForDef(RegDefPos, TLI, TII, TRI, RCId, Cost);
1962 RegPressure[RCId] += Cost;
Andrew Trickd0548ae2011-02-04 03:18:17 +00001963 break;
Andrew Trick9ccce772011-01-14 21:11:41 +00001964 }
1965 }
1966
Andrew Trickd0548ae2011-02-04 03:18:17 +00001967 // We should have this assert, but there may be dead SDNodes that never
1968 // materialize as SUnits, so they don't appear to generate liveness.
1969 //assert(SU->NumRegDefsLeft == 0 && "not all regdefs have scheduled uses");
1970 int SkipRegDefs = (int)SU->NumRegDefsLeft;
1971 for (ScheduleDAGSDNodes::RegDefIter RegDefPos(SU, scheduleDAG);
1972 RegDefPos.IsValid(); RegDefPos.Advance(), --SkipRegDefs) {
1973 if (SkipRegDefs > 0)
1974 continue;
Owen Anderson96adc4a2011-06-15 23:35:18 +00001975 unsigned RCId, Cost;
1976 GetCostForDef(RegDefPos, TLI, TII, TRI, RCId, Cost);
1977 if (RegPressure[RCId] < Cost) {
Andrew Trickd0548ae2011-02-04 03:18:17 +00001978 // Register pressure tracking is imprecise. This can happen. But we try
1979 // hard not to let it happen because it likely results in poor scheduling.
1980 DEBUG(dbgs() << " SU(" << SU->NodeNum << ") has too many regdefs\n");
1981 RegPressure[RCId] = 0;
1982 }
1983 else {
Owen Anderson96adc4a2011-06-15 23:35:18 +00001984 RegPressure[RCId] -= Cost;
Andrew Trick9ccce772011-01-14 21:11:41 +00001985 }
1986 }
Andrew Trick9ccce772011-01-14 21:11:41 +00001987 dumpRegPressure();
1988}
1989
1990void RegReductionPQBase::UnscheduledNode(SUnit *SU) {
1991 if (!TracksRegPressure)
1992 return;
1993
1994 const SDNode *N = SU->getNode();
Eric Christopher7238cba2011-03-08 19:35:47 +00001995 if (!N) return;
Andrew Tricka8846e02011-03-23 20:40:18 +00001996
Andrew Trick9ccce772011-01-14 21:11:41 +00001997 if (!N->isMachineOpcode()) {
1998 if (N->getOpcode() != ISD::CopyToReg)
1999 return;
2000 } else {
2001 unsigned Opc = N->getMachineOpcode();
2002 if (Opc == TargetOpcode::EXTRACT_SUBREG ||
2003 Opc == TargetOpcode::INSERT_SUBREG ||
2004 Opc == TargetOpcode::SUBREG_TO_REG ||
2005 Opc == TargetOpcode::REG_SEQUENCE ||
2006 Opc == TargetOpcode::IMPLICIT_DEF)
2007 return;
2008 }
2009
2010 for (SUnit::pred_iterator I = SU->Preds.begin(), E = SU->Preds.end();
2011 I != E; ++I) {
2012 if (I->isCtrl())
2013 continue;
2014 SUnit *PredSU = I->getSUnit();
Andrew Trick2cd1f0b2011-01-20 06:21:59 +00002015 // NumSuccsLeft counts all deps. Don't compare it with NumSuccs which only
2016 // counts data deps.
2017 if (PredSU->NumSuccsLeft != PredSU->Succs.size())
Andrew Trick9ccce772011-01-14 21:11:41 +00002018 continue;
2019 const SDNode *PN = PredSU->getNode();
2020 if (!PN->isMachineOpcode()) {
2021 if (PN->getOpcode() == ISD::CopyFromReg) {
2022 EVT VT = PN->getValueType(0);
2023 unsigned RCId = TLI->getRepRegClassFor(VT)->getID();
2024 RegPressure[RCId] += TLI->getRepRegClassCostFor(VT);
2025 }
2026 continue;
2027 }
2028 unsigned POpc = PN->getMachineOpcode();
2029 if (POpc == TargetOpcode::IMPLICIT_DEF)
2030 continue;
2031 if (POpc == TargetOpcode::EXTRACT_SUBREG) {
2032 EVT VT = PN->getOperand(0).getValueType();
2033 unsigned RCId = TLI->getRepRegClassFor(VT)->getID();
2034 RegPressure[RCId] += TLI->getRepRegClassCostFor(VT);
2035 continue;
2036 } else if (POpc == TargetOpcode::INSERT_SUBREG ||
2037 POpc == TargetOpcode::SUBREG_TO_REG) {
2038 EVT VT = PN->getValueType(0);
2039 unsigned RCId = TLI->getRepRegClassFor(VT)->getID();
2040 RegPressure[RCId] += TLI->getRepRegClassCostFor(VT);
2041 continue;
2042 }
2043 unsigned NumDefs = TII->get(PN->getMachineOpcode()).getNumDefs();
2044 for (unsigned i = 0; i != NumDefs; ++i) {
2045 EVT VT = PN->getValueType(i);
2046 if (!PN->hasAnyUseOfValue(i))
2047 continue;
2048 unsigned RCId = TLI->getRepRegClassFor(VT)->getID();
2049 if (RegPressure[RCId] < TLI->getRepRegClassCostFor(VT))
2050 // Register pressure tracking is imprecise. This can happen.
2051 RegPressure[RCId] = 0;
2052 else
2053 RegPressure[RCId] -= TLI->getRepRegClassCostFor(VT);
2054 }
2055 }
2056
2057 // Check for isMachineOpcode() as PrescheduleNodesWithMultipleUses()
2058 // may transfer data dependencies to CopyToReg.
2059 if (SU->NumSuccs && N->isMachineOpcode()) {
2060 unsigned NumDefs = TII->get(N->getMachineOpcode()).getNumDefs();
2061 for (unsigned i = NumDefs, e = N->getNumValues(); i != e; ++i) {
2062 EVT VT = N->getValueType(i);
2063 if (VT == MVT::Glue || VT == MVT::Other)
2064 continue;
2065 if (!N->hasAnyUseOfValue(i))
2066 continue;
2067 unsigned RCId = TLI->getRepRegClassFor(VT)->getID();
2068 RegPressure[RCId] += TLI->getRepRegClassCostFor(VT);
2069 }
2070 }
2071
2072 dumpRegPressure();
2073}
2074
2075//===----------------------------------------------------------------------===//
2076// Dynamic Node Priority for Register Pressure Reduction
2077//===----------------------------------------------------------------------===//
2078
Evan Chengb9e3db62007-03-14 22:43:40 +00002079/// closestSucc - Returns the scheduled cycle of the successor which is
Dan Gohmana19c6622009-03-12 23:55:10 +00002080/// closest to the current cycle.
Evan Cheng28748552007-03-13 23:25:11 +00002081static unsigned closestSucc(const SUnit *SU) {
Dan Gohmandddc1ac2008-12-16 03:25:46 +00002082 unsigned MaxHeight = 0;
Evan Cheng28748552007-03-13 23:25:11 +00002083 for (SUnit::const_succ_iterator I = SU->Succs.begin(), E = SU->Succs.end();
Evan Chengb9e3db62007-03-14 22:43:40 +00002084 I != E; ++I) {
Evan Chengce3bbe52009-02-10 08:30:11 +00002085 if (I->isCtrl()) continue; // ignore chain succs
Dan Gohmandddc1ac2008-12-16 03:25:46 +00002086 unsigned Height = I->getSUnit()->getHeight();
Evan Chengb9e3db62007-03-14 22:43:40 +00002087 // If there are bunch of CopyToRegs stacked up, they should be considered
2088 // to be at the same position.
Dan Gohman2d170892008-12-09 22:54:47 +00002089 if (I->getSUnit()->getNode() &&
2090 I->getSUnit()->getNode()->getOpcode() == ISD::CopyToReg)
Dan Gohmandddc1ac2008-12-16 03:25:46 +00002091 Height = closestSucc(I->getSUnit())+1;
2092 if (Height > MaxHeight)
2093 MaxHeight = Height;
Evan Chengb9e3db62007-03-14 22:43:40 +00002094 }
Dan Gohmandddc1ac2008-12-16 03:25:46 +00002095 return MaxHeight;
Evan Cheng28748552007-03-13 23:25:11 +00002096}
2097
Evan Cheng61bc51e2007-12-20 02:22:36 +00002098/// calcMaxScratches - Returns an cost estimate of the worse case requirement
Evan Cheng3a14efa2009-02-12 08:59:45 +00002099/// for scratch registers, i.e. number of data dependencies.
Evan Cheng61bc51e2007-12-20 02:22:36 +00002100static unsigned calcMaxScratches(const SUnit *SU) {
2101 unsigned Scratches = 0;
2102 for (SUnit::const_pred_iterator I = SU->Preds.begin(), E = SU->Preds.end();
Evan Chengb5704992009-02-12 09:52:13 +00002103 I != E; ++I) {
Dan Gohman2d170892008-12-09 22:54:47 +00002104 if (I->isCtrl()) continue; // ignore chain preds
Evan Chengb5704992009-02-12 09:52:13 +00002105 Scratches++;
2106 }
Evan Cheng61bc51e2007-12-20 02:22:36 +00002107 return Scratches;
2108}
2109
Andrew Trickb53a00d2011-04-13 00:38:32 +00002110/// hasOnlyLiveInOpers - Return true if SU has only value predecessors that are
2111/// CopyFromReg from a virtual register.
2112static bool hasOnlyLiveInOpers(const SUnit *SU) {
2113 bool RetVal = false;
2114 for (SUnit::const_pred_iterator I = SU->Preds.begin(), E = SU->Preds.end();
2115 I != E; ++I) {
2116 if (I->isCtrl()) continue;
2117 const SUnit *PredSU = I->getSUnit();
2118 if (PredSU->getNode() &&
2119 PredSU->getNode()->getOpcode() == ISD::CopyFromReg) {
2120 unsigned Reg =
2121 cast<RegisterSDNode>(PredSU->getNode()->getOperand(1))->getReg();
2122 if (TargetRegisterInfo::isVirtualRegister(Reg)) {
2123 RetVal = true;
2124 continue;
2125 }
2126 }
2127 return false;
2128 }
2129 return RetVal;
2130}
2131
2132/// hasOnlyLiveOutUses - Return true if SU has only value successors that are
Evan Cheng6c1414f2010-10-29 18:09:28 +00002133/// CopyToReg to a virtual register. This SU def is probably a liveout and
2134/// it has no other use. It should be scheduled closer to the terminator.
2135static bool hasOnlyLiveOutUses(const SUnit *SU) {
2136 bool RetVal = false;
2137 for (SUnit::const_succ_iterator I = SU->Succs.begin(), E = SU->Succs.end();
2138 I != E; ++I) {
2139 if (I->isCtrl()) continue;
2140 const SUnit *SuccSU = I->getSUnit();
2141 if (SuccSU->getNode() && SuccSU->getNode()->getOpcode() == ISD::CopyToReg) {
2142 unsigned Reg =
2143 cast<RegisterSDNode>(SuccSU->getNode()->getOperand(1))->getReg();
2144 if (TargetRegisterInfo::isVirtualRegister(Reg)) {
2145 RetVal = true;
2146 continue;
2147 }
2148 }
2149 return false;
2150 }
2151 return RetVal;
2152}
2153
Andrew Trickb53a00d2011-04-13 00:38:32 +00002154// Set isVRegCycle for a node with only live in opers and live out uses. Also
2155// set isVRegCycle for its CopyFromReg operands.
2156//
2157// This is only relevant for single-block loops, in which case the VRegCycle
2158// node is likely an induction variable in which the operand and target virtual
2159// registers should be coalesced (e.g. pre/post increment values). Setting the
2160// isVRegCycle flag helps the scheduler prioritize other uses of the same
2161// CopyFromReg so that this node becomes the virtual register "kill". This
2162// avoids interference between the values live in and out of the block and
2163// eliminates a copy inside the loop.
2164static void initVRegCycle(SUnit *SU) {
2165 if (DisableSchedVRegCycle)
2166 return;
2167
2168 if (!hasOnlyLiveInOpers(SU) || !hasOnlyLiveOutUses(SU))
2169 return;
2170
2171 DEBUG(dbgs() << "VRegCycle: SU(" << SU->NodeNum << ")\n");
2172
2173 SU->isVRegCycle = true;
2174
2175 for (SUnit::const_pred_iterator I = SU->Preds.begin(), E = SU->Preds.end();
Andrew Trickc5dd24a2011-04-12 19:54:36 +00002176 I != E; ++I) {
Andrew Trickb53a00d2011-04-13 00:38:32 +00002177 if (I->isCtrl()) continue;
2178 I->getSUnit()->isVRegCycle = true;
Andrew Trickc5dd24a2011-04-12 19:54:36 +00002179 }
Andrew Trick1b60ad62011-04-12 20:14:07 +00002180}
2181
Andrew Trickb53a00d2011-04-13 00:38:32 +00002182// After scheduling the definition of a VRegCycle, clear the isVRegCycle flag of
2183// CopyFromReg operands. We should no longer penalize other uses of this VReg.
2184static void resetVRegCycle(SUnit *SU) {
2185 if (!SU->isVRegCycle)
2186 return;
2187
2188 for (SUnit::const_pred_iterator I = SU->Preds.begin(),E = SU->Preds.end();
2189 I != E; ++I) {
Andrew Trick1b60ad62011-04-12 20:14:07 +00002190 if (I->isCtrl()) continue; // ignore chain preds
Andrew Trickb53a00d2011-04-13 00:38:32 +00002191 SUnit *PredSU = I->getSUnit();
2192 if (PredSU->isVRegCycle) {
2193 assert(PredSU->getNode()->getOpcode() == ISD::CopyFromReg &&
2194 "VRegCycle def must be CopyFromReg");
2195 I->getSUnit()->isVRegCycle = 0;
2196 }
2197 }
2198}
2199
2200// Return true if this SUnit uses a CopyFromReg node marked as a VRegCycle. This
2201// means a node that defines the VRegCycle has not been scheduled yet.
2202static bool hasVRegCycleUse(const SUnit *SU) {
2203 // If this SU also defines the VReg, don't hoist it as a "use".
2204 if (SU->isVRegCycle)
2205 return false;
2206
2207 for (SUnit::const_pred_iterator I = SU->Preds.begin(),E = SU->Preds.end();
2208 I != E; ++I) {
2209 if (I->isCtrl()) continue; // ignore chain preds
2210 if (I->getSUnit()->isVRegCycle &&
2211 I->getSUnit()->getNode()->getOpcode() == ISD::CopyFromReg) {
2212 DEBUG(dbgs() << " VReg cycle use: SU (" << SU->NodeNum << ")\n");
2213 return true;
Andrew Trick2ad0b372011-04-07 19:54:57 +00002214 }
2215 }
2216 return false;
2217}
2218
Andrew Trick9ccce772011-01-14 21:11:41 +00002219// Check for either a dependence (latency) or resource (hazard) stall.
2220//
2221// Note: The ScheduleHazardRecognizer interface requires a non-const SU.
2222static bool BUHasStall(SUnit *SU, int Height, RegReductionPQBase *SPQ) {
2223 if ((int)SPQ->getCurCycle() < Height) return true;
2224 if (SPQ->getHazardRec()->getHazardType(SU, 0)
2225 != ScheduleHazardRecognizer::NoHazard)
2226 return true;
2227 return false;
2228}
2229
2230// Return -1 if left has higher priority, 1 if right has higher priority.
2231// Return 0 if latency-based priority is equivalent.
2232static int BUCompareLatency(SUnit *left, SUnit *right, bool checkPref,
2233 RegReductionPQBase *SPQ) {
Andrew Trickb53a00d2011-04-13 00:38:32 +00002234 // Scheduling an instruction that uses a VReg whose postincrement has not yet
2235 // been scheduled will induce a copy. Model this as an extra cycle of latency.
2236 int LPenalty = hasVRegCycleUse(left) ? 1 : 0;
2237 int RPenalty = hasVRegCycleUse(right) ? 1 : 0;
2238 int LHeight = (int)left->getHeight() + LPenalty;
2239 int RHeight = (int)right->getHeight() + RPenalty;
Andrew Trick9ccce772011-01-14 21:11:41 +00002240
2241 bool LStall = (!checkPref || left->SchedulingPref == Sched::Latency) &&
2242 BUHasStall(left, LHeight, SPQ);
2243 bool RStall = (!checkPref || right->SchedulingPref == Sched::Latency) &&
2244 BUHasStall(right, RHeight, SPQ);
2245
2246 // If scheduling one of the node will cause a pipeline stall, delay it.
2247 // If scheduling either one of the node will cause a pipeline stall, sort
2248 // them according to their height.
2249 if (LStall) {
Andrew Trickb53a00d2011-04-13 00:38:32 +00002250 if (!RStall) {
2251 DEBUG(++FactorCount[FactStall]);
Andrew Trick9ccce772011-01-14 21:11:41 +00002252 return 1;
Andrew Trickb53a00d2011-04-13 00:38:32 +00002253 }
2254 if (LHeight != RHeight) {
2255 DEBUG(++FactorCount[FactStall]);
Andrew Trick9ccce772011-01-14 21:11:41 +00002256 return LHeight > RHeight ? 1 : -1;
Andrew Trickb53a00d2011-04-13 00:38:32 +00002257 }
2258 } else if (RStall) {
2259 DEBUG(++FactorCount[FactStall]);
Andrew Trick9ccce772011-01-14 21:11:41 +00002260 return -1;
Andrew Trickb53a00d2011-04-13 00:38:32 +00002261 }
Andrew Trick9ccce772011-01-14 21:11:41 +00002262
Andrew Trick47ff14b2011-01-21 05:51:33 +00002263 // If either node is scheduling for latency, sort them by height/depth
Andrew Trick9ccce772011-01-14 21:11:41 +00002264 // and latency.
2265 if (!checkPref || (left->SchedulingPref == Sched::Latency ||
2266 right->SchedulingPref == Sched::Latency)) {
Andrew Trick47ff14b2011-01-21 05:51:33 +00002267 if (DisableSchedCycles) {
Andrew Trickb53a00d2011-04-13 00:38:32 +00002268 if (LHeight != RHeight) {
2269 DEBUG(++FactorCount[FactHeight]);
Andrew Trick9ccce772011-01-14 21:11:41 +00002270 return LHeight > RHeight ? 1 : -1;
Andrew Trickb53a00d2011-04-13 00:38:32 +00002271 }
Andrew Trick9ccce772011-01-14 21:11:41 +00002272 }
Andrew Trick47ff14b2011-01-21 05:51:33 +00002273 else {
2274 // If neither instruction stalls (!LStall && !RStall) then
Eric Christopher9cb33de2011-03-06 21:13:45 +00002275 // its height is already covered so only its depth matters. We also reach
Andrew Trick47ff14b2011-01-21 05:51:33 +00002276 // this if both stall but have the same height.
Andrew Trickb53a00d2011-04-13 00:38:32 +00002277 int LDepth = left->getDepth() - LPenalty;
2278 int RDepth = right->getDepth() - RPenalty;
Andrew Trick47ff14b2011-01-21 05:51:33 +00002279 if (LDepth != RDepth) {
Andrew Trickb53a00d2011-04-13 00:38:32 +00002280 DEBUG(++FactorCount[FactDepth]);
Andrew Trick47ff14b2011-01-21 05:51:33 +00002281 DEBUG(dbgs() << " Comparing latency of SU (" << left->NodeNum
2282 << ") depth " << LDepth << " vs SU (" << right->NodeNum
2283 << ") depth " << RDepth << "\n");
2284 return LDepth < RDepth ? 1 : -1;
2285 }
2286 }
Andrew Trickb53a00d2011-04-13 00:38:32 +00002287 if (left->Latency != right->Latency) {
2288 DEBUG(++FactorCount[FactOther]);
Andrew Trick9ccce772011-01-14 21:11:41 +00002289 return left->Latency > right->Latency ? 1 : -1;
Andrew Trickb53a00d2011-04-13 00:38:32 +00002290 }
Andrew Trick9ccce772011-01-14 21:11:41 +00002291 }
2292 return 0;
2293}
2294
2295static bool BURRSort(SUnit *left, SUnit *right, RegReductionPQBase *SPQ) {
Andrew Trickbfbd9722011-04-14 05:15:06 +00002296 // Schedule physical register definitions close to their use. This is
2297 // motivated by microarchitectures that can fuse cmp+jump macro-ops. But as
2298 // long as shortening physreg live ranges is generally good, we can defer
2299 // creating a subtarget hook.
2300 if (!DisableSchedPhysRegJoin) {
2301 bool LHasPhysReg = left->hasPhysRegDefs;
2302 bool RHasPhysReg = right->hasPhysRegDefs;
2303 if (LHasPhysReg != RHasPhysReg) {
2304 DEBUG(++FactorCount[FactRegUses]);
2305 #ifndef NDEBUG
2306 const char *PhysRegMsg[] = {" has no physreg", " defines a physreg"};
2307 #endif
2308 DEBUG(dbgs() << " SU (" << left->NodeNum << ") "
2309 << PhysRegMsg[LHasPhysReg] << " SU(" << right->NodeNum << ") "
2310 << PhysRegMsg[RHasPhysReg] << "\n");
2311 return LHasPhysReg < RHasPhysReg;
2312 }
2313 }
2314
Evan Cheng2f647542011-04-26 04:57:37 +00002315 // Prioritize by Sethi-Ulmann number and push CopyToReg nodes down.
Evan Cheng6730f032007-01-08 23:55:53 +00002316 unsigned LPriority = SPQ->getNodePriority(left);
2317 unsigned RPriority = SPQ->getNodePriority(right);
Evan Cheng1355bbd2011-04-26 21:31:35 +00002318
2319 // Be really careful about hoisting call operands above previous calls.
2320 // Only allows it if it would reduce register pressure.
2321 if (left->isCall && right->isCallOp) {
2322 unsigned RNumVals = right->getNode()->getNumValues();
2323 RPriority = (RPriority > RNumVals) ? (RPriority - RNumVals) : 0;
2324 }
2325 if (right->isCall && left->isCallOp) {
2326 unsigned LNumVals = left->getNode()->getNumValues();
2327 LPriority = (LPriority > LNumVals) ? (LPriority - LNumVals) : 0;
2328 }
2329
Andrew Trick641e2d42011-03-05 08:00:22 +00002330 if (LPriority != RPriority) {
Andrew Trick52b3e382011-03-08 01:51:56 +00002331 DEBUG(++FactorCount[FactStatic]);
Evan Cheng73bdf042008-03-01 00:39:47 +00002332 return LPriority > RPriority;
Andrew Trick641e2d42011-03-05 08:00:22 +00002333 }
Andrew Trick52b3e382011-03-08 01:51:56 +00002334
Evan Cheng1355bbd2011-04-26 21:31:35 +00002335 // One or both of the nodes are calls and their sethi-ullman numbers are the
2336 // same, then keep source order.
2337 if (left->isCall || right->isCall) {
2338 unsigned LOrder = SPQ->getNodeOrdering(left);
2339 unsigned ROrder = SPQ->getNodeOrdering(right);
2340
2341 // Prefer an ordering where the lower the non-zero order number, the higher
2342 // the preference.
2343 if ((LOrder || ROrder) && LOrder != ROrder)
2344 return LOrder != 0 && (LOrder < ROrder || ROrder == 0);
2345 }
2346
Evan Cheng73bdf042008-03-01 00:39:47 +00002347 // Try schedule def + use closer when Sethi-Ullman numbers are the same.
2348 // e.g.
2349 // t1 = op t2, c1
2350 // t3 = op t4, c2
2351 //
2352 // and the following instructions are both ready.
2353 // t2 = op c3
2354 // t4 = op c4
2355 //
2356 // Then schedule t2 = op first.
2357 // i.e.
2358 // t4 = op c4
2359 // t2 = op c3
2360 // t1 = op t2, c1
2361 // t3 = op t4, c2
2362 //
2363 // This creates more short live intervals.
2364 unsigned LDist = closestSucc(left);
2365 unsigned RDist = closestSucc(right);
Andrew Trickb53a00d2011-04-13 00:38:32 +00002366 if (LDist != RDist) {
2367 DEBUG(++FactorCount[FactOther]);
Evan Cheng73bdf042008-03-01 00:39:47 +00002368 return LDist < RDist;
Andrew Trickb53a00d2011-04-13 00:38:32 +00002369 }
Evan Cheng73bdf042008-03-01 00:39:47 +00002370
Evan Cheng3a14efa2009-02-12 08:59:45 +00002371 // How many registers becomes live when the node is scheduled.
Evan Cheng73bdf042008-03-01 00:39:47 +00002372 unsigned LScratch = calcMaxScratches(left);
2373 unsigned RScratch = calcMaxScratches(right);
Andrew Trickb53a00d2011-04-13 00:38:32 +00002374 if (LScratch != RScratch) {
2375 DEBUG(++FactorCount[FactOther]);
Evan Cheng73bdf042008-03-01 00:39:47 +00002376 return LScratch > RScratch;
Andrew Trickb53a00d2011-04-13 00:38:32 +00002377 }
Evan Cheng73bdf042008-03-01 00:39:47 +00002378
Evan Cheng1355bbd2011-04-26 21:31:35 +00002379 // Comparing latency against a call makes little sense unless the node
2380 // is register pressure-neutral.
2381 if ((left->isCall && RPriority > 0) || (right->isCall && LPriority > 0))
2382 return (left->NodeQueueId > right->NodeQueueId);
2383
2384 // Do not compare latencies when one or both of the nodes are calls.
2385 if (!DisableSchedCycles &&
2386 !(left->isCall || right->isCall)) {
Andrew Trick9ccce772011-01-14 21:11:41 +00002387 int result = BUCompareLatency(left, right, false /*checkPref*/, SPQ);
2388 if (result != 0)
2389 return result > 0;
2390 }
2391 else {
Andrew Trickb53a00d2011-04-13 00:38:32 +00002392 if (left->getHeight() != right->getHeight()) {
2393 DEBUG(++FactorCount[FactHeight]);
Andrew Trick9ccce772011-01-14 21:11:41 +00002394 return left->getHeight() > right->getHeight();
Andrew Trickb53a00d2011-04-13 00:38:32 +00002395 }
Andrew Trick2085a962010-12-21 22:25:04 +00002396
Andrew Trickb53a00d2011-04-13 00:38:32 +00002397 if (left->getDepth() != right->getDepth()) {
2398 DEBUG(++FactorCount[FactDepth]);
Andrew Trick9ccce772011-01-14 21:11:41 +00002399 return left->getDepth() < right->getDepth();
Andrew Trickb53a00d2011-04-13 00:38:32 +00002400 }
Andrew Trick9ccce772011-01-14 21:11:41 +00002401 }
Evan Cheng73bdf042008-03-01 00:39:47 +00002402
Andrew Trick2085a962010-12-21 22:25:04 +00002403 assert(left->NodeQueueId && right->NodeQueueId &&
Roman Levenstein6b371142008-04-29 09:07:59 +00002404 "NodeQueueId cannot be zero");
Andrew Trickb53a00d2011-04-13 00:38:32 +00002405 DEBUG(++FactorCount[FactOther]);
Roman Levenstein6b371142008-04-29 09:07:59 +00002406 return (left->NodeQueueId > right->NodeQueueId);
Evan Chengd38c22b2006-05-11 23:55:42 +00002407}
2408
Bill Wendling8cbc25d2010-01-23 10:26:57 +00002409// Bottom up
Andrew Trick9ccce772011-01-14 21:11:41 +00002410bool bu_ls_rr_sort::operator()(SUnit *left, SUnit *right) const {
Andrew Trickbfbd9722011-04-14 05:15:06 +00002411 if (int res = checkSpecialNodes(left, right))
2412 return res > 0;
2413
Bill Wendling8cbc25d2010-01-23 10:26:57 +00002414 return BURRSort(left, right, SPQ);
2415}
2416
2417// Source order, otherwise bottom up.
Andrew Trick9ccce772011-01-14 21:11:41 +00002418bool src_ls_rr_sort::operator()(SUnit *left, SUnit *right) const {
Andrew Trickbfbd9722011-04-14 05:15:06 +00002419 if (int res = checkSpecialNodes(left, right))
2420 return res > 0;
2421
Bill Wendling8cbc25d2010-01-23 10:26:57 +00002422 unsigned LOrder = SPQ->getNodeOrdering(left);
2423 unsigned ROrder = SPQ->getNodeOrdering(right);
2424
2425 // Prefer an ordering where the lower the non-zero order number, the higher
2426 // the preference.
2427 if ((LOrder || ROrder) && LOrder != ROrder)
2428 return LOrder != 0 && (LOrder < ROrder || ROrder == 0);
2429
2430 return BURRSort(left, right, SPQ);
2431}
2432
Andrew Trick9ccce772011-01-14 21:11:41 +00002433// If the time between now and when the instruction will be ready can cover
2434// the spill code, then avoid adding it to the ready queue. This gives long
2435// stalls highest priority and allows hoisting across calls. It should also
2436// speed up processing the available queue.
2437bool hybrid_ls_rr_sort::isReady(SUnit *SU, unsigned CurCycle) const {
2438 static const unsigned ReadyDelay = 3;
2439
2440 if (SPQ->MayReduceRegPressure(SU)) return true;
2441
2442 if (SU->getHeight() > (CurCycle + ReadyDelay)) return false;
2443
2444 if (SPQ->getHazardRec()->getHazardType(SU, -ReadyDelay)
2445 != ScheduleHazardRecognizer::NoHazard)
2446 return false;
2447
2448 return true;
2449}
2450
2451// Return true if right should be scheduled with higher priority than left.
2452bool hybrid_ls_rr_sort::operator()(SUnit *left, SUnit *right) const {
Andrew Trickbfbd9722011-04-14 05:15:06 +00002453 if (int res = checkSpecialNodes(left, right))
2454 return res > 0;
2455
Evan Chengdebf9c52010-11-03 00:45:17 +00002456 if (left->isCall || right->isCall)
2457 // No way to compute latency of calls.
2458 return BURRSort(left, right, SPQ);
2459
Evan Chenge6d6c5d2010-07-26 21:49:07 +00002460 bool LHigh = SPQ->HighRegPressure(left);
2461 bool RHigh = SPQ->HighRegPressure(right);
Evan Cheng37b740c2010-07-24 00:39:05 +00002462 // Avoid causing spills. If register pressure is high, schedule for
2463 // register pressure reduction.
Andrew Trick2cd1f0b2011-01-20 06:21:59 +00002464 if (LHigh && !RHigh) {
Andrew Trickb53a00d2011-04-13 00:38:32 +00002465 DEBUG(++FactorCount[FactPressureDiff]);
Andrew Trick2cd1f0b2011-01-20 06:21:59 +00002466 DEBUG(dbgs() << " pressure SU(" << left->NodeNum << ") > SU("
2467 << right->NodeNum << ")\n");
Evan Cheng28590382010-07-21 23:53:58 +00002468 return true;
Andrew Trick2cd1f0b2011-01-20 06:21:59 +00002469 }
2470 else if (!LHigh && RHigh) {
Andrew Trickb53a00d2011-04-13 00:38:32 +00002471 DEBUG(++FactorCount[FactPressureDiff]);
Andrew Trick2cd1f0b2011-01-20 06:21:59 +00002472 DEBUG(dbgs() << " pressure SU(" << right->NodeNum << ") > SU("
2473 << left->NodeNum << ")\n");
Evan Cheng28590382010-07-21 23:53:58 +00002474 return false;
Andrew Trick2cd1f0b2011-01-20 06:21:59 +00002475 }
Andrew Trickb53a00d2011-04-13 00:38:32 +00002476 if (!LHigh && !RHigh) {
2477 int result = BUCompareLatency(left, right, true /*checkPref*/, SPQ);
2478 if (result != 0)
2479 return result > 0;
Evan Chengcc2efe12010-05-28 23:26:21 +00002480 }
Evan Chengbdd062d2010-05-20 06:13:19 +00002481 return BURRSort(left, right, SPQ);
2482}
2483
Andrew Trick10ffc2b2010-12-24 05:03:26 +00002484// Schedule as many instructions in each cycle as possible. So don't make an
2485// instruction available unless it is ready in the current cycle.
2486bool ilp_ls_rr_sort::isReady(SUnit *SU, unsigned CurCycle) const {
Andrew Trick9ccce772011-01-14 21:11:41 +00002487 if (SU->getHeight() > CurCycle) return false;
2488
2489 if (SPQ->getHazardRec()->getHazardType(SU, 0)
2490 != ScheduleHazardRecognizer::NoHazard)
2491 return false;
2492
Andrew Trickc88b7ec2011-03-04 02:03:45 +00002493 return true;
Andrew Trick10ffc2b2010-12-24 05:03:26 +00002494}
2495
Benjamin Kramerb2e4d842011-03-09 16:19:12 +00002496static bool canEnableCoalescing(SUnit *SU) {
Andrew Trick52b3e382011-03-08 01:51:56 +00002497 unsigned Opc = SU->getNode() ? SU->getNode()->getOpcode() : 0;
2498 if (Opc == ISD::TokenFactor || Opc == ISD::CopyToReg)
2499 // CopyToReg should be close to its uses to facilitate coalescing and
2500 // avoid spilling.
2501 return true;
2502
2503 if (Opc == TargetOpcode::EXTRACT_SUBREG ||
2504 Opc == TargetOpcode::SUBREG_TO_REG ||
2505 Opc == TargetOpcode::INSERT_SUBREG)
2506 // EXTRACT_SUBREG, INSERT_SUBREG, and SUBREG_TO_REG nodes should be
2507 // close to their uses to facilitate coalescing.
2508 return true;
2509
2510 if (SU->NumPreds == 0 && SU->NumSuccs != 0)
2511 // If SU does not have a register def, schedule it close to its uses
2512 // because it does not lengthen any live ranges.
2513 return true;
2514
2515 return false;
2516}
2517
Andrew Trickb8390b72011-03-05 08:04:11 +00002518// list-ilp is currently an experimental scheduler that allows various
2519// heuristics to be enabled prior to the normal register reduction logic.
Andrew Trick9ccce772011-01-14 21:11:41 +00002520bool ilp_ls_rr_sort::operator()(SUnit *left, SUnit *right) const {
Andrew Trickbfbd9722011-04-14 05:15:06 +00002521 if (int res = checkSpecialNodes(left, right))
2522 return res > 0;
2523
Evan Chengdebf9c52010-11-03 00:45:17 +00002524 if (left->isCall || right->isCall)
2525 // No way to compute latency of calls.
2526 return BURRSort(left, right, SPQ);
2527
Andrew Trick52b3e382011-03-08 01:51:56 +00002528 unsigned LLiveUses = 0, RLiveUses = 0;
2529 int LPDiff = 0, RPDiff = 0;
2530 if (!DisableSchedRegPressure || !DisableSchedLiveUses) {
2531 LPDiff = SPQ->RegPressureDiff(left, LLiveUses);
2532 RPDiff = SPQ->RegPressureDiff(right, RLiveUses);
2533 }
Andrew Trick641e2d42011-03-05 08:00:22 +00002534 if (!DisableSchedRegPressure && LPDiff != RPDiff) {
2535 DEBUG(++FactorCount[FactPressureDiff]);
Andrew Trick52b3e382011-03-08 01:51:56 +00002536 DEBUG(dbgs() << "RegPressureDiff SU(" << left->NodeNum << "): " << LPDiff
2537 << " != SU(" << right->NodeNum << "): " << RPDiff << "\n");
Andrew Trick641e2d42011-03-05 08:00:22 +00002538 return LPDiff > RPDiff;
2539 }
2540
Andrew Trick52b3e382011-03-08 01:51:56 +00002541 if (!DisableSchedRegPressure && (LPDiff > 0 || RPDiff > 0)) {
Benjamin Kramerb2e4d842011-03-09 16:19:12 +00002542 bool LReduce = canEnableCoalescing(left);
2543 bool RReduce = canEnableCoalescing(right);
Andrew Trick52b3e382011-03-08 01:51:56 +00002544 DEBUG(if (LReduce != RReduce) ++FactorCount[FactPressureDiff]);
2545 if (LReduce && !RReduce) return false;
2546 if (RReduce && !LReduce) return true;
2547 }
2548
2549 if (!DisableSchedLiveUses && (LLiveUses != RLiveUses)) {
2550 DEBUG(dbgs() << "Live uses SU(" << left->NodeNum << "): " << LLiveUses
2551 << " != SU(" << right->NodeNum << "): " << RLiveUses << "\n");
Andrew Trick641e2d42011-03-05 08:00:22 +00002552 DEBUG(++FactorCount[FactRegUses]);
2553 return LLiveUses < RLiveUses;
2554 }
2555
Andrew Trick52b3e382011-03-08 01:51:56 +00002556 if (!DisableSchedStalls) {
2557 bool LStall = BUHasStall(left, left->getHeight(), SPQ);
2558 bool RStall = BUHasStall(right, right->getHeight(), SPQ);
2559 if (LStall != RStall) {
2560 DEBUG(++FactorCount[FactHeight]);
2561 return left->getHeight() > right->getHeight();
2562 }
Andrew Trick641e2d42011-03-05 08:00:22 +00002563 }
2564
Andrew Trick25cedf32011-03-05 10:29:25 +00002565 if (!DisableSchedCriticalPath) {
2566 int spread = (int)left->getDepth() - (int)right->getDepth();
2567 if (std::abs(spread) > MaxReorderWindow) {
Andrew Trick52b3e382011-03-08 01:51:56 +00002568 DEBUG(dbgs() << "Depth of SU(" << left->NodeNum << "): "
2569 << left->getDepth() << " != SU(" << right->NodeNum << "): "
2570 << right->getDepth() << "\n");
Andrew Trick25cedf32011-03-05 10:29:25 +00002571 DEBUG(++FactorCount[FactDepth]);
2572 return left->getDepth() < right->getDepth();
2573 }
Andrew Trick641e2d42011-03-05 08:00:22 +00002574 }
2575
2576 if (!DisableSchedHeight && left->getHeight() != right->getHeight()) {
Andrew Trick52b3e382011-03-08 01:51:56 +00002577 int spread = (int)left->getHeight() - (int)right->getHeight();
2578 if (std::abs(spread) > MaxReorderWindow) {
2579 DEBUG(++FactorCount[FactHeight]);
2580 return left->getHeight() > right->getHeight();
2581 }
Evan Cheng37b740c2010-07-24 00:39:05 +00002582 }
2583
2584 return BURRSort(left, right, SPQ);
2585}
2586
Andrew Trickb53a00d2011-04-13 00:38:32 +00002587void RegReductionPQBase::initNodes(std::vector<SUnit> &sunits) {
2588 SUnits = &sunits;
2589 // Add pseudo dependency edges for two-address nodes.
2590 AddPseudoTwoAddrDeps();
2591 // Reroute edges to nodes with multiple uses.
2592 if (!TracksRegPressure)
2593 PrescheduleNodesWithMultipleUses();
2594 // Calculate node priorities.
2595 CalculateSethiUllmanNumbers();
2596
2597 // For single block loops, mark nodes that look like canonical IV increments.
2598 if (scheduleDAG->BB->isSuccessor(scheduleDAG->BB)) {
2599 for (unsigned i = 0, e = sunits.size(); i != e; ++i) {
2600 initVRegCycle(&sunits[i]);
2601 }
2602 }
2603}
2604
Andrew Trick9ccce772011-01-14 21:11:41 +00002605//===----------------------------------------------------------------------===//
2606// Preschedule for Register Pressure
2607//===----------------------------------------------------------------------===//
2608
2609bool RegReductionPQBase::canClobber(const SUnit *SU, const SUnit *Op) {
Evan Chengfd2c5dd2006-11-04 09:44:31 +00002610 if (SU->isTwoAddress) {
Dan Gohman1ddfcba2008-11-13 21:36:12 +00002611 unsigned Opc = SU->getNode()->getMachineOpcode();
Chris Lattner03ad8852008-01-07 07:27:27 +00002612 const TargetInstrDesc &TID = TII->get(Opc);
Chris Lattnerfd2e3382008-01-07 06:47:00 +00002613 unsigned NumRes = TID.getNumDefs();
Dan Gohman0340d1e2008-02-15 20:50:13 +00002614 unsigned NumOps = TID.getNumOperands() - NumRes;
Evan Chengfd2c5dd2006-11-04 09:44:31 +00002615 for (unsigned i = 0; i != NumOps; ++i) {
Chris Lattnerfd2e3382008-01-07 06:47:00 +00002616 if (TID.getOperandConstraint(i+NumRes, TOI::TIED_TO) != -1) {
Dan Gohman1ddfcba2008-11-13 21:36:12 +00002617 SDNode *DU = SU->getNode()->getOperand(i).getNode();
Dan Gohman46520a22008-06-21 19:18:17 +00002618 if (DU->getNodeId() != -1 &&
2619 Op->OrigNode == &(*SUnits)[DU->getNodeId()])
Evan Chengfd2c5dd2006-11-04 09:44:31 +00002620 return true;
2621 }
2622 }
Evan Chengd38c22b2006-05-11 23:55:42 +00002623 }
Evan Chengd38c22b2006-05-11 23:55:42 +00002624 return false;
2625}
2626
Evan Chengf9891412007-12-20 09:25:31 +00002627/// canClobberPhysRegDefs - True if SU would clobber one of SuccSU's
Dan Gohmanea045202008-06-21 22:05:24 +00002628/// physical register defs.
Dan Gohmane955c482008-08-05 14:45:15 +00002629static bool canClobberPhysRegDefs(const SUnit *SuccSU, const SUnit *SU,
Evan Chengf9891412007-12-20 09:25:31 +00002630 const TargetInstrInfo *TII,
Dan Gohman3a4be0f2008-02-10 18:45:23 +00002631 const TargetRegisterInfo *TRI) {
Dan Gohman1ddfcba2008-11-13 21:36:12 +00002632 SDNode *N = SuccSU->getNode();
Dan Gohman17059682008-07-17 19:10:17 +00002633 unsigned NumDefs = TII->get(N->getMachineOpcode()).getNumDefs();
2634 const unsigned *ImpDefs = TII->get(N->getMachineOpcode()).getImplicitDefs();
Dan Gohmanea045202008-06-21 22:05:24 +00002635 assert(ImpDefs && "Caller should check hasPhysRegDefs");
Dan Gohmana366da12009-03-23 16:23:01 +00002636 for (const SDNode *SUNode = SU->getNode(); SUNode;
Chris Lattner11a33812010-12-23 17:24:32 +00002637 SUNode = SUNode->getGluedNode()) {
Dan Gohmana366da12009-03-23 16:23:01 +00002638 if (!SUNode->isMachineOpcode())
Evan Chengf9891412007-12-20 09:25:31 +00002639 continue;
Dan Gohmana366da12009-03-23 16:23:01 +00002640 const unsigned *SUImpDefs =
2641 TII->get(SUNode->getMachineOpcode()).getImplicitDefs();
2642 if (!SUImpDefs)
2643 return false;
2644 for (unsigned i = NumDefs, e = N->getNumValues(); i != e; ++i) {
Owen Anderson53aa7a92009-08-10 22:56:29 +00002645 EVT VT = N->getValueType(i);
Chris Lattner3e5fbd72010-12-21 02:38:05 +00002646 if (VT == MVT::Glue || VT == MVT::Other)
Dan Gohmana366da12009-03-23 16:23:01 +00002647 continue;
2648 if (!N->hasAnyUseOfValue(i))
2649 continue;
2650 unsigned Reg = ImpDefs[i - NumDefs];
2651 for (;*SUImpDefs; ++SUImpDefs) {
2652 unsigned SUReg = *SUImpDefs;
2653 if (TRI->regsOverlap(Reg, SUReg))
2654 return true;
2655 }
Evan Chengf9891412007-12-20 09:25:31 +00002656 }
2657 }
2658 return false;
2659}
2660
Dan Gohman9a658d72009-03-24 00:49:12 +00002661/// PrescheduleNodesWithMultipleUses - Nodes with multiple uses
2662/// are not handled well by the general register pressure reduction
2663/// heuristics. When presented with code like this:
2664///
2665/// N
2666/// / |
2667/// / |
2668/// U store
2669/// |
2670/// ...
2671///
2672/// the heuristics tend to push the store up, but since the
2673/// operand of the store has another use (U), this would increase
2674/// the length of that other use (the U->N edge).
2675///
2676/// This function transforms code like the above to route U's
2677/// dependence through the store when possible, like this:
2678///
2679/// N
2680/// ||
2681/// ||
2682/// store
2683/// |
2684/// U
2685/// |
2686/// ...
2687///
2688/// This results in the store being scheduled immediately
2689/// after N, which shortens the U->N live range, reducing
2690/// register pressure.
2691///
Andrew Trick9ccce772011-01-14 21:11:41 +00002692void RegReductionPQBase::PrescheduleNodesWithMultipleUses() {
Dan Gohman9a658d72009-03-24 00:49:12 +00002693 // Visit all the nodes in topological order, working top-down.
2694 for (unsigned i = 0, e = SUnits->size(); i != e; ++i) {
2695 SUnit *SU = &(*SUnits)[i];
2696 // For now, only look at nodes with no data successors, such as stores.
2697 // These are especially important, due to the heuristics in
2698 // getNodePriority for nodes with no data successors.
2699 if (SU->NumSuccs != 0)
2700 continue;
2701 // For now, only look at nodes with exactly one data predecessor.
2702 if (SU->NumPreds != 1)
2703 continue;
2704 // Avoid prescheduling copies to virtual registers, which don't behave
2705 // like other nodes from the perspective of scheduling heuristics.
2706 if (SDNode *N = SU->getNode())
2707 if (N->getOpcode() == ISD::CopyToReg &&
2708 TargetRegisterInfo::isVirtualRegister
2709 (cast<RegisterSDNode>(N->getOperand(1))->getReg()))
2710 continue;
2711
2712 // Locate the single data predecessor.
2713 SUnit *PredSU = 0;
2714 for (SUnit::const_pred_iterator II = SU->Preds.begin(),
2715 EE = SU->Preds.end(); II != EE; ++II)
2716 if (!II->isCtrl()) {
2717 PredSU = II->getSUnit();
2718 break;
2719 }
2720 assert(PredSU);
2721
2722 // Don't rewrite edges that carry physregs, because that requires additional
2723 // support infrastructure.
2724 if (PredSU->hasPhysRegDefs)
2725 continue;
2726 // Short-circuit the case where SU is PredSU's only data successor.
2727 if (PredSU->NumSuccs == 1)
2728 continue;
2729 // Avoid prescheduling to copies from virtual registers, which don't behave
Andrew Trickd0548ae2011-02-04 03:18:17 +00002730 // like other nodes from the perspective of scheduling heuristics.
Dan Gohman9a658d72009-03-24 00:49:12 +00002731 if (SDNode *N = SU->getNode())
2732 if (N->getOpcode() == ISD::CopyFromReg &&
2733 TargetRegisterInfo::isVirtualRegister
2734 (cast<RegisterSDNode>(N->getOperand(1))->getReg()))
2735 continue;
2736
2737 // Perform checks on the successors of PredSU.
2738 for (SUnit::const_succ_iterator II = PredSU->Succs.begin(),
2739 EE = PredSU->Succs.end(); II != EE; ++II) {
2740 SUnit *PredSuccSU = II->getSUnit();
2741 if (PredSuccSU == SU) continue;
2742 // If PredSU has another successor with no data successors, for
2743 // now don't attempt to choose either over the other.
2744 if (PredSuccSU->NumSuccs == 0)
2745 goto outer_loop_continue;
2746 // Don't break physical register dependencies.
2747 if (SU->hasPhysRegClobbers && PredSuccSU->hasPhysRegDefs)
2748 if (canClobberPhysRegDefs(PredSuccSU, SU, TII, TRI))
2749 goto outer_loop_continue;
2750 // Don't introduce graph cycles.
2751 if (scheduleDAG->IsReachable(SU, PredSuccSU))
2752 goto outer_loop_continue;
2753 }
2754
2755 // Ok, the transformation is safe and the heuristics suggest it is
2756 // profitable. Update the graph.
Evan Chengbdd062d2010-05-20 06:13:19 +00002757 DEBUG(dbgs() << " Prescheduling SU #" << SU->NodeNum
2758 << " next to PredSU #" << PredSU->NodeNum
Chris Lattner4dc3edd2009-08-23 06:35:02 +00002759 << " to guide scheduling in the presence of multiple uses\n");
Dan Gohman9a658d72009-03-24 00:49:12 +00002760 for (unsigned i = 0; i != PredSU->Succs.size(); ++i) {
2761 SDep Edge = PredSU->Succs[i];
2762 assert(!Edge.isAssignedRegDep());
2763 SUnit *SuccSU = Edge.getSUnit();
2764 if (SuccSU != SU) {
2765 Edge.setSUnit(PredSU);
2766 scheduleDAG->RemovePred(SuccSU, Edge);
2767 scheduleDAG->AddPred(SU, Edge);
2768 Edge.setSUnit(SU);
2769 scheduleDAG->AddPred(SuccSU, Edge);
2770 --i;
2771 }
2772 }
2773 outer_loop_continue:;
2774 }
2775}
2776
Evan Chengd38c22b2006-05-11 23:55:42 +00002777/// AddPseudoTwoAddrDeps - If two nodes share an operand and one of them uses
2778/// it as a def&use operand. Add a pseudo control edge from it to the other
2779/// node (if it won't create a cycle) so the two-address one will be scheduled
Evan Chenga5e595d2007-09-28 22:32:30 +00002780/// first (lower in the schedule). If both nodes are two-address, favor the
2781/// one that has a CopyToReg use (more likely to be a loop induction update).
2782/// If both are two-address, but one is commutable while the other is not
2783/// commutable, favor the one that's not commutable.
Andrew Trick9ccce772011-01-14 21:11:41 +00002784void RegReductionPQBase::AddPseudoTwoAddrDeps() {
Evan Chengfd2c5dd2006-11-04 09:44:31 +00002785 for (unsigned i = 0, e = SUnits->size(); i != e; ++i) {
Dan Gohmane955c482008-08-05 14:45:15 +00002786 SUnit *SU = &(*SUnits)[i];
Evan Chengfd2c5dd2006-11-04 09:44:31 +00002787 if (!SU->isTwoAddress)
2788 continue;
2789
Dan Gohman1ddfcba2008-11-13 21:36:12 +00002790 SDNode *Node = SU->getNode();
Chris Lattner11a33812010-12-23 17:24:32 +00002791 if (!Node || !Node->isMachineOpcode() || SU->getNode()->getGluedNode())
Evan Chengfd2c5dd2006-11-04 09:44:31 +00002792 continue;
2793
Evan Cheng6c1414f2010-10-29 18:09:28 +00002794 bool isLiveOut = hasOnlyLiveOutUses(SU);
Dan Gohman17059682008-07-17 19:10:17 +00002795 unsigned Opc = Node->getMachineOpcode();
Chris Lattner03ad8852008-01-07 07:27:27 +00002796 const TargetInstrDesc &TID = TII->get(Opc);
Chris Lattnerfd2e3382008-01-07 06:47:00 +00002797 unsigned NumRes = TID.getNumDefs();
Dan Gohman0340d1e2008-02-15 20:50:13 +00002798 unsigned NumOps = TID.getNumOperands() - NumRes;
Evan Chengfd2c5dd2006-11-04 09:44:31 +00002799 for (unsigned j = 0; j != NumOps; ++j) {
Dan Gohman82016c22008-11-19 02:00:32 +00002800 if (TID.getOperandConstraint(j+NumRes, TOI::TIED_TO) == -1)
2801 continue;
2802 SDNode *DU = SU->getNode()->getOperand(j).getNode();
2803 if (DU->getNodeId() == -1)
2804 continue;
2805 const SUnit *DUSU = &(*SUnits)[DU->getNodeId()];
2806 if (!DUSU) continue;
2807 for (SUnit::const_succ_iterator I = DUSU->Succs.begin(),
2808 E = DUSU->Succs.end(); I != E; ++I) {
Dan Gohman2d170892008-12-09 22:54:47 +00002809 if (I->isCtrl()) continue;
2810 SUnit *SuccSU = I->getSUnit();
Dan Gohman82016c22008-11-19 02:00:32 +00002811 if (SuccSU == SU)
Evan Cheng1bf166312007-11-09 01:27:11 +00002812 continue;
Dan Gohman82016c22008-11-19 02:00:32 +00002813 // Be conservative. Ignore if nodes aren't at roughly the same
2814 // depth and height.
Dan Gohmandddc1ac2008-12-16 03:25:46 +00002815 if (SuccSU->getHeight() < SU->getHeight() &&
2816 (SU->getHeight() - SuccSU->getHeight()) > 1)
Dan Gohman82016c22008-11-19 02:00:32 +00002817 continue;
Dan Gohmaneefba6b2009-04-16 20:59:02 +00002818 // Skip past COPY_TO_REGCLASS nodes, so that the pseudo edge
2819 // constrains whatever is using the copy, instead of the copy
2820 // itself. In the case that the copy is coalesced, this
2821 // preserves the intent of the pseudo two-address heurietics.
2822 while (SuccSU->Succs.size() == 1 &&
2823 SuccSU->getNode()->isMachineOpcode() &&
2824 SuccSU->getNode()->getMachineOpcode() ==
Chris Lattnerb06015a2010-02-09 19:54:29 +00002825 TargetOpcode::COPY_TO_REGCLASS)
Dan Gohmaneefba6b2009-04-16 20:59:02 +00002826 SuccSU = SuccSU->Succs.front().getSUnit();
2827 // Don't constrain non-instruction nodes.
Dan Gohman82016c22008-11-19 02:00:32 +00002828 if (!SuccSU->getNode() || !SuccSU->getNode()->isMachineOpcode())
2829 continue;
2830 // Don't constrain nodes with physical register defs if the
2831 // predecessor can clobber them.
Dan Gohmanf3746cb2009-03-24 00:50:07 +00002832 if (SuccSU->hasPhysRegDefs && SU->hasPhysRegClobbers) {
Dan Gohman82016c22008-11-19 02:00:32 +00002833 if (canClobberPhysRegDefs(SuccSU, SU, TII, TRI))
Evan Cheng5924bf72007-09-25 01:54:36 +00002834 continue;
Dan Gohman82016c22008-11-19 02:00:32 +00002835 }
Dan Gohman3027bb62009-04-16 20:57:10 +00002836 // Don't constrain EXTRACT_SUBREG, INSERT_SUBREG, and SUBREG_TO_REG;
2837 // these may be coalesced away. We want them close to their uses.
Dan Gohman82016c22008-11-19 02:00:32 +00002838 unsigned SuccOpc = SuccSU->getNode()->getMachineOpcode();
Chris Lattnerb06015a2010-02-09 19:54:29 +00002839 if (SuccOpc == TargetOpcode::EXTRACT_SUBREG ||
2840 SuccOpc == TargetOpcode::INSERT_SUBREG ||
2841 SuccOpc == TargetOpcode::SUBREG_TO_REG)
Dan Gohman82016c22008-11-19 02:00:32 +00002842 continue;
2843 if ((!canClobber(SuccSU, DUSU) ||
Evan Cheng6c1414f2010-10-29 18:09:28 +00002844 (isLiveOut && !hasOnlyLiveOutUses(SuccSU)) ||
Dan Gohman82016c22008-11-19 02:00:32 +00002845 (!SU->isCommutable && SuccSU->isCommutable)) &&
2846 !scheduleDAG->IsReachable(SuccSU, SU)) {
Evan Chengbdd062d2010-05-20 06:13:19 +00002847 DEBUG(dbgs() << " Adding a pseudo-two-addr edge from SU #"
Chris Lattner4dc3edd2009-08-23 06:35:02 +00002848 << SU->NodeNum << " to SU #" << SuccSU->NodeNum << "\n");
Dan Gohman79c35162009-01-06 01:19:04 +00002849 scheduleDAG->AddPred(SU, SDep(SuccSU, SDep::Order, /*Latency=*/0,
Dan Gohmanbf8e5202009-01-06 01:28:56 +00002850 /*Reg=*/0, /*isNormalMemory=*/false,
2851 /*isMustAlias=*/false,
Dan Gohman2d170892008-12-09 22:54:47 +00002852 /*isArtificial=*/true));
Evan Chengfd2c5dd2006-11-04 09:44:31 +00002853 }
2854 }
2855 }
2856 }
Evan Chengd38c22b2006-05-11 23:55:42 +00002857}
2858
Roman Levenstein30d09512008-03-27 09:44:37 +00002859/// LimitedSumOfUnscheduledPredsOfSuccs - Compute the sum of the unscheduled
Roman Levensteinbc674502008-03-27 09:14:57 +00002860/// predecessors of the successors of the SUnit SU. Stop when the provided
2861/// limit is exceeded.
Andrew Trick2085a962010-12-21 22:25:04 +00002862static unsigned LimitedSumOfUnscheduledPredsOfSuccs(const SUnit *SU,
Roman Levensteinbc674502008-03-27 09:14:57 +00002863 unsigned Limit) {
2864 unsigned Sum = 0;
2865 for (SUnit::const_succ_iterator I = SU->Succs.begin(), E = SU->Succs.end();
2866 I != E; ++I) {
Dan Gohman2d170892008-12-09 22:54:47 +00002867 const SUnit *SuccSU = I->getSUnit();
Roman Levensteinbc674502008-03-27 09:14:57 +00002868 for (SUnit::const_pred_iterator II = SuccSU->Preds.begin(),
2869 EE = SuccSU->Preds.end(); II != EE; ++II) {
Dan Gohman2d170892008-12-09 22:54:47 +00002870 SUnit *PredSU = II->getSUnit();
Evan Cheng16d72072008-03-29 18:34:22 +00002871 if (!PredSU->isScheduled)
2872 if (++Sum > Limit)
2873 return Sum;
Roman Levensteinbc674502008-03-27 09:14:57 +00002874 }
2875 }
2876 return Sum;
2877}
2878
Evan Chengd38c22b2006-05-11 23:55:42 +00002879
2880// Top down
2881bool td_ls_rr_sort::operator()(const SUnit *left, const SUnit *right) const {
Andrew Trickbfbd9722011-04-14 05:15:06 +00002882 if (int res = checkSpecialNodes(left, right))
2883 return res < 0;
2884
Evan Cheng6730f032007-01-08 23:55:53 +00002885 unsigned LPriority = SPQ->getNodePriority(left);
2886 unsigned RPriority = SPQ->getNodePriority(right);
Dan Gohman1ddfcba2008-11-13 21:36:12 +00002887 bool LIsTarget = left->getNode() && left->getNode()->isMachineOpcode();
2888 bool RIsTarget = right->getNode() && right->getNode()->isMachineOpcode();
Evan Chengd38c22b2006-05-11 23:55:42 +00002889 bool LIsFloater = LIsTarget && left->NumPreds == 0;
2890 bool RIsFloater = RIsTarget && right->NumPreds == 0;
Roman Levensteinbc674502008-03-27 09:14:57 +00002891 unsigned LBonus = (LimitedSumOfUnscheduledPredsOfSuccs(left,1) == 1) ? 2 : 0;
2892 unsigned RBonus = (LimitedSumOfUnscheduledPredsOfSuccs(right,1) == 1) ? 2 : 0;
Evan Chengd38c22b2006-05-11 23:55:42 +00002893
2894 if (left->NumSuccs == 0 && right->NumSuccs != 0)
2895 return false;
2896 else if (left->NumSuccs != 0 && right->NumSuccs == 0)
2897 return true;
2898
Evan Chengd38c22b2006-05-11 23:55:42 +00002899 if (LIsFloater)
2900 LBonus -= 2;
2901 if (RIsFloater)
2902 RBonus -= 2;
2903 if (left->NumSuccs == 1)
2904 LBonus += 2;
2905 if (right->NumSuccs == 1)
2906 RBonus += 2;
2907
Evan Cheng73bdf042008-03-01 00:39:47 +00002908 if (LPriority+LBonus != RPriority+RBonus)
2909 return LPriority+LBonus < RPriority+RBonus;
Anton Korobeynikov035eaac2008-02-20 11:10:28 +00002910
Dan Gohmandddc1ac2008-12-16 03:25:46 +00002911 if (left->getDepth() != right->getDepth())
2912 return left->getDepth() < right->getDepth();
Evan Cheng73bdf042008-03-01 00:39:47 +00002913
2914 if (left->NumSuccsLeft != right->NumSuccsLeft)
2915 return left->NumSuccsLeft > right->NumSuccsLeft;
2916
Andrew Trick2085a962010-12-21 22:25:04 +00002917 assert(left->NodeQueueId && right->NodeQueueId &&
Roman Levenstein6b371142008-04-29 09:07:59 +00002918 "NodeQueueId cannot be zero");
2919 return (left->NodeQueueId > right->NodeQueueId);
Evan Chengd38c22b2006-05-11 23:55:42 +00002920}
2921
Evan Chengd38c22b2006-05-11 23:55:42 +00002922//===----------------------------------------------------------------------===//
2923// Public Constructor Functions
2924//===----------------------------------------------------------------------===//
2925
Dan Gohmandfaf6462009-02-11 04:27:20 +00002926llvm::ScheduleDAGSDNodes *
Andrew Trick10ffc2b2010-12-24 05:03:26 +00002927llvm::createBURRListDAGScheduler(SelectionDAGISel *IS,
2928 CodeGenOpt::Level OptLevel) {
Dan Gohman619ef482009-01-15 19:20:50 +00002929 const TargetMachine &TM = IS->TM;
2930 const TargetInstrInfo *TII = TM.getInstrInfo();
2931 const TargetRegisterInfo *TRI = TM.getRegisterInfo();
Andrew Trick2085a962010-12-21 22:25:04 +00002932
Evan Chenga77f3d32010-07-21 06:09:07 +00002933 BURegReductionPriorityQueue *PQ =
Evan Chengbf32e542010-07-22 06:24:48 +00002934 new BURegReductionPriorityQueue(*IS->MF, false, TII, TRI, 0);
Andrew Trick10ffc2b2010-12-24 05:03:26 +00002935 ScheduleDAGRRList *SD = new ScheduleDAGRRList(*IS->MF, false, PQ, OptLevel);
Evan Cheng7e4abde2008-07-02 09:23:51 +00002936 PQ->setScheduleDAG(SD);
Andrew Trick2085a962010-12-21 22:25:04 +00002937 return SD;
Evan Chengd38c22b2006-05-11 23:55:42 +00002938}
2939
Dan Gohmandfaf6462009-02-11 04:27:20 +00002940llvm::ScheduleDAGSDNodes *
Andrew Trick10ffc2b2010-12-24 05:03:26 +00002941llvm::createTDRRListDAGScheduler(SelectionDAGISel *IS,
2942 CodeGenOpt::Level OptLevel) {
Dan Gohman619ef482009-01-15 19:20:50 +00002943 const TargetMachine &TM = IS->TM;
2944 const TargetInstrInfo *TII = TM.getInstrInfo();
2945 const TargetRegisterInfo *TRI = TM.getRegisterInfo();
Andrew Trick2085a962010-12-21 22:25:04 +00002946
Evan Chenga77f3d32010-07-21 06:09:07 +00002947 TDRegReductionPriorityQueue *PQ =
2948 new TDRegReductionPriorityQueue(*IS->MF, false, TII, TRI, 0);
Andrew Trick10ffc2b2010-12-24 05:03:26 +00002949 ScheduleDAGRRList *SD = new ScheduleDAGRRList(*IS->MF, false, PQ, OptLevel);
Dan Gohman3f656df2008-11-20 02:45:51 +00002950 PQ->setScheduleDAG(SD);
2951 return SD;
Evan Chengd38c22b2006-05-11 23:55:42 +00002952}
Bill Wendling8cbc25d2010-01-23 10:26:57 +00002953
2954llvm::ScheduleDAGSDNodes *
Andrew Trick10ffc2b2010-12-24 05:03:26 +00002955llvm::createSourceListDAGScheduler(SelectionDAGISel *IS,
2956 CodeGenOpt::Level OptLevel) {
Bill Wendling8cbc25d2010-01-23 10:26:57 +00002957 const TargetMachine &TM = IS->TM;
2958 const TargetInstrInfo *TII = TM.getInstrInfo();
2959 const TargetRegisterInfo *TRI = TM.getRegisterInfo();
Andrew Trick2085a962010-12-21 22:25:04 +00002960
Evan Chenga77f3d32010-07-21 06:09:07 +00002961 SrcRegReductionPriorityQueue *PQ =
Evan Chengbf32e542010-07-22 06:24:48 +00002962 new SrcRegReductionPriorityQueue(*IS->MF, false, TII, TRI, 0);
Andrew Trick10ffc2b2010-12-24 05:03:26 +00002963 ScheduleDAGRRList *SD = new ScheduleDAGRRList(*IS->MF, false, PQ, OptLevel);
Evan Chengbdd062d2010-05-20 06:13:19 +00002964 PQ->setScheduleDAG(SD);
Andrew Trick2085a962010-12-21 22:25:04 +00002965 return SD;
Evan Chengbdd062d2010-05-20 06:13:19 +00002966}
2967
2968llvm::ScheduleDAGSDNodes *
Andrew Trick10ffc2b2010-12-24 05:03:26 +00002969llvm::createHybridListDAGScheduler(SelectionDAGISel *IS,
2970 CodeGenOpt::Level OptLevel) {
Evan Chengbdd062d2010-05-20 06:13:19 +00002971 const TargetMachine &TM = IS->TM;
2972 const TargetInstrInfo *TII = TM.getInstrInfo();
2973 const TargetRegisterInfo *TRI = TM.getRegisterInfo();
Evan Chenga77f3d32010-07-21 06:09:07 +00002974 const TargetLowering *TLI = &IS->getTargetLowering();
Andrew Trick2085a962010-12-21 22:25:04 +00002975
Evan Chenga77f3d32010-07-21 06:09:07 +00002976 HybridBURRPriorityQueue *PQ =
Evan Chengdf907f42010-07-23 22:39:59 +00002977 new HybridBURRPriorityQueue(*IS->MF, true, TII, TRI, TLI);
Andrew Trick10ffc2b2010-12-24 05:03:26 +00002978
2979 ScheduleDAGRRList *SD = new ScheduleDAGRRList(*IS->MF, true, PQ, OptLevel);
Bill Wendling8cbc25d2010-01-23 10:26:57 +00002980 PQ->setScheduleDAG(SD);
Andrew Trick2085a962010-12-21 22:25:04 +00002981 return SD;
Bill Wendling8cbc25d2010-01-23 10:26:57 +00002982}
Evan Cheng37b740c2010-07-24 00:39:05 +00002983
2984llvm::ScheduleDAGSDNodes *
Andrew Trick10ffc2b2010-12-24 05:03:26 +00002985llvm::createILPListDAGScheduler(SelectionDAGISel *IS,
2986 CodeGenOpt::Level OptLevel) {
Evan Cheng37b740c2010-07-24 00:39:05 +00002987 const TargetMachine &TM = IS->TM;
2988 const TargetInstrInfo *TII = TM.getInstrInfo();
2989 const TargetRegisterInfo *TRI = TM.getRegisterInfo();
2990 const TargetLowering *TLI = &IS->getTargetLowering();
Andrew Trick2085a962010-12-21 22:25:04 +00002991
Evan Cheng37b740c2010-07-24 00:39:05 +00002992 ILPBURRPriorityQueue *PQ =
2993 new ILPBURRPriorityQueue(*IS->MF, true, TII, TRI, TLI);
Andrew Trick10ffc2b2010-12-24 05:03:26 +00002994 ScheduleDAGRRList *SD = new ScheduleDAGRRList(*IS->MF, true, PQ, OptLevel);
Evan Cheng37b740c2010-07-24 00:39:05 +00002995 PQ->setScheduleDAG(SD);
Andrew Trick2085a962010-12-21 22:25:04 +00002996 return SD;
Evan Cheng37b740c2010-07-24 00:39:05 +00002997}