CMakeLists.txt 2.5 KB
Newer Older
1 2
unset(analysis_deps CACHE)
set(analysis_deps # analysis_deps can be extended accross the project
3
        framework_proto proto_desc graph pass paddle_inference_io executor pretty_log
4 5
        ir_pass_manager
        CACHE INTERNAL "")
6

7 8 9
add_subdirectory(ir_passes)
add_subdirectory(passes)

10
cc_library(analysis_helper SRCS helper.cc DEPS framework_proto proto_desc graph paddle_inference_io)
N
nhzlx 已提交
11 12

cc_library(ir_pass_manager SRCS ir_pass_manager.cc DEPS graph pass ${INFER_IR_PASSES} analysis_helper)
13

14 15
cc_library(argument INTERFACE SRCS argument.cc DEPS scope proto_desc)
cc_library(analysis_pass INTERFACE SRCS analysis_pass.cc DEPS proto_desc)
16

17
cc_library(analysis SRCS analyzer.cc
N
nhzlx 已提交
18
  DEPS ${analysis_deps} analysis_helper
19
  analysis_pass ${INFER_IR_PASSES}
20
  )
21

22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45
function(inference_analysis_test_build TARGET)
  if(WITH_TESTING)
     set(options "")
     set(oneValueArgs "")
     set(multiValueArgs SRCS EXTRA_DEPS)
     cmake_parse_arguments(analysis_test "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
     inference_base_test_build(${TARGET}
             SRCS ${analysis_test_SRCS}
             DEPS analysis pass ${GLOB_PASS_LIB} ${analysis_test_EXTRA_DEPS})
  endif()
endfunction()

function(inference_analysis_test_run TARGET)
  if(WITH_TESTING)
     set(options "")
     set(oneValueArgs "")
     set(multiValueArgs COMMAND ARGS)
     cmake_parse_arguments(analysis_test "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
     inference_base_test_run(${TARGET}
	     COMMAND ${analysis_test_COMMAND}
             ARGS ${analysis_test_ARGS})
  endif()
endfunction()

46 47 48 49 50 51
function(inference_analysis_test TARGET)
  if(WITH_TESTING)
     set(options "")
     set(oneValueArgs "")
     set(multiValueArgs SRCS ARGS EXTRA_DEPS)
     cmake_parse_arguments(analysis_test "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
52
     inference_base_test_build(${TARGET}
53
             SRCS ${analysis_test_SRCS}
54 55 56 57
             DEPS analysis pass ${GLOB_PASS_LIB} ${analysis_test_EXTRA_DEPS})
     inference_base_test_run(${TARGET}
	     COMMAND ${TARGET}
             ARGS ${analysis_test_ARGS})
58
  endif()
Y
Yan Chunwei 已提交
59
endfunction(inference_analysis_test)
Y
Yan Chunwei 已提交
60

61

62 63 64
if (NOT APPLE AND NOT WIN32)
  inference_analysis_test(test_analyzer
    SRCS analyzer_tester.cc
65
    EXTRA_DEPS reset_tensor_array paddle_inference_shared
66
    ARGS --inference_model_dir=${WORD2VEC_MODEL_DIR})
67 68 69 70 71
elseif(WIN32)
    inference_analysis_test(test_analyzer
      SRCS analyzer_tester.cc
      EXTRA_DEPS reset_tensor_array paddle_inference_api
      ARGS --inference_model_dir=${WORD2VEC_MODEL_DIR})
72
endif()