LoopVectorizer: Don't count the induction variable multiple times
When estimating register pressure, don't count the induction variable mulitple times. It is unlikely to be unrolled. This is currently disabled and hidden behind a flag ("enable-ind-var-reg-heur"). llvm-svn: 200371
This commit is contained in:
parent
7a578026ae
commit
1aab75ab49
|
@ -188,6 +188,10 @@ static cl::opt<unsigned> NumberOfStoresToPredicate(
|
|||
"vectorize-num-stores-pred", cl::init(0), cl::Hidden,
|
||||
cl::desc("Max number of stores to be predicated behind an if."));
|
||||
|
||||
static cl::opt<bool> EnableIndVarRegisterHeur(
|
||||
"enable-ind-var-reg-heur", cl::init(false), cl::Hidden,
|
||||
cl::desc("Count the induction variable only once when unrolling"));
|
||||
|
||||
static cl::opt<bool> EnableCondStoresVectorization(
|
||||
"enable-cond-stores-vec", cl::init(false), cl::Hidden,
|
||||
cl::desc("Enable if predication of stores during vectorization."));
|
||||
|
@ -5155,6 +5159,11 @@ LoopVectorizationCostModel::selectUnrollFactor(bool OptForSize,
|
|||
unsigned UF = PowerOf2Floor((TargetNumRegisters - R.LoopInvariantRegs) /
|
||||
R.MaxLocalUsers);
|
||||
|
||||
// Don't count the induction variable as unrolled.
|
||||
if (EnableIndVarRegisterHeur)
|
||||
UF = PowerOf2Floor((TargetNumRegisters - R.LoopInvariantRegs - 1) /
|
||||
std::max(1U, (R.MaxLocalUsers - 1)));
|
||||
|
||||
// Clamp the unroll factor ranges to reasonable factors.
|
||||
unsigned MaxUnrollSize = TTI.getMaximumUnrollFactor();
|
||||
|
||||
|
|
Loading…
Reference in New Issue