Skip to content

Commit 42b3837

Browse files
committed
feat(llama-gguf): Print out the tensor type in llama-gguf r
Branch: Mamba2Perf Signed-off-by: Gabe Goodhart <ghart@us.ibm.com>
1 parent 92bb84f commit 42b3837

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

examples/gguf/gguf.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -184,8 +184,9 @@ static bool gguf_ex_read_1(const std::string & fname, bool check_data) {
184184
const char * name = gguf_get_tensor_name (ctx, i);
185185
const size_t size = gguf_get_tensor_size (ctx, i);
186186
const size_t offset = gguf_get_tensor_offset(ctx, i);
187+
const char * type = ggml_type_name(gguf_get_tensor_type(ctx, i));
187188

188-
printf("%s: tensor[%d]: name = %s, size = %zu, offset = %zu\n", __func__, i, name, size, offset);
189+
printf("%s: tensor[%d]: name = %s, size = %zu, offset = %zu, type = %s\n", __func__, i, name, size, offset, type);
189190
}
190191
}
191192

0 commit comments

Comments
 (0)