Add LUMI support (#1546)

Author: shuds13

libensemble/resources/platforms.py

@@ -162,6 +162,19 @@ class GenericROCm(Platform):
     scheduler_match_slots: bool = True
 
 
+class Lumi(Platform):
+    mpi_runner: str = "srun"
+    cores_per_node: int = 64
+    logical_cores_per_node: int = 128
+
+
+class LumiGPU(Lumi):
+    gpus_per_node: int = 8
+    gpu_setting_type: str = "env"
+    gpu_setting_name: str = "ROCR_VISIBLE_DEVICES"
+    scheduler_match_slots: bool = True
+
+
 class Perlmutter(Platform):
     mpi_runner: str = "srun"
 
@@ -243,6 +256,8 @@ class Known_platforms(BaseModel):
     aurora: Aurora = Aurora()
     generic_rocm: GenericROCm = GenericROCm()
     frontier: Frontier = Frontier()
+    lumi: Lumi = Lumi()
+    lumi_g: LumiGPU = LumiGPU()
     perlmutter: Perlmutter = Perlmutter()
     perlmutter_c: PerlmutterCPU = PerlmutterCPU()
     perlmutter_g: PerlmutterGPU = PerlmutterGPU()
@@ -272,6 +287,16 @@ def known_envs():
         else:
             name = "perlmutter"
             logger.manager_warning("Perlmutter detected, but no compute partition detected. Are you on login nodes?")
+    if os.environ.get("SLURM_CLUSTER_NAME") == "lumi":
+        partition = os.environ.get("SLURM_JOB_PARTITION")
+        print(f"Lumi partition: {partition}")
+        if not partition:
+            logger.manager_warning("LUMI detected, but no compute partition detected. Are you on login nodes?")
+        if partition and partition.endswith("-g"):
+            name = "lumi_g"
+            print(f"Lumi GPU detected: {name}")
+        else:
+            name = "lumi"
     return name