Add new LC systems (#2474)

LLNL · Sep 27, 2024 · cb07681 · cb07681
1 parent cae84af
commit cb07681
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/python/lbann/contrib/lc/launcher.py b/python/lbann/contrib/lc/launcher.py
@@ -94,7 +94,7 @@ def prepend_environment_path(key, prefix):
     set_environment('MV2_USE_THREAD_WARNING', 0)
 
     # Optimizations for Tioga
-    if system in ('tioga', 'rzvernal'):
+    if system in ('tioga', 'rzvernal', 'tuolumne', 'rzadams'):
         #set_environment('NCCL_SOCKET_IFNAME', 'hsi')
         set_environment('NCCL_NET_GDR_LEVEL', '3') # From HPE to avoid hangs
         set_environment('MIOPEN_DEBUG_DISABLE_FIND_DB', '0')

diff --git a/python/lbann/contrib/lc/systems.py b/python/lbann/contrib/lc/systems.py
@@ -24,6 +24,8 @@ def __init__(self, cores_per_node, gpus_per_node, scheduler):
     'rzvernal': SystemParams(64, 8, 'flux'),
     'sierra':   SystemParams(44, 4, 'lsf'),
     'tioga':    SystemParams(64, 8, 'flux'),
+    'tuolumne': SystemParams(96, 4, 'flux'),
+    'rzadams':  SystemParams(96, 4, 'flux'),
 }
 
 # Detect system