[BugFix]: Fix HiMamba HiCache prefetch hang after L3 sidecar transfer (#27366)

2026-06-05 20:37:40 +08:00
parent e1955bf57a
commit faa6286946
1 changed files with 3 additions and 0 deletions
@@ -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(