Implement merging of blocks with the same condition if the block has multiple

predecessors.  This implements branch-phi-thread.ll::test1


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@23395 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Chris Lattner 2005-09-20 00:43:16 +00:00
parent ecd7b6d110
commit 2e42e36698

View File

@ -890,6 +890,26 @@ HoistTerminator:
return true; return true;
} }
/// BlockIsSimpleEnoughToThreadThrough - Return true if we can thread a branch
/// across this block.
static bool BlockIsSimpleEnoughToThreadThrough(BasicBlock *BB) {
BranchInst *BI = cast<BranchInst>(BB->getTerminator());
Value *Cond = BI->getCondition();
// If this basic block contains anything other than a PHI (which controls the
// branch) and branch itself, bail out. FIXME: improve this in the future.
for (BasicBlock::iterator BBI = BB->begin(); &*BBI != BI; ++BBI) {
if (!isa<PHINode>(BBI)) return false;
if (&*BBI != Cond || !Cond->hasOneUse())
return false;
// Looks ok, continue checking.
}
return true;
}
/// FoldCondBranchOnPHI - If we have a conditional branch on a PHI node value /// FoldCondBranchOnPHI - If we have a conditional branch on a PHI node value
/// that is defined in the same block as the branch and if any PHI entries are /// that is defined in the same block as the branch and if any PHI entries are
/// constants, thread edges corresponding to that entry to be branches to their /// constants, thread edges corresponding to that entry to be branches to their
@ -899,7 +919,8 @@ static bool FoldCondBranchOnPHI(BranchInst *BI) {
PHINode *PN = dyn_cast<PHINode>(BI->getCondition()); PHINode *PN = dyn_cast<PHINode>(BI->getCondition());
// NOTE: we currently cannot transform this case if the PHI node is used // NOTE: we currently cannot transform this case if the PHI node is used
// outside of the block. // outside of the block.
if (!PN || PN->getParent() != BB || !PN->hasOneUse()) return false; if (!PN || PN->getParent() != BB || !PN->hasOneUse())
return false;
// Degenerate case of a single entry PHI. // Degenerate case of a single entry PHI.
if (PN->getNumIncomingValues() == 1) { if (PN->getNumIncomingValues() == 1) {
@ -912,12 +933,7 @@ static bool FoldCondBranchOnPHI(BranchInst *BI) {
} }
// Now we know that this block has multiple preds and two succs. // Now we know that this block has multiple preds and two succs.
if (!BlockIsSimpleEnoughToThreadThrough(BB)) return false;
// If this basic block contains anything other than the PHI and branch, bail
// out. FIXME: improve this in the future.
BasicBlock::iterator BBI = BB->begin();
if (&*BBI != PN || &*++BBI != BI)
return false;
// Okay, this is a simple enough basic block. See if any phi values are // Okay, this is a simple enough basic block. See if any phi values are
// constants. // constants.
@ -1272,22 +1288,44 @@ bool llvm::SimplifyCFG(BasicBlock *BB) {
} }
} }
// If this block ends with a branch instruction, and if there is one // If this block ends with a branch instruction, and if there is a
// predecessor, see if the previous block ended with a branch on the same // predecessor that ends on a branch of the same condition, make this
// condition, which makes this conditional branch redundant. // conditional branch redundant.
if (BasicBlock *OnlyPred = BB->getSinglePredecessor()) for (pred_iterator PI = pred_begin(BB), E = pred_end(BB); PI != E; ++PI)
if (BranchInst *PBI = dyn_cast<BranchInst>(OnlyPred->getTerminator())) if (BranchInst *PBI = dyn_cast<BranchInst>((*PI)->getTerminator()))
if (PBI->isConditional() && if (PBI != BI && PBI->isConditional() &&
PBI->getCondition() == BI->getCondition() && PBI->getCondition() == BI->getCondition() &&
(PBI->getSuccessor(0) != BB || PBI->getSuccessor(1) != BB)) { PBI->getSuccessor(0) != PBI->getSuccessor(1)) {
// Okay, the outcome of this conditional branch is statically // Okay, the outcome of this conditional branch is statically
// knowable. Delete the outgoing CFG edge that is impossible to // knowable. If this block had a single pred, handle specially.
// execute. if (BB->getSinglePredecessor()) {
bool CondIsTrue = PBI->getSuccessor(0) == BB; // Turn this into a branch on constant.
BI->getSuccessor(CondIsTrue)->removePredecessor(BB); bool CondIsTrue = PBI->getSuccessor(0) == BB;
new BranchInst(BI->getSuccessor(!CondIsTrue), BB); BI->setCondition(ConstantBool::get(CondIsTrue));
BB->getInstList().erase(BI); return SimplifyCFG(BB); // Nuke the branch on constant.
return SimplifyCFG(BB) | true; }
// Otherwise, if there are multiple predecessors, insert a PHI that
// merges in the constant and simplify the block result.
if (BlockIsSimpleEnoughToThreadThrough(BB)) {
PHINode *NewPN = new PHINode(Type::BoolTy,
BI->getCondition()->getName()+".pr",
BB->begin());
for (PI = pred_begin(BB), E = pred_end(BB); PI != E; ++PI)
if ((PBI = dyn_cast<BranchInst>((*PI)->getTerminator())) &&
PBI != BI && PBI->isConditional() &&
PBI->getCondition() == BI->getCondition() &&
PBI->getSuccessor(0) != PBI->getSuccessor(1)) {
bool CondIsTrue = PBI->getSuccessor(0) == BB;
NewPN->addIncoming(ConstantBool::get(CondIsTrue), *PI);
} else {
NewPN->addIncoming(BI->getCondition(), *PI);
}
BI->setCondition(NewPN);
// This will thread the branch.
return SimplifyCFG(BB) | true;
}
} }
} }
} else if (isa<UnreachableInst>(BB->getTerminator())) { } else if (isa<UnreachableInst>(BB->getTerminator())) {