Homec4science
Diffusion Introduction to CUDA (master) Browse

introCUDA/exercises/gradients/test_gradients.F90
e768432c3298master

Last Change
Raw File

test_gradients.F90
View Options

subroutine initial_setup
use space_grid
use cuda_interface
use gradients
use iso_c_binding
implicit none
integer :: i,j,k
izsg=-1
ixsg=-1
iysg=-1
izeg=3
ixeg=1202
iyeg=2002
izs=1
ixs=1
iys=1
ize=1
ixe=1200
iye=2000
deltazi=0.03
deltayi=0.003
deltayi=0.005
call gbs_allocate_cuda(strmf_cpu,iysg,iyeg,ixsg,ixeg,izsg,izeg)
call gbs_allocate_cuda(strmf_gpu,iysg,iyeg,ixsg,ixeg,izsg,izeg)
call gbs_allocate_cuda(indata,iysg,iyeg,ixsg,ixeg,izsg,izeg)
call gbs_allocate_cuda(gradpar_y_n,iysg,iyeg,ixsg,ixeg)
call gbs_allocate_cuda(gradpar_x_n,iysg,iyeg,ixsg,ixeg)
call gbs_allocate_cuda(gradpar_y_n_gpu,iysg,iyeg,ixsg,ixeg)
call gbs_allocate_cuda(gradpar_x_n_gpu,iysg,iyeg,ixsg,ixeg)
call gbs_allocate_cuda(f_z,iysg,iyeg,ixsg,ixeg,izsg,izeg)
call gbs_allocate_cuda(f_y,iysg,iyeg,ixsg,ixeg,izsg,izeg)
call gbs_allocate_cuda(f_x,iysg,iyeg,ixsg,ixeg,izsg,izeg)
call gbs_allocate_cuda(f_n,iysg,iyeg,ixsg,ixeg,izsg,izeg)
call gbs_allocate_cuda(f_x_n,iysg,iyeg,ixsg,ixeg,izsg,izeg)
end subroutine initial_setup
subroutine fill_arrays
use space_grid
use cuda_interface
use gradients
use iso_c_binding
implicit none
integer :: i,j,k
call initial_setup
call intialize_coefficients
do k=izsg,izeg
do i=ixsg,ixeg
do j=iysg,iyeg
indata(j, i, k) = dble(i*123.0+j*5.1+k*45.12)
end do
end do
end do
nx_cuda = ixeg-ixsg+1
ny_cuda = iyeg-iysg+1
nz_cuda = izeg-izsg+1
do i=ixsg,ixeg
do j=iysg,iyeg
gradpar_x_n(j, i) = dble(i*123.0+j*5.1)
gradpar_y_n(j, i) = dble(i*12.0+j*15.1)
end do
end do
gradpar_x_n_gpu(:,:) = gradpar_x_n(:,:)
gradpar_y_n_gpu(:,:) = gradpar_y_n(:,:)
gradpar_z = 0.5
end subroutine fill_arrays
program main
use space_grid
use cuda_interface
use gradients
use iso_c_binding
implicit none
call initial_setup
call intialize_coefficients
call fill_arrays
call gradz_n2n_fd4(indata(:,:,:),strmf_cpu(:,:,:))
call gradz_n2n_fd4cuda(indata(:,:,:),strmf_gpu(:,:,:))
write(*,*) 'gradz_n2n_fd4 L2_norm(strmf_cpu-strmf_gpu):', norm2(strmf_cpu(iys:iye,ixs:ixe,izs:ize) - strmf_gpu(iys:iye,ixs:ixe,izs:ize))
call gradz_v2n_fd4(indata(:,:,:),strmf_cpu(:,:,:))
call gradz_v2n_fd4cuda(indata(:,:,:),strmf_gpu(:,:,:))
write(*,*) 'gradz_v2n_fd4 L2_norm(strmf_cpu-strmf_gpu):', norm2(strmf_cpu(iys:iye,ixs:ixe,izs:ize) - strmf_gpu(iys:iye,ixs:ixe,izs:ize))
call grady_v2n_fd4(indata(:,:,:),strmf_cpu(:,:,:))
call grady_v2n_fd4cuda(indata(:,:,:),strmf_gpu(:,:,:))
write(*,*) 'grady_v2n_fd4 L2_norm(strmf_cpu-strmf_gpu):', norm2(strmf_cpu(iys:iye,ixs:ixe,izs:ize) - strmf_gpu(iys:iye,ixs:ixe,izs:ize))
call interp_v2n_fd4(indata(:,:,:),strmf_cpu(:,:,:))
call interp_v2n_fd4cuda(indata(:,:,:),strmf_gpu(:,:,:))
write(*,*) 'interp_v2n_fd4 L2_norm(strmf_cpu-strmf_gpu):', norm2(strmf_cpu(iys:iye,ixs:ixe,izs:ize) - strmf_gpu(iys:iye,ixs:ixe,izs:ize))
call gradx_n2n_fd4(indata(:,:,:),strmf_cpu(:,:,:))
call gradx_n2n_fd4cuda(indata(:,:,:),strmf_gpu(:,:,:))
write(*,*) 'gradx_n2n_fd4 L2_norm(strmf_cpu-strmf_gpu):', norm2(strmf_cpu(iys:iye,ixs:ixe,izs:ize) - strmf_gpu(iys:iye,ixs:ixe,izs:ize))
CALL gradpar_v2n_fd4(indata(:,:,:),strmf_cpu(:,:,:))
CALL gradpar_v2n_fd4cuda(indata(:,:,:),strmf_gpu(:,:,:), gradpar_z)
write(*,*) 'gradpar_v2n_fd4 L2_norm(strmf_cpu-strmf_gpu):', norm2(strmf_cpu(iys:iye,ixs:ixe,izs:ize) - strmf_gpu(iys:iye,ixs:ixe,izs:ize))
end program main
c4science · Help