Changeset 5908 for LMDZ6/branches/PBLSURF_GPUPORT/libf/phylmd/surf_landice_mod.F90

Timestamp:

Dec 4, 2025, 1:38:46 PM (3 weeks ago)

Author:

yann meurdesoif

Message:

gpu port of surf_landice

YM

File:

• LMDZ6/branches/PBLSURF_GPUPORT/libf/phylmd/surf_landice_mod.F90 (modified) (10 diffs)

LMDZ6/branches/PBLSURF_GPUPORT/libf/phylmd/surf_landice_mod.F90

@@ -3 +3 @@
   
   IMPLICIT NONE
+  PRIVATE
+
+   REAL,SAVE :: alb_vis_sno_lic
+  !$OMP THREADPRIVATE(alb_vis_sno_lic)
+   REAL,SAVE :: alb_nir_sno_lic
+  !$OMP THREADPRIVATE(alb_nir_sno_lic)
+  
+  LOGICAL, SAVE :: firstcall = .TRUE.
+  !$OMP THREADPRIVATE(firstcall)
+
+
+  PUBLIC :: surf_landice_init, surf_landice, surf_landice_precall, surf_landice_postcall
 
 CONTAINS
@@ -8 +20 @@
 !****************************************************************************************
 !
+  SUBROUTINE surf_landice_init
+  USE ioipsl_getin_p_mod, ONLY : getin_p
+  IMPLICIT NONE
+  
+    alb_vis_sno_lic=0.77
+    CALL getin_p('alb_vis_sno_lic',alb_vis_sno_lic)
+    PRINT*, 'alb_vis_sno_lic',alb_vis_sno_lic
+    alb_nir_sno_lic=0.77
+    CALL getin_p('alb_nir_sno_lic',alb_nir_sno_lic)
+    PRINT*, 'alb_nir_sno_lic',alb_nir_sno_lic
+
+  END SUBROUTINE surf_landice_init
+
+
+  SUBROUTINE surf_landice_precall
+  IMPLICIT NONE
+
+  END SUBROUTINE surf_landice_precall
+
+
+  SUBROUTINE surf_landice_postcall
+  IMPLICIT NONE
+    IF (firstcall) firstcall=.FALSE.
+  END SUBROUTINE surf_landice_postcall
+  
+
+
   SUBROUTINE surf_landice(itime, dtime, knon, knindex, &
        rlon, rlat, debut, lafin, &
@@ -29 +68 @@
 #endif               
            &    )
-
+!$gpum horizontal knon klon  
     USE dimphy
     USE geometry_mod,     ONLY : longitude,latitude
@@ -50 +89 @@
     USE clesphys_mod_h
     USE yomcst_mod_h
-    USE ioipsl_getin_p_mod, ONLY : getin_p
     USE lmdz_blowing_snow_ini, ONLY : c_esalt_bs, zeta_bs, pbst_bs, prt_bs, rhoice_bs, rhohard_bs
     USE lmdz_blowing_snow_ini, ONLY : rhofresh_bs, tau_eqsalt_bs, tau_dens0_bs, tau_densmin_bs
@@ -194 +232 @@
     LOGICAL, DIMENSION(knon) :: ok_remaining_freshsnow
     REAL  :: ta1, ta2, ta3, z01, z02, z03, coefa, coefb, coefc, coefd
+    REAl :: lon(knon), lat(knon)   ! for indexation
+
 
 
@@ -200 +240 @@
 !FC 
 !FC
-   REAL,SAVE :: alb_vis_sno_lic
-  !$OMP THREADPRIVATE(alb_vis_sno_lic)
-   REAL,SAVE :: alb_nir_sno_lic
-  !$OMP THREADPRIVATE(alb_nir_sno_lic)
-  LOGICAL, SAVE :: firstcall = .TRUE.
-  !$OMP THREADPRIVATE(firstcall)
+
 
 
@@ -223 +258 @@
 
   IF (firstcall) THEN
-  alb_vis_sno_lic=0.77
-  CALL getin_p('alb_vis_sno_lic',alb_vis_sno_lic)
-           PRINT*, 'alb_vis_sno_lic',alb_vis_sno_lic
-  alb_nir_sno_lic=0.77
-  CALL getin_p('alb_nir_sno_lic',alb_nir_sno_lic)
-           PRINT*, 'alb_nir_sno_lic',alb_nir_sno_lic
-  
-  DO j=1,knon
+ 
+    DO j=1,knon
        i = knindex(j)
        tempsmoothlic(i) = temp_air(j)
-  ENDDO
-  firstcall=.false.
+    ENDDO
+  
   ENDIF
 !******************************************************************************************
@@ -313 +342 @@
             lafin_is=.true.
           END IF
-
+          
+          !ym surf_inlandsis not ported on gpu for now
+          !$gpum nocall
           CALL surf_inlandsis(knon, rlon, rlat, knindex, itime, dtis, debut_is, lafin_is,&
             rmu0, swdown, lwdown, albedo, pexner, ps, p1lay, precip_rain, precip_snow,   &
@@ -361 +392 @@
     ! use soil model and recalculate properly cal
     IF (soil_model) THEN 
+       lon(1:knon) = longitude(knindex(1:knon))
+       lat(1:knon) = latitude(knindex(1:knon))
        CALL soil(dtime, is_lic, knon, snow, tsurf, qsol, &
-        & longitude(knindex(1:knon)), latitude(knindex(1:knon)), tsoil, soilcap, soilflux)
+        & lon, lat, tsoil, soilcap, soilflux)
        cal(1:knon) = RCPD / soilcap(1:knon)
        radsol(1:knon)  = radsol(1:knon) + soilflux(1:knon)
@@ -662 +695 @@
        ENDDO
 
+       !$gpum nocall
        CALL cpl_send_landice_fields(itime, knon, knindex, run_off_lic, run_off_lic_frac)
     ENDIF
 
  ! transfer runoff rate [kg/m2/s](!) to physiq for output
-    runoff(1:knon)=run_off_lic(1:knon)/dtime
+    runoff(1:knon)=run_off_lic(knindex(1:knon))/dtime
 
 !ym WARNING snow_o, zfrac_o => en klon !!!