| 1845 | } |
| 1846 | |
| 1847 | bool ggml_backend_sched_reserve(ggml_backend_sched_t sched, struct ggml_cgraph * measure_graph) { |
| 1848 | GGML_ASSERT(sched); |
| 1849 | GGML_ASSERT((int)sched->hash_set.size >= measure_graph->n_nodes + measure_graph->n_leafs); |
| 1850 | |
| 1851 | ggml_backend_sched_synchronize(sched); |
| 1852 | |
| 1853 | ggml_backend_sched_split_graph(sched, measure_graph); |
| 1854 | |
| 1855 | if (!ggml_gallocr_reserve_n(sched->galloc, &sched->graph, sched->node_backend_ids, sched->leaf_backend_ids)) { |
| 1856 | return false; |
| 1857 | } |
| 1858 | |
| 1859 | ggml_backend_sched_reset(sched); |
| 1860 | |
| 1861 | return true; |
| 1862 | } |
| 1863 | |
| 1864 | bool ggml_backend_sched_alloc_graph(ggml_backend_sched_t sched, struct ggml_cgraph * graph) { |
| 1865 | GGML_ASSERT(sched); |
no test coverage detected