Switch the GPU buffer ops on when update is on GPU

[alexxy/gromacs.git] / src / gromacs / taskassignment / decidesimulationworkload.cpp

diff --git a/src/gromacs/taskassignment/decidesimulationworkload.cpp b/src/gromacs/taskassignment/decidesimulationworkload.cpp
index 597cf1a81580119c6a48907574c9aa9f43661953..ea82f95f42bccd1eef0cef622a3d343a0204ec28 100644 (file)
--- a/src/gromacs/taskassignment/decidesimulationworkload.cpp
+++ b/src/gromacs/taskassignment/decidesimulationworkload.cpp
@@ -53,7 +53,7 @@ namespace gmx
 SimulationWorkload createSimulationWorkload(bool       useGpuForNonbonded,
                                             PmeRunMode pmeRunMode,
                                             bool       useGpuForBonded,
-                                            bool       useGpuForUpdateConstraints,
+                                            bool       useGpuForUpdate,
                                             bool       useGpuForBufferOps,
                                             bool       useGpuHaloExchange,
                                             bool       useGpuPmePpComm)
@@ -65,8 +65,8 @@ SimulationWorkload createSimulationWorkload(bool       useGpuForNonbonded,
     simulationWorkload.useGpuPme                 = (pmeRunMode == PmeRunMode::GPU || pmeRunMode == PmeRunMode::Mixed);
     simulationWorkload.useGpuPmeFft              = (pmeRunMode == PmeRunMode::Mixed);
     simulationWorkload.useGpuBonded              = useGpuForBonded;
-    simulationWorkload.useGpuUpdate              = useGpuForUpdateConstraints;
-    simulationWorkload.useGpuBufferOps           = useGpuForBufferOps;
+    simulationWorkload.useGpuUpdate              = useGpuForUpdate;
+    simulationWorkload.useGpuBufferOps           = useGpuForBufferOps || useGpuForUpdate;
     simulationWorkload.useGpuHaloExchange        = useGpuHaloExchange;
     simulationWorkload.useGpuPmePpCommunication  = useGpuPmePpComm;
     simulationWorkload.useGpuDirectCommunication = useGpuHaloExchange || useGpuPmePpComm;