MPIAnisoMortarExchangeAsync_MappedVector3D Subroutine

public subroutine MPIAnisoMortarExchangeAsync_MappedVector3D(this, mesh)

GPU-resident analogue of the base-class anisotropic mortar message posting for vectors; one message per (variable, direction) on device memory.

Arguments

TypeIntentOptionalAttributesName
class(MappedVector3D), intent(inout) :: this
type(Mesh3D), intent(inout) :: mesh

Calls

proc~~mpianisomortarexchangeasync_mappedvector3d~~CallsGraph proc~mpianisomortarexchangeasync_mappedvector3d MPIAnisoMortarExchangeAsync_MappedVector3D mpi_irecv mpi_irecv proc~mpianisomortarexchangeasync_mappedvector3d->mpi_irecv mpi_isend mpi_isend proc~mpianisomortarexchangeasync_mappedvector3d->mpi_isend

Contents


Source Code

  subroutine MPIAnisoMortarExchangeAsync_MappedVector3D(this,mesh)
    !! GPU-resident analogue of the base-class anisotropic mortar message posting for
    !! vectors; one message per (variable, direction) on device memory.
    implicit none
    class(MappedVector3D),intent(inout) :: this
    type(Mesh3D),intent(inout) :: mesh
    ! Local
    integer :: m,h,ivar,idir
    integer :: eB,sB,rB,eS,sS,rS
    integer :: globalSideId,tag
    integer :: offset
    integer :: iError
    integer :: msgCount
    real(prec),pointer :: boundary(:,:,:,:,:,:)
    real(prec),pointer :: anisoMortarBuff(:,:,:,:,:,:)

    ! A send and a receive per rank-crossing half, per variable and direction.
    call mesh%decomp%ReserveMessages(2*3*this%nvar* &
                                     mesh%decomp%CountRemoteMortarSides(mesh%anisoMortarInfo,mesh%nAnisoMortars,2))

    msgCount = 0
    offset = mesh%decomp%offsetElem(mesh%decomp%rankId+1)
    call c_f_pointer(this%boundary_gpu,boundary, &
                     [this%interp%N+1,this%interp%N+1,6,this%nelem,this%nvar,3])
    call c_f_pointer(this%anisoMortarBuff_gpu,anisoMortarBuff, &
                     [this%interp%N+1,this%interp%N+1,4,mesh%nAnisoMortars,this%nvar,3])

    do idir = 1,3
      do ivar = 1,this%nvar
        do m = 1,mesh%nAnisoMortars

          eB = mesh%anisoMortarInfo(1,m)
          sB = mesh%anisoMortarInfo(2,m)
          rB = mesh%decomp%elemToRank(eB)

          do h = 1,2

            eS = mesh%anisoMortarInfo(2*h+1,m)
            sS = mesh%anisoMortarInfo(2*h+2,m)/10
            rS = mesh%decomp%elemToRank(eS)
            globalSideId = mesh%anisoMortarInfo(6+h,m)
            tag = globalSideId+mesh%nUniqueSides*(ivar-1+this%nvar*(idir-1))

            if(rB == mesh%decomp%rankId .and. rS /= mesh%decomp%rankId) then

              msgCount = msgCount+1
              call MPI_IRECV(anisoMortarBuff(:,:,2+h,m,ivar,idir), &
                             (this%interp%N+1)*(this%interp%N+1), &
                             mesh%decomp%mpiPrec, &
                             rS,tag, &
                             mesh%decomp%mpiComm, &
                             mesh%decomp%requests(msgCount),iError)

              msgCount = msgCount+1
              call MPI_ISEND(boundary(:,:,sB,eB-offset,ivar,idir), &
                             (this%interp%N+1)*(this%interp%N+1), &
                             mesh%decomp%mpiPrec, &
                             rS,tag, &
                             mesh%decomp%mpiComm, &
                             mesh%decomp%requests(msgCount),iError)

            elseif(rS == mesh%decomp%rankId .and. rB /= mesh%decomp%rankId) then

              msgCount = msgCount+1
              call MPI_IRECV(anisoMortarBuff(:,:,h,m,ivar,idir), &
                             (this%interp%N+1)*(this%interp%N+1), &
                             mesh%decomp%mpiPrec, &
                             rB,tag, &
                             mesh%decomp%mpiComm, &
                             mesh%decomp%requests(msgCount),iError)

              msgCount = msgCount+1
              call MPI_ISEND(boundary(:,:,sS,eS-offset,ivar,idir), &
                             (this%interp%N+1)*(this%interp%N+1), &
                             mesh%decomp%mpiPrec, &
                             rB,tag, &
                             mesh%decomp%mpiComm, &
                             mesh%decomp%requests(msgCount),iError)

            endif

          enddo
        enddo
      enddo
    enddo

    mesh%decomp%msgCount = msgCount

  endsubroutine MPIAnisoMortarExchangeAsync_MappedVector3D