d6524c8dfa
This reapplies 4f0325873fa (and follow up patches 26fc07d5d88, a001cc0e6cdc, c9bc242e387, and fd174f0ff3e), which were reverted in 212cdc9a377 to investigate bot failures (e.g. https://lab.llvm.org/buildbot/#/builders/108/builds/8502) The fix to address the bot failures was landed in d0052ebbe2e. This patch also restricts construction of the UnwindInfoManager object to Apple platforms (as it won't be used on other platforms).
97 lines
2.9 KiB
C++
97 lines
2.9 KiB
C++
//===------ CompileUtils.cpp - Utilities for compiling IR in the JIT ------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/ExecutionEngine/Orc/CompileUtils.h"
|
|
|
|
#include "llvm/ADT/SmallVector.h"
|
|
#include "llvm/ExecutionEngine/ObjectCache.h"
|
|
#include "llvm/IR/LegacyPassManager.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/MC/MCContext.h"
|
|
#include "llvm/Object/ObjectFile.h"
|
|
#include "llvm/Support/Error.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
#include "llvm/Support/SmallVectorMemoryBuffer.h"
|
|
#include "llvm/Target/TargetMachine.h"
|
|
|
|
namespace llvm {
|
|
namespace orc {
|
|
|
|
IRSymbolMapper::ManglingOptions
|
|
irManglingOptionsFromTargetOptions(const TargetOptions &Opts) {
|
|
IRSymbolMapper::ManglingOptions MO;
|
|
|
|
MO.EmulatedTLS = Opts.EmulatedTLS;
|
|
|
|
return MO;
|
|
}
|
|
|
|
/// Compile a Module to an ObjectFile.
|
|
Expected<SimpleCompiler::CompileResult> SimpleCompiler::operator()(Module &M) {
|
|
if (M.getDataLayout().isDefault())
|
|
M.setDataLayout(TM.createDataLayout());
|
|
|
|
CompileResult CachedObject = tryToLoadFromObjectCache(M);
|
|
if (CachedObject)
|
|
return std::move(CachedObject);
|
|
|
|
SmallVector<char, 0> ObjBufferSV;
|
|
|
|
{
|
|
raw_svector_ostream ObjStream(ObjBufferSV);
|
|
|
|
legacy::PassManager PM;
|
|
MCContext *Ctx;
|
|
if (TM.addPassesToEmitMC(PM, Ctx, ObjStream))
|
|
return make_error<StringError>("Target does not support MC emission",
|
|
inconvertibleErrorCode());
|
|
PM.run(M);
|
|
}
|
|
|
|
auto ObjBuffer = std::make_unique<SmallVectorMemoryBuffer>(
|
|
std::move(ObjBufferSV), M.getModuleIdentifier() + "-jitted-objectbuffer",
|
|
/*RequiresNullTerminator=*/false);
|
|
|
|
auto Obj = object::ObjectFile::createObjectFile(ObjBuffer->getMemBufferRef());
|
|
|
|
if (!Obj)
|
|
return Obj.takeError();
|
|
|
|
notifyObjectCompiled(M, *ObjBuffer);
|
|
return std::move(ObjBuffer);
|
|
}
|
|
|
|
SimpleCompiler::CompileResult
|
|
SimpleCompiler::tryToLoadFromObjectCache(const Module &M) {
|
|
if (!ObjCache)
|
|
return CompileResult();
|
|
|
|
return ObjCache->getObject(&M);
|
|
}
|
|
|
|
void SimpleCompiler::notifyObjectCompiled(const Module &M,
|
|
const MemoryBuffer &ObjBuffer) {
|
|
if (ObjCache)
|
|
ObjCache->notifyObjectCompiled(&M, ObjBuffer.getMemBufferRef());
|
|
}
|
|
|
|
ConcurrentIRCompiler::ConcurrentIRCompiler(JITTargetMachineBuilder JTMB,
|
|
ObjectCache *ObjCache)
|
|
: IRCompiler(irManglingOptionsFromTargetOptions(JTMB.getOptions())),
|
|
JTMB(std::move(JTMB)), ObjCache(ObjCache) {}
|
|
|
|
Expected<std::unique_ptr<MemoryBuffer>>
|
|
ConcurrentIRCompiler::operator()(Module &M) {
|
|
auto TM = cantFail(JTMB.createTargetMachine());
|
|
SimpleCompiler C(*TM, ObjCache);
|
|
return C(M);
|
|
}
|
|
|
|
} // end namespace orc
|
|
} // end namespace llvm
|