From b8007f4334eb335d5d97cd0cdfd3dadfbe096ec1 Mon Sep 17 00:00:00 2001 From: Sam Foreman Date: Mon, 25 Nov 2024 15:57:59 -0600 Subject: [PATCH] fix: `GRAD_ACC_STEPS` on 32 nodes of Aurora --- ALCF/helpers.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ALCF/helpers.sh b/ALCF/helpers.sh index 54c7b104de..a52fc2bbb1 100644 --- a/ALCF/helpers.sh +++ b/ALCF/helpers.sh @@ -522,9 +522,9 @@ get_grad_acc_steps_on_aurora() { gas=1 elif [[ 128 -le "${nhosts}" && "${nhosts}" -lt 256 ]]; then gas=2 - elif [[ 32 -le "${nhosts}" && "${nhosts}" -lt 128 ]]; then + elif [[ 32 -lt "${nhosts}" && "${nhosts}" -lt 129 ]]; then gas=4 - elif [[ 16 -le "${nhosts}" && "${nhosts}" -lt 32 ]]; then + elif [[ 16 -le "${nhosts}" && "${nhosts}" -le 32 ]]; then gas=8 else gas=16