xref: /freebsd/contrib/llvm-project/llvm/lib/Target/AMDGPU/GCNIterativeScheduler.cpp (revision 5ffd83dbcc34f10e07f6d3e968ae6365869615f4)
10b57cec5SDimitry Andric //===- GCNIterativeScheduler.cpp ------------------------------------------===//
20b57cec5SDimitry Andric //
30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
60b57cec5SDimitry Andric //
70b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
8*5ffd83dbSDimitry Andric ///
9*5ffd83dbSDimitry Andric /// \file
10*5ffd83dbSDimitry Andric /// This file implements the class GCNIterativeScheduler.
11*5ffd83dbSDimitry Andric ///
12*5ffd83dbSDimitry Andric //===----------------------------------------------------------------------===//
130b57cec5SDimitry Andric 
140b57cec5SDimitry Andric #include "GCNIterativeScheduler.h"
150b57cec5SDimitry Andric #include "AMDGPUSubtarget.h"
160b57cec5SDimitry Andric #include "GCNRegPressure.h"
170b57cec5SDimitry Andric #include "GCNSchedStrategy.h"
180b57cec5SDimitry Andric #include "SIMachineFunctionInfo.h"
190b57cec5SDimitry Andric #include "llvm/ADT/ArrayRef.h"
200b57cec5SDimitry Andric #include "llvm/ADT/STLExtras.h"
210b57cec5SDimitry Andric #include "llvm/ADT/SmallVector.h"
220b57cec5SDimitry Andric #include "llvm/CodeGen/LiveIntervals.h"
230b57cec5SDimitry Andric #include "llvm/CodeGen/MachineBasicBlock.h"
240b57cec5SDimitry Andric #include "llvm/CodeGen/MachineFunction.h"
250b57cec5SDimitry Andric #include "llvm/CodeGen/RegisterPressure.h"
260b57cec5SDimitry Andric #include "llvm/CodeGen/ScheduleDAG.h"
270b57cec5SDimitry Andric #include "llvm/Config/llvm-config.h"
280b57cec5SDimitry Andric #include "llvm/Support/Compiler.h"
290b57cec5SDimitry Andric #include "llvm/Support/Debug.h"
300b57cec5SDimitry Andric #include "llvm/Support/raw_ostream.h"
310b57cec5SDimitry Andric #include <algorithm>
320b57cec5SDimitry Andric #include <cassert>
330b57cec5SDimitry Andric #include <iterator>
340b57cec5SDimitry Andric #include <limits>
350b57cec5SDimitry Andric #include <memory>
360b57cec5SDimitry Andric #include <type_traits>
370b57cec5SDimitry Andric #include <vector>
380b57cec5SDimitry Andric 
390b57cec5SDimitry Andric using namespace llvm;
400b57cec5SDimitry Andric 
410b57cec5SDimitry Andric #define DEBUG_TYPE "machine-scheduler"
420b57cec5SDimitry Andric 
430b57cec5SDimitry Andric namespace llvm {
440b57cec5SDimitry Andric 
450b57cec5SDimitry Andric std::vector<const SUnit *> makeMinRegSchedule(ArrayRef<const SUnit *> TopRoots,
460b57cec5SDimitry Andric                                               const ScheduleDAG &DAG);
470b57cec5SDimitry Andric 
480b57cec5SDimitry Andric   std::vector<const SUnit*> makeGCNILPScheduler(ArrayRef<const SUnit*> BotRoots,
490b57cec5SDimitry Andric     const ScheduleDAG &DAG);
500b57cec5SDimitry Andric }
510b57cec5SDimitry Andric 
520b57cec5SDimitry Andric // shim accessors for different order containers
530b57cec5SDimitry Andric static inline MachineInstr *getMachineInstr(MachineInstr *MI) {
540b57cec5SDimitry Andric   return MI;
550b57cec5SDimitry Andric }
560b57cec5SDimitry Andric static inline MachineInstr *getMachineInstr(const SUnit *SU) {
570b57cec5SDimitry Andric   return SU->getInstr();
580b57cec5SDimitry Andric }
590b57cec5SDimitry Andric static inline MachineInstr *getMachineInstr(const SUnit &SU) {
600b57cec5SDimitry Andric   return SU.getInstr();
610b57cec5SDimitry Andric }
620b57cec5SDimitry Andric 
630b57cec5SDimitry Andric #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
640b57cec5SDimitry Andric LLVM_DUMP_METHOD
650b57cec5SDimitry Andric static void printRegion(raw_ostream &OS,
660b57cec5SDimitry Andric                         MachineBasicBlock::iterator Begin,
670b57cec5SDimitry Andric                         MachineBasicBlock::iterator End,
680b57cec5SDimitry Andric                         const LiveIntervals *LIS,
690b57cec5SDimitry Andric                         unsigned MaxInstNum =
700b57cec5SDimitry Andric                           std::numeric_limits<unsigned>::max()) {
710b57cec5SDimitry Andric   auto BB = Begin->getParent();
720b57cec5SDimitry Andric   OS << BB->getParent()->getName() << ":" << printMBBReference(*BB) << ' '
730b57cec5SDimitry Andric      << BB->getName() << ":\n";
740b57cec5SDimitry Andric   auto I = Begin;
750b57cec5SDimitry Andric   MaxInstNum = std::max(MaxInstNum, 1u);
760b57cec5SDimitry Andric   for (; I != End && MaxInstNum; ++I, --MaxInstNum) {
770b57cec5SDimitry Andric     if (!I->isDebugInstr() && LIS)
780b57cec5SDimitry Andric       OS << LIS->getInstructionIndex(*I);
790b57cec5SDimitry Andric     OS << '\t' << *I;
800b57cec5SDimitry Andric   }
810b57cec5SDimitry Andric   if (I != End) {
820b57cec5SDimitry Andric     OS << "\t...\n";
830b57cec5SDimitry Andric     I = std::prev(End);
840b57cec5SDimitry Andric     if (!I->isDebugInstr() && LIS)
850b57cec5SDimitry Andric       OS << LIS->getInstructionIndex(*I);
860b57cec5SDimitry Andric     OS << '\t' << *I;
870b57cec5SDimitry Andric   }
880b57cec5SDimitry Andric   if (End != BB->end()) { // print boundary inst if present
890b57cec5SDimitry Andric     OS << "----\n";
900b57cec5SDimitry Andric     if (LIS) OS << LIS->getInstructionIndex(*End) << '\t';
910b57cec5SDimitry Andric     OS << *End;
920b57cec5SDimitry Andric   }
930b57cec5SDimitry Andric }
940b57cec5SDimitry Andric 
950b57cec5SDimitry Andric LLVM_DUMP_METHOD
960b57cec5SDimitry Andric static void printLivenessInfo(raw_ostream &OS,
970b57cec5SDimitry Andric                               MachineBasicBlock::iterator Begin,
980b57cec5SDimitry Andric                               MachineBasicBlock::iterator End,
990b57cec5SDimitry Andric                               const LiveIntervals *LIS) {
1000b57cec5SDimitry Andric   const auto BB = Begin->getParent();
1010b57cec5SDimitry Andric   const auto &MRI = BB->getParent()->getRegInfo();
1020b57cec5SDimitry Andric 
1030b57cec5SDimitry Andric   const auto LiveIns = getLiveRegsBefore(*Begin, *LIS);
1040b57cec5SDimitry Andric   OS << "LIn RP: ";
1050b57cec5SDimitry Andric   getRegPressure(MRI, LiveIns).print(OS);
1060b57cec5SDimitry Andric 
1070b57cec5SDimitry Andric   const auto BottomMI = End == BB->end() ? std::prev(End) : End;
1080b57cec5SDimitry Andric   const auto LiveOuts = getLiveRegsAfter(*BottomMI, *LIS);
1090b57cec5SDimitry Andric   OS << "LOt RP: ";
1100b57cec5SDimitry Andric   getRegPressure(MRI, LiveOuts).print(OS);
1110b57cec5SDimitry Andric }
1120b57cec5SDimitry Andric 
1130b57cec5SDimitry Andric LLVM_DUMP_METHOD
1140b57cec5SDimitry Andric void GCNIterativeScheduler::printRegions(raw_ostream &OS) const {
1150b57cec5SDimitry Andric   const auto &ST = MF.getSubtarget<GCNSubtarget>();
1160b57cec5SDimitry Andric   for (const auto R : Regions) {
1170b57cec5SDimitry Andric     OS << "Region to schedule ";
1180b57cec5SDimitry Andric     printRegion(OS, R->Begin, R->End, LIS, 1);
1190b57cec5SDimitry Andric     printLivenessInfo(OS, R->Begin, R->End, LIS);
1200b57cec5SDimitry Andric     OS << "Max RP: ";
1210b57cec5SDimitry Andric     R->MaxPressure.print(OS, &ST);
1220b57cec5SDimitry Andric   }
1230b57cec5SDimitry Andric }
1240b57cec5SDimitry Andric 
1250b57cec5SDimitry Andric LLVM_DUMP_METHOD
1260b57cec5SDimitry Andric void GCNIterativeScheduler::printSchedResult(raw_ostream &OS,
1270b57cec5SDimitry Andric                                              const Region *R,
1280b57cec5SDimitry Andric                                              const GCNRegPressure &RP) const {
1290b57cec5SDimitry Andric   OS << "\nAfter scheduling ";
1300b57cec5SDimitry Andric   printRegion(OS, R->Begin, R->End, LIS);
1310b57cec5SDimitry Andric   printSchedRP(OS, R->MaxPressure, RP);
1320b57cec5SDimitry Andric   OS << '\n';
1330b57cec5SDimitry Andric }
1340b57cec5SDimitry Andric 
1350b57cec5SDimitry Andric LLVM_DUMP_METHOD
1360b57cec5SDimitry Andric void GCNIterativeScheduler::printSchedRP(raw_ostream &OS,
1370b57cec5SDimitry Andric                                          const GCNRegPressure &Before,
1380b57cec5SDimitry Andric                                          const GCNRegPressure &After) const {
1390b57cec5SDimitry Andric   const auto &ST = MF.getSubtarget<GCNSubtarget>();
1400b57cec5SDimitry Andric   OS << "RP before: ";
1410b57cec5SDimitry Andric   Before.print(OS, &ST);
1420b57cec5SDimitry Andric   OS << "RP after:  ";
1430b57cec5SDimitry Andric   After.print(OS, &ST);
1440b57cec5SDimitry Andric }
1450b57cec5SDimitry Andric #endif
1460b57cec5SDimitry Andric 
1470b57cec5SDimitry Andric // DAG builder helper
1480b57cec5SDimitry Andric class GCNIterativeScheduler::BuildDAG {
1490b57cec5SDimitry Andric   GCNIterativeScheduler &Sch;
1500b57cec5SDimitry Andric   SmallVector<SUnit *, 8> TopRoots;
1510b57cec5SDimitry Andric 
1520b57cec5SDimitry Andric   SmallVector<SUnit*, 8> BotRoots;
1530b57cec5SDimitry Andric public:
1540b57cec5SDimitry Andric   BuildDAG(const Region &R, GCNIterativeScheduler &_Sch)
1550b57cec5SDimitry Andric     : Sch(_Sch) {
1560b57cec5SDimitry Andric     auto BB = R.Begin->getParent();
1570b57cec5SDimitry Andric     Sch.BaseClass::startBlock(BB);
1580b57cec5SDimitry Andric     Sch.BaseClass::enterRegion(BB, R.Begin, R.End, R.NumRegionInstrs);
1590b57cec5SDimitry Andric 
1600b57cec5SDimitry Andric     Sch.buildSchedGraph(Sch.AA, nullptr, nullptr, nullptr,
1610b57cec5SDimitry Andric                         /*TrackLaneMask*/true);
1620b57cec5SDimitry Andric     Sch.Topo.InitDAGTopologicalSorting();
1630b57cec5SDimitry Andric     Sch.findRootsAndBiasEdges(TopRoots, BotRoots);
1640b57cec5SDimitry Andric   }
1650b57cec5SDimitry Andric 
1660b57cec5SDimitry Andric   ~BuildDAG() {
1670b57cec5SDimitry Andric     Sch.BaseClass::exitRegion();
1680b57cec5SDimitry Andric     Sch.BaseClass::finishBlock();
1690b57cec5SDimitry Andric   }
1700b57cec5SDimitry Andric 
1710b57cec5SDimitry Andric   ArrayRef<const SUnit *> getTopRoots() const {
1720b57cec5SDimitry Andric     return TopRoots;
1730b57cec5SDimitry Andric   }
1740b57cec5SDimitry Andric   ArrayRef<SUnit*> getBottomRoots() const {
1750b57cec5SDimitry Andric     return BotRoots;
1760b57cec5SDimitry Andric   }
1770b57cec5SDimitry Andric };
1780b57cec5SDimitry Andric 
1790b57cec5SDimitry Andric class GCNIterativeScheduler::OverrideLegacyStrategy {
1800b57cec5SDimitry Andric   GCNIterativeScheduler &Sch;
1810b57cec5SDimitry Andric   Region &Rgn;
1820b57cec5SDimitry Andric   std::unique_ptr<MachineSchedStrategy> SaveSchedImpl;
1830b57cec5SDimitry Andric   GCNRegPressure SaveMaxRP;
1840b57cec5SDimitry Andric 
1850b57cec5SDimitry Andric public:
1860b57cec5SDimitry Andric   OverrideLegacyStrategy(Region &R,
1870b57cec5SDimitry Andric                          MachineSchedStrategy &OverrideStrategy,
1880b57cec5SDimitry Andric                          GCNIterativeScheduler &_Sch)
1890b57cec5SDimitry Andric     : Sch(_Sch)
1900b57cec5SDimitry Andric     , Rgn(R)
1910b57cec5SDimitry Andric     , SaveSchedImpl(std::move(_Sch.SchedImpl))
1920b57cec5SDimitry Andric     , SaveMaxRP(R.MaxPressure) {
1930b57cec5SDimitry Andric     Sch.SchedImpl.reset(&OverrideStrategy);
1940b57cec5SDimitry Andric     auto BB = R.Begin->getParent();
1950b57cec5SDimitry Andric     Sch.BaseClass::startBlock(BB);
1960b57cec5SDimitry Andric     Sch.BaseClass::enterRegion(BB, R.Begin, R.End, R.NumRegionInstrs);
1970b57cec5SDimitry Andric   }
1980b57cec5SDimitry Andric 
1990b57cec5SDimitry Andric   ~OverrideLegacyStrategy() {
2000b57cec5SDimitry Andric     Sch.BaseClass::exitRegion();
2010b57cec5SDimitry Andric     Sch.BaseClass::finishBlock();
2020b57cec5SDimitry Andric     Sch.SchedImpl.release();
2030b57cec5SDimitry Andric     Sch.SchedImpl = std::move(SaveSchedImpl);
2040b57cec5SDimitry Andric   }
2050b57cec5SDimitry Andric 
2060b57cec5SDimitry Andric   void schedule() {
2070b57cec5SDimitry Andric     assert(Sch.RegionBegin == Rgn.Begin && Sch.RegionEnd == Rgn.End);
2080b57cec5SDimitry Andric     LLVM_DEBUG(dbgs() << "\nScheduling ";
2090b57cec5SDimitry Andric                printRegion(dbgs(), Rgn.Begin, Rgn.End, Sch.LIS, 2));
2100b57cec5SDimitry Andric     Sch.BaseClass::schedule();
2110b57cec5SDimitry Andric 
2120b57cec5SDimitry Andric     // Unfortunatelly placeDebugValues incorrectly modifies RegionEnd, restore
2130b57cec5SDimitry Andric     Sch.RegionEnd = Rgn.End;
2140b57cec5SDimitry Andric     //assert(Rgn.End == Sch.RegionEnd);
2150b57cec5SDimitry Andric     Rgn.Begin = Sch.RegionBegin;
2160b57cec5SDimitry Andric     Rgn.MaxPressure.clear();
2170b57cec5SDimitry Andric   }
2180b57cec5SDimitry Andric 
2190b57cec5SDimitry Andric   void restoreOrder() {
2200b57cec5SDimitry Andric     assert(Sch.RegionBegin == Rgn.Begin && Sch.RegionEnd == Rgn.End);
2210b57cec5SDimitry Andric     // DAG SUnits are stored using original region's order
2220b57cec5SDimitry Andric     // so just use SUnits as the restoring schedule
2230b57cec5SDimitry Andric     Sch.scheduleRegion(Rgn, Sch.SUnits, SaveMaxRP);
2240b57cec5SDimitry Andric   }
2250b57cec5SDimitry Andric };
2260b57cec5SDimitry Andric 
2270b57cec5SDimitry Andric namespace {
2280b57cec5SDimitry Andric 
2290b57cec5SDimitry Andric // just a stub to make base class happy
2300b57cec5SDimitry Andric class SchedStrategyStub : public MachineSchedStrategy {
2310b57cec5SDimitry Andric public:
2320b57cec5SDimitry Andric   bool shouldTrackPressure() const override { return false; }
2330b57cec5SDimitry Andric   bool shouldTrackLaneMasks() const override { return false; }
2340b57cec5SDimitry Andric   void initialize(ScheduleDAGMI *DAG) override {}
2350b57cec5SDimitry Andric   SUnit *pickNode(bool &IsTopNode) override { return nullptr; }
2360b57cec5SDimitry Andric   void schedNode(SUnit *SU, bool IsTopNode) override {}
2370b57cec5SDimitry Andric   void releaseTopNode(SUnit *SU) override {}
2380b57cec5SDimitry Andric   void releaseBottomNode(SUnit *SU) override {}
2390b57cec5SDimitry Andric };
2400b57cec5SDimitry Andric 
2410b57cec5SDimitry Andric } // end anonymous namespace
2420b57cec5SDimitry Andric 
2430b57cec5SDimitry Andric GCNIterativeScheduler::GCNIterativeScheduler(MachineSchedContext *C,
2440b57cec5SDimitry Andric                                              StrategyKind S)
2458bcb0991SDimitry Andric   : BaseClass(C, std::make_unique<SchedStrategyStub>())
2460b57cec5SDimitry Andric   , Context(C)
2470b57cec5SDimitry Andric   , Strategy(S)
2480b57cec5SDimitry Andric   , UPTracker(*LIS) {
2490b57cec5SDimitry Andric }
2500b57cec5SDimitry Andric 
2510b57cec5SDimitry Andric // returns max pressure for a region
2520b57cec5SDimitry Andric GCNRegPressure
2530b57cec5SDimitry Andric GCNIterativeScheduler::getRegionPressure(MachineBasicBlock::iterator Begin,
2540b57cec5SDimitry Andric                                          MachineBasicBlock::iterator End)
2550b57cec5SDimitry Andric   const {
2560b57cec5SDimitry Andric   // For the purpose of pressure tracking bottom inst of the region should
2570b57cec5SDimitry Andric   // be also processed. End is either BB end, BB terminator inst or sched
2580b57cec5SDimitry Andric   // boundary inst.
2590b57cec5SDimitry Andric   auto const BBEnd = Begin->getParent()->end();
2600b57cec5SDimitry Andric   auto const BottomMI = End == BBEnd ? std::prev(End) : End;
2610b57cec5SDimitry Andric 
2620b57cec5SDimitry Andric   // scheduleRegions walks bottom to top, so its likely we just get next
2630b57cec5SDimitry Andric   // instruction to track
2640b57cec5SDimitry Andric   auto AfterBottomMI = std::next(BottomMI);
2650b57cec5SDimitry Andric   if (AfterBottomMI == BBEnd ||
2660b57cec5SDimitry Andric       &*AfterBottomMI != UPTracker.getLastTrackedMI()) {
2670b57cec5SDimitry Andric     UPTracker.reset(*BottomMI);
2680b57cec5SDimitry Andric   } else {
2690b57cec5SDimitry Andric     assert(UPTracker.isValid());
2700b57cec5SDimitry Andric   }
2710b57cec5SDimitry Andric 
2720b57cec5SDimitry Andric   for (auto I = BottomMI; I != Begin; --I)
2730b57cec5SDimitry Andric     UPTracker.recede(*I);
2740b57cec5SDimitry Andric 
2750b57cec5SDimitry Andric   UPTracker.recede(*Begin);
2760b57cec5SDimitry Andric 
2770b57cec5SDimitry Andric   assert(UPTracker.isValid() ||
2780b57cec5SDimitry Andric          (dbgs() << "Tracked region ",
2790b57cec5SDimitry Andric           printRegion(dbgs(), Begin, End, LIS), false));
2800b57cec5SDimitry Andric   return UPTracker.moveMaxPressure();
2810b57cec5SDimitry Andric }
2820b57cec5SDimitry Andric 
2830b57cec5SDimitry Andric // returns max pressure for a tentative schedule
2840b57cec5SDimitry Andric template <typename Range> GCNRegPressure
2850b57cec5SDimitry Andric GCNIterativeScheduler::getSchedulePressure(const Region &R,
2860b57cec5SDimitry Andric                                            Range &&Schedule) const {
2870b57cec5SDimitry Andric   auto const BBEnd = R.Begin->getParent()->end();
2880b57cec5SDimitry Andric   GCNUpwardRPTracker RPTracker(*LIS);
2890b57cec5SDimitry Andric   if (R.End != BBEnd) {
2900b57cec5SDimitry Andric     // R.End points to the boundary instruction but the
2910b57cec5SDimitry Andric     // schedule doesn't include it
2920b57cec5SDimitry Andric     RPTracker.reset(*R.End);
2930b57cec5SDimitry Andric     RPTracker.recede(*R.End);
2940b57cec5SDimitry Andric   } else {
2950b57cec5SDimitry Andric     // R.End doesn't point to the boundary instruction
2960b57cec5SDimitry Andric     RPTracker.reset(*std::prev(BBEnd));
2970b57cec5SDimitry Andric   }
2980b57cec5SDimitry Andric   for (auto I = Schedule.end(), B = Schedule.begin(); I != B;) {
2990b57cec5SDimitry Andric     RPTracker.recede(*getMachineInstr(*--I));
3000b57cec5SDimitry Andric   }
3010b57cec5SDimitry Andric   return RPTracker.moveMaxPressure();
3020b57cec5SDimitry Andric }
3030b57cec5SDimitry Andric 
3040b57cec5SDimitry Andric void GCNIterativeScheduler::enterRegion(MachineBasicBlock *BB, // overriden
3050b57cec5SDimitry Andric                                         MachineBasicBlock::iterator Begin,
3060b57cec5SDimitry Andric                                         MachineBasicBlock::iterator End,
3070b57cec5SDimitry Andric                                         unsigned NumRegionInstrs) {
3080b57cec5SDimitry Andric   BaseClass::enterRegion(BB, Begin, End, NumRegionInstrs);
3090b57cec5SDimitry Andric   if (NumRegionInstrs > 2) {
3100b57cec5SDimitry Andric     Regions.push_back(
3110b57cec5SDimitry Andric       new (Alloc.Allocate())
3120b57cec5SDimitry Andric       Region { Begin, End, NumRegionInstrs,
3130b57cec5SDimitry Andric                getRegionPressure(Begin, End), nullptr });
3140b57cec5SDimitry Andric   }
3150b57cec5SDimitry Andric }
3160b57cec5SDimitry Andric 
3170b57cec5SDimitry Andric void GCNIterativeScheduler::schedule() { // overriden
3180b57cec5SDimitry Andric   // do nothing
3190b57cec5SDimitry Andric   LLVM_DEBUG(printLivenessInfo(dbgs(), RegionBegin, RegionEnd, LIS);
3200b57cec5SDimitry Andric              if (!Regions.empty() && Regions.back()->Begin == RegionBegin) {
3210b57cec5SDimitry Andric                dbgs() << "Max RP: ";
3220b57cec5SDimitry Andric                Regions.back()->MaxPressure.print(
3230b57cec5SDimitry Andric                    dbgs(), &MF.getSubtarget<GCNSubtarget>());
3240b57cec5SDimitry Andric              } dbgs()
3250b57cec5SDimitry Andric              << '\n';);
3260b57cec5SDimitry Andric }
3270b57cec5SDimitry Andric 
3280b57cec5SDimitry Andric void GCNIterativeScheduler::finalizeSchedule() { // overriden
3290b57cec5SDimitry Andric   if (Regions.empty())
3300b57cec5SDimitry Andric     return;
3310b57cec5SDimitry Andric   switch (Strategy) {
3320b57cec5SDimitry Andric   case SCHEDULE_MINREGONLY: scheduleMinReg(); break;
3330b57cec5SDimitry Andric   case SCHEDULE_MINREGFORCED: scheduleMinReg(true); break;
3340b57cec5SDimitry Andric   case SCHEDULE_LEGACYMAXOCCUPANCY: scheduleLegacyMaxOccupancy(); break;
3350b57cec5SDimitry Andric   case SCHEDULE_ILP: scheduleILP(false); break;
3360b57cec5SDimitry Andric   }
3370b57cec5SDimitry Andric }
3380b57cec5SDimitry Andric 
3390b57cec5SDimitry Andric // Detach schedule from SUnits and interleave it with debug values.
3400b57cec5SDimitry Andric // Returned schedule becomes independent of DAG state.
3410b57cec5SDimitry Andric std::vector<MachineInstr*>
3420b57cec5SDimitry Andric GCNIterativeScheduler::detachSchedule(ScheduleRef Schedule) const {
3430b57cec5SDimitry Andric   std::vector<MachineInstr*> Res;
3440b57cec5SDimitry Andric   Res.reserve(Schedule.size() * 2);
3450b57cec5SDimitry Andric 
3460b57cec5SDimitry Andric   if (FirstDbgValue)
3470b57cec5SDimitry Andric     Res.push_back(FirstDbgValue);
3480b57cec5SDimitry Andric 
3490b57cec5SDimitry Andric   const auto DbgB = DbgValues.begin(), DbgE = DbgValues.end();
3500b57cec5SDimitry Andric   for (auto SU : Schedule) {
3510b57cec5SDimitry Andric     Res.push_back(SU->getInstr());
3520b57cec5SDimitry Andric     const auto &D = std::find_if(DbgB, DbgE, [SU](decltype(*DbgB) &P) {
3530b57cec5SDimitry Andric       return P.second == SU->getInstr();
3540b57cec5SDimitry Andric     });
3550b57cec5SDimitry Andric     if (D != DbgE)
3560b57cec5SDimitry Andric       Res.push_back(D->first);
3570b57cec5SDimitry Andric   }
3580b57cec5SDimitry Andric   return Res;
3590b57cec5SDimitry Andric }
3600b57cec5SDimitry Andric 
3610b57cec5SDimitry Andric void GCNIterativeScheduler::setBestSchedule(Region &R,
3620b57cec5SDimitry Andric                                             ScheduleRef Schedule,
3630b57cec5SDimitry Andric                                             const GCNRegPressure &MaxRP) {
3640b57cec5SDimitry Andric   R.BestSchedule.reset(
3650b57cec5SDimitry Andric     new TentativeSchedule{ detachSchedule(Schedule), MaxRP });
3660b57cec5SDimitry Andric }
3670b57cec5SDimitry Andric 
3680b57cec5SDimitry Andric void GCNIterativeScheduler::scheduleBest(Region &R) {
3690b57cec5SDimitry Andric   assert(R.BestSchedule.get() && "No schedule specified");
3700b57cec5SDimitry Andric   scheduleRegion(R, R.BestSchedule->Schedule, R.BestSchedule->MaxPressure);
3710b57cec5SDimitry Andric   R.BestSchedule.reset();
3720b57cec5SDimitry Andric }
3730b57cec5SDimitry Andric 
3740b57cec5SDimitry Andric // minimal required region scheduler, works for ranges of SUnits*,
3750b57cec5SDimitry Andric // SUnits or MachineIntrs*
3760b57cec5SDimitry Andric template <typename Range>
3770b57cec5SDimitry Andric void GCNIterativeScheduler::scheduleRegion(Region &R, Range &&Schedule,
3780b57cec5SDimitry Andric                                            const GCNRegPressure &MaxRP) {
3790b57cec5SDimitry Andric   assert(RegionBegin == R.Begin && RegionEnd == R.End);
3800b57cec5SDimitry Andric   assert(LIS != nullptr);
3810b57cec5SDimitry Andric #ifndef NDEBUG
3820b57cec5SDimitry Andric   const auto SchedMaxRP = getSchedulePressure(R, Schedule);
3830b57cec5SDimitry Andric #endif
3840b57cec5SDimitry Andric   auto BB = R.Begin->getParent();
3850b57cec5SDimitry Andric   auto Top = R.Begin;
3860b57cec5SDimitry Andric   for (const auto &I : Schedule) {
3870b57cec5SDimitry Andric     auto MI = getMachineInstr(I);
3880b57cec5SDimitry Andric     if (MI != &*Top) {
3890b57cec5SDimitry Andric       BB->remove(MI);
3900b57cec5SDimitry Andric       BB->insert(Top, MI);
3910b57cec5SDimitry Andric       if (!MI->isDebugInstr())
3920b57cec5SDimitry Andric         LIS->handleMove(*MI, true);
3930b57cec5SDimitry Andric     }
3940b57cec5SDimitry Andric     if (!MI->isDebugInstr()) {
3950b57cec5SDimitry Andric       // Reset read - undef flags and update them later.
3960b57cec5SDimitry Andric       for (auto &Op : MI->operands())
3970b57cec5SDimitry Andric         if (Op.isReg() && Op.isDef())
3980b57cec5SDimitry Andric           Op.setIsUndef(false);
3990b57cec5SDimitry Andric 
4000b57cec5SDimitry Andric       RegisterOperands RegOpers;
4010b57cec5SDimitry Andric       RegOpers.collect(*MI, *TRI, MRI, /*ShouldTrackLaneMasks*/true,
4020b57cec5SDimitry Andric                                        /*IgnoreDead*/false);
4030b57cec5SDimitry Andric       // Adjust liveness and add missing dead+read-undef flags.
4040b57cec5SDimitry Andric       auto SlotIdx = LIS->getInstructionIndex(*MI).getRegSlot();
4050b57cec5SDimitry Andric       RegOpers.adjustLaneLiveness(*LIS, MRI, SlotIdx, MI);
4060b57cec5SDimitry Andric     }
4070b57cec5SDimitry Andric     Top = std::next(MI->getIterator());
4080b57cec5SDimitry Andric   }
4090b57cec5SDimitry Andric   RegionBegin = getMachineInstr(Schedule.front());
4100b57cec5SDimitry Andric 
4110b57cec5SDimitry Andric   // Schedule consisting of MachineInstr* is considered 'detached'
4120b57cec5SDimitry Andric   // and already interleaved with debug values
4130b57cec5SDimitry Andric   if (!std::is_same<decltype(*Schedule.begin()), MachineInstr*>::value) {
4140b57cec5SDimitry Andric     placeDebugValues();
4150b57cec5SDimitry Andric     // Unfortunatelly placeDebugValues incorrectly modifies RegionEnd, restore
4160b57cec5SDimitry Andric     //assert(R.End == RegionEnd);
4170b57cec5SDimitry Andric     RegionEnd = R.End;
4180b57cec5SDimitry Andric   }
4190b57cec5SDimitry Andric 
4200b57cec5SDimitry Andric   R.Begin = RegionBegin;
4210b57cec5SDimitry Andric   R.MaxPressure = MaxRP;
4220b57cec5SDimitry Andric 
4230b57cec5SDimitry Andric #ifndef NDEBUG
4240b57cec5SDimitry Andric   const auto RegionMaxRP = getRegionPressure(R);
4250b57cec5SDimitry Andric   const auto &ST = MF.getSubtarget<GCNSubtarget>();
4260b57cec5SDimitry Andric #endif
4270b57cec5SDimitry Andric   assert((SchedMaxRP == RegionMaxRP && (MaxRP.empty() || SchedMaxRP == MaxRP))
4280b57cec5SDimitry Andric   || (dbgs() << "Max RP mismatch!!!\n"
4290b57cec5SDimitry Andric                 "RP for schedule (calculated): ",
4300b57cec5SDimitry Andric       SchedMaxRP.print(dbgs(), &ST),
4310b57cec5SDimitry Andric       dbgs() << "RP for schedule (reported): ",
4320b57cec5SDimitry Andric       MaxRP.print(dbgs(), &ST),
4330b57cec5SDimitry Andric       dbgs() << "RP after scheduling: ",
4340b57cec5SDimitry Andric       RegionMaxRP.print(dbgs(), &ST),
4350b57cec5SDimitry Andric       false));
4360b57cec5SDimitry Andric }
4370b57cec5SDimitry Andric 
4380b57cec5SDimitry Andric // Sort recorded regions by pressure - highest at the front
4390b57cec5SDimitry Andric void GCNIterativeScheduler::sortRegionsByPressure(unsigned TargetOcc) {
4400b57cec5SDimitry Andric   const auto &ST = MF.getSubtarget<GCNSubtarget>();
4410b57cec5SDimitry Andric   llvm::sort(Regions, [&ST, TargetOcc](const Region *R1, const Region *R2) {
4420b57cec5SDimitry Andric     return R2->MaxPressure.less(ST, R1->MaxPressure, TargetOcc);
4430b57cec5SDimitry Andric   });
4440b57cec5SDimitry Andric }
4450b57cec5SDimitry Andric 
4460b57cec5SDimitry Andric ///////////////////////////////////////////////////////////////////////////////
4470b57cec5SDimitry Andric // Legacy MaxOccupancy Strategy
4480b57cec5SDimitry Andric 
4490b57cec5SDimitry Andric // Tries to increase occupancy applying minreg scheduler for a sequence of
4500b57cec5SDimitry Andric // most demanding regions. Obtained schedules are saved as BestSchedule for a
4510b57cec5SDimitry Andric // region.
4520b57cec5SDimitry Andric // TargetOcc is the best achievable occupancy for a kernel.
4530b57cec5SDimitry Andric // Returns better occupancy on success or current occupancy on fail.
4540b57cec5SDimitry Andric // BestSchedules aren't deleted on fail.
4550b57cec5SDimitry Andric unsigned GCNIterativeScheduler::tryMaximizeOccupancy(unsigned TargetOcc) {
4560b57cec5SDimitry Andric   // TODO: assert Regions are sorted descending by pressure
4570b57cec5SDimitry Andric   const auto &ST = MF.getSubtarget<GCNSubtarget>();
4580b57cec5SDimitry Andric   const auto Occ = Regions.front()->MaxPressure.getOccupancy(ST);
4590b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "Trying to improve occupancy, target = " << TargetOcc
4600b57cec5SDimitry Andric                     << ", current = " << Occ << '\n');
4610b57cec5SDimitry Andric 
4620b57cec5SDimitry Andric   auto NewOcc = TargetOcc;
4630b57cec5SDimitry Andric   for (auto R : Regions) {
4640b57cec5SDimitry Andric     if (R->MaxPressure.getOccupancy(ST) >= NewOcc)
4650b57cec5SDimitry Andric       break;
4660b57cec5SDimitry Andric 
4670b57cec5SDimitry Andric     LLVM_DEBUG(printRegion(dbgs(), R->Begin, R->End, LIS, 3);
4680b57cec5SDimitry Andric                printLivenessInfo(dbgs(), R->Begin, R->End, LIS));
4690b57cec5SDimitry Andric 
4700b57cec5SDimitry Andric     BuildDAG DAG(*R, *this);
4710b57cec5SDimitry Andric     const auto MinSchedule = makeMinRegSchedule(DAG.getTopRoots(), *this);
4720b57cec5SDimitry Andric     const auto MaxRP = getSchedulePressure(*R, MinSchedule);
4730b57cec5SDimitry Andric     LLVM_DEBUG(dbgs() << "Occupancy improvement attempt:\n";
4740b57cec5SDimitry Andric                printSchedRP(dbgs(), R->MaxPressure, MaxRP));
4750b57cec5SDimitry Andric 
4760b57cec5SDimitry Andric     NewOcc = std::min(NewOcc, MaxRP.getOccupancy(ST));
4770b57cec5SDimitry Andric     if (NewOcc <= Occ)
4780b57cec5SDimitry Andric       break;
4790b57cec5SDimitry Andric 
4800b57cec5SDimitry Andric     setBestSchedule(*R, MinSchedule, MaxRP);
4810b57cec5SDimitry Andric   }
4820b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "New occupancy = " << NewOcc
4830b57cec5SDimitry Andric                     << ", prev occupancy = " << Occ << '\n');
4840b57cec5SDimitry Andric   if (NewOcc > Occ) {
4850b57cec5SDimitry Andric     SIMachineFunctionInfo *MFI = MF.getInfo<SIMachineFunctionInfo>();
4860b57cec5SDimitry Andric     MFI->increaseOccupancy(MF, NewOcc);
4870b57cec5SDimitry Andric   }
4880b57cec5SDimitry Andric 
4890b57cec5SDimitry Andric   return std::max(NewOcc, Occ);
4900b57cec5SDimitry Andric }
4910b57cec5SDimitry Andric 
4920b57cec5SDimitry Andric void GCNIterativeScheduler::scheduleLegacyMaxOccupancy(
4930b57cec5SDimitry Andric   bool TryMaximizeOccupancy) {
4940b57cec5SDimitry Andric   const auto &ST = MF.getSubtarget<GCNSubtarget>();
4950b57cec5SDimitry Andric   SIMachineFunctionInfo *MFI = MF.getInfo<SIMachineFunctionInfo>();
4960b57cec5SDimitry Andric   auto TgtOcc = MFI->getMinAllowedOccupancy();
4970b57cec5SDimitry Andric 
4980b57cec5SDimitry Andric   sortRegionsByPressure(TgtOcc);
4990b57cec5SDimitry Andric   auto Occ = Regions.front()->MaxPressure.getOccupancy(ST);
5000b57cec5SDimitry Andric 
5010b57cec5SDimitry Andric   if (TryMaximizeOccupancy && Occ < TgtOcc)
5020b57cec5SDimitry Andric     Occ = tryMaximizeOccupancy(TgtOcc);
5030b57cec5SDimitry Andric 
5040b57cec5SDimitry Andric   // This is really weird but for some magic scheduling regions twice
5050b57cec5SDimitry Andric   // gives performance improvement
5060b57cec5SDimitry Andric   const int NumPasses = Occ < TgtOcc ? 2 : 1;
5070b57cec5SDimitry Andric 
5080b57cec5SDimitry Andric   TgtOcc = std::min(Occ, TgtOcc);
5090b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "Scheduling using default scheduler, "
5100b57cec5SDimitry Andric                        "target occupancy = "
5110b57cec5SDimitry Andric                     << TgtOcc << '\n');
5120b57cec5SDimitry Andric   GCNMaxOccupancySchedStrategy LStrgy(Context);
5130b57cec5SDimitry Andric   unsigned FinalOccupancy = std::min(Occ, MFI->getOccupancy());
5140b57cec5SDimitry Andric 
5150b57cec5SDimitry Andric   for (int I = 0; I < NumPasses; ++I) {
5160b57cec5SDimitry Andric     // running first pass with TargetOccupancy = 0 mimics previous scheduling
5170b57cec5SDimitry Andric     // approach and is a performance magic
5180b57cec5SDimitry Andric     LStrgy.setTargetOccupancy(I == 0 ? 0 : TgtOcc);
5190b57cec5SDimitry Andric     for (auto R : Regions) {
5200b57cec5SDimitry Andric       OverrideLegacyStrategy Ovr(*R, LStrgy, *this);
5210b57cec5SDimitry Andric 
5220b57cec5SDimitry Andric       Ovr.schedule();
5230b57cec5SDimitry Andric       const auto RP = getRegionPressure(*R);
5240b57cec5SDimitry Andric       LLVM_DEBUG(printSchedRP(dbgs(), R->MaxPressure, RP));
5250b57cec5SDimitry Andric 
5260b57cec5SDimitry Andric       if (RP.getOccupancy(ST) < TgtOcc) {
5270b57cec5SDimitry Andric         LLVM_DEBUG(dbgs() << "Didn't fit into target occupancy O" << TgtOcc);
5280b57cec5SDimitry Andric         if (R->BestSchedule.get() &&
5290b57cec5SDimitry Andric             R->BestSchedule->MaxPressure.getOccupancy(ST) >= TgtOcc) {
5300b57cec5SDimitry Andric           LLVM_DEBUG(dbgs() << ", scheduling minimal register\n");
5310b57cec5SDimitry Andric           scheduleBest(*R);
5320b57cec5SDimitry Andric         } else {
5330b57cec5SDimitry Andric           LLVM_DEBUG(dbgs() << ", restoring\n");
5340b57cec5SDimitry Andric           Ovr.restoreOrder();
5350b57cec5SDimitry Andric           assert(R->MaxPressure.getOccupancy(ST) >= TgtOcc);
5360b57cec5SDimitry Andric         }
5370b57cec5SDimitry Andric       }
5380b57cec5SDimitry Andric       FinalOccupancy = std::min(FinalOccupancy, RP.getOccupancy(ST));
5390b57cec5SDimitry Andric     }
5400b57cec5SDimitry Andric   }
5410b57cec5SDimitry Andric   MFI->limitOccupancy(FinalOccupancy);
5420b57cec5SDimitry Andric }
5430b57cec5SDimitry Andric 
5440b57cec5SDimitry Andric ///////////////////////////////////////////////////////////////////////////////
5450b57cec5SDimitry Andric // Minimal Register Strategy
5460b57cec5SDimitry Andric 
5470b57cec5SDimitry Andric void GCNIterativeScheduler::scheduleMinReg(bool force) {
5480b57cec5SDimitry Andric   const auto &ST = MF.getSubtarget<GCNSubtarget>();
5490b57cec5SDimitry Andric   const SIMachineFunctionInfo *MFI = MF.getInfo<SIMachineFunctionInfo>();
5500b57cec5SDimitry Andric   const auto TgtOcc = MFI->getOccupancy();
5510b57cec5SDimitry Andric   sortRegionsByPressure(TgtOcc);
5520b57cec5SDimitry Andric 
5530b57cec5SDimitry Andric   auto MaxPressure = Regions.front()->MaxPressure;
5540b57cec5SDimitry Andric   for (auto R : Regions) {
5550b57cec5SDimitry Andric     if (!force && R->MaxPressure.less(ST, MaxPressure, TgtOcc))
5560b57cec5SDimitry Andric       break;
5570b57cec5SDimitry Andric 
5580b57cec5SDimitry Andric     BuildDAG DAG(*R, *this);
5590b57cec5SDimitry Andric     const auto MinSchedule = makeMinRegSchedule(DAG.getTopRoots(), *this);
5600b57cec5SDimitry Andric 
5610b57cec5SDimitry Andric     const auto RP = getSchedulePressure(*R, MinSchedule);
5620b57cec5SDimitry Andric     LLVM_DEBUG(if (R->MaxPressure.less(ST, RP, TgtOcc)) {
5630b57cec5SDimitry Andric       dbgs() << "\nWarning: Pressure becomes worse after minreg!";
5640b57cec5SDimitry Andric       printSchedRP(dbgs(), R->MaxPressure, RP);
5650b57cec5SDimitry Andric     });
5660b57cec5SDimitry Andric 
5670b57cec5SDimitry Andric     if (!force && MaxPressure.less(ST, RP, TgtOcc))
5680b57cec5SDimitry Andric       break;
5690b57cec5SDimitry Andric 
5700b57cec5SDimitry Andric     scheduleRegion(*R, MinSchedule, RP);
5710b57cec5SDimitry Andric     LLVM_DEBUG(printSchedResult(dbgs(), R, RP));
5720b57cec5SDimitry Andric 
5730b57cec5SDimitry Andric     MaxPressure = RP;
5740b57cec5SDimitry Andric   }
5750b57cec5SDimitry Andric }
5760b57cec5SDimitry Andric 
5770b57cec5SDimitry Andric ///////////////////////////////////////////////////////////////////////////////
5780b57cec5SDimitry Andric // ILP scheduler port
5790b57cec5SDimitry Andric 
5800b57cec5SDimitry Andric void GCNIterativeScheduler::scheduleILP(
5810b57cec5SDimitry Andric   bool TryMaximizeOccupancy) {
5820b57cec5SDimitry Andric   const auto &ST = MF.getSubtarget<GCNSubtarget>();
5830b57cec5SDimitry Andric   SIMachineFunctionInfo *MFI = MF.getInfo<SIMachineFunctionInfo>();
5840b57cec5SDimitry Andric   auto TgtOcc = MFI->getMinAllowedOccupancy();
5850b57cec5SDimitry Andric 
5860b57cec5SDimitry Andric   sortRegionsByPressure(TgtOcc);
5870b57cec5SDimitry Andric   auto Occ = Regions.front()->MaxPressure.getOccupancy(ST);
5880b57cec5SDimitry Andric 
5890b57cec5SDimitry Andric   if (TryMaximizeOccupancy && Occ < TgtOcc)
5900b57cec5SDimitry Andric     Occ = tryMaximizeOccupancy(TgtOcc);
5910b57cec5SDimitry Andric 
5920b57cec5SDimitry Andric   TgtOcc = std::min(Occ, TgtOcc);
5930b57cec5SDimitry Andric   LLVM_DEBUG(dbgs() << "Scheduling using default scheduler, "
5940b57cec5SDimitry Andric                        "target occupancy = "
5950b57cec5SDimitry Andric                     << TgtOcc << '\n');
5960b57cec5SDimitry Andric 
5970b57cec5SDimitry Andric   unsigned FinalOccupancy = std::min(Occ, MFI->getOccupancy());
5980b57cec5SDimitry Andric   for (auto R : Regions) {
5990b57cec5SDimitry Andric     BuildDAG DAG(*R, *this);
6000b57cec5SDimitry Andric     const auto ILPSchedule = makeGCNILPScheduler(DAG.getBottomRoots(), *this);
6010b57cec5SDimitry Andric 
6020b57cec5SDimitry Andric     const auto RP = getSchedulePressure(*R, ILPSchedule);
6030b57cec5SDimitry Andric     LLVM_DEBUG(printSchedRP(dbgs(), R->MaxPressure, RP));
6040b57cec5SDimitry Andric 
6050b57cec5SDimitry Andric     if (RP.getOccupancy(ST) < TgtOcc) {
6060b57cec5SDimitry Andric       LLVM_DEBUG(dbgs() << "Didn't fit into target occupancy O" << TgtOcc);
6070b57cec5SDimitry Andric       if (R->BestSchedule.get() &&
6080b57cec5SDimitry Andric         R->BestSchedule->MaxPressure.getOccupancy(ST) >= TgtOcc) {
6090b57cec5SDimitry Andric         LLVM_DEBUG(dbgs() << ", scheduling minimal register\n");
6100b57cec5SDimitry Andric         scheduleBest(*R);
6110b57cec5SDimitry Andric       }
6120b57cec5SDimitry Andric     } else {
6130b57cec5SDimitry Andric       scheduleRegion(*R, ILPSchedule, RP);
6140b57cec5SDimitry Andric       LLVM_DEBUG(printSchedResult(dbgs(), R, RP));
6150b57cec5SDimitry Andric       FinalOccupancy = std::min(FinalOccupancy, RP.getOccupancy(ST));
6160b57cec5SDimitry Andric     }
6170b57cec5SDimitry Andric   }
6180b57cec5SDimitry Andric   MFI->limitOccupancy(FinalOccupancy);
6190b57cec5SDimitry Andric }
620