Skip to content

Commit dea6a00

Browse files
committed
[SLP][REVEC] Apply comments.
1 parent 7b503e0 commit dea6a00

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13772,9 +13772,12 @@ Value *BoUpSLP::vectorizeTree(TreeEntry *E, bool PostponedPHIs) {
1377213772
"Cannot expand getelementptr.");
1377313773
unsigned VF = VecTyNumElements / ScalarTyNumElements;
1377413774
SmallVector<Constant *> Indices(VecTyNumElements);
13775-
transform(seq(VecTyNumElements), Indices.begin(), [=](unsigned I) {
13776-
return Builder.getInt64(I % ScalarTyNumElements);
13777-
});
13775+
transform(seq(ScalarTyNumElements), Indices.begin(),
13776+
[=](unsigned I) { return Builder.getInt64(I); });
13777+
for (int Pos : createStrideMask(ScalarTyNumElements,
13778+
ScalarTyNumElements, VF - 1))
13779+
std::copy_n(Indices.begin(), ScalarTyNumElements,
13780+
Indices.begin() + Pos);
1377813781
VecPtr = Builder.CreateGEP(
1377913782
VecTy->getElementType(),
1378013783
Builder.CreateShuffleVector(

0 commit comments

Comments
 (0)