56 lines
1.1 KiB
Fortran
56 lines
1.1 KiB
Fortran
|
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
|