richardson_device.F90

Go to the documentation of this file.

 
 
module richardson_device
  use num_types, only : rp, c_rp
  use, intrinsic :: iso_c_binding, only : c_ptr
  use utils, only : neko_error
  implicit none
  private
 
#ifdef HAVE_HIP
  interface
     subroutine hip_richardson_compute(u_d, v_d, w_d, temp_d, &
          ind_r_d, ind_s_d, ind_t_d, ind_e_d, &
          n_x_d, n_y_d, n_z_d, h_d, &
          tau_x_d, tau_y_d, tau_z_d, n_nodes, lx, &
          kappa, mu_w_d, rho_w_d, g, Pr, z0, z0h_in, bc_type_int, &
          bc_value, tstep, &
          Ri_b_diagn, L_ob_diagn, utau_diagn, magu_diagn, ti_diagn, &
          ts_diagn, q_diagn, h_x_idx, h_y_idx, h_z_idx) &
          bind(c, name = 'hip_richardson_compute')
       use, intrinsic :: iso_c_binding, only : c_ptr, c_int
       use num_types, only : c_rp
       implicit none
       type(c_ptr), value :: u_d, v_d, w_d, temp_d
       type(c_ptr), value :: ind_r_d, ind_s_d, ind_t_d, ind_e_d
       type(c_ptr), value :: n_x_d, n_y_d, n_z_d, h_d
       real(c_rp) :: kappa, z0, z0h_in, bc_value, Pr
       type(c_ptr), value :: mu_w_d, rho_w_d
       real(c_rp) :: g(3)
       type(c_ptr), value :: tau_x_d, tau_y_d, tau_z_d
       integer(c_int) :: n_nodes, lx, tstep, bc_type_int
       type(c_ptr), value :: Ri_b_diagn, L_ob_diagn
       type(c_ptr), value :: utau_diagn, magu_diagn
       type(c_ptr), value :: ti_diagn, ts_diagn, q_diagn
       type(c_ptr), value :: h_x_idx, h_y_idx, h_z_idx
     end subroutine hip_richardson_compute
  end interface
#elif HAVE_CUDA
  interface
     subroutine cuda_richardson_compute(u_d, v_d, w_d, temp_d, &
          ind_r_d, ind_s_d, ind_t_d, ind_e_d, &
          n_x_d, n_y_d, n_z_d, h_d, &
          tau_x_d, tau_y_d, tau_z_d, n_nodes, lx, &
          kappa, mu_w_d, rho_w_d, g, Pr, z0, z0h_in, bc_type_int, &
          bc_value, tstep, &
          Ri_b_diagn, L_ob_diagn, utau_diagn, magu_diagn, ti_diagn, &
          ts_diagn, q_diagn, h_x_idx, h_y_idx, h_z_idx) &
          bind(c, name = 'cuda_richardson_compute')
       use, intrinsic :: iso_c_binding, only : c_ptr, c_int
       use num_types, only : c_rp
       implicit none
       type(c_ptr), value :: u_d, v_d, w_d, temp_d
       type(c_ptr), value :: ind_r_d, ind_s_d, ind_t_d, ind_e_d
       type(c_ptr), value :: n_x_d, n_y_d, n_z_d, h_d
       type(c_ptr), value :: mu_w_d, rho_w_d
       real(c_rp) :: kappa, z0, z0h_in, bc_value, Pr
       real(c_rp) :: g(3)
       type(c_ptr), value :: tau_x_d, tau_y_d, tau_z_d
       integer(c_int) :: n_nodes, lx, tstep, bc_type_int
       type(c_ptr), value :: Ri_b_diagn, L_ob_diagn
       type(c_ptr), value :: utau_diagn, magu_diagn
       type(c_ptr), value :: ti_diagn, ts_diagn, q_diagn
       type(c_ptr), value :: h_x_idx, h_y_idx, h_z_idx
     end subroutine cuda_richardson_compute
  end interface
#elif HAVE_OPENCL
#endif
  public :: richardson_compute_device
 
contains
  subroutine richardson_compute_device(u_d, v_d, w_d, temp_d, &
       ind_r_d, ind_s_d, ind_t_d, ind_e_d, &
       n_x_d, n_y_d, n_z_d, h_d, tau_x_d, tau_y_d, tau_z_d, &
       n_nodes, lx, kappa, mu_w_d, rho_w_d, g, Pr, z0, z0h_in, bc_type, &
       bc_value, tstep, Ri_b_diagn, L_ob_diagn, utau_diagn, &
       magu_diagn, ti_diagn, ts_diagn, q_diagn, &
       h_x_idx, h_y_idx, h_z_idx)
    integer, intent(in) :: n_nodes, lx, tstep
    type(c_ptr), intent(in) :: u_d, v_d, w_d, temp_d
    type(c_ptr), intent(in) :: ind_r_d, ind_s_d, ind_t_d, ind_e_d
    type(c_ptr), intent(in) :: n_x_d, n_y_d, n_z_d, h_d
    type(c_ptr), intent(inout) :: tau_x_d, tau_y_d, tau_z_d
    type(c_ptr), intent(in) :: mu_w_d, rho_w_d
    real(kind=rp), intent(in) :: kappa, z0, z0h_in, bc_value, pr
    real(kind=rp) :: g(3)
    character(len=*), intent(in) :: bc_type ! passed as a normal Fortran string
    integer :: bc_type_int
    type(c_ptr), value :: ri_b_diagn, l_ob_diagn
    type(c_ptr), value :: utau_diagn, magu_diagn
    type(c_ptr), value :: ti_diagn, ts_diagn, q_diagn
    type(c_ptr), value :: h_x_idx, h_y_idx, h_z_idx
 
    ! convert bc_type to integer to avoid cross-language passing of strings
    select case (trim(adjustl(bc_type))) ! (trimmed, lowercase-consistent)
    case ("neumann")
       bc_type_int = 0
    case ("dirichlet")
       bc_type_int = 1
    case default
       call neko_error("Neumann/Dirichlet bc not specified correctly " // &
            "(richardson_device)")
    end select
 
#if HAVE_HIP
    call hip_richardson_compute(u_d, v_d, w_d,temp_d, &
         ind_r_d, ind_s_d, ind_t_d, ind_e_d, &
         n_x_d, n_y_d, n_z_d, h_d, &
         tau_x_d, tau_y_d, tau_z_d, n_nodes, &
         lx, kappa, mu_w_d, rho_w_d, g, pr, z0, z0h_in, &
         bc_type_int, bc_value, tstep, ri_b_diagn, &
         l_ob_diagn, utau_diagn, magu_diagn, ti_diagn, &
         ts_diagn, q_diagn, h_x_idx, h_y_idx, h_z_idx)
#elif HAVE_CUDA
    call cuda_richardson_compute(u_d, v_d, w_d,temp_d, &
         ind_r_d, ind_s_d, ind_t_d, ind_e_d, &
         n_x_d, n_y_d, n_z_d, h_d, &
         tau_x_d, tau_y_d, tau_z_d, n_nodes, &
         lx, kappa, mu_w_d, rho_w_d, g, pr, z0, z0h_in, &
         bc_type_int, bc_value, tstep, ri_b_diagn, &
         l_ob_diagn, utau_diagn, magu_diagn, ti_diagn, &
         ts_diagn, q_diagn, h_x_idx, h_y_idx, h_z_idx)
#elif HAVE_OPENCL
    call neko_error("OPENCL is not implemented for the richardson wall model")
#else
    call neko_error('No device backend configured')
#endif
 
  end subroutine richardson_compute_device
end module richardson_device