LLVM 20.0.0git
Public Member Functions | Protected Member Functions | List of all members
llvm::RegPressureTracker Class Reference

Track the current register pressure at some position in the instruction stream, and remember the high water mark within the region traversed. More...

#include "llvm/CodeGen/RegisterPressure.h"

Public Member Functions

 RegPressureTracker (IntervalPressure &rp)
 
 RegPressureTracker (RegionPressure &rp)
 
void reset ()
 
void init (const MachineFunction *mf, const RegisterClassInfo *rci, const LiveIntervals *lis, const MachineBasicBlock *mbb, MachineBasicBlock::const_iterator pos, bool TrackLaneMasks, bool TrackUntiedDefs)
 Setup the RegPressureTracker.
 
void addLiveRegs (ArrayRef< RegisterMaskPair > Regs)
 Force liveness of virtual registers or physical register units.
 
MachineBasicBlock::const_iterator getPos () const
 Get the MI position corresponding to this register pressure.
 
void setPos (MachineBasicBlock::const_iterator Pos)
 
void recede (SmallVectorImpl< RegisterMaskPair > *LiveUses=nullptr)
 Recede across the previous instruction.
 
void recede (const RegisterOperands &RegOpers, SmallVectorImpl< RegisterMaskPair > *LiveUses=nullptr)
 Recede across the previous instruction.
 
void recedeSkipDebugValues ()
 Recede until we find an instruction which is not a DebugValue.
 
void advance ()
 Advance across the current instruction.
 
void advance (const RegisterOperands &RegOpers)
 Advance across the current instruction.
 
void closeRegion ()
 Finalize the region boundaries and recored live ins and live outs.
 
void initLiveThru (const RegPressureTracker &RPTracker)
 Initialize the LiveThru pressure set based on the untied defs found in RPTracker.
 
void initLiveThru (ArrayRef< unsigned > PressureSet)
 Copy an existing live thru pressure result.
 
ArrayRef< unsignedgetLiveThru () const
 
RegisterPressuregetPressure ()
 Get the resulting register pressure over the traversed region.
 
const RegisterPressuregetPressure () const
 
const std::vector< unsigned > & getRegSetPressureAtPos () const
 Get the register set pressure at the current position, which may be less than the pressure across the traversed region.
 
bool isTopClosed () const
 Does this pressure result have a valid top position and live ins.
 
bool isBottomClosed () const
 Does this pressure result have a valid bottom position and live outs.
 
void closeTop ()
 Set the boundary for the top of the region and summarize live ins.
 
void closeBottom ()
 Set the boundary for the bottom of the region and summarize live outs.
 
void getMaxUpwardPressureDelta (const MachineInstr *MI, PressureDiff *PDiff, RegPressureDelta &Delta, ArrayRef< PressureChange > CriticalPSets, ArrayRef< unsigned > MaxPressureLimit)
 Consider the pressure increase caused by traversing this instruction bottom-up.
 
void getUpwardPressureDelta (const MachineInstr *MI, PressureDiff &PDiff, RegPressureDelta &Delta, ArrayRef< PressureChange > CriticalPSets, ArrayRef< unsigned > MaxPressureLimit) const
 This is the fast version of querying register pressure that does not directly depend on current liveness.
 
void getMaxDownwardPressureDelta (const MachineInstr *MI, RegPressureDelta &Delta, ArrayRef< PressureChange > CriticalPSets, ArrayRef< unsigned > MaxPressureLimit)
 Consider the pressure increase caused by traversing this instruction top-down.
 
void getMaxPressureDelta (const MachineInstr *MI, RegPressureDelta &Delta, ArrayRef< PressureChange > CriticalPSets, ArrayRef< unsigned > MaxPressureLimit)
 Find the pressure set with the most change beyond its pressure limit after traversing this instruction either upward or downward depending on the closed end of the current region.
 
void getUpwardPressure (const MachineInstr *MI, std::vector< unsigned > &PressureResult, std::vector< unsigned > &MaxPressureResult)
 Get the pressure of each PSet after traversing this instruction bottom-up.
 
void getDownwardPressure (const MachineInstr *MI, std::vector< unsigned > &PressureResult, std::vector< unsigned > &MaxPressureResult)
 Get the pressure of each PSet after traversing this instruction top-down.
 
void getPressureAfterInst (const MachineInstr *MI, std::vector< unsigned > &PressureResult, std::vector< unsigned > &MaxPressureResult)
 
bool hasUntiedDef (Register VirtReg) const
 
void dump () const
 
void increaseRegPressure (Register RegUnit, LaneBitmask PreviousMask, LaneBitmask NewMask)
 
void decreaseRegPressure (Register RegUnit, LaneBitmask PreviousMask, LaneBitmask NewMask)
 

Protected Member Functions

void discoverLiveOut (RegisterMaskPair Pair)
 Add Reg to the live out set and increase max pressure.
 
void discoverLiveIn (RegisterMaskPair Pair)
 Add Reg to the live in set and increase max pressure.
 
SlotIndex getCurrSlot () const
 Get the SlotIndex for the first nondebug instruction including or after the current position.
 
void bumpDeadDefs (ArrayRef< RegisterMaskPair > DeadDefs)
 
void bumpUpwardPressure (const MachineInstr *MI)
 Record the upward impact of a single instruction on current register pressure.
 
void bumpDownwardPressure (const MachineInstr *MI)
 Record the downward impact of a single instruction on current register pressure.
 
void discoverLiveInOrOut (RegisterMaskPair Pair, SmallVectorImpl< RegisterMaskPair > &LiveInOrOut)
 
LaneBitmask getLastUsedLanes (Register RegUnit, SlotIndex Pos) const
 
LaneBitmask getLiveLanesAt (Register RegUnit, SlotIndex Pos) const
 
LaneBitmask getLiveThroughAt (Register RegUnit, SlotIndex Pos) const
 

Detailed Description

Track the current register pressure at some position in the instruction stream, and remember the high water mark within the region traversed.

This does not automatically consider live-through ranges. The client may independently adjust for global liveness.

Each RegPressureTracker only works within a MachineBasicBlock. Pressure can be tracked across a larger region by storing a RegisterPressure result at each block boundary and explicitly adjusting pressure to account for block live-in and live-out register sets.

RegPressureTracker holds a reference to a RegisterPressure result that it computes incrementally. During downward tracking, P.BottomIdx or P.BottomPos is invalid until it reaches the end of the block or closeRegion() is explicitly called. Similarly, P.TopIdx is invalid during upward tracking. Changing direction has the side effect of closing region, and traversing past TopIdx or BottomIdx reopens it.

Definition at line 359 of file RegisterPressure.h.

Constructor & Destructor Documentation

◆ RegPressureTracker() [1/2]

llvm::RegPressureTracker::RegPressureTracker ( IntervalPressure rp)
inline

Definition at line 399 of file RegisterPressure.h.

References P.

◆ RegPressureTracker() [2/2]

llvm::RegPressureTracker::RegPressureTracker ( RegionPressure rp)
inline

Definition at line 400 of file RegisterPressure.h.

References P.

Member Function Documentation

◆ addLiveRegs()

void RegPressureTracker::addLiveRegs ( ArrayRef< RegisterMaskPair Regs)

Force liveness of virtual registers or physical register units.

Force liveness of registers.

Particularly useful to initialize the livein/out state of the tracker before the first call to advance/recede.

Definition at line 697 of file RegisterPressure.cpp.

References increaseRegPressure(), llvm::LiveRegSet::insert(), and P.

Referenced by computeLiveOuts(), and llvm::ScheduleDAGMILive::initRegPressure().

◆ advance() [1/2]

void RegPressureTracker::advance ( )

◆ advance() [2/2]

void RegPressureTracker::advance ( const RegisterOperands RegOpers)

◆ bumpDeadDefs()

void RegPressureTracker::bumpDeadDefs ( ArrayRef< RegisterMaskPair DeadDefs)
protected

◆ bumpDownwardPressure()

void RegPressureTracker::bumpDownwardPressure ( const MachineInstr MI)
protected

Record the downward impact of a single instruction on current register pressure.

Unlike the advance/recede pressure tracking interface, this does not discover live in/outs.

This is intended for speculative queries. It leaves pressure inconsistent with the current position, so must be restored by the caller.

Definition at line 1277 of file RegisterPressure.cpp.

References llvm::RegisterOperands::adjustLaneLiveness(), assert(), bumpDeadDefs(), llvm::RegisterOperands::collect(), llvm::LiveRegSet::contains(), llvm::RegisterOperands::DeadDefs, decreaseRegPressure(), llvm::RegisterOperands::Defs, findUseBetween(), getCurrSlot(), llvm::LiveIntervals::getInstructionIndex(), getLastUsedLanes(), llvm::SlotIndex::getRegSlot(), increaseRegPressure(), MI, MRI, llvm::LaneBitmask::none(), Reg, TRI, and llvm::RegisterOperands::Uses.

Referenced by getDownwardPressure(), and getMaxDownwardPressureDelta().

◆ bumpUpwardPressure()

void RegPressureTracker::bumpUpwardPressure ( const MachineInstr MI)
protected

Record the upward impact of a single instruction on current register pressure.

Unlike the advance/recede pressure tracking interface, this does not discover live in/outs.

This is intended for speculative queries. It leaves pressure inconsistent with the current position, so must be restored by the caller.

Definition at line 1033 of file RegisterPressure.cpp.

References llvm::RegisterOperands::adjustLaneLiveness(), assert(), bumpDeadDefs(), llvm::RegisterOperands::collect(), llvm::LiveRegSet::contains(), llvm::RegisterOperands::DeadDefs, decreaseRegPressure(), llvm::RegisterOperands::Defs, llvm::RegisterOperands::detectDeadDefs(), llvm::LiveIntervals::getInstructionIndex(), getRegLanes(), llvm::SlotIndex::getRegSlot(), increaseRegPressure(), MI, MRI, P, Reg, TRI, and llvm::RegisterOperands::Uses.

Referenced by getMaxUpwardPressureDelta(), and getUpwardPressure().

◆ closeBottom()

void RegPressureTracker::closeBottom ( )

Set the boundary for the bottom of the region and summarize live outs.

Definition at line 330 of file RegisterPressure.cpp.

References llvm::LiveRegSet::appendTo(), assert(), getCurrSlot(), llvm::RegisterPressure::LiveOutRegs, and llvm::LiveRegSet::size().

Referenced by closeRegion(), llvm::ScheduleDAGMILive::initRegPressure(), and recedeSkipDebugValues().

◆ closeRegion()

void RegPressureTracker::closeRegion ( )

Finalize the region boundaries and recored live ins and live outs.

Finalize the region boundaries and record live ins and live outs.

Definition at line 342 of file RegisterPressure.cpp.

References assert(), closeBottom(), closeTop(), isBottomClosed(), isTopClosed(), and llvm::LiveRegSet::size().

Referenced by llvm::ScheduleDAGMILive::initRegPressure(), and llvm::PPCInstrInfo::shouldReduceRegisterPressure().

◆ closeTop()

void RegPressureTracker::closeTop ( )

Set the boundary for the top of the region and summarize live ins.

Definition at line 318 of file RegisterPressure.cpp.

References llvm::LiveRegSet::appendTo(), assert(), getCurrSlot(), llvm::RegisterPressure::LiveInRegs, and llvm::LiveRegSet::size().

Referenced by advance(), closeRegion(), and llvm::ScheduleDAGMILive::initRegPressure().

◆ decreaseRegPressure()

void RegPressureTracker::decreaseRegPressure ( Register  RegUnit,
LaneBitmask  PreviousMask,
LaneBitmask  NewMask 
)

◆ discoverLiveIn()

void RegPressureTracker::discoverLiveIn ( RegisterMaskPair  Pair)
protected

Add Reg to the live in set and increase max pressure.

Definition at line 727 of file RegisterPressure.cpp.

References discoverLiveInOrOut(), and P.

Referenced by advance().

◆ discoverLiveInOrOut()

void RegPressureTracker::discoverLiveInOrOut ( RegisterMaskPair  Pair,
SmallVectorImpl< RegisterMaskPair > &  LiveInOrOut 
)
protected

◆ discoverLiveOut()

void RegPressureTracker::discoverLiveOut ( RegisterMaskPair  Pair)
protected

Add Reg to the live out set and increase max pressure.

Definition at line 731 of file RegisterPressure.cpp.

References discoverLiveInOrOut(), and P.

Referenced by recede().

◆ dump()

LLVM_DUMP_METHOD void RegPressureTracker::dump ( ) const

◆ getCurrSlot()

SlotIndex RegPressureTracker::getCurrSlot ( ) const
protected

◆ getDownwardPressure()

void RegPressureTracker::getDownwardPressure ( const MachineInstr MI,
std::vector< unsigned > &  PressureResult,
std::vector< unsigned > &  MaxPressureResult 
)

Get the pressure of each PSet after traversing this instruction top-down.

Definition at line 1375 of file RegisterPressure.cpp.

References bumpDownwardPressure(), MI, and P.

Referenced by getPressureAfterInst(), and getRegisterPressures().

◆ getLastUsedLanes()

LaneBitmask RegPressureTracker::getLastUsedLanes ( Register  RegUnit,
SlotIndex  Pos 
) const
protected

◆ getLiveLanesAt()

LaneBitmask RegPressureTracker::getLiveLanesAt ( Register  RegUnit,
SlotIndex  Pos 
) const
protected

◆ getLiveThroughAt()

LaneBitmask RegPressureTracker::getLiveThroughAt ( Register  RegUnit,
SlotIndex  Pos 
) const
protected

◆ getLiveThru()

ArrayRef< unsigned > llvm::RegPressureTracker::getLiveThru ( ) const
inline

◆ getMaxDownwardPressureDelta()

void RegPressureTracker::getMaxDownwardPressureDelta ( const MachineInstr MI,
RegPressureDelta Delta,
ArrayRef< PressureChange CriticalPSets,
ArrayRef< unsigned MaxPressureLimit 
)

Consider the pressure increase caused by traversing this instruction top-down.

Find the pressure set with the most change beyond its pressure limit based on the tracker's current pressure, and record the number of excess register units of that pressure set introduced by this instruction.

Find the register class with the most change in its pressure limit based on the tracker's current pressure, and return the number of excess register units of that pressure set introduced by this instruction.

This assumes that the current LiveIn set is sufficient.

This is expensive for an on-the-fly query because it calls bumpDownwardPressure to recompute the pressure sets based on current liveness. We don't yet have a fast version of downward pressure tracking analogous to getUpwardPressureDelta.

Definition at line 1336 of file RegisterPressure.cpp.

References assert(), bumpDownwardPressure(), computeExcessPressureDelta(), computeMaxPressureDelta(), llvm::RegPressureDelta::CriticalMax, llvm::RegPressureDelta::CurrentMax, llvm::PressureChange::getUnitInc(), MI, and P.

Referenced by getMaxPressureDelta(), and llvm::GenericScheduler::initCandidate().

◆ getMaxPressureDelta()

void llvm::RegPressureTracker::getMaxPressureDelta ( const MachineInstr MI,
RegPressureDelta Delta,
ArrayRef< PressureChange CriticalPSets,
ArrayRef< unsigned MaxPressureLimit 
)
inline

Find the pressure set with the most change beyond its pressure limit after traversing this instruction either upward or downward depending on the closed end of the current region.

Definition at line 503 of file RegisterPressure.h.

References assert(), getMaxDownwardPressureDelta(), getMaxUpwardPressureDelta(), isBottomClosed(), isTopClosed(), and MI.

Referenced by llvm::ConvergingVLIWScheduler::readyQueueVerboseDump().

◆ getMaxUpwardPressureDelta()

void RegPressureTracker::getMaxUpwardPressureDelta ( const MachineInstr MI,
PressureDiff PDiff,
RegPressureDelta Delta,
ArrayRef< PressureChange CriticalPSets,
ArrayRef< unsigned MaxPressureLimit 
)

Consider the pressure increase caused by traversing this instruction bottom-up.

Find the pressure set with the most change beyond its pressure limit based on the tracker's current pressure, and record the number of excess register units of that pressure set introduced by this instruction.

Find the pressure set with the most change beyond its pressure limit based on the tracker's current pressure, and return the change in number of register units of that pressure set introduced by this instruction.

This assumes that the current LiveOut set is sufficient.

This is expensive for an on-the-fly query because it calls bumpUpwardPressure to recompute the pressure sets based on current liveness. This mainly exists to verify correctness, e.g. with -verify-misched. getUpwardPressureDelta is the fast version of this query that uses the per-SUnit cache of the PressureDiff.

Definition at line 1087 of file RegisterPressure.cpp.

References assert(), bumpUpwardPressure(), computeExcessPressureDelta(), computeMaxPressureDelta(), llvm::RegPressureDelta::CriticalMax, llvm::RegPressureDelta::CurrentMax, llvm::dbgs(), llvm::PressureDiff::dump(), llvm::RegPressureDelta::Excess, llvm::PressureChange::getPSet(), llvm::PressureChange::getUnitInc(), getUpwardPressureDelta(), llvm::PressureChange::isValid(), llvm_unreachable, MI, P, and TRI.

Referenced by getMaxPressureDelta(), and llvm::GenericScheduler::initCandidate().

◆ getPos()

MachineBasicBlock::const_iterator llvm::RegPressureTracker::getPos ( ) const
inline

◆ getPressure() [1/2]

RegisterPressure & llvm::RegPressureTracker::getPressure ( )
inline

◆ getPressure() [2/2]

const RegisterPressure & llvm::RegPressureTracker::getPressure ( ) const
inline

Definition at line 461 of file RegisterPressure.h.

References P.

◆ getPressureAfterInst()

void llvm::RegPressureTracker::getPressureAfterInst ( const MachineInstr MI,
std::vector< unsigned > &  PressureResult,
std::vector< unsigned > &  MaxPressureResult 
)
inline

◆ getRegSetPressureAtPos()

const std::vector< unsigned > & llvm::RegPressureTracker::getRegSetPressureAtPos ( ) const
inline

Get the register set pressure at the current position, which may be less than the pressure across the traversed region.

Definition at line 465 of file RegisterPressure.h.

Referenced by llvm::ScheduleDAGMILive::initRegPressure(), llvm::GCNSchedStrategy::pickNodeFromQueue(), and llvm::ScheduleDAGMILive::scheduleMI().

◆ getUpwardPressure()

void RegPressureTracker::getUpwardPressure ( const MachineInstr MI,
std::vector< unsigned > &  PressureResult,
std::vector< unsigned > &  MaxPressureResult 
)

Get the pressure of each PSet after traversing this instruction bottom-up.

Definition at line 1359 of file RegisterPressure.cpp.

References bumpUpwardPressure(), MI, and P.

Referenced by getPressureAfterInst(), and getRegisterPressures().

◆ getUpwardPressureDelta()

void RegPressureTracker::getUpwardPressureDelta ( const MachineInstr MI,
PressureDiff PDiff,
RegPressureDelta Delta,
ArrayRef< PressureChange CriticalPSets,
ArrayRef< unsigned MaxPressureLimit 
) const

This is the fast version of querying register pressure that does not directly depend on current liveness.

Parameters
Deltacaptures information needed for heuristics.
CriticalPSetsAre the pressure sets that are known to exceed some limit within the region, not necessarily at the current position.
MaxPressureLimitIs the max pressure within the region, not necessarily at the current position.

Definition at line 1155 of file RegisterPressure.cpp.

References assert(), llvm::PressureDiff::begin(), llvm::RegPressureDelta::CriticalMax, llvm::RegPressureDelta::CurrentMax, llvm::PressureDiff::end(), llvm::RegPressureDelta::Excess, llvm::RegisterClassInfo::getRegPressureSetLimit(), llvm::PressureChange::isValid(), P, llvm::PressureChange::setUnitInc(), and llvm::ArrayRef< T >::size().

Referenced by getMaxUpwardPressureDelta(), and llvm::GenericScheduler::initCandidate().

◆ hasUntiedDef()

bool llvm::RegPressureTracker::hasUntiedDef ( Register  VirtReg) const
inline

◆ increaseRegPressure()

void RegPressureTracker::increaseRegPressure ( Register  RegUnit,
LaneBitmask  PreviousMask,
LaneBitmask  NewMask 
)

◆ init()

void RegPressureTracker::init ( const MachineFunction mf,
const RegisterClassInfo rci,
const LiveIntervals lis,
const MachineBasicBlock mbb,
MachineBasicBlock::const_iterator  pos,
bool  TrackLaneMasks,
bool  TrackUntiedDefs 
)

◆ initLiveThru() [1/2]

void llvm::RegPressureTracker::initLiveThru ( ArrayRef< unsigned PressureSet)
inline

Copy an existing live thru pressure result.

Definition at line 452 of file RegisterPressure.h.

References llvm::ArrayRef< T >::begin(), and llvm::ArrayRef< T >::end().

◆ initLiveThru() [2/2]

void RegPressureTracker::initLiveThru ( const RegPressureTracker RPTracker)

Initialize the LiveThru pressure set based on the untied defs found in RPTracker.

The register tracker is unaware of global liveness so ignores normal live-thru ranges.

However, two-address or coalesced chains can also lead to live ranges with no holes. Count these to inform heuristics that we can never drop below this pressure.

Definition at line 358 of file RegisterPressure.cpp.

References assert(), llvm::LaneBitmask::getNone(), llvm::TargetRegisterInfo::getNumRegPressureSets(), hasUntiedDef(), increaseSetPressure(), isBottomClosed(), llvm::Register::isVirtual(), llvm::RegisterMaskPair::LaneMask, llvm::RegisterPressure::LiveOutRegs, and llvm::RegisterMaskPair::RegUnit.

Referenced by llvm::ScheduleDAGMILive::initRegPressure().

◆ isBottomClosed()

bool RegPressureTracker::isBottomClosed ( ) const

Does this pressure result have a valid bottom position and live outs.

Definition at line 302 of file RegisterPressure.cpp.

References llvm::RegionPressure::BottomPos.

Referenced by advance(), closeRegion(), dump(), getMaxPressureDelta(), getPressureAfterInst(), initLiveThru(), and recedeSkipDebugValues().

◆ isTopClosed()

bool RegPressureTracker::isTopClosed ( ) const

Does this pressure result have a valid top position and live ins.

Definition at line 294 of file RegisterPressure.cpp.

References llvm::RegionPressure::TopPos.

Referenced by advance(), closeRegion(), dump(), getMaxPressureDelta(), getPressureAfterInst(), and recedeSkipDebugValues().

◆ recede() [1/2]

void RegPressureTracker::recede ( const RegisterOperands RegOpers,
SmallVectorImpl< RegisterMaskPair > *  LiveUses = nullptr 
)

Recede across the previous instruction.

This "low-level" variant assumes that recedeSkipDebugValues() was called previously and takes precomputed RegisterOperands for the instruction.

If LiveUses is provided, record any RegUnits that are made live by the current instruction's uses. This includes registers that are both defined and used by the instruction. If a pressure difference pointer is provided record the changes is pressure caused by this instruction independent of liveness.

Definition at line 755 of file RegisterPressure.cpp.

References addRegLanes(), llvm::LaneBitmask::any(), assert(), bumpDeadDefs(), llvm::LiveRegSet::contains(), llvm::RegisterOperands::DeadDefs, decreaseRegPressure(), llvm::RegisterOperands::Defs, discoverLiveOut(), llvm::SmallVectorTemplateCommon< T, typename >::end(), llvm::LiveRegSet::erase(), llvm::find_if(), llvm::LiveIntervals::getInstructionIndex(), getLiveThroughAt(), llvm::LaneBitmask::getNone(), llvm::SlotIndex::getRegSlot(), I, increaseRegPressure(), increaseSetPressure(), llvm::LiveRegSet::insert(), llvm::Register::isVirtual(), MRI, llvm::LaneBitmask::none(), llvm::Other, Reg, removeRegLanes(), setRegZero(), and llvm::RegisterOperands::Uses.

◆ recede() [2/2]

void RegPressureTracker::recede ( SmallVectorImpl< RegisterMaskPair > *  LiveUses = nullptr)

◆ recedeSkipDebugValues()

void RegPressureTracker::recedeSkipDebugValues ( )

◆ reset()

void RegPressureTracker::reset ( )

◆ setPos()

void llvm::RegPressureTracker::setPos ( MachineBasicBlock::const_iterator  Pos)
inline

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