[llvm] [LoopVectorize] Enable vectorisation of early exit loops with live-outs (PR #120567)

[David Sherwood via llvm-commits]

================
@@ -306,6 +306,24 @@ template <> struct GraphTraits<VPlan *> {
   }
 };
 
+inline bool VPlan::isExitBlock(VPBlockBase *VPBB) {
+  if (!isa<VPIRBasicBlock>(VPBB) || VPBB->getNumSuccessors() ||
+      VPBB == getScalarHeader())
+    return false;
+
+  VPRegionBlock *RegionBlock = getVectorLoopRegion();
+  if (!RegionBlock)
+    return false;
+
+  // The block must be a successor of the region block.
+  for (auto *OtherVPBB :
----------------
david-arm wrote:

That requires a lambda function, which I'm not sure is any fewer lines of code, right? Perhaps llvm::is_contained would be more compact than the current version

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