I will save before I commit.

I will save before I commit.
I will save before I commit.
I will save before I commit.

llvm-svn: 136007

2 files changed, 3 insertions, 1 deletions

diff --git a/llvm/tools/llvm-objdump/MCFunction.h b/llvm/tools/llvm-objdump/MCFunction.h
index 60f64292092..f156e940118 100644
--- a/llvm/tools/llvm-objdump/MCFunction.h
+++ b/llvm/tools/llvm-objdump/MCFunction.h
@@ -48,6 +48,8 @@ public:
   succ_iterator succ_begin() const { return Succs.begin(); }
   succ_iterator succ_end() const { return Succs.end(); }
 
+  bool contains(MCBasicBlock *BB) const { return Succs.count(BB); }
+
   void addInst(const MCDecodedInst &Inst) { Insts.push_back(Inst); }
   void addSucc(MCBasicBlock *BB) { Succs.insert(BB); }
 };
diff --git a/llvm/tools/llvm-objdump/llvm-objdump.cpp b/llvm/tools/llvm-objdump/llvm-objdump.cpp
index 99549266c24..b1c74d298b3 100644
--- a/llvm/tools/llvm-objdump/llvm-objdump.cpp
+++ b/llvm/tools/llvm-objdump/llvm-objdump.cpp
@@ -286,7 +286,7 @@ static void DisassembleInput(const StringRef &Filename) {
           // FIXME: Slow.
           for (MCFunction::iterator pi = f.begin(), pe = f.end(); pi != pe;
                ++pi)
-            for (pi->second->contains(&i->second)) {
+            if (pi->second.contains(&i->second)) {
               hasPreds = true;
               break;
             }