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

llama : update llama_state_get_size after v_trans field

Browse source
This commit is contained in:
Georgi Gerganov 2024-04-25 20:06:23 +03:00
parent 0fc5c5eb74
commit 1e590ac3c9
No known key found for this signature in database
GPG key ID: 449E073F9DC10735
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama.cpp
View file

@ -16157,6 +16157,7 @@ size_t llama_state_get_size(const struct llama_context * ctx) {
const size_t s_kv_head = sizeof(uint32_t); const size_t s_kv_head = sizeof(uint32_t);
const size_t s_kv_size = sizeof(uint32_t); const size_t s_kv_size = sizeof(uint32_t);
const size_t s_kv_used = sizeof(uint32_t); const size_t s_kv_used = sizeof(uint32_t);
const size_t s_v_trans = sizeof(uint32_t);
const size_t s_kv = ctx->kv_self.total_size(); const size_t s_kv = ctx->kv_self.total_size();
const size_t s_kv_cell = sizeof(llama_pos) + sizeof(size_t) + cparams.n_seq_max*sizeof(llama_seq_id); const size_t s_kv_cell = sizeof(llama_pos) + sizeof(size_t) + cparams.n_seq_max*sizeof(llama_seq_id);
const size_t s_kv_cells = ctx->kv_self.size * s_kv_cell; const size_t s_kv_cells = ctx->kv_self.size * s_kv_cell;
@ -16174,6 +16175,7 @@ size_t llama_state_get_size(const struct llama_context * ctx) {
+ s_kv_head + s_kv_head
+ s_kv_size + s_kv_size
+ s_kv_used + s_kv_used
+ s_v_trans
+ s_kv + s_kv
+ s_kv_cells + s_kv_cells
); );