141 lines
4.6 KiB
C++
141 lines
4.6 KiB
C++
//===- CompileOnDemandLayer.h - Compile each function on demand -*- C++ -*-===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// JIT layer for breaking up modules and inserting callbacks to allow
|
|
// individual functions to be compiled on demand.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_EXECUTIONENGINE_ORC_COMPILEONDEMANDLAYER_H
|
|
#define LLVM_EXECUTIONENGINE_ORC_COMPILEONDEMANDLAYER_H
|
|
|
|
#include "llvm/ADT/APInt.h"
|
|
#include "llvm/ADT/Optional.h"
|
|
#include "llvm/ADT/STLExtras.h"
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/ExecutionEngine/JITSymbol.h"
|
|
#include "llvm/ExecutionEngine/Orc/IndirectionUtils.h"
|
|
#include "llvm/ExecutionEngine/Orc/Layer.h"
|
|
#include "llvm/ExecutionEngine/Orc/LazyReexports.h"
|
|
#include "llvm/ExecutionEngine/Orc/Speculation.h"
|
|
#include "llvm/ExecutionEngine/Orc/Shared/OrcError.h"
|
|
#include "llvm/ExecutionEngine/RuntimeDyld.h"
|
|
#include "llvm/IR/Attributes.h"
|
|
#include "llvm/IR/Constant.h"
|
|
#include "llvm/IR/Constants.h"
|
|
#include "llvm/IR/DataLayout.h"
|
|
#include "llvm/IR/Function.h"
|
|
#include "llvm/IR/GlobalAlias.h"
|
|
#include "llvm/IR/GlobalValue.h"
|
|
#include "llvm/IR/GlobalVariable.h"
|
|
#include "llvm/IR/Instruction.h"
|
|
#include "llvm/IR/Mangler.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/IR/Type.h"
|
|
#include "llvm/Support/Casting.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include "llvm/Transforms/Utils/ValueMapper.h"
|
|
#include <algorithm>
|
|
#include <cassert>
|
|
#include <functional>
|
|
#include <iterator>
|
|
#include <list>
|
|
#include <memory>
|
|
#include <set>
|
|
#include <string>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
namespace llvm {
|
|
|
|
class Value;
|
|
|
|
namespace orc {
|
|
|
|
class ExtractingIRMaterializationUnit;
|
|
|
|
class CompileOnDemandLayer : public IRLayer {
|
|
friend class PartitioningIRMaterializationUnit;
|
|
|
|
public:
|
|
/// Builder for IndirectStubsManagers.
|
|
using IndirectStubsManagerBuilder =
|
|
std::function<std::unique_ptr<IndirectStubsManager>()>;
|
|
|
|
using GlobalValueSet = std::set<const GlobalValue *>;
|
|
|
|
/// Partitioning function.
|
|
using PartitionFunction =
|
|
std::function<Optional<GlobalValueSet>(GlobalValueSet Requested)>;
|
|
|
|
/// Off-the-shelf partitioning which compiles all requested symbols (usually
|
|
/// a single function at a time).
|
|
static Optional<GlobalValueSet> compileRequested(GlobalValueSet Requested);
|
|
|
|
/// Off-the-shelf partitioning which compiles whole modules whenever any
|
|
/// symbol in them is requested.
|
|
static Optional<GlobalValueSet> compileWholeModule(GlobalValueSet Requested);
|
|
|
|
/// Construct a CompileOnDemandLayer.
|
|
CompileOnDemandLayer(ExecutionSession &ES, IRLayer &BaseLayer,
|
|
LazyCallThroughManager &LCTMgr,
|
|
IndirectStubsManagerBuilder BuildIndirectStubsManager);
|
|
|
|
/// Sets the partition function.
|
|
void setPartitionFunction(PartitionFunction Partition);
|
|
|
|
/// Sets the ImplSymbolMap
|
|
void setImplMap(ImplSymbolMap *Imp);
|
|
|
|
/// Emits the given module. This should not be called by clients: it will be
|
|
/// called by the JIT when a definition added via the add method is requested.
|
|
void emit(std::unique_ptr<MaterializationResponsibility> R,
|
|
ThreadSafeModule TSM) override;
|
|
|
|
private:
|
|
struct PerDylibResources {
|
|
public:
|
|
PerDylibResources(JITDylib &ImplD,
|
|
std::unique_ptr<IndirectStubsManager> ISMgr)
|
|
: ImplD(ImplD), ISMgr(std::move(ISMgr)) {}
|
|
JITDylib &getImplDylib() { return ImplD; }
|
|
IndirectStubsManager &getISManager() { return *ISMgr; }
|
|
|
|
private:
|
|
JITDylib &ImplD;
|
|
std::unique_ptr<IndirectStubsManager> ISMgr;
|
|
};
|
|
|
|
using PerDylibResourcesMap = std::map<const JITDylib *, PerDylibResources>;
|
|
|
|
PerDylibResources &getPerDylibResources(JITDylib &TargetD);
|
|
|
|
void cleanUpModule(Module &M);
|
|
|
|
void expandPartition(GlobalValueSet &Partition);
|
|
|
|
void emitPartition(std::unique_ptr<MaterializationResponsibility> R,
|
|
ThreadSafeModule TSM,
|
|
IRMaterializationUnit::SymbolNameToDefinitionMap Defs);
|
|
|
|
mutable std::mutex CODLayerMutex;
|
|
|
|
IRLayer &BaseLayer;
|
|
LazyCallThroughManager &LCTMgr;
|
|
IndirectStubsManagerBuilder BuildIndirectStubsManager;
|
|
PerDylibResourcesMap DylibResources;
|
|
PartitionFunction Partition = compileRequested;
|
|
SymbolLinkagePromoter PromoteSymbols;
|
|
ImplSymbolMap *AliaseeImpls = nullptr;
|
|
};
|
|
|
|
} // end namespace orc
|
|
} // end namespace llvm
|
|
|
|
#endif // LLVM_EXECUTIONENGINE_ORC_COMPILEONDEMANDLAYER_H
|