diff --git a/paddle/fluid/platform/CMakeLists.txt b/paddle/fluid/platform/CMakeLists.txt index 5695fd03bacf384400035103b32df675bcde9f70..550278950c39170f46cf52d32ca9f50968f3e1ca 100644 --- a/paddle/fluid/platform/CMakeLists.txt +++ b/paddle/fluid/platform/CMakeLists.txt @@ -120,7 +120,7 @@ cc_library(cudnn_workspace_helper SRCS cudnn_workspace_helper.cc DEPS boost) # avoiding cycle dependencies cc_library(device_context SRCS device_context.cc init.cc DEPS simple_threadpool malloc xxhash ${STREAM_CALLBACK_DEPS} place pten_place eigen3 stringpiece cpu_helper cpu_info framework_proto ${IPU_CTX_DEPS} ${GPU_CTX_DEPS} ${NPU_CTX_DEPS} ${MKLDNN_CTX_DEPS} - ${dgc_deps} dlpack cudnn_workspace_helper ${XPU_CTX_DEPS} ${MLU_CTX_DEPS} cpu_context) + ${dgc_deps} dlpack cudnn_workspace_helper ${XPU_CTX_DEPS} ${MLU_CTX_DEPS} eigen3 cpu_context) if(WITH_XPU) target_link_libraries(device_context xpu_context) endif() diff --git a/paddle/pten/backends/cpu/CMakeLists.txt b/paddle/pten/backends/cpu/CMakeLists.txt index 62eff2dedc99c8dcc54c0f1372e3b65e36c3e9f9..965b33f3800edf9597b07ad2446637d2c505fe0f 100644 --- a/paddle/pten/backends/cpu/CMakeLists.txt +++ b/paddle/pten/backends/cpu/CMakeLists.txt @@ -1,6 +1,6 @@ if(WITH_MKLDNN) # TODO(wilber): support mkldnn context. - cc_library(cpu_context SRCS cpu_context.cc DEPS pten_device_context mkldnn) + cc_library(cpu_context SRCS cpu_context.cc DEPS pten_device_context mkldnn eigen3) else() - cc_library(cpu_context SRCS cpu_context.cc DEPS pten_device_context) + cc_library(cpu_context SRCS cpu_context.cc DEPS pten_device_context eigen3) endif()