[mxnet] branch acl-build updated: package lib arm compute in wheel

[ma...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

manuseth pushed a commit to branch acl-build
in repository https://gitbox.apache.org/repos/asf/mxnet.git


The following commit(s) were added to refs/heads/acl-build by this push:
     new 961cc87689 package lib arm compute in wheel
961cc87689 is described below

commit 961cc8768902a34437f3c89134bf5608549196f0
Author: Ubuntu <ub...@ip-172-31-8-26.us-west-2.compute.internal>
AuthorDate: Fri Jun 23 19:06:18 2023 +0000

    package lib arm compute in wheel
---
 tools/pip/setup.py                   | 3 +++
 tools/staticbuild/build_lib_cmake.sh | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/tools/pip/setup.py b/tools/pip/setup.py
index d5b3bcfeb2..96a3eb753a 100644
--- a/tools/pip/setup.py
+++ b/tools/pip/setup.py
@@ -89,6 +89,9 @@ if platform.system() == 'Linux':
     if os.path.exists(os.path.join(libdir, 'libopenblas.so.0')):
         shutil.copy(os.path.join(libdir, 'libopenblas.so.0'), mxdir)
         package_data['mxnet'].append('mxnet/libopenblas.so.0')
+    if os.path.exists(os.path.join(libdir, 'libarm_compute.so')):
+        shutil.copy(os.path.join(libdir, 'libarm_compute.so'), mxdir)
+        package_data['mxnet'].append('mxnet/libarm_compute.so')
 
 # copy license and notice
 shutil.copytree(os.path.join(CURRENT_DIR, 'mxnet-build/licenses'),
diff --git a/tools/staticbuild/build_lib_cmake.sh b/tools/staticbuild/build_lib_cmake.sh
index 5b9f39cdb6..545d14fd31 100755
--- a/tools/staticbuild/build_lib_cmake.sh
+++ b/tools/staticbuild/build_lib_cmake.sh
@@ -45,6 +45,9 @@ if [[ $PLATFORM == 'linux' ]]; then
     cp -L build/libmxnet.so lib/libmxnet.so
     cp -L $(ldd lib/libmxnet.so | grep libgfortran | awk '{print $3}') lib/
     cp -L staticdeps/lib/libopenblas.so lib/libopenblas.so.0
+if [[ $ARCH == 'aarch64' ]]; then
+    cp -L /work/deps/acl/ComputeLibrary/build/libarm_compute.so lib/libarm_compute.so
+fi
 elif [[ $PLATFORM == 'darwin' ]]; then
     cp -L build/libmxnet.dylib lib/libmxnet.dylib
 fi