| 1 |
jmc |
1.2 |
C $Header: /u/gcmpack/MITgcm/pkg/exch2/w2_map_procs.F,v 1.1 2012/09/04 00:44:30 jmc Exp $ |
| 2 |
jmc |
1.1 |
C $Name: $ |
| 3 |
|
|
|
| 4 |
|
|
#include "CPP_EEOPTIONS.h" |
| 5 |
|
|
|
| 6 |
|
|
CBOP |
| 7 |
|
|
C !ROUTINE: W2_MAP_PROCS |
| 8 |
|
|
|
| 9 |
|
|
C !INTERFACE: |
| 10 |
|
|
SUBROUTINE W2_MAP_PROCS( myThid ) |
| 11 |
|
|
|
| 12 |
|
|
C !DESCRIPTION: |
| 13 |
|
|
C *==========================================================* |
| 14 |
|
|
C | SUBROUTINE W2_MAP_PROCS |
| 15 |
|
|
C | o Setup Mapping of W2-topology tiles to processes |
| 16 |
|
|
C *==========================================================* |
| 17 |
|
|
C | Set which process "own" which tiles |
| 18 |
|
|
C | and store the 2-way relation between, on one side, |
| 19 |
|
|
C | tile Id from W2-topology and, on the other side, |
| 20 |
|
|
C | process Id with local tile indices bi,bj. |
| 21 |
|
|
C *==========================================================* |
| 22 |
|
|
|
| 23 |
|
|
C !USES: |
| 24 |
|
|
IMPLICIT NONE |
| 25 |
|
|
|
| 26 |
|
|
#include "SIZE.h" |
| 27 |
|
|
#include "EEPARAMS.h" |
| 28 |
|
|
#include "W2_EXCH2_SIZE.h" |
| 29 |
|
|
#include "W2_EXCH2_TOPOLOGY.h" |
| 30 |
|
|
#include "W2_EXCH2_PARAMS.h" |
| 31 |
|
|
|
| 32 |
|
|
C !INPUT PARAMETERS: |
| 33 |
|
|
C myThid :: my Thread Id number |
| 34 |
|
|
C (Note: not relevant since threading has not yet started) |
| 35 |
|
|
INTEGER myThid |
| 36 |
|
|
CEOP |
| 37 |
|
|
|
| 38 |
|
|
C !FUNCTIONS: |
| 39 |
|
|
|
| 40 |
|
|
C !LOCAL VARIABLES: |
| 41 |
|
|
INTEGER thisProc |
| 42 |
|
|
CHARACTER*(MAX_LEN_MBUF) msgBuf |
| 43 |
|
|
CHARACTER commFlag |
| 44 |
|
|
INTEGER myTileId |
| 45 |
|
|
INTEGER I, J |
| 46 |
|
|
INTEGER np, ii, jj, bi, bj |
| 47 |
|
|
INTEGER iErr, tNx, tNy |
| 48 |
|
|
|
| 49 |
|
|
C-- Initialise common blocs W2_MAP_TILE2PROC & W2_EXCH2_COMMFLAG: |
| 50 |
|
|
DO I = 1,W2_maxNbTiles |
| 51 |
|
|
W2_tileProc(I) = 0 |
| 52 |
jmc |
1.2 |
W2_tileIndex(I) = 0 |
| 53 |
jmc |
1.1 |
c W2_tileRank(I) = 0 |
| 54 |
|
|
ENDDO |
| 55 |
|
|
DO bj=1,nSy |
| 56 |
|
|
DO bi=1,nSx |
| 57 |
|
|
W2_myTileList(bi,bj) = 0 |
| 58 |
|
|
DO np=1,nPx*nPy |
| 59 |
|
|
W2_procTileList(bi,bj,np) = 0 |
| 60 |
|
|
ENDDO |
| 61 |
|
|
DO J=1,W2_maxNeighbours |
| 62 |
|
|
W2_myCommFlag(J,bi,bj) = ' ' |
| 63 |
|
|
ENDDO |
| 64 |
|
|
ENDDO |
| 65 |
|
|
ENDDO |
| 66 |
|
|
|
| 67 |
|
|
C-- Decide which tiles this process handles - do this inline for now, but |
| 68 |
|
|
C should go in subroutine. |
| 69 |
|
|
C Set which rank processes "own" which tiles. This should probably |
| 70 |
|
|
C be queried as part of some hand-shaking but for now we use the |
| 71 |
|
|
C functional relationship that was used above. |
| 72 |
|
|
C Fill also W2_procTileList for Single-CPU-IO. |
| 73 |
|
|
|
| 74 |
|
|
C Number of tiles I handle is nSx*nSy |
| 75 |
|
|
thisProc = 1 + myProcId |
| 76 |
|
|
J = 0 |
| 77 |
|
|
DO I=1,exch2_nTiles |
| 78 |
|
|
IF ( exch2_myFace(I) .NE. 0 ) THEN |
| 79 |
|
|
C-- old ordering (makes no difference if nSy*nPy=1 ) |
| 80 |
|
|
c np = 1 + J/(nSx*nSy) |
| 81 |
|
|
c jj = MOD(J,nSx*nSy) |
| 82 |
|
|
c bj = 1 + jj/nSx |
| 83 |
|
|
c bi = 1 + MOD(jj,nSx) |
| 84 |
|
|
C-- new ordering: for single sub-domain (nFacets=1) case, match default setting |
| 85 |
|
|
jj = J/(nSx*nPx) |
| 86 |
|
|
ii = MOD(J,nSx*nPx) |
| 87 |
|
|
C-- natural way to order processors: |
| 88 |
|
|
c np = 1 + ii/nSx + (jj/nSy)*nPx |
| 89 |
|
|
C-- switch processor order to match MPI_CART set-up |
| 90 |
|
|
np = 1 + jj/nSy + (ii/nSx)*nPy |
| 91 |
|
|
bj = 1 + MOD(jj,nSy) |
| 92 |
|
|
bi = 1 + MOD(ii,nSx) |
| 93 |
|
|
C-- |
| 94 |
|
|
W2_tileProc(I) = np |
| 95 |
jmc |
1.2 |
W2_tileIndex(I)= bi + (bj-1)*nSx |
| 96 |
jmc |
1.1 |
W2_procTileList(bi,bj,np) = I |
| 97 |
|
|
IF ( np.EQ.thisProc ) W2_myTileList(bi,bj) = I |
| 98 |
|
|
J = J + 1 |
| 99 |
|
|
c W2_tileRank(I) = J |
| 100 |
|
|
ENDIF |
| 101 |
|
|
ENDDO |
| 102 |
|
|
IF ( J .NE. nSx*nSy*nPx*nPy ) THEN |
| 103 |
|
|
STOP |
| 104 |
|
|
& 'ERROR W2_MAP_PROCS: number of active tiles not =nPx*nSx*nPy*nSy' |
| 105 |
|
|
ENDIF |
| 106 |
|
|
|
| 107 |
|
|
C-- Check tile sizes |
| 108 |
|
|
iErr = 0 |
| 109 |
|
|
DO bj=1,nSy |
| 110 |
|
|
DO bi=1,nSx |
| 111 |
|
|
myTileId = W2_myTileList(bi,bj) |
| 112 |
|
|
tNx = exch2_tNx(myTileId) |
| 113 |
|
|
tNy = exch2_tNy(myTileId) |
| 114 |
|
|
IF ( tNx .NE. sNx ) THEN |
| 115 |
|
|
WRITE(msgBuf,'(3(A,I5))') |
| 116 |
|
|
& 'ERROR: S/R W2_MAP_PROCS Topology for tile', myTileId, |
| 117 |
|
|
& 'tNx=', tNx, ' is not equal to subgrid size sNx=', sNx |
| 118 |
|
|
CALL PRINT_MESSAGE(msgBuf, |
| 119 |
|
|
& errorMessageUnit, SQUEEZE_RIGHT, 1 ) |
| 120 |
|
|
iErr = iErr+1 |
| 121 |
|
|
ENDIF |
| 122 |
|
|
IF ( tNy .NE. sNy ) THEN |
| 123 |
|
|
WRITE(msgBuf,'(3(A,I5))') |
| 124 |
|
|
& 'ERROR: S/R W2_MAP_PROCS Topology for tile', myTileId, |
| 125 |
|
|
& 'tNy=', tNy, ' is not equal to subgrid size sNy=', sNy |
| 126 |
|
|
CALL PRINT_MESSAGE(msgBuf, |
| 127 |
|
|
& errorMessageUnit, SQUEEZE_RIGHT, 1 ) |
| 128 |
|
|
iErr = iErr+1 |
| 129 |
|
|
ENDIF |
| 130 |
|
|
ENDDO |
| 131 |
|
|
ENDDO |
| 132 |
|
|
IF ( iErr .NE. 0 ) THEN |
| 133 |
|
|
STOP 'ABNORMAL END: W2_MAP_PROCS' |
| 134 |
|
|
ENDIF |
| 135 |
|
|
|
| 136 |
|
|
C-- Print tiles connection for this process and set myCommonFlag : |
| 137 |
|
|
WRITE(msgBuf,'(A)') '===== W2 TILE TOPOLOGY =====' |
| 138 |
|
|
CALL PRINT_MESSAGE( msgBuf, W2_oUnit, SQUEEZE_BOTH, myThid ) |
| 139 |
|
|
DO bj=1,nSy |
| 140 |
|
|
DO bi=1,nSx |
| 141 |
|
|
myTileId = W2_myTileList(bi,bj) |
| 142 |
jmc |
1.2 |
WRITE(msgBuf,'(A,I5,A,2I4,2A,I3)') |
| 143 |
|
|
& ' TILE: ', myTileId,' (bi,bj=', bi, bj, ' )', |
| 144 |
|
|
& ', Nb of Neighbours =', exch2_nNeighbours(myTileId) |
| 145 |
jmc |
1.1 |
c WRITE(msgBuf,'(2(A,I5),A,I3)') ' TILE: ', myTileId, |
| 146 |
|
|
c & ' , rank=', W2_tileRank(myTileId), |
| 147 |
|
|
c & ' , Nb of Neighbours =', exch2_nNeighbours(myTileId) |
| 148 |
|
|
CALL PRINT_MESSAGE( msgBuf, W2_oUnit, SQUEEZE_RIGHT, myThid ) |
| 149 |
|
|
DO J=1,exch2_nNeighbours(myTileId) |
| 150 |
|
|
commFlag = 'M' |
| 151 |
jmc |
1.2 |
jj = exch2_neighbourId(J,myTileId) |
| 152 |
|
|
IF ( W2_tileProc(jj).EQ.thisProc ) commFlag = 'P' |
| 153 |
jmc |
1.1 |
IF ( commFlag .EQ. 'M' ) THEN |
| 154 |
|
|
WRITE(msgBuf,'(A,I3,A,I5,A,I3,2A,I5,A)') |
| 155 |
|
|
& ' NEIGHBOUR',J,' = TILE', exch2_neighbourId(J,myTileId), |
| 156 |
|
|
& ' (n=', exch2_opposingSend(J,myTileId), ') Comm = MSG', |
| 157 |
|
|
& ' (PROC=',W2_tileProc(exch2_neighbourId(J,myTileId)),')' |
| 158 |
|
|
CALL PRINT_MESSAGE( msgBuf, W2_oUnit, SQUEEZE_RIGHT, myThid ) |
| 159 |
|
|
ENDIF |
| 160 |
|
|
IF ( commFlag .EQ. 'P' ) THEN |
| 161 |
|
|
WRITE(msgBuf,'(A,I3,A,I5,A,I3,2A,I5,A)') |
| 162 |
|
|
& ' NEIGHBOUR',J,' = TILE', exch2_neighbourId(J,myTileId), |
| 163 |
|
|
& ' (n=', exch2_opposingSend(J,myTileId), ') Comm = PUT', |
| 164 |
|
|
& ' (PROC=',W2_tileProc(exch2_neighbourId(J,myTileId)),')' |
| 165 |
|
|
CALL PRINT_MESSAGE( msgBuf, W2_oUnit, SQUEEZE_RIGHT, myThid ) |
| 166 |
|
|
ENDIF |
| 167 |
|
|
W2_myCommFlag(J,bi,bj) = commFlag |
| 168 |
|
|
ENDDO |
| 169 |
|
|
ENDDO |
| 170 |
|
|
ENDDO |
| 171 |
|
|
|
| 172 |
|
|
RETURN |
| 173 |
|
|
END |
Parent Directory
| 
Revision Log
| 
Revision Graph