From 8fb82f8effa3db84ea1790aadb9ee4f47f70aef6 Mon Sep 17 00:00:00 2001
From: noel <ndkeen@lbl.gov>
Date: Wed, 13 Mar 2024 10:44:58 -0700
Subject: [PATCH] use gpus per node construct for pm-gpu and muller-gpu

---
 cime_config/machines/config_batch.xml | 14 ++++++--------
 1 file changed, 6 insertions(+), 8 deletions(-)

diff --git a/cime_config/machines/config_batch.xml b/cime_config/machines/config_batch.xml
index 61d2cd364a69..ae9b73c7582d 100644
--- a/cime_config/machines/config_batch.xml
+++ b/cime_config/machines/config_batch.xml
@@ -433,17 +433,16 @@
     <directives>
       <directive> --constraint=gpu</directive>
     </directives>
-    <directives compiler="gnugpu">
-      <directive> --gpus-per-task=1</directive>
-    </directives>
     <directives COMPSET="!.*MMF.*" compiler="gnugpu">
+      <directive> --gpus-per-node=4</directive>
       <directive> --gpu-bind=none</directive>
     </directives>
     <directives COMPSET=".*MMF.*" compiler="gnugpu">
+      <directive> --gpus-per-task=1</directive>
       <directive> --gpu-bind=map_gpu:0,1,2,3</directive>
     </directives>
     <directives compiler="nvidiagpu">
-      <directive> --gpus-per-task=1</directive>
+      <directive> --gpus-per-node=4</directive>
       <directive> --gpu-bind=none</directive>
     </directives>
     <directives compiler="gnu">
@@ -475,17 +474,16 @@
     <directives>
       <directive> --constraint=gpu</directive>
     </directives>
-    <directives compiler="gnugpu">
-      <directive> --gpus-per-task=1</directive>
-    </directives>
     <directives COMPSET="!.*MMF.*" compiler="gnugpu">
+      <directive> --gpus-per-node=4</directive>
       <directive> --gpu-bind=none</directive>
     </directives>
     <directives COMPSET=".*MMF.*" compiler="gnugpu">
+      <directive> --gpus-per-task=1</directive>
       <directive> --gpu-bind=map_gpu:0,1,2,3</directive>
     </directives>
     <directives compiler="nvidiagpu">
-      <directive> --gpus-per-task=1</directive>
+      <directive> --gpus-per-node=4</directive>
       <directive> --gpu-bind=none</directive>
     </directives>
     <directives compiler="gnu">