CudaResidualOp.Dispatch.ixx File Reference

#include <cublasLt.h>
#include <cuda_bf16.h>
#include <vector>
#include <memory>
#include <string>
#include <stdexcept>
#include <cstdint>
#include <type_traits>
#include <sstream>
#include <cassert>
#include "Kernels/Residual.cuh"

Classes
struct	Mila::Dnn::Compute::Cuda::Residual::Detail::cuda_residual_impl< float >
struct	Mila::Dnn::Compute::Cuda::Residual::Detail::cuda_residual_impl< nv_bfloat16 >

Namespaces
namespace	Mila
	Mila main API namespace.
namespace	Mila::Dnn
namespace	Mila::Dnn::Compute
namespace	Mila::Dnn::Compute::Cuda
namespace	Mila::Dnn::Compute::Cuda::Residual
namespace	Mila::Dnn::Compute::Cuda::Residual::Detail
	CUDA residual kernel dispatch implementations.