Set up workload data structures

[alexxy/gromacs.git] / src / gromacs / listed_forces / gpubondedkernels.cu

diff --git a/src/gromacs/listed_forces/gpubondedkernels.cu b/src/gromacs/listed_forces/gpubondedkernels.cu
index 709d99cc9afa892536ba32c8ab78ae40464e413e..e78b32c6fd66cf6a778a0ff271fd5624f6a2ff7a 100644 (file)
--- a/src/gromacs/listed_forces/gpubondedkernels.cu
+++ b/src/gromacs/listed_forces/gpubondedkernels.cu
@@ -57,8 +57,8 @@
 #include "gromacs/listed_forces/gpubonded.h"
 #include "gromacs/math/units.h"
 #include "gromacs/mdlib/force_flags.h"
-#include "gromacs/mdlib/ppforceworkload.h"
 #include "gromacs/mdtypes/forcerec.h"
+#include "gromacs/mdtypes/simulation_workload.h"
 #include "gromacs/pbcutil/pbc.h"
 #include "gromacs/pbcutil/pbc_aiuc_cuda.cuh"
 #include "gromacs/utility/gmxassert.h"
@@ -862,17 +862,17 @@ GpuBonded::Impl::launchKernel(const t_forcerec *fr,
 }
 
 void
-GpuBonded::launchKernel(const t_forcerec      *fr,
-                        const gmx::ForceFlags &forceFlags,
-                        const matrix           box)
+GpuBonded::launchKernel(const t_forcerec        *fr,
+                        const gmx::StepWorkload &stepWork,
+                        const matrix             box)
 {
-    if (forceFlags.computeEnergy)
+    if (stepWork.computeEnergy)
     {
         // When we need the energy, we also need the virial
         impl_->launchKernel<true, true>
             (fr, box);
     }
-    else if (forceFlags.computeVirial)
+    else if (stepWork.computeVirial)
     {
         impl_->launchKernel<true, false>
             (fr, box);