Unverified Commit 4a79262e authored by liuzhenwei's avatar liuzhenwei Committed by GitHub
Browse files

[UT][Hardware] let torchrun example tests use the default backend (#39879)


Signed-off-by: default avatarzhenwei-intel <zhenwei.liu@intel.com>
parent 3ed5231c
...@@ -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 = [
......
...@@ -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 = [
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment