Tweak per Chris' comments.

llvm-svn: 108736

1 files changed, 8 insertions, 7 deletions

diff --git a/llvm/lib/Transforms/InstCombine/InstCombineCasts.cpp b/llvm/lib/Transforms/InstCombine/InstCombineCasts.cpp
index 042bf1a9829..dbe5200d000 100644
--- a/llvm/lib/Transforms/InstCombine/InstCombineCasts.cpp
+++ b/llvm/lib/Transforms/InstCombine/InstCombineCasts.cpp
@@ -1106,19 +1106,20 @@ Instruction *InstCombiner::visitFPTrunc(FPTruncInst &CI) {
       Call->getNumArgOperands() == 1) {
     CastInst *Arg = dyn_cast<CastInst>(Call->getArgOperand(0));
     if (Arg && Arg->getOpcode() == Instruction::FPExt &&
-        CI.getType() == Builder->getFloatTy() &&
-        Call->getType() == Builder->getDoubleTy() &&
-        Arg->getType() == Builder->getDoubleTy() &&
-        Arg->getOperand(0)->getType() == Builder->getFloatTy()) {
-      Module* M = CI.getParent()->getParent()->getParent();
+        CI.getType()->isFloatTy() &&
+        Call->getType()->isDoubleTy() &&
+        Arg->getType()->isDoubleTy() &&
+        Arg->getOperand(0)->getType()->isFloatTy()) {
+      Function *Callee = Call->getCalledFunction();
+      Module *M = CI.getParent()->getParent()->getParent();
       Constant* SqrtfFunc = M->getOrInsertFunction("sqrtf", 
-                                                   Call->getAttributes(),
+                                                   Callee->getAttributes(),
                                                    Builder->getFloatTy(),
                                                    Builder->getFloatTy(),
                                                    NULL);
       CallInst *ret = CallInst::Create(SqrtfFunc, Arg->getOperand(0),
                                        "sqrtfcall");
-      ret->setAttributes(Call->getAttributes());
+      ret->setAttributes(Callee->getAttributes());
       return ret;
     }
   }