37
38
39
40
41 USE spmd_comm_world_mod, ONLY : spmd_comm_world
42#include "implicit_f.inc"
43
44
45
46#include "spmd.inc"
47
48
49
50#include "com01_c.inc"
51#include "task_c.inc"
52
53
54
55 INTEGER IAD_ELEM(2,*),FR_ELEM(*), SIZE, LENR
57
58
59
60#ifdef MPI
61 INTEGER MSGTYP,I,NOD,LOC_PROC,IERROR,
62 . SIZ,J,K,L,NB_NOD,
63 . STATUS(MPI_STATUS_SIZE),
64 . IAD_SEND(NSPMD+1),IAD_RECV(NSPMD+1),
65 . REQ_R(NSPMD),REQ_S(NSPMD),MSGOFF
66
67 my_real,
DIMENSION(:),
ALLOCATABLE :: sbuf,rbuf
68
69 DATA msgoff/7000/
70
71
72
73 ALLOCATE(sbuf(lenr*3))
74 ALLOCATE(rbuf(lenr*3))
75
76 loc_proc = ispmd+1
77 l = 1
78 iad_recv(1)=1
79
80
81 DO i=1,nspmd
82 siz = 3*(iad_elem(1,i+1)-iad_elem(1,i))
83 IF(siz/=0) THEN
84 msgtyp = msgoff
86 s rbuf(l),siz,real,it_spmd(i),msgtyp,
87 g spmd_comm_world,req_r(i),ierror)
88 l = l+siz
89 END IF
90 iad_recv(i+1) = l
91 END DO
92
93
94 l=1
95 iad_send(1)=1
96 DO i=1,nspmd
97 DO j=iad_elem(1,i),iad_elem(1,i+1)-1
98 nod = fr_elem(j)
99 sbuf(l ) = xnorm(1,nod)
100 sbuf(l+1) = xnorm(2,nod)
101 sbuf(l+2) = xnorm(3,nod)
102 l = l+3
103 END DO
104 iad_send(i+1) = l
105 END DO
106
107 DO i=1,nspmd
108
109 IF(iad_elem(1,i+1)-iad_elem(1,i)>0)THEN
110
111 msgtyp = msgoff
112 siz = iad_send(i+1)-iad_send(i)
113 l = iad_send(i)
115 s sbuf(l),siz,real,it_spmd(i),msgtyp,
116 g spmd_comm_world,req_s(i),ierror)
117 ENDIF
118 END DO
119
120
121 DO i = 1, nspmd
122
123 nb_nod = iad_elem(1,i+1)-iad_elem(1,i)
124 IF(nb_nod>0)THEN
125 CALL mpi_wait(req_r(i),status,ierror)
126 l = iad_recv(i)
127
128 DO j=iad_elem(1,i),iad_elem(1,i+1)-1
129
130 nod = fr_elem(j)
131 xnorm(1,nod) = xnorm(1,nod) + rbuf(l)
132 xnorm(2,nod) = xnorm(2,nod) + rbuf(l+1)
133 xnorm(3,nod) = xnorm(3,nod) + rbuf(l+2)
134 l = l+3
135 ENDDO
136 ENDIF
137 ENDDO
138
139 DO i = 1, nspmd
140 IF(iad_elem(1,i+1)-iad_elem(1,i)>0)THEN
141 CALL mpi_wait(req_s(i),status,ierror)
142 ENDIF
143 ENDDO
144
145 DEALLOCATE(sbuf)
146 DEALLOCATE(rbuf)
147
148#endif
149 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)