Add new LC systems

fiedorowicz1 · fiedorowicz1 · commit fd02a8cc6004 · 2024-09-27T14:38:13.000-07:00
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

Original file line number	Diff line number	Diff line change
`@@ -24,6 +24,8 @@ def __init__(self, cores_per_node, gpus_per_node, scheduler):`
`24`	`24`	`'rzvernal': SystemParams(64, 8, 'flux'),`
`25`	`25`	`'sierra': SystemParams(44, 4, 'lsf'),`
`26`	`26`	`'tioga': SystemParams(64, 8, 'flux'),`
	`27`	`+ 'tuolumne': SystemParams(96, 4, 'flux'),`
	`28`	`+ 'rzadams': SystemParams(96, 4, 'flux'),`
`27`	`29`	`}`
`28`	`30`
`29`	`31`	`# Detect system`