فهرست منبع

src: make tail invalid when kv cell is intersection for mamba (#9249)

Zhenwei Jin 1 سال پیش
والد
کامیت
f1485161e5
1فایلهای تغییر یافته به همراه2 افزوده شده و 1 حذف شده
  1. 2 1
      src/llama.cpp

+ 2 - 1
src/llama.cpp

@@ -3810,7 +3810,8 @@ static bool llama_kv_cache_seq_rm(
                 if ((0 < p0 && p0 <= cell.pos) || (0 < p1 && p1 <= cell.pos)) {
                 if ((0 < p0 && p0 <= cell.pos) || (0 < p1 && p1 <= cell.pos)) {
                     return false;
                     return false;
                 }
                 }
-                if (p0 <= cell.pos && p1 < cell.pos) {
+                // invalidate tails which will be cleared
+                if (p0 <= cell.pos && cell.pos < p1) {
                     tail_id = -1;
                     tail_id = -1;
                 }
                 }
             }
             }