From d45e3ae52b45e87b64ae1ea1d372b3e7b4bcfc67 Mon Sep 17 00:00:00 2001 From: yeyunpeng Date: Fri, 21 Aug 2020 10:01:42 +0800 Subject: [PATCH] Change Source Code Compilation in Converter CMakeList --- mindspore/lite/tools/converter/CMakeLists.txt | 41 ++++++++++++++++++- 1 file changed, 40 insertions(+), 1 deletion(-) diff --git a/mindspore/lite/tools/converter/CMakeLists.txt b/mindspore/lite/tools/converter/CMakeLists.txt index 69c13fa575..06a863ca34 100644 --- a/mindspore/lite/tools/converter/CMakeLists.txt +++ b/mindspore/lite/tools/converter/CMakeLists.txt @@ -113,6 +113,45 @@ add_subdirectory(parser/onnx) add_subdirectory(legacy_optimizer) add_subdirectory(quantizer) +set(SRC_DIR ${CMAKE_CURRENT_SOURCE_DIR}/../../src) +set(LITE_SRC + ${SRC_DIR}/common/graph_util.cc + ${SRC_DIR}/common/ms_tensor_utils.cc + ${SRC_DIR}/runtime/allocator.cc + ${SRC_DIR}/runtime/runtime_api.cc + ${SRC_DIR}/runtime/thread_pool.cc + ${SRC_DIR}/runtime/workspace_pool.cc + ${SRC_DIR}/ir/tensor.cc + ${SRC_DIR}/context.cc + ${SRC_DIR}/kernel_registry.cc + ${SRC_DIR}/lite_kernel.cc + ${SRC_DIR}/populate_parameter.cc + ${SRC_DIR}/scheduler.cc + ${SRC_DIR}/lite_session.cc + ${SRC_DIR}/executor.cc + ${SRC_DIR}/model.cc + ) + +set(ARM_DIR ${CMAKE_CURRENT_SOURCE_DIR}/../../src/runtime/kernel/arm) +file(GLOB KERNEL_SRC + ${ARM_DIR}/base/*.cc + ${ARM_DIR}/nnacl/*.c + ${ARM_DIR}/nnacl/fp32/*.c + ${ARM_DIR}/nnacl/int8/*.c + ${ARM_DIR}/nnacl/quantization/*.c + ${ARM_DIR}/fp32/*.cc + ${ARM_DIR}/int8/*.cc + ) +list(REMOVE_ITEM KERNEL_SRC ${ARM_DIR}/nnacl/opt_op_handler.c) + +if (PLATFORM_ARM64) + # assembly + file(GLOB ASSEMBLY_SRC nnacl/assembly/arm64/*.s + nnacl/assembly/arm64/*.S) + set_property(SOURCE ${ASSEMBLY_SRC} PROPERTY LANGUAGE C) + set(KERNEL_SRC ${KERNEL_SRC} ${ASSEMBLY_SRC}) +endif () + if (WIN32) set(PROTO_LIBRARY mindspore::protobuf) else() @@ -124,6 +163,7 @@ add_executable(converter_lite ${ANF_SRC} ${CONVERTER_SRC} ${OPS_SRC} + ${KERNEL_SRC} ${LITE_SRC} ) @@ -139,7 +179,6 @@ target_link_libraries(converter_lite PRIVATE quantizer_mid pthread ${PROTO_LIBRARY} - mindspore-lite ${SECUREC_LIBRARY} mindspore::json mindspore::eigen