Changeset 562

Timestamp:

09/28/17 21:58:12 (7 years ago)

Author:

dubos

Message:

devel : per-kernel activation of macro-generated code

Location:

codes/icosagcm/devel/src/dynamics

Files:

• caldyn.f90 (modified) (2 diffs)
• caldyn_hevi.f90 (modified) (3 diffs)
• caldyn_kernels_base.F90 (modified) (2 diffs)
• caldyn_kernels_hevi.F90 (modified) (13 diffs)

codes/icosagcm/devel/src/dynamics/caldyn.f90

@@ -2 +2 @@
   USE icosa
   PRIVATE
+  SAVE
   CHARACTER(LEN=255),SAVE :: caldyn_type
 !$OMP THREADPRIVATE(caldyn_type)
-  
-  PUBLIC init_caldyn, caldyn, caldyn_BC
+ 
+  PUBLIC init_caldyn, caldyn, caldyn_BC, dysl
   
 CONTAINS
@@ -28 +29 @@
     END SELECT
         
-      
+    dysl=.FALSE.
+    CALL getin("dysl",dysl)
+    dysl_geopot=dysl
+    CALL getin("dysl_geopot",dysl_geopot)
+    dysl_caldyn_fast=dysl
+    CALL getin("dysl_caldyn_fast",dysl_caldyn_fast)
+    dysl_slow_hydro=dysl
+    CALL getin("dysl_slow_hydro",dysl_slow_hydro)
+    dysl_pvort_only=dysl
+    CALL getin("dysl_pvort_only",dysl_pvort_only)
+    dysl_caldyn_coriolis=dysl
+    CALL getin("dysl_caldyn_coriolis",dysl_caldyn_coriolis)
+
   END SUBROUTINE init_caldyn
 

codes/icosagcm/devel/src/dynamics/caldyn_hevi.f90

@@ -48 +48 @@
     TYPE(t_field) :: f_dPhi_fast(:)
     
-    REAL(rstd),POINTER :: ps(:), dps(:)
+    REAL(rstd),POINTER :: ps(:), dps(:), phis(:)
     REAL(rstd),POINTER :: mass(:,:), theta_rhodz(:,:,:), dtheta_rhodz(:,:,:)
     REAL(rstd),POINTER :: du(:,:), dW(:,:), dPhi(:,:), hflux(:,:), wflux(:,:)
@@ -115 +115 @@
           CALL compute_geopot(mass,theta, ps,pk,geopot)
        ELSE
+          phis = f_phis(ind)
           W = f_W(ind)
           dW = f_dW_fast(ind)
@@ -121 +122 @@
           m_il = f_wil(ind)
           pres = f_gradPhi2(ind)
-          CALL compute_caldyn_solver(tau,mass,theta,pk,geopot,W, m_il,pres, dPhi,dW,du) ! computes d(Phi,W,du)_fast and updates Phi,W
+          CALL compute_caldyn_solver(tau,phis, mass,theta,pk,geopot,W, m_il,pres, dPhi,dW,du) ! computes d(Phi,W,du)_fast and updates Phi,W
        END IF
        u=f_u(ind)

codes/icosagcm/devel/src/dynamics/caldyn_kernels_base.F90

@@ -49 +49 @@
     Rd = kappa*cpp
 
+    IF(dysl_geopot) THEN
 #ifdef CPP_DYSL
-!#if 0
 #include "../kernels/compute_geopot.k90"
 #else
-
+    STOP 'dysl_geopot=.TRUE. but DYNAMICO has been compiled without the -dysl option'
+#endif
+    ELSE
     ! Pressure is computed first top-down (temporarily stored in pk)
     ! Then Exner pressure and geopotential are computed bottom-up
@@ -164 +166 @@
     END IF
 
-#endif
+    END IF ! dysl
 
     !ym flush geopot

codes/icosagcm/devel/src/dynamics/caldyn_kernels_hevi.F90

@@ -9 +9 @@
   PRIVATE
 
-  REAL(rstd), PARAMETER :: pbot=1e5, Phi_bot=0., rho_bot=1e6 ! FIXME
+  REAL(rstd), PARAMETER :: pbot=1e5, rho_bot=1e6
 
   LOGICAL, SAVE :: debug_hevi_solver = .FALSE.
-  LOGICAL, PARAMETER :: rigid=.TRUE.
 
   PUBLIC :: compute_theta, compute_pvort_only, compute_caldyn_Coriolis, &
@@ -62 +61 @@
     CALL trace_start("compute_pvort_only")  
 !!! Compute shallow-water potential vorticity
+    IF(dysl_pvort_only) THEN
 #ifdef CPP_DYSL
 #include "../kernels/pvort_only.k90"
 #else
+        STOP 'dysl_pvort_only=.TRUE. but DYNAMICO is compiled without the -dysl option'
+#endif
+    ELSE
+
     radius_m2=radius**(-2)
     DO l = ll_begin,ll_end
@@ -94 +98 @@
 
     ENDDO
-#endif
+   
+    END IF ! dysl
     CALL trace_end("compute_pvort_only")
 
   END SUBROUTINE compute_pvort_only
 
-  SUBROUTINE compute_NH_geopot(tau, m_ik, m_il, theta, W_il, Phi_il)
+  SUBROUTINE compute_NH_geopot(tau, phis, m_ik, m_il, theta, W_il, Phi_il)
     REAL(rstd),INTENT(IN)    :: tau ! solve Phi-tau*dPhi/dt = Phi_rhs
+    REAL(rstd),INTENT(IN)    :: phis(iim*jjm)
     REAL(rstd),INTENT(IN)    :: m_ik(iim*jjm,llm)
     REAL(rstd),INTENT(IN)    :: m_il(iim*jjm,llm+1)
@@ -125 +131 @@
 
 #ifdef CPP_DYSL
+#define PHI_BOT(ij) phis(ij)
 #include "../kernels/compute_NH_geopot.k90"
 #else
@@ -264 +271 @@
   END SUBROUTINE compute_NH_geopot
 
-  SUBROUTINE compute_caldyn_solver(tau,rhodz,theta,pk, geopot,W, m_il,pres, dPhi,dW,du)
+  SUBROUTINE compute_caldyn_solver(tau,phis, rhodz,theta,pk, geopot,W, m_il,pres, dPhi,dW,du)
     REAL(rstd),INTENT(IN) :: tau ! "solve" Phi-tau*dPhi/dt = Phi_rhs
+    REAL(rstd),INTENT(IN)    :: phis(iim*jjm)
     REAL(rstd),INTENT(IN)    :: rhodz(iim*jjm,llm)
     REAL(rstd),INTENT(IN)    :: theta(iim*jjm,llm,nqdyn)
@@ -287 +295 @@
 #ifdef CPP_DYSL
 !$OMP BARRIER
+#define PHI_BOT(ij) phis(ij)
+#define PHI_BOT_VAR phis
 #include "../kernels/caldyn_solver.k90"
 !$OMP BARRIER
@@ -396 +406 @@
     Rd=cpp*kappa
 
+    IF(dysl_caldyn_fast) THEN
 #ifdef CPP_DYSL
 #include "../kernels/caldyn_fast.k90"
 #else
+    STOP 'dysl_caldyn_fast=.TRUE. but DYNAMICO is compiled without the -dysl option'
+#endif
+    ELSE
+
     ! Compute Bernoulli term
     IF(boussinesq) THEN
@@ -496 +511 @@
        END IF
     END DO
-#endif       
+
+    END IF ! dysl
     CALL trace_end("compute_caldyn_fast")
 
@@ -515 +531 @@
     CALL trace_start("compute_caldyn_Coriolis")
 
+    IF(dysl_caldyn_coriolis) THEN
 #ifdef CPP_DYSL
 #include "../kernels/coriolis.k90"
 #else
+    STOP 'dysl_caldyn_coriolis is .TRUE. but DYNAMICO is compiled without the -dysl option'
+#endif
+
+    ELSE
 #define FTHETA(ij) Ftheta(ij,1)
 
@@ -653 +674 @@
     END SELECT
 #undef FTHETA
-#endif
+
+    END IF ! dysl
 
     CALL trace_end("compute_caldyn_Coriolis")
@@ -672 +694 @@
     CALL trace_start("compute_caldyn_slow_hydro")
 
+    IF(dysl_slow_hydro) THEN
 #ifdef CPP_DYSL
 #define BERNI(ij,l) berni(ij,l)
@@ -677 +700 @@
 #undef BERNI
 #else
+        STOP 'dysl_slow_hydro=.TRUE. but DYNAMICO is compiled without the -dysl option'
+#endif
+
+     ELSE
+
 #define BERNI(ij) berni(ij,1)
 
@@ -726 +754 @@
     END DO
 #undef BERNI
-#endif
+    END IF ! dysl
     CALL trace_end("compute_caldyn_slow_hydro")    
   END SUBROUTINE compute_caldyn_slow_hydro