Alter all config files to allow all scales by default

EESSI · Feb 2, 2024 · b52fd19 · b52fd19
1 parent 9f9b380
commit b52fd19
Show file tree

Hide file tree

Showing 6 changed files with 16 additions and 14 deletions.
diff --git a/config/aws_mc.py b/config/aws_mc.py
@@ -13,7 +13,7 @@
 import os
 
 from eessi.testsuite.common_config import common_logging_config, common_eessi_init
-from eessi.testsuite.constants import FEATURES
+from eessi.testsuite.constants import FEATURES, SCALES
 
 # This config will write all staging, output and logging to subdirs under this prefix
 # Override with RFM_PREFIX environment variable
@@ -97,7 +97,7 @@
     'environs': ['default'],
     'features': [
         FEATURES['CPU']
-    ],
+    ] + list(SCALES.keys()),
     'prepare_cmds': [
         'source %s' % common_eessi_init(),
         # Required when using srun as launcher with --export=NONE in partition access, in order to ensure job

diff --git a/config/github_actions.py b/config/github_actions.py
@@ -17,7 +17,7 @@
                     'scheduler': 'local',
                     'launcher': 'local',
                     'environs': ['default'],
-                    'features': [FEATURES[CPU]],
+                    'features': [FEATURES[CPU]] + list(SCALES.keys()),
                     'processor': {'num_cpus': 2},
                     'resources': [
                         {

diff --git a/config/it4i_karolina.py b/config/it4i_karolina.py
@@ -52,7 +52,7 @@
                     'max_jobs': 120,
                     'features': [
                         FEATURES[CPU],
-                    ],
+                    ] + list(SCALES.keys()),
                     'descr': 'CPU Universal Compute Nodes, see https://docs.it4i.cz/karolina/hardware-overview/'
                 },
 # We don't have GPU budget on Karolina at this time
@@ -88,7 +88,7 @@
 #                     ],
 #                     'features': [
 #                         FEATURES[GPU],
-#                     ],
+#                     ] + list(SCALES.keys()),
 #                     'descr': 'GPU partition with accelerated nodes, see https://docs.it4i.cz/karolina/hardware-overview/'
 #                 },
             ]

diff --git a/config/izum_vega.py b/config/izum_vega.py
@@ -58,7 +58,7 @@
                     ],
                     'features': [
                         FEATURES[CPU],
-                    ],
+                    ] + list(SCALES.keys()),
                     'descr': 'CPU partition Standard, see https://en-doc.vega.izum.si/architecture/'
                 },
                 {
@@ -97,7 +97,7 @@
                     ],
                     'features': [
                         FEATURES[GPU],
-                    ],
+                    ] + list(SCALES.keys()),
                     'descr': 'GPU partition, see https://en-doc.vega.izum.si/architecture/'
                 },
             ]

diff --git a/config/settings_example.py b/config/settings_example.py
@@ -56,7 +56,9 @@
                             'options': ['--mem={size}'],
                         }
                     ],
-                    'features': [FEATURES[CPU]],
+                    # list(SCALES.keys()) adds all the scales from eessi.testsuite.constants as valid for thi partition
+                    # Can be modified if not all scales can run on this partition, see e.g. the surf_snellius.py config
+                    'features': [FEATURES[CPU]] + list(SCALES.keys()),
                 },
                 {
                     'name': 'gpu_partition',
@@ -94,7 +96,7 @@
                     'features': [
                         FEATURES[CPU],
                         FEATURES[GPU],
-                    ],
+                    ] + list(SCALES.keys()),
                     'extras': {
                         GPU_VENDOR: GPU_VENDORS[NVIDIA],
                     },

diff --git a/config/vsc_hortense.py b/config/vsc_hortense.py
@@ -53,7 +53,7 @@ def command(self, job):
                     ],
                     'features': [
                         FEATURES[CPU],
-                    ],
+                    ] + list(SCALES.keys()),
                 },
                 {
                     'name': 'cpu_rome_512gb',
@@ -80,7 +80,7 @@ def command(self, job):
                     ],
                     'features': [
                         FEATURES[CPU],
-                    ],
+                    ] + list(SCALES.keys()),
                 },
                 {
                     'name': 'cpu_milan',
@@ -107,7 +107,7 @@ def command(self, job):
                     ],
                     'features': [
                         FEATURES[CPU],
-                    ],
+                    ] + list(SCALES.keys()),
                 },
                 {
                     'name': 'gpu_rome_a100_40gb',
@@ -128,7 +128,7 @@ def command(self, job):
                     },
                     'features': [
                         FEATURES[GPU],
-                    ],
+                    ] + list(SCALES.keys()),
                     'extras': {
                         GPU_VENDOR: GPU_VENDORS[NVIDIA],
                     },
@@ -169,7 +169,7 @@ def command(self, job):
                     },
                     'features': [
                         FEATURES[GPU],
-                    ],
+                    ] + list(SCALES.keys()),
                     'extras': {
                         GPU_VENDOR: GPU_VENDORS[NVIDIA],
                     },