summaryrefslogtreecommitdiff
path: root/lib/Transforms/IPO
diff options
context:
space:
mode:
authorOwen Anderson <resistor@mac.com>2008-09-05 23:36:01 +0000
committerOwen Anderson <resistor@mac.com>2008-09-05 23:36:01 +0000
commit6e6074508c2f781c3e52dfe0e301cb1c7f395a91 (patch)
tree78acc35e61261c50375bff374e13984acdcdce54 /lib/Transforms/IPO
parentd0c0fae63bbbdb2d26e62736f0654bfdc5193ccc (diff)
downloadllvm-6e6074508c2f781c3e52dfe0e301cb1c7f395a91.tar.gz
llvm-6e6074508c2f781c3e52dfe0e301cb1c7f395a91.tar.bz2
llvm-6e6074508c2f781c3e52dfe0e301cb1c7f395a91.tar.xz
Revert r55859. This is breaking the build in the abscence of its companion commit.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@55865 91177308-0d34-0410-b5e6-96231b3b80d8
Diffstat (limited to 'lib/Transforms/IPO')
-rw-r--r--lib/Transforms/IPO/PruneEH.cpp11
1 files changed, 7 insertions, 4 deletions
diff --git a/lib/Transforms/IPO/PruneEH.cpp b/lib/Transforms/IPO/PruneEH.cpp
index 94d6331ddd..8fb904b1ea 100644
--- a/lib/Transforms/IPO/PruneEH.cpp
+++ b/lib/Transforms/IPO/PruneEH.cpp
@@ -222,10 +222,13 @@ void PruneEH::DeleteBasicBlock(BasicBlock *BB) {
CallGraphNode *CGN = CG[BB->getParent()];
for (BasicBlock::iterator I = BB->end(), E = BB->begin(); I != E; ) {
--I;
- if (CallInst *CI = dyn_cast<CallInst>(I))
- CGN->removeCallEdgeFor(CI);
- else if (InvokeInst *II = dyn_cast<InvokeInst>(I))
- CGN->removeCallEdgeFor(II);
+ if (CallInst *CI = dyn_cast<CallInst>(I)) {
+ if (Function *Callee = CI->getCalledFunction())
+ CGN->removeCallEdgeTo(CG[Callee]);
+ } else if (InvokeInst *II = dyn_cast<InvokeInst>(I)) {
+ if (Function *Callee = II->getCalledFunction())
+ CGN->removeCallEdgeTo(CG[Callee]);
+ }
if (!I->use_empty())
I->replaceAllUsesWith(UndefValue::get(I->getType()));
}