1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
! { dg-do run }
module e_51_3_mod
contains
subroutine init (Q, rows, cols)
integer :: i, k, rows, cols
double precision :: Q(rows,cols)
do k = 1, cols
do i = 1, rows
Q(i,k) = 10 * i + k
end do
end do
end subroutine
subroutine check (P, Q, rows, cols)
integer :: i, k, rows, cols
double precision, parameter :: EPS = 0.00001
double precision :: P(rows,cols), Q(rows,cols), diff
do k = 1, cols
do i = 1, rows
diff = P(i,k) - Q(i,k)
if (diff > EPS .or. -diff > EPS) call abort
end do
end do
end subroutine
subroutine gramSchmidt_ref (Q, rows, cols)
integer :: i, k, rows, cols
double precision :: Q(rows,cols), tmp
do k = 1, cols
tmp = 0.0d0
do i = 1, rows
tmp = tmp + (Q(i,k) * Q(i,k))
end do
tmp = 1.0d0 / sqrt (tmp)
do i = 1, rows
Q(i,k) = Q(i,k) * tmp
end do
end do
end subroutine
subroutine gramSchmidt (Q, rows, cols)
integer :: i, k, rows, cols
double precision :: Q(rows,cols), tmp
!$omp target data map(Q)
do k = 1, cols
tmp = 0.0d0
!$omp target
!$omp parallel do reduction(+:tmp)
do i = 1, rows
tmp = tmp + (Q(i,k) * Q(i,k))
end do
!$omp end target
tmp = 1.0d0 / sqrt (tmp)
!$omp target
!$omp parallel do
do i = 1, rows
Q(i,k) = Q(i,k) * tmp
end do
!$omp end target
end do
!$omp end target data
end subroutine
end module
program e_51_3
use e_51_3_mod, only : init, check, gramSchmidt, gramSchmidt_ref
integer :: cols, rows
double precision, pointer :: P(:,:), Q(:,:)
cols = 5
rows = 5
allocate (P(rows,cols), Q(rows,cols))
call init (P, rows, cols)
call init (Q, rows, cols)
call gramSchmidt_ref (P, rows, cols)
call gramSchmidt (Q, rows, cols)
call check (P, Q, rows, cols)
deallocate (P, Q)
end program
|
