Fix path issue when building aarch64 wheels (#1560)

pytorch · Oct 2, 2023 · aa6df6f · aa6df6f
1 parent b2b429b
commit aa6df6f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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}..")