Changeset 5942 for LMDZ6/trunk/libf/phylmd/surf_land_bucket_mod.F90

Timestamp:

Dec 17, 2025, 7:33:39 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.

merge of commit r5868

YM

File:

• LMDZ6/trunk/libf/phylmd/surf_land_bucket_mod.F90 (modified) (3 diffs)

LMDZ6/trunk/libf/phylmd/surf_land_bucket_mod.F90

@@ -54 +54 @@
 !****************************************************************************************
     INTEGER, INTENT(IN)                     :: itime, jour, knon
-    INTEGER, DIMENSION(klon), INTENT(IN)    :: knindex
+    INTEGER, DIMENSION(knon), INTENT(IN)    :: knindex
     LOGICAL, INTENT(IN)                     :: debut
     REAL, INTENT(IN)                        :: dtime
-    REAL, DIMENSION(klon), INTENT(IN)       :: tsurf
-    REAL, DIMENSION(klon), INTENT(IN)       :: p1lay
-    REAL, DIMENSION(klon), INTENT(IN)       :: tq_cdrag
-    REAL, DIMENSION(klon), INTENT(IN)       :: precip_rain, precip_snow
-    REAL, DIMENSION(klon), INTENT(IN)       :: temp_air, spechum
-    REAL, DIMENSION(klon), INTENT(IN)       :: petAcoef, peqAcoef
-    REAL, DIMENSION(klon), INTENT(IN)       :: petBcoef, peqBcoef
-    REAL, DIMENSION(klon), INTENT(IN)       :: pref
-    REAL, DIMENSION(klon), INTENT(IN)       :: u1, v1, gustiness
-    REAL, DIMENSION(klon), INTENT(IN)       :: rugoro
-    REAL, DIMENSION(klon), INTENT(IN)       :: swnet, lwnet
-#ifdef ISO
-    REAL, DIMENSION(ntiso,klon), INTENT(IN) :: xtprecip_rain, xtprecip_snow
-    REAL, DIMENSION(ntiso,klon), INTENT(IN) :: xtspechum    
+    REAL, DIMENSION(knon), INTENT(IN)       :: tsurf
+    REAL, DIMENSION(knon), INTENT(IN)       :: p1lay
+    REAL, DIMENSION(knon), INTENT(IN)       :: tq_cdrag
+    REAL, DIMENSION(knon), INTENT(IN)       :: precip_rain, precip_snow
+    REAL, DIMENSION(knon), INTENT(IN)       :: temp_air, spechum
+    REAL, DIMENSION(knon), INTENT(IN)       :: petAcoef, peqAcoef
+    REAL, DIMENSION(knon), INTENT(IN)       :: petBcoef, peqBcoef
+    REAL, DIMENSION(knon), INTENT(IN)       :: pref
+    REAL, DIMENSION(knon), INTENT(IN)       :: u1, v1, gustiness
+    REAL, DIMENSION(knon), INTENT(IN)       :: rugoro
+    REAL, DIMENSION(knon), INTENT(IN)       :: swnet, lwnet
+#ifdef ISO
+    REAL, DIMENSION(ntiso,knon), INTENT(IN) :: xtprecip_rain, xtprecip_snow
+    REAL, DIMENSION(ntiso,knon), INTENT(IN) :: xtspechum    
 #endif
 
 ! In/Output variables
 !****************************************************************************************
-    REAL, DIMENSION(klon), INTENT(INOUT)          :: snow, qsol
-    REAL, DIMENSION(klon), INTENT(INOUT)          :: agesno
-    REAL, DIMENSION(klon, nsoilmx), INTENT(INOUT) :: tsoil
-#ifdef ISO
-    REAL, DIMENSION(niso,klon), INTENT(INOUT)       :: xtsnow,xtsol
+    REAL, DIMENSION(knon), INTENT(INOUT)          :: snow, qsol
+    REAL, DIMENSION(knon), INTENT(INOUT)          :: agesno
+    REAL, DIMENSION(knon, nsoilmx), INTENT(INOUT) :: tsoil
+#ifdef ISO
+    REAL, DIMENSION(niso,knon), INTENT(INOUT)       :: xtsnow,xtsol
 #endif
 
 ! Output variables
 !****************************************************************************************
-    REAL, DIMENSION(klon), INTENT(OUT)       :: qsurf
-    REAL, DIMENSION(klon), INTENT(OUT)       :: z0_new
-    REAL, DIMENSION(klon), INTENT(OUT)       :: alb1_new, alb2_new
-    REAL, DIMENSION(klon), INTENT(OUT)       :: evap, fluxsens, fluxlat
-    REAL, DIMENSION(klon), INTENT(OUT)       :: tsurf_new
-    REAL, DIMENSION(klon), INTENT(OUT)       :: dflux_s, dflux_l      
-#ifdef ISO
-    REAL, DIMENSION(ntiso,klon), INTENT(OUT) :: xtevap
-    REAL, DIMENSION(klon),       INTENT(OUT) :: h1
-    REAL, DIMENSION(niso,klon),  INTENT(OUT) :: xtrunoff_diag
-    REAL, DIMENSION(klon),       INTENT(OUT) :: runoff_diag
-    REAL, DIMENSION(niso,klon),  INTENT(IN)  :: Rland_ice
+    REAL, DIMENSION(knon), INTENT(OUT)       :: qsurf
+    REAL, DIMENSION(knon), INTENT(OUT)       :: z0_new
+    REAL, DIMENSION(knon), INTENT(OUT)       :: alb1_new, alb2_new
+    REAL, DIMENSION(knon), INTENT(OUT)       :: evap, fluxsens, fluxlat
+    REAL, DIMENSION(knon), INTENT(OUT)       :: tsurf_new
+    REAL, DIMENSION(knon), INTENT(OUT)       :: dflux_s, dflux_l      
+#ifdef ISO
+    REAL, DIMENSION(ntiso,knon), INTENT(OUT) :: xtevap
+    REAL, DIMENSION(knon),       INTENT(OUT) :: h1
+    REAL, DIMENSION(niso,knon),  INTENT(OUT) :: xtrunoff_diag
+    REAL, DIMENSION(knon),       INTENT(OUT) :: runoff_diag
+    REAL, DIMENSION(niso,knon),  INTENT(IN)  :: Rland_ice
 #endif
 
 ! Local variables
 !****************************************************************************************
-    REAL, DIMENSION(klon) :: soilcap, soilflux
-    REAL, DIMENSION(klon) :: cal, beta, dif_grnd, capsol
-    REAL, DIMENSION(klon) :: alb_neig, alb_lim, icesub
-    REAL, DIMENSION(klon) :: zfra
-    REAL, DIMENSION(klon) :: radsol       ! total net radiance at surface
-    REAL, DIMENSION(klon) :: u0, v0, u1_lay, v1_lay
-    REAL, DIMENSION(klon) :: dummy_riverflow,dummy_coastalflow 
+    REAL, DIMENSION(knon) :: soilcap, soilflux
+    REAL, DIMENSION(knon) :: cal, beta, dif_grnd, capsol
+    REAL, DIMENSION(knon) :: alb_neig, alb_lim, icesub
+    REAL, DIMENSION(knon) :: zfra
+    REAL, DIMENSION(knon) :: radsol       ! total net radiance at surface
+    REAL, DIMENSION(knon) :: u0, v0, u1_lay, v1_lay
+    REAL, DIMENSION(knon) :: dummy_riverflow,dummy_coastalflow 
     INTEGER               :: i
 #ifdef ISO
     INTEGER               :: ixt
-    REAL, DIMENSION(niso,klon) :: xtsnow_prec,xtsol_prec
-    REAL, DIMENSION(klon) :: snow_prec,qsol_prec
+    REAL, DIMENSION(niso,knon) :: xtsnow_prec,xtsol_prec
+    REAL, DIMENSION(knon) :: snow_prec,qsol_prec
     REAL, PARAMETER       :: t_coup = 273.15
-    REAL, DIMENSION(klon) :: fq_fonte_diag
-    REAL, DIMENSION(klon) :: fqfonte_diag
-    REAL, DIMENSION(klon) :: snow_evap_diag 
-    REAL, DIMENSION(klon) :: fqcalving_diag 
+    REAL, DIMENSION(knon) :: fq_fonte_diag
+    REAL, DIMENSION(knon) :: fqfonte_diag
+    REAL, DIMENSION(knon) :: snow_evap_diag 
+    REAL, DIMENSION(knon) :: fqcalving_diag 
     REAL                  :: max_eau_sol_diag  
-    REAL, DIMENSION(klon) :: run_off_lic_diag 
+    REAL, DIMENSION(knon) :: run_off_lic_diag 
     REAL :: coeff_rel_diag
 #endif 
@@ -284 +284 @@
         !write(*,*) 'snow_evap_diag(1)=',snow_evap_diag(1)
 #endif          
-        CALL calcul_iso_surf_ter_vectall(klon,knon, &
+        CALL calcul_iso_surf_ter_vectall(knon,knon, &
      &           evap,snow_evap_diag,snow, &
      &           fq_fonte_diag,fqfonte_diag,dtime,precip_rain,xtprecip_rain, &
@@ -302 +302 @@
 !* Calculate the age of snow
 !
-    CALL albsno(klon,knon,dtime,agesno(:),alb_neig(:), precip_snow(:))  
+    CALL albsno(knon,knon,dtime,agesno(:),alb_neig(:), precip_snow(:))  
     
     WHERE (snow(1 : knon) .LT. 0.0001) agesno(1 : knon) = 0.