|  | //===- GCNIterativeScheduler.cpp ------------------------------------------===// | 
|  | // | 
|  | //                     The LLVM Compiler Infrastructure | 
|  | // | 
|  | // This file is distributed under the University of Illinois Open Source | 
|  | // License. See LICENSE.TXT for details. | 
|  | // | 
|  | //===----------------------------------------------------------------------===// | 
|  |  | 
|  | #include "GCNIterativeScheduler.h" | 
|  | #include "AMDGPUSubtarget.h" | 
|  | #include "GCNRegPressure.h" | 
|  | #include "GCNSchedStrategy.h" | 
|  | #include "SIMachineFunctionInfo.h" | 
|  | #include "llvm/ADT/ArrayRef.h" | 
|  | #include "llvm/ADT/STLExtras.h" | 
|  | #include "llvm/ADT/SmallVector.h" | 
|  | #include "llvm/CodeGen/LiveIntervals.h" | 
|  | #include "llvm/CodeGen/MachineBasicBlock.h" | 
|  | #include "llvm/CodeGen/MachineFunction.h" | 
|  | #include "llvm/CodeGen/RegisterPressure.h" | 
|  | #include "llvm/CodeGen/ScheduleDAG.h" | 
|  | #include "llvm/Config/llvm-config.h" | 
|  | #include "llvm/Support/Compiler.h" | 
|  | #include "llvm/Support/Debug.h" | 
|  | #include "llvm/Support/raw_ostream.h" | 
|  | #include <algorithm> | 
|  | #include <cassert> | 
|  | #include <iterator> | 
|  | #include <limits> | 
|  | #include <memory> | 
|  | #include <type_traits> | 
|  | #include <vector> | 
|  |  | 
|  | using namespace llvm; | 
|  |  | 
|  | #define DEBUG_TYPE "machine-scheduler" | 
|  |  | 
|  | namespace llvm { | 
|  |  | 
|  | std::vector<const SUnit *> makeMinRegSchedule(ArrayRef<const SUnit *> TopRoots, | 
|  | const ScheduleDAG &DAG); | 
|  |  | 
|  | std::vector<const SUnit*> makeGCNILPScheduler(ArrayRef<const SUnit*> BotRoots, | 
|  | const ScheduleDAG &DAG); | 
|  | } | 
|  |  | 
|  | // shim accessors for different order containers | 
|  | static inline MachineInstr *getMachineInstr(MachineInstr *MI) { | 
|  | return MI; | 
|  | } | 
|  | static inline MachineInstr *getMachineInstr(const SUnit *SU) { | 
|  | return SU->getInstr(); | 
|  | } | 
|  | static inline MachineInstr *getMachineInstr(const SUnit &SU) { | 
|  | return SU.getInstr(); | 
|  | } | 
|  |  | 
|  | #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) | 
|  | LLVM_DUMP_METHOD | 
|  | static void printRegion(raw_ostream &OS, | 
|  | MachineBasicBlock::iterator Begin, | 
|  | MachineBasicBlock::iterator End, | 
|  | const LiveIntervals *LIS, | 
|  | unsigned MaxInstNum = | 
|  | std::numeric_limits<unsigned>::max()) { | 
|  | auto BB = Begin->getParent(); | 
|  | OS << BB->getParent()->getName() << ":" << printMBBReference(*BB) << ' ' | 
|  | << BB->getName() << ":\n"; | 
|  | auto I = Begin; | 
|  | MaxInstNum = std::max(MaxInstNum, 1u); | 
|  | for (; I != End && MaxInstNum; ++I, --MaxInstNum) { | 
|  | if (!I->isDebugInstr() && LIS) | 
|  | OS << LIS->getInstructionIndex(*I); | 
|  | OS << '\t' << *I; | 
|  | } | 
|  | if (I != End) { | 
|  | OS << "\t...\n"; | 
|  | I = std::prev(End); | 
|  | if (!I->isDebugInstr() && LIS) | 
|  | OS << LIS->getInstructionIndex(*I); | 
|  | OS << '\t' << *I; | 
|  | } | 
|  | if (End != BB->end()) { // print boundary inst if present | 
|  | OS << "----\n"; | 
|  | if (LIS) OS << LIS->getInstructionIndex(*End) << '\t'; | 
|  | OS << *End; | 
|  | } | 
|  | } | 
|  |  | 
|  | LLVM_DUMP_METHOD | 
|  | static void printLivenessInfo(raw_ostream &OS, | 
|  | MachineBasicBlock::iterator Begin, | 
|  | MachineBasicBlock::iterator End, | 
|  | const LiveIntervals *LIS) { | 
|  | const auto BB = Begin->getParent(); | 
|  | const auto &MRI = BB->getParent()->getRegInfo(); | 
|  |  | 
|  | const auto LiveIns = getLiveRegsBefore(*Begin, *LIS); | 
|  | OS << "LIn RP: "; | 
|  | getRegPressure(MRI, LiveIns).print(OS); | 
|  |  | 
|  | const auto BottomMI = End == BB->end() ? std::prev(End) : End; | 
|  | const auto LiveOuts = getLiveRegsAfter(*BottomMI, *LIS); | 
|  | OS << "LOt RP: "; | 
|  | getRegPressure(MRI, LiveOuts).print(OS); | 
|  | } | 
|  |  | 
|  | LLVM_DUMP_METHOD | 
|  | void GCNIterativeScheduler::printRegions(raw_ostream &OS) const { | 
|  | const auto &ST = MF.getSubtarget<GCNSubtarget>(); | 
|  | for (const auto R : Regions) { | 
|  | OS << "Region to schedule "; | 
|  | printRegion(OS, R->Begin, R->End, LIS, 1); | 
|  | printLivenessInfo(OS, R->Begin, R->End, LIS); | 
|  | OS << "Max RP: "; | 
|  | R->MaxPressure.print(OS, &ST); | 
|  | } | 
|  | } | 
|  |  | 
|  | LLVM_DUMP_METHOD | 
|  | void GCNIterativeScheduler::printSchedResult(raw_ostream &OS, | 
|  | const Region *R, | 
|  | const GCNRegPressure &RP) const { | 
|  | OS << "\nAfter scheduling "; | 
|  | printRegion(OS, R->Begin, R->End, LIS); | 
|  | printSchedRP(OS, R->MaxPressure, RP); | 
|  | OS << '\n'; | 
|  | } | 
|  |  | 
|  | LLVM_DUMP_METHOD | 
|  | void GCNIterativeScheduler::printSchedRP(raw_ostream &OS, | 
|  | const GCNRegPressure &Before, | 
|  | const GCNRegPressure &After) const { | 
|  | const auto &ST = MF.getSubtarget<GCNSubtarget>(); | 
|  | OS << "RP before: "; | 
|  | Before.print(OS, &ST); | 
|  | OS << "RP after:  "; | 
|  | After.print(OS, &ST); | 
|  | } | 
|  | #endif | 
|  |  | 
|  | // DAG builder helper | 
|  | class GCNIterativeScheduler::BuildDAG { | 
|  | GCNIterativeScheduler &Sch; | 
|  | SmallVector<SUnit *, 8> TopRoots; | 
|  |  | 
|  | SmallVector<SUnit*, 8> BotRoots; | 
|  | public: | 
|  | BuildDAG(const Region &R, GCNIterativeScheduler &_Sch) | 
|  | : Sch(_Sch) { | 
|  | auto BB = R.Begin->getParent(); | 
|  | Sch.BaseClass::startBlock(BB); | 
|  | Sch.BaseClass::enterRegion(BB, R.Begin, R.End, R.NumRegionInstrs); | 
|  |  | 
|  | Sch.buildSchedGraph(Sch.AA, nullptr, nullptr, nullptr, | 
|  | /*TrackLaneMask*/true); | 
|  | Sch.Topo.InitDAGTopologicalSorting(); | 
|  | Sch.findRootsAndBiasEdges(TopRoots, BotRoots); | 
|  | } | 
|  |  | 
|  | ~BuildDAG() { | 
|  | Sch.BaseClass::exitRegion(); | 
|  | Sch.BaseClass::finishBlock(); | 
|  | } | 
|  |  | 
|  | ArrayRef<const SUnit *> getTopRoots() const { | 
|  | return TopRoots; | 
|  | } | 
|  | ArrayRef<SUnit*> getBottomRoots() const { | 
|  | return BotRoots; | 
|  | } | 
|  | }; | 
|  |  | 
|  | class GCNIterativeScheduler::OverrideLegacyStrategy { | 
|  | GCNIterativeScheduler &Sch; | 
|  | Region &Rgn; | 
|  | std::unique_ptr<MachineSchedStrategy> SaveSchedImpl; | 
|  | GCNRegPressure SaveMaxRP; | 
|  |  | 
|  | public: | 
|  | OverrideLegacyStrategy(Region &R, | 
|  | MachineSchedStrategy &OverrideStrategy, | 
|  | GCNIterativeScheduler &_Sch) | 
|  | : Sch(_Sch) | 
|  | , Rgn(R) | 
|  | , SaveSchedImpl(std::move(_Sch.SchedImpl)) | 
|  | , SaveMaxRP(R.MaxPressure) { | 
|  | Sch.SchedImpl.reset(&OverrideStrategy); | 
|  | auto BB = R.Begin->getParent(); | 
|  | Sch.BaseClass::startBlock(BB); | 
|  | Sch.BaseClass::enterRegion(BB, R.Begin, R.End, R.NumRegionInstrs); | 
|  | } | 
|  |  | 
|  | ~OverrideLegacyStrategy() { | 
|  | Sch.BaseClass::exitRegion(); | 
|  | Sch.BaseClass::finishBlock(); | 
|  | Sch.SchedImpl.release(); | 
|  | Sch.SchedImpl = std::move(SaveSchedImpl); | 
|  | } | 
|  |  | 
|  | void schedule() { | 
|  | assert(Sch.RegionBegin == Rgn.Begin && Sch.RegionEnd == Rgn.End); | 
|  | LLVM_DEBUG(dbgs() << "\nScheduling "; | 
|  | printRegion(dbgs(), Rgn.Begin, Rgn.End, Sch.LIS, 2)); | 
|  | Sch.BaseClass::schedule(); | 
|  |  | 
|  | // Unfortunatelly placeDebugValues incorrectly modifies RegionEnd, restore | 
|  | Sch.RegionEnd = Rgn.End; | 
|  | //assert(Rgn.End == Sch.RegionEnd); | 
|  | Rgn.Begin = Sch.RegionBegin; | 
|  | Rgn.MaxPressure.clear(); | 
|  | } | 
|  |  | 
|  | void restoreOrder() { | 
|  | assert(Sch.RegionBegin == Rgn.Begin && Sch.RegionEnd == Rgn.End); | 
|  | // DAG SUnits are stored using original region's order | 
|  | // so just use SUnits as the restoring schedule | 
|  | Sch.scheduleRegion(Rgn, Sch.SUnits, SaveMaxRP); | 
|  | } | 
|  | }; | 
|  |  | 
|  | namespace { | 
|  |  | 
|  | // just a stub to make base class happy | 
|  | class SchedStrategyStub : public MachineSchedStrategy { | 
|  | public: | 
|  | bool shouldTrackPressure() const override { return false; } | 
|  | bool shouldTrackLaneMasks() const override { return false; } | 
|  | void initialize(ScheduleDAGMI *DAG) override {} | 
|  | SUnit *pickNode(bool &IsTopNode) override { return nullptr; } | 
|  | void schedNode(SUnit *SU, bool IsTopNode) override {} | 
|  | void releaseTopNode(SUnit *SU) override {} | 
|  | void releaseBottomNode(SUnit *SU) override {} | 
|  | }; | 
|  |  | 
|  | } // end anonymous namespace | 
|  |  | 
|  | GCNIterativeScheduler::GCNIterativeScheduler(MachineSchedContext *C, | 
|  | StrategyKind S) | 
|  | : BaseClass(C, llvm::make_unique<SchedStrategyStub>()) | 
|  | , Context(C) | 
|  | , Strategy(S) | 
|  | , UPTracker(*LIS) { | 
|  | } | 
|  |  | 
|  | // returns max pressure for a region | 
|  | GCNRegPressure | 
|  | GCNIterativeScheduler::getRegionPressure(MachineBasicBlock::iterator Begin, | 
|  | MachineBasicBlock::iterator End) | 
|  | const { | 
|  | // For the purpose of pressure tracking bottom inst of the region should | 
|  | // be also processed. End is either BB end, BB terminator inst or sched | 
|  | // boundary inst. | 
|  | auto const BBEnd = Begin->getParent()->end(); | 
|  | auto const BottomMI = End == BBEnd ? std::prev(End) : End; | 
|  |  | 
|  | // scheduleRegions walks bottom to top, so its likely we just get next | 
|  | // instruction to track | 
|  | auto AfterBottomMI = std::next(BottomMI); | 
|  | if (AfterBottomMI == BBEnd || | 
|  | &*AfterBottomMI != UPTracker.getLastTrackedMI()) { | 
|  | UPTracker.reset(*BottomMI); | 
|  | } else { | 
|  | assert(UPTracker.isValid()); | 
|  | } | 
|  |  | 
|  | for (auto I = BottomMI; I != Begin; --I) | 
|  | UPTracker.recede(*I); | 
|  |  | 
|  | UPTracker.recede(*Begin); | 
|  |  | 
|  | assert(UPTracker.isValid() || | 
|  | (dbgs() << "Tracked region ", | 
|  | printRegion(dbgs(), Begin, End, LIS), false)); | 
|  | return UPTracker.moveMaxPressure(); | 
|  | } | 
|  |  | 
|  | // returns max pressure for a tentative schedule | 
|  | template <typename Range> GCNRegPressure | 
|  | GCNIterativeScheduler::getSchedulePressure(const Region &R, | 
|  | Range &&Schedule) const { | 
|  | auto const BBEnd = R.Begin->getParent()->end(); | 
|  | GCNUpwardRPTracker RPTracker(*LIS); | 
|  | if (R.End != BBEnd) { | 
|  | // R.End points to the boundary instruction but the | 
|  | // schedule doesn't include it | 
|  | RPTracker.reset(*R.End); | 
|  | RPTracker.recede(*R.End); | 
|  | } else { | 
|  | // R.End doesn't point to the boundary instruction | 
|  | RPTracker.reset(*std::prev(BBEnd)); | 
|  | } | 
|  | for (auto I = Schedule.end(), B = Schedule.begin(); I != B;) { | 
|  | RPTracker.recede(*getMachineInstr(*--I)); | 
|  | } | 
|  | return RPTracker.moveMaxPressure(); | 
|  | } | 
|  |  | 
|  | void GCNIterativeScheduler::enterRegion(MachineBasicBlock *BB, // overriden | 
|  | MachineBasicBlock::iterator Begin, | 
|  | MachineBasicBlock::iterator End, | 
|  | unsigned NumRegionInstrs) { | 
|  | BaseClass::enterRegion(BB, Begin, End, NumRegionInstrs); | 
|  | if (NumRegionInstrs > 2) { | 
|  | Regions.push_back( | 
|  | new (Alloc.Allocate()) | 
|  | Region { Begin, End, NumRegionInstrs, | 
|  | getRegionPressure(Begin, End), nullptr }); | 
|  | } | 
|  | } | 
|  |  | 
|  | void GCNIterativeScheduler::schedule() { // overriden | 
|  | // do nothing | 
|  | LLVM_DEBUG(printLivenessInfo(dbgs(), RegionBegin, RegionEnd, LIS); | 
|  | if (!Regions.empty() && Regions.back()->Begin == RegionBegin) { | 
|  | dbgs() << "Max RP: "; | 
|  | Regions.back()->MaxPressure.print( | 
|  | dbgs(), &MF.getSubtarget<GCNSubtarget>()); | 
|  | } dbgs() | 
|  | << '\n';); | 
|  | } | 
|  |  | 
|  | void GCNIterativeScheduler::finalizeSchedule() { // overriden | 
|  | if (Regions.empty()) | 
|  | return; | 
|  | switch (Strategy) { | 
|  | case SCHEDULE_MINREGONLY: scheduleMinReg(); break; | 
|  | case SCHEDULE_MINREGFORCED: scheduleMinReg(true); break; | 
|  | case SCHEDULE_LEGACYMAXOCCUPANCY: scheduleLegacyMaxOccupancy(); break; | 
|  | case SCHEDULE_ILP: scheduleILP(false); break; | 
|  | } | 
|  | } | 
|  |  | 
|  | // Detach schedule from SUnits and interleave it with debug values. | 
|  | // Returned schedule becomes independent of DAG state. | 
|  | std::vector<MachineInstr*> | 
|  | GCNIterativeScheduler::detachSchedule(ScheduleRef Schedule) const { | 
|  | std::vector<MachineInstr*> Res; | 
|  | Res.reserve(Schedule.size() * 2); | 
|  |  | 
|  | if (FirstDbgValue) | 
|  | Res.push_back(FirstDbgValue); | 
|  |  | 
|  | const auto DbgB = DbgValues.begin(), DbgE = DbgValues.end(); | 
|  | for (auto SU : Schedule) { | 
|  | Res.push_back(SU->getInstr()); | 
|  | const auto &D = std::find_if(DbgB, DbgE, [SU](decltype(*DbgB) &P) { | 
|  | return P.second == SU->getInstr(); | 
|  | }); | 
|  | if (D != DbgE) | 
|  | Res.push_back(D->first); | 
|  | } | 
|  | return Res; | 
|  | } | 
|  |  | 
|  | void GCNIterativeScheduler::setBestSchedule(Region &R, | 
|  | ScheduleRef Schedule, | 
|  | const GCNRegPressure &MaxRP) { | 
|  | R.BestSchedule.reset( | 
|  | new TentativeSchedule{ detachSchedule(Schedule), MaxRP }); | 
|  | } | 
|  |  | 
|  | void GCNIterativeScheduler::scheduleBest(Region &R) { | 
|  | assert(R.BestSchedule.get() && "No schedule specified"); | 
|  | scheduleRegion(R, R.BestSchedule->Schedule, R.BestSchedule->MaxPressure); | 
|  | R.BestSchedule.reset(); | 
|  | } | 
|  |  | 
|  | // minimal required region scheduler, works for ranges of SUnits*, | 
|  | // SUnits or MachineIntrs* | 
|  | template <typename Range> | 
|  | void GCNIterativeScheduler::scheduleRegion(Region &R, Range &&Schedule, | 
|  | const GCNRegPressure &MaxRP) { | 
|  | assert(RegionBegin == R.Begin && RegionEnd == R.End); | 
|  | assert(LIS != nullptr); | 
|  | #ifndef NDEBUG | 
|  | const auto SchedMaxRP = getSchedulePressure(R, Schedule); | 
|  | #endif | 
|  | auto BB = R.Begin->getParent(); | 
|  | auto Top = R.Begin; | 
|  | for (const auto &I : Schedule) { | 
|  | auto MI = getMachineInstr(I); | 
|  | if (MI != &*Top) { | 
|  | BB->remove(MI); | 
|  | BB->insert(Top, MI); | 
|  | if (!MI->isDebugInstr()) | 
|  | LIS->handleMove(*MI, true); | 
|  | } | 
|  | if (!MI->isDebugInstr()) { | 
|  | // Reset read - undef flags and update them later. | 
|  | for (auto &Op : MI->operands()) | 
|  | if (Op.isReg() && Op.isDef()) | 
|  | Op.setIsUndef(false); | 
|  |  | 
|  | RegisterOperands RegOpers; | 
|  | RegOpers.collect(*MI, *TRI, MRI, /*ShouldTrackLaneMasks*/true, | 
|  | /*IgnoreDead*/false); | 
|  | // Adjust liveness and add missing dead+read-undef flags. | 
|  | auto SlotIdx = LIS->getInstructionIndex(*MI).getRegSlot(); | 
|  | RegOpers.adjustLaneLiveness(*LIS, MRI, SlotIdx, MI); | 
|  | } | 
|  | Top = std::next(MI->getIterator()); | 
|  | } | 
|  | RegionBegin = getMachineInstr(Schedule.front()); | 
|  |  | 
|  | // Schedule consisting of MachineInstr* is considered 'detached' | 
|  | // and already interleaved with debug values | 
|  | if (!std::is_same<decltype(*Schedule.begin()), MachineInstr*>::value) { | 
|  | placeDebugValues(); | 
|  | // Unfortunatelly placeDebugValues incorrectly modifies RegionEnd, restore | 
|  | //assert(R.End == RegionEnd); | 
|  | RegionEnd = R.End; | 
|  | } | 
|  |  | 
|  | R.Begin = RegionBegin; | 
|  | R.MaxPressure = MaxRP; | 
|  |  | 
|  | #ifndef NDEBUG | 
|  | const auto RegionMaxRP = getRegionPressure(R); | 
|  | const auto &ST = MF.getSubtarget<GCNSubtarget>(); | 
|  | #endif | 
|  | assert((SchedMaxRP == RegionMaxRP && (MaxRP.empty() || SchedMaxRP == MaxRP)) | 
|  | || (dbgs() << "Max RP mismatch!!!\n" | 
|  | "RP for schedule (calculated): ", | 
|  | SchedMaxRP.print(dbgs(), &ST), | 
|  | dbgs() << "RP for schedule (reported): ", | 
|  | MaxRP.print(dbgs(), &ST), | 
|  | dbgs() << "RP after scheduling: ", | 
|  | RegionMaxRP.print(dbgs(), &ST), | 
|  | false)); | 
|  | } | 
|  |  | 
|  | // Sort recorded regions by pressure - highest at the front | 
|  | void GCNIterativeScheduler::sortRegionsByPressure(unsigned TargetOcc) { | 
|  | const auto &ST = MF.getSubtarget<GCNSubtarget>(); | 
|  | llvm::sort(Regions.begin(), Regions.end(), | 
|  | [&ST, TargetOcc](const Region *R1, const Region *R2) { | 
|  | return R2->MaxPressure.less(ST, R1->MaxPressure, TargetOcc); | 
|  | }); | 
|  | } | 
|  |  | 
|  | /////////////////////////////////////////////////////////////////////////////// | 
|  | // Legacy MaxOccupancy Strategy | 
|  |  | 
|  | // Tries to increase occupancy applying minreg scheduler for a sequence of | 
|  | // most demanding regions. Obtained schedules are saved as BestSchedule for a | 
|  | // region. | 
|  | // TargetOcc is the best achievable occupancy for a kernel. | 
|  | // Returns better occupancy on success or current occupancy on fail. | 
|  | // BestSchedules aren't deleted on fail. | 
|  | unsigned GCNIterativeScheduler::tryMaximizeOccupancy(unsigned TargetOcc) { | 
|  | // TODO: assert Regions are sorted descending by pressure | 
|  | const auto &ST = MF.getSubtarget<GCNSubtarget>(); | 
|  | const auto Occ = Regions.front()->MaxPressure.getOccupancy(ST); | 
|  | LLVM_DEBUG(dbgs() << "Trying to improve occupancy, target = " << TargetOcc | 
|  | << ", current = " << Occ << '\n'); | 
|  |  | 
|  | auto NewOcc = TargetOcc; | 
|  | for (auto R : Regions) { | 
|  | if (R->MaxPressure.getOccupancy(ST) >= NewOcc) | 
|  | break; | 
|  |  | 
|  | LLVM_DEBUG(printRegion(dbgs(), R->Begin, R->End, LIS, 3); | 
|  | printLivenessInfo(dbgs(), R->Begin, R->End, LIS)); | 
|  |  | 
|  | BuildDAG DAG(*R, *this); | 
|  | const auto MinSchedule = makeMinRegSchedule(DAG.getTopRoots(), *this); | 
|  | const auto MaxRP = getSchedulePressure(*R, MinSchedule); | 
|  | LLVM_DEBUG(dbgs() << "Occupancy improvement attempt:\n"; | 
|  | printSchedRP(dbgs(), R->MaxPressure, MaxRP)); | 
|  |  | 
|  | NewOcc = std::min(NewOcc, MaxRP.getOccupancy(ST)); | 
|  | if (NewOcc <= Occ) | 
|  | break; | 
|  |  | 
|  | setBestSchedule(*R, MinSchedule, MaxRP); | 
|  | } | 
|  | LLVM_DEBUG(dbgs() << "New occupancy = " << NewOcc | 
|  | << ", prev occupancy = " << Occ << '\n'); | 
|  | if (NewOcc > Occ) { | 
|  | SIMachineFunctionInfo *MFI = MF.getInfo<SIMachineFunctionInfo>(); | 
|  | MFI->increaseOccupancy(MF, NewOcc); | 
|  | } | 
|  |  | 
|  | return std::max(NewOcc, Occ); | 
|  | } | 
|  |  | 
|  | void GCNIterativeScheduler::scheduleLegacyMaxOccupancy( | 
|  | bool TryMaximizeOccupancy) { | 
|  | const auto &ST = MF.getSubtarget<GCNSubtarget>(); | 
|  | SIMachineFunctionInfo *MFI = MF.getInfo<SIMachineFunctionInfo>(); | 
|  | auto TgtOcc = MFI->getMinAllowedOccupancy(); | 
|  |  | 
|  | sortRegionsByPressure(TgtOcc); | 
|  | auto Occ = Regions.front()->MaxPressure.getOccupancy(ST); | 
|  |  | 
|  | if (TryMaximizeOccupancy && Occ < TgtOcc) | 
|  | Occ = tryMaximizeOccupancy(TgtOcc); | 
|  |  | 
|  | // This is really weird but for some magic scheduling regions twice | 
|  | // gives performance improvement | 
|  | const int NumPasses = Occ < TgtOcc ? 2 : 1; | 
|  |  | 
|  | TgtOcc = std::min(Occ, TgtOcc); | 
|  | LLVM_DEBUG(dbgs() << "Scheduling using default scheduler, " | 
|  | "target occupancy = " | 
|  | << TgtOcc << '\n'); | 
|  | GCNMaxOccupancySchedStrategy LStrgy(Context); | 
|  | unsigned FinalOccupancy = std::min(Occ, MFI->getOccupancy()); | 
|  |  | 
|  | for (int I = 0; I < NumPasses; ++I) { | 
|  | // running first pass with TargetOccupancy = 0 mimics previous scheduling | 
|  | // approach and is a performance magic | 
|  | LStrgy.setTargetOccupancy(I == 0 ? 0 : TgtOcc); | 
|  | for (auto R : Regions) { | 
|  | OverrideLegacyStrategy Ovr(*R, LStrgy, *this); | 
|  |  | 
|  | Ovr.schedule(); | 
|  | const auto RP = getRegionPressure(*R); | 
|  | LLVM_DEBUG(printSchedRP(dbgs(), R->MaxPressure, RP)); | 
|  |  | 
|  | if (RP.getOccupancy(ST) < TgtOcc) { | 
|  | LLVM_DEBUG(dbgs() << "Didn't fit into target occupancy O" << TgtOcc); | 
|  | if (R->BestSchedule.get() && | 
|  | R->BestSchedule->MaxPressure.getOccupancy(ST) >= TgtOcc) { | 
|  | LLVM_DEBUG(dbgs() << ", scheduling minimal register\n"); | 
|  | scheduleBest(*R); | 
|  | } else { | 
|  | LLVM_DEBUG(dbgs() << ", restoring\n"); | 
|  | Ovr.restoreOrder(); | 
|  | assert(R->MaxPressure.getOccupancy(ST) >= TgtOcc); | 
|  | } | 
|  | } | 
|  | FinalOccupancy = std::min(FinalOccupancy, RP.getOccupancy(ST)); | 
|  | } | 
|  | } | 
|  | MFI->limitOccupancy(FinalOccupancy); | 
|  | } | 
|  |  | 
|  | /////////////////////////////////////////////////////////////////////////////// | 
|  | // Minimal Register Strategy | 
|  |  | 
|  | void GCNIterativeScheduler::scheduleMinReg(bool force) { | 
|  | const auto &ST = MF.getSubtarget<GCNSubtarget>(); | 
|  | const SIMachineFunctionInfo *MFI = MF.getInfo<SIMachineFunctionInfo>(); | 
|  | const auto TgtOcc = MFI->getOccupancy(); | 
|  | sortRegionsByPressure(TgtOcc); | 
|  |  | 
|  | auto MaxPressure = Regions.front()->MaxPressure; | 
|  | for (auto R : Regions) { | 
|  | if (!force && R->MaxPressure.less(ST, MaxPressure, TgtOcc)) | 
|  | break; | 
|  |  | 
|  | BuildDAG DAG(*R, *this); | 
|  | const auto MinSchedule = makeMinRegSchedule(DAG.getTopRoots(), *this); | 
|  |  | 
|  | const auto RP = getSchedulePressure(*R, MinSchedule); | 
|  | LLVM_DEBUG(if (R->MaxPressure.less(ST, RP, TgtOcc)) { | 
|  | dbgs() << "\nWarning: Pressure becomes worse after minreg!"; | 
|  | printSchedRP(dbgs(), R->MaxPressure, RP); | 
|  | }); | 
|  |  | 
|  | if (!force && MaxPressure.less(ST, RP, TgtOcc)) | 
|  | break; | 
|  |  | 
|  | scheduleRegion(*R, MinSchedule, RP); | 
|  | LLVM_DEBUG(printSchedResult(dbgs(), R, RP)); | 
|  |  | 
|  | MaxPressure = RP; | 
|  | } | 
|  | } | 
|  |  | 
|  | /////////////////////////////////////////////////////////////////////////////// | 
|  | // ILP scheduler port | 
|  |  | 
|  | void GCNIterativeScheduler::scheduleILP( | 
|  | bool TryMaximizeOccupancy) { | 
|  | const auto &ST = MF.getSubtarget<GCNSubtarget>(); | 
|  | SIMachineFunctionInfo *MFI = MF.getInfo<SIMachineFunctionInfo>(); | 
|  | auto TgtOcc = MFI->getMinAllowedOccupancy(); | 
|  |  | 
|  | sortRegionsByPressure(TgtOcc); | 
|  | auto Occ = Regions.front()->MaxPressure.getOccupancy(ST); | 
|  |  | 
|  | if (TryMaximizeOccupancy && Occ < TgtOcc) | 
|  | Occ = tryMaximizeOccupancy(TgtOcc); | 
|  |  | 
|  | TgtOcc = std::min(Occ, TgtOcc); | 
|  | LLVM_DEBUG(dbgs() << "Scheduling using default scheduler, " | 
|  | "target occupancy = " | 
|  | << TgtOcc << '\n'); | 
|  |  | 
|  | unsigned FinalOccupancy = std::min(Occ, MFI->getOccupancy()); | 
|  | for (auto R : Regions) { | 
|  | BuildDAG DAG(*R, *this); | 
|  | const auto ILPSchedule = makeGCNILPScheduler(DAG.getBottomRoots(), *this); | 
|  |  | 
|  | const auto RP = getSchedulePressure(*R, ILPSchedule); | 
|  | LLVM_DEBUG(printSchedRP(dbgs(), R->MaxPressure, RP)); | 
|  |  | 
|  | if (RP.getOccupancy(ST) < TgtOcc) { | 
|  | LLVM_DEBUG(dbgs() << "Didn't fit into target occupancy O" << TgtOcc); | 
|  | if (R->BestSchedule.get() && | 
|  | R->BestSchedule->MaxPressure.getOccupancy(ST) >= TgtOcc) { | 
|  | LLVM_DEBUG(dbgs() << ", scheduling minimal register\n"); | 
|  | scheduleBest(*R); | 
|  | } | 
|  | } else { | 
|  | scheduleRegion(*R, ILPSchedule, RP); | 
|  | LLVM_DEBUG(printSchedResult(dbgs(), R, RP)); | 
|  | FinalOccupancy = std::min(FinalOccupancy, RP.getOccupancy(ST)); | 
|  | } | 
|  | } | 
|  | MFI->limitOccupancy(FinalOccupancy); | 
|  | } |