diff --git a/cmake/generic.cmake b/cmake/generic.cmake index 43350333a5f28e721c623a61297db90a7323dc03..39ebaf7d9513b8ced2b03352c8c2287dec1842fc 100644 --- a/cmake/generic.cmake +++ b/cmake/generic.cmake @@ -437,6 +437,7 @@ function(lite_cc_test args) message(STATUS "building lite raw test: ${args}") raw_cc_test(${args} ${ARGN}) else() + message(STATUS "building lite heavy test: ${args}") cc_test(${args} ${ARGN}) endif() endfunction() diff --git a/paddle/fluid/lite/core/context.h b/paddle/fluid/lite/core/context.h index a7ec521dacfb2bb40827169c25bcc10410a306f9..50ec1708d24cbfaba6a532f9c5d2a0a8b63d66a9 100644 --- a/paddle/fluid/lite/core/context.h +++ b/paddle/fluid/lite/core/context.h @@ -13,15 +13,15 @@ // limitations under the License. #pragma once + +#ifdef LITE_WITH_CUDA #include +#include "paddle/fluid/lite/cuda/cuda_utils.h" +#endif #include #include #include "paddle/fluid/lite/core/target_wrapper.h" -#ifdef PADDLE_WITH_CUDA -#include "paddle/fluid/lite/cuda/cuda_utils.h" -#endif - namespace paddle { namespace lite { diff --git a/paddle/fluid/lite/core/mir/runtime_context_assign_pass.cc b/paddle/fluid/lite/core/mir/runtime_context_assign_pass.cc index 48149ec1a5b29cc89864c34717635bd07263eb76..2919de96f98f11368b7ff90386a1780b43ae9cd7 100644 --- a/paddle/fluid/lite/core/mir/runtime_context_assign_pass.cc +++ b/paddle/fluid/lite/core/mir/runtime_context_assign_pass.cc @@ -68,12 +68,16 @@ class RuntimeContextAssignPass : public StmtPass { } #endif +#ifdef LITE_WITH_CUDA void InitCudaBlas() { cublas_fp32_ = std::make_shared>(); } +#endif private: +#ifdef LITE_WITH_CUDA std::shared_ptr> cublas_fp32_; +#endif }; } // namespace mir