@@ -22,7 +22,7 @@ function (inference_analysis_test TARGET)
22
22
if (WITH_TESTING )
23
23
set (options "" )
24
24
set (oneValueArgs "" )
25
- set (multiValueArgs SRCS )
25
+ set (multiValueArgs SRCS EXTRA_DEPS )
26
26
cmake_parse_arguments (analysis_test "${options} " "${oneValueArgs} " "${multiValueArgs} " ${ARGN} )
27
27
28
28
set (mem_opt "" )
@@ -31,22 +31,43 @@ function (inference_analysis_test TARGET)
31
31
endif ()
32
32
cc_test (${TARGET}
33
33
SRCS "${analysis_test_SRCS} "
34
- DEPS analysis graph fc_fuse_pass graph_viz_pass infer_clean_graph_pass graph_pattern_detecter pass
34
+ DEPS analysis graph fc_fuse_pass graph_viz_pass infer_clean_graph_pass graph_pattern_detecter pass ${analysis_test_EXTRA_DEPS}
35
35
ARGS --inference_model_dir=${PYTHON_TESTS_DIR}/book/word2vec.inference.model ${mem_opt} )
36
36
set_tests_properties (${TARGET} PROPERTIES DEPENDS test_word2vec )
37
37
endif (WITH_TESTING )
38
38
endfunction (inference_analysis_test )
39
39
40
- cc_test (test_analyzer SRCS analyzer_tester.cc DEPS paddle_inference_api paddle_fluid_api ir_pass_manager analysis
41
- # ir
42
- fc_fuse_pass
43
- graph_viz_pass
44
- infer_clean_graph_pass
45
- graph_pattern_detecter
46
- pass
47
- ARGS --inference_model_dir=${PYTHON_TESTS_DIR}/book/word2vec.inference.model )
48
- #set_tests_properties(test_analyzer PROPERTIES DEPENDS test_word2vec)
49
- #inference_api_test(test_analyzer SRC analyzer_tester.cc ARGS test_word2vec)
40
+ set (DITU_RNN_MODEL_URL "http://paddle-inference-dist.bj.bcebos.com/ditu_rnn_fluid%2Fmodel.tar.gz" )
41
+ set (DITU_RNN_DATA_URL "http://paddle-inference-dist.bj.bcebos.com/ditu_rnn_fluid%2Fdata.txt.tar.gz" )
42
+ set (DITU_INSTALL_DIR "${THIRD_PARTY_PATH} /install/ditu_rnn" CACHE PATH "Ditu RNN model and data root." FORCE )
43
+ set (DITU_RNN_MODEL ${DITU_INSTALL_DIR} /model )
44
+ set (DITU_RNN_DATA ${DITU_INSTALL_DIR} /data.txt )
45
+
46
+ function (inference_download_and_uncompress target url gz_filename )
47
+ message (STATUS "Download inference test stuff ${gz_filename} from ${url} " )
48
+ execute_process (COMMAND bash -c "mkdir -p ${DITU_INSTALL_DIR} " )
49
+ execute_process (COMMAND bash -c "cd ${DITU_INSTALL_DIR} && wget -q ${url} " )
50
+ execute_process (COMMAND bash -c "cd ${DITU_INSTALL_DIR} && tar xzf ${gz_filename} " )
51
+ message (STATUS "finish downloading ${gz_filename} " )
52
+ endfunction (inference_download_and_uncompress )
53
+
54
+ if (NOT EXISTS ${DITU_INSTALL_DIR} )
55
+ inference_download_and_uncompress (ditu_rnn_model ${DITU_RNN_MODEL_URL} "ditu_rnn_fluid%2Fmodel.tar.gz" )
56
+ inference_download_and_uncompress (ditu_rnn_data ${DITU_RNN_DATA_URL} "ditu_rnn_fluid%2Fdata.txt.tar.gz" )
57
+ endif ()
58
+
59
+ inference_analysis_test (test_analyzer SRCS analyzer_tester.cc
60
+ EXTRA_DEPS paddle_inference_api paddle_fluid_api ir_pass_manager analysis
61
+ # ir
62
+ fc_fuse_pass
63
+ graph_viz_pass
64
+ infer_clean_graph_pass
65
+ graph_pattern_detecter
66
+ infer_clean_graph_pass
67
+ pass
68
+ ARGS --inference_model_dir=${PYTHON_TESTS_DIR}/book/word2vec.inference.model
69
+ --infer_ditu_rnn_model=${DITU_INSTALL_DIR}/model
70
+ --infer_ditu_rnn_data=${DITU_INSTALL_DIR}/data.txt )
50
71
51
72
inference_analysis_test (test_data_flow_graph SRCS data_flow_graph_tester.cc )
52
73
inference_analysis_test (test_data_flow_graph_to_fluid_pass SRCS data_flow_graph_to_fluid_pass_tester.cc )
0 commit comments