LLVM: llvm::PeelingModuloScheduleExpander Class Reference (original) (raw)

#include "[llvm/CodeGen/ModuloSchedule.h](ModuloSchedule%5F8h%5Fsource.html)"

Public Member Functions
	PeelingModuloScheduleExpander (MachineFunction &MF, ModuloSchedule &S, LiveIntervals *LIS)

void	expand ()

void	validateAgainstModuloScheduleExpander ()
	Runs ModuloScheduleExpander and treats it as a golden input to validate aspects of the code generated by PeelingModuloScheduleExpander.

Protected Member Functions
void	rewriteKernel ()
	Converts BB from the original loop body to the rewritten, pipelined steady-state.

MachineBasicBlock *	peelKernel (LoopPeelDirection LPD)
	Peels one iteration of the rewritten kernel (BB) in the specified direction.

void	filterInstructions (MachineBasicBlock *MB, int MinStage)

void	moveStageBetweenBlocks (MachineBasicBlock DestBB, MachineBasicBlock SourceBB, unsigned Stage)

void	peelPrologAndEpilogs ()
	Peel the kernel forwards and backwards to produce prologs and epilogs, and stitch them together.

Register	getEquivalentRegisterIn (Register Reg, MachineBasicBlock *BB)
	All prolog and epilog blocks are clones of the kernel, so any produced register in one block has an corollary in all other blocks.

void	rewriteUsesOf (MachineInstr *MI)
	Change all users of MI, if MI is predicated out (LiveStages[MI->getParent()] == false).

void	fixupBranches ()
	Insert branches between prologs, kernel and epilogs.

MachineBasicBlock *	CreateLCSSAExitingBlock ()
	Create a poor-man's LCSSA by cloning only the PHIs from the kernel block to a block dominated by all prologs and epilogs.

unsigned	getStage (MachineInstr *MI)
	Helper to get the stage of an instruction in the schedule.

Register	getPhiCanonicalReg (MachineInstr CanonicalPhi, MachineInstr Phi)
	Helper function to find the right canonical register for a phi instruction coming from a peeled out prologue.

Protected Attributes
ModuloSchedule &	Schedule

MachineFunction &	MF

const TargetSubtargetInfo &	ST

MachineRegisterInfo &	MRI

const TargetInstrInfo *	TII = nullptr

LiveIntervals *	LIS = nullptr

MachineBasicBlock *	BB = nullptr
	The original loop block that gets rewritten in-place.

MachineBasicBlock *	Preheader = nullptr
	The original loop preheader.

SmallVector< MachineBasicBlock *, 4 >	Prologs
	All prolog and epilog blocks.

SmallVector< MachineBasicBlock *, 4 >	Epilogs

DenseMap< MachineBasicBlock *, BitVector >	LiveStages
	For every block, the stages that are produced.

DenseMap< MachineBasicBlock *, BitVector >	AvailableStages
	For every block, the stages that are available.

DenseMap< MachineInstr *, unsigned >	PhiNodeLoopIteration
	When peeling the epilogue keep track of the distance between the phi nodes and the kernel.

DenseMap< MachineInstr , MachineInstr >	CanonicalMIs
	CanonicalMIs and BlockMIs form a bidirectional map between any of the loop kernel clones.

DenseMap< std::pair< MachineBasicBlock , MachineInstr >, MachineInstr * >	BlockMIs

std::deque< MachineBasicBlock * >	PeeledFront
	State passed from peelKernel to peelPrologAndEpilogs().

std::deque< MachineBasicBlock * >	PeeledBack

SmallVector< MachineInstr *, 4 >	IllegalPhisToDelete
	Illegal phis that need to be deleted once we re-link stages.

std::unique_ptr< TargetInstrInfo::PipelinerLoopInfo >	LoopInfo
	Target loop info before kernel peeling.