zhangyiss/gguf-tools
1
0
Fork 0
mirror of https://github.com/antirez/gguf-tools.git synced 2025-09-16 17:48:08 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Inspect tensor taking dims into consideration

Browse Source
This commit is contained in:
Juarez Bochi
2024-01-04 11:44:13 -05:00
parent b1f32c4088
commit 58a0479bb4
1 changed files with 33 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
gguf-tools.c
View File

@@ -361,14 +361,44 @@ void gguf_tools_inspect_weights(const char *filename, const char *tname, uint64_
exit(1);
}
uint64_t strides[GGUF_TENSOR_MAX_DIM] = {0};
strides[tensor.ndim-1] = 1;
for (int j = tensor.ndim - 2; j >= 0; j--) {
strides[j] = tensor.dim[j + 1] * strides[j + 1];
}
const int ident = 4;
uint64_t j = 0;
int broke = 1;
while (j < tensor.num_weights) {
printf("%f, ", weights[j]);
int last = j + 1 == tensor.num_weights;
for (int k = 0; k < (int) tensor.ndim - 1; k++) {
if (j % strides[k] == 0) {
printf("%*s\n", k * ident, "[");
}
}
if (broke) {
printf("%*s", tensor.ndim * ident, "");
}
printf("%f%s", weights[j], last ? "" : ", ");
broke = 0;
j++;
if (j % 4 == 0) printf("\n");
for (int k = (int) tensor.ndim - 2; k >= 0; k--) {
if (j % strides[k] == 0) {
if (!broke) {
broke = 1;
printf("\n");
}
printf("%*s%s\n", k * ident, "]", last ? "" : ",");
}
}
if (!broke && j % 4 == 0) {
broke = 1;
printf("\n");
}
if (j == count) break;
}
if (j % 4 != 0) printf("\n");
if (!broke) printf("\n");
free(weights);
return;
}