generate train runtime

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})