diff options
Diffstat (limited to 'lib/Transforms/Utils/UnifyFunctionExitNodes.cpp')
-rw-r--r-- | lib/Transforms/Utils/UnifyFunctionExitNodes.cpp | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/lib/Transforms/Utils/UnifyFunctionExitNodes.cpp b/lib/Transforms/Utils/UnifyFunctionExitNodes.cpp index 15e37e3993..da7b1f609a 100644 --- a/lib/Transforms/Utils/UnifyFunctionExitNodes.cpp +++ b/lib/Transforms/Utils/UnifyFunctionExitNodes.cpp @@ -23,16 +23,15 @@ AnalysisID UnifyMethodExitNodes::ID(AnalysisID::create<UnifyMethodExitNodes>()); // If there are no return stmts in the Method, a null pointer is returned. // bool UnifyMethodExitNodes::doit(Method *M, BasicBlock *&ExitNode) { - vector<BasicBlock*> ReturningBlocks; - // Loop over all of the blocks in a method, tracking all of the blocks that // return. // + vector<BasicBlock*> ReturningBlocks; for(Method::iterator I = M->begin(), E = M->end(); I != E; ++I) - if ((*I)->getTerminator()->getOpcode() == Instruction::Ret) + if (isa<ReturnInst>((*I)->getTerminator())) ReturningBlocks.push_back(*I); - if (ReturningBlocks.size() == 0) { + if (ReturningBlocks.empty()) { ExitNode = 0; return false; // No blocks return } else if (ReturningBlocks.size() == 1) { |