Revert "[Release Only] Build triton using pinned version rather branc…

…h" (pytorch#1751) This reverts commit ba3e20b.
atalman · Mar 21, 2024 · ced930a · ced930a
1 parent ba3e20b
commit ced930a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/conda/build_pytorch.sh b/conda/build_pytorch.sh
@@ -282,7 +282,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" ]]; then
+        if [[ -n "$OVERRIDE_PACKAGE_VERSION" && "$OVERRIDE_PACKAGE_VERSION" =~ .*dev.* ]]; then
             TRITON_SHORTHASH=$(cut -c1-10 $pytorch_rootdir/.github/ci_commit_pins/triton.txt)
             export CONDA_TRITON_CONSTRAINT="    - torchtriton==${TRITON_VERSION}+${TRITON_SHORTHASH} # [py < 312]"
         else