danbev commited on
Commit
4427ede
·
1 Parent(s): 43720b1

ggml : improve inputs log sched_print_assignments (ggml/1053)

Browse files

This commit attempts to improve the log message for the inputs of the
splits in the sched_print_assignments function.

The motivation for this change is that currently even if there are no
inputs a colon is displayed at the end of the line, which can make it a
little confusing when reading the output as it could be interpreted as
the line below are inputs when they are in fact nodes. With this change
the colon will only be printed if there actually are inputs.

Files changed (1) hide show
  1. ggml/src/ggml-backend.cpp +4 -1
ggml/src/ggml-backend.cpp CHANGED
@@ -795,9 +795,12 @@ static void ggml_backend_sched_print_assignments(ggml_backend_sched_t sched, str
795
  for (int i = 0; i < graph->n_nodes; i++) {
796
  if (cur_split < sched->n_splits && i == sched->splits[cur_split].i_start) {
797
  ggml_backend_t split_backend = sched->backends[sched->splits[cur_split].backend_id];
798
- GGML_LOG_DEBUG("\n## SPLIT #%d: %s # %d inputs: ", cur_split, ggml_backend_name(split_backend),
799
  sched->splits[cur_split].n_inputs);
800
  for (int j = 0; j < sched->splits[cur_split].n_inputs; j++) {
 
 
 
801
  GGML_LOG_DEBUG("[%s (%5.5s)] ", sched->splits[cur_split].inputs[j]->name,
802
  fmt_size(ggml_nbytes(sched->splits[cur_split].inputs[j])));
803
  }
 
795
  for (int i = 0; i < graph->n_nodes; i++) {
796
  if (cur_split < sched->n_splits && i == sched->splits[cur_split].i_start) {
797
  ggml_backend_t split_backend = sched->backends[sched->splits[cur_split].backend_id];
798
+ GGML_LOG_DEBUG("\n## SPLIT #%d: %s # %d inputs", cur_split, ggml_backend_name(split_backend),
799
  sched->splits[cur_split].n_inputs);
800
  for (int j = 0; j < sched->splits[cur_split].n_inputs; j++) {
801
+ if (j == 0) {
802
+ GGML_LOG_DEBUG(": ");
803
+ }
804
  GGML_LOG_DEBUG("[%s (%5.5s)] ", sched->splits[cur_split].inputs[j]->name,
805
  fmt_size(ggml_nbytes(sched->splits[cur_split].inputs[j])));
806
  }