CudaTensorDataType-Specializations.ixx File Reference

CUDA concrete type trait specializations. More...

#include <cstdint>
#include <string_view>
#include <cuda_fp16.h>
#include <cuda_bf16.h>
#include <cuda_fp8.h>
import Compute.DeviceType;
import Dnn.TensorDataTypeMap;
import Dnn.TensorDataType;

Classes
struct	Mila::Dnn::TensorDataTypeMap< __nv_fp8_e4m3 >
struct	Mila::Dnn::TensorDataTypeMap< __nv_fp8_e5m2 >
struct	Mila::Dnn::TensorDataTypeMap< half >
struct	Mila::Dnn::TensorDataTypeMap< nv_bfloat16 >

Namespaces
namespace	Mila
	Mila main API namespace.

Detailed Description

CUDA concrete type trait specializations.