Check the address is defined outside vector region.

llvm · Sep 10, 2024 · 206be4f · 206be4f
1 parent 8274b71
commit 206be4f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/llvm/lib/Transforms/Vectorize/LoopVectorize.cpp b/llvm/lib/Transforms/Vectorize/LoopVectorize.cpp
@@ -9422,7 +9422,10 @@ void LoopVectorizationPlanner::adjustRecipesForReductions(
                                   m_VPValue(), m_VPValue())) ||
                  (isa<VPReplicateRecipe>(&User) &&
                   cast<VPReplicateRecipe>(&User)->getUnderlyingValue() ==
-                      IntermediateStore);
+                      IntermediateStore &&
+                  cast<VPReplicateRecipe>(&User)
+                      ->getOperand(1)
+                      ->isDefinedOutsideVectorRegions());
         });
   }