Skip to content

Commit d6fd53a

Browse files
authored
llama.cpp : use ggml_elements()
1 parent 5a0a2c5 commit d6fd53a

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

llama.cpp

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1051,11 +1051,7 @@ struct llama_model_loader {
10511051
for (int i = 0; i < n_tensors; i++) {
10521052
const char * name = gguf_get_tensor_name(ctx_gguf, i);
10531053
struct ggml_tensor * t = ggml_get_tensor(ctx_meta, name);
1054-
size_t elem = 1;
1055-
for (int j = 0; j < t->n_dims; j++) {
1056-
elem *= t->ne[j];
1057-
}
1058-
n_tot_elements += elem;
1054+
n_tot_elements += ggml_nelements(t);
10591055
}
10601056

10611057
// print meta data

0 commit comments

Comments
 (0)