未验证 提交 bd5c962d 编写于 作者: W Wilber 提交者: GitHub

[PTEN] cpu_context add eigen deps (#39234)

* add eigen deps

* update
上级 5df78366
......@@ -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()
......
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()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册