Make DeviceStream into a class

[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 8672e450ca175fe46608e8672013980c89808139..ce70d0b049adbb5cc8dc099d144ecfc1c5bb0e64 100644 (file)
--- a/src/gromacs/gpu_utils/CMakeLists.txt
+++ b/src/gromacs/gpu_utils/CMakeLists.txt
@@ -45,6 +45,7 @@ gmx_add_libgromacs_sources(
 if(GMX_USE_OPENCL)
     gmx_add_libgromacs_sources(
         device_context_ocl.cpp
+        device_stream_ocl.cpp
         gpu_utils_ocl.cpp
         ocl_compiler.cpp
         ocl_caching.cpp
@@ -53,10 +54,15 @@ if(GMX_USE_OPENCL)
 elseif(GMX_USE_CUDA)
     gmx_add_libgromacs_sources(
         cudautils.cu
+        device_stream.cu
         gpu_utils.cu
         pinning.cu
         pmalloc_cuda.cu
         )
+elseif()
+    gmx_add_libgromacs_sources(
+        device_stream.cpp
+    )
 endif()
 
 if (BUILD_TESTING)