mirror of
https://github.com/c64scene-ar/llvm-6502.git
synced 2025-02-09 13:33:17 +00:00
0d40b2b30f
This patch replaces most of the Orc indirection utils API with a new class: JITCompileCallbackManager, which creates and manages JIT callbacks. Exposing this functionality directly allows the user to create callbacks that are associated with user supplied compilation actions. For example, you can create a callback to lazyily IR-gen something from an AST. (A kaleidoscope example demonstrating this will be committed shortly). This patch also refactors the CompileOnDemand layer to use the JITCompileCallbackManager API. git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@229461 91177308-0d34-0410-b5e6-96231b3b80d8
105 lines
3.8 KiB
C++
105 lines
3.8 KiB
C++
#include "llvm/ExecutionEngine/Orc/CloneSubModule.h"
|
|
#include "llvm/IR/Function.h"
|
|
#include "llvm/IR/GlobalVariable.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/Transforms/Utils/Cloning.h"
|
|
|
|
using namespace llvm;
|
|
|
|
void llvm::copyGVInitializer(GlobalVariable &New, const GlobalVariable &Orig,
|
|
ValueToValueMapTy &VMap) {
|
|
if (Orig.hasInitializer())
|
|
New.setInitializer(MapValue(Orig.getInitializer(), VMap));
|
|
}
|
|
|
|
void llvm::copyFunctionBody(Function &New, const Function &Orig,
|
|
ValueToValueMapTy &VMap) {
|
|
if (!Orig.isDeclaration()) {
|
|
Function::arg_iterator DestI = New.arg_begin();
|
|
for (Function::const_arg_iterator J = Orig.arg_begin(); J != Orig.arg_end();
|
|
++J) {
|
|
DestI->setName(J->getName());
|
|
VMap[J] = DestI++;
|
|
}
|
|
|
|
SmallVector<ReturnInst *, 8> Returns; // Ignore returns cloned.
|
|
CloneFunctionInto(&New, &Orig, VMap, /*ModuleLevelChanges=*/true, Returns);
|
|
}
|
|
}
|
|
|
|
void llvm::CloneSubModule(llvm::Module &Dst, const Module &Src,
|
|
HandleGlobalVariableFtor HandleGlobalVariable,
|
|
HandleFunctionFtor HandleFunction, bool CloneInlineAsm) {
|
|
|
|
ValueToValueMapTy VMap;
|
|
|
|
if (CloneInlineAsm)
|
|
Dst.appendModuleInlineAsm(Src.getModuleInlineAsm());
|
|
|
|
// Copy global variables (but not initializers, yet).
|
|
for (Module::const_global_iterator I = Src.global_begin(), E = Src.global_end();
|
|
I != E; ++I) {
|
|
GlobalVariable *GV = new GlobalVariable(
|
|
Dst, I->getType()->getElementType(), I->isConstant(), I->getLinkage(),
|
|
(Constant *)nullptr, I->getName(), (GlobalVariable *)nullptr,
|
|
I->getThreadLocalMode(), I->getType()->getAddressSpace());
|
|
GV->copyAttributesFrom(I);
|
|
VMap[I] = GV;
|
|
}
|
|
|
|
// Loop over the functions in the module, making external functions as before
|
|
for (Module::const_iterator I = Src.begin(), E = Src.end(); I != E; ++I) {
|
|
Function *NF =
|
|
Function::Create(cast<FunctionType>(I->getType()->getElementType()),
|
|
I->getLinkage(), I->getName(), &Dst);
|
|
NF->copyAttributesFrom(I);
|
|
VMap[I] = NF;
|
|
}
|
|
|
|
// Loop over the aliases in the module
|
|
for (Module::const_alias_iterator I = Src.alias_begin(), E = Src.alias_end();
|
|
I != E; ++I) {
|
|
auto *PTy = cast<PointerType>(I->getType());
|
|
auto *GA =
|
|
GlobalAlias::create(PTy->getElementType(), PTy->getAddressSpace(),
|
|
I->getLinkage(), I->getName(), &Dst);
|
|
GA->copyAttributesFrom(I);
|
|
VMap[I] = GA;
|
|
}
|
|
|
|
// Now that all of the things that global variable initializer can refer to
|
|
// have been created, loop through and copy the global variable referrers
|
|
// over... We also set the attributes on the global now.
|
|
for (Module::const_global_iterator I = Src.global_begin(), E = Src.global_end();
|
|
I != E; ++I) {
|
|
GlobalVariable &GV = *cast<GlobalVariable>(VMap[I]);
|
|
HandleGlobalVariable(GV, *I, VMap);
|
|
}
|
|
|
|
// Similarly, copy over function bodies now...
|
|
//
|
|
for (Module::const_iterator I = Src.begin(), E = Src.end(); I != E; ++I) {
|
|
Function &F = *cast<Function>(VMap[I]);
|
|
HandleFunction(F, *I, VMap);
|
|
}
|
|
|
|
// And aliases
|
|
for (Module::const_alias_iterator I = Src.alias_begin(), E = Src.alias_end();
|
|
I != E; ++I) {
|
|
GlobalAlias *GA = cast<GlobalAlias>(VMap[I]);
|
|
if (const Constant *C = I->getAliasee())
|
|
GA->setAliasee(MapValue(C, VMap));
|
|
}
|
|
|
|
// And named metadata....
|
|
for (Module::const_named_metadata_iterator I = Src.named_metadata_begin(),
|
|
E = Src.named_metadata_end();
|
|
I != E; ++I) {
|
|
const NamedMDNode &NMD = *I;
|
|
NamedMDNode *NewNMD = Dst.getOrInsertNamedMetadata(NMD.getName());
|
|
for (unsigned i = 0, e = NMD.getNumOperands(); i != e; ++i)
|
|
NewNMD->addOperand(MapMetadata(NMD.getOperand(i), VMap));
|
|
}
|
|
|
|
}
|