diff --git a/conda/build_pytorch.sh b/conda/build_pytorch.sh index 1cc09db85..3688d130e 100755 --- 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" && "$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 < 312]" else