// allocated out of different pools of memory, increasing locality and shrinking
// pointer size.
//
-// This pass requires a DCE & instcombine pass to be run after it for best
-// results.
-//
//===----------------------------------------------------------------------===//
-#include "llvm/Transforms/IPO/PoolAllocate.h"
-#include "llvm/Transforms/CloneFunction.h"
+#if 0
+#include "llvm/Transforms/IPO.h"
+#include "llvm/Transforms/Utils/Cloning.h"
#include "llvm/Analysis/DataStructure.h"
-#include "llvm/Analysis/DataStructureGraph.h"
#include "llvm/Module.h"
-#include "llvm/Function.h"
-#include "llvm/BasicBlock.h"
#include "llvm/iMemory.h"
#include "llvm/iTerminators.h"
#include "llvm/iPHINode.h"
#include "llvm/iOther.h"
#include "llvm/DerivedTypes.h"
-#include "llvm/ConstantVals.h"
+#include "llvm/Constants.h"
#include "llvm/Target/TargetData.h"
#include "llvm/Support/InstVisitor.h"
-#include "llvm/Argument.h"
#include "Support/DepthFirstIterator.h"
#include "Support/STLExtras.h"
#include <algorithm>
+using std::vector;
+using std::cerr;
+using std::map;
+using std::string;
+using std::set;
// DEBUG_CREATE_POOLS - Enable this to turn on debug output for the pool
// creation phase in the top level function of a transformed data structure.
Ptr8bits, Ptr16bits, Ptr32bits
};
-static cl::Enum<enum PtrSize> ReqPointerSize("ptrsize", 0,
- "Set pointer size for -poolalloc pass",
+static cl::opt<PtrSize>
+ReqPointerSize("poolalloc-ptr-size",
+ cl::desc("Set pointer size for -poolalloc pass"),
+ cl::values(
clEnumValN(Ptr32bits, "32", "Use 32 bit indices for pointers"),
clEnumValN(Ptr16bits, "16", "Use 16 bit indices for pointers"),
- clEnumValN(Ptr8bits , "8", "Use 8 bit indices for pointers"), 0);
+ clEnumValN(Ptr8bits , "8", "Use 8 bit indices for pointers"),
+ 0));
-static cl::Flag DisableRLE("no-pool-load-elim", "Disable pool load elimination after poolalloc pass", cl::Hidden);
+static cl::opt<bool>
+DisableRLE("no-pool-load-elim", cl::Hidden,
+ cl::desc("Disable pool load elimination after poolalloc pass"));
const Type *POINTERTYPE;
static TargetData TargetData("test");
static const Type *getPointerTransformedType(const Type *Ty) {
- if (PointerType *PT = dyn_cast<PointerType>(Ty)) {
+ if (const PointerType *PT = dyn_cast<PointerType>(Ty)) {
return POINTERTYPE;
- } else if (StructType *STy = dyn_cast<StructType>(Ty)) {
+ } else if (const StructType *STy = dyn_cast<StructType>(Ty)) {
vector<const Type *> NewElTypes;
NewElTypes.reserve(STy->getElementTypes().size());
for (StructType::ElementTypes::const_iterator
E = STy->getElementTypes().end(); I != E; ++I)
NewElTypes.push_back(getPointerTransformedType(*I));
return StructType::get(NewElTypes);
- } else if (ArrayType *ATy = dyn_cast<ArrayType>(Ty)) {
+ } else if (const ArrayType *ATy = dyn_cast<ArrayType>(Ty)) {
return ArrayType::get(getPointerTransformedType(ATy->getElementType()),
ATy->getNumElements());
} else {
// argument records, in order. Note that this must be a stable sort so
// that the entries with the same sorting criteria (ie they are multiple
// pool entries for the same argument) are kept in depth first order.
- stable_sort(ArgInfo.begin(), ArgInfo.end());
+ std::stable_sort(ArgInfo.begin(), ArgInfo.end());
}
// addCallInfo - For a specified function call CI, figure out which pool
return Result;
}
- bool run(Module *M);
+ bool run(Module &M);
- // getAnalysisUsageInfo - This function requires data structure information
+ // getAnalysisUsage - This function requires data structure information
// to be able to see what is pool allocatable.
//
- virtual void getAnalysisUsageInfo(Pass::AnalysisSet &Required,
- Pass::AnalysisSet &,Pass::AnalysisSet &) {
- Required.push_back(DataStructure::ID);
+ virtual void getAnalysisUsage(AnalysisUsage &AU) const {
+ AU.addRequired<DataStructure>();
}
public:
// specified module and update the Pool* instance variables to point to
// them.
//
- void addPoolPrototypes(Module *M);
+ void addPoolPrototypes(Module &M);
// CreatePools - Insert instructions into the function we are processing to
map<DSNode*, PoolInfo> &PoolDescs);
};
+
+ RegisterOpt<PoolAllocate> X("poolalloc",
+ "Pool allocate disjoint datastructures");
}
// isNotPoolableAlloc - This is a predicate that returns true if the specified
// variable sized array allocations and alloca's (which we do not want to
// pool allocate)
//
- Allocs.erase(remove_if(Allocs.begin(), Allocs.end(), isNotPoolableAlloc),
+ Allocs.erase(std::remove_if(Allocs.begin(), Allocs.end(), isNotPoolableAlloc),
Allocs.end());
return 0;
}
- BasicBlock::iterator ReplaceInstWith(Instruction *I, Instruction *New) {
- BasicBlock *BB = I->getParent();
- BasicBlock::iterator RI = find(BB->begin(), BB->end(), I);
- BB->getInstList().replaceWith(RI, New);
- XFormMap[I] = New;
- return RI;
+ BasicBlock::iterator ReplaceInstWith(Instruction &I, Instruction *New) {
+ BasicBlock *BB = I.getParent();
+ BasicBlock::iterator RI = &I;
+ BB->getInstList().remove(RI);
+ BB->getInstList().insert(RI, New);
+ XFormMap[&I] = New;
+ return New;
}
- LoadInst *createPoolBaseInstruction(Value *PtrVal) {
+ Instruction *createPoolBaseInstruction(Value *PtrVal) {
const ScalarInfo &SC = getScalarRef(PtrVal);
vector<Value*> Args(3);
Args[0] = ConstantUInt::get(Type::UIntTy, 0); // No pointer offset
Args[1] = ConstantUInt::get(Type::UByteTy, 0); // Field #0 of pool descriptr
Args[2] = ConstantUInt::get(Type::UByteTy, 0); // Field #0 of poolalloc val
- return new LoadInst(SC.Pool.Handle, Args, PtrVal->getName()+".poolbase");
+ return new LoadInst(SC.Pool.Handle, Args, PtrVal->getName()+".poolbase");
}
// NewInstructionCreator instance...
//===--------------------------------------------------------------------===//
- void visitGetElementPtrInst(GetElementPtrInst *I) {
+ void visitGetElementPtrInst(GetElementPtrInst &I) {
assert(0 && "Cannot transform get element ptr instructions yet!");
}
// Replace the load instruction with a new one.
- void visitLoadInst(LoadInst *I) {
- Instruction *PoolBase = createPoolBaseInstruction(I->getOperand(0));
+ void visitLoadInst(LoadInst &I) {
+ vector<Instruction *> BeforeInsts;
// Cast our index to be a UIntTy so we can use it to index into the pool...
CastInst *Index = new CastInst(Constant::getNullValue(POINTERTYPE),
- Type::UIntTy, I->getOperand(0)->getName());
-
- ReferencesToUpdate.push_back(RefToUpdate(Index, 0, I->getOperand(0)));
+ Type::UIntTy, I.getOperand(0)->getName());
+ BeforeInsts.push_back(Index);
+ ReferencesToUpdate.push_back(RefToUpdate(Index, 0, I.getOperand(0)));
+
+ // Include the pool base instruction...
+ Instruction *PoolBase = createPoolBaseInstruction(I.getOperand(0));
+ BeforeInsts.push_back(PoolBase);
- vector<Value*> Indices(I->idx_begin(), I->idx_end());
Instruction *IdxInst =
- BinaryOperator::create(Instruction::Add, Indices[0], Index,
- I->getName()+".idx");
+ BinaryOperator::create(Instruction::Add, *I.idx_begin(), Index,
+ I.getName()+".idx");
+ BeforeInsts.push_back(IdxInst);
+
+ vector<Value*> Indices(I.idx_begin(), I.idx_end());
Indices[0] = IdxInst;
- Instruction *NewLoad = new LoadInst(PoolBase, Indices, I->getName());
+ Instruction *Address = new GetElementPtrInst(PoolBase, Indices,
+ I.getName()+".addr");
+ BeforeInsts.push_back(Address);
+
+ Instruction *NewLoad = new LoadInst(Address, I.getName());
// Replace the load instruction with the new load instruction...
BasicBlock::iterator II = ReplaceInstWith(I, NewLoad);
- // Add the pool base calculator instruction before the load...
- II = NewLoad->getParent()->getInstList().insert(II, PoolBase) + 1;
-
- // Add the idx calculator instruction before the load...
- II = NewLoad->getParent()->getInstList().insert(II, Index) + 1;
-
- // Add the cast before the load instruction...
- NewLoad->getParent()->getInstList().insert(II, IdxInst);
+ // Add all of the instructions before the load...
+ NewLoad->getParent()->getInstList().insert(II, BeforeInsts.begin(),
+ BeforeInsts.end());
// If not yielding a pool allocated pointer, use the new load value as the
// value in the program instead of the old load value...
//
- if (!getScalar(I))
- I->replaceAllUsesWith(NewLoad);
+ if (!getScalar(&I))
+ I.replaceAllUsesWith(NewLoad);
}
// Replace the store instruction with a new one. In the store instruction,
// the value stored could be a pointer type, meaning that the new store may
// have to change one or both of it's operands.
//
- void visitStoreInst(StoreInst *I) {
- assert(getScalar(I->getOperand(1)) &&
+ void visitStoreInst(StoreInst &I) {
+ assert(getScalar(I.getOperand(1)) &&
"Store inst found only storing pool allocated pointer. "
"Not imp yet!");
- Value *Val = I->getOperand(0); // The value to store...
+ Value *Val = I.getOperand(0); // The value to store...
+
// Check to see if the value we are storing is a data structure pointer...
- //if (const ScalarInfo *ValScalar = getScalar(I->getOperand(0)))
- if (isa<PointerType>(I->getOperand(0)->getType()))
+ //if (const ScalarInfo *ValScalar = getScalar(I.getOperand(0)))
+ if (isa<PointerType>(I.getOperand(0)->getType()))
Val = Constant::getNullValue(POINTERTYPE); // Yes, store a dummy
- Instruction *PoolBase = createPoolBaseInstruction(I->getOperand(1));
+ Instruction *PoolBase = createPoolBaseInstruction(I.getOperand(1));
// Cast our index to be a UIntTy so we can use it to index into the pool...
CastInst *Index = new CastInst(Constant::getNullValue(POINTERTYPE),
- Type::UIntTy, I->getOperand(1)->getName());
- ReferencesToUpdate.push_back(RefToUpdate(Index, 0, I->getOperand(1)));
+ Type::UIntTy, I.getOperand(1)->getName());
+ ReferencesToUpdate.push_back(RefToUpdate(Index, 0, I.getOperand(1)));
+
+ // Instructions to add after the Index...
+ vector<Instruction*> AfterInsts;
- vector<Value*> Indices(I->idx_begin(), I->idx_end());
Instruction *IdxInst =
- BinaryOperator::create(Instruction::Add, Indices[0], Index, "idx");
- Indices[0] = IdxInst;
+ BinaryOperator::create(Instruction::Add, *I.idx_begin(), Index, "idx");
+ AfterInsts.push_back(IdxInst);
- Instruction *NewStore = new StoreInst(Val, PoolBase, Indices);
+ vector<Value*> Indices(I.idx_begin(), I.idx_end());
+ Indices[0] = IdxInst;
+ Instruction *Address = new GetElementPtrInst(PoolBase, Indices,
+ I.getName()+"storeaddr");
+ AfterInsts.push_back(Address);
- if (Val != I->getOperand(0)) // Value stored was a pointer?
- ReferencesToUpdate.push_back(RefToUpdate(NewStore, 0, I->getOperand(0)));
+ Instruction *NewStore = new StoreInst(Val, Address);
+ AfterInsts.push_back(NewStore);
+ if (Val != I.getOperand(0)) // Value stored was a pointer?
+ ReferencesToUpdate.push_back(RefToUpdate(NewStore, 0, I.getOperand(0)));
// Replace the store instruction with the cast instruction...
BasicBlock::iterator II = ReplaceInstWith(I, Index);
// Add the pool base calculator instruction before the index...
- II = Index->getParent()->getInstList().insert(II, PoolBase) + 2;
-
- // Add the indexing instruction...
- II = Index->getParent()->getInstList().insert(II, IdxInst) + 1;
+ II = ++Index->getParent()->getInstList().insert(II, PoolBase);
+ ++II;
- // Add the store after the cast instruction...
- Index->getParent()->getInstList().insert(II, NewStore);
+ // Add the instructions that go after the index...
+ Index->getParent()->getInstList().insert(II, AfterInsts.begin(),
+ AfterInsts.end());
}
// Create call to poolalloc for every malloc instruction
- void visitMallocInst(MallocInst *I) {
- const ScalarInfo &SCI = getScalarRef(I);
+ void visitMallocInst(MallocInst &I) {
+ const ScalarInfo &SCI = getScalarRef(&I);
vector<Value*> Args;
CallInst *Call;
- if (!I->isArrayAllocation()) {
+ if (!I.isArrayAllocation()) {
Args.push_back(SCI.Pool.Handle);
- Call = new CallInst(PoolAllocator.PoolAlloc, Args, I->getName());
+ Call = new CallInst(PoolAllocator.PoolAlloc, Args, I.getName());
} else {
- Args.push_back(I->getArraySize());
+ Args.push_back(I.getArraySize());
Args.push_back(SCI.Pool.Handle);
- Call = new CallInst(PoolAllocator.PoolAllocArray, Args, I->getName());
+ Call = new CallInst(PoolAllocator.PoolAllocArray, Args, I.getName());
}
ReplaceInstWith(I, Call);
}
// Convert a call to poolfree for every free instruction...
- void visitFreeInst(FreeInst *I) {
+ void visitFreeInst(FreeInst &I) {
// Create a new call to poolfree before the free instruction
vector<Value*> Args;
Args.push_back(Constant::getNullValue(POINTERTYPE));
- Args.push_back(getScalarRef(I->getOperand(0)).Pool.Handle);
+ Args.push_back(getScalarRef(I.getOperand(0)).Pool.Handle);
Instruction *NewCall = new CallInst(PoolAllocator.PoolFree, Args);
ReplaceInstWith(I, NewCall);
- ReferencesToUpdate.push_back(RefToUpdate(NewCall, 1, I->getOperand(0)));
+ ReferencesToUpdate.push_back(RefToUpdate(NewCall, 1, I.getOperand(0)));
}
// visitCallInst - Create a new call instruction with the extra arguments for
// all of the memory pools that the call needs.
//
- void visitCallInst(CallInst *I) {
- TransformFunctionInfo &TI = CallMap[I];
+ void visitCallInst(CallInst &I) {
+ TransformFunctionInfo &TI = CallMap[&I];
// Start with all of the old arguments...
- vector<Value*> Args(I->op_begin()+1, I->op_end());
+ vector<Value*> Args(I.op_begin()+1, I.op_end());
for (unsigned i = 0, e = TI.ArgInfo.size(); i != e; ++i) {
// Replace all of the pointer arguments with our new pointer typed values.
}
Function *NF = PoolAllocator.getTransformedFunction(TI);
- Instruction *NewCall = new CallInst(NF, Args, I->getName());
+ Instruction *NewCall = new CallInst(NF, Args, I.getName());
ReplaceInstWith(I, NewCall);
// Keep track of the mapping of operands so that we can resolve them to real
for (unsigned i = 0, e = TI.ArgInfo.size(); i != e; ++i)
if (TI.ArgInfo[i].ArgNo != -1)
ReferencesToUpdate.push_back(RefToUpdate(NewCall, TI.ArgInfo[i].ArgNo+1,
- I->getOperand(TI.ArgInfo[i].ArgNo+1)));
+ I.getOperand(TI.ArgInfo[i].ArgNo+1)));
else
RetVal = 0; // If returning a pointer, don't change retval...
// instead of the old call...
//
if (RetVal)
- I->replaceAllUsesWith(RetVal);
+ I.replaceAllUsesWith(RetVal);
}
// visitPHINode - Create a new PHI node of POINTERTYPE for all of the old Phi
// nodes...
//
- void visitPHINode(PHINode *PN) {
+ void visitPHINode(PHINode &PN) {
Value *DummyVal = Constant::getNullValue(POINTERTYPE);
- PHINode *NewPhi = new PHINode(POINTERTYPE, PN->getName());
- for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i) {
- NewPhi->addIncoming(DummyVal, PN->getIncomingBlock(i));
+ PHINode *NewPhi = new PHINode(POINTERTYPE, PN.getName());
+ for (unsigned i = 0, e = PN.getNumIncomingValues(); i != e; ++i) {
+ NewPhi->addIncoming(DummyVal, PN.getIncomingBlock(i));
ReferencesToUpdate.push_back(RefToUpdate(NewPhi, i*2,
- PN->getIncomingValue(i)));
+ PN.getIncomingValue(i)));
}
ReplaceInstWith(PN, NewPhi);
}
// visitReturnInst - Replace ret instruction with a new return...
- void visitReturnInst(ReturnInst *I) {
+ void visitReturnInst(ReturnInst &I) {
Instruction *Ret = new ReturnInst(Constant::getNullValue(POINTERTYPE));
ReplaceInstWith(I, Ret);
- ReferencesToUpdate.push_back(RefToUpdate(Ret, 0, I->getOperand(0)));
+ ReferencesToUpdate.push_back(RefToUpdate(Ret, 0, I.getOperand(0)));
}
// visitSetCondInst - Replace a conditional test instruction with a new one
- void visitSetCondInst(SetCondInst *SCI) {
- BinaryOperator *I = (BinaryOperator*)SCI;
+ void visitSetCondInst(SetCondInst &SCI) {
+ BinaryOperator &I = (BinaryOperator&)SCI;
Value *DummyVal = Constant::getNullValue(POINTERTYPE);
- BinaryOperator *New = BinaryOperator::create(I->getOpcode(), DummyVal,
- DummyVal, I->getName());
+ BinaryOperator *New = BinaryOperator::create(I.getOpcode(), DummyVal,
+ DummyVal, I.getName());
ReplaceInstWith(I, New);
- ReferencesToUpdate.push_back(RefToUpdate(New, 0, I->getOperand(0)));
- ReferencesToUpdate.push_back(RefToUpdate(New, 1, I->getOperand(1)));
+ ReferencesToUpdate.push_back(RefToUpdate(New, 0, I.getOperand(0)));
+ ReferencesToUpdate.push_back(RefToUpdate(New, 1, I.getOperand(1)));
// Make sure branches refer to the new condition...
- I->replaceAllUsesWith(New);
+ I.replaceAllUsesWith(New);
}
- void visitInstruction(Instruction *I) {
+ void visitInstruction(Instruction &I) {
cerr << "Unknown instruction to FunctionBodyTransformer:\n" << I;
}
};
}
#ifdef DEBUG_POOLBASE_LOAD_ELIMINATOR
- void visitFunction(Function *F) {
- cerr << "Pool Load Elim '" << F->getName() << "'\t";
+ void visitFunction(Function &F) {
+ cerr << "Pool Load Elim '" << F.getName() << "'\t";
}
~PoolBaseLoadEliminator() {
unsigned Total = Eliminated+Remaining;
// local transformation, we reset all of our state when we enter a new basic
// block.
//
- void visitBasicBlock(BasicBlock *) {
+ void visitBasicBlock(BasicBlock &) {
PoolDescMap.clear(); // Forget state.
}
// indicating that we have a value available to recycle next time we see the
// poolbase of this instruction being loaded.
//
- void visitLoadInst(LoadInst *LI) {
- Value *LoadAddr = LI->getPointerOperand();
+ void visitLoadInst(LoadInst &LI) {
+ Value *LoadAddr = LI.getPointerOperand();
map<Value*, LoadInst*>::iterator VIt = PoolDescMap.find(LoadAddr);
if (VIt != PoolDescMap.end()) { // We already have a value for this load?
- LI->replaceAllUsesWith(VIt->second); // Make the current load dead
+ LI.replaceAllUsesWith(VIt->second); // Make the current load dead
++Eliminated;
} else {
// This load might not be a load of a pool pointer, check to see if it is
- if (LI->getNumOperands() == 4 && // load pool, uint 0, ubyte 0, ubyte 0
+ if (LI.getNumOperands() == 4 && // load pool, uint 0, ubyte 0, ubyte 0
find(PoolDescValues.begin(), PoolDescValues.end(), LoadAddr) !=
PoolDescValues.end()) {
assert("Make sure it's a load of the pool base, not a chaining field" &&
- LI->getOperand(1) == Constant::getNullValue(Type::UIntTy) &&
- LI->getOperand(2) == Constant::getNullValue(Type::UByteTy) &&
- LI->getOperand(3) == Constant::getNullValue(Type::UByteTy));
+ LI.getOperand(1) == Constant::getNullValue(Type::UIntTy) &&
+ LI.getOperand(2) == Constant::getNullValue(Type::UByteTy) &&
+ LI.getOperand(3) == Constant::getNullValue(Type::UByteTy));
// If it is a load of a pool base, keep track of it for future reference
- PoolDescMap.insert(make_pair(LoadAddr, LI));
+ PoolDescMap.insert(std::make_pair(LoadAddr, &LI));
++Remaining;
}
}
// function might call one of these functions, so be conservative. Through
// more analysis, this could be improved in the future.
//
- void visitCallInst(CallInst *) {
+ void visitCallInst(CallInst &) {
PoolDescMap.clear();
}
};
NodeMapping);
} else {
// Figure out which node argument # ArgNo points to in the called graph.
- Value *Arg = F->getArgumentList()[TFI.ArgInfo[i].ArgNo];
- addNodeMapping(TFI.ArgInfo[i].Node, CalledGraph.getValueMap()[Arg],
+ Function::aiterator AI = F->abegin();
+ std::advance(AI, TFI.ArgInfo[i].ArgNo);
+ addNodeMapping(TFI.ArgInfo[i].Node, CalledGraph.getValueMap()[AI],
NodeMapping);
}
LastArgNo = TFI.ArgInfo[i].ArgNo;
Done = false;
}
- for (unsigned i = 0, e = Func->getArgumentList().size(); i != e; ++i) {
- Argument *Arg = Func->getArgumentList()[i];
- if (isa<PointerType>(Arg->getType())) {
+ unsigned i = 0;
+ for (Function::aiterator I = Func->abegin(), E = Func->aend(); I!=E; ++I,++i){
+ if (isa<PointerType>(I->getType())) {
if (PtrNo < ArgInfo.size() && ArgInfo[PtrNo++].ArgNo == (int)i) {
// We DO transform this arg... skip all possible entries for argument
while (PtrNo < ArgInfo.size() && ArgInfo[PtrNo].ArgNo == (int)i)
if (i == 0) // Only process retvals once (performance opt)
markReachableNodes(CalledDS.getRetNodes(), ReachableNodes);
} else { // If it's an argument value...
- Argument *Arg = Func->getArgumentList()[ArgInfo[i].ArgNo];
- if (isa<PointerType>(Arg->getType()))
- markReachableNodes(CalledDS.getValueMap()[Arg], ReachableNodes);
+ Function::aiterator AI = Func->abegin();
+ std::advance(AI, ArgInfo[i].ArgNo);
+ if (isa<PointerType>(AI->getType()))
+ markReachableNodes(CalledDS.getValueMap()[AI], ReachableNodes);
}
}
}
}
- for (unsigned i = 0, e = Func->getArgumentList().size(); i != e; ++i) {
- Argument *Arg = Func->getArgumentList()[i];
- if (isa<PointerType>(Arg->getType())) {
+ i = 0;
+ for (Function::aiterator I = Func->abegin(), E = Func->aend(); I!=E; ++I, ++i)
+ if (isa<PointerType>(I->getType())) {
if (PtrNo < ArgInfo.size() && ArgInfo[PtrNo++].ArgNo == (int)i) {
// We DO transform this arg... skip all possible entries for argument
while (PtrNo < ArgInfo.size() && ArgInfo[PtrNo].ArgNo == (int)i)
} else {
// This should generalize to any number of nodes, just see if any are
// reachable.
- assert(CalledDS.getValueMap()[Arg].size() == 1 &&
+ assert(CalledDS.getValueMap()[I].size() == 1 &&
"Only handle case where pointing to one node so far!");
// If the arg is not marked as being passed in, but it NEEDS to
// be transformed, then make it known now.
//
- DSNode *N = CalledDS.getValueMap()[Arg][0].Node;
+ DSNode *N = CalledDS.getValueMap()[I][0].Node;
if (ReachableNodes.count(N)) {
#ifdef DEBUG_TRANSFORM_PROGRESS
cerr << "ensure dependant arguments adds for arg #" << i << "\n";
}
}
}
- }
}
if (PoolDescs.count(RetNode.Node)) {
// Loop over all of the basic blocks, adding return instructions...
for (Function::iterator I = F->begin(), E = F->end(); I != E; ++I)
- if (ReturnInst *RI = dyn_cast<ReturnInst>((*I)->getTerminator()))
+ if (ReturnInst *RI = dyn_cast<ReturnInst>(I->getTerminator()))
InstToFix.push_back(RI);
}
}
#ifdef DEBUG_TRANSFORM_PROGRESS
for (unsigned i = 0, e = InstToFix.size(); i != e; ++i) {
cerr << "Fixing: " << InstToFix[i];
- NIC.visit(InstToFix[i]);
+ NIC.visit(*InstToFix[i]);
}
#else
NIC.visit(InstToFix.begin(), InstToFix.end());
// we can safely delete Arguments whose types have changed...
//
for_each(InstToFix.begin(), InstToFix.end(),
- mem_fun(&Instruction::dropAllReferences));
+ std::mem_fun(&Instruction::dropAllReferences));
// Loop through all of the pointer arguments coming into the function,
// replacing them with arguments of POINTERTYPE to match the function type of
//
FunctionType::ParamTypes::const_iterator TI =
F->getFunctionType()->getParamTypes().begin();
- for (Function::ArgumentListType::iterator I = F->getArgumentList().begin(),
- E = F->getArgumentList().end(); I != E; ++I, ++TI) {
- Argument *Arg = *I;
- if (Arg->getType() != *TI) {
- assert(isa<PointerType>(Arg->getType()) && *TI == POINTERTYPE);
- Argument *NewArg = new Argument(*TI, Arg->getName());
- XFormMap[Arg] = NewArg; // Map old arg into new arg...
+ for (Function::aiterator I = F->abegin(), E = F->aend(); I != E; ++I, ++TI) {
+ if (I->getType() != *TI) {
+ assert(isa<PointerType>(I->getType()) && *TI == POINTERTYPE);
+ Argument *NewArg = new Argument(*TI, I->getName());
+ XFormMap[I] = NewArg; // Map old arg into new arg...
// Replace the old argument and then delete it...
- delete F->getArgumentList().replaceWith(I, NewArg);
+ I = F->getArgumentList().erase(I);
+ I = F->getArgumentList().insert(I, NewArg);
}
}
// Add arguments to the function... starting with all of the old arguments
vector<Value*> ArgMap;
- for (unsigned i = 0, e = TFI.Func->getArgumentList().size(); i != e; ++i) {
- const Argument *OFA = TFI.Func->getArgumentList()[i];
- Argument *NFA = new Argument(OFA->getType(), OFA->getName());
+ for (Function::const_aiterator I = TFI.Func->abegin(), E = TFI.Func->aend();
+ I != E; ++I) {
+ Argument *NFA = new Argument(I->getType(), I->getName());
NewFunc->getArgumentList().push_back(NFA);
ArgMap.push_back(NFA); // Keep track of the arguments
}
#ifdef DEBUG_TRANSFORM_PROGRESS
cerr << "Should be argument #: " << ArgNo << "[i = " << a << "]\n";
#endif
- assert(ArgNo < NewFunc->getArgumentList().size() &&
+ assert(ArgNo < NewFunc->asize() &&
"Call already has pool arguments added??");
// Map the pool argument into the called function...
- CalleeValue = NewFunc->getArgumentList()[ArgNo];
+ Function::aiterator AI = NewFunc->abegin();
+ std::advance(AI, ArgNo);
+ CalleeValue = AI;
break; // Found value, quit loop
}
// Add the descriptor. We already know everything about it by now, much
// of it is the same as the caller info.
//
- PoolDescs.insert(make_pair(CalleeNode,
+ PoolDescs.insert(std::make_pair(CalleeNode,
PoolInfo(CalleeNode, CalleeValue,
CallerPI.NewType,
CallerPI.PoolType)));
static unsigned countPointerTypes(const Type *Ty) {
if (isa<PointerType>(Ty)) {
return 1;
- } else if (StructType *STy = dyn_cast<StructType>(Ty)) {
+ } else if (const StructType *STy = dyn_cast<StructType>(Ty)) {
unsigned Num = 0;
for (unsigned i = 0, e = STy->getElementTypes().size(); i != e; ++i)
Num += countPointerTypes(STy->getElementTypes()[i]);
return Num;
- } else if (ArrayType *ATy = dyn_cast<ArrayType>(Ty)) {
+ } else if (const ArrayType *ATy = dyn_cast<ArrayType>(Ty)) {
return countPointerTypes(ATy->getElementType());
} else {
assert(Ty->isPrimitiveType() && "Unknown derived type!");
// Find all of the return nodes in the function...
vector<BasicBlock*> ReturnNodes;
for (Function::iterator I = F->begin(), E = F->end(); I != E; ++I)
- if (isa<ReturnInst>((*I)->getTerminator()))
- ReturnNodes.push_back(*I);
+ if (isa<ReturnInst>(I->getTerminator()))
+ ReturnNodes.push_back(I);
#ifdef DEBUG_CREATE_POOLS
cerr << "Allocs that we are pool allocating:\n";
// except the node & NewType fields.
//
map<DSNode*, PoolInfo>::iterator PI =
- PoolDescs.insert(make_pair(Allocs[i], PoolInfo(Allocs[i]))).first;
+ PoolDescs.insert(std::make_pair(Allocs[i], PoolInfo(Allocs[i]))).first;
// Add a symbol table entry for the new type if there was one for the old
// type...
CurModule->addTypeName(OldName+".pool", PoolType);
// Create the pool type, with opaque values for pointers...
- AbsPoolTyMap.insert(make_pair(Allocs[i], PoolType));
+ AbsPoolTyMap.insert(std::make_pair(Allocs[i], PoolType));
#ifdef DEBUG_CREATE_POOLS
cerr << "POOL TY: " << AbsPoolTyMap.find(Allocs[i])->second.get() << "\n";
#endif
// The actual struct type could change each time through the loop, so it's
// NOT loop invariant.
- StructType *PoolTy = cast<StructType>(PoolTyH.get());
+ const StructType *PoolTy = cast<StructType>(PoolTyH.get());
// Get the opaque type...
- DerivedType *ElTy =
- cast<DerivedType>(PoolTy->getElementTypes()[p+1].get());
+ DerivedType *ElTy = (DerivedType*)(PoolTy->getElementTypes()[p+1].get());
#ifdef DEBUG_CREATE_POOLS
cerr << "Refining " << ElTy << " of " << PoolTy << " to "
"Pool type should not be abstract anymore!");
// Add an allocation and a free for each pool...
- AllocaInst *PoolAlloc
- = new AllocaInst(PointerType::get(PI.PoolType), 0,
- CurModule->getTypeName(PI.PoolType));
+ AllocaInst *PoolAlloc = new AllocaInst(PI.PoolType, 0,
+ CurModule->getTypeName(PI.PoolType));
PI.Handle = PoolAlloc;
EntryNodeInsts.push_back(PoolAlloc);
AllocationInst *AI = Allocs[i]->getAllocation();
// Insert it before the return instruction...
BasicBlock *RetNode = ReturnNodes[EN];
- RetNode->getInstList().insert(RetNode->end()-1, Destroy);
+ RetNode->getInstList().insert(RetNode->end()--, Destroy);
}
}
}
// Insert the entry node code into the entry block...
- F->getEntryNode()->getInstList().insert(F->getEntryNode()->begin()+1,
+ F->getEntryNode().getInstList().insert(++F->getEntryNode().begin(),
EntryNodeInsts.begin(),
EntryNodeInsts.end());
}
// addPoolPrototypes - Add prototypes for the pool functions to the specified
// module and update the Pool* instance variables to point to them.
//
-void PoolAllocate::addPoolPrototypes(Module *M) {
+void PoolAllocate::addPoolPrototypes(Module &M) {
// Get poolinit function...
vector<const Type*> Args;
Args.push_back(Type::UIntTy); // Num bytes per element
FunctionType *PoolInitTy = FunctionType::get(Type::VoidTy, Args, true);
- PoolInit = M->getOrInsertFunction("poolinit", PoolInitTy);
+ PoolInit = M.getOrInsertFunction("poolinit", PoolInitTy);
// Get pooldestroy function...
Args.pop_back(); // Only takes a pool...
FunctionType *PoolDestroyTy = FunctionType::get(Type::VoidTy, Args, true);
- PoolDestroy = M->getOrInsertFunction("pooldestroy", PoolDestroyTy);
+ PoolDestroy = M.getOrInsertFunction("pooldestroy", PoolDestroyTy);
// Get the poolalloc function...
FunctionType *PoolAllocTy = FunctionType::get(POINTERTYPE, Args, true);
- PoolAlloc = M->getOrInsertFunction("poolalloc", PoolAllocTy);
+ PoolAlloc = M.getOrInsertFunction("poolalloc", PoolAllocTy);
// Get the poolfree function...
Args.push_back(POINTERTYPE); // Pointer to free
FunctionType *PoolFreeTy = FunctionType::get(Type::VoidTy, Args, true);
- PoolFree = M->getOrInsertFunction("poolfree", PoolFreeTy);
+ PoolFree = M.getOrInsertFunction("poolfree", PoolFreeTy);
Args[0] = Type::UIntTy; // Number of slots to allocate
FunctionType *PoolAllocArrayTy = FunctionType::get(POINTERTYPE, Args, true);
- PoolAllocArray = M->getOrInsertFunction("poolallocarray", PoolAllocArrayTy);
+ PoolAllocArray = M.getOrInsertFunction("poolallocarray", PoolAllocArrayTy);
}
-bool PoolAllocate::run(Module *M) {
+bool PoolAllocate::run(Module &M) {
addPoolPrototypes(M);
- CurModule = M;
+ CurModule = &M;
DS = &getAnalysis<DataStructure>();
bool Changed = false;
- // We cannot use an iterator here because it will get invalidated when we add
- // functions to the module later...
- for (unsigned i = 0; i != M->size(); ++i)
- if (!M->getFunctionList()[i]->isExternal()) {
- Changed |= processFunction(M->getFunctionList()[i]);
+ for (Module::iterator I = M.begin(); I != M.end(); ++I)
+ if (!I->isExternal()) {
+ Changed |= processFunction(I);
if (Changed) {
cerr << "Only processing one function\n";
break;
return false;
}
-
// createPoolAllocatePass - Global function to access the functionality of this
// pass...
//
-Pass *createPoolAllocatePass() { return new PoolAllocate(); }
+Pass *createPoolAllocatePass() {
+ assert(0 && "Pool allocator disabled!");
+ return 0;
+ //return new PoolAllocate();
+}
+#endif