1 | ! !== IN: ptab is an array ==! |
---|
2 | # if defined REAL_TYPE |
---|
3 | # define ARRAY_TYPE(i) REAL(wp) , INTENT(inout) :: ARRAY_IN(i) |
---|
4 | # define TMP_TYPE(i) REAL(wp) , ALLOCATABLE :: work(i) |
---|
5 | # define MPI_TYPE mpi_double_precision |
---|
6 | # endif |
---|
7 | # if defined INTEGER_TYPE |
---|
8 | # define ARRAY_TYPE(i) INTEGER , INTENT(inout) :: ARRAY_IN(i) |
---|
9 | # define TMP_TYPE(i) INTEGER , ALLOCATABLE :: work(i) |
---|
10 | # define MPI_TYPE mpi_integer |
---|
11 | # endif |
---|
12 | # if defined COMPLEX_TYPE |
---|
13 | # define ARRAY_TYPE(i) COMPLEX , INTENT(inout) :: ARRAY_IN(i) |
---|
14 | # define TMP_TYPE(i) COMPLEX , ALLOCATABLE :: work(i) |
---|
15 | # define MPI_TYPE mpi_double_complex |
---|
16 | # endif |
---|
17 | # if defined DIM_0d |
---|
18 | # define ARRAY_IN(i) ptab |
---|
19 | # define I_SIZE(ptab) 1 |
---|
20 | # endif |
---|
21 | # if defined DIM_1d |
---|
22 | # define ARRAY_IN(i) ptab(i) |
---|
23 | # define I_SIZE(ptab) SIZE(ptab,1) |
---|
24 | # endif |
---|
25 | # if defined OPERATION_MAX |
---|
26 | # define MPI_OPERATION mpi_max |
---|
27 | # endif |
---|
28 | # if defined OPERATION_MIN |
---|
29 | # define MPI_OPERATION mpi_min |
---|
30 | # endif |
---|
31 | # if defined OPERATION_SUM |
---|
32 | # define MPI_OPERATION mpi_sum |
---|
33 | # endif |
---|
34 | # if defined OPERATION_SUM_DD |
---|
35 | # define MPI_OPERATION mpi_sumdd |
---|
36 | # endif |
---|
37 | |
---|
38 | SUBROUTINE ROUTINE_ALLREDUCE( cdname, ptab, kdim, kcom ) |
---|
39 | !!---------------------------------------------------------------------- |
---|
40 | CHARACTER(len=*), INTENT(in ) :: cdname ! name of the calling subroutine |
---|
41 | ARRAY_TYPE(:) ! array or pointer of arrays on which the boundary condition is applied |
---|
42 | INTEGER, OPTIONAL, INTENT(in ) :: kdim ! optional pointer dimension |
---|
43 | INTEGER, OPTIONAL, INTENT(in ) :: kcom ! optional communicator |
---|
44 | #if defined key_mpp_mpi |
---|
45 | ! |
---|
46 | INTEGER :: ipi, ii, ierr |
---|
47 | INTEGER :: ierror, ilocalcomm |
---|
48 | TMP_TYPE(:) |
---|
49 | !!----------------------------------------------------------------------- |
---|
50 | ! |
---|
51 | IF( narea == 1 .AND. numcom == -1 ) CALL mpp_report( cdname, ld_glb = .TRUE. ) |
---|
52 | ! |
---|
53 | ilocalcomm = mpi_comm_oce |
---|
54 | IF( PRESENT(kcom) ) ilocalcomm = kcom |
---|
55 | ! |
---|
56 | IF( PRESENT(kdim) ) then |
---|
57 | ipi = kdim |
---|
58 | ELSE |
---|
59 | ipi = I_SIZE(ptab) ! 1st dimension |
---|
60 | ENDIF |
---|
61 | ! |
---|
62 | ALLOCATE(work(ipi)) |
---|
63 | CALL mpi_allreduce( ARRAY_IN(:), work, ipi, MPI_TYPE, MPI_OPERATION, ilocalcomm, ierror ) |
---|
64 | DO ii = 1, ipi |
---|
65 | ARRAY_IN(ii) = work(ii) |
---|
66 | ENDDO |
---|
67 | DEALLOCATE(work) |
---|
68 | #else |
---|
69 | ! nothing to do if non-mpp case |
---|
70 | RETURN |
---|
71 | #endif |
---|
72 | |
---|
73 | END SUBROUTINE ROUTINE_ALLREDUCE |
---|
74 | |
---|
75 | #undef ARRAY_TYPE |
---|
76 | #undef ARRAY_IN |
---|
77 | #undef I_SIZE |
---|
78 | #undef MPI_OPERATION |
---|
79 | #undef TMP_TYPE |
---|
80 | #undef MPI_TYPE |
---|