@@ -723,7 +723,7 @@ static ggml_backend_t ggml_backend_reg_cpu_init(const char * params, void * user
723
723
724
724
// scheduler
725
725
726
- #define GGML_MAX_BACKENDS 4
726
+ #define GGML_MAX_BACKENDS 16
727
727
#define GGML_MAX_SPLITS 256
728
728
#define GGML_MAX_SPLIT_INPUTS 16
729
729
@@ -899,7 +899,7 @@ static void sched_print_assignments(ggml_backend_sched_t sched, struct ggml_cgra
899
899
}
900
900
ggml_tallocr_t node_allocr = node_allocr (node );
901
901
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 ,
903
903
fmt_size (ggml_nbytes (node )), node_allocr ? ggml_backend_name (node_backend ) : "NULL" , GET_CAUSE (node ));
904
904
for (int j = 0 ; j < GGML_MAX_SRC ; j ++ ) {
905
905
struct ggml_tensor * src = node -> src [j ];
@@ -908,7 +908,7 @@ static void sched_print_assignments(ggml_backend_sched_t sched, struct ggml_cgra
908
908
}
909
909
ggml_tallocr_t src_allocr = node_allocr (src );
910
910
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 ,
912
912
fmt_size (ggml_nbytes (src )), src_backend ? ggml_backend_name (src_backend ) : "NULL" , GET_CAUSE (src ));
913
913
}
914
914
fprintf (stderr , "\n" );
@@ -1087,7 +1087,7 @@ static void sched_split_graph(ggml_backend_sched_t sched, struct ggml_cgraph * g
1087
1087
sched -> n_splits = cur_split + 1 ;
1088
1088
}
1089
1089
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);
1091
1091
1092
1092
#ifndef NDEBUG
1093
1093
// sanity check: all sources should have the same backend as the node
0 commit comments