diff --git a/llvm/lib/Transforms/Vectorize/VectorCombine.cpp b/llvm/lib/Transforms/Vectorize/VectorCombine.cpp --- a/llvm/lib/Transforms/Vectorize/VectorCombine.cpp +++ b/llvm/lib/Transforms/Vectorize/VectorCombine.cpp @@ -831,8 +831,14 @@ Builder.Insert(GEP); StoreInst *NSI = Builder.CreateStore(NewElement, GEP); NSI->copyMetadata(*SI); - if (SI->getAlign() < NSI->getAlign()) - NSI->setAlignment(SI->getAlign()); + Align NewAlignment(1); + if (auto *C = dyn_cast(Idx)) { + NewAlignment = std::max(SI->getAlign(), Load->getAlign()); + NewAlignment = commonAlignment( + NewAlignment, + C->getZExtValue() * DL.getTypeStoreSize(NewElement->getType())); + } + NSI->setAlignment(NewAlignment); replaceValue(I, *NSI); // Need erasing the store manually. I.eraseFromParent(); diff --git a/llvm/test/Transforms/VectorCombine/load-insert-store.ll b/llvm/test/Transforms/VectorCombine/load-insert-store.ll --- a/llvm/test/Transforms/VectorCombine/load-insert-store.ll +++ b/llvm/test/Transforms/VectorCombine/load-insert-store.ll @@ -20,7 +20,7 @@ ; CHECK-LABEL: @insert_store_i16_align1( ; CHECK-NEXT: entry: ; CHECK-NEXT: [[TMP0:%.*]] = getelementptr inbounds <8 x i16>, <8 x i16>* [[Q:%.*]], i32 0, i32 3 -; CHECK-NEXT: store i16 [[S:%.*]], i16* [[TMP0]], align 1 +; CHECK-NEXT: store i16 [[S:%.*]], i16* [[TMP0]], align 2 ; CHECK-NEXT: ret void ; entry: @@ -125,6 +125,19 @@ ret void } +define void @nonconst_large_alignment(<8 x i32>* %q, i32 zeroext %s, i32 %idx) { +; CHECK-LABEL: @nonconst_large_alignment( +; CHECK-NEXT: [[I:%.*]] = load <8 x i32>, <8 x i32>* [[Q:%.*]], align 128 +; CHECK-NEXT: [[VECINS:%.*]] = insertelement <8 x i32> [[I]], i32 [[S:%.*]], i32 [[IDX:%.*]] +; CHECK-NEXT: store <8 x i32> [[VECINS]], <8 x i32>* [[Q]], align 128 +; CHECK-NEXT: ret void +; + %i = load <8 x i32>, <8 x i32>* %q, align 128 + %vecins = insertelement <8 x i32> %i, i32 %s, i32 %idx + store <8 x i32> %vecins, <8 x i32>* %q, align 128 + ret void +} + define void @insert_store_nonconst_index_known_valid_by_assume(<16 x i8>* %q, i8 zeroext %s, i32 %idx) { ; CHECK-LABEL: @insert_store_nonconst_index_known_valid_by_assume( ; CHECK-NEXT: entry: