mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
36124f6205
* check in cmake * move backend_ops to csrc/backend_ops * check in preprocess, model, some codebase and their c-apis * check in CMakeLists.txt * check in parts of test_csrc * commit everything else * add readme * update core's BUILD_INTERFACE directory * skip codespell on third_party * update trt_net and ort_net's CMakeLists * ignore clion's build directory * check in pybind11 * add onnx.proto. Remove MMDeploy's dependency on ncnn's source code * export MMDeployTargets only when MMDEPLOY_BUILD_SDK is ON * remove useless message * target include directory is wrong * change target name from mmdeploy_ppl_net to mmdeploy_pplnn_net * skip install directory * update project's cmake * remove useless code * set CMAKE_BUILD_TYPE to Release by force if it isn't set by user * update custom ops CMakeLists * pass object target's source lists * fix lint end-of-file * fix lint: trailing whitespace * fix codespell hook * remove bicubic_interpolate to csrc/backend_ops/ * set MMDEPLOY_BUILD_SDK OFF * change custom ops build command * add spdlog installation command * update docs on how to checkout pybind11 * move bicubic_interpolate to backend_ops/tensorrt directory * remove useless code * correct cmake * fix typo * fix typo * fix install directory * correct sdk's readme * set cub dir when cuda version < 11.0 * change directory where clang-format will apply to * fix build command * add .clang-format * change clang-format style from google to file * reformat csrc/backend_ops * format sdk's code * turn off clang-format for some files * add -Xcompiler=-fno-gnu-unique * fix trt topk initialize * check in config for sdk demo * update cmake script and csrc's readme * correct config's path * add cuda include directory, otherwise compile failed in case of tensorrt8.2 * clang-format onnx2ncnn.cpp Co-authored-by: zhangli <lzhang329@gmail.com> Co-authored-by: grimoire <yaoqian@sensetime.com>
85 lines
3.4 KiB
CMake
85 lines
3.4 KiB
CMake
# Copyright (c) OpenMMLab. All rights reserved.
|
|
if (NOT DEFINED CMAKE_INSTALL_PREFIX)
|
|
set(CMAKE_INSTALL_PREFIX "${CMAKE_BINARY_DIR}/install" CACHE PATH "MMDeploy's installation directory")
|
|
endif ()
|
|
message(STATUS "CMAKE_INSTALL_PREFIX: ${CMAKE_INSTALL_PREFIX}")
|
|
|
|
cmake_minimum_required(VERSION 3.14)
|
|
project(MMDeploy VERSION 0.1.0)
|
|
|
|
set(CMAKE_CXX_STANDARD 17)
|
|
|
|
set(CMAKE_ARCHIVE_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/lib)
|
|
set(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/lib)
|
|
set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/bin)
|
|
|
|
# options
|
|
option(MMDEPLOY_BUILD_MODEL_OPTIMIZER "build MMDeploy's model optimizer" ON)
|
|
option(MMDEPLOY_BUILD_SDK "build MMDeploy's SDK" OFF)
|
|
option(MMDEPLOY_ZIP_MODEL "support sdk model in zip format" OFF)
|
|
option(MMDEPLOY_BUILD_TEST "build MMDeploy's csrc's unittest" OFF)
|
|
set(MMDEPLOY_TARGET_DEVICES
|
|
"cpu" CACHE STRING "MMDeploy's target devices")
|
|
set(MMDEPLOY_TARGET_BACKENDS "" CACHE STRING "MMDeploy's target inference engines")
|
|
set(MMDEPLOY_CODEBASES "" CACHE STRING "select OpenMMLab's codebases")
|
|
#set(MMDEPLOY_TRANSFORM_PLATFORMS "cpu" CACHE STRING "select transform operations' platform")
|
|
|
|
if (NOT CMAKE_BUILD_TYPE)
|
|
set(CMAKE_BUILD_TYPE Release CACHE STRING "choose 'Release' as default build type" FORCE)
|
|
endif ()
|
|
|
|
include(${CMAKE_SOURCE_DIR}/cmake/common.cmake)
|
|
# set INTERFACE target to gather linked modules
|
|
add_library(MMDeployStaticModules INTERFACE)
|
|
add_library(MMDeployDynamicModules INTERFACE)
|
|
|
|
add_subdirectory(csrc)
|
|
if (MMDEPLOY_BUILD_TEST)
|
|
add_subdirectory(tests/test_csrc)
|
|
endif ()
|
|
|
|
if (MMDEPLOY_BUILD_SDK)
|
|
# get static modules and dynamic modules from MMDeployStaticModules and MMDeployDynamicModules, respectively
|
|
set(STATIC_MODULES)
|
|
get_target_property(STATIC_MODULES MMDeployStaticModules INTERFACE_LINK_LIBRARIES)
|
|
get_target_list("${STATIC_MODULES}" FILTERED_MODULES)
|
|
set(MMDEPLOY_STATIC_MODULES "${FILTERED_MODULES}" CACHE STRING "MMDeploy's static modules")
|
|
message(STATUS "MMDEPLOY_STATIC_MODULES: ${MMDEPLOY_STATIC_MODULES}")
|
|
|
|
set(DYNAMIC_MODULES)
|
|
get_target_property(DYNAMIC_MODULES MMDeployDynamicModules INTERFACE_LINK_LIBRARIES)
|
|
get_target_list("${DYNAMIC_MODULES}" FILTERED_MODULES)
|
|
set(MMDEPLOY_DYNAMIC_MODULES "${FILTERED_MODULES}" CACHE STRING "MMDeploy's dynamic modules")
|
|
message(STATUS "MMDEPLOY_DYNAMIC_MODULES: ${MMDEPLOY_DYNAMIC_MODULES}")
|
|
|
|
# export MMDeploy package
|
|
install(EXPORT MMDeployTargets
|
|
# NAMESPACE mmdeploy::
|
|
FILE MMDeployTargets.cmake
|
|
#EXPORT_LINK_INTERFACE_LIBRARIES
|
|
DESTINATION lib/cmake/MMDeploy)
|
|
|
|
include(CMakePackageConfigHelpers)
|
|
# generate the config file that is includes the exports
|
|
configure_package_config_file(${CMAKE_SOURCE_DIR}/cmake/MMDeployConfig.cmake.in
|
|
"${CMAKE_CURRENT_BINARY_DIR}/MMDeployConfig.cmake"
|
|
INSTALL_DESTINATION "lib/cmake"
|
|
NO_SET_AND_CHECK_MACRO
|
|
NO_CHECK_REQUIRED_COMPONENTS_MACRO
|
|
)
|
|
|
|
write_basic_package_version_file(
|
|
"${CMAKE_CURRENT_BINARY_DIR}/MMDeployConfigVersion.cmake"
|
|
VERSION "${MMDeploy_VERSION_MAJOR}.${MMDeploy_VERSION_MINOR}"
|
|
COMPATIBILITY AnyNewerVersion
|
|
)
|
|
|
|
install(FILES
|
|
${CMAKE_CURRENT_BINARY_DIR}/MMDeployConfig.cmake
|
|
${CMAKE_CURRENT_BINARY_DIR}/MMDeployConfigVersion.cmake
|
|
DESTINATION lib/cmake/MMDeploy
|
|
)
|
|
|
|
install(DIRECTORY ${CMAKE_SOURCE_DIR}/demo/csrc/ DESTINATION example)
|
|
endif ()
|