35
36
37
38
39 USE spmd_comm_world_mod, ONLY : spmd_comm_world
40#include "implicit_f.inc"
41
42
43
44#include "spmd.inc"
45
46
47
48#include "com01_c.inc"
49#include "task_c.inc"
50
51
52
53 INTEGER LCOMM, ISIZE, FR_M(*), IAD_M(*),IROT
55 . a(3,*), ar(3,*), ms(*), in(*),
56 . stifn(*), stifr(*)
57
58
59
60#ifdef MPI
61 INTEGER MSGTYP,LOC_PROC,NOD,I,J,L,IDEB,IAD,LEN,
62 . NBINDEX,INDEX,MSGOFF,SIZ,IERROR,
63 . STATUS(MPI_STATUS_SIZE),
64 . REQ_S(NSPMD),REQ_R(NSPMD),INDEXI(NSPMD)
65 DATA msgoff/207/
67 . sbuf(lcomm*isize), rbuf(lcomm*isize)
68
69
70
71 loc_proc = ispmd + 1
72
73 ideb = 1
74 l = 0
75 DO i = 1, nspmd
76 len = iad_m(i+1)-iad_m(i)
77 IF(len>0) THEN
78 siz = len*isize
79 l=l+1
80 indexi(l)=i
81 msgtyp = msgoff
83 s rbuf(ideb),siz,real,it_spmd(i),msgtyp,
84 g spmd_comm_world,req_r(l),ierror)
85 ideb = ideb + siz
86 ENDIF
87 ENDDO
88 nbindex = l
89
90 ideb = 1
91 DO l = 1, nbindex
92 i = indexi(l)
93 len = iad_m(i+1) - iad_m(i)
94 iad = iad_m(i)-1
95 IF (irot==0) THEN
96#include "vectorize.inc"
97 DO j = 1, len
98 nod = fr_m(iad+j)
99 sbuf(ideb) = a(1,nod)
100 sbuf(ideb+1) = a(2,nod)
101 sbuf(ideb+2) = a(3,nod)
102 sbuf(ideb+3) = ms(nod)
103 sbuf(ideb+4) = stifn(nod)
104 ideb = ideb + isize
105 ENDDO
106 ELSE
107#include "vectorize.inc"
108 DO j = 1, len
109 nod = fr_m(iad+j)
110 sbuf(ideb) = a(1,nod)
111 sbuf(ideb+1) = a(2,nod)
112 sbuf(ideb+2) = a(3,nod)
113 sbuf(ideb+3) = ar(1,nod)
114 sbuf(ideb+4) = ar(2,nod)
115 sbuf(ideb+5) = ar(3,nod)
116 sbuf(ideb+6) = ms(nod)
117 sbuf(ideb+7) = in(nod)
118 sbuf(ideb+8) = stifn(nod)
119 sbuf(ideb+9) = stifr(nod)
120 ideb = ideb + isize
121 ENDDO
122 ENDIF
123 ENDDO
124
125 ideb = 1
126 DO l=1,nbindex
127 i = indexi(l)
128 len = iad_m(i+1)-iad_m(i)
129 siz = len*isize
130 msgtyp = msgoff
132 s sbuf(ideb),siz,real,it_spmd(i),msgtyp,
133 g spmd_comm_world,req_s(l),ierror)
134 ideb = ideb + siz
135 ENDDO
136
137 DO l=1,nbindex
138 CALL mpi_waitany(nbindex,req_r,index,status,ierror)
139 i = indexi(index)
140 ideb = 1+(iad_m(i)-1)*isize
141 len = iad_m(i+1)-iad_m(i)
142 iad = iad_m(i)-1
143 IF (irot==0) THEN
144#include "vectorize.inc"
145 DO j = 1, len
146 nod = fr_m(iad+j)
147 a(1,nod) = a(1,nod) + rbuf(ideb)
148 a(2,nod) = a(2,nod) + rbuf(ideb+1)
149 a(3,nod) = a(3,nod) + rbuf(ideb+2)
150 ms(nod) = ms(nod) + rbuf(ideb+3)
151 stifn(nod) = stifn(nod)+rbuf(ideb+4)
152 ideb = ideb + isize
153 ENDDO
154 ELSE
155#include "vectorize.inc"
156 DO j = 1, len
157 nod = fr_m(iad+j)
158 a(1,nod) = a(1,nod) + rbuf(ideb)
159 a(2,nod) = a(2,nod) + rbuf(ideb+1)
160 a(3,nod) = a(3,nod) + rbuf(ideb+2)
161 ar(1,nod) = ar(1,nod)+ rbuf(ideb+3)
162 ar(2,nod) = ar(2,nod)+ rbuf(ideb+4)
163 ar(3,nod) = ar(3,nod)+ rbuf(ideb+5)
164 ms(nod) = ms(nod) + rbuf(ideb+6)
165 in(nod) = in(nod) + rbuf(ideb+7)
166 stifn(nod) = stifn(nod)+rbuf(ideb+8)
167 stifr(nod) = stifr(nod)+rbuf(ideb+9)
168 ideb = ideb + isize
169 END DO
170 END IF
171 ENDDO
172
173 DO l=1,nbindex
174 CALL mpi_waitany(nbindex,req_s,index,status,ierror)
175 ENDDO
176
177#endif
178 RETURN
subroutine mpi_isend(buf, cnt, datatype, dest, tag, comm, ireq, ierr)
subroutine mpi_waitany(cnt, array_of_requests, index, status, ierr)
subroutine mpi_irecv(buf, cnt, datatype, source, tag, comm, ireq, ierr)