"vscode:/vscode.git/clone" did not exist on "774d1035e4000fe3a40e01ccbf017a0f763fd6f2"
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:
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
......
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