diff --git a/torch/distributed/distributed_c10d.py b/torch/distributed/distributed_c10d.py index 2b5461d0bc33d..74f2ed5845a21 100644 --- a/torch/distributed/distributed_c10d.py +++ b/torch/distributed/distributed_c10d.py @@ -1878,6 +1878,9 @@ def send(tensor: torch.Tensor, dst: int, group: Optional[ProcessGroup] = None, t """ Send a tensor synchronously. + .. warning:: + ``tag`` is not supported with the NCCL backend. + Args: tensor (Tensor): Tensor to send. dst (int): Destination rank on global process group (regardless of ``group`` argument). @@ -1913,6 +1916,9 @@ def recv(tensor: torch.Tensor, src: Optional[int] = None, group: Optional[Proces """ Receives a tensor synchronously. + .. warning:: + ``tag`` is not supported with the NCCL backend. + Args: tensor (Tensor): Tensor to fill with received data. src (int, optional): Source rank on global process group (regardless of ``group`` argument).