fix: GRAD_ACC_STEPS on 32 nodes of Aurora

argonne-lcf · Nov 25, 2024 · b8007f4 · b8007f4
1 parent 8d0b43b
commit b8007f4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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