add gpu partition #225

feelpp · Feb 6, 2025 · 3ce8441 · 3ce8441
1 parent 983839a
commit 3ce8441
Showing 1 changed file with 30 additions and 1 deletion.
diff --git a/src/feelpp/benchmarking/reframe/config/machineConfigs/gaya.py b/src/feelpp/benchmarking/reframe/config/machineConfigs/gaya.py
@@ -41,6 +41,35 @@
                         'memory_per_node':500
                     }
                 },
+                {
+                    'name':'gpu',
+                    'scheduler':'squeue',
+                    'launcher':'local',
+                    'max_jobs':4,
+                    'access': ['--partition=gpu'],
+                    'environs': ['default'],
+                    'prepare_cmds': [
+                        'source /etc/profile.d/modules.sh',
+                        "export PATH=/opt/apptainer/v1.3.5/apptainer/bin/:$PATH"
+                    ],
+                    'processor': {
+                        'num_cpus': 32
+                    },
+                    'devices': [
+                        {
+                            'type': 'gpu',
+                            'num_devices': 3
+                        }
+                    ],
+                    'container_platforms':[
+                        {
+                            'type': 'Apptainer'
+                        }
+                    ],
+                    'extras':{
+                        'memory_per_node':256
+                    }
+                }
             ]
         }
     ],
@@ -50,7 +79,7 @@
             'modules': [],
             'cc': 'clang',
             'cxx': 'clang++',
-            'target_systems': ['gaya:production']
+            'target_systems': ['gaya:production','gaya:gpu']
         },
         {
             'name': 'hpcx',