6502/llvm-6502
1
0
Fork 0
mirror of https://github.com/c64scene-ar/llvm-6502.git synced 2025-01-05 12:31:33 +00:00
Code Issues Projects Releases Wiki Activity

Preserve paramater attributes.

Browse Source 
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@47847 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Devang Patel 2008-03-03 18:36:03 +00:00
parent 8f3153e217
commit 2a4821b2b3
1 changed files with 31 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
lib/Transforms/IPO/StructRetPromotion.cpp
View File

@ -17,6 +17,7 @@
#include "llvm/Module.h" #include "llvm/Module.h"
#include "llvm/CallGraphSCCPass.h" #include "llvm/CallGraphSCCPass.h"
#include "llvm/Instructions.h" #include "llvm/Instructions.h"
#include "llvm/ParamAttrsList.h"
#include "llvm/Analysis/CallGraph.h" #include "llvm/Analysis/CallGraph.h"
#include "llvm/Support/CallSite.h" #include "llvm/Support/CallSite.h"
#include "llvm/Support/CFG.h" #include "llvm/Support/CFG.h"
@ -181,21 +182,34 @@ bool SRETPromotion::isSafeToUpdateAllCallers(Function *F) {
Function *SRETPromotion::cloneFunctionBody(Function *F, Function *SRETPromotion::cloneFunctionBody(Function *F,
const StructType *STy) { const StructType *STy) {
// FIXME : Do not drop param attributes on the floor.
const FunctionType *FTy = F->getFunctionType(); const FunctionType *FTy = F->getFunctionType();
std::vector<const Type*> Params; std::vector<const Type*> Params;
// ParamAttrs - Keep track of the parameter attributes for the arguments.
ParamAttrsVector ParamAttrsVec;
const ParamAttrsList *PAL = F->getParamAttrs();
// Add any return attributes.
if (ParameterAttributes attrs = PAL ? PAL->getParamAttrs(0) : ParamAttr::None)
ParamAttrsVec.push_back(ParamAttrsWithIndex::get(0, attrs));
// Skip first argument. // Skip first argument.
Function::arg_iterator I = F->arg_begin(), E = F->arg_end(); Function::arg_iterator I = F->arg_begin(), E = F->arg_end();
++I; ++I;
unsigned ParamIndex = 1; // 0th parameter attribute is reserved for return type.
while (I != E) { while (I != E) {
Params.push_back(I->getType()); Params.push_back(I->getType());
if (ParameterAttributes attrs = PAL ? PAL->getParamAttrs(ParamIndex) :
ParamAttr::None)
ParamAttrsVec.push_back(ParamAttrsWithIndex::get(Params.size(), attrs));
++I; ++I;
++ParamIndex;
} }
FunctionType *NFTy = FunctionType::get(STy, Params, FTy->isVarArg()); FunctionType *NFTy = FunctionType::get(STy, Params, FTy->isVarArg());
Function *NF = new Function(NFTy, F->getLinkage(), F->getName()); Function *NF = new Function(NFTy, F->getLinkage(), F->getName());
NF->setCallingConv(F->getCallingConv()); NF->setCallingConv(F->getCallingConv());
NF->setParamAttrs(ParamAttrsList::get(ParamAttrsVec));
F->getParent()->getFunctionList().insert(F, NF); F->getParent()->getFunctionList().insert(F, NF);
NF->getBasicBlockList().splice(NF->begin(), F->getBasicBlockList()); NF->getBasicBlockList().splice(NF->begin(), F->getBasicBlockList());
@ -217,20 +231,32 @@ Function *SRETPromotion::cloneFunctionBody(Function *F,
/// updateCallSites - Update all sites that call F to use NF. /// updateCallSites - Update all sites that call F to use NF.
void SRETPromotion::updateCallSites(Function *F, Function *NF) { void SRETPromotion::updateCallSites(Function *F, Function *NF) {
// FIXME : Handle parameter attributes
SmallVector<Value*, 16> Args; SmallVector<Value*, 16> Args;
// ParamAttrs - Keep track of the parameter attributes for the arguments.
ParamAttrsVector ParamAttrsVec;
for (Value::use_iterator FUI = F->use_begin(), FUE = F->use_end(); FUI != FUE;) { for (Value::use_iterator FUI = F->use_begin(), FUE = F->use_end(); FUI != FUE;) {
CallSite CS = CallSite::get(*FUI); CallSite CS = CallSite::get(*FUI);
++FUI; ++FUI;
Instruction *Call = CS.getInstruction(); Instruction *Call = CS.getInstruction();
const ParamAttrsList *PAL = F->getParamAttrs();
// Add any return attributes.
if (ParameterAttributes attrs = PAL ? PAL->getParamAttrs(0) : ParamAttr::None)
ParamAttrsVec.push_back(ParamAttrsWithIndex::get(0, attrs));
// Copy arguments, however skip first one. // Copy arguments, however skip first one.
CallSite::arg_iterator AI = CS.arg_begin(), AE = CS.arg_end(); CallSite::arg_iterator AI = CS.arg_begin(), AE = CS.arg_end();
Value *FirstCArg = *AI; Value *FirstCArg = *AI;
++AI; ++AI;
unsigned ParamIndex = 1; // 0th parameter attribute is reserved for return type.
while (AI != AE) { while (AI != AE) {
Args.push_back(*AI); Args.push_back(*AI);
if (ParameterAttributes Attrs = PAL ? PAL->getParamAttrs(ParamIndex) :
ParamAttr::None)
ParamAttrsVec.push_back(ParamAttrsWithIndex::get(Args.size(), Attrs));
++ParamIndex;
++AI; ++AI;
} }
@ -240,13 +266,16 @@ void SRETPromotion::updateCallSites(Function *F, Function *NF) {
New = new InvokeInst(NF, II->getNormalDest(), II->getUnwindDest(), New = new InvokeInst(NF, II->getNormalDest(), II->getUnwindDest(),
Args.begin(), Args.end(), "", Call); Args.begin(), Args.end(), "", Call);
cast<InvokeInst>(New)->setCallingConv(CS.getCallingConv()); cast<InvokeInst>(New)->setCallingConv(CS.getCallingConv());
cast<InvokeInst>(New)->setParamAttrs(ParamAttrsList::get(ParamAttrsVec));
} else { } else {
New = new CallInst(NF, Args.begin(), Args.end(), "", Call); New = new CallInst(NF, Args.begin(), Args.end(), "", Call);
cast<CallInst>(New)->setCallingConv(CS.getCallingConv()); cast<CallInst>(New)->setCallingConv(CS.getCallingConv());
cast<CallInst>(New)->setParamAttrs(ParamAttrsList::get(ParamAttrsVec));
if (cast<CallInst>(Call)->isTailCall()) if (cast<CallInst>(Call)->isTailCall())
cast<CallInst>(New)->setTailCall(); cast<CallInst>(New)->setTailCall();
} }
Args.clear(); Args.clear();
ParamAttrsVec.clear();
New->takeName(Call); New->takeName(Call);
// Update all users of sret parameter to extract value using getresult. // Update all users of sret parameter to extract value using getresult.