diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index d6a5084429..c1c5f5f90b 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -126,7 +126,6 @@ repos: -c, requirements/xpu.txt, -o, requirements/test/xpu.txt, --index-strategy, unsafe-best-match, - --torch-backend, xpu, --python-platform, x86_64-manylinux_2_39, --python-version, "3.12", ] diff --git a/requirements/test/xpu.txt b/requirements/test/xpu.txt index 39ef171115..f05459a3d5 100644 --- a/requirements/test/xpu.txt +++ b/requirements/test/xpu.txt @@ -1,5 +1,5 @@ # This file was autogenerated by uv via the following command: -# uv pip compile requirements/test/xpu.in -c requirements/xpu.txt -o requirements/test/xpu.txt --index-strategy unsafe-best-match --torch-backend xpu --python-platform x86_64-manylinux_2_39 --python-version 3.12 +# uv pip compile requirements/test/xpu.in -c requirements/xpu.txt -o requirements/test/xpu.txt --index-strategy unsafe-best-match --python-platform x86_64-manylinux_2_39 --python-version 3.12 absl-py==2.4.0 # via # -r requirements/test/xpu.in @@ -948,10 +948,14 @@ transformers==5.15.0 # compressed-tensors # sentence-transformers # xgrammar -triton==3.7.1 - # via xgrammar +triton==3.7.2+xpu + # via + # -c requirements/xpu.txt + # xgrammar triton-xpu==3.7.2 - # via torch + # via + # torch + # triton typepy==1.3.4 # via # dataproperty diff --git a/requirements/xpu.txt b/requirements/xpu.txt index 4fd0d18b2d..aa33fe3f0e 100644 --- a/requirements/xpu.txt +++ b/requirements/xpu.txt @@ -1,5 +1,7 @@ # Common dependencies -r common.txt +--extra-index-url=https://wheels.vllm.ai/xpu/ +triton==3.7.2+xpu ray>=2.9 cmake>=3.26.1