You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
2.8 KiB
61 lines
2.8 KiB
cc_library(ir_pass_manager SRCS ir_pass_manager.cc DEPS graph pass)
|
|
cc_library(analysis SRCS pass_manager.cc dot.cc node.cc data_flow_graph.cc graph_traits.cc subgraph_splitter.cc
|
|
analyzer.cc
|
|
helper.cc
|
|
# passes
|
|
fluid_to_data_flow_graph_pass.cc
|
|
data_flow_graph_to_fluid_pass.cc
|
|
dfg_graphviz_draw_pass.cc
|
|
tensorrt_subgraph_pass.cc
|
|
tensorrt_subgraph_node_mark_pass.cc
|
|
fluid_to_ir_pass.cc
|
|
model_store_pass.cc
|
|
DEPS framework_proto proto_desc ir_pass_manager graph pass)
|
|
|
|
cc_test(test_node SRCS node_tester.cc DEPS analysis)
|
|
cc_test(test_dot SRCS dot_tester.cc DEPS analysis)
|
|
cc_binary(inference_analyzer SRCS analyzer_main.cc DEPS analysis)
|
|
|
|
set(PYTHON_TESTS_DIR ${PADDLE_BINARY_DIR}/python/paddle/fluid/tests)
|
|
|
|
function (inference_analysis_test TARGET)
|
|
if(WITH_TESTING)
|
|
set(options "")
|
|
set(oneValueArgs "")
|
|
set(multiValueArgs SRCS)
|
|
cmake_parse_arguments(analysis_test "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
|
|
|
|
set(mem_opt "")
|
|
if(WITH_GPU)
|
|
set(mem_opt "--fraction_of_gpu_memory_to_use=0.5")
|
|
endif()
|
|
cc_test(${TARGET}
|
|
SRCS "${analysis_test_SRCS}"
|
|
DEPS analysis graph fc_fuse_pass graph_viz_pass infer_clean_graph_pass graph_pattern_detecter pass
|
|
ARGS --inference_model_dir=${PYTHON_TESTS_DIR}/book/word2vec.inference.model ${mem_opt})
|
|
set_tests_properties(${TARGET} PROPERTIES DEPENDS test_word2vec)
|
|
endif(WITH_TESTING)
|
|
endfunction(inference_analysis_test)
|
|
|
|
cc_test(test_analyzer SRCS analyzer_tester.cc DEPS paddle_inference_api paddle_fluid_api ir_pass_manager analysis
|
|
# ir
|
|
fc_fuse_pass
|
|
graph_viz_pass
|
|
infer_clean_graph_pass
|
|
graph_pattern_detecter
|
|
pass
|
|
ARGS --inference_model_dir=${PYTHON_TESTS_DIR}/book/word2vec.inference.model)
|
|
#set_tests_properties(test_analyzer PROPERTIES DEPENDS test_word2vec)
|
|
#inference_api_test(test_analyzer SRC analyzer_tester.cc ARGS test_word2vec)
|
|
|
|
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_ir_pass SRCS fluid_to_ir_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)
|
|
inference_analysis_test(test_tensorrt_subgraph_pass SRCS tensorrt_subgraph_pass_tester.cc)
|
|
inference_analysis_test(test_pass_manager SRCS pass_manager_tester.cc)
|
|
inference_analysis_test(test_tensorrt_subgraph_node_mark_pass SRCS tensorrt_subgraph_node_mark_pass_tester.cc)
|
|
inference_analysis_test(test_model_store_pass SRCS model_store_pass_tester.cc)
|