From: Diego Devesa <redacted>
Date: Sun, 17 Nov 2024 06:31:17 +0000 (+0100)
Subject: ggml : fix possible buffer use after free in sched reserve (#9930)
X-Git-Tag: upstream/0.0.4488~375
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=eda7e1d4f54711de1c9b40502d6c88bbc217da60;p=pkg%2Fggml%2Fsources%2Fllama.cpp

ggml : fix possible buffer use after free in sched reserve (#9930)
---

diff --git a/ggml/src/ggml-backend.cpp b/ggml/src/ggml-backend.cpp
index 634fe38e..9a6010d3 100644
--- a/ggml/src/ggml-backend.cpp
+++ b/ggml/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;
 }