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
4a79262e
Unverified
Commit
4a79262e
authored
Apr 23, 2026
by
liuzhenwei
Committed by
GitHub
Apr 23, 2026
Browse files
[UT][Hardware] let torchrun example tests use the default backend (#39879)
Signed-off-by:
zhenwei-intel
<
zhenwei.liu@intel.com
>
parent
3ed5231c
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
2 deletions
+4
-2
tests/distributed/test_torchrun_example.py
tests/distributed/test_torchrun_example.py
+2
-1
tests/distributed/test_torchrun_example_moe.py
tests/distributed/test_torchrun_example_moe.py
+2
-1
No files found.
tests/distributed/test_torchrun_example.py
View file @
4a79262e
...
@@ -10,7 +10,8 @@ import torch.distributed as dist
...
@@ -10,7 +10,8 @@ import torch.distributed as dist
from
vllm
import
LLM
,
SamplingParams
from
vllm
import
LLM
,
SamplingParams
from
vllm.distributed.parallel_state
import
get_world_group
from
vllm.distributed.parallel_state
import
get_world_group
dist
.
init_process_group
(
backend
=
"gloo"
)
# Let PyTorch choose the WORLD backend for the current device type.
dist
.
init_process_group
()
# Create prompts
# Create prompts
prompts
=
[
prompts
=
[
...
...
tests/distributed/test_torchrun_example_moe.py
View file @
4a79262e
...
@@ -10,7 +10,8 @@ import torch.distributed as dist
...
@@ -10,7 +10,8 @@ import torch.distributed as dist
from
vllm
import
LLM
,
SamplingParams
from
vllm
import
LLM
,
SamplingParams
from
vllm.distributed.parallel_state
import
get_tp_group
,
get_world_group
from
vllm.distributed.parallel_state
import
get_tp_group
,
get_world_group
dist
.
init_process_group
(
backend
=
"gloo"
)
# Let PyTorch choose the WORLD backend for the current device type.
dist
.
init_process_group
()
# Create prompts
# Create prompts
prompts
=
[
prompts
=
[
...
...
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