99
1010class Cp2kCheck (rfm .RunOnlyRegressionTest ):
1111 def __init__ (self ):
12- self .valid_prog_environs = ['builtin' ]
12+ if self .current_system .name == 'pilatus' :
13+ self .valid_prog_environs = ['cpeGNU' ]
14+ else :
15+ self .valid_prog_environs = ['builtin' ]
16+
1317 self .modules = ['CP2K' ]
1418 self .executable = 'cp2k.psmp'
1519 self .executable_opts = ['H2O-256.inp' ]
@@ -51,13 +55,13 @@ class Cp2kCpuCheck(Cp2kCheck):
5155 def __init__ (self , scale , variant ):
5256 super ().__init__ ()
5357 self .descr = 'CP2K CPU check (version: %s, %s)' % (scale , variant )
54- self .valid_systems = ['daint:mc' , 'eiger:mc' ]
58+ self .valid_systems = ['daint:mc' , 'eiger:mc' , 'pilatus:mc' ]
5559 if scale == 'small' :
5660 self .valid_systems += ['dom:mc' ]
5761 if self .current_system .name in ['daint' , 'dom' ]:
5862 self .num_tasks = 216
5963 self .num_tasks_per_node = 36
60- elif self .current_system .name == 'eiger' :
64+ elif self .current_system .name in [ 'eiger' , 'pilatus' ] :
6165 self .num_tasks = 96
6266 self .num_tasks_per_node = 16
6367 self .num_cpus_per_task = 16
@@ -74,7 +78,7 @@ def __init__(self, scale, variant):
7478 if self .current_system .name in ['daint' , 'dom' ]:
7579 self .num_tasks = 576
7680 self .num_tasks_per_node = 36
77- elif self .current_system .name in ['eiger' ]:
81+ elif self .current_system .name in ['eiger' , 'pilatus' ]:
7882 self .num_tasks = 256
7983 self .num_tasks_per_node = 16
8084 self .num_cpus_per_task = 16
@@ -92,22 +96,26 @@ def __init__(self, scale, variant):
9296 'small' : {
9397 'dom:mc' : {'time' : (202.2 , None , 0.05 , 's' )},
9498 'daint:mc' : {'time' : (180.9 , None , 0.08 , 's' )},
95- 'eiger:mc' : {'time' : (70.0 , None , 0.08 , 's' )}
99+ 'eiger:mc' : {'time' : (70.0 , None , 0.08 , 's' )},
100+ 'pilatus:mc' : {'time' : (70.0 , None , 0.08 , 's' )}
96101 },
97102 'large' : {
98103 'daint:mc' : {'time' : (141.0 , None , 0.05 , 's' )},
99- 'eiger:mc' : {'time' : (46.0 , None , 0.05 , 's' )}
104+ 'eiger:mc' : {'time' : (46.0 , None , 0.05 , 's' )},
105+ 'pilatus:mc' : {'time' : (46.0 , None , 0.05 , 's' )}
100106 }
101107 },
102108 'prod' : {
103109 'small' : {
104110 'dom:mc' : {'time' : (202.2 , None , 0.05 , 's' )},
105111 'daint:mc' : {'time' : (180.9 , None , 0.08 , 's' )},
106- 'eiger:mc' : {'time' : (70.0 , None , 0.08 , 's' )}
112+ 'eiger:mc' : {'time' : (70.0 , None , 0.08 , 's' )},
113+ 'pilatus:mc' : {'time' : (70.0 , None , 0.08 , 's' )}
107114 },
108115 'large' : {
109116 'daint:mc' : {'time' : (113.0 , None , 0.05 , 's' )},
110- 'eiger:mc' : {'time' : (46.0 , None , 0.05 , 's' )}
117+ 'eiger:mc' : {'time' : (46.0 , None , 0.05 , 's' )},
118+ 'pilatus:mc' : {'time' : (46.0 , None , 0.05 , 's' )}
111119 }
112120 }
113121 }
0 commit comments