From 997cf78308d292b03c8a1e68d8d1a1f599551937 Mon Sep 17 00:00:00 2001 From: Rui Qiao <161574667+ruisearch42@users.noreply.github.com> Date: Mon, 5 Aug 2024 11:10:16 -0700 Subject: [PATCH] [Misc] Fix typo in GroupCoordinator.recv() (#7167) Signed-off-by: Rui Qiao --- vllm/distributed/parallel_state.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vllm/distributed/parallel_state.py b/vllm/distributed/parallel_state.py index d7ca8fd82e1a2..a20b92de81cda 100644 --- a/vllm/distributed/parallel_state.py +++ b/vllm/distributed/parallel_state.py @@ -713,8 +713,8 @@ def recv(self, 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