@@ -25,17 +25,16 @@ function (inference_analysis_test TARGET)
25
25
if (WITH_TESTING )
26
26
set (options "" )
27
27
set (oneValueArgs "" )
28
- set (multiValueArgs SRCS EXTRA_DEPS )
28
+ set (multiValueArgs SRCS ARGS EXTRA_DEPS )
29
29
cmake_parse_arguments (analysis_test "${options} " "${oneValueArgs} " "${multiValueArgs} " ${ARGN} )
30
-
31
30
set (mem_opt "" )
32
31
if (WITH_GPU )
33
32
set (mem_opt "--fraction_of_gpu_memory_to_use=0.5" )
34
33
endif ()
35
34
cc_test (${TARGET}
36
35
SRCS "${analysis_test_SRCS} "
37
36
DEPS analysis graph fc_fuse_pass graph_viz_pass infer_clean_graph_pass graph_pattern_detector pass ${analysis_test_EXTRA_DEPS}
38
- ARGS --inference_model_dir=${PYTHON_TESTS_DIR}/book/word2vec.inference.model ${mem_opt} )
37
+ ARGS --inference_model_dir=${PYTHON_TESTS_DIR}/book/word2vec.inference.model ${mem_opt} ${analysis_test_ARGS} )
39
38
set_tests_properties (${TARGET} PROPERTIES DEPENDS test_word2vec )
40
39
endif (WITH_TESTING )
41
40
endfunction (inference_analysis_test )
@@ -70,8 +69,7 @@ inference_analysis_test(test_analyzer SRCS analyzer_tester.cc
70
69
attention_lstm_fuse_pass
71
70
paddle_inference_api
72
71
pass
73
- ARGS --inference_model_dir=${PYTHON_TESTS_DIR}/book/word2vec.inference.model
74
- --infer_ditu_rnn_model=${DITU_INSTALL_DIR}/model
72
+ ARGS --infer_ditu_rnn_model=${DITU_INSTALL_DIR}/model
75
73
--infer_ditu_rnn_data=${DITU_INSTALL_DIR}/data.txt )
76
74
77
75
inference_analysis_test (test_data_flow_graph SRCS data_flow_graph_tester.cc )
@@ -93,8 +91,7 @@ if (NOT EXISTS ${CHINESE_NER_INSTALL_DIR})
93
91
inference_download_and_uncompress (${CHINESE_NER_INSTALL_DIR} ${CHINESE_NER_DATA_URL} "chinese_ner-data.txt.tar.gz" )
94
92
endif ()
95
93
96
- inference_analysis_test (test_chinese_ner SRCS chinese_ner_tester .cc
94
+ inference_analysis_test (test_analyzer_ner SRCS analyzer_ner_tester .cc
97
95
EXTRA_DEPS paddle_inference_api paddle_fluid_api
98
- ARGS --inference_model_dir=${PYTHON_TESTS_DIR}/book/word2vec.inference.model
99
- --infer_model=${CHINESE_NER_INSTALL_DIR}/model
96
+ ARGS --infer_model=${CHINESE_NER_INSTALL_DIR}/model
100
97
--infer_data=${CHINESE_NER_INSTALL_DIR}/data.txt )
0 commit comments