fix(hiradix): wait for extra pool IO (#27550)
Co-authored-by: Zhangheng <hzh0425@apache.org>
This commit is contained in:
@@ -1342,6 +1342,13 @@ class HiRadixCache(RadixCache):
|
|||||||
# unknown prefetch stop policy, just return True
|
# unknown prefetch stop policy, just return True
|
||||||
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()
|
operation_terminated = operation.is_terminated()
|
||||||
states = torch.tensor(
|
states = torch.tensor(
|
||||||
[1 - int(can_terminate), int(operation_terminated)],
|
[1 - int(can_terminate), int(operation_terminated)],
|
||||||
|
|||||||
Reference in New Issue
Block a user