spmd_vfi_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_vfi_sms (v, size, vfi, fr_rms, fr_sms, list_rms, list_sms, iflag, iad_send, iad_recv, req_r, req_s, rbuf, sbuf)

Function/Subroutine Documentation

spmd_vfi_sms()

subroutine spmd_vfi_sms	(		v,
		integer	size,
			vfi,
		integer, dimension(nspmd+1)	fr_rms,
		integer, dimension(nspmd+1)	fr_sms,
		integer, dimension(*)	list_rms,
		integer, dimension(*)	list_sms,
		integer	iflag,
		integer, dimension(nspmd+1)	iad_send,
		integer, dimension(nspmd+1)	iad_recv,
		integer, dimension(nspmd)	req_r,
		integer, dimension(nspmd)	req_s,
			rbuf,
			sbuf )

Definition at line 31 of file spmd_vfi_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 SIZE, IFLAG,
     .        FR_RMS(NSPMD+1), FR_SMS(NSPMD+1),LIST_RMS(*), LIST_SMS(*),
     .        IAD_SEND(NSPMD+1),IAD_RECV(NSPMD+1),
     .        REQ_R(NSPMD),REQ_S(NSPMD)
      my_real
     .        v(SIZE,*), vfi(SIZE,*)
      my_real
     .        rbuf(size*(fr_rms(nspmd+1)+fr_sms(nspmd+1))),
     .        sbuf(size*(fr_rms(nspmd+1)+fr_sms(nspmd+1)))
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,M,KK,
     .        STATUS(MPI_STATUS_SIZE)
      DATA msgoff/217/
C-----------------------------------------------
C   S o u r c e  L i n e s
C-----------------------------------------------
      loc_proc = ispmd + 1
 
      IF(iflag==1)THEN
        l = 1
        iad_recv(1) = 1
        DO i=1,nspmd
 
          siz = fr_rms(i+1) - fr_rms(i)
          IF(i/=loc_proc)THEN
            siz = siz + fr_sms(i+1) - fr_sms(i)
          ENDIF
          siz = size*siz
 
          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)
          ENDIF
          l = l + siz
          iad_recv(i+1) = l
 
        END DO
 
        l = 1
        iad_send(1) = 1
        m = 1
        DO i=1,nspmd
          IF(i/=loc_proc)THEN
#include      "vectorize.inc"
            DO j=fr_sms(i),fr_sms(i+1)-1
              nod = list_sms(m)
              IF(nod/=0)THEN
                DO k=1,SIZE
                  sbuf(l+k-1) =  v(k,nod)
                END DO
              ELSE
                DO k=1,SIZE
                  sbuf(l+k-1) =  zero
                END DO
              END IF
              l = l + SIZE
              m = m + 1
            END DO
          END IF
#include      "vectorize.inc"
          DO j=fr_rms(i),fr_rms(i+1)-1
            nod = list_rms(j)
            IF(nod/=0)THEN
              DO k=1,SIZE
                sbuf(l+k-1) =  v(k,nod)
              END DO
            ELSE
              DO k=1,SIZE
                sbuf(l+k-1) =  zero
              END DO
            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_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
      ELSE ! IFLAG==1
C
        m = 1
        kk= 1
        DO i = 1, nspmd
          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=fr_rms(i),fr_rms(i+1)-1
              IF(list_rms(j)/=0)THEN
                DO k=1,SIZE
                  vfi(k,kk) = rbuf(l+k-1)
                END DO
              END IF
              l = l + SIZE
              kk= kk + 1
            END DO
            IF(i/=loc_proc)THEN
#include      "vectorize.inc"
              DO j=fr_sms(i),fr_sms(i+1)-1
                IF(list_sms(m)/=0)THEN
                  DO k=1,SIZE
                    vfi(k,kk) = rbuf(l+k-1)
                  END DO
                END IF
                l = l + SIZE
                m = m + 1
                kk= kk + 1
              END DO
            END IF
          ENDIF
        END DO
C
        DO i = 1, nspmd
          IF(iad_send(i+1)-iad_send(i)>0)THEN
            CALL mpi_wait(req_s(i),status,ierror)
          ENDIF
        ENDDO
C
      END IF
#endif
      RETURN