| @@ -54,6 +54,7 @@ endif () | |||||
| if (PLATFORM_ARM64) | if (PLATFORM_ARM64) | ||||
| install(FILES ${TOP_DIR}/mindspore/lite/build/src/libmindspore-lite.so DESTINATION ${LIB_DIR} COMPONENT ${COMPONENT_NAME}) | install(FILES ${TOP_DIR}/mindspore/lite/build/src/libmindspore-lite.so DESTINATION ${LIB_DIR} COMPONENT ${COMPONENT_NAME}) | ||||
| install(FILES ${TOP_DIR}/mindspore/lite/build/src/libmindspore-lite.a DESTINATION ${LIB_DIR} COMPONENT ${COMPONENT_NAME}) | |||||
| install(FILES ${TOP_DIR}/mindspore/core/ir/dtype/type_id.h DESTINATION ${INC_DIR}/ir/dtype COMPONENT ${COMPONENT_NAME}) | install(FILES ${TOP_DIR}/mindspore/core/ir/dtype/type_id.h DESTINATION ${INC_DIR}/ir/dtype COMPONENT ${COMPONENT_NAME}) | ||||
| install(DIRECTORY ${TOP_DIR}/mindspore/lite/include/ DESTINATION ${INC_DIR} COMPONENT ${COMPONENT_NAME} FILES_MATCHING PATTERN "*.h") | install(DIRECTORY ${TOP_DIR}/mindspore/lite/include/ DESTINATION ${INC_DIR} COMPONENT ${COMPONENT_NAME} FILES_MATCHING PATTERN "*.h") | ||||
| install(DIRECTORY ${TOP_DIR}/mindspore/lite/schema/ DESTINATION ${INC_DIR}/schema COMPONENT ${COMPONENT_NAME} FILES_MATCHING PATTERN "*.h" PATTERN "inner" EXCLUDE) | install(DIRECTORY ${TOP_DIR}/mindspore/lite/schema/ DESTINATION ${INC_DIR}/schema COMPONENT ${COMPONENT_NAME} FILES_MATCHING PATTERN "*.h" PATTERN "inner" EXCLUDE) | ||||
| @@ -61,6 +62,7 @@ if (PLATFORM_ARM64) | |||||
| install(DIRECTORY ${TOP_DIR}/third_party/flatbuffers/include DESTINATION ${FLATBF_DIR} COMPONENT ${COMPONENT_NAME}) | install(DIRECTORY ${TOP_DIR}/third_party/flatbuffers/include DESTINATION ${FLATBF_DIR} COMPONENT ${COMPONENT_NAME}) | ||||
| elseif (PLATFORM_ARM32) | elseif (PLATFORM_ARM32) | ||||
| install(FILES ${TOP_DIR}/mindspore/lite/build/src/libmindspore-lite.so DESTINATION ${LIB_DIR} COMPONENT ${COMPONENT_NAME}) | install(FILES ${TOP_DIR}/mindspore/lite/build/src/libmindspore-lite.so DESTINATION ${LIB_DIR} COMPONENT ${COMPONENT_NAME}) | ||||
| install(FILES ${TOP_DIR}/mindspore/lite/build/src/libmindspore-lite.a DESTINATION ${LIB_DIR} COMPONENT ${COMPONENT_NAME}) | |||||
| install(FILES ${TOP_DIR}/mindspore/core/ir/dtype/type_id.h DESTINATION ${INC_DIR}/ir/dtype COMPONENT ${COMPONENT_NAME}) | install(FILES ${TOP_DIR}/mindspore/core/ir/dtype/type_id.h DESTINATION ${INC_DIR}/ir/dtype COMPONENT ${COMPONENT_NAME}) | ||||
| install(DIRECTORY ${TOP_DIR}/mindspore/lite/include/ DESTINATION ${INC_DIR} COMPONENT ${COMPONENT_NAME} FILES_MATCHING PATTERN "*.h") | install(DIRECTORY ${TOP_DIR}/mindspore/lite/include/ DESTINATION ${INC_DIR} COMPONENT ${COMPONENT_NAME} FILES_MATCHING PATTERN "*.h") | ||||
| install(DIRECTORY ${TOP_DIR}/mindspore/lite/schema/ DESTINATION ${INC_DIR}/schema COMPONENT ${COMPONENT_NAME} FILES_MATCHING PATTERN "*.h" PATTERN "inner" EXCLUDE) | install(DIRECTORY ${TOP_DIR}/mindspore/lite/schema/ DESTINATION ${INC_DIR}/schema COMPONENT ${COMPONENT_NAME} FILES_MATCHING PATTERN "*.h" PATTERN "inner" EXCLUDE) | ||||
| @@ -77,7 +79,7 @@ else () | |||||
| install(FILES ${TOP_DIR}/mindspore/core/ir/dtype/type_id.h DESTINATION ${INC_DIR_RUN_X86}/ir/dtype COMPONENT ${RUN_X86_COMPONENT_NAME}) | install(FILES ${TOP_DIR}/mindspore/core/ir/dtype/type_id.h DESTINATION ${INC_DIR_RUN_X86}/ir/dtype COMPONENT ${RUN_X86_COMPONENT_NAME}) | ||||
| install(DIRECTORY ${TOP_DIR}/third_party/flatbuffers/include DESTINATION ${FLATBF_DIR_RUN_X86} COMPONENT ${RUN_X86_COMPONENT_NAME}) | install(DIRECTORY ${TOP_DIR}/third_party/flatbuffers/include DESTINATION ${FLATBF_DIR_RUN_X86} COMPONENT ${RUN_X86_COMPONENT_NAME}) | ||||
| install(FILES ${TOP_DIR}/mindspore/lite/build/src/libmindspore-lite.so DESTINATION ${LIB_DIR_RUN_X86} COMPONENT ${RUN_X86_COMPONENT_NAME}) | install(FILES ${TOP_DIR}/mindspore/lite/build/src/libmindspore-lite.so DESTINATION ${LIB_DIR_RUN_X86} COMPONENT ${RUN_X86_COMPONENT_NAME}) | ||||
| install(FILES ${TOP_DIR}/mindspore/lite/build/src/libmindspore-lite.a DESTINATION ${LIB_DIR_RUN_X86} COMPONENT ${RUN_X86_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}) | 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}) | ||||
| install(FILES ${TOP_DIR}/third_party/flatbuffers/build/libflatbuffers.so.1.11.0 DESTINATION ${FLATBF_DIR}/lib RENAME libflatbuffers.so.1 COMPONENT ${COMPONENT_NAME}) | install(FILES ${TOP_DIR}/third_party/flatbuffers/build/libflatbuffers.so.1.11.0 DESTINATION ${FLATBF_DIR}/lib RENAME libflatbuffers.so.1 COMPONENT ${COMPONENT_NAME}) | ||||
| endif () | endif () | ||||
| @@ -181,7 +181,7 @@ if (BUILD_MINDDATA STREQUAL "full") | |||||
| elseif (BUILD_MINDDATA STREQUAL "lite") | elseif (BUILD_MINDDATA STREQUAL "lite") | ||||
| list(REMOVE_ITEM MINDDATA_CORE_SRC_FILES "${MINDDATA_DIR}/core/client.cc") | list(REMOVE_ITEM MINDDATA_CORE_SRC_FILES "${MINDDATA_DIR}/core/client.cc") | ||||
| list(REMOVE_ITEM MINDDATA_KERNELS_SRC_FILES "${MINDDATA_DIR}/kernels/py_func_op.cc") | list(REMOVE_ITEM MINDDATA_KERNELS_SRC_FILES "${MINDDATA_DIR}/kernels/py_func_op.cc") | ||||
| add_library(minddata-eager OBJECT | |||||
| add_library(minddata_eager_mid OBJECT | |||||
| ${MINDDATA_DIR}/api/de_tensor.cc | ${MINDDATA_DIR}/api/de_tensor.cc | ||||
| ${MINDDATA_DIR}/api/execute.cc | ${MINDDATA_DIR}/api/execute.cc | ||||
| ) | ) | ||||
| @@ -26,55 +26,67 @@ set(LITE_SRC | |||||
| ) | ) | ||||
| if (SUPPORT_GPU) | if (SUPPORT_GPU) | ||||
| set(LITE_SRC | |||||
| ${LITE_SRC} | |||||
| ${CMAKE_CURRENT_SOURCE_DIR}/runtime/kernel/opencl/subgraph_opencl_kernel.cc | |||||
| ${CMAKE_CURRENT_SOURCE_DIR}/runtime/kernel/opencl/utils.cc | |||||
| ${CMAKE_CURRENT_SOURCE_DIR}/runtime/opencl/opencl_executor.cc | |||||
| ${CMAKE_CURRENT_SOURCE_DIR}/runtime/opencl/opencl_allocator.cc | |||||
| ${CMAKE_CURRENT_SOURCE_DIR}/runtime/opencl/opencl_runtime.cc | |||||
| ${CMAKE_CURRENT_SOURCE_DIR}/runtime/opencl/opencl_wrapper.cc | |||||
| ) | |||||
| set(LITE_SRC | |||||
| ${LITE_SRC} | |||||
| ${CMAKE_CURRENT_SOURCE_DIR}/runtime/kernel/opencl/subgraph_opencl_kernel.cc | |||||
| ${CMAKE_CURRENT_SOURCE_DIR}/runtime/kernel/opencl/utils.cc | |||||
| ${CMAKE_CURRENT_SOURCE_DIR}/runtime/opencl/opencl_executor.cc | |||||
| ${CMAKE_CURRENT_SOURCE_DIR}/runtime/opencl/opencl_allocator.cc | |||||
| ${CMAKE_CURRENT_SOURCE_DIR}/runtime/opencl/opencl_runtime.cc | |||||
| ${CMAKE_CURRENT_SOURCE_DIR}/runtime/opencl/opencl_wrapper.cc | |||||
| ) | |||||
| endif() | endif() | ||||
| if (SUPPORT_TRAIN) | if (SUPPORT_TRAIN) | ||||
| set(ANF_SRC | set(ANF_SRC | ||||
| ${ANF_SRC} | |||||
| ${ANF_SRC} | |||||
| ) | ) | ||||
| set(PASS_SRC) | set(PASS_SRC) | ||||
| set(LITE_SRC | set(LITE_SRC | ||||
| ${LITE_SRC} | ${LITE_SRC} | ||||
| ${ANF_SRC} | ${ANF_SRC} | ||||
| # ${CMAKE_CURRENT_SOURCE_DIR}/train/ops/train_ops.cc | |||||
| ${CMAKE_CURRENT_SOURCE_DIR}/train/train_populate_parameter.cc | ${CMAKE_CURRENT_SOURCE_DIR}/train/train_populate_parameter.cc | ||||
| ${CMAKE_CURRENT_SOURCE_DIR}/train/train_session.cc | ${CMAKE_CURRENT_SOURCE_DIR}/train/train_session.cc | ||||
| ${CMAKE_CURRENT_SOURCE_DIR}/lite_session.cc | ${CMAKE_CURRENT_SOURCE_DIR}/lite_session.cc | ||||
| ) | ) | ||||
| endif () | endif () | ||||
| file(GLOB_RECURSE C_OPS_SRC ${CMAKE_CURRENT_SOURCE_DIR}/ops/*.cc) | file(GLOB_RECURSE C_OPS_SRC ${CMAKE_CURRENT_SOURCE_DIR}/ops/*.cc) | ||||
| add_subdirectory(runtime/kernel/arm) | |||||
| add_library(mindspore-lite SHARED ${LITE_SRC} ${C_OPS_SRC}) | add_library(mindspore-lite SHARED ${LITE_SRC} ${C_OPS_SRC}) | ||||
| set_target_properties(mindspore-lite PROPERTIES CLEAN_DIRECT_OUTPUT 1) | |||||
| add_library(mindspore-lite_static STATIC ${LITE_SRC} ${C_OPS_SRC}) | |||||
| set_target_properties(mindspore-lite_static PROPERTIES OUTPUT_NAME "mindspore-lite") | |||||
| set_target_properties(mindspore-lite_static PROPERTIES CLEAN_DIRECT_OUTPUT 1) | |||||
| set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-unused-private-field") | set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-unused-private-field") | ||||
| if (SUPPORT_GPU) | if (SUPPORT_GPU) | ||||
| add_subdirectory(runtime/kernel/opencl) | add_subdirectory(runtime/kernel/opencl) | ||||
| target_link_libraries(mindspore-lite | target_link_libraries(mindspore-lite | ||||
| cpu_kernel_mid_ | |||||
| opencl_kernel_lib_ | |||||
| cpu_kernel_mid | |||||
| opencl_kernel_mid | |||||
| ) | |||||
| target_link_libraries(mindspore-lite_static | |||||
| cpu_kernel_mid | |||||
| opencl_kernel_mid | |||||
| ) | ) | ||||
| else () | else () | ||||
| target_link_libraries(mindspore-lite | target_link_libraries(mindspore-lite | ||||
| cpu_kernel_mid_ | |||||
| cpu_kernel_mid | |||||
| ) | |||||
| target_link_libraries(mindspore-lite_static | |||||
| cpu_kernel_mid | |||||
| ) | ) | ||||
| endif () | endif () | ||||
| add_subdirectory(runtime/kernel/arm) | |||||
| if (PLATFORM_ARM32 OR PLATFORM_ARM64) | if (PLATFORM_ARM32 OR PLATFORM_ARM64) | ||||
| target_link_libraries(mindspore-lite log) | target_link_libraries(mindspore-lite log) | ||||
| endif () | endif () | ||||
| if (BUILD_MINDDATA STREQUAL "lite") | if (BUILD_MINDDATA STREQUAL "lite") | ||||
| target_link_libraries(mindspore-lite minddata-eager minddata-lite) | |||||
| target_link_libraries(mindspore-lite minddata_eager_mid minddata-lite) | |||||
| target_link_libraries(mindspore-lite_static minddata_eager_mid) | |||||
| endif () | endif () | ||||
| if ("${CMAKE_BUILD_TYPE}" STREQUAL "Release" AND (PLATFORM_ARM64 OR PLATFORM_ARM32)) | if ("${CMAKE_BUILD_TYPE}" STREQUAL "Release" AND (PLATFORM_ARM64 OR PLATFORM_ARM32)) | ||||
| @@ -16,6 +16,7 @@ file (GLOB TRAIN_KERNEL_SRC | |||||
| ${CMAKE_CURRENT_SOURCE_DIR}/fp32_grad/*.cc | ${CMAKE_CURRENT_SOURCE_DIR}/fp32_grad/*.cc | ||||
| ${CMAKE_CURRENT_SOURCE_DIR}/../../../../nnacl/fp32_grad/*.c | ${CMAKE_CURRENT_SOURCE_DIR}/../../../../nnacl/fp32_grad/*.c | ||||
| ) | ) | ||||
| set(KERNEL_SRC ${KERNEL_SRC} ${TRAIN_KERNEL_SRC}) | |||||
| endif() | endif() | ||||
| if (PLATFORM_ARM64) | if (PLATFORM_ARM64) | ||||
| @@ -35,4 +36,4 @@ if (PLATFORM_ARM32) | |||||
| set(KERNEL_SRC ${KERNEL_SRC} ${ASSEMBLY_SRC}) | set(KERNEL_SRC ${KERNEL_SRC} ${ASSEMBLY_SRC}) | ||||
| endif() | endif() | ||||
| add_library(cpu_kernel_mid_ OBJECT ${KERNEL_SRC} ${TRAIN_KERNEL_SRC}) | |||||
| add_library(cpu_kernel_mid OBJECT ${KERNEL_SRC}) | |||||
| @@ -38,9 +38,6 @@ class PoolingGradCPUKernel : public LiteKernel { | |||||
| int Init() override; | int Init() override; | ||||
| int ReSize() override; | int ReSize() override; | ||||
| int Run() override; | int Run() override; | ||||
| private: | |||||
| uint8_t data_shape_{0}; | |||||
| }; | }; | ||||
| } // namespace mindspore::kernel | } // namespace mindspore::kernel | ||||
| @@ -1,2 +1,2 @@ | |||||
| file(GLOB_RECURSE OPENCL_KERNEL_SRC ${CMAKE_CURRENT_SOURCE_DIR}/kernel/*.cc) | file(GLOB_RECURSE OPENCL_KERNEL_SRC ${CMAKE_CURRENT_SOURCE_DIR}/kernel/*.cc) | ||||
| add_library(opencl_kernel_lib_ OBJECT ${KERNEL_SRC} ${OPENCL_KERNEL_SRC}) | |||||
| add_library(opencl_kernel_mid OBJECT ${OPENCL_KERNEL_SRC}) | |||||
| @@ -336,8 +336,8 @@ endif() | |||||
| if (BUILD_MINDDATA STREQUAL "lite") | if (BUILD_MINDDATA STREQUAL "lite") | ||||
| target_link_libraries(lite-test | target_link_libraries(lite-test | ||||
| minddata_eager_mid | |||||
| minddata-lite | minddata-lite | ||||
| minddata-eager | |||||
| ) | ) | ||||
| if (PLATFORM_ARM32 OR PLATFORM_ARM64) | if (PLATFORM_ARM32 OR PLATFORM_ARM64) | ||||
| target_link_libraries(lite-test log) | target_link_libraries(lite-test log) | ||||