Make use of the DeviceStreamManager

[alexxy/gromacs.git] / src / gromacs / ewald / pme_gpu_types_host_impl.h

diff --git a/src/gromacs/ewald/pme_gpu_types_host_impl.h b/src/gromacs/ewald/pme_gpu_types_host_impl.h
index a019a7c0310b599ba6dda7c2e4fac04b65355f43..e134d2c0a79d306d4a6793c24c5ba6bfe2767b63 100644 (file)
--- a/src/gromacs/ewald/pme_gpu_types_host_impl.h
+++ b/src/gromacs/ewald/pme_gpu_types_host_impl.h
@@ -71,9 +71,14 @@ struct PmeGpuSpecific
 {
     /*! \brief Constructor
      *
-     * \param[in] deviceContext GPU device context.
+     * \param[in] deviceContext  GPU device context
+     * \param[in] pmeStream      GPU pme stream.
      */
-    PmeGpuSpecific(const DeviceContext& deviceContext) : deviceContext_(deviceContext) {}
+    PmeGpuSpecific(const DeviceContext& deviceContext, const DeviceStream& pmeStream) :
+        deviceContext_(deviceContext),
+        pmeStream_(pmeStream)
+    {
+    }
 
     /*! \brief
      * A handle to the GPU context.
@@ -84,7 +89,7 @@ struct PmeGpuSpecific
     const DeviceContext& deviceContext_;
 
     /*! \brief The GPU stream where everything related to the PME happens. */
-    DeviceStream pmeStream_;
+    const DeviceStream& pmeStream_;
 
     /* Synchronization events */
     /*! \brief Triggered after the PME Force Calculations have been completed */