@@ -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