Spaces:
Sleeping
Sleeping
Diego Devesa
commited on
Commit
·
4703ea3
1
Parent(s):
6ba2c8f
ggml : fix possible buffer use after free in sched reserve (llama/9930)
Browse files
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 |
}
|