Skip to content

Commit 6a0f547

Browse files
authored
Add hardware print to TPU V1 test (vllm-project#16792)
1 parent 30ed81b commit 6a0f547

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

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

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,10 +17,12 @@ source /etc/environment
1717
docker run --privileged --net host --shm-size=16G -it \
1818
-e "HF_TOKEN=$HF_TOKEN" --name tpu-test \
1919
vllm-tpu /bin/bash -c "python3 -m pip install git+https://github.com/thuml/depyf.git \
20-
&& python3 -m pip install pytest \
20+
&& python3 -m pip install pytest tpu-info \
2121
&& python3 -m pip install lm_eval[api]==0.4.4 \
2222
&& export VLLM_USE_V1=1 \
2323
&& export VLLM_XLA_CHECK_RECOMPILATION=1 \
24+
&& echo HARDWARE \
25+
&& tpu-info \
2426
&& echo TEST_0 \
2527
&& pytest -v -s /workspace/vllm/tests/v1/tpu/test_perf.py \
2628
&& echo TEST_1 \

0 commit comments

Comments
 (0)