[XPU] upgrade triton-xpu version to 3.7.1 (#25730)

Co-authored-by: Ma Mingfei <mingfei.ma@intel.com>
2026-05-21 10:29:20 +08:00
parent 4868b92d47
commit 34479c19bd
2 changed files with 2 additions and 3 deletions
@@ -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