33 USE spmd_mod
34
35
36
37#include "implicit_f.inc"
38
39
40
41#include "spmd.inc"
42
43
44
45#include "com01_c.inc"
46#include "task_c.inc"
47
48
49
50 INTEGER NODNX_SMS(*), SIZE,
51 . FR_RMS(+1), FR_SMS(NSPMD+1),LIST_RMS(*), LIST_SMS(*)
53 . v(SIZE,*), vfi(SIZE,*)
54
55
56
57#ifdef MPI
58 INTEGER MSGTYP,I,NOD,LOC_PROC,IERROR,MSGOFF,
59 . SIZ,J,L,NB_NOD,K,M,KK,
60 . STATUS(MPI_STATUS_SIZE),
61 . IAD_SEND(NSPMD+1),IAD_RECV(NSPMD+1),IADFI_SEND(NSPMD+1),
62 . REQ_R(NSPMD),REQ_S()
64 . rbuf(size*(fr_rms(nspmd+1)+fr_sms(nspmd+1))),
65 . sbuf(size*(fr_rms(nspmd+1)+fr_sms(nspmd+1)))
66 DATA msgoff/215/
67
68
69
70 loc_proc = ispmd + 1
71 l = 1
72 iad_recv(1) = 1
73 DO i=1,nspmd
74
75 siz = fr_rms(i+1) - fr_rms(i)
76 IF(i/=loc_proc)THEN
77 siz = siz + fr_sms(i+1) - fr_sms(i)
78 ENDIF
79 siz = size*siz
80
81 IF(siz/=0)THEN
82 msgtyp = msgoff
84 s rbuf(l),siz,real,it_spmd(i),msgtyp,
85 g spmd_comm_world,req_r(i),ierror)
86 ENDIF
87 l = l + siz
88 iad_recv(i+1) = l
89
90 END DO
91
92 l = 1
93 iad_send(1) = 1
94 m = 1
95 DO i=1,nspmd
96 IF(i/=loc_proc)THEN
97#include "vectorize.inc"
98 DO j=fr_sms(i),fr_sms(i+1)-1
99 nod = list_sms(m)
100 IF(nod/=0)THEN
101 DO k=1,SIZE
102 sbuf(l+k-1) = v(k,nod)
103 END DO
104 ELSE
105 DO k=1,SIZE
106 sbuf(l+k-1) = zero
107 END DO
108 END IF
109 l = l + SIZE
110 m = m + 1
111 END DO
112 END IF
113#include "vectorize.inc"
114 DO j=fr_rms(i),fr_rms(i+1)-1
115 nod = list_rms(j)
116 IF(nod/=0)THEN
117 DO k=1,SIZE
118 sbuf(l+k-1) = v(k,nod)
119 END DO
120 ELSE
121 DO k=1,SIZE
122 sbuf(l+k-1) = zero
123 END DO
124 END IF
125 l = l + SIZE
126 END DO
127 iad_send(i+1) = l
128 ENDDO
129
130
131
132 DO i=1,nspmd
133
134 IF(iad_send(i+1)-iad_send(i)>0)THEN
135 msgtyp = msgoff
136 siz = iad_send(i+1)-iad_send(i)
137 l = iad_send(i)
139 s sbuf(l),siz,real,it_spmd(i),msgtyp,
140 g spmd_comm_world,req_s(i),ierror)
141 ENDIF
142
143 ENDDO
144
145 m = 1
146 kk= 1
147 DO i = 1, nspmd
148 IF(iad_recv(i+1)-iad_recv(i)>0)THEN
149 CALL mpi_wait(req_r(i),status,ierror)
150 l = iad_recv(i)
151#include "vectorize.inc"
152 DO j=fr_rms(i),fr_rms(i+1)-1
153 IF(list_rms(j)/=0)THEN
154 DO k=1,SIZE
155 vfi(k,kk) = rbuf(l+k-1)
156 END DO
157 END IF
158 l = l + SIZE
159 kk= kk + 1
160 END DO
161 IF(i/=loc_proc)THEN
162#include "vectorize.inc"
163 DO j=fr_sms(i),fr_sms(i+1)-1
164 IF(list_sms(m)/=0)THEN
165 DO k=1,SIZE
166 vfi(k,kk) = rbuf(l+k-1)
167 END DO
168 END IF
169 l = l + SIZE
170 m = m + 1
171 kk= kk + 1
172 END DO
173 END IF
174 ENDIF
175 END DO
176
177 DO i = 1, nspmd
178 IF(iad_send(i+1)-iad_send(i)>0)THEN
179 CALL mpi_wait(req_s(i),status,ierror)
180 ENDIF
181 ENDDO
182
183#endif
184 RETURN
subroutine mpi_isend(buf, cnt, datatype, dest, tag, comm, ireq, ierr)
subroutine mpi_wait(ireq, status, ierr)
subroutine mpi_irecv(buf, cnt, datatype, source, tag, comm, ireq, ierr)