diff --git a/eb_hooks.py b/eb_hooks.py index 6b284ca7..de50e29e 100644 --- a/eb_hooks.py +++ b/eb_hooks.py @@ -2475,6 +2475,9 @@ def set_maximum(parallel, max_value): # Deucalion has 32GB HBM for 48 cores per node CPU_TARGET_A64FX: (divide_by_factor, 4), # software-specific limits + #'hipBLASLt': { + # '*': (set_maximum, 1), + #}, 'libxc': { '*': (divide_by_factor, 2), CPU_TARGET_A64FX: (set_maximum, 6), @@ -2494,6 +2497,9 @@ def set_maximum(parallel, max_value): 'QuantumESPRESSO': { CPU_TARGET_A64FX: (set_maximum, 6), }, + 'rocBLAS': { + '*': (divide_by_factor, 2), + }, 'TensorFlow': { '*': (divide_by_factor, 2), CPU_TARGET_A64FX: (set_maximum, 8),