Correct num_device value in initialize_GPU

SPECFEM · Jan 22, 2025 · b4a8a95 · b4a8a95
1 parent 8f39006
commit b4a8a95
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/specfem3D/initialize_simulation.F90 b/src/specfem3D/initialize_simulation.F90
@@ -583,6 +583,8 @@ subroutine initialize_GPU()
 
   if (NPROC == 1 .and. NUMBER_OF_SIMULTANEOUS_RUNS > 1 ) then
     num_device = mygroup
+  else if (NPROC > 1 .and. NUMBER_OF_SIMULTANEOUS_RUNS > 1 ) then
+    num_device = myrank + mygroup * NPROC
   else
     num_device = myrank
   endif