ソースを参照

context : fix pos_min initialization upon error decode (#14008)

ggml-ci
Georgi Gerganov 7 ヶ月 前
コミット
9e31bec4fd
1 ファイル変更4 行追加1 行削除
  1. 4 1
      src/llama-context.cpp

+ 4 - 1
src/llama-context.cpp

@@ -1058,7 +1058,10 @@ int llama_context::decode(llama_batch & inp_batch) {
 
         if (!res) {
             // the last ubatch failed or was aborted -> remove all positions of that ubatch from the KV cache
-            llama_pos pos_min[LLAMA_MAX_PARALLEL_SEQUENCES] = { std::numeric_limits<llama_pos>::max() };
+            llama_pos pos_min[LLAMA_MAX_PARALLEL_SEQUENCES];
+            for (int s = 0; s < LLAMA_MAX_PARALLEL_SEQUENCES; ++s) {
+                pos_min[s] = std::numeric_limits<llama_pos>::max();
+            }
 
             for (uint32_t i = 0; i < ubatch.n_tokens; ++i) {
                 const auto & seq_id = ubatch.seq_id[i][0];