Skip to content

Commit 31aa543

Browse files
kraushmteojgo
kraushm
and
teojgo
authored
Add Arolla configuration/fix hook.py (eth-cscs#207)
* Adding Arolla configuration * Change name of system in hooks.py --------- Co-authored-by: Theofilos Manitaras <manitaras@cscs.ch>
1 parent 33ef7ea commit 31aa543

File tree

2 files changed

+174
-2
lines changed

2 files changed

+174
-2
lines changed

‎checks/microbenchmarks/gpu/hooks.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ def set_gpu_arch(self):
2222
if self.current_environ.name not in {'PrgEnv-nvidia'}:
2323
self.modules = ['craype-accel-nvidia60', 'cdt-cuda']
2424

25-
elif cs in {'arola', 'tsa'}:
25+
elif cs in {'arolla', 'tsa'}:
2626
self.gpu_arch = '70'
2727
self.modules = ['cuda/10.1.243']
2828
elif cs in {'ault'}:
@@ -46,7 +46,7 @@ def set_num_gpus_per_node(self):
4646
cp = self.current_partition.fullname
4747
if cs in {'dom', 'daint'}:
4848
self.num_gpus_per_node = 1
49-
elif cs in {'arola', 'tsa'}:
49+
elif cs in {'arolla', 'tsa'}:
5050
self.num_gpus_per_node = 8
5151
elif cp in {'ault:amda100', 'ault:intelv100'}:
5252
self.num_gpus_per_node = 4

‎config/systems/arolla.py

+172
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,172 @@
1+
# Copyright 2016-2023 Swiss National Supercomputing Centre (CSCS/ETH Zurich)
2+
# ReFrame Project Developers. See the top-level LICENSE file for details.
3+
#
4+
# SPDX-License-Identifier: BSD-3-Clause
5+
#
6+
# ReFrame CSCS settings
7+
#
8+
9+
import reframe.utility.osext as osext
10+
import copy
11+
12+
base_config = {
13+
'modules_system': 'tmod',
14+
'resourcesdir': '/apps/common/UES/reframe/resources',
15+
'partitions': [
16+
{
17+
'name': 'login',
18+
'scheduler': 'local',
19+
'environs': [
20+
'PrgEnv-pgi',
21+
'PrgEnv-pgi-nompi',
22+
'PrgEnv-pgi-nocuda',
23+
'PrgEnv-pgi-nompi-nocuda',
24+
'PrgEnv-gnu',
25+
'PrgEnv-gnu-nompi',
26+
'PrgEnv-gnu-nocuda',
27+
'PrgEnv-gnu-nompi-nocuda'
28+
],
29+
'descr': 'Tsa login nodes',
30+
'max_jobs': 4,
31+
'launcher': 'local'
32+
},
33+
{
34+
'name': 'pn',
35+
'scheduler': 'slurm',
36+
'access': ['--partition=pn-regression'],
37+
'environs': [
38+
'PrgEnv-pgi',
39+
'PrgEnv-pgi-nompi',
40+
'PrgEnv-pgi-nocuda',
41+
'PrgEnv-pgi-nompi-nocuda',
42+
'PrgEnv-gnu',
43+
'PrgEnv-gnu-nompi',
44+
'PrgEnv-gnu-nocuda',
45+
'PrgEnv-gnu-nompi-nocuda'
46+
],
47+
'descr': 'Tsa post-processing nodes',
48+
'max_jobs': 20,
49+
'extras': {
50+
'cn_memory': 377,
51+
},
52+
'launcher': 'srun',
53+
'features': ['remote'],
54+
},
55+
{
56+
'name': 'cn',
57+
'scheduler': 'slurm',
58+
'access': ['--partition=cn-regression'],
59+
'environs': [
60+
'PrgEnv-gnu',
61+
'PrgEnv-gnu-nompi',
62+
'PrgEnv-gnu-nocuda',
63+
'PrgEnv-gnu-nompi-nocuda',
64+
'PrgEnv-pgi',
65+
'PrgEnv-pgi-nompi',
66+
'PrgEnv-pgi-nocuda',
67+
'PrgEnv-pgi-nompi-nocuda'
68+
],
69+
'descr': 'Tsa compute nodes',
70+
'max_jobs': 20,
71+
'extras': {
72+
'cn_memory': 377,
73+
},
74+
'features': ['gpu', 'nvgpu', 'remote'],
75+
'resources': [
76+
{
77+
'name': '_rfm_gpu',
78+
'options': ['--gres=gpu:{num_gpus_per_node}']
79+
}
80+
],
81+
'devices': [
82+
{
83+
'type': 'gpu',
84+
'arch': 'sm_70',
85+
'num_devices': 8
86+
}
87+
],
88+
'launcher': 'srun'
89+
}
90+
]
91+
}
92+
93+
arolla_sys = copy.deepcopy(base_config)
94+
arolla_sys['name'] = 'arolla'
95+
arolla_sys['descr'] = 'Tsa MCH'
96+
arolla_sys['hostnames'] = [r'arolla-\w+\d+']
97+
98+
site_configuration = {
99+
'systems': [
100+
arolla_sys,
101+
],
102+
'environments': [
103+
{
104+
'name': 'PrgEnv-pgi-nompi-nocuda',
105+
'target_systems': ['arolla'],
106+
'modules': ['PrgEnv-pgi/20.4-nocuda'],
107+
'cc': 'pgcc',
108+
'cxx': 'pgc++',
109+
'ftn': 'pgf90'
110+
},
111+
{
112+
'name': 'PrgEnv-pgi-nompi',
113+
'target_systems': ['arolla'],
114+
'modules': ['PrgEnv-pgi/20.4'],
115+
'features': ['cuda'],
116+
'cc': 'pgcc',
117+
'cxx': 'pgc++',
118+
'ftn': 'pgf90'
119+
},
120+
{
121+
'name': 'PrgEnv-pgi',
122+
'target_systems': ['arolla'],
123+
'modules': ['PrgEnv-pgi/20.4'],
124+
'features': ['cuda'],
125+
'cc': 'mpicc',
126+
'cxx': 'mpicxx',
127+
'ftn': 'mpifort'
128+
},
129+
{
130+
'name': 'PrgEnv-pgi-nocuda',
131+
'target_systems': ['arolla'],
132+
'modules': ['PrgEnv-pgi/20.4-nocuda'],
133+
'cc': 'mpicc',
134+
'cxx': 'mpicxx',
135+
'ftn': 'mpifort'
136+
},
137+
{
138+
'name': 'PrgEnv-gnu',
139+
'target_systems': ['arolla'],
140+
'modules': ['PrgEnv-gnu/19.2'],
141+
'features': ['cuda'],
142+
'cc': 'mpicc',
143+
'cxx': 'mpicxx',
144+
'ftn': 'mpifort'
145+
},
146+
{
147+
'name': 'PrgEnv-gnu-nocuda',
148+
'target_systems': ['arolla'],
149+
'modules': ['PrgEnv-gnu/19.2-nocuda'],
150+
'cc': 'mpicc',
151+
'cxx': 'mpicxx',
152+
'ftn': 'mpifort'
153+
},
154+
{
155+
'name': 'PrgEnv-gnu-nompi',
156+
'target_systems': ['arolla'],
157+
'modules': ['PrgEnv-gnu/19.2'],
158+
'features': ['cuda'],
159+
'cc': 'gcc',
160+
'cxx': 'g++',
161+
'ftn': 'gfortran'
162+
},
163+
{
164+
'name': 'PrgEnv-gnu-nompi-nocuda',
165+
'target_systems': ['arolla'],
166+
'modules': ['PrgEnv-gnu/19.2-nocuda'],
167+
'cc': 'gcc',
168+
'cxx': 'g++',
169+
'ftn': 'gfortran'
170+
},
171+
]
172+
}

0 commit comments

Comments
 (0)