feat(off-topic): print the number of elements in tensors with llama-gguf

Branch: Mamba2SSD

Signed-off-by: Gabe Goodhart <ghart@us.ibm.com>

Author: gabe-l-hart

examples/gguf/gguf.cpp

@@ -184,9 +184,12 @@ static bool gguf_ex_read_1(const std::string & fname, bool check_data) {
             const char * name   = gguf_get_tensor_name  (ctx, i);
             const size_t size   = gguf_get_tensor_size  (ctx, i);
             const size_t offset = gguf_get_tensor_offset(ctx, i);
-            const char * type   = ggml_type_name(gguf_get_tensor_type(ctx, i));
+            const auto type = gguf_get_tensor_type(ctx, i);
+            const char * type_name   = ggml_type_name(type);
+            const size_t type_size = ggml_type_size(type);
+            const size_t n_elements = size / type_size;
 
-            printf("%s: tensor[%d]: name = %s, size = %zu, offset = %zu, type = %s\n", __func__, i, name, size, offset, type);
+            printf("%s: tensor[%d]: name = %s, size = %zu, offset = %zu, type = %s, n_elts = %zu\n", __func__, i, name, size, offset, type_name, n_elements);
         }
     }