Homec4science
Diffusion miniapp_gbs_gradients (master) Browse

R11986/gradients_cuda_mod.F90
00e807a5cf27master

Last Change
Raw File

gradients_cuda_mod.F90
View Options

module gradients_cuda
use space_grid
use gradients
use prec_const
use iso_c_binding
real(dp), pointer :: gradpar_y_n_cuda(:,:),gradpar_x_n_cuda(:,:)
contains
subroutine gradx_n2n_cuda(f_in, f_out)
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, ixeg-ixsg+1, iyeg-iysg+1, &
izeg-izsg+1, deltaxi)
end subroutine gradx_n2n_cuda
subroutine gradz_n2n_cuda(f_in, f_out)
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, ixeg-ixsg+1, iyeg-iysg+1, &
izeg-izsg+1, deltazi)
end subroutine gradz_n2n_cuda
subroutine gradz_v2n_cuda(f_in, f_out)
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, ixeg-ixsg+1, iyeg-iysg+1, &
izeg-izsg+1, deltazi)
end subroutine gradz_v2n_cuda
subroutine grady_v2n_cuda(f_in, f_out)
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, ixeg-ixsg+1, iyeg-iysg+1, &
izeg-izsg+1, deltayi)
end subroutine grady_v2n_cuda
subroutine interp_v2n_fd4cuda(f_in, f_out)
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, ixeg-ixsg+1, iyeg-iysg+1, &
izeg-izsg+1)
end subroutine interp_v2n_fd4cuda
subroutine gradpar_v2n_cuda(f_in, f_out)
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
real(dp), dimension(:,:,:), pointer :: f_1,f_2,f_3,f_4
call gbs_allocate_cuda(f_1,iysg,iyeg,ixsg,ixeg,izsg,izeg)
call gbs_allocate_cuda(f_2,iysg,iyeg,ixsg,ixeg,izsg,izeg)
call gbs_allocate_cuda(f_3,iysg,iyeg,ixsg,ixeg,izsg,izeg)
call gbs_allocate_cuda(f_4,iysg,iyeg,ixsg,ixeg,izsg,izeg)
call gradz_v2n_cuda(f_in, f_1) !call gradz_v2n_fd4(f, f_z)
call grady_v2n_cuda(f_in, f_2) !call grady_v2n_fd4(f, f_y)
call interp_v2n_fd4cuda(f_in, f_3) !call interp_v2n_fd4(f, flu, frd, f_n )
call gradx_n2n_cuda(f_3, f_4) !call gradx_n2n_fd4(f_n, f_x_n)
call gradpar_fd4_cuda_(f_1, f_2, gradpar_y_n_cuda, gradpar_x_n_cuda, f_4,&
gradpar_z, f_out, ixeg-ixsg+1, iyeg-iysg+1, izeg-izsg+1)
end subroutine gradpar_v2n_cuda
end module gradients_cuda
c4science · Help