Merge pull request #1969 from jjotero/feat/hook-from-meta

[feat] Make hook-related machinery available as class directives

Author: Vasileios Karakasis

cscs-checks/apps/cp2k/cp2k_check.py

@@ -123,11 +123,11 @@ def __init__(self, scale, variant):
         self.reference = references[variant][scale]
         self.tags |= {'maintenance' if variant == 'maint' else 'production'}
 
-    @rfm.run_before('run')
+    @run_before('run')
     def set_task_distribution(self):
         self.job.options = ['--distribution=block:block']
 
-    @rfm.run_before('run')
+    @run_before('run')
     def set_cpu_binding(self):
         self.job.launcher.options = ['--cpu-bind=cores']
 

cscs-checks/apps/greasy/greasy_check.py

@@ -90,13 +90,13 @@ def __init__(self, variant, partition, num_greasy_tasks, nworkes_per_node,
         self.maintainers = ['VH', 'SK']
         self.tags = {'production'}
 
-    @rfm.run_before('run')
+    @run_before('run')
     def generate_tasks_file(self):
         with open(os.path.join(self.stagedir, self.tasks_file), 'w') as fp:
             for i in range(self.num_greasy_tasks):
                 fp.write(f'./{self.executable} output-{i}\n')
 
-    @rfm.run_before('run')
+    @run_before('run')
     def daint_dom_gpu_specific_workaround(self):
         if self.current_partition.fullname in ['daint:gpu', 'dom:gpu']:
             self.variables['CRAY_CUDA_MPS'] = '1'
@@ -121,13 +121,13 @@ def daint_dom_gpu_specific_workaround(self):
                     }
                 }
 
-    @rfm.run_before('run')
+    @run_before('run')
     def change_executable_name(self):
         # After compiling the code we can change the executable to be
         # the greasy one
         self.executable = 'greasy'
 
-    @rfm.run_before('run')
+    @run_before('run')
     def set_launcher(self):
         # The job launcher has to be changed to local since greasy
         # make calls to srun

cscs-checks/apps/jupyter/check_ipcmagic.py

@@ -58,7 +58,7 @@ def __init__(self):
         self.maintainers = ['RS', 'TR']
         self.tags = {'production'}
 
-    @rfm.run_before('run')
+    @run_before('run')
     def prepare_run(self):
         # Change the job launcher since `ipython`
         # needs to be launched without `srun`.

cscs-checks/apps/paraview/paraview_check.py

@@ -26,7 +26,7 @@ def __init__(self):
         self.maintainers = ['JF', 'TM']
         self.tags = {'scs', 'production'}
 
-    @rfm.run_before('sanity')
+    @run_before('sanity')
     def set_sanity(self):
         if self.current_partition.name == 'mc':
             self.sanity_patterns = sn.all([

cscs-checks/apps/quantumespresso/quantumespresso_check.py

@@ -123,11 +123,11 @@ def __init__(self):
             'maintenance' if self.variant == 'maint' else 'production'
         }
 
-    @rfm.run_before('run')
+    @run_before('run')
     def set_task_distribution(self):
         self.job.options = ['--distribution=block:block']
 
-    @rfm.run_before('run')
+    @run_before('run')
     def set_cpu_binding(self):
         self.job.launcher.options = ['--cpu-bind=cores']
 

cscs-checks/apps/spark/spark_check.py

@@ -30,7 +30,7 @@ def __init__(self):
         self.maintainers = ['TM', 'RS']
         self.tags = {'production'}
 
-    @rfm.run_before('run')
+    @run_before('run')
     def prepare_run(self):
         if self.current_partition.fullname in ['daint:gpu', 'dom:gpu']:
             num_workers = 12

cscs-checks/apps/vasp/vasp_check.py

@@ -82,11 +82,11 @@ def __init__(self, variant):
         self.reference = references[variant]
         self.tags |= {'maintenance' if variant == 'maint' else 'production'}
 
-    @rfm.run_before('run')
+    @run_before('run')
     def set_task_distribution(self):
         self.job.options = ['--distribution=block:block']
 
-    @rfm.run_before('run')
+    @run_before('run')
     def set_cpu_binding(self):
         self.job.launcher.options = ['--cpu-bind=cores']
 

cscs-checks/compile/haswell_fma_check.py

@@ -31,7 +31,7 @@ def __init__(self):
         self.maintainers = ['AJ', 'CB']
         self.tags = {'production', 'craype'}
 
-    @rfm.run_before('compile')
+    @run_before('compile')
     def setflags(self):
         if self.current_environ.name == 'PrgEnv-cray':
             self.build_system.cflags = ['-Ofast', '-S']

cscs-checks/compile/libsci_resolve.py

@@ -40,7 +40,7 @@ def __init__(self, module_name):
         }
         self.postbuild_cmds = [f'readelf -d {self.executable}']
 
-    @rfm.run_before('sanity')
+    @run_before('sanity')
     def set_sanity(self):
         # here lib_name is in the format: libsci_acc_gnu_48_nv35.so or
         #                                 libsci_acc_cray_nv35.so

cscs-checks/libraries/gridtools/gridtools_tests.py

@@ -131,7 +131,7 @@ def __init__(self, variant):
         self.tags = {'scs', 'benchmark'}
         self.maintainers = ['CB']
 
-    @rfm.require_deps
+    @require_deps
     def set_executable(self, GridToolsCPUBuildCheck):
         self.executable = os.path.join(
             GridToolsCPUBuildCheck().stagedir,
@@ -176,7 +176,7 @@ def __init__(self, variant):
         self.tags = {'scs', 'benchmark'}
         self.maintainers = ['CB']
 
-    @rfm.require_deps
+    @require_deps
     def set_executable(self, GridToolsGPUBuildCheck):
         self.executable = os.path.join(
             GridToolsGPUBuildCheck().stagedir,