diff --git a/python/sglang/srt/mem_cache/hi_mamba_radix_cache.py b/python/sglang/srt/mem_cache/hi_mamba_radix_cache.py index 6b5b904f1..b028574b6 100644 --- a/python/sglang/srt/mem_cache/hi_mamba_radix_cache.py +++ b/python/sglang/srt/mem_cache/hi_mamba_radix_cache.py @@ -1662,6 +1662,9 @@ class HiMambaRadixCache(MambaRadixCache): else: return True + if completed and operation.pool_transfers and not operation.pool_transfers_done: + can_terminate = False + operation_terminated = operation.is_terminated() if self.tp_world_size > 1: states = torch.tensor(