Changeset 5820 for LMDZ6

Timestamp:

Sep 17, 2025, 2:50:16 PM (2 months ago)

Author:

rkazeroni

Message:

For GPU porting of ratqs_main routine:

• Add "horizontal" comment to specify possible names of horizontal variables
• Move computation of maxval into an initialization routine as this cannot be GPU ported
• Fix wrong dimensionality of wake_s in the argument declaration
• Change problematic variable initializations at declaration (SAVE attribute implied) by either adding a PARAMETER attribute or moving the initialization

Location:

LMDZ6/trunk/libf/phylmd

Files:

• lmdz_ratqs_main.f90 (modified) (4 diffs)
• lmdz_ratqs_multi.f90 (modified) (1 diff)
• physiq_mod.F90 (modified) (2 diffs)

LMDZ6/trunk/libf/phylmd/lmdz_ratqs_main.f90

@@ +1 @@
+!$gpum horizontal klon ngrid
 MODULE lmdz_ratqs_main
+  PRIVATE
+
+  LOGICAL, SAVE :: first=.TRUE.  ! fisrt call to ratqs_main
+  !$OMP THREADPRIVATE(first)
+
+  REAL, SAVE :: resolmax
+
+  REAL, SAVE, ALLOCATABLE :: resol(:)
+  !$OMP THREADPRIVATE(resol)
+
+  PUBLIC ratqs_main_first, ratqs_main
 
 CONTAINS
+
+SUBROUTINE ratqs_main_first(klon, cell_area)
+
+  IMPLICIT NONE
+  INTEGER, INTENT(in) :: klon
+  REAL, DIMENSION(klon), INTENT(in) :: cell_area
+  INTEGER :: i
+
+  IF (first) THEN
+     ALLOCATE(resol(klon))
+     resolmax=sqrt(maxval(cell_area))
+     DO i=1,klon
+        resol(i) = sqrt(cell_area(i))
+     END DO
+     first = .FALSE.
+  END IF
+
+END SUBROUTINE ratqs_main_first
 
 SUBROUTINE ratqs_main(klon,klev,nbsrf,prt_level,lunout,       &
@@ -45 +75 @@
 logical, dimension(klon,klev),intent(in) :: ptconv
 real, dimension(klon,klev),intent(in) :: rnebcon0th,clwcon0th
-real, dimension(klon,klev),intent(in) :: wake_deltaq,wake_s
+real, dimension(klon,klev),intent(in) :: wake_deltaq
+real, dimension(klon),intent(in) :: wake_s
 real, dimension(klon,nbsrf),intent(in) :: t2m,q2m
 real, dimension(klon), intent(in) :: cell_area
@@ -62 +93 @@
 real facteur,zfratqs1,zfratqs2
 real, dimension(klon,klev) :: ratqs_hetero_,ratqs_oro_,ratqs_tke_
-real resol,resolmax,fact
+real, dimension(klon) :: fact
 
 !-------------------------------------------------------------------------
@@ -163 +194 @@
 ! Audran, Meryl, Lea, Gwendal and Etienne
 ! April 2023 
-        resolmax=sqrt(maxval(cell_area))
          do k=1,klev
             do i=1,klon
-              resol=sqrt(cell_area(i))
-              fact=sqrt(resol/resolmax)
-              ratqss(i,k)=ratqsbas*fact+0.5*(ratqshaut-ratqsbas)*fact &
+              fact(i) = sqrt(resol(i)/resolmax)
+              ratqss(i,k)=ratqsbas*fact(i)+0.5*(ratqshaut-ratqsbas)*fact(i) &
               *( tanh( (ratqsp0-pplay(i,k))/ratqsdp) + 1.)
            enddo

LMDZ6/trunk/libf/phylmd/lmdz_ratqs_multi.f90

@@ -58 +58 @@
 
 ! local
-LOGICAL :: klein = .false.
-LOGICAL :: klein_conv = .true.
-REAL :: taup0 = 70000
-REAL :: taudp = 500
-integer :: lev_out=10
+LOGICAL, PARAMETER :: klein = .false.
+LOGICAL, PARAMETER :: klein_conv = .true.
+REAL, PARAMETER :: taup0 = 70000
+REAL, PARAMETER :: taudp = 500
+integer :: lev_out
 REAL, DIMENSION (klon,klev) :: zmasse,entr0,detr0,detraincv,dqp,detrain_p,q0,qd0,tau_diss
 REAL, DIMENSION (klon,klev+1) :: fm0

LMDZ6/trunk/libf/phylmd/physiq_mod.F90

@@ -99 +99 @@
     USE lmdz_blowing_snow_ini, ONLY : blowing_snow_ini , qbst_bs
     USE lmdz_lscp_ini, ONLY : lscp_ini
-    USE lmdz_ratqs_main, ONLY : ratqs_main
+    USE lmdz_ratqs_main, ONLY : ratqs_main, ratqs_main_first
     USE lmdz_ratqs_ini, ONLY : ratqs_ini
     USE lmdz_cloud_optics_prop_ini, ONLY : cloud_optics_prop_ini
@@ -3931 +3931 @@
     ENDDO
 
+    CALL ratqs_main_first(klon, cell_area)
     CALL ratqs_main(klon,klev,nbsrf,prt_level,lunout,        &
          iflag_ratqs,iflag_con,iflag_cld_th,pdtphys,  &