From 34479c19bd037e7a68326371f74002f5cf25bad7 Mon Sep 17 00:00:00 2001 From: jiayisunx Date: Thu, 21 May 2026 10:29:20 +0800 Subject: [PATCH] [XPU] upgrade triton-xpu version to 3.7.1 (#25730) Co-authored-by: Ma Mingfei --- docker/xpu.Dockerfile | 3 +-- docs_new/docs/hardware-platforms/xpu.mdx | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/docker/xpu.Dockerfile b/docker/xpu.Dockerfile index 2c9b765c3..1a3de356b 100644 --- a/docker/xpu.Dockerfile +++ b/docker/xpu.Dockerfile @@ -68,8 +68,7 @@ RUN --mount=type=secret,id=github_token \ cp pyproject_xpu.toml pyproject.toml && \ pip install . --extra-index-url https://download.pytorch.org/whl/xpu && \ pip install --no-deps xgrammar==0.1.33 && \ - # index will change after torch 2.12 release - pip install triton-xpu --index-url https://download.pytorch.org/whl/test/xpu --force-reinstall && \ + pip install triton-xpu==3.7.1 --index-url https://download.pytorch.org/whl/xpu --force-reinstall && \ # Add environment setup commands to .bashrc again (in case it was overwritten) echo ". /home/sdp/miniforge3/bin/activate; conda activate py${PYTHON_VERSION}; cd /home/sdp" >> /home/sdp/.bashrc diff --git a/docs_new/docs/hardware-platforms/xpu.mdx b/docs_new/docs/hardware-platforms/xpu.mdx index 8eae4fd67..b5a6c8a2c 100644 --- a/docs_new/docs/hardware-platforms/xpu.mdx +++ b/docs_new/docs/hardware-platforms/xpu.mdx @@ -68,7 +68,7 @@ cp pyproject_xpu.toml pyproject.toml pip install --upgrade pip setuptools pip install -v . --extra-index-url https://download.pytorch.org/whl/xpu # Using this version of triton-xpu to avoid a bug in the version shipped with torch 2.11.0+xpu -pip install triton-xpu --index-url https://download.pytorch.org/whl/test/xpu --force-reinstall # index will change after torch 2.12 release +pip install triton-xpu==3.7.1 --index-url https://download.pytorch.org/whl/xpu --force-reinstall ``` ### Install Using Docker