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: Tyler Michael Smith <tyler@neuralmagic.com>
  • Loading branch information
@xuechendi @tlrmchlsmth
xuechendi authored and tlrmchlsmth committed Nov 23, 2024
1 parent 8d7744b commit f5478b1
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 f5478b1

Please sign in to comment.