|
4 | 4 | # SPDX-License-Identifier: BSD-3-Clause |
5 | 5 |
|
6 | 6 | import contextlib |
7 | | -import itertools |
8 | 7 | import os |
9 | 8 |
|
10 | 9 | import reframe as rfm |
@@ -100,27 +99,37 @@ def __init__(self, scale, variant): |
100 | 99 | class GromacsCPUCheck(GromacsBaseCheck): |
101 | 100 | def __init__(self, scale, variant): |
102 | 101 | super().__init__('md.log') |
103 | | - self.valid_systems = ['daint:mc'] |
| 102 | + self.valid_systems = ['daint:mc', 'eiger:mc'] |
104 | 103 | self.descr = 'GROMACS CPU check' |
105 | 104 | self.executable_opts = ['mdrun', '-dlb yes', '-ntomp 1', '-npme -1', |
106 | 105 | '-nb cpu', '-s herflat.tpr'] |
107 | 106 |
|
108 | 107 | if scale == 'small': |
109 | 108 | self.valid_systems += ['dom:mc'] |
110 | | - self.num_tasks = 216 |
111 | | - self.num_tasks_per_node = 36 |
| 109 | + if (self.current_system.name in ['daint', 'dom']): |
| 110 | + self.num_tasks = 216 |
| 111 | + self.num_tasks_per_node = 36 |
| 112 | + elif (self.current_system.name in ['eiger']): |
| 113 | + self.num_tasks = 768 |
| 114 | + self.num_tasks_per_node = 128 |
112 | 115 | else: |
113 | | - self.num_tasks = 576 |
114 | | - self.num_tasks_per_node = 36 |
| 116 | + if (self.current_system.name in ['daint', 'dom']): |
| 117 | + self.num_tasks = 576 |
| 118 | + self.num_tasks_per_node = 36 |
| 119 | + elif (self.current_system.name in ['eiger']): |
| 120 | + self.num_tasks = 2048 |
| 121 | + self.num_tasks_per_node = 128 |
115 | 122 |
|
116 | 123 | references = { |
117 | 124 | 'prod': { |
118 | 125 | 'small': { |
119 | 126 | 'dom:mc': {'perf': (40.0, -0.05, None, 'ns/day')}, |
120 | | - 'daint:mc': {'perf': (38.8, -0.10, None, 'ns/day')} |
| 127 | + 'daint:mc': {'perf': (38.8, -0.10, None, 'ns/day')}, |
| 128 | + 'eiger:mc': {'perf': (103.00, -0.10, None, 'ns/day')} |
121 | 129 | }, |
122 | 130 | 'large': { |
123 | | - 'daint:mc': {'perf': (68.0, -0.20, None, 'ns/day')} |
| 131 | + 'daint:mc': {'perf': (68.0, -0.20, None, 'ns/day')}, |
| 132 | + 'eiger:mc': {'perf': (146.00, -0.20, None, 'ns/day')} |
124 | 133 | } |
125 | 134 | }, |
126 | 135 | } |
|
0 commit comments