diff options
author | Georgi Gerganov <ggerganov@gmail.com> | 2023-05-21 11:56:23 +0300 |
---|---|---|
committer | Georgi Gerganov <ggerganov@gmail.com> | 2023-05-21 11:56:23 +0300 |
commit | 265db9834e761b7c8210ea1888117efcd3262f52 (patch) | |
tree | c7ec05f2e28c23e2b3a4ff40acfe19c1a0e78ea4 /ggml.c | |
parent | fab49c685e09d95942de34e3eadd72f880de21d5 (diff) |
ggml : output 3d sizes in ggml_graph_dump_dot()
Diffstat (limited to 'ggml.c')
-rw-r--r-- | ggml.c | 9 |
1 files changed, 6 insertions, 3 deletions
@@ -14631,9 +14631,12 @@ void ggml_graph_dump_dot(const struct ggml_cgraph * gb, const struct ggml_cgraph fprintf(fp, "%s |", node->name); } - fprintf(fp, "%d [%" PRId64 ", %" PRId64 "] | <x>%s", - i, node->ne[0], node->ne[1], - GGML_OP_SYMBOL[node->op]); + if (node->n_dims == 2) { + fprintf(fp, "%d [%" PRId64 ", %" PRId64 "] | <x>%s", i, node->ne[0], node->ne[1], GGML_OP_SYMBOL[node->op]); + } else { + fprintf(fp, "%d [%" PRId64 ", %" PRId64 ", %" PRId64 "] | <x>%s", i, node->ne[0], node->ne[1], node->ne[2], GGML_OP_SYMBOL[node->op]); + } + if (node->grad) { fprintf(fp, " | <g>%s\"; ]\n", GGML_OP_SYMBOL[node->grad->op]); |