fix the usage of wrapper script for Derecho

ESMCI · Aug 7, 2023 · 2341850 · 2341850
1 parent 011e807
commit 2341850
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/CIME/case/case.py b/CIME/case/case.py
@@ -2107,7 +2107,7 @@ def get_mpirun_cmd(self, job=None, allow_unresolved_envvars=True, overrides=None
 
         ngpus_per_node = self.get_value("NGPUS_PER_NODE")
         if ngpus_per_node and ngpus_per_node > 0 and config.gpus_use_set_device_rank:
-            if self.get_value("MACH") == "gust":
+            if self.get_value("MACH") == "gust" or self.get_value("MACH") == "derecho":
                 mpi_arg_string = mpi_arg_string + " get_local_rank "
             else:
                 # this wrapper script only works with OpenMPI library