mmdeploy/CMakeLists.txt

cmake_minimum_required(VERSION 3.14)
project(mmdeploy_backend_ops)

# ONNXRUNTIME config

# enable onnxruntime
option(BUILD_ONNXRUNTIME_OPS "enable ONNXRUNTIME ops" OFF)
# ONNXRUNTIME search path
if(BUILD_ONNXRUNTIME_OPS)
  if(NOT DEFINED ONNXRUNTIME_DIR)
    set(ONNXRUNTIME_DIR $ENV{ONNXRUNTIME_DIR})
  endif()
  if(NOT ONNXRUNTIME_DIR)
    message(ERROR " ONNXRUNTIME_DIR is not found.")
  endif()
endif()

# TensorRT config

# enable tensorrt
option(BUILD_TENSORRT_OPS "enable TensorRT ops" OFF)
# TensorRT search path
if(BUILD_TENSORRT_OPS)
  if(NOT DEFINED TENSORRT_DIR)
    set(TENSORRT_DIR $ENV{TENSORRT_DIR})
  endif()
endif()

# NCNN config

# enable ncnn
option(BUILD_NCNN_OPS "enable NCNN ops" OFF)
# NCNN search path
if(BUILD_NCNN_OPS)
  if(NOT DEFINED NCNN_DIR)
    set(NCNN_DIR $ENV{NCNN_DIR})
  endif()
  if(NOT NCNN_DIR)
    message(ERROR " NCNN_DIR is not found.")
  endif()
endif()

add_subdirectory(backend_ops)