Skip to content

Commit 431fdfd

Browse files
Move test_async_tp.py to Distributed Tests (2 GPUs) section
Moved compile/test_async_tp.py from Compilation Tests to Distributed Tests (2 GPUs) section as it requires 2 GPUs to run (@multi_gpu_test decorator). Also added tests/compile/test_async_tp.py to source_file_dependencies. Signed-off-by: jasonlizhengjian <[email protected]>
1 parent 49a3b8a commit 431fdfd

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

.buildkite/test-pipeline.yaml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -399,7 +399,6 @@ steps:
399399
- pytest -v -s compile/test_fusion_attn.py
400400
- pytest -v -s compile/test_silu_mul_quant_fusion.py
401401
- pytest -v -s compile/test_sequence_parallelism.py
402-
- pytest -v -s compile/test_async_tp.py
403402
- pytest -v -s compile/test_fusion_all_reduce.py
404403
- pytest -v -s compile/test_decorator.py
405404
- pytest -v -s compile/test_noop_elimination.py
@@ -919,6 +918,7 @@ steps:
919918
- vllm/worker/worker_base.py
920919
- vllm/v1/engine/
921920
- vllm/v1/worker/
921+
- tests/compile/test_async_tp.py
922922
- tests/compile/test_basic_correctness.py
923923
- tests/compile/test_wrapper.py
924924
- tests/distributed/
@@ -932,6 +932,7 @@ steps:
932932
- TP_SIZE=1 DP_SIZE=2 pytest -v -s v1/distributed/test_external_lb_dp.py
933933
- DP_SIZE=2 pytest -v -s v1/entrypoints/openai/test_multi_api_servers.py
934934
- pytest -v -s entrypoints/llm/test_collective_rpc.py
935+
- pytest -v -s ./compile/test_async_tp.py
935936
- pytest -v -s ./compile/test_basic_correctness.py
936937
- pytest -v -s ./compile/test_wrapper.py
937938
- VLLM_TEST_SAME_HOST=1 torchrun --nproc-per-node=4 distributed/test_same_node.py | grep 'Same node test passed'

0 commit comments

Comments
 (0)