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

NexMing · Mel-Chen · web-flow · commit a80f972b90b8 · 2025-03-24T18:31:44.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
@@ -2435,7 +2435,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
`@@ -2435,7 +2435,7 @@ InnerLoopVectorizer::getOrCreateVectorTripCount(BasicBlock *InsertBlock) {`
`2435`	`2435`	`assert(!Cost->requiresScalarEpilogue(true) &&`
`2436`	`2436`	`"Use predicated EVL instructions for tail-folding does not allow "`
`2437`	`2437`	`"scalar epilogue");`
`2438`		`- return TC;`
	`2438`	`+ return VectorTripCount = TC;`
`2439`	`2439`	`}`
`2440`	`2440`
`2441`	`2441`	`// This is where we can make the step a runtime constant.`