spmd_exch_wave.F File Reference

#include "implicit_f.inc"
#include "spmd.inc"
#include "com01_c.inc"
#include "task_c.inc"

Go to the source code of this file.

Functions/Subroutines
subroutine	spmd_exch_wave (fr_wave, iad_elem, fr_elem, size, lenr)

Function/Subroutine Documentation

spmd_exch_wave()

subroutine spmd_exch_wave	(		fr_wave,
		integer, dimension(2,*)	iad_elem,
		integer, dimension(*)	fr_elem,
		integer	size,
		integer	lenr )

Definition at line 32 of file spmd_exch_wave.F.

C-----------------------------------------------
C   I m p l i c i t   T y p e s
C-----------------------------------------------
        USE spmd_comm_world_mod, ONLY : spmd_comm_world
#include      "implicit_f.inc"
C-----------------------------------------------------------------
C   M e s s a g e   P a s s i n g
C-----------------------------------------------
#include "spmd.inc"
C-----------------------------------------------
C   C o m m o n   B l o c k s
C-----------------------------------------------
#include      "com01_c.inc"
#include      "task_c.inc"
C-----------------------------------------------
C   D u m m y   A r g u m e n t s
C-----------------------------------------------
      INTEGER IAD_ELEM(2,*), FR_ELEM(*), SIZE, LENR
      my_real
     .        fr_wave(*)
C-----------------------------------------------
C   L o c a l  V a r i a b l e s
C-----------------------------------------------
#ifdef MPI
      INTEGER MSGTYP,I,NOD,LOC_PROC,IERROR, MSGOFF,
     .        SIZ,J,L,NB_NOD,
     .        STATUS(MPI_STATUS_SIZE),
     .        IAD_SEND(NSPMD+1),IAD_RECV(NSPMD+1),
     .        REQ_R(NSPMD),REQ_S(NSPMD)
      DATA msgoff/128/
      my_real
     .        rbuf(size*lenr), sbuf(size*lenr)
C-----------------------------------------------
C   S o u r c e  L i n e s
C-----------------------------------------------
      loc_proc = ispmd + 1
      l = 1
      iad_recv(1) = 1
      DO i=1,nspmd
        siz = size*(iad_elem(1,i+1)-iad_elem(1,i))
        IF(siz/=0)THEN
          msgtyp = msgoff
          CALL mpi_irecv(
     s      rbuf(l),siz,real,it_spmd(i),msgtyp,
     g      spmd_comm_world,req_r(i),ierror)
          l = l + siz
        ENDIF
        iad_recv(i+1) = l
      END DO
      l = 1
      iad_send(1) = 1
      DO i=1,nspmd
#include      "vectorize.inc"
        DO j=iad_elem(1,i),iad_elem(1,i+1)-1
          nod = fr_elem(j)
          IF(fr_wave(nod)<0)THEN
            sbuf(l  ) =  fr_wave(nod)
          ELSE
            sbuf(l) = zero
          END IF
          l = l + SIZE
        END DO
        iad_send(i+1) = l
      ENDDO
C
C   echange messages
C
      DO i=1,nspmd
C--------------------------------------------------------------------
       IF(iad_elem(1,i+1)-iad_elem(1,i)>0)THEN
          msgtyp = msgoff
          siz = iad_send(i+1)-iad_send(i)
          l = iad_send(i)
          CALL mpi_isend(
     s      sbuf(l),siz,real,it_spmd(i),msgtyp,
     g      spmd_comm_world,req_s(i),ierror)
       ENDIF
C--------------------------------------------------------------------
      ENDDO
C
      DO i = 1, nspmd
        nb_nod = iad_elem(1,i+1)-iad_elem(1,i)
        IF(nb_nod>0)THEN
          CALL mpi_wait(req_r(i),status,ierror)
          l = iad_recv(i)
#include      "vectorize.inc"
          DO j=iad_elem(1,i),iad_elem(1,i+1)-1
            nod = fr_elem(j)
            IF(rbuf(l)<zero) THEN
              fr_wave(nod) = rbuf(l)
            END IF
            l = l + SIZE
          END DO
        ENDIF
      END DO
C
      DO i = 1, nspmd
        IF(iad_elem(1,i+1)-iad_elem(1,i)>0)THEN
          CALL mpi_wait(req_s(i),status,ierror)
        ENDIF
      ENDDO
C
#endif
      RETURN