From 6643b3ac3580ba56dfc5011b7b550d679c4c5078 Mon Sep 17 00:00:00 2001
From: Jian Sun <sjsprecious@gmail.com>
Date: Tue, 9 Jan 2024 21:17:52 -0700
Subject: [PATCH] update gpu wrapper script on Derecho

---
 machines/derecho/config_machines.xml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/machines/derecho/config_machines.xml b/machines/derecho/config_machines.xml
index 6f9d6fd6..4d757cfb 100644
--- a/machines/derecho/config_machines.xml
+++ b/machines/derecho/config_machines.xml
@@ -18,7 +18,7 @@
     <MAX_CPUTASKS_PER_GPU_NODE>64</MAX_CPUTASKS_PER_GPU_NODE>
     <GPU_TYPE>none,a100</GPU_TYPE>
     <GPU_OFFLOAD>none,openacc,openmp,combined</GPU_OFFLOAD>
-    <MPI_GPU_WRAPPER_SCRIPT>get_local_rank</MPI_GPU_WRAPPER_SCRIPT>
+    <MPI_GPU_WRAPPER_SCRIPT>set_gpu_rank</MPI_GPU_WRAPPER_SCRIPT>
     <PROJECT_REQUIRED>TRUE</PROJECT_REQUIRED>
     <mpirun mpilib="default">
       <executable>mpibind</executable>