diff options
| author | Michael Podvitskiy <podvitskiymichael@gmail.com> | 2024-02-09 10:42:27 +0100 |
|---|---|---|
| committer | Georgi Gerganov <ggerganov@gmail.com> | 2024-02-10 09:29:21 +0200 |
| commit | 4633d93af08d890ecd00fa6e4f61d76f21cded4c (patch) | |
| tree | 3c1bc3f11afdca41b89959cd20b52dc29a157bcc /ggml-backend.h | |
| parent | 4b7b38bef5addbd31f453871d79647fbae6bec8a (diff) | |
ggml : add abort_callback for cpu backend (ggml/725)
* a way to use abort_callback with the cpu backend
* whisper update
Diffstat (limited to 'ggml-backend.h')
| -rw-r--r-- | ggml-backend.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/ggml-backend.h b/ggml-backend.h index 8b8160fc..282b3a9b 100644 --- a/ggml-backend.h +++ b/ggml-backend.h @@ -83,8 +83,9 @@ extern "C" { GGML_API ggml_backend_t ggml_backend_cpu_init(void); - GGML_API GGML_CALL bool ggml_backend_is_cpu (ggml_backend_t backend); - GGML_API void ggml_backend_cpu_set_n_threads(ggml_backend_t backend_cpu, int n_threads); + GGML_API GGML_CALL bool ggml_backend_is_cpu (ggml_backend_t backend); + GGML_API void ggml_backend_cpu_set_n_threads (ggml_backend_t backend_cpu, int n_threads); + GGML_API void ggml_backend_cpu_set_abort_callback(ggml_backend_t backend_cpu, ggml_abort_callback abort_callback, void * abort_callback_data); // Create a backend buffer from an existing pointer GGML_API GGML_CALL ggml_backend_buffer_t ggml_backend_cpu_buffer_from_ptr(void * ptr, size_t size); |