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

enable lite ut. (#30890)

上级 ad50fa71
...@@ -311,6 +311,7 @@ if(WITH_DGC) ...@@ -311,6 +311,7 @@ if(WITH_DGC)
endif() endif()
if (WITH_LITE) if (WITH_LITE)
message(STATUS "Compile Paddle with Lite Engine.")
include(external/lite) include(external/lite)
endif (WITH_LITE) endif (WITH_LITE)
......
...@@ -599,13 +599,12 @@ endif() ...@@ -599,13 +599,12 @@ endif()
set(LITE_MODEL_INSTALL_DIR "${INFERENCE_DEMO_INSTALL_DIR}/lite") set(LITE_MODEL_INSTALL_DIR "${INFERENCE_DEMO_INSTALL_DIR}/lite")
download_data(${LITE_MODEL_INSTALL_DIR} "mul_model_fp32.tgz") 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
#inference_analysis_test(lite_mul_model_test SRCS lite_mul_model_test.cc EXTRA_DEPS ${INFERENCE_EXTRA_DEPS}
# EXTRA_DEPS ${INFERENCE_EXTRA_DEPS} ARGS --infer_model=${LITE_MODEL_INSTALL_DIR})
# ARGS --infer_model=${LITE_MODEL_INSTALL_DIR}) inference_analysis_test(lite_resnet50_test SRCS lite_resnet50_test.cc
#inference_analysis_test(lite_resnet50_test SRCS lite_resnet50_test.cc EXTRA_DEPS ${INFERENCE_EXTRA_DEPS}
# EXTRA_DEPS ${INFERENCE_EXTRA_DEPS} ARGS --infer_model=${RESNET50_MODEL_DIR})
# ARGS --infer_model=${RESNET50_MODEL_DIR})
inference_analysis_test(test_analyzer_capi SRCS analyzer_capi_tester.cc inference_analysis_test(test_analyzer_capi SRCS analyzer_capi_tester.cc
EXTRA_DEPS ${INFERENCE_EXTRA_DEPS} paddle_inference_c EXTRA_DEPS ${INFERENCE_EXTRA_DEPS} paddle_inference_c
...@@ -657,8 +656,7 @@ if(WITH_MKLDNN) ...@@ -657,8 +656,7 @@ if(WITH_MKLDNN)
set_tests_properties(test_analyzer_int8_mobilenetv1 PROPERTIES TIMEOUT 120) set_tests_properties(test_analyzer_int8_mobilenetv1 PROPERTIES TIMEOUT 120)
endif() 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_mobilenet_transpose PROPERTIES TIMEOUT 120)
set_tests_properties(test_analyzer_resnet50 PROPERTIES TIMEOUT 120) set_tests_properties(test_analyzer_resnet50 PROPERTIES TIMEOUT 120)
set_tests_properties(test_analyzer_ner PROPERTIES TIMEOUT 120) set_tests_properties(test_analyzer_ner PROPERTIES TIMEOUT 120)
......
...@@ -81,6 +81,7 @@ TEST(Predictor, use_gpu) { ...@@ -81,6 +81,7 @@ TEST(Predictor, use_gpu) {
config.EnableLiteEngine(PrecisionType::kFloat32); config.EnableLiteEngine(PrecisionType::kFloat32);
auto predictor = CreatePredictor(config); auto predictor = CreatePredictor(config);
const int batch = 1; const int batch = 1;
const int channel = 3; const int channel = 3;
const int height = 318; const int height = 318;
......
op_library(lite_engine_op DEPS lite_engine lite_tensor_utils) 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.
先完成此消息的编辑!
想要评论请 注册