Skip to content

Commit d107459

Browse files
committed
ggml-backend : increase GGML_MAX_BACKENDS
1 parent 863ef45 commit d107459

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

ggml-backend.c

+4-4
Original file line numberDiff line numberDiff line change
@@ -727,7 +727,7 @@ static ggml_backend_t ggml_backend_reg_cpu_init(const char * params, void * user
727727

728728
// scheduler
729729

730-
#define GGML_MAX_BACKENDS 4
730+
#define GGML_MAX_BACKENDS 16
731731
#define GGML_MAX_SPLITS 256
732732
#define GGML_MAX_SPLIT_INPUTS 16
733733

@@ -903,7 +903,7 @@ static void sched_print_assignments(ggml_backend_sched_t sched, struct ggml_cgra
903903
}
904904
ggml_tallocr_t node_allocr = node_allocr(node);
905905
ggml_backend_t node_backend = node_allocr ? get_allocr_backend(sched, node_allocr) : NULL; // FIXME:
906-
fprintf(stderr, "node #%3d (%10.10s): %20.20s (%4.4s) [%4.4s %8.8s]:", i, ggml_op_name(node->op), node->name,
906+
fprintf(stderr, "node #%3d (%10.10s): %20.20s (%5.5s) [%5.5s %8.8s]:", i, ggml_op_name(node->op), node->name,
907907
fmt_size(ggml_nbytes(node)), node_allocr ? ggml_backend_name(node_backend) : "NULL", GET_CAUSE(node));
908908
for (int j = 0; j < GGML_MAX_SRC; j++) {
909909
struct ggml_tensor * src = node->src[j];
@@ -912,7 +912,7 @@ static void sched_print_assignments(ggml_backend_sched_t sched, struct ggml_cgra
912912
}
913913
ggml_tallocr_t src_allocr = node_allocr(src);
914914
ggml_backend_t src_backend = src_allocr ? get_allocr_backend(sched, src_allocr) : NULL;
915-
fprintf(stderr, " %20.20s (%4.4s) [%4.4s %8.8s]", src->name,
915+
fprintf(stderr, " %20.20s (%5.5s) [%5.5s %8.8s]", src->name,
916916
fmt_size(ggml_nbytes(src)), src_backend ? ggml_backend_name(src_backend) : "NULL", GET_CAUSE(src));
917917
}
918918
fprintf(stderr, "\n");
@@ -1091,7 +1091,7 @@ static void sched_split_graph(ggml_backend_sched_t sched, struct ggml_cgraph * g
10911091
sched->n_splits = cur_split + 1;
10921092
}
10931093

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

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

0 commit comments

Comments
 (0)