llvm_jit.h

Go to the documentation of this file.

#pragma once

#include "llvm/ExecutionEngine/Orc/CompileUtils.h"
#include "llvm/ExecutionEngine/Orc/IRCompileLayer.h"
#include "llvm/ExecutionEngine/Orc/IRTransformLayer.h"
#include "llvm/ExecutionEngine/Orc/RTDyldObjectLinkingLayer.h"
#include "llvm/Target/TargetMachine.h"

namespace tc {

namespace polyhedral {
class Scop;
}

class Jit {
 private:
  std::unique_ptr<llvm::TargetMachine> TM_;
  const llvm::DataLayout DL_;
  llvm::orc::RTDyldObjectLinkingLayer objectLayer_;
  llvm::orc::IRCompileLayer<decltype(objectLayer_), llvm::orc::SimpleCompiler>
      compileLayer_;

 public:
  Jit();

  void codegenScop(
      const std::string& specializedName,
      const polyhedral::Scop& scop);

  using ModuleHandle = decltype(compileLayer_)::ModuleHandleT;
  ModuleHandle addModule(std::unique_ptr<llvm::Module> M);
  void removeModule(ModuleHandle H);

  llvm::JITSymbol findSymbol(const std::string name);
  llvm::JITTargetAddress getSymbolAddress(const std::string name);

  llvm::TargetMachine& getTargetMachine();
};

} // namespace tc