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

[Florian Hahn via llvm-commits]

================
@@ -1375,6 +1375,17 @@ bool LoopVectorizationLegality::isFixedOrderRecurrence(
 }
 
 bool LoopVectorizationLegality::blockNeedsPredication(BasicBlock *BB) const {
+  // When vectorizing early exits, create predicates for the latch block. The
+  // early exiting block must be a direct predecessor of the latch at the
+  // moment.
+  BasicBlock *Latch = TheLoop->getLoopLatch();
+  if (hasUncountableEarlyExit()) {
+    assert(
+        getUncountableExitingBlocks().size() == 1 &&
----------------
fhahn wrote:

Updated to use the existing `getUncountableEarlyExitingBlock` which I missed originally, thanks

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