diff options
Diffstat (limited to 'llama.cpp/ggml/src/ggml-musa/mudnn.cuh')
| -rw-r--r-- | llama.cpp/ggml/src/ggml-musa/mudnn.cuh | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/llama.cpp/ggml/src/ggml-musa/mudnn.cuh b/llama.cpp/ggml/src/ggml-musa/mudnn.cuh new file mode 100644 index 0000000..c301285 --- /dev/null +++ b/llama.cpp/ggml/src/ggml-musa/mudnn.cuh | |||
| @@ -0,0 +1,12 @@ | |||
| 1 | #pragma once | ||
| 2 | |||
| 3 | #include "ggml-cuda/common.cuh" | ||
| 4 | #include "ggml.h" | ||
| 5 | |||
| 6 | // Asynchronously copies data from src tensor to dst tensor using the provided context. | ||
| 7 | // Returns a musaError_t indicating success or failure. | ||
| 8 | musaError_t mudnnMemcpyAsync( | ||
| 9 | ggml_backend_cuda_context &ctx, | ||
| 10 | const ggml_tensor *dst, | ||
| 11 | const ggml_tensor *src | ||
| 12 | ); | ||
