cmake_minimum_required (VERSION 3.10)
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)