From 54224908794d8d4294d1778cb00d32c1004c7919 Mon Sep 17 00:00:00 2001
From: Lara Ramona Peeters <49882639+laraPPr@users.noreply.github.com>
Date: Tue, 17 Dec 2024 18:55:47 +0100
Subject: [PATCH] Adding missing info for running test-suites with GPUS

---
 reframe_config_bot.py.tmpl | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/reframe_config_bot.py.tmpl b/reframe_config_bot.py.tmpl
index 8f77429434..b63d5928b3 100644
--- 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
                     }
                 ]