Added unfinished inliner that uses type feedback info to inline instance methods.
Made class method inliner also inline message sends to super. Various updates caused by clang now attaching more sensible metadata to the IR.main
theraven
16 years ago
parent
160d1ce829
commit
6a58d3d49d
@ -0,0 +1,150 @@
|
||||
#include "llvm/Constants.h"
|
||||
#include "llvm/Pass.h"
|
||||
#include "llvm/Module.h"
|
||||
#include "llvm/Function.h"
|
||||
#include "llvm/Instructions.h"
|
||||
#include "llvm/Support/IRBuilder.h"
|
||||
#include "llvm/Linker.h"
|
||||
#include <vector>
|
||||
|
||||
using namespace llvm;
|
||||
|
||||
namespace {
|
||||
struct GNUObjCTypeFeedbackDrivenInliner : public ModulePass {
|
||||
|
||||
typedef std::pair<CallInst*,CallInst*> callPair;
|
||||
typedef std::vector<callPair > replacementVector;
|
||||
static char ID;
|
||||
uint32_t callsiteCount;
|
||||
const IntegerType *Int32Ty;
|
||||
GNUObjCTypeFeedbackDrivenInliner() : ModulePass(&ID), callsiteCount(0) {}
|
||||
|
||||
void profileFunction(Function &F, Constant *ModuleID) {
|
||||
for (Function::iterator i=F.begin(), e=F.end() ;
|
||||
i != e ; ++i) {
|
||||
|
||||
replacementVector replacements;
|
||||
for (BasicBlock::iterator b=i->begin(), last=i->end() ;
|
||||
b != last ; ++b) {
|
||||
|
||||
Module *M = F.getParent();
|
||||
if (CallInst *call = dyn_cast<CallInst>(b)) {
|
||||
if (Function *callee = call->getCalledFunction()) {
|
||||
if (callee->getName() == "objc_msg_lookup_sender") {
|
||||
llvm::Value *args[] = { call->getOperand(1),
|
||||
call->getOperand(2), call->getOperand(3),
|
||||
ModuleID, ConstantInt::get(Int32Ty,
|
||||
callsiteCount++) };
|
||||
Function *profile = cast<Function>(
|
||||
M->getOrInsertFunction("objc_msg_lookup_profile",
|
||||
callee->getFunctionType()->getReturnType(),
|
||||
args[0]->getType(), args[1]->getType(),
|
||||
args[2]->getType(),
|
||||
ModuleID->getType(), Int32Ty, NULL));
|
||||
llvm::CallInst *profileCall =
|
||||
CallInst::Create(profile, args, args+5, "", call);
|
||||
replacements.push_back(callPair(call, profileCall));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
for (replacementVector::iterator r=replacements.begin(),
|
||||
e=replacements.end() ; e!=r ; r++) {
|
||||
r->first->replaceAllUsesWith(r->second);
|
||||
r->second->getParent()->getInstList().erase(r->first);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public:
|
||||
virtual bool runOnModule(Module &M)
|
||||
{
|
||||
LLVMContext &VMContext = M.getContext();
|
||||
Int32Ty = IntegerType::get(VMContext, 32);
|
||||
const PointerType *PtrTy = Type::getInt8PtrTy(VMContext);
|
||||
Constant *moduleName =
|
||||
ConstantArray::get(VMContext, M.getModuleIdentifier(), true);
|
||||
moduleName = new GlobalVariable(M, moduleName->getType(), true,
|
||||
GlobalValue::InternalLinkage, moduleName,
|
||||
".objc_profile_module_name");
|
||||
std::vector<Constant*> functions;
|
||||
|
||||
llvm::Constant *Zeros[2];
|
||||
Zeros[0] = ConstantInt::get(Type::getInt32Ty(VMContext), 0);
|
||||
Zeros[1] = Zeros[0];
|
||||
|
||||
moduleName = ConstantExpr::getGetElementPtr(moduleName, Zeros, 2);
|
||||
functions.push_back(moduleName);;
|
||||
functions.push_back(moduleName);;
|
||||
|
||||
for (Module::iterator F=M.begin(), e=M.end() ;
|
||||
F != e ; ++F) {
|
||||
if (F->isDeclaration()) { continue; }
|
||||
functions.push_back(ConstantExpr::getBitCast(F, PtrTy));
|
||||
|
||||
Constant * ConstStr =
|
||||
llvm::ConstantArray::get(VMContext, F->getName());
|
||||
ConstStr = new GlobalVariable(M, ConstStr->getType(), true,
|
||||
GlobalValue::PrivateLinkage, ConstStr, "str");
|
||||
functions.push_back(
|
||||
ConstantExpr::getGetElementPtr(ConstStr, Zeros, 2));
|
||||
|
||||
profileFunction(*F, moduleName);
|
||||
}
|
||||
functions.push_back(ConstantPointerNull::get(PtrTy));
|
||||
Constant *symtab = ConstantArray::get(ArrayType::get(PtrTy,
|
||||
functions.size()), functions);
|
||||
Value *symbolTable = new GlobalVariable(M, symtab->getType(), true,
|
||||
GlobalValue::InternalLinkage, symtab, "symtab");
|
||||
|
||||
Function *init =
|
||||
Function::Create(FunctionType::get(Type::getVoidTy(VMContext), false),
|
||||
GlobalValue::PrivateLinkage, "load_symbol_table", &M);
|
||||
BasicBlock * EntryBB = BasicBlock::Create(VMContext, "entry", init);
|
||||
IRBuilder<> B = IRBuilder<>(EntryBB);
|
||||
Value *syms = B.CreateStructGEP(symbolTable, 0);
|
||||
B.CreateCall(M.getOrInsertFunction("objc_profile_write_symbols",
|
||||
Type::getVoidTy(VMContext), syms->getType(), NULL),
|
||||
syms);
|
||||
B.CreateRetVoid();
|
||||
|
||||
GlobalVariable *GCL = M.getGlobalVariable("llvm.global_ctors");
|
||||
|
||||
std::vector<Constant*> ctors;
|
||||
|
||||
ConstantArray *CA = cast<ConstantArray>(GCL->getInitializer());
|
||||
|
||||
for (User::op_iterator i = CA->op_begin(), e = CA->op_end(); i != e; ++i) {
|
||||
ctors.push_back(cast<ConstantStruct>(*i));
|
||||
}
|
||||
|
||||
// Type of one ctor
|
||||
const Type *ctorTy =
|
||||
cast<ArrayType>(GCL->getType()->getElementType())->getElementType();
|
||||
// Add the
|
||||
std::vector<Constant*> CSVals;
|
||||
CSVals.push_back(ConstantInt::get(Type::getInt32Ty(VMContext),65535));
|
||||
CSVals.push_back(init);
|
||||
ctors.push_back(ConstantStruct::get(GCL->getContext(), CSVals, false));
|
||||
// Create the array initializer.
|
||||
CA = cast<ConstantArray>(ConstantArray::get(ArrayType::get(ctorTy,
|
||||
ctors.size()), ctors));
|
||||
// Create the new global and replace the old one
|
||||
GlobalVariable *NGV = new GlobalVariable(CA->getType(),
|
||||
GCL->isConstant(), GCL->getLinkage(), CA, "", GCL->isThreadLocal());
|
||||
GCL->getParent()->getGlobalList().insert(GCL, NGV);
|
||||
NGV->takeName(GCL);
|
||||
GCL->replaceAllUsesWith(NGV);
|
||||
GCL->eraseFromParent();
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
};
|
||||
|
||||
char GNUObjCTypeFeedbackDrivenInliner::ID = 0;
|
||||
RegisterPass<GNUObjCTypeFeedbackDrivenInliner> X("gnu-objc-feedback-driven-inline",
|
||||
"Objective-C type feedback-driven inliner for the GNU runtime.", false,
|
||||
true);
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue