Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
7de45db9
Unverified
Commit
7de45db9
authored
Aug 03, 2025
by
Ning Xie
Committed by
GitHub
Aug 03, 2025
Browse files
[Misc] update doc comment for send (#22026)
Signed-off-by:
Andy Xie
<
andy.xning@gmail.com
>
parent
789562c2
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
3 additions
and
3 deletions
+3
-3
vllm/distributed/device_communicators/base_device_communicator.py
...tributed/device_communicators/base_device_communicator.py
+1
-1
vllm/distributed/device_communicators/cuda_communicator.py
vllm/distributed/device_communicators/cuda_communicator.py
+1
-1
vllm/distributed/parallel_state.py
vllm/distributed/parallel_state.py
+1
-1
No files found.
vllm/distributed/device_communicators/base_device_communicator.py
View file @
7de45db9
...
...
@@ -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
...
...
vllm/distributed/device_communicators/cuda_communicator.py
View file @
7de45db9
...
...
@@ -179,7 +179,7 @@ class CudaCommunicator(DeviceCommunicatorBase):
return
output
.
movedim
(
0
,
dim
).
contiguous
()
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
...
...
vllm/distributed/parallel_state.py
View file @
7de45db9
...
...
@@ -782,7 +782,7 @@ class GroupCoordinator:
torch
.
distributed
.
barrier
(
group
=
self
.
cpu_group
)
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."""
self
.
device_communicator
.
send
(
tensor
,
dst
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment