Просмотр исходного кода

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

Diego Devesa 1 год назад
Родитель
Сommit
eda7e1d4f5
1 измененных файлов с 2 добавлено и 1 удалено
  1. 2 1
      ggml/src/ggml-backend.cpp

+ 2 - 1
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;
 }