proteus::JitEngineDeviceCUDA Class Reference

#include <JitEngineDeviceCUDA.hpp>

Inheritance diagram for proteus::JitEngineDeviceCUDA:

Collaboration diagram for proteus::JitEngineDeviceCUDA:

Public Member Functions
void	setLaunchBoundsForKernel (Module &M, Function &F, size_t GridSize, int BlockSize)
void	extractModules (BinaryInfo &BinInfo)
std::unique_ptr< Module >	tryExtractKernelModule (BinaryInfo &BinInfo, StringRef KernelName, LLVMContext &Ctx)
void	codegenPTX (Module &M, StringRef DeviceArch, SmallVectorImpl< char > &PTXStr)
HashT	getModuleHash (BinaryInfo &BinInfo)
std::unique_ptr< MemoryBuffer >	compileOnly (Module &M, bool DisableIROpt=false)
Public Member Functions inherited from proteus::JitEngineDevice< JitEngineDeviceCUDA >
DeviceError_t	compileAndRun (JITKernelInfo &KernelInfo, dim3 GridDim, dim3 BlockDim, void **KernelArgs, uint64_t ShmemSize, typename DeviceTraits< JitEngineDeviceCUDA >::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 (void *Handle, const char *VarName, const void *Addr, uint64_t VarSize)
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 ()
StringRef	getDeviceArch () const
Public Member Functions inherited from proteus::JitEngine
bool	isProteusDisabled ()
void	enable ()
void	disable ()

Static Public Member Functions
static JitEngineDeviceCUDA &	instance ()

Additional Inherited Members
Public Types inherited from proteus::JitEngineDevice< JitEngineDeviceCUDA >
using	DeviceError_t = typename DeviceTraits< JitEngineDeviceCUDA >::DeviceError_t
using	DeviceStream_t = typename DeviceTraits< JitEngineDeviceCUDA >::DeviceStream_t
using	KernelFunction_t = typename DeviceTraits< JitEngineDeviceCUDA >::KernelFunction_t
Public Attributes inherited from proteus::JitEngineDevice< JitEngineDeviceCUDA >
void *	CurHandle
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 inherited from proteus::JitEngineDevice< JitEngineDeviceCUDA >
	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 inherited from proteus::JitEngineDevice< JitEngineDeviceCUDA >
MemoryCache< KernelFunction_t >	CodeCache
StorageCache	ObjectCache
std::string	DeviceArch
DenseMap< const void *, JITKernelInfo >	JITKernelInfoMap

Member Function Documentation

codegenPTX()

void proteus::JitEngineDeviceCUDA::codegenPTX	(	Module &	M,
		StringRef	DeviceArch,
		SmallVectorImpl< char > &	PTXStr
	)

compileOnly()

std::unique_ptr< MemoryBuffer > JitEngineDeviceCUDA::compileOnly	(	Module &	M,
		bool	DisableIROpt = false
	)

extractModules()

void JitEngineDeviceCUDA::extractModules

(

BinaryInfo &

BinInfo

)

getModuleHash()

HashT JitEngineDeviceCUDA::getModuleHash

(

BinaryInfo &

BinInfo

)

instance()

JitEngineDeviceCUDA & JitEngineDeviceCUDA::instance ( )

static

setLaunchBoundsForKernel()

void proteus::JitEngineDeviceCUDA::setLaunchBoundsForKernel	(	Module &	M,
		Function &	F,
		size_t	GridSize,
		int	BlockSize
	)

tryExtractKernelModule()

std::unique_ptr< Module > JitEngineDeviceCUDA::tryExtractKernelModule	(	BinaryInfo &	BinInfo,
		StringRef	KernelName,
		LLVMContext &	Ctx
	)

---

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

• include/proteus/JitEngineDeviceCUDA.hpp
• src/lib/JitEngineDeviceCUDA.cpp