Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
6a0f5475
Unverified
Commit
6a0f5475
authored
Apr 17, 2025
by
Michael Goin
Committed by
GitHub
Apr 17, 2025
Browse files
Add hardware print to TPU V1 test (#16792)
parent
30ed81b7
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
1 deletion
+3
-1
.buildkite/scripts/hardware_ci/run-tpu-v1-test.sh
.buildkite/scripts/hardware_ci/run-tpu-v1-test.sh
+3
-1
No files found.
.buildkite/scripts/hardware_ci/run-tpu-v1-test.sh
View file @
6a0f5475
...
...
@@ -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
\
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment