8000 Fix make triton command on release branch · pytorch/pytorch@9bb6ce0 · GitHub
[go: up one dir, main page]
Skip to content

Commit 9bb6ce0

Browse files
atalmanatalman
committed
Fix make triton command on release branch
1 parent eba28a6 commit 9bb6ce0

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

‎scripts/install_triton_wheel.sh

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,11 @@
11
#!/bin/bash
22
# Updates Triton to the pinned version for this copy of PyTorch
3-
pip install --index-url https://download.pytorch.org/whl/nightly/ "pytorch-triton==$(cat .ci/docker/triton_version.txt)+$(head -c 10 .ci/docker/ci_commit_pins/triton.txt)"
3+
BRANCH=$(git rev-parse --abbrev-ref HEAD)
4+
TRITON_VERSION="pytorch-triton==$(cat .ci/docker/triton_version.txt)"
5+
DOWNLOAD_PYTORCH_ORG="https://download.pytorch.org/whl"
6+
7+
if [[ "$BRANCH" =~ .*release.* ]]; then
8+
pip install --index-url ${DOWNLOAD_PYTORCH_ORG}/test/ $TRITON_VERSION
9+
else
10+
pip install --index-url ${DOWNLOAD_PYTORCH_ORG}/nightly/ $TRITON_VERSION+$(head -c 10 .ci/docker/ci_commit_pins/triton.txt)
11+
fi

0 commit comments

Comments
 (0)
0