Adding missing info for running test-suites with GPUS

EESSI · Dec 17, 2024 · 5422490 · 5422490
1 parent d76fea0
commit 5422490
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/reframe_config_bot.py.tmpl b/reframe_config_bot.py.tmpl
@@ -23,6 +23,10 @@ site_configuration = {
                         FEATURES[GPU]
                     ] + list(SCALES.keys()),
                     'resources': [
+                        {
+                            'name': '_rfm_gpu',
+                            'options': ['--gpus-per-node={num_gpus_per_node}'],
+                        },
                         {
                             'name': 'memory',
                             'options': ['--mem={size}'],
@@ -34,6 +38,12 @@ site_configuration = {
                         'mem_per_node': __MEM_PER_NODE__,
                         GPU_VENDOR: GPU_VENDORS[NVIDIA],
                     },
+                    'devices': [
+                        {
+                            'type': DEVICE_TYPES[GPU],
+                            'num_devices': 4,
+                        }
+                    ],
                     'max_jobs': 1
                     }
                 ]