[llvm] r276943 - Fix the assertion error in collectLoopUniforms caused by empty Worklist before expanding.

[Wei Mi via llvm-commits]

Author: wmi
Date: Wed Jul 27 18:53:58 2016
New Revision: 276943

URL: http://llvm.org/viewvc/llvm-project?rev=276943&view=rev
Log:
Fix the assertion error in collectLoopUniforms caused by empty Worklist before expanding.

Contributed-by: David Callahan

Differential Revision: https://reviews.llvm.org/D22886

Added:
    llvm/trunk/test/Transforms/LoopVectorize/2016-07-27-loop-vec.ll
Modified:
    llvm/trunk/lib/Transforms/Vectorize/LoopVectorize.cpp

Modified: llvm/trunk/lib/Transforms/Vectorize/LoopVectorize.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Transforms/Vectorize/LoopVectorize.cpp?rev=276943&r1=276942&r2=276943&view=diff
==============================================================================
--- llvm/trunk/lib/Transforms/Vectorize/LoopVectorize.cpp (original)
+++ llvm/trunk/lib/Transforms/Vectorize/LoopVectorize.cpp Wed Jul 27 18:53:58 2016
@@ -4859,7 +4859,7 @@ void LoopVectorizationLegality::collectL
   // out of scope. It ensures a uniform instruction will only be used
   // by uniform instructions or out of scope instructions.
   unsigned idx = 0;
-  do {
+  while (idx != Worklist.size()) {
     Instruction *I = Worklist[idx++];
 
     for (auto OV : I->operand_values()) {
@@ -4873,7 +4873,7 @@ void LoopVectorizationLegality::collectL
         DEBUG(dbgs() << "LV: Found uniform instruction: " << *OI << "\n");
       }
     }
-  } while (idx != Worklist.size());
+  }
 
   // For an instruction to be added into Worklist above, all its users inside
   // the current loop should be already added into Worklist. This condition

Added: llvm/trunk/test/Transforms/LoopVectorize/2016-07-27-loop-vec.ll
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/Transforms/LoopVectorize/2016-07-27-loop-vec.ll?rev=276943&view=auto
==============================================================================
--- llvm/trunk/test/Transforms/LoopVectorize/2016-07-27-loop-vec.ll (added)
+++ llvm/trunk/test/Transforms/LoopVectorize/2016-07-27-loop-vec.ll Wed Jul 27 18:53:58 2016
@@ -0,0 +1,19 @@
+; RUN: opt < %s -loop-vectorize -S
+
+define void @foo() local_unnamed_addr {
+entry:
+  %exitcond = icmp eq i64 3, 3
+  br label %for.body
+
+for.body:                                         ; preds = %entry
+  %i.05 = phi i64 [ %inc, %for.body ], [ 0, %entry ]
+  %total1 = add nsw i64 %i.05, 3
+  %inc = add nuw nsw i64 %i.05, 1
+  br i1 %exitcond, label %for.end, label %for.body, !llvm.loop !0
+
+for.end:                                          ; preds = %for.body
+  ret void
+}
+
+!0 = distinct !{!0, !1}
+!1 = !{!"llvm.loop.vectorize.enable", i1 true}