MUDA
Loading...
Searching...
No Matches
kernel_assign.h
1#pragma once
2#include <cuda.h>
3#include <muda/buffer/buffer_fwd.h>
4
5namespace muda::details::buffer
6{
7// assign 0D
8template <typename T>
9MUDA_HOST void kernel_assign(cudaStream_t stream, VarView<T> dst, CVarView<T> src);
10
11// assign 1D
12template <typename T>
13MUDA_HOST void kernel_assign(int grid_dim,
14 int block_dim,
15 cudaStream_t stream,
16 BufferView<T> dst,
17 CBufferView<T> src);
18
19// assign 2D
20template <typename T>
21MUDA_HOST void kernel_assign(int grid_dim,
22 int block_dim,
23 cudaStream_t stream,
24 Buffer2DView<T> dst,
25 CBuffer2DView<T> src);
26
27// assign 3D
28template <typename T>
29MUDA_HOST void kernel_assign(int grid_dim,
30 int block_dim,
31 cudaStream_t stream,
32 Buffer3DView<T> dst,
33 CBuffer3DView<T> src);
34} // namespace muda::details::buffer
35
36#include "details/kernel_assign.inl"