From: Diego Devesa Date: Sun, 17 Nov 2024 06:31:17 +0000 (+0100) Subject: ggml : fix possible buffer use after free in sched reserve (llama/9930) X-Git-Tag: upstream/1.7.4~193 X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=24ad19d0e91750611994c7782f26143af411d633;p=pkg%2Fggml%2Fsources%2Fwhisper.cpp ggml : fix possible buffer use after free in sched reserve (llama/9930) --- diff --git a/ggml/src/ggml-backend.cpp b/ggml/src/ggml-backend.cpp index bcbe0b6c..9d40d7a8 100644 --- a/ggml/src/ggml-backend.cpp +++ b/ggml/src/ggml-backend.cpp @@ -1729,12 +1729,13 @@ bool ggml_backend_sched_reserve(ggml_backend_sched_t sched, struct ggml_cgraph * ggml_backend_sched_split_graph(sched, measure_graph); + ggml_backend_sched_synchronize(sched); + if (!ggml_gallocr_reserve_n(sched->galloc, &sched->graph, sched->node_backend_ids, sched->leaf_backend_ids)) { return false; } ggml_backend_sched_reset(sched); - ggml_backend_sched_synchronize(sched); return true; }