diff options
Diffstat (limited to 'ggml/src/ggml-cuda/cpy.cuh')
-rw-r--r-- | ggml/src/ggml-cuda/cpy.cuh | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/ggml/src/ggml-cuda/cpy.cuh b/ggml/src/ggml-cuda/cpy.cuh new file mode 100644 index 00000000..79616742 --- /dev/null +++ b/ggml/src/ggml-cuda/cpy.cuh @@ -0,0 +1,9 @@ +#include "common.cuh" + +#define CUDA_CPY_BLOCK_SIZE 32 + +void ggml_cuda_cpy(ggml_backend_cuda_context & ctx, const ggml_tensor * src0, ggml_tensor * src1); + +void ggml_cuda_dup(ggml_backend_cuda_context & ctx, ggml_tensor * dst); + +void* ggml_cuda_cpy_fn(const ggml_tensor * src0, ggml_tensor * src1); |