OpenACC-course/lab1/FORTRAN/task4/task4_4096_solution.f90

56 lines
1.1 KiB
Fortran
Raw Normal View History

program main
use openacc
implicit real(4) (A-H,O-Z)
integer, parameter :: NN = 4096
integer, parameter :: NM = 4096
real(4) A(NN,NM), Anew(NN,NM)
iter_max = 1000
tol = 1.0e-6
error = 1.0
A(1,:) = 1.0
A(2:NN,:) = 0.0
Anew(1,:) = 1.0
Anew(2:NN,:) = 0.0
print 100,NN,NM
call cpu_time(t1)
iter = 0
!$acc data copy(A), create(Anew)
do while ( (error > tol) .and. (iter < iter_max) )
error = 0.0
!$acc kernels loop
do j = 2, NM-1
!$acc loop gang(8) vector(32)
do i = 2, NN-1
Anew(i,j) = 0.25 * ( A(i+1,j) + A(i-1,j) + &
A(i,j-1) + A(i,j+1) )
error = max( error, abs(Anew(i,j) - A(i,j)) )
end do
end do
!$acc end kernels
!$acc kernels loop
do j = 2, NM-1
!$acc loop gang(8) vector(32)
do i = 2, NN-1
A(i,j) = Anew(i,j)
end do
end do
!$acc end kernels
if(mod(iter,100) == 0) print 101,iter,error
iter = iter + 1
end do
!$acc end data
call cpu_time(t2)
print 102,t2-t1
100 format("Jacobi relaxation Calculation: ",i4," x ",i4," mesh")
101 format(2x,i4,2x,f9.6)
102 format("total: ",f9.6," s")
end program