[Misc] update doc comment for send (#22026)

Signed-off-by: Andy Xie <andy.xning@gmail.com>
This commit is contained in:
Ning Xie
2025-08-03 15:55:20 +08:00
committed by GitHub
parent 789562c28c
commit 7de45db9a5
3 changed files with 3 additions and 3 deletions

View File

@@ -219,7 +219,7 @@ class DeviceCommunicatorBase:
return output_tensor
def send(self, tensor: torch.Tensor, dst: Optional[int] = None) -> None:
"""Sends a tensor to the destination rank in a non-blocking way"""
"""Sends a tensor to the destination rank in a blocking way"""
"""NOTE: `dst` is the local rank of the destination rank."""
if dst is None:
dst = (self.rank_in_group + 1) % self.world_size