/
machine_compiler.hpp
122 lines (98 loc) · 3.45 KB
/
machine_compiler.hpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
#ifndef RBX_MACHINE_JIT_HPP
#define RBX_MACHINE_JIT_HPP
#include <llvm/Config/llvm-config.h>
#if LLVM_VERSION_MAJOR == 9
#include <llvm/ADT/StringRef.h>
#include <llvm/ExecutionEngine/JITSymbol.h>
#include <llvm/ExecutionEngine/Orc/CompileUtils.h>
#include <llvm/ExecutionEngine/Orc/Core.h>
#include <llvm/ExecutionEngine/Orc/ExecutionUtils.h>
#include <llvm/ExecutionEngine/Orc/IRCompileLayer.h>
#include <llvm/ExecutionEngine/Orc/IRTransformLayer.h>
#include <llvm/ExecutionEngine/Orc/JITTargetMachineBuilder.h>
#include <llvm/ExecutionEngine/Orc/RTDyldObjectLinkingLayer.h>
#include <llvm/ExecutionEngine/Orc/ThreadSafeModule.h>
#include <llvm/ExecutionEngine/SectionMemoryManager.h>
#include <llvm/IR/DataLayout.h>
#include <llvm/IR/LLVMContext.h>
#include <llvm/IR/LegacyPassManager.h>
#include <llvm/Transforms/InstCombine/InstCombine.h>
#include <llvm/Transforms/Scalar.h>
#include <llvm/Transforms/Scalar/GVN.h>
#endif
#include "machine_thread.hpp"
#include <list>
#include <memory>
#include <mutex>
#include <condition_variable>
namespace rubinius {
namespace jit {
#if LLVM_VERSION_MAJOR == 9
using namespace llvm;
using namespace llvm::orc;
#endif
class MachineCompiler {
public:
MachineCompiler(STATE);
virtual ~MachineCompiler() { }
#if LLVM_VERSION_MAJOR == 9
ExecutionSession ES;
RTDyldObjectLinkingLayer ObjectLayer;
IRCompileLayer CompileLayer;
IRTransformLayer OptimizeLayer;
DataLayout DL;
MangleAndInterner Mangle;
ThreadSafeContext Ctx;
public:
Compiler(JITTargetMachineBuilder JTMB, DataLayout DL)
: ObjectLayer(ES, []() { return llvm::make_unique<SectionMemoryManager>(); })
, CompileLayer(ES, ObjectLayer, ConcurrentIRCompiler(std::move(JTMB)))
, OptimizeLayer(ES, CompileLayer, optimizeModule)
, DL(std::move(DL)), Mangle(ES, this->DL)
, Ctx(llvm::make_unique<LLVMContext>())
{
ES.getMainJITDylib().setGenerator(
cantFail(DynamicLibrarySearchGenerator::GetForCurrentProcess(
DL.getGlobalPrefix())));
}
virtual ~Compiler() { }
const DataLayout &getDataLayout() const { return DL; }
LLVMContext &getContext() { return *Ctx.getContext(); }
static Expected<std::unique_ptr<Compiler>> Create() {
auto JTMB = JITTargetMachineBuilder::detectHost();
if (!JTMB)
return JTMB.takeError();
auto DL = JTMB->getDefaultDataLayoutForTarget();
if (!DL)
return DL.takeError();
return llvm::make_unique<Compiler>(std::move(*JTMB), std::move(*DL));
}
Error addModule(std::unique_ptr<llvm::Module> M) {
return OptimizeLayer.add(ES.getMainJITDylib(),
ThreadSafeModule(std::move(M), Ctx));
}
Expected<JITEvaluatedSymbol> lookup(StringRef Name) {
return ES.lookup({&ES.getMainJITDylib()}, Mangle(Name.str()));
}
private:
static Expected<ThreadSafeModule>
optimizeModule(ThreadSafeModule TSM, const MaterializationResponsibility &R) {
// Create a function pass manager.
auto FPM = llvm::make_unique<legacy::FunctionPassManager>(TSM.getModule());
// Add some optimizations.
FPM->add(createInstructionCombiningPass());
FPM->add(createReassociatePass());
FPM->add(createGVNPass());
FPM->add(createCFGSimplificationPass());
FPM->doInitialization();
// Run the optimizations over all functions in the module being added to
// the JIT.
for (auto &F : *TSM.getModule())
FPM->run(F);
return TSM;
}
#endif
};
}
}
#endif