Unverified Commit 997cf783 authored by Rui Qiao's avatar Rui Qiao Committed by GitHub
Browse files

[Misc] Fix typo in GroupCoordinator.recv() (#7167)


Signed-off-by: default avatarRui Qiao <ruisearch42@gmail.com>
parent 57f560aa
...@@ -713,8 +713,8 @@ class GroupCoordinator: ...@@ -713,8 +713,8 @@ class GroupCoordinator:
size: torch.Size, size: torch.Size,
dtype: torch.dtype, dtype: torch.dtype,
src: Optional[int] = None) -> torch.Tensor: src: Optional[int] = None) -> torch.Tensor:
"""Receives a tensor from the src rank.""" """Receives a tensor from the source rank."""
"""NOTE: `src` is the local rank of the destination rank.""" """NOTE: `src` is the local rank of the source rank."""
if src is None: if src is None:
src = (self.rank_in_group - 1) % self.world_size src = (self.rank_in_group - 1) % self.world_size
......
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