xref: /llvm-project/llvm/lib/CodeGen/MachineScheduler.cpp (revision c27a18f39fa155583a5a124549137016cb8c7712)
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 #include "llvm/CodeGen/MachineScheduler.h"
16 #include "llvm/ADT/PriorityQueue.h"
17 #include "llvm/Analysis/AliasAnalysis.h"
18 #include "llvm/CodeGen/LiveIntervalAnalysis.h"
19 #include "llvm/CodeGen/MachineDominators.h"
20 #include "llvm/CodeGen/MachineLoopInfo.h"
21 #include "llvm/CodeGen/MachineRegisterInfo.h"
22 #include "llvm/CodeGen/Passes.h"
23 #include "llvm/CodeGen/RegisterClassInfo.h"
24 #include "llvm/CodeGen/ScheduleDFS.h"
25 #include "llvm/CodeGen/ScheduleHazardRecognizer.h"
26 #include "llvm/Support/CommandLine.h"
27 #include "llvm/Support/Debug.h"
28 #include "llvm/Support/ErrorHandling.h"
29 #include "llvm/Support/GraphWriter.h"
30 #include "llvm/Support/raw_ostream.h"
31 #include "llvm/Target/TargetInstrInfo.h"
32 #include <queue>
33 
34 using namespace llvm;
35 
36 #define DEBUG_TYPE "misched"
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 cl::opt<bool>
44 DumpCriticalPathLength("misched-dcpl", cl::Hidden,
45                        cl::desc("Print critical path length to stdout"));
46 }
47 
48 #ifndef NDEBUG
49 static cl::opt<bool> ViewMISchedDAGs("view-misched-dags", cl::Hidden,
50   cl::desc("Pop up a window to show MISched dags after they are processed"));
51 
52 /// In some situations a few uninteresting nodes depend on nearly all other
53 /// nodes in the graph, provide a cutoff to hide them.
54 static cl::opt<unsigned> ViewMISchedCutoff("view-misched-cutoff", cl::Hidden,
55   cl::desc("Hide nodes with more predecessor/successor than cutoff"));
56 
57 static cl::opt<unsigned> MISchedCutoff("misched-cutoff", cl::Hidden,
58   cl::desc("Stop scheduling after N instructions"), cl::init(~0U));
59 
60 static cl::opt<std::string> SchedOnlyFunc("misched-only-func", cl::Hidden,
61   cl::desc("Only schedule this function"));
62 static cl::opt<unsigned> SchedOnlyBlock("misched-only-block", cl::Hidden,
63   cl::desc("Only schedule this MBB#"));
64 #else
65 static bool ViewMISchedDAGs = false;
66 #endif // NDEBUG
67 
68 static cl::opt<bool> EnableRegPressure("misched-regpressure", cl::Hidden,
69   cl::desc("Enable register pressure scheduling."), cl::init(true));
70 
71 static cl::opt<bool> EnableCyclicPath("misched-cyclicpath", cl::Hidden,
72   cl::desc("Enable cyclic critical path analysis."), cl::init(true));
73 
74 static cl::opt<bool> EnableLoadCluster("misched-cluster", cl::Hidden,
75   cl::desc("Enable load clustering."), cl::init(true));
76 
77 // Experimental heuristics
78 static cl::opt<bool> EnableMacroFusion("misched-fusion", cl::Hidden,
79   cl::desc("Enable scheduling for macro fusion."), cl::init(true));
80 
81 static cl::opt<bool> VerifyScheduling("verify-misched", cl::Hidden,
82   cl::desc("Verify machine instrs before and after machine scheduling"));
83 
84 // DAG subtrees must have at least this many nodes.
85 static const unsigned MinSubtreeSize = 8;
86 
87 // Pin the vtables to this file.
88 void MachineSchedStrategy::anchor() {}
89 void ScheduleDAGMutation::anchor() {}
90 
91 //===----------------------------------------------------------------------===//
92 // Machine Instruction Scheduling Pass and Registry
93 //===----------------------------------------------------------------------===//
94 
95 MachineSchedContext::MachineSchedContext():
96     MF(nullptr), MLI(nullptr), MDT(nullptr), PassConfig(nullptr), AA(nullptr), LIS(nullptr) {
97   RegClassInfo = new RegisterClassInfo();
98 }
99 
100 MachineSchedContext::~MachineSchedContext() {
101   delete RegClassInfo;
102 }
103 
104 namespace {
105 /// Base class for a machine scheduler class that can run at any point.
106 class MachineSchedulerBase : public MachineSchedContext,
107                              public MachineFunctionPass {
108 public:
109   MachineSchedulerBase(char &ID): MachineFunctionPass(ID) {}
110 
111   void print(raw_ostream &O, const Module* = nullptr) const override;
112 
113 protected:
114   void scheduleRegions(ScheduleDAGInstrs &Scheduler, bool FixKillFlags);
115 };
116 
117 /// MachineScheduler runs after coalescing and before register allocation.
118 class MachineScheduler : public MachineSchedulerBase {
119 public:
120   MachineScheduler();
121 
122   void getAnalysisUsage(AnalysisUsage &AU) const override;
123 
124   bool runOnMachineFunction(MachineFunction&) override;
125 
126   static char ID; // Class identification, replacement for typeinfo
127 
128 protected:
129   ScheduleDAGInstrs *createMachineScheduler();
130 };
131 
132 /// PostMachineScheduler runs after shortly before code emission.
133 class PostMachineScheduler : public MachineSchedulerBase {
134 public:
135   PostMachineScheduler();
136 
137   void getAnalysisUsage(AnalysisUsage &AU) const override;
138 
139   bool runOnMachineFunction(MachineFunction&) override;
140 
141   static char ID; // Class identification, replacement for typeinfo
142 
143 protected:
144   ScheduleDAGInstrs *createPostMachineScheduler();
145 };
146 } // namespace
147 
148 char MachineScheduler::ID = 0;
149 
150 char &llvm::MachineSchedulerID = MachineScheduler::ID;
151 
152 INITIALIZE_PASS_BEGIN(MachineScheduler, "machine-scheduler",
153                       "Machine Instruction Scheduler", false, false)
154 INITIALIZE_PASS_DEPENDENCY(AAResultsWrapperPass)
155 INITIALIZE_PASS_DEPENDENCY(SlotIndexes)
156 INITIALIZE_PASS_DEPENDENCY(LiveIntervals)
157 INITIALIZE_PASS_END(MachineScheduler, "machine-scheduler",
158                     "Machine Instruction Scheduler", false, false)
159 
160 MachineScheduler::MachineScheduler()
161 : MachineSchedulerBase(ID) {
162   initializeMachineSchedulerPass(*PassRegistry::getPassRegistry());
163 }
164 
165 void MachineScheduler::getAnalysisUsage(AnalysisUsage &AU) const {
166   AU.setPreservesCFG();
167   AU.addRequiredID(MachineDominatorsID);
168   AU.addRequired<MachineLoopInfo>();
169   AU.addRequired<AAResultsWrapperPass>();
170   AU.addRequired<TargetPassConfig>();
171   AU.addRequired<SlotIndexes>();
172   AU.addPreserved<SlotIndexes>();
173   AU.addRequired<LiveIntervals>();
174   AU.addPreserved<LiveIntervals>();
175   MachineFunctionPass::getAnalysisUsage(AU);
176 }
177 
178 char PostMachineScheduler::ID = 0;
179 
180 char &llvm::PostMachineSchedulerID = PostMachineScheduler::ID;
181 
182 INITIALIZE_PASS(PostMachineScheduler, "postmisched",
183                 "PostRA Machine Instruction Scheduler", false, false)
184 
185 PostMachineScheduler::PostMachineScheduler()
186 : MachineSchedulerBase(ID) {
187   initializePostMachineSchedulerPass(*PassRegistry::getPassRegistry());
188 }
189 
190 void PostMachineScheduler::getAnalysisUsage(AnalysisUsage &AU) const {
191   AU.setPreservesCFG();
192   AU.addRequiredID(MachineDominatorsID);
193   AU.addRequired<MachineLoopInfo>();
194   AU.addRequired<TargetPassConfig>();
195   MachineFunctionPass::getAnalysisUsage(AU);
196 }
197 
198 MachinePassRegistry MachineSchedRegistry::Registry;
199 
200 /// A dummy default scheduler factory indicates whether the scheduler
201 /// is overridden on the command line.
202 static ScheduleDAGInstrs *useDefaultMachineSched(MachineSchedContext *C) {
203   return nullptr;
204 }
205 
206 /// MachineSchedOpt allows command line selection of the scheduler.
207 static cl::opt<MachineSchedRegistry::ScheduleDAGCtor, false,
208                RegisterPassParser<MachineSchedRegistry> >
209 MachineSchedOpt("misched",
210                 cl::init(&useDefaultMachineSched), cl::Hidden,
211                 cl::desc("Machine instruction scheduler to use"));
212 
213 static MachineSchedRegistry
214 DefaultSchedRegistry("default", "Use the target's default scheduler choice.",
215                      useDefaultMachineSched);
216 
217 static cl::opt<bool> EnableMachineSched(
218     "enable-misched",
219     cl::desc("Enable the machine instruction scheduling pass."), cl::init(true),
220     cl::Hidden);
221 
222 static cl::opt<bool> EnablePostRAMachineSched(
223     "enable-post-misched",
224     cl::desc("Enable the post-ra machine instruction scheduling pass."),
225     cl::init(true), cl::Hidden);
226 
227 /// Forward declare the standard machine scheduler. This will be used as the
228 /// default scheduler if the target does not set a default.
229 static ScheduleDAGInstrs *createGenericSchedLive(MachineSchedContext *C);
230 static ScheduleDAGInstrs *createGenericSchedPostRA(MachineSchedContext *C);
231 
232 /// Decrement this iterator until reaching the top or a non-debug instr.
233 static MachineBasicBlock::const_iterator
234 priorNonDebug(MachineBasicBlock::const_iterator I,
235               MachineBasicBlock::const_iterator Beg) {
236   assert(I != Beg && "reached the top of the region, cannot decrement");
237   while (--I != Beg) {
238     if (!I->isDebugValue())
239       break;
240   }
241   return I;
242 }
243 
244 /// Non-const version.
245 static MachineBasicBlock::iterator
246 priorNonDebug(MachineBasicBlock::iterator I,
247               MachineBasicBlock::const_iterator Beg) {
248   return const_cast<MachineInstr*>(
249     &*priorNonDebug(MachineBasicBlock::const_iterator(I), Beg));
250 }
251 
252 /// If this iterator is a debug value, increment until reaching the End or a
253 /// non-debug instruction.
254 static MachineBasicBlock::const_iterator
255 nextIfDebug(MachineBasicBlock::const_iterator I,
256             MachineBasicBlock::const_iterator End) {
257   for(; I != End; ++I) {
258     if (!I->isDebugValue())
259       break;
260   }
261   return I;
262 }
263 
264 /// Non-const version.
265 static MachineBasicBlock::iterator
266 nextIfDebug(MachineBasicBlock::iterator I,
267             MachineBasicBlock::const_iterator End) {
268   // Cast the return value to nonconst MachineInstr, then cast to an
269   // instr_iterator, which does not check for null, finally return a
270   // bundle_iterator.
271   return MachineBasicBlock::instr_iterator(
272     const_cast<MachineInstr*>(
273       &*nextIfDebug(MachineBasicBlock::const_iterator(I), End)));
274 }
275 
276 /// Instantiate a ScheduleDAGInstrs that will be owned by the caller.
277 ScheduleDAGInstrs *MachineScheduler::createMachineScheduler() {
278   // Select the scheduler, or set the default.
279   MachineSchedRegistry::ScheduleDAGCtor Ctor = MachineSchedOpt;
280   if (Ctor != useDefaultMachineSched)
281     return Ctor(this);
282 
283   // Get the default scheduler set by the target for this function.
284   ScheduleDAGInstrs *Scheduler = PassConfig->createMachineScheduler(this);
285   if (Scheduler)
286     return Scheduler;
287 
288   // Default to GenericScheduler.
289   return createGenericSchedLive(this);
290 }
291 
292 /// Instantiate a ScheduleDAGInstrs for PostRA scheduling that will be owned by
293 /// the caller. We don't have a command line option to override the postRA
294 /// scheduler. The Target must configure it.
295 ScheduleDAGInstrs *PostMachineScheduler::createPostMachineScheduler() {
296   // Get the postRA scheduler set by the target for this function.
297   ScheduleDAGInstrs *Scheduler = PassConfig->createPostMachineScheduler(this);
298   if (Scheduler)
299     return Scheduler;
300 
301   // Default to GenericScheduler.
302   return createGenericSchedPostRA(this);
303 }
304 
305 /// Top-level MachineScheduler pass driver.
306 ///
307 /// Visit blocks in function order. Divide each block into scheduling regions
308 /// and visit them bottom-up. Visiting regions bottom-up is not required, but is
309 /// consistent with the DAG builder, which traverses the interior of the
310 /// scheduling regions bottom-up.
311 ///
312 /// This design avoids exposing scheduling boundaries to the DAG builder,
313 /// simplifying the DAG builder's support for "special" target instructions.
314 /// At the same time the design allows target schedulers to operate across
315 /// scheduling boundaries, for example to bundle the boudary instructions
316 /// without reordering them. This creates complexity, because the target
317 /// scheduler must update the RegionBegin and RegionEnd positions cached by
318 /// ScheduleDAGInstrs whenever adding or removing instructions. A much simpler
319 /// design would be to split blocks at scheduling boundaries, but LLVM has a
320 /// general bias against block splitting purely for implementation simplicity.
321 bool MachineScheduler::runOnMachineFunction(MachineFunction &mf) {
322   if (skipOptnoneFunction(*mf.getFunction()))
323     return false;
324 
325   if (EnableMachineSched.getNumOccurrences()) {
326     if (!EnableMachineSched)
327       return false;
328   } else if (!mf.getSubtarget().enableMachineScheduler())
329     return false;
330 
331   DEBUG(dbgs() << "Before MISched:\n"; mf.print(dbgs()));
332 
333   // Initialize the context of the pass.
334   MF = &mf;
335   MLI = &getAnalysis<MachineLoopInfo>();
336   MDT = &getAnalysis<MachineDominatorTree>();
337   PassConfig = &getAnalysis<TargetPassConfig>();
338   AA = &getAnalysis<AAResultsWrapperPass>().getAAResults();
339 
340   LIS = &getAnalysis<LiveIntervals>();
341 
342   if (VerifyScheduling) {
343     DEBUG(LIS->dump());
344     MF->verify(this, "Before machine scheduling.");
345   }
346   RegClassInfo->runOnMachineFunction(*MF);
347 
348   // Instantiate the selected scheduler for this target, function, and
349   // optimization level.
350   std::unique_ptr<ScheduleDAGInstrs> Scheduler(createMachineScheduler());
351   scheduleRegions(*Scheduler, false);
352 
353   DEBUG(LIS->dump());
354   if (VerifyScheduling)
355     MF->verify(this, "After machine scheduling.");
356   return true;
357 }
358 
359 bool PostMachineScheduler::runOnMachineFunction(MachineFunction &mf) {
360   if (skipOptnoneFunction(*mf.getFunction()))
361     return false;
362 
363   if (EnablePostRAMachineSched.getNumOccurrences()) {
364     if (!EnablePostRAMachineSched)
365       return false;
366   } else if (!mf.getSubtarget().enablePostRAScheduler()) {
367     DEBUG(dbgs() << "Subtarget disables post-MI-sched.\n");
368     return false;
369   }
370   DEBUG(dbgs() << "Before post-MI-sched:\n"; mf.print(dbgs()));
371 
372   // Initialize the context of the pass.
373   MF = &mf;
374   PassConfig = &getAnalysis<TargetPassConfig>();
375 
376   if (VerifyScheduling)
377     MF->verify(this, "Before post machine scheduling.");
378 
379   // Instantiate the selected scheduler for this target, function, and
380   // optimization level.
381   std::unique_ptr<ScheduleDAGInstrs> Scheduler(createPostMachineScheduler());
382   scheduleRegions(*Scheduler, true);
383 
384   if (VerifyScheduling)
385     MF->verify(this, "After post machine scheduling.");
386   return true;
387 }
388 
389 /// Return true of the given instruction should not be included in a scheduling
390 /// region.
391 ///
392 /// MachineScheduler does not currently support scheduling across calls. To
393 /// handle calls, the DAG builder needs to be modified to create register
394 /// anti/output dependencies on the registers clobbered by the call's regmask
395 /// operand. In PreRA scheduling, the stack pointer adjustment already prevents
396 /// scheduling across calls. In PostRA scheduling, we need the isCall to enforce
397 /// the boundary, but there would be no benefit to postRA scheduling across
398 /// calls this late anyway.
399 static bool isSchedBoundary(MachineBasicBlock::iterator MI,
400                             MachineBasicBlock *MBB,
401                             MachineFunction *MF,
402                             const TargetInstrInfo *TII) {
403   return MI->isCall() || TII->isSchedulingBoundary(MI, MBB, *MF);
404 }
405 
406 /// Main driver for both MachineScheduler and PostMachineScheduler.
407 void MachineSchedulerBase::scheduleRegions(ScheduleDAGInstrs &Scheduler,
408                                            bool FixKillFlags) {
409   const TargetInstrInfo *TII = MF->getSubtarget().getInstrInfo();
410 
411   // Visit all machine basic blocks.
412   //
413   // TODO: Visit blocks in global postorder or postorder within the bottom-up
414   // loop tree. Then we can optionally compute global RegPressure.
415   for (MachineFunction::iterator MBB = MF->begin(), MBBEnd = MF->end();
416        MBB != MBBEnd; ++MBB) {
417 
418     Scheduler.startBlock(&*MBB);
419 
420 #ifndef NDEBUG
421     if (SchedOnlyFunc.getNumOccurrences() && SchedOnlyFunc != MF->getName())
422       continue;
423     if (SchedOnlyBlock.getNumOccurrences()
424         && (int)SchedOnlyBlock != MBB->getNumber())
425       continue;
426 #endif
427 
428     // Break the block into scheduling regions [I, RegionEnd), and schedule each
429     // region as soon as it is discovered. RegionEnd points the scheduling
430     // boundary at the bottom of the region. The DAG does not include RegionEnd,
431     // but the region does (i.e. the next RegionEnd is above the previous
432     // RegionBegin). If the current block has no terminator then RegionEnd ==
433     // MBB->end() for the bottom region.
434     //
435     // The Scheduler may insert instructions during either schedule() or
436     // exitRegion(), even for empty regions. So the local iterators 'I' and
437     // 'RegionEnd' are invalid across these calls.
438     //
439     // MBB::size() uses instr_iterator to count. Here we need a bundle to count
440     // as a single instruction.
441     unsigned RemainingInstrs = std::distance(MBB->begin(), MBB->end());
442     for(MachineBasicBlock::iterator RegionEnd = MBB->end();
443         RegionEnd != MBB->begin(); RegionEnd = Scheduler.begin()) {
444 
445       // Avoid decrementing RegionEnd for blocks with no terminator.
446       if (RegionEnd != MBB->end() ||
447           isSchedBoundary(&*std::prev(RegionEnd), &*MBB, MF, TII)) {
448         --RegionEnd;
449         // Count the boundary instruction.
450         --RemainingInstrs;
451       }
452 
453       // The next region starts above the previous region. Look backward in the
454       // instruction stream until we find the nearest boundary.
455       unsigned NumRegionInstrs = 0;
456       MachineBasicBlock::iterator I = RegionEnd;
457       for(;I != MBB->begin(); --I, --RemainingInstrs) {
458         if (isSchedBoundary(&*std::prev(I), &*MBB, MF, TII))
459           break;
460         if (!I->isDebugValue())
461           ++NumRegionInstrs;
462       }
463       // Notify the scheduler of the region, even if we may skip scheduling
464       // it. Perhaps it still needs to be bundled.
465       Scheduler.enterRegion(&*MBB, I, RegionEnd, NumRegionInstrs);
466 
467       // Skip empty scheduling regions (0 or 1 schedulable instructions).
468       if (I == RegionEnd || I == std::prev(RegionEnd)) {
469         // Close the current region. Bundle the terminator if needed.
470         // This invalidates 'RegionEnd' and 'I'.
471         Scheduler.exitRegion();
472         continue;
473       }
474       DEBUG(dbgs() << "********** MI Scheduling **********\n");
475       DEBUG(dbgs() << MF->getName()
476             << ":BB#" << MBB->getNumber() << " " << MBB->getName()
477             << "\n  From: " << *I << "    To: ";
478             if (RegionEnd != MBB->end()) dbgs() << *RegionEnd;
479             else dbgs() << "End";
480             dbgs() << " RegionInstrs: " << NumRegionInstrs
481             << " Remaining: " << RemainingInstrs << "\n");
482       if (DumpCriticalPathLength) {
483         errs() << MF->getName();
484         errs() << ":BB# " << MBB->getNumber();
485         errs() << " " << MBB->getName() << " \n";
486       }
487 
488       // Schedule a region: possibly reorder instructions.
489       // This invalidates 'RegionEnd' and 'I'.
490       Scheduler.schedule();
491 
492       // Close the current region.
493       Scheduler.exitRegion();
494 
495       // Scheduling has invalidated the current iterator 'I'. Ask the
496       // scheduler for the top of it's scheduled region.
497       RegionEnd = Scheduler.begin();
498     }
499     assert(RemainingInstrs == 0 && "Instruction count mismatch!");
500     Scheduler.finishBlock();
501     // FIXME: Ideally, no further passes should rely on kill flags. However,
502     // thumb2 size reduction is currently an exception, so the PostMIScheduler
503     // needs to do this.
504     if (FixKillFlags)
505         Scheduler.fixupKills(&*MBB);
506   }
507   Scheduler.finalizeSchedule();
508 }
509 
510 void MachineSchedulerBase::print(raw_ostream &O, const Module* m) const {
511   // unimplemented
512 }
513 
514 LLVM_DUMP_METHOD
515 void ReadyQueue::dump() {
516   dbgs() << "Queue " << Name << ": ";
517   for (unsigned i = 0, e = Queue.size(); i < e; ++i)
518     dbgs() << Queue[i]->NodeNum << " ";
519   dbgs() << "\n";
520 }
521 
522 //===----------------------------------------------------------------------===//
523 // ScheduleDAGMI - Basic machine instruction scheduling. This is
524 // independent of PreRA/PostRA scheduling and involves no extra book-keeping for
525 // virtual registers.
526 // ===----------------------------------------------------------------------===/
527 
528 // Provide a vtable anchor.
529 ScheduleDAGMI::~ScheduleDAGMI() {
530 }
531 
532 bool ScheduleDAGMI::canAddEdge(SUnit *SuccSU, SUnit *PredSU) {
533   return SuccSU == &ExitSU || !Topo.IsReachable(PredSU, SuccSU);
534 }
535 
536 bool ScheduleDAGMI::addEdge(SUnit *SuccSU, const SDep &PredDep) {
537   if (SuccSU != &ExitSU) {
538     // Do not use WillCreateCycle, it assumes SD scheduling.
539     // If Pred is reachable from Succ, then the edge creates a cycle.
540     if (Topo.IsReachable(PredDep.getSUnit(), SuccSU))
541       return false;
542     Topo.AddPred(SuccSU, PredDep.getSUnit());
543   }
544   SuccSU->addPred(PredDep, /*Required=*/!PredDep.isArtificial());
545   // Return true regardless of whether a new edge needed to be inserted.
546   return true;
547 }
548 
549 /// ReleaseSucc - Decrement the NumPredsLeft count of a successor. When
550 /// NumPredsLeft reaches zero, release the successor node.
551 ///
552 /// FIXME: Adjust SuccSU height based on MinLatency.
553 void ScheduleDAGMI::releaseSucc(SUnit *SU, SDep *SuccEdge) {
554   SUnit *SuccSU = SuccEdge->getSUnit();
555 
556   if (SuccEdge->isWeak()) {
557     --SuccSU->WeakPredsLeft;
558     if (SuccEdge->isCluster())
559       NextClusterSucc = SuccSU;
560     return;
561   }
562 #ifndef NDEBUG
563   if (SuccSU->NumPredsLeft == 0) {
564     dbgs() << "*** Scheduling failed! ***\n";
565     SuccSU->dump(this);
566     dbgs() << " has been released too many times!\n";
567     llvm_unreachable(nullptr);
568   }
569 #endif
570   // SU->TopReadyCycle was set to CurrCycle when it was scheduled. However,
571   // CurrCycle may have advanced since then.
572   if (SuccSU->TopReadyCycle < SU->TopReadyCycle + SuccEdge->getLatency())
573     SuccSU->TopReadyCycle = SU->TopReadyCycle + SuccEdge->getLatency();
574 
575   --SuccSU->NumPredsLeft;
576   if (SuccSU->NumPredsLeft == 0 && SuccSU != &ExitSU)
577     SchedImpl->releaseTopNode(SuccSU);
578 }
579 
580 /// releaseSuccessors - Call releaseSucc on each of SU's successors.
581 void ScheduleDAGMI::releaseSuccessors(SUnit *SU) {
582   for (SUnit::succ_iterator I = SU->Succs.begin(), E = SU->Succs.end();
583        I != E; ++I) {
584     releaseSucc(SU, &*I);
585   }
586 }
587 
588 /// ReleasePred - Decrement the NumSuccsLeft count of a predecessor. When
589 /// NumSuccsLeft reaches zero, release the predecessor node.
590 ///
591 /// FIXME: Adjust PredSU height based on MinLatency.
592 void ScheduleDAGMI::releasePred(SUnit *SU, SDep *PredEdge) {
593   SUnit *PredSU = PredEdge->getSUnit();
594 
595   if (PredEdge->isWeak()) {
596     --PredSU->WeakSuccsLeft;
597     if (PredEdge->isCluster())
598       NextClusterPred = PredSU;
599     return;
600   }
601 #ifndef NDEBUG
602   if (PredSU->NumSuccsLeft == 0) {
603     dbgs() << "*** Scheduling failed! ***\n";
604     PredSU->dump(this);
605     dbgs() << " has been released too many times!\n";
606     llvm_unreachable(nullptr);
607   }
608 #endif
609   // SU->BotReadyCycle was set to CurrCycle when it was scheduled. However,
610   // CurrCycle may have advanced since then.
611   if (PredSU->BotReadyCycle < SU->BotReadyCycle + PredEdge->getLatency())
612     PredSU->BotReadyCycle = SU->BotReadyCycle + PredEdge->getLatency();
613 
614   --PredSU->NumSuccsLeft;
615   if (PredSU->NumSuccsLeft == 0 && PredSU != &EntrySU)
616     SchedImpl->releaseBottomNode(PredSU);
617 }
618 
619 /// releasePredecessors - Call releasePred on each of SU's predecessors.
620 void ScheduleDAGMI::releasePredecessors(SUnit *SU) {
621   for (SUnit::pred_iterator I = SU->Preds.begin(), E = SU->Preds.end();
622        I != E; ++I) {
623     releasePred(SU, &*I);
624   }
625 }
626 
627 /// enterRegion - Called back from MachineScheduler::runOnMachineFunction after
628 /// crossing a scheduling boundary. [begin, end) includes all instructions in
629 /// the region, including the boundary itself and single-instruction regions
630 /// that don't get scheduled.
631 void ScheduleDAGMI::enterRegion(MachineBasicBlock *bb,
632                                      MachineBasicBlock::iterator begin,
633                                      MachineBasicBlock::iterator end,
634                                      unsigned regioninstrs)
635 {
636   ScheduleDAGInstrs::enterRegion(bb, begin, end, regioninstrs);
637 
638   SchedImpl->initPolicy(begin, end, regioninstrs);
639 }
640 
641 /// This is normally called from the main scheduler loop but may also be invoked
642 /// by the scheduling strategy to perform additional code motion.
643 void ScheduleDAGMI::moveInstruction(
644   MachineInstr *MI, MachineBasicBlock::iterator InsertPos) {
645   // Advance RegionBegin if the first instruction moves down.
646   if (&*RegionBegin == MI)
647     ++RegionBegin;
648 
649   // Update the instruction stream.
650   BB->splice(InsertPos, BB, MI);
651 
652   // Update LiveIntervals
653   if (LIS)
654     LIS->handleMove(*MI, /*UpdateFlags=*/true);
655 
656   // Recede RegionBegin if an instruction moves above the first.
657   if (RegionBegin == InsertPos)
658     RegionBegin = MI;
659 }
660 
661 bool ScheduleDAGMI::checkSchedLimit() {
662 #ifndef NDEBUG
663   if (NumInstrsScheduled == MISchedCutoff && MISchedCutoff != ~0U) {
664     CurrentTop = CurrentBottom;
665     return false;
666   }
667   ++NumInstrsScheduled;
668 #endif
669   return true;
670 }
671 
672 /// Per-region scheduling driver, called back from
673 /// MachineScheduler::runOnMachineFunction. This is a simplified driver that
674 /// does not consider liveness or register pressure. It is useful for PostRA
675 /// scheduling and potentially other custom schedulers.
676 void ScheduleDAGMI::schedule() {
677   DEBUG(dbgs() << "ScheduleDAGMI::schedule starting\n");
678   DEBUG(SchedImpl->dumpPolicy());
679 
680   // Build the DAG.
681   buildSchedGraph(AA);
682 
683   Topo.InitDAGTopologicalSorting();
684 
685   postprocessDAG();
686 
687   SmallVector<SUnit*, 8> TopRoots, BotRoots;
688   findRootsAndBiasEdges(TopRoots, BotRoots);
689 
690   // Initialize the strategy before modifying the DAG.
691   // This may initialize a DFSResult to be used for queue priority.
692   SchedImpl->initialize(this);
693 
694   DEBUG(for (unsigned su = 0, e = SUnits.size(); su != e; ++su)
695           SUnits[su].dumpAll(this));
696   if (ViewMISchedDAGs) viewGraph();
697 
698   // Initialize ready queues now that the DAG and priority data are finalized.
699   initQueues(TopRoots, BotRoots);
700 
701   bool IsTopNode = false;
702   while (true) {
703     DEBUG(dbgs() << "** ScheduleDAGMI::schedule picking next node\n");
704     SUnit *SU = SchedImpl->pickNode(IsTopNode);
705     if (!SU) break;
706 
707     assert(!SU->isScheduled && "Node already scheduled");
708     if (!checkSchedLimit())
709       break;
710 
711     MachineInstr *MI = SU->getInstr();
712     if (IsTopNode) {
713       assert(SU->isTopReady() && "node still has unscheduled dependencies");
714       if (&*CurrentTop == MI)
715         CurrentTop = nextIfDebug(++CurrentTop, CurrentBottom);
716       else
717         moveInstruction(MI, CurrentTop);
718     }
719     else {
720       assert(SU->isBottomReady() && "node still has unscheduled dependencies");
721       MachineBasicBlock::iterator priorII =
722         priorNonDebug(CurrentBottom, CurrentTop);
723       if (&*priorII == MI)
724         CurrentBottom = priorII;
725       else {
726         if (&*CurrentTop == MI)
727           CurrentTop = nextIfDebug(++CurrentTop, priorII);
728         moveInstruction(MI, CurrentBottom);
729         CurrentBottom = MI;
730       }
731     }
732     // Notify the scheduling strategy before updating the DAG.
733     // This sets the scheduled node's ReadyCycle to CurrCycle. When updateQueues
734     // runs, it can then use the accurate ReadyCycle time to determine whether
735     // newly released nodes can move to the readyQ.
736     SchedImpl->schedNode(SU, IsTopNode);
737 
738     updateQueues(SU, IsTopNode);
739   }
740   assert(CurrentTop == CurrentBottom && "Nonempty unscheduled zone.");
741 
742   placeDebugValues();
743 
744   DEBUG({
745       unsigned BBNum = begin()->getParent()->getNumber();
746       dbgs() << "*** Final schedule for BB#" << BBNum << " ***\n";
747       dumpSchedule();
748       dbgs() << '\n';
749     });
750 }
751 
752 /// Apply each ScheduleDAGMutation step in order.
753 void ScheduleDAGMI::postprocessDAG() {
754   for (unsigned i = 0, e = Mutations.size(); i < e; ++i) {
755     Mutations[i]->apply(this);
756   }
757 }
758 
759 void ScheduleDAGMI::
760 findRootsAndBiasEdges(SmallVectorImpl<SUnit*> &TopRoots,
761                       SmallVectorImpl<SUnit*> &BotRoots) {
762   for (std::vector<SUnit>::iterator
763          I = SUnits.begin(), E = SUnits.end(); I != E; ++I) {
764     SUnit *SU = &(*I);
765     assert(!SU->isBoundaryNode() && "Boundary node should not be in SUnits");
766 
767     // Order predecessors so DFSResult follows the critical path.
768     SU->biasCriticalPath();
769 
770     // A SUnit is ready to top schedule if it has no predecessors.
771     if (!I->NumPredsLeft)
772       TopRoots.push_back(SU);
773     // A SUnit is ready to bottom schedule if it has no successors.
774     if (!I->NumSuccsLeft)
775       BotRoots.push_back(SU);
776   }
777   ExitSU.biasCriticalPath();
778 }
779 
780 /// Identify DAG roots and setup scheduler queues.
781 void ScheduleDAGMI::initQueues(ArrayRef<SUnit*> TopRoots,
782                                ArrayRef<SUnit*> BotRoots) {
783   NextClusterSucc = nullptr;
784   NextClusterPred = nullptr;
785 
786   // Release all DAG roots for scheduling, not including EntrySU/ExitSU.
787   //
788   // Nodes with unreleased weak edges can still be roots.
789   // Release top roots in forward order.
790   for (SmallVectorImpl<SUnit*>::const_iterator
791          I = TopRoots.begin(), E = TopRoots.end(); I != E; ++I) {
792     SchedImpl->releaseTopNode(*I);
793   }
794   // Release bottom roots in reverse order so the higher priority nodes appear
795   // first. This is more natural and slightly more efficient.
796   for (SmallVectorImpl<SUnit*>::const_reverse_iterator
797          I = BotRoots.rbegin(), E = BotRoots.rend(); I != E; ++I) {
798     SchedImpl->releaseBottomNode(*I);
799   }
800 
801   releaseSuccessors(&EntrySU);
802   releasePredecessors(&ExitSU);
803 
804   SchedImpl->registerRoots();
805 
806   // Advance past initial DebugValues.
807   CurrentTop = nextIfDebug(RegionBegin, RegionEnd);
808   CurrentBottom = RegionEnd;
809 }
810 
811 /// Update scheduler queues after scheduling an instruction.
812 void ScheduleDAGMI::updateQueues(SUnit *SU, bool IsTopNode) {
813   // Release dependent instructions for scheduling.
814   if (IsTopNode)
815     releaseSuccessors(SU);
816   else
817     releasePredecessors(SU);
818 
819   SU->isScheduled = true;
820 }
821 
822 /// Reinsert any remaining debug_values, just like the PostRA scheduler.
823 void ScheduleDAGMI::placeDebugValues() {
824   // If first instruction was a DBG_VALUE then put it back.
825   if (FirstDbgValue) {
826     BB->splice(RegionBegin, BB, FirstDbgValue);
827     RegionBegin = FirstDbgValue;
828   }
829 
830   for (std::vector<std::pair<MachineInstr *, MachineInstr *> >::iterator
831          DI = DbgValues.end(), DE = DbgValues.begin(); DI != DE; --DI) {
832     std::pair<MachineInstr *, MachineInstr *> P = *std::prev(DI);
833     MachineInstr *DbgValue = P.first;
834     MachineBasicBlock::iterator OrigPrevMI = P.second;
835     if (&*RegionBegin == DbgValue)
836       ++RegionBegin;
837     BB->splice(++OrigPrevMI, BB, DbgValue);
838     if (OrigPrevMI == std::prev(RegionEnd))
839       RegionEnd = DbgValue;
840   }
841   DbgValues.clear();
842   FirstDbgValue = nullptr;
843 }
844 
845 #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
846 void ScheduleDAGMI::dumpSchedule() const {
847   for (MachineBasicBlock::iterator MI = begin(), ME = end(); MI != ME; ++MI) {
848     if (SUnit *SU = getSUnit(&(*MI)))
849       SU->dump(this);
850     else
851       dbgs() << "Missing SUnit\n";
852   }
853 }
854 #endif
855 
856 //===----------------------------------------------------------------------===//
857 // ScheduleDAGMILive - Base class for MachineInstr scheduling with LiveIntervals
858 // preservation.
859 //===----------------------------------------------------------------------===//
860 
861 ScheduleDAGMILive::~ScheduleDAGMILive() {
862   delete DFSResult;
863 }
864 
865 /// enterRegion - Called back from MachineScheduler::runOnMachineFunction after
866 /// crossing a scheduling boundary. [begin, end) includes all instructions in
867 /// the region, including the boundary itself and single-instruction regions
868 /// that don't get scheduled.
869 void ScheduleDAGMILive::enterRegion(MachineBasicBlock *bb,
870                                 MachineBasicBlock::iterator begin,
871                                 MachineBasicBlock::iterator end,
872                                 unsigned regioninstrs)
873 {
874   // ScheduleDAGMI initializes SchedImpl's per-region policy.
875   ScheduleDAGMI::enterRegion(bb, begin, end, regioninstrs);
876 
877   // For convenience remember the end of the liveness region.
878   LiveRegionEnd = (RegionEnd == bb->end()) ? RegionEnd : std::next(RegionEnd);
879 
880   SUPressureDiffs.clear();
881 
882   ShouldTrackPressure = SchedImpl->shouldTrackPressure();
883   ShouldTrackLaneMasks = SchedImpl->shouldTrackLaneMasks();
884 
885   if (ShouldTrackLaneMasks) {
886     if (!ShouldTrackPressure)
887       report_fatal_error("ShouldTrackLaneMasks requires ShouldTrackPressure");
888     // Dead subregister defs have no users and therefore no dependencies,
889     // moving them around may cause liveintervals to degrade into multiple
890     // components. Change independent components to have their own vreg to avoid
891     // this.
892     if (!DisconnectedComponentsRenamed)
893       LIS->renameDisconnectedComponents();
894   }
895 }
896 
897 // Setup the register pressure trackers for the top scheduled top and bottom
898 // scheduled regions.
899 void ScheduleDAGMILive::initRegPressure() {
900   TopRPTracker.init(&MF, RegClassInfo, LIS, BB, RegionBegin,
901                     ShouldTrackLaneMasks, false);
902   BotRPTracker.init(&MF, RegClassInfo, LIS, BB, LiveRegionEnd,
903                     ShouldTrackLaneMasks, false);
904 
905   // Close the RPTracker to finalize live ins.
906   RPTracker.closeRegion();
907 
908   DEBUG(RPTracker.dump());
909 
910   // Initialize the live ins and live outs.
911   TopRPTracker.addLiveRegs(RPTracker.getPressure().LiveInRegs);
912   BotRPTracker.addLiveRegs(RPTracker.getPressure().LiveOutRegs);
913 
914   // Close one end of the tracker so we can call
915   // getMaxUpward/DownwardPressureDelta before advancing across any
916   // instructions. This converts currently live regs into live ins/outs.
917   TopRPTracker.closeTop();
918   BotRPTracker.closeBottom();
919 
920   BotRPTracker.initLiveThru(RPTracker);
921   if (!BotRPTracker.getLiveThru().empty()) {
922     TopRPTracker.initLiveThru(BotRPTracker.getLiveThru());
923     DEBUG(dbgs() << "Live Thru: ";
924           dumpRegSetPressure(BotRPTracker.getLiveThru(), TRI));
925   };
926 
927   // For each live out vreg reduce the pressure change associated with other
928   // uses of the same vreg below the live-out reaching def.
929   updatePressureDiffs(RPTracker.getPressure().LiveOutRegs);
930 
931   // Account for liveness generated by the region boundary.
932   if (LiveRegionEnd != RegionEnd) {
933     SmallVector<RegisterMaskPair, 8> LiveUses;
934     BotRPTracker.recede(&LiveUses);
935     updatePressureDiffs(LiveUses);
936   }
937 
938   DEBUG(
939     dbgs() << "Top Pressure:\n";
940     dumpRegSetPressure(TopRPTracker.getRegSetPressureAtPos(), TRI);
941     dbgs() << "Bottom Pressure:\n";
942     dumpRegSetPressure(BotRPTracker.getRegSetPressureAtPos(), TRI);
943   );
944 
945   assert(BotRPTracker.getPos() == RegionEnd && "Can't find the region bottom");
946 
947   // Cache the list of excess pressure sets in this region. This will also track
948   // the max pressure in the scheduled code for these sets.
949   RegionCriticalPSets.clear();
950   const std::vector<unsigned> &RegionPressure =
951     RPTracker.getPressure().MaxSetPressure;
952   for (unsigned i = 0, e = RegionPressure.size(); i < e; ++i) {
953     unsigned Limit = RegClassInfo->getRegPressureSetLimit(i);
954     if (RegionPressure[i] > Limit) {
955       DEBUG(dbgs() << TRI->getRegPressureSetName(i)
956             << " Limit " << Limit
957             << " Actual " << RegionPressure[i] << "\n");
958       RegionCriticalPSets.push_back(PressureChange(i));
959     }
960   }
961   DEBUG(dbgs() << "Excess PSets: ";
962         for (unsigned i = 0, e = RegionCriticalPSets.size(); i != e; ++i)
963           dbgs() << TRI->getRegPressureSetName(
964             RegionCriticalPSets[i].getPSet()) << " ";
965         dbgs() << "\n");
966 }
967 
968 void ScheduleDAGMILive::
969 updateScheduledPressure(const SUnit *SU,
970                         const std::vector<unsigned> &NewMaxPressure) {
971   const PressureDiff &PDiff = getPressureDiff(SU);
972   unsigned CritIdx = 0, CritEnd = RegionCriticalPSets.size();
973   for (PressureDiff::const_iterator I = PDiff.begin(), E = PDiff.end();
974        I != E; ++I) {
975     if (!I->isValid())
976       break;
977     unsigned ID = I->getPSet();
978     while (CritIdx != CritEnd && RegionCriticalPSets[CritIdx].getPSet() < ID)
979       ++CritIdx;
980     if (CritIdx != CritEnd && RegionCriticalPSets[CritIdx].getPSet() == ID) {
981       if ((int)NewMaxPressure[ID] > RegionCriticalPSets[CritIdx].getUnitInc()
982           && NewMaxPressure[ID] <= INT16_MAX)
983         RegionCriticalPSets[CritIdx].setUnitInc(NewMaxPressure[ID]);
984     }
985     unsigned Limit = RegClassInfo->getRegPressureSetLimit(ID);
986     if (NewMaxPressure[ID] >= Limit - 2) {
987       DEBUG(dbgs() << "  " << TRI->getRegPressureSetName(ID) << ": "
988             << NewMaxPressure[ID]
989             << ((NewMaxPressure[ID] > Limit) ? " > " : " <= ") << Limit
990             << "(+ " << BotRPTracker.getLiveThru()[ID] << " livethru)\n");
991     }
992   }
993 }
994 
995 /// Update the PressureDiff array for liveness after scheduling this
996 /// instruction.
997 void ScheduleDAGMILive::updatePressureDiffs(
998     ArrayRef<RegisterMaskPair> LiveUses) {
999   for (const RegisterMaskPair &P : LiveUses) {
1000     unsigned Reg = P.RegUnit;
1001     /// FIXME: Currently assuming single-use physregs.
1002     if (!TRI->isVirtualRegister(Reg))
1003       continue;
1004 
1005     if (ShouldTrackLaneMasks) {
1006       // If the register has just become live then other uses won't change
1007       // this fact anymore => decrement pressure.
1008       // If the register has just become dead then other uses make it come
1009       // back to life => increment pressure.
1010       bool Decrement = P.LaneMask != 0;
1011 
1012       for (const VReg2SUnit &V2SU
1013            : make_range(VRegUses.find(Reg), VRegUses.end())) {
1014         SUnit &SU = *V2SU.SU;
1015         if (SU.isScheduled || &SU == &ExitSU)
1016           continue;
1017 
1018         PressureDiff &PDiff = getPressureDiff(&SU);
1019         PDiff.addPressureChange(Reg, Decrement, &MRI);
1020         DEBUG(
1021           dbgs() << "  UpdateRegP: SU(" << SU.NodeNum << ") "
1022                  << PrintReg(Reg, TRI) << ':' << PrintLaneMask(P.LaneMask)
1023                  << ' ' << *SU.getInstr();
1024           dbgs() << "              to ";
1025           PDiff.dump(*TRI);
1026         );
1027       }
1028     } else {
1029       assert(P.LaneMask != 0);
1030       DEBUG(dbgs() << "  LiveReg: " << PrintVRegOrUnit(Reg, TRI) << "\n");
1031       // This may be called before CurrentBottom has been initialized. However,
1032       // BotRPTracker must have a valid position. We want the value live into the
1033       // instruction or live out of the block, so ask for the previous
1034       // instruction's live-out.
1035       const LiveInterval &LI = LIS->getInterval(Reg);
1036       VNInfo *VNI;
1037       MachineBasicBlock::const_iterator I =
1038         nextIfDebug(BotRPTracker.getPos(), BB->end());
1039       if (I == BB->end())
1040         VNI = LI.getVNInfoBefore(LIS->getMBBEndIdx(BB));
1041       else {
1042         LiveQueryResult LRQ = LI.Query(LIS->getInstructionIndex(*I));
1043         VNI = LRQ.valueIn();
1044       }
1045       // RegisterPressureTracker guarantees that readsReg is true for LiveUses.
1046       assert(VNI && "No live value at use.");
1047       for (const VReg2SUnit &V2SU
1048            : make_range(VRegUses.find(Reg), VRegUses.end())) {
1049         SUnit *SU = V2SU.SU;
1050         // If this use comes before the reaching def, it cannot be a last use,
1051         // so decrease its pressure change.
1052         if (!SU->isScheduled && SU != &ExitSU) {
1053           LiveQueryResult LRQ =
1054               LI.Query(LIS->getInstructionIndex(*SU->getInstr()));
1055           if (LRQ.valueIn() == VNI) {
1056             PressureDiff &PDiff = getPressureDiff(SU);
1057             PDiff.addPressureChange(Reg, true, &MRI);
1058             DEBUG(
1059               dbgs() << "  UpdateRegP: SU(" << SU->NodeNum << ") "
1060                      << *SU->getInstr();
1061               dbgs() << "              to ";
1062               PDiff.dump(*TRI);
1063             );
1064           }
1065         }
1066       }
1067     }
1068   }
1069 }
1070 
1071 /// schedule - Called back from MachineScheduler::runOnMachineFunction
1072 /// after setting up the current scheduling region. [RegionBegin, RegionEnd)
1073 /// only includes instructions that have DAG nodes, not scheduling boundaries.
1074 ///
1075 /// This is a skeletal driver, with all the functionality pushed into helpers,
1076 /// so that it can be easily extended by experimental schedulers. Generally,
1077 /// implementing MachineSchedStrategy should be sufficient to implement a new
1078 /// scheduling algorithm. However, if a scheduler further subclasses
1079 /// ScheduleDAGMILive then it will want to override this virtual method in order
1080 /// to update any specialized state.
1081 void ScheduleDAGMILive::schedule() {
1082   DEBUG(dbgs() << "ScheduleDAGMILive::schedule starting\n");
1083   DEBUG(SchedImpl->dumpPolicy());
1084   buildDAGWithRegPressure();
1085 
1086   Topo.InitDAGTopologicalSorting();
1087 
1088   postprocessDAG();
1089 
1090   SmallVector<SUnit*, 8> TopRoots, BotRoots;
1091   findRootsAndBiasEdges(TopRoots, BotRoots);
1092 
1093   // Initialize the strategy before modifying the DAG.
1094   // This may initialize a DFSResult to be used for queue priority.
1095   SchedImpl->initialize(this);
1096 
1097   DEBUG(
1098     for (const SUnit &SU : SUnits) {
1099       SU.dumpAll(this);
1100       if (ShouldTrackPressure) {
1101         dbgs() << "  Pressure Diff      : ";
1102         getPressureDiff(&SU).dump(*TRI);
1103       }
1104       dbgs() << '\n';
1105     }
1106   );
1107   if (ViewMISchedDAGs) viewGraph();
1108 
1109   // Initialize ready queues now that the DAG and priority data are finalized.
1110   initQueues(TopRoots, BotRoots);
1111 
1112   if (ShouldTrackPressure) {
1113     assert(TopRPTracker.getPos() == RegionBegin && "bad initial Top tracker");
1114     TopRPTracker.setPos(CurrentTop);
1115   }
1116 
1117   bool IsTopNode = false;
1118   while (true) {
1119     DEBUG(dbgs() << "** ScheduleDAGMILive::schedule picking next node\n");
1120     SUnit *SU = SchedImpl->pickNode(IsTopNode);
1121     if (!SU) break;
1122 
1123     assert(!SU->isScheduled && "Node already scheduled");
1124     if (!checkSchedLimit())
1125       break;
1126 
1127     scheduleMI(SU, IsTopNode);
1128 
1129     if (DFSResult) {
1130       unsigned SubtreeID = DFSResult->getSubtreeID(SU);
1131       if (!ScheduledTrees.test(SubtreeID)) {
1132         ScheduledTrees.set(SubtreeID);
1133         DFSResult->scheduleTree(SubtreeID);
1134         SchedImpl->scheduleTree(SubtreeID);
1135       }
1136     }
1137 
1138     // Notify the scheduling strategy after updating the DAG.
1139     SchedImpl->schedNode(SU, IsTopNode);
1140 
1141     updateQueues(SU, IsTopNode);
1142   }
1143   assert(CurrentTop == CurrentBottom && "Nonempty unscheduled zone.");
1144 
1145   placeDebugValues();
1146 
1147   DEBUG({
1148       unsigned BBNum = begin()->getParent()->getNumber();
1149       dbgs() << "*** Final schedule for BB#" << BBNum << " ***\n";
1150       dumpSchedule();
1151       dbgs() << '\n';
1152     });
1153 }
1154 
1155 /// Build the DAG and setup three register pressure trackers.
1156 void ScheduleDAGMILive::buildDAGWithRegPressure() {
1157   if (!ShouldTrackPressure) {
1158     RPTracker.reset();
1159     RegionCriticalPSets.clear();
1160     buildSchedGraph(AA);
1161     return;
1162   }
1163 
1164   // Initialize the register pressure tracker used by buildSchedGraph.
1165   RPTracker.init(&MF, RegClassInfo, LIS, BB, LiveRegionEnd,
1166                  ShouldTrackLaneMasks, /*TrackUntiedDefs=*/true);
1167 
1168   // Account for liveness generate by the region boundary.
1169   if (LiveRegionEnd != RegionEnd)
1170     RPTracker.recede();
1171 
1172   // Build the DAG, and compute current register pressure.
1173   buildSchedGraph(AA, &RPTracker, &SUPressureDiffs, LIS, ShouldTrackLaneMasks);
1174 
1175   // Initialize top/bottom trackers after computing region pressure.
1176   initRegPressure();
1177 }
1178 
1179 void ScheduleDAGMILive::computeDFSResult() {
1180   if (!DFSResult)
1181     DFSResult = new SchedDFSResult(/*BottomU*/true, MinSubtreeSize);
1182   DFSResult->clear();
1183   ScheduledTrees.clear();
1184   DFSResult->resize(SUnits.size());
1185   DFSResult->compute(SUnits);
1186   ScheduledTrees.resize(DFSResult->getNumSubtrees());
1187 }
1188 
1189 /// Compute the max cyclic critical path through the DAG. The scheduling DAG
1190 /// only provides the critical path for single block loops. To handle loops that
1191 /// span blocks, we could use the vreg path latencies provided by
1192 /// MachineTraceMetrics instead. However, MachineTraceMetrics is not currently
1193 /// available for use in the scheduler.
1194 ///
1195 /// The cyclic path estimation identifies a def-use pair that crosses the back
1196 /// edge and considers the depth and height of the nodes. For example, consider
1197 /// the following instruction sequence where each instruction has unit latency
1198 /// and defines an epomymous virtual register:
1199 ///
1200 /// a->b(a,c)->c(b)->d(c)->exit
1201 ///
1202 /// The cyclic critical path is a two cycles: b->c->b
1203 /// The acyclic critical path is four cycles: a->b->c->d->exit
1204 /// LiveOutHeight = height(c) = len(c->d->exit) = 2
1205 /// LiveOutDepth = depth(c) + 1 = len(a->b->c) + 1 = 3
1206 /// LiveInHeight = height(b) + 1 = len(b->c->d->exit) + 1 = 4
1207 /// LiveInDepth = depth(b) = len(a->b) = 1
1208 ///
1209 /// LiveOutDepth - LiveInDepth = 3 - 1 = 2
1210 /// LiveInHeight - LiveOutHeight = 4 - 2 = 2
1211 /// CyclicCriticalPath = min(2, 2) = 2
1212 ///
1213 /// This could be relevant to PostRA scheduling, but is currently implemented
1214 /// assuming LiveIntervals.
1215 unsigned ScheduleDAGMILive::computeCyclicCriticalPath() {
1216   // This only applies to single block loop.
1217   if (!BB->isSuccessor(BB))
1218     return 0;
1219 
1220   unsigned MaxCyclicLatency = 0;
1221   // Visit each live out vreg def to find def/use pairs that cross iterations.
1222   for (const RegisterMaskPair &P : RPTracker.getPressure().LiveOutRegs) {
1223     unsigned Reg = P.RegUnit;
1224     if (!TRI->isVirtualRegister(Reg))
1225         continue;
1226     const LiveInterval &LI = LIS->getInterval(Reg);
1227     const VNInfo *DefVNI = LI.getVNInfoBefore(LIS->getMBBEndIdx(BB));
1228     if (!DefVNI)
1229       continue;
1230 
1231     MachineInstr *DefMI = LIS->getInstructionFromIndex(DefVNI->def);
1232     const SUnit *DefSU = getSUnit(DefMI);
1233     if (!DefSU)
1234       continue;
1235 
1236     unsigned LiveOutHeight = DefSU->getHeight();
1237     unsigned LiveOutDepth = DefSU->getDepth() + DefSU->Latency;
1238     // Visit all local users of the vreg def.
1239     for (const VReg2SUnit &V2SU
1240          : make_range(VRegUses.find(Reg), VRegUses.end())) {
1241       SUnit *SU = V2SU.SU;
1242       if (SU == &ExitSU)
1243         continue;
1244 
1245       // Only consider uses of the phi.
1246       LiveQueryResult LRQ = LI.Query(LIS->getInstructionIndex(*SU->getInstr()));
1247       if (!LRQ.valueIn()->isPHIDef())
1248         continue;
1249 
1250       // Assume that a path spanning two iterations is a cycle, which could
1251       // overestimate in strange cases. This allows cyclic latency to be
1252       // estimated as the minimum slack of the vreg's depth or height.
1253       unsigned CyclicLatency = 0;
1254       if (LiveOutDepth > SU->getDepth())
1255         CyclicLatency = LiveOutDepth - SU->getDepth();
1256 
1257       unsigned LiveInHeight = SU->getHeight() + DefSU->Latency;
1258       if (LiveInHeight > LiveOutHeight) {
1259         if (LiveInHeight - LiveOutHeight < CyclicLatency)
1260           CyclicLatency = LiveInHeight - LiveOutHeight;
1261       }
1262       else
1263         CyclicLatency = 0;
1264 
1265       DEBUG(dbgs() << "Cyclic Path: SU(" << DefSU->NodeNum << ") -> SU("
1266             << SU->NodeNum << ") = " << CyclicLatency << "c\n");
1267       if (CyclicLatency > MaxCyclicLatency)
1268         MaxCyclicLatency = CyclicLatency;
1269     }
1270   }
1271   DEBUG(dbgs() << "Cyclic Critical Path: " << MaxCyclicLatency << "c\n");
1272   return MaxCyclicLatency;
1273 }
1274 
1275 /// Move an instruction and update register pressure.
1276 void ScheduleDAGMILive::scheduleMI(SUnit *SU, bool IsTopNode) {
1277   // Move the instruction to its new location in the instruction stream.
1278   MachineInstr *MI = SU->getInstr();
1279 
1280   if (IsTopNode) {
1281     assert(SU->isTopReady() && "node still has unscheduled dependencies");
1282     if (&*CurrentTop == MI)
1283       CurrentTop = nextIfDebug(++CurrentTop, CurrentBottom);
1284     else {
1285       moveInstruction(MI, CurrentTop);
1286       TopRPTracker.setPos(MI);
1287     }
1288 
1289     if (ShouldTrackPressure) {
1290       // Update top scheduled pressure.
1291       RegisterOperands RegOpers;
1292       RegOpers.collect(*MI, *TRI, MRI, ShouldTrackLaneMasks, false);
1293       if (ShouldTrackLaneMasks) {
1294         // Adjust liveness and add missing dead+read-undef flags.
1295         SlotIndex SlotIdx = LIS->getInstructionIndex(*MI).getRegSlot();
1296         RegOpers.adjustLaneLiveness(*LIS, MRI, SlotIdx, MI);
1297       } else {
1298         // Adjust for missing dead-def flags.
1299         RegOpers.detectDeadDefs(*MI, *LIS);
1300       }
1301 
1302       TopRPTracker.advance(RegOpers);
1303       assert(TopRPTracker.getPos() == CurrentTop && "out of sync");
1304       DEBUG(
1305         dbgs() << "Top Pressure:\n";
1306         dumpRegSetPressure(TopRPTracker.getRegSetPressureAtPos(), TRI);
1307       );
1308 
1309       updateScheduledPressure(SU, TopRPTracker.getPressure().MaxSetPressure);
1310     }
1311   }
1312   else {
1313     assert(SU->isBottomReady() && "node still has unscheduled dependencies");
1314     MachineBasicBlock::iterator priorII =
1315       priorNonDebug(CurrentBottom, CurrentTop);
1316     if (&*priorII == MI)
1317       CurrentBottom = priorII;
1318     else {
1319       if (&*CurrentTop == MI) {
1320         CurrentTop = nextIfDebug(++CurrentTop, priorII);
1321         TopRPTracker.setPos(CurrentTop);
1322       }
1323       moveInstruction(MI, CurrentBottom);
1324       CurrentBottom = MI;
1325     }
1326     if (ShouldTrackPressure) {
1327       RegisterOperands RegOpers;
1328       RegOpers.collect(*MI, *TRI, MRI, ShouldTrackLaneMasks, false);
1329       if (ShouldTrackLaneMasks) {
1330         // Adjust liveness and add missing dead+read-undef flags.
1331         SlotIndex SlotIdx = LIS->getInstructionIndex(*MI).getRegSlot();
1332         RegOpers.adjustLaneLiveness(*LIS, MRI, SlotIdx, MI);
1333       } else {
1334         // Adjust for missing dead-def flags.
1335         RegOpers.detectDeadDefs(*MI, *LIS);
1336       }
1337 
1338       BotRPTracker.recedeSkipDebugValues();
1339       SmallVector<RegisterMaskPair, 8> LiveUses;
1340       BotRPTracker.recede(RegOpers, &LiveUses);
1341       assert(BotRPTracker.getPos() == CurrentBottom && "out of sync");
1342       DEBUG(
1343         dbgs() << "Bottom Pressure:\n";
1344         dumpRegSetPressure(BotRPTracker.getRegSetPressureAtPos(), TRI);
1345       );
1346 
1347       updateScheduledPressure(SU, BotRPTracker.getPressure().MaxSetPressure);
1348       updatePressureDiffs(LiveUses);
1349     }
1350   }
1351 }
1352 
1353 //===----------------------------------------------------------------------===//
1354 // LoadClusterMutation - DAG post-processing to cluster loads.
1355 //===----------------------------------------------------------------------===//
1356 
1357 namespace {
1358 /// \brief Post-process the DAG to create cluster edges between neighboring
1359 /// loads.
1360 class LoadClusterMutation : public ScheduleDAGMutation {
1361   struct LoadInfo {
1362     SUnit *SU;
1363     unsigned BaseReg;
1364     int64_t Offset;
1365     LoadInfo(SUnit *su, unsigned reg, unsigned ofs)
1366       : SU(su), BaseReg(reg), Offset(ofs) {}
1367 
1368     bool operator<(const LoadInfo &RHS) const {
1369       return std::tie(BaseReg, Offset) < std::tie(RHS.BaseReg, RHS.Offset);
1370     }
1371   };
1372 
1373   const TargetInstrInfo *TII;
1374   const TargetRegisterInfo *TRI;
1375 public:
1376   LoadClusterMutation(const TargetInstrInfo *tii,
1377                       const TargetRegisterInfo *tri)
1378     : TII(tii), TRI(tri) {}
1379 
1380   void apply(ScheduleDAGInstrs *DAGInstrs) override;
1381 protected:
1382   void clusterNeighboringLoads(ArrayRef<SUnit*> Loads, ScheduleDAGMI *DAG);
1383 };
1384 } // anonymous
1385 
1386 void LoadClusterMutation::clusterNeighboringLoads(ArrayRef<SUnit*> Loads,
1387                                                   ScheduleDAGMI *DAG) {
1388   SmallVector<LoadClusterMutation::LoadInfo,32> LoadRecords;
1389   for (unsigned Idx = 0, End = Loads.size(); Idx != End; ++Idx) {
1390     SUnit *SU = Loads[Idx];
1391     unsigned BaseReg;
1392     int64_t Offset;
1393     if (TII->getMemOpBaseRegImmOfs(SU->getInstr(), BaseReg, Offset, TRI))
1394       LoadRecords.push_back(LoadInfo(SU, BaseReg, Offset));
1395   }
1396   if (LoadRecords.size() < 2)
1397     return;
1398   std::sort(LoadRecords.begin(), LoadRecords.end());
1399   unsigned ClusterLength = 1;
1400   for (unsigned Idx = 0, End = LoadRecords.size(); Idx < (End - 1); ++Idx) {
1401     if (LoadRecords[Idx].BaseReg != LoadRecords[Idx+1].BaseReg) {
1402       ClusterLength = 1;
1403       continue;
1404     }
1405 
1406     SUnit *SUa = LoadRecords[Idx].SU;
1407     SUnit *SUb = LoadRecords[Idx+1].SU;
1408     if (TII->shouldClusterLoads(SUa->getInstr(), SUb->getInstr(), ClusterLength)
1409         && DAG->addEdge(SUb, SDep(SUa, SDep::Cluster))) {
1410 
1411       DEBUG(dbgs() << "Cluster loads SU(" << SUa->NodeNum << ") - SU("
1412             << SUb->NodeNum << ")\n");
1413       // Copy successor edges from SUa to SUb. Interleaving computation
1414       // dependent on SUa can prevent load combining due to register reuse.
1415       // Predecessor edges do not need to be copied from SUb to SUa since nearby
1416       // loads should have effectively the same inputs.
1417       for (SUnit::const_succ_iterator
1418              SI = SUa->Succs.begin(), SE = SUa->Succs.end(); SI != SE; ++SI) {
1419         if (SI->getSUnit() == SUb)
1420           continue;
1421         DEBUG(dbgs() << "  Copy Succ SU(" << SI->getSUnit()->NodeNum << ")\n");
1422         DAG->addEdge(SI->getSUnit(), SDep(SUb, SDep::Artificial));
1423       }
1424       ++ClusterLength;
1425     }
1426     else
1427       ClusterLength = 1;
1428   }
1429 }
1430 
1431 /// \brief Callback from DAG postProcessing to create cluster edges for loads.
1432 void LoadClusterMutation::apply(ScheduleDAGInstrs *DAGInstrs) {
1433   ScheduleDAGMI *DAG = static_cast<ScheduleDAGMI*>(DAGInstrs);
1434 
1435   // Map DAG NodeNum to store chain ID.
1436   DenseMap<unsigned, unsigned> StoreChainIDs;
1437   // Map each store chain to a set of dependent loads.
1438   SmallVector<SmallVector<SUnit*,4>, 32> StoreChainDependents;
1439   for (unsigned Idx = 0, End = DAG->SUnits.size(); Idx != End; ++Idx) {
1440     SUnit *SU = &DAG->SUnits[Idx];
1441     if (!SU->getInstr()->mayLoad())
1442       continue;
1443     unsigned ChainPredID = DAG->SUnits.size();
1444     for (SUnit::const_pred_iterator
1445            PI = SU->Preds.begin(), PE = SU->Preds.end(); PI != PE; ++PI) {
1446       if (PI->isCtrl()) {
1447         ChainPredID = PI->getSUnit()->NodeNum;
1448         break;
1449       }
1450     }
1451     // Check if this chain-like pred has been seen
1452     // before. ChainPredID==MaxNodeID for loads at the top of the schedule.
1453     unsigned NumChains = StoreChainDependents.size();
1454     std::pair<DenseMap<unsigned, unsigned>::iterator, bool> Result =
1455       StoreChainIDs.insert(std::make_pair(ChainPredID, NumChains));
1456     if (Result.second)
1457       StoreChainDependents.resize(NumChains + 1);
1458     StoreChainDependents[Result.first->second].push_back(SU);
1459   }
1460   // Iterate over the store chains.
1461   for (unsigned Idx = 0, End = StoreChainDependents.size(); Idx != End; ++Idx)
1462     clusterNeighboringLoads(StoreChainDependents[Idx], DAG);
1463 }
1464 
1465 //===----------------------------------------------------------------------===//
1466 // MacroFusion - DAG post-processing to encourage fusion of macro ops.
1467 //===----------------------------------------------------------------------===//
1468 
1469 namespace {
1470 /// \brief Post-process the DAG to create cluster edges between instructions
1471 /// that may be fused by the processor into a single operation.
1472 class MacroFusion : public ScheduleDAGMutation {
1473   const TargetInstrInfo &TII;
1474   const TargetRegisterInfo &TRI;
1475 public:
1476   MacroFusion(const TargetInstrInfo &TII, const TargetRegisterInfo &TRI)
1477     : TII(TII), TRI(TRI) {}
1478 
1479   void apply(ScheduleDAGInstrs *DAGInstrs) override;
1480 };
1481 } // anonymous
1482 
1483 /// Returns true if \p MI reads a register written by \p Other.
1484 static bool HasDataDep(const TargetRegisterInfo &TRI, const MachineInstr &MI,
1485                        const MachineInstr &Other) {
1486   for (const MachineOperand &MO : MI.uses()) {
1487     if (!MO.isReg() || !MO.readsReg())
1488       continue;
1489 
1490     unsigned Reg = MO.getReg();
1491     if (Other.modifiesRegister(Reg, &TRI))
1492       return true;
1493   }
1494   return false;
1495 }
1496 
1497 /// \brief Callback from DAG postProcessing to create cluster edges to encourage
1498 /// fused operations.
1499 void MacroFusion::apply(ScheduleDAGInstrs *DAGInstrs) {
1500   ScheduleDAGMI *DAG = static_cast<ScheduleDAGMI*>(DAGInstrs);
1501 
1502   // For now, assume targets can only fuse with the branch.
1503   SUnit &ExitSU = DAG->ExitSU;
1504   MachineInstr *Branch = ExitSU.getInstr();
1505   if (!Branch)
1506     return;
1507 
1508   for (SUnit &SU : DAG->SUnits) {
1509     // SUnits with successors can't be schedule in front of the ExitSU.
1510     if (!SU.Succs.empty())
1511       continue;
1512     // We only care if the node writes to a register that the branch reads.
1513     MachineInstr *Pred = SU.getInstr();
1514     if (!HasDataDep(TRI, *Branch, *Pred))
1515       continue;
1516 
1517     if (!TII.shouldScheduleAdjacent(Pred, Branch))
1518       continue;
1519 
1520     // Create a single weak edge from SU to ExitSU. The only effect is to cause
1521     // bottom-up scheduling to heavily prioritize the clustered SU.  There is no
1522     // need to copy predecessor edges from ExitSU to SU, since top-down
1523     // scheduling cannot prioritize ExitSU anyway. To defer top-down scheduling
1524     // of SU, we could create an artificial edge from the deepest root, but it
1525     // hasn't been needed yet.
1526     bool Success = DAG->addEdge(&ExitSU, SDep(&SU, SDep::Cluster));
1527     (void)Success;
1528     assert(Success && "No DAG nodes should be reachable from ExitSU");
1529 
1530     DEBUG(dbgs() << "Macro Fuse SU(" << SU.NodeNum << ")\n");
1531     break;
1532   }
1533 }
1534 
1535 //===----------------------------------------------------------------------===//
1536 // CopyConstrain - DAG post-processing to encourage copy elimination.
1537 //===----------------------------------------------------------------------===//
1538 
1539 namespace {
1540 /// \brief Post-process the DAG to create weak edges from all uses of a copy to
1541 /// the one use that defines the copy's source vreg, most likely an induction
1542 /// variable increment.
1543 class CopyConstrain : public ScheduleDAGMutation {
1544   // Transient state.
1545   SlotIndex RegionBeginIdx;
1546   // RegionEndIdx is the slot index of the last non-debug instruction in the
1547   // scheduling region. So we may have RegionBeginIdx == RegionEndIdx.
1548   SlotIndex RegionEndIdx;
1549 public:
1550   CopyConstrain(const TargetInstrInfo *, const TargetRegisterInfo *) {}
1551 
1552   void apply(ScheduleDAGInstrs *DAGInstrs) override;
1553 
1554 protected:
1555   void constrainLocalCopy(SUnit *CopySU, ScheduleDAGMILive *DAG);
1556 };
1557 } // anonymous
1558 
1559 /// constrainLocalCopy handles two possibilities:
1560 /// 1) Local src:
1561 /// I0:     = dst
1562 /// I1: src = ...
1563 /// I2:     = dst
1564 /// I3: dst = src (copy)
1565 /// (create pred->succ edges I0->I1, I2->I1)
1566 ///
1567 /// 2) Local copy:
1568 /// I0: dst = src (copy)
1569 /// I1:     = dst
1570 /// I2: src = ...
1571 /// I3:     = dst
1572 /// (create pred->succ edges I1->I2, I3->I2)
1573 ///
1574 /// Although the MachineScheduler is currently constrained to single blocks,
1575 /// this algorithm should handle extended blocks. An EBB is a set of
1576 /// contiguously numbered blocks such that the previous block in the EBB is
1577 /// always the single predecessor.
1578 void CopyConstrain::constrainLocalCopy(SUnit *CopySU, ScheduleDAGMILive *DAG) {
1579   LiveIntervals *LIS = DAG->getLIS();
1580   MachineInstr *Copy = CopySU->getInstr();
1581 
1582   // Check for pure vreg copies.
1583   unsigned SrcReg = Copy->getOperand(1).getReg();
1584   if (!TargetRegisterInfo::isVirtualRegister(SrcReg))
1585     return;
1586 
1587   unsigned DstReg = Copy->getOperand(0).getReg();
1588   if (!TargetRegisterInfo::isVirtualRegister(DstReg))
1589     return;
1590 
1591   // Check if either the dest or source is local. If it's live across a back
1592   // edge, it's not local. Note that if both vregs are live across the back
1593   // edge, we cannot successfully contrain the copy without cyclic scheduling.
1594   // If both the copy's source and dest are local live intervals, then we
1595   // should treat the dest as the global for the purpose of adding
1596   // constraints. This adds edges from source's other uses to the copy.
1597   unsigned LocalReg = SrcReg;
1598   unsigned GlobalReg = DstReg;
1599   LiveInterval *LocalLI = &LIS->getInterval(LocalReg);
1600   if (!LocalLI->isLocal(RegionBeginIdx, RegionEndIdx)) {
1601     LocalReg = DstReg;
1602     GlobalReg = SrcReg;
1603     LocalLI = &LIS->getInterval(LocalReg);
1604     if (!LocalLI->isLocal(RegionBeginIdx, RegionEndIdx))
1605       return;
1606   }
1607   LiveInterval *GlobalLI = &LIS->getInterval(GlobalReg);
1608 
1609   // Find the global segment after the start of the local LI.
1610   LiveInterval::iterator GlobalSegment = GlobalLI->find(LocalLI->beginIndex());
1611   // If GlobalLI does not overlap LocalLI->start, then a copy directly feeds a
1612   // local live range. We could create edges from other global uses to the local
1613   // start, but the coalescer should have already eliminated these cases, so
1614   // don't bother dealing with it.
1615   if (GlobalSegment == GlobalLI->end())
1616     return;
1617 
1618   // If GlobalSegment is killed at the LocalLI->start, the call to find()
1619   // returned the next global segment. But if GlobalSegment overlaps with
1620   // LocalLI->start, then advance to the next segement. If a hole in GlobalLI
1621   // exists in LocalLI's vicinity, GlobalSegment will be the end of the hole.
1622   if (GlobalSegment->contains(LocalLI->beginIndex()))
1623     ++GlobalSegment;
1624 
1625   if (GlobalSegment == GlobalLI->end())
1626     return;
1627 
1628   // Check if GlobalLI contains a hole in the vicinity of LocalLI.
1629   if (GlobalSegment != GlobalLI->begin()) {
1630     // Two address defs have no hole.
1631     if (SlotIndex::isSameInstr(std::prev(GlobalSegment)->end,
1632                                GlobalSegment->start)) {
1633       return;
1634     }
1635     // If the prior global segment may be defined by the same two-address
1636     // instruction that also defines LocalLI, then can't make a hole here.
1637     if (SlotIndex::isSameInstr(std::prev(GlobalSegment)->start,
1638                                LocalLI->beginIndex())) {
1639       return;
1640     }
1641     // If GlobalLI has a prior segment, it must be live into the EBB. Otherwise
1642     // it would be a disconnected component in the live range.
1643     assert(std::prev(GlobalSegment)->start < LocalLI->beginIndex() &&
1644            "Disconnected LRG within the scheduling region.");
1645   }
1646   MachineInstr *GlobalDef = LIS->getInstructionFromIndex(GlobalSegment->start);
1647   if (!GlobalDef)
1648     return;
1649 
1650   SUnit *GlobalSU = DAG->getSUnit(GlobalDef);
1651   if (!GlobalSU)
1652     return;
1653 
1654   // GlobalDef is the bottom of the GlobalLI hole. Open the hole by
1655   // constraining the uses of the last local def to precede GlobalDef.
1656   SmallVector<SUnit*,8> LocalUses;
1657   const VNInfo *LastLocalVN = LocalLI->getVNInfoBefore(LocalLI->endIndex());
1658   MachineInstr *LastLocalDef = LIS->getInstructionFromIndex(LastLocalVN->def);
1659   SUnit *LastLocalSU = DAG->getSUnit(LastLocalDef);
1660   for (SUnit::const_succ_iterator
1661          I = LastLocalSU->Succs.begin(), E = LastLocalSU->Succs.end();
1662        I != E; ++I) {
1663     if (I->getKind() != SDep::Data || I->getReg() != LocalReg)
1664       continue;
1665     if (I->getSUnit() == GlobalSU)
1666       continue;
1667     if (!DAG->canAddEdge(GlobalSU, I->getSUnit()))
1668       return;
1669     LocalUses.push_back(I->getSUnit());
1670   }
1671   // Open the top of the GlobalLI hole by constraining any earlier global uses
1672   // to precede the start of LocalLI.
1673   SmallVector<SUnit*,8> GlobalUses;
1674   MachineInstr *FirstLocalDef =
1675     LIS->getInstructionFromIndex(LocalLI->beginIndex());
1676   SUnit *FirstLocalSU = DAG->getSUnit(FirstLocalDef);
1677   for (SUnit::const_pred_iterator
1678          I = GlobalSU->Preds.begin(), E = GlobalSU->Preds.end(); I != E; ++I) {
1679     if (I->getKind() != SDep::Anti || I->getReg() != GlobalReg)
1680       continue;
1681     if (I->getSUnit() == FirstLocalSU)
1682       continue;
1683     if (!DAG->canAddEdge(FirstLocalSU, I->getSUnit()))
1684       return;
1685     GlobalUses.push_back(I->getSUnit());
1686   }
1687   DEBUG(dbgs() << "Constraining copy SU(" << CopySU->NodeNum << ")\n");
1688   // Add the weak edges.
1689   for (SmallVectorImpl<SUnit*>::const_iterator
1690          I = LocalUses.begin(), E = LocalUses.end(); I != E; ++I) {
1691     DEBUG(dbgs() << "  Local use SU(" << (*I)->NodeNum << ") -> SU("
1692           << GlobalSU->NodeNum << ")\n");
1693     DAG->addEdge(GlobalSU, SDep(*I, SDep::Weak));
1694   }
1695   for (SmallVectorImpl<SUnit*>::const_iterator
1696          I = GlobalUses.begin(), E = GlobalUses.end(); I != E; ++I) {
1697     DEBUG(dbgs() << "  Global use SU(" << (*I)->NodeNum << ") -> SU("
1698           << FirstLocalSU->NodeNum << ")\n");
1699     DAG->addEdge(FirstLocalSU, SDep(*I, SDep::Weak));
1700   }
1701 }
1702 
1703 /// \brief Callback from DAG postProcessing to create weak edges to encourage
1704 /// copy elimination.
1705 void CopyConstrain::apply(ScheduleDAGInstrs *DAGInstrs) {
1706   ScheduleDAGMI *DAG = static_cast<ScheduleDAGMI*>(DAGInstrs);
1707   assert(DAG->hasVRegLiveness() && "Expect VRegs with LiveIntervals");
1708 
1709   MachineBasicBlock::iterator FirstPos = nextIfDebug(DAG->begin(), DAG->end());
1710   if (FirstPos == DAG->end())
1711     return;
1712   RegionBeginIdx = DAG->getLIS()->getInstructionIndex(*FirstPos);
1713   RegionEndIdx = DAG->getLIS()->getInstructionIndex(
1714       *priorNonDebug(DAG->end(), DAG->begin()));
1715 
1716   for (unsigned Idx = 0, End = DAG->SUnits.size(); Idx != End; ++Idx) {
1717     SUnit *SU = &DAG->SUnits[Idx];
1718     if (!SU->getInstr()->isCopy())
1719       continue;
1720 
1721     constrainLocalCopy(SU, static_cast<ScheduleDAGMILive*>(DAG));
1722   }
1723 }
1724 
1725 //===----------------------------------------------------------------------===//
1726 // MachineSchedStrategy helpers used by GenericScheduler, GenericPostScheduler
1727 // and possibly other custom schedulers.
1728 //===----------------------------------------------------------------------===//
1729 
1730 static const unsigned InvalidCycle = ~0U;
1731 
1732 SchedBoundary::~SchedBoundary() { delete HazardRec; }
1733 
1734 void SchedBoundary::reset() {
1735   // A new HazardRec is created for each DAG and owned by SchedBoundary.
1736   // Destroying and reconstructing it is very expensive though. So keep
1737   // invalid, placeholder HazardRecs.
1738   if (HazardRec && HazardRec->isEnabled()) {
1739     delete HazardRec;
1740     HazardRec = nullptr;
1741   }
1742   Available.clear();
1743   Pending.clear();
1744   CheckPending = false;
1745   NextSUs.clear();
1746   CurrCycle = 0;
1747   CurrMOps = 0;
1748   MinReadyCycle = UINT_MAX;
1749   ExpectedLatency = 0;
1750   DependentLatency = 0;
1751   RetiredMOps = 0;
1752   MaxExecutedResCount = 0;
1753   ZoneCritResIdx = 0;
1754   IsResourceLimited = false;
1755   ReservedCycles.clear();
1756 #ifndef NDEBUG
1757   // Track the maximum number of stall cycles that could arise either from the
1758   // latency of a DAG edge or the number of cycles that a processor resource is
1759   // reserved (SchedBoundary::ReservedCycles).
1760   MaxObservedStall = 0;
1761 #endif
1762   // Reserve a zero-count for invalid CritResIdx.
1763   ExecutedResCounts.resize(1);
1764   assert(!ExecutedResCounts[0] && "nonzero count for bad resource");
1765 }
1766 
1767 void SchedRemainder::
1768 init(ScheduleDAGMI *DAG, const TargetSchedModel *SchedModel) {
1769   reset();
1770   if (!SchedModel->hasInstrSchedModel())
1771     return;
1772   RemainingCounts.resize(SchedModel->getNumProcResourceKinds());
1773   for (std::vector<SUnit>::iterator
1774          I = DAG->SUnits.begin(), E = DAG->SUnits.end(); I != E; ++I) {
1775     const MCSchedClassDesc *SC = DAG->getSchedClass(&*I);
1776     RemIssueCount += SchedModel->getNumMicroOps(I->getInstr(), SC)
1777       * SchedModel->getMicroOpFactor();
1778     for (TargetSchedModel::ProcResIter
1779            PI = SchedModel->getWriteProcResBegin(SC),
1780            PE = SchedModel->getWriteProcResEnd(SC); PI != PE; ++PI) {
1781       unsigned PIdx = PI->ProcResourceIdx;
1782       unsigned Factor = SchedModel->getResourceFactor(PIdx);
1783       RemainingCounts[PIdx] += (Factor * PI->Cycles);
1784     }
1785   }
1786 }
1787 
1788 void SchedBoundary::
1789 init(ScheduleDAGMI *dag, const TargetSchedModel *smodel, SchedRemainder *rem) {
1790   reset();
1791   DAG = dag;
1792   SchedModel = smodel;
1793   Rem = rem;
1794   if (SchedModel->hasInstrSchedModel()) {
1795     ExecutedResCounts.resize(SchedModel->getNumProcResourceKinds());
1796     ReservedCycles.resize(SchedModel->getNumProcResourceKinds(), InvalidCycle);
1797   }
1798 }
1799 
1800 /// Compute the stall cycles based on this SUnit's ready time. Heuristics treat
1801 /// these "soft stalls" differently than the hard stall cycles based on CPU
1802 /// resources and computed by checkHazard(). A fully in-order model
1803 /// (MicroOpBufferSize==0) will not make use of this since instructions are not
1804 /// available for scheduling until they are ready. However, a weaker in-order
1805 /// model may use this for heuristics. For example, if a processor has in-order
1806 /// behavior when reading certain resources, this may come into play.
1807 unsigned SchedBoundary::getLatencyStallCycles(SUnit *SU) {
1808   if (!SU->isUnbuffered)
1809     return 0;
1810 
1811   unsigned ReadyCycle = (isTop() ? SU->TopReadyCycle : SU->BotReadyCycle);
1812   if (ReadyCycle > CurrCycle)
1813     return ReadyCycle - CurrCycle;
1814   return 0;
1815 }
1816 
1817 /// Compute the next cycle at which the given processor resource can be
1818 /// scheduled.
1819 unsigned SchedBoundary::
1820 getNextResourceCycle(unsigned PIdx, unsigned Cycles) {
1821   unsigned NextUnreserved = ReservedCycles[PIdx];
1822   // If this resource has never been used, always return cycle zero.
1823   if (NextUnreserved == InvalidCycle)
1824     return 0;
1825   // For bottom-up scheduling add the cycles needed for the current operation.
1826   if (!isTop())
1827     NextUnreserved += Cycles;
1828   return NextUnreserved;
1829 }
1830 
1831 /// Does this SU have a hazard within the current instruction group.
1832 ///
1833 /// The scheduler supports two modes of hazard recognition. The first is the
1834 /// ScheduleHazardRecognizer API. It is a fully general hazard recognizer that
1835 /// supports highly complicated in-order reservation tables
1836 /// (ScoreboardHazardRecognizer) and arbitraty target-specific logic.
1837 ///
1838 /// The second is a streamlined mechanism that checks for hazards based on
1839 /// simple counters that the scheduler itself maintains. It explicitly checks
1840 /// for instruction dispatch limitations, including the number of micro-ops that
1841 /// can dispatch per cycle.
1842 ///
1843 /// TODO: Also check whether the SU must start a new group.
1844 bool SchedBoundary::checkHazard(SUnit *SU) {
1845   if (HazardRec->isEnabled()
1846       && HazardRec->getHazardType(SU) != ScheduleHazardRecognizer::NoHazard) {
1847     return true;
1848   }
1849   unsigned uops = SchedModel->getNumMicroOps(SU->getInstr());
1850   if ((CurrMOps > 0) && (CurrMOps + uops > SchedModel->getIssueWidth())) {
1851     DEBUG(dbgs() << "  SU(" << SU->NodeNum << ") uops="
1852           << SchedModel->getNumMicroOps(SU->getInstr()) << '\n');
1853     return true;
1854   }
1855   if (SchedModel->hasInstrSchedModel() && SU->hasReservedResource) {
1856     const MCSchedClassDesc *SC = DAG->getSchedClass(SU);
1857     for (TargetSchedModel::ProcResIter
1858            PI = SchedModel->getWriteProcResBegin(SC),
1859            PE = SchedModel->getWriteProcResEnd(SC); PI != PE; ++PI) {
1860       unsigned NRCycle = getNextResourceCycle(PI->ProcResourceIdx, PI->Cycles);
1861       if (NRCycle > CurrCycle) {
1862 #ifndef NDEBUG
1863         MaxObservedStall = std::max(PI->Cycles, MaxObservedStall);
1864 #endif
1865         DEBUG(dbgs() << "  SU(" << SU->NodeNum << ") "
1866               << SchedModel->getResourceName(PI->ProcResourceIdx)
1867               << "=" << NRCycle << "c\n");
1868         return true;
1869       }
1870     }
1871   }
1872   return false;
1873 }
1874 
1875 // Find the unscheduled node in ReadySUs with the highest latency.
1876 unsigned SchedBoundary::
1877 findMaxLatency(ArrayRef<SUnit*> ReadySUs) {
1878   SUnit *LateSU = nullptr;
1879   unsigned RemLatency = 0;
1880   for (ArrayRef<SUnit*>::iterator I = ReadySUs.begin(), E = ReadySUs.end();
1881        I != E; ++I) {
1882     unsigned L = getUnscheduledLatency(*I);
1883     if (L > RemLatency) {
1884       RemLatency = L;
1885       LateSU = *I;
1886     }
1887   }
1888   if (LateSU) {
1889     DEBUG(dbgs() << Available.getName() << " RemLatency SU("
1890           << LateSU->NodeNum << ") " << RemLatency << "c\n");
1891   }
1892   return RemLatency;
1893 }
1894 
1895 // Count resources in this zone and the remaining unscheduled
1896 // instruction. Return the max count, scaled. Set OtherCritIdx to the critical
1897 // resource index, or zero if the zone is issue limited.
1898 unsigned SchedBoundary::
1899 getOtherResourceCount(unsigned &OtherCritIdx) {
1900   OtherCritIdx = 0;
1901   if (!SchedModel->hasInstrSchedModel())
1902     return 0;
1903 
1904   unsigned OtherCritCount = Rem->RemIssueCount
1905     + (RetiredMOps * SchedModel->getMicroOpFactor());
1906   DEBUG(dbgs() << "  " << Available.getName() << " + Remain MOps: "
1907         << OtherCritCount / SchedModel->getMicroOpFactor() << '\n');
1908   for (unsigned PIdx = 1, PEnd = SchedModel->getNumProcResourceKinds();
1909        PIdx != PEnd; ++PIdx) {
1910     unsigned OtherCount = getResourceCount(PIdx) + Rem->RemainingCounts[PIdx];
1911     if (OtherCount > OtherCritCount) {
1912       OtherCritCount = OtherCount;
1913       OtherCritIdx = PIdx;
1914     }
1915   }
1916   if (OtherCritIdx) {
1917     DEBUG(dbgs() << "  " << Available.getName() << " + Remain CritRes: "
1918           << OtherCritCount / SchedModel->getResourceFactor(OtherCritIdx)
1919           << " " << SchedModel->getResourceName(OtherCritIdx) << "\n");
1920   }
1921   return OtherCritCount;
1922 }
1923 
1924 void SchedBoundary::releaseNode(SUnit *SU, unsigned ReadyCycle) {
1925   assert(SU->getInstr() && "Scheduled SUnit must have instr");
1926 
1927 #ifndef NDEBUG
1928   // ReadyCycle was been bumped up to the CurrCycle when this node was
1929   // scheduled, but CurrCycle may have been eagerly advanced immediately after
1930   // scheduling, so may now be greater than ReadyCycle.
1931   if (ReadyCycle > CurrCycle)
1932     MaxObservedStall = std::max(ReadyCycle - CurrCycle, MaxObservedStall);
1933 #endif
1934 
1935   if (ReadyCycle < MinReadyCycle)
1936     MinReadyCycle = ReadyCycle;
1937 
1938   // Check for interlocks first. For the purpose of other heuristics, an
1939   // instruction that cannot issue appears as if it's not in the ReadyQueue.
1940   bool IsBuffered = SchedModel->getMicroOpBufferSize() != 0;
1941   if ((!IsBuffered && ReadyCycle > CurrCycle) || checkHazard(SU))
1942     Pending.push(SU);
1943   else
1944     Available.push(SU);
1945 
1946   // Record this node as an immediate dependent of the scheduled node.
1947   NextSUs.insert(SU);
1948 }
1949 
1950 void SchedBoundary::releaseTopNode(SUnit *SU) {
1951   if (SU->isScheduled)
1952     return;
1953 
1954   releaseNode(SU, SU->TopReadyCycle);
1955 }
1956 
1957 void SchedBoundary::releaseBottomNode(SUnit *SU) {
1958   if (SU->isScheduled)
1959     return;
1960 
1961   releaseNode(SU, SU->BotReadyCycle);
1962 }
1963 
1964 /// Move the boundary of scheduled code by one cycle.
1965 void SchedBoundary::bumpCycle(unsigned NextCycle) {
1966   if (SchedModel->getMicroOpBufferSize() == 0) {
1967     assert(MinReadyCycle < UINT_MAX && "MinReadyCycle uninitialized");
1968     if (MinReadyCycle > NextCycle)
1969       NextCycle = MinReadyCycle;
1970   }
1971   // Update the current micro-ops, which will issue in the next cycle.
1972   unsigned DecMOps = SchedModel->getIssueWidth() * (NextCycle - CurrCycle);
1973   CurrMOps = (CurrMOps <= DecMOps) ? 0 : CurrMOps - DecMOps;
1974 
1975   // Decrement DependentLatency based on the next cycle.
1976   if ((NextCycle - CurrCycle) > DependentLatency)
1977     DependentLatency = 0;
1978   else
1979     DependentLatency -= (NextCycle - CurrCycle);
1980 
1981   if (!HazardRec->isEnabled()) {
1982     // Bypass HazardRec virtual calls.
1983     CurrCycle = NextCycle;
1984   }
1985   else {
1986     // Bypass getHazardType calls in case of long latency.
1987     for (; CurrCycle != NextCycle; ++CurrCycle) {
1988       if (isTop())
1989         HazardRec->AdvanceCycle();
1990       else
1991         HazardRec->RecedeCycle();
1992     }
1993   }
1994   CheckPending = true;
1995   unsigned LFactor = SchedModel->getLatencyFactor();
1996   IsResourceLimited =
1997     (int)(getCriticalCount() - (getScheduledLatency() * LFactor))
1998     > (int)LFactor;
1999 
2000   DEBUG(dbgs() << "Cycle: " << CurrCycle << ' ' << Available.getName() << '\n');
2001 }
2002 
2003 void SchedBoundary::incExecutedResources(unsigned PIdx, unsigned Count) {
2004   ExecutedResCounts[PIdx] += Count;
2005   if (ExecutedResCounts[PIdx] > MaxExecutedResCount)
2006     MaxExecutedResCount = ExecutedResCounts[PIdx];
2007 }
2008 
2009 /// Add the given processor resource to this scheduled zone.
2010 ///
2011 /// \param Cycles indicates the number of consecutive (non-pipelined) cycles
2012 /// during which this resource is consumed.
2013 ///
2014 /// \return the next cycle at which the instruction may execute without
2015 /// oversubscribing resources.
2016 unsigned SchedBoundary::
2017 countResource(unsigned PIdx, unsigned Cycles, unsigned NextCycle) {
2018   unsigned Factor = SchedModel->getResourceFactor(PIdx);
2019   unsigned Count = Factor * Cycles;
2020   DEBUG(dbgs() << "  " << SchedModel->getResourceName(PIdx)
2021         << " +" << Cycles << "x" << Factor << "u\n");
2022 
2023   // Update Executed resources counts.
2024   incExecutedResources(PIdx, Count);
2025   assert(Rem->RemainingCounts[PIdx] >= Count && "resource double counted");
2026   Rem->RemainingCounts[PIdx] -= Count;
2027 
2028   // Check if this resource exceeds the current critical resource. If so, it
2029   // becomes the critical resource.
2030   if (ZoneCritResIdx != PIdx && (getResourceCount(PIdx) > getCriticalCount())) {
2031     ZoneCritResIdx = PIdx;
2032     DEBUG(dbgs() << "  *** Critical resource "
2033           << SchedModel->getResourceName(PIdx) << ": "
2034           << getResourceCount(PIdx) / SchedModel->getLatencyFactor() << "c\n");
2035   }
2036   // For reserved resources, record the highest cycle using the resource.
2037   unsigned NextAvailable = getNextResourceCycle(PIdx, Cycles);
2038   if (NextAvailable > CurrCycle) {
2039     DEBUG(dbgs() << "  Resource conflict: "
2040           << SchedModel->getProcResource(PIdx)->Name << " reserved until @"
2041           << NextAvailable << "\n");
2042   }
2043   return NextAvailable;
2044 }
2045 
2046 /// Move the boundary of scheduled code by one SUnit.
2047 void SchedBoundary::bumpNode(SUnit *SU) {
2048   // Update the reservation table.
2049   if (HazardRec->isEnabled()) {
2050     if (!isTop() && SU->isCall) {
2051       // Calls are scheduled with their preceding instructions. For bottom-up
2052       // scheduling, clear the pipeline state before emitting.
2053       HazardRec->Reset();
2054     }
2055     HazardRec->EmitInstruction(SU);
2056   }
2057   // checkHazard should prevent scheduling multiple instructions per cycle that
2058   // exceed the issue width.
2059   const MCSchedClassDesc *SC = DAG->getSchedClass(SU);
2060   unsigned IncMOps = SchedModel->getNumMicroOps(SU->getInstr());
2061   assert(
2062       (CurrMOps == 0 || (CurrMOps + IncMOps) <= SchedModel->getIssueWidth()) &&
2063       "Cannot schedule this instruction's MicroOps in the current cycle.");
2064 
2065   unsigned ReadyCycle = (isTop() ? SU->TopReadyCycle : SU->BotReadyCycle);
2066   DEBUG(dbgs() << "  Ready @" << ReadyCycle << "c\n");
2067 
2068   unsigned NextCycle = CurrCycle;
2069   switch (SchedModel->getMicroOpBufferSize()) {
2070   case 0:
2071     assert(ReadyCycle <= CurrCycle && "Broken PendingQueue");
2072     break;
2073   case 1:
2074     if (ReadyCycle > NextCycle) {
2075       NextCycle = ReadyCycle;
2076       DEBUG(dbgs() << "  *** Stall until: " << ReadyCycle << "\n");
2077     }
2078     break;
2079   default:
2080     // We don't currently model the OOO reorder buffer, so consider all
2081     // scheduled MOps to be "retired". We do loosely model in-order resource
2082     // latency. If this instruction uses an in-order resource, account for any
2083     // likely stall cycles.
2084     if (SU->isUnbuffered && ReadyCycle > NextCycle)
2085       NextCycle = ReadyCycle;
2086     break;
2087   }
2088   RetiredMOps += IncMOps;
2089 
2090   // Update resource counts and critical resource.
2091   if (SchedModel->hasInstrSchedModel()) {
2092     unsigned DecRemIssue = IncMOps * SchedModel->getMicroOpFactor();
2093     assert(Rem->RemIssueCount >= DecRemIssue && "MOps double counted");
2094     Rem->RemIssueCount -= DecRemIssue;
2095     if (ZoneCritResIdx) {
2096       // Scale scheduled micro-ops for comparing with the critical resource.
2097       unsigned ScaledMOps =
2098         RetiredMOps * SchedModel->getMicroOpFactor();
2099 
2100       // If scaled micro-ops are now more than the previous critical resource by
2101       // a full cycle, then micro-ops issue becomes critical.
2102       if ((int)(ScaledMOps - getResourceCount(ZoneCritResIdx))
2103           >= (int)SchedModel->getLatencyFactor()) {
2104         ZoneCritResIdx = 0;
2105         DEBUG(dbgs() << "  *** Critical resource NumMicroOps: "
2106               << ScaledMOps / SchedModel->getLatencyFactor() << "c\n");
2107       }
2108     }
2109     for (TargetSchedModel::ProcResIter
2110            PI = SchedModel->getWriteProcResBegin(SC),
2111            PE = SchedModel->getWriteProcResEnd(SC); PI != PE; ++PI) {
2112       unsigned RCycle =
2113         countResource(PI->ProcResourceIdx, PI->Cycles, NextCycle);
2114       if (RCycle > NextCycle)
2115         NextCycle = RCycle;
2116     }
2117     if (SU->hasReservedResource) {
2118       // For reserved resources, record the highest cycle using the resource.
2119       // For top-down scheduling, this is the cycle in which we schedule this
2120       // instruction plus the number of cycles the operations reserves the
2121       // resource. For bottom-up is it simply the instruction's cycle.
2122       for (TargetSchedModel::ProcResIter
2123              PI = SchedModel->getWriteProcResBegin(SC),
2124              PE = SchedModel->getWriteProcResEnd(SC); PI != PE; ++PI) {
2125         unsigned PIdx = PI->ProcResourceIdx;
2126         if (SchedModel->getProcResource(PIdx)->BufferSize == 0) {
2127           if (isTop()) {
2128             ReservedCycles[PIdx] =
2129               std::max(getNextResourceCycle(PIdx, 0), NextCycle + PI->Cycles);
2130           }
2131           else
2132             ReservedCycles[PIdx] = NextCycle;
2133         }
2134       }
2135     }
2136   }
2137   // Update ExpectedLatency and DependentLatency.
2138   unsigned &TopLatency = isTop() ? ExpectedLatency : DependentLatency;
2139   unsigned &BotLatency = isTop() ? DependentLatency : ExpectedLatency;
2140   if (SU->getDepth() > TopLatency) {
2141     TopLatency = SU->getDepth();
2142     DEBUG(dbgs() << "  " << Available.getName()
2143           << " TopLatency SU(" << SU->NodeNum << ") " << TopLatency << "c\n");
2144   }
2145   if (SU->getHeight() > BotLatency) {
2146     BotLatency = SU->getHeight();
2147     DEBUG(dbgs() << "  " << Available.getName()
2148           << " BotLatency SU(" << SU->NodeNum << ") " << BotLatency << "c\n");
2149   }
2150   // If we stall for any reason, bump the cycle.
2151   if (NextCycle > CurrCycle) {
2152     bumpCycle(NextCycle);
2153   }
2154   else {
2155     // After updating ZoneCritResIdx and ExpectedLatency, check if we're
2156     // resource limited. If a stall occurred, bumpCycle does this.
2157     unsigned LFactor = SchedModel->getLatencyFactor();
2158     IsResourceLimited =
2159       (int)(getCriticalCount() - (getScheduledLatency() * LFactor))
2160       > (int)LFactor;
2161   }
2162   // Update CurrMOps after calling bumpCycle to handle stalls, since bumpCycle
2163   // resets CurrMOps. Loop to handle instructions with more MOps than issue in
2164   // one cycle.  Since we commonly reach the max MOps here, opportunistically
2165   // bump the cycle to avoid uselessly checking everything in the readyQ.
2166   CurrMOps += IncMOps;
2167   while (CurrMOps >= SchedModel->getIssueWidth()) {
2168     DEBUG(dbgs() << "  *** Max MOps " << CurrMOps
2169           << " at cycle " << CurrCycle << '\n');
2170     bumpCycle(++NextCycle);
2171   }
2172   DEBUG(dumpScheduledState());
2173 }
2174 
2175 /// Release pending ready nodes in to the available queue. This makes them
2176 /// visible to heuristics.
2177 void SchedBoundary::releasePending() {
2178   // If the available queue is empty, it is safe to reset MinReadyCycle.
2179   if (Available.empty())
2180     MinReadyCycle = UINT_MAX;
2181 
2182   // Check to see if any of the pending instructions are ready to issue.  If
2183   // so, add them to the available queue.
2184   bool IsBuffered = SchedModel->getMicroOpBufferSize() != 0;
2185   for (unsigned i = 0, e = Pending.size(); i != e; ++i) {
2186     SUnit *SU = *(Pending.begin()+i);
2187     unsigned ReadyCycle = isTop() ? SU->TopReadyCycle : SU->BotReadyCycle;
2188 
2189     if (ReadyCycle < MinReadyCycle)
2190       MinReadyCycle = ReadyCycle;
2191 
2192     if (!IsBuffered && ReadyCycle > CurrCycle)
2193       continue;
2194 
2195     if (checkHazard(SU))
2196       continue;
2197 
2198     Available.push(SU);
2199     Pending.remove(Pending.begin()+i);
2200     --i; --e;
2201   }
2202   DEBUG(if (!Pending.empty()) Pending.dump());
2203   CheckPending = false;
2204 }
2205 
2206 /// Remove SU from the ready set for this boundary.
2207 void SchedBoundary::removeReady(SUnit *SU) {
2208   if (Available.isInQueue(SU))
2209     Available.remove(Available.find(SU));
2210   else {
2211     assert(Pending.isInQueue(SU) && "bad ready count");
2212     Pending.remove(Pending.find(SU));
2213   }
2214 }
2215 
2216 /// If this queue only has one ready candidate, return it. As a side effect,
2217 /// defer any nodes that now hit a hazard, and advance the cycle until at least
2218 /// one node is ready. If multiple instructions are ready, return NULL.
2219 SUnit *SchedBoundary::pickOnlyChoice() {
2220   if (CheckPending)
2221     releasePending();
2222 
2223   if (CurrMOps > 0) {
2224     // Defer any ready instrs that now have a hazard.
2225     for (ReadyQueue::iterator I = Available.begin(); I != Available.end();) {
2226       if (checkHazard(*I)) {
2227         Pending.push(*I);
2228         I = Available.remove(I);
2229         continue;
2230       }
2231       ++I;
2232     }
2233   }
2234   for (unsigned i = 0; Available.empty(); ++i) {
2235 //  FIXME: Re-enable assert once PR20057 is resolved.
2236 //    assert(i <= (HazardRec->getMaxLookAhead() + MaxObservedStall) &&
2237 //           "permanent hazard");
2238     (void)i;
2239     bumpCycle(CurrCycle + 1);
2240     releasePending();
2241   }
2242   if (Available.size() == 1)
2243     return *Available.begin();
2244   return nullptr;
2245 }
2246 
2247 #ifndef NDEBUG
2248 // This is useful information to dump after bumpNode.
2249 // Note that the Queue contents are more useful before pickNodeFromQueue.
2250 void SchedBoundary::dumpScheduledState() {
2251   unsigned ResFactor;
2252   unsigned ResCount;
2253   if (ZoneCritResIdx) {
2254     ResFactor = SchedModel->getResourceFactor(ZoneCritResIdx);
2255     ResCount = getResourceCount(ZoneCritResIdx);
2256   }
2257   else {
2258     ResFactor = SchedModel->getMicroOpFactor();
2259     ResCount = RetiredMOps * SchedModel->getMicroOpFactor();
2260   }
2261   unsigned LFactor = SchedModel->getLatencyFactor();
2262   dbgs() << Available.getName() << " @" << CurrCycle << "c\n"
2263          << "  Retired: " << RetiredMOps;
2264   dbgs() << "\n  Executed: " << getExecutedCount() / LFactor << "c";
2265   dbgs() << "\n  Critical: " << ResCount / LFactor << "c, "
2266          << ResCount / ResFactor << " "
2267          << SchedModel->getResourceName(ZoneCritResIdx)
2268          << "\n  ExpectedLatency: " << ExpectedLatency << "c\n"
2269          << (IsResourceLimited ? "  - Resource" : "  - Latency")
2270          << " limited.\n";
2271 }
2272 #endif
2273 
2274 //===----------------------------------------------------------------------===//
2275 // GenericScheduler - Generic implementation of MachineSchedStrategy.
2276 //===----------------------------------------------------------------------===//
2277 
2278 void GenericSchedulerBase::SchedCandidate::
2279 initResourceDelta(const ScheduleDAGMI *DAG,
2280                   const TargetSchedModel *SchedModel) {
2281   if (!Policy.ReduceResIdx && !Policy.DemandResIdx)
2282     return;
2283 
2284   const MCSchedClassDesc *SC = DAG->getSchedClass(SU);
2285   for (TargetSchedModel::ProcResIter
2286          PI = SchedModel->getWriteProcResBegin(SC),
2287          PE = SchedModel->getWriteProcResEnd(SC); PI != PE; ++PI) {
2288     if (PI->ProcResourceIdx == Policy.ReduceResIdx)
2289       ResDelta.CritResources += PI->Cycles;
2290     if (PI->ProcResourceIdx == Policy.DemandResIdx)
2291       ResDelta.DemandedResources += PI->Cycles;
2292   }
2293 }
2294 
2295 /// Set the CandPolicy given a scheduling zone given the current resources and
2296 /// latencies inside and outside the zone.
2297 void GenericSchedulerBase::setPolicy(CandPolicy &Policy,
2298                                      bool IsPostRA,
2299                                      SchedBoundary &CurrZone,
2300                                      SchedBoundary *OtherZone) {
2301   // Apply preemptive heuristics based on the total latency and resources
2302   // inside and outside this zone. Potential stalls should be considered before
2303   // following this policy.
2304 
2305   // Compute remaining latency. We need this both to determine whether the
2306   // overall schedule has become latency-limited and whether the instructions
2307   // outside this zone are resource or latency limited.
2308   //
2309   // The "dependent" latency is updated incrementally during scheduling as the
2310   // max height/depth of scheduled nodes minus the cycles since it was
2311   // scheduled:
2312   //   DLat = max (N.depth - (CurrCycle - N.ReadyCycle) for N in Zone
2313   //
2314   // The "independent" latency is the max ready queue depth:
2315   //   ILat = max N.depth for N in Available|Pending
2316   //
2317   // RemainingLatency is the greater of independent and dependent latency.
2318   unsigned RemLatency = CurrZone.getDependentLatency();
2319   RemLatency = std::max(RemLatency,
2320                         CurrZone.findMaxLatency(CurrZone.Available.elements()));
2321   RemLatency = std::max(RemLatency,
2322                         CurrZone.findMaxLatency(CurrZone.Pending.elements()));
2323 
2324   // Compute the critical resource outside the zone.
2325   unsigned OtherCritIdx = 0;
2326   unsigned OtherCount =
2327     OtherZone ? OtherZone->getOtherResourceCount(OtherCritIdx) : 0;
2328 
2329   bool OtherResLimited = false;
2330   if (SchedModel->hasInstrSchedModel()) {
2331     unsigned LFactor = SchedModel->getLatencyFactor();
2332     OtherResLimited = (int)(OtherCount - (RemLatency * LFactor)) > (int)LFactor;
2333   }
2334   // Schedule aggressively for latency in PostRA mode. We don't check for
2335   // acyclic latency during PostRA, and highly out-of-order processors will
2336   // skip PostRA scheduling.
2337   if (!OtherResLimited) {
2338     if (IsPostRA || (RemLatency + CurrZone.getCurrCycle() > Rem.CriticalPath)) {
2339       Policy.ReduceLatency |= true;
2340       DEBUG(dbgs() << "  " << CurrZone.Available.getName()
2341             << " RemainingLatency " << RemLatency << " + "
2342             << CurrZone.getCurrCycle() << "c > CritPath "
2343             << Rem.CriticalPath << "\n");
2344     }
2345   }
2346   // If the same resource is limiting inside and outside the zone, do nothing.
2347   if (CurrZone.getZoneCritResIdx() == OtherCritIdx)
2348     return;
2349 
2350   DEBUG(
2351     if (CurrZone.isResourceLimited()) {
2352       dbgs() << "  " << CurrZone.Available.getName() << " ResourceLimited: "
2353              << SchedModel->getResourceName(CurrZone.getZoneCritResIdx())
2354              << "\n";
2355     }
2356     if (OtherResLimited)
2357       dbgs() << "  RemainingLimit: "
2358              << SchedModel->getResourceName(OtherCritIdx) << "\n";
2359     if (!CurrZone.isResourceLimited() && !OtherResLimited)
2360       dbgs() << "  Latency limited both directions.\n");
2361 
2362   if (CurrZone.isResourceLimited() && !Policy.ReduceResIdx)
2363     Policy.ReduceResIdx = CurrZone.getZoneCritResIdx();
2364 
2365   if (OtherResLimited)
2366     Policy.DemandResIdx = OtherCritIdx;
2367 }
2368 
2369 #ifndef NDEBUG
2370 const char *GenericSchedulerBase::getReasonStr(
2371   GenericSchedulerBase::CandReason Reason) {
2372   switch (Reason) {
2373   case NoCand:         return "NOCAND    ";
2374   case PhysRegCopy:    return "PREG-COPY";
2375   case RegExcess:      return "REG-EXCESS";
2376   case RegCritical:    return "REG-CRIT  ";
2377   case Stall:          return "STALL     ";
2378   case Cluster:        return "CLUSTER   ";
2379   case Weak:           return "WEAK      ";
2380   case RegMax:         return "REG-MAX   ";
2381   case ResourceReduce: return "RES-REDUCE";
2382   case ResourceDemand: return "RES-DEMAND";
2383   case TopDepthReduce: return "TOP-DEPTH ";
2384   case TopPathReduce:  return "TOP-PATH  ";
2385   case BotHeightReduce:return "BOT-HEIGHT";
2386   case BotPathReduce:  return "BOT-PATH  ";
2387   case NextDefUse:     return "DEF-USE   ";
2388   case NodeOrder:      return "ORDER     ";
2389   };
2390   llvm_unreachable("Unknown reason!");
2391 }
2392 
2393 void GenericSchedulerBase::traceCandidate(const SchedCandidate &Cand) {
2394   PressureChange P;
2395   unsigned ResIdx = 0;
2396   unsigned Latency = 0;
2397   switch (Cand.Reason) {
2398   default:
2399     break;
2400   case RegExcess:
2401     P = Cand.RPDelta.Excess;
2402     break;
2403   case RegCritical:
2404     P = Cand.RPDelta.CriticalMax;
2405     break;
2406   case RegMax:
2407     P = Cand.RPDelta.CurrentMax;
2408     break;
2409   case ResourceReduce:
2410     ResIdx = Cand.Policy.ReduceResIdx;
2411     break;
2412   case ResourceDemand:
2413     ResIdx = Cand.Policy.DemandResIdx;
2414     break;
2415   case TopDepthReduce:
2416     Latency = Cand.SU->getDepth();
2417     break;
2418   case TopPathReduce:
2419     Latency = Cand.SU->getHeight();
2420     break;
2421   case BotHeightReduce:
2422     Latency = Cand.SU->getHeight();
2423     break;
2424   case BotPathReduce:
2425     Latency = Cand.SU->getDepth();
2426     break;
2427   }
2428   dbgs() << "  Cand SU(" << Cand.SU->NodeNum << ") " << getReasonStr(Cand.Reason);
2429   if (P.isValid())
2430     dbgs() << " " << TRI->getRegPressureSetName(P.getPSet())
2431            << ":" << P.getUnitInc() << " ";
2432   else
2433     dbgs() << "      ";
2434   if (ResIdx)
2435     dbgs() << " " << SchedModel->getProcResource(ResIdx)->Name << " ";
2436   else
2437     dbgs() << "         ";
2438   if (Latency)
2439     dbgs() << " " << Latency << " cycles ";
2440   else
2441     dbgs() << "          ";
2442   dbgs() << '\n';
2443 }
2444 #endif
2445 
2446 /// Return true if this heuristic determines order.
2447 static bool tryLess(int TryVal, int CandVal,
2448                     GenericSchedulerBase::SchedCandidate &TryCand,
2449                     GenericSchedulerBase::SchedCandidate &Cand,
2450                     GenericSchedulerBase::CandReason Reason) {
2451   if (TryVal < CandVal) {
2452     TryCand.Reason = Reason;
2453     return true;
2454   }
2455   if (TryVal > CandVal) {
2456     if (Cand.Reason > Reason)
2457       Cand.Reason = Reason;
2458     return true;
2459   }
2460   Cand.setRepeat(Reason);
2461   return false;
2462 }
2463 
2464 static bool tryGreater(int TryVal, int CandVal,
2465                        GenericSchedulerBase::SchedCandidate &TryCand,
2466                        GenericSchedulerBase::SchedCandidate &Cand,
2467                        GenericSchedulerBase::CandReason Reason) {
2468   if (TryVal > CandVal) {
2469     TryCand.Reason = Reason;
2470     return true;
2471   }
2472   if (TryVal < CandVal) {
2473     if (Cand.Reason > Reason)
2474       Cand.Reason = Reason;
2475     return true;
2476   }
2477   Cand.setRepeat(Reason);
2478   return false;
2479 }
2480 
2481 static bool tryLatency(GenericSchedulerBase::SchedCandidate &TryCand,
2482                        GenericSchedulerBase::SchedCandidate &Cand,
2483                        SchedBoundary &Zone) {
2484   if (Zone.isTop()) {
2485     if (Cand.SU->getDepth() > Zone.getScheduledLatency()) {
2486       if (tryLess(TryCand.SU->getDepth(), Cand.SU->getDepth(),
2487                   TryCand, Cand, GenericSchedulerBase::TopDepthReduce))
2488         return true;
2489     }
2490     if (tryGreater(TryCand.SU->getHeight(), Cand.SU->getHeight(),
2491                    TryCand, Cand, GenericSchedulerBase::TopPathReduce))
2492       return true;
2493   }
2494   else {
2495     if (Cand.SU->getHeight() > Zone.getScheduledLatency()) {
2496       if (tryLess(TryCand.SU->getHeight(), Cand.SU->getHeight(),
2497                   TryCand, Cand, GenericSchedulerBase::BotHeightReduce))
2498         return true;
2499     }
2500     if (tryGreater(TryCand.SU->getDepth(), Cand.SU->getDepth(),
2501                    TryCand, Cand, GenericSchedulerBase::BotPathReduce))
2502       return true;
2503   }
2504   return false;
2505 }
2506 
2507 static void tracePick(const GenericSchedulerBase::SchedCandidate &Cand,
2508                       bool IsTop) {
2509   DEBUG(dbgs() << "Pick " << (IsTop ? "Top " : "Bot ")
2510         << GenericSchedulerBase::getReasonStr(Cand.Reason) << '\n');
2511 }
2512 
2513 void GenericScheduler::initialize(ScheduleDAGMI *dag) {
2514   assert(dag->hasVRegLiveness() &&
2515          "(PreRA)GenericScheduler needs vreg liveness");
2516   DAG = static_cast<ScheduleDAGMILive*>(dag);
2517   SchedModel = DAG->getSchedModel();
2518   TRI = DAG->TRI;
2519 
2520   Rem.init(DAG, SchedModel);
2521   Top.init(DAG, SchedModel, &Rem);
2522   Bot.init(DAG, SchedModel, &Rem);
2523 
2524   // Initialize resource counts.
2525 
2526   // Initialize the HazardRecognizers. If itineraries don't exist, are empty, or
2527   // are disabled, then these HazardRecs will be disabled.
2528   const InstrItineraryData *Itin = SchedModel->getInstrItineraries();
2529   if (!Top.HazardRec) {
2530     Top.HazardRec =
2531         DAG->MF.getSubtarget().getInstrInfo()->CreateTargetMIHazardRecognizer(
2532             Itin, DAG);
2533   }
2534   if (!Bot.HazardRec) {
2535     Bot.HazardRec =
2536         DAG->MF.getSubtarget().getInstrInfo()->CreateTargetMIHazardRecognizer(
2537             Itin, DAG);
2538   }
2539 }
2540 
2541 /// Initialize the per-region scheduling policy.
2542 void GenericScheduler::initPolicy(MachineBasicBlock::iterator Begin,
2543                                   MachineBasicBlock::iterator End,
2544                                   unsigned NumRegionInstrs) {
2545   const MachineFunction &MF = *Begin->getParent()->getParent();
2546   const TargetLowering *TLI = MF.getSubtarget().getTargetLowering();
2547 
2548   // Avoid setting up the register pressure tracker for small regions to save
2549   // compile time. As a rough heuristic, only track pressure when the number of
2550   // schedulable instructions exceeds half the integer register file.
2551   RegionPolicy.ShouldTrackPressure = true;
2552   for (unsigned VT = MVT::i32; VT > (unsigned)MVT::i1; --VT) {
2553     MVT::SimpleValueType LegalIntVT = (MVT::SimpleValueType)VT;
2554     if (TLI->isTypeLegal(LegalIntVT)) {
2555       unsigned NIntRegs = Context->RegClassInfo->getNumAllocatableRegs(
2556         TLI->getRegClassFor(LegalIntVT));
2557       RegionPolicy.ShouldTrackPressure = NumRegionInstrs > (NIntRegs / 2);
2558     }
2559   }
2560 
2561   // For generic targets, we default to bottom-up, because it's simpler and more
2562   // compile-time optimizations have been implemented in that direction.
2563   RegionPolicy.OnlyBottomUp = true;
2564 
2565   // Allow the subtarget to override default policy.
2566   MF.getSubtarget().overrideSchedPolicy(RegionPolicy, Begin, End,
2567                                         NumRegionInstrs);
2568 
2569   // After subtarget overrides, apply command line options.
2570   if (!EnableRegPressure)
2571     RegionPolicy.ShouldTrackPressure = false;
2572 
2573   // Check -misched-topdown/bottomup can force or unforce scheduling direction.
2574   // e.g. -misched-bottomup=false allows scheduling in both directions.
2575   assert((!ForceTopDown || !ForceBottomUp) &&
2576          "-misched-topdown incompatible with -misched-bottomup");
2577   if (ForceBottomUp.getNumOccurrences() > 0) {
2578     RegionPolicy.OnlyBottomUp = ForceBottomUp;
2579     if (RegionPolicy.OnlyBottomUp)
2580       RegionPolicy.OnlyTopDown = false;
2581   }
2582   if (ForceTopDown.getNumOccurrences() > 0) {
2583     RegionPolicy.OnlyTopDown = ForceTopDown;
2584     if (RegionPolicy.OnlyTopDown)
2585       RegionPolicy.OnlyBottomUp = false;
2586   }
2587 }
2588 
2589 void GenericScheduler::dumpPolicy() {
2590   dbgs() << "GenericScheduler RegionPolicy: "
2591          << " ShouldTrackPressure=" << RegionPolicy.ShouldTrackPressure
2592          << " OnlyTopDown=" << RegionPolicy.OnlyTopDown
2593          << " OnlyBottomUp=" << RegionPolicy.OnlyBottomUp
2594          << "\n";
2595 }
2596 
2597 /// Set IsAcyclicLatencyLimited if the acyclic path is longer than the cyclic
2598 /// critical path by more cycles than it takes to drain the instruction buffer.
2599 /// We estimate an upper bounds on in-flight instructions as:
2600 ///
2601 /// CyclesPerIteration = max( CyclicPath, Loop-Resource-Height )
2602 /// InFlightIterations = AcyclicPath / CyclesPerIteration
2603 /// InFlightResources = InFlightIterations * LoopResources
2604 ///
2605 /// TODO: Check execution resources in addition to IssueCount.
2606 void GenericScheduler::checkAcyclicLatency() {
2607   if (Rem.CyclicCritPath == 0 || Rem.CyclicCritPath >= Rem.CriticalPath)
2608     return;
2609 
2610   // Scaled number of cycles per loop iteration.
2611   unsigned IterCount =
2612     std::max(Rem.CyclicCritPath * SchedModel->getLatencyFactor(),
2613              Rem.RemIssueCount);
2614   // Scaled acyclic critical path.
2615   unsigned AcyclicCount = Rem.CriticalPath * SchedModel->getLatencyFactor();
2616   // InFlightCount = (AcyclicPath / IterCycles) * InstrPerLoop
2617   unsigned InFlightCount =
2618     (AcyclicCount * Rem.RemIssueCount + IterCount-1) / IterCount;
2619   unsigned BufferLimit =
2620     SchedModel->getMicroOpBufferSize() * SchedModel->getMicroOpFactor();
2621 
2622   Rem.IsAcyclicLatencyLimited = InFlightCount > BufferLimit;
2623 
2624   DEBUG(dbgs() << "IssueCycles="
2625         << Rem.RemIssueCount / SchedModel->getLatencyFactor() << "c "
2626         << "IterCycles=" << IterCount / SchedModel->getLatencyFactor()
2627         << "c NumIters=" << (AcyclicCount + IterCount-1) / IterCount
2628         << " InFlight=" << InFlightCount / SchedModel->getMicroOpFactor()
2629         << "m BufferLim=" << SchedModel->getMicroOpBufferSize() << "m\n";
2630         if (Rem.IsAcyclicLatencyLimited)
2631           dbgs() << "  ACYCLIC LATENCY LIMIT\n");
2632 }
2633 
2634 void GenericScheduler::registerRoots() {
2635   Rem.CriticalPath = DAG->ExitSU.getDepth();
2636 
2637   // Some roots may not feed into ExitSU. Check all of them in case.
2638   for (std::vector<SUnit*>::const_iterator
2639          I = Bot.Available.begin(), E = Bot.Available.end(); I != E; ++I) {
2640     if ((*I)->getDepth() > Rem.CriticalPath)
2641       Rem.CriticalPath = (*I)->getDepth();
2642   }
2643   DEBUG(dbgs() << "Critical Path(GS-RR ): " << Rem.CriticalPath << '\n');
2644   if (DumpCriticalPathLength) {
2645     errs() << "Critical Path(GS-RR ): " << Rem.CriticalPath << " \n";
2646   }
2647 
2648   if (EnableCyclicPath) {
2649     Rem.CyclicCritPath = DAG->computeCyclicCriticalPath();
2650     checkAcyclicLatency();
2651   }
2652 }
2653 
2654 static bool tryPressure(const PressureChange &TryP,
2655                         const PressureChange &CandP,
2656                         GenericSchedulerBase::SchedCandidate &TryCand,
2657                         GenericSchedulerBase::SchedCandidate &Cand,
2658                         GenericSchedulerBase::CandReason Reason,
2659                         const TargetRegisterInfo *TRI,
2660                         const MachineFunction &MF) {
2661   unsigned TryPSet = TryP.getPSetOrMax();
2662   unsigned CandPSet = CandP.getPSetOrMax();
2663   // If both candidates affect the same set, go with the smallest increase.
2664   if (TryPSet == CandPSet) {
2665     return tryLess(TryP.getUnitInc(), CandP.getUnitInc(), TryCand, Cand,
2666                    Reason);
2667   }
2668   // If one candidate decreases and the other increases, go with it.
2669   // Invalid candidates have UnitInc==0.
2670   if (tryGreater(TryP.getUnitInc() < 0, CandP.getUnitInc() < 0, TryCand, Cand,
2671                  Reason)) {
2672     return true;
2673   }
2674 
2675   int TryRank = TryP.isValid() ? TRI->getRegPressureSetScore(MF, TryPSet) :
2676                                  std::numeric_limits<int>::max();
2677 
2678   int CandRank = CandP.isValid() ? TRI->getRegPressureSetScore(MF, CandPSet) :
2679                                    std::numeric_limits<int>::max();
2680 
2681   // If the candidates are decreasing pressure, reverse priority.
2682   if (TryP.getUnitInc() < 0)
2683     std::swap(TryRank, CandRank);
2684   return tryGreater(TryRank, CandRank, TryCand, Cand, Reason);
2685 }
2686 
2687 static unsigned getWeakLeft(const SUnit *SU, bool isTop) {
2688   return (isTop) ? SU->WeakPredsLeft : SU->WeakSuccsLeft;
2689 }
2690 
2691 /// Minimize physical register live ranges. Regalloc wants them adjacent to
2692 /// their physreg def/use.
2693 ///
2694 /// FIXME: This is an unnecessary check on the critical path. Most are root/leaf
2695 /// copies which can be prescheduled. The rest (e.g. x86 MUL) could be bundled
2696 /// with the operation that produces or consumes the physreg. We'll do this when
2697 /// regalloc has support for parallel copies.
2698 static int biasPhysRegCopy(const SUnit *SU, bool isTop) {
2699   const MachineInstr *MI = SU->getInstr();
2700   if (!MI->isCopy())
2701     return 0;
2702 
2703   unsigned ScheduledOper = isTop ? 1 : 0;
2704   unsigned UnscheduledOper = isTop ? 0 : 1;
2705   // If we have already scheduled the physreg produce/consumer, immediately
2706   // schedule the copy.
2707   if (TargetRegisterInfo::isPhysicalRegister(
2708         MI->getOperand(ScheduledOper).getReg()))
2709     return 1;
2710   // If the physreg is at the boundary, defer it. Otherwise schedule it
2711   // immediately to free the dependent. We can hoist the copy later.
2712   bool AtBoundary = isTop ? !SU->NumSuccsLeft : !SU->NumPredsLeft;
2713   if (TargetRegisterInfo::isPhysicalRegister(
2714         MI->getOperand(UnscheduledOper).getReg()))
2715     return AtBoundary ? -1 : 1;
2716   return 0;
2717 }
2718 
2719 /// Apply a set of heursitics to a new candidate. Heuristics are currently
2720 /// hierarchical. This may be more efficient than a graduated cost model because
2721 /// we don't need to evaluate all aspects of the model for each node in the
2722 /// queue. But it's really done to make the heuristics easier to debug and
2723 /// statistically analyze.
2724 ///
2725 /// \param Cand provides the policy and current best candidate.
2726 /// \param TryCand refers to the next SUnit candidate, otherwise uninitialized.
2727 /// \param Zone describes the scheduled zone that we are extending.
2728 /// \param RPTracker describes reg pressure within the scheduled zone.
2729 /// \param TempTracker is a scratch pressure tracker to reuse in queries.
2730 void GenericScheduler::tryCandidate(SchedCandidate &Cand,
2731                                     SchedCandidate &TryCand,
2732                                     SchedBoundary &Zone,
2733                                     const RegPressureTracker &RPTracker,
2734                                     RegPressureTracker &TempTracker) {
2735 
2736   if (DAG->isTrackingPressure()) {
2737     // Always initialize TryCand's RPDelta.
2738     if (Zone.isTop()) {
2739       TempTracker.getMaxDownwardPressureDelta(
2740         TryCand.SU->getInstr(),
2741         TryCand.RPDelta,
2742         DAG->getRegionCriticalPSets(),
2743         DAG->getRegPressure().MaxSetPressure);
2744     }
2745     else {
2746       if (VerifyScheduling) {
2747         TempTracker.getMaxUpwardPressureDelta(
2748           TryCand.SU->getInstr(),
2749           &DAG->getPressureDiff(TryCand.SU),
2750           TryCand.RPDelta,
2751           DAG->getRegionCriticalPSets(),
2752           DAG->getRegPressure().MaxSetPressure);
2753       }
2754       else {
2755         RPTracker.getUpwardPressureDelta(
2756           TryCand.SU->getInstr(),
2757           DAG->getPressureDiff(TryCand.SU),
2758           TryCand.RPDelta,
2759           DAG->getRegionCriticalPSets(),
2760           DAG->getRegPressure().MaxSetPressure);
2761       }
2762     }
2763   }
2764   DEBUG(if (TryCand.RPDelta.Excess.isValid())
2765           dbgs() << "  Try  SU(" << TryCand.SU->NodeNum << ") "
2766                  << TRI->getRegPressureSetName(TryCand.RPDelta.Excess.getPSet())
2767                  << ":" << TryCand.RPDelta.Excess.getUnitInc() << "\n");
2768 
2769   // Initialize the candidate if needed.
2770   if (!Cand.isValid()) {
2771     TryCand.Reason = NodeOrder;
2772     return;
2773   }
2774 
2775   if (tryGreater(biasPhysRegCopy(TryCand.SU, Zone.isTop()),
2776                  biasPhysRegCopy(Cand.SU, Zone.isTop()),
2777                  TryCand, Cand, PhysRegCopy))
2778     return;
2779 
2780   // Avoid exceeding the target's limit.
2781   if (DAG->isTrackingPressure() && tryPressure(TryCand.RPDelta.Excess,
2782                                                Cand.RPDelta.Excess,
2783                                                TryCand, Cand, RegExcess, TRI,
2784                                                DAG->MF))
2785     return;
2786 
2787   // Avoid increasing the max critical pressure in the scheduled region.
2788   if (DAG->isTrackingPressure() && tryPressure(TryCand.RPDelta.CriticalMax,
2789                                                Cand.RPDelta.CriticalMax,
2790                                                TryCand, Cand, RegCritical, TRI,
2791                                                DAG->MF))
2792     return;
2793 
2794   // For loops that are acyclic path limited, aggressively schedule for latency.
2795   // This can result in very long dependence chains scheduled in sequence, so
2796   // once every cycle (when CurrMOps == 0), switch to normal heuristics.
2797   if (Rem.IsAcyclicLatencyLimited && !Zone.getCurrMOps()
2798       && tryLatency(TryCand, Cand, Zone))
2799     return;
2800 
2801   // Prioritize instructions that read unbuffered resources by stall cycles.
2802   if (tryLess(Zone.getLatencyStallCycles(TryCand.SU),
2803               Zone.getLatencyStallCycles(Cand.SU), TryCand, Cand, Stall))
2804     return;
2805 
2806   // Keep clustered nodes together to encourage downstream peephole
2807   // optimizations which may reduce resource requirements.
2808   //
2809   // This is a best effort to set things up for a post-RA pass. Optimizations
2810   // like generating loads of multiple registers should ideally be done within
2811   // the scheduler pass by combining the loads during DAG postprocessing.
2812   const SUnit *NextClusterSU =
2813     Zone.isTop() ? DAG->getNextClusterSucc() : DAG->getNextClusterPred();
2814   if (tryGreater(TryCand.SU == NextClusterSU, Cand.SU == NextClusterSU,
2815                  TryCand, Cand, Cluster))
2816     return;
2817 
2818   // Weak edges are for clustering and other constraints.
2819   if (tryLess(getWeakLeft(TryCand.SU, Zone.isTop()),
2820               getWeakLeft(Cand.SU, Zone.isTop()),
2821               TryCand, Cand, Weak)) {
2822     return;
2823   }
2824   // Avoid increasing the max pressure of the entire region.
2825   if (DAG->isTrackingPressure() && tryPressure(TryCand.RPDelta.CurrentMax,
2826                                                Cand.RPDelta.CurrentMax,
2827                                                TryCand, Cand, RegMax, TRI,
2828                                                DAG->MF))
2829     return;
2830 
2831   // Avoid critical resource consumption and balance the schedule.
2832   TryCand.initResourceDelta(DAG, SchedModel);
2833   if (tryLess(TryCand.ResDelta.CritResources, Cand.ResDelta.CritResources,
2834               TryCand, Cand, ResourceReduce))
2835     return;
2836   if (tryGreater(TryCand.ResDelta.DemandedResources,
2837                  Cand.ResDelta.DemandedResources,
2838                  TryCand, Cand, ResourceDemand))
2839     return;
2840 
2841   // Avoid serializing long latency dependence chains.
2842   // For acyclic path limited loops, latency was already checked above.
2843   if (!RegionPolicy.DisableLatencyHeuristic && Cand.Policy.ReduceLatency &&
2844       !Rem.IsAcyclicLatencyLimited && tryLatency(TryCand, Cand, Zone)) {
2845     return;
2846   }
2847 
2848   // Prefer immediate defs/users of the last scheduled instruction. This is a
2849   // local pressure avoidance strategy that also makes the machine code
2850   // readable.
2851   if (tryGreater(Zone.isNextSU(TryCand.SU), Zone.isNextSU(Cand.SU),
2852                  TryCand, Cand, NextDefUse))
2853     return;
2854 
2855   // Fall through to original instruction order.
2856   if ((Zone.isTop() && TryCand.SU->NodeNum < Cand.SU->NodeNum)
2857       || (!Zone.isTop() && TryCand.SU->NodeNum > Cand.SU->NodeNum)) {
2858     TryCand.Reason = NodeOrder;
2859   }
2860 }
2861 
2862 /// Pick the best candidate from the queue.
2863 ///
2864 /// TODO: getMaxPressureDelta results can be mostly cached for each SUnit during
2865 /// DAG building. To adjust for the current scheduling location we need to
2866 /// maintain the number of vreg uses remaining to be top-scheduled.
2867 void GenericScheduler::pickNodeFromQueue(SchedBoundary &Zone,
2868                                          const RegPressureTracker &RPTracker,
2869                                          SchedCandidate &Cand) {
2870   ReadyQueue &Q = Zone.Available;
2871 
2872   DEBUG(Q.dump());
2873 
2874   // getMaxPressureDelta temporarily modifies the tracker.
2875   RegPressureTracker &TempTracker = const_cast<RegPressureTracker&>(RPTracker);
2876 
2877   for (ReadyQueue::iterator I = Q.begin(), E = Q.end(); I != E; ++I) {
2878 
2879     SchedCandidate TryCand(Cand.Policy);
2880     TryCand.SU = *I;
2881     tryCandidate(Cand, TryCand, Zone, RPTracker, TempTracker);
2882     if (TryCand.Reason != NoCand) {
2883       // Initialize resource delta if needed in case future heuristics query it.
2884       if (TryCand.ResDelta == SchedResourceDelta())
2885         TryCand.initResourceDelta(DAG, SchedModel);
2886       Cand.setBest(TryCand);
2887       DEBUG(traceCandidate(Cand));
2888     }
2889   }
2890 }
2891 
2892 /// Pick the best candidate node from either the top or bottom queue.
2893 SUnit *GenericScheduler::pickNodeBidirectional(bool &IsTopNode) {
2894   // Schedule as far as possible in the direction of no choice. This is most
2895   // efficient, but also provides the best heuristics for CriticalPSets.
2896   if (SUnit *SU = Bot.pickOnlyChoice()) {
2897     IsTopNode = false;
2898     DEBUG(dbgs() << "Pick Bot ONLY1\n");
2899     return SU;
2900   }
2901   if (SUnit *SU = Top.pickOnlyChoice()) {
2902     IsTopNode = true;
2903     DEBUG(dbgs() << "Pick Top ONLY1\n");
2904     return SU;
2905   }
2906   CandPolicy NoPolicy;
2907   SchedCandidate BotCand(NoPolicy);
2908   SchedCandidate TopCand(NoPolicy);
2909   // Set the bottom-up policy based on the state of the current bottom zone and
2910   // the instructions outside the zone, including the top zone.
2911   setPolicy(BotCand.Policy, /*IsPostRA=*/false, Bot, &Top);
2912   // Set the top-down policy based on the state of the current top zone and
2913   // the instructions outside the zone, including the bottom zone.
2914   setPolicy(TopCand.Policy, /*IsPostRA=*/false, Top, &Bot);
2915 
2916   // Prefer bottom scheduling when heuristics are silent.
2917   pickNodeFromQueue(Bot, DAG->getBotRPTracker(), BotCand);
2918   assert(BotCand.Reason != NoCand && "failed to find the first candidate");
2919 
2920   // If either Q has a single candidate that provides the least increase in
2921   // Excess pressure, we can immediately schedule from that Q.
2922   //
2923   // RegionCriticalPSets summarizes the pressure within the scheduled region and
2924   // affects picking from either Q. If scheduling in one direction must
2925   // increase pressure for one of the excess PSets, then schedule in that
2926   // direction first to provide more freedom in the other direction.
2927   if ((BotCand.Reason == RegExcess && !BotCand.isRepeat(RegExcess))
2928       || (BotCand.Reason == RegCritical
2929           && !BotCand.isRepeat(RegCritical)))
2930   {
2931     IsTopNode = false;
2932     tracePick(BotCand, IsTopNode);
2933     return BotCand.SU;
2934   }
2935   // Check if the top Q has a better candidate.
2936   pickNodeFromQueue(Top, DAG->getTopRPTracker(), TopCand);
2937   assert(TopCand.Reason != NoCand && "failed to find the first candidate");
2938 
2939   // Choose the queue with the most important (lowest enum) reason.
2940   if (TopCand.Reason < BotCand.Reason) {
2941     IsTopNode = true;
2942     tracePick(TopCand, IsTopNode);
2943     return TopCand.SU;
2944   }
2945   // Otherwise prefer the bottom candidate, in node order if all else failed.
2946   IsTopNode = false;
2947   tracePick(BotCand, IsTopNode);
2948   return BotCand.SU;
2949 }
2950 
2951 /// Pick the best node to balance the schedule. Implements MachineSchedStrategy.
2952 SUnit *GenericScheduler::pickNode(bool &IsTopNode) {
2953   if (DAG->top() == DAG->bottom()) {
2954     assert(Top.Available.empty() && Top.Pending.empty() &&
2955            Bot.Available.empty() && Bot.Pending.empty() && "ReadyQ garbage");
2956     return nullptr;
2957   }
2958   SUnit *SU;
2959   do {
2960     if (RegionPolicy.OnlyTopDown) {
2961       SU = Top.pickOnlyChoice();
2962       if (!SU) {
2963         CandPolicy NoPolicy;
2964         SchedCandidate TopCand(NoPolicy);
2965         pickNodeFromQueue(Top, DAG->getTopRPTracker(), TopCand);
2966         assert(TopCand.Reason != NoCand && "failed to find a candidate");
2967         tracePick(TopCand, true);
2968         SU = TopCand.SU;
2969       }
2970       IsTopNode = true;
2971     }
2972     else if (RegionPolicy.OnlyBottomUp) {
2973       SU = Bot.pickOnlyChoice();
2974       if (!SU) {
2975         CandPolicy NoPolicy;
2976         SchedCandidate BotCand(NoPolicy);
2977         pickNodeFromQueue(Bot, DAG->getBotRPTracker(), BotCand);
2978         assert(BotCand.Reason != NoCand && "failed to find a candidate");
2979         tracePick(BotCand, false);
2980         SU = BotCand.SU;
2981       }
2982       IsTopNode = false;
2983     }
2984     else {
2985       SU = pickNodeBidirectional(IsTopNode);
2986     }
2987   } while (SU->isScheduled);
2988 
2989   if (SU->isTopReady())
2990     Top.removeReady(SU);
2991   if (SU->isBottomReady())
2992     Bot.removeReady(SU);
2993 
2994   DEBUG(dbgs() << "Scheduling SU(" << SU->NodeNum << ") " << *SU->getInstr());
2995   return SU;
2996 }
2997 
2998 void GenericScheduler::reschedulePhysRegCopies(SUnit *SU, bool isTop) {
2999 
3000   MachineBasicBlock::iterator InsertPos = SU->getInstr();
3001   if (!isTop)
3002     ++InsertPos;
3003   SmallVectorImpl<SDep> &Deps = isTop ? SU->Preds : SU->Succs;
3004 
3005   // Find already scheduled copies with a single physreg dependence and move
3006   // them just above the scheduled instruction.
3007   for (SmallVectorImpl<SDep>::iterator I = Deps.begin(), E = Deps.end();
3008        I != E; ++I) {
3009     if (I->getKind() != SDep::Data || !TRI->isPhysicalRegister(I->getReg()))
3010       continue;
3011     SUnit *DepSU = I->getSUnit();
3012     if (isTop ? DepSU->Succs.size() > 1 : DepSU->Preds.size() > 1)
3013       continue;
3014     MachineInstr *Copy = DepSU->getInstr();
3015     if (!Copy->isCopy())
3016       continue;
3017     DEBUG(dbgs() << "  Rescheduling physreg copy ";
3018           I->getSUnit()->dump(DAG));
3019     DAG->moveInstruction(Copy, InsertPos);
3020   }
3021 }
3022 
3023 /// Update the scheduler's state after scheduling a node. This is the same node
3024 /// that was just returned by pickNode(). However, ScheduleDAGMILive needs to
3025 /// update it's state based on the current cycle before MachineSchedStrategy
3026 /// does.
3027 ///
3028 /// FIXME: Eventually, we may bundle physreg copies rather than rescheduling
3029 /// them here. See comments in biasPhysRegCopy.
3030 void GenericScheduler::schedNode(SUnit *SU, bool IsTopNode) {
3031   if (IsTopNode) {
3032     SU->TopReadyCycle = std::max(SU->TopReadyCycle, Top.getCurrCycle());
3033     Top.bumpNode(SU);
3034     if (SU->hasPhysRegUses)
3035       reschedulePhysRegCopies(SU, true);
3036   }
3037   else {
3038     SU->BotReadyCycle = std::max(SU->BotReadyCycle, Bot.getCurrCycle());
3039     Bot.bumpNode(SU);
3040     if (SU->hasPhysRegDefs)
3041       reschedulePhysRegCopies(SU, false);
3042   }
3043 }
3044 
3045 /// Create the standard converging machine scheduler. This will be used as the
3046 /// default scheduler if the target does not set a default.
3047 static ScheduleDAGInstrs *createGenericSchedLive(MachineSchedContext *C) {
3048   ScheduleDAGMILive *DAG = new ScheduleDAGMILive(C, make_unique<GenericScheduler>(C));
3049   // Register DAG post-processors.
3050   //
3051   // FIXME: extend the mutation API to allow earlier mutations to instantiate
3052   // data and pass it to later mutations. Have a single mutation that gathers
3053   // the interesting nodes in one pass.
3054   DAG->addMutation(make_unique<CopyConstrain>(DAG->TII, DAG->TRI));
3055   if (EnableLoadCluster && DAG->TII->enableClusterLoads())
3056     DAG->addMutation(make_unique<LoadClusterMutation>(DAG->TII, DAG->TRI));
3057   if (EnableMacroFusion)
3058     DAG->addMutation(make_unique<MacroFusion>(*DAG->TII, *DAG->TRI));
3059   return DAG;
3060 }
3061 
3062 static MachineSchedRegistry
3063 GenericSchedRegistry("converge", "Standard converging scheduler.",
3064                      createGenericSchedLive);
3065 
3066 //===----------------------------------------------------------------------===//
3067 // PostGenericScheduler - Generic PostRA implementation of MachineSchedStrategy.
3068 //===----------------------------------------------------------------------===//
3069 
3070 void PostGenericScheduler::initialize(ScheduleDAGMI *Dag) {
3071   DAG = Dag;
3072   SchedModel = DAG->getSchedModel();
3073   TRI = DAG->TRI;
3074 
3075   Rem.init(DAG, SchedModel);
3076   Top.init(DAG, SchedModel, &Rem);
3077   BotRoots.clear();
3078 
3079   // Initialize the HazardRecognizers. If itineraries don't exist, are empty,
3080   // or are disabled, then these HazardRecs will be disabled.
3081   const InstrItineraryData *Itin = SchedModel->getInstrItineraries();
3082   if (!Top.HazardRec) {
3083     Top.HazardRec =
3084         DAG->MF.getSubtarget().getInstrInfo()->CreateTargetMIHazardRecognizer(
3085             Itin, DAG);
3086   }
3087 }
3088 
3089 
3090 void PostGenericScheduler::registerRoots() {
3091   Rem.CriticalPath = DAG->ExitSU.getDepth();
3092 
3093   // Some roots may not feed into ExitSU. Check all of them in case.
3094   for (SmallVectorImpl<SUnit*>::const_iterator
3095          I = BotRoots.begin(), E = BotRoots.end(); I != E; ++I) {
3096     if ((*I)->getDepth() > Rem.CriticalPath)
3097       Rem.CriticalPath = (*I)->getDepth();
3098   }
3099   DEBUG(dbgs() << "Critical Path: (PGS-RR) " << Rem.CriticalPath << '\n');
3100   if (DumpCriticalPathLength) {
3101     errs() << "Critical Path(PGS-RR ): " << Rem.CriticalPath << " \n";
3102   }
3103 }
3104 
3105 /// Apply a set of heursitics to a new candidate for PostRA scheduling.
3106 ///
3107 /// \param Cand provides the policy and current best candidate.
3108 /// \param TryCand refers to the next SUnit candidate, otherwise uninitialized.
3109 void PostGenericScheduler::tryCandidate(SchedCandidate &Cand,
3110                                         SchedCandidate &TryCand) {
3111 
3112   // Initialize the candidate if needed.
3113   if (!Cand.isValid()) {
3114     TryCand.Reason = NodeOrder;
3115     return;
3116   }
3117 
3118   // Prioritize instructions that read unbuffered resources by stall cycles.
3119   if (tryLess(Top.getLatencyStallCycles(TryCand.SU),
3120               Top.getLatencyStallCycles(Cand.SU), TryCand, Cand, Stall))
3121     return;
3122 
3123   // Avoid critical resource consumption and balance the schedule.
3124   if (tryLess(TryCand.ResDelta.CritResources, Cand.ResDelta.CritResources,
3125               TryCand, Cand, ResourceReduce))
3126     return;
3127   if (tryGreater(TryCand.ResDelta.DemandedResources,
3128                  Cand.ResDelta.DemandedResources,
3129                  TryCand, Cand, ResourceDemand))
3130     return;
3131 
3132   // Avoid serializing long latency dependence chains.
3133   if (Cand.Policy.ReduceLatency && tryLatency(TryCand, Cand, Top)) {
3134     return;
3135   }
3136 
3137   // Fall through to original instruction order.
3138   if (TryCand.SU->NodeNum < Cand.SU->NodeNum)
3139     TryCand.Reason = NodeOrder;
3140 }
3141 
3142 void PostGenericScheduler::pickNodeFromQueue(SchedCandidate &Cand) {
3143   ReadyQueue &Q = Top.Available;
3144 
3145   DEBUG(Q.dump());
3146 
3147   for (ReadyQueue::iterator I = Q.begin(), E = Q.end(); I != E; ++I) {
3148     SchedCandidate TryCand(Cand.Policy);
3149     TryCand.SU = *I;
3150     TryCand.initResourceDelta(DAG, SchedModel);
3151     tryCandidate(Cand, TryCand);
3152     if (TryCand.Reason != NoCand) {
3153       Cand.setBest(TryCand);
3154       DEBUG(traceCandidate(Cand));
3155     }
3156   }
3157 }
3158 
3159 /// Pick the next node to schedule.
3160 SUnit *PostGenericScheduler::pickNode(bool &IsTopNode) {
3161   if (DAG->top() == DAG->bottom()) {
3162     assert(Top.Available.empty() && Top.Pending.empty() && "ReadyQ garbage");
3163     return nullptr;
3164   }
3165   SUnit *SU;
3166   do {
3167     SU = Top.pickOnlyChoice();
3168     if (!SU) {
3169       CandPolicy NoPolicy;
3170       SchedCandidate TopCand(NoPolicy);
3171       // Set the top-down policy based on the state of the current top zone and
3172       // the instructions outside the zone, including the bottom zone.
3173       setPolicy(TopCand.Policy, /*IsPostRA=*/true, Top, nullptr);
3174       pickNodeFromQueue(TopCand);
3175       assert(TopCand.Reason != NoCand && "failed to find a candidate");
3176       tracePick(TopCand, true);
3177       SU = TopCand.SU;
3178     }
3179   } while (SU->isScheduled);
3180 
3181   IsTopNode = true;
3182   Top.removeReady(SU);
3183 
3184   DEBUG(dbgs() << "Scheduling SU(" << SU->NodeNum << ") " << *SU->getInstr());
3185   return SU;
3186 }
3187 
3188 /// Called after ScheduleDAGMI has scheduled an instruction and updated
3189 /// scheduled/remaining flags in the DAG nodes.
3190 void PostGenericScheduler::schedNode(SUnit *SU, bool IsTopNode) {
3191   SU->TopReadyCycle = std::max(SU->TopReadyCycle, Top.getCurrCycle());
3192   Top.bumpNode(SU);
3193 }
3194 
3195 /// Create a generic scheduler with no vreg liveness or DAG mutation passes.
3196 static ScheduleDAGInstrs *createGenericSchedPostRA(MachineSchedContext *C) {
3197   return new ScheduleDAGMI(C, make_unique<PostGenericScheduler>(C), /*IsPostRA=*/true);
3198 }
3199 
3200 //===----------------------------------------------------------------------===//
3201 // ILP Scheduler. Currently for experimental analysis of heuristics.
3202 //===----------------------------------------------------------------------===//
3203 
3204 namespace {
3205 /// \brief Order nodes by the ILP metric.
3206 struct ILPOrder {
3207   const SchedDFSResult *DFSResult;
3208   const BitVector *ScheduledTrees;
3209   bool MaximizeILP;
3210 
3211   ILPOrder(bool MaxILP)
3212     : DFSResult(nullptr), ScheduledTrees(nullptr), MaximizeILP(MaxILP) {}
3213 
3214   /// \brief Apply a less-than relation on node priority.
3215   ///
3216   /// (Return true if A comes after B in the Q.)
3217   bool operator()(const SUnit *A, const SUnit *B) const {
3218     unsigned SchedTreeA = DFSResult->getSubtreeID(A);
3219     unsigned SchedTreeB = DFSResult->getSubtreeID(B);
3220     if (SchedTreeA != SchedTreeB) {
3221       // Unscheduled trees have lower priority.
3222       if (ScheduledTrees->test(SchedTreeA) != ScheduledTrees->test(SchedTreeB))
3223         return ScheduledTrees->test(SchedTreeB);
3224 
3225       // Trees with shallower connections have have lower priority.
3226       if (DFSResult->getSubtreeLevel(SchedTreeA)
3227           != DFSResult->getSubtreeLevel(SchedTreeB)) {
3228         return DFSResult->getSubtreeLevel(SchedTreeA)
3229           < DFSResult->getSubtreeLevel(SchedTreeB);
3230       }
3231     }
3232     if (MaximizeILP)
3233       return DFSResult->getILP(A) < DFSResult->getILP(B);
3234     else
3235       return DFSResult->getILP(A) > DFSResult->getILP(B);
3236   }
3237 };
3238 
3239 /// \brief Schedule based on the ILP metric.
3240 class ILPScheduler : public MachineSchedStrategy {
3241   ScheduleDAGMILive *DAG;
3242   ILPOrder Cmp;
3243 
3244   std::vector<SUnit*> ReadyQ;
3245 public:
3246   ILPScheduler(bool MaximizeILP): DAG(nullptr), Cmp(MaximizeILP) {}
3247 
3248   void initialize(ScheduleDAGMI *dag) override {
3249     assert(dag->hasVRegLiveness() && "ILPScheduler needs vreg liveness");
3250     DAG = static_cast<ScheduleDAGMILive*>(dag);
3251     DAG->computeDFSResult();
3252     Cmp.DFSResult = DAG->getDFSResult();
3253     Cmp.ScheduledTrees = &DAG->getScheduledTrees();
3254     ReadyQ.clear();
3255   }
3256 
3257   void registerRoots() override {
3258     // Restore the heap in ReadyQ with the updated DFS results.
3259     std::make_heap(ReadyQ.begin(), ReadyQ.end(), Cmp);
3260   }
3261 
3262   /// Implement MachineSchedStrategy interface.
3263   /// -----------------------------------------
3264 
3265   /// Callback to select the highest priority node from the ready Q.
3266   SUnit *pickNode(bool &IsTopNode) override {
3267     if (ReadyQ.empty()) return nullptr;
3268     std::pop_heap(ReadyQ.begin(), ReadyQ.end(), Cmp);
3269     SUnit *SU = ReadyQ.back();
3270     ReadyQ.pop_back();
3271     IsTopNode = false;
3272     DEBUG(dbgs() << "Pick node " << "SU(" << SU->NodeNum << ") "
3273           << " ILP: " << DAG->getDFSResult()->getILP(SU)
3274           << " Tree: " << DAG->getDFSResult()->getSubtreeID(SU) << " @"
3275           << DAG->getDFSResult()->getSubtreeLevel(
3276             DAG->getDFSResult()->getSubtreeID(SU)) << '\n'
3277           << "Scheduling " << *SU->getInstr());
3278     return SU;
3279   }
3280 
3281   /// \brief Scheduler callback to notify that a new subtree is scheduled.
3282   void scheduleTree(unsigned SubtreeID) override {
3283     std::make_heap(ReadyQ.begin(), ReadyQ.end(), Cmp);
3284   }
3285 
3286   /// Callback after a node is scheduled. Mark a newly scheduled tree, notify
3287   /// DFSResults, and resort the priority Q.
3288   void schedNode(SUnit *SU, bool IsTopNode) override {
3289     assert(!IsTopNode && "SchedDFSResult needs bottom-up");
3290   }
3291 
3292   void releaseTopNode(SUnit *) override { /*only called for top roots*/ }
3293 
3294   void releaseBottomNode(SUnit *SU) override {
3295     ReadyQ.push_back(SU);
3296     std::push_heap(ReadyQ.begin(), ReadyQ.end(), Cmp);
3297   }
3298 };
3299 } // namespace
3300 
3301 static ScheduleDAGInstrs *createILPMaxScheduler(MachineSchedContext *C) {
3302   return new ScheduleDAGMILive(C, make_unique<ILPScheduler>(true));
3303 }
3304 static ScheduleDAGInstrs *createILPMinScheduler(MachineSchedContext *C) {
3305   return new ScheduleDAGMILive(C, make_unique<ILPScheduler>(false));
3306 }
3307 static MachineSchedRegistry ILPMaxRegistry(
3308   "ilpmax", "Schedule bottom-up for max ILP", createILPMaxScheduler);
3309 static MachineSchedRegistry ILPMinRegistry(
3310   "ilpmin", "Schedule bottom-up for min ILP", createILPMinScheduler);
3311 
3312 //===----------------------------------------------------------------------===//
3313 // Machine Instruction Shuffler for Correctness Testing
3314 //===----------------------------------------------------------------------===//
3315 
3316 #ifndef NDEBUG
3317 namespace {
3318 /// Apply a less-than relation on the node order, which corresponds to the
3319 /// instruction order prior to scheduling. IsReverse implements greater-than.
3320 template<bool IsReverse>
3321 struct SUnitOrder {
3322   bool operator()(SUnit *A, SUnit *B) const {
3323     if (IsReverse)
3324       return A->NodeNum > B->NodeNum;
3325     else
3326       return A->NodeNum < B->NodeNum;
3327   }
3328 };
3329 
3330 /// Reorder instructions as much as possible.
3331 class InstructionShuffler : public MachineSchedStrategy {
3332   bool IsAlternating;
3333   bool IsTopDown;
3334 
3335   // Using a less-than relation (SUnitOrder<false>) for the TopQ priority
3336   // gives nodes with a higher number higher priority causing the latest
3337   // instructions to be scheduled first.
3338   PriorityQueue<SUnit*, std::vector<SUnit*>, SUnitOrder<false> >
3339     TopQ;
3340   // When scheduling bottom-up, use greater-than as the queue priority.
3341   PriorityQueue<SUnit*, std::vector<SUnit*>, SUnitOrder<true> >
3342     BottomQ;
3343 public:
3344   InstructionShuffler(bool alternate, bool topdown)
3345     : IsAlternating(alternate), IsTopDown(topdown) {}
3346 
3347   void initialize(ScheduleDAGMI*) override {
3348     TopQ.clear();
3349     BottomQ.clear();
3350   }
3351 
3352   /// Implement MachineSchedStrategy interface.
3353   /// -----------------------------------------
3354 
3355   SUnit *pickNode(bool &IsTopNode) override {
3356     SUnit *SU;
3357     if (IsTopDown) {
3358       do {
3359         if (TopQ.empty()) return nullptr;
3360         SU = TopQ.top();
3361         TopQ.pop();
3362       } while (SU->isScheduled);
3363       IsTopNode = true;
3364     }
3365     else {
3366       do {
3367         if (BottomQ.empty()) return nullptr;
3368         SU = BottomQ.top();
3369         BottomQ.pop();
3370       } while (SU->isScheduled);
3371       IsTopNode = false;
3372     }
3373     if (IsAlternating)
3374       IsTopDown = !IsTopDown;
3375     return SU;
3376   }
3377 
3378   void schedNode(SUnit *SU, bool IsTopNode) override {}
3379 
3380   void releaseTopNode(SUnit *SU) override {
3381     TopQ.push(SU);
3382   }
3383   void releaseBottomNode(SUnit *SU) override {
3384     BottomQ.push(SU);
3385   }
3386 };
3387 } // namespace
3388 
3389 static ScheduleDAGInstrs *createInstructionShuffler(MachineSchedContext *C) {
3390   bool Alternate = !ForceTopDown && !ForceBottomUp;
3391   bool TopDown = !ForceBottomUp;
3392   assert((TopDown || !ForceTopDown) &&
3393          "-misched-topdown incompatible with -misched-bottomup");
3394   return new ScheduleDAGMILive(C, make_unique<InstructionShuffler>(Alternate, TopDown));
3395 }
3396 static MachineSchedRegistry ShufflerRegistry(
3397   "shuffle", "Shuffle machine instructions alternating directions",
3398   createInstructionShuffler);
3399 #endif // !NDEBUG
3400 
3401 //===----------------------------------------------------------------------===//
3402 // GraphWriter support for ScheduleDAGMILive.
3403 //===----------------------------------------------------------------------===//
3404 
3405 #ifndef NDEBUG
3406 namespace llvm {
3407 
3408 template<> struct GraphTraits<
3409   ScheduleDAGMI*> : public GraphTraits<ScheduleDAG*> {};
3410 
3411 template<>
3412 struct DOTGraphTraits<ScheduleDAGMI*> : public DefaultDOTGraphTraits {
3413 
3414   DOTGraphTraits (bool isSimple=false) : DefaultDOTGraphTraits(isSimple) {}
3415 
3416   static std::string getGraphName(const ScheduleDAG *G) {
3417     return G->MF.getName();
3418   }
3419 
3420   static bool renderGraphFromBottomUp() {
3421     return true;
3422   }
3423 
3424   static bool isNodeHidden(const SUnit *Node) {
3425     if (ViewMISchedCutoff == 0)
3426       return false;
3427     return (Node->Preds.size() > ViewMISchedCutoff
3428          || Node->Succs.size() > ViewMISchedCutoff);
3429   }
3430 
3431   /// If you want to override the dot attributes printed for a particular
3432   /// edge, override this method.
3433   static std::string getEdgeAttributes(const SUnit *Node,
3434                                        SUnitIterator EI,
3435                                        const ScheduleDAG *Graph) {
3436     if (EI.isArtificialDep())
3437       return "color=cyan,style=dashed";
3438     if (EI.isCtrlDep())
3439       return "color=blue,style=dashed";
3440     return "";
3441   }
3442 
3443   static std::string getNodeLabel(const SUnit *SU, const ScheduleDAG *G) {
3444     std::string Str;
3445     raw_string_ostream SS(Str);
3446     const ScheduleDAGMI *DAG = static_cast<const ScheduleDAGMI*>(G);
3447     const SchedDFSResult *DFS = DAG->hasVRegLiveness() ?
3448       static_cast<const ScheduleDAGMILive*>(G)->getDFSResult() : nullptr;
3449     SS << "SU:" << SU->NodeNum;
3450     if (DFS)
3451       SS << " I:" << DFS->getNumInstrs(SU);
3452     return SS.str();
3453   }
3454   static std::string getNodeDescription(const SUnit *SU, const ScheduleDAG *G) {
3455     return G->getGraphNodeLabel(SU);
3456   }
3457 
3458   static std::string getNodeAttributes(const SUnit *N, const ScheduleDAG *G) {
3459     std::string Str("shape=Mrecord");
3460     const ScheduleDAGMI *DAG = static_cast<const ScheduleDAGMI*>(G);
3461     const SchedDFSResult *DFS = DAG->hasVRegLiveness() ?
3462       static_cast<const ScheduleDAGMILive*>(G)->getDFSResult() : nullptr;
3463     if (DFS) {
3464       Str += ",style=filled,fillcolor=\"#";
3465       Str += DOT::getColorString(DFS->getSubtreeID(N));
3466       Str += '"';
3467     }
3468     return Str;
3469   }
3470 };
3471 } // namespace llvm
3472 #endif // NDEBUG
3473 
3474 /// viewGraph - Pop up a ghostview window with the reachable parts of the DAG
3475 /// rendered using 'dot'.
3476 ///
3477 void ScheduleDAGMI::viewGraph(const Twine &Name, const Twine &Title) {
3478 #ifndef NDEBUG
3479   ViewGraph(this, Name, false, Title);
3480 #else
3481   errs() << "ScheduleDAGMI::viewGraph is only available in debug builds on "
3482          << "systems with Graphviz or gv!\n";
3483 #endif  // NDEBUG
3484 }
3485 
3486 /// Out-of-line implementation with no arguments is handy for gdb.
3487 void ScheduleDAGMI::viewGraph() {
3488   viewGraph(getDAGName(), "Scheduling-Units Graph for " + getDAGName());
3489 }
3490