mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
5116293f7e
Triton XPU shares its version file with the community one. When the community updates Triton version, it will temporarily break the XPU CI/CD because they use different repositories and commits. To decouple Triton version bumps between the community and XPU, we propose splitting the version into two separate files. Refer the latest community triton version bump PR #153117 Pull Request resolved: https://github.com/pytorch/pytorch/pull/155313 Approved by: https://github.com/etaf, https://github.com/EikanWang, https://github.com/atalman
34 lines
1.4 KiB
Bash
Executable File
34 lines
1.4 KiB
Bash
Executable File
#!/bin/bash
|
|
# Updates Triton to the pinned version for this copy of PyTorch
|
|
PYTHON="python3"
|
|
PIP="$PYTHON -m pip"
|
|
BRANCH=$(git rev-parse --abbrev-ref HEAD)
|
|
DOWNLOAD_PYTORCH_ORG="https://download.pytorch.org/whl"
|
|
|
|
if [[ -z "${USE_XPU}" ]]; then
|
|
# Default install from PyTorch source
|
|
|
|
TRITON_VERSION="pytorch-triton==$(cat .ci/docker/triton_version.txt)"
|
|
if [[ "$BRANCH" =~ .*release.* ]]; then
|
|
${PIP} install --index-url ${DOWNLOAD_PYTORCH_ORG}/test/ $TRITON_VERSION
|
|
else
|
|
${PIP} install --index-url ${DOWNLOAD_PYTORCH_ORG}/nightly/ $TRITON_VERSION+git$(head -c 8 .ci/docker/ci_commit_pins/triton.txt)
|
|
fi
|
|
else
|
|
# The Triton xpu logic is as follows:
|
|
# 1. By default, install pre-built whls.
|
|
# 2. [Not exposed to user] If the user set `TRITON_XPU_BUILD_FROM_SOURCE=1` flag,
|
|
# it will install Triton from the source.
|
|
|
|
TRITON_VERSION="pytorch-triton-xpu==$(cat .ci/docker/triton_xpu_version.txt)"
|
|
TRITON_XPU_COMMIT_ID="$(head -c 8 .ci/docker/ci_commit_pins/triton-xpu.txt)"
|
|
if [[ -z "${TRITON_XPU_BUILD_FROM_SOURCE}" ]]; then
|
|
${PIP} install --index-url ${DOWNLOAD_PYTORCH_ORG}/nightly/ ${TRITON_VERSION}+git${TRITON_XPU_COMMIT_ID}
|
|
else
|
|
TRITON_XPU_REPO="https://github.com/intel/intel-xpu-backend-for-triton"
|
|
|
|
# force-reinstall to ensure the pinned version is installed
|
|
${PIP} install --force-reinstall "git+${TRITON_XPU_REPO}@${TRITON_XPU_COMMIT_ID}#subdirectory=python"
|
|
fi
|
|
fi
|