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 NRTM,NUMNOD,LENR,LENT,MAXCC,
52 . IRECT(4,NRTM),ITAG(*),
53 . IAD_ELEM(2,*), FR_ELEM(*), ISDSIZ(*), IRCSIZ(*)
54
55
56
57
58#ifdef MPI
59 INTEGER I ,J ,N1, N2, N3, N4, IERROR, IAD, SIZ, NB,
60 . MSGTYP, LOC_PROC, MSGOFF,
61 . IBUFS(LENR), IBUFR(LENR),
62 . STATUS(MPI_STATUS_SIZE),REQ_R(NSPMD)
63 DATA msgoff/193/
64
65
66
67 loc_proc = ispmd+1
68
69 iad = 1
70 DO i=1,nspmd
71 ircsiz(i) = 0
72 siz = iad_elem(1,i+1)-iad_elem(1,i)
73 IF(siz>0)THEN
74 msgtyp = msgoff
76 s ibufr(iad),siz,mpi_integer,it_spmd(i),msgtyp,
77 g spmd_comm_world,req_r(i),ierror)
78 iad = iad + siz
79 ENDIF
80 END DO
81 DO n1=1,numnod
82 itag(n1) = 0
83 END DO
84
85 DO i=1,nrtm
86 n1 = irect(1,i)
87 n2 = irect(2,i)
88 n3 = irect(3,i)
89 n4 = irect(4,i)
90 itag(n1) = itag(n1) + 1
91 itag(n2) = itag(n2) + 1
92 itag(n3) = itag(n3) + 1
93 itag(n4) = itag(n4) + 1
94 END DO
95 DO i=1,nspmd
96 isdsiz(i) = 0
97 siz = iad_elem(1,i+1)-iad_elem(1,i)
98 IF(siz>0)THEN
99 iad = 1
100#include "vectorize.inc"
101 DO j=iad_elem(1,i),iad_elem(1,i+1)-1
102 n1 = fr_elem(j)
103 isdsiz(i) = isdsiz
104 ibufs(iad) = itag(n1)
105 iad = iad + 1
106 END DO
107 msgtyp = msgoff
109 s ibufs,siz,mpi_integer,it_spmd(i),msgtyp,
110 g spmd_comm_world,ierror)
111 END IF
112 END DO
113
114 isdsiz(nspmd+1) = 0
115 ircsiz(nspmd+1) = 0
116 iad = 0
117 DO i = 1, nspmd
118 siz = iad_elem(1,i+1)-iad_elem(1,i)
119 IF(siz>0)THEN
120 CALL mpi_wait(req_r(i),status,ierror)
121 DO j=iad_elem(1,i),iad_elem(1,i+1)-1
122 n1 = fr_elem(j)
123 iad = iad + 1
124 nb = ibufr(iad)
125 ircsiz(i) = ircsiz(i) + nb
126 itag(n1) = itag(n1) + nb
127 END DO
128 isdsiz(nspmd+1) = isdsiz(nspmd+1) + isdsiz(i)
129 ircsiz(nspmd+1) = ircsiz(nspmd+1) + ircsiz(i)
130 END IF
131 END DO
132
133 lent = 0
134 maxcc = 0
135 DO n1=1,numnod
136 lent = lent + itag(n1)
137 maxcc =
max(maxcc,itag(n1))
138 END DO
139
140#endif
141 RETURN
subroutine mpi_wait(ireq, status, ierr)
subroutine mpi_send(buf, cnt, datatype, dest, tag, comm, ierr)
subroutine mpi_irecv(buf, cnt, datatype, source, tag, comm, ireq, ierr)