mirror of
https://github.com/capstone-engine/llvm-capstone.git
synced 2025-04-15 21:01:29 +00:00
98440293fb
CompileOnDemandLayer2 now supports user-supplied partition functions (the original CompileOnDemandLayer already supported these). Partition functions are called with the list of requested global values (i.e. global values that currently have queries waiting on them) and have an opportunity to select extra global values to materialize at the same time. Also adds testing infrastructure for the new feature to lli. llvm-svn: 343396
66 lines
2.0 KiB
C++
66 lines
2.0 KiB
C++
//===-- ThreadSafeModule.cpp - Thread safe Module, Context, and Utilities
|
|
//h-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/ExecutionEngine/Orc/ThreadSafeModule.h"
|
|
#include "llvm/Bitcode/BitcodeReader.h"
|
|
#include "llvm/Bitcode/BitcodeWriter.h"
|
|
#include "llvm/Transforms/Utils/Cloning.h"
|
|
|
|
namespace llvm {
|
|
namespace orc {
|
|
|
|
ThreadSafeModule cloneToNewContext(ThreadSafeModule &TSM,
|
|
GVPredicate ShouldCloneDef,
|
|
GVModifier UpdateClonedDefSource) {
|
|
assert(TSM && "Can not clone null module");
|
|
|
|
if (!ShouldCloneDef)
|
|
ShouldCloneDef = [](const GlobalValue &) { return true; };
|
|
|
|
auto Lock = TSM.getContextLock();
|
|
|
|
SmallVector<char, 1> ClonedModuleBuffer;
|
|
|
|
{
|
|
std::set<GlobalValue *> ClonedDefsInSrc;
|
|
ValueToValueMapTy VMap;
|
|
auto Tmp = CloneModule(*TSM.getModule(), VMap, [&](const GlobalValue *GV) {
|
|
if (ShouldCloneDef(*GV)) {
|
|
ClonedDefsInSrc.insert(const_cast<GlobalValue *>(GV));
|
|
return true;
|
|
}
|
|
return false;
|
|
});
|
|
|
|
if (UpdateClonedDefSource)
|
|
for (auto *GV : ClonedDefsInSrc)
|
|
UpdateClonedDefSource(*GV);
|
|
|
|
BitcodeWriter BCWriter(ClonedModuleBuffer);
|
|
|
|
BCWriter.writeModule(*Tmp);
|
|
BCWriter.writeSymtab();
|
|
BCWriter.writeStrtab();
|
|
}
|
|
|
|
MemoryBufferRef ClonedModuleBufferRef(
|
|
StringRef(ClonedModuleBuffer.data(), ClonedModuleBuffer.size()),
|
|
"cloned module buffer");
|
|
ThreadSafeContext NewTSCtx(llvm::make_unique<LLVMContext>());
|
|
|
|
auto ClonedModule =
|
|
cantFail(parseBitcodeFile(ClonedModuleBufferRef, *NewTSCtx.getContext()));
|
|
ClonedModule->setModuleIdentifier(TSM.getModule()->getName());
|
|
return ThreadSafeModule(std::move(ClonedModule), std::move(NewTSCtx));
|
|
}
|
|
|
|
} // end namespace orc
|
|
} // end namespace llvm
|