diff options
| -rw-r--r-- | lib/Transforms/Scalar/TailDuplication.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/lib/Transforms/Scalar/TailDuplication.cpp b/lib/Transforms/Scalar/TailDuplication.cpp index 7d5aac4fe7..315b90697c 100644 --- a/lib/Transforms/Scalar/TailDuplication.cpp +++ b/lib/Transforms/Scalar/TailDuplication.cpp @@ -125,6 +125,7 @@ bool TailDup::shouldEliminateUnconditionalBranch(TerminatorInst *TI) { bool TailDup::canEliminateUnconditionalBranch(TerminatorInst *TI) { // Basically, we refuse to make the transformation if any of the values // computed in the 'tail' are used in any other basic blocks. + BasicBlock *BB = TI->getParent(); BasicBlock *Tail = TI->getSuccessor(0); assert(isa<BranchInst>(TI) && cast<BranchInst>(TI)->isUnconditional()); @@ -132,7 +133,7 @@ bool TailDup::canEliminateUnconditionalBranch(TerminatorInst *TI) { for (Value::use_iterator UI = I->use_begin(), E = I->use_end(); UI != E; ++UI) { Instruction *User = cast<Instruction>(*UI); - if (User->getParent() != Tail || isa<PHINode>(User)) + if (User->getParent() != Tail && User->getParent() != BB) return false; } return true; |