From 5692731db3f4fdddeb5541122a7b8d8db8c5f73b Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Wed, 8 Mar 2023 16:29:00 -0500 Subject: [PATCH] Aarch64 changes for PyTorch release 2.0 (#1336) * Aarch64 changes for PyTorch release 2.0 * Fix spacing * Update aarch64_linux/build_aarch64_wheel.py Co-authored-by: Nikita Shulga * Update aarch64_linux/build_aarch64_wheel.py Co-authored-by: Nikita Shulga --------- Co-authored-by: Nikita Shulga --- aarch64_linux/build_aarch64_wheel.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/aarch64_linux/build_aarch64_wheel.py b/aarch64_linux/build_aarch64_wheel.py index ac943baf6..ad2e3f2f7 100755 --- a/aarch64_linux/build_aarch64_wheel.py +++ b/aarch64_linux/build_aarch64_wheel.py @@ -294,6 +294,7 @@ def build_torchvision(host: RemoteHost, *, "v1.12.1": ("0.13.1", "rc6"), "v1.13.0": ("0.14.0", "rc4"), "v1.13.1": ("0.14.1", "rc2"), + "v2.0.0": ("0.15.0", "rc6"), }) print('Building TorchVision wheel') build_vars = "" @@ -333,6 +334,7 @@ def build_torchdata(host: RemoteHost, *, git_clone_flags=git_clone_flags, mapping={ "v1.13.1": ("0.5.1", ""), + "v2.0.0": ("0.6.0", "rc4"), }) print('Building TorchData wheel') build_vars = "" @@ -375,6 +377,7 @@ def build_torchtext(host: RemoteHost, *, "v1.12.1": ("0.13.1", "rc5"), "v1.13.0": ("0.14.0", "rc3"), "v1.13.1": ("0.14.1", "rc1"), + "v2.0.0": ("0.15.0", "rc4"), }) print('Building TorchText wheel') build_vars = "" @@ -417,6 +420,7 @@ def build_torchaudio(host: RemoteHost, *, "v1.12.1": ("0.12.1", "rc5"), "v1.13.0": ("0.13.0", "rc4"), "v1.13.1": ("0.13.1", "rc2"), + "v2.0.0": ("2.0.0", "rc5"), }) print('Building TorchAudio wheel') build_vars = "" @@ -515,7 +519,7 @@ def start_build(host: RemoteHost, *, build_date = host.check_output("cd pytorch ; git log --pretty=format:%s -1").strip().split()[0].replace("-", "") version = host.check_output("cat pytorch/version.txt").strip()[:-2] build_vars += f"BUILD_TEST=0 PYTORCH_BUILD_VERSION={version}.dev{build_date} PYTORCH_BUILD_NUMBER=1" - if branch.startswith("v1."): + if branch.startswith("v1.") or branch.startswith("v2."): build_vars += f"BUILD_TEST=0 PYTORCH_BUILD_VERSION={branch[1:branch.find('-')]} PYTORCH_BUILD_NUMBER=1" if host.using_docker(): build_vars += " CMAKE_SHARED_LINKER_FLAGS=-Wl,-z,max-page-size=0x10000"