zzy34407230
/
mindspore2022
Not watched
1
0
Fork 0
Code
Releases 22 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: bc738d4ec3
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'bc738d4ec3'
${ noResults }
mindspore2022/mindspore/lite/test/CMakeLists.txt

307 lines
10 kB
Raw Blame History 

  1. set(TOP_DIR ${CMAKE_CURRENT_SOURCE_DIR}/../../..)

  2. set(TEST_DIR ${TOP_DIR}/mindspore/lite/test)

  3. set(LITE_DIR ${TOP_DIR}/mindspore/lite)

  4. set(CCSRC_DIR ${TOP_DIR}/mindspore/ccsrc)

  5. include_directories(${TOP_DIR})

  6. include_directories(${TEST_DIR})

  7. include(${CMAKE_CURRENT_SOURCE_DIR}/../../../cmake/dependency_gtest.cmake)

  8. 

  9. STRING(REPLACE " -fvisibility=hidden " " -fvisibility=default " CMAKE_C_FLAGS "${CMAKE_C_FLAGS}")

  10. STRING(REPLACE " -fvisibility=hidden " " -fvisibility=default " CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")

  11. 

  12. if(ENABLE_CONVERTER)

  13. set(CCSRC_SRC

  14.         ## ccsrc

  15.         ${CCSRC_DIR}/backend/optimizer/common/pattern_engine.cc

  16.         ${CCSRC_DIR}/backend/optimizer/common/visit.cc

  17.         ${CCSRC_DIR}/backend/optimizer/common/optimizer.cc

  18.         )

  19. else(ENABLE_CONVERTER)

  20.     set(TEST_LITE_SRC ${LITE_DIR}/src/common/log_adapter.cc)

  21.     add_compile_definitions(USE_ANDROID_LOG)

  22. endif()

  23. 

  24. ### cpu kernel

  25. file(GLOB KERNEL_OP_SRC

  26.         ${LITE_DIR}/src/runtime/kernel/arm/base/*.cc

  27.         ${LITE_DIR}/src/runtime/kernel/arm/fp32/*.cc

  28.         ${LITE_DIR}/src/runtime/kernel/arm/int8/*.cc

  29.         ${LITE_DIR}/src/runtime/kernel/arm/string/*.cc

  30.         ${LITE_DIR}/nnacl/*.c

  31.         ${LITE_DIR}/nnacl/fp32/*.c

  32.         ${LITE_DIR}/nnacl/int8/*.c

  33.         ${LITE_DIR}/nnacl/quantization/*.c

  34.         )

  35. 

  36. file(GLOB KERNEL_OP_TRAIN_SRC

  37.         ${LITE_DIR}/nnacl/fp32_grad/*.c

  38.         ${LITE_DIR}/src/runtime/kernel/arm/fp32_grad/*.cc

  39. )

  40. 

  41. if (SUPPORT_TRAIN)

  42.         list(APPEND KERNEL_OP_SRC ${KERNEL_OP_TRAIN_SRC})

  43. endif()

  44. 

  45. if (PLATFORM_ARM64)

  46.     # assembly

  47.     file(GLOB TEST_ASSEMBLY_SRC ${LITE_DIR}/nnacl/assembly/arm64/*.s

  48.             ${LITE_DIR}/nnacl/assembly/arm64/*.S)

  49. 

  50.     set_property(SOURCE ${TEST_ASSEMBLY_SRC} PROPERTY LANGUAGE C)

  51.     set(KERNEL_OP_SRC

  52.             ${KERNEL_OP_SRC}

  53.             ${TEST_ASSEMBLY_SRC}

  54.             )

  55. endif()

  56. 

  57. if (PLATFORM_ARM32)

  58.     # assembly

  59.     file(GLOB TEST_ASSEMBLY_SRC

  60.             ${LITE_DIR}/nnacl/assembly/arm32/*.S

  61.             ${LITE_DIR}/nnacl/assembly/arm32/*.s)

  62.     set_property(SOURCE ${TEST_ASSEMBLY_SRC} PROPERTY LANGUAGE C)

  63.     set(KERNEL_OP_SRC

  64.             ${KERNEL_OP_SRC}

  65.             ${TEST_ASSEMBLY_SRC}

  66.             )

  67. endif()

  68. 

  69. if (ENABLE_FP16)

  70.     file(GLOB KERNEL_OP_FP16_SRC

  71.             ${LITE_DIR}/src/runtime/kernel/arm/fp16/*.cc

  72.             ${LITE_DIR}/nnacl/fp16/*.c

  73.             )

  74.     set(KERNEL_OP_SRC

  75.             ${KERNEL_OP_SRC}

  76.             ${KERNEL_OP_FP16_SRC}

  77.             )

  78. endif ()

  79. ### gpu kernel

  80. if (SUPPORT_GPU)

  81.     file(GLOB GPU_KERNEL_OP_SRC

  82.             ${LITE_DIR}/src/runtime/kernel/opencl/kernel/*.cc

  83.             )

  84.     set(KERNEL_OP_SRC

  85.             ${KERNEL_OP_SRC}

  86.             ${GPU_KERNEL_OP_SRC}

  87.             ${LITE_DIR}/src/runtime/kernel/opencl/subgraph_opencl_kernel.cc

  88.             ${LITE_DIR}/src/runtime/kernel/opencl/utils.cc

  89.             )

  90. endif()

  91. ### minddata lite

  92. if (BUILD_MINDDATA STREQUAL "lite")

  93.     include_directories(${CCSRC_DIR}/minddata)

  94.     set(DATASET_TEST_DIR ${TEST_DIR}/ut/src/dataset)

  95.     set(TEST_MINDDATA_SRC

  96.                 ${DATASET_TEST_DIR}/de_tensor_test.cc

  97.                 ${DATASET_TEST_DIR}/eager_test.cc

  98.             )

  99. elseif (BUILD_MINDDATA STREQUAL "full")

  100. endif()

  101. ### runtime framework

  102. file(GLOB_RECURSE OPS_SRC ${LITE_DIR}/src/ops/*.cc ${LITE_DIR}/src/ops/populate/*.cc)

  103. set(TEST_LITE_SRC

  104.         ${TEST_LITE_SRC}

  105.         ${CCSRC_SRC}

  106.         ${OPS_SRC}

  107.         ${KERNEL_OP_SRC}

  108.         ${LITE_DIR}/src/runtime/allocator.cc

  109.         ${LITE_DIR}/src/runtime/runtime_api.cc

  110.         ${LITE_DIR}/src/runtime/thread_pool.c

  111.         ${LITE_DIR}/src/runtime/workspace_pool.cc

  112.         ${LITE_DIR}/src/runtime/parallel_executor.cc

  113.         ${LITE_DIR}/src/tensor.cc

  114.         ${LITE_DIR}/src/executor.cc

  115.         ${LITE_DIR}/src/inner_context.cc

  116.         ${LITE_DIR}/src/kernel_registry.cc

  117.         ${LITE_DIR}/src/lite_kernel.cc

  118.         ${LITE_DIR}/src/lite_session.cc

  119.         ${LITE_DIR}/src/sub_graph_kernel.cc

  120.         ${LITE_DIR}/src/model.cc

  121.         ${LITE_DIR}/src/model_common.cc

  122.         ${LITE_DIR}/src/scheduler.cc

  123.         ${LITE_DIR}/src/common/graph_util.cc

  124.         ${LITE_DIR}/src/common/file_utils.cc

  125.         ${LITE_DIR}/src/common/file_utils_ext.cc

  126.         ${LITE_DIR}/src/common/utils.cc

  127.         ${LITE_DIR}/src/common/string_util.cc

  128.         ${LITE_DIR}/tools/common/graph_util.cc

  129.         ${LITE_DIR}/tools/common/tensor_util.cc

  130.         ${LITE_DIR}/tools/common/node_util.cc

  131.         ${LITE_DIR}/tools/common/flag_parser.cc

  132.         ${LITE_DIR}/tools/common/storage.cc

  133.         ${LITE_DIR}/tools/benchmark/benchmark.cc

  134.         ${LITE_DIR}/test/st/benchmark_test.cc

  135.         ${LITE_DIR}/src/errorcode.cc

  136.         )

  137. ### gpu runtime

  138. if (SUPPORT_GPU)

  139.     include_directories(${TOP_DIR}/third_party/OpenCL-Headers)

  140.     include_directories(${TOP_DIR}/third_party/OpenCL-CLHPP/include)

  141.     set(OPENCL_RUNTIME_SRC

  142.             ${LITE_DIR}/src/runtime/opencl/opencl_allocator.cc

  143.             ${LITE_DIR}/src/runtime/opencl/opencl_executor.cc

  144.             ${LITE_DIR}/src/runtime/opencl/opencl_runtime.cc

  145.             ${LITE_DIR}/src/runtime/opencl/opencl_wrapper.cc

  146.             )

  147.     set(TEST_LITE_SRC

  148.             ${TEST_LITE_SRC}

  149.             ${OPENCL_RUNTIME_SRC}

  150.             )

  151. endif()

  152. ### converter

  153. if(ENABLE_CONVERTER)

  154.     add_definitions(-DPRIMITIVE_WRITEABLE)

  155.     add_definitions(-DUSE_GLOG)

  156.     file(GLOB_RECURSE TEST_CASE_TFLITE_PARSERS_SRC

  157.             ${TEST_DIR}/ut/tools/converter/parser/tflite/*.cc

  158.             )

  159.     set(TEST_LITE_SRC

  160.             ${TEST_LITE_SRC}

  161.             ${TEST_CASE_TFLITE_PARSERS_SRC}

  162.             ${TOP_DIR}/mindspore/core/utils/flags.cc

  163.             ${LITE_DIR}/tools/common/protobuf_utils.cc

  164.             ${LITE_DIR}/tools/converter/optimizer.cc

  165.             ${LITE_DIR}/tools/converter/anf_transform.cc

  166.             ${LITE_DIR}/tools/converter/graphdef_transform.cc

  167.             ${LITE_DIR}/tools/converter/converter_flags.cc

  168.             ${LITE_DIR}/tools/converter/converter.cc

  169.             ${LITE_DIR}/tools/optimizer/common/node_pass_extends.cc

  170.             ${LITE_DIR}/tools/optimizer/common/pass_manager_extends.cc

  171.             ${LITE_DIR}/tools/optimizer/common/gllo_utils.cc

  172.             ${LITE_DIR}/tools/optimizer/fusion/conv_biasadd_fusion.cc

  173.             ${LITE_DIR}/tools/optimizer/fusion/conv_activation_fusion.cc

  174.             ${LITE_DIR}/tools/optimizer/fusion/conv_tuple_activation_fusion.cc

  175.             ${LITE_DIR}/tools/optimizer/fusion/conv_transform_fusion.cc

  176.             ${LITE_DIR}/tools/optimizer/fusion/conv_scale_fusion.cc

  177.             ${LITE_DIR}/tools/optimizer/fusion/conv_bn_fusion.cc

  178.             ${LITE_DIR}/tools/optimizer/fusion/constant_folding_fusion.cc

  179.             ${LITE_DIR}/tools/optimizer/fusion/quant_dtype_cast_fusion.cc

  180.             ${LITE_DIR}/tools/optimizer/fusion/layer_norm_fusion.cc

  181.             ${LITE_DIR}/tools/optimizer/fusion/batchmatmul_fusion.cc

  182.             ${LITE_DIR}/tools/optimizer/graph/weight_format_transform_pass.cc

  183.             ${LITE_DIR}/tools/optimizer/graph/weight_format_hardcode_pass.cc

  184.             ${LITE_DIR}/tools/optimizer/graph/clip_convert_activation_pass.cc

  185.             ${LITE_DIR}/tools/optimizer/graph/unused_cast_node_remove_pass.cc

  186.             ${LITE_DIR}/tools/optimizer/graph/identity_remove_pass.cc

  187.             )

  188. endif()

  189. ### train

  190. if (SUPPORT_TRAIN)

  191.     set(TEST_LITE_SRC

  192.             ${TEST_LITE_SRC}

  193.            # ${LITE_DIR}/src/train/ops/train_ops.cc

  194.             ${LITE_DIR}/src/train/train_populate_parameter.cc

  195.             ${LITE_DIR}/src/train/train_session.cc

  196.             ${LITE_DIR}/src/train/train_model.cc

  197.             ${LITE_DIR}/src/lite_session.cc

  198.             )

  199. else()

  200.     set(TEST_LITE_SRC

  201.             ${TEST_LITE_SRC}

  202.             ${LITE_DIR}/src/lite_session.cc

  203.             )

  204. endif()

  205. ### test src

  206. file(GLOB_RECURSE TEST_CASE_KERNEL_SRC

  207.     ${TEST_DIR}/ut/src/runtime/kernel/arm/common/*.cc

  208.     ${TEST_DIR}/ut/src/runtime/kernel/arm/fp32/*.cc

  209.     ${TEST_DIR}/ut/src/runtime/kernel/arm/int8/*.cc

  210.     ${TEST_DIR}/ut/src/runtime/kernel/arm/string/*.cc

  211. )

  212. 

  213. file(GLOB_RECURSE TEST_CASE_KERNEL_TRAIN_SRC

  214.     ${TEST_DIR}/ut/src/runtime/kernel/arm/fp32_grad/*.cc

  215. )

  216. 

  217. set(TEST_SRC

  218.         ${TEST_LITE_SRC}

  219.         ${TEST_MINDDATA_SRC}

  220.         ${TEST_CASE_KERNEL_SRC}

  221.         ${TEST_DIR}/main.cc

  222.         ${TEST_DIR}/common/common_test.cc

  223.         ${TEST_DIR}/ut/src/infer_test.cc

  224.         ${TEST_DIR}/ut/src/utils_test.cc

  225.         ${TEST_DIR}/ut/src/scheduler_test.cc

  226. )

  227. 

  228. if (ENABLE_CONVERTER)

  229.     set(TEST_SRC

  230.             ${TEST_SRC}

  231.             ${TEST_DIR}/st/converter_test.cc

  232.             ${TEST_DIR}/ut/tools/optimizer/fusion/conv_biasadd_fusion_test.cc

  233.             ${TEST_DIR}/ut/tools/optimizer/fusion/conv_bn_fusion_test.cc

  234.             ${TEST_DIR}/ut/tools/optimizer/fusion/conv_scale_fusion_test.cc

  235.             ${TEST_DIR}/ut/tools/optimizer/fusion/conv_activation_fusion_test.cc

  236.             ${TEST_DIR}/ut/tools/optimizer/fusion/constant_folding_fusion_test.cc

  237.             )

  238. endif()

  239. 

  240. if (SUPPORT_TRAIN)

  241.     set(TEST_SRC

  242.             ${TEST_SRC}

  243.             ${TEST_CASE_KERNEL_TRAIN_SRC}

  244.             ${TEST_DIR}/ut/src/infer_test.cc  # temporary

  245.             )

  246. else()

  247.     set(TEST_SRC

  248.             ${TEST_SRC}

  249.             ${TEST_DIR}/ut/src/infer_test.cc

  250.             )

  251. endif()

  252. 

  253. if (SUPPORT_GPU)

  254.     file(GLOB_RECURSE TEST_CASE_KERNEL_GPU_SRC

  255.             ${TEST_DIR}/ut/src/runtime/kernel/opencl/*.cc

  256.             )

  257.     set(TEST_SRC

  258.             ${TEST_SRC}

  259.             ${TEST_CASE_KERNEL_GPU_SRC}

  260.             )

  261. endif()

  262. 

  263. if (ENABLE_FP16)

  264.     set(TEST_SRC

  265.             ${TEST_SRC}

  266.             ${TEST_DIR}/ut/src/runtime/kernel/arm/fp16/convolution_fp16_tests.cc)

  267. endif ()

  268. 

  269. 

  270. add_executable(lite-test ${TEST_SRC})

  271. 

  272. target_link_libraries(lite-test dl ${GTEST_LIBRARY})

  273. if (PLATFORM_ARM64)

  274.     target_link_libraries(lite-test mslite_internal)

  275. endif()

  276. 

  277. if (PLATFORM_ARM)

  278.     target_link_libraries(lite-test log)

  279. endif()

  280. 

  281. if (BUILD_MINDDATA STREQUAL "lite")

  282.     target_link_libraries(lite-test

  283.     	   minddata_eager_mid

  284.            minddata-lite

  285.            )

  286. endif()

  287. if (ENABLE_CONVERTER)

  288.     target_link_libraries(lite-test

  289.             anf_importer_mid

  290.             anf_exporter_mid

  291.             tflite_parser_mid

  292.             caffe_parser_mid

  293.             onnx_parser_mid

  294.             graph_pass_mid

  295.             fusion_mid

  296.             quantizer_mid

  297.             proto_mid

  298.             pthread

  299.             mindspore::protobuf

  300.             mindspore::eigen

  301.             mindspore::json

  302.             mindspore_core

  303.             mindspore::glog

  304.             ${SECUREC_LIBRARY}

  305.             )

  306. endif()