mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
93408c7cd8
add onnxruntime plugin support, add more custom ops update function_rewriter, enable use origin function add setup.cfg update setup.cfg
19 lines
509 B
CMake
19 lines
509 B
CMake
add_definitions(-std=c++11)
|
|
set(CMAKE_CXX_FLAGS_RELEASE "-O3")
|
|
|
|
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)
|
|
|
|
# build TensorRT ops
|
|
if (BUILD_ONNXRUNTIME_OPS)
|
|
message("Build ONNXRUNTIME custom ops.")
|
|
add_subdirectory (onnxruntime)
|
|
endif()
|
|
|
|
# build TensorRT ops
|
|
if (BUILD_TENSORRT_OPS)
|
|
message("Build TensorRT custom ops.")
|
|
add_subdirectory (tensorrt)
|
|
endif()
|