CudaTensorOps.h File Reference

CUDA tensor operation kernel function declarations. More...

#include <cuda_runtime.h>
#include <cstddef>

Go to the source code of this file.

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

Detailed Description

CUDA tensor operation kernel function declarations.

This header provides C-style function declarations for CUDA kernel wrappers used by CUDA tensor operations. All functions are implemented in corresponding .cu files and linked during compilation.