diff --git a/tools/pip/setup.py b/tools/pip/setup.py index d5b3bcfeb2c..96a3eb753a4 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 5b9f39cdb6e..545d14fd316 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