CublasLt.Utils.ixx File Reference

#include <cuda_runtime.h>
#include <cuda_bf16.h>
#include <cuda_fp16.h>
#include <cuda_fp8.h>
#include <cublasLt.h>
#include <type_traits>

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

Functions
template<typename NativeType>
void	Mila::Dnn::Compute::Cuda::cublaslt_compute_types (cublasComputeType_t &compute_type, cudaDataType_t &scale_type)
template<typename NativeType>
cudaDataType_t	Mila::Dnn::Compute::Cuda::cublaslt_cuda_data_type ()