summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGeorgi Gerganov <ggerganov@gmail.com>2024-02-18 18:21:52 +0200
committerGeorgi Gerganov <ggerganov@gmail.com>2024-02-18 18:21:52 +0200
commit1dcc3fde004787e6fc4d84c9de0bb34cd2901a3e (patch)
treed271845d3a5043d3aaf7ad7500fe4b2fbac322a5
parent5d3de51f972055702a1859186fe7acb8f0b43dc4 (diff)
common : fix ub (#5530)
-rw-r--r--common/common.cpp3
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; }