diff --git a/python/lbann/contrib/lc/launcher.py b/python/lbann/contrib/lc/launcher.py index 5a0231acaa..9b5ce42f9b 100644 --- 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 index 0daeb63ce2..4a7370879c 100644 --- 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