spmd_exch_sms.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_sms (v, nodnx_sms, iad_elem, fr_elem, size, lenr)

Function/Subroutine Documentation

spmd_exch_sms()

subroutine spmd_exch_sms	(		v,
		integer, dimension(*)	nodnx_sms,
		integer, dimension(2,*)	iad_elem,
		integer, dimension(*)	fr_elem,
		integer	size,
		integer	lenr )

Definition at line 34 of file spmd_exch_sms.F.

        USE spmd_mod
C-----------------------------------------------
C   I m p l i c i t   T y p e s
C-----------------------------------------------
#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 NODNX_SMS(*), IAD_ELEM(2,*), FR_ELEM(*), SIZE, LENR
      my_real
     .        v(SIZE,*)
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,K,
     .        STATUS(MPI_STATUS_SIZE),
     .        IAD_SEND(NSPMD+1),IAD_RECV(NSPMD+1),
     .        REQ_R(NSPMD),REQ_S(NSPMD)
      my_real
     .        rbuf(size*lenr), sbuf(size*lenr), w(SIZE,lenr)
      DATA msgoff/213/
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 = 0
        DO j=iad_elem(1,i),iad_elem(1,i+1)-1
          nod = fr_elem(j)
          IF(nodnx_sms(nod)/=0)THEN
            siz = siz + SIZE
          END IF
        END DO
 
        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(nodnx_sms(nod)/=0)THEN
            DO k=1,SIZE
              sbuf(l+k-1) =  v(k,nod)
            END DO
            l = l + SIZE
          END IF
        END DO
        iad_send(i+1) = l
      ENDDO
C
C   echange messages
C
      DO i=1,nspmd
C--------------------------------------------------------------------
ccc       IF(IAD_ELEM(1,I+1)-IAD_ELEM(1,I)>0)THEN
       IF(iad_send(i+1)-iad_send(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 j=1,iad_elem(1,nspmd+1)-1
        nod = fr_elem(j)
        IF(nodnx_sms(nod)/=0)THEN
          DO k=1,SIZE
            w(k,j)   = v(k,nod)
            v(k,nod) = zero
          END DO
        END IF
      END DO
C
      DO i = 1, loc_proc-1
ccc        IF(IAD_ELEM(1,I+1)-IAD_ELEM(1,I)>0)THEN
        IF(iad_recv(i+1)-iad_recv(i)>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(nodnx_sms(nod)/=0) THEN
              DO k=1,SIZE
                v(k,nod) = v(k,nod)+rbuf(l+k-1)
              END DO
              l = l + SIZE
            END IF
          END DO
        ENDIF
      END DO
C
      DO j=1,iad_elem(1,nspmd+1)-1
        nod = fr_elem(j)
        IF(nodnx_sms(nod)/=0)THEN
          DO k=1,SIZE
            v(k,nod) = v(k,nod) + w(k,j)
          END DO
        END IF
      END DO
C
      DO i = loc_proc+1,nspmd
ccc        IF(IAD_ELEM(1,I+1)-IAD_ELEM(1,I)>0)THEN
        IF(iad_recv(i+1)-iad_recv(i)>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(nodnx_sms(nod)/=0) THEN
              DO k=1,SIZE
                v(k,nod) = v(k,nod)+rbuf(l+k-1)
              END DO
              l = l + SIZE
            END IF
          END DO
        ENDIF
      END DO
C
      DO i = 1, nspmd
ccc        IF(IAD_ELEM(1,I+1)-IAD_ELEM(1,I)>0)THEN
        IF(iad_send(i+1)-iad_send(i)>0)THEN
          CALL mpi_wait(req_s(i),status,ierror)
        ENDIF
      ENDDO
C
#endif
      RETURN