Take over management of OpenCL context from PME and NBNXM

[alexxy/gromacs.git] / src / gromacs / listed_forces / gpubonded.h

diff --git a/src/gromacs/listed_forces/gpubonded.h b/src/gromacs/listed_forces/gpubonded.h
index 1a231d2c2cc83289379bed789dd9a0247e1cddfa..b1c69d45721fd00ec7aa8b67e45578e93a479fb9 100644 (file)
--- a/src/gromacs/listed_forces/gpubonded.h
+++ b/src/gromacs/listed_forces/gpubonded.h
@@ -55,6 +55,7 @@
 #include "gromacs/utility/arrayref.h"
 #include "gromacs/utility/classhelpers.h"
 
+class DeviceContext;
 struct gmx_enerdata_t;
 struct gmx_ffparams_t;
 struct gmx_mtop_t;
@@ -106,7 +107,10 @@ class GpuBonded
 {
 public:
     //! Construct the manager with constant data and the stream to use.
-    GpuBonded(const gmx_ffparams_t& ffparams, void* streamPtr, gmx_wallcycle* wcycle);
+    GpuBonded(const gmx_ffparams_t& ffparams,
+              const DeviceContext&  deviceContext,
+              void*                 streamPtr,
+              gmx_wallcycle*        wcycle);
     //! Destructor
     ~GpuBonded();