TensorOps.Fill.h File Reference

#include <cuda_runtime.h>
#include <cuda_fp16.h>
#include <cuda_bf16.h>
#include <cuda_fp8.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

Functions
template<typename TargetType, typename HostType>
void	Mila::Dnn::Compute::Cuda::launch_array_fill_typed (void *dst, const HostType *host_values, size_t count, cudaStream_t stream)
template<typename TargetType, typename HostType>
void	Mila::Dnn::Compute::Cuda::launch_constant_fill_typed (void *dst, size_t count, HostType host_value, cudaStream_t stream)
	Templated fill operations with compile-time type dispatch.