Add hardware print to TPU V1 test (#16792)

This commit is contained in:
Michael Goin 2025-04-17 23:13:26 -06:00 committed by GitHub
parent 30ed81b7ca
commit 6a0f547561
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -17,10 +17,12 @@ source /etc/environment
docker run --privileged --net host --shm-size=16G -it \
-e "HF_TOKEN=$HF_TOKEN" --name tpu-test \
vllm-tpu /bin/bash -c "python3 -m pip install git+https://github.com/thuml/depyf.git \
&& python3 -m pip install pytest \
&& python3 -m pip install pytest tpu-info \
&& python3 -m pip install lm_eval[api]==0.4.4 \
&& export VLLM_USE_V1=1 \
&& export VLLM_XLA_CHECK_RECOMPILATION=1 \
&& echo HARDWARE \
&& tpu-info \
&& echo TEST_0 \
&& pytest -v -s /workspace/vllm/tests/v1/tpu/test_perf.py \
&& echo TEST_1 \