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

This commit is contained in:
Zhangheng
2026-06-05 20:37:40 +08:00
committed by GitHub
parent e1955bf57a
commit faa6286946
@@ -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(