Huawei_Technology
/
mindspore

 
			
							set(TOP_DIR ${CMAKE_CURRENT_SOURCE_DIR}/../../..)
set(TEST_DIR ${TOP_DIR}/mindspore/lite/test)
set(LITE_DIR ${TOP_DIR}/mindspore/lite)
set(CCSRC_DIR ${TOP_DIR}/mindspore/ccsrc)
include_directories(${TOP_DIR})
include_directories(${TEST_DIR})
include(${CMAKE_CURRENT_SOURCE_DIR}/../../../cmake/dependency_gtest.cmake)

string(REPLACE " -Werror " " " CMAKE_C_FLAGS "${CMAKE_C_FLAGS}")
string(REPLACE " -Werror " " " CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
STRING(REPLACE " -fvisibility=hidden " " -fvisibility=default " CMAKE_C_FLAGS "${CMAKE_C_FLAGS}")
STRING(REPLACE " -fvisibility=hidden " " -fvisibility=default " CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")


if(BUILD_CONVERTER)
file (GLOB_RECURSE CORE_SRC
        ${CMAKE_CURRENT_SOURCE_DIR}/../../core/*.cc
        )
set(CCSRC_SRC
        ## ccsrc
        ${CCSRC_DIR}/backend/optimizer/common/pattern_engine.cc
        ${CCSRC_DIR}/backend/optimizer/common/visit.cc
        ${CCSRC_DIR}/backend/optimizer/common/optimizer.cc
        )
else(BUILD_CONVERTER)
    set(CORE_SRC
            ${CMAKE_CURRENT_SOURCE_DIR}/../../core/abstract/utils.cc
            ${CMAKE_CURRENT_SOURCE_DIR}/../../core/abstract/dshape.cc
            ${CMAKE_CURRENT_SOURCE_DIR}/../../core/abstract/abstract_value.cc
            ${CMAKE_CURRENT_SOURCE_DIR}/../../core/abstract/abstract_function.cc
            ${CMAKE_CURRENT_SOURCE_DIR}/../../core/ir/primitive.cc
            ${CMAKE_CURRENT_SOURCE_DIR}/../../core/ir/named.cc
            ${CMAKE_CURRENT_SOURCE_DIR}/../../core/ir/value.cc
            ${CMAKE_CURRENT_SOURCE_DIR}/../../core/ir/value_extends.cc
            ${CMAKE_CURRENT_SOURCE_DIR}/../../core/ir/dtype.cc
            ${CMAKE_CURRENT_SOURCE_DIR}/../../core/ir/dtype_extends.cc
            ${CMAKE_CURRENT_SOURCE_DIR}/../../core/ir/dtype/container.cc
            ${CMAKE_CURRENT_SOURCE_DIR}/../../core/ir/dtype/type.cc
            ${CMAKE_CURRENT_SOURCE_DIR}/../../core/ir/dtype/empty.cc
            ${CMAKE_CURRENT_SOURCE_DIR}/../../core/ir/dtype/tensor_type.cc
            ${CMAKE_CURRENT_SOURCE_DIR}/../../core/ir/dtype/type_extends.cc
            ${CMAKE_CURRENT_SOURCE_DIR}/../../core/ir/dtype/number.cc
            ${CMAKE_CURRENT_SOURCE_DIR}/../../core/gvar/logging_level.cc
            ${CMAKE_CURRENT_SOURCE_DIR}/../../core/gvar/typeid_manager.cc
            ${CMAKE_CURRENT_SOURCE_DIR}/../../core/base/base.cc
            ${CMAKE_CURRENT_SOURCE_DIR}/../../core/base/base_ref.cc
            ${CMAKE_CURRENT_SOURCE_DIR}/../../core/utils/log_adapter.cc
            )
endif()

### cpu kernel
file(GLOB KERNEL_OP_SRC
        ${LITE_DIR}/src/runtime/kernel/arm/base/*.cc
        ${LITE_DIR}/src/runtime/kernel/arm/fp32/*.cc
        ${LITE_DIR}/src/runtime/kernel/arm/int8/*.cc
        ${LITE_DIR}/nnacl/*.c
        ${LITE_DIR}/nnacl/fp32/*.c
        ${LITE_DIR}/nnacl/int8/*.c
        ${LITE_DIR}/nnacl/quantization/*.c
        )

file(GLOB KERNEL_OP_TRAIN_SRC
        ${LITE_DIR}/nnacl/fp32_grad/*.c
        ${LITE_DIR}/src/runtime/kernel/arm/fp32_grad/*.cc
)

if (SUPPORT_TRAIN)
        list(APPEND KERNEL_OP_SRC ${KERNEL_OP_TRAIN_SRC})
endif()

if (PLATFORM_ARM64)
    # assembly
    file(GLOB TEST_ASSEMBLY_SRC ${LITE_DIR}/nnacl/assembly/arm64/*.s
            ${LITE_DIR}/nnacl/assembly/arm64/*.S)

    set_property(SOURCE ${TEST_ASSEMBLY_SRC} PROPERTY LANGUAGE C)
    set(KERNEL_OP_SRC
            ${KERNEL_OP_SRC}
            ${TEST_ASSEMBLY_SRC}
            )
endif()

if (PLATFORM_ARM32)
    # assembly
    file(GLOB TEST_ASSEMBLY_SRC
            ${LITE_DIR}/nnacl/assembly/arm32/*.S
            ${LITE_DIR}/nnacl/assembly/arm32/*.s)
    set_property(SOURCE ${TEST_ASSEMBLY_SRC} PROPERTY LANGUAGE C)
    set(KERNEL_OP_SRC
            ${KERNEL_OP_SRC}
            ${TEST_ASSEMBLY_SRC}
            )
endif()

if (ENABLE_FP16)
    file(GLOB KERNEL_OP_FP16_SRC
            ${LITE_DIR}/src/runtime/kernel/arm/fp16/*.cc
            ${LITE_DIR}/nnacl/fp16/*.c
            )
    set(KERNEL_OP_SRC
            ${KERNEL_OP_SRC}
            ${KERNEL_OP_FP16_SRC}
            )
endif ()
### gpu kernel
if (SUPPORT_GPU)
    file(GLOB GPU_KERNEL_OP_SRC
            ${LITE_DIR}/src/runtime/kernel/opencl/kernel/*.cc
            )
    set(KERNEL_OP_SRC
            ${KERNEL_OP_SRC}
            ${GPU_KERNEL_OP_SRC}
            ${LITE_DIR}/src/runtime/kernel/opencl/subgraph_opencl_kernel.cc
            ${LITE_DIR}/src/runtime/kernel/opencl/utils.cc
            )
endif()
### minddata lite
if (BUILD_MINDDATA STREQUAL "lite")
    include_directories(${CCSRC_DIR}/minddata)
    set(DATASET_TEST_DIR ${TEST_DIR}/ut/src/dataset)
    set(TEST_MINDDATA_SRC
                ${DATASET_TEST_DIR}/de_tensor_test.cc
                ${DATASET_TEST_DIR}/eager_test.cc
            )
elseif (BUILD_MINDDATA STREQUAL "full")
endif()
### runtime framework
file(GLOB_RECURSE OPS_SRC ${LITE_DIR}/src/ops/*.cc)
set(TEST_LITE_SRC
        ${CORE_SRC}
        ${CCSRC_SRC}
        ${OPS_SRC}
        ${KERNEL_OP_SRC}
        ${LITE_DIR}/src/runtime/allocator.cc
        ${LITE_DIR}/src/runtime/runtime_api.cc
        ${LITE_DIR}/src/runtime/thread_pool.c
        ${LITE_DIR}/src/runtime/workspace_pool.cc
        ${LITE_DIR}/src/runtime/parallel_executor.cc
        ${LITE_DIR}/src/tensor.cc
        ${LITE_DIR}/src/executor.cc
        ${LITE_DIR}/src/kernel_registry.cc
        ${LITE_DIR}/src/lite_kernel.cc
        ${LITE_DIR}/src/lite_session.cc
        ${LITE_DIR}/src/model.cc
        ${LITE_DIR}/src/populate_parameter.cc
        ${LITE_DIR}/src/scheduler.cc
        ${LITE_DIR}/src/common/graph_util.cc
        ${LITE_DIR}/src/common/file_utils.cc
        ${LITE_DIR}/src/common/file_utils_ext.cc
        ${LITE_DIR}/src/common/utils.cc
        ${LITE_DIR}/tools/common/graph_util.cc
        ${LITE_DIR}/tools/common/tensor_util.cc
        ${LITE_DIR}/tools/common/node_util.cc
        ${LITE_DIR}/tools/common/flag_parser.cc
        ${LITE_DIR}/tools/common/storage.cc
        ${LITE_DIR}/tools/benchmark/benchmark.cc
        ${LITE_DIR}/test/st/benchmark_test.cc
        )
### gpu runtime
if (SUPPORT_GPU)
    include_directories(${TOP_DIR}/third_party/OpenCL-Headers)
    include_directories(${TOP_DIR}/third_party/OpenCL-CLHPP/include)
    set(OPENCL_RUNTIME_SRC
            ${LITE_DIR}/src/runtime/opencl/opencl_allocator.cc
            ${LITE_DIR}/src/runtime/opencl/opencl_executor.cc
            ${LITE_DIR}/src/runtime/opencl/opencl_runtime.cc
            ${LITE_DIR}/src/runtime/opencl/opencl_wrapper.cc
            )
    set(TEST_LITE_SRC
            ${TEST_LITE_SRC}
            ${OPENCL_RUNTIME_SRC}
            )
endif()
### converter
if(BUILD_CONVERTER)
    add_definitions(-DPRIMITIVE_WRITEABLE)
    file(GLOB_RECURSE TEST_CASE_TFLITE_PARSERS_SRC
            ${TEST_DIR}/ut/tools/converter/parser/tflite/*.cc
            )
    set(TEST_LITE_SRC
            ${TEST_LITE_SRC}
            ${TEST_CASE_TFLITE_PARSERS_SRC}
            ${TOP_DIR}/mindspore/core/utils/flags.cc
            ${LITE_DIR}/tools/common/protobuf_utils.cc
            ${LITE_DIR}/tools/converter/optimizer.cc
            ${LITE_DIR}/tools/converter/anf_transform.cc
            ${LITE_DIR}/tools/converter/graphdef_transform.cc
            ${LITE_DIR}/tools/converter/converter_flags.cc
            ${LITE_DIR}/tools/converter/converter.cc
            ${LITE_DIR}/test/st/converter_test.cc
            ${LITE_DIR}/test/ut/tools/optimizer/fusion/conv_activation_fusion_test.cc
            ${LITE_DIR}/test/ut/tools/optimizer/fusion/conv_biasadd_fusion_test.cc
            ${LITE_DIR}/test/ut/tools/optimizer/fusion/conv_bn_fusion_test.cc
            ${LITE_DIR}/test/ut/tools/optimizer/fusion/conv_scale_fusion_test.cc
            ${LITE_DIR}/test/ut/tools/optimizer/fusion/constant_folding_fusion_test.cc
            ${LITE_DIR}/tools/optimizer/common/node_pass_extends.cc
            ${LITE_DIR}/tools/optimizer/common/pass_manager_extends.cc
            ${LITE_DIR}/tools/optimizer/common/gllo_utils.cc
            ${LITE_DIR}/tools/optimizer/fusion/conv_biasadd_fusion.cc
            ${LITE_DIR}/tools/optimizer/fusion/conv_activation_fusion.cc
             ${LITE_DIR}/tools/optimizer/fusion/conv_tuple_activation_fusion.cc
            ${LITE_DIR}/tools/optimizer/fusion/conv_transform_fusion.cc
            ${LITE_DIR}/tools/optimizer/fusion/conv_scale_fusion.cc
            ${LITE_DIR}/tools/optimizer/fusion/conv_bn_fusion.cc
            ${LITE_DIR}/tools/optimizer/fusion/constant_folding_fusion.cc
            ${LITE_DIR}/tools/optimizer/fusion/quant_dtype_cast_fusion.cc
            )
endif()
### train
if (SUPPORT_TRAIN)
    set(TEST_LITE_SRC
            ${TEST_LITE_SRC}
           # ${LITE_DIR}/src/train/ops/train_ops.cc
            ${LITE_DIR}/src/train/train_populate_parameter.cc
            ${LITE_DIR}/src/train/train_session.cc
            ${LITE_DIR}/src/lite_session.cc
            #            ${SRC_DIR}/common/trans.cc
            #            ${SRC_DIR}/common/lite/trans_extends.cc
            #            ${SRC_DIR}/kernel/kernel_build_info.cc
            #            ${SRC_DIR}/utils/lite/base_ref_utils.cc
            #            ${SRC_DIR}/session/lite/anf_runtime_algorithm_extends.cc
            #            ${SRC_DIR}/session/lite/session_basic_extends.cc
            #            ${SRC_DIR}/session/anf_runtime_algorithm.cc
            #            ${SRC_DIR}/session/anf_runtime_algorithm.cc
            #            ${SRC_DIR}/session/session_basic.cc
            #            ${SRC_DIR}/session/kernel_graph.cc
            #            ${SRC_DIR}/session/session_factory.cc
            #            ${SRC_DIR}/device/kernel_info.cc
            #            ${SRC_DIR}/device/kernel_runtime.cc
            #            ${SRC_DIR}/device/lite/kernel_runtime_extends.cc
    #        ${LITE_DIR}/src/common/anf_importer/anf_importer.cc
    #       ${LITE_DIR}/src/common/anf_importer/import_from_meta_graph.cc
    #       ${LITE_DIR}/src/ir/primitive_value.cc
    #       ${LITE_DIR}/src/train/lite_kernel_runtime.cc
    #       ${LITE_DIR}/src/train/train_session.cc
    #       ${LITE_DIR}/src/train/model.cc
            ${LITE_DIR}/src/lite_session.cc  # temporary
            )
else()
    set(TEST_LITE_SRC
            ${TEST_LITE_SRC}
            ${LITE_DIR}/src/lite_session.cc
            )
endif()
### test src
file(GLOB_RECURSE TEST_CASE_KERNEL_SRC
    ${TEST_DIR}/ut/src/runtime/kernel/arm/common/*.cc
    ${TEST_DIR}/ut/src/runtime/kernel/arm/fp32/*.cc
    ${TEST_DIR}/ut/src/runtime/kernel/arm/int8/*.cc
)

file(GLOB_RECURSE TEST_CASE_KERNEL_TRAIN_SRC
    ${TEST_DIR}/ut/src/runtime/kernel/arm/fp32_grad/*.cc
)

set(TEST_SRC
    ${TEST_LITE_SRC}
    ${TEST_MINDDATA_SRC}
    ${TEST_CASE_KERNEL_SRC}
    ${TEST_DIR}/common/common_test.cc
    ${TEST_DIR}/main.cc
    ${TEST_DIR}/ut/src/runtime/kernel/arm/common/pack_tests.cc
    ${TEST_DIR}/ut/src/infer_test.cc
    ${TEST_DIR}/ut/src/utils_test.cc
    #${TEST_DIR}/ut/internal/infer_test.cc
)

if (SUPPORT_TRAIN)
    set(TEST_SRC
            ${TEST_SRC}
            ${TEST_CASE_KERNEL_TRAIN_SRC}
            ${TEST_DIR}/ut/src/infer_test.cc  # temporary
            )
else()
    set(TEST_SRC
            ${TEST_SRC}
            ${TEST_DIR}/ut/src/infer_test.cc
            )
endif()

if (SUPPORT_GPU)
    file(GLOB_RECURSE TEST_CASE_KERNEL_GPU_SRC
            ${TEST_DIR}/ut/src/runtime/kernel/opencl/*.cc
            )
    set(TEST_SRC
            ${TEST_SRC}
            ${TEST_CASE_KERNEL_GPU_SRC}
            )
endif()

if (ENABLE_FP16)
    set(TEST_SRC
            ${TEST_SRC}
            ${TEST_DIR}/ut/src/runtime/kernel/arm/fp16/convolution_fp16_tests.cc)
endif ()


add_executable(lite-test ${TEST_SRC})

target_link_libraries(lite-test dl ${GTEST_LIBRARY})
if (PLATFORM_ARM64)
    target_link_libraries(lite-test mslite_internal)
endif()

if (BUILD_MINDDATA STREQUAL "lite")
    target_link_libraries(lite-test
	   minddata_eager_mid
           minddata-lite
           )
    if (PLATFORM_ARM32 OR PLATFORM_ARM64)
           target_link_libraries(lite-test log)
    endif()
elseif(BUILD_MINDDATA STREQUAL "full")
    if (PLATFORM_ARM32 OR PLATFORM_ARM64)
           target_link_libraries(lite-test log)
    endif()     
endif()
if (BUILD_CONVERTER)
    target_link_libraries(lite-test
            anf_importer_mid
            anf_exporter_mid
            tflite_parser_mid
            caffe_parser_mid
            onnx_parser_mid
            graph_pass_mid
            fusion_mid
            quantizer_mid
            pthread
            protobuf
            mindspore::eigen
            mindspore::json
            ${SECUREC_LIBRARY}
            )
endif()