Skip to content

Commit

Permalink
[MISC][XPU] quick fix for XPU CI (vllm-project#10859)
Browse files Browse the repository at this point in the history
Signed-off-by: yan ma <[email protected]>
  • Loading branch information
yma11 authored Dec 3, 2024
1 parent 3bc94ca commit 2f2cdc7
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions .buildkite/run-xpu-test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@ trap remove_docker_container EXIT
remove_docker_container

# Run the image and test offline inference/tensor parallel
docker run -it -d --name xpu-test --device /dev/dri -v /dev/dri/by-path:/dev/dri/by-path xpu-test /bin/bash
docker exec xpu-test bash -c "python3 examples/offline_inference.py"
docker exec xpu-test bash -c "python3 examples/offline_inference_cli.py -tp 2"
docker run --name xpu-test --device /dev/dri -v /dev/dri/by-path:/dev/dri/by-path --entrypoint="" xpu-test sh -c '
python3 examples/offline_inference.py
python3 examples/offline_inference_cli.py -tp 2
'

0 comments on commit 2f2cdc7

Please sign in to comment.