CMakeLists.txt 1.9 KB
Newer Older
Y
Yan Chunwei 已提交
1 2 3 4
cc_library(analysis SRCS pass_manager.cc dot.cc node.cc data_flow_graph.cc graph_traits.cc subgraph_splitter.cc
  fluid_to_data_flow_graph_pass.cc
  data_flow_graph_to_fluid_pass.cc
  dfg_graphviz_draw_pass.cc
5 6 7 8 9
  tensorrt_subgraph_pass.cc
  tensorrt_subgraph_node_mark_pass.cc
  analyzer.cc
  helper.cc
  DEPS framework_proto proto_desc)
10
cc_test(test_node SRCS node_tester.cc DEPS analysis)
11 12 13 14
cc_test(test_dot SRCS dot_tester.cc DEPS analysis)

set(PYTHON_TESTS_DIR ${PADDLE_BINARY_DIR}/python/paddle/fluid/tests)

Y
Yan Chunwei 已提交
15
function (inference_analysis_test TARGET)
S
update  
superjomn 已提交
16 17 18 19 20
    if(WITH_TESTING)
        set(options "")
        set(oneValueArgs "")
        set(multiValueArgs SRCS)
        cmake_parse_arguments(analysis_test "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
21

S
update  
superjomn 已提交
22 23 24 25 26 27
        cc_test(${TARGET}
                SRCS "${analysis_test_SRCS}"
                DEPS analysis
                ARGS --inference_model_dir=${PYTHON_TESTS_DIR}/book/word2vec.inference.model --fraction_of_gpu_memory_to_use=0.5)
        set_tests_properties(${TARGET} PROPERTIES DEPENDS test_word2vec)
    endif(WITH_TESTING)
Y
Yan Chunwei 已提交
28
endfunction(inference_analysis_test)
Y
Yan Chunwei 已提交
29

Y
Yan Chunwei 已提交
30 31 32 33 34
inference_analysis_test(test_data_flow_graph SRCS data_flow_graph_tester.cc)
inference_analysis_test(test_data_flow_graph_to_fluid_pass SRCS data_flow_graph_to_fluid_pass_tester.cc)
inference_analysis_test(test_fluid_to_data_flow_graph_pass SRCS fluid_to_data_flow_graph_pass_tester.cc)
inference_analysis_test(test_subgraph_splitter SRCS subgraph_splitter_tester.cc)
inference_analysis_test(test_dfg_graphviz_draw_pass SRCS dfg_graphviz_draw_pass_tester.cc)
35
inference_analysis_test(test_tensorrt_subgraph_pass SRCS tensorrt_subgraph_pass_tester.cc)
Y
Yan Chunwei 已提交
36
inference_analysis_test(test_pass_manager SRCS pass_manager_tester.cc)
37 38
inference_analysis_test(test_tensorrt_subgraph_node_mark_pass SRCS tensorrt_subgraph_node_mark_pass_tester.cc)
inference_analysis_test(test_analyzer SRCS analyzer_tester.cc)