Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::Compute::CudaExecutionContext Member List

This is the complete list of members for Mila::Dnn::Compute::CudaExecutionContext, including all inherited members.

CudaExecutionContext(DeviceId device_id)Mila::Dnn::Compute::CudaExecutionContextinlineexplicit
CudaExecutionContext(const CudaExecutionContext &)=delete (defined in Mila::Dnn::Compute::CudaExecutionContext)Mila::Dnn::Compute::CudaExecutionContext
CudaExecutionContext(CudaExecutionContext &&)=delete (defined in Mila::Dnn::Compute::CudaExecutionContext)Mila::Dnn::Compute::CudaExecutionContext
getActivationObserver() const noexcept (defined in Mila::Dnn::Compute::IExecutionContext)Mila::Dnn::Compute::IExecutionContextinline
getCublasLtHandle() constMila::Dnn::Compute::CudaExecutionContextinline
getCublasLtWorkspace() const noexceptMila::Dnn::Compute::CudaExecutionContextinline
getCublasLtWorkspaceSize() const noexceptMila::Dnn::Compute::CudaExecutionContextinline
getCurandGenerator() constMila::Dnn::Compute::CudaExecutionContextinline
getDeviceId() const noexcept overrideMila::Dnn::Compute::CudaExecutionContextinlinevirtual
getDeviceScratchBuffer(size_t required_bytes) constMila::Dnn::Compute::CudaExecutionContextinline
getLoadStagingBuffer(size_t required_bytes) constMila::Dnn::Compute::CudaExecutionContextinline
getLoadStagingLimitBytes() noexceptMila::Dnn::Compute::CudaExecutionContextinlinestatic
getPinnedStagingBuffer(size_t required_bytes) constMila::Dnn::Compute::CudaExecutionContextinline
getStream() const noexceptMila::Dnn::Compute::CudaExecutionContextinline
hasActivationObserver() const noexcept (defined in Mila::Dnn::Compute::IExecutionContext)Mila::Dnn::Compute::IExecutionContextinline
IExecutionContext()=default (defined in Mila::Dnn::Compute::IExecutionContext)Mila::Dnn::Compute::IExecutionContextprotected
operator=(const CudaExecutionContext &)=delete (defined in Mila::Dnn::Compute::CudaExecutionContext)Mila::Dnn::Compute::CudaExecutionContext
operator=(CudaExecutionContext &&)=delete (defined in Mila::Dnn::Compute::CudaExecutionContext)Mila::Dnn::Compute::CudaExecutionContext
releaseLoadStaging() noexcept overrideMila::Dnn::Compute::CudaExecutionContextinlinevirtual
reserveScratch(std::size_t bytes) overrideMila::Dnn::Compute::CudaExecutionContextinlinevirtual
setActivationObserver(ActivationObserver observer)Mila::Dnn::Compute::IExecutionContextinline
synchronize() overrideMila::Dnn::Compute::CudaExecutionContextinlinevirtual
~CudaExecutionContext()Mila::Dnn::Compute::CudaExecutionContextinline
~IExecutionContext()=default (defined in Mila::Dnn::Compute::IExecutionContext)Mila::Dnn::Compute::IExecutionContextvirtual