proteus::JitEngineDevice< ImplT > Class Template Reference

#include <JitEngineDevice.h>

Inheritance diagram for proteus::JitEngineDevice< ImplT >:

Collaboration diagram for proteus::JitEngineDevice< ImplT >:

Public Types
using	DeviceError_t = typename DeviceTraits< ImplT >::DeviceError_t
using	DeviceStream_t = typename DeviceTraits< ImplT >::DeviceStream_t
using	KernelFunction_t = typename DeviceTraits< ImplT >::KernelFunction_t

Public Member Functions
DeviceError_t	compileAndRun (JITKernelInfo &KernelInfo, dim3 GridDim, dim3 BlockDim, void **KernelArgs, uint64_t ShmemSize, typename DeviceTraits< ImplT >::DeviceStream_t Stream)
std::pair< std::unique_ptr< Module >, std::unique_ptr< MemoryBuffer > >	extractKernelModule (BinaryInfo &BinInfo, StringRef KernelName, LLVMContext &Ctx)
void	extractModuleAndBitcode (JITKernelInfo &KernelInfo)
Module &	getModule (JITKernelInfo &KernelInfo)
MemoryBufferRef	getBitcode (JITKernelInfo &KernelInfo)
void	getLambdaJitValues (JITKernelInfo &KernelInfo, SmallVector< RuntimeConstant > &LambdaJitValuesVec)
void	registerVar (void *Handle, const char *VarName, const void *HostAddr, uint64_t VarSize)
void	registerLinkedBinary (void *Handle, FatbinWrapperT *FatbinWrapper, const char *ModuleId)
void	registerFatBinary (void *Handle, FatbinWrapperT *FatbinWrapper, const char *ModuleId)
void	finalizeRegistration ()
void	registerFunction (void *Handle, void *Kernel, char *KernelName, ArrayRef< RuntimeConstantInfo * > RCInfoArray)
bool	containsJITKernelInfo (const void *Func)
std::optional< std::reference_wrapper< JITKernelInfo > >	getJITKernelInfo (const void *Func)
HashT	getStaticHash (JITKernelInfo &KernelInfo)
StringRef	getDeviceArch () const
Public Member Functions inherited from proteus::JitEngine
bool	isProteusDisabled ()
void	enable ()
void	disable ()

Public Attributes
std::unordered_map< std::string, FatbinWrapperT * >	ModuleIdToFatBinary
std::unordered_map< const void *, BinaryInfo >	HandleToBinaryInfo
SmallPtrSet< void *, 8 >	GlobalLinkedBinaries
Public Attributes inherited from proteus::JitEngine
InitLLVMTargets	Init

Protected Member Functions
	JitEngineDevice ()
	~JitEngineDevice ()
Protected Member Functions inherited from proteus::JitEngine
SmallVector< RuntimeConstant >	getRuntimeConstantValues (void **KernelArgs, ArrayRef< RuntimeConstantInfo * > RCInfoArray)
	JitEngine ()

Protected Attributes
MemoryCache< KernelFunction_t >	CodeCache {"JitEngineDevice"}
std::optional< ObjectCacheChain >	CacheChain
std::string	DeviceArch
DenseMap< const void *, JITKernelInfo >	JITKernelInfoMap
std::unique_ptr< CompilerAsync >	AsyncCompiler

Member Typedef Documentation

DeviceError_t

template<typename ImplT >

using proteus::JitEngineDevice< ImplT >::DeviceError_t = typename DeviceTraits<ImplT>::DeviceError_t

DeviceStream_t

template<typename ImplT >

using proteus::JitEngineDevice< ImplT >::DeviceStream_t = typename DeviceTraits<ImplT>::DeviceStream_t

KernelFunction_t

template<typename ImplT >

using proteus::JitEngineDevice< ImplT >::KernelFunction_t = typename DeviceTraits<ImplT>::KernelFunction_t

Constructor & Destructor Documentation

JitEngineDevice()

template<typename ImplT >

proteus::JitEngineDevice< ImplT >::JitEngineDevice ( )

inlineprotected

~JitEngineDevice()

template<typename ImplT >

proteus::JitEngineDevice< ImplT >::~JitEngineDevice ( )

inlineprotected

Member Function Documentation

compileAndRun()

template<typename ImplT >

DeviceTraits< ImplT >::DeviceError_t proteus::JitEngineDevice< ImplT >::compileAndRun	(	JITKernelInfo &	KernelInfo,
		dim3	GridDim,
		dim3	BlockDim,
		void **	KernelArgs,
		uint64_t	ShmemSize,
		typename DeviceTraits< ImplT >::DeviceStream_t	Stream
	)

containsJITKernelInfo()

template<typename ImplT >

bool proteus::JitEngineDevice< ImplT >::containsJITKernelInfo ( const void *  Func )

inline

extractKernelModule()

template<typename ImplT >

std::pair< std::unique_ptr< Module >, std::unique_ptr< MemoryBuffer > > proteus::JitEngineDevice< ImplT >::extractKernelModule ( BinaryInfo &  BinInfo, StringRef  KernelName, LLVMContext &  Ctx  )

inline

extractModuleAndBitcode()

template<typename ImplT >

void proteus::JitEngineDevice< ImplT >::extractModuleAndBitcode ( JITKernelInfo &  KernelInfo )

inline

finalizeRegistration()

template<typename ImplT >

void proteus::JitEngineDevice< ImplT >::finalizeRegistration

(

)

getBitcode()

template<typename ImplT >

MemoryBufferRef proteus::JitEngineDevice< ImplT >::getBitcode ( JITKernelInfo &  KernelInfo )

inline

getDeviceArch()

template<typename ImplT >

StringRef proteus::JitEngineDevice< ImplT >::getDeviceArch ( ) const

inline

getJITKernelInfo()

template<typename ImplT >

std::optional< std::reference_wrapper< JITKernelInfo > > proteus::JitEngineDevice< ImplT >::getJITKernelInfo ( const void *  Func )

inline

getLambdaJitValues()

template<typename ImplT >

void proteus::JitEngineDevice< ImplT >::getLambdaJitValues ( JITKernelInfo &  KernelInfo, SmallVector< RuntimeConstant > &  LambdaJitValuesVec  )

inline

getModule()

template<typename ImplT >

Module & proteus::JitEngineDevice< ImplT >::getModule ( JITKernelInfo &  KernelInfo )

inline

getStaticHash()

template<typename ImplT >

HashT proteus::JitEngineDevice< ImplT >::getStaticHash ( JITKernelInfo &  KernelInfo )

inline

registerFatBinary()

template<typename ImplT >

void proteus::JitEngineDevice< ImplT >::registerFatBinary	(	void *	Handle,
		FatbinWrapperT *	FatbinWrapper,
		const char *	ModuleId
	)

registerFunction()

template<typename ImplT >

void proteus::JitEngineDevice< ImplT >::registerFunction	(	void *	Handle,
		void *	Kernel,
		char *	KernelName,
		ArrayRef< RuntimeConstantInfo * >	RCInfoArray
	)

registerLinkedBinary()

template<typename ImplT >

void proteus::JitEngineDevice< ImplT >::registerLinkedBinary	(	void *	Handle,
		FatbinWrapperT *	FatbinWrapper,
		const char *	ModuleId
	)

registerVar()

template<typename ImplT >

void proteus::JitEngineDevice< ImplT >::registerVar ( void *  Handle, const char *  VarName, const void *  HostAddr, uint64_t  VarSize  )

inline

Member Data Documentation

AsyncCompiler

template<typename ImplT >

std::unique_ptr<CompilerAsync> proteus::JitEngineDevice< ImplT >::AsyncCompiler

protected

CacheChain

template<typename ImplT >

std::optional<ObjectCacheChain> proteus::JitEngineDevice< ImplT >::CacheChain

protected

CodeCache

template<typename ImplT >

MemoryCache<KernelFunction_t> proteus::JitEngineDevice< ImplT >::CodeCache {"JitEngineDevice"}

protected

DeviceArch

template<typename ImplT >

std::string proteus::JitEngineDevice< ImplT >::DeviceArch

protected

GlobalLinkedBinaries

template<typename ImplT >

SmallPtrSet<void *, 8> proteus::JitEngineDevice< ImplT >::GlobalLinkedBinaries

HandleToBinaryInfo

template<typename ImplT >

std::unordered_map<const void *, BinaryInfo> proteus::JitEngineDevice< ImplT >::HandleToBinaryInfo

JITKernelInfoMap

template<typename ImplT >

DenseMap<const void *, JITKernelInfo> proteus::JitEngineDevice< ImplT >::JITKernelInfoMap

protected

ModuleIdToFatBinary

template<typename ImplT >

std::unordered_map<std::string, FatbinWrapperT *> proteus::JitEngineDevice< ImplT >::ModuleIdToFatBinary

---

The documentation for this class was generated from the following file:

• src/include/proteus/impl/JitEngineDevice.h