[CI] Fix nonexistent dependency for data-parallel example test selection (#53026)
Signed-off-by: Taneem Ibrahim <taneem.ibrahim@gmail.com>
Taneem Ibrahim committed
Aug 20, 2026 at 04:11 UTC
5d4d470470aed5a4a2f2be1268176c3199cc4f8e
2 files changed
+3
-3
.buildkite/test-amd.yaml
+1
-1
@@ -858,7 +858,7 @@ steps:
858
- tests/distributed/test_torchrun_example.py
859
- tests/distributed/test_torchrun_example_moe.py
860
- examples/rl/
861
- - tests/examples/features/data_parallel/data_parallel_offline.py
861
+ - examples/features/data_parallel/data_parallel_offline.py
862
- vllm/platforms/rocm.py
863
commands:
864
- torchrun --nproc-per-node=4 distributed/test_torchrun_example.py
.buildkite/test_areas/distributed.yaml
+2
-2
@@ -116,7 +116,7 @@ steps:
116
- tests/distributed/test_torchrun_example.py
117
- tests/distributed/test_torchrun_example_moe.py
118
- examples/rl/
119
- - tests/examples/features/data_parallel/data_parallel_offline.py
119
+ - examples/features/data_parallel/data_parallel_offline.py
120
commands:
121
# https://github.com/NVIDIA/nccl/issues/1838
122
- export NCCL_CUMEM_HOST_ENABLE=0
@@ -265,7 +265,7 @@ steps:
265
- vllm/executor/
266
- vllm/model_executor/models/
267
- tests/distributed/
268
- - tests/examples/features/data_parallel/data_parallel_offline.py
268
+ - examples/features/data_parallel/data_parallel_offline.py
269
commands:
270
- ./.buildkite/scripts/run-multi-node-test.sh /vllm-workspace/tests 2 2 $IMAGE_TAG "VLLM_TEST_SAME_HOST=0 torchrun --nnodes 2 --nproc-per-node=2 --rdzv_backend=c10d --rdzv_endpoint=192.168.10.10 distributed/test_same_node.py | grep 'Same node test passed' && NUM_NODES=2 torchrun --nnodes 2 --nproc-per-node=2 --rdzv_backend=c10d --rdzv_endpoint=192.168.10.10 distributed/test_node_count.py | grep 'Node count test passed' && python3 ../examples/features/data_parallel/data_parallel_offline.py -dp=2 -tp=1 --dp-num-nodes=2 --dp-node-rank=0 --dp-master-addr=192.168.10.10 --dp-master-port=12345 --enforce-eager --trust-remote-code && VLLM_MULTI_NODE=1 pytest -v -s distributed/test_multi_node_assignment.py && VLLM_MULTI_NODE=1 pytest -v -s distributed/test_pipeline_parallel.py" "VLLM_TEST_SAME_HOST=0 torchrun --nnodes 2 --nproc-per-node=2 --rdzv_backend=c10d --rdzv_endpoint=192.168.10.10 distributed/test_same_node.py | grep 'Same node test passed' && NUM_NODES=2 torchrun --nnodes 2 --nproc-per-node=2 --rdzv_backend=c10d --rdzv_endpoint=192.168.10.10 distributed/test_node_count.py | grep 'Node count test passed' && python3 ../examples/features/data_parallel/data_parallel_offline.py -dp=2 -tp=1 --dp-num-nodes=2 --dp-node-rank=1 --dp-master-addr=192.168.10.10 --dp-master-port=12345 --enforce-eager --trust-remote-code"
271