Changeset 1950 for trunk/LMDZ.TITAN/libf/phytitan

Timestamp:

Jun 20, 2018, 4:10:30 PM (7 years ago)

Author:

jvatant

Message:

Optimization for chemistry firstcall C routine disso.c
-> Row-major-friendly declaration and nesting of loops for huge array krpd
-> ~factor 4 for this routine who was ~ 1/2 of firstcall -> ~ gain ~5% total run-time for standard runs
--JVO

File:

• trunk/LMDZ.TITAN/libf/phytitan/calchim.F90 (modified) (2 diffs)

trunk/LMDZ.TITAN/libf/phytitan/calchim.F90

@@ -171 +171 @@
      
      ALLOCATE(krate(nlaykim_tot,nr_kim))
-     ALLOCATE(krpd(15,nd_kim+1,nlrt_kim,nlat_actfluxes))
+     ALLOCATE(krpd(nd_kim+1,nlrt_kim,15,nlat_actfluxes))
 
      ALLOCATE(nom_prod(nkim))
@@ -420 +420 @@
            DO i=1,nd_kim+1 ! nd_kim+1 is dissociation of N2 by GCR
 
-                 krpddec   =   (   krpd(idec+1,i,ialt  ,klat)   * (1.0-factalt)                   &
-                                 + krpd(idec+1,i,ialt+1,klat)   * factalt       ) * (1.0-factlat) &
-                             + (   krpd(idec+1,i,ialt  ,klat+1) * (1.0-factalt)                   &
-                                 + krpd(idec+1,i,ialt+1,klat+1) * factalt       ) * factlat
+                 krpddec   =   (   krpd(i,ialt  ,idec+1,klat)   * (1.0-factalt)                   &
+                                 + krpd(i,ialt+1,idec+1,klat)   * factalt       ) * (1.0-factlat) &
+                             + (   krpd(i,ialt  ,idec+1,klat+1) * (1.0-factalt)                   &
+                                 + krpd(i,ialt+1,idec+1,klat+1) * factalt       ) * factlat
 
               if      ( factdec.lt.0. ) then 
-                 krpddecm1 =   (   krpd(idec  ,i,ialt  ,klat)   * (1.0-factalt)                   &
-                                 + krpd(idec  ,i,ialt+1,klat)   * factalt       ) * (1.0-factlat) &
-                             + (   krpd(idec  ,i,ialt  ,klat+1) * (1.0-factalt)                   &
-                                 + krpd(idec  ,i,ialt+1,klat+1) * factalt       ) * factlat
+                 krpddecm1 =   (   krpd(i,ialt  ,idec  ,klat)   * (1.0-factalt)                   &
+                                 + krpd(i,ialt+1,idec  ,klat)   * factalt       ) * (1.0-factlat) &
+                             + (   krpd(i,ialt  ,idec  ,klat+1) * (1.0-factalt)                   &
+                                 + krpd(i,ialt+1,idec  ,klat+1) * factalt       ) * factlat
                  krate(l,i) = krpddecm1 * abs(factdec) + krpddec   * ( 1.0 + factdec)
               else if ( factdec.gt.0. ) then
-                 krpddecp1 =   (   krpd(idec+2,i,ialt  ,klat)   * (1.0-factalt)                   &
-                                 + krpd(idec+2,i,ialt+1,klat)   * factalt       ) * (1.0-factlat) &
-                             + (   krpd(idec+2,i,ialt  ,klat+1) * (1.0-factalt)                   &
-                                 + krpd(idec+2,i,ialt+1,klat+1) * factalt       ) * factlat
+                 krpddecp1 =   (   krpd(i,ialt  ,idec+2,klat)   * (1.0-factalt)                   &
+                                 + krpd(i,ialt+1,idec+2,klat)   * factalt       ) * (1.0-factlat) &
+                             + (   krpd(i,ialt  ,idec+2,klat+1) * (1.0-factalt)                   &
+                                 + krpd(i,ialt+1,idec+2,klat+1) * factalt       ) * factlat
                  krate(l,i) = krpddecp1 * factdec      + krpddec   * ( 1.0 - factdec)
               else if ( factdec.eq.0. ) then