Home | History | Annotate | Download | only in Orc 
      1 //===-- CompileUtils.h - Utilities for compiling IR in the JIT --*- C++ -*-===//
      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 // Contains utilities for compiling IR to object files.
     11 //
     12 //===----------------------------------------------------------------------===//
     13 
     14 #ifndef LLVM_EXECUTIONENGINE_ORC_COMPILEUTILS_H
     15 #define LLVM_EXECUTIONENGINE_ORC_COMPILEUTILS_H
     16 
     17 #include "llvm/ExecutionEngine/ObjectMemoryBuffer.h"
     18 #include "llvm/IR/LegacyPassManager.h"
     19 #include "llvm/MC/MCContext.h"
     20 #include "llvm/Object/ObjectFile.h"
     21 #include "llvm/Target/TargetMachine.h"
     22 
     23 namespace llvm {
     24 namespace orc {
     25 
     26 /// @brief Simple compile functor: Takes a single IR module and returns an
     27 ///        ObjectFile.
     28 class SimpleCompiler {
     29 public:
     30   /// @brief Construct a simple compile functor with the given target.
     31   SimpleCompiler(TargetMachine &TM) : TM(TM) {}
     32 
     33   /// @brief Compile a Module to an ObjectFile.
     34   object::OwningBinary<object::ObjectFile> operator()(Module &M) const {
     35     SmallVector<char, 0> ObjBufferSV;
     36     raw_svector_ostream ObjStream(ObjBufferSV);
     37 
     38     legacy::PassManager PM;
     39     MCContext *Ctx;
     40     if (TM.addPassesToEmitMC(PM, Ctx, ObjStream))
     41       llvm_unreachable("Target does not support MC emission.");
     42     PM.run(M);
     43     std::unique_ptr<MemoryBuffer> ObjBuffer(
     44         new ObjectMemoryBuffer(std::move(ObjBufferSV)));
     45     Expected<std::unique_ptr<object::ObjectFile>> Obj =
     46         object::ObjectFile::createObjectFile(ObjBuffer->getMemBufferRef());
     47     typedef object::OwningBinary<object::ObjectFile> OwningObj;
     48     if (Obj)
     49       return OwningObj(std::move(*Obj), std::move(ObjBuffer));
     50     // TODO: Actually report errors helpfully.
     51     consumeError(Obj.takeError());
     52     return OwningObj(nullptr, nullptr);
     53   }
     54 
     55 private:
     56   TargetMachine &TM;
     57 };
     58 
     59 } // End namespace orc.
     60 } // End namespace llvm.
     61 
     62 #endif // LLVM_EXECUTIONENGINE_ORC_COMPILEUTILS_H
     63