CudaTensorDataTypes-CublasLtTypes.ixx File Reference

#include <cuda_runtime.h>
#include <cublasLt.h>
#include <cuda_fp16.h>
#include <cuda_bf16.h>
#include <cuda_fp8.h>
#include <type_traits>
import Dnn.TensorDataType;

Classes
struct	Mila::Dnn::Compute::Cuda::CudaDataTypeTraits< TDataType >
	Compile-time mapping from TensorDataType -> cudaDataType_t. More...
struct	Mila::Dnn::Compute::Cuda::CudaDataTypeTraits< TensorDataType::BF16 >
struct	Mila::Dnn::Compute::Cuda::CudaDataTypeTraits< TensorDataType::FP16 >
struct	Mila::Dnn::Compute::Cuda::CudaDataTypeTraits< TensorDataType::FP32 >
struct	Mila::Dnn::Compute::Cuda::CudaDataTypeTraits< TensorDataType::FP8_E4M3 >
struct	Mila::Dnn::Compute::Cuda::CudaDataTypeTraits< TensorDataType::FP8_E5M2 >
struct	Mila::Dnn::Compute::Cuda::CudaDataTypeTraits< TensorDataType::INT32 >
struct	Mila::Dnn::Compute::Cuda::CudaDataTypeTraits< TensorDataType::INT8 >

Namespaces
namespace	Mila
	Mila main API namespace.
namespace	Mila::Dnn
namespace	Mila::Dnn::Compute
namespace	Mila::Dnn::Compute::Cuda

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.