Diego Devesa commited on
Commit
4703ea3
·
1 Parent(s): 6ba2c8f

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

Browse files
Files changed (1) hide show
  1. ggml/src/ggml-backend.cpp +2 -1
ggml/src/ggml-backend.cpp CHANGED
@@ -1729,12 +1729,13 @@ bool ggml_backend_sched_reserve(ggml_backend_sched_t sched, struct ggml_cgraph *
1729
 
1730
  ggml_backend_sched_split_graph(sched, measure_graph);
1731
 
 
 
1732
  if (!ggml_gallocr_reserve_n(sched->galloc, &sched->graph, sched->node_backend_ids, sched->leaf_backend_ids)) {
1733
  return false;
1734
  }
1735
 
1736
  ggml_backend_sched_reset(sched);
1737
- ggml_backend_sched_synchronize(sched);
1738
 
1739
  return true;
1740
  }
 
1729
 
1730
  ggml_backend_sched_split_graph(sched, measure_graph);
1731
 
1732
+ ggml_backend_sched_synchronize(sched);
1733
+
1734
  if (!ggml_gallocr_reserve_n(sched->galloc, &sched->graph, sched->node_backend_ids, sched->leaf_backend_ids)) {
1735
  return false;
1736
  }
1737
 
1738
  ggml_backend_sched_reset(sched);
 
1739
 
1740
  return true;
1741
  }