From a7b2e851cf85c19462c8f10b16bcca0672e8b119 Mon Sep 17 00:00:00 2001 From: atalman Date: Mon, 15 Dec 2025 10:57:42 -0800 Subject: [PATCH] Fix VERSION_SUFFIX in aarch64 builds --- tools/pkg-helpers/pytorch_pkg_helpers/cuda.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/pkg-helpers/pytorch_pkg_helpers/cuda.py b/tools/pkg-helpers/pytorch_pkg_helpers/cuda.py index a346af8e21..bff75ec1b1 100644 --- a/tools/pkg-helpers/pytorch_pkg_helpers/cuda.py +++ b/tools/pkg-helpers/pytorch_pkg_helpers/cuda.py @@ -31,7 +31,7 @@ def get_cuda_variables( version_suffix = "" pytorch_version_suffix = "" wheel_dir = "" - if package_type == "wheel" and platform != "darwin" and platform != "linux-aarch64": + if package_type == "wheel" and platform != "darwin": version_suffix = f"+{gpu_arch_version}" pytorch_version_suffix = f"+{gpu_arch_version}" wheel_dir = f"{gpu_arch_version}/"