未验证 提交 7d91974c 编写于 作者: W Wilber 提交者: GitHub

enable lite ut. (#30890)

上级 ad50fa71
......@@ -311,6 +311,7 @@ if(WITH_DGC)
endif()
if (WITH_LITE)
message(STATUS "Compile Paddle with Lite Engine.")
include(external/lite)
endif (WITH_LITE)
......
......@@ -599,13 +599,12 @@ endif()
set(LITE_MODEL_INSTALL_DIR "${INFERENCE_DEMO_INSTALL_DIR}/lite")
download_data(${LITE_MODEL_INSTALL_DIR} "mul_model_fp32.tgz")
#TODO(wilber): tmp disable ut.
#inference_analysis_test(lite_mul_model_test SRCS lite_mul_model_test.cc
# EXTRA_DEPS ${INFERENCE_EXTRA_DEPS}
# ARGS --infer_model=${LITE_MODEL_INSTALL_DIR})
#inference_analysis_test(lite_resnet50_test SRCS lite_resnet50_test.cc
# EXTRA_DEPS ${INFERENCE_EXTRA_DEPS}
# ARGS --infer_model=${RESNET50_MODEL_DIR})
inference_analysis_test(lite_mul_model_test SRCS lite_mul_model_test.cc
EXTRA_DEPS ${INFERENCE_EXTRA_DEPS}
ARGS --infer_model=${LITE_MODEL_INSTALL_DIR})
inference_analysis_test(lite_resnet50_test SRCS lite_resnet50_test.cc
EXTRA_DEPS ${INFERENCE_EXTRA_DEPS}
ARGS --infer_model=${RESNET50_MODEL_DIR})
inference_analysis_test(test_analyzer_capi SRCS analyzer_capi_tester.cc
EXTRA_DEPS ${INFERENCE_EXTRA_DEPS} paddle_inference_c
......@@ -657,8 +656,7 @@ if(WITH_MKLDNN)
set_tests_properties(test_analyzer_int8_mobilenetv1 PROPERTIES TIMEOUT 120)
endif()
#TODO(wilber): tmp disable ut
#set_tests_properties(lite_resnet50_test PROPERTIES TIMEOUT 120)
set_tests_properties(lite_resnet50_test PROPERTIES TIMEOUT 120)
set_tests_properties(test_analyzer_mobilenet_transpose PROPERTIES TIMEOUT 120)
set_tests_properties(test_analyzer_resnet50 PROPERTIES TIMEOUT 120)
set_tests_properties(test_analyzer_ner PROPERTIES TIMEOUT 120)
......
......@@ -81,6 +81,7 @@ TEST(Predictor, use_gpu) {
config.EnableLiteEngine(PrecisionType::kFloat32);
auto predictor = CreatePredictor(config);
const int batch = 1;
const int channel = 3;
const int height = 318;
......
op_library(lite_engine_op DEPS lite_engine lite_tensor_utils)
# TODO(wilber): fix the ut.
#cc_test(test_lite_engine_op SRCS lite_engine_op_test.cc DEPS lite_engine_op analysis)
cc_test(test_lite_engine_op SRCS lite_engine_op_test.cc DEPS lite_engine_op analysis)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册