Skip to content

Commit

Permalink
Merge pull request #2350 from rsarm/update-syntax/cp2k
Browse files Browse the repository at this point in the history 
[test] Update syntax of CP2K check
  • Loading branch information
Vasileios Karakasis authored Dec 20, 2021
2 parents d4021cc + ea82578 commit c79d952
Showing 1 changed file with 100 additions and 95 deletions.
195 changes: 100 additions & 95 deletions cscs-checks/apps/cp2k/cp2k_check.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8,55 +8,88 @@


class Cp2kCheck(rfm.RunOnlyRegressionTest):
def __init__(self):
modules = ['CP2K']
executable = 'cp2k.psmp'
executable_opts = ['H2O-256.inp']
maintainers = ['LM']
tags = {'scs'}
strict_check = False
extra_resources = {
'switches': {
'num_switches': 1
}
}

@run_after('init')
def set_prgenv(self):
if self.current_system.name in ['eiger', 'pilatus']:
self.valid_prog_environs = ['cpeGNU']
else:
self.valid_prog_environs = ['builtin']

self.modules = ['CP2K']
self.executable = 'cp2k.psmp'
self.executable_opts = ['H2O-256.inp']

@sanity_function
def assert_energy_diff(self):
energy = sn.extractsingle(
r'\s+ENERGY\| Total FORCE_EVAL \( QS \) '
r'energy [\[\(]a\.u\.[\]\)]:\s+(?P<energy>\S+)',
self.stdout, 'energy', float, item=-1
)
energy_reference = -4404.2323
energy_diff = sn.abs(energy-energy_reference)
self.sanity_patterns = sn.all([
return sn.all([
sn.assert_found(r'PROGRAM STOPPED IN', self.stdout),
sn.assert_eq(sn.count(sn.extractall(
r'(?i)(?P<step_count>STEP NUMBER)',
self.stdout, 'step_count')), 10),
sn.assert_lt(energy_diff, 1e-4)
])

self.perf_patterns = {
'time': sn.extractsingle(r'^ CP2K(\s+[\d\.]+){4}\s+(?P<perf>\S+)',
self.stdout, 'perf', float)
}

self.maintainers = ['LM']
self.tags = {'scs'}
self.strict_check = False
self.extra_resources = {
'switches': {
'num_switches': 1
}
}
@performance_function('s')
def time(self):
return sn.extractsingle(r'^ CP2K(\s+[\d\.]+){4}\s+(?P<perf>\S+)',
self.stdout, 'perf', float)


@rfm.parameterized_test(*([s, v]
for s in ['small', 'large']
for v in ['maint', 'prod']))
@rfm.simple_test
class Cp2kCpuCheck(Cp2kCheck):
def __init__(self, scale, variant):
super().__init__()
self.descr = 'CP2K CPU check (version: %s, %s)' % (scale, variant)
self.valid_systems = ['daint:mc', 'eiger:mc', 'pilatus:mc']
if scale == 'small':
scale = parameter(['small', 'large'])
variant = parameter(['maint', 'prod'])
valid_systems = ['daint:mc', 'eiger:mc', 'pilatus:mc']
references_by_variant = {
'maint': {
'small': {
'dom:mc': {'time': (202.2, None, 0.05, 's')},
'daint:mc': {'time': (180.9, None, 0.08, 's')},
'eiger:mc': {'time': (70.0, None, 0.08, 's')},
'pilatus:mc': {'time': (70.0, None, 0.08, 's')}
},
'large': {
'daint:mc': {'time': (141.0, None, 0.05, 's')},
'eiger:mc': {'time': (46.0, None, 0.05, 's')},
'pilatus:mc': {'time': (46.0, None, 0.05, 's')}
}
},
'prod': {
'small': {
'dom:mc': {'time': (202.2, None, 0.05, 's')},
'daint:mc': {'time': (180.9, None, 0.08, 's')},
'eiger:mc': {'time': (70.0, None, 0.08, 's')},
'pilatus:mc': {'time': (70.0, None, 0.08, 's')}
},
'large': {
'daint:mc': {'time': (113.0, None, 0.05, 's')},
'eiger:mc': {'time': (46.0, None, 0.05, 's')},
'pilatus:mc': {'time': (46.0, None, 0.05, 's')}
}
}
}

@run_after('init')
def setup_by_variant_and_scale(self):
self.descr = f'CP2K CPU check (version: {self.scale}, {self.variant})'
self.tags |= {'maintenance'
if self.variant == 'maint' else 'production'}
if self.scale == 'small':
self.valid_systems += ['dom:mc']
if self.current_system.name in ['daint', 'dom']:
self.num_tasks = 216
Expand Down Expand Up @@ -91,37 +124,7 @@ def __init__(self, scale, variant):
'OMP_PROC_BIND': 'close'
}

references = {
'maint': {
'small': {
'dom:mc': {'time': (202.2, None, 0.05, 's')},
'daint:mc': {'time': (180.9, None, 0.08, 's')},
'eiger:mc': {'time': (70.0, None, 0.08, 's')},
'pilatus:mc': {'time': (70.0, None, 0.08, 's')}
},
'large': {
'daint:mc': {'time': (141.0, None, 0.05, 's')},
'eiger:mc': {'time': (46.0, None, 0.05, 's')},
'pilatus:mc': {'time': (46.0, None, 0.05, 's')}
}
},
'prod': {
'small': {
'dom:mc': {'time': (202.2, None, 0.05, 's')},
'daint:mc': {'time': (180.9, None, 0.08, 's')},
'eiger:mc': {'time': (70.0, None, 0.08, 's')},
'pilatus:mc': {'time': (70.0, None, 0.08, 's')}
},
'large': {
'daint:mc': {'time': (113.0, None, 0.05, 's')},
'eiger:mc': {'time': (46.0, None, 0.05, 's')},
'pilatus:mc': {'time': (46.0, None, 0.05, 's')}
}
}
}

self.reference = references[variant][scale]
self.tags |= {'maintenance' if variant == 'maint' else 'production'}
self.reference = self.references_by_variant[self.variant][self.scale]

@run_before('run')
def set_task_distribution(self):
Expand All @@ -132,46 +135,48 @@ def set_cpu_binding(self):
self.job.launcher.options = ['--cpu-bind=cores']


@rfm.parameterized_test(*([s, v]
for s in ['small', 'large']
for v in ['maint', 'prod']))
@rfm.simple_test
class Cp2kGpuCheck(Cp2kCheck):
def __init__(self, scale, variant):
super().__init__()
self.descr = 'CP2K GPU check (version: %s, %s)' % (scale, variant)
self.valid_systems = ['daint:gpu']
self.num_gpus_per_node = 1
if scale == 'small':
scale = parameter(['small', 'large'])
variant = parameter(['maint', 'prod'])
valid_systems = ['daint:gpu']
num_gpus_per_node = 1
num_tasks_per_node = 6
num_cpus_per_task = 2
variables = {
'CRAY_CUDA_MPS': '1',
'OMP_NUM_THREADS': str(num_cpus_per_task)
}
references_by_variant = {
'maint': {
'small': {
'dom:gpu': {'time': (251.8, None, 0.15, 's')},
'daint:gpu': {'time': (241.3, None, 0.05, 's')}
},
'large': {
'daint:gpu': {'time': (199.6, None, 0.06, 's')}
}
},
'prod': {
'small': {
'dom:gpu': {'time': (240.0, None, 0.05, 's')},
'daint:gpu': {'time': (241.3, None, 0.05, 's')}
},
'large': {
'daint:gpu': {'time': (199.6, None, 0.06, 's')}
}
}
}

@run_after('init')
def setup_by_variant_and_scale(self):
self.descr = f'CP2K GPU check (version: {self.scale}, {self.variant})'
if self.scale == 'small':
self.valid_systems += ['dom:gpu']
self.num_tasks = 36
else:
self.num_tasks = 96

self.num_tasks_per_node = 6
self.num_cpus_per_task = 2
self.variables = {
'CRAY_CUDA_MPS': '1',
'OMP_NUM_THREADS': str(self.num_cpus_per_task)
}
references = {
'maint': {
'small': {
'dom:gpu': {'time': (251.8, None, 0.15, 's')},
'daint:gpu': {'time': (241.3, None, 0.05, 's')}
},
'large': {
'daint:gpu': {'time': (199.6, None, 0.06, 's')}
}
},
'prod': {
'small': {
'dom:gpu': {'time': (240.0, None, 0.05, 's')},
'daint:gpu': {'time': (241.3, None, 0.05, 's')}
},
'large': {
'daint:gpu': {'time': (199.6, None, 0.06, 's')}
}
}
}
self.reference = references[variant][scale]
self.tags |= {'maintenance' if variant == 'maint' else 'production'}
self.reference = self.references_by_variant[self.variant][self.scale]
self.tags |= {'maintenance'
if self.variant == 'maint' else 'production'}

0 comments on commit c79d952

Please sign in to comment.