From 0ebc6388cf13744d185c24badbf2cddb69e643f7 Mon Sep 17 00:00:00 2001 From: PyTorch MergeBot Date: Mon, 23 Dec 2024 19:37:35 +0000 Subject: [PATCH] Revert "Exclude py 31.3t triton package from PyTorch 3.13t wheel (#143218)" This reverts commit 3bfdf6f0633e6feb067e032009256c740a2a2665. Reverted https://github.com/pytorch/pytorch/pull/143218 on behalf of https://github.com/atalman due to this constrain is ignored see https://github.com/pytorch/pytorch/issues/143654 ([comment](https://github.com/pytorch/pytorch/pull/143218#issuecomment-2560208992)) --- .circleci/scripts/binary_populate_env.sh | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/.circleci/scripts/binary_populate_env.sh b/.circleci/scripts/binary_populate_env.sh index 0c5b16e2c6a..b83ba302acd 100755 --- a/.circleci/scripts/binary_populate_env.sh +++ b/.circleci/scripts/binary_populate_env.sh @@ -75,10 +75,9 @@ export PYTORCH_BUILD_NUMBER=1 TRITON_VERSION=$(cat $PYTORCH_ROOT/.ci/docker/triton_version.txt) # Here PYTORCH_EXTRA_INSTALL_REQUIREMENTS is already set for the all the wheel builds hence append TRITON_CONSTRAINT -TRITON_CONSTRAINT="platform_system == 'Linux' and platform_machine == 'x86_64' and python_version != '3.13t'" +TRITON_CONSTRAINT="platform_system == 'Linux' and platform_machine == 'x86_64'" if [[ "$PACKAGE_TYPE" =~ .*wheel.* && -n "${PYTORCH_EXTRA_INSTALL_REQUIREMENTS:-}" ]]; then TRITON_REQUIREMENT="triton==${TRITON_VERSION}; ${TRITON_CONSTRAINT}" - # Only linux Python <= 3.13, not 3.13t are supported wheels for triton if [[ -n "$PYTORCH_BUILD_VERSION" && "$PYTORCH_BUILD_VERSION" =~ .*dev.* ]]; then TRITON_SHORTHASH=$(cut -c1-8 $PYTORCH_ROOT/.ci/docker/ci_commit_pins/triton.txt) TRITON_REQUIREMENT="pytorch-triton==${TRITON_VERSION}+git${TRITON_SHORTHASH}; ${TRITON_CONSTRAINT}"