From 599be9ee6d51a532f608bb62c98bf5e21ce6bc76 Mon Sep 17 00:00:00 2001 From: Huy Do Date: Fri, 17 Oct 2025 06:47:40 -0700 Subject: [PATCH] Build CUDA 12.9 aarch64 for domains (#7364) I missed this in https://github.com/pytorch/test-infra/pull/7352 which covered only cu129 x86 builds :( Signed-off-by: Huy Do --- tools/scripts/generate_binary_build_matrix.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py index 2b91272c15..b516367bda 100755 --- a/tools/scripts/generate_binary_build_matrix.py +++ b/tools/scripts/generate_binary_build_matrix.py @@ -161,6 +161,7 @@ def initialize_globals(channel: str, os: str, build_python_only: bool) -> None: # TODO (huydhn): Only build CUDA 12.9 for Linux. This logic is to be cleaned up # in 2.10 CUDA_ARCHES.append("12.9") + CUDA_AARCH64_ARCHES.append("12.9-aarch64") ROCM_ARCHES = ROCM_ARCHES_DICT[channel] if build_python_only: # Only select the oldest version of python if building a python only package