Introduce DeviceStreamManager

[alexxy/gromacs.git] / src / gromacs / gpu_utils / CMakeLists.txt

diff --git a/src/gromacs/gpu_utils/CMakeLists.txt b/src/gromacs/gpu_utils/CMakeLists.txt
index ce70d0b049adbb5cc8dc099d144ecfc1c5bb0e64..e8c02ccf31e2f28e4937f6a170ef5a2017d108cd 100644 (file)
--- a/src/gromacs/gpu_utils/CMakeLists.txt
+++ b/src/gromacs/gpu_utils/CMakeLists.txt
@@ -38,6 +38,7 @@
 
 gmx_add_libgromacs_sources(
         clfftinitializer.cpp
+        device_stream_manager.cpp
         hostallocator.cpp
         gpu_utils.cpp
         gpu_testutils.cpp
@@ -59,7 +60,10 @@ elseif(GMX_USE_CUDA)
         pinning.cu
         pmalloc_cuda.cu
         )
-elseif()
+    gmx_compile_cpp_as_cuda(
+        device_stream_manager.cpp
+        )
+else()
     gmx_add_libgromacs_sources(
         device_stream.cpp
     )