use ArgOperand API and CallSite to access arguments of CallInst

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@106833 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Gabor Greif 2010-06-25 09:03:52 +00:00
parent 89c4cead3c
commit 9e38531d95

View File

@ -16,6 +16,7 @@
#include "llvm/Module.h" #include "llvm/Module.h"
#include "llvm/Type.h" #include "llvm/Type.h"
#include "llvm/CodeGen/IntrinsicLowering.h" #include "llvm/CodeGen/IntrinsicLowering.h"
#include "llvm/Support/CallSite.h"
#include "llvm/Support/ErrorHandling.h" #include "llvm/Support/ErrorHandling.h"
#include "llvm/Support/IRBuilder.h" #include "llvm/Support/IRBuilder.h"
#include "llvm/Support/raw_ostream.h" #include "llvm/Support/raw_ostream.h"
@ -314,21 +315,22 @@ static Value *LowerCTLZ(LLVMContext &Context, Value *V, Instruction *IP) {
static void ReplaceFPIntrinsicWithCall(CallInst *CI, const char *Fname, static void ReplaceFPIntrinsicWithCall(CallInst *CI, const char *Fname,
const char *Dname, const char *Dname,
const char *LDname) { const char *LDname) {
switch (CI->getOperand(1)->getType()->getTypeID()) { CallSite CS(CI);
switch (CI->getArgOperand(0)->getType()->getTypeID()) {
default: llvm_unreachable("Invalid type in intrinsic"); default: llvm_unreachable("Invalid type in intrinsic");
case Type::FloatTyID: case Type::FloatTyID:
ReplaceCallWith(Fname, CI, CI->op_begin() + 1, CI->op_end(), ReplaceCallWith(Fname, CI, CS.arg_begin(), CS.arg_end(),
Type::getFloatTy(CI->getContext())); Type::getFloatTy(CI->getContext()));
break; break;
case Type::DoubleTyID: case Type::DoubleTyID:
ReplaceCallWith(Dname, CI, CI->op_begin() + 1, CI->op_end(), ReplaceCallWith(Dname, CI, CS.arg_begin(), CS.arg_end(),
Type::getDoubleTy(CI->getContext())); Type::getDoubleTy(CI->getContext()));
break; break;
case Type::X86_FP80TyID: case Type::X86_FP80TyID:
case Type::FP128TyID: case Type::FP128TyID:
case Type::PPC_FP128TyID: case Type::PPC_FP128TyID:
ReplaceCallWith(LDname, CI, CI->op_begin() + 1, CI->op_end(), ReplaceCallWith(LDname, CI, CS.arg_begin(), CS.arg_end(),
CI->getOperand(1)->getType()); CI->getArgOperand(0)->getType());
break; break;
} }
} }
@ -340,6 +342,7 @@ void IntrinsicLowering::LowerIntrinsicCall(CallInst *CI) {
const Function *Callee = CI->getCalledFunction(); const Function *Callee = CI->getCalledFunction();
assert(Callee && "Cannot lower an indirect call!"); assert(Callee && "Cannot lower an indirect call!");
CallSite CS(CI);
switch (Callee->getIntrinsicID()) { switch (Callee->getIntrinsicID()) {
case Intrinsic::not_intrinsic: case Intrinsic::not_intrinsic:
report_fatal_error("Cannot lower a call to a non-intrinsic function '"+ report_fatal_error("Cannot lower a call to a non-intrinsic function '"+
@ -353,7 +356,7 @@ void IntrinsicLowering::LowerIntrinsicCall(CallInst *CI) {
// by the lowerinvoke pass. In both cases, the right thing to do is to // by the lowerinvoke pass. In both cases, the right thing to do is to
// convert the call to an explicit setjmp or longjmp call. // convert the call to an explicit setjmp or longjmp call.
case Intrinsic::setjmp: { case Intrinsic::setjmp: {
Value *V = ReplaceCallWith("setjmp", CI, CI->op_begin() + 1, CI->op_end(), Value *V = ReplaceCallWith("setjmp", CI, CS.arg_begin(), CS.arg_end(),
Type::getInt32Ty(Context)); Type::getInt32Ty(Context));
if (!CI->getType()->isVoidTy()) if (!CI->getType()->isVoidTy())
CI->replaceAllUsesWith(V); CI->replaceAllUsesWith(V);
@ -365,7 +368,7 @@ void IntrinsicLowering::LowerIntrinsicCall(CallInst *CI) {
break; break;
case Intrinsic::longjmp: { case Intrinsic::longjmp: {
ReplaceCallWith("longjmp", CI, CI->op_begin() + 1, CI->op_end(), ReplaceCallWith("longjmp", CI, CS.arg_begin(), CS.arg_end(),
Type::getVoidTy(Context)); Type::getVoidTy(Context));
break; break;
} }
@ -377,20 +380,20 @@ void IntrinsicLowering::LowerIntrinsicCall(CallInst *CI) {
break; break;
} }
case Intrinsic::ctpop: case Intrinsic::ctpop:
CI->replaceAllUsesWith(LowerCTPOP(Context, CI->getOperand(1), CI)); CI->replaceAllUsesWith(LowerCTPOP(Context, CI->getArgOperand(0), CI));
break; break;
case Intrinsic::bswap: case Intrinsic::bswap:
CI->replaceAllUsesWith(LowerBSWAP(Context, CI->getOperand(1), CI)); CI->replaceAllUsesWith(LowerBSWAP(Context, CI->getArgOperand(0), CI));
break; break;
case Intrinsic::ctlz: case Intrinsic::ctlz:
CI->replaceAllUsesWith(LowerCTLZ(Context, CI->getOperand(1), CI)); CI->replaceAllUsesWith(LowerCTLZ(Context, CI->getArgOperand(0), CI));
break; break;
case Intrinsic::cttz: { case Intrinsic::cttz: {
// cttz(x) -> ctpop(~X & (X-1)) // cttz(x) -> ctpop(~X & (X-1))
Value *Src = CI->getOperand(1); Value *Src = CI->getArgOperand(0);
Value *NotSrc = Builder.CreateNot(Src); Value *NotSrc = Builder.CreateNot(Src);
NotSrc->setName(Src->getName() + ".not"); NotSrc->setName(Src->getName() + ".not");
Value *SrcM1 = ConstantInt::get(Src->getType(), 1); Value *SrcM1 = ConstantInt::get(Src->getType(), 1);
@ -451,37 +454,37 @@ void IntrinsicLowering::LowerIntrinsicCall(CallInst *CI) {
case Intrinsic::memcpy: { case Intrinsic::memcpy: {
const IntegerType *IntPtr = TD.getIntPtrType(Context); const IntegerType *IntPtr = TD.getIntPtrType(Context);
Value *Size = Builder.CreateIntCast(CI->getOperand(3), IntPtr, Value *Size = Builder.CreateIntCast(CI->getArgOperand(2), IntPtr,
/* isSigned */ false); /* isSigned */ false);
Value *Ops[3]; Value *Ops[3];
Ops[0] = CI->getOperand(1); Ops[0] = CI->getArgOperand(0);
Ops[1] = CI->getOperand(2); Ops[1] = CI->getArgOperand(1);
Ops[2] = Size; Ops[2] = Size;
ReplaceCallWith("memcpy", CI, Ops, Ops+3, CI->getOperand(1)->getType()); ReplaceCallWith("memcpy", CI, Ops, Ops+3, CI->getArgOperand(0)->getType());
break; break;
} }
case Intrinsic::memmove: { case Intrinsic::memmove: {
const IntegerType *IntPtr = TD.getIntPtrType(Context); const IntegerType *IntPtr = TD.getIntPtrType(Context);
Value *Size = Builder.CreateIntCast(CI->getOperand(3), IntPtr, Value *Size = Builder.CreateIntCast(CI->getArgOperand(2), IntPtr,
/* isSigned */ false); /* isSigned */ false);
Value *Ops[3]; Value *Ops[3];
Ops[0] = CI->getOperand(1); Ops[0] = CI->getArgOperand(0);
Ops[1] = CI->getOperand(2); Ops[1] = CI->getArgOperand(1);
Ops[2] = Size; Ops[2] = Size;
ReplaceCallWith("memmove", CI, Ops, Ops+3, CI->getOperand(1)->getType()); ReplaceCallWith("memmove", CI, Ops, Ops+3, CI->getArgOperand(0)->getType());
break; break;
} }
case Intrinsic::memset: { case Intrinsic::memset: {
const IntegerType *IntPtr = TD.getIntPtrType(Context); const IntegerType *IntPtr = TD.getIntPtrType(Context);
Value *Size = Builder.CreateIntCast(CI->getOperand(3), IntPtr, Value *Size = Builder.CreateIntCast(CI->getArgOperand(2), IntPtr,
/* isSigned */ false); /* isSigned */ false);
Value *Ops[3]; Value *Ops[3];
Ops[0] = CI->getOperand(1); Ops[0] = CI->getArgOperand(0);
// Extend the amount to i32. // Extend the amount to i32.
Ops[1] = Builder.CreateIntCast(CI->getOperand(2), Type::getInt32Ty(Context), Ops[1] = Builder.CreateIntCast(CI->getArgOperand(1), Type::getInt32Ty(Context),
/* isSigned */ false); /* isSigned */ false);
Ops[2] = Size; Ops[2] = Size;
ReplaceCallWith("memset", CI, Ops, Ops+3, CI->getOperand(1)->getType()); ReplaceCallWith("memset", CI, Ops, Ops+3, CI->getArgOperand(0)->getType());
break; break;
} }
case Intrinsic::sqrt: { case Intrinsic::sqrt: {