Make DeviceStream into a class

[alexxy/gromacs.git] / src / gromacs / mdlib / update_constrain_gpu.h

diff --git a/src/gromacs/mdlib/update_constrain_gpu.h b/src/gromacs/mdlib/update_constrain_gpu.h
index 61f8537efa24bc052851f016b86f8583bfb0a087..c0a5e9b21aa4c607361d1d4fabe081ef365b67cb 100644 (file)
--- a/src/gromacs/mdlib/update_constrain_gpu.h
+++ b/src/gromacs/mdlib/update_constrain_gpu.h
@@ -50,8 +50,8 @@
 #include "gromacs/utility/classhelpers.h"
 
 class DeviceContext;
+class DeviceStream;
 class GpuEventSynchronizer;
-
 struct gmx_mtop_t;
 enum class PbcType : int;
 class InteractionDefinitions;
@@ -68,7 +68,7 @@ class UpdateConstrainGpu
 public:
     /*! \brief Create Update-Constrain object.
      *
-     * The constructor is given a non-nullptr \p commandStream, in which all the update and constrain
+     * The constructor is given a non-nullptr \p deviceStream, in which all the update and constrain
      * routines are executed. \p xUpdatedOnDevice should mark the completion of all kernels that modify
      * coordinates. The event is maintained outside this class and also passed to all (if any) consumers
      * of the updated coordinates. The \p xUpdatedOnDevice also can not be a nullptr because the
@@ -79,13 +79,13 @@ public:
      * \param[in] mtop              Topology of the system: SETTLE gets the masses for O and H atoms
      *                              and target O-H and H-H distances from this object.
      * \param[in] deviceContext     GPU device context.
-     * \param[in] commandStream     GPU stream to use. Can be nullptr.
+     * \param[in] deviceStream      GPU stream to use.
      * \param[in] xUpdatedOnDevice  The event synchronizer to use to mark that update is done on the GPU.
      */
     UpdateConstrainGpu(const t_inputrec&     ir,
                        const gmx_mtop_t&     mtop,
                        const DeviceContext&  deviceContext,
-                       const void*           commandStream,
+                       const DeviceStream&   deviceStream,
                        GpuEventSynchronizer* xUpdatedOnDevice);
 
     ~UpdateConstrainGpu();