1 //===- MachineScheduler.cpp - Machine Instruction Scheduler ---------------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // 10 // MachineScheduler schedules machine instructions after phi elimination. It 11 // preserves LiveIntervals so it can be invoked before register allocation. 12 // 13 //===----------------------------------------------------------------------===// 14 15 #define DEBUG_TYPE "misched" 16 17 #include "RegisterClassInfo.h" 18 #include "RegisterPressure.h" 19 #include "llvm/CodeGen/LiveIntervalAnalysis.h" 20 #include "llvm/CodeGen/MachineScheduler.h" 21 #include "llvm/CodeGen/Passes.h" 22 #include "llvm/CodeGen/ScheduleDAGInstrs.h" 23 #include "llvm/CodeGen/ScheduleHazardRecognizer.h" 24 #include "llvm/Analysis/AliasAnalysis.h" 25 #include "llvm/Target/TargetInstrInfo.h" 26 #include "llvm/Support/CommandLine.h" 27 #include "llvm/Support/Debug.h" 28 #include "llvm/Support/ErrorHandling.h" 29 #include "llvm/Support/raw_ostream.h" 30 #include "llvm/ADT/OwningPtr.h" 31 #include "llvm/ADT/PriorityQueue.h" 32 33 #include <queue> 34 35 using namespace llvm; 36 37 static cl::opt<bool> ForceTopDown("misched-topdown", cl::Hidden, 38 cl::desc("Force top-down list scheduling")); 39 static cl::opt<bool> ForceBottomUp("misched-bottomup", cl::Hidden, 40 cl::desc("Force bottom-up list scheduling")); 41 42 #ifndef NDEBUG 43 static cl::opt<bool> ViewMISchedDAGs("view-misched-dags", cl::Hidden, 44 cl::desc("Pop up a window to show MISched dags after they are processed")); 45 46 static cl::opt<unsigned> MISchedCutoff("misched-cutoff", cl::Hidden, 47 cl::desc("Stop scheduling after N instructions"), cl::init(~0U)); 48 #else 49 static bool ViewMISchedDAGs = false; 50 #endif // NDEBUG 51 52 //===----------------------------------------------------------------------===// 53 // Machine Instruction Scheduling Pass and Registry 54 //===----------------------------------------------------------------------===// 55 56 MachineSchedContext::MachineSchedContext(): 57 MF(0), MLI(0), MDT(0), PassConfig(0), AA(0), LIS(0) { 58 RegClassInfo = new RegisterClassInfo(); 59 } 60 61 MachineSchedContext::~MachineSchedContext() { 62 delete RegClassInfo; 63 } 64 65 namespace { 66 /// MachineScheduler runs after coalescing and before register allocation. 67 class MachineScheduler : public MachineSchedContext, 68 public MachineFunctionPass { 69 public: 70 MachineScheduler(); 71 72 virtual void getAnalysisUsage(AnalysisUsage &AU) const; 73 74 virtual void releaseMemory() {} 75 76 virtual bool runOnMachineFunction(MachineFunction&); 77 78 virtual void print(raw_ostream &O, const Module* = 0) const; 79 80 static char ID; // Class identification, replacement for typeinfo 81 }; 82 } // namespace 83 84 char MachineScheduler::ID = 0; 85 86 char &llvm::MachineSchedulerID = MachineScheduler::ID; 87 88 INITIALIZE_PASS_BEGIN(MachineScheduler, "misched", 89 "Machine Instruction Scheduler", false, false) 90 INITIALIZE_AG_DEPENDENCY(AliasAnalysis) 91 INITIALIZE_PASS_DEPENDENCY(SlotIndexes) 92 INITIALIZE_PASS_DEPENDENCY(LiveIntervals) 93 INITIALIZE_PASS_END(MachineScheduler, "misched", 94 "Machine Instruction Scheduler", false, false) 95 96 MachineScheduler::MachineScheduler() 97 : MachineFunctionPass(ID) { 98 initializeMachineSchedulerPass(*PassRegistry::getPassRegistry()); 99 } 100 101 void MachineScheduler::getAnalysisUsage(AnalysisUsage &AU) const { 102 AU.setPreservesCFG(); 103 AU.addRequiredID(MachineDominatorsID); 104 AU.addRequired<MachineLoopInfo>(); 105 AU.addRequired<AliasAnalysis>(); 106 AU.addRequired<TargetPassConfig>(); 107 AU.addRequired<SlotIndexes>(); 108 AU.addPreserved<SlotIndexes>(); 109 AU.addRequired<LiveIntervals>(); 110 AU.addPreserved<LiveIntervals>(); 111 MachineFunctionPass::getAnalysisUsage(AU); 112 } 113 114 MachinePassRegistry MachineSchedRegistry::Registry; 115 116 /// A dummy default scheduler factory indicates whether the scheduler 117 /// is overridden on the command line. 118 static ScheduleDAGInstrs *useDefaultMachineSched(MachineSchedContext *C) { 119 return 0; 120 } 121 122 /// MachineSchedOpt allows command line selection of the scheduler. 123 static cl::opt<MachineSchedRegistry::ScheduleDAGCtor, false, 124 RegisterPassParser<MachineSchedRegistry> > 125 MachineSchedOpt("misched", 126 cl::init(&useDefaultMachineSched), cl::Hidden, 127 cl::desc("Machine instruction scheduler to use")); 128 129 static MachineSchedRegistry 130 DefaultSchedRegistry("default", "Use the target's default scheduler choice.", 131 useDefaultMachineSched); 132 133 /// Forward declare the standard machine scheduler. This will be used as the 134 /// default scheduler if the target does not set a default. 135 static ScheduleDAGInstrs *createConvergingSched(MachineSchedContext *C); 136 137 138 /// Decrement this iterator until reaching the top or a non-debug instr. 139 static MachineBasicBlock::iterator 140 priorNonDebug(MachineBasicBlock::iterator I, MachineBasicBlock::iterator Beg) { 141 assert(I != Beg && "reached the top of the region, cannot decrement"); 142 while (--I != Beg) { 143 if (!I->isDebugValue()) 144 break; 145 } 146 return I; 147 } 148 149 /// If this iterator is a debug value, increment until reaching the End or a 150 /// non-debug instruction. 151 static MachineBasicBlock::iterator 152 nextIfDebug(MachineBasicBlock::iterator I, MachineBasicBlock::iterator End) { 153 for(; I != End; ++I) { 154 if (!I->isDebugValue()) 155 break; 156 } 157 return I; 158 } 159 160 /// Top-level MachineScheduler pass driver. 161 /// 162 /// Visit blocks in function order. Divide each block into scheduling regions 163 /// and visit them bottom-up. Visiting regions bottom-up is not required, but is 164 /// consistent with the DAG builder, which traverses the interior of the 165 /// scheduling regions bottom-up. 166 /// 167 /// This design avoids exposing scheduling boundaries to the DAG builder, 168 /// simplifying the DAG builder's support for "special" target instructions. 169 /// At the same time the design allows target schedulers to operate across 170 /// scheduling boundaries, for example to bundle the boudary instructions 171 /// without reordering them. This creates complexity, because the target 172 /// scheduler must update the RegionBegin and RegionEnd positions cached by 173 /// ScheduleDAGInstrs whenever adding or removing instructions. A much simpler 174 /// design would be to split blocks at scheduling boundaries, but LLVM has a 175 /// general bias against block splitting purely for implementation simplicity. 176 bool MachineScheduler::runOnMachineFunction(MachineFunction &mf) { 177 DEBUG(dbgs() << "Before MISsched:\n"; mf.print(dbgs())); 178 179 // Initialize the context of the pass. 180 MF = &mf; 181 MLI = &getAnalysis<MachineLoopInfo>(); 182 MDT = &getAnalysis<MachineDominatorTree>(); 183 PassConfig = &getAnalysis<TargetPassConfig>(); 184 AA = &getAnalysis<AliasAnalysis>(); 185 186 LIS = &getAnalysis<LiveIntervals>(); 187 const TargetInstrInfo *TII = MF->getTarget().getInstrInfo(); 188 189 RegClassInfo->runOnMachineFunction(*MF); 190 191 // Select the scheduler, or set the default. 192 MachineSchedRegistry::ScheduleDAGCtor Ctor = MachineSchedOpt; 193 if (Ctor == useDefaultMachineSched) { 194 // Get the default scheduler set by the target. 195 Ctor = MachineSchedRegistry::getDefault(); 196 if (!Ctor) { 197 Ctor = createConvergingSched; 198 MachineSchedRegistry::setDefault(Ctor); 199 } 200 } 201 // Instantiate the selected scheduler. 202 OwningPtr<ScheduleDAGInstrs> Scheduler(Ctor(this)); 203 204 // Visit all machine basic blocks. 205 // 206 // TODO: Visit blocks in global postorder or postorder within the bottom-up 207 // loop tree. Then we can optionally compute global RegPressure. 208 for (MachineFunction::iterator MBB = MF->begin(), MBBEnd = MF->end(); 209 MBB != MBBEnd; ++MBB) { 210 211 Scheduler->startBlock(MBB); 212 213 // Break the block into scheduling regions [I, RegionEnd), and schedule each 214 // region as soon as it is discovered. RegionEnd points the the scheduling 215 // boundary at the bottom of the region. The DAG does not include RegionEnd, 216 // but the region does (i.e. the next RegionEnd is above the previous 217 // RegionBegin). If the current block has no terminator then RegionEnd == 218 // MBB->end() for the bottom region. 219 // 220 // The Scheduler may insert instructions during either schedule() or 221 // exitRegion(), even for empty regions. So the local iterators 'I' and 222 // 'RegionEnd' are invalid across these calls. 223 unsigned RemainingCount = MBB->size(); 224 for(MachineBasicBlock::iterator RegionEnd = MBB->end(); 225 RegionEnd != MBB->begin(); RegionEnd = Scheduler->begin()) { 226 227 // Avoid decrementing RegionEnd for blocks with no terminator. 228 if (RegionEnd != MBB->end() 229 || TII->isSchedulingBoundary(llvm::prior(RegionEnd), MBB, *MF)) { 230 --RegionEnd; 231 // Count the boundary instruction. 232 --RemainingCount; 233 } 234 235 // The next region starts above the previous region. Look backward in the 236 // instruction stream until we find the nearest boundary. 237 MachineBasicBlock::iterator I = RegionEnd; 238 for(;I != MBB->begin(); --I, --RemainingCount) { 239 if (TII->isSchedulingBoundary(llvm::prior(I), MBB, *MF)) 240 break; 241 } 242 // Notify the scheduler of the region, even if we may skip scheduling 243 // it. Perhaps it still needs to be bundled. 244 Scheduler->enterRegion(MBB, I, RegionEnd, RemainingCount); 245 246 // Skip empty scheduling regions (0 or 1 schedulable instructions). 247 if (I == RegionEnd || I == llvm::prior(RegionEnd)) { 248 // Close the current region. Bundle the terminator if needed. 249 // This invalidates 'RegionEnd' and 'I'. 250 Scheduler->exitRegion(); 251 continue; 252 } 253 DEBUG(dbgs() << "********** MI Scheduling **********\n"); 254 DEBUG(dbgs() << MF->getFunction()->getName() 255 << ":BB#" << MBB->getNumber() << "\n From: " << *I << " To: "; 256 if (RegionEnd != MBB->end()) dbgs() << *RegionEnd; 257 else dbgs() << "End"; 258 dbgs() << " Remaining: " << RemainingCount << "\n"); 259 260 // Schedule a region: possibly reorder instructions. 261 // This invalidates 'RegionEnd' and 'I'. 262 Scheduler->schedule(); 263 264 // Close the current region. 265 Scheduler->exitRegion(); 266 267 // Scheduling has invalidated the current iterator 'I'. Ask the 268 // scheduler for the top of it's scheduled region. 269 RegionEnd = Scheduler->begin(); 270 } 271 assert(RemainingCount == 0 && "Instruction count mismatch!"); 272 Scheduler->finishBlock(); 273 } 274 Scheduler->finalizeSchedule(); 275 DEBUG(LIS->print(dbgs())); 276 return true; 277 } 278 279 void MachineScheduler::print(raw_ostream &O, const Module* m) const { 280 // unimplemented 281 } 282 283 //===----------------------------------------------------------------------===// 284 // MachineSchedStrategy - Interface to a machine scheduling algorithm. 285 //===----------------------------------------------------------------------===// 286 287 namespace { 288 class ScheduleDAGMI; 289 290 /// MachineSchedStrategy - Interface used by ScheduleDAGMI to drive the selected 291 /// scheduling algorithm. 292 /// 293 /// If this works well and targets wish to reuse ScheduleDAGMI, we may expose it 294 /// in ScheduleDAGInstrs.h 295 class MachineSchedStrategy { 296 public: 297 virtual ~MachineSchedStrategy() {} 298 299 /// Initialize the strategy after building the DAG for a new region. 300 virtual void initialize(ScheduleDAGMI *DAG) = 0; 301 302 /// Pick the next node to schedule, or return NULL. Set IsTopNode to true to 303 /// schedule the node at the top of the unscheduled region. Otherwise it will 304 /// be scheduled at the bottom. 305 virtual SUnit *pickNode(bool &IsTopNode) = 0; 306 307 /// Notify MachineSchedStrategy that ScheduleDAGMI has scheduled a node. 308 virtual void schedNode(SUnit *SU, bool IsTopNode) = 0; 309 310 /// When all predecessor dependencies have been resolved, free this node for 311 /// top-down scheduling. 312 virtual void releaseTopNode(SUnit *SU) = 0; 313 /// When all successor dependencies have been resolved, free this node for 314 /// bottom-up scheduling. 315 virtual void releaseBottomNode(SUnit *SU) = 0; 316 }; 317 } // namespace 318 319 //===----------------------------------------------------------------------===// 320 // ScheduleDAGMI - Base class for MachineInstr scheduling with LiveIntervals 321 // preservation. 322 //===----------------------------------------------------------------------===// 323 324 namespace { 325 /// ScheduleDAGMI is an implementation of ScheduleDAGInstrs that schedules 326 /// machine instructions while updating LiveIntervals. 327 class ScheduleDAGMI : public ScheduleDAGInstrs { 328 AliasAnalysis *AA; 329 RegisterClassInfo *RegClassInfo; 330 MachineSchedStrategy *SchedImpl; 331 332 MachineBasicBlock::iterator LiveRegionEnd; 333 334 /// Register pressure in this region computed by buildSchedGraph. 335 IntervalPressure RegPressure; 336 RegPressureTracker RPTracker; 337 338 /// List of pressure sets that exceed the target's pressure limit before 339 /// scheduling, listed in increasing set ID order. Each pressure set is paired 340 /// with its max pressure in the currently scheduled regions. 341 std::vector<PressureElement> RegionCriticalPSets; 342 343 /// The top of the unscheduled zone. 344 MachineBasicBlock::iterator CurrentTop; 345 IntervalPressure TopPressure; 346 RegPressureTracker TopRPTracker; 347 348 /// The bottom of the unscheduled zone. 349 MachineBasicBlock::iterator CurrentBottom; 350 IntervalPressure BotPressure; 351 RegPressureTracker BotRPTracker; 352 353 /// The number of instructions scheduled so far. Used to cut off the 354 /// scheduler at the point determined by misched-cutoff. 355 unsigned NumInstrsScheduled; 356 public: 357 ScheduleDAGMI(MachineSchedContext *C, MachineSchedStrategy *S): 358 ScheduleDAGInstrs(*C->MF, *C->MLI, *C->MDT, /*IsPostRA=*/false, C->LIS), 359 AA(C->AA), RegClassInfo(C->RegClassInfo), SchedImpl(S), 360 RPTracker(RegPressure), CurrentTop(), TopRPTracker(TopPressure), 361 CurrentBottom(), BotRPTracker(BotPressure), NumInstrsScheduled(0) {} 362 363 ~ScheduleDAGMI() { 364 delete SchedImpl; 365 } 366 367 MachineBasicBlock::iterator top() const { return CurrentTop; } 368 MachineBasicBlock::iterator bottom() const { return CurrentBottom; } 369 370 /// Implement the ScheduleDAGInstrs interface for handling the next scheduling 371 /// region. This covers all instructions in a block, while schedule() may only 372 /// cover a subset. 373 void enterRegion(MachineBasicBlock *bb, 374 MachineBasicBlock::iterator begin, 375 MachineBasicBlock::iterator end, 376 unsigned endcount); 377 378 /// Implement ScheduleDAGInstrs interface for scheduling a sequence of 379 /// reorderable instructions. 380 void schedule(); 381 382 /// Get current register pressure for the top scheduled instructions. 383 const IntervalPressure &getTopPressure() const { return TopPressure; } 384 const RegPressureTracker &getTopRPTracker() const { return TopRPTracker; } 385 386 /// Get current register pressure for the bottom scheduled instructions. 387 const IntervalPressure &getBotPressure() const { return BotPressure; } 388 const RegPressureTracker &getBotRPTracker() const { return BotRPTracker; } 389 390 /// Get register pressure for the entire scheduling region before scheduling. 391 const IntervalPressure &getRegPressure() const { return RegPressure; } 392 393 const std::vector<PressureElement> &getRegionCriticalPSets() const { 394 return RegionCriticalPSets; 395 } 396 397 protected: 398 void initRegPressure(); 399 void updateScheduledPressure(std::vector<unsigned> NewMaxPressure); 400 401 void moveInstruction(MachineInstr *MI, MachineBasicBlock::iterator InsertPos); 402 bool checkSchedLimit(); 403 404 void releaseRoots(); 405 406 void releaseSucc(SUnit *SU, SDep *SuccEdge); 407 void releaseSuccessors(SUnit *SU); 408 void releasePred(SUnit *SU, SDep *PredEdge); 409 void releasePredecessors(SUnit *SU); 410 411 void placeDebugValues(); 412 }; 413 } // namespace 414 415 /// ReleaseSucc - Decrement the NumPredsLeft count of a successor. When 416 /// NumPredsLeft reaches zero, release the successor node. 417 /// 418 /// FIXME: Adjust SuccSU height based on MinLatency. 419 void ScheduleDAGMI::releaseSucc(SUnit *SU, SDep *SuccEdge) { 420 SUnit *SuccSU = SuccEdge->getSUnit(); 421 422 #ifndef NDEBUG 423 if (SuccSU->NumPredsLeft == 0) { 424 dbgs() << "*** Scheduling failed! ***\n"; 425 SuccSU->dump(this); 426 dbgs() << " has been released too many times!\n"; 427 llvm_unreachable(0); 428 } 429 #endif 430 --SuccSU->NumPredsLeft; 431 if (SuccSU->NumPredsLeft == 0 && SuccSU != &ExitSU) 432 SchedImpl->releaseTopNode(SuccSU); 433 } 434 435 /// releaseSuccessors - Call releaseSucc on each of SU's successors. 436 void ScheduleDAGMI::releaseSuccessors(SUnit *SU) { 437 for (SUnit::succ_iterator I = SU->Succs.begin(), E = SU->Succs.end(); 438 I != E; ++I) { 439 releaseSucc(SU, &*I); 440 } 441 } 442 443 /// ReleasePred - Decrement the NumSuccsLeft count of a predecessor. When 444 /// NumSuccsLeft reaches zero, release the predecessor node. 445 /// 446 /// FIXME: Adjust PredSU height based on MinLatency. 447 void ScheduleDAGMI::releasePred(SUnit *SU, SDep *PredEdge) { 448 SUnit *PredSU = PredEdge->getSUnit(); 449 450 #ifndef NDEBUG 451 if (PredSU->NumSuccsLeft == 0) { 452 dbgs() << "*** Scheduling failed! ***\n"; 453 PredSU->dump(this); 454 dbgs() << " has been released too many times!\n"; 455 llvm_unreachable(0); 456 } 457 #endif 458 --PredSU->NumSuccsLeft; 459 if (PredSU->NumSuccsLeft == 0 && PredSU != &EntrySU) 460 SchedImpl->releaseBottomNode(PredSU); 461 } 462 463 /// releasePredecessors - Call releasePred on each of SU's predecessors. 464 void ScheduleDAGMI::releasePredecessors(SUnit *SU) { 465 for (SUnit::pred_iterator I = SU->Preds.begin(), E = SU->Preds.end(); 466 I != E; ++I) { 467 releasePred(SU, &*I); 468 } 469 } 470 471 void ScheduleDAGMI::moveInstruction(MachineInstr *MI, 472 MachineBasicBlock::iterator InsertPos) { 473 // Advance RegionBegin if the first instruction moves down. 474 if (&*RegionBegin == MI) 475 ++RegionBegin; 476 477 // Update the instruction stream. 478 BB->splice(InsertPos, BB, MI); 479 480 // Update LiveIntervals 481 LIS->handleMove(MI); 482 483 // Recede RegionBegin if an instruction moves above the first. 484 if (RegionBegin == InsertPos) 485 RegionBegin = MI; 486 } 487 488 bool ScheduleDAGMI::checkSchedLimit() { 489 #ifndef NDEBUG 490 if (NumInstrsScheduled == MISchedCutoff && MISchedCutoff != ~0U) { 491 CurrentTop = CurrentBottom; 492 return false; 493 } 494 ++NumInstrsScheduled; 495 #endif 496 return true; 497 } 498 499 /// enterRegion - Called back from MachineScheduler::runOnMachineFunction after 500 /// crossing a scheduling boundary. [begin, end) includes all instructions in 501 /// the region, including the boundary itself and single-instruction regions 502 /// that don't get scheduled. 503 void ScheduleDAGMI::enterRegion(MachineBasicBlock *bb, 504 MachineBasicBlock::iterator begin, 505 MachineBasicBlock::iterator end, 506 unsigned endcount) 507 { 508 ScheduleDAGInstrs::enterRegion(bb, begin, end, endcount); 509 510 // For convenience remember the end of the liveness region. 511 LiveRegionEnd = 512 (RegionEnd == bb->end()) ? RegionEnd : llvm::next(RegionEnd); 513 } 514 515 // Setup the register pressure trackers for the top scheduled top and bottom 516 // scheduled regions. 517 void ScheduleDAGMI::initRegPressure() { 518 TopRPTracker.init(&MF, RegClassInfo, LIS, BB, RegionBegin); 519 BotRPTracker.init(&MF, RegClassInfo, LIS, BB, LiveRegionEnd); 520 521 // Close the RPTracker to finalize live ins. 522 RPTracker.closeRegion(); 523 524 DEBUG(RPTracker.getPressure().dump(TRI)); 525 526 // Initialize the live ins and live outs. 527 TopRPTracker.addLiveRegs(RPTracker.getPressure().LiveInRegs); 528 BotRPTracker.addLiveRegs(RPTracker.getPressure().LiveOutRegs); 529 530 // Close one end of the tracker so we can call 531 // getMaxUpward/DownwardPressureDelta before advancing across any 532 // instructions. This converts currently live regs into live ins/outs. 533 TopRPTracker.closeTop(); 534 BotRPTracker.closeBottom(); 535 536 // Account for liveness generated by the region boundary. 537 if (LiveRegionEnd != RegionEnd) 538 BotRPTracker.recede(); 539 540 assert(BotRPTracker.getPos() == RegionEnd && "Can't find the region bottom"); 541 542 // Cache the list of excess pressure sets in this region. This will also track 543 // the max pressure in the scheduled code for these sets. 544 RegionCriticalPSets.clear(); 545 std::vector<unsigned> RegionPressure = RPTracker.getPressure().MaxSetPressure; 546 for (unsigned i = 0, e = RegionPressure.size(); i < e; ++i) { 547 unsigned Limit = TRI->getRegPressureSetLimit(i); 548 if (RegionPressure[i] > Limit) 549 RegionCriticalPSets.push_back(PressureElement(i, 0)); 550 } 551 DEBUG(dbgs() << "Excess PSets: "; 552 for (unsigned i = 0, e = RegionCriticalPSets.size(); i != e; ++i) 553 dbgs() << TRI->getRegPressureSetName( 554 RegionCriticalPSets[i].PSetID) << " "; 555 dbgs() << "\n"); 556 } 557 558 // FIXME: When the pressure tracker deals in pressure differences then we won't 559 // iterate over all RegionCriticalPSets[i]. 560 void ScheduleDAGMI:: 561 updateScheduledPressure(std::vector<unsigned> NewMaxPressure) { 562 for (unsigned i = 0, e = RegionCriticalPSets.size(); i < e; ++i) { 563 unsigned ID = RegionCriticalPSets[i].PSetID; 564 int &MaxUnits = RegionCriticalPSets[i].UnitIncrease; 565 if ((int)NewMaxPressure[ID] > MaxUnits) 566 MaxUnits = NewMaxPressure[ID]; 567 } 568 } 569 570 // Release all DAG roots for scheduling. 571 void ScheduleDAGMI::releaseRoots() { 572 SmallVector<SUnit*, 16> BotRoots; 573 574 for (std::vector<SUnit>::iterator 575 I = SUnits.begin(), E = SUnits.end(); I != E; ++I) { 576 // A SUnit is ready to top schedule if it has no predecessors. 577 if (I->Preds.empty()) 578 SchedImpl->releaseTopNode(&(*I)); 579 // A SUnit is ready to bottom schedule if it has no successors. 580 if (I->Succs.empty()) 581 BotRoots.push_back(&(*I)); 582 } 583 // Release bottom roots in reverse order so the higher priority nodes appear 584 // first. This is more natural and slightly more efficient. 585 for (SmallVectorImpl<SUnit*>::const_reverse_iterator 586 I = BotRoots.rbegin(), E = BotRoots.rend(); I != E; ++I) 587 SchedImpl->releaseBottomNode(*I); 588 } 589 590 /// schedule - Called back from MachineScheduler::runOnMachineFunction 591 /// after setting up the current scheduling region. [RegionBegin, RegionEnd) 592 /// only includes instructions that have DAG nodes, not scheduling boundaries. 593 void ScheduleDAGMI::schedule() { 594 // Initialize the register pressure tracker used by buildSchedGraph. 595 RPTracker.init(&MF, RegClassInfo, LIS, BB, LiveRegionEnd); 596 597 // Account for liveness generate by the region boundary. 598 if (LiveRegionEnd != RegionEnd) 599 RPTracker.recede(); 600 601 // Build the DAG, and compute current register pressure. 602 buildSchedGraph(AA, &RPTracker); 603 604 // Initialize top/bottom trackers after computing region pressure. 605 initRegPressure(); 606 607 DEBUG(for (unsigned su = 0, e = SUnits.size(); su != e; ++su) 608 SUnits[su].dumpAll(this)); 609 610 if (ViewMISchedDAGs) viewGraph(); 611 612 SchedImpl->initialize(this); 613 614 // Release edges from the special Entry node or to the special Exit node. 615 releaseSuccessors(&EntrySU); 616 releasePredecessors(&ExitSU); 617 618 // Release all DAG roots for scheduling. 619 releaseRoots(); 620 621 CurrentTop = nextIfDebug(RegionBegin, RegionEnd); 622 CurrentBottom = RegionEnd; 623 bool IsTopNode = false; 624 while (SUnit *SU = SchedImpl->pickNode(IsTopNode)) { 625 DEBUG(dbgs() << "*** " << (IsTopNode ? "Top" : "Bottom") 626 << " Scheduling Instruction"); 627 if (!checkSchedLimit()) 628 break; 629 630 // Move the instruction to its new location in the instruction stream. 631 MachineInstr *MI = SU->getInstr(); 632 633 if (IsTopNode) { 634 assert(SU->isTopReady() && "node still has unscheduled dependencies"); 635 if (&*CurrentTop == MI) 636 CurrentTop = nextIfDebug(++CurrentTop, CurrentBottom); 637 else { 638 moveInstruction(MI, CurrentTop); 639 TopRPTracker.setPos(MI); 640 } 641 642 // Update top scheduled pressure. 643 TopRPTracker.advance(); 644 assert(TopRPTracker.getPos() == CurrentTop && "out of sync"); 645 updateScheduledPressure(TopRPTracker.getPressure().MaxSetPressure); 646 647 // Release dependent instructions for scheduling. 648 releaseSuccessors(SU); 649 } 650 else { 651 assert(SU->isBottomReady() && "node still has unscheduled dependencies"); 652 MachineBasicBlock::iterator priorII = 653 priorNonDebug(CurrentBottom, CurrentTop); 654 if (&*priorII == MI) 655 CurrentBottom = priorII; 656 else { 657 if (&*CurrentTop == MI) { 658 CurrentTop = nextIfDebug(++CurrentTop, priorII); 659 TopRPTracker.setPos(CurrentTop); 660 } 661 moveInstruction(MI, CurrentBottom); 662 CurrentBottom = MI; 663 } 664 // Update bottom scheduled pressure. 665 BotRPTracker.recede(); 666 assert(BotRPTracker.getPos() == CurrentBottom && "out of sync"); 667 updateScheduledPressure(BotRPTracker.getPressure().MaxSetPressure); 668 669 // Release dependent instructions for scheduling. 670 releasePredecessors(SU); 671 } 672 SU->isScheduled = true; 673 SchedImpl->schedNode(SU, IsTopNode); 674 DEBUG(SU->dump(this)); 675 } 676 assert(CurrentTop == CurrentBottom && "Nonempty unscheduled zone."); 677 678 placeDebugValues(); 679 } 680 681 /// Reinsert any remaining debug_values, just like the PostRA scheduler. 682 void ScheduleDAGMI::placeDebugValues() { 683 // If first instruction was a DBG_VALUE then put it back. 684 if (FirstDbgValue) { 685 BB->splice(RegionBegin, BB, FirstDbgValue); 686 RegionBegin = FirstDbgValue; 687 } 688 689 for (std::vector<std::pair<MachineInstr *, MachineInstr *> >::iterator 690 DI = DbgValues.end(), DE = DbgValues.begin(); DI != DE; --DI) { 691 std::pair<MachineInstr *, MachineInstr *> P = *prior(DI); 692 MachineInstr *DbgValue = P.first; 693 MachineBasicBlock::iterator OrigPrevMI = P.second; 694 BB->splice(++OrigPrevMI, BB, DbgValue); 695 if (OrigPrevMI == llvm::prior(RegionEnd)) 696 RegionEnd = DbgValue; 697 } 698 DbgValues.clear(); 699 FirstDbgValue = NULL; 700 } 701 702 //===----------------------------------------------------------------------===// 703 // ConvergingScheduler - Implementation of the standard MachineSchedStrategy. 704 //===----------------------------------------------------------------------===// 705 706 namespace { 707 /// ReadyQ encapsulates vector of "ready" SUnits with basic convenience methods 708 /// for pushing and removing nodes. ReadyQ's are uniquely identified by an 709 /// ID. SUnit::NodeQueueId us a mask of the ReadyQs that the SUnit is in. 710 class ReadyQueue { 711 unsigned ID; 712 std::string Name; 713 std::vector<SUnit*> Queue; 714 715 public: 716 ReadyQueue(unsigned id, const Twine &name): ID(id), Name(name.str()) {} 717 718 unsigned getID() const { return ID; } 719 720 StringRef getName() const { return Name; } 721 722 // SU is in this queue if it's NodeQueueID is a superset of this ID. 723 bool isInQueue(SUnit *SU) const { return (SU->NodeQueueId & ID); } 724 725 bool empty() const { return Queue.empty(); } 726 727 unsigned size() const { return Queue.size(); } 728 729 typedef std::vector<SUnit*>::iterator iterator; 730 731 iterator begin() { return Queue.begin(); } 732 733 iterator end() { return Queue.end(); } 734 735 iterator find(SUnit *SU) { 736 return std::find(Queue.begin(), Queue.end(), SU); 737 } 738 739 void push(SUnit *SU) { 740 Queue.push_back(SU); 741 SU->NodeQueueId |= ID; 742 } 743 744 void remove(iterator I) { 745 (*I)->NodeQueueId &= ~ID; 746 *I = Queue.back(); 747 Queue.pop_back(); 748 } 749 750 void dump() { 751 dbgs() << Name << ": "; 752 for (unsigned i = 0, e = Queue.size(); i < e; ++i) 753 dbgs() << Queue[i]->NodeNum << " "; 754 dbgs() << "\n"; 755 } 756 }; 757 758 /// ConvergingScheduler shrinks the unscheduled zone using heuristics to balance 759 /// the schedule. 760 class ConvergingScheduler : public MachineSchedStrategy { 761 762 /// Store the state used by ConvergingScheduler heuristics, required for the 763 /// lifetime of one invocation of pickNode(). 764 struct SchedCandidate { 765 // The best SUnit candidate. 766 SUnit *SU; 767 768 // Register pressure values for the best candidate. 769 RegPressureDelta RPDelta; 770 771 SchedCandidate(): SU(NULL) {} 772 }; 773 /// Represent the type of SchedCandidate found within a single queue. 774 enum CandResult { 775 NoCand, NodeOrder, SingleExcess, SingleCritical, SingleMax, MultiPressure }; 776 777 /// Each Scheduling boundary is associated with ready queues. It tracks the 778 /// current cycle in whichever direction at has moved, and maintains the state 779 /// of "hazards" and other interlocks at the current cycle. 780 struct SchedBoundary { 781 ReadyQueue Available; 782 ReadyQueue Pending; 783 bool CheckPending; 784 785 ScheduleHazardRecognizer *HazardRec; 786 787 unsigned CurrCycle; 788 unsigned IssueCount; 789 790 /// MinReadyCycle - Cycle of the soonest available instruction. 791 unsigned MinReadyCycle; 792 793 /// Pending queues extend the ready queues with the same ID and the 794 /// PendingFlag set. 795 SchedBoundary(unsigned ID, const Twine &Name): 796 Available(ID, Name+".A"), 797 Pending(ID << ConvergingScheduler::LogMaxQID, Name+".P"), 798 CheckPending(false), HazardRec(0), CurrCycle(0), IssueCount(0), 799 MinReadyCycle(UINT_MAX) {} 800 801 ~SchedBoundary() { delete HazardRec; } 802 803 bool isTop() const { 804 return Available.getID() == ConvergingScheduler::TopQID; 805 } 806 807 void releaseNode(SUnit *SU, unsigned ReadyCycle); 808 809 void bumpCycle(); 810 811 void releasePending(); 812 813 void removeReady(SUnit *SU); 814 815 SUnit *pickOnlyChoice(); 816 }; 817 818 ScheduleDAGMI *DAG; 819 const TargetRegisterInfo *TRI; 820 821 // State of the top and bottom scheduled instruction boundaries. 822 SchedBoundary Top; 823 SchedBoundary Bot; 824 825 public: 826 /// SUnit::NodeQueueId: 0 (none), 1 (top), 2 (bot), 3 (both) 827 enum { 828 TopQID = 1, 829 BotQID = 2, 830 LogMaxQID = 2 831 }; 832 833 ConvergingScheduler(): 834 DAG(0), TRI(0), Top(TopQID, "TopQ"), Bot(BotQID, "BotQ") {} 835 836 virtual void initialize(ScheduleDAGMI *dag); 837 838 virtual SUnit *pickNode(bool &IsTopNode); 839 840 virtual void schedNode(SUnit *SU, bool IsTopNode); 841 842 virtual void releaseTopNode(SUnit *SU); 843 844 virtual void releaseBottomNode(SUnit *SU); 845 846 protected: 847 SUnit *pickNodeBidrectional(bool &IsTopNode); 848 849 CandResult pickNodeFromQueue(ReadyQueue &Q, 850 const RegPressureTracker &RPTracker, 851 SchedCandidate &Candidate); 852 #ifndef NDEBUG 853 void traceCandidate(const char *Label, const ReadyQueue &Q, SUnit *SU, 854 PressureElement P = PressureElement()); 855 #endif 856 }; 857 } // namespace 858 859 void ConvergingScheduler::initialize(ScheduleDAGMI *dag) { 860 DAG = dag; 861 TRI = DAG->TRI; 862 863 // Initialize the HazardRecognizers. 864 const TargetMachine &TM = DAG->MF.getTarget(); 865 const InstrItineraryData *Itin = TM.getInstrItineraryData(); 866 Top.HazardRec = TM.getInstrInfo()->CreateTargetMIHazardRecognizer(Itin, DAG); 867 Bot.HazardRec = TM.getInstrInfo()->CreateTargetMIHazardRecognizer(Itin, DAG); 868 869 assert((!ForceTopDown || !ForceBottomUp) && 870 "-misched-topdown incompatible with -misched-bottomup"); 871 } 872 873 void ConvergingScheduler::releaseTopNode(SUnit *SU) { 874 Top.releaseNode(SU, SU->getDepth()); 875 } 876 877 void ConvergingScheduler::releaseBottomNode(SUnit *SU) { 878 Bot.releaseNode(SU, SU->getHeight()); 879 } 880 881 void ConvergingScheduler::SchedBoundary::releaseNode(SUnit *SU, 882 unsigned ReadyCycle) { 883 if (SU->isScheduled) 884 return; 885 886 if (ReadyCycle < MinReadyCycle) 887 MinReadyCycle = ReadyCycle; 888 889 // Check for interlocks first. For the purpose of other heuristics, an 890 // instruction that cannot issue appears as if it's not in the ReadyQueue. 891 if (HazardRec->isEnabled() 892 && HazardRec->getHazardType(SU) != ScheduleHazardRecognizer::NoHazard) 893 Pending.push(SU); 894 else 895 Available.push(SU); 896 } 897 898 /// Move the boundary of scheduled code by one cycle. 899 void ConvergingScheduler::SchedBoundary::bumpCycle() { 900 IssueCount = 0; 901 902 assert(MinReadyCycle < UINT_MAX && "MinReadyCycle uninitialized"); 903 unsigned NextCycle = std::max(CurrCycle + 1, MinReadyCycle); 904 905 if (!HazardRec->isEnabled()) { 906 // Bypass lots of virtual calls in case of long latency. 907 CurrCycle = NextCycle; 908 } 909 else { 910 for (; CurrCycle != NextCycle; ++CurrCycle) { 911 if (isTop()) 912 HazardRec->AdvanceCycle(); 913 else 914 HazardRec->RecedeCycle(); 915 } 916 } 917 CheckPending = true; 918 919 DEBUG(dbgs() << "*** " << Available.getName() << " cycle " 920 << CurrCycle << '\n'); 921 } 922 923 /// Release pending ready nodes in to the available queue. This makes them 924 /// visible to heuristics. 925 void ConvergingScheduler::SchedBoundary::releasePending() { 926 // If the available queue is empty, it is safe to reset MinReadyCycle. 927 if (Available.empty()) 928 MinReadyCycle = UINT_MAX; 929 930 // Check to see if any of the pending instructions are ready to issue. If 931 // so, add them to the available queue. 932 for (unsigned i = 0, e = Pending.size(); i != e; ++i) { 933 SUnit *SU = *(Pending.begin()+i); 934 unsigned ReadyCycle = isTop() ? SU->getHeight() : SU->getDepth(); 935 936 if (ReadyCycle < MinReadyCycle) 937 MinReadyCycle = ReadyCycle; 938 939 if (ReadyCycle > CurrCycle) 940 continue; 941 942 if (HazardRec->isEnabled() 943 && HazardRec->getHazardType(SU) != ScheduleHazardRecognizer::NoHazard) 944 continue; 945 946 Available.push(SU); 947 Pending.remove(Pending.begin()+i); 948 --i; --e; 949 } 950 CheckPending = false; 951 } 952 953 /// Remove SU from the ready set for this boundary. 954 void ConvergingScheduler::SchedBoundary::removeReady(SUnit *SU) { 955 if (Available.isInQueue(SU)) 956 Available.remove(Available.find(SU)); 957 else { 958 assert(Pending.isInQueue(SU) && "bad ready count"); 959 Pending.remove(Pending.find(SU)); 960 } 961 } 962 963 /// If this queue only has one ready candidate, return it. As a side effect, 964 /// advance the cycle until at least one node is ready. If multiple instructions 965 /// are ready, return NULL. 966 SUnit *ConvergingScheduler::SchedBoundary::pickOnlyChoice() { 967 if (CheckPending) 968 releasePending(); 969 970 for (unsigned i = 0; Available.empty(); ++i) { 971 assert(i <= HazardRec->getMaxLookAhead() && "permanent hazard"); (void)i; 972 bumpCycle(); 973 releasePending(); 974 } 975 if (Available.size() == 1) 976 return *Available.begin(); 977 return NULL; 978 } 979 980 #ifndef NDEBUG 981 void ConvergingScheduler::traceCandidate(const char *Label, const ReadyQueue &Q, 982 SUnit *SU, PressureElement P) { 983 dbgs() << Label << " " << Q.getName() << " "; 984 if (P.isValid()) 985 dbgs() << TRI->getRegPressureSetName(P.PSetID) << ":" << P.UnitIncrease 986 << " "; 987 else 988 dbgs() << " "; 989 SU->dump(DAG); 990 } 991 #endif 992 993 /// pickNodeFromQueue helper that returns true if the LHS reg pressure effect is 994 /// more desirable than RHS from scheduling standpoint. 995 static bool compareRPDelta(const RegPressureDelta &LHS, 996 const RegPressureDelta &RHS) { 997 // Compare each component of pressure in decreasing order of importance 998 // without checking if any are valid. Invalid PressureElements are assumed to 999 // have UnitIncrease==0, so are neutral. 1000 1001 // Avoid increasing the max critical pressure in the scheduled region. 1002 if (LHS.Excess.UnitIncrease != RHS.Excess.UnitIncrease) 1003 return LHS.Excess.UnitIncrease < RHS.Excess.UnitIncrease; 1004 1005 // Avoid increasing the max critical pressure in the scheduled region. 1006 if (LHS.CriticalMax.UnitIncrease != RHS.CriticalMax.UnitIncrease) 1007 return LHS.CriticalMax.UnitIncrease < RHS.CriticalMax.UnitIncrease; 1008 1009 // Avoid increasing the max pressure of the entire region. 1010 if (LHS.CurrentMax.UnitIncrease != RHS.CurrentMax.UnitIncrease) 1011 return LHS.CurrentMax.UnitIncrease < RHS.CurrentMax.UnitIncrease; 1012 1013 return false; 1014 } 1015 1016 /// Pick the best candidate from the top queue. 1017 /// 1018 /// TODO: getMaxPressureDelta results can be mostly cached for each SUnit during 1019 /// DAG building. To adjust for the current scheduling location we need to 1020 /// maintain the number of vreg uses remaining to be top-scheduled. 1021 ConvergingScheduler::CandResult ConvergingScheduler:: 1022 pickNodeFromQueue(ReadyQueue &Q, const RegPressureTracker &RPTracker, 1023 SchedCandidate &Candidate) { 1024 DEBUG(Q.dump()); 1025 1026 // getMaxPressureDelta temporarily modifies the tracker. 1027 RegPressureTracker &TempTracker = const_cast<RegPressureTracker&>(RPTracker); 1028 1029 // BestSU remains NULL if no top candidates beat the best existing candidate. 1030 CandResult FoundCandidate = NoCand; 1031 for (ReadyQueue::iterator I = Q.begin(), E = Q.end(); I != E; ++I) { 1032 RegPressureDelta RPDelta; 1033 TempTracker.getMaxPressureDelta((*I)->getInstr(), RPDelta, 1034 DAG->getRegionCriticalPSets(), 1035 DAG->getRegPressure().MaxSetPressure); 1036 1037 // Initialize the candidate if needed. 1038 if (!Candidate.SU) { 1039 Candidate.SU = *I; 1040 Candidate.RPDelta = RPDelta; 1041 FoundCandidate = NodeOrder; 1042 continue; 1043 } 1044 // Avoid exceeding the target's limit. 1045 if (RPDelta.Excess.UnitIncrease < Candidate.RPDelta.Excess.UnitIncrease) { 1046 DEBUG(traceCandidate("ECAND", Q, *I, RPDelta.Excess)); 1047 Candidate.SU = *I; 1048 Candidate.RPDelta = RPDelta; 1049 FoundCandidate = SingleExcess; 1050 continue; 1051 } 1052 if (RPDelta.Excess.UnitIncrease > Candidate.RPDelta.Excess.UnitIncrease) 1053 continue; 1054 if (FoundCandidate == SingleExcess) 1055 FoundCandidate = MultiPressure; 1056 1057 // Avoid increasing the max critical pressure in the scheduled region. 1058 if (RPDelta.CriticalMax.UnitIncrease 1059 < Candidate.RPDelta.CriticalMax.UnitIncrease) { 1060 DEBUG(traceCandidate("PCAND", Q, *I, RPDelta.CriticalMax)); 1061 Candidate.SU = *I; 1062 Candidate.RPDelta = RPDelta; 1063 FoundCandidate = SingleCritical; 1064 continue; 1065 } 1066 if (RPDelta.CriticalMax.UnitIncrease 1067 > Candidate.RPDelta.CriticalMax.UnitIncrease) 1068 continue; 1069 if (FoundCandidate == SingleCritical) 1070 FoundCandidate = MultiPressure; 1071 1072 // Avoid increasing the max pressure of the entire region. 1073 if (RPDelta.CurrentMax.UnitIncrease 1074 < Candidate.RPDelta.CurrentMax.UnitIncrease) { 1075 DEBUG(traceCandidate("MCAND", Q, *I, RPDelta.CurrentMax)); 1076 Candidate.SU = *I; 1077 Candidate.RPDelta = RPDelta; 1078 FoundCandidate = SingleMax; 1079 continue; 1080 } 1081 if (RPDelta.CurrentMax.UnitIncrease 1082 > Candidate.RPDelta.CurrentMax.UnitIncrease) 1083 continue; 1084 if (FoundCandidate == SingleMax) 1085 FoundCandidate = MultiPressure; 1086 1087 // Fall through to original instruction order. 1088 // Only consider node order if Candidate was chosen from this Q. 1089 if (FoundCandidate == NoCand) 1090 continue; 1091 1092 if ((Q.getID() == TopQID && (*I)->NodeNum < Candidate.SU->NodeNum) 1093 || (Q.getID() == BotQID && (*I)->NodeNum > Candidate.SU->NodeNum)) { 1094 DEBUG(traceCandidate("NCAND", Q, *I)); 1095 Candidate.SU = *I; 1096 Candidate.RPDelta = RPDelta; 1097 FoundCandidate = NodeOrder; 1098 } 1099 } 1100 return FoundCandidate; 1101 } 1102 1103 /// Pick the best candidate node from either the top or bottom queue. 1104 SUnit *ConvergingScheduler::pickNodeBidrectional(bool &IsTopNode) { 1105 // Schedule as far as possible in the direction of no choice. This is most 1106 // efficient, but also provides the best heuristics for CriticalPSets. 1107 if (SUnit *SU = Bot.pickOnlyChoice()) { 1108 IsTopNode = false; 1109 return SU; 1110 } 1111 if (SUnit *SU = Top.pickOnlyChoice()) { 1112 IsTopNode = true; 1113 return SU; 1114 } 1115 SchedCandidate BotCand; 1116 // Prefer bottom scheduling when heuristics are silent. 1117 CandResult BotResult = pickNodeFromQueue(Bot.Available, 1118 DAG->getBotRPTracker(), BotCand); 1119 assert(BotResult != NoCand && "failed to find the first candidate"); 1120 1121 // If either Q has a single candidate that provides the least increase in 1122 // Excess pressure, we can immediately schedule from that Q. 1123 // 1124 // RegionCriticalPSets summarizes the pressure within the scheduled region and 1125 // affects picking from either Q. If scheduling in one direction must 1126 // increase pressure for one of the excess PSets, then schedule in that 1127 // direction first to provide more freedom in the other direction. 1128 if (BotResult == SingleExcess || BotResult == SingleCritical) { 1129 IsTopNode = false; 1130 return BotCand.SU; 1131 } 1132 // Check if the top Q has a better candidate. 1133 SchedCandidate TopCand; 1134 CandResult TopResult = pickNodeFromQueue(Top.Available, 1135 DAG->getTopRPTracker(), TopCand); 1136 assert(TopResult != NoCand && "failed to find the first candidate"); 1137 1138 if (TopResult == SingleExcess || TopResult == SingleCritical) { 1139 IsTopNode = true; 1140 return TopCand.SU; 1141 } 1142 // If either Q has a single candidate that minimizes pressure above the 1143 // original region's pressure pick it. 1144 if (BotResult == SingleMax) { 1145 IsTopNode = false; 1146 return BotCand.SU; 1147 } 1148 if (TopResult == SingleMax) { 1149 IsTopNode = true; 1150 return TopCand.SU; 1151 } 1152 // Check for a salient pressure difference and pick the best from either side. 1153 if (compareRPDelta(TopCand.RPDelta, BotCand.RPDelta)) { 1154 IsTopNode = true; 1155 return TopCand.SU; 1156 } 1157 // Otherwise prefer the bottom candidate in node order. 1158 IsTopNode = false; 1159 return BotCand.SU; 1160 } 1161 1162 /// Pick the best node to balance the schedule. Implements MachineSchedStrategy. 1163 SUnit *ConvergingScheduler::pickNode(bool &IsTopNode) { 1164 if (DAG->top() == DAG->bottom()) { 1165 assert(Top.Available.empty() && Top.Pending.empty() && 1166 Bot.Available.empty() && Bot.Pending.empty() && "ReadyQ garbage"); 1167 return NULL; 1168 } 1169 SUnit *SU; 1170 if (ForceTopDown) { 1171 SU = Top.pickOnlyChoice(); 1172 if (!SU) { 1173 SchedCandidate TopCand; 1174 CandResult TopResult = 1175 pickNodeFromQueue(Top.Available, DAG->getTopRPTracker(), TopCand); 1176 assert(TopResult != NoCand && "failed to find the first candidate"); 1177 SU = TopCand.SU; 1178 } 1179 IsTopNode = true; 1180 } 1181 else if (ForceBottomUp) { 1182 SU = Bot.pickOnlyChoice(); 1183 if (!SU) { 1184 SchedCandidate BotCand; 1185 CandResult BotResult = 1186 pickNodeFromQueue(Bot.Available, DAG->getBotRPTracker(), BotCand); 1187 assert(BotResult != NoCand && "failed to find the first candidate"); 1188 SU = BotCand.SU; 1189 } 1190 IsTopNode = false; 1191 } 1192 else { 1193 SU = pickNodeBidrectional(IsTopNode); 1194 } 1195 if (SU->isTopReady()) 1196 Top.removeReady(SU); 1197 if (SU->isBottomReady()) 1198 Bot.removeReady(SU); 1199 return SU; 1200 } 1201 1202 /// Update the scheduler's state after scheduling a node. This is the same node 1203 /// that was just returned by pickNode(). However, ScheduleDAGMI needs to update 1204 /// it's state based on the current cycle before MachineSchedStrategy. 1205 void ConvergingScheduler::schedNode(SUnit *SU, bool IsTopNode) { 1206 DEBUG(dbgs() << " in cycle " << (IsTopNode ? Top.CurrCycle : Bot.CurrCycle) 1207 << '\n'); 1208 1209 // Update the reservation table. 1210 if (IsTopNode && Top.HazardRec->isEnabled()) { 1211 Top.HazardRec->EmitInstruction(SU); 1212 if (Top.HazardRec->atIssueLimit()) { 1213 DEBUG(dbgs() << "*** Max instrs at cycle " << Top.CurrCycle << '\n'); 1214 Top.bumpCycle(); 1215 } 1216 } 1217 else if (Bot.HazardRec->isEnabled()) { 1218 if (SU->isCall) { 1219 // Calls are scheduled with their preceding instructions. For bottom-up 1220 // scheduling, clear the pipeline state before emitting. 1221 Bot.HazardRec->Reset(); 1222 } 1223 Bot.HazardRec->EmitInstruction(SU); 1224 if (Bot.HazardRec->atIssueLimit()) { 1225 DEBUG(dbgs() << "*** Max instrs at cycle " << Bot.CurrCycle << '\n'); 1226 Bot.bumpCycle(); 1227 } 1228 } 1229 } 1230 1231 /// Create the standard converging machine scheduler. This will be used as the 1232 /// default scheduler if the target does not set a default. 1233 static ScheduleDAGInstrs *createConvergingSched(MachineSchedContext *C) { 1234 assert((!ForceTopDown || !ForceBottomUp) && 1235 "-misched-topdown incompatible with -misched-bottomup"); 1236 return new ScheduleDAGMI(C, new ConvergingScheduler()); 1237 } 1238 static MachineSchedRegistry 1239 ConvergingSchedRegistry("converge", "Standard converging scheduler.", 1240 createConvergingSched); 1241 1242 //===----------------------------------------------------------------------===// 1243 // Machine Instruction Shuffler for Correctness Testing 1244 //===----------------------------------------------------------------------===// 1245 1246 #ifndef NDEBUG 1247 namespace { 1248 /// Apply a less-than relation on the node order, which corresponds to the 1249 /// instruction order prior to scheduling. IsReverse implements greater-than. 1250 template<bool IsReverse> 1251 struct SUnitOrder { 1252 bool operator()(SUnit *A, SUnit *B) const { 1253 if (IsReverse) 1254 return A->NodeNum > B->NodeNum; 1255 else 1256 return A->NodeNum < B->NodeNum; 1257 } 1258 }; 1259 1260 /// Reorder instructions as much as possible. 1261 class InstructionShuffler : public MachineSchedStrategy { 1262 bool IsAlternating; 1263 bool IsTopDown; 1264 1265 // Using a less-than relation (SUnitOrder<false>) for the TopQ priority 1266 // gives nodes with a higher number higher priority causing the latest 1267 // instructions to be scheduled first. 1268 PriorityQueue<SUnit*, std::vector<SUnit*>, SUnitOrder<false> > 1269 TopQ; 1270 // When scheduling bottom-up, use greater-than as the queue priority. 1271 PriorityQueue<SUnit*, std::vector<SUnit*>, SUnitOrder<true> > 1272 BottomQ; 1273 public: 1274 InstructionShuffler(bool alternate, bool topdown) 1275 : IsAlternating(alternate), IsTopDown(topdown) {} 1276 1277 virtual void initialize(ScheduleDAGMI *) { 1278 TopQ.clear(); 1279 BottomQ.clear(); 1280 } 1281 1282 /// Implement MachineSchedStrategy interface. 1283 /// ----------------------------------------- 1284 1285 virtual SUnit *pickNode(bool &IsTopNode) { 1286 SUnit *SU; 1287 if (IsTopDown) { 1288 do { 1289 if (TopQ.empty()) return NULL; 1290 SU = TopQ.top(); 1291 TopQ.pop(); 1292 } while (SU->isScheduled); 1293 IsTopNode = true; 1294 } 1295 else { 1296 do { 1297 if (BottomQ.empty()) return NULL; 1298 SU = BottomQ.top(); 1299 BottomQ.pop(); 1300 } while (SU->isScheduled); 1301 IsTopNode = false; 1302 } 1303 if (IsAlternating) 1304 IsTopDown = !IsTopDown; 1305 return SU; 1306 } 1307 1308 virtual void schedNode(SUnit *SU, bool IsTopNode) {} 1309 1310 virtual void releaseTopNode(SUnit *SU) { 1311 TopQ.push(SU); 1312 } 1313 virtual void releaseBottomNode(SUnit *SU) { 1314 BottomQ.push(SU); 1315 } 1316 }; 1317 } // namespace 1318 1319 static ScheduleDAGInstrs *createInstructionShuffler(MachineSchedContext *C) { 1320 bool Alternate = !ForceTopDown && !ForceBottomUp; 1321 bool TopDown = !ForceBottomUp; 1322 assert((TopDown || !ForceTopDown) && 1323 "-misched-topdown incompatible with -misched-bottomup"); 1324 return new ScheduleDAGMI(C, new InstructionShuffler(Alternate, TopDown)); 1325 } 1326 static MachineSchedRegistry ShufflerRegistry( 1327 "shuffle", "Shuffle machine instructions alternating directions", 1328 createInstructionShuffler); 1329 #endif // !NDEBUG 1330