]> git.djapps.eu Git - pkg/ggml/sources/whisper.cpp/commitdiff
ggml : fix possible buffer use after free in sched reserve (llama/9930)
authorDiego Devesa <redacted>
Sun, 17 Nov 2024 06:31:17 +0000 (07:31 +0100)
committerGeorgi Gerganov <redacted>
Wed, 20 Nov 2024 19:00:08 +0000 (21:00 +0200)
ggml/src/ggml-backend.cpp

index bcbe0b6c78c8c8c1e24d5509b942ac3525f9adce..9d40d7a87f94f7b1cea5bc4a55880b984469b8de 100644 (file)
@@ -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;
 }