From ddaa33d2322c084b85387fbac9b2d3b40efe1d46 Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 21 Mar 2023 08:49:09 -0700 Subject: [PATCH 1/3] Fix torchvision image extension compilation --- aarch64_linux/build_aarch64_wheel.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/aarch64_linux/build_aarch64_wheel.py b/aarch64_linux/build_aarch64_wheel.py index d4e846192..61b00d602 100755 --- a/aarch64_linux/build_aarch64_wheel.py +++ b/aarch64_linux/build_aarch64_wheel.py @@ -301,6 +301,11 @@ def build_torchvision(host: RemoteHost, *, "v2.0.0": ("0.15.1", "rc2"), }) print('Building TorchVision wheel') + + # Please note libnpg and jpeg is required to build image.so extension + if(use_conda): + host.run_cmd("conda install -y libpng jpeg") + build_vars = "" if branch == 'nightly': version = host.check_output(["if [ -f vision/version.txt ]; then cat vision/version.txt; fi"]).strip() From b3dfc27b128d60b3ad518dcd7750253fd076313a Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 21 Mar 2023 08:49:47 -0700 Subject: [PATCH 2/3] Fix torchvision image extension compilation --- aarch64_linux/build_aarch64_wheel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aarch64_linux/build_aarch64_wheel.py b/aarch64_linux/build_aarch64_wheel.py index 61b00d602..19a2e449e 100755 --- a/aarch64_linux/build_aarch64_wheel.py +++ b/aarch64_linux/build_aarch64_wheel.py @@ -302,7 +302,7 @@ def build_torchvision(host: RemoteHost, *, }) print('Building TorchVision wheel') - # Please note libnpg and jpeg is required to build image.so extension + # Please note libnpg and jpeg are required to build image.so extension if(use_conda): host.run_cmd("conda install -y libpng jpeg") From 47a08d4f73969fbba6f6766bbe063f96441f4906 Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 21 Mar 2023 09:49:30 -0700 Subject: [PATCH 3/3] Set enable_mkldnn to pypi build --- aarch64_linux/build_aarch64_wheel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aarch64_linux/build_aarch64_wheel.py b/aarch64_linux/build_aarch64_wheel.py index 19a2e449e..499845f4f 100755 --- a/aarch64_linux/build_aarch64_wheel.py +++ b/aarch64_linux/build_aarch64_wheel.py @@ -773,7 +773,7 @@ def parse_arguments(): configure_system(host, compiler=args.compiler, python_version=python_version, - enable_mkldnn=False) + enable_mkldnn=not args.disable_mkldnn) print("Installing PyTorch wheel") host.run_cmd("pip3 install torch") build_domains(host,