millaguie/llama.cpp
1
0 Fork 0
Code Issues Pull Requests Actions 29 Packages Projects Releases Wiki Activity

llama-batch: add n_keep_tail in split_equal for recurrent models (#25278)

Browse Source
This commit is contained in:
Aman Gupta authored and GitHub committed 2026-07-08 15:55:19 +08:00
1 parent f296fdfbed
commit 230ea9d214
10 files changed
+102 -38
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/llama-kv-cache-iswa.cpp
+1 -1
View File
@@ -206,7 +206,7 @@ llama_memory_context_ptr llama_kv_cache_iswa::init_batch(llama_batch_allocr & ba
std::vector<llama_ubatch> ubatches;
while (true) {
auto ubatch = balloc.split_equal(n_ubatch, !unified);
auto ubatch = balloc.split_equal(n_ubatch, !unified, 0);
if (ubatch.n_tokens == 0) {
break;