From: Georgi Gerganov <redacted>
Date: Sun, 18 Feb 2024 16:21:52 +0000 (+0200)
Subject: common : fix ub (#5530)
X-Git-Tag: upstream/0.0.4488~2312
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=1dcc3fde004787e6fc4d84c9de0bb34cd2901a3e;p=pkg%2Fggml%2Fsources%2Fllama.cpp

common : fix ub (#5530)
---

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; }