diff --git a/CMakeLists.txt b/CMakeLists.txt index d0a5bdc46d0cfa2d2f484da53667a44371ae9a30..1a0b0b368616059a9a7e4b75badd6327271f483e 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -80,7 +80,7 @@ option(WITH_FAST_MATH "Make use of fast math library, might affect the precisi option(WITH_DGC "Use DGC(Deep Gradient Compression) or not" ON) # for lite, both server and mobile framework. -option(WITH_LITE "Enable lite framework" ON) +option(WITH_LITE "Enable lite framework" OFF) option(LITE_WITH_CUDA "Enable CUDA in lite mode" OFF) option(LITE_WITH_X86 "Enable X86 in lite mode" ON) option(LITE_WITH_LIGHT_WEIGHT_FRAMEWORK "Enable light-weight framework" OFF) diff --git a/paddle/fluid/lite/api/light_api_test.cc b/paddle/fluid/lite/api/light_api_test.cc index 6dd596bbee5b0c1285ba2558069363a7071bb65a..ad0d87cf00bd98924a432455261bb3f0d7ff9c69 100644 --- a/paddle/fluid/lite/api/light_api_test.cc +++ b/paddle/fluid/lite/api/light_api_test.cc @@ -26,7 +26,7 @@ TEST(LightAPI, load) { predictor.Build(FLAGS_optimized_model); auto* input_tensor = predictor.GetInput(0); - input_tensor->Resize(DDimLite(std::vector({100, 100}))); + input_tensor->Resize(DDim(std::vector({100, 100}))); auto* data = input_tensor->mutable_data(); for (int i = 0; i < 100 * 100; i++) { data[i] = i; @@ -39,13 +39,13 @@ TEST(LightAPI, load) { } // namespace paddle USE_LITE_OP(mul); -// USE_LITE_OP(fc); -// USE_LITE_OP(scale); +USE_LITE_OP(fc); +USE_LITE_OP(scale); USE_LITE_OP(feed); USE_LITE_OP(fetch); USE_LITE_OP(io_copy); -// USE_LITE_KERNEL(fc, kHost, kFloat, kNCHW, def); -// USE_LITE_KERNEL(mul, kHost, kFloat, kNCHW, def); -// USE_LITE_KERNEL(scale, kHost, kFloat, kNCHW, def); +USE_LITE_KERNEL(fc, kHost, kFloat, kNCHW, def); +USE_LITE_KERNEL(mul, kHost, kFloat, kNCHW, def); +USE_LITE_KERNEL(scale, kHost, kFloat, kNCHW, def); USE_LITE_KERNEL(feed, kHost, kAny, kAny, def); USE_LITE_KERNEL(fetch, kHost, kAny, kAny, def); diff --git a/paddle/fluid/lite/core/CMakeLists.txt b/paddle/fluid/lite/core/CMakeLists.txt index e5b118aa66a339302c728e2a1cfac0bc34f90750..f6c3006844616511eee6a039fc552425941d3fc7 100644 --- a/paddle/fluid/lite/core/CMakeLists.txt +++ b/paddle/fluid/lite/core/CMakeLists.txt @@ -1,4 +1,4 @@ -cc_library(lite_gtest_main SRCS lite_gtest_main.cc) +cc_library(lite_gtest_main SRCS lite_gtest_main.cc DEPS gtest) cc_library(memory_lite SRCS memory.cc DEPS target_wrapper_lite target_wrapper_host) cc_library(target_wrapper_lite SRCS target_wrapper.cc)