Make DeviceStream into a class

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

diff --git a/src/gromacs/mdlib/leapfrog_gpu.cuh b/src/gromacs/mdlib/leapfrog_gpu.cuh
index 26a6fc7399a4c373dcacbcad524f2198d8e91150..6097a9a87e0abd6244248c8d87a202cb2d08a4d6 100644 (file)
--- a/src/gromacs/mdlib/leapfrog_gpu.cuh
+++ b/src/gromacs/mdlib/leapfrog_gpu.cuh
@@ -64,9 +64,9 @@ public:
     /*! \brief Constructor.
      *
      * \param[in] deviceContext  Device context (dummy in CUDA).
-     * \param[in] commandStream  Device command stream to use.
+     * \param[in] deviceStream   Device stream to use.
      */
-    LeapFrogGpu(const DeviceContext& deviceContext, CommandStream commandStream);
+    LeapFrogGpu(const DeviceContext& deviceContext, const DeviceStream& deviceStream);
     ~LeapFrogGpu();
 
     /*! \brief Integrate
@@ -115,7 +115,7 @@ private:
     //! GPU context object
     const DeviceContext& deviceContext_;
     //! GPU stream
-    CommandStream commandStream_;
+    const DeviceStream& deviceStream_;
     //! GPU kernel launch config
     KernelLaunchConfig kernelLaunchConfig_;
     //! Number of atoms