From b18c8ad3a3c6a87a02f8f53dfa71280b0e08cd61 Mon Sep 17 00:00:00 2001 From: atalman Date: Mon, 2 Oct 2023 07:41:20 -0700 Subject: [PATCH] Fix path issue when building aarch64 wheels --- aarch64_linux/aarch64_wheel_ci_build.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/aarch64_linux/aarch64_wheel_ci_build.py b/aarch64_linux/aarch64_wheel_ci_build.py index da789a231..bdc6717ef 100755 --- a/aarch64_linux/aarch64_wheel_ci_build.py +++ b/aarch64_linux/aarch64_wheel_ci_build.py @@ -106,8 +106,8 @@ def parse_arguments(): print("build pytorch without mkldnn backend") # work around to fix Raspberry pie crash - os.system(f"cd $HOME && git clone https://github.com/pytorch/builder.git") - os.system(f"cd $HOME/pytorch/third_party/ideep/mkl-dnn && patch -p1 < $HOME/builder/mkldnn_fix/aarch64-fix-default-build-flags-to-armv8-a.patch") + print("Applying mkl-dnn patch to fix Raspberry pie crash") + os.system(f"cd /pytorch/third_party/ideep/mkl-dnn && patch -p1 < /builder/mkldnn_fix/aarch64-fix-default-build-flags-to-armv8-a.patch") os.system(f"cd /pytorch; {build_vars} python3 setup.py bdist_wheel") pytorch_wheel_name = complete_wheel("pytorch") print(f"Build Compelete. Created {pytorch_wheel_name}..")