!
! Copyright (c) 2019, Advanced Micro Devices, Inc. All rights reserved.
!
! F2008 Compliance Tests: Flang-F2008-Impure elemental procedures
!
! Date of Modification: Fri Oct 18
!

! Purpose:
! Elemental procedures take scalar arguments and produce a scalar
! result. Else raise error.
IMPURE ELEMENTAL FUNCTION VADD(A, B, C)
	IMPLICIT NONE
	INTEGER, INTENT(IN) :: A(:), B(:), C(:)
	INTEGER VADD(SIZE(A)) 

	PRINT *, 'A = ', A
	PRINT *, 'B = ', B
	PRINT *, 'B = ', C

	VADD = A + B + C
END

! Only works with gfortran now. how to enable scalar ops in flang?
PROGRAM IMPURE06
	IMPLICIT NONE
	INTEGER, DIMENSION(100) :: X, Y, Z
	INTEGER :: I
	INTERFACE
		FUNCTION VADD(A, B, C)
			INTEGER, INTENT(IN) :: A(:), B(:), C(:)
			INTEGER  VADD(SIZE(A)) 
		END FUNCTION
	END INTERFACE
	INTEGER, PARAMETER :: N = 1
  LOGICAL exp(N), res(N)
  CALL check(res, exp, N)

	DO I = 1, 100
		X(I) = 2 * I
		Y(I) = 3 * I
		Z(I) = 4 * I
	END DO

	PRINT *, 'X + Y + Z = ', VADD(X, Y, Z)
END PROGRAM