jit_utils.hpp

#pragma once
 
#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/JITTargetMachineBuilder.h"
#include "llvm/ExecutionEngine/Orc/RTDyldObjectLinkingLayer.h"
#include "llvm/ExecutionEngine/SectionMemoryManager.h"
#include "llvm/IR/DataLayout.h"
#include "llvm/IR/LLVMContext.h"
#include "llvm/Support/DynamicLibrary.h"
#include "llvm/Support/Error.h"
#include "llvm/Support/raw_ostream.h"
#include <memory>
 
using namespace llvm;
using namespace llvm::orc;
 
namespace qcor {
 
class XACCJIT {
private:
  ExecutionSession ES;
  RTDyldObjectLinkingLayer ObjectLayer;
  IRCompileLayer CompileLayer;
 
  DataLayout DL;
  MangleAndInterner Mangle;
  ThreadSafeContext Ctx;
 
  JITDylib &MainJD;
 
public:
  XACCJIT(JITTargetMachineBuilder JTMB, DataLayout DL)
      : ObjectLayer(ES,
                    []() { return std::make_unique<SectionMemoryManager>(); }),
        CompileLayer(ES, ObjectLayer, ConcurrentIRCompiler(std::move(JTMB))),
        DL(std::move(DL)), Mangle(ES, this->DL),
        Ctx(std::make_unique<LLVMContext>()),
        MainJD(ES.createJITDylib("<main>")) {
    MainJD.addGenerator(
        cantFail(DynamicLibrarySearchGenerator::GetForCurrentProcess(
            DL.getGlobalPrefix())));
    llvm::sys::DynamicLibrary::LoadLibraryPermanently(nullptr);
  }
 
  static Expected<std::unique_ptr<XACCJIT>> Create() {
    auto JTMB = JITTargetMachineBuilder::detectHost();
 
    if (!JTMB)
      return JTMB.takeError();
 
    auto DL = JTMB->getDefaultDataLayoutForTarget();
    if (!DL)
      return DL.takeError();
 
    return std::make_unique<XACCJIT>(std::move(*JTMB), std::move(*DL));
  }
 
  const DataLayout &getDataLayout() const { return DL; }
 
  LLVMContext &getContext() { return *Ctx.getContext(); }
 
  Error addModule(std::unique_ptr<Module> M) {
 
    MainJD.addGenerator(cantFail(DynamicLibrarySearchGenerator::Load(
        "/home/cades/.xacc/lib/libxacc.so", DL.getGlobalPrefix())));
    MainJD.addGenerator(cantFail(DynamicLibrarySearchGenerator::Load(
        "/home/cades/.xacc/lib/libqrt.so", DL.getGlobalPrefix())));
    MainJD.addGenerator(cantFail(DynamicLibrarySearchGenerator::Load(
        "/home/cades/.xacc/lib/libqcor.so", DL.getGlobalPrefix())));
    MainJD.addGenerator(cantFail(DynamicLibrarySearchGenerator::Load(
        "/home/cades/.xacc/lib/libCppMicroServices.so", DL.getGlobalPrefix())));
 
    return CompileLayer.add(MainJD, ThreadSafeModule(std::move(M), Ctx));
  }
 
  Expected<JITEvaluatedSymbol> lookup(StringRef Name) {
    return ES.lookup({&MainJD}, Mangle(Name.str()));
  }
 
 
};
 
} // namespace qcor