Skip to content

Commit dcc0dad

Browse files
committed
ggml-backend : increase GGML_MAX_BACKENDS
1 parent 370f1c2 commit dcc0dad

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

ggml-backend.c

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -723,7 +723,7 @@ static ggml_backend_t ggml_backend_reg_cpu_init(const char * params, void * user
723723

724724
// scheduler
725725

726-
#define GGML_MAX_BACKENDS 4
726+
#define GGML_MAX_BACKENDS 16
727727
#define GGML_MAX_SPLITS 256
728728
#define GGML_MAX_SPLIT_INPUTS 16
729729

@@ -899,7 +899,7 @@ static void sched_print_assignments(ggml_backend_sched_t sched, struct ggml_cgra
899899
}
900900
ggml_tallocr_t node_allocr = node_allocr(node);
901901
ggml_backend_t node_backend = node_allocr ? get_allocr_backend(sched, node_allocr) : NULL; // FIXME:
902-
fprintf(stderr, "node #%3d (%10.10s): %20.20s (%4.4s) [%4.4s %8.8s]:", i, ggml_op_name(node->op), node->name,
902+
fprintf(stderr, "node #%3d (%10.10s): %20.20s (%5.5s) [%5.5s %8.8s]:", i, ggml_op_name(node->op), node->name,
903903
fmt_size(ggml_nbytes(node)), node_allocr ? ggml_backend_name(node_backend) : "NULL", GET_CAUSE(node));
904904
for (int j = 0; j < GGML_MAX_SRC; j++) {
905905
struct ggml_tensor * src = node->src[j];
@@ -908,7 +908,7 @@ static void sched_print_assignments(ggml_backend_sched_t sched, struct ggml_cgra
908908
}
909909
ggml_tallocr_t src_allocr = node_allocr(src);
910910
ggml_backend_t src_backend = src_allocr ? get_allocr_backend(sched, src_allocr) : NULL;
911-
fprintf(stderr, " %20.20s (%4.4s) [%4.4s %8.8s]", src->name,
911+
fprintf(stderr, " %20.20s (%5.5s) [%5.5s %8.8s]", src->name,
912912
fmt_size(ggml_nbytes(src)), src_backend ? ggml_backend_name(src_backend) : "NULL", GET_CAUSE(src));
913913
}
914914
fprintf(stderr, "\n");
@@ -1087,7 +1087,7 @@ static void sched_split_graph(ggml_backend_sched_t sched, struct ggml_cgraph * g
10871087
sched->n_splits = cur_split + 1;
10881088
}
10891089

1090-
//fprintf(stderr, "PASS 4 ASSIGNMENTS\n"); sched_print_assignments(sched, graph); fflush(stdout);
1090+
//fprintf(stderr, "PASS 4 ASSIGNMENTS\n"); sched_print_assignments(sched, graph);
10911091

10921092
#ifndef NDEBUG
10931093
// sanity check: all sources should have the same backend as the node

0 commit comments

Comments
 (0)