mirror of
https://github.com/opencv/opencv.git
synced 2024-11-28 05:06:29 +08:00
a817813b50
G-API: Support DirectML Execution Provider for ONNXRT Backend #24045 ### Pull Request Readiness Checklist See details at https://github.com/opencv/opencv/wiki/How_to_contribute#making-a-good-pull-request - [ ] I agree to contribute to the project under Apache 2 License. - [ ] To the best of my knowledge, the proposed patch is not based on a code under GPL or another license that is incompatible with OpenCV - [ ] The PR is proposed to the proper branch - [ ] There is a reference to the original bug report and related work - [ ] There is accuracy test, performance test and test data in opencv_extra repository, if applicable Patch to opencv_extra has the same branch name. - [ ] The feature is well documented and sample code can be built with the project CMake
52 lines
1.6 KiB
CMake
52 lines
1.6 KiB
CMake
ocv_clear_vars(HAVE_ONNX)
|
|
|
|
set(ONNXRT_ROOT_DIR "" CACHE PATH "ONNX Runtime install directory")
|
|
|
|
# For now, check the old name ORT_INSTALL_DIR
|
|
if(ORT_INSTALL_DIR AND NOT ONNXRT_ROOT_DIR)
|
|
set(ONNXRT_ROOT_DIR ${ORT_INSTALL_DIR})
|
|
endif()
|
|
|
|
if(ONNXRT_ROOT_DIR)
|
|
find_library(ORT_LIB onnxruntime
|
|
${ONNXRT_ROOT_DIR}/lib
|
|
CMAKE_FIND_ROOT_PATH_BOTH)
|
|
find_path(ORT_INCLUDE onnxruntime_cxx_api.h
|
|
${ONNXRT_ROOT_DIR}/include/onnxruntime/core/session
|
|
CMAKE_FIND_ROOT_PATH_BOTH)
|
|
endif()
|
|
|
|
macro(detect_onxxrt_ep filename dir have_ep_var)
|
|
find_path(ORT_EP_INCLUDE ${filename} ${dir} CMAKE_FIND_ROOT_PATH_BOTH)
|
|
if(ORT_EP_INCLUDE)
|
|
set(${have_ep_var} TRUE)
|
|
endif()
|
|
endmacro()
|
|
|
|
if(ORT_LIB AND ORT_INCLUDE)
|
|
# Check DirectML Execution Provider availability
|
|
get_filename_component(dml_dir ${ONNXRT_ROOT_DIR}/include/onnxruntime/core/providers/dml ABSOLUTE)
|
|
detect_onxxrt_ep(
|
|
dml_provider_factory.h
|
|
${dml_dir}
|
|
HAVE_ONNX_DML
|
|
)
|
|
|
|
set(HAVE_ONNX TRUE)
|
|
# For CMake output only
|
|
set(ONNX_LIBRARIES "${ORT_LIB}" CACHE STRING "ONNX Runtime libraries")
|
|
set(ONNX_INCLUDE_DIR "${ORT_INCLUDE}" CACHE STRING "ONNX Runtime include path")
|
|
|
|
# Link target with associated interface headers
|
|
set(ONNX_LIBRARY "onnxruntime" CACHE STRING "ONNX Link Target")
|
|
ocv_add_library(${ONNX_LIBRARY} SHARED IMPORTED)
|
|
set_target_properties(${ONNX_LIBRARY} PROPERTIES
|
|
INTERFACE_INCLUDE_DIRECTORIES ${ORT_INCLUDE}
|
|
IMPORTED_LOCATION ${ORT_LIB}
|
|
IMPORTED_IMPLIB ${ORT_LIB})
|
|
endif()
|
|
|
|
if(NOT HAVE_ONNX)
|
|
ocv_clear_vars(HAVE_ONNX ORT_LIB ORT_INCLUDE_DIR)
|
|
endif()
|