Skip to content

Commit 566398e

Browse files
CAROLZXYZXYlsy323
authored andcommitted
[CI/Build] Fix TPU V1 Test mixed use of & and && across tests (vllm-project#17968)
Signed-off-by: Siyuan Liu <[email protected]>
1 parent 72a3f6b commit 566398e

File tree

1 file changed

+21
-21
lines changed

1 file changed

+21
-21
lines changed

.buildkite/scripts/hardware_ci/run-tpu-v1-test.sh

Lines changed: 21 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -26,27 +26,27 @@ docker run --privileged --net host --shm-size=16G -it \
2626
&& tpu-info \
2727
&& { \
2828
echo TEST_0: Running test_perf.py; \
29-
pytest -s -v /workspace/vllm/tests/tpu/test_perf.py; \
29+
python3 -m pytest -s -v /workspace/vllm/tests/tpu/test_perf.py; \
3030
echo TEST_0_EXIT_CODE: \$?; \
3131
} & \
32-
&& { \
32+
{ \
3333
echo TEST_1: Running test_compilation.py; \
34-
pytest -s -v /workspace/vllm/tests/tpu/test_compilation.py; \
34+
python3 -m pytest -s -v /workspace/vllm/tests/tpu/test_compilation.py; \
3535
echo TEST_1_EXIT_CODE: \$?; \
3636
} & \
3737
{ \
3838
echo TEST_2: Running test_basic.py; \
39-
pytest -s -v /workspace/vllm/tests/v1/tpu/test_basic.py; \
39+
python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_basic.py; \
4040
echo TEST_2_EXIT_CODE: \$?; \
4141
} & \
4242
{ \
4343
echo TEST_3: Running test_accuracy.py::test_lm_eval_accuracy_v1_engine; \
44-
pytest -s -v /workspace/vllm/tests/entrypoints/llm/test_accuracy.py::test_lm_eval_accuracy_v1_engine; \
44+
python3 -m pytest -s -v /workspace/vllm/tests/entrypoints/llm/test_accuracy.py::test_lm_eval_accuracy_v1_engine; \
4545
echo TEST_3_EXIT_CODE: \$?; \
4646
} & \
4747
{ \
4848
echo TEST_4: Running test_quantization_accuracy.py; \
49-
pytest -s -v /workspace/vllm/tests/tpu/test_quantization_accuracy.py; \
49+
python3 -m pytest -s -v /workspace/vllm/tests/tpu/test_quantization_accuracy.py; \
5050
echo TEST_4_EXIT_CODE: \$?; \
5151
} & \
5252
{ \
@@ -56,43 +56,43 @@ docker run --privileged --net host --shm-size=16G -it \
5656
} & \
5757
{ \
5858
echo TEST_6: Running test_tpu_model_runner.py; \
59-
pytest -s -v /workspace/vllm/tests/tpu/worker/test_tpu_model_runner.py; \
59+
python3 -m pytest -s -v /workspace/vllm/tests/tpu/worker/test_tpu_model_runner.py; \
6060
echo TEST_6_EXIT_CODE: \$?; \
6161
} & \
62-
&& { \
62+
{ \
6363
echo TEST_7: Running test_sampler.py; \
64-
pytest -s -v /workspace/vllm/tests/v1/tpu/test_sampler.py; \
64+
python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_sampler.py; \
6565
echo TEST_7_EXIT_CODE: \$?; \
6666
} & \
67-
&& { \
67+
{ \
6868
echo TEST_8: Running test_topk_topp_sampler.py; \
69-
pytest -s -v /workspace/vllm/tests/v1/tpu/test_topk_topp_sampler.py; \
69+
python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_topk_topp_sampler.py; \
7070
echo TEST_8_EXIT_CODE: \$?; \
7171
} & \
72-
&& { \
72+
{ \
7373
echo TEST_9: Running test_multimodal.py; \
74-
pytest -s -v /workspace/vllm/tests/v1/tpu/test_multimodal.py; \
74+
python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_multimodal.py; \
7575
echo TEST_9_EXIT_CODE: \$?; \
7676
} & \
77-
&& { \
77+
{ \
7878
echo TEST_10: Running test_pallas.py; \
79-
pytest -s -v /workspace/vllm/tests/v1/tpu/test_pallas.py; \
79+
python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_pallas.py; \
8080
echo TEST_10_EXIT_CODE: \$?; \
8181
} & \
82-
&& { \
82+
{ \
8383
echo TEST_11: Running test_struct_output_generate.py; \
84-
pytest -s -v /workspace/vllm/tests/v1/entrypoints/llm/test_struct_output_generate.py; \
84+
python3 -m pytest -s -v /workspace/vllm/tests/v1/entrypoints/llm/test_struct_output_generate.py; \
8585
echo TEST_11_EXIT_CODE: \$?; \
8686
} & \
87-
&& { \
87+
{ \
8888
echo TEST_12: Running test_moe_pallas.py; \
89-
pytest -s -v /workspace/vllm/tests/tpu/test_moe_pallas.py; \
89+
python3 -m pytest -s -v /workspace/vllm/tests/tpu/test_moe_pallas.py; \
9090
echo TEST_12_EXIT_CODE: \$?; \
9191
} & \
9292
# Disable the TPU LoRA tests until the feature is activated
93-
# && { \
93+
# & { \
9494
# echo TEST_13: Running test_moe_pallas.py; \
95-
# pytest -s -v /workspace/vllm/tests/tpu/lora/; \
95+
# python3 -m pytest -s -v /workspace/vllm/tests/tpu/lora/; \
9696
# echo TEST_13_EXIT_CODE: \$?; \
9797
# } & \
9898
wait \

0 commit comments

Comments
 (0)