From bd5c962dab7cdcba9816f7c1dd70bad754a70cca Mon Sep 17 00:00:00 2001 From: Wilber Date: Wed, 26 Jan 2022 16:43:45 +0800 Subject: [PATCH] [PTEN] cpu_context add eigen deps (#39234) * add eigen deps * update --- paddle/fluid/platform/CMakeLists.txt | 2 +- paddle/pten/backends/cpu/CMakeLists.txt | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/paddle/fluid/platform/CMakeLists.txt b/paddle/fluid/platform/CMakeLists.txt index 5695fd03bac..550278950c3 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 62eff2dedc9..965b33f3800 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() -- GitLab