We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8f7bace commit 1766fadCopy full SHA for 1766fad
.buildkite/scripts/hardware_ci/run-tpu-v1-test.sh
@@ -16,9 +16,11 @@ source /etc/environment
16
# Run a simple end-to-end example.
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 \
+ vllm-tpu /bin/bash -c "rm -r ~/.cache/vllm/xla_cache \
20
+ && python3 -m pip install git+https://github.com/thuml/depyf.git \
21
&& python3 -m pip install pytest pytest-asyncio tpu-info \
22
&& python3 -m pip install lm_eval[api]==0.4.4 \
23
+ && export VLLM_XLA_CACHE_PATH= \
24
&& export VLLM_USE_V1=1 \
25
&& export VLLM_XLA_CHECK_RECOMPILATION=1 \
26
&& echo HARDWARE \
0 commit comments