corrected dynamic load balancing when sharing GPUs

[alexxy/gromacs.git] / src / kernel / runner.c

diff --git a/src/kernel/runner.c b/src/kernel/runner.c
index 42ef770eda2eb4d88ce4ce3f2c2a97fb661e98c2..17b5f351ad044b5756ad317285078c0d1c55671c 100644 (file)
--- a/src/kernel/runner.c
+++ b/src/kernel/runner.c
@@ -1485,6 +1485,12 @@ int mdrunner(gmx_hw_opt_t *hw_opt,
     /* check consistency of CPU acceleration and number of GPUs selected */
     gmx_check_hw_runconf_consistency(fplog, hwinfo, cr, hw_opt, bUseGPU);
 
+    if (DOMAINDECOMP(cr))
+    {
+        /* When we share GPUs over ranks, we need to know this for the DLB */
+        dd_setup_dlb_resource_sharing(cr, hwinfo, hw_opt);
+    }
+
     /* getting number of PP/PME threads
        PME: env variable should be read only on one node to make sure it is
        identical everywhere;