[llvm] [LV] Transform to handle exits in the scalar loop (PR #148626)

[David Sherwood via llvm-commits]

================
@@ -588,10 +588,13 @@ void VPlanTransforms::handleEarlyExits(VPlan &Plan,
         handleUncountableEarlyExit(cast<VPBasicBlock>(Pred), EB, Plan,
                                    cast<VPBasicBlock>(HeaderVPB), LatchVPBB);
         HandledUncountableEarlyExit = true;
-      } else {
+      }
+
+      if (!HasUncountableEarlyExit ||
+          Plan.shouldEarlyExitContinueInScalarLoop())
----------------
david-arm wrote:

Can't you just pass this flag in as an argument to handleEarlyExits, similar to `HasUncountableEarlyExit`? If you follow @sdesmalen-arm's suggestion for an enum, then you could just pass in a single enum here.

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