Changeset 5876 for LMDZ6/branches/PBLSURF_GPUPORT/libf/phylmd/cdrag_mod.f90

Timestamp:

Nov 21, 2025, 3:58:45 PM (16 hours ago)

Author:

yann meurdesoif

Message:

GPU port of cdrag + call_atke + coef_diff_turb

YM

File:

• LMDZ6/branches/PBLSURF_GPUPORT/libf/phylmd/cdrag_mod.f90 (modified) (11 diffs)

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

@@ -9 +9 @@
 !
   IMPLICIT NONE
+  
+  PRIVATE
+
+  LOGICAL, SAVE :: firstcall = .TRUE.
+  !$OMP THREADPRIVATE(firstcall)
+  INTEGER, SAVE :: iflag_corr_sta
+  !$OMP THREADPRIVATE(iflag_corr_sta)
+  INTEGER, SAVE :: iflag_corr_insta
+  !$OMP THREADPRIVATE(iflag_corr_insta)
+  LOGICAL, SAVE :: ok_cdrag_iter
+  !$OMP THREADPRIVATE(ok_cdrag_iter)
+  
+  PUBLIC :: cdrag_init, cdrag
 
 CONTAINS
@@ -16 +29 @@
 !r original routine svn3623
 !
+  SUBROUTINE cdrag_init
+  USE ioipsl_getin_p_mod, ONLY : getin_p
+  USE modd_csts, ONLY : ini_csts
+  IMPLICIT NONE
+
+! On choisit les fonctions de stabilite utilisees au premier appel
+!**************************************************************************
+    IF (firstcall) THEN
+      iflag_corr_sta=2
+      iflag_corr_insta=2
+      ok_cdrag_iter = .FALSE.
+ 
+      CALL getin_p('iflag_corr_sta',iflag_corr_sta)
+      CALL getin_p('iflag_corr_insta',iflag_corr_insta)
+      CALL getin_p('ok_cdrag_iter',ok_cdrag_iter)
+  
+      CALL ini_csts
+
+      firstcall = .FALSE.
+    ENDIF
+    
+
+  END SUBROUTINE cdrag_init
+
+
  SUBROUTINE cdrag(knon,  nsrf,   &
      speed, t1,    q1,    zgeop1, &
@@ -21 +59 @@
      ri_in, iri_in, &
      cdm,  cdh,  zri,   pref, prain, tsol , pat1)
+!$gpum horizontal knon
 
   USE coare_cp_mod, ONLY: coare_cp
@@ -26 +65 @@
   USE indice_sol_mod
   USE print_control_mod, ONLY: lunout, prt_level
-  USE ioipsl_getin_p_mod, ONLY : getin_p
   USE lmdz_atke_turbulence_ini, ONLY : smmin, ric, cinf, cepsilon, pr_slope, pr_asym, pr_neut
   USE yomcst_mod_h
   USE clesphys_mod_h
   USE yoethf_mod_h
+  USE clc_core_cp_mod, ONLY : clc_core_cp
+  USE ecumev6_flux_mod, ONLY : ecumev6_flux
   IMPLICIT NONE
 ! ================================================================= c
@@ -169 +209 @@
 
 
-
-  LOGICAL, SAVE :: firstcall = .TRUE.
-  !$OMP THREADPRIVATE(firstcall)
-  INTEGER, SAVE :: iflag_corr_sta
-  !$OMP THREADPRIVATE(iflag_corr_sta)
-  INTEGER, SAVE :: iflag_corr_insta
-  !$OMP THREADPRIVATE(iflag_corr_insta)
-  LOGICAL, SAVE :: ok_cdrag_iter
-  !$OMP THREADPRIVATE(ok_cdrag_iter)
 
 !===================================================================c
@@ -249 +280 @@
 !=============================================================================c
 
-! On choisit les fonctions de stabilite utilisees au premier appel
-!**************************************************************************
- IF (firstcall) THEN
-   iflag_corr_sta=2
-   iflag_corr_insta=2
-   ok_cdrag_iter = .FALSE.
- 
-   CALL getin_p('iflag_corr_sta',iflag_corr_sta)
-   CALL getin_p('iflag_corr_insta',iflag_corr_insta)
-   CALL getin_p('ok_cdrag_iter',ok_cdrag_iter)
-
-   firstcall = .FALSE.
- ENDIF
-
 !------------------ Rajout (OT2018) --------------------
 !---------    Rajout pour itération sur rugosité     ----------------
@@ -302 +319 @@
             mixte = .true.
          ENDIF
+
          call clc_core_cp ( sqrt(zdu2),t1(i)-tsurf(i),q1(i)-qsurf(i),t1(i),q1(i),&
              zgeop1(i)/RG, zgeop1(i)/RG,  zgeop1(i)/RG,&
@@ -320 +338 @@
            PSSS = 0.0
          ENDIF
-         call ini_csts
-         call ecumev6_flux( z_0m,t1(i),tsurf(i),&
+!         call ini_csts
+!$gpum nocall 
+          call ecumev6_flux( knon, 1, z_0m,t1(i),tsurf(i),&
              q1(i),qsurf(i),sqrt(zdu2),zgeop1(i)/RG,PSSS,zgeop1(i)/RG,&
              psol(i),pat1(i), OPRECIP, OPWEBB,&
@@ -339 +358 @@
          LPRECIP = .false.
          LPWG    = .false.
-         call ini_csts
+!ym         call ini_csts
 
            z0m_1d = z0m
@@ -345 +364 @@
            sqrt_zdu2_1d = sqrt(zdu2)
            zgeop1_rg_1d=zgeop1(i)/RG
-           call coare30_flux_cnrm(z0m_1d,t1(i),tsurf(i), q1(i),  &
+
+!$gpum nocall
+           call coare30_flux_cnrm(1, z0m_1d,t1(i),tsurf(i), q1(i),  &
                sqrt_zdu2_1d,zgeop1_rg_1d,zgeop1_rg_1d,psol(i),qsurf(i),PQSAT, &
                PSFTH,PFSTQ,PUSTAR,PCD,PCDN,PCH,PCE,PRI, &
@@ -362 +383 @@
          ENDIF
          write(*,*) "debug size",size(coeffs)
+
+!$gpum nocall
          call coare_cp(sqrt(zdu2),t1(i)-tsurf(i),q1(i)-qsurf(i),&
                t1(i),q1(i),&