CMakeLists.txt 236 字节
Newer Older
W
Wilber 已提交
1 2
if(WITH_MKLDNN)
  # TODO(wilber): support mkldnn context.
3
  cc_library(cpu_context SRCS cpu_context.cc DEPS pten_device_context mkldnn eigen3)
W
Wilber 已提交
4
else()
5
  cc_library(cpu_context SRCS cpu_context.cc DEPS pten_device_context eigen3)
W
Wilber 已提交
6
endif()