|
|
|
|
@ -477,9 +477,10 @@ if(WITH_GPU AND TENSORRT_FOUND)
|
|
|
|
|
inference_download_and_uncompress(${TEST_TRT_ERNIE_MODEL} ${INFERENCE_URL}/tensorrt_test "ernie_model_4_unserialized.tgz")
|
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
inference_analysis_test(test_trt_dynamic_shape_ernie_ser_deser SRCS trt_dynamic_shape_ernie_deserialize_test.cc
|
|
|
|
|
EXTRA_DEPS ${INFERENCE_EXTRA_DEPS}
|
|
|
|
|
ARGS --infer_model=${TEST_TRT_ERNIE_MODEL}/ernie_model_4_unserialized)
|
|
|
|
|
# disable test_trt_dynamic_shape_ernie_ser_deser temporary
|
|
|
|
|
#inference_analysis_test(test_trt_dynamic_shape_ernie_ser_deser SRCS trt_dynamic_shape_ernie_deserialize_test.cc
|
|
|
|
|
# EXTRA_DEPS ${INFERENCE_EXTRA_DEPS}
|
|
|
|
|
# ARGS --infer_model=${TEST_TRT_ERNIE_MODEL}/ernie_model_4_unserialized)
|
|
|
|
|
|
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
|