Make DeviceContext into a proper 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 ba6d1be3f209437744959e4fbe18de8ddc4bb649..98703c05b94121b8dcaaff96910d6e8b0451a909 100644 (file)
--- a/src/gromacs/mdlib/leapfrog_gpu.cuh
+++ b/src/gromacs/mdlib/leapfrog_gpu.cuh
@@ -44,6 +44,7 @@
 #ifndef GMX_MDLIB_LEAPFROG_GPU_CUH
 #define GMX_MDLIB_LEAPFROG_GPU_CUH
 
+#include "gromacs/gpu_utils/device_context.h"
 #include "gromacs/gpu_utils/gputraits.cuh"
 #include "gromacs/gpu_utils/hostallocator.h"
 #include "gromacs/mdtypes/group.h"
@@ -62,9 +63,10 @@ class LeapFrogGpu
 public:
     /*! \brief Constructor.
      *
+     * \param[in] deviceContext  Device context (dummy in CUDA).
      * \param[in] commandStream  Device command stream to use.
      */
-    LeapFrogGpu(CommandStream commandStream);
+    LeapFrogGpu(const DeviceContext& deviceContext, CommandStream commandStream);
     ~LeapFrogGpu();
 
     /*! \brief Integrate
@@ -110,6 +112,8 @@ public:
     class Impl;
 
 private:
+    //! Dummy GPU context object
+    const DeviceContext& deviceContext_;
     //! GPU stream
     CommandStream commandStream_;
     //! GPU kernel launch config