From 6eb2e890160a6462f7a45cd72dc3e8c0a9cb40e9 Mon Sep 17 00:00:00 2001
From: grimoire <yaoqian@sensetime.com>
Date: Mon, 21 Jun 2021 10:34:48 +0800
Subject: [PATCH] update cmakelists

---
 backend_ops/onnxruntime/CMakeLists.txt             | 4 +++-
 backend_ops/onnxruntime/grid_sample/CMakeLists.txt | 4 ----
 backend_ops/onnxruntime/roi_align/CMakeLists.txt   | 4 ----
 backend_ops/tensorrt/CMakeLists.txt                | 8 ++++----
 backend_ops/tensorrt/common_impl/CMakeLists.txt    | 4 ----
 backend_ops/tensorrt/nms/CMakeLists.txt            | 4 ----
 backend_ops/tensorrt/roi_align/CMakeLists.txt      | 4 ----
 backend_ops/tensorrt/scatternd/CMakeLists.txt      | 4 ----
 8 files changed, 7 insertions(+), 29 deletions(-)
 delete mode 100644 backend_ops/onnxruntime/grid_sample/CMakeLists.txt
 delete mode 100644 backend_ops/onnxruntime/roi_align/CMakeLists.txt
 delete mode 100644 backend_ops/tensorrt/common_impl/CMakeLists.txt
 delete mode 100644 backend_ops/tensorrt/nms/CMakeLists.txt
 delete mode 100644 backend_ops/tensorrt/roi_align/CMakeLists.txt
 delete mode 100644 backend_ops/tensorrt/scatternd/CMakeLists.txt

diff --git a/backend_ops/onnxruntime/CMakeLists.txt b/backend_ops/onnxruntime/CMakeLists.txt
index 02e61a78f..588fc7f47 100644
--- a/backend_ops/onnxruntime/CMakeLists.txt
+++ b/backend_ops/onnxruntime/CMakeLists.txt
@@ -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)
diff --git a/backend_ops/onnxruntime/grid_sample/CMakeLists.txt b/backend_ops/onnxruntime/grid_sample/CMakeLists.txt
deleted file mode 100644
index c3eba9a06..000000000
--- a/backend_ops/onnxruntime/grid_sample/CMakeLists.txt
+++ /dev/null
@@ -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)
diff --git a/backend_ops/onnxruntime/roi_align/CMakeLists.txt b/backend_ops/onnxruntime/roi_align/CMakeLists.txt
deleted file mode 100644
index c3eba9a06..000000000
--- a/backend_ops/onnxruntime/roi_align/CMakeLists.txt
+++ /dev/null
@@ -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)
diff --git a/backend_ops/tensorrt/CMakeLists.txt b/backend_ops/tensorrt/CMakeLists.txt
index 393674f28..eb32ede00 100644
--- a/backend_ops/tensorrt/CMakeLists.txt
+++ b/backend_ops/tensorrt/CMakeLists.txt
@@ -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})
 
diff --git a/backend_ops/tensorrt/common_impl/CMakeLists.txt b/backend_ops/tensorrt/common_impl/CMakeLists.txt
deleted file mode 100644
index c3eba9a06..000000000
--- a/backend_ops/tensorrt/common_impl/CMakeLists.txt
+++ /dev/null
@@ -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)
diff --git a/backend_ops/tensorrt/nms/CMakeLists.txt b/backend_ops/tensorrt/nms/CMakeLists.txt
deleted file mode 100644
index c3eba9a06..000000000
--- a/backend_ops/tensorrt/nms/CMakeLists.txt
+++ /dev/null
@@ -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)
diff --git a/backend_ops/tensorrt/roi_align/CMakeLists.txt b/backend_ops/tensorrt/roi_align/CMakeLists.txt
deleted file mode 100644
index c3eba9a06..000000000
--- a/backend_ops/tensorrt/roi_align/CMakeLists.txt
+++ /dev/null
@@ -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)
diff --git a/backend_ops/tensorrt/scatternd/CMakeLists.txt b/backend_ops/tensorrt/scatternd/CMakeLists.txt
deleted file mode 100644
index c3eba9a06..000000000
--- a/backend_ops/tensorrt/scatternd/CMakeLists.txt
+++ /dev/null
@@ -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)