patch ci script with mkldnn fix (pytorch#1556)

atalman · Oct 2, 2023 · 7259dfa · 7259dfa
1 parent 1ffc178
commit 7259dfa
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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}..")