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