1 files changed, 5 insertions, 4 deletions
diff --git a/llvm/lib/Transforms/IPO/PartialInlining.cpp b/llvm/lib/Transforms/IPO/PartialInlining.cpp
index 5957e84ffab..76d6dfa8e88 100644
--- a/llvm/lib/Transforms/IPO/PartialInlining.cpp
+++ b/llvm/lib/Transforms/IPO/PartialInlining.cpp
@@ -58,12 +58,13 @@ Function* PartialInliner::unswitchFunction(Function* F) {
   BasicBlock* returnBlock = nullptr;
   BasicBlock* nonReturnBlock = nullptr;
   unsigned returnCount = 0;
-  for (BasicBlock *Succ : successors(entryBlock))
-    if (isa<ReturnInst>(Succ->getTerminator())) {
-      returnBlock = Succ;
+  for (succ_iterator SI = succ_begin(entryBlock), SE = succ_end(entryBlock);
+       SI != SE; ++SI)
+    if (isa<ReturnInst>((*SI)->getTerminator())) {
+      returnBlock = *SI;
       returnCount++;
     } else
-      nonReturnBlock = Succ;
+      nonReturnBlock = *SI;
   
   if (returnCount != 1)
     return nullptr;