Skip to content

Commit

Permalink
[CI/Build] Fix IDC hpu [Device not found] issue (vllm-project#10384)
Browse files Browse the repository at this point in the history 
Signed-off-by: Chendi Xue <chendi.xue@intel.com>
Signed-off-by: Linkun Chen <lkchen+anyscale@github.com>
  • Loading branch information
@xuechendi
xuechendi authored and Linkun Chen committed Nov 18, 2024
1 parent 7fa97cf commit 629f512
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion .buildkite/run-hpu-test.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -13,4 +13,4 @@ trap remove_docker_container EXIT
remove_docker_container

# Run the image and launch offline inference
docker run --runtime=habana --name=hpu-test --network=host -e VLLM_SKIP_WARMUP=true --entrypoint="" hpu-test-env python3 examples/offline_inference.py
docker run --runtime=habana --name=hpu-test --network=host -e HABANA_VISIBLE_DEVICES=all -e VLLM_SKIP_WARMUP=true --entrypoint="" hpu-test-env python3 examples/offline_inference.py

0 comments on commit 629f512

Please sign in to comment.