vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

ggml-debug: printing also the sum of each tensor

Browse source
This commit is contained in:
Pierrick HYMBERT 2024-04-11 13:05:45 +02:00
parent bb359cdd7d
commit 8d7be2c986
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
examples/ggml-debug/ggml-debug.cpp
View file

@ -30,6 +30,7 @@ static std::string ggml_ne_string(const ggml_tensor * t) {
static void ggml_print_tensor(uint8_t * data, ggml_type type, const int64_t * ne, const size_t * nb, int64_t n) {
size_t bs = ggml_blck_size(type);
float sum = 0;
for (int64_t i3 = 0; i3 < ne[3]; i3++) {
printf(" [\n");
for (int64_t i2 = 0; i2 < ne[2] && i2 < n; i2++) {
@ -53,6 +54,7 @@ static void ggml_print_tensor(uint8_t * data, ggml_type type, const int64_t * ne
GGML_ASSERT(false);
}
printf("%8.4f", v);
sum += v;
if (i0 < ne[0] - 1 && i0 < n - 1) printf(", ");
}
if (ne[0] > n) printf(", ...");
@ -63,6 +65,7 @@ static void ggml_print_tensor(uint8_t * data, ggml_type type, const int64_t * ne
}
if (ne[2] > n) printf(" ...\n");
printf(" ]\n");
printf(" sum = %f\n", sum);
}
}