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)