|
|
|
@@ -84,25 +84,26 @@ add_compile_options(-fPIC) |
|
|
|
|
|
|
|
if(SUPPORT_TRAIN) |
|
|
|
set(BUILD_MINDDATA "full") |
|
|
|
if(PLATFORM_ARM64) |
|
|
|
set(RUNTIME_COMPONENT_NAME train-android-aarch64) |
|
|
|
elseif(PLATFORM_ARM32) |
|
|
|
set(RUNTIME_COMPONENT_NAME train-android-aarch32) |
|
|
|
elseif(WIN32) |
|
|
|
set(RUNTIME_COMPONENT_NAME train-win-x64) |
|
|
|
else() |
|
|
|
set(RUNTIME_COMPONENT_NAME train-linux-x64) |
|
|
|
endif() |
|
|
|
set(TRAIN_OR_INFERENCE "train") |
|
|
|
else() |
|
|
|
if(PLATFORM_ARM64) |
|
|
|
set(RUNTIME_COMPONENT_NAME inference-android-aarch64) |
|
|
|
elseif(PLATFORM_ARM32) |
|
|
|
set(RUNTIME_COMPONENT_NAME inference-android-aarch32) |
|
|
|
elseif(WIN32) |
|
|
|
set(RUNTIME_COMPONENT_NAME inference-win-x64) |
|
|
|
set(TRAIN_OR_INFERENCE "inference") |
|
|
|
endif() |
|
|
|
|
|
|
|
if(PLATFORM_ARM64) |
|
|
|
set(RUNTIME_COMPONENT_NAME ${TRAIN_OR_INFERENCE}-android-aarch64) |
|
|
|
elseif(PLATFORM_ARM32) |
|
|
|
set(RUNTIME_COMPONENT_NAME ${TRAIN_OR_INFERENCE}-android-aarch32) |
|
|
|
elseif(WIN32) |
|
|
|
execute_process(COMMAND "${CMAKE_C_COMPILER}" -dumpmachine |
|
|
|
OUTPUT_VARIABLE i686_or_x86_64 |
|
|
|
) |
|
|
|
if(i686_or_x86_64 MATCHES "^i686-") |
|
|
|
set(RUNTIME_COMPONENT_NAME ${TRAIN_OR_INFERENCE}-win-x86) |
|
|
|
else() |
|
|
|
set(RUNTIME_COMPONENT_NAME inference-linux-x64) |
|
|
|
set(RUNTIME_COMPONENT_NAME ${TRAIN_OR_INFERENCE}-win-x64) |
|
|
|
endif() |
|
|
|
else() |
|
|
|
set(RUNTIME_COMPONENT_NAME ${TRAIN_OR_INFERENCE}-linux-x64) |
|
|
|
endif() |
|
|
|
|
|
|
|
string(REPLACE "/mindspore/lite" "" TOP_DIR ${CMAKE_CURRENT_SOURCE_DIR}) |
|
|
|
|