steepest_descent_math.f90

Go to the documentation of this file.

 
 
module steepest_descent_math
  use num_types, only: rp
  use vector, only: vector_t
  use neko_config, only: neko_bcknd_device
  use steepest_descent_cpu, only: sd_direction_cpu, &
       sd_update_move_limits_cpu, sd_update_design_cpu
  use device_steepest_descent, only: device_sd_direction, &
       device_sd_update_move_limits, device_sd_update_design
  implicit none
  private
 
  public :: sd_direction, sd_update_move_limits, sd_update_design
 
contains
 
  subroutine sd_direction(s, x, xmin, xmax, n)
    type(vector_t), intent(inout) :: s
    type(vector_t), intent(in) :: x
    real(kind=rp), intent(in) :: xmin, xmax
    integer, intent(in), optional :: n
    integer :: n_local
 
    if (present(n)) then
       n_local = n
    else
       n_local = s%size()
    end if
 
    ! An empty range has nothing to do, and its host array may be unallocated
    if (n_local .lt. 1) return
 
    if (neko_bcknd_device .eq. 1) then
       call device_sd_direction(s%x_d, x%x_d, xmin, xmax, n_local)
    else
       call sd_direction_cpu(s%x, x%x, xmin, xmax, n_local)
    end if
 
  end subroutine sd_direction
 
  subroutine sd_update_move_limits(delta, x, x1, x2, dec, inc, dmin, dmax, n)
    type(vector_t), intent(inout) :: delta
    type(vector_t), intent(in) :: x, x1, x2
    real(kind=rp), intent(in) :: dec, inc, dmin, dmax
    integer, intent(in), optional :: n
    integer :: n_local
 
    if (present(n)) then
       n_local = n
    else
       n_local = delta%size()
    end if
 
    ! An empty range has nothing to do, and its host array may be unallocated
    if (n_local .lt. 1) return
 
    if (neko_bcknd_device .eq. 1) then
       call device_sd_update_move_limits(delta%x_d, x%x_d, x1%x_d, x2%x_d, &
            dec, inc, dmin, dmax, n_local)
    else
       call sd_update_move_limits_cpu(delta%x, x%x, x1%x, x2%x, &
            dec, inc, dmin, dmax, n_local)
    end if
 
  end subroutine sd_update_move_limits
 
  subroutine sd_update_design(x_new, x, s, delta, psi, smax, xmin, xmax, n)
    type(vector_t), intent(inout) :: x_new
    type(vector_t), intent(in) :: x, s, delta
    real(kind=rp), intent(in) :: psi, smax, xmin, xmax
    integer, intent(in), optional :: n
    integer :: n_local
 
    if (present(n)) then
       n_local = n
    else
       n_local = x_new%size()
    end if
 
    ! An empty range has nothing to do, and its host array may be unallocated
    if (n_local .lt. 1) return
 
    if (neko_bcknd_device .eq. 1) then
       call device_sd_update_design(x_new%x_d, x%x_d, s%x_d, delta%x_d, &
            psi, smax, xmin, xmax, n_local)
    else
       call sd_update_design_cpu(x_new%x, x%x, s%x, delta%x, &
            psi, smax, xmin, xmax, n_local)
    end if
 
  end subroutine sd_update_design
 
end module steepest_descent_math