Skip to content

File kernel_fill.h

File List > agent > kernel_fill.h

Go to the documentation of this file

#pragma once
#include <cuda.h>
#include <muda/buffer/buffer_fwd.h>

namespace muda::details::buffer
{
// fill 0D
template <typename T>
MUDA_HOST void kernel_fill(cudaStream_t stream, VarView<T> dst, const T& val);

// fill 1D
template <typename T>
MUDA_HOST void kernel_fill(
    int grid_dim, int block_dim, cudaStream_t stream, BufferView<T> dst, const T& val);

// fill 2D
template <typename T>
MUDA_HOST void kernel_fill(int             grid_dim,
                           int             block_dim,
                           cudaStream_t    stream,
                           Buffer2DView<T> dst,
                           const T&        val);

// fill 3D
template <typename T>
MUDA_HOST void kernel_fill(int             grid_dim,
                           int             block_dim,
                           cudaStream_t    stream,
                           Buffer3DView<T> dst,
                           const T&        val);
}  // namespace muda::details::buffer

#include "details/kernel_fill.inl"