[llvm] [VPlan] Dispatch to multiple exit blocks via middle blocks. (PR #112138)

Florian Hahn via llvm-commits llvm-commits at lists.llvm.org
Tue Dec 10 06:08:16 PST 2024


================
@@ -9238,11 +9259,25 @@ LoopVectorizationPlanner::tryToBuildVPlanWithVPRecipes(VFRange &Range) {
          "VPBasicBlock");
   RecipeBuilder.fixHeaderPhis();
 
+  if (Legal->hasUncountableEarlyExit()) {
+    assert(Legal->getUncountableExitingBlocks().size() == 1 &&
----------------
fhahn wrote:

Updated, it required updating `getUncountableEarlyExitingBlock` to return nullptr in if `!UncountableEarlyExit` and some slight adjustments on how it is set (and clear the entries if it is false).

WDYT @david-arm ?

https://github.com/llvm/llvm-project/pull/112138


More information about the llvm-commits mailing list