[Release 2.5] Temp change to build triton from pin (#1982)

pytorch · Sep 9, 2024 · b19cd6e · b19cd6e
1 parent 1cdd6fa
commit b19cd6e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/conda/build_pytorch.sh b/conda/build_pytorch.sh
@@ -285,7 +285,7 @@ else
     if [[ "$OSTYPE" != "msys" ]]; then
         # TODO: Remove me when Triton has a proper release channel
         TRITON_VERSION=$(cat $pytorch_rootdir/.ci/docker/triton_version.txt)
-        if [[ -n "$OVERRIDE_PACKAGE_VERSION" && "$OVERRIDE_PACKAGE_VERSION" =~ .*dev.* ]]; then
+        if [[ -n "$OVERRIDE_PACKAGE_VERSION" ]]; then
             TRITON_SHORTHASH=$(cut -c1-10 $pytorch_rootdir/.github/ci_commit_pins/triton.txt)
             export CONDA_TRITON_CONSTRAINT="    - torchtriton==${TRITON_VERSION}+${TRITON_SHORTHASH} # [py < 313]"
         else