Skip to content

Commit aa6df6f

Browse files
authored
Fix path issue when building aarch64 wheels (#1560)
1 parent b2b429b commit aa6df6f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

aarch64_linux/aarch64_wheel_ci_build.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -106,8 +106,8 @@ def parse_arguments():
106106
print("build pytorch without mkldnn backend")
107107

108108
# work around to fix Raspberry pie crash
109-
os.system(f"cd $HOME && git clone https://github.com/pytorch/builder.git")
110-
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")
109+
print("Applying mkl-dnn patch to fix Raspberry pie crash")
110+
os.system(f"cd /pytorch/third_party/ideep/mkl-dnn && patch -p1 < /builder/mkldnn_fix/aarch64-fix-default-build-flags-to-armv8-a.patch")
111111
os.system(f"cd /pytorch; {build_vars} python3 setup.py bdist_wheel")
112112
pytorch_wheel_name = complete_wheel("pytorch")
113113
print(f"Build Compelete. Created {pytorch_wheel_name}..")

0 commit comments

Comments
 (0)