Skip to content

Commit eda7e1d

Browse files
authored
ggml : fix possible buffer use after free in sched reserve (#9930)
1 parent 24203e9 commit eda7e1d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

ggml/src/ggml-backend.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1538,12 +1538,13 @@ bool ggml_backend_sched_reserve(ggml_backend_sched_t sched, struct ggml_cgraph *
15381538

15391539
ggml_backend_sched_split_graph(sched, measure_graph);
15401540

1541+
ggml_backend_sched_synchronize(sched);
1542+
15411543
if (!ggml_gallocr_reserve_n(sched->galloc, &sched->graph, sched->node_backend_ids, sched->leaf_backend_ids)) {
15421544
return false;
15431545
}
15441546

15451547
ggml_backend_sched_reset(sched);
1546-
ggml_backend_sched_synchronize(sched);
15471548

15481549
return true;
15491550
}

0 commit comments

Comments
 (0)