package lib arm compute in wheel

apache · Jun 23, 2023 · 961cc87 · 961cc87
1 parent e9f92b5
commit 961cc87
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/tools/pip/setup.py b/tools/pip/setup.py
@@ -89,6 +89,9 @@
     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
@@ -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