Skip to content

Commit f668734

Browse files
committed
dyn graphs : remove n_tasks from ggml_cplan
1 parent af8849a commit f668734

File tree

3 files changed

+261
-253
lines changed

3 files changed

+261
-253
lines changed

include/ggml/ggml.h

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -532,9 +532,6 @@ extern "C" {
532532

533533
int n_threads;
534534

535-
// the `n_tasks` of nodes, 1:1 mapping to cgraph nodes
536-
int n_tasks[GGML_DEFAULT_GRAPH_SIZE]; // FIXME: need to allocate ggml_cplan dynamically?
537-
538535
// abort ggml_graph_compute when true
539536
bool (*abort_callback)(void * data);
540537
void * abort_callback_data;

src/ggml-backend.c

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -862,6 +862,7 @@ ggml_backend_sched_t ggml_backend_sched_new(ggml_backend_t * backends, int n_bac
862862
GGML_ASSERT(n_backends <= GGML_MAX_BACKENDS);
863863

864864
struct ggml_backend_sched * sched = malloc(sizeof(struct ggml_backend_sched));
865+
memset(sched, 0, sizeof(struct ggml_backend_sched));
865866

866867
fprintf(stderr, "ggml_backend_sched size: %lu KB\n", sizeof(struct ggml_backend_sched)/1024);
867868

@@ -885,6 +886,9 @@ void ggml_backend_sched_free(ggml_backend_sched_t sched) {
885886
for (int i = 0; i < sched->n_backends; i++) {
886887
ggml_allocr_free(sched->allocs[i]);
887888
}
889+
free(sched->hash_set.keys);
890+
free(sched->node_allocr);
891+
free(sched->node_copies);
888892
free(sched);
889893
}
890894

0 commit comments

Comments
 (0)