source: codes/icosagcm/trunk/src/transfert_mpi.f90 @ 392

Last change on this file since 392 was 392, checked in by ymipsl, 8 years ago

Add mpi global fonction transfert : bcast_field

YM

File size: 59.5 KB
Line 
1MODULE transfert_mpi_mod
2USE genmod
3USE field_mod
4 
5  TYPE array
6    INTEGER,POINTER :: value(:)
7    INTEGER,POINTER :: sign(:)
8    INTEGER         :: domain
9    INTEGER         :: rank
10    INTEGER         :: tag
11    INTEGER         :: size
12    INTEGER         :: offset
13    INTEGER         :: ireq
14    INTEGER,POINTER :: buffer(:)
15    REAL,POINTER    :: buffer_r(:)
16    INTEGER,POINTER :: src_value(:)
17  END TYPE array
18 
19  TYPE t_buffer
20    REAL,POINTER    :: r(:)
21    INTEGER         :: size
22    INTEGER         :: rank
23  END TYPE t_buffer   
24   
25  TYPE t_request
26    INTEGER :: type_field
27    INTEGER :: max_size
28    INTEGER :: size
29    LOGICAL :: vector
30    INTEGER,POINTER :: src_domain(:)
31    INTEGER,POINTER :: src_i(:)
32    INTEGER,POINTER :: src_j(:)
33    INTEGER,POINTER :: src_ind(:)
34    INTEGER,POINTER :: target_ind(:)
35    INTEGER,POINTER :: target_i(:)
36    INTEGER,POINTER :: target_j(:)
37    INTEGER,POINTER :: target_sign(:)
38    INTEGER :: nrecv
39    TYPE(ARRAY),POINTER :: recv(:)
40    INTEGER :: nsend
41    INTEGER :: nreq_mpi
42    INTEGER :: nreq_send
43    INTEGER :: nreq_recv
44    TYPE(ARRAY),POINTER :: send(:)
45  END TYPE t_request
46 
47  TYPE(t_request),SAVE,POINTER :: req_i1(:)
48  TYPE(t_request),SAVE,POINTER :: req_e1_scal(:)
49  TYPE(t_request),SAVE,POINTER :: req_e1_vect(:)
50 
51  TYPE(t_request),SAVE,POINTER :: req_i0(:)
52  TYPE(t_request),SAVE,POINTER :: req_e0_scal(:)
53  TYPE(t_request),SAVE,POINTER :: req_e0_vect(:)
54
55  TYPE t_reorder
56    INTEGER :: ind
57    INTEGER :: rank
58    INTEGER :: tag
59    INTEGER :: isend
60  END TYPE t_reorder 
61 
62  TYPE t_message
63    CHARACTER(LEN=100) :: name ! for debug
64    TYPE(t_request), POINTER :: request(:)
65    INTEGER :: nreq
66    INTEGER :: nreq_send
67    INTEGER :: nreq_recv
68    TYPE(t_reorder), POINTER :: reorder(:)
69    INTEGER, POINTER :: mpi_req(:)
70    INTEGER, POINTER :: status(:,:)
71    TYPE(t_buffer),POINTER :: buffers(:) 
72    TYPE(t_field),POINTER :: field(:)
73    LOGICAL :: completed
74    LOGICAL :: pending
75    LOGICAL :: open      ! for debug
76    INTEGER :: number
77  END TYPE t_message
78
79
80  INTERFACE bcast_mpi
81    MODULE PROCEDURE bcast_mpi_c,                                                     &
82                     bcast_mpi_i,bcast_mpi_i1,bcast_mpi_i2,bcast_mpi_i3,bcast_mpi_i4, &
83                     bcast_mpi_r,bcast_mpi_r1,bcast_mpi_r2,bcast_mpi_r3,bcast_mpi_r4, &
84                     bcast_mpi_l,bcast_mpi_l1,bcast_mpi_l2,bcast_mpi_l3,bcast_mpi_l4
85  END INTERFACE
86
87
88 
89CONTAINS
90       
91     
92  SUBROUTINE init_transfert
93  USE domain_mod
94  USE dimensions
95  USE field_mod
96  USE metric
97  USE mpipara
98  USE mpi_mod
99  IMPLICIT NONE
100  INTEGER :: ind,i,j
101  LOGICAL ::ok
102 
103    CALL create_request(field_t,req_i1)
104
105    DO ind=1,ndomain
106      CALL swap_dimensions(ind)
107      DO i=ii_begin,ii_end+1
108        CALL request_add_point(ind,i,jj_begin-1,req_i1)
109      ENDDO
110
111      DO j=jj_begin,jj_end
112        CALL request_add_point(ind,ii_end+1,j,req_i1)
113      ENDDO   
114      DO i=ii_begin,ii_end
115        CALL request_add_point(ind,i,jj_end+1,req_i1)
116      ENDDO   
117
118      DO j=jj_begin,jj_end+1
119        CALL request_add_point(ind,ii_begin-1,j,req_i1)
120      ENDDO   
121   
122    ENDDO
123 
124    CALL finalize_request(req_i1)
125
126
127    CALL create_request(field_t,req_i0)
128
129    DO ind=1,ndomain
130      CALL swap_dimensions(ind)
131   
132      DO i=ii_begin,ii_end
133        CALL request_add_point(ind,i,jj_begin,req_i0)
134      ENDDO
135
136      DO j=jj_begin,jj_end
137        CALL request_add_point(ind,ii_end,j,req_i0)
138      ENDDO   
139   
140      DO i=ii_begin,ii_end
141        CALL request_add_point(ind,i,jj_end,req_i0)
142      ENDDO   
143
144      DO j=jj_begin,jj_end
145        CALL request_add_point(ind,ii_begin,j,req_i0)
146      ENDDO   
147   
148    ENDDO
149 
150    CALL finalize_request(req_i0) 
151
152
153    CALL create_request(field_u,req_e1_scal)
154    DO ind=1,ndomain
155      CALL swap_dimensions(ind)
156      DO i=ii_begin,ii_end
157        CALL request_add_point(ind,i,jj_begin-1,req_e1_scal,rup)
158        CALL request_add_point(ind,i+1,jj_begin-1,req_e1_scal,lup)
159      ENDDO
160
161      DO j=jj_begin,jj_end
162        CALL request_add_point(ind,ii_end+1,j,req_e1_scal,left)
163      ENDDO   
164      DO j=jj_begin,jj_end
165        CALL request_add_point(ind,ii_end+1,j-1,req_e1_scal,lup)
166      ENDDO   
167   
168      DO i=ii_begin,ii_end
169        CALL request_add_point(ind,i,jj_end+1,req_e1_scal,ldown)
170        CALL request_add_point(ind,i-1,jj_end+1,req_e1_scal,rdown)
171      ENDDO   
172
173      DO j=jj_begin,jj_end
174        CALL request_add_point(ind,ii_begin-1,j,req_e1_scal,right)
175      ENDDO   
176      DO j=jj_begin,jj_end
177        CALL request_add_point(ind,ii_begin-1,j+1,req_e1_scal,rdown)
178      ENDDO   
179
180    ENDDO
181
182    CALL finalize_request(req_e1_scal)
183
184
185    CALL create_request(field_u,req_e0_scal)
186    DO ind=1,ndomain
187      CALL swap_dimensions(ind)
188
189
190      DO i=ii_begin+1,ii_end-1
191        CALL request_add_point(ind,i,jj_begin,req_e0_scal,right)
192        CALL request_add_point(ind,i,jj_end,req_e0_scal,right)
193      ENDDO
194   
195      DO j=jj_begin+1,jj_end-1
196        CALL request_add_point(ind,ii_begin,j,req_e0_scal,rup)
197        CALL request_add_point(ind,ii_end,j,req_e0_scal,rup)
198      ENDDO   
199
200      CALL request_add_point(ind,ii_begin+1,jj_begin,req_e0_scal,left)
201      CALL request_add_point(ind,ii_begin,jj_begin+1,req_e0_scal,ldown)
202      CALL request_add_point(ind,ii_begin+1,jj_end,req_e0_scal,left)
203      CALL request_add_point(ind,ii_end,jj_begin+1,req_e0_scal,ldown)
204
205    ENDDO
206
207    CALL finalize_request(req_e0_scal)
208
209
210   
211    CALL create_request(field_u,req_e1_vect,.TRUE.)
212    DO ind=1,ndomain
213      CALL swap_dimensions(ind)
214      DO i=ii_begin,ii_end
215        CALL request_add_point(ind,i,jj_begin-1,req_e1_vect,rup)
216        CALL request_add_point(ind,i+1,jj_begin-1,req_e1_vect,lup)
217      ENDDO
218
219      DO j=jj_begin,jj_end
220        CALL request_add_point(ind,ii_end+1,j,req_e1_vect,left)
221      ENDDO   
222      DO j=jj_begin,jj_end
223        CALL request_add_point(ind,ii_end+1,j-1,req_e1_vect,lup)
224      ENDDO   
225   
226      DO i=ii_begin,ii_end
227        CALL request_add_point(ind,i,jj_end+1,req_e1_vect,ldown)
228        CALL request_add_point(ind,i-1,jj_end+1,req_e1_vect,rdown)
229      ENDDO   
230
231      DO j=jj_begin,jj_end
232        CALL request_add_point(ind,ii_begin-1,j,req_e1_vect,right)
233      ENDDO   
234      DO j=jj_begin,jj_end
235        CALL request_add_point(ind,ii_begin-1,j+1,req_e1_vect,rdown)
236      ENDDO   
237
238 
239    ENDDO 
240
241    CALL finalize_request(req_e1_vect)
242   
243   
244    CALL create_request(field_u,req_e0_vect,.TRUE.)
245    DO ind=1,ndomain
246      CALL swap_dimensions(ind)
247 
248      DO i=ii_begin+1,ii_end-1
249        CALL request_add_point(ind,i,jj_begin,req_e0_vect,right)
250        CALL request_add_point(ind,i,jj_end,req_e0_vect,right)
251      ENDDO
252   
253      DO j=jj_begin+1,jj_end-1
254        CALL request_add_point(ind,ii_begin,j,req_e0_vect,rup)
255        CALL request_add_point(ind,ii_end,j,req_e0_vect,rup)
256      ENDDO   
257
258      CALL request_add_point(ind,ii_begin+1,jj_begin,req_e0_vect,left)
259      CALL request_add_point(ind,ii_begin,jj_begin+1,req_e0_vect,ldown)
260      CALL request_add_point(ind,ii_begin+1,jj_end,req_e0_vect,left)
261      CALL request_add_point(ind,ii_end,jj_begin+1,req_e0_vect,ldown)
262 
263    ENDDO 
264
265    CALL finalize_request(req_e0_vect)
266
267
268  END SUBROUTINE init_transfert
269 
270  SUBROUTINE create_request(type_field,request,vector)
271  USE domain_mod
272  USE field_mod
273  IMPLICIT NONE
274    INTEGER :: type_field
275    TYPE(t_request),POINTER :: request(:)
276    LOGICAL,OPTIONAL :: vector
277   
278    TYPE(t_request),POINTER :: req
279    TYPE(t_domain),POINTER :: d
280    INTEGER :: ind
281    INTEGER :: max_size
282       
283    ALLOCATE(request(ndomain))
284
285    DO ind=1,ndomain
286      req=>request(ind)
287      d=>domain(ind)
288      IF (type_field==field_t) THEN
289        Max_size=2*(d%iim+2)+2*(d%jjm+2)
290      ELSE IF (type_field==field_u) THEN
291        Max_size=3*(2*(d%iim+2)+2*(d%jjm+2))
292      ELSE IF (type_field==field_z) THEN
293        Max_size=2*(2*(d%iim+2)+2*(d%jjm+2))
294      ENDIF
295
296      req%type_field=type_field
297      req%max_size=max_size*2
298      req%size=0
299      req%vector=.FALSE.
300      IF (PRESENT(vector)) req%vector=vector
301      ALLOCATE(req%src_domain(req%max_size))
302      ALLOCATE(req%src_ind(req%max_size))
303      ALLOCATE(req%target_ind(req%max_size))
304      ALLOCATE(req%src_i(req%max_size))
305      ALLOCATE(req%src_j(req%max_size))
306      ALLOCATE(req%target_i(req%max_size))
307      ALLOCATE(req%target_j(req%max_size))
308      ALLOCATE(req%target_sign(req%max_size))
309    ENDDO
310 
311  END SUBROUTINE create_request
312
313  SUBROUTINE reallocate_request(req)
314  IMPLICIT NONE
315    TYPE(t_request),POINTER :: req
316     
317    INTEGER,POINTER :: src_domain(:)
318    INTEGER,POINTER :: src_ind(:)
319    INTEGER,POINTER :: target_ind(:)
320    INTEGER,POINTER :: src_i(:)
321    INTEGER,POINTER :: src_j(:)
322    INTEGER,POINTER :: target_i(:)
323    INTEGER,POINTER :: target_j(:)
324    INTEGER,POINTER :: target_sign(:)
325
326    PRINT *,"REALLOCATE_REQUEST"
327    src_domain=>req%src_domain
328    src_ind=>req%src_ind
329    target_ind=>req%target_ind
330    src_i=>req%src_i
331    src_j=>req%src_j
332    target_i=>req%target_i
333    target_j=>req%target_j
334    target_sign=>req%target_sign
335
336    ALLOCATE(req%src_domain(req%max_size*2))
337    ALLOCATE(req%src_ind(req%max_size*2))
338    ALLOCATE(req%target_ind(req%max_size*2))
339    ALLOCATE(req%src_i(req%max_size*2))
340    ALLOCATE(req%src_j(req%max_size*2))
341    ALLOCATE(req%target_i(req%max_size*2))
342    ALLOCATE(req%target_j(req%max_size*2))
343    ALLOCATE(req%target_sign(req%max_size*2))
344   
345    req%src_domain(1:req%max_size)=src_domain(:)
346    req%src_ind(1:req%max_size)=src_ind(:)
347    req%target_ind(1:req%max_size)=target_ind(:)
348    req%src_i(1:req%max_size)=src_i(:)
349    req%src_j(1:req%max_size)=src_j(:)
350    req%target_i(1:req%max_size)=target_i(:)
351    req%target_j(1:req%max_size)=target_j(:)
352    req%target_sign(1:req%max_size)=target_sign(:)
353   
354    req%max_size=req%max_size*2
355         
356    DEALLOCATE(src_domain)
357    DEALLOCATE(src_ind)
358    DEALLOCATE(target_ind)
359    DEALLOCATE(src_i)
360    DEALLOCATE(src_j)
361    DEALLOCATE(target_i)
362    DEALLOCATE(target_j)
363    DEALLOCATE(target_sign)
364
365  END SUBROUTINE reallocate_request
366
367     
368    SUBROUTINE request_add_point(ind,i,j,request,pos)
369    USE domain_mod
370    USE field_mod
371    IMPLICIT NONE
372      INTEGER,INTENT(IN)            ::  ind
373      INTEGER,INTENT(IN)            :: i
374      INTEGER,INTENT(IN)            :: j
375      TYPE(t_request),POINTER :: request(:)
376      INTEGER,INTENT(IN),OPTIONAL  :: pos
377     
378      INTEGER :: src_domain
379      INTEGER :: src_iim,src_i,src_j,src_n,src_pos,src_delta
380      TYPE(t_request),POINTER :: req
381      TYPE(t_domain),POINTER :: d
382     
383      req=>request(ind)
384      d=>domain(ind)
385     
386      IF (req%max_size==req%size) CALL reallocate_request(req)
387      req%size=req%size+1
388      IF (req%type_field==field_t) THEN
389        src_domain=domain(ind)%assign_domain(i,j)
390        src_iim=domain_glo(src_domain)%iim
391        src_i=domain(ind)%assign_i(i,j)
392        src_j=domain(ind)%assign_j(i,j)
393
394        req%target_ind(req%size)=(j-1)*d%iim+i
395        req%target_sign(req%size)=1
396        req%src_domain(req%size)=src_domain
397        req%src_ind(req%size)=(src_j-1)*src_iim+src_i
398      ELSE IF (req%type_field==field_u) THEN
399        IF (.NOT. PRESENT(pos)) STOP 'argument request_add_point non conforme'
400
401        src_domain=domain(ind)%edge_assign_domain(pos-1,i,j)
402        src_iim=domain_glo(src_domain)%iim
403        src_i=domain(ind)%edge_assign_i(pos-1,i,j)
404        src_j=domain(ind)%edge_assign_j(pos-1,i,j)
405        src_n=(src_j-1)*src_iim+src_i
406        src_delta=domain(ind)%delta(i,j)
407        src_pos=domain(ind)%edge_assign_pos(pos-1,i,j)+1
408               
409        req%target_ind(req%size)=(j-1)*d%iim+i+d%u_pos(pos)
410
411        req%target_sign(req%size)= 1
412        IF (req%vector) req%target_sign(req%size)= domain(ind)%edge_assign_sign(pos-1,i,j)
413
414        req%src_domain(req%size)=src_domain
415        req%src_ind(req%size)=src_n+domain_glo(src_domain)%u_pos(src_pos)
416
417      ELSE IF (req%type_field==field_z) THEN
418        IF (.NOT. PRESENT(pos)) STOP 'argument request_add_point non conforme'
419
420        src_domain=domain(ind)%assign_domain(i,j)
421        src_iim=domain_glo(src_domain)%iim
422        src_i=domain(ind)%assign_i(i,j)
423        src_j=domain(ind)%assign_j(i,j)
424        src_n=(src_j-1)*src_iim+src_i
425        src_delta=domain(ind)%delta(i,j)
426       
427        src_pos=MOD(pos-1+src_delta+6,6)+1
428       
429        req%target_ind(req%size)=(j-1)*d%iim+i+d%z_pos(pos)
430        req%target_sign(req%size)=1
431        req%src_domain(req%size)=src_domain
432        req%src_ind(req%size)=src_n+domain_glo(src_domain)%z_pos(src_pos)
433      ENDIF
434  END SUBROUTINE request_add_point
435 
436 
437  SUBROUTINE Finalize_request(request)
438  USE mpipara
439  USE domain_mod
440  USE mpi_mod
441  IMPLICIT NONE
442    TYPE(t_request),POINTER :: request(:)
443    TYPE(t_request),POINTER :: req, req_src
444    INTEGER :: nb_domain_recv(0:mpi_size-1)
445    INTEGER :: nb_domain_send(0:mpi_size-1)
446    INTEGER :: tag_rank(0:mpi_size-1)
447    INTEGER :: nb_data_domain_recv(ndomain_glo)
448    INTEGER :: list_domain_recv(ndomain_glo)
449    INTEGER,ALLOCATABLE :: list_domain_send(:)
450    INTEGER             :: list_domain(ndomain)
451
452    INTEGER :: rank,i,j,pos
453    INTEGER :: size_,ind_glo,ind_loc, ind_src
454    INTEGER :: isend, irecv, ireq, nreq, nsend, nrecv
455    INTEGER, ALLOCATABLE :: mpi_req(:)
456    INTEGER, ALLOCATABLE :: status(:,:)
457    INTEGER, ALLOCATABLE :: rank_list(:)
458    INTEGER, ALLOCATABLE :: offset(:)
459    LOGICAL,PARAMETER :: debug = .FALSE.
460
461 
462    IF (.NOT. using_mpi) RETURN
463   
464    DO ind_loc=1,ndomain
465      req=>request(ind_loc)
466     
467      nb_data_domain_recv(:) = 0
468      nb_domain_recv(:) = 0
469      tag_rank(:)=0
470     
471      DO i=1,req%size
472        ind_glo=req%src_domain(i)
473        nb_data_domain_recv(ind_glo)=nb_data_domain_recv(ind_glo)+1
474      ENDDO
475 
476      DO ind_glo=1,ndomain_glo
477        IF ( nb_data_domain_recv(ind_glo) > 0 )  nb_domain_recv(domglo_rank(ind_glo))=nb_domain_recv(domglo_rank(ind_glo))+1
478      ENDDO
479
480      req%nrecv=sum(nb_domain_recv(:))
481      ALLOCATE(req%recv(req%nrecv))
482
483      irecv=0
484      DO ind_glo=1,ndomain_glo
485        IF (nb_data_domain_recv(ind_glo)>0) THEN
486          irecv=irecv+1
487          list_domain_recv(ind_glo)=irecv
488          req%recv(irecv)%rank=domglo_rank(ind_glo)
489          req%recv(irecv)%size=nb_data_domain_recv(ind_glo)
490          req%recv(irecv)%domain=domglo_loc_ind(ind_glo)
491          ALLOCATE(req%recv(irecv)%value(req%recv(irecv)%size))
492          ALLOCATE(req%recv(irecv)%sign(req%recv(irecv)%size))
493          ALLOCATE(req%recv(irecv)%buffer(req%recv(irecv)%size))
494        ENDIF
495      ENDDO
496     
497      req%recv(:)%size=0
498      irecv=0
499      DO i=1,req%size
500        irecv=list_domain_recv(req%src_domain(i))
501        req%recv(irecv)%size=req%recv(irecv)%size+1
502        size_=req%recv(irecv)%size
503        req%recv(irecv)%value(size_)=req%src_ind(i)
504        req%recv(irecv)%buffer(size_)=req%target_ind(i)
505        req%recv(irecv)%sign(size_)=req%target_sign(i)
506      ENDDO
507    ENDDO
508
509    nb_domain_recv(:) = 0   
510    DO ind_loc=1,ndomain
511      req=>request(ind_loc)
512     
513      DO irecv=1,req%nrecv
514        rank=req%recv(irecv)%rank
515        nb_domain_recv(rank)=nb_domain_recv(rank)+1
516      ENDDO
517    ENDDO
518   
519    CALL MPI_Alltoall(nb_domain_recv,1,MPI_INTEGER,nb_domain_send,1,MPI_INTEGER,comm_icosa,ierr)     
520   
521
522    ALLOCATE(list_domain_send(sum(nb_domain_send)))
523   
524    nreq=sum(nb_domain_recv(:))+sum(nb_domain_send(:))
525    ALLOCATE(mpi_req(nreq))
526    ALLOCATE(status(MPI_STATUS_SIZE,nreq))
527   
528
529    ireq=0
530    DO ind_loc=1,ndomain
531      req=>request(ind_loc)
532      DO irecv=1,req%nrecv
533        ireq=ireq+1
534        CALL MPI_ISEND(req%recv(irecv)%domain,1,MPI_INTEGER,req%recv(irecv)%rank,0,comm_icosa, mpi_req(ireq),ierr)
535        IF (debug) PRINT *,"Isend ",req%recv(irecv)%domain, "from ", mpi_rank, "to ",req%recv(irecv)%rank, "tag ",0
536      ENDDO
537    ENDDO
538
539    IF (debug) PRINT *,"------------"   
540    j=0
541    DO rank=0,mpi_size-1
542      DO i=1,nb_domain_send(rank)
543        j=j+1
544        ireq=ireq+1
545        CALL MPI_IRECV(list_domain_send(j),1,MPI_INTEGER,rank,0,comm_icosa, mpi_req(ireq),ierr)
546        IF (debug) PRINT *,"IRecv ",list_domain_send(j), "from ", rank, "to ",mpi_rank, "tag ",0
547      ENDDO
548    ENDDO
549    IF (debug) PRINT *,"------------"   
550   
551    CALL MPI_WAITALL(nreq,mpi_req,status,ierr)
552   
553    list_domain(:)=0
554    DO i=1,sum(nb_domain_send)
555      ind_loc=list_domain_send(i)
556      list_domain(ind_loc)=list_domain(ind_loc)+1
557    ENDDO
558   
559    DO ind_loc=1,ndomain
560      req=>request(ind_loc)
561      req%nsend=list_domain(ind_loc)
562      ALLOCATE(req%send(req%nsend))
563    ENDDO
564
565    IF (debug) PRINT *,"------------"   
566   
567   ireq=0 
568   DO ind_loc=1,ndomain
569     req=>request(ind_loc)
570     
571     DO irecv=1,req%nrecv
572       ireq=ireq+1
573       CALL MPI_ISEND(mpi_rank,1,MPI_INTEGER,req%recv(irecv)%rank,req%recv(irecv)%domain,comm_icosa, mpi_req(ireq),ierr)
574       IF (debug) PRINT *,"Isend ",mpi_rank, "from ", mpi_rank, "to ",req%recv(irecv)%rank,"tag ",req%recv(irecv)%domain
575     ENDDO
576    IF (debug) PRINT *,"------------"   
577     
578     DO isend=1,req%nsend
579       ireq=ireq+1
580       CALL MPI_IRECV(req%send(isend)%rank,1,MPI_INTEGER,MPI_ANY_SOURCE,ind_loc,comm_icosa, mpi_req(ireq),ierr)
581       IF (debug) PRINT *,"IRecv ",req%send(isend)%rank, "from ", "xxx", "to ",mpi_rank, "tag ",ind_loc
582     ENDDO
583   ENDDO
584
585   IF (debug) PRINT *,"------------"   
586
587   CALL MPI_WAITALL(nreq,mpi_req,status,ierr)
588   CALL MPI_BARRIER(comm_icosa,ierr)
589
590   IF (debug) PRINT *,"------------"   
591
592   ireq=0 
593   DO ind_loc=1,ndomain
594     req=>request(ind_loc)
595     
596     DO irecv=1,req%nrecv
597       ireq=ireq+1
598       CALL MPI_ISEND(ind_loc,1,MPI_INTEGER,req%recv(irecv)%rank,req%recv(irecv)%domain,comm_icosa, mpi_req(ireq),ierr)
599       IF (debug) PRINT *,"Isend ",ind_loc, "from ", mpi_rank, "to ",req%recv(irecv)%rank,"tag ",req%recv(irecv)%domain
600     ENDDO
601
602     IF (debug) PRINT *,"------------"   
603     
604     DO isend=1,req%nsend
605       ireq=ireq+1
606       CALL MPI_IRECV(req%send(isend)%domain,1,MPI_INTEGER,req%send(isend)%rank,ind_loc,comm_icosa, mpi_req(ireq),ierr)
607       IF (debug) PRINT *,"IRecv ",req%send(isend)%domain, "from ", req%send(isend)%rank, "to ",mpi_rank, "tag ",ind_loc
608     ENDDO
609   ENDDO
610   IF (debug) PRINT *,"------------"   
611   
612   CALL MPI_WAITALL(nreq,mpi_req,status,ierr)
613   CALL MPI_BARRIER(comm_icosa,ierr)
614   IF (debug) PRINT *,"------------"   
615
616   ireq=0
617   DO ind_loc=1,ndomain
618     req=>request(ind_loc)
619     
620     DO irecv=1,req%nrecv
621       ireq=ireq+1
622       req%recv(irecv)%tag=tag_rank(req%recv(irecv)%rank)
623       tag_rank(req%recv(irecv)%rank)=tag_rank(req%recv(irecv)%rank)+1
624       CALL MPI_ISEND(req%recv(irecv)%tag,1,MPI_INTEGER,req%recv(irecv)%rank,req%recv(irecv)%domain,comm_icosa, mpi_req(ireq),ierr)
625       IF (debug) PRINT *,"Isend ",req%recv(irecv)%tag, "from ", mpi_rank, "to ",req%recv(irecv)%rank,"tag ",req%recv(irecv)%domain
626     ENDDO
627   IF (debug) PRINT *,"------------"   
628     
629     DO isend=1,req%nsend
630       ireq=ireq+1
631       CALL MPI_IRECV(req%send(isend)%tag,1,MPI_INTEGER,req%send(isend)%rank,ind_loc,comm_icosa, mpi_req(ireq),ierr)
632       IF (debug) PRINT *,"IRecv ",req%send(isend)%tag, "from ", req%send(isend)%rank, "to ",mpi_rank, "tag ",ind_loc
633     ENDDO
634   ENDDO
635   IF (debug) PRINT *,"------------"   
636   
637   CALL MPI_WAITALL(nreq,mpi_req,status,ierr)
638   CALL MPI_BARRIER(comm_icosa,ierr)
639
640
641   IF (debug) PRINT *,"------------"   
642
643   ireq=0 
644   DO ind_loc=1,ndomain
645     req=>request(ind_loc)
646     
647     DO irecv=1,req%nrecv
648       ireq=ireq+1
649       CALL MPI_ISEND(req%recv(irecv)%size,1,MPI_INTEGER,req%recv(irecv)%rank,req%recv(irecv)%tag,comm_icosa, mpi_req(ireq),ierr)
650       IF (debug) PRINT *,"Isend ",req%recv(irecv)%size, "from ", mpi_rank, "to ",req%recv(irecv)%rank,"tag ",req%recv(irecv)%domain
651     ENDDO
652     IF (debug) PRINT *,"------------"   
653     
654     DO isend=1,req%nsend
655       ireq=ireq+1
656       CALL MPI_IRECV(req%send(isend)%size,1,MPI_INTEGER,req%send(isend)%rank,req%send(isend)%tag,comm_icosa, mpi_req(ireq),ierr)
657       IF (debug) PRINT *,"IRecv ",req%send(isend)%size, "from ", req%send(isend)%rank, "to ",mpi_rank, "tag ",ind_loc
658     ENDDO
659   ENDDO
660
661   CALL MPI_WAITALL(nreq,mpi_req,status,ierr)
662
663   ireq=0 
664   DO ind_loc=1,ndomain
665     req=>request(ind_loc)
666     
667     DO irecv=1,req%nrecv
668       ireq=ireq+1
669       CALL MPI_ISEND(req%recv(irecv)%value,req%recv(irecv)%size,MPI_INTEGER,&
670            req%recv(irecv)%rank,req%recv(irecv)%tag,comm_icosa, mpi_req(ireq),ierr)
671     ENDDO
672     
673     DO isend=1,req%nsend
674       ireq=ireq+1
675       ALLOCATE(req%send(isend)%value(req%send(isend)%size))
676       CALL MPI_IRECV(req%send(isend)%value,req%send(isend)%size,MPI_INTEGER,&
677            req%send(isend)%rank,req%send(isend)%tag,comm_icosa, mpi_req(ireq),ierr)
678     ENDDO
679   ENDDO
680
681   CALL MPI_WAITALL(nreq,mpi_req,status,ierr)
682
683   DO ind_loc=1,ndomain
684     req=>request(ind_loc)
685     
686     DO irecv=1,req%nrecv
687       req%recv(irecv)%value(:)=req%recv(irecv)%buffer(:)
688       req%recv(irecv)%sign(:) =req%recv(irecv)%sign(:)
689       DEALLOCATE(req%recv(irecv)%buffer)
690     ENDDO
691   ENDDO 
692   
693
694! domain is on the same mpi process => copie memory to memory
695   
696   DO ind_loc=1,ndomain
697     req=>request(ind_loc)
698     
699     DO irecv=1,req%nrecv
700   
701       IF (req%recv(irecv)%rank==mpi_rank) THEN
702           req_src=>request(req%recv(irecv)%domain)
703           DO isend=1,req_src%nsend
704             IF (req_src%send(isend)%rank==mpi_rank .AND. req_src%send(isend)%tag==req%recv(irecv)%tag) THEN
705               req%recv(irecv)%src_value => req_src%send(isend)%value
706               IF ( size(req%recv(irecv)%value) /= size(req_src%send(isend)%value)) THEN
707                 PRINT *,ind_loc, irecv, isend, size(req%recv(irecv)%value), size(req_src%send(isend)%value)
708                 STOP "size(req%recv(irecv)%value) /= size(req_src%send(isend)%value"
709               ENDIF
710             ENDIF
711           ENDDO
712       ENDIF
713     
714     ENDDO
715   ENDDO
716   
717! true number of mpi request
718
719   request(:)%nreq_mpi=0
720   request(:)%nreq_send=0
721   request(:)%nreq_recv=0
722   ALLOCATE(rank_list(sum(request(:)%nsend)))
723   ALLOCATE(offset(sum(request(:)%nsend)))
724   offset(:)=0
725   
726   nsend=0
727   DO ind_loc=1,ndomain
728     req=>request(ind_loc)
729
730     DO isend=1,req%nsend
731       IF (req%send(isend)%rank/=mpi_rank) THEN
732         pos=0
733         DO i=1,nsend
734           IF (req%send(isend)%rank==rank_list(i)) EXIT
735           pos=pos+1
736         ENDDO
737       
738         IF (pos==nsend) THEN
739           nsend=nsend+1
740           req%nreq_mpi=req%nreq_mpi+1
741           req%nreq_send=req%nreq_send+1
742           IF (mpi_threading_mode==MPI_THREAD_FUNNELED .OR. mpi_threading_mode==MPI_THREAD_SINGLE) THEN
743             rank_list(nsend)=req%send(isend)%rank
744           ELSE
745             rank_list(nsend)=-1
746           ENDIF
747         ENDIF
748         
749         pos=pos+1
750         req%send(isend)%ireq=pos
751         req%send(isend)%offset=offset(pos)
752         offset(pos)=offset(pos)+req%send(isend)%size
753       ENDIF
754     ENDDO
755   ENDDO
756
757   DEALLOCATE(rank_list)
758   DEALLOCATE(offset)
759     
760   ALLOCATE(rank_list(sum(request(:)%nrecv)))
761   ALLOCATE(offset(sum(request(:)%nrecv)))
762   offset(:)=0
763   
764   nrecv=0
765   DO ind_loc=1,ndomain
766     req=>request(ind_loc)
767
768     DO irecv=1,req%nrecv
769       IF (req%recv(irecv)%rank/=mpi_rank) THEN
770         pos=0
771         DO i=1,nrecv
772           IF (req%recv(irecv)%rank==rank_list(i)) EXIT
773           pos=pos+1
774         ENDDO
775       
776         IF (pos==nrecv) THEN
777           nrecv=nrecv+1
778           req%nreq_mpi=req%nreq_mpi+1
779           req%nreq_recv=req%nreq_recv+1
780           IF (mpi_threading_mode==MPI_THREAD_FUNNELED .OR. mpi_threading_mode==MPI_THREAD_SINGLE) THEN
781             rank_list(nrecv)=req%recv(irecv)%rank
782           ELSE
783             rank_list(nrecv)=-1
784           ENDIF
785         ENDIF
786       
787         pos=pos+1
788         req%recv(irecv)%ireq=nsend+pos
789         req%recv(irecv)%offset=offset(pos)
790         offset(pos)=offset(pos)+req%recv(irecv)%size
791       ENDIF
792     ENDDO
793   ENDDO 
794
795! get the offsets   
796
797   ireq=0 
798   DO ind_loc=1,ndomain
799     req=>request(ind_loc)
800     
801     DO irecv=1,req%nrecv
802       ireq=ireq+1
803       CALL MPI_ISEND(req%recv(irecv)%offset,1,MPI_INTEGER,&
804            req%recv(irecv)%rank,req%recv(irecv)%tag,comm_icosa, mpi_req(ireq),ierr)
805     ENDDO
806     
807     DO isend=1,req%nsend
808       ireq=ireq+1
809       CALL MPI_IRECV(req%send(isend)%offset,1,MPI_INTEGER,&
810            req%send(isend)%rank,req%send(isend)%tag,comm_icosa, mpi_req(ireq),ierr)
811     ENDDO
812   ENDDO
813
814   CALL MPI_WAITALL(nreq,mpi_req,status,ierr)
815     
816       
817  END SUBROUTINE Finalize_request 
818
819
820  SUBROUTINE init_message_seq(field, request, message, name)
821  USE field_mod
822  USE domain_mod
823  USE mpi_mod
824  USE mpipara
825  USE mpi_mod
826  IMPLICIT NONE
827    TYPE(t_field),POINTER :: field(:)
828    TYPE(t_request),POINTER :: request(:)
829    TYPE(t_message) :: message
830    CHARACTER(LEN=*), INTENT(IN), OPTIONAL :: name
831!$OMP MASTER   
832    message%request=>request
833    IF(PRESENT(name)) THEN
834       message%name = TRIM(name)
835    ELSE
836       message%name = 'unknown'
837    END IF
838!$OMP END MASTER   
839!$OMP BARRIER   
840
841  END SUBROUTINE init_message_seq
842
843  SUBROUTINE send_message_seq(field,message)
844  USE field_mod
845  USE domain_mod
846  USE mpi_mod
847  USE mpipara
848  USE omp_para
849  USE trace
850  IMPLICIT NONE
851    TYPE(t_field),POINTER :: field(:)
852    TYPE(t_message) :: message
853
854    CALL transfert_request_seq(field,message%request)
855   
856  END SUBROUTINE send_message_seq
857 
858  SUBROUTINE test_message_seq(message)
859  IMPLICIT NONE
860    TYPE(t_message) :: message
861  END SUBROUTINE  test_message_seq
862 
863   
864  SUBROUTINE wait_message_seq(message)
865  IMPLICIT NONE
866    TYPE(t_message) :: message
867   
868  END SUBROUTINE wait_message_seq   
869
870  SUBROUTINE transfert_message_seq(field,message)
871  USE field_mod
872  USE domain_mod
873  USE mpi_mod
874  USE mpipara
875  USE omp_para
876  USE trace
877  IMPLICIT NONE
878    TYPE(t_field),POINTER :: field(:)
879    TYPE(t_message) :: message
880
881   CALL send_message_seq(field,message)
882   
883  END SUBROUTINE transfert_message_seq   
884   
885
886
887   
888  SUBROUTINE init_message_mpi(field,request, message, name)
889  USE field_mod
890  USE domain_mod
891  USE mpi_mod
892  USE mpipara
893  USE mpi_mod
894  IMPLICIT NONE
895 
896    TYPE(t_field),POINTER :: field(:)
897    TYPE(t_request),POINTER :: request(:)
898    TYPE(t_message) :: message
899    CHARACTER(LEN=*), INTENT(IN),OPTIONAL :: name
900
901    TYPE(ARRAY),POINTER :: recv,send 
902    TYPE(t_request),POINTER :: req
903    INTEGER :: irecv,isend
904    INTEGER :: ireq,nreq, nreq_send
905    INTEGER :: ind
906    INTEGER :: dim3,dim4
907    INTEGER :: i,j
908    INTEGER,SAVE :: message_number=0
909!    TYPE(t_reorder),POINTER :: reorder(:)
910!    TYPE(t_reorder) :: reorder_swap
911
912!$OMP BARRIER
913!$OMP MASTER
914    IF(PRESENT(name)) THEN
915       message%name = TRIM(name)
916    ELSE
917       message%name = 'unknown'
918    END IF
919    message%number=message_number
920    message_number=message_number+1
921    IF (message_number==100) message_number=0
922
923 
924    message%request=>request
925    message%nreq=sum(message%request(:)%nreq_mpi)
926    message%nreq_send=sum(message%request(:)%nreq_send)
927    message%nreq_recv=sum(message%request(:)%nreq_recv)
928    nreq=message%nreq
929
930    ALLOCATE(message%mpi_req(nreq))
931    ALLOCATE(message%buffers(nreq))
932    ALLOCATE(message%status(MPI_STATUS_SIZE,nreq))
933    message%buffers(:)%size=0
934    message%pending=.FALSE.
935    message%completed=.FALSE.
936    message%open=.FALSE.
937
938    DO ind=1,ndomain
939      req=>request(ind)
940      DO isend=1,req%nsend
941        IF (req%send(isend)%rank/=mpi_rank) THEN
942          ireq=req%send(isend)%ireq 
943          message%buffers(ireq)%size=message%buffers(ireq)%size+req%send(isend)%size
944          message%buffers(ireq)%rank=req%send(isend)%rank
945        ENDIF
946      ENDDO
947      DO irecv=1,req%nrecv
948        IF (req%recv(irecv)%rank/=mpi_rank) THEN
949          ireq=req%recv(irecv)%ireq 
950          message%buffers(ireq)%size=message%buffers(ireq)%size+req%recv(irecv)%size
951          message%buffers(ireq)%rank=req%recv(irecv)%rank
952        ENDIF
953      ENDDO
954    ENDDO
955
956
957    IF (field(1)%data_type==type_real) THEN
958
959      IF (field(1)%ndim==2) THEN
960     
961        DO ireq=1,message%nreq
962          CALL allocate_mpi_buffer(message%buffers(ireq)%r,message%buffers(ireq)%size)
963        ENDDO
964     
965      ELSE  IF (field(1)%ndim==3) THEN
966     
967        dim3=size(field(1)%rval3d,2)
968        DO ireq=1,message%nreq
969          message%buffers(ireq)%size=message%buffers(ireq)%size*dim3
970          CALL allocate_mpi_buffer(message%buffers(ireq)%r,message%buffers(ireq)%size)
971        ENDDO
972     
973      ELSE  IF (field(1)%ndim==4) THEN
974        dim3=size(field(1)%rval4d,2)
975        dim4=size(field(1)%rval4d,3)
976        DO ireq=1,message%nreq
977          message%buffers(ireq)%size=message%buffers(ireq)%size*dim3*dim4
978          CALL allocate_mpi_buffer(message%buffers(ireq)%r,message%buffers(ireq)%size)
979        ENDDO
980      ENDIF     
981    ENDIF
982     
983         
984   
985! ! Reorder the request, so recv request are done in the same order than send request
986
987!    nreq_send=sum(request(:)%nsend) 
988!    message%nreq_send=nreq_send
989!    ALLOCATE(message%reorder(nreq_send))
990!    reorder=>message%reorder
991!    ireq=0
992!    DO ind=1,ndomain
993!      req=>request(ind)
994!      DO isend=1,req%nsend
995!        ireq=ireq+1
996!        reorder(ireq)%ind=ind
997!        reorder(ireq)%isend=isend
998!        reorder(ireq)%tag=req%send(isend)%tag
999!      ENDDO
1000!    ENDDO
1001
1002! ! do a very very bad sort
1003!    DO i=1,nreq_send-1
1004!      DO j=i+1,nreq_send
1005!        IF (reorder(i)%tag > reorder(j)%tag) THEN
1006!          reorder_swap=reorder(i)
1007!          reorder(i)=reorder(j)
1008!          reorder(j)=reorder_swap
1009!        ENDIF
1010!      ENDDO
1011!    ENDDO
1012!    PRINT *,"reorder ",reorder(:)%tag
1013   
1014 
1015!$OMP END MASTER
1016!$OMP BARRIER   
1017
1018  END SUBROUTINE init_message_mpi
1019 
1020  SUBROUTINE Finalize_message_mpi(field,message)
1021  USE field_mod
1022  USE domain_mod
1023  USE mpi_mod
1024  USE mpipara
1025  USE mpi_mod
1026  IMPLICIT NONE
1027    TYPE(t_field),POINTER :: field(:)
1028    TYPE(t_message) :: message
1029
1030    TYPE(t_request),POINTER :: req
1031    INTEGER :: irecv,isend
1032    INTEGER :: ireq,nreq
1033    INTEGER :: ind
1034
1035!$OMP BARRIER
1036!$OMP MASTER
1037
1038
1039    IF (field(1)%data_type==type_real) THEN
1040
1041      IF (field(1)%ndim==2) THEN
1042     
1043        DO ireq=1,message%nreq
1044          CALL free_mpi_buffer(message%buffers(ireq)%r)
1045        ENDDO
1046     
1047      ELSE  IF (field(1)%ndim==3) THEN
1048
1049        DO ireq=1,message%nreq
1050          CALL free_mpi_buffer(message%buffers(ireq)%r)
1051        ENDDO
1052     
1053      ELSE  IF (field(1)%ndim==4) THEN
1054
1055        DO ireq=1,message%nreq
1056          CALL free_mpi_buffer(message%buffers(ireq)%r)
1057        ENDDO
1058
1059      ENDIF     
1060    ENDIF
1061   
1062    DEALLOCATE(message%mpi_req)
1063    DEALLOCATE(message%buffers)
1064    DEALLOCATE(message%status)
1065
1066!$OMP END MASTER
1067!$OMP BARRIER
1068
1069     
1070  END SUBROUTINE Finalize_message_mpi
1071
1072
1073 
1074  SUBROUTINE barrier
1075  USE mpi_mod
1076  USE mpipara
1077  IMPLICIT NONE
1078   
1079    CALL MPI_BARRIER(comm_icosa,ierr)
1080   
1081  END SUBROUTINE barrier 
1082   
1083  SUBROUTINE transfert_message_mpi(field,message)
1084  USE field_mod
1085  IMPLICIT NONE
1086    TYPE(t_field),POINTER :: field(:)
1087    TYPE(t_message) :: message
1088   
1089    CALL send_message_mpi(field,message)
1090    CALL wait_message_mpi(message)
1091   
1092  END SUBROUTINE transfert_message_mpi
1093
1094
1095  SUBROUTINE send_message_mpi(field,message)
1096  USE field_mod
1097  USE domain_mod
1098  USE mpi_mod
1099  USE mpipara
1100  USE omp_para
1101  USE trace
1102  IMPLICIT NONE
1103    TYPE(t_field),POINTER :: field(:)
1104    TYPE(t_message) :: message
1105    REAL(rstd),POINTER :: rval2d(:), src_rval2d(:) 
1106    REAL(rstd),POINTER :: rval3d(:,:), src_rval3d(:,:) 
1107    REAL(rstd),POINTER :: rval4d(:,:,:), src_rval4d(:,:,:) 
1108    REAL(rstd),POINTER :: buffer_r(:) 
1109    INTEGER,POINTER :: value(:) 
1110    INTEGER,POINTER :: sgn(:) 
1111    TYPE(ARRAY),POINTER :: recv,send 
1112    TYPE(t_request),POINTER :: req
1113    INTEGER, ALLOCATABLE :: mpi_req(:)
1114    INTEGER, ALLOCATABLE :: status(:,:)
1115    INTEGER :: irecv,isend
1116    INTEGER :: ireq,nreq
1117    INTEGER :: ind,i,n,l,m
1118    INTEGER :: dim3,dim4,d3,d4
1119    INTEGER,POINTER :: src_value(:)
1120    INTEGER,POINTER :: sign(:)
1121    INTEGER :: offset,msize,rank
1122    INTEGER :: lbegin, lend
1123
1124!    CALL trace_start("send_message_mpi")
1125
1126!$OMP BARRIER
1127
1128
1129!$OMP MASTER
1130    IF(message%open) THEN
1131       PRINT *, 'send_message_mpi : message ' // TRIM(message%name) // &
1132            ' is still open, no call to wait_message_mpi after last send_message_mpi'
1133       CALL ABORT
1134    END IF
1135    message%open=.TRUE. ! will be set to .FALSE. by wait_message_mpi
1136
1137    message%field=>field
1138
1139    IF (message%nreq>0) THEN
1140      message%completed=.FALSE.
1141      message%pending=.TRUE.
1142    ELSE
1143      message%completed=.TRUE.
1144      message%pending=.FALSE.
1145    ENDIF
1146!$OMP END MASTER
1147!$OMP BARRIER
1148     
1149    IF (field(1)%data_type==type_real) THEN
1150      IF (field(1)%ndim==2) THEN
1151
1152        DO ind=1,ndomain
1153          IF (.NOT. assigned_domain(ind) .OR. .NOT. is_omp_level_master) CYCLE
1154         
1155          rval2d=>field(ind)%rval2d
1156       
1157          req=>message%request(ind)
1158          DO isend=1,req%nsend
1159            send=>req%send(isend)
1160            value=>send%value
1161
1162           
1163            IF (send%rank/=mpi_rank) THEN
1164              ireq=send%ireq
1165
1166              buffer_r=>message%buffers(ireq)%r
1167              offset=send%offset
1168                                           
1169              DO n=1,send%size
1170                buffer_r(offset+n)=rval2d(value(n))
1171              ENDDO
1172
1173              IF (mpi_threading_mode==MPI_THREAD_SERIALIZED) THEN
1174                !$OMP CRITICAL           
1175                CALL MPI_ISEND(buffer_r,send%size,MPI_REAL8,send%rank,               &
1176                  send%tag+1000*message%number,comm_icosa, message%mpi_req(ireq),ierr)
1177                !$OMP END CRITICAL
1178              ELSE IF (mpi_threading_mode==MPI_THREAD_MULTIPLE) THEN
1179                CALL MPI_ISEND(buffer_r,send%size,MPI_REAL8,send%rank,               &
1180                  send%tag+1000*message%number,comm_icosa, message%mpi_req(ireq),ierr)
1181              ENDIF
1182             
1183            ENDIF
1184          ENDDO
1185        ENDDO
1186       
1187        DO ind=1,ndomain
1188          IF (.NOT. assigned_domain(ind) .OR. .NOT. is_omp_level_master) CYCLE
1189          rval2d=>field(ind)%rval2d
1190          req=>message%request(ind)       
1191
1192          DO irecv=1,req%nrecv
1193            recv=>req%recv(irecv)
1194
1195            IF (recv%rank==mpi_rank) THEN
1196
1197              value=>recv%value
1198              src_value => recv%src_value
1199              src_rval2d=>field(recv%domain)%rval2d
1200              sgn=>recv%sign
1201
1202              DO n=1,recv%size
1203                rval2d(value(n))=src_rval2d(src_value(n))*sgn(n)
1204              ENDDO
1205               
1206                   
1207            ELSE
1208           
1209              ireq=recv%ireq
1210              buffer_r=>message%buffers(ireq)%r
1211              IF (mpi_threading_mode==MPI_THREAD_SERIALIZED) THEN
1212               !$OMP CRITICAL           
1213                CALL MPI_IRECV(buffer_r,recv%size,MPI_REAL8,recv%rank,               &
1214                  recv%tag+1000*message%number,comm_icosa, message%mpi_req(ireq),ierr)
1215               !$OMP END CRITICAL
1216              ELSE IF (mpi_threading_mode==MPI_THREAD_MULTIPLE) THEN
1217                 CALL MPI_IRECV(buffer_r,recv%size,MPI_REAL8,recv%rank,              &
1218                   recv%tag+1000*message%number,comm_icosa, message%mpi_req(ireq),ierr)
1219              ENDIF
1220           
1221            ENDIF
1222          ENDDO
1223       
1224        ENDDO
1225       
1226     
1227      ELSE  IF (field(1)%ndim==3) THEN
1228     
1229        DO ind=1,ndomain
1230          IF (.NOT. assigned_domain(ind) ) CYCLE
1231
1232          dim3=size(field(ind)%rval3d,2)
1233          CALL distrib_level(dim3,lbegin,lend)
1234
1235          rval3d=>field(ind)%rval3d
1236          req=>message%request(ind)
1237 
1238          DO isend=1,req%nsend
1239            send=>req%send(isend)
1240            value=>send%value
1241
1242            IF (send%rank/=mpi_rank) THEN
1243              ireq=send%ireq
1244              buffer_r=>message%buffers(ireq)%r
1245
1246              offset=send%offset*dim3 + (lbegin-1)*send%size
1247         
1248              CALL trace_start("copy_to_buffer")
1249
1250              DO d3=lbegin,lend
1251                DO n=1,send%size
1252                  buffer_r(n+offset)=rval3d(value(n),d3)
1253                ENDDO
1254                offset=offset+send%size
1255              ENDDO
1256              CALL trace_end("copy_to_buffer")
1257
1258              IF (is_omp_level_master) THEN
1259                IF (mpi_threading_mode==MPI_THREAD_SERIALIZED) THEN
1260                  !$OMP CRITICAL   
1261                  CALL MPI_ISEND(buffer_r,send%size*dim3,MPI_REAL8,send%rank,        &
1262                    send%tag+1000*message%number,comm_icosa, message%mpi_req(ireq),ierr)
1263                  !$OMP END CRITICAL
1264                ELSE IF (mpi_threading_mode==MPI_THREAD_MULTIPLE) THEN
1265                  CALL MPI_ISEND(buffer_r,send%size*dim3,MPI_REAL8,send%rank,        &
1266                    send%tag+1000*message%number,comm_icosa, message%mpi_req(ireq),ierr)
1267                ENDIF
1268              ENDIF
1269            ENDIF
1270          ENDDO
1271        ENDDO
1272         
1273        DO ind=1,ndomain
1274          IF (.NOT. assigned_domain(ind) ) CYCLE
1275          dim3=size(field(ind)%rval3d,2)
1276          CALL distrib_level(dim3,lbegin,lend)
1277          rval3d=>field(ind)%rval3d
1278          req=>message%request(ind)
1279
1280          DO irecv=1,req%nrecv
1281            recv=>req%recv(irecv)
1282
1283            IF (recv%rank==mpi_rank) THEN
1284              value=>recv%value
1285              src_value => recv%src_value
1286              src_rval3d=>field(recv%domain)%rval3d
1287              sgn=>recv%sign
1288
1289              CALL trace_start("copy_data")
1290
1291              DO d3=lbegin,lend
1292                DO n=1,recv%size
1293                  rval3d(value(n),d3)=src_rval3d(src_value(n),d3)*sgn(n)
1294                ENDDO
1295              ENDDO
1296              CALL trace_end("copy_data")
1297
1298            ELSE
1299              ireq=recv%ireq
1300              buffer_r=>message%buffers(ireq)%r
1301 
1302              IF (is_omp_level_master) THEN
1303                IF (mpi_threading_mode==MPI_THREAD_SERIALIZED) THEN
1304                  !$OMP CRITICAL
1305                  CALL MPI_IRECV(buffer_r,recv%size*dim3,MPI_REAL8,recv%rank,        &
1306                    recv%tag+1000*message%number,comm_icosa, message%mpi_req(ireq),ierr)
1307                  !$OMP END CRITICAL
1308                ELSE IF (mpi_threading_mode==MPI_THREAD_MULTIPLE) THEN
1309                  CALL MPI_IRECV(buffer_r,recv%size*dim3,MPI_REAL8,recv%rank,        &
1310                    recv%tag+1000*message%number,comm_icosa, message%mpi_req(ireq),ierr)
1311                ENDIF
1312              ENDIF
1313            ENDIF 
1314          ENDDO
1315       
1316        ENDDO
1317
1318      ELSE  IF (field(1)%ndim==4) THEN
1319   
1320        DO ind=1,ndomain
1321          IF (.NOT. assigned_domain(ind) ) CYCLE
1322
1323          dim3=size(field(ind)%rval4d,2)
1324          CALL distrib_level(dim3,lbegin,lend)
1325          dim4=size(field(ind)%rval4d,3)
1326          rval4d=>field(ind)%rval4d
1327          req=>message%request(ind)
1328
1329          DO isend=1,req%nsend
1330            send=>req%send(isend)
1331            value=>send%value
1332
1333            IF (send%rank/=mpi_rank) THEN
1334
1335              ireq=send%ireq
1336              buffer_r=>message%buffers(ireq)%r
1337
1338              CALL trace_start("copy_to_buffer")
1339              DO d4=1,dim4
1340                offset=send%offset*dim3*dim4 + dim3*send%size*(d4-1) +               &
1341                  (lbegin-1)*send%size
1342
1343                DO d3=lbegin,lend
1344                  DO n=1,send%size
1345                    buffer_r(n+offset)=rval4d(value(n),d3,d4)
1346                  ENDDO
1347                  offset=offset+send%size
1348                ENDDO
1349              ENDDO
1350              CALL trace_end("copy_to_buffer")
1351
1352              IF (is_omp_level_master) THEN
1353                IF (mpi_threading_mode==MPI_THREAD_SERIALIZED) THEN
1354                  !$OMP CRITICAL
1355                  CALL MPI_ISEND(buffer_r,send%size*dim3*dim4,MPI_REAL8,send%rank,   &
1356                    send%tag+1000*message%number,comm_icosa, message%mpi_req(ireq),ierr)
1357                  !$OMP END CRITICAL
1358                ELSE IF (mpi_threading_mode==MPI_THREAD_MULTIPLE) THEN
1359                  CALL MPI_ISEND(buffer_r,send%size*dim3*dim4,MPI_REAL8,send%rank,   &
1360                    send%tag+1000*message%number,comm_icosa, message%mpi_req(ireq),ierr)
1361                ENDIF
1362              ENDIF
1363
1364            ENDIF
1365          ENDDO
1366        ENDDO
1367       
1368        DO ind=1,ndomain
1369          IF (.NOT. assigned_domain(ind) ) CYCLE
1370         
1371          dim3=size(field(ind)%rval4d,2)
1372          CALL distrib_level(dim3,lbegin,lend)
1373          dim4=size(field(ind)%rval4d,3)
1374          rval4d=>field(ind)%rval4d
1375          req=>message%request(ind)
1376
1377          DO irecv=1,req%nrecv
1378            recv=>req%recv(irecv)
1379            IF (recv%rank==mpi_rank) THEN
1380              value=>recv%value
1381              src_value => recv%src_value
1382              src_rval4d=>field(recv%domain)%rval4d
1383              sgn=>recv%sign
1384
1385              CALL trace_start("copy_data")
1386              DO d4=1,dim4
1387                DO d3=lbegin,lend
1388                  DO n=1,recv%size
1389                    rval4d(value(n),d3,d4)=src_rval4d(src_value(n),d3,d4)*sgn(n)
1390                  ENDDO
1391                ENDDO
1392              ENDDO
1393               
1394              CALL trace_end("copy_data")
1395                   
1396            ELSE
1397
1398              ireq=recv%ireq
1399              buffer_r=>message%buffers(ireq)%r
1400              IF (is_omp_level_master) THEN
1401                IF (mpi_threading_mode==MPI_THREAD_SERIALIZED) THEN
1402                 !$OMP CRITICAL           
1403                  CALL MPI_IRECV(buffer_r,recv%size*dim3*dim4,MPI_REAL8,recv%rank,   &
1404                    recv%tag+1000*message%number,comm_icosa, message%mpi_req(ireq),ierr) 
1405                  !$OMP END CRITICAL
1406                ELSE IF (mpi_threading_mode==MPI_THREAD_MULTIPLE) THEN
1407                  CALL MPI_IRECV(buffer_r,recv%size*dim3*dim4,MPI_REAL8,recv%rank,   &
1408                    recv%tag+1000*message%number,comm_icosa, message%mpi_req(ireq),ierr) 
1409                ENDIF
1410              ENDIF
1411            ENDIF
1412          ENDDO
1413        ENDDO
1414
1415      ENDIF     
1416
1417      IF (mpi_threading_mode==MPI_THREAD_FUNNELED .OR. mpi_threading_mode==MPI_THREAD_SINGLE) THEN
1418!$OMP BARRIER
1419!$OMP MASTER       
1420
1421        DO ireq=1,message%nreq_send
1422          buffer_r=>message%buffers(ireq)%r
1423          msize=message%buffers(ireq)%size
1424          rank=message%buffers(ireq)%rank
1425          CALL MPI_ISEND(buffer_r,msize,MPI_REAL8,rank,1000*message%number,          &
1426            comm_icosa, message%mpi_req(ireq),ierr)
1427        ENDDO
1428
1429        DO ireq=message%nreq_send+1,message%nreq_send+message%nreq_recv
1430          buffer_r=>message%buffers(ireq)%r
1431          msize=message%buffers(ireq)%size
1432          rank=message%buffers(ireq)%rank
1433          CALL MPI_IRECV(buffer_r,msize,MPI_REAL8,rank,1000*message%number,          &
1434            comm_icosa, message%mpi_req(ireq),ierr)
1435        ENDDO
1436
1437!$OMP END MASTER
1438      ENDIF             
1439    ENDIF
1440   
1441!$OMP BARRIER
1442!    CALL trace_end("send_message_mpi")
1443   
1444  END SUBROUTINE send_message_mpi
1445 
1446  SUBROUTINE test_message_mpi(message)
1447  IMPLICIT NONE
1448    TYPE(t_message) :: message
1449   
1450    INTEGER :: ierr
1451
1452!$OMP MASTER
1453    IF (message%pending .AND. .NOT. message%completed) CALL MPI_TESTALL(message%nreq,&
1454      message%mpi_req,message%completed,message%status,ierr)
1455!$OMP END MASTER
1456  END SUBROUTINE  test_message_mpi
1457 
1458   
1459  SUBROUTINE wait_message_mpi(message)
1460  USE field_mod
1461  USE domain_mod
1462  USE mpi_mod
1463  USE mpipara
1464  USE omp_para
1465  USE trace
1466  IMPLICIT NONE
1467    TYPE(t_message) :: message
1468
1469    TYPE(t_field),POINTER :: field(:)
1470    REAL(rstd),POINTER :: rval2d(:) 
1471    REAL(rstd),POINTER :: rval3d(:,:) 
1472    REAL(rstd),POINTER :: rval4d(:,:,:) 
1473    REAL(rstd),POINTER :: buffer_r(:) 
1474    INTEGER,POINTER :: value(:) 
1475    INTEGER,POINTER :: sgn(:) 
1476    TYPE(ARRAY),POINTER :: recv,send 
1477    TYPE(t_request),POINTER :: req
1478    INTEGER, ALLOCATABLE :: mpi_req(:)
1479    INTEGER, ALLOCATABLE :: status(:,:)
1480    INTEGER :: irecv,isend
1481    INTEGER :: ireq,nreq
1482    INTEGER :: ind,n,l,m,i
1483    INTEGER :: dim3,dim4,d3,d4,lbegin,lend
1484    INTEGER :: offset
1485
1486    message%open=.FALSE.
1487    IF (.NOT. message%pending) RETURN
1488
1489!    CALL trace_start("wait_message_mpi")
1490
1491    field=>message%field
1492    nreq=message%nreq
1493   
1494    IF (field(1)%data_type==type_real) THEN
1495      IF (field(1)%ndim==2) THEN
1496
1497!$OMP MASTER
1498        IF (.NOT. message%completed) CALL MPI_WAITALL(nreq,message%mpi_req,          &
1499          message%status,ierr)
1500!$OMP END MASTER
1501!$OMP BARRIER
1502       
1503        DO ind=1,ndomain
1504          IF (.NOT. assigned_domain(ind) .OR. .NOT. is_omp_level_master) CYCLE
1505         
1506          rval2d=>field(ind)%rval2d
1507          req=>message%request(ind)
1508          DO irecv=1,req%nrecv
1509            recv=>req%recv(irecv)
1510            IF (recv%rank/=mpi_rank) THEN
1511              ireq=recv%ireq
1512              buffer_r=>message%buffers(ireq)%r
1513              value=>recv%value
1514              sgn=>recv%sign
1515              offset=recv%offset
1516              DO n=1,recv%size
1517                rval2d(value(n))=buffer_r(n+offset)*sgn(n) 
1518              ENDDO
1519
1520            ENDIF
1521          ENDDO
1522       
1523        ENDDO
1524     
1525     
1526      ELSE  IF (field(1)%ndim==3) THEN
1527
1528!$OMP MASTER
1529        IF (.NOT. message%completed) CALL MPI_WAITALL(nreq,message%mpi_req,          &
1530          message%status,ierr)
1531!$OMP END MASTER
1532!$OMP BARRIER
1533
1534       
1535        DO ind=1,ndomain
1536          IF (.NOT. assigned_domain(ind) ) CYCLE
1537
1538          rval3d=>field(ind)%rval3d
1539          req=>message%request(ind)
1540          DO irecv=1,req%nrecv
1541            recv=>req%recv(irecv)
1542            IF (recv%rank/=mpi_rank) THEN
1543              ireq=recv%ireq
1544              buffer_r=>message%buffers(ireq)%r
1545              value=>recv%value
1546              sgn=>recv%sign
1547             
1548              dim3=size(rval3d,2)
1549   
1550              CALL distrib_level(dim3,lbegin,lend)
1551              offset=recv%offset*dim3 + (lbegin-1)*recv%size
1552              CALL trace_start("copy_from_buffer")
1553             
1554              IF (req%vector) THEN
1555                DO d3=lbegin,lend
1556                  DO n=1,recv%size
1557                    rval3d(value(n),d3)=buffer_r(n+offset)*sgn(n) 
1558                  ENDDO
1559                  offset=offset+recv%size
1560                ENDDO
1561              ELSE
1562                DO d3=lbegin,lend
1563                  DO n=1,recv%size
1564                    rval3d(value(n),d3)=buffer_r(n+offset) 
1565                  ENDDO
1566                  offset=offset+recv%size
1567                ENDDO
1568              ENDIF
1569               
1570              CALL trace_end("copy_from_buffer")
1571            ENDIF
1572          ENDDO
1573       
1574        ENDDO
1575
1576      ELSE  IF (field(1)%ndim==4) THEN
1577!$OMP MASTER
1578        IF (.NOT. message%completed) CALL MPI_WAITALL(nreq,message%mpi_req,          &
1579          message%status,ierr)
1580!$OMP END MASTER
1581!$OMP BARRIER
1582
1583               
1584        DO ind=1,ndomain
1585          IF (.NOT. assigned_domain(ind) ) CYCLE
1586
1587          rval4d=>field(ind)%rval4d
1588          req=>message%request(ind)
1589          DO irecv=1,req%nrecv
1590            recv=>req%recv(irecv)
1591            IF (recv%rank/=mpi_rank) THEN
1592              ireq=recv%ireq
1593              buffer_r=>message%buffers(ireq)%r
1594              value=>recv%value
1595              sgn=>recv%sign
1596
1597              dim3=size(rval4d,2)
1598              CALL distrib_level(dim3,lbegin,lend)
1599              dim4=size(rval4d,3)
1600              CALL trace_start("copy_from_buffer")
1601              DO d4=1,dim4
1602                offset=recv%offset*dim3*dim4 + dim3*recv%size*(d4-1) +               &
1603                  (lbegin-1)*recv%size
1604                DO d3=lbegin,lend
1605                  DO n=1,recv%size
1606                    rval4d(value(n),d3,d4)=buffer_r(n+offset)*sgn(n) 
1607                  ENDDO
1608                  offset=offset+recv%size
1609                ENDDO
1610              ENDDO
1611              CALL trace_end("copy_from_buffer")
1612            ENDIF
1613          ENDDO
1614       
1615        ENDDO
1616     
1617      ENDIF     
1618     
1619    ENDIF
1620
1621!$OMP MASTER
1622    message%pending=.FALSE.
1623!$OMP END MASTER
1624
1625!    CALL trace_end("wait_message_mpi")
1626!$OMP BARRIER
1627   
1628  END SUBROUTINE wait_message_mpi
1629
1630  SUBROUTINE transfert_request_mpi(field,request)
1631  USE field_mod
1632  IMPLICIT NONE
1633    TYPE(t_field),POINTER :: field(:)
1634    TYPE(t_request),POINTER :: request(:)
1635
1636    TYPE(t_message),SAVE :: message
1637   
1638   
1639    CALL init_message_mpi(field,request, message)
1640    CALL transfert_message_mpi(field,message)
1641    CALL finalize_message_mpi(field,message)
1642   
1643  END SUBROUTINE transfert_request_mpi
1644 
1645   
1646   
1647  SUBROUTINE transfert_request_seq(field,request)
1648  USE field_mod
1649  USE domain_mod
1650  IMPLICIT NONE
1651    TYPE(t_field),POINTER :: field(:)
1652    TYPE(t_request),POINTER :: request(:)
1653    REAL(rstd),POINTER :: rval2d(:) 
1654    REAL(rstd),POINTER :: rval3d(:,:) 
1655    REAL(rstd),POINTER :: rval4d(:,:,:) 
1656    INTEGER :: ind
1657    TYPE(t_request),POINTER :: req
1658    INTEGER :: n
1659    REAL(rstd) :: var1,var2
1660   
1661    DO ind=1,ndomain
1662      req=>request(ind)
1663      rval2d=>field(ind)%rval2d
1664      rval3d=>field(ind)%rval3d
1665      rval4d=>field(ind)%rval4d
1666     
1667      IF (field(ind)%data_type==type_real) THEN
1668        IF (field(ind)%ndim==2) THEN
1669          DO n=1,req%size
1670            rval2d(req%target_ind(n))=field(req%src_domain(n))%rval2d(req%src_ind(n))*&
1671              req%target_sign(n)
1672          ENDDO
1673        ELSE IF (field(ind)%ndim==3) THEN
1674          DO n=1,req%size
1675            rval3d(req%target_ind(n),:)=field(req%src_domain(n))%rval3d(req%src_ind(n),:)*&
1676              req%target_sign(n)
1677          ENDDO
1678        ELSE IF (field(ind)%ndim==4) THEN
1679          DO n=1,req%size
1680            rval4d(req%target_ind(n),:,:)=field(req%src_domain(n))%rval4d(req%src_ind(n),:,:)*&
1681              req%target_sign(n)
1682          ENDDO
1683        ENDIF
1684      ENDIF       
1685
1686    ENDDO
1687   
1688  END SUBROUTINE transfert_request_seq
1689 
1690 
1691  SUBROUTINE gather_field(field_loc,field_glo)
1692  USE field_mod
1693  USE domain_mod
1694  USE mpi_mod
1695  USE mpipara
1696  IMPLICIT NONE
1697    TYPE(t_field),POINTER :: field_loc(:)
1698    TYPE(t_field),POINTER :: field_glo(:)
1699    INTEGER, ALLOCATABLE :: mpi_req(:)
1700    INTEGER, ALLOCATABLE :: status(:,:)
1701    INTEGER :: ireq,nreq
1702    INTEGER :: ind_glo,ind_loc   
1703 
1704    IF (.NOT. using_mpi) THEN
1705   
1706      DO ind_loc=1,ndomain
1707        IF (field_loc(ind_loc)%ndim==2) field_glo(ind_loc)%rval2d=field_loc(ind_loc)%rval2d
1708        IF (field_loc(ind_loc)%ndim==3) field_glo(ind_loc)%rval3d=field_loc(ind_loc)%rval3d
1709        IF (field_loc(ind_loc)%ndim==4) field_glo(ind_loc)%rval4d=field_loc(ind_loc)%rval4d
1710      ENDDO
1711   
1712    ELSE
1713         
1714      nreq=ndomain
1715      IF (mpi_rank==0) nreq=nreq+ndomain_glo 
1716      ALLOCATE(mpi_req(nreq))
1717      ALLOCATE(status(MPI_STATUS_SIZE,nreq))
1718   
1719   
1720      ireq=0
1721      IF (mpi_rank==0) THEN
1722        DO ind_glo=1,ndomain_glo
1723          ireq=ireq+1
1724
1725          IF (field_glo(ind_glo)%ndim==2) THEN
1726            CALL MPI_IRECV(field_glo(ind_glo)%rval2d,size(field_glo(ind_glo)%rval2d) , MPI_REAL8 ,   &
1727                         domglo_rank(ind_glo),domglo_loc_ind(ind_glo), comm_icosa, mpi_req(ireq), ierr)
1728   
1729          ELSE IF (field_glo(ind_glo)%ndim==3) THEN
1730            CALL MPI_IRECV(field_glo(ind_glo)%rval3d,size(field_glo(ind_glo)%rval3d) , MPI_REAL8 ,   &
1731                         domglo_rank(ind_glo),domglo_loc_ind(ind_glo), comm_icosa, mpi_req(ireq), ierr)
1732
1733          ELSE IF (field_glo(ind_glo)%ndim==4) THEN
1734            CALL MPI_IRECV(field_glo(ind_glo)%rval4d,size(field_glo(ind_glo)%rval4d) , MPI_REAL8 ,   &
1735                         domglo_rank(ind_glo),domglo_loc_ind(ind_glo), comm_icosa, mpi_req(ireq), ierr)
1736          ENDIF
1737         
1738        ENDDO
1739      ENDIF
1740 
1741      DO ind_loc=1,ndomain
1742        ireq=ireq+1
1743
1744        IF (field_loc(ind_loc)%ndim==2) THEN
1745          CALL MPI_ISEND(field_loc(ind_loc)%rval2d,size(field_loc(ind_loc)%rval2d) , MPI_REAL8 ,   &
1746                         0, ind_loc, comm_icosa, mpi_req(ireq), ierr)
1747        ELSE IF (field_loc(ind_loc)%ndim==3) THEN
1748          CALL MPI_ISEND(field_loc(ind_loc)%rval3d,size(field_loc(ind_loc)%rval3d) , MPI_REAL8 ,   &
1749                         0, ind_loc, comm_icosa, mpi_req(ireq), ierr)
1750        ELSE IF (field_loc(ind_loc)%ndim==4) THEN
1751          CALL MPI_ISEND(field_loc(ind_loc)%rval4d,size(field_loc(ind_loc)%rval4d) , MPI_REAL8 ,   &
1752                         0, ind_loc, comm_icosa, mpi_req(ireq), ierr)
1753        ENDIF
1754     
1755      ENDDO
1756   
1757      CALL MPI_WAITALL(nreq,mpi_req,status,ierr)
1758
1759    ENDIF
1760       
1761  END SUBROUTINE gather_field
1762
1763  SUBROUTINE bcast_field(field_glo)
1764  USE field_mod
1765  USE domain_mod
1766  USE mpi_mod
1767  USE mpipara
1768  IMPLICIT NONE
1769    TYPE(t_field),POINTER :: field_glo(:)
1770    INTEGER :: ind_glo   
1771 
1772    IF (.NOT. using_mpi) THEN
1773   
1774! nothing to do
1775   
1776    ELSE
1777         
1778      DO ind_glo=1,ndomain_glo
1779
1780          IF (field_glo(ind_glo)%ndim==2) THEN
1781            CALL MPI_BCAST(field_glo(ind_glo)%rval2d, size(field_glo(ind_glo)%rval2d) , MPI_REAL8, 0, comm_icosa, ierr)
1782          ELSE IF (field_glo(ind_glo)%ndim==3) THEN
1783            CALL MPI_BCAST(field_glo(ind_glo)%rval3d, size(field_glo(ind_glo)%rval3d) , MPI_REAL8, 0, comm_icosa, ierr)
1784          ELSE IF (field_glo(ind_glo)%ndim==4) THEN
1785            CALL MPI_BCAST(field_glo(ind_glo)%rval4d, size(field_glo(ind_glo)%rval4d) , MPI_REAL8, 0, comm_icosa, ierr)
1786          ENDIF
1787         
1788        ENDDO
1789      ENDIF
1790       
1791  END SUBROUTINE bcast_field
1792
1793  SUBROUTINE scatter_field(field_glo,field_loc)
1794  USE field_mod
1795  USE domain_mod
1796  USE mpi_mod
1797  USE mpipara
1798  IMPLICIT NONE
1799    TYPE(t_field),POINTER :: field_glo(:)
1800    TYPE(t_field),POINTER :: field_loc(:)
1801    INTEGER, ALLOCATABLE :: mpi_req(:)
1802    INTEGER, ALLOCATABLE :: status(:,:)
1803    INTEGER :: ireq,nreq
1804    INTEGER :: ind_glo,ind_loc   
1805 
1806    IF (.NOT. using_mpi) THEN
1807   
1808      DO ind_loc=1,ndomain
1809        IF (field_loc(ind_loc)%ndim==2) field_loc(ind_loc)%rval2d=field_glo(ind_loc)%rval2d
1810        IF (field_loc(ind_loc)%ndim==3) field_loc(ind_loc)%rval3d=field_glo(ind_loc)%rval3d
1811        IF (field_loc(ind_loc)%ndim==4) field_loc(ind_loc)%rval4d=field_glo(ind_loc)%rval4d
1812      ENDDO
1813   
1814    ELSE
1815         
1816      nreq=ndomain
1817      IF (mpi_rank==0) nreq=nreq+ndomain_glo 
1818      ALLOCATE(mpi_req(nreq))
1819      ALLOCATE(status(MPI_STATUS_SIZE,nreq))
1820   
1821   
1822      ireq=0
1823      IF (mpi_rank==0) THEN
1824        DO ind_glo=1,ndomain_glo
1825          ireq=ireq+1
1826
1827          IF (field_glo(ind_glo)%ndim==2) THEN
1828            CALL MPI_ISEND(field_glo(ind_glo)%rval2d,size(field_glo(ind_glo)%rval2d) , MPI_REAL8 ,   &
1829                         domglo_rank(ind_glo),domglo_loc_ind(ind_glo), comm_icosa, mpi_req(ireq), ierr)
1830   
1831          ELSE IF (field_glo(ind_glo)%ndim==3) THEN
1832            CALL MPI_ISEND(field_glo(ind_glo)%rval3d,size(field_glo(ind_glo)%rval3d) , MPI_REAL8 ,   &
1833                         domglo_rank(ind_glo),domglo_loc_ind(ind_glo), comm_icosa, mpi_req(ireq), ierr)
1834
1835          ELSE IF (field_glo(ind_glo)%ndim==4) THEN
1836            CALL MPI_ISEND(field_glo(ind_glo)%rval4d,size(field_glo(ind_glo)%rval4d) , MPI_REAL8 ,   &
1837                         domglo_rank(ind_glo),domglo_loc_ind(ind_glo), comm_icosa, mpi_req(ireq), ierr)
1838          ENDIF
1839         
1840        ENDDO
1841      ENDIF
1842 
1843      DO ind_loc=1,ndomain
1844        ireq=ireq+1
1845
1846        IF (field_loc(ind_loc)%ndim==2) THEN
1847          CALL MPI_IRECV(field_loc(ind_loc)%rval2d,size(field_loc(ind_loc)%rval2d) , MPI_REAL8 ,   &
1848                         0, ind_loc, comm_icosa, mpi_req(ireq), ierr)
1849        ELSE IF (field_loc(ind_loc)%ndim==3) THEN
1850          CALL MPI_IRECV(field_loc(ind_loc)%rval3d,size(field_loc(ind_loc)%rval3d) , MPI_REAL8 ,   &
1851                         0, ind_loc, comm_icosa, mpi_req(ireq), ierr)
1852        ELSE IF (field_loc(ind_loc)%ndim==4) THEN
1853          CALL MPI_IRECV(field_loc(ind_loc)%rval4d,size(field_loc(ind_loc)%rval4d) , MPI_REAL8 ,   &
1854                         0, ind_loc, comm_icosa, mpi_req(ireq), ierr)
1855        ENDIF
1856     
1857      ENDDO
1858   
1859      CALL MPI_WAITALL(nreq,mpi_req,status,ierr)
1860
1861    ENDIF
1862       
1863  END SUBROUTINE scatter_field
1864 
1865  SUBROUTINE trace_in
1866  USE trace
1867  IMPLICIT NONE
1868 
1869    CALL trace_start("transfert_buffer")
1870  END SUBROUTINE trace_in             
1871
1872  SUBROUTINE trace_out
1873  USE trace
1874  IMPLICIT NONE
1875 
1876    CALL trace_end("transfert_buffer")
1877  END SUBROUTINE trace_out             
1878
1879
1880
1881
1882!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
1883!! Definition des Broadcast --> 4D   !!
1884!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
1885
1886!! -- Les chaine de charactï¿œre -- !!
1887
1888  SUBROUTINE bcast_mpi_c(var1)
1889  IMPLICIT NONE
1890    CHARACTER(LEN=*),INTENT(INOUT) :: Var1
1891   
1892    CALL bcast_mpi_cgen(Var1,len(Var1))
1893
1894  END SUBROUTINE bcast_mpi_c
1895
1896!! -- Les entiers -- !!
1897 
1898  SUBROUTINE bcast_mpi_i(var)
1899  USE mpipara
1900  IMPLICIT NONE
1901    INTEGER,INTENT(INOUT) :: Var
1902   
1903    INTEGER               :: var_tmp(1)
1904   
1905    IF (is_mpi_master) var_tmp(1)=var
1906    CALL bcast_mpi_igen(Var_tmp,1)
1907    var=var_tmp(1)
1908   
1909  END SUBROUTINE bcast_mpi_i
1910
1911  SUBROUTINE bcast_mpi_i1(var)
1912  IMPLICIT NONE
1913    INTEGER,INTENT(INOUT) :: Var(:)
1914
1915    CALL bcast_mpi_igen(Var,size(Var))
1916   
1917  END SUBROUTINE bcast_mpi_i1
1918
1919  SUBROUTINE bcast_mpi_i2(var)
1920  IMPLICIT NONE
1921    INTEGER,INTENT(INOUT) :: Var(:,:)
1922   
1923    CALL bcast_mpi_igen(Var,size(Var))
1924 
1925  END SUBROUTINE bcast_mpi_i2
1926
1927  SUBROUTINE bcast_mpi_i3(var)
1928  IMPLICIT NONE
1929    INTEGER,INTENT(INOUT) :: Var(:,:,:)
1930   
1931    CALL bcast_mpi_igen(Var,size(Var))
1932
1933  END SUBROUTINE bcast_mpi_i3
1934
1935  SUBROUTINE bcast_mpi_i4(var)
1936  IMPLICIT NONE
1937    INTEGER,INTENT(INOUT) :: Var(:,:,:,:)
1938   
1939    CALL bcast_mpi_igen(Var,size(Var))
1940
1941  END SUBROUTINE bcast_mpi_i4
1942
1943
1944!! -- Les reels -- !!
1945
1946  SUBROUTINE bcast_mpi_r(var)
1947  USE mpipara
1948  IMPLICIT NONE
1949    REAL,INTENT(INOUT) :: Var
1950    REAL               :: var_tmp(1)
1951   
1952    IF (is_mpi_master) var_tmp(1)=var
1953    CALL bcast_mpi_rgen(Var_tmp,1)
1954    var=var_tmp(1)   
1955
1956  END SUBROUTINE bcast_mpi_r
1957
1958  SUBROUTINE bcast_mpi_r1(var)
1959  IMPLICIT NONE
1960    REAL,INTENT(INOUT) :: Var(:)
1961   
1962    CALL bcast_mpi_rgen(Var,size(Var))
1963
1964  END SUBROUTINE bcast_mpi_r1
1965
1966  SUBROUTINE bcast_mpi_r2(var)
1967  IMPLICIT NONE
1968    REAL,INTENT(INOUT) :: Var(:,:)
1969   
1970    CALL bcast_mpi_rgen(Var,size(Var))
1971
1972  END SUBROUTINE bcast_mpi_r2
1973
1974  SUBROUTINE bcast_mpi_r3(var)
1975  IMPLICIT NONE
1976    REAL,INTENT(INOUT) :: Var(:,:,:)
1977   
1978    CALL bcast_mpi_rgen(Var,size(Var))
1979
1980  END SUBROUTINE bcast_mpi_r3
1981
1982  SUBROUTINE bcast_mpi_r4(var)
1983  IMPLICIT NONE
1984    REAL,INTENT(INOUT) :: Var(:,:,:,:)
1985   
1986    CALL bcast_mpi_rgen(Var,size(Var))
1987
1988  END SUBROUTINE bcast_mpi_r4
1989 
1990!! -- Les booleans -- !!
1991
1992  SUBROUTINE bcast_mpi_l(var)
1993  USE mpipara
1994  IMPLICIT NONE
1995    LOGICAL,INTENT(INOUT) :: Var
1996    LOGICAL               :: var_tmp(1)
1997   
1998    IF (is_mpi_master) var_tmp(1)=var
1999    CALL bcast_mpi_lgen(Var_tmp,1)
2000    var=var_tmp(1)   
2001
2002  END SUBROUTINE bcast_mpi_l
2003
2004  SUBROUTINE bcast_mpi_l1(var)
2005  IMPLICIT NONE
2006    LOGICAL,INTENT(INOUT) :: Var(:)
2007   
2008    CALL bcast_mpi_lgen(Var,size(Var))
2009
2010  END SUBROUTINE bcast_mpi_l1
2011
2012  SUBROUTINE bcast_mpi_l2(var)
2013  IMPLICIT NONE
2014    LOGICAL,INTENT(INOUT) :: Var(:,:)
2015   
2016    CALL bcast_mpi_lgen(Var,size(Var))
2017
2018  END SUBROUTINE bcast_mpi_l2
2019
2020  SUBROUTINE bcast_mpi_l3(var)
2021  IMPLICIT NONE
2022    LOGICAL,INTENT(INOUT) :: Var(:,:,:)
2023   
2024    CALL bcast_mpi_lgen(Var,size(Var))
2025
2026  END SUBROUTINE bcast_mpi_l3
2027
2028  SUBROUTINE bcast_mpi_l4(var)
2029  IMPLICIT NONE
2030    LOGICAL,INTENT(INOUT) :: Var(:,:,:,:)
2031   
2032    CALL bcast_mpi_lgen(Var,size(Var))
2033
2034  END SUBROUTINE bcast_mpi_l4
2035 
2036!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
2037!! DEFINITION DES FONCTIONS DE TRANSFERT GENERIQUES !
2038!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
2039
2040  SUBROUTINE bcast_mpi_cgen(var,nb)
2041    USE mpi_mod
2042    USE mpipara
2043    IMPLICIT NONE
2044   
2045    CHARACTER(LEN=*),INTENT(INOUT) :: Var
2046    INTEGER,INTENT(IN) :: nb
2047
2048    IF (.NOT. using_mpi) RETURN
2049   
2050    CALL MPI_BCAST(Var,nb,MPI_CHARACTER,mpi_master,comm_icosa,ierr)
2051       
2052  END SUBROUTINE bcast_mpi_cgen
2053
2054
2055     
2056  SUBROUTINE bcast_mpi_igen(var,nb)
2057    USE mpi_mod
2058    USE mpipara
2059    IMPLICIT NONE
2060   
2061    INTEGER,DIMENSION(nb),INTENT(INOUT) :: Var
2062    INTEGER,INTENT(IN) :: nb
2063
2064    IF (.NOT. using_mpi) RETURN
2065
2066    CALL MPI_BCAST(Var,nb,MPI_INTEGER,mpi_master,comm_icosa,ierr)
2067       
2068  END SUBROUTINE bcast_mpi_igen
2069
2070
2071
2072 
2073  SUBROUTINE bcast_mpi_rgen(var,nb)
2074    USE mpi_mod
2075    USE mpipara
2076    IMPLICIT NONE
2077   
2078    REAL,DIMENSION(nb),INTENT(INOUT) :: Var
2079    INTEGER,INTENT(IN) :: nb
2080
2081    IF (.NOT. using_mpi) RETURN
2082
2083    CALL MPI_BCAST(Var,nb,MPI_REAL8,mpi_master,comm_icosa,ierr)
2084   
2085  END SUBROUTINE bcast_mpi_rgen
2086 
2087
2088
2089
2090  SUBROUTINE bcast_mpi_lgen(var,nb)
2091    USE mpi_mod
2092    USE mpipara
2093    IMPLICIT NONE
2094   
2095    LOGICAL,DIMENSION(nb),INTENT(INOUT) :: Var
2096    INTEGER,INTENT(IN) :: nb
2097
2098    IF (.NOT. using_mpi) RETURN
2099
2100    CALL MPI_BCAST(Var,nb,MPI_LOGICAL,mpi_master,comm_icosa,ierr)
2101
2102  END SUBROUTINE bcast_mpi_lgen
2103 
2104   
2105END MODULE transfert_mpi_mod
2106     
2107       
2108       
2109       
2110     
Note: See TracBrowser for help on using the repository browser.