Swap exit condition operands if it works.


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@41817 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/lib/Transforms/Scalar/LoopIndexSplit.cpp b/lib/Transforms/Scalar/LoopIndexSplit.cpp
index 3a04665..0f55a91 100644
--- a/lib/Transforms/Scalar/LoopIndexSplit.cpp
+++ b/lib/Transforms/Scalar/LoopIndexSplit.cpp
@@ -328,14 +328,24 @@
   if (!CI)
     return;
 
-  // FIXME 
+  // FIXME
+  if (CI->getPredicate() == ICmpInst::ICMP_EQ
+      || CI->getPredicate() == ICmpInst::ICMP_NE)
+    return;
+
   if (CI->getPredicate() == ICmpInst::ICMP_SGT
       || CI->getPredicate() == ICmpInst::ICMP_UGT
       || CI->getPredicate() == ICmpInst::ICMP_SGE
-      || CI->getPredicate() == ICmpInst::ICMP_UGE
-      || CI->getPredicate() == ICmpInst::ICMP_EQ
-      || CI->getPredicate() == ICmpInst::ICMP_NE)
-    return;
+      || CI->getPredicate() == ICmpInst::ICMP_UGE) {
+
+    BasicBlock *FirstSuccessor = BR->getSuccessor(0);
+    // splitLoop() is expecting LT/LE as exit condition predicate.
+    // Swap operands here if possible to meet this requirement.
+    if (!L->contains(FirstSuccessor)) 
+      CI->swapOperands();
+    else
+      return;
+  }
 
   ExitCondition = CI;