LLVM 22.0.0git
llvm::GCNIterativeScheduler Class Reference

#include "Target/AMDGPU/GCNIterativeScheduler.h"

Inheritance diagram for llvm::GCNIterativeScheduler:
[legend]

Classes

class  BuildDAG
class  OverrideLegacyStrategy
struct  Region
struct  TentativeSchedule

Public Types

enum  StrategyKind { SCHEDULE_MINREGONLY , SCHEDULE_MINREGFORCED , SCHEDULE_LEGACYMAXOCCUPANCY , SCHEDULE_ILP }
Public Types inherited from llvm::ScheduleDAGInstrs
enum  DumpDirection { TopDown , BottomUp , Bidirectional , NotSet }
 The direction that should be used to dump the scheduled Sequence. More...
using SUList = std::list<SUnit *>
 A list of SUnits, used in Value2SUsMap, during DAG construction.

Public Member Functions

 GCNIterativeScheduler (MachineSchedContext *C, StrategyKind S)
void schedule () override
 Orders nodes according to selected style.
void enterRegion (MachineBasicBlock *BB, MachineBasicBlock::iterator Begin, MachineBasicBlock::iterator End, unsigned RegionInstrs) override
 Initialize the DAG and common scheduler state for a new scheduling region.
void finalizeSchedule () override
 Allow targets to perform final scheduling actions at the level of the whole MachineFunction.
Public Member Functions inherited from llvm::ScheduleDAGMILive
 ScheduleDAGMILive (MachineSchedContext *C, std::unique_ptr< MachineSchedStrategy > S)
 ~ScheduleDAGMILive () override
bool hasVRegLiveness () const override
 Return true if this DAG supports VReg liveness and RegPressure.
bool isTrackingPressure () const
 Return true if register pressure tracking is enabled.
const IntervalPressuregetTopPressure () const
 Get current register pressure for the top scheduled instructions.
const RegPressureTrackergetTopRPTracker () const
const IntervalPressuregetBotPressure () const
 Get current register pressure for the bottom scheduled instructions.
const RegPressureTrackergetBotRPTracker () const
const IntervalPressuregetRegPressure () const
 Get register pressure for the entire scheduling region before scheduling.
const std::vector< PressureChange > & getRegionCriticalPSets () const
PressureDiffgetPressureDiff (const SUnit *SU)
const PressureDiffgetPressureDiff (const SUnit *SU) const
void computeDFSResult ()
 Compute a DFSResult after DAG building is complete, and before any queue comparisons.
const SchedDFSResultgetDFSResult () const
 Return a non-null DFS result if the scheduling strategy initialized it.
BitVectorgetScheduledTrees ()
void enterRegion (MachineBasicBlock *bb, MachineBasicBlock::iterator begin, MachineBasicBlock::iterator end, unsigned regioninstrs) override
 Implement the ScheduleDAGInstrs interface for handling the next scheduling region.
void schedule () override
 Implement ScheduleDAGInstrs interface for scheduling a sequence of reorderable instructions.
unsigned computeCyclicCriticalPath ()
 Compute the cyclic critical path through the DAG.
void dump () const override
Public Member Functions inherited from llvm::ScheduleDAGMI
 ScheduleDAGMI (MachineSchedContext *C, std::unique_ptr< MachineSchedStrategy > S, bool RemoveKillFlags)
 ~ScheduleDAGMI () override
bool doMBBSchedRegionsTopDown () const override
 If this method returns true, handling of the scheduling regions themselves (in case of a scheduling boundary in MBB) will be done beginning with the topmost region of MBB.
LiveIntervalsgetLIS () const
void addMutation (std::unique_ptr< ScheduleDAGMutation > Mutation)
 Add a postprocessing step to the DAG builder.
MachineBasicBlock::iterator top () const
MachineBasicBlock::iterator bottom () const
void startBlock (MachineBasicBlock *bb) override
 Prepares to perform scheduling in the given block.
void finishBlock () override
 Cleans up after scheduling in the given block.
void moveInstruction (MachineInstr *MI, MachineBasicBlock::iterator InsertPos)
 Change the position of an instruction within the basic block and update live ranges and region boundary iterators.
void viewGraph (const Twine &Name, const Twine &Title) override
 viewGraph - Pop up a ghostview window with the reachable parts of the DAG rendered using 'dot'.
void viewGraph () override
 Out-of-line implementation with no arguments is handy for gdb.
Public Member Functions inherited from llvm::ScheduleDAGInstrs
void setDumpDirection (DumpDirection D)
 ScheduleDAGInstrs (MachineFunction &mf, const MachineLoopInfo *mli, bool RemoveKillFlags=false)
 ~ScheduleDAGInstrs () override=default
const TargetSchedModelgetSchedModel () const
 Gets the machine model for instruction scheduling.
const MCSchedClassDescgetSchedClass (SUnit *SU) const
 Resolves and cache a resolved scheduling class for an SUnit.
bool IsReachable (SUnit *SU, SUnit *TargetSU)
 IsReachable - Checks if SU is reachable from TargetSU.
bool shouldScheduleSingleMIRegions () const
 Whether regions with a single MI should be scheduled.
MachineBasicBlock::iterator begin () const
 Returns an iterator to the top of the current scheduling region.
MachineBasicBlock::iterator end () const
 Returns an iterator to the bottom of the current scheduling region.
SUnitnewSUnit (MachineInstr *MI)
 Creates a new SUnit and return a ptr to it.
SUnitgetSUnit (MachineInstr *MI) const
 Returns an existing SUnit for this MI, or nullptr.
virtual void exitRegion ()
 Called when the scheduler has finished scheduling the current region.
void buildSchedGraph (AAResults *AA, RegPressureTracker *RPTracker=nullptr, PressureDiffs *PDiffs=nullptr, LiveIntervals *LIS=nullptr, bool TrackLaneMasks=false)
 Builds SUnits for the current region.
void addSchedBarrierDeps ()
 Adds dependencies from instructions in the current list of instructions being scheduled to scheduling barrier.
void dumpNode (const SUnit &SU) const override
std::string getGraphNodeLabel (const SUnit *SU) const override
 Returns a label for a DAG node that points to an instruction.
std::string getDAGName () const override
 Returns a label for the region of code covered by the DAG.
void fixupKills (MachineBasicBlock &MBB)
 Fixes register kill flags that scheduling has made invalid.
bool canAddEdge (SUnit *SuccSU, SUnit *PredSU)
 True if an edge can be added from PredSU to SuccSU without creating a cycle.
bool addEdge (SUnit *SuccSU, const SDep &PredDep)
 Add a DAG edge to the given SU with the given predecessor dependence data.
SmallVector< ClusterInfo > & getClusters ()
 Returns the array of the clusters.
ClusterInfogetCluster (unsigned Idx)
 Get the specific cluster, return nullptr for InvalidClusterId.
Public Member Functions inherited from llvm::ScheduleDAG
 ScheduleDAG (const ScheduleDAG &)=delete
ScheduleDAGoperator= (const ScheduleDAG &)=delete
 ScheduleDAG (MachineFunction &mf)
virtual ~ScheduleDAG ()
void clearDAG ()
 Clears the DAG state (between regions).
const MCInstrDescgetInstrDesc (const SUnit *SU) const
 Returns the MCInstrDesc of this SUnit.
void dumpNodeName (const SUnit &SU) const
virtual void addCustomGraphFeatures (GraphWriter< ScheduleDAG * > &) const
 Adds custom features for a visualization of the ScheduleDAG.
unsigned VerifyScheduledDAG (bool isBottomUp)
 Verifies that all SUnits were scheduled and that their state is consistent.

Protected Types

using ScheduleRef = ArrayRef<const SUnit *>
Protected Types inherited from llvm::ScheduleDAGInstrs
using DbgValueVector

Protected Member Functions

template<typename Range>
GCNRegPressure getSchedulePressure (const Region &R, Range &&Schedule) const
GCNRegPressure getRegionPressure (MachineBasicBlock::iterator Begin, MachineBasicBlock::iterator End) const
GCNRegPressure getRegionPressure (const Region &R) const
void swapIGLPMutations (const Region &R, bool IsReentry)
void setBestSchedule (Region &R, ScheduleRef Schedule, const GCNRegPressure &MaxRP=GCNRegPressure())
void scheduleBest (Region &R)
std::vector< MachineInstr * > detachSchedule (ScheduleRef Schedule) const
void sortRegionsByPressure (unsigned TargetOcc)
template<typename Range>
void scheduleRegion (Region &R, Range &&Schedule, const GCNRegPressure &MaxRP=GCNRegPressure())
unsigned tryMaximizeOccupancy (unsigned TargetOcc=std::numeric_limits< unsigned >::max())
void scheduleLegacyMaxOccupancy (bool TryMaximizeOccupancy=true)
void scheduleMinReg (bool force=false)
void scheduleILP (bool TryMaximizeOccupancy=true)
void printRegions (raw_ostream &OS) const
void printSchedResult (raw_ostream &OS, const Region *R, const GCNRegPressure &RP) const
void printSchedRP (raw_ostream &OS, const GCNRegPressure &Before, const GCNRegPressure &After) const
Protected Member Functions inherited from llvm::ScheduleDAGMILive
void buildDAGWithRegPressure ()
 Call ScheduleDAGInstrs::buildSchedGraph with register pressure tracking enabled.
void initQueues (ArrayRef< SUnit * > TopRoots, ArrayRef< SUnit * > BotRoots)
 Release ExitSU predecessors and setup scheduler queues.
void scheduleMI (SUnit *SU, bool IsTopNode)
 Move an instruction and update register pressure.
void initRegPressure ()
void updatePressureDiffs (ArrayRef< VRegMaskOrUnit > LiveUses)
 Update the PressureDiff array for liveness after scheduling this instruction.
void updateScheduledPressure (const SUnit *SU, const std::vector< unsigned > &NewMaxPressure)
void collectVRegUses (SUnit &SU)
Protected Member Functions inherited from llvm::ScheduleDAGMI
void postProcessDAG ()
 Apply each ScheduleDAGMutation step in order.
void initQueues (ArrayRef< SUnit * > TopRoots, ArrayRef< SUnit * > BotRoots)
 Release ExitSU predecessors and setup scheduler queues.
void updateQueues (SUnit *SU, bool IsTopNode)
 Update scheduler DAG and queues after scheduling an instruction.
void placeDebugValues ()
 Reinsert debug_values recorded in ScheduleDAGInstrs::DbgValues.
void dumpSchedule () const
 dump the scheduled Sequence.
void dumpScheduleTraceTopDown () const
 Print execution trace of the schedule top-down or bottom-up.
void dumpScheduleTraceBottomUp () const
bool checkSchedLimit ()
void findRootsAndBiasEdges (SmallVectorImpl< SUnit * > &TopRoots, SmallVectorImpl< SUnit * > &BotRoots)
void releaseSucc (SUnit *SU, SDep *SuccEdge)
 ReleaseSucc - Decrement the NumPredsLeft count of a successor.
void releaseSuccessors (SUnit *SU)
 releaseSuccessors - Call releaseSucc on each of SU's successors.
void releasePred (SUnit *SU, SDep *PredEdge)
 ReleasePred - Decrement the NumSuccsLeft count of a predecessor.
void releasePredecessors (SUnit *SU)
 releasePredecessors - Call releasePred on each of SU's predecessors.
Protected Member Functions inherited from llvm::ScheduleDAGInstrs
BatchAAResultsgetAAForDep () const
 Returns a (possibly null) pointer to the current BatchAAResults.
void reduceHugeMemNodeMaps (Value2SUsMap &stores, Value2SUsMap &loads, unsigned N)
 Reduces maps in FIFO order, by N SUs.
void addChainDependency (SUnit *SUa, SUnit *SUb, unsigned Latency=0)
 Adds a chain edge between SUa and SUb, but only if both AAResults and Target fail to deny the dependency.
void addChainDependencies (SUnit *SU, SUList &SUs, unsigned Latency)
 Adds dependencies as needed from all SUs in list to SU.
void addChainDependencies (SUnit *SU, Value2SUsMap &Val2SUsMap)
 Adds dependencies as needed from all SUs in map, to SU.
void addChainDependencies (SUnit *SU, Value2SUsMap &Val2SUsMap, ValueType V)
 Adds dependencies as needed to SU, from all SUs mapped to V.
void addBarrierChain (Value2SUsMap &map)
 Adds barrier chain edges from all SUs in map, and then clear the map.
void insertBarrierChain (Value2SUsMap &map)
 Inserts a barrier chain in a huge region, far below current SU.
void initSUnits ()
 Creates an SUnit for each real instruction, numbered in top-down topological order.
void addPhysRegDataDeps (SUnit *SU, unsigned OperIdx)
 MO is an operand of SU's instruction that defines a physical register.
void addPhysRegDeps (SUnit *SU, unsigned OperIdx)
 Adds register dependencies (data, anti, and output) from this SUnit to following instructions in the same scheduling region that depend the physical register referenced at OperIdx.
void addVRegDefDeps (SUnit *SU, unsigned OperIdx)
 Adds register output and data dependencies from this SUnit to instructions that occur later in the same scheduling region if they read from or write to the virtual register defined at OperIdx.
void addVRegUseDeps (SUnit *SU, unsigned OperIdx)
 Adds a register data dependency if the instruction that defines the virtual register used at OperIdx is mapped to an SUnit.
LaneBitmask getLaneMaskForMO (const MachineOperand &MO) const
 Returns a mask for which lanes get read/written by the given (register) machine operand.
bool deadDefHasNoUse (const MachineOperand &MO)
 Returns true if the def register in MO has no uses.
Protected Member Functions inherited from llvm::ScheduleDAG
void dumpNodeAll (const SUnit &SU) const

Protected Attributes

SpecificBumpPtrAllocator< RegionAlloc
std::vector< Region * > Regions
MachineSchedContextContext
const StrategyKind Strategy
GCNUpwardRPTracker UPTracker
std::vector< std::unique_ptr< ScheduleDAGMutation > > SavedMutations
Protected Attributes inherited from llvm::ScheduleDAGMILive
RegisterClassInfoRegClassInfo
SchedDFSResultDFSResult = nullptr
 Information about DAG subtrees.
BitVector ScheduledTrees
MachineBasicBlock::iterator LiveRegionEnd
VReg2SUnitMultiMap VRegUses
 Maps vregs to the SUnits of their uses in the current scheduling region.
PressureDiffs SUPressureDiffs
bool ShouldTrackPressure = false
 Register pressure in this region computed by initRegPressure.
bool ShouldTrackLaneMasks = false
IntervalPressure RegPressure
RegPressureTracker RPTracker
std::vector< PressureChangeRegionCriticalPSets
 List of pressure sets that exceed the target's pressure limit before scheduling, listed in increasing set ID order.
IntervalPressure TopPressure
 The top of the unscheduled zone.
RegPressureTracker TopRPTracker
IntervalPressure BotPressure
 The bottom of the unscheduled zone.
RegPressureTracker BotRPTracker
Protected Attributes inherited from llvm::ScheduleDAGMI
AAResultsAA
LiveIntervalsLIS
std::unique_ptr< MachineSchedStrategySchedImpl
std::vector< std::unique_ptr< ScheduleDAGMutation > > Mutations
 Ordered list of DAG postprocessing steps.
MachineBasicBlock::iterator CurrentTop
 The top of the unscheduled zone.
MachineBasicBlock::iterator CurrentBottom
 The bottom of the unscheduled zone.
Protected Attributes inherited from llvm::ScheduleDAGInstrs
const MachineLoopInfoMLI = nullptr
const MachineFrameInfoMFI
TargetSchedModel SchedModel
 TargetSchedModel provides an interface to the machine model.
bool RemoveKillFlags
 True if the DAG builder should remove kill flags (in preparation for rescheduling).
bool ScheduleSingleMIRegions = false
 True if regions with a single MI should be scheduled.
bool CanHandleTerminators = false
 The standard DAG builder does not normally include terminators as DAG nodes because it does not create the necessary dependencies to prevent reordering.
bool TrackLaneMasks = false
 Whether lane masks should get tracked.
MachineBasicBlockBB = nullptr
 The block in which to insert instructions.
MachineBasicBlock::iterator RegionBegin
 The beginning of the range to be scheduled.
MachineBasicBlock::iterator RegionEnd
 The end of the range to be scheduled.
unsigned NumRegionInstrs = 0
 Instructions in this region (distance(RegionBegin, RegionEnd)).
DenseMap< MachineInstr *, SUnit * > MISUnitMap
 After calling BuildSchedGraph, each machine instruction in the current scheduling region is mapped to an SUnit.
RegUnit2SUnitsMap Defs
 Defs, Uses - Remember where defs and uses of each register are as we iterate upward through the instructions.
RegUnit2SUnitsMap Uses
VReg2SUnitMultiMap CurrentVRegDefs
 Tracks the last instruction(s) in this region defining each virtual register.
VReg2SUnitOperIdxMultiMap CurrentVRegUses
 Tracks the last instructions in this region using each virtual register.
std::optional< BatchAAResultsAAForDep
SUnitBarrierChain = nullptr
 Remember a generic side-effecting instruction as we proceed.
SmallVector< ClusterInfoClusters
DumpDirection DumpDir = NotSet
UndefValueUnknownValue
 For an unanalyzable memory access, this Value is used in maps.
ScheduleDAGTopologicalSort Topo
 Topo - A topological ordering for SUnits which permits fast IsReachable and similar queries.
DbgValueVector DbgValues
 Remember instruction that precedes DBG_VALUE.
MachineInstrFirstDbgValue = nullptr
LiveRegUnits LiveRegs
 Set of live physical registers for updating kill flags.

Additional Inherited Members

Public Attributes inherited from llvm::ScheduleDAG
const TargetMachineTM
 Target processor.
const TargetInstrInfoTII
 Target instruction information.
const TargetRegisterInfoTRI
 Target processor register info.
MachineFunctionMF
 Machine function.
MachineRegisterInfoMRI
 Virtual/real register map.
std::vector< SUnitSUnits
 The scheduling units.
SUnit EntrySU
 Special node for the region entry.
SUnit ExitSU
 Special node for the region exit.
bool StressSched

Detailed Description

Definition at line 29 of file GCNIterativeScheduler.h.

Member Typedef Documentation

◆ ScheduleRef

Definition at line 53 of file GCNIterativeScheduler.h.

Member Enumeration Documentation

◆ StrategyKind

Enumerator
SCHEDULE_MINREGONLY 
SCHEDULE_MINREGFORCED 
SCHEDULE_LEGACYMAXOCCUPANCY 
SCHEDULE_ILP 

Definition at line 33 of file GCNIterativeScheduler.h.

Constructor & Destructor Documentation

◆ GCNIterativeScheduler()

Member Function Documentation

◆ detachSchedule()

std::vector< MachineInstr * > GCNIterativeScheduler::detachSchedule ( ScheduleRef Schedule) const
protected

◆ enterRegion()

void GCNIterativeScheduler::enterRegion ( MachineBasicBlock * bb,
MachineBasicBlock::iterator begin,
MachineBasicBlock::iterator end,
unsigned regioninstrs )
overridevirtual

Initialize the DAG and common scheduler state for a new scheduling region.

This does not actually create the DAG, only clears it. The scheduling driver may call BuildSchedGraph multiple times per scheduling region.

Reimplemented from llvm::ScheduleDAGInstrs.

Definition at line 300 of file GCNIterativeScheduler.cpp.

References Alloc, llvm::ScheduleDAGInstrs::BB, llvm::ScheduleDAGMILive::enterRegion(), llvm::ScheduleDAGInstrs::NumRegionInstrs, and Regions.

◆ finalizeSchedule()

void GCNIterativeScheduler::finalizeSchedule ( )
overridevirtual

Allow targets to perform final scheduling actions at the level of the whole MachineFunction.

By default does nothing.

Reimplemented from llvm::ScheduleDAGInstrs.

Definition at line 324 of file GCNIterativeScheduler.cpp.

References Regions, SCHEDULE_ILP, SCHEDULE_LEGACYMAXOCCUPANCY, SCHEDULE_MINREGFORCED, SCHEDULE_MINREGONLY, scheduleILP(), scheduleLegacyMaxOccupancy(), scheduleMinReg(), and Strategy.

◆ getRegionPressure() [1/2]

GCNRegPressure llvm::GCNIterativeScheduler::getRegionPressure ( const Region & R) const
inlineprotected

Definition at line 92 of file GCNIterativeScheduler.h.

References getRegionPressure().

◆ getRegionPressure() [2/2]

GCNRegPressure GCNIterativeScheduler::getRegionPressure ( MachineBasicBlock::iterator Begin,
MachineBasicBlock::iterator End ) const
protected

◆ getSchedulePressure()

template<typename Range>
GCNRegPressure GCNIterativeScheduler::getSchedulePressure ( const Region & R,
Range && Schedule ) const
protected

◆ printRegions()

LLVM_DUMP_METHOD void GCNIterativeScheduler::printRegions ( raw_ostream & OS) const
protected

◆ printSchedResult()

LLVM_DUMP_METHOD void GCNIterativeScheduler::printSchedResult ( raw_ostream & OS,
const Region * R,
const GCNRegPressure & RP ) const
protected

Definition at line 103 of file GCNIterativeScheduler.cpp.

References llvm::ScheduleDAGMI::LIS, printRegion(), and printSchedRP().

Referenced by scheduleILP(), and scheduleMinReg().

◆ printSchedRP()

LLVM_DUMP_METHOD void GCNIterativeScheduler::printSchedRP ( raw_ostream & OS,
const GCNRegPressure & Before,
const GCNRegPressure & After ) const
protected

◆ schedule()

void GCNIterativeScheduler::schedule ( )
overridevirtual

Orders nodes according to selected style.

Typically, a scheduling algorithm will implement schedule() without overriding enterRegion() or exitRegion().

Implements llvm::ScheduleDAGInstrs.

Definition at line 313 of file GCNIterativeScheduler.cpp.

References llvm::dbgs(), llvm::ScheduleDAGMI::LIS, LLVM_DEBUG, printLivenessInfo(), llvm::ScheduleDAGInstrs::RegionBegin, llvm::ScheduleDAGInstrs::RegionEnd, and Regions.

◆ scheduleBest()

void GCNIterativeScheduler::scheduleBest ( Region & R)
protected

Definition at line 364 of file GCNIterativeScheduler.cpp.

References assert(), and scheduleRegion().

Referenced by scheduleILP(), and scheduleLegacyMaxOccupancy().

◆ scheduleILP()

◆ scheduleLegacyMaxOccupancy()

◆ scheduleMinReg()

◆ scheduleRegion()

◆ setBestSchedule()

void GCNIterativeScheduler::setBestSchedule ( Region & R,
ScheduleRef Schedule,
const GCNRegPressure & MaxRP = GCNRegPressure() )
protected

Definition at line 357 of file GCNIterativeScheduler.cpp.

References detachSchedule().

Referenced by tryMaximizeOccupancy().

◆ sortRegionsByPressure()

void GCNIterativeScheduler::sortRegionsByPressure ( unsigned TargetOcc)
protected

◆ swapIGLPMutations()

◆ tryMaximizeOccupancy()

Member Data Documentation

◆ Alloc

SpecificBumpPtrAllocator<Region> llvm::GCNIterativeScheduler::Alloc
protected

Definition at line 73 of file GCNIterativeScheduler.h.

Referenced by enterRegion().

◆ Context

MachineSchedContext* llvm::GCNIterativeScheduler::Context
protected

Definition at line 76 of file GCNIterativeScheduler.h.

Referenced by GCNIterativeScheduler(), and scheduleLegacyMaxOccupancy().

◆ Regions

std::vector<Region*> llvm::GCNIterativeScheduler::Regions
protected

◆ SavedMutations

std::vector<std::unique_ptr<ScheduleDAGMutation> > llvm::GCNIterativeScheduler::SavedMutations
protected

Definition at line 80 of file GCNIterativeScheduler.h.

Referenced by swapIGLPMutations().

◆ Strategy

const StrategyKind llvm::GCNIterativeScheduler::Strategy
protected

Definition at line 77 of file GCNIterativeScheduler.h.

Referenced by finalizeSchedule(), and GCNIterativeScheduler().

◆ UPTracker

GCNUpwardRPTracker llvm::GCNIterativeScheduler::UPTracker
mutableprotected

Definition at line 78 of file GCNIterativeScheduler.h.

Referenced by GCNIterativeScheduler(), and getRegionPressure().


The documentation for this class was generated from the following files: