Address review comment

Author: david-arm

llvm/lib/Transforms/Vectorize/LoopVectorize.cpp

@@ -10213,15 +10213,19 @@ bool LoopVectorizePass::processLoop(Loop *L) {
     // of a loop-defined variable, but it ignores induction variables which are
     // handled by InnerLoopVectorizer::fixupIVUsers. We need to bail out if we
     // encounter induction variables too otherwise fixupIVUsers will crash.
-    for (BasicBlock *BB : L->blocks()) {
-      for (Instruction &I : *BB) {
-        for (User *U : I.users()) {
+    BasicBlock *LoopLatch = L->getLoopLatch();
+    for (const auto &Induction : LVL.getInductionVars()) {
+      PHINode *Ind = Induction.first;
+      Instruction *IndUpdate =
+          cast<Instruction>(Ind->getIncomingValueForBlock(LoopLatch));
+      for (Instruction *I : {cast<Instruction>(Ind), IndUpdate}) {
+        for (User *U : I->users()) {
           Instruction *UI = cast<Instruction>(U);
           if (!L->contains(UI)) {
             reportVectorizationFailure(
-                "Auto-vectorization of loops with uncountable "
-                "early exit and live-outs is not supported",
-                "UncountableEarlyExitLoopLiveOutsUnsupported", ORE, L);
+                "Auto-vectorization of loops with uncountable early exits and "
+                "outside uses of induction variables unsupported",
+                "UncountableEarlyExitLoopIndLiveOutsUnsupported", ORE, L);
             return false;
           }
         }

llvm/test/Transforms/LoopVectorize/early_exit_legality.ll

@@ -49,7 +49,7 @@ define i64 @same_exit_block_pre_inc_use1() {
 ; CHECK-LABEL: LV: Checking a loop in 'same_exit_block_pre_inc_use1'
 ; CHECK:       LV: Found an early exit loop with symbolic max backedge taken count: 63
 ; CHECK-NEXT:  LV: We can vectorize this loop!
-; CHECK-NEXT:  LV: Not vectorizing: Auto-vectorization of loops with uncountable early exit and live-outs is not supported.
+; CHECK-NEXT:  LV: Not vectorizing: Auto-vectorization of loops with uncountable early exits and outside uses of induction variables unsupported
 entry:
   %p1 = alloca [1024 x i8]
   %p2 = alloca [1024 x i8]
@@ -141,7 +141,7 @@ define i64 @loop_contains_load_after_early_exit(ptr dereferenceable(1024) align(
 ; CHECK-LABEL: LV: Checking a loop in 'loop_contains_load_after_early_exit'
 ; CHECK:       LV: Found an early exit loop with symbolic max backedge taken count: 63
 ; CHECK-NEXT:  LV: We can vectorize this loop!
-; CHECK:       LV: Not vectorizing: Auto-vectorization of loops with uncountable early exit and live-outs is not supported.
+; CHECK:       LV: Not vectorizing: Auto-vectorization of loops with uncountable early exits and outside uses of induction variables unsupported
 entry:
   %p1 = alloca [1024 x i8]
   call void @init_mem(ptr %p1, i64 1024)