|
|
|
@ -258,8 +258,10 @@ checkopts()
|
|
|
|
|
I)
|
|
|
|
|
COMPILE_LITE="on"
|
|
|
|
|
if [[ "$OPTARG" == "arm64" ]]; then
|
|
|
|
|
ENABLE_CONVERTER="off"
|
|
|
|
|
LITE_PLATFORM="arm64"
|
|
|
|
|
elif [[ "$OPTARG" == "arm32" ]]; then
|
|
|
|
|
ENABLE_CONVERTER="off"
|
|
|
|
|
LITE_PLATFORM="arm32"
|
|
|
|
|
elif [[ "$OPTARG" == "x86_64" ]]; then
|
|
|
|
|
ENABLE_CONVERTER="on"
|
|
|
|
@ -291,7 +293,7 @@ checkopts()
|
|
|
|
|
ENABLE_IBVERBS="on"
|
|
|
|
|
echo "enable IBVERBS for parameter server"
|
|
|
|
|
;;
|
|
|
|
|
T)
|
|
|
|
|
T)
|
|
|
|
|
check_on_off $OPTARG T
|
|
|
|
|
SUPPORT_TRAIN=$OPTARG
|
|
|
|
|
echo "support train on device "
|
|
|
|
@ -566,7 +568,7 @@ build_opencv() {
|
|
|
|
|
OPENCV_BIN="${BASEPATH}"/third_party/opencv/build/lib/armeabi-v7a/libopencv_core.so
|
|
|
|
|
else
|
|
|
|
|
OPENCV_BIN="${BASEPATH}"/third_party/opencv/build/lib/arm64-v8a/libopencv_core.so
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
fi
|
|
|
|
|
if [[ ! -f "${OPENCV_BIN}" ]]; then
|
|
|
|
|
if [[ "${MSLIBS_SERVER}" ]]; then
|
|
|
|
@ -665,7 +667,7 @@ build_lite()
|
|
|
|
|
fi
|
|
|
|
|
build_flatbuffer
|
|
|
|
|
build_gtest
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if [ "${COMPILE_MINDDATA_LITE}" == "lite" ] || [ "${COMPILE_MINDDATA_LITE}" == "full" ]; then
|
|
|
|
|
build_minddata_lite_deps
|
|
|
|
|
fi
|
|
|
|
@ -686,27 +688,30 @@ build_lite()
|
|
|
|
|
cmake -DCMAKE_TOOLCHAIN_FILE="${ANDROID_NDK}/build/cmake/android.toolchain.cmake" -DANDROID_NATIVE_API_LEVEL="19" \
|
|
|
|
|
-DANDROID_NDK="${ANDROID_NDK}" -DANDROID_ABI="arm64-v8a" -DANDROID_TOOLCHAIN_NAME="aarch64-linux-android-clang" \
|
|
|
|
|
-DANDROID_STL="c++_static" -DCMAKE_BUILD_TYPE=${BUILD_TYPE} -DSUPPORT_TRAIN=${SUPPORT_TRAIN} \
|
|
|
|
|
-DBUILD_DEVICE=on -DPLATFORM_ARM64=on -DBUILD_CONVERTER=off -DENABLE_NEON=on -DENABLE_FP16="off" \
|
|
|
|
|
-DBUILD_DEVICE=on -DPLATFORM_ARM64=on -DENABLE_NEON=on -DENABLE_FP16="off" \
|
|
|
|
|
-DENABLE_TOOLS=${ENABLE_TOOLS} -DENABLE_CONVERTER=${ENABLE_CONVERTER} -DBUILD_TESTCASES=${RUN_TESTCASES} \
|
|
|
|
|
-DSUPPORT_GPU=${ENABLE_GPU} -DOFFLINE_COMPILE=${OPENCL_OFFLINE_COMPILE} -DBUILD_MINDDATA=${COMPILE_MINDDATA_LITE} \
|
|
|
|
|
-DCMAKE_INSTALL_PREFIX=${BASEPATH}/output/tmp -DMS_VERSION_MAJOR=${VERSION_MAJOR} \
|
|
|
|
|
-DMS_VERSION_MINOR=${VERSION_MINOR} -DMS_VERSION_REVISION=${VERSION_REVISION} -DENABLE_VERBOSE=${ENABLE_VERBOSE} \
|
|
|
|
|
-DENABLE_TOOLS=${ENABLE_TOOLS} -DENABLE_CONVERTER=${ENABLE_CONVERTER} "${BASEPATH}/mindspore/lite"
|
|
|
|
|
"${BASEPATH}/mindspore/lite"
|
|
|
|
|
elif [[ "${LITE_PLATFORM}" == "arm32" ]]; then
|
|
|
|
|
checkndk
|
|
|
|
|
cmake -DCMAKE_TOOLCHAIN_FILE="${ANDROID_NDK}/build/cmake/android.toolchain.cmake" -DANDROID_NATIVE_API_LEVEL="19" \
|
|
|
|
|
-DANDROID_NDK="${ANDROID_NDK}" -DANDROID_ABI="armeabi-v7a" -DANDROID_TOOLCHAIN_NAME="clang" \
|
|
|
|
|
-DANDROID_STL="c++_static" -DCMAKE_BUILD_TYPE=${BUILD_TYPE} \
|
|
|
|
|
-DBUILD_DEVICE=on -DPLATFORM_ARM32=on -DENABLE_NEON=on -DSUPPORT_TRAIN=${SUPPORT_TRAIN} -DBUILD_CONVERTER=off \
|
|
|
|
|
-DBUILD_DEVICE=on -DPLATFORM_ARM32=on -DENABLE_NEON=on -DSUPPORT_TRAIN=${SUPPORT_TRAIN} \
|
|
|
|
|
-DENABLE_TOOLS=${ENABLE_TOOLS} -DENABLE_CONVERTER=${ENABLE_CONVERTER} -DBUILD_TESTCASES=${RUN_TESTCASES} \
|
|
|
|
|
-DSUPPORT_GPU=${ENABLE_GPU} -DOFFLINE_COMPILE=${OPENCL_OFFLINE_COMPILE} -DBUILD_MINDDATA=${COMPILE_MINDDATA_LITE} \
|
|
|
|
|
-DCMAKE_INSTALL_PREFIX=${BASEPATH}/output/tmp -DMS_VERSION_MAJOR=${VERSION_MAJOR} \
|
|
|
|
|
-DMS_VERSION_MINOR=${VERSION_MINOR} -DMS_VERSION_REVISION=${VERSION_REVISION} -DENABLE_VERBOSE=${ENABLE_VERBOSE} \
|
|
|
|
|
-DENABLE_TOOLS=${ENABLE_TOOLS} -DENABLE_CONVERTER=${ENABLE_CONVERTER} "${BASEPATH}/mindspore/lite"
|
|
|
|
|
"${BASEPATH}/mindspore/lite"
|
|
|
|
|
else
|
|
|
|
|
cmake -DBUILD_DEVICE=on -DPLATFORM_ARM64=off -DBUILD_CONVERTER=${ENABLE_CONVERTER} -DSUPPORT_TRAIN=${SUPPORT_TRAIN} \
|
|
|
|
|
cmake -DBUILD_DEVICE=on -DPLATFORM_ARM64=off -DSUPPORT_TRAIN=${SUPPORT_TRAIN} \
|
|
|
|
|
-DENABLE_TOOLS=${ENABLE_TOOLS} -DENABLE_CONVERTER=${ENABLE_CONVERTER} -DBUILD_TESTCASES=${RUN_TESTCASES} \
|
|
|
|
|
-DCMAKE_BUILD_TYPE=${BUILD_TYPE} -DSUPPORT_GPU=${ENABLE_GPU} -DBUILD_MINDDATA=${COMPILE_MINDDATA_LITE} \
|
|
|
|
|
-DOFFLINE_COMPILE=${OPENCL_OFFLINE_COMPILE} -DCMAKE_INSTALL_PREFIX=${BASEPATH}/output/tmp \
|
|
|
|
|
-DMS_VERSION_MAJOR=${VERSION_MAJOR} -DMS_VERSION_MINOR=${VERSION_MINOR} -DMS_VERSION_REVISION=${VERSION_REVISION} \
|
|
|
|
|
-DENABLE_VERBOSE=${ENABLE_VERBOSE} -DENABLE_TOOLS=${ENABLE_TOOLS} -DENABLE_CONVERTER=${ENABLE_CONVERTER} "${BASEPATH}/mindspore/lite"
|
|
|
|
|
-DENABLE_VERBOSE=${ENABLE_VERBOSE} "${BASEPATH}/mindspore/lite"
|
|
|
|
|
fi
|
|
|
|
|
make -j$THREAD_NUM && make install && make package
|
|
|
|
|
COMPILE_RET=$?
|
|
|
|
|