onnxruntime

1
# Copyright (c) Microsoft Corporation. All rights reserved.
2
# Licensed under the MIT License.
3

4
  add_definitions(-DUSE_CANN=1)
5

6
  file(GLOB_RECURSE onnxruntime_providers_cann_cc_srcs CONFIGURE_DEPENDS
7
    "${ONNXRUNTIME_ROOT}/core/providers/cann/*.h"
8
    "${ONNXRUNTIME_ROOT}/core/providers/cann/*.cc"
9
  )
10

11
  # The shared_library files are in a separate list since they use precompiled headers, and the above files have them disabled.
12
  file(GLOB_RECURSE onnxruntime_providers_cann_shared_srcs CONFIGURE_DEPENDS
13
    "${ONNXRUNTIME_ROOT}/core/providers/shared_library/*.h"
14
    "${ONNXRUNTIME_ROOT}/core/providers/shared_library/*.cc"
15
  )
16

17
  source_group(TREE ${ONNXRUNTIME_ROOT}/core FILES ${onnxruntime_providers_cann_cc_srcs} ${onnxruntime_providers_cann_shared_srcs})
18
  set(onnxruntime_providers_cann_src ${onnxruntime_providers_cann_cc_srcs} ${onnxruntime_providers_cann_shared_srcs})
19

20
  onnxruntime_add_shared_library_module(onnxruntime_providers_cann ${onnxruntime_providers_cann_src})
21
  onnxruntime_add_include_to_target(onnxruntime_providers_cann onnxruntime_common onnxruntime_framework onnx onnx_proto ${PROTOBUF_LIB} flatbuffers::flatbuffers Boost::mp11 safeint_interface)
22

23
  add_dependencies(onnxruntime_providers_cann onnxruntime_providers_shared ${onnxruntime_EXTERNAL_DEPENDENCIES})
24
  target_link_libraries(onnxruntime_providers_cann PRIVATE ascendcl acl_op_compiler fmk_onnx_parser nsync::nsync_cpp ${ABSEIL_LIBS} ${ONNXRUNTIME_PROVIDERS_SHARED})
25
  target_link_directories(onnxruntime_providers_cann PRIVATE ${onnxruntime_CANN_HOME}/lib64)
26
  target_include_directories(onnxruntime_providers_cann PRIVATE ${ONNXRUNTIME_ROOT} ${CMAKE_CURRENT_BINARY_DIR} ${eigen_INCLUDE_DIRS} ${onnxruntime_CANN_HOME} ${onnxruntime_CANN_HOME}/include)
27

28
  set_target_properties(onnxruntime_providers_cann PROPERTIES LINKER_LANGUAGE CXX)
29
  set_target_properties(onnxruntime_providers_cann PROPERTIES FOLDER "ONNXRuntime")
30

31
  install(TARGETS onnxruntime_providers_cann
32
          ARCHIVE  DESTINATION ${CMAKE_INSTALL_LIBDIR}
33
          LIBRARY  DESTINATION ${CMAKE_INSTALL_LIBDIR}
34
          RUNTIME  DESTINATION ${CMAKE_INSTALL_BINDIR})