Compute.CudaTensorDataType Module Reference

Classes
struct	Mila::Dnn::Compute::Cuda::CudaDataTypeTraits< TDataType >
	Compile-time mapping from TensorDataType -> cudaDataType_t. More...
struct	Mila::Dnn::Compute::Cuda::TensorDataTypeMap< TDataType >
	Compile-time mapping from abstract TensorDataType -> CUDA native device type. More...

Typedefs
template<TensorDataType TDataType>
using	Mila::Dnn::Compute::Cuda::device_type_t = typename TensorDataTypeMap<TDataType>::device_type
	Convenience alias for accessing CUDA native type.

Variables
template<TensorDataType TDataType>
constexpr cudaDataType_t	Mila::Dnn::Compute::Cuda::cuda_data_type_v = CudaDataTypeTraits<TDataType>::cuda_data_type
	Convenience alias for accessing the cudaDataType_t mapping directly.

Files
file	/__w/Mila/Mila/Mila/Src/Dnn/Compute/Devices/Cuda/Tensors/CudaTensorDataType.ixx
	CUDA-specific tensor data type trait system - Primary module interface.
file	/__w/Mila/Mila/Mila/Src/Dnn/Compute/Devices/Cuda/Tensors/CudaTensorDataType-Maps.ixx
	CUDA-specific mappings between abstract TensorDataType and concrete CUDA native types.
file	/__w/Mila/Mila/Mila/Src/Dnn/Compute/Devices/Cuda/Tensors/CudaTensorDataType-Specializations.ixx
file	/__w/Mila/Mila/Mila/Src/Dnn/Compute/Devices/Cuda/Tensors/CudaTensorDataTypes-CublasLtTypes.ixx