Make DeviceContext into a proper 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 3816579d638a0ac5cc956ea54938726b5941501c..f07af017e33fdaba6b53382465b6da76d340a7a3 100644 (file)
--- a/src/gromacs/mdlib/settle_gpu.cuh
+++ b/src/gromacs/mdlib/settle_gpu.cuh
@@ -45,6 +45,7 @@
 
 #include "gmxpre.h"
 
+#include "gromacs/gpu_utils/device_context.h"
 #include "gromacs/gpu_utils/gputraits.cuh"
 #include "gromacs/math/functions.h"
 #include "gromacs/math/invertmatrix.h"
@@ -200,9 +201,10 @@ public:
      * \param[in] mtop           Topology of the system to gen the masses for O and H atoms and
      *                           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.
      */
-    SettleGpu(const gmx_mtop_t& mtop, CommandStream commandStream);
+    SettleGpu(const gmx_mtop_t& mtop, const DeviceContext& deviceContext, CommandStream commandStream);
 
     ~SettleGpu();
 
@@ -250,6 +252,8 @@ public:
     void set(const InteractionDefinitions& idef, const t_mdatoms& md);
 
 private:
+    //! Dummy GPU context object
+    const DeviceContext& deviceContext_;
     //! GPU stream
     CommandStream commandStream_;