From 75ad55095689f524947fbfac2251201abd80c37c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Micka=C3=ABl=20Seznec?= Date: Wed, 24 Feb 2021 16:30:53 +0100 Subject: [PATCH] Use right CMake variable for cu files in GroupNorm MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit In groupNormalizationPlugin, the .cu files are not added to PLUGIN_CU_SOURCES, contrary to other plugins. This is problematic because the NVCC flags are not set correctly when compiling those .cu files, e.g. GENCODES. While this is not a direct issue for this plugin in particular, it should be fixed as the documentation refers to it for building new plugings. Signed-off-by: Mickaël Seznec --- plugin/groupNormalizationPlugin/CMakeLists.txt | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/plugin/groupNormalizationPlugin/CMakeLists.txt b/plugin/groupNormalizationPlugin/CMakeLists.txt index 23b6d1b06..53b70a7ec 100644 --- a/plugin/groupNormalizationPlugin/CMakeLists.txt +++ b/plugin/groupNormalizationPlugin/CMakeLists.txt @@ -13,6 +13,9 @@ # See the License for the specific language governing permissions and # limitations under the License. # -file(GLOB SRCS *.cpp *.cu) +file(GLOB SRCS *.cpp) set(PLUGIN_SOURCES ${PLUGIN_SOURCES} ${SRCS}) set(PLUGIN_SOURCES ${PLUGIN_SOURCES} PARENT_SCOPE) +file(GLOB CU_SRCS *.cu) +set(PLUGIN_CU_SOURCES ${PLUGIN_CU_SOURCES} ${CU_SRCS}) +set(PLUGIN_CU_SOURCES ${PLUGIN_CU_SOURCES} PARENT_SCOPE)