upstream-archive/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2026-07-23 11:10:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

gguf : add tensor shape accessor (#24405)

Browse Source 
* gguf : add tensor shape accessors

* gguf : return tensor shape as const int64_t *

* gguf : remove n_dims accessor, keep only gguf_get_tensor_ne
This commit is contained in:
QuintinShaw
2026-07-13 18:55:15 +08:00
committed by GitHub
parent 91c631b21d
commit ad8d821991
3 changed files with 19 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/test-gguf.cpp
View File

@@ -662,6 +662,13 @@ static bool handcrafted_check_tensors(const gguf_context * gguf_ctx, const unsig
if (gguf_get_tensor_type(gguf_ctx, id) != type) {
ok = false;
}
const int64_t * ne = gguf_get_tensor_ne(gguf_ctx, id);
for (int j = 0; j < GGML_MAX_DIMS; ++j) {
if (ne[j] != shape[j]) {
ok = false;
}
}
} else {
ok = false;
continue;