[llvm-commits] [poolalloc] r130261 - /poolalloc/trunk/lib/AssistDS/LoadArgs.cpp
Arushi Aggarwal
aggarwa4 at illinois.edu
Tue Apr 26 17:11:17 PDT 2011
Author: aggarwa4
Date: Tue Apr 26 19:11:17 2011
New Revision: 130261
URL: http://llvm.org/viewvc/llvm-project?rev=130261&view=rev
Log:
Restructure it to be similiar to GEPExprArgs.
Iterate over CallInsts, instead of LoadInst.
Modified:
poolalloc/trunk/lib/AssistDS/LoadArgs.cpp
Modified: poolalloc/trunk/lib/AssistDS/LoadArgs.cpp
URL: http://llvm.org/viewvc/llvm-project/poolalloc/trunk/lib/AssistDS/LoadArgs.cpp?rev=130261&r1=130260&r2=130261&view=diff
==============================================================================
--- poolalloc/trunk/lib/AssistDS/LoadArgs.cpp (original)
+++ poolalloc/trunk/lib/AssistDS/LoadArgs.cpp Tue Apr 26 19:11:17 2011
@@ -9,7 +9,7 @@
//
//
//===----------------------------------------------------------------------===//
-#define DEBUG_TYPE "ldargs"
+#define DEBUG_TYPE "ld-args"
#include "llvm/Instructions.h"
#include "llvm/Module.h"
@@ -26,7 +26,8 @@
#include <map>
using namespace llvm;
-STATISTIC(numSimplified, "Number of Calls Changed");
+STATISTIC(numSimplified, "Number of Calls Modified");
+
namespace {
class LoadArgs : public ModulePass {
@@ -35,123 +36,121 @@
LoadArgs() : ModulePass(&ID) {}
bool runOnModule(Module& M) {
for (Module::iterator F = M.begin(); F != M.end(); ++F){
- for (Function::iterator B = F->begin(), FE = F->end(); B != FE; ++B) {
- for (BasicBlock::iterator I = B->begin(), BE = B->end(); I != BE; I++) {
- if(!(isa<LoadInst>(I)))
- continue;
- // we are only interested in GEPs
- LoadInst *LI = cast<LoadInst>(I);
- // If the GEP is not doing structure indexing, dont care.
- for (Value::use_iterator UI = LI->use_begin(),UE = LI->use_end(); UI != UE; ) {
- // check if GEP is used in a Call Inst
- CallInst *CI = dyn_cast<CallInst>(UI++);
- if(!CI)
- continue;
+ for (Function::iterator B = F->begin(), FE = F->end(); B != FE; ++B) {
+ for (BasicBlock::iterator I = B->begin(), BE = B->end(); I != BE;) {
+ CallInst *CI = dyn_cast<CallInst>(I++);
+ if(!CI)
+ continue;
- if(CI->hasByValArgument())
- continue;
+ if(CI->hasByValArgument())
+ continue;
+ // if the GEP calls a function, that is externally defined,
+ // or might be changed, ignore this call site.
+ Function *F = CI->getCalledFunction();
- // It should be in the same basic block
- if(LI->getParent() != CI->getParent())
- continue;
+ if (!F || (F->isDeclaration() || F->mayBeOverridden()))
+ continue;
+ if(F->hasStructRetAttr())
+ continue;
+ if(F->isVarArg())
+ continue;
- // Also check that there is no store after the load.
- // TODO: Check if the load/store do not alias.
- BasicBlock::iterator bii = LI->getParent()->begin();
- Instruction *BII = bii;
- while(BII != LI) {
- ++bii;
- BII = bii;
- }
- while(BII != CI) {
- if(isa<StoreInst>(BII))
- break;
- ++bii;
- BII = bii;
- }
- if(isa<StoreInst>(bii)){
+ // find the argument we must replace
+ Function::arg_iterator ai = F->arg_begin(), ae = F->arg_end();
+ unsigned argNum = 1;
+ for(; argNum < CI->getNumOperands();argNum++, ++ai) {
+ if(ai->use_empty())
continue;
- }
+ if(F->paramHasAttr(argNum, Attribute::SExt) ||
+ F->paramHasAttr(argNum, Attribute::ZExt))
+ continue;
+ if (isa<LoadInst>(CI->getOperand(argNum)))
+ break;
+ }
+ // if no argument was a GEP operator to be changed
+ if(ai == ae)
+ continue;
- // if the GEP calls a function, that is externally defined,
- // or might be changed, ignore this call site.
- Function *F = CI->getCalledFunction();
+ LoadInst *LI = dyn_cast<LoadInst>(CI->getOperand(argNum));
+ if(LI->getParent() != CI->getParent())
+ continue;
+ // Also check that there is no store after the load.
+ // TODO: Check if the load/store do not alias.
+ BasicBlock::iterator bii = LI->getParent()->begin();
+ Instruction *BII = bii;
+ while(BII != LI) {
+ ++bii;
+ BII = bii;
+ }
+ while(BII != CI) {
+ if(isa<StoreInst>(BII))
+ break;
+ ++bii;
+ BII = bii;
+ }
+ if(isa<StoreInst>(bii)){
+ continue;
+ }
- if (!F || (F->isDeclaration() || F->mayBeOverridden()))
- continue;
- if(F->hasStructRetAttr())
- continue;
- if(F->isVarArg())
- continue;
+ // Construct the new Type
+ // Appends the struct Type at the beginning
+ std::vector<const Type*>TP;
+ TP.push_back(LI->getOperand(0)->getType());
+ for(unsigned c = 1; c < CI->getNumOperands();c++) {
+ TP.push_back(CI->getOperand(c)->getType());
+ }
- // find the argument we must replace
- unsigned argNum = 1;
- for(; argNum < CI->getNumOperands();argNum++)
- if(LI == CI->getOperand(argNum))
- break;
+ //return type is same as that of original instruction
+ const FunctionType *NewFTy = FunctionType::get(CI->getType(), TP, false);
+ Function *NewF;
+ numSimplified++;
+ if(numSimplified > 500) //26
+ return true;
+
+ NewF = Function::Create(NewFTy,
+ GlobalValue::InternalLinkage,
+ F->getNameStr() + ".TEST",
+ &M);
+
+ Function::arg_iterator NI = NewF->arg_begin();
+ NI->setName("Sarg");
+ ++NI;
+
+ DenseMap<const Value*, Value*> ValueMap;
+
+ for (Function::arg_iterator II = F->arg_begin(); NI != NewF->arg_end(); ++II, ++NI) {
+ ValueMap[II] = NI;
+ NI->setName(II->getName());
+ }
+ // Perform the cloning.
+ SmallVector<ReturnInst*,100> Returns;
+ CloneFunctionInto(NewF, F, ValueMap, Returns);
+ std::vector<Value*> fargs;
+ for(Function::arg_iterator ai = NewF->arg_begin(),
+ ae= NewF->arg_end(); ai != ae; ++ai) {
+ fargs.push_back(ai);
+ }
- if(F->paramHasAttr(argNum, Attribute::SExt) ||
- F->paramHasAttr(argNum, Attribute::ZExt))
- continue;
- // Construct the new Type
- // Appends the struct Type at the beginning
- std::vector<const Type*>TP;
- TP.push_back(LI->getOperand(0)->getType());
- for(unsigned c = 1; c < CI->getNumOperands();c++) {
- TP.push_back(CI->getOperand(c)->getType());
- }
-
- //return type is same as that of original instruction
- const FunctionType *NewFTy = FunctionType::get(CI->getType(), TP, false);
- Function *NewF;
- numSimplified++;
- if(numSimplified > 50) //26
- return true;
-
- NewF = Function::Create(NewFTy,
- GlobalValue::InternalLinkage,
- F->getNameStr() + ".TEST",
- &M);
-
- Function::arg_iterator NI = NewF->arg_begin();
- NI->setName("Sarg");
- ++NI;
-
- DenseMap<const Value*, Value*> ValueMap;
-
- for (Function::arg_iterator II = F->arg_begin(); NI != NewF->arg_end(); ++II, ++NI) {
- ValueMap[II] = NI;
- NI->setName(II->getName());
- }
- // Perform the cloning.
- SmallVector<ReturnInst*,100> Returns;
- CloneFunctionInto(NewF, F, ValueMap, Returns);
- std::vector<Value*> fargs;
- for(Function::arg_iterator ai = NewF->arg_begin(),
- ae= NewF->arg_end(); ai != ae; ++ai) {
- fargs.push_back(ai);
- }
-
- NewF->setAlignment(F->getAlignment());
- //Get the point to insert the GEP instr.
- NI = NewF->arg_begin();
- Instruction *InsertPoint;
- for (BasicBlock::iterator insrt = NewF->front().begin(); isa<AllocaInst>(InsertPoint = insrt); ++insrt) {;}
-
- LoadInst *LI_new = new LoadInst(cast<Value>(NI), "", InsertPoint);
- fargs.at(argNum)->replaceAllUsesWith(LI_new);
-
- SmallVector<Value*, 8> Args;
- Args.push_back(LI->getOperand(0));
- for(unsigned j =1;j<CI->getNumOperands();j++) {
- Args.push_back(CI->getOperand(j));
- }
- CallInst *CallI = CallInst::Create(NewF,Args.begin(), Args.end(),"", CI);
- CallI->setCallingConv(CI->getCallingConv());
- CI->replaceAllUsesWith(CallI);
- CI->eraseFromParent();
+ NewF->setAlignment(F->getAlignment());
+ //Get the point to insert the GEP instr.
+ NI = NewF->arg_begin();
+ SmallVector<Value*, 8> Ops(CI->op_begin()+1, CI->op_end());
+ Instruction *InsertPoint;
+ for (BasicBlock::iterator insrt = NewF->front().begin(); isa<AllocaInst>(InsertPoint = insrt); ++insrt) {;}
+
+ LoadInst *LI_new = new LoadInst(cast<Value>(NI), "", InsertPoint);
+ fargs.at(argNum)->replaceAllUsesWith(LI_new);
+
+ SmallVector<Value*, 8> Args;
+ Args.push_back(LI->getOperand(0));
+ for(unsigned j =1;j<CI->getNumOperands();j++) {
+ Args.push_back(CI->getOperand(j));
}
+ CallInst *CallI = CallInst::Create(NewF,Args.begin(), Args.end(),"", CI);
+ CallI->setCallingConv(CI->getCallingConv());
+ CI->replaceAllUsesWith(CallI);
+ CI->eraseFromParent();
}
}
}
@@ -162,4 +161,4 @@
char LoadArgs::ID = 0;
static RegisterPass<LoadArgs>
-X("ld-args", "Find GEP into structs passed as args");
+X("ld-args", "Find Load Inst passed as args");
More information about the llvm-commits
mailing list