mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
31 lines
803 B
CMake
31 lines
803 B
CMake
|
set(CMAKE_CXX_STANDARD 14)
|
||
|
|
set(CMAKE_CXX_FLAGS_RELEASE "-O3")
|
||
|
|
|
||
|
|
# build ONNXRUNTIME ops
|
||
|
|
if ("ort" IN_LIST MMDEPLOY_TARGET_BACKENDS)
|
||
|
|
if (NOT DEFINED ONNXRUNTIME_DIR)
|
||
|
|
set(ONNXRUNTIME_DIR $ENV{ONNXRUNTIME_DIR})
|
||
|
|
endif ()
|
||
|
|
if (NOT ONNXRUNTIME_DIR)
|
||
|
|
message(FATAL_ERROR " ONNXRUNTIME_DIR is not found.")
|
||
|
|
else ()
|
||
|
|
message(STATUS "Build ONNXRUNTIME custom ops.")
|
||
|
|
add_subdirectory(onnxruntime)
|
||
|
|
endif ()
|
||
|
|
endif ()
|
||
|
|
|
||
|
|
# build TensorRT ops
|
||
|
|
if ("trt" IN_LIST MMDEPLOY_TARGET_BACKENDS)
|
||
|
|
if (NOT DEFINED TENSORRT_DIR)
|
||
|
|
set(TENSORRT_DIR $ENV{TENSORRT_DIR})
|
||
|
|
endif ()
|
||
|
|
message(STATUS "Build TensorRT custom ops.")
|
||
|
|
add_subdirectory(tensorrt)
|
||
|
|
endif ()
|
||
|
|
|
||
|
|
# build NCNN ops
|
||
|
|
if ("ncnn" IN_LIST MMDEPLOY_TARGET_BACKENDS)
|
||
|
|
message(STATUS "Build NCNN custom ops")
|
||
|
|
add_subdirectory(ncnn)
|
||
|
|
endif ()
|