mirror of
https://github.com/RPCS3/llvm.git
synced 2026-01-31 01:25:19 +01:00
e87aaa1093
MaterializationResponsibility. VModuleKeys are intended to enable selective removal of modules from a JIT session, however for a wide variety of use cases selective removal is not needed and introduces unnecessary overhead. As of this commit, the default constructed VModuleKey value is reserved as a "do not track" value, and becomes the default when adding a new module to the JIT. This commit also changes the propagation of VModuleKeys. They were passed alongside the MaterializationResponsibity instance in XXLayer::emit methods, but are now propagated as part of the MaterializationResponsibility instance itself (and as part of MaterializationUnit when stored in a JITDylib). Associating VModuleKeys with MaterializationUnits in this way should allow for a thread-safe module removal mechanism in the future, even when a module is in the process of being compiled, by having the MaterializationResponsibility object check in on its VModuleKey's state before commiting its results to the JITDylib. git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@344643 91177308-0d34-0410-b5e6-96231b3b80d8
115 lines
4.2 KiB
C++
115 lines
4.2 KiB
C++
//===- ObjectTransformLayer.h - Run all objects through functor -*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Run all objects passed in through a user supplied functor.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_EXECUTIONENGINE_ORC_OBJECTTRANSFORMLAYER_H
|
|
#define LLVM_EXECUTIONENGINE_ORC_OBJECTTRANSFORMLAYER_H
|
|
|
|
#include "llvm/ExecutionEngine/JITSymbol.h"
|
|
#include "llvm/ExecutionEngine/Orc/Layer.h"
|
|
#include <algorithm>
|
|
#include <memory>
|
|
#include <string>
|
|
|
|
namespace llvm {
|
|
namespace orc {
|
|
|
|
class ObjectTransformLayer : public ObjectLayer {
|
|
public:
|
|
using TransformFunction =
|
|
std::function<Expected<std::unique_ptr<MemoryBuffer>>(
|
|
std::unique_ptr<MemoryBuffer>)>;
|
|
|
|
ObjectTransformLayer(ExecutionSession &ES, ObjectLayer &BaseLayer,
|
|
TransformFunction Transform);
|
|
|
|
void emit(MaterializationResponsibility R,
|
|
std::unique_ptr<MemoryBuffer> O) override;
|
|
|
|
private:
|
|
ObjectLayer &BaseLayer;
|
|
TransformFunction Transform;
|
|
};
|
|
|
|
/// Object mutating layer.
|
|
///
|
|
/// This layer accepts sets of ObjectFiles (via addObject). It
|
|
/// immediately applies the user supplied functor to each object, then adds
|
|
/// the set of transformed objects to the layer below.
|
|
template <typename BaseLayerT, typename TransformFtor>
|
|
class LegacyObjectTransformLayer {
|
|
public:
|
|
/// Construct an ObjectTransformLayer with the given BaseLayer
|
|
LegacyObjectTransformLayer(BaseLayerT &BaseLayer,
|
|
TransformFtor Transform = TransformFtor())
|
|
: BaseLayer(BaseLayer), Transform(std::move(Transform)) {}
|
|
|
|
/// Apply the transform functor to each object in the object set, then
|
|
/// add the resulting set of objects to the base layer, along with the
|
|
/// memory manager and symbol resolver.
|
|
///
|
|
/// @return A handle for the added objects.
|
|
template <typename ObjectPtr> Error addObject(VModuleKey K, ObjectPtr Obj) {
|
|
return BaseLayer.addObject(std::move(K), Transform(std::move(Obj)));
|
|
}
|
|
|
|
/// Remove the object set associated with the VModuleKey K.
|
|
Error removeObject(VModuleKey K) { return BaseLayer.removeObject(K); }
|
|
|
|
/// Search for the given named symbol.
|
|
/// @param Name The name of the symbol to search for.
|
|
/// @param ExportedSymbolsOnly If true, search only for exported symbols.
|
|
/// @return A handle for the given named symbol, if it exists.
|
|
JITSymbol findSymbol(const std::string &Name, bool ExportedSymbolsOnly) {
|
|
return BaseLayer.findSymbol(Name, ExportedSymbolsOnly);
|
|
}
|
|
|
|
/// Get the address of the given symbol in the context of the set of
|
|
/// objects represented by the VModuleKey K. This call is forwarded to
|
|
/// the base layer's implementation.
|
|
/// @param K The VModuleKey associated with the object set to search in.
|
|
/// @param Name The name of the symbol to search for.
|
|
/// @param ExportedSymbolsOnly If true, search only for exported symbols.
|
|
/// @return A handle for the given named symbol, if it is found in the
|
|
/// given object set.
|
|
JITSymbol findSymbolIn(VModuleKey K, const std::string &Name,
|
|
bool ExportedSymbolsOnly) {
|
|
return BaseLayer.findSymbolIn(K, Name, ExportedSymbolsOnly);
|
|
}
|
|
|
|
/// Immediately emit and finalize the object set represented by the
|
|
/// given VModuleKey K.
|
|
Error emitAndFinalize(VModuleKey K) { return BaseLayer.emitAndFinalize(K); }
|
|
|
|
/// Map section addresses for the objects associated with the
|
|
/// VModuleKey K.
|
|
void mapSectionAddress(VModuleKey K, const void *LocalAddress,
|
|
JITTargetAddress TargetAddr) {
|
|
BaseLayer.mapSectionAddress(K, LocalAddress, TargetAddr);
|
|
}
|
|
|
|
/// Access the transform functor directly.
|
|
TransformFtor &getTransform() { return Transform; }
|
|
|
|
/// Access the mumate functor directly.
|
|
const TransformFtor &getTransform() const { return Transform; }
|
|
|
|
private:
|
|
BaseLayerT &BaseLayer;
|
|
TransformFtor Transform;
|
|
};
|
|
|
|
} // end namespace orc
|
|
} // end namespace llvm
|
|
|
|
#endif // LLVM_EXECUTIONENGINE_ORC_OBJECTTRANSFORMLAYER_H
|