|
|
|
@ -1,5 +1,9 @@
|
|
|
|
|
set(INFERENCE_EXTRA_DEPS paddle_inference_api paddle_fluid_api ir_pass_manager analysis_predictor)
|
|
|
|
|
|
|
|
|
|
if(WITH_GPU AND TENSORRT_FOUND)
|
|
|
|
|
set(INFERENCE_EXTRA_DEPS ${INFERENCE_EXTRA_DEPS} analysis ${analysis_deps} ir_pass_manager analysis_predictor)
|
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
function(download_model install_dir model_name)
|
|
|
|
|
if (NOT EXISTS ${install_dir})
|
|
|
|
|
inference_download_and_uncompress(${install_dir} ${INFERENCE_URL} ${model_name})
|
|
|
|
@ -109,6 +113,6 @@ if(WITH_GPU AND TENSORRT_FOUND)
|
|
|
|
|
inference_download_and_uncompress(${TRT_MODEL_INSTALL_DIR} ${INFERENCE_URL}/tensorrt_test "trt_test_models.tar.gz")
|
|
|
|
|
endif()
|
|
|
|
|
inference_analysis_test(test_trt_models SRCS trt_models_tester.cc
|
|
|
|
|
EXTRA_DEPS ${INFERENCE_EXTRA_DEPS} analysis ${analysis_deps} ir_pass_manager analysis_predictor
|
|
|
|
|
EXTRA_DEPS ${INFERENCE_EXTRA_DEPS}
|
|
|
|
|
ARGS --infer_model=${TRT_MODEL_INSTALL_DIR}/trt_test_models SERIAL)
|
|
|
|
|
endif()
|
|
|
|
|