Unverified Commit 1d20c347 authored by Ilya Markov's avatar Ilya Markov Committed by GitHub
Browse files

[CI] Fix `tests/distributed/test_ca_buffer_sharing.py` (#22849)


Signed-off-by: default avatarilmarkov <imarkov@redhat.com>
Co-authored-by: default avatarilmarkov <imarkov@redhat.com>
Co-authored-by: default avatarWentao Ye <44945378+yewentao256@users.noreply.github.com>
parent b6af24fb
...@@ -297,7 +297,7 @@ class CustomAllreduce: ...@@ -297,7 +297,7 @@ class CustomAllreduce:
@staticmethod @staticmethod
def free_shared_buffer(pointers: list[int], def free_shared_buffer(pointers: list[int],
group: Optional[ProcessGroup] = None, group: Optional[ProcessGroup] = None,
rank: Optional[int] = 0) -> None: rank: Optional[int] = None) -> None:
if rank is None: if rank is None:
rank = dist.get_rank(group=group) rank = dist.get_rank(group=group)
if ops is not None: if ops is not None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment