mpl_stats_mod.F90

Go to the documentation of this file.

MODULE mpl_stats_mod

PRIVATE
PUBLIC :: mpl_statsinit, mpl_statson, mpl_statsread, mpl_sendstats, mpl_recvstats

CONTAINS

SUBROUTINE mpl_statsinit

!**** MPL_STATSINIT - Initialise collection of mpl statistics

!     Purpose.
!     --------
!     Initialises the mpl_stats package 

!**   Interface.
!     ----------
!        CALL MPL_STATSINIT

!        Input required arguments :
!        -------------------------
!           none

!        Input optional arguments :
!        -------------------------
!           none

!        Output required arguments :
!        -------------------------
!           none

!        Output optional arguments :
!        -------------------------
!           none 
 
!     Author.
!     -------
!        P.Towers             ECMWF

!     Modifications.
!     --------------
!        Original: 2011-04-06
!      F. Vana  05-Mar-2015  Support for single precision

!     ------------------------------------------------------------------

USE parkind1  ,ONLY : jpim

USE yommplstats

IMPLICIT NONE

INTEGER(KIND=JPIM) :: ITHR,OMP_GET_MAX_THREADS

lmplstats=.true.

ithr = 1
!$ ITHR = OMP_GET_MAX_THREADS()
ithr = ithr-1

ALLOCATE(mplsendbytes(0:ithr))
ALLOCATE(mplrecvbytes(0:ithr))
ALLOCATE(mplsendnum(0:ithr))
ALLOCATE(mplrecvnum(0:ithr))
  
mplsendbytes(:) = 0
mplrecvbytes(:) = 0
mplsendnum(:) = 0
mplrecvnum(:) = 0

RETURN
END SUBROUTINE mpl_statsinit

SUBROUTINE mpl_statson(SENDNUM,SENDBYTES,RECVNUM,RECVBYTES)

!**** MPL_STATSON - Reset mpl statistics counters

!     Purpose.
!     --------
!     Returns the mpl statistics counter values
!     and sets them back to zero
!     non zero returned values correspond to messages that have
!     been sent/received outside of a GSTATS MPL region

!**   Interface.
!     ----------
!        CALL MPL_STATSON(SENDNUM,SENDBYTES,RECVNUM,RECVBYTES)

!        Input required arguments :
!        -------------------------
!           none

!        Input optional arguments :
!        -------------------------
!           none

!        Output required arguments :
!        -------------------------
!           SENDNUM     - number of unknown messages sent
!           SENDBYTES   - number of unknown bytes sent
!           RECVNUM     - number of unknown messages received
!           RECVBYTES   - number of unknown bytes received

!        Output optional arguments :
!        -------------------------
!           none 
 
!     Author.
!     -------
!        P.Towers             ECMWF

!     Modifications.
!     --------------
!        Original: 2011-04-06

!     ------------------------------------------------------------------

USE parkind1  ,ONLY : jprd, jpim

USE yommplstats

IMPLICIT NONE

REAL(KIND=JPRD), INTENT(OUT)    :: SENDBYTES,RECVBYTES
INTEGER(KIND=JPIM), INTENT(OUT) :: SENDNUM,RECVNUM


 sendbytes = sum(mplsendbytes(:))
 recvbytes = sum(mplrecvbytes(:))
 sendnum   = sum(mplsendnum(:))
 recvnum   = sum(mplrecvnum(:))
 
 mplsendbytes(:)=0.0_jprd
 mplrecvbytes(:)=0.0_jprd
 mplsendnum(:)=0
 mplrecvnum(:)=0
 
RETURN
END SUBROUTINE mpl_statson

SUBROUTINE mpl_statsread(SENDNUM,SENDBYTES,RECVNUM,RECVBYTES)

!**** MPL_STATSREAD - read mpl statistics counters

!     Purpose.
!     --------
!     returns the mpl statistics counter values
!     before setting them back to zero

!**   Interface.
!     ----------
!        CALL MPL_STATSREAD(SENDNUM,SENDBYTES,RECVNUM,RECVBYTES)

!        Input required arguments :
!        -------------------------
!           none

!        Input optional arguments :
!        -------------------------
!           none

!        Output required arguments :
!        -------------------------
!           SENDNUM     - number of messages sent
!           SENDBYTES   - number of bytes sent
!           RECVNUM     - number of messages received
!           RECVBYTES   - number of bytes received

!        Output optional arguments :
!        -------------------------
!           none 
 
!     Author.
!     -------
!        P.Towers             ECMWF

!     Modifications.
!     --------------
!        Original: 2011-04-06

!     ------------------------------------------------------------------

USE parkind1  ,ONLY : jprd, jpim

USE yommplstats

IMPLICIT NONE

REAL(KIND=JPRD), INTENT(OUT)    :: SENDBYTES,RECVBYTES
INTEGER(KIND=JPIM), INTENT(OUT) :: SENDNUM,RECVNUM

  sendnum=sum(mplsendnum(:))
  recvnum=sum(mplrecvnum(:))
  sendbytes=sum(mplsendbytes(:))
  recvbytes=sum(mplrecvbytes(:))

  mplsendnum(:)=0
  mplrecvnum(:)=0
  mplsendbytes(:)=0.0_jprd
  mplrecvbytes(:)=0.0_jprd

RETURN
END SUBROUTINE mpl_statsread

SUBROUTINE mpl_sendstats(ICOUNT,ITYPE)

!**** MPL_SENDSTATS - collect mpl send statistics

!     Purpose.
!     --------
!     counts the number of messages and volume sent

!**   Interface.
!     ----------
!        CALL MPL_SENDSTATS(ICOUNT,ITYPE)

!        Input required arguments :
!        -------------------------
!           ICOUNT   - The number of elements sent
!           ITYPE    - The type of an element 

!        Input optional arguments :
!        -------------------------
!           none

!        Output required arguments :
!        -------------------------
!           none

!        Output optional arguments :
!        -------------------------
!           none 
 
!     Author.
!     -------
!        P.Towers             ECMWF

!     Modifications.
!     --------------
!        Original: 2011-04-06

!     ------------------------------------------------------------------

USE parkind1  ,ONLY : jpim

USE mpl_data_module
USE mpl_message_mod
USE yommplstats

IMPLICIT NONE

INTEGER(KIND=JPIM),INTENT(IN)            :: ICOUNT
INTEGER(KIND=JPIM),INTENT(IN)            :: ITYPE

INTEGER(KIND=JPIM)  ISIZE,IERR,ITH,OMP_GET_THREAD_NUM

ith = 0
!$ ITH = OMP_GET_THREAD_NUM()

mplsendnum(ith) = mplsendnum(ith) + 1

CALL mpi_type_size(itype,isize,ierr)

mplsendbytes(ith)=mplsendbytes(ith) + float(isize * icount)

  
RETURN
END SUBROUTINE mpl_sendstats

SUBROUTINE mpl_recvstats(ICOUNT,ITYPE)

!**** MPL_RECVSTATS - collect mpl recv statistics

!     Purpose.
!     --------
!     counts the number of messages and volume received

!**   Interface.
!     ----------
!        CALL MPL_RECVSTATS(ICOUNT,ITYPE)

!        Input required arguments :
!        -------------------------
!           ICOUNT   - The number of elements received
!           ITYPE    - The type of an element 

!        Input optional arguments :
!        -------------------------
!           none

!        Output required arguments :
!        -------------------------
!           none

!        Output optional arguments :
!        -------------------------
!           none 
 
!     Author.
!     -------
!        P.Towers             ECMWF

!     Modifications.
!     --------------
!        Original: 2011-04-06

!     ------------------------------------------------------------------

USE parkind1  ,ONLY : jpim

USE mpl_data_module
USE mpl_message_mod
USE yommplstats

IMPLICIT NONE

INTEGER(KIND=JPIM),INTENT(IN)            :: ICOUNT
INTEGER(KIND=JPIM),INTENT(IN)            :: ITYPE

INTEGER(KIND=JPIM) ISIZE,IERR,ITH,OMP_GET_THREAD_NUM

ith = 0
!$ ITH = OMP_GET_THREAD_NUM()

mplrecvnum(ith) = mplrecvnum(ith) + 1

CALL mpi_type_size(itype,isize,ierr)

mplrecvbytes(ith)=mplrecvbytes(ith) + float(isize * icount)
  
RETURN
END SUBROUTINE mpl_recvstats

END MODULE mpl_stats_mod