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
1b44aaf4
Unverified
Commit
1b44aaf4
authored
Jun 17, 2024
by
youkaichao
Committed by
GitHub
Jun 17, 2024
Browse files
[bugfix][distributed] fix 16 gpus local rank arrangement (#5604)
parent
9e4e6fe2
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
0 deletions
+6
-0
vllm/executor/ray_gpu_executor.py
vllm/executor/ray_gpu_executor.py
+6
-0
No files found.
vllm/executor/ray_gpu_executor.py
View file @
1b44aaf4
...
@@ -137,6 +137,12 @@ class RayGPUExecutor(DistributedGPUExecutor):
...
@@ -137,6 +137,12 @@ class RayGPUExecutor(DistributedGPUExecutor):
for
i
,
(
node_id
,
gpu_ids
)
in
enumerate
(
worker_node_and_gpu_ids
):
for
i
,
(
node_id
,
gpu_ids
)
in
enumerate
(
worker_node_and_gpu_ids
):
node_workers
[
node_id
].
append
(
i
)
node_workers
[
node_id
].
append
(
i
)
# `gpu_ids` can be a list of strings or integers.
# convert them to integers for consistency.
# NOTE: gpu_ids can be larger than 9 (e.g. 16 GPUs),
# string sorting is not sufficient.
# see https://github.com/vllm-project/vllm/issues/5590
gpu_ids
=
[
int
(
x
)
for
x
in
gpu_ids
]
node_gpus
[
node_id
].
extend
(
gpu_ids
)
node_gpus
[
node_id
].
extend
(
gpu_ids
)
for
node_id
,
gpu_ids
in
node_gpus
.
items
():
for
node_id
,
gpu_ids
in
node_gpus
.
items
():
node_gpus
[
node_id
]
=
sorted
(
gpu_ids
)
node_gpus
[
node_id
]
=
sorted
(
gpu_ids
)
...
...
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