Homec4science
Diffusion Introduction to CUDA (master) Browse

introCUDA/exercises/gradients/cuda_interface_mod.F90
e768432c3298master

Last Change
Raw File

cuda_interface_mod.F90
View Options

MODULE cuda_interface
use prec_const
INTERFACE
FUNCTION allocate_cuda_memory(n) BIND(C, NAME='allocate_cuda_memory')
use iso_c_binding
TYPE(C_PTR) :: allocate_cuda_memory
integer :: n
END FUNCTION allocate_cuda_memory
END INTERFACE
INTERFACE gbs_allocate_cuda
MODULE PROCEDURE gbs_allocate_cuda_dp3, gbs_allocate_cuda_dp2, gbs_allocate_cuda_dp1
END INTERFACE gbs_allocate_cuda
real(dp), pointer, save :: strmf_cpu(:,:,:),strmf_gpu(:,:,:), indata(:,:,:)
real(dp), pointer, save :: strmf1_cpu(:,:,:), strmf1_gpu(:,:,:)
real(dp), pointer, save :: grady_f(:,:,:), grady_diff(:,:,:), diff(:,:,:), f_x(:,:,:)
real(dp), pointer, save :: f_z(:,:,:), f_y(:,:,:), f_n(:,:,:), f_x_n(:,:,:)
real(dp), pointer, save :: f_v(:,:,:), f_x_v(:,:,:), f_xx_n(:,:,:),f_xx_v(:,:,:)
real(dp), pointer, save :: f_zz(:,:,:), f_yy(:,:,:), f_xx(:,:,:)
real(dp), pointer, save :: f_zy(:,:,:), f_yz(:,:,:), f_zx(:,:,:)
real(dp), pointer, save :: f_xz(:,:,:), f_xy(:,:,:), f_yx(:,:,:)
real(dp), pointer, save :: gradpar_x_n_gpu(:,:), gradpar_x_v_gpu(:,:), gradpar_y_n_gpu(:,:), gradpar_y_v_gpu(:,:)
real(dp), pointer, save :: grad2par_yy_n_gpu(:,:), grad2par_xx_n_gpu(:,:), grad2par_x_n_gpu(:,:), grad2par_y_n_gpu(:,:)
real(dp), pointer, save :: grad2par_zy_n_gpu(:,:), grad2par_zx_n_gpu(:,:), grad2par_xy_n_gpu(:,:)
real(dp), pointer, save :: grad2par_yy_v_gpu(:,:), grad2par_xx_v_gpu(:,:), grad2par_x_v_gpu(:,:), grad2par_y_v_gpu(:,:)
real(dp), pointer, save :: grad2par_zy_v_gpu(:,:), grad2par_zx_v_gpu(:,:), grad2par_xy_v_gpu(:,:)
real(dp), pointer, save :: cur_x_n_gpu(:), cur_x_v_gpu(:), ccur_xx_n_gpu(:), ccur_xx_v_gpu(:)
real(dp), pointer, save :: cur_y_n_gpu(:,:), cur_y_v_gpu(:,:), ccur_yy_n_gpu(:,:), ccur_xy_n_gpu(:,:)
real(dp), pointer, save :: ccur_yy_v_gpu(:,:), ccur_xy_v_gpu(:,:)
real(dp), pointer, dimension(:,:,:), save :: A_x_n_gpu, A_x_v_gpu, A_y_v_gpu, B_x_v_gpu, B_y_v_gpu
real(dp), pointer, dimension(:,:,:), save :: B_x_n_gpu, B_y_n_gpu, A_y_n_gpu
real(dp), pointer, dimension(:,:), save :: gradcur_xx_v_gpu, gradcur_yy_v_gpu, gradcur_xy_v_gpu, gradcur_yx_v_gpu
real(dp), pointer, dimension(:,:), save :: gradcur_zx_v_gpu, gradcur_zy_v_gpu, gradcur_x_v_gpu, gradcur_y_v_gpu
real(dp), pointer, dimension(:,:), save :: curgrad_xx_n_gpu, curgrad_yy_n_gpu, curgrad_xy_n_gpu, curgrad_yx_n_gpu
real(dp), pointer, dimension(:,:), save :: curgrad_zx_n_gpu, curgrad_zy_n_gpu, curgrad_x_n_gpu, curgrad_y_n_gpu
!real(dp), pointer, save :: gradpar_x_n_gpu(:,:,:), gradpar_x_v_gpu(:,:,:), gradpar_y_n_gpu(:,:,:), gradpar_y_v_gpu(:,:,:)
integer :: nx_cuda, ny_cuda, nz_cuda
contains
SUBROUTINE gbs_allocate_cuda_dp3(a,is1,ie1,is2,ie2,is3,ie3)
use iso_c_binding
real(dp), DIMENSION(:,:,:), pointer, INTENT(INOUT) :: a
INTEGER, INTENT(IN) :: is1,ie1,is2,ie2,is3,ie3
integer ndata
ndata=(ie3-is3+1)*(ie2-is2+1)*(ie1-is1+1)
call c_f_pointer(allocate_cuda_memory(ndata), a, \
[(ie1-is1+1),(ie2-is2+1),(ie3-is3+1)])
a(is1:,is2:,is3:) => a
a=0.0_dp
END SUBROUTINE gbs_allocate_cuda_dp3
SUBROUTINE gbs_allocate_cuda_dp2(a,is1,ie1,is2,ie2)
use iso_c_binding
real(dp), DIMENSION(:,:), pointer, INTENT(INOUT) :: a
INTEGER, INTENT(IN) :: is1,ie1,is2,ie2
integer ndata
ndata=(ie2-is2+1)*(ie1-is1+1)
call c_f_pointer(allocate_cuda_memory(ndata), a, \
[(ie1-is1+1),(ie2-is2+1)])
a(is1:,is2:) => a
a=0.0_dp
END SUBROUTINE gbs_allocate_cuda_dp2
SUBROUTINE gbs_allocate_cuda_dp1(a,is1,ie1)
use iso_c_binding
real(dp), DIMENSION(:), pointer, INTENT(INOUT) :: a
INTEGER, INTENT(IN) :: is1,ie1
integer ndata
ndata=(ie1-is1+1)
call c_f_pointer(allocate_cuda_memory(ndata), a, \
[(ie1-is1+1)])
a(is1:) => a
a=0.0_dp
END SUBROUTINE gbs_allocate_cuda_dp1
subroutine gradz_n2n_fd4cuda(f_in, f_out)
use space_grid
real(dp), DIMENSION(iysg:iyeg,ixsg:ixeg,izsg:izeg), INTENT(in) :: f_in
real(dp), DIMENSION(iysg:iyeg,ixsg:ixeg,izsg:izeg), intent(out) :: f_out
call gradz_n2n_fd4_cuda(f_in, f_out, nz_cuda, nx_cuda, ny_cuda, deltazi)
end subroutine gradz_n2n_fd4cuda
subroutine gradz_v2n_fd4cuda(f_in, f_out)
use space_grid
real(dp), DIMENSION(iysg:iyeg,ixsg:ixeg,izsg:izeg), INTENT(in) :: f_in
real(dp), DIMENSION(iysg:iyeg,ixsg:ixeg,izsg:izeg), intent(out) :: f_out
call gradz_v2n_fd4_cuda(f_in, f_out, nz_cuda, nx_cuda, ny_cuda, deltazi)
end subroutine gradz_v2n_fd4cuda
subroutine grady_v2n_fd4cuda(f_in, f_out)
use space_grid
real(dp), DIMENSION(iysg:iyeg,ixsg:ixeg,izsg:izeg), INTENT(in) :: f_in
real(dp), DIMENSION(iysg:iyeg,ixsg:ixeg,izsg:izeg), intent(out) :: f_out
call grady_v2n_fd4_cuda(f_in, f_out, nz_cuda, nx_cuda, ny_cuda, deltayi)
end subroutine grady_v2n_fd4cuda
subroutine interp_v2n_fd4cuda(f_in, f_out)
use space_grid
real(dp), DIMENSION(iysg:iyeg,ixsg:ixeg,izsg:izeg), INTENT(in) :: f_in
real(dp), DIMENSION(iysg:iyeg,ixsg:ixeg,izsg:izeg), intent(out) :: f_out
call interp_v2n_fd4_cuda(f_in, f_out, nz_cuda, nx_cuda, ny_cuda)
end subroutine interp_v2n_fd4cuda
subroutine gradx_n2n_fd4cuda(f_in, f_out)
use space_grid
real(dp), DIMENSION(iysg:iyeg,ixsg:ixeg,izsg:izeg), INTENT(in) :: f_in
real(dp), DIMENSION(iysg:iyeg,ixsg:ixeg,izsg:izeg), intent(out) :: f_out
call gradx_n2n_fd4_cuda(f_in, f_out, nz_cuda, nx_cuda, ny_cuda, deltaxi)
end subroutine gradx_n2n_fd4cuda
subroutine gradpar_v2n_fd4cuda(f_in, f_out, gradpar_z)
use space_grid
real(dp), DIMENSION(iysg:iyeg,ixsg:ixeg,izsg:izeg), INTENT(in) :: f_in
real(dp), INTENT(in) :: gradpar_z
real(dp), DIMENSION(iysg:iyeg,ixsg:ixeg,izsg:izeg), intent(out) :: f_out
CALL gradpar_v2n_fd4_cuda(f_in, f_out, nz_cuda, nx_cuda, ny_cuda, f_z, f_y, f_n, \
f_x_n, deltazi, deltayi, deltaxi, gradpar_z, gradpar_x_n_gpu, gradpar_y_n_gpu)
end subroutine gradpar_v2n_fd4cuda
END MODULE cuda_interface
c4science · Help