Merge pull request #137 from EESSI/fix_compact_process_binding

Fix compact process binding for OpenMPI mpirun
EESSI · May 5, 2024 · 0e37c60 · 0e37c60
2 parents 2dfe951 + 6ecad3a
commit 0e37c60
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/eessi/testsuite/hooks.py b/eessi/testsuite/hooks.py
@@ -436,7 +436,7 @@ def set_compact_process_binding(test: rfm.RegressionTest):
     # Other launchers may or may not do the correct binding
     test.env_vars['I_MPI_PIN_CELL'] = 'core'  # Don't bind to hyperthreads, only to physcial cores
     test.env_vars['I_MPI_PIN_DOMAIN'] = '%s:compact' % physical_cpus_per_task
-    test.env_vars['OMPI_MCA_rmaps_base_mapping_policy'] = 'node:PE=%s' % physical_cpus_per_task
+    test.env_vars['OMPI_MCA_rmaps_base_mapping_policy'] = 'slot:PE=%s' % physical_cpus_per_task
     # Default binding for SLURM. Only effective if the task/affinity plugin is enabled
     # and when number of tasks times cpus per task equals either socket, core or thread count
     test.env_vars['SLURM_CPU_BIND'] = 'verbose'