1 | MODULE mpp_map |
---|
2 | !!====================================================================== |
---|
3 | !! *** MODULE mpp_mpa *** |
---|
4 | !! NEMOVAR: MPP global grid point mapping to processors |
---|
5 | !!====================================================================== |
---|
6 | !! History : 2.0 ! 2007-08 (K. Mogensen) Original code |
---|
7 | !!---------------------------------------------------------------------- |
---|
8 | |
---|
9 | !!---------------------------------------------------------------------- |
---|
10 | !! mppmap_init : Initialize mppmap. |
---|
11 | !!---------------------------------------------------------------------- |
---|
12 | USE par_kind, ONLY : wp ! Precision variables |
---|
13 | USE par_oce , ONLY : jpi, jpj, Nis0, Nie0, Njs0, Nje0 ! Ocean parameters |
---|
14 | USE dom_oce , ONLY : mig, mjg, narea ! Ocean space and time domain variables |
---|
15 | #if defined key_mpp_mpi |
---|
16 | USE lib_mpp , ONLY : mpi_comm_oce ! MPP library |
---|
17 | #endif |
---|
18 | USE in_out_manager ! I/O manager |
---|
19 | |
---|
20 | IMPLICIT NONE |
---|
21 | PRIVATE |
---|
22 | |
---|
23 | PUBLIC :: mppmap_init, mppmap !: ??? |
---|
24 | |
---|
25 | INTEGER, DIMENSION(:,:), ALLOCATABLE :: mppmap ! ??? |
---|
26 | |
---|
27 | !!---------------------------------------------------------------------- |
---|
28 | !! NEMO/OCE 4.0 , NEMO Consortium (2018) |
---|
29 | !! $Id$ |
---|
30 | !! Software governed by the CeCILL license (see ./LICENSE) |
---|
31 | !!---------------------------------------------------------------------- |
---|
32 | CONTAINS |
---|
33 | |
---|
34 | SUBROUTINE mppmap_init |
---|
35 | !!---------------------------------------------------------------------- |
---|
36 | !! *** ROUTINE mppmap_init *** |
---|
37 | !! |
---|
38 | !! ** Purpose : Setup a global map of processor rank for all gridpoints |
---|
39 | !! |
---|
40 | !! ** Method : MPI all reduce. |
---|
41 | !! |
---|
42 | !! ** Action : This does only work for MPI. |
---|
43 | !! |
---|
44 | !! References : http://www.mpi-forum.org |
---|
45 | !!---------------------------------------------------------------------- |
---|
46 | INTEGER, DIMENSION(:,:), ALLOCATABLE :: imppmap ! |
---|
47 | #if defined key_mpp_mpi |
---|
48 | INTEGER :: ierr |
---|
49 | |
---|
50 | INCLUDE 'mpif.h' |
---|
51 | #endif |
---|
52 | !!---------------------------------------------------------------------- |
---|
53 | |
---|
54 | IF (.NOT. ALLOCATED(mppmap)) THEN |
---|
55 | ALLOCATE( & |
---|
56 | & mppmap(jpiglo,jpjglo) & |
---|
57 | & ) |
---|
58 | ENDIF |
---|
59 | ! Initialize local imppmap |
---|
60 | |
---|
61 | ALLOCATE( & |
---|
62 | & imppmap(jpiglo,jpjglo) & |
---|
63 | & ) |
---|
64 | imppmap(:,:) = 0 |
---|
65 | |
---|
66 | ! ! Setup local grid points |
---|
67 | imppmap(mig(1):mig(jpi),mjg(1):mjg(jpj)) = narea |
---|
68 | |
---|
69 | ! Get global data |
---|
70 | |
---|
71 | #if defined key_mpp_mpi |
---|
72 | |
---|
73 | ! Call the MPI library to find the max across processors |
---|
74 | CALL mpi_allreduce( imppmap, mppmap, jpiglo*jpjglo, mpi_integer, & |
---|
75 | & mpi_max, mpi_comm_oce, ierr ) |
---|
76 | #else |
---|
77 | |
---|
78 | ! No MPP: Just copy the data |
---|
79 | mppmap(:,:) = imppmap(:,:) |
---|
80 | #endif |
---|
81 | ! |
---|
82 | END SUBROUTINE mppmap_init |
---|
83 | |
---|
84 | !!====================================================================== |
---|
85 | END MODULE mpp_map |
---|