Skip to content

Commit b1b962d

Browse files
author
Vasileios Karakasis
authored
Merge pull request #2503 from vkarak/testlib/gpuburn
[testlib] Make GPU burn test a library test
2 parents ed094e6 + 86986ee commit b1b962d

File tree

26 files changed

+455
-342
lines changed

26 files changed

+455
-342
lines changed

‎config/cscs.py‎

Lines changed: 101 additions & 104 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
{
2222
'name': 'login',
2323
'scheduler': 'local',
24-
'environs': ['builtin', 'PrgEnv-gnu'],
24+
'environs': ['gnu'],
2525
'descr': 'Login nodes',
2626
'max_jobs': 4,
2727
'launcher': 'local'
@@ -30,7 +30,7 @@
3030
'name': 'a64fx',
3131
'scheduler': 'slurm',
3232
'access': ['-pa64fx'],
33-
'environs': ['builtin', 'PrgEnv-gnu'],
33+
'environs': ['gnu'],
3434
'descr': 'Fujitsu A64FX CPUs',
3535
'max_jobs': 100,
3636
'launcher': 'srun'
@@ -39,85 +39,81 @@
3939
'name': 'amda100',
4040
'scheduler': 'slurm',
4141
'access': ['-pamda100'],
42-
'environs': ['builtin', 'PrgEnv-gnu'],
42+
'environs': ['gnu', 'cuda'],
4343
'descr': 'AMD Naples 32c + 4x NVIDIA A100',
4444
'max_jobs': 100,
45-
'launcher': 'srun'
45+
'launcher': 'srun',
46+
'features': ['gpu'],
47+
'devices': [
48+
{
49+
'type': 'gpu',
50+
'arch': 'sm_80',
51+
'num_devices': 4
52+
}
53+
]
4654
},
4755
{
4856
'name': 'amdv100',
4957
'scheduler': 'slurm',
5058
'access': ['-pamdv100'],
51-
'environs': ['builtin', 'PrgEnv-gnu'],
59+
'environs': ['gnu', 'cuda'],
5260
'descr': 'AMD Naples 32c + 2x NVIDIA V100',
5361
'max_jobs': 100,
54-
'launcher': 'srun'
62+
'launcher': 'srun',
63+
'features': ['gpu'],
64+
'devices': [
65+
{
66+
'type': 'gpu',
67+
'arch': 'sm_70',
68+
'num_devices': 2
69+
}
70+
]
5571
},
5672
{
5773
'name': 'amdvega',
5874
'scheduler': 'slurm',
5975
'access': ['-pamdvega'],
60-
'environs': ['builtin', 'PrgEnv-gnu'],
76+
'environs': ['gnu', 'rocm'],
6177
'descr': 'AMD Naples 32c + 3x AMD GFX900',
6278
'max_jobs': 100,
63-
'launcher': 'srun'
79+
'launcher': 'srun',
80+
'features': ['gpu'],
81+
'devices': [
82+
{
83+
'type': 'gpu',
84+
'arch': 'gfx900,gfx906',
85+
'num_devices': 3
86+
}
87+
]
6488
},
6589
{
6690
'name': 'intelv100',
6791
'scheduler': 'slurm',
6892
'access': ['-pintelv100'],
69-
'environs': ['builtin', 'PrgEnv-gnu'],
93+
'environs': ['gnu', 'cuda'],
7094
'descr': 'Intel Skylake 36c + 4x NVIDIA V100',
7195
'max_jobs': 100,
72-
'launcher': 'srun'
96+
'launcher': 'srun',
97+
'features': ['gpu'],
98+
'devices': [
99+
{
100+
'type': 'gpu',
101+
'arch': 'sm_70',
102+
'num_devices': 4
103+
}
104+
]
73105
},
74106
{
75107
'name': 'intel',
76108
'scheduler': 'slurm',
77109
'access': ['-pintel'],
78-
'environs': ['builtin', 'PrgEnv-gnu'],
110+
'environs': ['gnu'],
79111
'descr': 'Intel Skylake 36c',
80112
'max_jobs': 100,
81113
'launcher': 'srun'
82114
}
83115
]
84116
},
85-
{
86-
'name': 'tave',
87-
'descr': 'Grand Tave',
88-
'hostnames': ['tave'],
89-
'modules_system': 'tmod',
90-
'resourcesdir': '/apps/common/UES/reframe/resources',
91-
'partitions': [
92-
{
93-
'name': 'login',
94-
'scheduler': 'local',
95-
'environs': [
96-
'builtin',
97-
'PrgEnv-cray',
98-
'PrgEnv-gnu',
99-
'PrgEnv-intel',
100-
'PrgEnv-pgi'
101-
],
102-
'descr': 'Login nodes',
103-
'max_jobs': 4,
104-
'launcher': 'local'
105-
},
106-
{
107-
'name': 'compute',
108-
'scheduler': 'slurm',
109-
'environs': [
110-
'PrgEnv-cray',
111-
'PrgEnv-gnu',
112-
'PrgEnv-intel',
113-
'PrgEnv-pgi'
114-
],
115-
'descr': 'Intel Xeon Phi',
116-
'max_jobs': 100,
117-
'launcher': 'srun'
118-
}
119-
]
120-
},
121117
{
122118
'name': 'daint',
123119
'descr': 'Piz Daint',
@@ -167,6 +163,7 @@
167163
],
168164
'descr': 'Hybrid nodes (Haswell/P100)',
169165
'max_jobs': 100,
166+
'features': ['gpu'],
170167
'resources': [
171168
{
172169
'name': 'switches',
@@ -177,7 +174,14 @@
177174
'options': ['--gres={gres}']
178175
}
179176
],
180-
'launcher': 'srun'
177+
'devices': [
178+
{
179+
'type': 'gpu',
180+
'arch': 'sm_60',
181+
'num_devices': 1
182+
}
183+
],
184+
'launcher': 'srun',
181185
},
182186
{
183187
'name': 'mc',
@@ -309,11 +313,19 @@
309313
'descr': 'Hybrid nodes (Haswell/P100)',
310314
'max_jobs': 100,
311315
'launcher': 'srun',
316+
'features': ['gpu'],
312317
'resources': [
313318
{
314319
'name': 'gres',
315320
'options': ['--gres={gres}']
316321
}
322+
],
323+
'devices': [
324+
{
325+
'type': 'gpu',
326+
'arch': 'sm_60',
327+
'num_devices': 1
328+
}
317329
]
318330
},
319331
{
@@ -392,46 +404,6 @@
392404
}
393405
]
394406
},
395-
{
396-
'name': 'fulen',
397-
'descr': 'Fulen',
398-
'hostnames': [r'fulen-ln\d+'],
399-
'modules_system': 'tmod',
400-
'resourcesdir': '/apps/common/UES/reframe/resources',
401-
'partitions': [
402-
{
403-
'name': 'login',
404-
'scheduler': 'local',
405-
'environs': ['PrgEnv-gnu'],
406-
'descr': 'Login nodes',
407-
'max_jobs': 1,
408-
'launcher': 'local'
409-
},
410-
{
411-
'name': 'normal',
412-
'scheduler': 'slurm',
413-
'environs': ['PrgEnv-gnu'],
414-
'descr': 'Compute nodes - default partition',
415-
'launcher': 'srun'
416-
},
417-
{
418-
'name': 'fat',
419-
'scheduler': 'slurm',
420-
'environs': ['PrgEnv-gnu'],
421-
'access': ['--partition fat'],
422-
'descr': 'High-memory compute nodes',
423-
'launcher': 'srun'
424-
},
425-
{
426-
'name': 'gpu',
427-
'scheduler': 'slurm',
428-
'environs': ['PrgEnv-gnu'],
429-
'access': ['--partition gpu'],
430-
'descr': 'Hybrid compute nodes',
431-
'launcher': 'srun'
432-
}
433-
]
434-
},
435407
{
436408
'name': 'arolla',
437409
'descr': 'Arolla MCH',
@@ -483,6 +455,14 @@
483455
'PrgEnv-pgi-nompi-nocuda'
484456
],
485457
'descr': 'Arolla compute nodes',
458+
'features': ['gpu'],
459+
'devices': [
460+
{
461+
'type': 'gpu',
462+
'arch': 'sm_70',
463+
'num_devices': 8
464+
}
465+
],
486466
'resources': [
487467
{
488468
'name': '_rfm_gpu',
@@ -552,12 +532,20 @@
552532
],
553533
'descr': 'Tsa compute nodes',
554534
'max_jobs': 20,
535+
'features': ['gpu'],
555536
'resources': [
556537
{
557538
'name': '_rfm_gpu',
558539
'options': ['--gres=gpu:{num_gpus_per_node}']
559540
}
560541
],
542+
'devices': [
543+
{
544+
'type': 'gpu',
545+
'arch': 'sm_70',
546+
'num_devices': 8
547+
}
548+
],
561549
'launcher': 'srun'
562550
}
563551
]
@@ -724,26 +712,30 @@
724712
],
725713
'environments': [
726714
{
727-
'name': 'PrgEnv-gnu',
728-
'target_systems': ['ault'],
729-
'modules': ['openmpi'],
730-
'cc': 'mpicc',
731-
'cxx': 'mpicxx',
732-
'ftn': 'mpif90'
715+
'name': 'gnu',
716+
'modules': ['gcc'],
717+
'cc': 'gcc',
718+
'cxx': 'g++',
719+
'ftn': 'gfortran',
720+
'target_systems': ['ault']
733721
},
734722
{
735-
'name': 'builtin',
723+
'name': 'cuda',
724+
'modules': ['gcc', 'cuda'],
725+
'cc': 'gcc',
726+
'cxx': 'g++',
727+
'ftn': 'gfortran',
736728
'target_systems': ['ault'],
737-
'cc': 'cc',
738-
'cxx': '',
739-
'ftn': ''
729+
'features': ['cuda']
740730
},
741731
{
742-
'name': 'builtin-gcc',
743-
'target_systems': ['ault'],
732+
'name': 'rocm',
733+
'modules': ['gcc', 'rocm'],
744734
'cc': 'gcc',
745735
'cxx': 'g++',
746-
'ftn': 'gfortran'
736+
'ftn': 'gfortran',
737+
'target_systems': ['ault'],
738+
'features': ['hip']
747739
},
748740
{
749741
'name': 'PrgEnv-pgi-nompi-nocuda',
@@ -765,6 +757,7 @@
765757
'name': 'PrgEnv-pgi-nompi',
766758
'target_systems': ['arolla'],
767759
'modules': ['PrgEnv-pgi/19.9'],
760+
'features': ['cuda'],
768761
'cc': 'pgcc',
769762
'cxx': 'pgc++',
770763
'ftn': 'pgf90'
@@ -773,6 +766,7 @@
773766
'name': 'PrgEnv-pgi-nompi',
774767
'target_systems': ['tsa'],
775768
'modules': ['PrgEnv-pgi/20.4'],
769+
'features': ['cuda'],
776770
'cc': 'pgcc',
777771
'cxx': 'pgc++',
778772
'ftn': 'pgf90'
@@ -781,6 +775,7 @@
781775
'name': 'PrgEnv-pgi',
782776
'target_systems': ['arolla'],
783777
'modules': ['PrgEnv-pgi/19.9'],
778+
'features': ['cuda'],
784779
'cc': 'mpicc',
785780
'cxx': 'mpicxx',
786781
'ftn': 'mpifort'
@@ -789,6 +784,7 @@
789784
'name': 'PrgEnv-pgi',
790785
'target_systems': ['tsa'],
791786
'modules': ['PrgEnv-pgi/20.4'],
787+
'features': ['cuda'],
792788
'cc': 'mpicc',
793789
'cxx': 'mpicxx',
794790
'ftn': 'mpifort'
@@ -813,6 +809,7 @@
813809
'name': 'PrgEnv-gnu',
814810
'target_systems': ['arolla', 'tsa'],
815811
'modules': ['PrgEnv-gnu/19.2'],
812+
'features': ['cuda'],
816813
'cc': 'mpicc',
817814
'cxx': 'mpicxx',
818815
'ftn': 'mpifort'
@@ -829,6 +826,7 @@
829826
'name': 'PrgEnv-gnu-nompi',
830827
'target_systems': ['arolla', 'tsa'],
831828
'modules': ['PrgEnv-gnu/19.2'],
829+
'features': ['cuda'],
832830
'cc': 'gcc',
833831
'cxx': 'g++',
834832
'ftn': 'gfortran'
@@ -908,10 +906,9 @@
908906
},
909907
{
910908
'name': 'PrgEnv-nvidia',
909+
'modules': ['PrgEnv-nvidia'],
910+
'features': ['cuda'],
911911
'target_systems': ['daint', 'dom'],
912-
'modules': [
913-
'PrgEnv-nvidia',
914-
]
915912
},
916913
{
917914
'name': 'builtin',

0 commit comments

Comments
 (0)