root/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/llama.cpp.git synced 2025-01-12 19:50:17 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

ggml : output 3d sizes in ggml_graph_dump_dot()

Browse Source
This commit is contained in:
Georgi Gerganov 2023-05-21 11:56:23 +03:00
parent fab49c685e
commit 265db9834e
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
ggml.c
View File

@ -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, "%s |", node->name);
} }
fprintf(fp, "%d [%" PRId64 ", %" PRId64 "] | <x>%s", if (node->n_dims == 2) {
i, node->ne[0], node->ne[1], fprintf(fp, "%d [%" PRId64 ", %" PRId64 "] | <x>%s", i, node->ne[0], node->ne[1], GGML_OP_SYMBOL[node->op]);
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) { if (node->grad) {
fprintf(fp, " | <g>%s\"; ]\n", GGML_OP_SYMBOL[node->grad->op]); fprintf(fp, " | <g>%s\"; ]\n", GGML_OP_SYMBOL[node->grad->op]);