Changeset 6090 for LMDZ6/trunk/libf/phylmd/lmdz_call_gwd.f90

Timestamp:

Mar 5, 2026, 2:30:00 PM (4 days ago)

Author:

rkazeroni

Message:

Edit call_gwd routine to enable automatic GPU porting with GPUM:

• add "nogpu directive" for call to diagnostic
• replace Forall statement not supported by GPUM by DO-loop
• move array operation before call instead of inside a call because it is not supproted on GPU

File:

• LMDZ6/trunk/libf/phylmd/lmdz_call_gwd.f90 (modified) (5 diffs)

LMDZ6/trunk/libf/phylmd/lmdz_call_gwd.f90

@@ -1 @@
-!$gpum horizontal klon, nbsrf
+!$gpum horizontal klon
 MODULE lmdz_call_gwd
 
@@ -144 +144 @@
       REAL, DIMENSION(klon, klev) :: dtadd, duadd, dvadd, exner
       REAL, DIMENSION(klon, klev) :: d_t_oro_tke, d_u_oro_tke, d_v_oro_tke
+      REAL, DIMENSION(klon) :: fall
 
 !===================================================================
@@ -373 +374 @@
       IF (ok_gwd_rando) THEN
 
+         fall = rain_fall + snow_fall
          CALL flott_gwd_rando(klon, klev, phys_tstep, pplay, presnivs, t_seri, u_seri, v_seri, &
-                              rain_fall + snow_fall, zustr_precip, zvstr_precip, &
+                              fall, zustr_precip, zvstr_precip, &
                               d_u_precip, d_v_precip, east_gwstress, west_gwstress)
 
@@ -415 +417 @@
          !  Mountain-induced torque and angular momentum calculation
          !-------------------------------------------------------------------------------------
+         !$gpum nocall
          CALL aaam_bud(27, klon, klev, jD_cur - jD_ref, jH_cur, &
                        ra, rg, romega, &
@@ -486 +489 @@
          ! TKE update from subgrid temperature and wind tendencies
          !----------------------------------------------------------
-         forall (k=1:klev) exner(:, k) = (pplay(:, k)/paprs(:, 1))**rkappa
+         DO k = 1, klev
+            exner(:, k) = (pplay(:, k)/paprs(:, 1))**rkappa
+         END DO
 
          CALL tend_to_tke(phys_tstep, klon, klev, nbsrf, is_ave, paprs, exner, t_seri, u_seri, v_seri, dtadd, duadd, dvadd, pctsrf, tke)