mirrors
/
mmdeploy
mirror of https://github.com/open-mmlab/mmdeploy.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

update cmakelists

pull/12/head
grimoire 2021-06-21 10:34:48 +08:00
parent 2bd96243e2
commit 6eb2e89016
8 changed files with 7 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend_ops/onnxruntime/CMakeLists.txt
View File

@ -11,7 +11,9 @@ set(PLUGIN_LISTS grid_sample
roi_align)
foreach(PLUGIN_ITER ${PLUGIN_LISTS})
add_subdirectory(${PLUGIN_ITER})
file(GLOB PLUGIN_OPS_SRCS ${PLUGIN_ITER}/*.cpp ${PLUGIN_ITER}/*.cu)
file(GLOB PLUGIN_OPS_HEADS ${PLUGIN_ITER}/*.h ${PLUGIN_ITER}/*.hpp ${PLUGIN_ITER}/*.cuh)
set(BACKEND_OPS_SRCS ${BACKEND_OPS_SRCS} ${PLUGIN_OPS_SRCS} ${PLUGIN_OPS_HEADS})
endforeach(PLUGIN_ITER)
set(INFER_PLUGIN_LIB onnxruntime)

4
backend_ops/onnxruntime/grid_sample/CMakeLists.txt
View File

@ -1,4 +0,0 @@
file(GLOB OPS_SRCS *.cpp *.cu)
file(GLOB OPS_HEADS *.h *.hpp *.cuh)
set(BACKEND_OPS_SRCS ${BACKEND_OPS_SRCS} ${OPS_SRCS} ${OPS_HEADS})
set(BACKEND_OPS_SRCS ${BACKEND_OPS_SRCS} PARENT_SCOPE)

4
backend_ops/onnxruntime/roi_align/CMakeLists.txt
View File

@ -1,4 +0,0 @@
file(GLOB OPS_SRCS *.cpp *.cu)
file(GLOB OPS_HEADS *.h *.hpp *.cuh)
set(BACKEND_OPS_SRCS ${BACKEND_OPS_SRCS} ${OPS_SRCS} ${OPS_HEADS})
set(BACKEND_OPS_SRCS ${BACKEND_OPS_SRCS} PARENT_SCOPE)

8
backend_ops/tensorrt/CMakeLists.txt
View File

@ -41,19 +41,19 @@ if(NOT TENSORRT_FOUND)
endif()
INCLUDE_DIRECTORIES(${TENSORRT_INCLUDE_DIR})
# include common
add_subdirectory(common_impl)
# add plugin source
set(PLUGIN_LISTS scatternd
nms
roi_align)
foreach(PLUGIN_ITER ${PLUGIN_LISTS})
add_subdirectory(${PLUGIN_ITER})
file(GLOB PLUGIN_OPS_SRCS ${PLUGIN_ITER}/*.cpp ${PLUGIN_ITER}/*.cu)
file(GLOB PLUGIN_OPS_HEADS ${PLUGIN_ITER}/*.h ${PLUGIN_ITER}/*.hpp ${PLUGIN_ITER}/*.cuh)
set(BACKEND_OPS_SRCS ${BACKEND_OPS_SRCS} ${PLUGIN_OPS_SRCS} ${PLUGIN_OPS_HEADS})
endforeach(PLUGIN_ITER)
list(APPEND BACKEND_OPS_SRCS "${CMAKE_CURRENT_SOURCE_DIR}/trt_plugin.cpp")
list(APPEND BACKEND_OPS_SRCS "${CMAKE_CURRENT_SOURCE_DIR}/common_impl/trt_cuda_helper.cu")
set(INFER_PLUGIN_LIB ${TENSORRT_LIBRARY})

4
backend_ops/tensorrt/common_impl/CMakeLists.txt
View File

@ -1,4 +0,0 @@
file(GLOB OPS_SRCS *.cpp *.cu)
file(GLOB OPS_HEADS *.h *.hpp *.cuh)
set(BACKEND_OPS_SRCS ${BACKEND_OPS_SRCS} ${OPS_SRCS} ${OPS_HEADS})
set(BACKEND_OPS_SRCS ${BACKEND_OPS_SRCS} PARENT_SCOPE)

4
backend_ops/tensorrt/nms/CMakeLists.txt
View File

@ -1,4 +0,0 @@
file(GLOB OPS_SRCS *.cpp *.cu)
file(GLOB OPS_HEADS *.h *.hpp *.cuh)
set(BACKEND_OPS_SRCS ${BACKEND_OPS_SRCS} ${OPS_SRCS} ${OPS_HEADS})
set(BACKEND_OPS_SRCS ${BACKEND_OPS_SRCS} PARENT_SCOPE)

4
backend_ops/tensorrt/roi_align/CMakeLists.txt
View File

@ -1,4 +0,0 @@
file(GLOB OPS_SRCS *.cpp *.cu)
file(GLOB OPS_HEADS *.h *.hpp *.cuh)
set(BACKEND_OPS_SRCS ${BACKEND_OPS_SRCS} ${OPS_SRCS} ${OPS_HEADS})
set(BACKEND_OPS_SRCS ${BACKEND_OPS_SRCS} PARENT_SCOPE)

4
backend_ops/tensorrt/scatternd/CMakeLists.txt
View File

@ -1,4 +0,0 @@
file(GLOB OPS_SRCS *.cpp *.cu)
file(GLOB OPS_HEADS *.h *.hpp *.cuh)
set(BACKEND_OPS_SRCS ${BACKEND_OPS_SRCS} ${OPS_SRCS} ${OPS_HEADS})
set(BACKEND_OPS_SRCS ${BACKEND_OPS_SRCS} PARENT_SCOPE)