Changeset 5868 for LMDZ6/branches/PBLSURF_GPUPORT/libf/phylmd/coef_diff_turb_mod.f90

Timestamp:

Nov 17, 2025, 3:50:06 PM (6 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.

YM

File:

• LMDZ6/branches/PBLSURF_GPUPORT/libf/phylmd/coef_diff_turb_mod.f90 (modified) (7 diffs)

LMDZ6/branches/PBLSURF_GPUPORT/libf/phylmd/coef_diff_turb_mod.f90

@@ -17 +17 @@
        ycoefm, ycoefh ,yq2, yeps, ydrgpro)
 
-    USE dimphy
+    USE dimphy, ONLY : klev
+    USE mod_grid_phy_lmdz, ONLY: klon_glo
     USE indice_sol_mod
     USE print_control_mod, ONLY: prt_level, lunout
@@ -34 +35 @@
     REAL, INTENT(IN)                           :: dtime
     INTEGER, INTENT(IN)                        :: nsrf, knon
-    INTEGER, DIMENSION(klon), INTENT(IN)       :: ni
-    REAL, DIMENSION(klon,klev+1), INTENT(IN)   :: ypaprs
-    REAL, DIMENSION(klon,klev), INTENT(IN)     :: ypplay
-    REAL, DIMENSION(klon,klev), INTENT(IN)     :: yu, yv
-    REAL, DIMENSION(klon,klev), INTENT(IN)     :: yq, yt
-    REAL, DIMENSION(klon), INTENT(IN)          :: yts, yqsurf
-    REAL, DIMENSION(klon), INTENT(IN)          :: ycdragm
+    INTEGER, DIMENSION(knon), INTENT(IN)       :: ni
+    REAL, DIMENSION(knon,klev+1), INTENT(IN)   :: ypaprs
+    REAL, DIMENSION(knon,klev), INTENT(IN)     :: ypplay
+    REAL, DIMENSION(knon,klev), INTENT(IN)     :: yu, yv
+    REAL, DIMENSION(knon,klev), INTENT(IN)     :: yq, yt
+    REAL, DIMENSION(knon), INTENT(IN)          :: yts, yqsurf
+    REAL, DIMENSION(knon), INTENT(IN)          :: ycdragm
 !FC
-    REAL, DIMENSION(klon,klev), INTENT(IN)     :: ydrgpro
+    REAL, DIMENSION(knon,klev), INTENT(IN)     :: ydrgpro
 
 
 ! InOutput arguments
 !****************************************************************************************
-    REAL, DIMENSION(klon,klev+1), INTENT(INOUT):: yq2
+    REAL, DIMENSION(knon,klev+1), INTENT(INOUT):: yq2
 
 ! Output arguments
 !****************************************************************************************
-    REAL, DIMENSION(klon,klev+1), INTENT(OUT)  :: yeps
-    REAL, DIMENSION(klon,klev), INTENT(OUT)    :: ycoefh
-    REAL, DIMENSION(klon,klev), INTENT(OUT)    :: ycoefm
+    REAL, DIMENSION(knon,klev+1), INTENT(OUT)  :: yeps
+    REAL, DIMENSION(knon,klev), INTENT(OUT)    :: ycoefh
+    REAL, DIMENSION(knon,klev), INTENT(OUT)    :: ycoefm
 
 ! Other local variables
 !****************************************************************************************
     INTEGER                                    :: k, i, j
-    REAL, DIMENSION(klon,klev)                 :: ycoefm0, ycoefh0, yzlay, yteta
-    REAL, DIMENSION(klon,klev+1)               :: yzlev, q2diag, ykmm, ykmn, ykmq
-    REAL, DIMENSION(klon)                      :: yustar
+    REAL, DIMENSION(knon,klev)                 :: ycoefm0, ycoefh0, yzlay, yteta
+    REAL, DIMENSION(knon,klev+1)               :: yzlev, q2diag, ykmm, ykmn, ykmq
+    REAL, DIMENSION(knon)                      :: yustar
 
     ykmm = 0 !ym missing init
@@ -161 +162 @@
        ! Mais ca fait planter le replay.
        ! En attendant une réécriture, on a joute des if (Fredho)
-       if ( klon>1 .or. (klon==1 .and. knon==1) ) then
+
+       if ( klon_glo>1 .or. (klon_glo==1 .and. knon==1) ) then
           CALL ustarhb(knon,klev,knon,yu,yv,ycdragm, yustar)
        endif
@@ -171 +173 @@
 !   iflag_pbl peut etre utilise comme longuer de melange
        IF (iflag_pbl.GE.31) THEN
-          if ( klon>1 .or. (klon==1 .and. knon==1) ) then
+
+          if ( klon_glo>1 .or. (klon_glo==1 .and. knon==1) ) then
           CALL vdif_kcay(knon,klev,knon,dtime,RG,RD,ypaprs,yt, &
                yzlev,yzlay,yu,yv,yteta, &
@@ -236 +239 @@
     INTEGER, INTENT(IN)                      :: knon, nsrf
     REAL, INTENT(IN)                         :: ksta, ksta_ter
-    REAL, DIMENSION(klon), INTENT(IN)        :: ts
-    REAL, DIMENSION(klon,klev+1), INTENT(IN) :: paprs
-    REAL, DIMENSION(klon,klev), INTENT(IN)   :: pplay
-    REAL, DIMENSION(klon,klev), INTENT(IN)   :: u, v, t, q
-    REAL, DIMENSION(klon), INTENT(IN)        :: qsurf
-
-    REAL, DIMENSION(klon,klev), INTENT(OUT)  :: pcfm, pcfh
+    REAL, DIMENSION(knon), INTENT(IN)        :: ts
+    REAL, DIMENSION(knon,klev+1), INTENT(IN) :: paprs
+    REAL, DIMENSION(knon,klev), INTENT(IN)   :: pplay
+    REAL, DIMENSION(knon,klev), INTENT(IN)   :: u, v, t, q
+    REAL, DIMENSION(knon), INTENT(IN)        :: qsurf
+
+    REAL, DIMENSION(knon,klev), INTENT(OUT)  :: pcfm, pcfh
 
 !
 ! Local variables:
 !
-    INTEGER, DIMENSION(klon)    :: itop ! numero de couche du sommet de la couche limite
+    INTEGER, DIMENSION(knon)    :: itop ! numero de couche du sommet de la couche limite
 !
 ! Quelques constantes et options:
@@ -278 +281 @@
 ! Variables locales:
     INTEGER i, k !IM 120704
-    REAL zgeop(klon,klev)
-    REAL zmgeom(klon)
-    REAL zri(klon)
-    REAL zl2(klon)
+    REAL zgeop(knon,klev)
+    REAL zmgeom(knon)
+    REAL zri(knon)
+    REAL zl2(knon)
     REAL zdphi, zdu2, ztvd, ztvu, zcdn
     REAL zscf
@@ -517 +520 @@
 !
     INTEGER, INTENT(IN)                       :: knon, nsrf
-    REAL, DIMENSION(klon, klev+1), INTENT(IN) ::  paprs
-    REAL, DIMENSION(klon, klev), INTENT(IN)   ::  pplay
-    REAL, DIMENSION(klon, klev), INTENT(IN)   :: t(klon,klev)
-
-    REAL, DIMENSION(klon, klev), INTENT(OUT)  :: pcfm, pcfh
+    REAL, DIMENSION(knon, klev+1), INTENT(IN) ::  paprs
+    REAL, DIMENSION(knon, klev), INTENT(IN)   ::  pplay
+    REAL, DIMENSION(knon, klev), INTENT(IN)   :: t(klon,klev)
+
+    REAL, DIMENSION(knon, klev), INTENT(OUT)  :: pcfm, pcfh
 !
 ! Quelques constantes et options: