Update llvm/lib/Transforms/Vectorize/LoopVectorize.cpp

NexMing · Mel-Chen · NexMing · commit ef2e77667cc6 · 2025-03-31T15:12:12.000+08:00
Co-authored-by: Mel Chen &lt;mel.chen@sifive.com&gt;
diff --git a/llvm/lib/Transforms/Vectorize/LoopVectorize.cpp b/llvm/lib/Transforms/Vectorize/LoopVectorize.cpp
@@ -2434,7 +2434,7 @@ InnerLoopVectorizer::getOrCreateVectorTripCount(BasicBlock *InsertBlock) {
     assert(!Cost->requiresScalarEpilogue(true) &&
            "Use predicated EVL instructions for tail-folding does not allow "
            "scalar epilogue");
-    return TC;
+    return VectorTripCount = TC;
   }
 
   // This is where we can make the step a runtime constant.

Original file line number	Diff line number	Diff line change
`@@ -2434,7 +2434,7 @@ InnerLoopVectorizer::getOrCreateVectorTripCount(BasicBlock *InsertBlock) {`
`2434`	`2434`	`assert(!Cost->requiresScalarEpilogue(true) &&`
`2435`	`2435`	`"Use predicated EVL instructions for tail-folding does not allow "`
`2436`	`2436`	`"scalar epilogue");`
`2437`		`- return TC;`
	`2437`	`+ return VectorTripCount = TC;`
`2438`	`2438`	`}`
`2439`	`2439`
`2440`	`2440`	`// This is where we can make the step a runtime constant.`