mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
25 lines
644 B
CMake
Executable File
25 lines
644 B
CMake
Executable File
# Copyright (c) OpenMMLab. All rights reserved.
|
|
cmake_minimum_required(VERSION 3.14)
|
|
|
|
# ncnn
|
|
find_package(ncnn)
|
|
|
|
if(ncnn_FOUND)
|
|
message(STATUS "ncnn library found!")
|
|
else()
|
|
message(FATAL_ERROR "Could not locate ncnn")
|
|
endif()
|
|
|
|
set_targets(mmdeploy_ncnn_ops NCNN_OPS_OBJ NCNN_OPS_STATIC NCNN_OPS_SHARED)
|
|
|
|
if(NOT ANDROID AND NOT IOS)
|
|
add_subdirectory(ops)
|
|
add_subdirectory(onnx2ncnn)
|
|
add_subdirectory(pyncnn_ext)
|
|
else()
|
|
# In case of embedded platform, like android, or ios, we only build custom ncnn
|
|
# ops, and leave the executable converter(onnx2ncnn, pyncnn_ext) built under
|
|
# the host platforms
|
|
add_subdirectory(ops)
|
|
endif()
|