CMakeLists.txt 652 字节
Newer Older
1
if(NOT WITH_MLU)
2
  return()
3
endif()
F
fwenguang 已提交
4

5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32
cc_test(
  mlu_enforce_test
  SRCS enforce_test.cc
  DEPS stringpiece)
cc_library(
  mlu_info
  SRCS mlu_info.cc
  DEPS enforce glog monitor neuware_lib)
cc_library(
  mlu_stream
  SRCS mlu_stream.cc
  DEPS boost mlu_info stream_callback_manager eigen3 ${MKLDNN_CTX_DEPS})
cc_library(
  mlu_device_context
  SRCS device_context.cc
  DEPS mlu_stream)
cc_test(
  mlu_device_context_test
  SRCS device_context_test.cc
  DEPS mlu_device_context)
cc_library(
  mlu_collective_helper
  SRCS mlu_collective_helper.cc
  DEPS mlu_stream mlu_info)
cc_library(
  mlu_resource_pool
  SRCS mlu_resource_pool.cc
  DEPS mlu_info)