Set up workload data structures

[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 b7a13fcae28d3f42e10a55e3c2d5a990244cf28f..8fe7b490214c2947365c499387e14364e9af6f01 100644 (file)
--- a/src/gromacs/listed_forces/gpubonded.h
+++ b/src/gromacs/listed_forces/gpubonded.h
@@ -65,7 +65,7 @@ struct gmx_wallcycle;
 namespace gmx
 {
 
-class ForceFlags;
+class StepWorkload;
 
 /*! \brief The number on bonded function types supported on GPUs */
 static constexpr int numFTypesOnGpu = 8;
@@ -137,9 +137,9 @@ class GpuBonded
          * assigned to the GPU */
         bool haveInteractions() const;
         /*! \brief Launches bonded kernel on a GPU */
-        void launchKernel(const t_forcerec      *fr,
-                          const gmx::ForceFlags &forceFlags,
-                          const matrix           box);
+        void launchKernel(const t_forcerec        *fr,
+                          const gmx::StepWorkload &stepWork,
+                          const matrix             box);
         /*! \brief Launches the transfer of computed bonded energies. */
         void launchEnergyTransfer();
         /*! \brief Waits on the energy transfer, and accumulates bonded energies to \c enerd. */