cont : fix prune check

ggerganov · ggerganov · commit edf9c536c1f3 · 2025-05-15T21:27:50.000+03:00
ggml-ci
diff --git a/src/llama-kv-cache.cpp b/src/llama-kv-cache.cpp
@@ -1225,6 +1225,10 @@ uint32_t llama_kv_cache_unified::cell_max() const {
 }
 
 bool llama_kv_cache_unified::is_masked_swa(llama_pos p0, llama_pos p1) const {
+    if (p0 < 0) {
+        return true;
+    }
+
     switch (swa_type) {
         case LLAMA_SWA_TYPE_NONE:
             {

Original file line number	Diff line number	Diff line change
`@@ -1225,6 +1225,10 @@ uint32_t llama_kv_cache_unified::cell_max() const {`
`1225`	`1225`	`}`
`1226`	`1226`
`1227`	`1227`	`bool llama_kv_cache_unified::is_masked_swa(llama_pos p0, llama_pos p1) const {`
	`1228`	`+ if (p0 < 0) {`
	`1229`	`+ return true;`
	`1230`	`+ }`
	`1231`	`+`
`1228`	`1232`	`switch (swa_type) {`
`1229`	`1233`	`case LLAMA_SWA_TYPE_NONE:`
`1230`	`1234`	`{`