proteus::JitEngineDevice< ImplT > Class Template Reference

#include <JitEngineDevice.hpp>

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	insertRegisterVar (const char *VarName, const void *Addr)
void	registerLinkedBinary (FatbinWrapperT *FatbinWrapper, const char *ModuleId)
void	registerFatBinary (void *Handle, FatbinWrapperT *FatbinWrapper, const char *ModuleId)
void	registerFatBinaryEnd ()
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)
void	finalize ()
Public Member Functions inherited from proteus::JitEngine
void	optimizeIR (Module &M, StringRef Arch, char OptLevel='3', unsigned CodegenOptLevel=3)
bool	isProteusDisabled ()
void	enable ()
void	disable ()

Public Attributes
void *	CurHandle = nullptr
std::unordered_map< std::string, FatbinWrapperT * >	ModuleIdToFatBinary
std::unordered_map< const void *, BinaryInfo >	HandleToBinaryInfo
SmallVector< std::string >	GlobalLinkedModuleIds
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 ()
std::string	mangleSuffix (HashT &HashValue)

Protected Attributes
JitCache< KernelFunction_t >	CodeCache
JitStorageCache< KernelFunction_t >	StorageCache
std::string	DeviceArch
std::unordered_map< std::string, const void * >	VarNameToDevPtr
DenseMap< const void *, JITKernelInfo >	JITKernelInfoMap

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

finalize()

template<typename ImplT >

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

inline

getBitcode()

template<typename ImplT >

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

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

insertRegisterVar()

template<typename ImplT >

void proteus::JitEngineDevice< ImplT >::insertRegisterVar ( const char *  VarName, const void *  Addr  )

inline

registerFatBinary()

template<typename ImplT >

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

registerFatBinaryEnd()

template<typename ImplT >

void proteus::JitEngineDevice< ImplT >::registerFatBinaryEnd

(

)

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	(	FatbinWrapperT *	FatbinWrapper,
		const char *	ModuleId
	)

Member Data Documentation

CodeCache

template<typename ImplT >

JitCache<KernelFunction_t> proteus::JitEngineDevice< ImplT >::CodeCache

protected

CurHandle

template<typename ImplT >

void* proteus::JitEngineDevice< ImplT >::CurHandle = nullptr

DeviceArch

template<typename ImplT >

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

protected

GlobalLinkedBinaries

template<typename ImplT >

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

GlobalLinkedModuleIds

template<typename ImplT >

SmallVector<std::string> proteus::JitEngineDevice< ImplT >::GlobalLinkedModuleIds

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

StorageCache

template<typename ImplT >

JitStorageCache<KernelFunction_t> proteus::JitEngineDevice< ImplT >::StorageCache

protected

VarNameToDevPtr

template<typename ImplT >

std::unordered_map<std::string, const void *> proteus::JitEngineDevice< ImplT >::VarNameToDevPtr

protected

---

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

• include/proteus/JitEngineDevice.hpp