ParallelSolution.f90

Go to the documentation of this file.

module parallelsolutionmodule
  use kindmodule, only: dp, lgp, i4b
  use constantsmodule, only: lenpakloc, done, dzero
  use profilermodule
  use numericalsolutionmodule, only: numericalsolutiontype
  use mpi
  use mpiworldmodule
  implicit none
  private
 
  public :: parallelsolutiontype
 
  type, extends(numericalsolutiontype) :: parallelsolutiontype
    integer(I4B) :: tmr_convergence = -1 !< timer for convergence check
    integer(I4B) :: tmr_pkg_cnvg = -1 !< timer for package convergence check
    integer(I4B) :: tmr_sync_nur = -1 !< timer for NUR synchronization
    integer(I4B) :: tmr_nur_cnvg = -1 !< timer for NUR convergence check
    integer(I4B) :: tmr_calcptc = -1 !< timer for PTC calculation
    integer(I4B) :: tmr_underrelax = -1 !< timer for underrelaxation
    integer(I4B) :: tmr_backtracking = -1 !< timer for backtracking
  contains
    ! override
    procedure :: sln_has_converged => par_has_converged
    procedure :: sln_package_convergence => par_package_convergence
    procedure :: sln_sync_newtonur_flag => par_sync_newtonur_flag
    procedure :: sln_nur_has_converged => par_nur_has_converged
    procedure :: sln_calc_ptc => par_calc_ptc
    procedure :: sln_underrelax => par_underrelax
    procedure :: sln_backtracking_xupdate => par_backtracking_xupdate
 
  end type parallelsolutiontype
 
contains
 
  !> @brief Check global convergence. The local maximum dependent
  !! variable change is reduced over MPI with all other processes
  !< that are running this parallel numerical solution.
  function par_has_converged(this, max_dvc) result(has_converged)
    class(parallelsolutiontype) :: this !< parallel solution
    real(dp) :: max_dvc !< the LOCAL maximum dependent variable change
    logical(LGP) :: has_converged !< True, when GLOBALLY converged
    ! local
    real(dp) :: global_max_dvc
    real(dp) :: abs_max_dvc
    integer :: ierr
    type(mpiworldtype), pointer :: mpi_world
 
    call g_prof%start("Parallel Solution (cnvg check)", this%tmr_convergence)
 
    mpi_world => get_mpi_world()
 
    has_converged = .false.
    abs_max_dvc = abs(max_dvc)
    call mpi_allreduce(abs_max_dvc, global_max_dvc, 1, mpi_double_precision, &
                       mpi_max, mpi_world%comm, ierr)
    call check_mpi(ierr)
    if (global_max_dvc <= this%dvclose) then
      has_converged = .true.
    end if
 
    call g_prof%stop(this%tmr_convergence)
 
  end function par_has_converged
 
  function par_package_convergence(this, dpak, cpakout, iend) &
    result(icnvg_global)
    class(parallelsolutiontype) :: this !< parallel solution instance
    real(dp), intent(in) :: dpak !< Newton Under-relaxation flag
    character(len=LENPAKLOC), intent(in) :: cpakout
    integer(I4B), intent(in) :: iend
    ! local
    integer(I4B) :: icnvg_global
    integer(I4B) :: icnvg_local
    integer :: ierr
    type(mpiworldtype), pointer :: mpi_world
 
    call g_prof%start("Parallel Solution (package cnvg)", this%tmr_pkg_cnvg)
 
    mpi_world => get_mpi_world()
 
    icnvg_local = &
      this%NumericalSolutionType%sln_package_convergence(dpak, cpakout, iend)
 
    call mpi_allreduce(icnvg_local, icnvg_global, 1, mpi_integer, &
                       mpi_min, mpi_world%comm, ierr)
    call check_mpi(ierr)
 
    call g_prof%stop(this%tmr_pkg_cnvg)
 
  end function par_package_convergence
 
  function par_sync_newtonur_flag(this, inewtonur) result(ivalue)
    class(parallelsolutiontype) :: this !< parallel solution instance
    integer(I4B), intent(in) :: inewtonur !< local Newton Under-relaxation flag
    ! local
    integer(I4B) :: ivalue !< Maximum of all local values (1 = under-relaxation applied)
    integer :: ierr
    type(mpiworldtype), pointer :: mpi_world
 
    call g_prof%start("Parallel Solution (NUR)", this%tmr_sync_nur)
 
    mpi_world => get_mpi_world()
    call mpi_allreduce(inewtonur, ivalue, 1, mpi_integer, &
                       mpi_max, mpi_world%comm, ierr)
    call check_mpi(ierr)
 
    call g_prof%stop(this%tmr_sync_nur)
 
  end function par_sync_newtonur_flag
 
  function par_nur_has_converged(this, dxold_max, hncg) &
    result(has_converged)
    class(parallelsolutiontype) :: this !< parallel solution instance
    real(dp), intent(in) :: dxold_max !< the maximum dependent variable change for cells not adjusted by NUR
    real(dp), intent(in) :: hncg !< largest dep. var. change at end of Picard iter.
    logical(LGP) :: has_converged !< True, when converged
    ! local
    integer(I4B) :: icnvg_local
    integer(I4B) :: icnvg_global
    integer :: ierr
    type(mpiworldtype), pointer :: mpi_world
 
    call g_prof%start("Parallel Solution (NUR cnvg)", this%tmr_nur_cnvg)
 
    mpi_world => get_mpi_world()
 
    has_converged = .false.
    icnvg_local = 0
    if (this%NumericalSolutionType%sln_nur_has_converged( &
        dxold_max, hncg)) then
      icnvg_local = 1
    end if
 
    call mpi_allreduce(icnvg_local, icnvg_global, 1, mpi_integer, &
                       mpi_min, mpi_world%comm, ierr)
    call check_mpi(ierr)
    if (icnvg_global == 1) has_converged = .true.
 
    call g_prof%stop(this%tmr_nur_cnvg)
 
  end function par_nur_has_converged
 
  !> @brief Calculate pseudo-transient continuation factor
  !< for the parallel case
  subroutine par_calc_ptc(this, iptc, ptcf)
    class(parallelsolutiontype) :: this !< parallel solution
    integer(I4B) :: iptc !< PTC (1) or not (0)
    real(DP) :: ptcf !< the (global) PTC factor calculated
    ! local
    integer(I4B) :: iptc_loc
    real(DP) :: ptcf_loc, ptcf_glo_max
    integer :: ierr
    type(mpiworldtype), pointer :: mpi_world
 
    call g_prof%start("Parallel Solution (PTC calc)", this%tmr_calcptc)
 
    mpi_world => get_mpi_world()
    call this%NumericalSolutionType%sln_calc_ptc(iptc_loc, ptcf_loc)
    if (iptc_loc == 0) ptcf_loc = dzero
 
    ! now reduce
    call mpi_allreduce(ptcf_loc, ptcf_glo_max, 1, mpi_double_precision, &
                       mpi_max, mpi_world%comm, ierr)
    call check_mpi(ierr)
 
    iptc = 0
    ptcf = dzero
    if (ptcf_glo_max > dzero) then
      iptc = 1
      ptcf = ptcf_glo_max
    end if
 
    call g_prof%stop(this%tmr_calcptc)
 
  end subroutine par_calc_ptc
 
  !> @brief apply under-relaxation in sync over all processes
  !<
  subroutine par_underrelax(this, kiter, bigch, neq, active, x, xtemp)
    class(parallelsolutiontype) :: this !< parallel instance
    integer(I4B), intent(in) :: kiter !< Picard iteration number
    real(DP), intent(in) :: bigch !< maximum dependent-variable change
    integer(I4B), intent(in) :: neq !< number of equations
    integer(I4B), dimension(neq), intent(in) :: active !< active cell flag (1)
    real(DP), dimension(neq), intent(inout) :: x !< current dependent-variable
    real(DP), dimension(neq), intent(in) :: xtemp !< previous dependent-variable
    ! local
    real(DP) :: dvc_global_max, dvc_global_min
    integer :: ierr
    type(mpiworldtype), pointer :: mpi_world
 
    call g_prof%start("Parallel Solution (underrelax)", this%tmr_underrelax)
 
    mpi_world => get_mpi_world()
 
    ! first reduce largest change over all processes
    call mpi_allreduce(bigch, dvc_global_max, 1, mpi_double_precision, &
                       mpi_max, mpi_world%comm, ierr)
    call check_mpi(ierr)
    call mpi_allreduce(bigch, dvc_global_min, 1, mpi_double_precision, &
                       mpi_min, mpi_world%comm, ierr)
    call check_mpi(ierr)
 
    if (abs(dvc_global_min) > abs(dvc_global_max)) then
      dvc_global_max = dvc_global_min
    end if
 
    ! call local underrelax routine
    call this%NumericalSolutionType%sln_underrelax(kiter, dvc_global_max, &
                                                   neq, active, x, xtemp)
 
    call g_prof%stop(this%tmr_underrelax)
 
  end subroutine par_underrelax
 
  !> @brief synchronize backtracking flag over processes
  !<
  subroutine par_backtracking_xupdate(this, bt_flag)
    ! -- dummy variables
    class(parallelsolutiontype), intent(inout) :: this !< ParallelSolutionType instance
    integer(I4B), intent(inout) :: bt_flag !< global backtracking flag (1) backtracking performed (0) backtracking not performed
    ! -- local variables
    integer(I4B) :: btflag_local
    type(mpiworldtype), pointer :: mpi_world
    integer :: ierr
 
    call g_prof%start("Parallel Solution (backtrack)", this%tmr_backtracking)
 
    mpi_world => get_mpi_world()
 
    ! get local bt flag
    btflag_local = this%NumericalSolutionType%get_backtracking_flag()
 
    ! reduce into global decision (if any, then all)
    call mpi_allreduce(btflag_local, bt_flag, 1, mpi_integer, &
                       mpi_max, mpi_world%comm, ierr)
    call check_mpi(ierr)
 
    ! perform backtracking if ...
    if (bt_flag > 0) then
      call this%NumericalSolutionType%apply_backtracking()
    end if
 
    call g_prof%stop(this%tmr_backtracking)
 
  end subroutine par_backtracking_xupdate
 
end module parallelsolutionmodule