if(WITH_MKLDNN)
  # TODO(wilber): support mkldnn context.
  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 eigen3)
endif()
