Use check_call for applying the patch

Otherwise, if it fails to apply one would never know
pytorch · Mar 7, 2024 · 0926610 · 0926610
1 parent 9329cbe
commit 0926610
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/aarch64_linux/aarch64_wheel_ci_build.py b/aarch64_linux/aarch64_wheel_ci_build.py
@@ -108,7 +108,8 @@ def parse_arguments():
 
  # patch mkldnn to fix aarch64 mac and aws lambda crash
  print("Applying mkl-dnn patch to fix crash due to /sys not accesible")
- os.system("cd /pytorch/third_party/ideep/mkl-dnn && patch -p1 < /builder/mkldnn_fix/fix-xbyak-failure.patch")
+ with open("/builder/mkldnn_fix/fix-xbyak-failure.patch") as f:
+ check_call(["patch", "-p1"], stdin=f, cwd="/pytorch/third_party/ideep/mkl-dnn")
 
  os.system(f"cd /pytorch; {build_vars} python3 setup.py bdist_wheel")
  pytorch_wheel_name = complete_wheel("pytorch")