root/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/llama.cpp.git synced 2024-12-25 10:54:36 +00:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Zhenwei Jin 2024-09-03 01:53:23 +08:00 committed by GitHub
parent 048de848ee
commit f1485161e5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/llama.cpp
View File

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