d6524c8dfa
This reapplies4f0325873f(and follow up patches26fc07d5d8,a001cc0e6c,c9bc242e38, andfd174f0ff3), which were reverted in212cdc9a37to investigate bot failures (e.g. https://lab.llvm.org/buildbot/#/builders/108/builds/8502) The fix to address the bot failures was landed ind0052ebbe2. 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
|