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

Signed-off-by: Rui Qiao <ruisearch42@gmail.com>
This commit is contained in:
Rui Qiao 2024-08-05 11:10:16 -07:00 committed by GitHub
parent 57f560aa23
commit 997cf78308
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -713,8 +713,8 @@ class GroupCoordinator:
size: torch.Size,
dtype: torch.dtype,
src: Optional[int] = None) -> torch.Tensor:
"""Receives a tensor from the src rank."""
"""NOTE: `src` is the local rank of the destination rank."""
"""Receives a tensor from the source rank."""
"""NOTE: `src` is the local rank of the source rank."""
if src is None:
src = (self.rank_in_group - 1) % self.world_size