Changeset 12765 for NEMO/branches/UKMO/dev_r12745_HPC-02_Daley_Tiling_trial_public/src/OCE/DOM/domain.F90

Timestamp:

2020-04-17T14:45:04+02:00 (4 years ago)

Author:

hadcv

Message:

tra_ldf_iso trial using public variables

File:

• NEMO/branches/UKMO/dev_r12745_HPC-02_Daley_Tiling_trial_public/src/OCE/DOM/domain.F90 (modified) (6 diffs)

NEMO/branches/UKMO/dev_r12745_HPC-02_Daley_Tiling_trial_public/src/OCE/DOM/domain.F90

@@ -49 +49 @@
 
    PUBLIC   dom_init     ! called by nemogcm.F90
+   PUBLIC   dom_tile     ! called by step.F90
    PUBLIC   domain_cfg   ! called by nemogcm.F90
 
@@ -122 +123 @@
       CALL dom_glo                     ! global domain versus local domain
       CALL dom_nam                     ! read namelist ( namrun, namdom )
+
+      ! Initialise tile to full domain
+      CALL dom_tile(0)
+
       !
       IF( lwxios ) THEN
@@ -270 +275 @@
 
 
+   SUBROUTINE dom_tile(kntile)
+      !!----------------------------------------------------------------------
+      !!                     ***  ROUTINE dom_tile  ***
+      !!
+      !! ** Purpose :   Set domain indices for specified tile
+      !!
+      !! ** Action  : - ntile          : current tile number
+      !!              - ntsi, ntsj     : start of internal part of domain
+      !!              - ntei, ntej     : end of internal part of domain
+      !!              - ntsim1, ntsjm1 : start of domain
+      !!              - nteip1, ntejp1 : end of domain
+      !!----------------------------------------------------------------------
+      INTEGER   , INTENT(in ) :: kntile               ! Tile number
+      INTEGER                 :: iitile, ijtile       ! Tile number in i and j
+      !!----------------------------------------------------------------------
+
+      IF( ln_tile .AND. kntile > 0 ) THEN          ! Tile domain
+         iitile = 1 + MOD( kntile - 1, jpnitile )
+         ijtile = 1 + (kntile - 1) / jpnitile
+
+         ntile = kntile
+         ntsi = 2 + (iitile - 1) * nn_tile_i
+         ntsj = 2 + (ijtile - 1) * nn_tile_j
+         ntei = MIN(ntsi + nn_tile_i - 1, jpim1)   ! Size of last tile limited by full domain
+         ntej = MIN(ntsj + nn_tile_j - 1, jpjm1)   !
+         ntsim1 = ntsi - 1
+         ntsjm1 = ntsj - 1
+         nteip1 = ntei + 1
+         ntejp1 = ntej + 1
+      ELSE                                         ! Full domain
+         ntile = 1
+         ntsi = 2
+         ntsj = 2
+         ntei = jpim1
+         ntej = jpjm1
+         ntsim1 = 1
+         ntsjm1 = 1
+         nteip1 = jpi
+         ntejp1 = jpj
+      ENDIF
+   END SUBROUTINE dom_tile
+
+
    SUBROUTINE dom_nam
       !!----------------------------------------------------------------------
@@ -278 +326 @@
       !! ** input   : - namrun namelist
       !!              - namdom namelist
+      !!              - namtile namelist
       !!              - namnc4 namelist   ! "key_netcdf4" only
       !!----------------------------------------------------------------------
@@ -290 +339 @@
          &             ln_cfmeta, ln_xios_read, nn_wxios
       NAMELIST/namdom/ ln_linssh, rn_Dt, rn_atfp, ln_crs, ln_meshmask
+      NAMELIST/namtile/ ln_tile, nn_tile_i, nn_tile_j
 #if defined key_netcdf4
       NAMELIST/namnc4/ nn_nchunks_i, nn_nchunks_j, nn_nchunks_k, ln_nc4zip
@@ -417 +467 @@
       r1_Dt = 1._wp / rDt
 
+      READ  ( numnam_ref, namtile, IOSTAT = ios, ERR = 905 )
+905   IF( ios /= 0 )   CALL ctl_nam ( ios , 'namtile in reference namelist' )
+      READ  ( numnam_cfg, namtile, IOSTAT = ios, ERR = 906 )
+906   IF( ios >  0 )   CALL ctl_nam ( ios , 'namtile in configuration namelist' )
+      IF(lwm) WRITE( numond, namtile )
+
+      ! Set tile decomposition
+      IF( ln_tile ) THEN
+         jpnitile = (jpi - 2) / nn_tile_i
+         jpnjtile = (jpj - 2) / nn_tile_j
+         IF( MOD( jpi - 2, nn_tile_i ) /= 0 ) jpnitile = jpnitile + 1
+         IF( MOD( jpj - 2, nn_tile_j ) /= 0 ) jpnjtile = jpnjtile + 1
+      ELSE
+         jpnitile = 1
+         jpnjtile = 1
+      ENDIF
+      jpnijtile = jpnitile * jpnjtile
+
+      IF(lwp) THEN
+         WRITE(numout,*)
+         WRITE(numout,*)    '   Namelist : namtile   ---   tiling decomposition'
+         WRITE(numout,*)    '      Tiling (T) or not (F)                ln_tile   = ', ln_tile
+         WRITE(numout,*)    '      Length of tile in i                  nn_tile_i = ', nn_tile_i
+         WRITE(numout,*)    '      Length of tile in j                  nn_tile_j = ', nn_tile_j
+         WRITE(numout,*)
+         IF( ln_tile ) THEN
+            WRITE(numout,*) '      The domain will be decomposed into', jpnijtile, 'tiles of size', nn_tile_i, 'x', nn_tile_j
+         ELSE
+            WRITE(numout,*) '      Domain tiling will NOT be used'
+         ENDIF
+      ENDIF
+
       IF( TRIM(Agrif_CFixed()) == '0' ) THEN
          lrxios = ln_xios_read.AND.ln_rstart