spmd_gather_wa.F File Reference

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

Go to the source code of this file.

Functions/Subroutines
subroutine	spmd_gather_wa (mode, size_buffer_s, size_buffer_r, sindex, rindex_proc, buffer_s, buffer_r, shift_r, nb_elem_r)

Function/Subroutine Documentation

spmd_gather_wa()

subroutine spmd_gather_wa	(	integer, intent(in)	mode,
		integer, intent(in)	size_buffer_s,
		integer, intent(inout)	size_buffer_r,
		integer, dimension(nsect+1,3), intent(in)	sindex,
		integer, dimension(nsect+1,3,nspmd), intent(inout)	rindex_proc,
		integer, dimension(size_buffer_s), intent(in)	buffer_s,
		integer, dimension(*), intent(inout)	buffer_r,
		integer, dimension(nspmd), intent(inout)	shift_r,
		integer, dimension(nspmd), intent(inout)	nb_elem_r )

Definition at line 31 of file spmd_gather_wa.F.

C-----------------------------------------------    
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"
#include "spmd.inc"
C-----------------------------------------------
C   C o m m o n   B l o c k s
C-----------------------------------------------
#include      "com01_c.inc"
#include      "com04_c.inc"
#include      "task_c.inc"
C-----------------------------------------------
C   D u m m y   A r g u m e n t s
C-----------------------------------------------
!       MODE = 0 --> collect the index array for each proc
!       MODE = 1 --> collect all the buffer arrays
!       SIZE_BUFFER_S is the size of the local sended array
        INTEGER, INTENT(IN) :: MODE,SIZE_BUFFER_S
!       SIZE_BUFFER_R is the size of the global received array, proc 0 
        INTEGER, INTENT(INOUT) :: SIZE_BUFFER_R
!       SINDEX : index buffer array, local on each proc                      
        INTEGER, DIMENSION(NSECT+1,3), INTENT(IN) :: SINDEX
!       RINDEX_PROC : global index buffer array, only on proc 0
        INTEGER, DIMENSION(NSECT+1,3,NSPMD), INTENT(INOUT) :: RINDEX_PROC
!       BUFFER_S : sended buffer
        INTEGER, DIMENSION(SIZE_BUFFER_S), INTENT(IN) :: BUFFER_S
!       BUFFER_R : received buffer, size = SIZE_BUFFER_R on proc 0
        INTEGER, DIMENSION(*), INTENT(INOUT) :: BUFFER_R
        INTEGER, DIMENSION(NSPMD), INTENT(INOUT) :: SHIFT_R,NB_ELEM_R
C-----------------------------------------------
C   L O C A L   V A R I A B L E S
C-----------------------------------------------
#ifdef MPI
      INTEGER IERROR,MSGOFF
      INTEGER I
 
C-----------------------------------------------
      DATA msgoff/7069/
C-----------------------------------------------
        
        IF(mode==0) THEN
        !       collect the local index arrays on the main processor
                CALL mpi_gather(sindex,3*(nsect+1),mpi_integer,
     .                   rindex_proc,3*(nsect+1),mpi_integer,0,spmd_comm_world,ierror)  
        !       sum all the size of local sended arrays in order to get the size 
        !       of the global array on proc 0                  
                IF(ispmd==0) THEN
                        nb_elem_r(1:nspmd) = rindex_proc(nsect+1,1,1:nspmd)
                        size_buffer_r = 0
                        DO i=1,nspmd
                                size_buffer_r = size_buffer_r + nb_elem_r(i)
                        ENDDO
        !       shift for the _gatherv comm
                        shift_r(1) = 0
                        DO i=2,nspmd
                                shift_r(i) = shift_r(i-1)+nb_elem_r(i-1)
                        ENDDO
                ELSE
                        size_buffer_r = 0
                ENDIF
        ELSEIF(mode==1) THEN
 
                CALL mpi_gatherv(buffer_s,size_buffer_s,mpi_integer,
     .                           buffer_r,nb_elem_r,shift_r,mpi_integer,
     .                           0,spmd_comm_world,ierror)   
        ENDIF
#endif
       RETURN