biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[TPU][CI] Fix TPUModelRunner Test (#15667)

Browse Source 
Signed-off-by: Robert Shaw <robshaw@redhat.com>
Co-authored-by: Robert Shaw <robshaw@redhat.com>
This commit is contained in:
Robert Shaw
2025-03-28 03:01:26 -04:00
committed by GitHub
parent 355f66348c
commit 2d9045fce8
2 changed files with 2 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.buildkite/run-tpu-v1-test.sh
View File

@@ -30,7 +30,7 @@ docker run --privileged --net host --shm-size=16G -it \
&& echo TEST_4 \
&& python3 /workspace/vllm/examples/offline_inference/tpu.py \
&& echo TEST_5 \
&& pytest -s -v /workspace/vllm/tests/tpu/worker/test_tpu_model_runner.py \
&& pytest -s -v /workspace/vllm/tests/v1/tpu/worker/test_tpu_model_runner.py \
&& echo TEST_6 \
&& pytest -s -v /workspace/vllm/tests/v1/tpu/test_sampler.py" \