From a76e39c51932f786e8015b2e89a1b430d7e1abb4 Mon Sep 17 00:00:00 2001 From: Ting Lu Date: Wed, 29 May 2024 06:00:25 -0700 Subject: [PATCH 1/2] add cuda version to arm nightly wheel --- aarch64_linux/aarch64_wheel_ci_build.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/aarch64_linux/aarch64_wheel_ci_build.py b/aarch64_linux/aarch64_wheel_ci_build.py index 61efa10f2..ce3ccdd33 100755 --- a/aarch64_linux/aarch64_wheel_ci_build.py +++ b/aarch64_linux/aarch64_wheel_ci_build.py @@ -218,7 +218,11 @@ def parse_arguments(): version = ( check_output(["cat", "version.txt"], cwd="/pytorch").decode().strip()[:-2] ) - build_vars += f"BUILD_TEST=0 PYTORCH_BUILD_VERSION={version}.dev{build_date} PYTORCH_BUILD_NUMBER=1 " + if enable_cuda: + desired_cuda = os.getenv("DESIRED_CUDA") + build_vars += f"BUILD_TEST=0 PYTORCH_BUILD_VERSION={version}.dev{build_date}+{desired_cuda} PYTORCH_BUILD_NUMBER=1 " + else: + build_vars += f"BUILD_TEST=0 PYTORCH_BUILD_VERSION={version}.dev{build_date} PYTORCH_BUILD_NUMBER=1 " elif branch.startswith(("v1.", "v2.")): build_vars += f"BUILD_TEST=0 PYTORCH_BUILD_VERSION={branch[1:branch.find('-')]} PYTORCH_BUILD_NUMBER=1 " From 170d3542c13a45b454bd922fde5f4ef5f4f39e69 Mon Sep 17 00:00:00 2001 From: Ting Lu Date: Wed, 29 May 2024 07:39:32 -0700 Subject: [PATCH 2/2] fix lint line too long --- aarch64_linux/aarch64_wheel_ci_build.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/aarch64_linux/aarch64_wheel_ci_build.py b/aarch64_linux/aarch64_wheel_ci_build.py index ce3ccdd33..fe612f326 100755 --- a/aarch64_linux/aarch64_wheel_ci_build.py +++ b/aarch64_linux/aarch64_wheel_ci_build.py @@ -220,7 +220,9 @@ def parse_arguments(): ) if enable_cuda: desired_cuda = os.getenv("DESIRED_CUDA") - build_vars += f"BUILD_TEST=0 PYTORCH_BUILD_VERSION={version}.dev{build_date}+{desired_cuda} PYTORCH_BUILD_NUMBER=1 " + build_vars += ( + f"BUILD_TEST=0 PYTORCH_BUILD_VERSION={version}.dev{build_date}+{desired_cuda} PYTORCH_BUILD_NUMBER=1 " + ) else: build_vars += f"BUILD_TEST=0 PYTORCH_BUILD_VERSION={version}.dev{build_date} PYTORCH_BUILD_NUMBER=1 " elif branch.startswith(("v1.", "v2.")):