[ci] feat: make the test_torchrun_example run with tp=2, external_dp=2 (#15172)

Signed-off-by: Chi Zhang <zhangchi.usc1992@bytedance.com>
Signed-off-by: youkaichao <youkaichao@gmail.com>
Co-authored-by: youkaichao <youkaichao@gmail.com>
This commit is contained in:
Chi Zhang
2025-03-21 00:30:04 +08:00
committed by GitHub
parent 5a0905ba2a
commit 086b56824c
2 changed files with 6 additions and 2 deletions

View File

@@ -9,6 +9,8 @@ import torch.distributed as dist
from vllm import LLM, SamplingParams
from vllm.distributed.parallel_state import get_world_group
dist.init_process_group(backend="gloo")
# Create prompts
prompts = [
"Hello, my name is",