xref: /llvm-project/llvm/lib/CodeGen/MachineScheduler.cpp (revision 80df8b837f7322f52fa0ee1f3506b7f39152d3a6)
1 //===- MachineScheduler.cpp - Machine Instruction Scheduler ---------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // MachineScheduler schedules machine instructions after phi elimination. It
11 // preserves LiveIntervals so it can be invoked before register allocation.
12 //
13 //===----------------------------------------------------------------------===//
14 
15 #define DEBUG_TYPE "misched"
16 
17 #include "llvm/CodeGen/MachineScheduler.h"
18 #include "llvm/ADT/OwningPtr.h"
19 #include "llvm/ADT/PriorityQueue.h"
20 #include "llvm/Analysis/AliasAnalysis.h"
21 #include "llvm/CodeGen/LiveIntervalAnalysis.h"
22 #include "llvm/CodeGen/MachineDominators.h"
23 #include "llvm/CodeGen/MachineLoopInfo.h"
24 #include "llvm/CodeGen/Passes.h"
25 #include "llvm/CodeGen/RegisterClassInfo.h"
26 #include "llvm/CodeGen/ScheduleDFS.h"
27 #include "llvm/CodeGen/ScheduleHazardRecognizer.h"
28 #include "llvm/Support/CommandLine.h"
29 #include "llvm/Support/Debug.h"
30 #include "llvm/Support/ErrorHandling.h"
31 #include "llvm/Support/GraphWriter.h"
32 #include "llvm/Support/raw_ostream.h"
33 #include "llvm/Target/TargetInstrInfo.h"
34 #include <queue>
35 
36 using namespace llvm;
37 
38 namespace llvm {
39 cl::opt<bool> ForceTopDown("misched-topdown", cl::Hidden,
40                            cl::desc("Force top-down list scheduling"));
41 cl::opt<bool> ForceBottomUp("misched-bottomup", cl::Hidden,
42                             cl::desc("Force bottom-up list scheduling"));
43 }
44 
45 #ifndef NDEBUG
46 static cl::opt<bool> ViewMISchedDAGs("view-misched-dags", cl::Hidden,
47   cl::desc("Pop up a window to show MISched dags after they are processed"));
48 
49 static cl::opt<unsigned> MISchedCutoff("misched-cutoff", cl::Hidden,
50   cl::desc("Stop scheduling after N instructions"), cl::init(~0U));
51 #else
52 static bool ViewMISchedDAGs = false;
53 #endif // NDEBUG
54 
55 // FIXME: remove this flag after initial testing. It should always be a good
56 // thing.
57 static cl::opt<bool> EnableCopyConstrain("misched-vcopy", cl::Hidden,
58     cl::desc("Constrain vreg copies."), cl::init(true));
59 
60 static cl::opt<bool> EnableLoadCluster("misched-cluster", cl::Hidden,
61   cl::desc("Enable load clustering."), cl::init(true));
62 
63 // Experimental heuristics
64 static cl::opt<bool> EnableMacroFusion("misched-fusion", cl::Hidden,
65   cl::desc("Enable scheduling for macro fusion."), cl::init(true));
66 
67 static cl::opt<bool> VerifyScheduling("verify-misched", cl::Hidden,
68   cl::desc("Verify machine instrs before and after machine scheduling"));
69 
70 // DAG subtrees must have at least this many nodes.
71 static const unsigned MinSubtreeSize = 8;
72 
73 //===----------------------------------------------------------------------===//
74 // Machine Instruction Scheduling Pass and Registry
75 //===----------------------------------------------------------------------===//
76 
77 MachineSchedContext::MachineSchedContext():
78     MF(0), MLI(0), MDT(0), PassConfig(0), AA(0), LIS(0) {
79   RegClassInfo = new RegisterClassInfo();
80 }
81 
82 MachineSchedContext::~MachineSchedContext() {
83   delete RegClassInfo;
84 }
85 
86 namespace {
87 /// MachineScheduler runs after coalescing and before register allocation.
88 class MachineScheduler : public MachineSchedContext,
89                          public MachineFunctionPass {
90 public:
91   MachineScheduler();
92 
93   virtual void getAnalysisUsage(AnalysisUsage &AU) const;
94 
95   virtual void releaseMemory() {}
96 
97   virtual bool runOnMachineFunction(MachineFunction&);
98 
99   virtual void print(raw_ostream &O, const Module* = 0) const;
100 
101   static char ID; // Class identification, replacement for typeinfo
102 };
103 } // namespace
104 
105 char MachineScheduler::ID = 0;
106 
107 char &llvm::MachineSchedulerID = MachineScheduler::ID;
108 
109 INITIALIZE_PASS_BEGIN(MachineScheduler, "misched",
110                       "Machine Instruction Scheduler", false, false)
111 INITIALIZE_AG_DEPENDENCY(AliasAnalysis)
112 INITIALIZE_PASS_DEPENDENCY(SlotIndexes)
113 INITIALIZE_PASS_DEPENDENCY(LiveIntervals)
114 INITIALIZE_PASS_END(MachineScheduler, "misched",
115                     "Machine Instruction Scheduler", false, false)
116 
117 MachineScheduler::MachineScheduler()
118 : MachineFunctionPass(ID) {
119   initializeMachineSchedulerPass(*PassRegistry::getPassRegistry());
120 }
121 
122 void MachineScheduler::getAnalysisUsage(AnalysisUsage &AU) const {
123   AU.setPreservesCFG();
124   AU.addRequiredID(MachineDominatorsID);
125   AU.addRequired<MachineLoopInfo>();
126   AU.addRequired<AliasAnalysis>();
127   AU.addRequired<TargetPassConfig>();
128   AU.addRequired<SlotIndexes>();
129   AU.addPreserved<SlotIndexes>();
130   AU.addRequired<LiveIntervals>();
131   AU.addPreserved<LiveIntervals>();
132   MachineFunctionPass::getAnalysisUsage(AU);
133 }
134 
135 MachinePassRegistry MachineSchedRegistry::Registry;
136 
137 /// A dummy default scheduler factory indicates whether the scheduler
138 /// is overridden on the command line.
139 static ScheduleDAGInstrs *useDefaultMachineSched(MachineSchedContext *C) {
140   return 0;
141 }
142 
143 /// MachineSchedOpt allows command line selection of the scheduler.
144 static cl::opt<MachineSchedRegistry::ScheduleDAGCtor, false,
145                RegisterPassParser<MachineSchedRegistry> >
146 MachineSchedOpt("misched",
147                 cl::init(&useDefaultMachineSched), cl::Hidden,
148                 cl::desc("Machine instruction scheduler to use"));
149 
150 static MachineSchedRegistry
151 DefaultSchedRegistry("default", "Use the target's default scheduler choice.",
152                      useDefaultMachineSched);
153 
154 /// Forward declare the standard machine scheduler. This will be used as the
155 /// default scheduler if the target does not set a default.
156 static ScheduleDAGInstrs *createConvergingSched(MachineSchedContext *C);
157 
158 
159 /// Decrement this iterator until reaching the top or a non-debug instr.
160 static MachineBasicBlock::iterator
161 priorNonDebug(MachineBasicBlock::iterator I, MachineBasicBlock::iterator Beg) {
162   assert(I != Beg && "reached the top of the region, cannot decrement");
163   while (--I != Beg) {
164     if (!I->isDebugValue())
165       break;
166   }
167   return I;
168 }
169 
170 /// If this iterator is a debug value, increment until reaching the End or a
171 /// non-debug instruction.
172 static MachineBasicBlock::iterator
173 nextIfDebug(MachineBasicBlock::iterator I, MachineBasicBlock::iterator End) {
174   for(; I != End; ++I) {
175     if (!I->isDebugValue())
176       break;
177   }
178   return I;
179 }
180 
181 /// Top-level MachineScheduler pass driver.
182 ///
183 /// Visit blocks in function order. Divide each block into scheduling regions
184 /// and visit them bottom-up. Visiting regions bottom-up is not required, but is
185 /// consistent with the DAG builder, which traverses the interior of the
186 /// scheduling regions bottom-up.
187 ///
188 /// This design avoids exposing scheduling boundaries to the DAG builder,
189 /// simplifying the DAG builder's support for "special" target instructions.
190 /// At the same time the design allows target schedulers to operate across
191 /// scheduling boundaries, for example to bundle the boudary instructions
192 /// without reordering them. This creates complexity, because the target
193 /// scheduler must update the RegionBegin and RegionEnd positions cached by
194 /// ScheduleDAGInstrs whenever adding or removing instructions. A much simpler
195 /// design would be to split blocks at scheduling boundaries, but LLVM has a
196 /// general bias against block splitting purely for implementation simplicity.
197 bool MachineScheduler::runOnMachineFunction(MachineFunction &mf) {
198   DEBUG(dbgs() << "Before MISsched:\n"; mf.print(dbgs()));
199 
200   // Initialize the context of the pass.
201   MF = &mf;
202   MLI = &getAnalysis<MachineLoopInfo>();
203   MDT = &getAnalysis<MachineDominatorTree>();
204   PassConfig = &getAnalysis<TargetPassConfig>();
205   AA = &getAnalysis<AliasAnalysis>();
206 
207   LIS = &getAnalysis<LiveIntervals>();
208   const TargetInstrInfo *TII = MF->getTarget().getInstrInfo();
209 
210   if (VerifyScheduling) {
211     DEBUG(LIS->print(dbgs()));
212     MF->verify(this, "Before machine scheduling.");
213   }
214   RegClassInfo->runOnMachineFunction(*MF);
215 
216   // Select the scheduler, or set the default.
217   MachineSchedRegistry::ScheduleDAGCtor Ctor = MachineSchedOpt;
218   if (Ctor == useDefaultMachineSched) {
219     // Get the default scheduler set by the target.
220     Ctor = MachineSchedRegistry::getDefault();
221     if (!Ctor) {
222       Ctor = createConvergingSched;
223       MachineSchedRegistry::setDefault(Ctor);
224     }
225   }
226   // Instantiate the selected scheduler.
227   OwningPtr<ScheduleDAGInstrs> Scheduler(Ctor(this));
228 
229   // Visit all machine basic blocks.
230   //
231   // TODO: Visit blocks in global postorder or postorder within the bottom-up
232   // loop tree. Then we can optionally compute global RegPressure.
233   for (MachineFunction::iterator MBB = MF->begin(), MBBEnd = MF->end();
234        MBB != MBBEnd; ++MBB) {
235 
236     Scheduler->startBlock(MBB);
237 
238     // Break the block into scheduling regions [I, RegionEnd), and schedule each
239     // region as soon as it is discovered. RegionEnd points the scheduling
240     // boundary at the bottom of the region. The DAG does not include RegionEnd,
241     // but the region does (i.e. the next RegionEnd is above the previous
242     // RegionBegin). If the current block has no terminator then RegionEnd ==
243     // MBB->end() for the bottom region.
244     //
245     // The Scheduler may insert instructions during either schedule() or
246     // exitRegion(), even for empty regions. So the local iterators 'I' and
247     // 'RegionEnd' are invalid across these calls.
248     unsigned RemainingInstrs = MBB->size();
249     for(MachineBasicBlock::iterator RegionEnd = MBB->end();
250         RegionEnd != MBB->begin(); RegionEnd = Scheduler->begin()) {
251 
252       // Avoid decrementing RegionEnd for blocks with no terminator.
253       if (RegionEnd != MBB->end()
254           || TII->isSchedulingBoundary(llvm::prior(RegionEnd), MBB, *MF)) {
255         --RegionEnd;
256         // Count the boundary instruction.
257         --RemainingInstrs;
258       }
259 
260       // The next region starts above the previous region. Look backward in the
261       // instruction stream until we find the nearest boundary.
262       MachineBasicBlock::iterator I = RegionEnd;
263       for(;I != MBB->begin(); --I, --RemainingInstrs) {
264         if (TII->isSchedulingBoundary(llvm::prior(I), MBB, *MF))
265           break;
266       }
267       // Notify the scheduler of the region, even if we may skip scheduling
268       // it. Perhaps it still needs to be bundled.
269       Scheduler->enterRegion(MBB, I, RegionEnd, RemainingInstrs);
270 
271       // Skip empty scheduling regions (0 or 1 schedulable instructions).
272       if (I == RegionEnd || I == llvm::prior(RegionEnd)) {
273         // Close the current region. Bundle the terminator if needed.
274         // This invalidates 'RegionEnd' and 'I'.
275         Scheduler->exitRegion();
276         continue;
277       }
278       DEBUG(dbgs() << "********** MI Scheduling **********\n");
279       DEBUG(dbgs() << MF->getName()
280             << ":BB#" << MBB->getNumber() << " " << MBB->getName()
281             << "\n  From: " << *I << "    To: ";
282             if (RegionEnd != MBB->end()) dbgs() << *RegionEnd;
283             else dbgs() << "End";
284             dbgs() << " Remaining: " << RemainingInstrs << "\n");
285 
286       // Schedule a region: possibly reorder instructions.
287       // This invalidates 'RegionEnd' and 'I'.
288       Scheduler->schedule();
289 
290       // Close the current region.
291       Scheduler->exitRegion();
292 
293       // Scheduling has invalidated the current iterator 'I'. Ask the
294       // scheduler for the top of it's scheduled region.
295       RegionEnd = Scheduler->begin();
296     }
297     assert(RemainingInstrs == 0 && "Instruction count mismatch!");
298     Scheduler->finishBlock();
299   }
300   Scheduler->finalizeSchedule();
301   DEBUG(LIS->print(dbgs()));
302   if (VerifyScheduling)
303     MF->verify(this, "After machine scheduling.");
304   return true;
305 }
306 
307 void MachineScheduler::print(raw_ostream &O, const Module* m) const {
308   // unimplemented
309 }
310 
311 #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
312 void ReadyQueue::dump() {
313   dbgs() << "  " << Name << ": ";
314   for (unsigned i = 0, e = Queue.size(); i < e; ++i)
315     dbgs() << Queue[i]->NodeNum << " ";
316   dbgs() << "\n";
317 }
318 #endif
319 
320 //===----------------------------------------------------------------------===//
321 // ScheduleDAGMI - Base class for MachineInstr scheduling with LiveIntervals
322 // preservation.
323 //===----------------------------------------------------------------------===//
324 
325 ScheduleDAGMI::~ScheduleDAGMI() {
326   delete DFSResult;
327   DeleteContainerPointers(Mutations);
328   delete SchedImpl;
329 }
330 
331 bool ScheduleDAGMI::canAddEdge(SUnit *SuccSU, SUnit *PredSU) {
332   return SuccSU == &ExitSU || !Topo.IsReachable(PredSU, SuccSU);
333 }
334 
335 bool ScheduleDAGMI::addEdge(SUnit *SuccSU, const SDep &PredDep) {
336   if (SuccSU != &ExitSU) {
337     // Do not use WillCreateCycle, it assumes SD scheduling.
338     // If Pred is reachable from Succ, then the edge creates a cycle.
339     if (Topo.IsReachable(PredDep.getSUnit(), SuccSU))
340       return false;
341     Topo.AddPred(SuccSU, PredDep.getSUnit());
342   }
343   SuccSU->addPred(PredDep, /*Required=*/!PredDep.isArtificial());
344   // Return true regardless of whether a new edge needed to be inserted.
345   return true;
346 }
347 
348 /// ReleaseSucc - Decrement the NumPredsLeft count of a successor. When
349 /// NumPredsLeft reaches zero, release the successor node.
350 ///
351 /// FIXME: Adjust SuccSU height based on MinLatency.
352 void ScheduleDAGMI::releaseSucc(SUnit *SU, SDep *SuccEdge) {
353   SUnit *SuccSU = SuccEdge->getSUnit();
354 
355   if (SuccEdge->isWeak()) {
356     --SuccSU->WeakPredsLeft;
357     if (SuccEdge->isCluster())
358       NextClusterSucc = SuccSU;
359     return;
360   }
361 #ifndef NDEBUG
362   if (SuccSU->NumPredsLeft == 0) {
363     dbgs() << "*** Scheduling failed! ***\n";
364     SuccSU->dump(this);
365     dbgs() << " has been released too many times!\n";
366     llvm_unreachable(0);
367   }
368 #endif
369   --SuccSU->NumPredsLeft;
370   if (SuccSU->NumPredsLeft == 0 && SuccSU != &ExitSU)
371     SchedImpl->releaseTopNode(SuccSU);
372 }
373 
374 /// releaseSuccessors - Call releaseSucc on each of SU's successors.
375 void ScheduleDAGMI::releaseSuccessors(SUnit *SU) {
376   for (SUnit::succ_iterator I = SU->Succs.begin(), E = SU->Succs.end();
377        I != E; ++I) {
378     releaseSucc(SU, &*I);
379   }
380 }
381 
382 /// ReleasePred - Decrement the NumSuccsLeft count of a predecessor. When
383 /// NumSuccsLeft reaches zero, release the predecessor node.
384 ///
385 /// FIXME: Adjust PredSU height based on MinLatency.
386 void ScheduleDAGMI::releasePred(SUnit *SU, SDep *PredEdge) {
387   SUnit *PredSU = PredEdge->getSUnit();
388 
389   if (PredEdge->isWeak()) {
390     --PredSU->WeakSuccsLeft;
391     if (PredEdge->isCluster())
392       NextClusterPred = PredSU;
393     return;
394   }
395 #ifndef NDEBUG
396   if (PredSU->NumSuccsLeft == 0) {
397     dbgs() << "*** Scheduling failed! ***\n";
398     PredSU->dump(this);
399     dbgs() << " has been released too many times!\n";
400     llvm_unreachable(0);
401   }
402 #endif
403   --PredSU->NumSuccsLeft;
404   if (PredSU->NumSuccsLeft == 0 && PredSU != &EntrySU)
405     SchedImpl->releaseBottomNode(PredSU);
406 }
407 
408 /// releasePredecessors - Call releasePred on each of SU's predecessors.
409 void ScheduleDAGMI::releasePredecessors(SUnit *SU) {
410   for (SUnit::pred_iterator I = SU->Preds.begin(), E = SU->Preds.end();
411        I != E; ++I) {
412     releasePred(SU, &*I);
413   }
414 }
415 
416 /// This is normally called from the main scheduler loop but may also be invoked
417 /// by the scheduling strategy to perform additional code motion.
418 void ScheduleDAGMI::moveInstruction(MachineInstr *MI,
419                                     MachineBasicBlock::iterator InsertPos) {
420   // Advance RegionBegin if the first instruction moves down.
421   if (&*RegionBegin == MI)
422     ++RegionBegin;
423 
424   // Update the instruction stream.
425   BB->splice(InsertPos, BB, MI);
426 
427   // Update LiveIntervals
428   LIS->handleMove(MI, /*UpdateFlags=*/true);
429 
430   // Recede RegionBegin if an instruction moves above the first.
431   if (RegionBegin == InsertPos)
432     RegionBegin = MI;
433 }
434 
435 bool ScheduleDAGMI::checkSchedLimit() {
436 #ifndef NDEBUG
437   if (NumInstrsScheduled == MISchedCutoff && MISchedCutoff != ~0U) {
438     CurrentTop = CurrentBottom;
439     return false;
440   }
441   ++NumInstrsScheduled;
442 #endif
443   return true;
444 }
445 
446 /// enterRegion - Called back from MachineScheduler::runOnMachineFunction after
447 /// crossing a scheduling boundary. [begin, end) includes all instructions in
448 /// the region, including the boundary itself and single-instruction regions
449 /// that don't get scheduled.
450 void ScheduleDAGMI::enterRegion(MachineBasicBlock *bb,
451                                 MachineBasicBlock::iterator begin,
452                                 MachineBasicBlock::iterator end,
453                                 unsigned endcount)
454 {
455   ScheduleDAGInstrs::enterRegion(bb, begin, end, endcount);
456 
457   // For convenience remember the end of the liveness region.
458   LiveRegionEnd =
459     (RegionEnd == bb->end()) ? RegionEnd : llvm::next(RegionEnd);
460 }
461 
462 // Setup the register pressure trackers for the top scheduled top and bottom
463 // scheduled regions.
464 void ScheduleDAGMI::initRegPressure() {
465   TopRPTracker.init(&MF, RegClassInfo, LIS, BB, RegionBegin);
466   BotRPTracker.init(&MF, RegClassInfo, LIS, BB, LiveRegionEnd);
467 
468   // Close the RPTracker to finalize live ins.
469   RPTracker.closeRegion();
470 
471   DEBUG(RPTracker.getPressure().dump(TRI));
472 
473   // Initialize the live ins and live outs.
474   TopRPTracker.addLiveRegs(RPTracker.getPressure().LiveInRegs);
475   BotRPTracker.addLiveRegs(RPTracker.getPressure().LiveOutRegs);
476 
477   // Close one end of the tracker so we can call
478   // getMaxUpward/DownwardPressureDelta before advancing across any
479   // instructions. This converts currently live regs into live ins/outs.
480   TopRPTracker.closeTop();
481   BotRPTracker.closeBottom();
482 
483   // Account for liveness generated by the region boundary.
484   if (LiveRegionEnd != RegionEnd)
485     BotRPTracker.recede();
486 
487   assert(BotRPTracker.getPos() == RegionEnd && "Can't find the region bottom");
488 
489   // Cache the list of excess pressure sets in this region. This will also track
490   // the max pressure in the scheduled code for these sets.
491   RegionCriticalPSets.clear();
492   const std::vector<unsigned> &RegionPressure =
493     RPTracker.getPressure().MaxSetPressure;
494   for (unsigned i = 0, e = RegionPressure.size(); i < e; ++i) {
495     unsigned Limit = TRI->getRegPressureSetLimit(i);
496     DEBUG(dbgs() << TRI->getRegPressureSetName(i)
497           << "Limit " << Limit
498           << " Actual " << RegionPressure[i] << "\n");
499     if (RegionPressure[i] > Limit)
500       RegionCriticalPSets.push_back(PressureElement(i, 0));
501   }
502   DEBUG(dbgs() << "Excess PSets: ";
503         for (unsigned i = 0, e = RegionCriticalPSets.size(); i != e; ++i)
504           dbgs() << TRI->getRegPressureSetName(
505             RegionCriticalPSets[i].PSetID) << " ";
506         dbgs() << "\n");
507 }
508 
509 // FIXME: When the pressure tracker deals in pressure differences then we won't
510 // iterate over all RegionCriticalPSets[i].
511 void ScheduleDAGMI::
512 updateScheduledPressure(const std::vector<unsigned> &NewMaxPressure) {
513   for (unsigned i = 0, e = RegionCriticalPSets.size(); i < e; ++i) {
514     unsigned ID = RegionCriticalPSets[i].PSetID;
515     int &MaxUnits = RegionCriticalPSets[i].UnitIncrease;
516     if ((int)NewMaxPressure[ID] > MaxUnits)
517       MaxUnits = NewMaxPressure[ID];
518   }
519   DEBUG(
520     for (unsigned i = 0, e = NewMaxPressure.size(); i < e; ++i) {
521       unsigned Limit = TRI->getRegPressureSetLimit(i);
522       if (NewMaxPressure[i] > Limit ) {
523         dbgs() << "  " << TRI->getRegPressureSetName(i) << ": "
524                << NewMaxPressure[i] << " > " << Limit << "\n";
525       }
526     });
527 }
528 
529 /// schedule - Called back from MachineScheduler::runOnMachineFunction
530 /// after setting up the current scheduling region. [RegionBegin, RegionEnd)
531 /// only includes instructions that have DAG nodes, not scheduling boundaries.
532 ///
533 /// This is a skeletal driver, with all the functionality pushed into helpers,
534 /// so that it can be easilly extended by experimental schedulers. Generally,
535 /// implementing MachineSchedStrategy should be sufficient to implement a new
536 /// scheduling algorithm. However, if a scheduler further subclasses
537 /// ScheduleDAGMI then it will want to override this virtual method in order to
538 /// update any specialized state.
539 void ScheduleDAGMI::schedule() {
540   buildDAGWithRegPressure();
541 
542   Topo.InitDAGTopologicalSorting();
543 
544   postprocessDAG();
545 
546   SmallVector<SUnit*, 8> TopRoots, BotRoots;
547   findRootsAndBiasEdges(TopRoots, BotRoots);
548 
549   // Initialize the strategy before modifying the DAG.
550   // This may initialize a DFSResult to be used for queue priority.
551   SchedImpl->initialize(this);
552 
553   DEBUG(for (unsigned su = 0, e = SUnits.size(); su != e; ++su)
554           SUnits[su].dumpAll(this));
555   if (ViewMISchedDAGs) viewGraph();
556 
557   // Initialize ready queues now that the DAG and priority data are finalized.
558   initQueues(TopRoots, BotRoots);
559 
560   bool IsTopNode = false;
561   while (SUnit *SU = SchedImpl->pickNode(IsTopNode)) {
562     assert(!SU->isScheduled && "Node already scheduled");
563     if (!checkSchedLimit())
564       break;
565 
566     scheduleMI(SU, IsTopNode);
567 
568     updateQueues(SU, IsTopNode);
569   }
570   assert(CurrentTop == CurrentBottom && "Nonempty unscheduled zone.");
571 
572   placeDebugValues();
573 
574   DEBUG({
575       unsigned BBNum = begin()->getParent()->getNumber();
576       dbgs() << "*** Final schedule for BB#" << BBNum << " ***\n";
577       dumpSchedule();
578       dbgs() << '\n';
579     });
580 }
581 
582 /// Build the DAG and setup three register pressure trackers.
583 void ScheduleDAGMI::buildDAGWithRegPressure() {
584   // Initialize the register pressure tracker used by buildSchedGraph.
585   RPTracker.init(&MF, RegClassInfo, LIS, BB, LiveRegionEnd);
586 
587   // Account for liveness generate by the region boundary.
588   if (LiveRegionEnd != RegionEnd)
589     RPTracker.recede();
590 
591   // Build the DAG, and compute current register pressure.
592   buildSchedGraph(AA, &RPTracker);
593 
594   // Initialize top/bottom trackers after computing region pressure.
595   initRegPressure();
596 }
597 
598 /// Apply each ScheduleDAGMutation step in order.
599 void ScheduleDAGMI::postprocessDAG() {
600   for (unsigned i = 0, e = Mutations.size(); i < e; ++i) {
601     Mutations[i]->apply(this);
602   }
603 }
604 
605 void ScheduleDAGMI::computeDFSResult() {
606   if (!DFSResult)
607     DFSResult = new SchedDFSResult(/*BottomU*/true, MinSubtreeSize);
608   DFSResult->clear();
609   ScheduledTrees.clear();
610   DFSResult->resize(SUnits.size());
611   DFSResult->compute(SUnits);
612   ScheduledTrees.resize(DFSResult->getNumSubtrees());
613 }
614 
615 void ScheduleDAGMI::findRootsAndBiasEdges(SmallVectorImpl<SUnit*> &TopRoots,
616                                           SmallVectorImpl<SUnit*> &BotRoots) {
617   for (std::vector<SUnit>::iterator
618          I = SUnits.begin(), E = SUnits.end(); I != E; ++I) {
619     SUnit *SU = &(*I);
620     assert(!SU->isBoundaryNode() && "Boundary node should not be in SUnits");
621 
622     // Order predecessors so DFSResult follows the critical path.
623     SU->biasCriticalPath();
624 
625     // A SUnit is ready to top schedule if it has no predecessors.
626     if (!I->NumPredsLeft)
627       TopRoots.push_back(SU);
628     // A SUnit is ready to bottom schedule if it has no successors.
629     if (!I->NumSuccsLeft)
630       BotRoots.push_back(SU);
631   }
632   ExitSU.biasCriticalPath();
633 }
634 
635 /// Identify DAG roots and setup scheduler queues.
636 void ScheduleDAGMI::initQueues(ArrayRef<SUnit*> TopRoots,
637                                ArrayRef<SUnit*> BotRoots) {
638   NextClusterSucc = NULL;
639   NextClusterPred = NULL;
640 
641   // Release all DAG roots for scheduling, not including EntrySU/ExitSU.
642   //
643   // Nodes with unreleased weak edges can still be roots.
644   // Release top roots in forward order.
645   for (SmallVectorImpl<SUnit*>::const_iterator
646          I = TopRoots.begin(), E = TopRoots.end(); I != E; ++I) {
647     SchedImpl->releaseTopNode(*I);
648   }
649   // Release bottom roots in reverse order so the higher priority nodes appear
650   // first. This is more natural and slightly more efficient.
651   for (SmallVectorImpl<SUnit*>::const_reverse_iterator
652          I = BotRoots.rbegin(), E = BotRoots.rend(); I != E; ++I) {
653     SchedImpl->releaseBottomNode(*I);
654   }
655 
656   releaseSuccessors(&EntrySU);
657   releasePredecessors(&ExitSU);
658 
659   SchedImpl->registerRoots();
660 
661   // Advance past initial DebugValues.
662   assert(TopRPTracker.getPos() == RegionBegin && "bad initial Top tracker");
663   CurrentTop = nextIfDebug(RegionBegin, RegionEnd);
664   TopRPTracker.setPos(CurrentTop);
665 
666   CurrentBottom = RegionEnd;
667 }
668 
669 /// Move an instruction and update register pressure.
670 void ScheduleDAGMI::scheduleMI(SUnit *SU, bool IsTopNode) {
671   // Move the instruction to its new location in the instruction stream.
672   MachineInstr *MI = SU->getInstr();
673 
674   if (IsTopNode) {
675     assert(SU->isTopReady() && "node still has unscheduled dependencies");
676     if (&*CurrentTop == MI)
677       CurrentTop = nextIfDebug(++CurrentTop, CurrentBottom);
678     else {
679       moveInstruction(MI, CurrentTop);
680       TopRPTracker.setPos(MI);
681     }
682 
683     // Update top scheduled pressure.
684     TopRPTracker.advance();
685     assert(TopRPTracker.getPos() == CurrentTop && "out of sync");
686     updateScheduledPressure(TopRPTracker.getPressure().MaxSetPressure);
687   }
688   else {
689     assert(SU->isBottomReady() && "node still has unscheduled dependencies");
690     MachineBasicBlock::iterator priorII =
691       priorNonDebug(CurrentBottom, CurrentTop);
692     if (&*priorII == MI)
693       CurrentBottom = priorII;
694     else {
695       if (&*CurrentTop == MI) {
696         CurrentTop = nextIfDebug(++CurrentTop, priorII);
697         TopRPTracker.setPos(CurrentTop);
698       }
699       moveInstruction(MI, CurrentBottom);
700       CurrentBottom = MI;
701     }
702     // Update bottom scheduled pressure.
703     BotRPTracker.recede();
704     assert(BotRPTracker.getPos() == CurrentBottom && "out of sync");
705     updateScheduledPressure(BotRPTracker.getPressure().MaxSetPressure);
706   }
707 }
708 
709 /// Update scheduler queues after scheduling an instruction.
710 void ScheduleDAGMI::updateQueues(SUnit *SU, bool IsTopNode) {
711   // Release dependent instructions for scheduling.
712   if (IsTopNode)
713     releaseSuccessors(SU);
714   else
715     releasePredecessors(SU);
716 
717   SU->isScheduled = true;
718 
719   if (DFSResult) {
720     unsigned SubtreeID = DFSResult->getSubtreeID(SU);
721     if (!ScheduledTrees.test(SubtreeID)) {
722       ScheduledTrees.set(SubtreeID);
723       DFSResult->scheduleTree(SubtreeID);
724       SchedImpl->scheduleTree(SubtreeID);
725     }
726   }
727 
728   // Notify the scheduling strategy after updating the DAG.
729   SchedImpl->schedNode(SU, IsTopNode);
730 }
731 
732 /// Reinsert any remaining debug_values, just like the PostRA scheduler.
733 void ScheduleDAGMI::placeDebugValues() {
734   // If first instruction was a DBG_VALUE then put it back.
735   if (FirstDbgValue) {
736     BB->splice(RegionBegin, BB, FirstDbgValue);
737     RegionBegin = FirstDbgValue;
738   }
739 
740   for (std::vector<std::pair<MachineInstr *, MachineInstr *> >::iterator
741          DI = DbgValues.end(), DE = DbgValues.begin(); DI != DE; --DI) {
742     std::pair<MachineInstr *, MachineInstr *> P = *prior(DI);
743     MachineInstr *DbgValue = P.first;
744     MachineBasicBlock::iterator OrigPrevMI = P.second;
745     if (&*RegionBegin == DbgValue)
746       ++RegionBegin;
747     BB->splice(++OrigPrevMI, BB, DbgValue);
748     if (OrigPrevMI == llvm::prior(RegionEnd))
749       RegionEnd = DbgValue;
750   }
751   DbgValues.clear();
752   FirstDbgValue = NULL;
753 }
754 
755 #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
756 void ScheduleDAGMI::dumpSchedule() const {
757   for (MachineBasicBlock::iterator MI = begin(), ME = end(); MI != ME; ++MI) {
758     if (SUnit *SU = getSUnit(&(*MI)))
759       SU->dump(this);
760     else
761       dbgs() << "Missing SUnit\n";
762   }
763 }
764 #endif
765 
766 //===----------------------------------------------------------------------===//
767 // LoadClusterMutation - DAG post-processing to cluster loads.
768 //===----------------------------------------------------------------------===//
769 
770 namespace {
771 /// \brief Post-process the DAG to create cluster edges between neighboring
772 /// loads.
773 class LoadClusterMutation : public ScheduleDAGMutation {
774   struct LoadInfo {
775     SUnit *SU;
776     unsigned BaseReg;
777     unsigned Offset;
778     LoadInfo(SUnit *su, unsigned reg, unsigned ofs)
779       : SU(su), BaseReg(reg), Offset(ofs) {}
780   };
781   static bool LoadInfoLess(const LoadClusterMutation::LoadInfo &LHS,
782                            const LoadClusterMutation::LoadInfo &RHS);
783 
784   const TargetInstrInfo *TII;
785   const TargetRegisterInfo *TRI;
786 public:
787   LoadClusterMutation(const TargetInstrInfo *tii,
788                       const TargetRegisterInfo *tri)
789     : TII(tii), TRI(tri) {}
790 
791   virtual void apply(ScheduleDAGMI *DAG);
792 protected:
793   void clusterNeighboringLoads(ArrayRef<SUnit*> Loads, ScheduleDAGMI *DAG);
794 };
795 } // anonymous
796 
797 bool LoadClusterMutation::LoadInfoLess(
798   const LoadClusterMutation::LoadInfo &LHS,
799   const LoadClusterMutation::LoadInfo &RHS) {
800   if (LHS.BaseReg != RHS.BaseReg)
801     return LHS.BaseReg < RHS.BaseReg;
802   return LHS.Offset < RHS.Offset;
803 }
804 
805 void LoadClusterMutation::clusterNeighboringLoads(ArrayRef<SUnit*> Loads,
806                                                   ScheduleDAGMI *DAG) {
807   SmallVector<LoadClusterMutation::LoadInfo,32> LoadRecords;
808   for (unsigned Idx = 0, End = Loads.size(); Idx != End; ++Idx) {
809     SUnit *SU = Loads[Idx];
810     unsigned BaseReg;
811     unsigned Offset;
812     if (TII->getLdStBaseRegImmOfs(SU->getInstr(), BaseReg, Offset, TRI))
813       LoadRecords.push_back(LoadInfo(SU, BaseReg, Offset));
814   }
815   if (LoadRecords.size() < 2)
816     return;
817   std::sort(LoadRecords.begin(), LoadRecords.end(), LoadInfoLess);
818   unsigned ClusterLength = 1;
819   for (unsigned Idx = 0, End = LoadRecords.size(); Idx < (End - 1); ++Idx) {
820     if (LoadRecords[Idx].BaseReg != LoadRecords[Idx+1].BaseReg) {
821       ClusterLength = 1;
822       continue;
823     }
824 
825     SUnit *SUa = LoadRecords[Idx].SU;
826     SUnit *SUb = LoadRecords[Idx+1].SU;
827     if (TII->shouldClusterLoads(SUa->getInstr(), SUb->getInstr(), ClusterLength)
828         && DAG->addEdge(SUb, SDep(SUa, SDep::Cluster))) {
829 
830       DEBUG(dbgs() << "Cluster loads SU(" << SUa->NodeNum << ") - SU("
831             << SUb->NodeNum << ")\n");
832       // Copy successor edges from SUa to SUb. Interleaving computation
833       // dependent on SUa can prevent load combining due to register reuse.
834       // Predecessor edges do not need to be copied from SUb to SUa since nearby
835       // loads should have effectively the same inputs.
836       for (SUnit::const_succ_iterator
837              SI = SUa->Succs.begin(), SE = SUa->Succs.end(); SI != SE; ++SI) {
838         if (SI->getSUnit() == SUb)
839           continue;
840         DEBUG(dbgs() << "  Copy Succ SU(" << SI->getSUnit()->NodeNum << ")\n");
841         DAG->addEdge(SI->getSUnit(), SDep(SUb, SDep::Artificial));
842       }
843       ++ClusterLength;
844     }
845     else
846       ClusterLength = 1;
847   }
848 }
849 
850 /// \brief Callback from DAG postProcessing to create cluster edges for loads.
851 void LoadClusterMutation::apply(ScheduleDAGMI *DAG) {
852   // Map DAG NodeNum to store chain ID.
853   DenseMap<unsigned, unsigned> StoreChainIDs;
854   // Map each store chain to a set of dependent loads.
855   SmallVector<SmallVector<SUnit*,4>, 32> StoreChainDependents;
856   for (unsigned Idx = 0, End = DAG->SUnits.size(); Idx != End; ++Idx) {
857     SUnit *SU = &DAG->SUnits[Idx];
858     if (!SU->getInstr()->mayLoad())
859       continue;
860     unsigned ChainPredID = DAG->SUnits.size();
861     for (SUnit::const_pred_iterator
862            PI = SU->Preds.begin(), PE = SU->Preds.end(); PI != PE; ++PI) {
863       if (PI->isCtrl()) {
864         ChainPredID = PI->getSUnit()->NodeNum;
865         break;
866       }
867     }
868     // Check if this chain-like pred has been seen
869     // before. ChainPredID==MaxNodeID for loads at the top of the schedule.
870     unsigned NumChains = StoreChainDependents.size();
871     std::pair<DenseMap<unsigned, unsigned>::iterator, bool> Result =
872       StoreChainIDs.insert(std::make_pair(ChainPredID, NumChains));
873     if (Result.second)
874       StoreChainDependents.resize(NumChains + 1);
875     StoreChainDependents[Result.first->second].push_back(SU);
876   }
877   // Iterate over the store chains.
878   for (unsigned Idx = 0, End = StoreChainDependents.size(); Idx != End; ++Idx)
879     clusterNeighboringLoads(StoreChainDependents[Idx], DAG);
880 }
881 
882 //===----------------------------------------------------------------------===//
883 // MacroFusion - DAG post-processing to encourage fusion of macro ops.
884 //===----------------------------------------------------------------------===//
885 
886 namespace {
887 /// \brief Post-process the DAG to create cluster edges between instructions
888 /// that may be fused by the processor into a single operation.
889 class MacroFusion : public ScheduleDAGMutation {
890   const TargetInstrInfo *TII;
891 public:
892   MacroFusion(const TargetInstrInfo *tii): TII(tii) {}
893 
894   virtual void apply(ScheduleDAGMI *DAG);
895 };
896 } // anonymous
897 
898 /// \brief Callback from DAG postProcessing to create cluster edges to encourage
899 /// fused operations.
900 void MacroFusion::apply(ScheduleDAGMI *DAG) {
901   // For now, assume targets can only fuse with the branch.
902   MachineInstr *Branch = DAG->ExitSU.getInstr();
903   if (!Branch)
904     return;
905 
906   for (unsigned Idx = DAG->SUnits.size(); Idx > 0;) {
907     SUnit *SU = &DAG->SUnits[--Idx];
908     if (!TII->shouldScheduleAdjacent(SU->getInstr(), Branch))
909       continue;
910 
911     // Create a single weak edge from SU to ExitSU. The only effect is to cause
912     // bottom-up scheduling to heavily prioritize the clustered SU.  There is no
913     // need to copy predecessor edges from ExitSU to SU, since top-down
914     // scheduling cannot prioritize ExitSU anyway. To defer top-down scheduling
915     // of SU, we could create an artificial edge from the deepest root, but it
916     // hasn't been needed yet.
917     bool Success = DAG->addEdge(&DAG->ExitSU, SDep(SU, SDep::Cluster));
918     (void)Success;
919     assert(Success && "No DAG nodes should be reachable from ExitSU");
920 
921     DEBUG(dbgs() << "Macro Fuse SU(" << SU->NodeNum << ")\n");
922     break;
923   }
924 }
925 
926 //===----------------------------------------------------------------------===//
927 // CopyConstrain - DAG post-processing to encourage copy elimination.
928 //===----------------------------------------------------------------------===//
929 
930 namespace {
931 /// \brief Post-process the DAG to create weak edges from all uses of a copy to
932 /// the one use that defines the copy's source vreg, most likely an induction
933 /// variable increment.
934 class CopyConstrain : public ScheduleDAGMutation {
935   // Transient state.
936   SlotIndex RegionBeginIdx;
937   // RegionEndIdx is the slot index of the last non-debug instruction in the
938   // scheduling region. So we may have RegionBeginIdx == RegionEndIdx.
939   SlotIndex RegionEndIdx;
940 public:
941   CopyConstrain(const TargetInstrInfo *, const TargetRegisterInfo *) {}
942 
943   virtual void apply(ScheduleDAGMI *DAG);
944 
945 protected:
946   void constrainLocalCopy(SUnit *CopySU, ScheduleDAGMI *DAG);
947 };
948 } // anonymous
949 
950 /// constrainLocalCopy handles two possibilities:
951 /// 1) Local src:
952 /// I0:     = dst
953 /// I1: src = ...
954 /// I2:     = dst
955 /// I3: dst = src (copy)
956 /// (create pred->succ edges I0->I1, I2->I1)
957 ///
958 /// 2) Local copy:
959 /// I0: dst = src (copy)
960 /// I1:     = dst
961 /// I2: src = ...
962 /// I3:     = dst
963 /// (create pred->succ edges I1->I2, I3->I2)
964 ///
965 /// Although the MachineScheduler is currently constrained to single blocks,
966 /// this algorithm should handle extended blocks. An EBB is a set of
967 /// contiguously numbered blocks such that the previous block in the EBB is
968 /// always the single predecessor.
969 void CopyConstrain::constrainLocalCopy(SUnit *CopySU, ScheduleDAGMI *DAG) {
970   LiveIntervals *LIS = DAG->getLIS();
971   MachineInstr *Copy = CopySU->getInstr();
972 
973   // Check for pure vreg copies.
974   unsigned SrcReg = Copy->getOperand(1).getReg();
975   if (!TargetRegisterInfo::isVirtualRegister(SrcReg))
976     return;
977 
978   unsigned DstReg = Copy->getOperand(0).getReg();
979   if (!TargetRegisterInfo::isVirtualRegister(DstReg))
980     return;
981 
982   // Check if either the dest or source is local. If it's live across a back
983   // edge, it's not local. Note that if both vregs are live across the back
984   // edge, we cannot successfully contrain the copy without cyclic scheduling.
985   unsigned LocalReg = DstReg;
986   unsigned GlobalReg = SrcReg;
987   LiveInterval *LocalLI = &LIS->getInterval(LocalReg);
988   if (!LocalLI->isLocal(RegionBeginIdx, RegionEndIdx)) {
989     LocalReg = SrcReg;
990     GlobalReg = DstReg;
991     LocalLI = &LIS->getInterval(LocalReg);
992     if (!LocalLI->isLocal(RegionBeginIdx, RegionEndIdx))
993       return;
994   }
995   LiveInterval *GlobalLI = &LIS->getInterval(GlobalReg);
996 
997   // Find the global segment after the start of the local LI.
998   LiveInterval::iterator GlobalSegment = GlobalLI->find(LocalLI->beginIndex());
999   // If GlobalLI does not overlap LocalLI->start, then a copy directly feeds a
1000   // local live range. We could create edges from other global uses to the local
1001   // start, but the coalescer should have already eliminated these cases, so
1002   // don't bother dealing with it.
1003   if (GlobalSegment == GlobalLI->end())
1004     return;
1005 
1006   // If GlobalSegment is killed at the LocalLI->start, the call to find()
1007   // returned the next global segment. But if GlobalSegment overlaps with
1008   // LocalLI->start, then advance to the next segement. If a hole in GlobalLI
1009   // exists in LocalLI's vicinity, GlobalSegment will be the end of the hole.
1010   if (GlobalSegment->contains(LocalLI->beginIndex()))
1011     ++GlobalSegment;
1012 
1013   if (GlobalSegment == GlobalLI->end())
1014     return;
1015 
1016   // Check if GlobalLI contains a hole in the vicinity of LocalLI.
1017   if (GlobalSegment != GlobalLI->begin()) {
1018     // Two address defs have no hole.
1019     if (SlotIndex::isSameInstr(llvm::prior(GlobalSegment)->end,
1020                                GlobalSegment->start)) {
1021       return;
1022     }
1023     // If GlobalLI has a prior segment, it must be live into the EBB. Otherwise
1024     // it would be a disconnected component in the live range.
1025     assert(llvm::prior(GlobalSegment)->start < LocalLI->beginIndex() &&
1026            "Disconnected LRG within the scheduling region.");
1027   }
1028   MachineInstr *GlobalDef = LIS->getInstructionFromIndex(GlobalSegment->start);
1029   if (!GlobalDef)
1030     return;
1031 
1032   SUnit *GlobalSU = DAG->getSUnit(GlobalDef);
1033   if (!GlobalSU)
1034     return;
1035 
1036   // GlobalDef is the bottom of the GlobalLI hole. Open the hole by
1037   // constraining the uses of the last local def to precede GlobalDef.
1038   SmallVector<SUnit*,8> LocalUses;
1039   const VNInfo *LastLocalVN = LocalLI->getVNInfoBefore(LocalLI->endIndex());
1040   MachineInstr *LastLocalDef = LIS->getInstructionFromIndex(LastLocalVN->def);
1041   SUnit *LastLocalSU = DAG->getSUnit(LastLocalDef);
1042   for (SUnit::const_succ_iterator
1043          I = LastLocalSU->Succs.begin(), E = LastLocalSU->Succs.end();
1044        I != E; ++I) {
1045     if (I->getKind() != SDep::Data || I->getReg() != LocalReg)
1046       continue;
1047     if (I->getSUnit() == GlobalSU)
1048       continue;
1049     if (!DAG->canAddEdge(GlobalSU, I->getSUnit()))
1050       return;
1051     LocalUses.push_back(I->getSUnit());
1052   }
1053   // Open the top of the GlobalLI hole by constraining any earlier global uses
1054   // to precede the start of LocalLI.
1055   SmallVector<SUnit*,8> GlobalUses;
1056   MachineInstr *FirstLocalDef =
1057     LIS->getInstructionFromIndex(LocalLI->beginIndex());
1058   SUnit *FirstLocalSU = DAG->getSUnit(FirstLocalDef);
1059   for (SUnit::const_pred_iterator
1060          I = GlobalSU->Preds.begin(), E = GlobalSU->Preds.end(); I != E; ++I) {
1061     if (I->getKind() != SDep::Anti || I->getReg() != GlobalReg)
1062       continue;
1063     if (I->getSUnit() == FirstLocalSU)
1064       continue;
1065     if (!DAG->canAddEdge(FirstLocalSU, I->getSUnit()))
1066       return;
1067     GlobalUses.push_back(I->getSUnit());
1068   }
1069   DEBUG(dbgs() << "Constraining copy SU(" << CopySU->NodeNum << ")\n");
1070   // Add the weak edges.
1071   for (SmallVectorImpl<SUnit*>::const_iterator
1072          I = LocalUses.begin(), E = LocalUses.end(); I != E; ++I) {
1073     DEBUG(dbgs() << "  Local use SU(" << (*I)->NodeNum << ") -> SU("
1074           << GlobalSU->NodeNum << ")\n");
1075     DAG->addEdge(GlobalSU, SDep(*I, SDep::Weak));
1076   }
1077   for (SmallVectorImpl<SUnit*>::const_iterator
1078          I = GlobalUses.begin(), E = GlobalUses.end(); I != E; ++I) {
1079     DEBUG(dbgs() << "  Global use SU(" << (*I)->NodeNum << ") -> SU("
1080           << FirstLocalSU->NodeNum << ")\n");
1081     DAG->addEdge(FirstLocalSU, SDep(*I, SDep::Weak));
1082   }
1083 }
1084 
1085 /// \brief Callback from DAG postProcessing to create weak edges to encourage
1086 /// copy elimination.
1087 void CopyConstrain::apply(ScheduleDAGMI *DAG) {
1088   MachineBasicBlock::iterator FirstPos = nextIfDebug(DAG->begin(), DAG->end());
1089   if (FirstPos == DAG->end())
1090     return;
1091   RegionBeginIdx = DAG->getLIS()->getInstructionIndex(&*FirstPos);
1092   RegionEndIdx = DAG->getLIS()->getInstructionIndex(
1093     &*priorNonDebug(DAG->end(), DAG->begin()));
1094 
1095   for (unsigned Idx = 0, End = DAG->SUnits.size(); Idx != End; ++Idx) {
1096     SUnit *SU = &DAG->SUnits[Idx];
1097     if (!SU->getInstr()->isCopy())
1098       continue;
1099 
1100     constrainLocalCopy(SU, DAG);
1101   }
1102 }
1103 
1104 //===----------------------------------------------------------------------===//
1105 // ConvergingScheduler - Implementation of the standard MachineSchedStrategy.
1106 //===----------------------------------------------------------------------===//
1107 
1108 namespace {
1109 /// ConvergingScheduler shrinks the unscheduled zone using heuristics to balance
1110 /// the schedule.
1111 class ConvergingScheduler : public MachineSchedStrategy {
1112 public:
1113   /// Represent the type of SchedCandidate found within a single queue.
1114   /// pickNodeBidirectional depends on these listed by decreasing priority.
1115   enum CandReason {
1116     NoCand, PhysRegCopy, SingleExcess, SingleCritical, Cluster, Weak,
1117     ResourceReduce, ResourceDemand, BotHeightReduce, BotPathReduce,
1118     TopDepthReduce, TopPathReduce, SingleMax, MultiPressure, NextDefUse,
1119     NodeOrder};
1120 
1121 #ifndef NDEBUG
1122   static const char *getReasonStr(ConvergingScheduler::CandReason Reason);
1123 #endif
1124 
1125   /// Policy for scheduling the next instruction in the candidate's zone.
1126   struct CandPolicy {
1127     bool ReduceLatency;
1128     unsigned ReduceResIdx;
1129     unsigned DemandResIdx;
1130 
1131     CandPolicy(): ReduceLatency(false), ReduceResIdx(0), DemandResIdx(0) {}
1132   };
1133 
1134   /// Status of an instruction's critical resource consumption.
1135   struct SchedResourceDelta {
1136     // Count critical resources in the scheduled region required by SU.
1137     unsigned CritResources;
1138 
1139     // Count critical resources from another region consumed by SU.
1140     unsigned DemandedResources;
1141 
1142     SchedResourceDelta(): CritResources(0), DemandedResources(0) {}
1143 
1144     bool operator==(const SchedResourceDelta &RHS) const {
1145       return CritResources == RHS.CritResources
1146         && DemandedResources == RHS.DemandedResources;
1147     }
1148     bool operator!=(const SchedResourceDelta &RHS) const {
1149       return !operator==(RHS);
1150     }
1151   };
1152 
1153   /// Store the state used by ConvergingScheduler heuristics, required for the
1154   /// lifetime of one invocation of pickNode().
1155   struct SchedCandidate {
1156     CandPolicy Policy;
1157 
1158     // The best SUnit candidate.
1159     SUnit *SU;
1160 
1161     // The reason for this candidate.
1162     CandReason Reason;
1163 
1164     // Register pressure values for the best candidate.
1165     RegPressureDelta RPDelta;
1166 
1167     // Critical resource consumption of the best candidate.
1168     SchedResourceDelta ResDelta;
1169 
1170     SchedCandidate(const CandPolicy &policy)
1171     : Policy(policy), SU(NULL), Reason(NoCand) {}
1172 
1173     bool isValid() const { return SU; }
1174 
1175     // Copy the status of another candidate without changing policy.
1176     void setBest(SchedCandidate &Best) {
1177       assert(Best.Reason != NoCand && "uninitialized Sched candidate");
1178       SU = Best.SU;
1179       Reason = Best.Reason;
1180       RPDelta = Best.RPDelta;
1181       ResDelta = Best.ResDelta;
1182     }
1183 
1184     void initResourceDelta(const ScheduleDAGMI *DAG,
1185                            const TargetSchedModel *SchedModel);
1186   };
1187 
1188   /// Summarize the unscheduled region.
1189   struct SchedRemainder {
1190     // Critical path through the DAG in expected latency.
1191     unsigned CriticalPath;
1192 
1193     // Unscheduled resources
1194     SmallVector<unsigned, 16> RemainingCounts;
1195     // Critical resource for the unscheduled zone.
1196     unsigned CritResIdx;
1197     // Number of micro-ops left to schedule.
1198     unsigned RemainingMicroOps;
1199 
1200     void reset() {
1201       CriticalPath = 0;
1202       RemainingCounts.clear();
1203       CritResIdx = 0;
1204       RemainingMicroOps = 0;
1205     }
1206 
1207     SchedRemainder() { reset(); }
1208 
1209     void init(ScheduleDAGMI *DAG, const TargetSchedModel *SchedModel);
1210 
1211     unsigned getMaxRemainingCount(const TargetSchedModel *SchedModel) const {
1212       if (!SchedModel->hasInstrSchedModel())
1213         return 0;
1214 
1215       return std::max(
1216         RemainingMicroOps * SchedModel->getMicroOpFactor(),
1217         RemainingCounts[CritResIdx]);
1218     }
1219   };
1220 
1221   /// Each Scheduling boundary is associated with ready queues. It tracks the
1222   /// current cycle in the direction of movement, and maintains the state
1223   /// of "hazards" and other interlocks at the current cycle.
1224   struct SchedBoundary {
1225     ScheduleDAGMI *DAG;
1226     const TargetSchedModel *SchedModel;
1227     SchedRemainder *Rem;
1228 
1229     ReadyQueue Available;
1230     ReadyQueue Pending;
1231     bool CheckPending;
1232 
1233     // For heuristics, keep a list of the nodes that immediately depend on the
1234     // most recently scheduled node.
1235     SmallPtrSet<const SUnit*, 8> NextSUs;
1236 
1237     ScheduleHazardRecognizer *HazardRec;
1238 
1239     unsigned CurrCycle;
1240     unsigned IssueCount;
1241 
1242     /// MinReadyCycle - Cycle of the soonest available instruction.
1243     unsigned MinReadyCycle;
1244 
1245     // The expected latency of the critical path in this scheduled zone.
1246     unsigned ExpectedLatency;
1247 
1248     // Resources used in the scheduled zone beyond this boundary.
1249     SmallVector<unsigned, 16> ResourceCounts;
1250 
1251     // Cache the critical resources ID in this scheduled zone.
1252     unsigned CritResIdx;
1253 
1254     // Is the scheduled region resource limited vs. latency limited.
1255     bool IsResourceLimited;
1256 
1257     unsigned ExpectedCount;
1258 
1259 #ifndef NDEBUG
1260     // Remember the greatest min operand latency.
1261     unsigned MaxMinLatency;
1262 #endif
1263 
1264     void reset() {
1265       // A new HazardRec is created for each DAG and owned by SchedBoundary.
1266       delete HazardRec;
1267 
1268       Available.clear();
1269       Pending.clear();
1270       CheckPending = false;
1271       NextSUs.clear();
1272       HazardRec = 0;
1273       CurrCycle = 0;
1274       IssueCount = 0;
1275       MinReadyCycle = UINT_MAX;
1276       ExpectedLatency = 0;
1277       ResourceCounts.resize(1);
1278       assert(!ResourceCounts[0] && "nonzero count for bad resource");
1279       CritResIdx = 0;
1280       IsResourceLimited = false;
1281       ExpectedCount = 0;
1282 #ifndef NDEBUG
1283       MaxMinLatency = 0;
1284 #endif
1285       // Reserve a zero-count for invalid CritResIdx.
1286       ResourceCounts.resize(1);
1287     }
1288 
1289     /// Pending queues extend the ready queues with the same ID and the
1290     /// PendingFlag set.
1291     SchedBoundary(unsigned ID, const Twine &Name):
1292       DAG(0), SchedModel(0), Rem(0), Available(ID, Name+".A"),
1293       Pending(ID << ConvergingScheduler::LogMaxQID, Name+".P"),
1294       HazardRec(0) {
1295       reset();
1296     }
1297 
1298     ~SchedBoundary() { delete HazardRec; }
1299 
1300     void init(ScheduleDAGMI *dag, const TargetSchedModel *smodel,
1301               SchedRemainder *rem);
1302 
1303     bool isTop() const {
1304       return Available.getID() == ConvergingScheduler::TopQID;
1305     }
1306 
1307     unsigned getUnscheduledLatency(SUnit *SU) const {
1308       if (isTop())
1309         return SU->getHeight();
1310       return SU->getDepth() + SU->Latency;
1311     }
1312 
1313     unsigned getCriticalCount() const {
1314       return ResourceCounts[CritResIdx];
1315     }
1316 
1317     bool checkHazard(SUnit *SU);
1318 
1319     void setLatencyPolicy(CandPolicy &Policy);
1320 
1321     void releaseNode(SUnit *SU, unsigned ReadyCycle);
1322 
1323     void bumpCycle();
1324 
1325     void countResource(unsigned PIdx, unsigned Cycles);
1326 
1327     void bumpNode(SUnit *SU);
1328 
1329     void releasePending();
1330 
1331     void removeReady(SUnit *SU);
1332 
1333     SUnit *pickOnlyChoice();
1334   };
1335 
1336 private:
1337   ScheduleDAGMI *DAG;
1338   const TargetSchedModel *SchedModel;
1339   const TargetRegisterInfo *TRI;
1340 
1341   // State of the top and bottom scheduled instruction boundaries.
1342   SchedRemainder Rem;
1343   SchedBoundary Top;
1344   SchedBoundary Bot;
1345 
1346 public:
1347   /// SUnit::NodeQueueId: 0 (none), 1 (top), 2 (bot), 3 (both)
1348   enum {
1349     TopQID = 1,
1350     BotQID = 2,
1351     LogMaxQID = 2
1352   };
1353 
1354   ConvergingScheduler():
1355     DAG(0), SchedModel(0), TRI(0), Top(TopQID, "TopQ"), Bot(BotQID, "BotQ") {}
1356 
1357   virtual void initialize(ScheduleDAGMI *dag);
1358 
1359   virtual SUnit *pickNode(bool &IsTopNode);
1360 
1361   virtual void schedNode(SUnit *SU, bool IsTopNode);
1362 
1363   virtual void releaseTopNode(SUnit *SU);
1364 
1365   virtual void releaseBottomNode(SUnit *SU);
1366 
1367   virtual void registerRoots();
1368 
1369 protected:
1370   void balanceZones(
1371     ConvergingScheduler::SchedBoundary &CriticalZone,
1372     ConvergingScheduler::SchedCandidate &CriticalCand,
1373     ConvergingScheduler::SchedBoundary &OppositeZone,
1374     ConvergingScheduler::SchedCandidate &OppositeCand);
1375 
1376   void checkResourceLimits(ConvergingScheduler::SchedCandidate &TopCand,
1377                            ConvergingScheduler::SchedCandidate &BotCand);
1378 
1379   void tryCandidate(SchedCandidate &Cand,
1380                     SchedCandidate &TryCand,
1381                     SchedBoundary &Zone,
1382                     const RegPressureTracker &RPTracker,
1383                     RegPressureTracker &TempTracker);
1384 
1385   SUnit *pickNodeBidirectional(bool &IsTopNode);
1386 
1387   void pickNodeFromQueue(SchedBoundary &Zone,
1388                          const RegPressureTracker &RPTracker,
1389                          SchedCandidate &Candidate);
1390 
1391   void reschedulePhysRegCopies(SUnit *SU, bool isTop);
1392 
1393 #ifndef NDEBUG
1394   void traceCandidate(const SchedCandidate &Cand);
1395 #endif
1396 };
1397 } // namespace
1398 
1399 void ConvergingScheduler::SchedRemainder::
1400 init(ScheduleDAGMI *DAG, const TargetSchedModel *SchedModel) {
1401   reset();
1402   if (!SchedModel->hasInstrSchedModel())
1403     return;
1404   RemainingCounts.resize(SchedModel->getNumProcResourceKinds());
1405   for (std::vector<SUnit>::iterator
1406          I = DAG->SUnits.begin(), E = DAG->SUnits.end(); I != E; ++I) {
1407     const MCSchedClassDesc *SC = DAG->getSchedClass(&*I);
1408     RemainingMicroOps += SchedModel->getNumMicroOps(I->getInstr(), SC);
1409     for (TargetSchedModel::ProcResIter
1410            PI = SchedModel->getWriteProcResBegin(SC),
1411            PE = SchedModel->getWriteProcResEnd(SC); PI != PE; ++PI) {
1412       unsigned PIdx = PI->ProcResourceIdx;
1413       unsigned Factor = SchedModel->getResourceFactor(PIdx);
1414       RemainingCounts[PIdx] += (Factor * PI->Cycles);
1415     }
1416   }
1417   for (unsigned PIdx = 0, PEnd = SchedModel->getNumProcResourceKinds();
1418        PIdx != PEnd; ++PIdx) {
1419     if ((int)(RemainingCounts[PIdx] - RemainingCounts[CritResIdx])
1420         >= (int)SchedModel->getLatencyFactor()) {
1421       CritResIdx = PIdx;
1422     }
1423   }
1424 }
1425 
1426 void ConvergingScheduler::SchedBoundary::
1427 init(ScheduleDAGMI *dag, const TargetSchedModel *smodel, SchedRemainder *rem) {
1428   reset();
1429   DAG = dag;
1430   SchedModel = smodel;
1431   Rem = rem;
1432   if (SchedModel->hasInstrSchedModel())
1433     ResourceCounts.resize(SchedModel->getNumProcResourceKinds());
1434 }
1435 
1436 void ConvergingScheduler::initialize(ScheduleDAGMI *dag) {
1437   DAG = dag;
1438   SchedModel = DAG->getSchedModel();
1439   TRI = DAG->TRI;
1440 
1441   Rem.init(DAG, SchedModel);
1442   Top.init(DAG, SchedModel, &Rem);
1443   Bot.init(DAG, SchedModel, &Rem);
1444 
1445   // Initialize resource counts.
1446 
1447   // Initialize the HazardRecognizers. If itineraries don't exist, are empty, or
1448   // are disabled, then these HazardRecs will be disabled.
1449   const InstrItineraryData *Itin = SchedModel->getInstrItineraries();
1450   const TargetMachine &TM = DAG->MF.getTarget();
1451   Top.HazardRec = TM.getInstrInfo()->CreateTargetMIHazardRecognizer(Itin, DAG);
1452   Bot.HazardRec = TM.getInstrInfo()->CreateTargetMIHazardRecognizer(Itin, DAG);
1453 
1454   assert((!ForceTopDown || !ForceBottomUp) &&
1455          "-misched-topdown incompatible with -misched-bottomup");
1456 }
1457 
1458 void ConvergingScheduler::releaseTopNode(SUnit *SU) {
1459   if (SU->isScheduled)
1460     return;
1461 
1462   for (SUnit::pred_iterator I = SU->Preds.begin(), E = SU->Preds.end();
1463        I != E; ++I) {
1464     unsigned PredReadyCycle = I->getSUnit()->TopReadyCycle;
1465     unsigned MinLatency = I->getMinLatency();
1466 #ifndef NDEBUG
1467     Top.MaxMinLatency = std::max(MinLatency, Top.MaxMinLatency);
1468 #endif
1469     if (SU->TopReadyCycle < PredReadyCycle + MinLatency)
1470       SU->TopReadyCycle = PredReadyCycle + MinLatency;
1471   }
1472   Top.releaseNode(SU, SU->TopReadyCycle);
1473 }
1474 
1475 void ConvergingScheduler::releaseBottomNode(SUnit *SU) {
1476   if (SU->isScheduled)
1477     return;
1478 
1479   assert(SU->getInstr() && "Scheduled SUnit must have instr");
1480 
1481   for (SUnit::succ_iterator I = SU->Succs.begin(), E = SU->Succs.end();
1482        I != E; ++I) {
1483     if (I->isWeak())
1484       continue;
1485     unsigned SuccReadyCycle = I->getSUnit()->BotReadyCycle;
1486     unsigned MinLatency = I->getMinLatency();
1487 #ifndef NDEBUG
1488     Bot.MaxMinLatency = std::max(MinLatency, Bot.MaxMinLatency);
1489 #endif
1490     if (SU->BotReadyCycle < SuccReadyCycle + MinLatency)
1491       SU->BotReadyCycle = SuccReadyCycle + MinLatency;
1492   }
1493   Bot.releaseNode(SU, SU->BotReadyCycle);
1494 }
1495 
1496 void ConvergingScheduler::registerRoots() {
1497   Rem.CriticalPath = DAG->ExitSU.getDepth();
1498   // Some roots may not feed into ExitSU. Check all of them in case.
1499   for (std::vector<SUnit*>::const_iterator
1500          I = Bot.Available.begin(), E = Bot.Available.end(); I != E; ++I) {
1501     if ((*I)->getDepth() > Rem.CriticalPath)
1502       Rem.CriticalPath = (*I)->getDepth();
1503   }
1504   DEBUG(dbgs() << "Critical Path: " << Rem.CriticalPath << '\n');
1505 }
1506 
1507 /// Does this SU have a hazard within the current instruction group.
1508 ///
1509 /// The scheduler supports two modes of hazard recognition. The first is the
1510 /// ScheduleHazardRecognizer API. It is a fully general hazard recognizer that
1511 /// supports highly complicated in-order reservation tables
1512 /// (ScoreboardHazardRecognizer) and arbitraty target-specific logic.
1513 ///
1514 /// The second is a streamlined mechanism that checks for hazards based on
1515 /// simple counters that the scheduler itself maintains. It explicitly checks
1516 /// for instruction dispatch limitations, including the number of micro-ops that
1517 /// can dispatch per cycle.
1518 ///
1519 /// TODO: Also check whether the SU must start a new group.
1520 bool ConvergingScheduler::SchedBoundary::checkHazard(SUnit *SU) {
1521   if (HazardRec->isEnabled())
1522     return HazardRec->getHazardType(SU) != ScheduleHazardRecognizer::NoHazard;
1523 
1524   unsigned uops = SchedModel->getNumMicroOps(SU->getInstr());
1525   if ((IssueCount > 0) && (IssueCount + uops > SchedModel->getIssueWidth())) {
1526     DEBUG(dbgs() << "  SU(" << SU->NodeNum << ") uops="
1527           << SchedModel->getNumMicroOps(SU->getInstr()) << '\n');
1528     return true;
1529   }
1530   return false;
1531 }
1532 
1533 /// Compute the remaining latency to determine whether ILP should be increased.
1534 void ConvergingScheduler::SchedBoundary::setLatencyPolicy(CandPolicy &Policy) {
1535   // FIXME: compile time. In all, we visit four queues here one we should only
1536   // need to visit the one that was last popped if we cache the result.
1537   unsigned RemLatency = 0;
1538   for (ReadyQueue::iterator I = Available.begin(), E = Available.end();
1539        I != E; ++I) {
1540     unsigned L = getUnscheduledLatency(*I);
1541     DEBUG(dbgs() << "  " << Available.getName()
1542           << " RemLatency SU(" << (*I)->NodeNum << ") " << L << '\n');
1543     if (L > RemLatency)
1544       RemLatency = L;
1545   }
1546   for (ReadyQueue::iterator I = Pending.begin(), E = Pending.end();
1547        I != E; ++I) {
1548     unsigned L = getUnscheduledLatency(*I);
1549     if (L > RemLatency)
1550       RemLatency = L;
1551   }
1552   unsigned CriticalPathLimit = Rem->CriticalPath + SchedModel->getILPWindow();
1553   DEBUG(dbgs() << "  " << Available.getName()
1554         << " ExpectedLatency " << ExpectedLatency
1555         << " CP Limit " << CriticalPathLimit << '\n');
1556   if (RemLatency + ExpectedLatency >= CriticalPathLimit
1557       && RemLatency > Rem->getMaxRemainingCount(SchedModel)) {
1558     Policy.ReduceLatency = true;
1559     DEBUG(dbgs() << "  Increase ILP: " << Available.getName() << '\n');
1560   }
1561 }
1562 
1563 void ConvergingScheduler::SchedBoundary::releaseNode(SUnit *SU,
1564                                                      unsigned ReadyCycle) {
1565 
1566   if (ReadyCycle < MinReadyCycle)
1567     MinReadyCycle = ReadyCycle;
1568 
1569   // Check for interlocks first. For the purpose of other heuristics, an
1570   // instruction that cannot issue appears as if it's not in the ReadyQueue.
1571   if (ReadyCycle > CurrCycle || checkHazard(SU))
1572     Pending.push(SU);
1573   else
1574     Available.push(SU);
1575 
1576   // Record this node as an immediate dependent of the scheduled node.
1577   NextSUs.insert(SU);
1578 }
1579 
1580 /// Move the boundary of scheduled code by one cycle.
1581 void ConvergingScheduler::SchedBoundary::bumpCycle() {
1582   unsigned Width = SchedModel->getIssueWidth();
1583   IssueCount = (IssueCount <= Width) ? 0 : IssueCount - Width;
1584 
1585   unsigned NextCycle = CurrCycle + 1;
1586   assert(MinReadyCycle < UINT_MAX && "MinReadyCycle uninitialized");
1587   if (MinReadyCycle > NextCycle) {
1588     IssueCount = 0;
1589     NextCycle = MinReadyCycle;
1590   }
1591 
1592   if (!HazardRec->isEnabled()) {
1593     // Bypass HazardRec virtual calls.
1594     CurrCycle = NextCycle;
1595   }
1596   else {
1597     // Bypass getHazardType calls in case of long latency.
1598     for (; CurrCycle != NextCycle; ++CurrCycle) {
1599       if (isTop())
1600         HazardRec->AdvanceCycle();
1601       else
1602         HazardRec->RecedeCycle();
1603     }
1604   }
1605   CheckPending = true;
1606   IsResourceLimited = getCriticalCount() > std::max(ExpectedLatency, CurrCycle);
1607 
1608   DEBUG(dbgs() << "  " << Available.getName()
1609         << " Cycle: " << CurrCycle << '\n');
1610 }
1611 
1612 /// Add the given processor resource to this scheduled zone.
1613 void ConvergingScheduler::SchedBoundary::countResource(unsigned PIdx,
1614                                                        unsigned Cycles) {
1615   unsigned Factor = SchedModel->getResourceFactor(PIdx);
1616   DEBUG(dbgs() << "  " << SchedModel->getProcResource(PIdx)->Name
1617         << " +(" << Cycles << "x" << Factor
1618         << ") / " << SchedModel->getLatencyFactor() << '\n');
1619 
1620   unsigned Count = Factor * Cycles;
1621   ResourceCounts[PIdx] += Count;
1622   assert(Rem->RemainingCounts[PIdx] >= Count && "resource double counted");
1623   Rem->RemainingCounts[PIdx] -= Count;
1624 
1625   // Check if this resource exceeds the current critical resource by a full
1626   // cycle. If so, it becomes the critical resource.
1627   if ((int)(ResourceCounts[PIdx] - ResourceCounts[CritResIdx])
1628       >= (int)SchedModel->getLatencyFactor()) {
1629     CritResIdx = PIdx;
1630     DEBUG(dbgs() << "  *** Critical resource "
1631           << SchedModel->getProcResource(PIdx)->Name << " x"
1632           << ResourceCounts[PIdx] << '\n');
1633   }
1634 }
1635 
1636 /// Move the boundary of scheduled code by one SUnit.
1637 void ConvergingScheduler::SchedBoundary::bumpNode(SUnit *SU) {
1638   // Update the reservation table.
1639   if (HazardRec->isEnabled()) {
1640     if (!isTop() && SU->isCall) {
1641       // Calls are scheduled with their preceding instructions. For bottom-up
1642       // scheduling, clear the pipeline state before emitting.
1643       HazardRec->Reset();
1644     }
1645     HazardRec->EmitInstruction(SU);
1646   }
1647   // Update resource counts and critical resource.
1648   if (SchedModel->hasInstrSchedModel()) {
1649     const MCSchedClassDesc *SC = DAG->getSchedClass(SU);
1650     Rem->RemainingMicroOps -= SchedModel->getNumMicroOps(SU->getInstr(), SC);
1651     for (TargetSchedModel::ProcResIter
1652            PI = SchedModel->getWriteProcResBegin(SC),
1653            PE = SchedModel->getWriteProcResEnd(SC); PI != PE; ++PI) {
1654       countResource(PI->ProcResourceIdx, PI->Cycles);
1655     }
1656   }
1657   if (isTop()) {
1658     if (SU->getDepth() > ExpectedLatency)
1659       ExpectedLatency = SU->getDepth();
1660   }
1661   else {
1662     if (SU->getHeight() > ExpectedLatency)
1663       ExpectedLatency = SU->getHeight();
1664   }
1665 
1666   IsResourceLimited = getCriticalCount() > std::max(ExpectedLatency, CurrCycle);
1667 
1668   // Check the instruction group dispatch limit.
1669   // TODO: Check if this SU must end a dispatch group.
1670   IssueCount += SchedModel->getNumMicroOps(SU->getInstr());
1671 
1672   // checkHazard prevents scheduling multiple instructions per cycle that exceed
1673   // issue width. However, we commonly reach the maximum. In this case
1674   // opportunistically bump the cycle to avoid uselessly checking everything in
1675   // the readyQ. Furthermore, a single instruction may produce more than one
1676   // cycle's worth of micro-ops.
1677   if (IssueCount >= SchedModel->getIssueWidth()) {
1678     DEBUG(dbgs() << "  *** Max instrs at cycle " << CurrCycle << '\n');
1679     bumpCycle();
1680   }
1681 }
1682 
1683 /// Release pending ready nodes in to the available queue. This makes them
1684 /// visible to heuristics.
1685 void ConvergingScheduler::SchedBoundary::releasePending() {
1686   // If the available queue is empty, it is safe to reset MinReadyCycle.
1687   if (Available.empty())
1688     MinReadyCycle = UINT_MAX;
1689 
1690   // Check to see if any of the pending instructions are ready to issue.  If
1691   // so, add them to the available queue.
1692   for (unsigned i = 0, e = Pending.size(); i != e; ++i) {
1693     SUnit *SU = *(Pending.begin()+i);
1694     unsigned ReadyCycle = isTop() ? SU->TopReadyCycle : SU->BotReadyCycle;
1695 
1696     if (ReadyCycle < MinReadyCycle)
1697       MinReadyCycle = ReadyCycle;
1698 
1699     if (ReadyCycle > CurrCycle)
1700       continue;
1701 
1702     if (checkHazard(SU))
1703       continue;
1704 
1705     Available.push(SU);
1706     Pending.remove(Pending.begin()+i);
1707     --i; --e;
1708   }
1709   DEBUG(if (!Pending.empty()) Pending.dump());
1710   CheckPending = false;
1711 }
1712 
1713 /// Remove SU from the ready set for this boundary.
1714 void ConvergingScheduler::SchedBoundary::removeReady(SUnit *SU) {
1715   if (Available.isInQueue(SU))
1716     Available.remove(Available.find(SU));
1717   else {
1718     assert(Pending.isInQueue(SU) && "bad ready count");
1719     Pending.remove(Pending.find(SU));
1720   }
1721 }
1722 
1723 /// If this queue only has one ready candidate, return it. As a side effect,
1724 /// defer any nodes that now hit a hazard, and advance the cycle until at least
1725 /// one node is ready. If multiple instructions are ready, return NULL.
1726 SUnit *ConvergingScheduler::SchedBoundary::pickOnlyChoice() {
1727   if (CheckPending)
1728     releasePending();
1729 
1730   if (IssueCount > 0) {
1731     // Defer any ready instrs that now have a hazard.
1732     for (ReadyQueue::iterator I = Available.begin(); I != Available.end();) {
1733       if (checkHazard(*I)) {
1734         Pending.push(*I);
1735         I = Available.remove(I);
1736         continue;
1737       }
1738       ++I;
1739     }
1740   }
1741   for (unsigned i = 0; Available.empty(); ++i) {
1742     assert(i <= (HazardRec->getMaxLookAhead() + MaxMinLatency) &&
1743            "permanent hazard"); (void)i;
1744     bumpCycle();
1745     releasePending();
1746   }
1747   if (Available.size() == 1)
1748     return *Available.begin();
1749   return NULL;
1750 }
1751 
1752 /// Record the candidate policy for opposite zones with different critical
1753 /// resources.
1754 ///
1755 /// If the CriticalZone is latency limited, don't force a policy for the
1756 /// candidates here. Instead, setLatencyPolicy sets ReduceLatency if needed.
1757 void ConvergingScheduler::balanceZones(
1758   ConvergingScheduler::SchedBoundary &CriticalZone,
1759   ConvergingScheduler::SchedCandidate &CriticalCand,
1760   ConvergingScheduler::SchedBoundary &OppositeZone,
1761   ConvergingScheduler::SchedCandidate &OppositeCand) {
1762 
1763   if (!CriticalZone.IsResourceLimited)
1764     return;
1765   assert(SchedModel->hasInstrSchedModel() && "required schedmodel");
1766 
1767   SchedRemainder *Rem = CriticalZone.Rem;
1768 
1769   // If the critical zone is overconsuming a resource relative to the
1770   // remainder, try to reduce it.
1771   unsigned RemainingCritCount =
1772     Rem->RemainingCounts[CriticalZone.CritResIdx];
1773   if ((int)(Rem->getMaxRemainingCount(SchedModel) - RemainingCritCount)
1774       > (int)SchedModel->getLatencyFactor()) {
1775     CriticalCand.Policy.ReduceResIdx = CriticalZone.CritResIdx;
1776     DEBUG(dbgs() << "  Balance " << CriticalZone.Available.getName()
1777           << " reduce "
1778           << SchedModel->getProcResource(CriticalZone.CritResIdx)->Name
1779           << '\n');
1780   }
1781   // If the other zone is underconsuming a resource relative to the full zone,
1782   // try to increase it.
1783   unsigned OppositeCount =
1784     OppositeZone.ResourceCounts[CriticalZone.CritResIdx];
1785   if ((int)(OppositeZone.ExpectedCount - OppositeCount)
1786       > (int)SchedModel->getLatencyFactor()) {
1787     OppositeCand.Policy.DemandResIdx = CriticalZone.CritResIdx;
1788     DEBUG(dbgs() << "  Balance " << OppositeZone.Available.getName()
1789           << " demand "
1790           << SchedModel->getProcResource(OppositeZone.CritResIdx)->Name
1791           << '\n');
1792   }
1793 }
1794 
1795 /// Determine if the scheduled zones exceed resource limits or critical path and
1796 /// set each candidate's ReduceHeight policy accordingly.
1797 void ConvergingScheduler::checkResourceLimits(
1798   ConvergingScheduler::SchedCandidate &TopCand,
1799   ConvergingScheduler::SchedCandidate &BotCand) {
1800 
1801   // Set ReduceLatency to true if needed.
1802   Bot.setLatencyPolicy(BotCand.Policy);
1803   Top.setLatencyPolicy(TopCand.Policy);
1804 
1805   // Handle resource-limited regions.
1806   if (Top.IsResourceLimited && Bot.IsResourceLimited
1807       && Top.CritResIdx == Bot.CritResIdx) {
1808     // If the scheduled critical resource in both zones is no longer the
1809     // critical remaining resource, attempt to reduce resource height both ways.
1810     if (Top.CritResIdx != Rem.CritResIdx) {
1811       TopCand.Policy.ReduceResIdx = Top.CritResIdx;
1812       BotCand.Policy.ReduceResIdx = Bot.CritResIdx;
1813       DEBUG(dbgs() << "  Reduce scheduled "
1814             << SchedModel->getProcResource(Top.CritResIdx)->Name << '\n');
1815     }
1816     return;
1817   }
1818   // Handle latency-limited regions.
1819   if (!Top.IsResourceLimited && !Bot.IsResourceLimited) {
1820     // If the total scheduled expected latency exceeds the region's critical
1821     // path then reduce latency both ways.
1822     //
1823     // Just because a zone is not resource limited does not mean it is latency
1824     // limited. Unbuffered resource, such as max micro-ops may cause CurrCycle
1825     // to exceed expected latency.
1826     if ((Top.ExpectedLatency + Bot.ExpectedLatency >= Rem.CriticalPath)
1827         && (Rem.CriticalPath > Top.CurrCycle + Bot.CurrCycle)) {
1828       TopCand.Policy.ReduceLatency = true;
1829       BotCand.Policy.ReduceLatency = true;
1830       DEBUG(dbgs() << "  Reduce scheduled latency " << Top.ExpectedLatency
1831             << " + " << Bot.ExpectedLatency << '\n');
1832     }
1833     return;
1834   }
1835   // The critical resource is different in each zone, so request balancing.
1836 
1837   // Compute the cost of each zone.
1838   Top.ExpectedCount = std::max(Top.ExpectedLatency, Top.CurrCycle);
1839   Top.ExpectedCount = std::max(
1840     Top.getCriticalCount(),
1841     Top.ExpectedCount * SchedModel->getLatencyFactor());
1842   Bot.ExpectedCount = std::max(Bot.ExpectedLatency, Bot.CurrCycle);
1843   Bot.ExpectedCount = std::max(
1844     Bot.getCriticalCount(),
1845     Bot.ExpectedCount * SchedModel->getLatencyFactor());
1846 
1847   balanceZones(Top, TopCand, Bot, BotCand);
1848   balanceZones(Bot, BotCand, Top, TopCand);
1849 }
1850 
1851 void ConvergingScheduler::SchedCandidate::
1852 initResourceDelta(const ScheduleDAGMI *DAG,
1853                   const TargetSchedModel *SchedModel) {
1854   if (!Policy.ReduceResIdx && !Policy.DemandResIdx)
1855     return;
1856 
1857   const MCSchedClassDesc *SC = DAG->getSchedClass(SU);
1858   for (TargetSchedModel::ProcResIter
1859          PI = SchedModel->getWriteProcResBegin(SC),
1860          PE = SchedModel->getWriteProcResEnd(SC); PI != PE; ++PI) {
1861     if (PI->ProcResourceIdx == Policy.ReduceResIdx)
1862       ResDelta.CritResources += PI->Cycles;
1863     if (PI->ProcResourceIdx == Policy.DemandResIdx)
1864       ResDelta.DemandedResources += PI->Cycles;
1865   }
1866 }
1867 
1868 /// Return true if this heuristic determines order.
1869 static bool tryLess(int TryVal, int CandVal,
1870                     ConvergingScheduler::SchedCandidate &TryCand,
1871                     ConvergingScheduler::SchedCandidate &Cand,
1872                     ConvergingScheduler::CandReason Reason) {
1873   if (TryVal < CandVal) {
1874     TryCand.Reason = Reason;
1875     return true;
1876   }
1877   if (TryVal > CandVal) {
1878     if (Cand.Reason > Reason)
1879       Cand.Reason = Reason;
1880     return true;
1881   }
1882   return false;
1883 }
1884 
1885 static bool tryGreater(int TryVal, int CandVal,
1886                        ConvergingScheduler::SchedCandidate &TryCand,
1887                        ConvergingScheduler::SchedCandidate &Cand,
1888                        ConvergingScheduler::CandReason Reason) {
1889   if (TryVal > CandVal) {
1890     TryCand.Reason = Reason;
1891     return true;
1892   }
1893   if (TryVal < CandVal) {
1894     if (Cand.Reason > Reason)
1895       Cand.Reason = Reason;
1896     return true;
1897   }
1898   return false;
1899 }
1900 
1901 static unsigned getWeakLeft(const SUnit *SU, bool isTop) {
1902   return (isTop) ? SU->WeakPredsLeft : SU->WeakSuccsLeft;
1903 }
1904 
1905 /// Minimize physical register live ranges. Regalloc wants them adjacent to
1906 /// their physreg def/use.
1907 ///
1908 /// FIXME: This is an unnecessary check on the critical path. Most are root/leaf
1909 /// copies which can be prescheduled. The rest (e.g. x86 MUL) could be bundled
1910 /// with the operation that produces or consumes the physreg. We'll do this when
1911 /// regalloc has support for parallel copies.
1912 static int biasPhysRegCopy(const SUnit *SU, bool isTop) {
1913   const MachineInstr *MI = SU->getInstr();
1914   if (!MI->isCopy())
1915     return 0;
1916 
1917   unsigned ScheduledOper = isTop ? 1 : 0;
1918   unsigned UnscheduledOper = isTop ? 0 : 1;
1919   // If we have already scheduled the physreg produce/consumer, immediately
1920   // schedule the copy.
1921   if (TargetRegisterInfo::isPhysicalRegister(
1922         MI->getOperand(ScheduledOper).getReg()))
1923     return 1;
1924   // If the physreg is at the boundary, defer it. Otherwise schedule it
1925   // immediately to free the dependent. We can hoist the copy later.
1926   bool AtBoundary = isTop ? !SU->NumSuccsLeft : !SU->NumPredsLeft;
1927   if (TargetRegisterInfo::isPhysicalRegister(
1928         MI->getOperand(UnscheduledOper).getReg()))
1929     return AtBoundary ? -1 : 1;
1930   return 0;
1931 }
1932 
1933 /// Apply a set of heursitics to a new candidate. Heuristics are currently
1934 /// hierarchical. This may be more efficient than a graduated cost model because
1935 /// we don't need to evaluate all aspects of the model for each node in the
1936 /// queue. But it's really done to make the heuristics easier to debug and
1937 /// statistically analyze.
1938 ///
1939 /// \param Cand provides the policy and current best candidate.
1940 /// \param TryCand refers to the next SUnit candidate, otherwise uninitialized.
1941 /// \param Zone describes the scheduled zone that we are extending.
1942 /// \param RPTracker describes reg pressure within the scheduled zone.
1943 /// \param TempTracker is a scratch pressure tracker to reuse in queries.
1944 void ConvergingScheduler::tryCandidate(SchedCandidate &Cand,
1945                                        SchedCandidate &TryCand,
1946                                        SchedBoundary &Zone,
1947                                        const RegPressureTracker &RPTracker,
1948                                        RegPressureTracker &TempTracker) {
1949 
1950   // Always initialize TryCand's RPDelta.
1951   TempTracker.getMaxPressureDelta(TryCand.SU->getInstr(), TryCand.RPDelta,
1952                                   DAG->getRegionCriticalPSets(),
1953                                   DAG->getRegPressure().MaxSetPressure);
1954 
1955   // Initialize the candidate if needed.
1956   if (!Cand.isValid()) {
1957     TryCand.Reason = NodeOrder;
1958     return;
1959   }
1960 
1961   if (tryGreater(biasPhysRegCopy(TryCand.SU, Zone.isTop()),
1962                  biasPhysRegCopy(Cand.SU, Zone.isTop()),
1963                  TryCand, Cand, PhysRegCopy))
1964     return;
1965 
1966   // Avoid exceeding the target's limit.
1967   if (tryLess(TryCand.RPDelta.Excess.UnitIncrease,
1968               Cand.RPDelta.Excess.UnitIncrease, TryCand, Cand, SingleExcess))
1969     return;
1970   if (Cand.Reason == SingleExcess)
1971     Cand.Reason = MultiPressure;
1972 
1973   // Avoid increasing the max critical pressure in the scheduled region.
1974   if (tryLess(TryCand.RPDelta.CriticalMax.UnitIncrease,
1975               Cand.RPDelta.CriticalMax.UnitIncrease,
1976               TryCand, Cand, SingleCritical))
1977     return;
1978   if (Cand.Reason == SingleCritical)
1979     Cand.Reason = MultiPressure;
1980 
1981   // Keep clustered nodes together to encourage downstream peephole
1982   // optimizations which may reduce resource requirements.
1983   //
1984   // This is a best effort to set things up for a post-RA pass. Optimizations
1985   // like generating loads of multiple registers should ideally be done within
1986   // the scheduler pass by combining the loads during DAG postprocessing.
1987   const SUnit *NextClusterSU =
1988     Zone.isTop() ? DAG->getNextClusterSucc() : DAG->getNextClusterPred();
1989   if (tryGreater(TryCand.SU == NextClusterSU, Cand.SU == NextClusterSU,
1990                  TryCand, Cand, Cluster))
1991     return;
1992 
1993   // Weak edges are for clustering and other constraints.
1994   //
1995   // Deferring TryCand here does not change Cand's reason. This is good in the
1996   // sense that a bad candidate shouldn't affect a previous candidate's
1997   // goodness, but bad in that it is assymetric and depends on queue order.
1998   CandReason OrigReason = Cand.Reason;
1999   if (tryLess(getWeakLeft(TryCand.SU, Zone.isTop()),
2000               getWeakLeft(Cand.SU, Zone.isTop()),
2001               TryCand, Cand, Weak)) {
2002     Cand.Reason = OrigReason;
2003     return;
2004   }
2005   // Avoid critical resource consumption and balance the schedule.
2006   TryCand.initResourceDelta(DAG, SchedModel);
2007   if (tryLess(TryCand.ResDelta.CritResources, Cand.ResDelta.CritResources,
2008               TryCand, Cand, ResourceReduce))
2009     return;
2010   if (tryGreater(TryCand.ResDelta.DemandedResources,
2011                  Cand.ResDelta.DemandedResources,
2012                  TryCand, Cand, ResourceDemand))
2013     return;
2014 
2015   // Avoid serializing long latency dependence chains.
2016   if (Cand.Policy.ReduceLatency) {
2017     if (Zone.isTop()) {
2018       if (Cand.SU->getDepth() * SchedModel->getLatencyFactor()
2019           > Zone.ExpectedCount) {
2020         if (tryLess(TryCand.SU->getDepth(), Cand.SU->getDepth(),
2021                     TryCand, Cand, TopDepthReduce))
2022           return;
2023       }
2024       if (tryGreater(TryCand.SU->getHeight(), Cand.SU->getHeight(),
2025                      TryCand, Cand, TopPathReduce))
2026         return;
2027     }
2028     else {
2029       if (Cand.SU->getHeight() * SchedModel->getLatencyFactor()
2030           > Zone.ExpectedCount) {
2031         if (tryLess(TryCand.SU->getHeight(), Cand.SU->getHeight(),
2032                     TryCand, Cand, BotHeightReduce))
2033           return;
2034       }
2035       if (tryGreater(TryCand.SU->getDepth(), Cand.SU->getDepth(),
2036                      TryCand, Cand, BotPathReduce))
2037         return;
2038     }
2039   }
2040 
2041   // Avoid increasing the max pressure of the entire region.
2042   if (tryLess(TryCand.RPDelta.CurrentMax.UnitIncrease,
2043               Cand.RPDelta.CurrentMax.UnitIncrease, TryCand, Cand, SingleMax))
2044     return;
2045   if (Cand.Reason == SingleMax)
2046     Cand.Reason = MultiPressure;
2047 
2048   // Prefer immediate defs/users of the last scheduled instruction. This is a
2049   // nice pressure avoidance strategy that also conserves the processor's
2050   // register renaming resources and keeps the machine code readable.
2051   if (tryGreater(Zone.NextSUs.count(TryCand.SU), Zone.NextSUs.count(Cand.SU),
2052                  TryCand, Cand, NextDefUse))
2053     return;
2054 
2055   // Fall through to original instruction order.
2056   if ((Zone.isTop() && TryCand.SU->NodeNum < Cand.SU->NodeNum)
2057       || (!Zone.isTop() && TryCand.SU->NodeNum > Cand.SU->NodeNum)) {
2058     TryCand.Reason = NodeOrder;
2059   }
2060 }
2061 
2062 /// pickNodeFromQueue helper that returns true if the LHS reg pressure effect is
2063 /// more desirable than RHS from scheduling standpoint.
2064 static bool compareRPDelta(const RegPressureDelta &LHS,
2065                            const RegPressureDelta &RHS) {
2066   // Compare each component of pressure in decreasing order of importance
2067   // without checking if any are valid. Invalid PressureElements are assumed to
2068   // have UnitIncrease==0, so are neutral.
2069 
2070   // Avoid increasing the max critical pressure in the scheduled region.
2071   if (LHS.Excess.UnitIncrease != RHS.Excess.UnitIncrease) {
2072     DEBUG(dbgs() << "  RP excess top - bot: "
2073           << (LHS.Excess.UnitIncrease - RHS.Excess.UnitIncrease) << '\n');
2074     return LHS.Excess.UnitIncrease < RHS.Excess.UnitIncrease;
2075   }
2076   // Avoid increasing the max critical pressure in the scheduled region.
2077   if (LHS.CriticalMax.UnitIncrease != RHS.CriticalMax.UnitIncrease) {
2078     DEBUG(dbgs() << "  RP critical top - bot: "
2079           << (LHS.CriticalMax.UnitIncrease - RHS.CriticalMax.UnitIncrease)
2080           << '\n');
2081     return LHS.CriticalMax.UnitIncrease < RHS.CriticalMax.UnitIncrease;
2082   }
2083   // Avoid increasing the max pressure of the entire region.
2084   if (LHS.CurrentMax.UnitIncrease != RHS.CurrentMax.UnitIncrease) {
2085     DEBUG(dbgs() << "  RP current top - bot: "
2086           << (LHS.CurrentMax.UnitIncrease - RHS.CurrentMax.UnitIncrease)
2087           << '\n');
2088     return LHS.CurrentMax.UnitIncrease < RHS.CurrentMax.UnitIncrease;
2089   }
2090   return false;
2091 }
2092 
2093 #ifndef NDEBUG
2094 const char *ConvergingScheduler::getReasonStr(
2095   ConvergingScheduler::CandReason Reason) {
2096   switch (Reason) {
2097   case NoCand:         return "NOCAND    ";
2098   case PhysRegCopy:    return "PREG-COPY";
2099   case SingleExcess:   return "REG-EXCESS";
2100   case SingleCritical: return "REG-CRIT  ";
2101   case Cluster:        return "CLUSTER   ";
2102   case Weak:           return "WEAK      ";
2103   case SingleMax:      return "REG-MAX   ";
2104   case MultiPressure:  return "REG-MULTI ";
2105   case ResourceReduce: return "RES-REDUCE";
2106   case ResourceDemand: return "RES-DEMAND";
2107   case TopDepthReduce: return "TOP-DEPTH ";
2108   case TopPathReduce:  return "TOP-PATH  ";
2109   case BotHeightReduce:return "BOT-HEIGHT";
2110   case BotPathReduce:  return "BOT-PATH  ";
2111   case NextDefUse:     return "DEF-USE   ";
2112   case NodeOrder:      return "ORDER     ";
2113   };
2114   llvm_unreachable("Unknown reason!");
2115 }
2116 
2117 void ConvergingScheduler::traceCandidate(const SchedCandidate &Cand) {
2118   PressureElement P;
2119   unsigned ResIdx = 0;
2120   unsigned Latency = 0;
2121   switch (Cand.Reason) {
2122   default:
2123     break;
2124   case SingleExcess:
2125     P = Cand.RPDelta.Excess;
2126     break;
2127   case SingleCritical:
2128     P = Cand.RPDelta.CriticalMax;
2129     break;
2130   case SingleMax:
2131     P = Cand.RPDelta.CurrentMax;
2132     break;
2133   case ResourceReduce:
2134     ResIdx = Cand.Policy.ReduceResIdx;
2135     break;
2136   case ResourceDemand:
2137     ResIdx = Cand.Policy.DemandResIdx;
2138     break;
2139   case TopDepthReduce:
2140     Latency = Cand.SU->getDepth();
2141     break;
2142   case TopPathReduce:
2143     Latency = Cand.SU->getHeight();
2144     break;
2145   case BotHeightReduce:
2146     Latency = Cand.SU->getHeight();
2147     break;
2148   case BotPathReduce:
2149     Latency = Cand.SU->getDepth();
2150     break;
2151   }
2152   dbgs() << "  SU(" << Cand.SU->NodeNum << ") " << getReasonStr(Cand.Reason);
2153   if (P.isValid())
2154     dbgs() << " " << TRI->getRegPressureSetName(P.PSetID)
2155            << ":" << P.UnitIncrease << " ";
2156   else
2157     dbgs() << "      ";
2158   if (ResIdx)
2159     dbgs() << " " << SchedModel->getProcResource(ResIdx)->Name << " ";
2160   else
2161     dbgs() << "         ";
2162   if (Latency)
2163     dbgs() << " " << Latency << " cycles ";
2164   else
2165     dbgs() << "          ";
2166   dbgs() << '\n';
2167 }
2168 #endif
2169 
2170 /// Pick the best candidate from the top queue.
2171 ///
2172 /// TODO: getMaxPressureDelta results can be mostly cached for each SUnit during
2173 /// DAG building. To adjust for the current scheduling location we need to
2174 /// maintain the number of vreg uses remaining to be top-scheduled.
2175 void ConvergingScheduler::pickNodeFromQueue(SchedBoundary &Zone,
2176                                             const RegPressureTracker &RPTracker,
2177                                             SchedCandidate &Cand) {
2178   ReadyQueue &Q = Zone.Available;
2179 
2180   DEBUG(Q.dump());
2181 
2182   // getMaxPressureDelta temporarily modifies the tracker.
2183   RegPressureTracker &TempTracker = const_cast<RegPressureTracker&>(RPTracker);
2184 
2185   for (ReadyQueue::iterator I = Q.begin(), E = Q.end(); I != E; ++I) {
2186 
2187     SchedCandidate TryCand(Cand.Policy);
2188     TryCand.SU = *I;
2189     tryCandidate(Cand, TryCand, Zone, RPTracker, TempTracker);
2190     if (TryCand.Reason != NoCand) {
2191       // Initialize resource delta if needed in case future heuristics query it.
2192       if (TryCand.ResDelta == SchedResourceDelta())
2193         TryCand.initResourceDelta(DAG, SchedModel);
2194       Cand.setBest(TryCand);
2195       DEBUG(traceCandidate(Cand));
2196     }
2197   }
2198 }
2199 
2200 static void tracePick(const ConvergingScheduler::SchedCandidate &Cand,
2201                       bool IsTop) {
2202   DEBUG(dbgs() << "Pick " << (IsTop ? "Top " : "Bot ")
2203         << ConvergingScheduler::getReasonStr(Cand.Reason) << '\n');
2204 }
2205 
2206 /// Pick the best candidate node from either the top or bottom queue.
2207 SUnit *ConvergingScheduler::pickNodeBidirectional(bool &IsTopNode) {
2208   // Schedule as far as possible in the direction of no choice. This is most
2209   // efficient, but also provides the best heuristics for CriticalPSets.
2210   if (SUnit *SU = Bot.pickOnlyChoice()) {
2211     IsTopNode = false;
2212     DEBUG(dbgs() << "Pick Top NOCAND\n");
2213     return SU;
2214   }
2215   if (SUnit *SU = Top.pickOnlyChoice()) {
2216     IsTopNode = true;
2217     DEBUG(dbgs() << "Pick Bot NOCAND\n");
2218     return SU;
2219   }
2220   CandPolicy NoPolicy;
2221   SchedCandidate BotCand(NoPolicy);
2222   SchedCandidate TopCand(NoPolicy);
2223   checkResourceLimits(TopCand, BotCand);
2224 
2225   // Prefer bottom scheduling when heuristics are silent.
2226   pickNodeFromQueue(Bot, DAG->getBotRPTracker(), BotCand);
2227   assert(BotCand.Reason != NoCand && "failed to find the first candidate");
2228 
2229   // If either Q has a single candidate that provides the least increase in
2230   // Excess pressure, we can immediately schedule from that Q.
2231   //
2232   // RegionCriticalPSets summarizes the pressure within the scheduled region and
2233   // affects picking from either Q. If scheduling in one direction must
2234   // increase pressure for one of the excess PSets, then schedule in that
2235   // direction first to provide more freedom in the other direction.
2236   if (BotCand.Reason == SingleExcess || BotCand.Reason == SingleCritical) {
2237     IsTopNode = false;
2238     tracePick(BotCand, IsTopNode);
2239     return BotCand.SU;
2240   }
2241   // Check if the top Q has a better candidate.
2242   pickNodeFromQueue(Top, DAG->getTopRPTracker(), TopCand);
2243   assert(TopCand.Reason != NoCand && "failed to find the first candidate");
2244 
2245   // If either Q has a single candidate that minimizes pressure above the
2246   // original region's pressure pick it.
2247   if (TopCand.Reason <= SingleMax || BotCand.Reason <= SingleMax) {
2248     if (TopCand.Reason < BotCand.Reason) {
2249       IsTopNode = true;
2250       tracePick(TopCand, IsTopNode);
2251       return TopCand.SU;
2252     }
2253     IsTopNode = false;
2254     tracePick(BotCand, IsTopNode);
2255     return BotCand.SU;
2256   }
2257   // Check for a salient pressure difference and pick the best from either side.
2258   if (compareRPDelta(TopCand.RPDelta, BotCand.RPDelta)) {
2259     IsTopNode = true;
2260     tracePick(TopCand, IsTopNode);
2261     return TopCand.SU;
2262   }
2263   // Otherwise prefer the bottom candidate, in node order if all else failed.
2264   if (TopCand.Reason < BotCand.Reason) {
2265     IsTopNode = true;
2266     tracePick(TopCand, IsTopNode);
2267     return TopCand.SU;
2268   }
2269   IsTopNode = false;
2270   tracePick(BotCand, IsTopNode);
2271   return BotCand.SU;
2272 }
2273 
2274 /// Pick the best node to balance the schedule. Implements MachineSchedStrategy.
2275 SUnit *ConvergingScheduler::pickNode(bool &IsTopNode) {
2276   if (DAG->top() == DAG->bottom()) {
2277     assert(Top.Available.empty() && Top.Pending.empty() &&
2278            Bot.Available.empty() && Bot.Pending.empty() && "ReadyQ garbage");
2279     return NULL;
2280   }
2281   SUnit *SU;
2282   do {
2283     if (ForceTopDown) {
2284       SU = Top.pickOnlyChoice();
2285       if (!SU) {
2286         CandPolicy NoPolicy;
2287         SchedCandidate TopCand(NoPolicy);
2288         pickNodeFromQueue(Top, DAG->getTopRPTracker(), TopCand);
2289         assert(TopCand.Reason != NoCand && "failed to find the first candidate");
2290         SU = TopCand.SU;
2291       }
2292       IsTopNode = true;
2293     }
2294     else if (ForceBottomUp) {
2295       SU = Bot.pickOnlyChoice();
2296       if (!SU) {
2297         CandPolicy NoPolicy;
2298         SchedCandidate BotCand(NoPolicy);
2299         pickNodeFromQueue(Bot, DAG->getBotRPTracker(), BotCand);
2300         assert(BotCand.Reason != NoCand && "failed to find the first candidate");
2301         SU = BotCand.SU;
2302       }
2303       IsTopNode = false;
2304     }
2305     else {
2306       SU = pickNodeBidirectional(IsTopNode);
2307     }
2308   } while (SU->isScheduled);
2309 
2310   if (SU->isTopReady())
2311     Top.removeReady(SU);
2312   if (SU->isBottomReady())
2313     Bot.removeReady(SU);
2314 
2315   DEBUG(dbgs() << "Scheduling SU(" << SU->NodeNum << ") " << *SU->getInstr());
2316   return SU;
2317 }
2318 
2319 void ConvergingScheduler::reschedulePhysRegCopies(SUnit *SU, bool isTop) {
2320 
2321   MachineBasicBlock::iterator InsertPos = SU->getInstr();
2322   if (!isTop)
2323     ++InsertPos;
2324   SmallVectorImpl<SDep> &Deps = isTop ? SU->Preds : SU->Succs;
2325 
2326   // Find already scheduled copies with a single physreg dependence and move
2327   // them just above the scheduled instruction.
2328   for (SmallVectorImpl<SDep>::iterator I = Deps.begin(), E = Deps.end();
2329        I != E; ++I) {
2330     if (I->getKind() != SDep::Data || !TRI->isPhysicalRegister(I->getReg()))
2331       continue;
2332     SUnit *DepSU = I->getSUnit();
2333     if (isTop ? DepSU->Succs.size() > 1 : DepSU->Preds.size() > 1)
2334       continue;
2335     MachineInstr *Copy = DepSU->getInstr();
2336     if (!Copy->isCopy())
2337       continue;
2338     DEBUG(dbgs() << "  Rescheduling physreg copy ";
2339           I->getSUnit()->dump(DAG));
2340     DAG->moveInstruction(Copy, InsertPos);
2341   }
2342 }
2343 
2344 /// Update the scheduler's state after scheduling a node. This is the same node
2345 /// that was just returned by pickNode(). However, ScheduleDAGMI needs to update
2346 /// it's state based on the current cycle before MachineSchedStrategy does.
2347 ///
2348 /// FIXME: Eventually, we may bundle physreg copies rather than rescheduling
2349 /// them here. See comments in biasPhysRegCopy.
2350 void ConvergingScheduler::schedNode(SUnit *SU, bool IsTopNode) {
2351   if (IsTopNode) {
2352     SU->TopReadyCycle = Top.CurrCycle;
2353     Top.bumpNode(SU);
2354     if (SU->hasPhysRegUses)
2355       reschedulePhysRegCopies(SU, true);
2356   }
2357   else {
2358     SU->BotReadyCycle = Bot.CurrCycle;
2359     Bot.bumpNode(SU);
2360     if (SU->hasPhysRegDefs)
2361       reschedulePhysRegCopies(SU, false);
2362   }
2363 }
2364 
2365 /// Create the standard converging machine scheduler. This will be used as the
2366 /// default scheduler if the target does not set a default.
2367 static ScheduleDAGInstrs *createConvergingSched(MachineSchedContext *C) {
2368   assert((!ForceTopDown || !ForceBottomUp) &&
2369          "-misched-topdown incompatible with -misched-bottomup");
2370   ScheduleDAGMI *DAG = new ScheduleDAGMI(C, new ConvergingScheduler());
2371   // Register DAG post-processors.
2372   //
2373   // FIXME: extend the mutation API to allow earlier mutations to instantiate
2374   // data and pass it to later mutations. Have a single mutation that gathers
2375   // the interesting nodes in one pass.
2376   if (EnableCopyConstrain)
2377     DAG->addMutation(new CopyConstrain(DAG->TII, DAG->TRI));
2378   if (EnableLoadCluster)
2379     DAG->addMutation(new LoadClusterMutation(DAG->TII, DAG->TRI));
2380   if (EnableMacroFusion)
2381     DAG->addMutation(new MacroFusion(DAG->TII));
2382   return DAG;
2383 }
2384 static MachineSchedRegistry
2385 ConvergingSchedRegistry("converge", "Standard converging scheduler.",
2386                         createConvergingSched);
2387 
2388 //===----------------------------------------------------------------------===//
2389 // ILP Scheduler. Currently for experimental analysis of heuristics.
2390 //===----------------------------------------------------------------------===//
2391 
2392 namespace {
2393 /// \brief Order nodes by the ILP metric.
2394 struct ILPOrder {
2395   const SchedDFSResult *DFSResult;
2396   const BitVector *ScheduledTrees;
2397   bool MaximizeILP;
2398 
2399   ILPOrder(bool MaxILP): DFSResult(0), ScheduledTrees(0), MaximizeILP(MaxILP) {}
2400 
2401   /// \brief Apply a less-than relation on node priority.
2402   ///
2403   /// (Return true if A comes after B in the Q.)
2404   bool operator()(const SUnit *A, const SUnit *B) const {
2405     unsigned SchedTreeA = DFSResult->getSubtreeID(A);
2406     unsigned SchedTreeB = DFSResult->getSubtreeID(B);
2407     if (SchedTreeA != SchedTreeB) {
2408       // Unscheduled trees have lower priority.
2409       if (ScheduledTrees->test(SchedTreeA) != ScheduledTrees->test(SchedTreeB))
2410         return ScheduledTrees->test(SchedTreeB);
2411 
2412       // Trees with shallower connections have have lower priority.
2413       if (DFSResult->getSubtreeLevel(SchedTreeA)
2414           != DFSResult->getSubtreeLevel(SchedTreeB)) {
2415         return DFSResult->getSubtreeLevel(SchedTreeA)
2416           < DFSResult->getSubtreeLevel(SchedTreeB);
2417       }
2418     }
2419     if (MaximizeILP)
2420       return DFSResult->getILP(A) < DFSResult->getILP(B);
2421     else
2422       return DFSResult->getILP(A) > DFSResult->getILP(B);
2423   }
2424 };
2425 
2426 /// \brief Schedule based on the ILP metric.
2427 class ILPScheduler : public MachineSchedStrategy {
2428   /// In case all subtrees are eventually connected to a common root through
2429   /// data dependence (e.g. reduction), place an upper limit on their size.
2430   ///
2431   /// FIXME: A subtree limit is generally good, but in the situation commented
2432   /// above, where multiple similar subtrees feed a common root, we should
2433   /// only split at a point where the resulting subtrees will be balanced.
2434   /// (a motivating test case must be found).
2435   static const unsigned SubtreeLimit = 16;
2436 
2437   ScheduleDAGMI *DAG;
2438   ILPOrder Cmp;
2439 
2440   std::vector<SUnit*> ReadyQ;
2441 public:
2442   ILPScheduler(bool MaximizeILP): DAG(0), Cmp(MaximizeILP) {}
2443 
2444   virtual void initialize(ScheduleDAGMI *dag) {
2445     DAG = dag;
2446     DAG->computeDFSResult();
2447     Cmp.DFSResult = DAG->getDFSResult();
2448     Cmp.ScheduledTrees = &DAG->getScheduledTrees();
2449     ReadyQ.clear();
2450   }
2451 
2452   virtual void registerRoots() {
2453     // Restore the heap in ReadyQ with the updated DFS results.
2454     std::make_heap(ReadyQ.begin(), ReadyQ.end(), Cmp);
2455   }
2456 
2457   /// Implement MachineSchedStrategy interface.
2458   /// -----------------------------------------
2459 
2460   /// Callback to select the highest priority node from the ready Q.
2461   virtual SUnit *pickNode(bool &IsTopNode) {
2462     if (ReadyQ.empty()) return NULL;
2463     std::pop_heap(ReadyQ.begin(), ReadyQ.end(), Cmp);
2464     SUnit *SU = ReadyQ.back();
2465     ReadyQ.pop_back();
2466     IsTopNode = false;
2467     DEBUG(dbgs() << "Pick node " << "SU(" << SU->NodeNum << ") "
2468           << " ILP: " << DAG->getDFSResult()->getILP(SU)
2469           << " Tree: " << DAG->getDFSResult()->getSubtreeID(SU) << " @"
2470           << DAG->getDFSResult()->getSubtreeLevel(
2471             DAG->getDFSResult()->getSubtreeID(SU)) << '\n'
2472           << "Scheduling " << *SU->getInstr());
2473     return SU;
2474   }
2475 
2476   /// \brief Scheduler callback to notify that a new subtree is scheduled.
2477   virtual void scheduleTree(unsigned SubtreeID) {
2478     std::make_heap(ReadyQ.begin(), ReadyQ.end(), Cmp);
2479   }
2480 
2481   /// Callback after a node is scheduled. Mark a newly scheduled tree, notify
2482   /// DFSResults, and resort the priority Q.
2483   virtual void schedNode(SUnit *SU, bool IsTopNode) {
2484     assert(!IsTopNode && "SchedDFSResult needs bottom-up");
2485   }
2486 
2487   virtual void releaseTopNode(SUnit *) { /*only called for top roots*/ }
2488 
2489   virtual void releaseBottomNode(SUnit *SU) {
2490     ReadyQ.push_back(SU);
2491     std::push_heap(ReadyQ.begin(), ReadyQ.end(), Cmp);
2492   }
2493 };
2494 } // namespace
2495 
2496 static ScheduleDAGInstrs *createILPMaxScheduler(MachineSchedContext *C) {
2497   return new ScheduleDAGMI(C, new ILPScheduler(true));
2498 }
2499 static ScheduleDAGInstrs *createILPMinScheduler(MachineSchedContext *C) {
2500   return new ScheduleDAGMI(C, new ILPScheduler(false));
2501 }
2502 static MachineSchedRegistry ILPMaxRegistry(
2503   "ilpmax", "Schedule bottom-up for max ILP", createILPMaxScheduler);
2504 static MachineSchedRegistry ILPMinRegistry(
2505   "ilpmin", "Schedule bottom-up for min ILP", createILPMinScheduler);
2506 
2507 //===----------------------------------------------------------------------===//
2508 // Machine Instruction Shuffler for Correctness Testing
2509 //===----------------------------------------------------------------------===//
2510 
2511 #ifndef NDEBUG
2512 namespace {
2513 /// Apply a less-than relation on the node order, which corresponds to the
2514 /// instruction order prior to scheduling. IsReverse implements greater-than.
2515 template<bool IsReverse>
2516 struct SUnitOrder {
2517   bool operator()(SUnit *A, SUnit *B) const {
2518     if (IsReverse)
2519       return A->NodeNum > B->NodeNum;
2520     else
2521       return A->NodeNum < B->NodeNum;
2522   }
2523 };
2524 
2525 /// Reorder instructions as much as possible.
2526 class InstructionShuffler : public MachineSchedStrategy {
2527   bool IsAlternating;
2528   bool IsTopDown;
2529 
2530   // Using a less-than relation (SUnitOrder<false>) for the TopQ priority
2531   // gives nodes with a higher number higher priority causing the latest
2532   // instructions to be scheduled first.
2533   PriorityQueue<SUnit*, std::vector<SUnit*>, SUnitOrder<false> >
2534     TopQ;
2535   // When scheduling bottom-up, use greater-than as the queue priority.
2536   PriorityQueue<SUnit*, std::vector<SUnit*>, SUnitOrder<true> >
2537     BottomQ;
2538 public:
2539   InstructionShuffler(bool alternate, bool topdown)
2540     : IsAlternating(alternate), IsTopDown(topdown) {}
2541 
2542   virtual void initialize(ScheduleDAGMI *) {
2543     TopQ.clear();
2544     BottomQ.clear();
2545   }
2546 
2547   /// Implement MachineSchedStrategy interface.
2548   /// -----------------------------------------
2549 
2550   virtual SUnit *pickNode(bool &IsTopNode) {
2551     SUnit *SU;
2552     if (IsTopDown) {
2553       do {
2554         if (TopQ.empty()) return NULL;
2555         SU = TopQ.top();
2556         TopQ.pop();
2557       } while (SU->isScheduled);
2558       IsTopNode = true;
2559     }
2560     else {
2561       do {
2562         if (BottomQ.empty()) return NULL;
2563         SU = BottomQ.top();
2564         BottomQ.pop();
2565       } while (SU->isScheduled);
2566       IsTopNode = false;
2567     }
2568     if (IsAlternating)
2569       IsTopDown = !IsTopDown;
2570     return SU;
2571   }
2572 
2573   virtual void schedNode(SUnit *SU, bool IsTopNode) {}
2574 
2575   virtual void releaseTopNode(SUnit *SU) {
2576     TopQ.push(SU);
2577   }
2578   virtual void releaseBottomNode(SUnit *SU) {
2579     BottomQ.push(SU);
2580   }
2581 };
2582 } // namespace
2583 
2584 static ScheduleDAGInstrs *createInstructionShuffler(MachineSchedContext *C) {
2585   bool Alternate = !ForceTopDown && !ForceBottomUp;
2586   bool TopDown = !ForceBottomUp;
2587   assert((TopDown || !ForceTopDown) &&
2588          "-misched-topdown incompatible with -misched-bottomup");
2589   return new ScheduleDAGMI(C, new InstructionShuffler(Alternate, TopDown));
2590 }
2591 static MachineSchedRegistry ShufflerRegistry(
2592   "shuffle", "Shuffle machine instructions alternating directions",
2593   createInstructionShuffler);
2594 #endif // !NDEBUG
2595 
2596 //===----------------------------------------------------------------------===//
2597 // GraphWriter support for ScheduleDAGMI.
2598 //===----------------------------------------------------------------------===//
2599 
2600 #ifndef NDEBUG
2601 namespace llvm {
2602 
2603 template<> struct GraphTraits<
2604   ScheduleDAGMI*> : public GraphTraits<ScheduleDAG*> {};
2605 
2606 template<>
2607 struct DOTGraphTraits<ScheduleDAGMI*> : public DefaultDOTGraphTraits {
2608 
2609   DOTGraphTraits (bool isSimple=false) : DefaultDOTGraphTraits(isSimple) {}
2610 
2611   static std::string getGraphName(const ScheduleDAG *G) {
2612     return G->MF.getName();
2613   }
2614 
2615   static bool renderGraphFromBottomUp() {
2616     return true;
2617   }
2618 
2619   static bool isNodeHidden(const SUnit *Node) {
2620     return (Node->NumPreds > 10 || Node->NumSuccs > 10);
2621   }
2622 
2623   static bool hasNodeAddressLabel(const SUnit *Node,
2624                                   const ScheduleDAG *Graph) {
2625     return false;
2626   }
2627 
2628   /// If you want to override the dot attributes printed for a particular
2629   /// edge, override this method.
2630   static std::string getEdgeAttributes(const SUnit *Node,
2631                                        SUnitIterator EI,
2632                                        const ScheduleDAG *Graph) {
2633     if (EI.isArtificialDep())
2634       return "color=cyan,style=dashed";
2635     if (EI.isCtrlDep())
2636       return "color=blue,style=dashed";
2637     return "";
2638   }
2639 
2640   static std::string getNodeLabel(const SUnit *SU, const ScheduleDAG *G) {
2641     std::string Str;
2642     raw_string_ostream SS(Str);
2643     SS << "SU(" << SU->NodeNum << ')';
2644     return SS.str();
2645   }
2646   static std::string getNodeDescription(const SUnit *SU, const ScheduleDAG *G) {
2647     return G->getGraphNodeLabel(SU);
2648   }
2649 
2650   static std::string getNodeAttributes(const SUnit *N,
2651                                        const ScheduleDAG *Graph) {
2652     std::string Str("shape=Mrecord");
2653     const SchedDFSResult *DFS =
2654       static_cast<const ScheduleDAGMI*>(Graph)->getDFSResult();
2655     if (DFS) {
2656       Str += ",style=filled,fillcolor=\"#";
2657       Str += DOT::getColorString(DFS->getSubtreeID(N));
2658       Str += '"';
2659     }
2660     return Str;
2661   }
2662 };
2663 } // namespace llvm
2664 #endif // NDEBUG
2665 
2666 /// viewGraph - Pop up a ghostview window with the reachable parts of the DAG
2667 /// rendered using 'dot'.
2668 ///
2669 void ScheduleDAGMI::viewGraph(const Twine &Name, const Twine &Title) {
2670 #ifndef NDEBUG
2671   ViewGraph(this, Name, false, Title);
2672 #else
2673   errs() << "ScheduleDAGMI::viewGraph is only available in debug builds on "
2674          << "systems with Graphviz or gv!\n";
2675 #endif  // NDEBUG
2676 }
2677 
2678 /// Out-of-line implementation with no arguments is handy for gdb.
2679 void ScheduleDAGMI::viewGraph() {
2680   viewGraph(getDAGName(), "Scheduling-Units Graph for " + getDAGName());
2681 }
2682