Make use of the DeviceStreamManager

[alexxy/gromacs.git] / src / gromacs / ewald / pme_only.h

diff --git a/src/gromacs/ewald/pme_only.h b/src/gromacs/ewald/pme_only.h
index 18edbb9b43ddf5bcdbe23127f58a2e79ece71027..1a71ea195c1b74ad44d168bd20753cfb0706e4d4 100644 (file)
--- a/src/gromacs/ewald/pme_only.h
+++ b/src/gromacs/ewald/pme_only.h
@@ -55,17 +55,20 @@ struct t_nrnb;
 struct gmx_pme_t;
 struct gmx_wallcycle;
 
-class DeviceContext;
 enum class PmeRunMode;
+namespace gmx
+{
+class DeviceStreamManager;
+}
 
 /*! \brief Called on the nodes that do PME exclusively */
-int gmx_pmeonly(gmx_pme_t*                pme,
-                const t_commrec*          cr,
-                t_nrnb*                   mynrnb,
-                gmx_wallcycle*            wcycle,
-                gmx_walltime_accounting_t walltime_accounting,
-                t_inputrec*               ir,
-                PmeRunMode                runMode,
-                const DeviceContext*      deviceContext);
+int gmx_pmeonly(gmx_pme_t*                      pme,
+                const t_commrec*                cr,
+                t_nrnb*                         mynrnb,
+                gmx_wallcycle*                  wcycle,
+                gmx_walltime_accounting_t       walltime_accounting,
+                t_inputrec*                     ir,
+                PmeRunMode                      runMode,
+                const gmx::DeviceStreamManager* deviceStreamManager);
 
 #endif