spmd_exch_v.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_v (xnorm, nsiz, iad_elem, fr_elem, lenr)

Function/Subroutine Documentation

spmd_exch_v()

subroutine spmd_exch_v	(		xnorm,
		integer	nsiz,
		integer, dimension(2,*)	iad_elem,
		integer, dimension(*)	fr_elem,
		integer	lenr )

Definition at line 32 of file spmd_exch_v.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(*), NSIZ, LENR
      my_real xnorm(nsiz,*)
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,
     .        SIZ,J,K,L,NB_NOD,
     .        STATUS(MPI_STATUS_SIZE),
     .        IAD_SEND(NSPMD+1),IAD_RECV(NSPMD+1),
     .        REQ_R(NSPMD),REQ_S(NSPMD),MSGOFF,NSIZ1
 
      my_real, DIMENSION(:),ALLOCATABLE :: sbuf,rbuf
 
      DATA msgoff/7000/
C-----------------------------------------------
C   S o u r c e  L i n e s
C-----------------------------------------------
      ALLOCATE(sbuf(lenr*nsiz))
      ALLOCATE(rbuf(lenr*nsiz))
C-----------------------------------------------
      loc_proc = ispmd+1
      l = 1
      iad_recv(1)=1
 
C   
      DO i=1,nspmd
         siz = nsiz*(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
         END IF
         iad_recv(i+1) = l
      END DO
 
C   
      l=1
      iad_send(1)=1
      nsiz1 = nsiz-1
      DO i=1,nspmd
         DO j=iad_elem(1,i),iad_elem(1,i+1)-1
            nod = fr_elem(j)
            sbuf(l:l+nsiz1) = xnorm(1:nsiz,nod)
            l = l+nsiz
         END DO
        iad_send(i+1) = l
      END DO
 
      DO i=1,nspmd
 
       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                
      END DO
 
C   
      DO i = 1, nspmd
C
        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)
 
          DO j=iad_elem(1,i),iad_elem(1,i+1)-1
 
             nod = fr_elem(j)
             xnorm(1:nsiz,nod) = xnorm(1:nsiz,nod) + rbuf(l:l+nsiz1)
             l = l+nsiz
          ENDDO
        ENDIF
      ENDDO
 
      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
 
      DEALLOCATE(sbuf)
      DEALLOCATE(rbuf)
 
#endif
      RETURN