Skip to content

Commit 0fabc3b

Browse files
pytorchbotatalman
andauthored
CUDA aarch64 12.6 and 12.8 builds fix triton constraints (pytorch#165022)
CUDA aarch64 12.6 and 12.8 builds fix triton constraints (pytorch#165013) Since we have introduced CUDA aarch64 builds for all cuda versions we need to remove this constraint. This was missed by pytorch#162364 Proper constraint on triton should be: ``` Requires-Dist: triton==3.5.0; platform_system == "Linux" ``` not: ``` Requires-Dist: triton==3.5.0; platform_system == "Linux" and platform_machine == "x86_64" ``` Pull Request resolved: pytorch#165013 Approved by: https://github.com/Camyll, https://github.com/nWEIdia, https://github.com/tinglvv (cherry picked from commit 81dbeb0) Co-authored-by: atalman <[email protected]>
1 parent 26e023a commit 0fabc3b

File tree

1 file changed

+1
-8
lines changed

1 file changed

+1
-8
lines changed

.circleci/scripts/binary_populate_env.sh

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -71,14 +71,7 @@ export PYTORCH_BUILD_NUMBER=1
7171

7272
# Set triton version as part of PYTORCH_EXTRA_INSTALL_REQUIREMENTS
7373
TRITON_VERSION=$(cat $PYTORCH_ROOT/.ci/docker/triton_version.txt)
74-
75-
# Here PYTORCH_EXTRA_INSTALL_REQUIREMENTS is already set for the all the wheel builds hence append TRITON_CONSTRAINT
76-
TRITON_CONSTRAINT="platform_system == 'Linux' and platform_machine == 'x86_64'"
77-
78-
# CUDA 12.9/13.0 builds have triton for Linux and Linux aarch64 binaries.
79-
if [[ "$DESIRED_CUDA" == "cu129" ]] || [[ "$DESIRED_CUDA" == "cu130" ]]; then
80-
TRITON_CONSTRAINT="platform_system == 'Linux'"
81-
fi
74+
TRITON_CONSTRAINT="platform_system == 'Linux'"
8275

8376
if [[ "$PACKAGE_TYPE" =~ .*wheel.* && -n "${PYTORCH_EXTRA_INSTALL_REQUIREMENTS:-}" && ! "$PYTORCH_BUILD_VERSION" =~ .*xpu.* ]]; then
8477
TRITON_REQUIREMENT="triton==${TRITON_VERSION}; ${TRITON_CONSTRAINT}"

0 commit comments

Comments
 (0)