contrib/llvm/lib/ExecutionEngine/Orc/IRCompileLayer.cpp

   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
  15 IRCompileLayer::IRCompileLayer(ExecutionSession &ES, ObjectLayer &BaseLayer,
  16                                  CompileFunction Compile)
  17     : IRLayer(ES), BaseLayer(BaseLayer), Compile(std::move(Compile)) {}
  18
  19 void IRCompileLayer::setNotifyCompiled(NotifyCompiledFunction NotifyCompiled) {
  20   std::lock_guard<std::mutex> Lock(IRLayerMutex);
  21   this->NotifyCompiled = std::move(NotifyCompiled);
  22 }
  23
  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.