diff --git a/cmake/external/lite.cmake b/cmake/external/lite.cmake index 70c11d37f911da9225a8609de756438c9b74c596..451c621eac1394743e189cedc36b28a713444d5e 100644 --- a/cmake/external/lite.cmake +++ b/cmake/external/lite.cmake @@ -28,6 +28,10 @@ if (NOT LITE_SOURCE_DIR OR NOT LITE_BINARY_DIR) set(LITE_GIT_TAG 34c29406c27ee00cef033a98887403443eb2565f) endif() + if(NOT CUDA_ARCH_NAME) + set(CUDA_ARCH_NAME "Auto") + 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 diff --git a/paddle/fluid/inference/lite/engine.cc b/paddle/fluid/inference/lite/engine.cc index 38230e4f18f6c7c3b1e17639e155152a4821f49d..6dbe748879c31523ade333dd7ca0b5c340915b3c 100644 --- a/paddle/fluid/inference/lite/engine.cc +++ b/paddle/fluid/inference/lite/engine.cc @@ -18,6 +18,10 @@ #include "paddle/fluid/inference/lite/engine.h" +#include "lite/api/paddle_use_kernels.h" +#include "lite/api/paddle_use_ops.h" +#include "lite/api/paddle_use_passes.h" + namespace paddle { namespace inference { namespace lite { diff --git a/paddle/fluid/inference/lite/engine.h b/paddle/fluid/inference/lite/engine.h index d30c2d31e12c8cc1c4cf52b974f2eaeeee24df50..5f11c51952bd3ce0bb0e09121dbd5e633c6fd3ae 100644 --- a/paddle/fluid/inference/lite/engine.h +++ b/paddle/fluid/inference/lite/engine.h @@ -24,9 +24,6 @@ #pragma GCC diagnostic ignored "-Wall" #include "lite/api/cxx_api.h" #include "lite/api/paddle_place.h" -#include "lite/api/paddle_use_kernels.h" -#include "lite/api/paddle_use_ops.h" -#include "lite/api/paddle_use_passes.h" #include "lite/core/context.h" #include "lite/core/device_info.h" #include "lite/core/op_registry.h"