libgomp/testsuite/libgomp.oacc-fortran/kernels-private-vars-loop-worker-3.f90

   1 ! Test of worker-private variables declared on a loop directive, broadcasting
   2 ! to vector-partitioned mode.  Back-to-back worker loops.
   3
   4 ! { dg-do run }
   5
   6 program main
   7   integer :: x, i, j, k, idx, arr(0:32*32*32)
   8
   9   do i = 0, 32*32*32-1
  10      arr(i) = i
  11   end do
  12
  13   !$acc kernels copy(arr)
  14   !$acc loop gang(num:32)
  15   do i = 0, 31
  16      !$acc loop worker(num:8) private(x)
  17      do j = 0, 31
  18         x = ieor(i, j * 3)
  19
  20         !$acc loop vector(length:32)
  21         do k = 0, 31
  22            arr(i * 1024 + j * 32 + k) = arr(i * 1024 + j * 32 + k) + x * k
  23         end do
  24      end do
  25
  26      !$acc loop worker(num:8) private(x)
  27      do j = 0, 31
  28         x = ior(i, j * 5)
  29
  30         !$acc loop vector(length:32)
  31         do k = 0, 31
  32            arr(i * 1024 + j * 32 + k) = arr(i * 1024 + j * 32 + k) + x * k
  33         end do
  34      end do
  35   end do
  36   !$acc end kernels
  37
  38   do i = 0, 32 - 1
  39      do j = 0, 32 -1
  40         do k = 0, 32 - 1
  41            idx = i * 1024 + j * 32 + k
  42            if (arr(idx) .ne. idx + ieor(i, j * 3) * k + ior(i, j * 5) * k) then
  43               call abort
  44            end if
  45         end do
  46      end do
  47   end do
  48 end program main