X-Git-Url: http://plrg.eecs.uci.edu/git/?p=oota-llvm.git;a=blobdiff_plain;f=lib%2FIR%2FFunction.cpp;h=bb6bef7bd026ab2a4f276deb9f3256d7752fc128;hp=970bbaeed8b524c46e852840f7cb137eec26f7af;hb=dfa550a1761a85417d0e42c8cd17cd08e753388b;hpb=4b70bfc905f3ac68a8429f9fe0016e30433b3b0c

diff --git a/lib/IR/Function.cpp b/lib/IR/Function.cpp
index 970bbaeed8b..bb6bef7bd02 100644
--- a/lib/IR/Function.cpp
+++ b/lib/IR/Function.cpp
@@ -736,10 +736,8 @@ bool Function::isDefTriviallyDead() const {
 bool Function::callsFunctionThatReturnsTwice() const {
   for (const_inst_iterator
          I = inst_begin(this), E = inst_end(this); I != E; ++I) {
-    const CallInst* callInst = dyn_cast<CallInst>(&*I);
-    if (!callInst)
-      continue;
-    if (callInst->canReturnTwice())
+    ImmutableCallSite CS(&*I);
+    if (CS && CS.hasFnAttr(Attribute::ReturnsTwice))
       return true;
   }