Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
TensorOps.Fill.h
Go to the documentation of this file.
1#pragma once
2
3#include <cuda_runtime.h>
4#include <cuda_fp16.h>
5#include <cuda_bf16.h>
6#include <cuda_fp8.h>
7#include <cstddef>
8
10{
11 // ============================================================================
12 // Tensor Fill Operations (TensorFill.cu)
13 // ============================================================================
14
18 template<typename TargetType, typename HostType>
19 void launch_constant_fill_typed( void* dst, size_t count, HostType host_value, cudaStream_t stream );
20
21 template<typename TargetType, typename HostType>
22 void launch_array_fill_typed( void* dst, const HostType* host_values, size_t count, cudaStream_t stream );
23}
Definition CublasLt.Utils.ixx:15
void launch_array_fill_typed(void *dst, const HostType *host_values, size_t count, cudaStream_t stream)
void launch_constant_fill_typed(void *dst, size_t count, HostType host_value, cudaStream_t stream)
Templated fill operations with compile-time type dispatch.