From f6dab8388d6de1e2faa6d9eb4c5cbaaaa8e0b580 Mon Sep 17 00:00:00 2001 From: Yi Sun <98612013+Boreas618@users.noreply.github.com> Date: Thu, 29 Jan 2026 14:38:11 +0800 Subject: [PATCH] [https://nvbugs/5813452][fix] Fix "Assertion failed: isLeaf() in kvCacheManager.cpp:465" (#10922) Signed-off-by: Yi Sun --- cpp/tensorrt_llm/batch_manager/kvCacheManager.cpp | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/cpp/tensorrt_llm/batch_manager/kvCacheManager.cpp b/cpp/tensorrt_llm/batch_manager/kvCacheManager.cpp index f20019a0d1..e533c00476 100644 --- a/cpp/tensorrt_llm/batch_manager/kvCacheManager.cpp +++ b/cpp/tensorrt_llm/batch_manager/kvCacheManager.cpp @@ -1636,6 +1636,11 @@ std::pair> WindowBlockManager::sto TLLM_CHECK_WITH_INFO(block->getBlockId() == bid, "Block id mismatch " + std::to_string(block->getBlockId()) + " != " + std::to_string(bid)); needMatch = false; // no matching needed for following blocks + + if (block->getPrevBlock() != nullptr) + { + block->getPrevBlock()->removeNextBlock(block->getBlockKey()); + } block->setBlockKey(blockKey, static_cast(blockKey.uniqueTokens.size()) == mTokensPerBlock); block->setPrevBlock(searchRoot); block->setPrevBlockInSeq(searchRoot);