From df411d856f4a1b6a940f773bee0fbcc59da0699a Mon Sep 17 00:00:00 2001 From: Sunita Nadampalli Date: Fri, 29 Sep 2023 22:53:34 -0500 Subject: [PATCH] patch ci script with mkldnn fix --- aarch64_linux/aarch64_wheel_ci_build.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/aarch64_linux/aarch64_wheel_ci_build.py b/aarch64_linux/aarch64_wheel_ci_build.py index f6797ce1c..da789a231 100755 --- a/aarch64_linux/aarch64_wheel_ci_build.py +++ b/aarch64_linux/aarch64_wheel_ci_build.py @@ -105,6 +105,9 @@ def parse_arguments(): else: 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") 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}..")