| @@ -576,39 +576,6 @@ build_minddata_lite_deps() | |||||
| build_jpeg_turbo | build_jpeg_turbo | ||||
| } | } | ||||
| prepare_md_lite() { | |||||
| if [ "${COMPILE_MINDDATA_LITE}" == "on" ]; then | |||||
| echo "packaging minddata" | |||||
| cp ${BASEPATH}/mindspore/ccsrc/minddata/dataset/include/*h ${OUTPUT_DIR}/include/ | |||||
| cp ${BASEPATH}/mindspore/lite/build/minddata/libminddata-lite.so ${OUTPUT_DIR}/lib/ | |||||
| if [[ "$LITE_PLATFORM" == "x86_64" ]]; then | |||||
| mkdir -p ${OUTPUT_DIR}/third_party/libjpeg-turbo/lib | |||||
| cp -r ${BASEPATH}/third_party/libjpeg-turbo/lib/libjpeg.so ${OUTPUT_DIR}/third_party/libjpeg-turbo/lib/ | |||||
| cp -r ${BASEPATH}/third_party/libjpeg-turbo/lib/libturbojpeg.so ${OUTPUT_DIR}/third_party/libjpeg-turbo/lib/ | |||||
| mkdir -p ${OUTPUT_DIR}/third_party/opencv/lib/ | |||||
| cp -r ${BASEPATH}/third_party/opencv/build/lib/libopencv_core.so ${OUTPUT_DIR}/third_party/opencv/lib/ | |||||
| cp -r ${BASEPATH}/third_party/opencv/build/lib/libopencv_imgcodecs.so ${OUTPUT_DIR}/third_party/opencv/lib/ | |||||
| cp -r ${BASEPATH}/third_party/opencv/build/lib/libopencv_imgproc.so ${OUTPUT_DIR}/third_party/opencv/lib/ | |||||
| elif [[ "$LITE_PLATFORM" == "arm64" ]]; then | |||||
| mkdir -p ${OUTPUT_DIR}/third_party/libjpeg-turbo/lib | |||||
| cp -r ${BASEPATH}/third_party/libjpeg-turbo/lib/libjpeg.so ${OUTPUT_DIR}/third_party/libjpeg-turbo/lib/ | |||||
| cp -r ${BASEPATH}/third_party/libjpeg-turbo/lib/libturbojpeg.so ${OUTPUT_DIR}/third_party/libjpeg-turbo/lib/ | |||||
| mkdir -p ${OUTPUT_DIR}/third_party/opencv/lib/arm64-v8a/ | |||||
| cp -r ${BASEPATH}/third_party/opencv/build/lib/arm64-v8a/libopencv_core.so ${OUTPUT_DIR}/third_party/opencv/lib/arm64-v8a/ | |||||
| cp -r ${BASEPATH}/third_party/opencv/build/lib/arm64-v8a/libopencv_imgcodecs.so ${OUTPUT_DIR}/third_party/opencv/lib/arm64-v8a/ | |||||
| cp -r ${BASEPATH}/third_party/opencv/build/lib/arm64-v8a/libopencv_imgproc.so ${OUTPUT_DIR}/third_party/opencv/lib/arm64-v8a/ | |||||
| elif [[ "$LITE_PLATFORM" == "arm32" ]]; then | |||||
| mkdir -p ${OUTPUT_DIR}/third_party/libjpeg-turbo/lib | |||||
| cp -r ${BASEPATH}/third_party/libjpeg-turbo/lib/libjpeg.so ${OUTPUT_DIR}/third_party/libjpeg-turbo/lib/ | |||||
| cp -r ${BASEPATH}/third_party/libjpeg-turbo/lib/libturbojpeg.so ${OUTPUT_DIR}/third_party/libjpeg-turbo/lib/ | |||||
| mkdir -p ${OUTPUT_DIR}/third_party/opencv/lib/armeabi-v7a/ | |||||
| cp -r ${BASEPATH}/third_party/opencv/build/lib/armeabi-v7a/libopencv_core.so ${OUTPUT_DIR}/third_party/opencv/lib/armeabi-v7a/ | |||||
| cp -r ${BASEPATH}/third_party/opencv/build/lib/armeabi-v7a/libopencv_imgcodecs.so ${OUTPUT_DIR}/third_party/opencv/lib/armeabi-v7a/ | |||||
| cp -r ${BASEPATH}/third_party/opencv/build/lib/armeabi-v7a/libopencv_imgproc.so ${OUTPUT_DIR}/third_party/opencv/lib/armeabi-v7a/ | |||||
| fi | |||||
| fi | |||||
| } | |||||
| build_lite() | build_lite() | ||||
| { | { | ||||
| echo "start build mindspore lite project" | echo "start build mindspore lite project" | ||||
| @@ -645,7 +612,7 @@ build_lite() | |||||
| -DANDROID_STL="c++_shared" -DCMAKE_BUILD_TYPE=${BUILD_TYPE} -DSUPPORT_TRAIN=${SUPPORT_TRAIN} \ | -DANDROID_STL="c++_shared" -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 -DBUILD_CONVERTER=off -DENABLE_NEON=on -DENABLE_FP16="off" \ | ||||
| -DSUPPORT_GPU=${ENABLE_GPU} -DOFFLINE_COMPILE=${OPENCL_OFFLINE_COMPILE} -DBUILD_MINDDATA=${COMPILE_MINDDATA_LITE} \ | -DSUPPORT_GPU=${ENABLE_GPU} -DOFFLINE_COMPILE=${OPENCL_OFFLINE_COMPILE} -DBUILD_MINDDATA=${COMPILE_MINDDATA_LITE} \ | ||||
| "${BASEPATH}/mindspore/lite" | |||||
| -DCMAKE_INSTALL_PREFIX=${BASEPATH}/output/tmp "${BASEPATH}/mindspore/lite" | |||||
| elif [[ "${LITE_PLATFORM}" == "arm32" ]]; then | elif [[ "${LITE_PLATFORM}" == "arm32" ]]; then | ||||
| checkndk | checkndk | ||||
| cmake -DCMAKE_TOOLCHAIN_FILE="${ANDROID_NDK}/build/cmake/android.toolchain.cmake" -DANDROID_NATIVE_API_LEVEL="19" \ | cmake -DCMAKE_TOOLCHAIN_FILE="${ANDROID_NDK}/build/cmake/android.toolchain.cmake" -DANDROID_NATIVE_API_LEVEL="19" \ | ||||
| @@ -653,88 +620,20 @@ build_lite() | |||||
| -DANDROID_STL="c++_shared" -DCMAKE_BUILD_TYPE=${BUILD_TYPE} \ | -DANDROID_STL="c++_shared" -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} -DBUILD_CONVERTER=off \ | ||||
| -DSUPPORT_GPU=${ENABLE_GPU} -DOFFLINE_COMPILE=${OPENCL_OFFLINE_COMPILE} -DBUILD_MINDDATA=${COMPILE_MINDDATA_LITE} \ | -DSUPPORT_GPU=${ENABLE_GPU} -DOFFLINE_COMPILE=${OPENCL_OFFLINE_COMPILE} -DBUILD_MINDDATA=${COMPILE_MINDDATA_LITE} \ | ||||
| "${BASEPATH}/mindspore/lite" | |||||
| -DCMAKE_INSTALL_PREFIX=${BASEPATH}/output/tmp "${BASEPATH}/mindspore/lite" | |||||
| else | else | ||||
| cmake -DBUILD_DEVICE=on -DPLATFORM_ARM64=off -DBUILD_CONVERTER=${ENABLE_CONVERTER} -DSUPPORT_TRAIN=${SUPPORT_TRAIN} \ | cmake -DBUILD_DEVICE=on -DPLATFORM_ARM64=off -DBUILD_CONVERTER=${ENABLE_CONVERTER} -DSUPPORT_TRAIN=${SUPPORT_TRAIN} \ | ||||
| -DCMAKE_BUILD_TYPE=${BUILD_TYPE} -DSUPPORT_GPU=${ENABLE_GPU} -DBUILD_MINDDATA=${COMPILE_MINDDATA_LITE} \ | -DCMAKE_BUILD_TYPE=${BUILD_TYPE} -DSUPPORT_GPU=${ENABLE_GPU} -DBUILD_MINDDATA=${COMPILE_MINDDATA_LITE} \ | ||||
| -DOFFLINE_COMPILE=${OPENCL_OFFLINE_COMPILE} "${BASEPATH}/mindspore/lite" | |||||
| -DOFFLINE_COMPILE=${OPENCL_OFFLINE_COMPILE} -DCMAKE_INSTALL_PREFIX=${BASEPATH}/output/tmp "${BASEPATH}/mindspore/lite" | |||||
| fi | fi | ||||
| VERBOSE=2 make -j$THREAD_NUM | |||||
| VERBOSE=2 make -j$THREAD_NUM && make install && make package | |||||
| COMPILE_RET=$? | COMPILE_RET=$? | ||||
| if [[ "${COMPILE_RET}" -ne 0 ]]; then | if [[ "${COMPILE_RET}" -ne 0 ]]; then | ||||
| echo "---------------- mindspore lite: build failed ----------------" | echo "---------------- mindspore lite: build failed ----------------" | ||||
| else | else | ||||
| mkdir -pv ${BASEPATH}/mindspore/lite/output/ | |||||
| if [[ "$LITE_PLATFORM" == "x86_64" ]]; then | |||||
| OUTPUT_DIR=${BASEPATH}/output/MSLite-0.6.0-linux_x86_64 | |||||
| rm -rf ${OUTPUT_DIR} && mkdir -p ${OUTPUT_DIR} && cd ${OUTPUT_DIR} | |||||
| mkdir -p ${OUTPUT_DIR}/converter && mkdir -p ${OUTPUT_DIR}/time_profile | |||||
| mkdir -p ${OUTPUT_DIR}/benchmark && mkdir -p ${OUTPUT_DIR}/include && mkdir -p ${OUTPUT_DIR}/lib | |||||
| mkdir -p ${OUTPUT_DIR}/third_party | |||||
| prepare_md_lite | |||||
| cp ${BASEPATH}/mindspore/lite/build/tools/converter/converter_lite ${OUTPUT_DIR}/converter/ | |||||
| cp ${BASEPATH}/mindspore/lite/build/tools/benchmark/benchmark ${OUTPUT_DIR}/benchmark/ | |||||
| cp ${BASEPATH}/mindspore/lite/build/tools/time_profile/timeprofile ${OUTPUT_DIR}/time_profile/ | |||||
| cp ${BASEPATH}/mindspore/lite/include/*.h ${OUTPUT_DIR}/include/ | |||||
| mkdir -p ${OUTPUT_DIR}/include/ir/dtype/ | |||||
| cp ${BASEPATH}/mindspore/core/ir/dtype/type_id.h ${OUTPUT_DIR}/include/ir/dtype/ | |||||
| mkdir -p ${OUTPUT_DIR}/include/schema/ | |||||
| cp ${BASEPATH}/mindspore/lite/schema/*.h ${OUTPUT_DIR}/include/schema/ | |||||
| cp ${BASEPATH}/mindspore/lite/build/src/libmindspore-lite.so ${OUTPUT_DIR}/lib/ | |||||
| mkdir -p ${OUTPUT_DIR}/third_party/protobuf/lib | |||||
| cp -r ${BASEPATH}/third_party/protobuf/build/include/ ${OUTPUT_DIR}/third_party/protobuf/ | |||||
| cp -r ${BASEPATH}/third_party/protobuf/build/lib/libprotobuf.so.19.0.0 ${OUTPUT_DIR}/third_party/protobuf/lib/libprotobuf.so.19 | |||||
| mkdir -p ${OUTPUT_DIR}/third_party/flatbuffers | |||||
| cp -r ${BASEPATH}/third_party/flatbuffers/include/ ${OUTPUT_DIR}/third_party/flatbuffers/ | |||||
| cd .. | |||||
| tar -czf MSLite-0.6.0-linux_x86_64.tar.gz MSLite-0.6.0-linux_x86_64/ --warning=no-file-changed | |||||
| sha256sum MSLite-0.6.0-linux_x86_64.tar.gz > MSLite-0.6.0-linux_x86_64.tar.gz.sha256 | |||||
| rm -rf MSLite-0.6.0-linux_x86_64/ | |||||
| elif [[ "$LITE_PLATFORM" == "arm64" ]]; then | |||||
| OUTPUT_DIR=${BASEPATH}/output/MSLite-0.6.0-linux_arm64 | |||||
| rm -rf ${OUTPUT_DIR} && mkdir -p ${OUTPUT_DIR} && cd ${OUTPUT_DIR} | |||||
| mkdir -p ${OUTPUT_DIR}/time_profile && mkdir -p ${OUTPUT_DIR}/benchmark | |||||
| mkdir -p ${OUTPUT_DIR}/include && mkdir -p ${OUTPUT_DIR}/lib | |||||
| mkdir -p ${OUTPUT_DIR}/third_party | |||||
| prepare_md_lite | |||||
| cp ${BASEPATH}/mindspore/lite/build/tools/benchmark/benchmark ${OUTPUT_DIR}/benchmark/ | |||||
| cp ${BASEPATH}/mindspore/lite/build/tools/time_profile/timeprofile ${OUTPUT_DIR}/time_profile/ | |||||
| cp ${BASEPATH}/mindspore/lite/include/*.h ${OUTPUT_DIR}/include/ | |||||
| mkdir -p ${OUTPUT_DIR}/include/ir/dtype/ | |||||
| cp ${BASEPATH}/mindspore/core/ir/dtype/type_id.h ${OUTPUT_DIR}/include/ir/dtype/ | |||||
| mkdir -p ${OUTPUT_DIR}/include/schema/ | |||||
| cp ${BASEPATH}/mindspore/lite/schema/*.h ${OUTPUT_DIR}/include/schema/ | |||||
| cp ${BASEPATH}/mindspore/lite/build/src/libmindspore-lite.so ${OUTPUT_DIR}/lib/ | |||||
| cp ${BASEPATH}/mindspore/lite/build/src/runtime/kernel/arm/nnacl/liboptimize.so ${OUTPUT_DIR}/lib/ | |||||
| mkdir -p ${OUTPUT_DIR}/third_party/flatbuffers | |||||
| cp -r ${BASEPATH}/third_party/flatbuffers/include/ ${OUTPUT_DIR}/third_party/flatbuffers/ | |||||
| cd .. | |||||
| tar -czf MSLite-0.6.0-linux_arm64.tar.gz MSLite-0.6.0-linux_arm64/ --warning=no-file-changed | |||||
| sha256sum MSLite-0.6.0-linux_arm64.tar.gz > MSLite-0.6.0-linux_arm64.tar.gz.sha256 | |||||
| rm -rf MSLite-0.6.0-linux_arm64/ | |||||
| elif [[ "$LITE_PLATFORM" == "arm32" ]]; then | |||||
| OUTPUT_DIR=${BASEPATH}/output/MSLite-0.6.0-linux_arm32 | |||||
| rm -rf ${OUTPUT_DIR} && mkdir -p ${OUTPUT_DIR} && cd ${OUTPUT_DIR} | |||||
| mkdir -p ${OUTPUT_DIR}/time_profile && mkdir -p ${OUTPUT_DIR}/benchmark | |||||
| mkdir -p ${OUTPUT_DIR}/include && mkdir -p ${OUTPUT_DIR}/lib | |||||
| mkdir -p ${OUTPUT_DIR}/third_party | |||||
| prepare_md_lite | |||||
| cp ${BASEPATH}/mindspore/lite/build/tools/benchmark/benchmark ${OUTPUT_DIR}/benchmark/ | |||||
| cp ${BASEPATH}/mindspore/lite/build/tools/time_profile/timeprofile ${OUTPUT_DIR}/time_profile/ | |||||
| cp ${BASEPATH}/mindspore/lite/include/*.h ${OUTPUT_DIR}/include/ | |||||
| mkdir -p ${OUTPUT_DIR}/include/ir/dtype/ | |||||
| cp ${BASEPATH}/mindspore/core/ir/dtype/type_id.h ${OUTPUT_DIR}/include/ir/dtype/ | |||||
| mkdir -p ${OUTPUT_DIR}/include/schema/ | |||||
| cp ${BASEPATH}/mindspore/lite/schema/*.h ${OUTPUT_DIR}/include/schema/ | |||||
| cp ${BASEPATH}/mindspore/lite/build/src/libmindspore-lite.so ${OUTPUT_DIR}/lib/ | |||||
| mkdir -p ${OUTPUT_DIR}/third_party/flatbuffers | |||||
| cp -r ${BASEPATH}/third_party/flatbuffers/include/ ${OUTPUT_DIR}/third_party/flatbuffers/ | |||||
| cd .. | |||||
| tar -czf MSLite-0.6.0-linux_arm32.tar.gz MSLite-0.6.0-linux_arm32/ --warning=no-file-changed | |||||
| sha256sum MSLite-0.6.0-linux_arm32.tar.gz > MSLite-0.6.0-linux_arm32.tar.gz.sha256 | |||||
| rm -rf MSLite-0.6.0-linux_arm32/ | |||||
| fi | |||||
| mv ${BASEPATH}/output/tmp/*.tar.gz* ${BASEPATH}/output/ | |||||
| rm -rf ${BASEPATH}/output/tmp/ | |||||
| echo "---------------- mindspore lite: build success ----------------" | echo "---------------- mindspore lite: build success ----------------" | ||||
| fi | fi | ||||
| } | } | ||||
| @@ -0,0 +1,53 @@ | |||||
| include(CMakePackageConfigHelpers) | |||||
| set(LIB_DIR ${MAIN_DIR}/lib) | |||||
| set(INC_DIR ${MAIN_DIR}/include) | |||||
| set(TURBO_DIR ${MAIN_DIR}/third_party/libjpeg-turbo) | |||||
| set(OPENCV_DIR ${MAIN_DIR}/third_party/opencv) | |||||
| set(PROTOBF_DIR ${MAIN_DIR}/third_party/protobuf) | |||||
| set(FLATBF_DIR ${MAIN_DIR}/third_party/flatbuffers) | |||||
| install(DIRECTORY ${TOP_DIR}/mindspore/lite/include/ DESTINATION ${INC_DIR} COMPONENT ${COMPONENT_NAME} PATTERN "*.h") | |||||
| install(DIRECTORY ${TOP_DIR}/mindspore/lite/schema/ DESTINATION ${INC_DIR}/schema COMPONENT ${COMPONENT_NAME} PATTERN "*.h") | |||||
| install(FILES ${TOP_DIR}/mindspore/core/ir/dtype/type_id.h DESTINATION ${INC_DIR}/ir/dtype COMPONENT ${COMPONENT_NAME}) | |||||
| install(FILES ${TOP_DIR}/mindspore/lite/build/src/libmindspore-lite.so DESTINATION ${LIB_DIR} COMPONENT ${COMPONENT_NAME}) | |||||
| if (BUILD_MINDDATA) | |||||
| install(DIRECTORY ${TOP_DIR}/mindspore/ccsrc/minddata/dataset/include/ DESTINATION ${INC_DIR} COMPONENT ${COMPONENT_NAME} PATTERN "*.h") | |||||
| install(FILES ${TOP_DIR}/mindspore/lite/build/minddata/libminddata-lite.so DESTINATION ${LIB_DIR} COMPONENT ${COMPONENT_NAME}) | |||||
| install(FILES ${TOP_DIR}/third_party/libjpeg-turbo/lib/libjpeg.so DESTINATION ${TURBO_DIR}/lib COMPONENT ${COMPONENT_NAME}) | |||||
| install(FILES ${TOP_DIR}/third_party/libjpeg-turbo/lib/libturbojpeg.so DESTINATION ${TURBO_DIR}/lib COMPONENT ${COMPONENT_NAME}) | |||||
| if (PLATFORM_ARM64) | |||||
| install(FILES ${TOP_DIR}/third_party/opencv/build/lib/arm64-v8a/libopencv_core.so DESTINATION ${OPENCV_DIR}/lib/arm64-v8a COMPONENT ${COMPONENT_NAME}) | |||||
| install(FILES ${TOP_DIR}/third_party/opencv/build/lib/arm64-v8a/libopencv_imgcodecs.so DESTINATION ${OPENCV_DIR}/lib/arm64-v8a COMPONENT ${COMPONENT_NAME}) | |||||
| install(FILES ${TOP_DIR}/third_party/opencv/build/lib/arm64-v8a/libopencv_imgproc.so DESTINATION ${OPENCV_DIR}/lib/arm64-v8a COMPONENT ${COMPONENT_NAME}) | |||||
| elseif (PLATFORM_ARM32) | |||||
| install(FILES ${TOP_DIR}/third_party/opencv/build/lib/armeabi-v7a/libopencv_core.so DESTINATION ${OPENCV_DIR}/lib/armeabi-v7a COMPONENT ${COMPONENT_NAME}) | |||||
| install(FILES ${TOP_DIR}/third_party/opencv/build/lib/armeabi-v7a/libopencv_imgcodecs.so DESTINATION ${OPENCV_DIR}/lib/armeabi-v7a COMPONENT ${COMPONENT_NAME}) | |||||
| install(FILES ${TOP_DIR}/third_party/opencv/build/lib/armeabi-v7a/libopencv_imgproc.so DESTINATION ${OPENCV_DIR}/lib/armeabi-v7a COMPONENT ${COMPONENT_NAME}) | |||||
| else () | |||||
| install(FILES ${TOP_DIR}/third_party/opencv/build/lib/libopencv_core.so DESTINATION ${OPENCV_DIR}/lib COMPONENT ${COMPONENT_NAME}) | |||||
| install(FILES ${TOP_DIR}/third_party/opencv/build/lib/libopencv_imgcodecs.so DESTINATION ${OPENCV_DIR}/lib COMPONENT ${COMPONENT_NAME}) | |||||
| install(FILES ${TOP_DIR}/third_party/opencv/build/lib/libopencv_imgproc.so DESTINATION ${OPENCV_DIR}/lib COMPONENT ${COMPONENT_NAME}) | |||||
| endif () | |||||
| endif () | |||||
| if (PLATFORM_ARM64) | |||||
| install(FILES ${TOP_DIR}/mindspore/lite/build/src/runtime/kernel/arm/nnacl/liboptimize.so DESTINATION ${LIB_DIR} COMPONENT ${COMPONENT_NAME}) | |||||
| elseif (NOT PLATFORM_ARM32) | |||||
| install(DIRECTORY ${TOP_DIR}/third_party/protobuf/build/include DESTINATION ${PROTOBF_DIR} COMPONENT ${COMPONENT_NAME}) | |||||
| install(FILES ${TOP_DIR}/third_party/protobuf/build/lib/libprotobuf.so.19.0.0 DESTINATION ${PROTOBF_DIR}/lib RENAME libprotobuf.so.19 COMPONENT ${COMPONENT_NAME}) | |||||
| endif () | |||||
| install(DIRECTORY ${TOP_DIR}/third_party/flatbuffers/include DESTINATION ${FLATBF_DIR} COMPONENT ${COMPONENT_NAME}) | |||||
| set(CPACK_GENERATOR TGZ) | |||||
| set(CPACK_ARCHIVE_COMPONENT_INSTALL ON) | |||||
| set(CPACK_COMPONENTS_ALL ${COMPONENT_NAME}) | |||||
| set(CPACK_PACKAGE_FILE_NAME ${MAIN_DIR}) | |||||
| set(CPACK_PACKAGE_DIRECTORY ${TOP_DIR}/output/tmp) | |||||
| set(CPACK_PACKAGE_CHECKSUM SHA256) | |||||
| include(CPack) | |||||
| @@ -5,6 +5,28 @@ if (CMAKE_CXX_COMPILER_ID STREQUAL "GNU" AND CMAKE_CXX_COMPILER_VERSION VERSION_ | |||||
| message(FATAL_ERROR "GCC vesion ${CMAKE_CXX_COMPILER_VERSION} must not be less than 7.3.0") | message(FATAL_ERROR "GCC vesion ${CMAKE_CXX_COMPILER_VERSION} must not be less than 7.3.0") | ||||
| endif () | endif () | ||||
| set(MS_VERSION_MAJOY 0) | |||||
| set(MS_VERSION_MINOR 6) | |||||
| set(MS_VERSION_REVISION 0) | |||||
| set(DIR_PREFIX mindspore-lite) | |||||
| set(MS_VERSION ${MS_VERSION_MAJOY}.${MS_VERSION_MINOR}.${MS_VERSION_REVISION}) | |||||
| set(MAIN_DIR ${DIR_PREFIX}-${MS_VERSION}) | |||||
| if (SUPPORT_GPU) | |||||
| set(PROCESS_UNIT gpu) | |||||
| else () | |||||
| set(PROCESS_UNIT cpu) | |||||
| endif () | |||||
| if (PLATFORM_ARM64) | |||||
| set(COMPONENT_NAME runtime-arm64-${PROCESS_UNIT}) | |||||
| elseif (PLATFORM_ARM32) | |||||
| set(COMPONENT_NAME runtime-arm32-${PROCESS_UNIT}) | |||||
| else () | |||||
| set(COMPONENT_NAME convert-ubuntu) | |||||
| endif() | |||||
| set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++17") | set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++17") | ||||
| set(TOP_DIR ${CMAKE_CURRENT_SOURCE_DIR}/../..) | set(TOP_DIR ${CMAKE_CURRENT_SOURCE_DIR}/../..) | ||||
| set(CORE_DIR ${TOP_DIR}/mindspore/core) | set(CORE_DIR ${TOP_DIR}/mindspore/core) | ||||
| @@ -157,3 +179,5 @@ if (BUILD_DEVICE) | |||||
| add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/test) | add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/test) | ||||
| add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/tools/time_profile) | add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/tools/time_profile) | ||||
| endif() | endif() | ||||
| include(${TOP_DIR}/cmake/package_lite.cmake) | |||||
| @@ -9,8 +9,8 @@ function Run_x86() { | |||||
| continue | continue | ||||
| fi | fi | ||||
| echo ${model_name} | echo ${model_name} | ||||
| echo 'cd '${convertor_path}'/MSLite-*-linux_x86_64' | |||||
| cd ${convertor_path}/MSLite-*-linux_x86_64 || return 1 | |||||
| echo 'cd '${convertor_path}'/mindspore-lite-'${version}'-convert-ubuntu' | |||||
| cd ${convertor_path}/mindspore-lite-${version}-convert-ubuntu || return 1 | |||||
| echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath='${ms_models_path}'/'${model_name}'.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/input/'${model_name}'.ms.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/'${model_name}'.ms.out --warmUpLoopCount=1 --loopCount=1' || return 1 | echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath='${ms_models_path}'/'${model_name}'.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/input/'${model_name}'.ms.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/'${model_name}'.ms.out --warmUpLoopCount=1 --loopCount=1' || return 1 | ||||
| export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath=${ms_models_path}/${model_name}.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/input/${model_name}.ms.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/${model_name}.ms.out --warmUpLoopCount=1 --loopCount=1 | export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath=${ms_models_path}/${model_name}.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/input/${model_name}.ms.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/${model_name}.ms.out --warmUpLoopCount=1 --loopCount=1 | ||||
| if [ $? = 0 ]; then | if [ $? = 0 ]; then | ||||
| @@ -30,8 +30,8 @@ function Run_x86() { | |||||
| continue | continue | ||||
| fi | fi | ||||
| echo ${model_name} | echo ${model_name} | ||||
| echo 'cd '${convertor_path}'/MSLite-*-linux_x86_64' | |||||
| cd ${convertor_path}/MSLite-*-linux_x86_64 || return 1 | |||||
| echo 'cd '${convertor_path}'/mindspore-lite-'${version}'-convert-ubuntu' | |||||
| cd ${convertor_path}/mindspore-lite-${version}-convert-ubuntu || return 1 | |||||
| echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath='${ms_models_path}'/'${model_name}'.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/input/'${model_name}'.ms.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/'${model_name}'.ms.out --warmUpLoopCount=1 --loopCount=1' || return 1 | echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath='${ms_models_path}'/'${model_name}'.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/input/'${model_name}'.ms.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/'${model_name}'.ms.out --warmUpLoopCount=1 --loopCount=1' || return 1 | ||||
| export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath=${ms_models_path}/${model_name}.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/input/${model_name}.ms.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/${model_name}.ms.out --warmUpLoopCount=1 --loopCount=1 | export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath=${ms_models_path}/${model_name}.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/input/${model_name}.ms.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/${model_name}.ms.out --warmUpLoopCount=1 --loopCount=1 | ||||
| if [ $? = 0 ]; then | if [ $? = 0 ]; then | ||||
| @@ -51,8 +51,8 @@ function Run_x86() { | |||||
| continue | continue | ||||
| fi | fi | ||||
| echo ${model_name} | echo ${model_name} | ||||
| echo 'cd '${convertor_path}'/MSLite-*-linux_x86_64' | |||||
| cd ${convertor_path}/MSLite-*-linux_x86_64 || return 1 | |||||
| echo 'cd '${convertor_path}'/mindspore-lite-'${version}'-convert-ubuntu' | |||||
| cd ${convertor_path}/mindspore-lite-${version}-convert-ubuntu || return 1 | |||||
| echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath='${ms_models_path}'/'${model_name}'.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/input/'${model_name}'.ms.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/'${model_name}'.ms.out --warmUpLoopCount=1 --loopCount=1' || return 1 | echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath='${ms_models_path}'/'${model_name}'.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/input/'${model_name}'.ms.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/'${model_name}'.ms.out --warmUpLoopCount=1 --loopCount=1' || return 1 | ||||
| export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath=${ms_models_path}/${model_name}.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/input/${model_name}.ms.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/${model_name}.ms.out --warmUpLoopCount=1 --loopCount=1 | export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath=${ms_models_path}/${model_name}.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/input/${model_name}.ms.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/${model_name}.ms.out --warmUpLoopCount=1 --loopCount=1 | ||||
| if [ $? = 0 ]; then | if [ $? = 0 ]; then | ||||
| @@ -72,8 +72,8 @@ function Run_x86() { | |||||
| continue | continue | ||||
| fi | fi | ||||
| echo ${model_name} | echo ${model_name} | ||||
| echo 'cd '${convertor_path}'/MSLite-*-linux_x86_64' | |||||
| cd ${convertor_path}/MSLite-*-linux_x86_64 || return 1 | |||||
| echo 'cd '${convertor_path}'/mindspore-lite-'${version}'-convert-ubuntu' | |||||
| cd ${convertor_path}/mindspore-lite-${version}-convert-ubuntu || return 1 | |||||
| echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath='${ms_models_path}'/'${model_name}'_posttraining.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/quantTraining/mnist_calibration_data/00099.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/'${model_name}'_posttraining.ms.out --warmUpLoopCount=1 --loopCount=1' || return 1 | echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath='${ms_models_path}'/'${model_name}'_posttraining.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/quantTraining/mnist_calibration_data/00099.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/'${model_name}'_posttraining.ms.out --warmUpLoopCount=1 --loopCount=1' || return 1 | ||||
| export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath=${ms_models_path}/${model_name}_posttraining.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/quantTraining/mnist_calibration_data/00099.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/${model_name}_posttraining.ms.out --warmUpLoopCount=1 --loopCount=1 | export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath=${ms_models_path}/${model_name}_posttraining.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/quantTraining/mnist_calibration_data/00099.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/${model_name}_posttraining.ms.out --warmUpLoopCount=1 --loopCount=1 | ||||
| if [ $? = 0 ]; then | if [ $? = 0 ]; then | ||||
| @@ -93,8 +93,8 @@ function Run_x86() { | |||||
| continue | continue | ||||
| fi | fi | ||||
| echo ${model_name} | echo ${model_name} | ||||
| echo 'cd '${convertor_path}'/MSLite-*-linux_x86_64' | |||||
| cd ${convertor_path}/MSLite-*-linux_x86_64 || return 1 | |||||
| echo 'cd '${convertor_path}'/mindspore-lite-'${version}'-convert-ubuntu' | |||||
| cd ${convertor_path}/mindspore-lite-${version}-convert-ubuntu || return 1 | |||||
| echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath='${ms_models_path}'/'${model_name}'.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/input/${model_name}.ms.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/'${model_name}'.ms.out --warmUpLoopCount=1 --loopCount=1 --numThreads=1' || return 1 | echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath='${ms_models_path}'/'${model_name}'.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/input/${model_name}.ms.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/'${model_name}'.ms.out --warmUpLoopCount=1 --loopCount=1 --numThreads=1' || return 1 | ||||
| export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath=${ms_models_path}/${model_name}.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/input/${model_name}.ms.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/${model_name}.ms.out --warmUpLoopCount=1 --loopCount=1 --numThreads=1 | export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath=${ms_models_path}/${model_name}.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/input/${model_name}.ms.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/${model_name}.ms.out --warmUpLoopCount=1 --loopCount=1 --numThreads=1 | ||||
| if [ $? = 0 ]; then | if [ $? = 0 ]; then | ||||
| @@ -114,8 +114,8 @@ function Run_x86() { | |||||
| continue | continue | ||||
| fi | fi | ||||
| echo ${model_name} | echo ${model_name} | ||||
| echo 'cd '${convertor_path}'/MSLite-*-linux_x86_64' | |||||
| cd ${convertor_path}/MSLite-*-linux_x86_64 || return 1 | |||||
| echo 'cd '${convertor_path}'/mindspore-lite-'${version}'-convert-ubuntu' | |||||
| cd ${convertor_path}/mindspore-lite-${version}-convert-ubuntu || return 1 | |||||
| echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath='${ms_models_path}'/'${model_name}'.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/input/'${model_name}'.ms.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/'${model_name}'.ms.out --warmUpLoopCount=1 --loopCount=1' || return 1 | echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath='${ms_models_path}'/'${model_name}'.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/input/'${model_name}'.ms.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/'${model_name}'.ms.out --warmUpLoopCount=1 --loopCount=1' || return 1 | ||||
| export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath=${ms_models_path}/${model_name}.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/input/${model_name}.ms.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/${model_name}.ms.out --warmUpLoopCount=1 --loopCount=1 --accuracyThreshold=1.5 | export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./lib;./benchmark/benchmark --modelPath=${ms_models_path}/${model_name}.ms --inDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/input/${model_name}.ms.bin --calibDataPath=/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/output/${model_name}.ms.out --warmUpLoopCount=1 --loopCount=1 --accuracyThreshold=1.5 | ||||
| if [ $? = 0 ]; then | if [ $? = 0 ]; then | ||||
| @@ -241,19 +241,27 @@ while getopts "a:c:m:d:" opt; do | |||||
| esac | esac | ||||
| done | done | ||||
| file_name=$(ls ${arm_path}/*runtime-arm64*.tar.gz) | |||||
| IFS="-" read -r -a file_name_array <<< "$file_name" | |||||
| version=${file_name_array[2]} | |||||
| IFS="." read -r -a suffix <<< "${file_name_array[-1]}" | |||||
| process_unit=${suffix[0]} | |||||
| # Unzip arm | # Unzip arm | ||||
| cd ${arm_path} || exit 1 | cd ${arm_path} || exit 1 | ||||
| tar -zxf MSLite-*-linux_arm64.tar.gz || exit 1 | |||||
| mkdir mindspore-lite-${version}-runtime-arm64-${process_unit} | |||||
| tar -zxf mindspore-lite-${version}-runtime-arm64-${process_unit}.tar.gz -C mindspore-lite-${version}-runtime-arm64-${process_unit} --strip-components 1 || exit 1 | |||||
| # Unzip convertor | # Unzip convertor | ||||
| cd ${convertor_path} || exit 1 | cd ${convertor_path} || exit 1 | ||||
| tar -zxf MSLite-*-linux_x86_64.tar.gz || exit 1 | |||||
| cd ${convertor_path}/MSLite-*-linux_x86_64 || exit 1 | |||||
| mkdir mindspore-lite-${version}-convert-ubuntu | |||||
| tar -zxf mindspore-lite-${version}-convert-ubuntu.tar.gz -C mindspore-lite-${version}-convert-ubuntu --strip-components 1 || exit 1 | |||||
| cd ${convertor_path}/mindspore-lite-${version}-convert-ubuntu || exit 1 | |||||
| cp converter/converter_lite ./ || exit 1 | cp converter/converter_lite ./ || exit 1 | ||||
| export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:./lib/:./third_party/protobuf/lib | export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:./lib/:./third_party/protobuf/lib | ||||
| # Convert the models | # Convert the models | ||||
| cd ${convertor_path}/MSLite-*-linux_x86_64 || exit 1 | |||||
| cd ${convertor_path}/mindspore-lite-${version}-convert-ubuntu || exit 1 | |||||
| # Set models config filepath | # Set models config filepath | ||||
| models_tflite_config=${basepath}/models_tflite.cfg | models_tflite_config=${basepath}/models_tflite.cfg | ||||
| @@ -342,8 +350,8 @@ rm -rf ${basepath}/benchmark_test | |||||
| mkdir -p ${basepath}/benchmark_test | mkdir -p ${basepath}/benchmark_test | ||||
| benchmark_test_path=${basepath}/benchmark_test | benchmark_test_path=${basepath}/benchmark_test | ||||
| cd ${benchmark_test_path} || exit 1 | cd ${benchmark_test_path} || exit 1 | ||||
| cp -a ${arm_path}/MSLite-*-linux_arm64/lib/libmindspore-lite.so ${benchmark_test_path}/libmindspore-lite.so || exit 1 | |||||
| cp -a ${arm_path}/MSLite-*-linux_arm64/benchmark/benchmark ${benchmark_test_path}/benchmark || exit 1 | |||||
| cp -a ${arm_path}/mindspore-lite-${version}-runtime-arm64-${process_unit}/lib/libmindspore-lite.so ${benchmark_test_path}/libmindspore-lite.so || exit 1 | |||||
| cp -a ${arm_path}/mindspore-lite-${version}-runtime-arm64-${process_unit}/benchmark/benchmark ${benchmark_test_path}/benchmark || exit 1 | |||||
| # Copy the MindSpore models: | # Copy the MindSpore models: | ||||
| cp -a ${ms_models_path}/*.ms ${benchmark_test_path} || exit 1 | cp -a ${ms_models_path}/*.ms ${benchmark_test_path} || exit 1 | ||||
| @@ -15,3 +15,6 @@ if (PLATFORM_ARM32 OR PLATFORM_ARM64) | |||||
| else() | else() | ||||
| target_link_libraries(benchmark mindspore-lite pthread) | target_link_libraries(benchmark mindspore-lite pthread) | ||||
| endif() | endif() | ||||
| install(TARGETS benchmark | |||||
| RUNTIME DESTINATION ${MAIN_DIR}/benchmark COMPONENT ${COMPONENT_NAME}) | |||||
| @@ -115,3 +115,7 @@ target_link_libraries(converter_lite PRIVATE | |||||
| mindspore::eigen | mindspore::eigen | ||||
| ) | ) | ||||
| if (NOT PLATFORM_arm64 AND NOT PLATFORM_arm32) | |||||
| install(TARGETS converter_lite | |||||
| RUNTIME DESTINATION ${MAIN_DIR}/converter COMPONENT ${COMPONENT_NAME}) | |||||
| endif () | |||||
| @@ -16,3 +16,6 @@ if (PLATFORM_ARM32 OR PLATFORM_ARM64) | |||||
| else() | else() | ||||
| target_link_libraries(timeprofile mindspore-lite pthread) | target_link_libraries(timeprofile mindspore-lite pthread) | ||||
| endif() | endif() | ||||
| install(TARGETS timeprofile | |||||
| RUNTIME DESTINATION ${MAIN_DIR}/time_profile COMPONENT ${COMPONENT_NAME}) | |||||