Changeset 5942 for LMDZ6/trunk/libf/phylmd/stdlevvar_mod.f90

Timestamp:

Dec 17, 2025, 7:33:39 PM (2 weeks ago)

Author:

yann meurdesoif

Message:

Separate pbl_surface into 3 subroutines for GPU port

pbl_surface_uncompress_pre : prepare computation for sub subsurface before compressing
pbl_surface_subsrf : each sub-surface is called one after other (horizontal = knon)
pbl_surface_uncompress_post : sub-surface are uncompressed, computation is done on whole domain (horizontal = klon)

pbl_surface_main becomes the driver, calling pbl_surface_uncompress_pre, and then looping under sub-surface (and calling pbl_surface_subsrf) and then calling pbl_surface_uncompress_post.

merge of commit r5868

YM

File:

• LMDZ6/trunk/libf/phylmd/stdlevvar_mod.f90 (modified) (7 diffs)

LMDZ6/trunk/libf/phylmd/stdlevvar_mod.f90

@@ -318 +318 @@
                            n2mout)
 !
+      USE indice_sol_mod, ONLY : nbsrf
       USE yomcst_mod_h
       USE ioipsl_getin_p_mod, ONLY : getin_p
       USE yoethf_mod_h
       USE flux_arp_mod_h
+      USE lmdz_checksum
       IMPLICIT NONE
 !-------------------------------------------------------------------------
@@ -370 +372 @@
       REAL, dimension(klon), intent(out) :: t_2m, q_2m, ustar
       REAL, dimension(klon), intent(out) :: u_10m, t_10m, q_10m
-      INTEGER, dimension(klon, 6), intent(out) :: n2mout
+      INTEGER, dimension(klon, nbsrf, 6), intent(out) :: n2mout
 !
       REAL, dimension(klon) :: u_2m
@@ -431 +433 @@
       INTEGER, dimension(klon, 6) :: n10mout
 
+      CALL checksum("u1", u1)
+      CALL checksum("v1", v1)
+      CALL checksum("t1", t1)
+      CALL checksum("q1", q1)
+      CALL checksum("z1", z1)
+      CALL checksum("ts1", ts1)
+      CALL checksum("qsurf", qsurf)
+      CALL checksum("z0m", z0m)
+      CALL checksum("z0h", z0h)
+      CALL checksum("psol", psol)
+      CALL checksum("pat1", pat1)
+      CALL checksum("t_2m", t_2m)
+      CALL checksum("q_2m", q_2m)
+      CALL checksum("t_10m", t_10m)
+      CALL checksum("q_10m", q_10m)
+      CALL checksum("u_10m", u_10m)
+      CALL checksum("ustar", ustar)
+      CALL checksum("s_pblh", s_pblh)
+      CALL checksum("prain", prain)
+      CALL checksum("tsol", tsol)
+
 !------------------------------------------------------------------------- 
       CEPDUE=0.1
@@ -439 +462 @@
 !          [tsurf, temp], [qsurf, q1] ou [0, speed]
 !
-      n2mout(:,:)=0
-      n10mout(:,:)=0
+      n2mout(:, nsrf, :)=0
+      n10mout(:, :)=0
       
       DO i=1, knon
@@ -503 +526 @@
 !
          IF(ok_t2m_toosmall(i).OR.ok_t2m_toobig(i)) THEN
-             n2mout(i,1)=n2mout(i,1)+1
+             n2mout(i, nsrf, 1)=n2mout(i, nsrf, 1)+1
          ENDIF
          IF(ok_q2m_toosmall(i).OR.ok_q2m_toobig(i)) THEN
-             n2mout(i,3)=n2mout(i,3)+1
+             n2mout(i, nsrf, 3)=n2mout(i, nsrf, 3)+1
          ENDIF
          IF(ok_u2m_toobig(i)) THEN
-             n2mout(i,5)=n2mout(i,5)+1
+             n2mout(i, nsrf, 5)=n2mout(i, nsrf, 5)+1
          ENDIF
 !
@@ -567 +590 @@
 !
           IF(ok_t2m_toosmall(i).OR.ok_t2m_toobig(i)) THEN
-              n2mout(i,2)=n2mout(i,2)+1
+              n2mout(i, nsrf, 2)=n2mout(i, nsrf, 2)+1
           ENDIF
           IF(ok_q2m_toosmall(i).OR.ok_q2m_toobig(i)) THEN
-              n2mout(i,4)=n2mout(i,4)+1
+              n2mout(i, nsrf, 4)=n2mout(i, nsrf, 4)+1
           ENDIF
           IF(ok_u2m_toobig(i)) THEN
-              n2mout(i,6)=n2mout(i,6)+1
+              n2mout(i, nsrf, 6)=n2mout(i, nsrf, 6)+1
           ENDIF
 !
@@ -750 +773 @@
       ENDDO
 !
+      CALL checksum("u1", u1)
+      CALL checksum("v1", v1)
+      CALL checksum("t1", t1)
+      CALL checksum("q1", q1)
+      CALL checksum("z1", z1)
+      CALL checksum("ts1", ts1)
+      CALL checksum("qsurf", qsurf)
+      CALL checksum("z0m", z0m)
+      CALL checksum("z0h", z0h)
+      CALL checksum("psol", psol)
+      CALL checksum("pat1", pat1)
+      CALL checksum("t_2m", t_2m)
+      CALL checksum("q_2m", q_2m)
+      CALL checksum("t_10m", t_10m)
+      CALL checksum("q_10m", q_10m)
+      CALL checksum("u_10m", u_10m)
+      CALL checksum("ustar", ustar)
+      CALL checksum("s_pblh", s_pblh)
+      CALL checksum("prain", prain)
+      CALL checksum("tsol", tsol)
       RETURN
       END subroutine stdlevvarn