use mixin class for gromacs

eessi/testsuite/tests/apps/gromacs.py

@@ -35,43 +35,44 @@
 from hpctestlib.sciapps.gromacs.benchmarks import gromacs_check
 
 from eessi.testsuite import hooks
-from eessi.testsuite.constants import SCALES, TAGS
+from eessi.testsuite.constants import COMPUTE_UNIT, DEVICE_TYPES, SCALES
+from eessi.testsuite.eessi_mixin import EESSI_Mixin
 from eessi.testsuite.utils import find_modules, log
 
 
+class EESSI_GROMACS_base(gromacs_check):
+    @run_after('init')
+    def set_device_type(self):
+        self.device_type = self.nb_impl
+
+
 @rfm.simple_test
-class EESSI_GROMACS(gromacs_check):
+class EESSI_GROMACS(EESSI_GROMACS_base, EESSI_Mixin):
     scale = parameter(SCALES.keys())
-    valid_prog_environs = ['default']
-    valid_systems = ['*']
     time_limit = '30m'
     module_name = parameter(find_modules('GROMACS'))
+    bench_name_ci = 'HECBioSim/Crambin'
 
-    @run_after('init')
-    def run_after_init(self):
-        """Hooks to run after the init phase"""
-
-        # Filter on which scales are supported by the partitions defined in the ReFrame configuration
-        hooks.filter_supported_scales(self)
+    def required_mem_per_node(self):
+        return self.num_tasks_per_node * 1024
 
-        # Make sure that GPU tests run in partitions that support running on a GPU,
-        # and that CPU-only tests run in partitions that support running CPU-only.
-        # Also support setting valid_systems on the cmd line.
-        hooks.filter_valid_systems_by_device_type(self, required_device_type=self.nb_impl)
-
-        # Support selecting modules on the cmd line.
-        hooks.set_modules(self)
-
-        # Support selecting scales on the cmd line via tags.
-        hooks.set_tag_scale(self)
+    def __init__(self):
+        # self.device_type must be set before the @run_after('init') hooks of the EESSI_Mixin class
+        self.device_type = self.nb_impl
 
     @run_after('init')
-    def set_tag_ci(self):
-        """Set tag CI on smallest benchmark, so it can be selected on the cmd line via --tag CI"""
-        # Crambin input is smallest input (20K atoms), cfr. https://www.hecbiosim.ac.uk/access-hpc/benchmarks
-        if self.benchmark_info[0] == 'HECBioSim/Crambin':
-            self.tags.add(TAGS['CI'])
-            log(f'tags set to {self.tags}')
+    def set_compute_unit(self):
+        """
+        Set the compute unit to which tasks will be assigned:
+        one task per CPU core for CPU runs, and one task per GPU for GPU runs.
+        """
+        if self.device_type == DEVICE_TYPES['CPU']:
+            self.compute_unit = COMPUTE_UNIT['CPU']
+        elif self.device_type == DEVICE_TYPES['GPU']:
+            self.compute_unit = COMPUTE_UNIT['GPU']
+        else:
+            msg = f"No mapping of device type {self.device_type} to a COMPUTE_UNIT was specified in this test"
+            raise NotImplementedError(msg)
 
     @run_after('setup')
     def set_executable_opts(self):
@@ -86,15 +87,6 @@ def set_executable_opts(self):
             self.executable_opts += ['-dlb', 'yes', '-npme', '-1']
             log(f'executable_opts set to {self.executable_opts}')
 
-    @run_after('setup')
-    def run_after_setup(self):
-        """Hooks to run after the setup phase"""
-
-        # Calculate default requested resources based on the scale:
-        # 1 task per CPU for CPU-only tests, 1 task per GPU for GPU tests.
-        # Also support setting the resources on the cmd line.
-        hooks.assign_tasks_per_compute_unit(test=self, compute_unit=self.nb_impl)
-
     @run_after('setup')
     def set_omp_num_threads(self):
         """
@@ -113,11 +105,3 @@ def set_omp_num_threads(self):
 
         self.env_vars['OMP_NUM_THREADS'] = omp_num_threads
         log(f'env_vars set to {self.env_vars}')
-
-    @run_after('setup')
-    def set_binding_policy(self):
-        """
-        Default process binding may depend on the launcher used. We've seen some variable performance.
-        Better set it explicitely to make sure process migration cannot cause such variations.
-        """
-        hooks.set_compact_process_binding(self)