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

fix wrong type in print

Browse source
This commit is contained in:
Johannes Gäßler 2024-12-04 14:16:05 +01:00
parent b88727009d
commit 096b847a0f
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ggml/src/ggml.c
View file

@ -6780,7 +6780,7 @@ struct gguf_context * gguf_init_from_file(const char * fname, struct gguf_init_p
break; break;
} }
const size_t type_size = ggml_type_size(info->t.type); const size_t type_size = ggml_type_size(info->t.type);
const size_t blck_size = ggml_blck_size(info->t.type); const int64_t blck_size = ggml_blck_size(info->t.type);
// check that row size is divisible by block size // check that row size is divisible by block size
if (blck_size == 0 || info->t.ne[0] % blck_size != 0) { if (blck_size == 0 || info->t.ne[0] % blck_size != 0) {
@ -7351,7 +7351,7 @@ void gguf_set_tensor_type(struct gguf_context * ctx, const char * name, enum ggm
} }
struct ggml_tensor * tensor = &ctx->info[idx].t; struct ggml_tensor * tensor = &ctx->info[idx].t;
const size_t type_size = ggml_type_size(type); const size_t type_size = ggml_type_size(type);
const int blck_size = ggml_blck_size(type); const int64_t blck_size = ggml_blck_size(type);
tensor->type = type; tensor->type = type;
GGML_ASSERT(tensor->ne[0] % blck_size == 0 && "tensor row size not divisible by block size of new type"); GGML_ASSERT(tensor->ne[0] % blck_size == 0 && "tensor row size not divisible by block size of new type");