From 72ee0b2fa13b125e49cfca3db1ec7ee557d30a28 Mon Sep 17 00:00:00 2001
From: Jian Sun <sunjian@ucar.edu>
Date: Tue, 8 Aug 2023 09:08:18 -0600
Subject: [PATCH] Add a new XML variable to apply the MPI wrapper script more
 generically.

---
 cime_config/config_component.xml | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/cime_config/config_component.xml b/cime_config/config_component.xml
index 0137597af..a329be743 100644
--- a/cime_config/config_component.xml
+++ b/cime_config/config_component.xml
@@ -802,6 +802,16 @@
     <desc>If set will compile and submit with this gpu offload method enabled </desc>
   </entry>
 
+  <entry id="MPI_GPU_WRAPPER_SCRIPT">
+    <type>char</type>
+    <valid_values></valid_values>
+    <default_value></default_value>
+    <group>build_def</group>
+    <file>env_build.xml</file>
+    <desc>If set will attach this script to the MPI run command, mapping
+    different MPI ranks to different GPUs within the same compute node</desc>
+  </entry>
+
   <entry id="SMP_PRESENT">
     <type>logical</type>
     <valid_values>TRUE,FALSE</valid_values>