Minor tweak

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@3985 91177308-0d34-0410-b5e6-96231b3b80d8

1 files changed, 3 insertions, 2 deletions

diff --git a/lib/Analysis/LoopInfo.cpp b/lib/Analysis/LoopInfo.cpp
index 1c3d5ee565..1f8e34ca7e 100644
--- a/lib/Analysis/LoopInfo.cpp
+++ b/lib/Analysis/LoopInfo.cpp
@@ -147,8 +147,9 @@ BasicBlock *Loop::getLoopPreheader() const {
   BasicBlock *Header = getHeader();
   for (pred_iterator PI = pred_begin(Header), PE = pred_end(Header);
        PI != PE; ++PI)
-    if (!contains(*PI)) { // If the block is not in the loop...
-      if (Out) return 0;  // Multiple predecessors outside the loop
+    if (!contains(*PI)) {     // If the block is not in the loop...
+      if (Out && Out != *PI)
+        return 0;             // Multiple predecessors outside the loop
       Out = *PI;
     }