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/0.0.1642~184 X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=351c72876d244a512df473d871c7307b3772059e;p=pkg%2Fggml%2Fsources%2Fggml ggml : fix possible buffer use after free in sched reserve (llama/9930) --- diff --git a/src/ggml-backend.cpp b/src/ggml-backend.cpp index 634fe38e..9a6010d3 100644 --- a/src/ggml-backend.cpp +++ b/src/ggml-backend.cpp @@ -1538,12 +1538,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; }