mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
ea3078aa56
* Update custom ops namespace * rename register, rename generated library name * rename domain to mmcv
17 lines
633 B
CMake
17 lines
633 B
CMake
set(TARGET_NAME mmdeploy_onnxruntime_ops)
|
|
set(SHARED_TARGET ${TARGET_NAME})
|
|
|
|
# onnxruntime (prebuilt and release)
|
|
include_directories(${ONNXRUNTIME_DIR}/include ${ONNXRUNTIME_DIR}/include/onnxruntime/core/session)
|
|
link_directories(${ONNXRUNTIME_DIR}/lib ${ONNXRUNTIME_DIR}/build/Linux/RelWithDebInfo/)
|
|
|
|
# add plugin source
|
|
file(GLOB_RECURSE BACKEND_OPS_SRCS *.cpp)
|
|
|
|
set(INFER_PLUGIN_LIB onnxruntime)
|
|
|
|
add_library(${SHARED_TARGET} SHARED ${BACKEND_OPS_SRCS})
|
|
target_link_libraries(${SHARED_TARGET} ${INFER_PLUGIN_LIB})
|
|
target_include_directories(${SHARED_TARGET}
|
|
PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/common)
|