MpiUnitCache.f90

Go to the documentation of this file.

module mpiunitcachemodule
  use kindmodule, only: i4b, lgp
  use listmodule
  use stlvecintmodule
  use simstagesmodule, only: nr_sim_stages
  use mpi
  implicit none
  private
 
  integer(I4B), public, parameter :: no_cached_value = -1
 
  type, public :: mpiunitcachetype
    ! private
    type(stlvecint), private :: cached_ranks
    type(stlvecint), private :: cached_messages
    integer(I4B), private :: nr_stages
    integer(I4B), private :: nr_msg_types
  contains
    procedure :: init => cc_init
    procedure :: get_cached => cc_get_cached
    procedure :: cache => mc_cache
    procedure :: destroy => cc_destroy
    ! private
    procedure, private :: is_rank_cached
    procedure, private :: add_rank_cache
    procedure, private :: get_rank_index
    procedure, private :: get_msg_index
  end type mpiunitcachetype
 
contains
 
  !> @brief Initialize the unit cache.
  !<
  subroutine cc_init(this, nr_stages, nr_msg_types)
    class(mpiunitcachetype) :: this
    integer(I4B) :: nr_stages !< number of (simulation) stages
    integer(I4B) :: nr_msg_types !< number of message types to be cached during a stage
 
    this%nr_stages = nr_stages
    this%nr_msg_types = nr_msg_types
    call this%cached_ranks%init()
    call this%cached_messages%init()
 
  end subroutine cc_init
 
  !> @brief Get the cached mpi type for this rank and
  !< stage. Equal to NO_CACHED_VALUE when not present.
  function cc_get_cached(this, rank, stage, msg_id) result(mpi_type)
    class(mpiunitcachetype) :: this
    integer(I4B) :: rank
    integer(I4B) :: stage
    integer(I4B) :: msg_id
    integer :: mpi_type
    ! local
    integer(I4B) :: msg_idx
 
    mpi_type = no_cached_value
    msg_idx = this%get_msg_index(rank, stage, msg_id)
    if (msg_idx > 0) then
      mpi_type = this%cached_messages%at(msg_idx)
    end if
 
  end function cc_get_cached
 
  !> @brief Cache the mpi datatype for this particular
  !! rank and stage. The datatype should be committed
  !< to the type database externally.
  subroutine mc_cache(this, rank, stage, msg_id, mpi_type)
    class(mpiunitcachetype) :: this
    integer(I4B) :: rank
    integer(I4B) :: stage
    integer(I4B) :: msg_id
    integer :: mpi_type
    ! local
    integer(I4B) :: msg_idx
 
    ! add if rank not present in cache yet
    if (.not. this%is_rank_cached(rank)) then
      call this%add_rank_cache(rank)
    end if
 
    ! rank has been added to cache, now set
    ! mpi datatype for this stage's message:
    msg_idx = this%get_msg_index(rank, stage, msg_id)
    call this%cached_messages%set(msg_idx, mpi_type)
 
  end subroutine mc_cache
 
  function is_rank_cached(this, rank) result(in_cache)
    class(mpiunitcachetype) :: this
    integer(I4B) :: rank
    logical(LGP) :: in_cache
 
    in_cache = this%cached_ranks%contains(rank)
 
  end function is_rank_cached
 
  subroutine add_rank_cache(this, rank)
    class(mpiunitcachetype) :: this
    integer(I4B) :: rank
    ! local
    integer(I4B) :: i, j
 
    call this%cached_ranks%push_back(rank)
    do i = 1, this%nr_stages
      do j = 1, this%nr_msg_types
        call this%cached_messages%push_back(no_cached_value)
      end do
    end do
 
  end subroutine add_rank_cache
 
  !> @Brief returns -1 when not present
  !<
  function get_rank_index(this, rank) result(rank_index)
    class(mpiunitcachetype) :: this
    integer(I4B) :: rank
    integer(I4B) :: rank_index
 
    rank_index = this%cached_ranks%get_index(rank)
 
  end function get_rank_index
 
  !> @Brief returns -1 when not present
  !<
  function get_msg_index(this, rank, stage, msg_id) result(msg_index)
    class(mpiunitcachetype) :: this
    integer(I4B) :: rank
    integer(I4B) :: stage
    integer(I4B) :: msg_id
    integer(I4B) :: msg_index
    ! local
    integer(I4B) :: rank_idx
    integer(I4B) :: rank_offset, stage_offset
 
    msg_index = -1
    rank_idx = this%get_rank_index(rank)
    if (rank_idx < 1) return
 
    rank_offset = (rank_idx - 1) * (this%nr_stages * this%nr_msg_types)
    stage_offset = (stage - 1) * this%nr_msg_types
    msg_index = rank_offset + stage_offset + msg_id
 
  end function get_msg_index
 
  !> @brief Clean up the unit cache.
  !<
  subroutine cc_destroy(this)
    class(mpiunitcachetype) :: this
    ! local
    integer(I4B) :: i
    integer :: mpi_type, ierr
 
    do i = 1, this%cached_messages%size
      mpi_type = this%cached_messages%at(i)
      if (mpi_type /= no_cached_value) then
        call mpi_type_free(mpi_type, ierr)
      end if
    end do
 
    call this%cached_ranks%destroy()
    call this%cached_messages%destroy()
 
  end subroutine cc_destroy
 
end module