From ef8aabd42422725026cb4dbf48aafa9efa226a04 Mon Sep 17 00:00:00 2001 From: Wei Wang Date: Tue, 2 Sep 2025 19:31:14 +0000 Subject: [PATCH] [CD][CUDA13][ARM] aarch64 binary seems to be missing Triton dependency (#161833) Requires: filelock, fsspec, jinja2, networkx, setuptools, sympy, typing-extensions Seems to be missing Triton. Pull Request resolved: https://github.com/pytorch/pytorch/pull/161833 Approved by: https://github.com/tinglvv, https://github.com/Skylion007, https://github.com/atalman --- .circleci/scripts/binary_populate_env.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.circleci/scripts/binary_populate_env.sh b/.circleci/scripts/binary_populate_env.sh index 87fea14b8d2..f5b949858d6 100755 --- a/.circleci/scripts/binary_populate_env.sh +++ b/.circleci/scripts/binary_populate_env.sh @@ -75,8 +75,8 @@ 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'" -# CUDA 12.9 builds have triton for Linux and Linux aarch64 binaries. -if [[ "$DESIRED_CUDA" == "cu129" ]]; then +# CUDA 12.9/13.0 builds have triton for Linux and Linux aarch64 binaries. +if [[ "$DESIRED_CUDA" == "cu129" ]] || [[ "$DESIRED_CUDA" == "cu130" ]]; then TRITON_CONSTRAINT="platform_system == 'Linux'" fi