Skip to content

Commit 7de45db

Browse files
authored
[Misc] update doc comment for send (#22026)
Signed-off-by: Andy Xie <[email protected]>
1 parent 789562c commit 7de45db

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

vllm/distributed/device_communicators/base_device_communicator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -219,7 +219,7 @@ def gather(self,
219219
return output_tensor
220220

221221
def send(self, tensor: torch.Tensor, dst: Optional[int] = None) -> None:
222-
"""Sends a tensor to the destination rank in a non-blocking way"""
222+
"""Sends a tensor to the destination rank in a blocking way"""
223223
"""NOTE: `dst` is the local rank of the destination rank."""
224224
if dst is None:
225225
dst = (self.rank_in_group + 1) % self.world_size

vllm/distributed/device_communicators/cuda_communicator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -179,7 +179,7 @@ def reduce_scatterv(self,
179179
return output.movedim(0, dim).contiguous()
180180

181181
def send(self, tensor: torch.Tensor, dst: Optional[int] = None) -> None:
182-
"""Sends a tensor to the destination rank in a non-blocking way"""
182+
"""Sends a tensor to the destination rank in a blocking way"""
183183
"""NOTE: `dst` is the local rank of the destination rank."""
184184
if dst is None:
185185
dst = (self.rank_in_group + 1) % self.world_size

vllm/distributed/parallel_state.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -782,7 +782,7 @@ def barrier(self):
782782
torch.distributed.barrier(group=self.cpu_group)
783783

784784
def send(self, tensor: torch.Tensor, dst: Optional[int] = None) -> None:
785-
"""Sends a tensor to the destination rank in a non-blocking way"""
785+
"""Sends a tensor to the destination rank in a blocking way"""
786786
"""NOTE: `dst` is the local rank of the destination rank."""
787787
self.device_communicator.send(tensor, dst)
788788

0 commit comments

Comments
 (0)