diff options
author | Georgi Gerganov <ggerganov@gmail.com> | 2024-02-18 18:21:52 +0200 |
---|---|---|
committer | Georgi Gerganov <ggerganov@gmail.com> | 2024-02-18 18:21:52 +0200 |
commit | 1dcc3fde004787e6fc4d84c9de0bb34cd2901a3e (patch) | |
tree | d271845d3a5043d3aaf7ad7500fe4b2fbac322a5 /common/common.cpp | |
parent | 5d3de51f972055702a1859186fe7acb8f0b43dc4 (diff) |
common : fix ub (#5530)
Diffstat (limited to 'common/common.cpp')
-rw-r--r-- | common/common.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/common/common.cpp b/common/common.cpp index 9ffc3951..489462b5 100644 --- a/common/common.cpp +++ b/common/common.cpp @@ -1801,7 +1801,8 @@ void dump_kv_cache_view_seqs(const llama_kv_cache_view & view, int row_size) { if (cs_curr[j] < 0) { continue; } if (seqs.find(cs_curr[j]) == seqs.end()) { if (seqs.size() + 1 >= sizeof(slot_chars)) { break; } - seqs[cs_curr[j]] = seqs.size(); + const size_t sz = seqs.size(); + seqs[cs_curr[j]] = sz; } } if (seqs.size() + 1 >= sizeof(slot_chars)) { break; } |