1
1
set (INFERENCE_EXTRA_DEPS paddle_inference_api paddle_fluid_api ir_pass_manager analysis_predictor )
2
2
3
+ if (WITH_GPU AND TENSORRT_FOUND )
4
+ set (INFERENCE_EXTRA_DEPS ${INFERENCE_EXTRA_DEPS} analysis ${analysis_deps} ir_pass_manager analysis_predictor )
5
+ endif ()
6
+
3
7
function (download_model install_dir model_name )
4
8
if (NOT EXISTS ${install_dir} )
5
9
inference_download_and_uncompress (${install_dir} ${INFERENCE_URL} ${model_name} )
@@ -75,11 +79,11 @@ endif()
75
79
inference_analysis_api_test (test_analyzer_ocr ${OCR_INSTALL_DIR} analyzer_vis_tester.cc )
76
80
77
81
# resnet50
78
- inference_analysis_api_test_with_fake_data (test_analyzer_resnet50
82
+ inference_analysis_api_test_with_fake_data (test_analyzer_resnet50
79
83
"${INFERENCE_DEMO_INSTALL_DIR} /resnet50" analyzer_resnet50_tester.cc "resnet50_model.tar.gz" )
80
84
81
85
# mobilenet with depthwise_conv op
82
- inference_analysis_api_test_with_fake_data (test_analyzer_mobilenet
86
+ inference_analysis_api_test_with_fake_data (test_analyzer_mobilenet
83
87
"${INFERENCE_DEMO_INSTALL_DIR} /mobilenet_depthwise_conv" analyzer_resnet50_tester.cc "mobilenet_model.tar.gz" )
84
88
85
89
# anakin
@@ -89,15 +93,15 @@ if (WITH_ANAKIN AND WITH_MKL) # only needed in CI
89
93
set (ANAKIN_RNN1_INSTALL_DIR "${ANAKIN_INSTALL_DIR} /rnn1" )
90
94
inference_download (${ANAKIN_RNN1_INSTALL_DIR} ${INFERENCE_URL} "anakin_test%2Fditu_rnn.anakin2.model.bin" )
91
95
inference_download (${ANAKIN_RNN1_INSTALL_DIR} ${INFERENCE_URL} "anakin_test%2Fditu_rnn_data.txt" )
92
- cc_test (test_anakin_rnn1 SRCS anakin_rnn1_tester.cc
93
- ARGS --model=${ANAKIN_RNN1_INSTALL_DIR}/anakin_test%2Fditu_rnn.anakin2.model.bin
96
+ cc_test (test_anakin_rnn1 SRCS anakin_rnn1_tester.cc
97
+ ARGS --model=${ANAKIN_RNN1_INSTALL_DIR}/anakin_test%2Fditu_rnn.anakin2.model.bin
94
98
--datapath=${ANAKIN_RNN1_INSTALL_DIR}/anakin_test%2Fditu_rnn_data.txt
95
99
DEPS inference_anakin_api_shared SERIAL )
96
100
# anakin mobilenet
97
101
if (WITH_GPU )
98
102
set (ANAKIN_MOBILENET_INSTALL_DIR "${ANAKIN_INSTALL_DIR} /mobilenet" )
99
103
inference_download (${ANAKIN_MOBILENET_INSTALL_DIR} ${INFERENCE_URL} "mobilenet_v2.anakin.bin" )
100
- cc_test (test_anakin_mobilenet SRCS anakin_mobilenet_tester.cc
104
+ cc_test (test_anakin_mobilenet SRCS anakin_mobilenet_tester.cc
101
105
ARGS --model=${ANAKIN_MOBILENET_INSTALL_DIR}/mobilenet_v2.anakin.bin
102
106
DEPS inference_anakin_api_shared dynload_cuda SERIAL )
103
107
endif ()
@@ -109,6 +113,6 @@ if(WITH_GPU AND TENSORRT_FOUND)
109
113
inference_download_and_uncompress (${TRT_MODEL_INSTALL_DIR} ${INFERENCE_URL} /tensorrt_test "trt_test_models.tar.gz" )
110
114
endif ()
111
115
inference_analysis_test (test_trt_models SRCS trt_models_tester.cc
112
- EXTRA_DEPS ${INFERENCE_EXTRA_DEPS} analysis ${analysis_deps} ir_pass_manager analysis_predictor
116
+ EXTRA_DEPS ${INFERENCE_EXTRA_DEPS}
113
117
ARGS --infer_model=${TRT_MODEL_INSTALL_DIR}/trt_test_models SERIAL )
114
118
endif ()
0 commit comments