mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-10-24 05:23:45 +02:00
27fb68307a
MCJIT.
This patch decouples the two responsibilities of the RTDyldMemoryManager class,
memory management and symbol resolution, into two new classes:
RuntimeDyld::MemoryManager and RuntimeDyld::SymbolResolver.
The symbol resolution interface is modified slightly, from:
uint64_t getSymbolAddress(const std::string &Name);
to:
RuntimeDyld::SymbolInfo findSymbol(const std::string &Name);
The latter passes symbol flags along with symbol addresses, allowing RuntimeDyld
and others to reason about non-strong/non-exported symbols.
The memory management interface removes the following method:
void notifyObjectLoaded(ExecutionEngine *EE,
const object::ObjectFile &) {}
as it is not related to memory management. (Note: Backwards compatibility *is*
maintained for this method in MCJIT and OrcMCJITReplacement, see below).
The RTDyldMemoryManager class remains in-tree for backwards compatibility.
It inherits directly from RuntimeDyld::SymbolResolver, and indirectly from
RuntimeDyld::MemoryManager via the new MCJITMemoryManager class, which
just subclasses RuntimeDyld::MemoryManager and reintroduces the
notifyObjectLoaded method for backwards compatibility).
The EngineBuilder class retains the existing method:
EngineBuilder&
setMCJITMemoryManager(std::unique_ptr<RTDyldMemoryManager> mcjmm);
and includes two new methods:
EngineBuilder&
setMemoryManager(std::unique_ptr<MCJITMemoryManager> MM);
EngineBuilder&
setSymbolResolver(std::unique_ptr<RuntimeDyld::SymbolResolver> SR);
Clients should use EITHER:
A single call to setMCJITMemoryManager with an RTDyldMemoryManager.
OR (exclusive)
One call each to each of setMemoryManager and setSymbolResolver.
This patch should be fully compatible with existing uses of RTDyldMemoryManager.
If it is not it should be considered a bug, and the patch either fixed or
reverted.
If clients find the new API to be an improvement the goal will be to deprecate
and eventually remove the RTDyldMemoryManager class in favor of the new classes.
llvm-svn: 233509
99 lines
3.0 KiB
C++
99 lines
3.0 KiB
C++
//===--- OrcLazyJIT.h - Basic Orc-based JIT for lazy execution --*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Simple Orc-based JIT. Uses the compile-on-demand layer to break up and
|
|
// lazily compile modules.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_TOOLS_LLI_ORCLAZYJIT_H
|
|
#define LLVM_TOOLS_LLI_ORCLAZYJIT_H
|
|
|
|
#include "llvm/ADT/Triple.h"
|
|
#include "llvm/ExecutionEngine/Orc/CompileOnDemandLayer.h"
|
|
#include "llvm/ExecutionEngine/Orc/CompileUtils.h"
|
|
#include "llvm/ExecutionEngine/Orc/IRCompileLayer.h"
|
|
#include "llvm/ExecutionEngine/Orc/LazyEmittingLayer.h"
|
|
#include "llvm/ExecutionEngine/Orc/ObjectLinkingLayer.h"
|
|
#include "llvm/IR/LLVMContext.h"
|
|
|
|
namespace llvm {
|
|
|
|
class OrcLazyJIT {
|
|
public:
|
|
|
|
typedef orc::JITCompileCallbackManagerBase CompileCallbackMgr;
|
|
typedef orc::ObjectLinkingLayer<> ObjLayerT;
|
|
typedef orc::IRCompileLayer<ObjLayerT> CompileLayerT;
|
|
typedef orc::LazyEmittingLayer<CompileLayerT> LazyEmitLayerT;
|
|
typedef orc::CompileOnDemandLayer<LazyEmitLayerT,
|
|
CompileCallbackMgr> CODLayerT;
|
|
typedef CODLayerT::ModuleSetHandleT ModuleHandleT;
|
|
|
|
OrcLazyJIT(std::unique_ptr<TargetMachine> TM, LLVMContext &Context)
|
|
: Error(false), TM(std::move(TM)),
|
|
Mang(this->TM->getDataLayout()),
|
|
ObjectLayer(),
|
|
CompileLayer(ObjectLayer, orc::SimpleCompiler(*this->TM)),
|
|
LazyEmitLayer(CompileLayer),
|
|
CCMgr(createCallbackMgr(Triple(this->TM->getTargetTriple()), Context)),
|
|
CODLayer(LazyEmitLayer, *CCMgr) { }
|
|
|
|
bool Ok() const { return !Error; }
|
|
|
|
ModuleHandleT addModule(std::unique_ptr<Module> M) {
|
|
// Attach a data-layout if one isn't already present.
|
|
if (M->getDataLayout().isDefault())
|
|
M->setDataLayout(*TM->getDataLayout());
|
|
|
|
std::vector<std::unique_ptr<Module>> S;
|
|
S.push_back(std::move(M));
|
|
return CODLayer.addModuleSet(std::move(S));
|
|
}
|
|
|
|
orc::JITSymbol findSymbol(const std::string &Name) {
|
|
return CODLayer.findSymbol(mangle(Name), true);
|
|
}
|
|
|
|
orc::JITSymbol findSymbolIn(ModuleHandleT H, const std::string &Name) {
|
|
return CODLayer.findSymbolIn(H, mangle(Name), true);
|
|
}
|
|
|
|
private:
|
|
|
|
std::unique_ptr<CompileCallbackMgr>
|
|
createCallbackMgr(Triple T, LLVMContext &Context);
|
|
|
|
std::string mangle(const std::string &Name) {
|
|
std::string MangledName;
|
|
{
|
|
raw_string_ostream MangledNameStream(MangledName);
|
|
Mang.getNameWithPrefix(MangledNameStream, Name);
|
|
}
|
|
return MangledName;
|
|
}
|
|
|
|
bool Error;
|
|
std::unique_ptr<TargetMachine> TM;
|
|
Mangler Mang;
|
|
SectionMemoryManager CCMgrMemMgr;
|
|
|
|
ObjLayerT ObjectLayer;
|
|
CompileLayerT CompileLayer;
|
|
LazyEmitLayerT LazyEmitLayer;
|
|
std::unique_ptr<CompileCallbackMgr> CCMgr;
|
|
CODLayerT CODLayer;
|
|
};
|
|
|
|
int runOrcLazyJIT(std::unique_ptr<Module> M, int ArgC, char* ArgV[]);
|
|
|
|
} // end namespace llvm
|
|
|
|
#endif
|