Changeset 12879 for NEMO/branches/UKMO/dev_r12745_HPC-02_Daley_Tiling_trial_public/src/OCE/DOM

Timestamp:

2020-05-06T14:38:01+02:00 (4 years ago)

Author:

hadcv

Message:

Changes following feedback

Location:

NEMO/branches/UKMO/dev_r12745_HPC-02_Daley_Tiling_trial_public/src/OCE/DOM

Files:

• dom_oce.F90 (modified) (1 diff)
• domain.F90 (modified) (5 diffs)

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

@@ -75 +75 @@
    ! Tiling namelist
    LOGICAL, PUBLIC ::   ln_tile
-   INTEGER         ::   nn_tile_i, nn_tile_j
+   INTEGER         ::   nn_ltile_i, nn_ltile_j
 
    !                                 !  domain MPP decomposition parameters

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
 
@@ -123 +122 @@
       CALL dom_glo                     ! global domain versus local domain
       CALL dom_nam                     ! read namelist ( namrun, namdom )
-
-      ! Initialise tile to full domain
-      CALL dom_tile(0)
+      CALL dom_tile                    ! Tile domains
 
       !
@@ -275 +272 @@
 
 
-   SUBROUTINE dom_tile(kntile)
+   SUBROUTINE dom_tile
       !!----------------------------------------------------------------------
       !!                     ***  ROUTINE dom_tile  ***
       !!
-      !! ** Purpose :   Set domain indices for specified tile
-      !!
-      !! ** Action  : - ntile          : current tile number
-      !!              - ntsi, ntsj     : start of internal part of domain
+      !! ** Purpose :   Set tile domain variables
+      !!
+      !! ** Action  : - 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
+      !!              - nijtile        : total number of tiles
+      !!----------------------------------------------------------------------
+      INTEGER ::   jt               ! dummy loop argument
+      INTEGER ::   iitile, ijtile   ! Local integers
+      !!----------------------------------------------------------------------
+      ntile = 0                     ! Initialise to full domain indices
+
+      IF( ln_tile ) THEN            ! Set tile decomposition
+         iitile = (jpi - 2 * nn_hls) / nn_ltile_i
+         ijtile = (jpj - 2 * nn_hls) / nn_ltile_j
+         IF( MOD( jpi - 2 * nn_hls, nn_ltile_i ) /= 0 ) iitile = iitile + 1
+         IF( MOD( jpj - 2 * nn_hls, nn_ltile_j ) /= 0 ) ijtile = ijtile + 1
+
+         nijtile = iitile * ijtile
+         ALLOCATE( ntsi(0:nijtile), ntsj(0:nijtile), ntei(0:nijtile), ntej(0:nijtile) )
+      ELSE
+         nijtile = 1
+         ALLOCATE( ntsi(0:0), ntsj(0:0), ntei(0:0), ntej(0:0) )
+      ENDIF
+
+      ntsi(0) = 1 + nn_hls          ! Full domain
+      ntsj(0) = 1 + nn_hls
+      ntei(0) = jpi - nn_hls
+      ntej(0) = jpj - nn_hls
+
+      IF( ln_tile ) THEN            ! Tile domains
+         DO jt = 1, nijtile
+            ntsi(jt) = ntsi(0) + nn_ltile_i * MOD(jt - 1, iitile)
+            ntsj(jt) = ntsj(0) + nn_ltile_j * ((jt - 1) / iitile)
+            ntei(jt) = MIN(ntsi(jt) + nn_ltile_i - 1, ntei(0))
+            ntej(jt) = MIN(ntsj(jt) + nn_ltile_j - 1, ntej(0))
+         ENDDO
+      ENDIF
+
+      IF(lwp) THEN                  ! control print
+         WRITE(numout,*)
+         WRITE(numout,*) 'dom_tile : Domain tiling decomposition'
+         WRITE(numout,*) '~~~~~~~~'
+         IF( ln_tile ) THEN
+            WRITE(numout,*) iitile, 'tiles in i'
+            WRITE(numout,*) '    Starting indices'
+            WRITE(numout,*) '        ', (ntsi(jt), jt=1, iitile)
+            WRITE(numout,*) '    Ending indices'
+            WRITE(numout,*) '        ', (ntei(jt), jt=1, iitile)
+            WRITE(numout,*) ijtile, 'tiles in j'
+            WRITE(numout,*) '    Starting indices'
+            WRITE(numout,*) '        ', (ntsj(jt), jt=1, nijtile, iitile)
+            WRITE(numout,*) '    Ending indices'
+            WRITE(numout,*) '        ', (ntej(jt), jt=1, nijtile, iitile)
+         ELSE
+            WRITE(numout,*) 'No domain tiling'
+            WRITE(numout,*) '    i indices =', ntsi(0), ':', ntei(0)
+            WRITE(numout,*) '    j indices =', ntsj(0), ':', ntej(0)
+         ENDIF
       ENDIF
    END SUBROUTINE dom_tile
@@ -339 +359 @@
          &             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
+      NAMELIST/namtile/ ln_tile, nn_ltile_i, nn_ltile_j
 #if defined key_netcdf4
       NAMELIST/namnc4/ nn_nchunks_i, nn_nchunks_j, nn_nchunks_k, ln_nc4zip
@@ -473 +493 @@
       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,*)    '   Namelist : namtile   ---   Domain 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,*)    '      Length of tile in i                  nn_ltile_i = ', nn_ltile_i
+         WRITE(numout,*)    '      Length of tile in j                  nn_ltile_j = ', nn_ltile_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
+            WRITE(numout,*) '      The domain will be decomposed into tiles of size', nn_ltile_i, 'x', nn_ltile_j
          ELSE
             WRITE(numout,*) '      Domain tiling will NOT be used'