mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
6fdf6b8616
* apply cmake-format and refactor cmake scripts of building onnx and tensorrt ops * add static target * suppress 'CMAKE_CUDA_ARCHITECTURES' warning when cmake with version 18 or later is used * fix typo
44 lines
932 B
CMake
44 lines
932 B
CMake
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)
|