Fix lowering to work correctly


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@14134 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/lib/CodeGen/IntrinsicLowering.cpp b/lib/CodeGen/IntrinsicLowering.cpp
index bc6f02e..b6ebe09 100644
--- a/lib/CodeGen/IntrinsicLowering.cpp
+++ b/lib/CodeGen/IntrinsicLowering.cpp
@@ -69,7 +69,14 @@
 
   std::string Name = CI->getName(); CI->setName("");
   if (FT->getReturnType() == Type::VoidTy) Name.clear();
-  return new CallInst(FCache, Operands, Name, CI);
+  CallInst *NewCI = new CallInst(FCache, Operands, Name, CI);
+  if (!CI->use_empty()) {
+    Value *V = NewCI;
+    if (CI->getType() != NewCI->getType())
+      V = new CastInst(NewCI, CI->getType(), Name, CI);
+    CI->replaceAllUsesWith(V);
+  }
+  return NewCI;
 }
 
 void DefaultIntrinsicLowering::AddPrototypes(Module &M) {
@@ -98,7 +105,9 @@
         EnsureFunctionExists(M, "memset", I->abegin(), --I->aend(),
                              I->abegin()->getType());
         break;
-
+      case Intrinsic::isnan:
+        EnsureFunctionExists(M, "isnan", I->abegin(), I->aend(), Type::BoolTy);
+        break;
       }
 
 }
@@ -196,7 +205,7 @@
     // multiple isnans for different FP arguments.
     static Function *isnanFCache = 0;
     ReplaceCallWith("isnan", CI, CI->op_begin()+1, CI->op_end(),
-                    (*(CI->op_begin()+1))->getType(), isnanFCache);
+                    Type::BoolTy, isnanFCache);
     break;
   }
   }
diff --git a/lib/VMCore/IntrinsicLowering.cpp b/lib/VMCore/IntrinsicLowering.cpp
index bc6f02e..b6ebe09 100644
--- a/lib/VMCore/IntrinsicLowering.cpp
+++ b/lib/VMCore/IntrinsicLowering.cpp
@@ -69,7 +69,14 @@
 
   std::string Name = CI->getName(); CI->setName("");
   if (FT->getReturnType() == Type::VoidTy) Name.clear();
-  return new CallInst(FCache, Operands, Name, CI);
+  CallInst *NewCI = new CallInst(FCache, Operands, Name, CI);
+  if (!CI->use_empty()) {
+    Value *V = NewCI;
+    if (CI->getType() != NewCI->getType())
+      V = new CastInst(NewCI, CI->getType(), Name, CI);
+    CI->replaceAllUsesWith(V);
+  }
+  return NewCI;
 }
 
 void DefaultIntrinsicLowering::AddPrototypes(Module &M) {
@@ -98,7 +105,9 @@
         EnsureFunctionExists(M, "memset", I->abegin(), --I->aend(),
                              I->abegin()->getType());
         break;
-
+      case Intrinsic::isnan:
+        EnsureFunctionExists(M, "isnan", I->abegin(), I->aend(), Type::BoolTy);
+        break;
       }
 
 }
@@ -196,7 +205,7 @@
     // multiple isnans for different FP arguments.
     static Function *isnanFCache = 0;
     ReplaceCallWith("isnan", CI, CI->op_begin()+1, CI->op_end(),
-                    (*(CI->op_begin()+1))->getType(), isnanFCache);
+                    Type::BoolTy, isnanFCache);
     break;
   }
   }