package lib arm compute in wheel
diff --git a/tools/pip/setup.py b/tools/pip/setup.py
index d5b3bcf..96a3eb7 100644
--- 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
index 5b9f39c..545d14f 100755
--- a/tools/staticbuild/build_lib_cmake.sh
+++ b/tools/staticbuild/build_lib_cmake.sh
@@ -45,6 +45,9 @@
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
