CMakeLists.txt 684 字节
Newer Older
1 2 3 4 5
# NPU

add_subdirectory(dynload)

if(WITH_ASCEND)
6 7 8 9
  cc_library(
    ascend_npu_info
    SRCS ascend_npu_info.cc
    DEPS gflags glog enforce atlas_acl)
10 11 12
endif()

if(WITH_ASCEND_CL)
13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32
  cc_library(
    npu_info
    SRCS npu_info.cc
    DEPS gflags glog enforce monitor ascendcl acl_op_compiler)
  cc_library(
    npu_resource_pool
    SRCS npu_resource_pool.cc
    DEPS npu_info)
  cc_library(
    npu_stream
    SRCS npu_stream.cc
    DEPS enforce boost stream_callback_manager)
  cc_library(
    npu_collective_helper
    SRCS npu_collective_helper.cc
    DEPS npu_stream npu_info data_type)
  cc_library(
    npu_op_runner
    SRCS npu_op_runner.cc
    DEPS operator npu_info)
33
endif()