From a6e9044ed2390ec462c6b13aa009ce8342f2ec5a Mon Sep 17 00:00:00 2001 From: l00421674 Date: Tue, 8 Sep 2020 10:48:03 +0800 Subject: [PATCH] generate train runtime --- mindspore/lite/CMakeLists.txt | 32 +++++++++++++++++++++++--------- 1 file changed, 23 insertions(+), 9 deletions(-) diff --git a/mindspore/lite/CMakeLists.txt b/mindspore/lite/CMakeLists.txt index 30c700f48f..32030375f0 100644 --- a/mindspore/lite/CMakeLists.txt +++ b/mindspore/lite/CMakeLists.txt @@ -21,17 +21,31 @@ 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}) -elseif (WIN32) - set(PARSER_NAME libconverter-parser-win-${PROCESS_UNIT}) - set(COMPONENT_NAME converter-win-${PROCESS_UNIT}) +if (SUPPORT_TRAIN) + if (PLATFORM_ARM64) + set(COMPONENT_NAME runtime-arm64-${PROCESS_UNIT}-train) + elseif (PLATFORM_ARM32) + set(COMPONENT_NAME runtime-arm32-${PROCESS_UNIT}-train) + elseif (WIN32) + set(PARSER_NAME libconverter-parser-win-${PROCESS_UNIT}) + set(COMPONENT_NAME converter-win-${PROCESS_UNIT}) + else () + set(COMPONENT_NAME converter-ubuntu) + endif() + set(RUN_X86_COMPONENT_NAME runtime-x86-${PROCESS_UNIT}-train) else () - set(COMPONENT_NAME converter-ubuntu) + if (PLATFORM_ARM64) + set(COMPONENT_NAME runtime-arm64-${PROCESS_UNIT}) + elseif (PLATFORM_ARM32) + set(COMPONENT_NAME runtime-arm32-${PROCESS_UNIT}) + elseif (WIN32) + set(PARSER_NAME libconverter-parser-win-${PROCESS_UNIT}) + set(COMPONENT_NAME converter-win-${PROCESS_UNIT}) + else () + set(COMPONENT_NAME converter-ubuntu) + endif() + set(RUN_X86_COMPONENT_NAME runtime-x86-${PROCESS_UNIT}) endif() -set(RUN_X86_COMPONENT_NAME runtime-x86-${PROCESS_UNIT}) set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++17") string(REPLACE "/mindspore/lite" "" TOP_DIR ${CMAKE_CURRENT_SOURCE_DIR})