We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 30ed81b commit 6a0f547Copy full SHA for 6a0f547
.buildkite/scripts/hardware_ci/run-tpu-v1-test.sh
@@ -17,10 +17,12 @@ source /etc/environment
17
docker run --privileged --net host --shm-size=16G -it \
18
-e "HF_TOKEN=$HF_TOKEN" --name tpu-test \
19
vllm-tpu /bin/bash -c "python3 -m pip install git+https://github.com/thuml/depyf.git \
20
- && python3 -m pip install pytest \
+ && python3 -m pip install pytest tpu-info \
21
&& python3 -m pip install lm_eval[api]==0.4.4 \
22
&& export VLLM_USE_V1=1 \
23
&& export VLLM_XLA_CHECK_RECOMPILATION=1 \
24
+ && echo HARDWARE \
25
+ && tpu-info \
26
&& echo TEST_0 \
27
&& pytest -v -s /workspace/vllm/tests/v1/tpu/test_perf.py \
28
&& echo TEST_1 \
0 commit comments