mirror of
https://github.com/MaaAssistantArknights/MaaAssistantArknights.git
synced 2026-07-16 01:40:46 +08:00
5c0445d3ab
Co-authored-by: status102 <102887808+status102@users.noreply.github.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
42 lines
1.8 KiB
CMake
42 lines
1.8 KiB
CMake
find_path(ONNXRuntime_INCLUDE_DIR NAMES onnxruntime/onnxruntime_c_api.h)
|
|
|
|
find_library(ONNXRuntime_LIBRARY_IMP NAMES onnxruntime)
|
|
|
|
if (WIN32)
|
|
get_filename_component(ONNXRuntime_PATH_LIB ${ONNXRuntime_LIBRARY_IMP} DIRECTORY)
|
|
find_file(ONNXRuntime_LIBRARY NAMES onnxruntime_maa.dll PATHS "${ONNXRuntime_PATH_LIB}/../bin")
|
|
|
|
find_file(ONNXRuntime_LIBRARY_IMP_DEBUG NAMES onnxruntime.lib PATHS "${ONNXRuntime_PATH_LIB}/../debug/lib")
|
|
find_file(ONNXRuntime_LIBRARY_DEBUG NAMES onnxruntime_maa.dll PATHS "${ONNXRuntime_PATH_LIB}/../debug/bin")
|
|
else ()
|
|
set(ONNXRuntime_LIBRARY ${ONNXRuntime_LIBRARY_IMP})
|
|
endif (WIN32)
|
|
|
|
include(FindPackageHandleStandardArgs)
|
|
find_package_handle_standard_args(
|
|
ONNXRuntime
|
|
REQUIRED_VARS ONNXRuntime_LIBRARY_IMP ONNXRuntime_INCLUDE_DIR
|
|
)
|
|
|
|
if(ONNXRuntime_FOUND)
|
|
set(ONNXRuntime_INCLUDE_DIRS ${ONNXRuntime_INCLUDE_DIR})
|
|
if(NOT TARGET ONNXRuntime::ONNXRuntime)
|
|
add_library(ONNXRuntime::ONNXRuntime SHARED IMPORTED)
|
|
set_property(TARGET ONNXRuntime::ONNXRuntime APPEND PROPERTY IMPORTED_CONFIGURATIONS RELEASE)
|
|
if (WIN32)
|
|
set_property(TARGET ONNXRuntime::ONNXRuntime APPEND PROPERTY IMPORTED_CONFIGURATIONS DEBUG)
|
|
set_target_properties(ONNXRuntime::ONNXRuntime PROPERTIES
|
|
IMPORTED_IMPLIB_RELEASE "${ONNXRuntime_LIBRARY_IMP}"
|
|
)
|
|
set_target_properties(ONNXRuntime::ONNXRuntime PROPERTIES
|
|
IMPORTED_IMPLIB_DEBUG "${ONNXRuntime_LIBRARY_IMP_DEBUG}"
|
|
IMPORTED_LOCATION_DEBUG "${ONNXRuntime_LIBRARY_DEBUG}"
|
|
)
|
|
endif (WIN32)
|
|
set_target_properties(ONNXRuntime::ONNXRuntime PROPERTIES
|
|
IMPORTED_LOCATION_RELEASE "${ONNXRuntime_LIBRARY}"
|
|
INTERFACE_INCLUDE_DIRECTORIES "${ONNXRuntime_INCLUDE_DIR}"
|
|
)
|
|
endif()
|
|
endif()
|