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