mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
5f59b4464f
* docs(docs/zh_cn): add doc and link checker * docs(REAME): update * docs(docs/zh_cn): update * docs(benchmark): update table * docs(zh_cn/benchmark): update link * CI(docs): update link check * ci(doc): update checker * docs(zh_cn): update * style(ci): remove useless para * style(ci): update * docs(zh_cn): update * docs(benchmark.md): fix mobilnet link error * docs(docs/zh_cn): add doc and link checker * docs(REAME): update * docs(docs/zh_cn): update * docs(benchmark): update table * docs(zh_cn/benchmark): update link * CI(docs): update link check * ci(doc): update checker * docs(zh_cn): update * style(ci): remove useless para * style(ci): update * docs(zh_cn): update * docs(benchmark.md): fix mobilnet link error * docs(zh_cn/do_regression_test.md): rebase * docs(docs/zh_cn): add doc and link checker * Update README_zh-CN.md * Update README_zh-CN.md * Update index.rst * Update check-doc-link.yml * [Fix] Fix ci (#426) * fix ci * add nvidia key * remote torch * recover pytorch * ci(codecov): ignore ci * docs(zh_cn): add get_started.md * docs(zh_cn): fix review advice * docs(readthedocs): update * docs(zh_CN): update * docs(zh_CN): revert * fix(docs): review advices * fix(docs): review advices * fix(docs): review Co-authored-by: q.yao <streetyao@live.com>
39 lines
1.0 KiB
CMake
39 lines
1.0 KiB
CMake
if (NOT MSVC)
|
|
set(CMAKE_CXX_STANDARD 14)
|
|
endif ()
|
|
set(CMAKE_CXX_FLAGS_RELEASE "-O3")
|
|
|
|
# build ONNXRUNTIME ops
|
|
if ("ort" IN_LIST MMDEPLOY_TARGET_BACKENDS)
|
|
if (NOT DEFINED ONNXRUNTIME_DIR)
|
|
set(ONNXRUNTIME_DIR $ENV{ONNXRUNTIME_DIR})
|
|
endif ()
|
|
if (NOT ONNXRUNTIME_DIR)
|
|
message(FATAL_ERROR " ONNXRUNTIME_DIR is not found.")
|
|
else ()
|
|
message(STATUS "Build ONNXRUNTIME custom ops.")
|
|
add_subdirectory(onnxruntime)
|
|
endif ()
|
|
endif ()
|
|
|
|
# build TensorRT ops
|
|
if ("trt" IN_LIST MMDEPLOY_TARGET_BACKENDS)
|
|
if (NOT DEFINED TENSORRT_DIR)
|
|
set(TENSORRT_DIR $ENV{TENSORRT_DIR})
|
|
endif ()
|
|
message(STATUS "Build TensorRT custom ops.")
|
|
add_subdirectory(tensorrt)
|
|
endif ()
|
|
|
|
# build ncnn ops
|
|
if ("ncnn" IN_LIST MMDEPLOY_TARGET_BACKENDS)
|
|
message(STATUS "Build ncnn custom ops")
|
|
add_subdirectory(ncnn)
|
|
endif ()
|
|
|
|
# build TorchScript ops
|
|
if ("torchscript" IN_LIST MMDEPLOY_TARGET_BACKENDS)
|
|
message(STATUS "Build torchsciprt custom ops")
|
|
add_subdirectory(torchscript)
|
|
endif ()
|