Skip to content

Commit 11dbbd0

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 89a10ce commit 11dbbd0

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
@@ -400,7 +400,6 @@ steps:
400400
- pytest -v -s compile/test_functionalization.py
401401
- pytest -v -s compile/test_silu_mul_quant_fusion.py
402402
- pytest -v -s compile/test_sequence_parallelism.py
403-
- pytest -v -s compile/test_async_tp.py
404403
- pytest -v -s compile/test_fusion_all_reduce.py
405404
- pytest -v -s compile/test_decorator.py
406405
- pytest -v -s compile/test_noop_elimination.py
@@ -920,6 +919,7 @@ steps:
920919
- vllm/worker/worker_base.py
921920
- vllm/v1/engine/
922921
- vllm/v1/worker/
922+
- tests/compile/test_async_tp.py
923923
- tests/compile/test_basic_correctness.py
924924
- tests/compile/test_wrapper.py
925925
- tests/distributed/
@@ -933,6 +933,7 @@ steps:
933933
- TP_SIZE=1 DP_SIZE=2 pytest -v -s v1/distributed/test_external_lb_dp.py
934934
- DP_SIZE=2 pytest -v -s v1/entrypoints/openai/test_multi_api_servers.py
935935
- pytest -v -s entrypoints/llm/test_collective_rpc.py
936+
- pytest -v -s ./compile/test_async_tp.py
936937
- pytest -v -s ./compile/test_basic_correctness.py
937938
- pytest -v -s ./compile/test_wrapper.py
938939
- 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)