Make DeviceStream into a class

[alexxy/gromacs.git] / src / gromacs / mdlib / settle_gpu.cuh

diff --git a/src/gromacs/mdlib/settle_gpu.cuh b/src/gromacs/mdlib/settle_gpu.cuh
index da8bafd8dfbf45ae0dcbcb13874525fa483da6a4..24584f7a4bb0065f2de54756cad0a347e9da73e8 100644 (file)
--- a/src/gromacs/mdlib/settle_gpu.cuh
+++ b/src/gromacs/mdlib/settle_gpu.cuh
@@ -202,9 +202,9 @@ public:
      *                           target O-H and H-H distances. These values are also checked for
      *                           consistency.
      * \param[in] deviceContext  Device context (dummy in CUDA).
-     * \param[in] commandStream  Device stream to use.
+     * \param[in] deviceStream   Device stream to use.
      */
-    SettleGpu(const gmx_mtop_t& mtop, const DeviceContext& deviceContext, CommandStream commandStream);
+    SettleGpu(const gmx_mtop_t& mtop, const DeviceContext& deviceContext, const DeviceStream& deviceStream);
 
     ~SettleGpu();
 
@@ -255,7 +255,7 @@ private:
     //! GPU context object
     const DeviceContext& deviceContext_;
     //! GPU stream
-    CommandStream commandStream_;
+    const DeviceStream& deviceStream_;
 
     //! Scaled virial tensor (9 reals, GPU)
     std::vector<float> h_virialScaled_;