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

L
Leo Chen 已提交
5
cc_test(mlu_enforce_test SRCS enforce_test.cc)
6 7 8
cc_library(
  mlu_info
  SRCS mlu_info.cc
9
  DEPS enforce glog malloc monitor neuware_lib)
10 11 12
cc_library(
  mlu_stream
  SRCS mlu_stream.cc
R
Ruibiao Chen 已提交
13
  DEPS mlu_info stream_callback_manager eigen3 ${MKLDNN_CTX_DEPS})
14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29
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)