Changeset 5820 for LMDZ6/trunk/libf/phylmd/lmdz_ratqs_main.f90

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

File:

• LMDZ6/trunk/libf/phylmd/lmdz_ratqs_main.f90 (modified) (4 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