Skip to content

Commit f6ea026

Browse files
authored
fix ACL_ROOT_DIR setting and upgrade the ACL version to 22.11 (#1291)
1 parent 4959070 commit f6ea026

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

build_aarch64_wheel.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -227,7 +227,7 @@ def build_OpenBLAS(host: RemoteHost, git_clone_flags: str = "") -> None:
227227
def build_ArmComputeLibrary(host: RemoteHost, git_clone_flags: str = "") -> None:
228228
print('Building Arm Compute Library')
229229
host.run_cmd("mkdir $HOME/acl")
230-
host.run_cmd(f"git clone https://github.com/ARM-software/ComputeLibrary.git -b v22.05 {git_clone_flags}")
230+
host.run_cmd(f"git clone https://github.com/ARM-software/ComputeLibrary.git -b v22.11 {git_clone_flags}")
231231
host.run_cmd(f"pushd ComputeLibrary; export acl_install_dir=$HOME/acl; scons Werror=1 -j8 debug=0 neon=1 opencl=0 os=linux openmp=1 cppthreads=0 arch=armv8.2-a multi_isa=1 build=native build_dir=$acl_install_dir/build; cp -r arm_compute $acl_install_dir; cp -r include $acl_install_dir; cp -r utils $acl_install_dir; cp -r support $acl_install_dir; popd")
232232

233233

@@ -488,7 +488,7 @@ def start_build(host: RemoteHost, *,
488488
build_ArmComputeLibrary(host, git_clone_flags)
489489
print("build pytorch with mkldnn+acl backend")
490490
build_vars += " USE_MKLDNN=ON USE_MKLDNN_ACL=ON"
491-
host.run_cmd(f"cd pytorch ; export ACL_ROOT_DIR=$HOME/acl; {build_vars} python3 setup.py bdist_wheel")
491+
host.run_cmd(f"cd pytorch ; export ACL_ROOT_DIR=$HOME/ComputeLibrary:$HOME/acl; {build_vars} python3 setup.py bdist_wheel")
492492
print('Repair the wheel')
493493
pytorch_wheel_name = host.list_dir("pytorch/dist")[0]
494494
host.run_cmd(f"export LD_LIBRARY_PATH=$HOME/acl/build:$HOME/pytorch/build/lib; auditwheel repair $HOME/pytorch/dist/{pytorch_wheel_name}")

0 commit comments

Comments
 (0)