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