RmsNormOp.Dispatch.ixx File Reference

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

Classes
struct	Mila::Dnn::Compute::Cuda::RmsNorm::Detail::cuda_rmsnorm_impl< float >
struct	Mila::Dnn::Compute::Cuda::RmsNorm::Detail::cuda_rmsnorm_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::RmsNorm
namespace	Mila::Dnn::Compute::Cuda::RmsNorm::Detail