Improve OpenMP parallelization in GC-Classic using collapse

Signed-off-by: Lizzie Lundgren <elundgren@seas.harvard.edu>
geoschem · Apr 18, 2024 · 2cd624f · 2cd624f
1 parent 5decf9e
commit 2cd624f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/GeosCore/aerosol_thermodynamics_mod.F90 b/GeosCore/aerosol_thermodynamics_mod.F90
@@ -498,7 +498,8 @@ SUBROUTINE DO_ATE( Input_Opt,  State_Chm, State_Diag, &
     !$OMP PRIVATE( HETP_Mg,    HETP_H,    HETP_OH,    HETP_LWC             ) &
     !$OMP PRIVATE( HETP_frNa,  HETP_frCa, HETP_frK,   HETP_frMg            ) &
     !$OMP PRIVATE( HETP_frSO4, HETP_num                                    ) &
-    !$OMP SCHEDULE( DYNAMIC, 1                                             )
+    !$OMP COLLAPSE( 3                                                      ) &
+    !$OMP SCHEDULE( DYNAMIC, 8                                             )
     DO L = 1, State_Grid%NZ
     DO J = 1, State_Grid%NY
     DO I = 1, State_Grid%NX