From 71772d14b84960eb3a0c2a8b6f91b86efd1593be Mon Sep 17 00:00:00 2001
From: snadampal <87143774+snadampal@users.noreply.github.com>
Date: Fri, 29 Sep 2023 23:07:27 -0500
Subject: [PATCH] patch ci script with mkldnn fix (#1556)

---
 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}..")