diff --git a/cmake/external/lite.cmake b/cmake/external/lite.cmake index 2617bdd834d1867eb6c2c40dac944aefe0f2a783..323a8ddc729f846217f6ea6ba1c05cc5a2d777d5 100644 --- a/cmake/external/lite.cmake +++ b/cmake/external/lite.cmake @@ -24,6 +24,10 @@ if (NOT LITE_SOURCE_DIR OR NOT LITE_BINARY_DIR) set(LITE_SOURCES_DIR ${THIRD_PARTY_PATH}/lite) set(LITE_INSTALL_DIR ${THIRD_PARTY_PATH}/install/lite) + if(NOT LITE_GIT_TAG) + set(LITE_GIT_TAG 34c29406c27ee00cef033a98887403443eb2565f) + endif() + # No quotes, so cmake can resolve it as a command with arguments. set(LITE_BUILD_COMMAND $(MAKE) publish_inference -j) set(LITE_OPTIONAL_ARGS -DWITH_MKL=ON @@ -38,13 +42,14 @@ if (NOT LITE_SOURCE_DIR OR NOT LITE_BINARY_DIR) -DLITE_BUILD_EXTRA=ON -DCUDNN_ROOT=${CUDNN_ROOT} -DLITE_WITH_STATIC_CUDA=OFF + -DCUDA_ARCH_NAME=${CUDA_ARCH_NAME} -DLITE_WITH_ARM=OFF) ExternalProject_Add( ${LITE_PROJECT} ${EXTERNAL_PROJECT_LOG_ARGS} GIT_REPOSITORY "https://github.com/PaddlePaddle/Paddle-Lite.git" - GIT_TAG 0f875ef367bd2dbfa2e557eb2a2fc841bacdf6cf + GIT_TAG ${LITE_GIT_TAG} PREFIX ${LITE_SOURCES_DIR} UPDATE_COMMAND "" BUILD_COMMAND ${LITE_BUILD_COMMAND} diff --git a/paddle/fluid/pybind/inference_api.cc b/paddle/fluid/pybind/inference_api.cc index 5d3d01b6b3371ee16e1d957a8aba013fdf4d0271..3a19ad2d0235fb42eb4a1e2d94f40b179cae9a4f 100644 --- a/paddle/fluid/pybind/inference_api.cc +++ b/paddle/fluid/pybind/inference_api.cc @@ -423,6 +423,11 @@ void BindAnalysisConfig(py::module *m) { std::map>({}), py::arg("disable_trt_plugin_fp16") = false) .def("tensorrt_engine_enabled", &AnalysisConfig::tensorrt_engine_enabled) + .def("enable_lite_engine", &AnalysisConfig::EnableLiteEngine, + py::arg("precision_mode") = AnalysisConfig::Precision::kFloat32, + py::arg("passes_filter") = std::vector(), + py::arg("ops_filter") = std::vector()) + .def("lite_engine_enabled", &AnalysisConfig::lite_engine_enabled) .def("switch_ir_debug", &AnalysisConfig::SwitchIrDebug, py::arg("x") = true) .def("enable_mkldnn", &AnalysisConfig::EnableMKLDNN)