fix(hiradix): wait for extra pool IO (#27550)

Co-authored-by: Zhangheng <hzh0425@apache.org>
This commit is contained in:
LJL36
2026-06-09 14:17:36 +08:00
committed by GitHub
co-authored by Zhangheng
parent dff695c76b
commit 991689fd0d
@@ -1342,6 +1342,13 @@ class HiRadixCache(RadixCache):
# unknown prefetch stop policy, just return True
return True
if (
completed
and getattr(operation, "pool_transfers", None)
and not getattr(operation, "pool_transfers_done", True)
):
can_terminate = False
operation_terminated = operation.is_terminated()
states = torch.tensor(
[1 - int(can_terminate), int(operation_terminated)],