|
|
|
@@ -65,14 +65,14 @@ if (SUPPORT_TRAIN) |
|
|
|
) |
|
|
|
endif () |
|
|
|
|
|
|
|
file(GLOB_RECURSE C_OPS_SRC ${CMAKE_CURRENT_SOURCE_DIR}/ops/*.cc ${CMAKE_CURRENT_SOURCE_DIR}/ops/populate/*.cc) |
|
|
|
add_subdirectory(ops) |
|
|
|
add_subdirectory(runtime/kernel/arm) |
|
|
|
|
|
|
|
add_library(mindspore-lite SHARED ${LITE_SRC} ${C_OPS_SRC}) |
|
|
|
add_library(mindspore-lite SHARED ${LITE_SRC}) |
|
|
|
add_dependencies(mindspore-lite fbs_src) |
|
|
|
set_target_properties(mindspore-lite PROPERTIES CLEAN_DIRECT_OUTPUT 1) |
|
|
|
|
|
|
|
add_library(mindspore-lite_static STATIC ${LITE_SRC} ${C_OPS_SRC}) |
|
|
|
add_library(mindspore-lite_static STATIC ${LITE_SRC}) |
|
|
|
add_dependencies(mindspore-lite_static fbs_src) |
|
|
|
set_target_properties(mindspore-lite_static PROPERTIES OUTPUT_NAME "mindspore-lite") |
|
|
|
set_target_properties(mindspore-lite_static PROPERTIES CLEAN_DIRECT_OUTPUT 1) |
|
|
|
@@ -80,11 +80,11 @@ set_target_properties(mindspore-lite_static PROPERTIES CLEAN_DIRECT_OUTPUT 1) |
|
|
|
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-unused-private-field") |
|
|
|
if (SUPPORT_GPU) |
|
|
|
add_subdirectory(runtime/kernel/opencl) |
|
|
|
target_link_libraries(mindspore-lite cpu_kernel_mid opencl_kernel_mid nnacl) |
|
|
|
target_link_libraries(mindspore-lite_static cpu_kernel_mid opencl_kernel_mid nnacl_mid) |
|
|
|
target_link_libraries(mindspore-lite cpu_kernel_mid opencl_kernel_mid nnacl cpu_ops_mid) |
|
|
|
target_link_libraries(mindspore-lite_static cpu_kernel_mid opencl_kernel_mid nnacl_mid cpu_ops_mid) |
|
|
|
else () |
|
|
|
target_link_libraries(mindspore-lite cpu_kernel_mid nnacl) |
|
|
|
target_link_libraries(mindspore-lite_static cpu_kernel_mid nnacl_mid) |
|
|
|
target_link_libraries(mindspore-lite cpu_kernel_mid nnacl cpu_ops_mid) |
|
|
|
target_link_libraries(mindspore-lite_static cpu_kernel_mid nnacl_mid cpu_ops_mid) |
|
|
|
endif () |
|
|
|
if (PLATFORM_ARM32 OR PLATFORM_ARM64) |
|
|
|
target_link_libraries(mindspore-lite log) |
|
|
|
|