summaryrefslogtreecommitdiff
path: root/ggml/src/ggml-cuda/cpy.cuh
diff options
context:
space:
mode:
Diffstat (limited to 'ggml/src/ggml-cuda/cpy.cuh')
-rw-r--r--ggml/src/ggml-cuda/cpy.cuh9
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);