From 6829350871eebb890edb10c090286e035d388ab9 Mon Sep 17 00:00:00 2001
From: Nuno Lopes <nunoplopes@sapo.pt>
Date: Fri, 22 Jun 2012 15:50:53 +0000
Subject: [PATCH] simplify code from previous commits (Thanks Duncan)

llvm-svn: 158999
---
 lib/Analysis/MemoryBuiltins.cpp | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/lib/Analysis/MemoryBuiltins.cpp b/lib/Analysis/MemoryBuiltins.cpp
index d1a9363b411..a9f12445350 100644
--- a/lib/Analysis/MemoryBuiltins.cpp
+++ b/lib/Analysis/MemoryBuiltins.cpp
@@ -66,13 +66,8 @@ static Function *getCalledFunction(const Value *V, bool LookThroughBitCast) {
   if (LookThroughBitCast)
     V = V->stripPointerCasts();
 
-  Value *I = const_cast<Value*>(V);
-  CallSite CS;
-  if (CallInst *CI = dyn_cast<CallInst>(I))
-    CS = CallSite(CI);
-  else if (InvokeInst *II = dyn_cast<InvokeInst>(I))
-    CS = CallSite(II);
-  else
+  CallSite CS(const_cast<Value*>(V));
+  if (!CS.getInstruction())
     return 0;
 
   Function *Callee = CS.getCalledFunction();