Skip to content

Commit 3ca240c

Browse files
committed
Separate the existing tests to 2 folders.
Signed-off-by: Qiliang Cui <derrhein@gmail.com>
1 parent f7dc6af commit 3ca240c

File tree

2 files changed

+6
-40
lines changed

2 files changed

+6
-40
lines changed

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

Lines changed: 6 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -127,39 +127,17 @@ run_and_track_test() {
127127
}
128128
129129
# --- Actual Test Execution ---
130-
run_and_track_test 0 "test_perf.py" \
131-
"python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_perf.py"
132-
run_and_track_test 1 "test_compilation.py" \
133-
"python3 -m pytest -s -v /workspace/vllm/tests/tpu/test_compilation.py"
134-
run_and_track_test 2 "test_basic.py" \
135-
"python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_basic.py"
136-
run_and_track_test 3 "test_accuracy.py::test_lm_eval_accuracy_v1_engine" \
137-
"python3 -m pytest -s -v /workspace/vllm/tests/entrypoints/llm/test_accuracy.py::test_lm_eval_accuracy_v1_engine"
138-
run_and_track_test 4 "test_quantization_accuracy.py" \
139-
"python3 -m pytest -s -v /workspace/vllm/tests/tpu/test_quantization_accuracy.py"
140-
run_and_track_test 5 "examples/offline_inference/tpu.py" \
141-
"python3 /workspace/vllm/examples/offline_inference/tpu.py"
142-
run_and_track_test 6 "test_tpu_model_runner.py" \
143-
"python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/worker/test_tpu_model_runner.py"
144-
run_and_track_test 7 "test_sampler.py" \
145-
"python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_sampler.py"
146-
run_and_track_test 8 "test_topk_topp_sampler.py" \
147-
"python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_topk_topp_sampler.py"
148-
run_and_track_test 9 "test_multimodal.py" \
149-
"python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_multimodal.py"
150-
run_and_track_test 10 "test_pallas.py" \
151-
"python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_pallas.py"
152-
run_and_track_test 11 "test_struct_output_generate.py" \
130+
run_and_track_test 1 "test_struct_output_generate.py" \
153131
"python3 -m pytest -s -v /workspace/vllm/tests/v1/entrypoints/llm/test_struct_output_generate.py -k \"not test_structured_output_with_reasoning_matrices\""
154-
run_and_track_test 12 "test_moe_pallas.py" \
132+
run_and_track_test 2 "test_moe_pallas.py" \
155133
"python3 -m pytest -s -v /workspace/vllm/tests/tpu/test_moe_pallas.py"
156-
run_and_track_test 13 "test_lora.py" \
134+
run_and_track_test 3 "test_lora.py" \
157135
"VLLM_XLA_CHECK_RECOMPILATION=0 python3 -m pytest -s -v /workspace/vllm/tests/tpu/lora/test_lora.py"
158-
run_and_track_test 14 "test_tpu_qkv_linear.py" \
136+
run_and_track_test 4 "test_tpu_qkv_linear.py" \
159137
"python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_tpu_qkv_linear.py"
160-
run_and_track_test 15 "test_spmd_model_weight_loading.py" \
138+
run_and_track_test 5 "test_spmd_model_weight_loading.py" \
161139
"python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_spmd_model_weight_loading.py"
162-
run_and_track_test 16 "test_kv_cache_update_kernel.py" \
140+
run_and_track_test 6 "test_kv_cache_update_kernel.py" \
163141
"python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_kv_cache_update_kernel.py"
164142
165143
# After all tests have been attempted, exit with the overall status.

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

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -149,18 +149,6 @@ run_and_track_test 9 "test_multimodal.py" \
149149
"python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_multimodal.py"
150150
run_and_track_test 10 "test_pallas.py" \
151151
"python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_pallas.py"
152-
run_and_track_test 11 "test_struct_output_generate.py" \
153-
"python3 -m pytest -s -v /workspace/vllm/tests/v1/entrypoints/llm/test_struct_output_generate.py -k \"not test_structured_output_with_reasoning_matrices\""
154-
run_and_track_test 12 "test_moe_pallas.py" \
155-
"python3 -m pytest -s -v /workspace/vllm/tests/tpu/test_moe_pallas.py"
156-
run_and_track_test 13 "test_lora.py" \
157-
"VLLM_XLA_CHECK_RECOMPILATION=0 python3 -m pytest -s -v /workspace/vllm/tests/tpu/lora/test_lora.py"
158-
run_and_track_test 14 "test_tpu_qkv_linear.py" \
159-
"python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_tpu_qkv_linear.py"
160-
run_and_track_test 15 "test_spmd_model_weight_loading.py" \
161-
"python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_spmd_model_weight_loading.py"
162-
run_and_track_test 16 "test_kv_cache_update_kernel.py" \
163-
"python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_kv_cache_update_kernel.py"
164152
165153
# After all tests have been attempted, exit with the overall status.
166154
if [ "$overall_script_exit_code" -ne 0 ]; then

0 commit comments

Comments
 (0)