build_ArmComputeLibrary: better copy algorithm

Just iterate over folders that needs to be copied
pytorch · Mar 7, 2024 · 9329cbe · 9329cbe
1 parent 9f4ba0a
commit 9329cbe
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/aarch64_linux/aarch64_wheel_ci_build.py b/aarch64_linux/aarch64_wheel_ci_build.py
@@ -29,11 +29,8 @@ def build_ArmComputeLibrary() -> None:
  "--depth", "1", "--shallow-submodules"])
  check_call(["scons", "Werror=1", "-j8", f"build_dir=/{acl_install_dir}/build"] + acl_build_flags,
  cwd=acl_checkout_dir)
- shutil.copytree(f"{acl_checkout_dir}/arm_compute", acl_install_dir)
- shutil.copytree(f"{acl_checkout_dir}/include", acl_install_dir)
- shutil.copytree(f"{acl_checkout_dir}/utils", acl_install_dir)
- shutil.copytree(f"{acl_checkout_dir}/support", acl_install_dir)
- shutil.copytree(f"{acl_checkout_dir}/src", acl_install_dir)
+ for d in ["arm_compute", "include", "utils", "support", "src"]:
+ shutil.copytree(f"{acl_checkout_dir}/{d}", f"{acl_install_dir}/{d}")
 
 
 def complete_wheel(folder: str) -> str: