onnxruntime

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

4
  add_definitions(-DUSE_MIGRAPHX=1)
5
  set(BUILD_LIBRARY_ONLY 1)
6
  add_definitions("-DONNX_ML=1")
7
  add_definitions("-DONNX_NAMESPACE=onnx")
8
  include_directories(${protobuf_SOURCE_DIR} ${eigen_SOURCE_DIR})
9
  set(MIGRAPHX_ROOT ${onnxruntime_MIGRAPHX_HOME})
10
  include_directories(${onnx_SOURCE_DIR})
11
  set(OLD_CMAKE_CXX_FLAGS ${CMAKE_CXX_FLAGS})
12
  if ( CMAKE_COMPILER_IS_GNUCC )
13
    set(CMAKE_CXX_FLAGS  "${CMAKE_CXX_FLAGS} -Wno-unused-parameter -Wno-missing-field-initializers")
14
  endif()
15
  set(CXX_VERSION_DEFINED TRUE)
16
  set(CMAKE_CXX_FLAGS ${OLD_CMAKE_CXX_FLAGS})
17
  if ( CMAKE_COMPILER_IS_GNUCC )
18
    set(CMAKE_CXX_FLAGS  "${CMAKE_CXX_FLAGS} -Wno-unused-parameter")
19
  endif()
20

21
  # Add search paths for default rocm installation
22
  list(APPEND CMAKE_PREFIX_PATH /opt/rocm/hcc /opt/rocm/hip /opt/rocm $ENV{HIP_PATH})
23

24
  # Suppress the warning about the small capitals of the package name - Enable when support to CMake 3.27.0 is used
25
  # cmake_policy(SET CMP0144 NEW)
26

27
  if(WIN32 AND NOT HIP_PLATFORM)
28
    set(HIP_PLATFORM "amd")
29
  endif()
30

31
  find_package(hip REQUIRED)
32
  find_package(migraphx REQUIRED PATHS ${AMD_MIGRAPHX_HOME})
33

34
  set(migraphx_libs migraphx::c hip::host)
35

36
  file(GLOB_RECURSE onnxruntime_providers_migraphx_cc_srcs CONFIGURE_DEPENDS
37
    "${ONNXRUNTIME_ROOT}/core/providers/migraphx/*.h"
38
    "${ONNXRUNTIME_ROOT}/core/providers/migraphx/*.cc"
39
    "${ONNXRUNTIME_ROOT}/core/providers/shared_library/*.h"
40
    "${ONNXRUNTIME_ROOT}/core/providers/shared_library/*.cc"
41
  )
42
  source_group(TREE ${ONNXRUNTIME_ROOT}/core FILES ${onnxruntime_providers_migraphx_cc_srcs})
43
  onnxruntime_add_shared_library_module(onnxruntime_providers_migraphx ${onnxruntime_providers_migraphx_cc_srcs})
44
  onnxruntime_add_include_to_target(onnxruntime_providers_migraphx onnxruntime_common onnx flatbuffers::flatbuffers Boost::mp11 safeint_interface)
45
  add_dependencies(onnxruntime_providers_migraphx onnxruntime_providers_shared ${onnxruntime_EXTERNAL_DEPENDENCIES})
46
  target_link_libraries(onnxruntime_providers_migraphx PRIVATE ${migraphx_libs} ${ONNXRUNTIME_PROVIDERS_SHARED} onnx flatbuffers::flatbuffers Boost::mp11 safeint_interface)
47
  target_include_directories(onnxruntime_providers_migraphx PRIVATE ${ONNXRUNTIME_ROOT} ${CMAKE_CURRENT_BINARY_DIR} ${CMAKE_CURRENT_BINARY_DIR}/amdgpu/onnxruntime)
48
  set_target_properties(onnxruntime_providers_migraphx PROPERTIES LINKER_LANGUAGE CXX)
49
  set_target_properties(onnxruntime_providers_migraphx PROPERTIES FOLDER "ONNXRuntime")
50
  target_compile_definitions(onnxruntime_providers_migraphx PRIVATE ONNXIFI_BUILD_LIBRARY=1)
51
  if(MSVC)
52
    set_property(TARGET onnxruntime_providers_migraphx APPEND_STRING PROPERTY LINK_FLAGS /DEF:${ONNXRUNTIME_ROOT}/core/providers/migraphx/symbols.def)
53
    target_link_libraries(onnxruntime_providers_migraphx PRIVATE ws2_32)
54
  else()
55
    target_compile_options(onnxruntime_providers_migraphx PRIVATE -Wno-error=sign-compare)
56
    set_property(TARGET onnxruntime_providers_migraphx APPEND_STRING PROPERTY COMPILE_FLAGS "-Wno-deprecated-declarations")
57
  endif()
58
  if(UNIX)
59
    set_property(TARGET onnxruntime_providers_migraphx APPEND_STRING PROPERTY LINK_FLAGS "-Xlinker --version-script=${ONNXRUNTIME_ROOT}/core/providers/migraphx/version_script.lds -Xlinker --gc-sections")
60
    target_link_libraries(onnxruntime_providers_migraphx PRIVATE nsync::nsync_cpp stdc++fs)
61
  endif()
62

63
  if (onnxruntime_ENABLE_TRAINING_OPS)
64
    onnxruntime_add_include_to_target(onnxruntime_providers_migraphx onnxruntime_training)
65
    target_link_libraries(onnxruntime_providers_migraphx PRIVATE onnxruntime_training)
66
    if (onnxruntime_ENABLE_TRAINING_TORCH_INTEROP)
67
      onnxruntime_add_include_to_target(onnxruntime_providers_migraphx Python::Module)
68
    endif()
69
  endif()
70

71
  if(CMAKE_SYSTEM_NAME STREQUAL "Windows")
72
    install(TARGETS onnxruntime_providers_migraphx
73
            ARCHIVE  DESTINATION ${CMAKE_INSTALL_LIBDIR}
74
            LIBRARY  DESTINATION ${CMAKE_INSTALL_BINDIR}
75
            RUNTIME  DESTINATION ${CMAKE_INSTALL_BINDIR}
76
    )
77
  else()
78
    install(TARGETS onnxruntime_providers_migraphx
79
            ARCHIVE  DESTINATION ${CMAKE_INSTALL_LIBDIR}
80
            LIBRARY  DESTINATION ${CMAKE_INSTALL_LIBDIR}
81
            RUNTIME  DESTINATION ${CMAKE_INSTALL_BINDIR}
82
    )
83
  endif()
84