1 //===--------------- IRCompileLayer.cpp - IR Compiling Layer --------------===//
2 //
3 // The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9
10 #include "llvm/ExecutionEngine/Orc/IRCompileLayer.h"
11
12 namespace llvm {
13 namespace orc {
14
IRCompileLayer(ExecutionSession & ES,ObjectLayer & BaseLayer,CompileFunction Compile)15 IRCompileLayer::IRCompileLayer(ExecutionSession &ES, ObjectLayer &BaseLayer,
16 CompileFunction Compile)
17 : IRLayer(ES), BaseLayer(BaseLayer), Compile(std::move(Compile)) {}
18
setNotifyCompiled(NotifyCompiledFunction NotifyCompiled)19 void IRCompileLayer::setNotifyCompiled(NotifyCompiledFunction NotifyCompiled) {
20 std::lock_guard<std::mutex> Lock(IRLayerMutex);
21 this->NotifyCompiled = std::move(NotifyCompiled);
22 }
23
emit(MaterializationResponsibility R,ThreadSafeModule TSM)24 void IRCompileLayer::emit(MaterializationResponsibility R,
25 ThreadSafeModule TSM) {
26 assert(TSM.getModule() && "Module must not be null");
27
28 if (auto Obj = Compile(*TSM.getModule())) {
29 {
30 std::lock_guard<std::mutex> Lock(IRLayerMutex);
31 if (NotifyCompiled)
32 NotifyCompiled(R.getVModuleKey(), std::move(TSM));
33 else
34 TSM = ThreadSafeModule();
35 }
36 BaseLayer.emit(std::move(R), std::move(*Obj));
37 } else {
38 R.failMaterialization();
39 getExecutionSession().reportError(Obj.takeError());
40 }
41 }
42
43 } // End namespace orc.
44 } // End namespace llvm.
45