summaryrefslogtreecommitdiff
path: root/libgomp/testsuite/libgomp.fortran/examples-4/e.55.1.f90
blob: 0dd00b4ba8c69af1d4d802cedfcb808aad4363e4 (plain)
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
! { dg-do run }

module e_55_1_mod
  integer, parameter :: N = 10000000, CHUNKSZ = 100000
  real :: Y(N), Z(N)
end module

subroutine init ()
  use e_55_1_mod, only : Y, Z, N
  integer :: i
  do i = 1, N
    Y(i) = 0.1 * i
    Z(i) = Y(i)
  end do
end subroutine

subroutine check ()
  use e_55_1_mod, only : Y, Z, N
  real :: err
  real, parameter :: EPS = 0.00001
  integer :: i
  do i = 1, N
    if (Y(i) == 0.0) then
      err = Z(i)
    else if (Z(i) == 0.0) then
      err = Y(i)
    else
      err = (Y(i) - Z(i)) / Z(i)
    end if
    if (err > EPS .or. err < -EPS) call abort
  end do
end subroutine

real function F (z)
  !$omp declare target
  real, intent(in) :: z
  F = -z
end function

subroutine pipedF ()
  use e_55_1_mod, only: Z, N, CHUNKSZ
  integer :: C, i
  real :: F
  do C = 1, N, CHUNKSZ
    !$omp task
      !$omp target map(Z(C:C+CHUNKSZ-1))
        !$omp parallel do
        do i = C, C+CHUNKSZ-1
          Z(i) = F (Z(i))
        end do
      !$omp end target
    !$omp end task
  end do
end subroutine

subroutine pipedF_ref ()
  use e_55_1_mod, only: Y, N
  integer :: i
  real :: F
  do i = 1, N
    Y(i) = F (Y(i))
  end do
end subroutine

program e_55_1
  call init ()
  call pipedF ()
  call pipedF_ref ()
  call check ()
end program