LLVM 22.0.0git
Public Member Functions | Friends | List of all members
llvm::VPlan Class Reference

VPlan models a candidate for vectorization, encoding various decisions take to produce efficient output IR, including which branches, basic-blocks and output IR instructions to generate, and their cost. More...

#include "Transforms/Vectorize/VPlan.h"

Public Member Functions

 VPlan (Loop *L)
 Construct a VPlan for L.
 
 VPlan (BasicBlock *ScalarHeaderBB, VPValue *TC)
 Construct a VPlan with a new VPBasicBlock as entry, a VPIRBasicBlock wrapping ScalarHeaderBB and a trip count of TC.
 
LLVM_ABI_FOR_TEST ~VPlan ()
 
void setEntry (VPBasicBlock *VPBB)
 
void execute (VPTransformState *State)
 Generate the IR code for this VPlan.
 
InstructionCost cost (ElementCount VF, VPCostContext &Ctx)
 Return the cost of this plan.
 
VPBasicBlockgetEntry ()
 
const VPBasicBlockgetEntry () const
 
VPBasicBlockgetVectorPreheader ()
 Returns the preheader of the vector loop region, if one exists, or null otherwise.
 
LLVM_ABI_FOR_TEST VPRegionBlockgetVectorLoopRegion ()
 Returns the VPRegionBlock of the vector loop.
 
LLVM_ABI_FOR_TEST const VPRegionBlockgetVectorLoopRegion () const
 
VPBasicBlockgetMiddleBlock ()
 Returns the 'middle' block of the plan, that is the block that selects whether to execute the scalar tail loop or the exit block from the loop latch.
 
const VPBasicBlockgetMiddleBlock () const
 
VPBasicBlockgetScalarPreheader () const
 Return the VPBasicBlock for the preheader of the scalar loop.
 
VPIRBasicBlockgetScalarHeader () const
 Return the VPIRBasicBlock wrapping the header of the scalar loop.
 
ArrayRef< VPIRBasicBlock * > getExitBlocks () const
 Return an ArrayRef containing VPIRBasicBlocks wrapping the exit blocks of the original scalar loop.
 
VPIRBasicBlockgetExitBlock (BasicBlock *IRBB) const
 Return the VPIRBasicBlock corresponding to IRBB.
 
bool isExitBlock (VPBlockBase *VPBB)
 Returns true if VPBB is an exit block.
 
VPValuegetTripCount () const
 The trip count of the original loop.
 
void setTripCount (VPValue *NewTripCount)
 Set the trip count assuming it is currently null; if it is not - use resetTripCount().
 
void resetTripCount (VPValue *NewTripCount)
 Resets the trip count for the VPlan.
 
VPValuegetOrCreateBackedgeTakenCount ()
 The backedge taken count of the original loop.
 
VPValuegetVectorTripCount ()
 The vector trip count.
 
VPValuegetVF ()
 Returns the VF of the vector loop region.
 
VPValuegetVFxUF ()
 Returns VF * UF of the vector loop region.
 
LLVMContextgetContext () const
 
void addVF (ElementCount VF)
 
void setVF (ElementCount VF)
 
bool hasVF (ElementCount VF) const
 
bool hasScalableVF () const
 
iterator_range< SmallSetVector< ElementCount, 2 >::iterator > vectorFactors () const
 Returns an iterator range over all VFs of the plan.
 
bool hasScalarVFOnly () const
 
bool hasUF (unsigned UF) const
 
unsigned getUF () const
 
void setUF (unsigned UF)
 
bool isUnrolled () const
 Returns true if the VPlan already has been unrolled, i.e.
 
std::string getName () const
 Return a string with the name of the plan and the applicable VFs and UFs.
 
void setName (const Twine &newName)
 
VPValuegetOrAddLiveIn (Value *V)
 Gets the live-in VPValue for V or adds a new live-in (if none exists yet) for V.
 
VPValuegetTrue ()
 Return a VPValue wrapping i1 true.
 
VPValuegetFalse ()
 Return a VPValue wrapping i1 false.
 
VPValuegetLiveIn (Value *V) const
 Return the live-in VPValue for V, if there is one or nullptr otherwise.
 
ArrayRef< VPValue * > getLiveIns () const
 Return the list of live-in VPValues available in the VPlan.
 
void printLiveIns (raw_ostream &O) const
 Print the live-ins of this VPlan to O.
 
void print (raw_ostream &O) const
 Print this VPlan to O.
 
void printDOT (raw_ostream &O) const
 Print this VPlan in DOT format to O.
 
LLVM_DUMP_METHOD void dump () const
 Dump the plan to stderr (for debugging).
 
VPCanonicalIVPHIRecipegetCanonicalIV ()
 Returns the canonical induction recipe of the vector loop.
 
VPValuegetSCEVExpansion (const SCEV *S) const
 
void addSCEVExpansion (const SCEV *S, VPValue *V)
 
VPlanduplicate ()
 Clone the current VPlan, update all VPValues of the new VPlan and cloned recipes to refer to the clones, and return it.
 
VPBasicBlockcreateVPBasicBlock (const Twine &Name, VPRecipeBase *Recipe=nullptr)
 Create a new VPBasicBlock with Name and containing Recipe if present.
 
VPRegionBlockcreateVPRegionBlock (VPBlockBase *Entry, VPBlockBase *Exiting, const std::string &Name="", bool IsReplicator=false)
 Create a new VPRegionBlock with Entry, Exiting and Name.
 
VPRegionBlockcreateVPRegionBlock (const std::string &Name="")
 Create a new loop VPRegionBlock with Name and entry and exiting blocks set to nullptr.
 
VPIRBasicBlockcreateEmptyVPIRBasicBlock (BasicBlock *IRBB)
 Create a VPIRBasicBlock wrapping IRBB, but do not create VPIRInstructions wrapping the instructions in tIRBB.
 
LLVM_ABI_FOR_TEST VPIRBasicBlockcreateVPIRBasicBlock (BasicBlock *IRBB)
 Create a VPIRBasicBlock from IRBB containing VPIRInstructions for all instructions in IRBB, except its terminator which is managed by the successors of the block in VPlan.
 
bool hasEarlyExit () const
 Returns true if the VPlan is based on a loop with an early exit.
 
bool hasScalarTail () const
 Returns true if the scalar tail may execute after the vector loop.
 

Friends

class VPlanPrinter
 
class VPSlotTracker
 

Detailed Description

VPlan models a candidate for vectorization, encoding various decisions take to produce efficient output IR, including which branches, basic-blocks and output IR instructions to generate, and their cost.

VPlan holds a Hierarchical-CFG of VPBasicBlocks and VPRegionBlocks rooted at an Entry VPBasicBlock.

Definition at line 3930 of file VPlan.h.

Constructor & Destructor Documentation

◆ VPlan() [1/2]

VPlan::VPlan ( Loop L)

Construct a VPlan for L.

This will create VPIRBasicBlocks wrapping the original preheader and scalar header of L, to be used as entry and scalar header blocks of the new VPlan.

Definition at line 907 of file VPlan.cpp.

References createVPIRBasicBlock(), and setEntry().

◆ VPlan() [2/2]

llvm::VPlan::VPlan ( BasicBlock ScalarHeaderBB,
VPValue TC 
)
inline

Construct a VPlan with a new VPBasicBlock as entry, a VPIRBasicBlock wrapping ScalarHeaderBB and a trip count of TC.

Definition at line 4010 of file VPlan.h.

References createVPBasicBlock(), createVPIRBasicBlock(), and setEntry().

◆ ~VPlan()

VPlan::~VPlan ( )

Definition at line 917 of file VPlan.cpp.

References getLiveIns(), and I.

Member Function Documentation

◆ addSCEVExpansion()

void llvm::VPlan::addSCEVExpansion ( const SCEV S,
VPValue V 
)
inline

◆ addVF()

void llvm::VPlan::addVF ( ElementCount  VF)
inline

Definition at line 4131 of file VPlan.h.

References llvm::SetVector< T, Vector, Set, N >::insert().

◆ cost()

InstructionCost VPlan::cost ( ElementCount  VF,
VPCostContext Ctx 
)

◆ createEmptyVPIRBasicBlock()

VPIRBasicBlock * VPlan::createEmptyVPIRBasicBlock ( BasicBlock IRBB)

Create a VPIRBasicBlock wrapping IRBB, but do not create VPIRInstructions wrapping the instructions in tIRBB.

The returned block is owned by the VPlan and deleted once the VPlan is destroyed.

Definition at line 1246 of file VPlan.cpp.

Referenced by llvm::VPIRBasicBlock::clone(), and createVPIRBasicBlock().

◆ createVPBasicBlock()

VPBasicBlock * llvm::VPlan::createVPBasicBlock ( const Twine Name,
VPRecipeBase Recipe = nullptr 
)
inline

Create a new VPBasicBlock with Name and containing Recipe if present.

The returned block is owned by the VPlan and deleted once the VPlan is destroyed.

Definition at line 4260 of file VPlan.h.

References Name, and llvm::SmallVectorTemplateBase< T, bool >::push_back().

Referenced by addInitialSkeleton(), llvm::VPBasicBlock::clone(), createReplicateRegion(), llvm::VPlanTransforms::handleUncountableEarlyExit(), llvm::VPBasicBlock::splitAt(), and VPlan().

◆ createVPIRBasicBlock()

VPIRBasicBlock * VPlan::createVPIRBasicBlock ( BasicBlock IRBB)

Create a VPIRBasicBlock from IRBB containing VPIRInstructions for all instructions in IRBB, except its terminator which is managed by the successors of the block in VPlan.

The returned block is owned by the VPlan and deleted once the VPlan is destroyed.

Definition at line 1252 of file VPlan.cpp.

References llvm::BasicBlock::begin(), llvm::VPIRInstruction::create(), createEmptyVPIRBasicBlock(), llvm::ilist_node_impl< OptionsT >::getIterator(), llvm::BasicBlock::getTerminator(), I, and llvm::make_range().

Referenced by llvm::VPlanTransforms::attachCheckBlock(), duplicate(), llvm::EpilogueVectorizerEpilogueLoop::emitMinimumVectorEpilogueIterCountCheck(), llvm::EpilogueVectorizerMainLoop::introduceCheckBlockInVPlan(), replaceVPBBWithIRVPBB(), and VPlan().

◆ createVPRegionBlock() [1/2]

VPRegionBlock * llvm::VPlan::createVPRegionBlock ( const std::string &  Name = "")
inline

Create a new loop VPRegionBlock with Name and entry and exiting blocks set to nullptr.

The returned block is owned by the VPlan and deleted once the VPlan is destroyed.

Definition at line 4281 of file VPlan.h.

References Name, and llvm::SmallVectorTemplateBase< T, bool >::push_back().

◆ createVPRegionBlock() [2/2]

VPRegionBlock * llvm::VPlan::createVPRegionBlock ( VPBlockBase Entry,
VPBlockBase Exiting,
const std::string &  Name = "",
bool  IsReplicator = false 
)
inline

Create a new VPRegionBlock with Entry, Exiting and Name.

If IsReplicator is true, the region is a replicate region. The returned block is owned by the VPlan and deleted once the VPlan is destroyed.

Definition at line 4270 of file VPlan.h.

References Name, and llvm::SmallVectorTemplateBase< T, bool >::push_back().

Referenced by llvm::VPRegionBlock::clone(), createLoopRegion(), and createReplicateRegion().

◆ dump()

LLVM_DUMP_METHOD void VPlan::dump ( ) const

Dump the plan to stderr (for debugging).

Definition at line 1138 of file VPlan.cpp.

References llvm::dbgs(), and print().

Referenced by execute().

◆ duplicate()

VPlan * VPlan::duplicate ( )

◆ execute()

void VPlan::execute ( VPTransformState State)

◆ getCanonicalIV()

VPCanonicalIVPHIRecipe * llvm::VPlan::getCanonicalIV ( )
inline

◆ getContext()

LLVMContext & llvm::VPlan::getContext ( ) const
inline

◆ getEntry() [1/2]

VPBasicBlock * llvm::VPlan::getEntry ( )
inline

◆ getEntry() [2/2]

const VPBasicBlock * llvm::VPlan::getEntry ( ) const
inline

Definition at line 4030 of file VPlan.h.

◆ getExitBlock()

VPIRBasicBlock * VPlan::getExitBlock ( BasicBlock IRBB) const

Return the VPIRBasicBlock corresponding to IRBB.

IRBB must be an exit block.

Definition at line 940 of file VPlan.cpp.

References assert(), llvm::find_if(), getExitBlocks(), and llvm::VPIRBasicBlock::getIRBasicBlock().

◆ getExitBlocks()

ArrayRef< VPIRBasicBlock * > llvm::VPlan::getExitBlocks ( ) const
inline

Return an ArrayRef containing VPIRBasicBlocks wrapping the exit blocks of the original scalar loop.

Definition at line 4081 of file VPlan.h.

Referenced by calculateEarlyExitCost(), createExtractsForLiveOuts(), llvm::LoopVectorizationPlanner::executePlan(), getExitBlock(), llvm::VPlanTransforms::handleEarlyExits(), and llvm::VPlanTransforms::optimizeInductionExitUsers().

◆ getFalse()

VPValue * llvm::VPlan::getFalse ( )
inline

◆ getLiveIn()

VPValue * llvm::VPlan::getLiveIn ( Value V) const
inline

Return the live-in VPValue for V, if there is one or nullptr otherwise.

Definition at line 4208 of file VPlan.h.

References llvm::DenseMapBase< DerivedT, KeyT, ValueT, KeyInfoT, BucketT >::lookup().

◆ getLiveIns()

ArrayRef< VPValue * > llvm::VPlan::getLiveIns ( ) const
inline

Return the list of live-in VPValues available in the VPlan.

Definition at line 4211 of file VPlan.h.

References llvm::all_of(), assert(), llvm::is_contained(), and P.

Referenced by duplicate(), llvm::VPlanTransforms::materializeBroadcasts(), and ~VPlan().

◆ getMiddleBlock() [1/2]

VPBasicBlock * llvm::VPlan::getMiddleBlock ( )
inline

Returns the 'middle' block of the plan, that is the block that selects whether to execute the scalar tail loop or the exit block from the loop latch.

If there is an early exit from the vector loop, the middle block conceptully has the early exit block as third successor, split accross 2 VPBBs. In that case, the second VPBB selects whether to execute the scalar tail loop or the exit bock. If the scalar tail loop or exit block are known to always execute, the middle block may branch directly to that block. This function cannot be called once the vector loop region has been removed.

Definition at line 4054 of file VPlan.h.

References assert(), getScalarPreheader(), llvm::VPBlockBase::getSingleSuccessor(), getVectorLoopRegion(), and llvm::is_contained().

Referenced by llvm::VPlanTransforms::addBranchWeightToMiddleTerminator(), addExitUsersForFirstOrderRecurrences(), addScalarResumePhis(), calculateEarlyExitCost(), getMiddleBlock(), llvm::VPlanTransforms::handleMaxMinNumReductions(), llvm::VPlanTransforms::materializeConstantVectorTripCount(), llvm::VPlanTransforms::optimizeInductionExitUsers(), preparePlanForMainVectorLoop(), and llvm::LoopVectorizePass::processLoop().

◆ getMiddleBlock() [2/2]

const VPBasicBlock * llvm::VPlan::getMiddleBlock ( ) const
inline

Definition at line 4067 of file VPlan.h.

References getMiddleBlock().

◆ getName()

std::string VPlan::getName ( ) const

Return a string with the name of the plan and the applicable VFs and UFs.

Definition at line 1108 of file VPlan.cpp.

References llvm::drop_begin(), and llvm::SetVector< T, Vector, Set, N >::empty().

Referenced by llvm::VPlanPrinter::dump(), and print().

◆ getOrAddLiveIn()

VPValue * llvm::VPlan::getOrAddLiveIn ( Value V)
inline

◆ getOrCreateBackedgeTakenCount()

VPValue * llvm::VPlan::getOrCreateBackedgeTakenCount ( )
inline

◆ getScalarHeader()

VPIRBasicBlock * llvm::VPlan::getScalarHeader ( ) const
inline

◆ getScalarPreheader()

VPBasicBlock * llvm::VPlan::getScalarPreheader ( ) const
inline

◆ getSCEVExpansion()

VPValue * llvm::VPlan::getSCEVExpansion ( const SCEV S) const
inline

◆ getTripCount()

VPValue * llvm::VPlan::getTripCount ( ) const
inline

◆ getTrue()

VPValue * llvm::VPlan::getTrue ( )
inline

◆ getUF()

unsigned llvm::VPlan::getUF ( ) const
inline

◆ getVectorLoopRegion() [1/2]

VPRegionBlock * VPlan::getVectorLoopRegion ( )

Returns the VPRegionBlock of the vector loop.

Definition at line 1037 of file VPlan.cpp.

References B, getEntry(), and llvm::vp_depth_first_shallow().

Referenced by addExitUsersForFirstOrderRecurrences(), llvm::VPlanTransforms::addExplicitVectorLength(), addScalarResumePhis(), addVPLaneMaskPhiAndUpdateExitBranch(), llvm::VPlanTransforms::adjustFixedOrderRecurrences(), llvm::LoopVectorizationPlanner::buildVPlans(), llvm::calculateRegisterUsageForPlan(), llvm::VPlanTransforms::clearReductionWrapFlags(), llvm::collectEphemeralRecipesForVPlan(), llvm::VPlanTransforms::convertToAbstractRecipes(), cost(), llvm::VPlanTransforms::createLoopRegions(), createScalarIVSteps(), llvm::LoopVectorizationPlanner::emitInvalidCostRemarks(), expandVPWidenIntOrFpInduction(), expandVPWidenPointerInduction(), findHeaderMask(), getCanonicalIV(), getMiddleBlock(), getVectorPreheader(), llvm::VPlanTransforms::handleMaxMinNumReductions(), hasReplicatorRegion(), llvm::VPlanTransforms::introduceMasksAndLinearize(), isDefinedInsideLoopRegions(), legalizeAndOptimizeInductions(), licm(), llvm::VPlanTransforms::materializeBuildVectors(), llvm::VPlanTransforms::narrowInterleaveGroups(), narrowToSingleScalarRecipes(), optimizeVectorInductionWidthForTCAndVFUF(), planContainsAdditionalSimplifications(), preparePlanForEpilogueVectorLoop(), preparePlanForMainVectorLoop(), removeRedundantCanonicalIVs(), removeRedundantInductionCasts(), llvm::VPlanTransforms::replicateByVF(), scalarizeInstruction(), llvm::LoopVectorizationPlanner::selectInterleaveCount(), simplifyBlends(), simplifyBranchConditionForVFAndUF(), transformRecipestoEVLRecipes(), llvm::VPlanTransforms::truncateToMinimalBitwidths(), llvm::VPlanTransforms::unrollByUF(), llvm::VPInterleavedAccessInfo::VPInterleavedAccessInfo(), llvm::VPTypeAnalysis::VPTypeAnalysis(), and willGenerateVectors().

◆ getVectorLoopRegion() [2/2]

const VPRegionBlock * VPlan::getVectorLoopRegion ( ) const

Definition at line 1045 of file VPlan.cpp.

References B, getEntry(), and llvm::vp_depth_first_shallow().

◆ getVectorPreheader()

VPBasicBlock * llvm::VPlan::getVectorPreheader ( )
inline

◆ getVectorTripCount()

VPValue & llvm::VPlan::getVectorTripCount ( )
inline

◆ getVF()

VPValue & llvm::VPlan::getVF ( )
inline

◆ getVFxUF()

VPValue & llvm::VPlan::getVFxUF ( )
inline

◆ hasEarlyExit()

bool llvm::VPlan::hasEarlyExit ( ) const
inline

Returns true if the VPlan is based on a loop with an early exit.

That is the case if the VPlan has either more than one exit block or a single exit block with multiple predecessors (one for the exit via the latch and one via the other early exit).

Definition at line 4302 of file VPlan.h.

References llvm::count_if(), llvm::VPBlockBase::getNumPredecessors(), and llvm::SmallVectorBase< Size_T >::size().

Referenced by llvm::LoopVectorizationPlanner::computeBestVF(), llvm::LoopVectorizationPlanner::executePlan(), and llvm::LoopVectorizationPlanner::selectInterleaveCount().

◆ hasScalableVF()

bool llvm::VPlan::hasScalableVF ( ) const
inline

◆ hasScalarTail()

bool llvm::VPlan::hasScalarTail ( ) const
inline

Returns true if the scalar tail may execute after the vector loop.

Note that this relies on unneeded branches to the scalar tail loop being removed.

Definition at line 4313 of file VPlan.h.

References getEntry(), llvm::VPBlockBase::getNumPredecessors(), getScalarPreheader(), and llvm::VPBlockBase::getSinglePredecessor().

Referenced by llvm::VPlanTransforms::handleMaxMinNumReductions(), and llvm::VPlanTransforms::materializeConstantVectorTripCount().

◆ hasScalarVFOnly()

bool llvm::VPlan::hasScalarVFOnly ( ) const
inline

◆ hasUF()

bool llvm::VPlan::hasUF ( unsigned  UF) const
inline

◆ hasVF()

bool llvm::VPlan::hasVF ( ElementCount  VF) const
inline

◆ isExitBlock()

bool VPlan::isExitBlock ( VPBlockBase VPBB)

Returns true if VPBB is an exit block.

Definition at line 948 of file VPlan.cpp.

References llvm::is_contained().

Referenced by llvm::VPBasicBlock::connectToPredecessors().

◆ isUnrolled()

bool llvm::VPlan::isUnrolled ( ) const
inline

Returns true if the VPlan already has been unrolled, i.e.

it has a single concrete UF.

Definition at line 4172 of file VPlan.h.

References llvm::SetVector< T, Vector, Set, N >::size().

Referenced by simplifyRecipe().

◆ print()

LLVM_DUMP_METHOD void VPlan::print ( raw_ostream O) const

Print this VPlan to O.

Definition at line 1091 of file VPlan.cpp.

References llvm::Block, getEntry(), getName(), and printLiveIns().

Referenced by dump(), llvm::operator<<(), and llvm::LoopVectorizationPlanner::printPlans().

◆ printDOT()

LLVM_DUMP_METHOD void VPlan::printDOT ( raw_ostream O) const

Print this VPlan in DOT format to O.

Definition at line 1132 of file VPlan.cpp.

References llvm::Metadata::dump(), and Printer.

Referenced by llvm::LoopVectorizationPlanner::printPlans().

◆ printLiveIns()

void VPlan::printLiveIns ( raw_ostream O) const

Print the live-ins of this VPlan to O.

Definition at line 1053 of file VPlan.cpp.

References llvm::VPValue::getNumUsers(), llvm::VPValue::isLiveIn(), and llvm::VPValue::printAsOperand().

Referenced by llvm::VPlanPrinter::dump(), and print().

◆ resetTripCount()

void llvm::VPlan::resetTripCount ( VPValue NewTripCount)
inline

Resets the trip count for the VPlan.

The caller must make sure all uses of the original trip count have been replaced.

Definition at line 4105 of file VPlan.h.

References assert(), and llvm::VPValue::getNumUsers().

Referenced by llvm::VPlanTransforms::expandSCEVs(), and preparePlanForEpilogueVectorLoop().

◆ setEntry()

void llvm::VPlan::setEntry ( VPBasicBlock VPBB)
inline

◆ setName()

void llvm::VPlan::setName ( const Twine newName)
inline

Definition at line 4177 of file VPlan.h.

References Name, and llvm::Twine::str().

Referenced by execute().

◆ setTripCount()

void llvm::VPlan::setTripCount ( VPValue NewTripCount)
inline

Set the trip count assuming it is currently null; if it is not - use resetTripCount().

Definition at line 4098 of file VPlan.h.

References assert().

Referenced by addInitialSkeleton().

◆ setUF()

void llvm::VPlan::setUF ( unsigned  UF)
inline

◆ setVF()

void llvm::VPlan::setVF ( ElementCount  VF)
inline

◆ vectorFactors()

iterator_range< SmallSetVector< ElementCount, 2 >::iterator > llvm::VPlan::vectorFactors ( ) const
inline

Returns an iterator range over all VFs of the plan.

Definition at line 4146 of file VPlan.h.

Referenced by llvm::LoopVectorizationPlanner::computeBestVF(), and llvm::LoopVectorizationPlanner::emitInvalidCostRemarks().

Friends And Related Function Documentation

◆ VPlanPrinter

friend class VPlanPrinter
friend

Definition at line 3931 of file VPlan.h.

◆ VPSlotTracker

friend class VPSlotTracker
friend

Definition at line 3932 of file VPlan.h.


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