[XPU] upgrade triton-xpu version to 3.7.1 (#25730)
Co-authored-by: Ma Mingfei <mingfei.ma@intel.com>
This commit is contained in:
@@ -68,8 +68,7 @@ RUN --mount=type=secret,id=github_token \
|
|||||||
cp pyproject_xpu.toml pyproject.toml && \
|
cp pyproject_xpu.toml pyproject.toml && \
|
||||||
pip install . --extra-index-url https://download.pytorch.org/whl/xpu && \
|
pip install . --extra-index-url https://download.pytorch.org/whl/xpu && \
|
||||||
pip install --no-deps xgrammar==0.1.33 && \
|
pip install --no-deps xgrammar==0.1.33 && \
|
||||||
# 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 && \
|
||||||
pip install triton-xpu --index-url https://download.pytorch.org/whl/test/xpu --force-reinstall && \
|
|
||||||
# Add environment setup commands to .bashrc again (in case it was overwritten)
|
# 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
|
echo ". /home/sdp/miniforge3/bin/activate; conda activate py${PYTHON_VERSION}; cd /home/sdp" >> /home/sdp/.bashrc
|
||||||
|
|
||||||
|
|||||||
@@ -68,7 +68,7 @@ cp pyproject_xpu.toml pyproject.toml
|
|||||||
pip install --upgrade pip setuptools
|
pip install --upgrade pip setuptools
|
||||||
pip install -v . --extra-index-url https://download.pytorch.org/whl/xpu
|
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
|
# 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
|
### Install Using Docker
|
||||||
|
|||||||
Reference in New Issue
Block a user