! RUN: %python %S/../test_errors.py %s %flang -fopenacc

! Check OpenACC clause validity for the following construct and directive:
!   2.9 Loop

program openacc_loop_validity

  implicit none

  type atype
    real(8), dimension(10) :: arr
    real(8) :: s
    integer :: n
  end type atype

  integer :: i, j, k, l, m, b, gang_size, vector_size, worker_size
  integer, parameter :: N = 256
  integer, dimension(N) :: c
  logical, dimension(N) :: d, e
  integer :: async1
  integer :: wait1, wait2
  real :: reduction_r
  logical :: reduction_l
  real(8), dimension(N, N) :: aa, bb, cc
  logical :: ifCondition = .TRUE.
  type(atype) :: t
  type(atype), dimension(10) :: ta

  real(8), dimension(N) :: a, f, g, h

  !$acc parallel
  !$acc loop tile(2)
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel device_type(*) num_gangs(2)
  !$acc loop
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !$acc loop seq
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !$acc loop independent
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !$acc loop auto
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !ERROR: At most one VECTOR clause can appear on the LOOP directive or in group separated by the DEVICE_TYPE clause
  !$acc loop vector vector(128)
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !$acc loop vector
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !$acc loop vector(10)
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !$acc loop vector(vector_size)
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !$acc loop vector(length: vector_size)
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !ERROR: At most one WORKER clause can appear on the LOOP directive or in group separated by the DEVICE_TYPE clause
  !$acc loop worker worker(10)
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !$acc loop worker
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !$acc loop worker(10)
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !$acc loop worker(worker_size)
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !$acc loop worker(num: worker_size)
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !ERROR: At most one GANG clause can appear on the LOOP directive or in group separated by the DEVICE_TYPE clause
  !$acc loop gang gang(gang_size)
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc loop gang device_type(default) gang(gang_size)
  do i = 1, N
    a(i) = 3.14d0
  end do

  !ERROR: At most one GANG clause can appear on the PARALLEL LOOP directive or in group separated by the DEVICE_TYPE clause
  !$acc parallel loop gang gang(gang_size)
  do i = 1, N
    a(i) = 3.14d0
  end do

  !$acc parallel loop gang device_type(default) gang(gang_size)
  do i = 1, N
    a(i) = 3.14d0
  end do

  !$acc parallel
  !$acc loop gang(gang_size)
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !$acc loop gang(num: gang_size)
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !$acc loop gang(gang_size, static:*)
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !$acc loop gang(num: gang_size, static:*)
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !$acc loop gang(num: gang_size, static: gang_size)
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !$acc loop private(b, a(:))
  do i = 1, N
    a(i) = b
  end do
  !$acc end parallel

  !$acc parallel
  !$acc loop tile(*)
  do i = 1, N
    a(i) = b
  end do
  !$acc end parallel

  !$acc parallel
  !$acc loop tile(2, 2)
  do i = 1, N
    do j = 1, N
      a(i) = b
    end do
  end do
  !$acc end parallel

  !$acc parallel
  !ERROR: The parameter of the COLLAPSE clause must be a constant positive integer expression
  !$acc loop collapse(-1)
  do i = 1, N
    do j = 1, N
      a(i) = 3.14d0 + j
    end do
  end do
  !$acc end parallel

  !$acc parallel
  !ERROR: Clause PRIVATE is not allowed after clause DEVICE_TYPE on the LOOP directive
  !$acc loop device_type(*) private(i)
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !ERROR: Clause GANG is not allowed if clause SEQ appears on the LOOP directive
  !$acc loop gang seq
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !ERROR: Clause WORKER is not allowed if clause SEQ appears on the LOOP directive
  !$acc loop worker seq
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !ERROR: Clause VECTOR is not allowed if clause SEQ appears on the LOOP directive
  !$acc loop vector seq
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc parallel
  !$acc loop gang
  do i = 1, n
    !$acc loop worker
    do j = 1, n
      !ERROR: GANG clause is not allowed in the region of a loop with the WORKER clause
      !ERROR: GANG clause is not allowed in the region of a loop with the GANG clause
      !$acc loop gang vector
      do k = 1, i
      end do
    end do
  end do
  !$acc end parallel

  !$acc parallel loop vector
  do  i = 1, n
    !ERROR: GANG clause is not allowed in the region of a loop with the VECTOR clause
    !$acc loop gang
    do j = 1, n
      !ERROR: WORKER clause is not allowed in the region of a loop with the VECTOR clause
      !$acc loop worker
      do k = 1, i
        !ERROR: VECTOR clause is not allowed in the region of a loop with the VECTOR clause
        !$acc loop vector
        do l = 1, 1
        end do
      end do
    end do
  end do
  !$acc end parallel loop

  !$acc kernels
  do  i = 1, n
    !$acc loop gang worker
    do j = 1, n
      !ERROR: WORKER clause is not allowed in the region of a loop with the WORKER clause
      !$acc loop worker vector
      do k = 1, i
      end do
    end do
  end do
  !$acc end kernels

  !$acc parallel
  !$acc loop gang(dim:1)
  do i = 1, n
    !ERROR: GANG(dim:1) clause is not allowed in the region of a loop with the GANG(dim:1) clause
    !$acc loop gang(dim:1)
    do j = 1, n
      !ERROR: GANG(dim:2) clause is not allowed in the region of a loop with the GANG(dim:1) clause
      !$acc loop gang(dim:2)
      do k = 1, i
        !ERROR: GANG(dim:3) clause is not allowed in the region of a loop with the GANG(dim:2) clause
        !ERROR: GANG(dim:3) clause is not allowed in the region of a loop with the GANG(dim:1) clause
        !$acc loop gang(dim:3)
        do l = 1, 1
          !ERROR: GANG(dim:3) clause is not allowed in the region of a loop with the GANG(dim:3) clause
          !ERROR: GANG(dim:3) clause is not allowed in the region of a loop with the GANG(dim:2) clause
          !ERROR: GANG(dim:3) clause is not allowed in the region of a loop with the GANG(dim:1) clause
          !$acc loop gang(dim:3)
          do m = 1, 1
          end do
        end do
      end do
    end do
  end do
  !$acc end parallel

  !$acc parallel loop gang(dim:3)
  do i = 1, n
    !$acc loop gang(dim:2)
    do j = 1, n
      !$acc loop gang(dim:1) worker vector
      do k = 1, i
      end do
    end do
  end do
  !$acc end parallel loop

  !$acc kernels loop gang(dim:3)
  do i = 1, n
    !ERROR: Nested GANG loops are not allowed in the region of a KERNELS construct
    !$acc loop gang(dim:2)
    do j = 1, n
      !ERROR: Nested GANG loops are not allowed in the region of a KERNELS construct
      !$acc loop gang(dim:1) worker vector
      do k = 1, i
      end do
    end do
  end do
  !$acc end kernels loop

  !ERROR: Clause IF is not allowed after clause DEVICE_TYPE on the PARALLEL directive
  !$acc parallel device_type(*) if(.TRUE.)
  !$acc loop
  do i = 1, N
    a(i) = 3.14d0
  end do
  !$acc end parallel

  !$acc loop collapse(2)
  do i = 1, N
    !ERROR: Loop control is not present in the DO LOOP
    do
      a(i) = 3.14d0
    end do
  end do

  !ERROR: The num argument is not allowed when dim is specified
  !$acc loop gang(1, dim: 2)
  do i = 1, N
  end do

  !$acc loop
  do i = 1, N
  end do
  !$acc end loop

  !$acc loop collapse(2)
  do i = 1, 10
    !ERROR: LOOP directive not expected in COLLAPSE loop nest
    !$acc loop
    do j = 1, 10
    end do
  end do

  !$acc parallel
  !$acc loop
  do i = 1, n
    if(i == 10) cycle
  end do
  !$acc end parallel

  !$acc loop gang device_type(nvidia) gang(num: 8)
  DO i = 1, n
  END DO

  !$acc loop vector device_type(default) vector(16)
  DO i = 1, n
  END DO

  !$acc loop worker device_type(*) worker(8)
  DO i = 1, n
  END DO

  !$acc loop device_type(multicore) collapse(2)
  DO i = 1, n
    DO j = 1, n
    END DO
  END DO

  !ERROR: Trip count must be computable and invariant
  !$acc loop collapse(2)
  DO i = 1, n
    DO j = 1, c(i)
    END DO
  END DO

  !ERROR: Trip count must be computable and invariant
  !$acc loop collapse(2)
  DO i = 1, n
    DO j = 1, i
    END DO
  END DO

  !ERROR: Trip count must be computable and invariant
  !$acc loop collapse(2)
  DO i = 1, n
    DO j = 1, ta(i)%n
    END DO
  END DO

  !ERROR: Trip count must be computable and invariant
  !$acc parallel loop collapse(2)
  DO i = 1, n
    DO j = 1, ta(i)%n
    END DO
  END DO

  !ERROR: Trip count must be computable and invariant
  !$acc loop collapse(3)
  DO i = 1, n
    DO j = 1, n
      DO k = 1, i
      END DO
    END DO
  END DO

  ! do concurrent: each index variable counts as one collapse level.

  ! Valid: collapse(2) covers both indices of a 2-index do concurrent.
  !$acc loop collapse(2)
  DO CONCURRENT (i = 1:n, j = 1:n)
    aa(i, j) = 3.14d0
  END DO

  ! Valid: collapse(3) covers both concurrent indices then one nested do.
  !$acc loop collapse(3)
  DO CONCURRENT (i = 1:n, j = 1:n)
    DO k = 1, n
      aa(i, j) = aa(i, j) + a(k)
    END DO
  END DO

  ! Valid: collapse(2) with single-index do concurrent followed by a nested do.
  !$acc loop collapse(2)
  DO CONCURRENT (i = 1:n)
    DO j = 1, n
      aa(i, j) = 3.14d0
    END DO
  END DO

  ! Valid: combined directive, collapse(2) with do concurrent.
  !$acc parallel loop collapse(2)
  DO CONCURRENT (i = 1:n, j = 1:n)
    aa(i, j) = 3.14d0
  END DO

  ! Valid: outer regular do followed by inner do concurrent covering the
  ! remaining collapse levels.
  !$acc loop collapse(3)
  DO i = 1, n
    DO CONCURRENT (j = 1:n, k = 1:n)
      aa(i, j) = aa(i, j) + a(k)
    END DO
  END DO

  ! Valid (more concurrent indices than collapse levels): collapse(2) consumes
  ! only the first two indices of a 3-index do concurrent; the third is outside
  ! the collapsed nest.
  !$acc loop collapse(2)
  DO CONCURRENT (i = 1:n, j = 1:n, k = 1:n)
    aa(i, j) = aa(i, j) + a(k)
  END DO

  ! Valid (more loops than collapse levels): collapse(1) consumes only the
  ! first index of a 2-index do concurrent; the second index is outside the
  ! collapsed nest.
  !$acc loop collapse(1)
  DO CONCURRENT (i = 1:n, j = 1:n)
    aa(i, j) = 3.14d0
  END DO

  ! Invalid: nested do's upper bound depends on a collapsed concurrent index.
  !ERROR: Trip count must be computable and invariant
  !$acc loop collapse(3)
  DO CONCURRENT (i = 1:n, j = 1:n)
    DO k = 1, i
      aa(i, j) = aa(i, j) + a(k)
    END DO
  END DO

  ! Invalid: nested do's upper bound depends on a collapsed concurrent index.
  !ERROR: Trip count must be computable and invariant
  !$acc loop collapse(2)
  DO CONCURRENT (i = 1:n)
    DO j = 1, i
      aa(i, j) = 3.14d0
    END DO
  END DO

  ! Invalid: inner concurrent index bound depends on the outer collapsed regular
  ! do index.
  !ERROR: Trip count must be computable and invariant
  !$acc loop collapse(3)
  DO i = 1, n
    DO CONCURRENT (j = 1:n, k = 1:i)
      aa(i, j) = aa(i, j) + a(k)
    END DO
  END DO

  ! Fewer loops than collapse(n): collapse(3) but only 2 levels exist.
  ! This exercises the loop-nest depth check.
  !ERROR: Not enough perfectly nested loops for COLLAPSE(3) clause, found 2, expected 1 more
  !$acc loop collapse(3)
  DO CONCURRENT (i = 1:n, j = 1:n)
    aa(i, j) = 3.14d0
  END DO

contains

  subroutine sub1()
    !$acc routine gang(dim:2)
    implicit none
    integer, parameter :: N = 256
    integer :: i, j

    !$acc loop gang(dim:2)
    DO j = 1, N
       !$acc loop gang(dim:1) vector
       DO i = 1, N
       END DO
    END DO
  end subroutine sub1

  subroutine sub2()
    !$acc routine gang(dim:2)
    implicit none
    integer, parameter :: N = 256
    integer :: i, j

    !$acc loop gang(dim:2)
    DO j = 1, N
       !ERROR: GANG(dim:2) clause is not allowed in the region of a loop with the GANG(dim:2) clause
       !$acc loop gang(dim:2) vector
       DO i = 1, N
       END DO
    END DO
  end subroutine sub2

end program openacc_loop_validity