cf1f871023
Implementation details: * Recognize prefetch directive in the parser as `!dir$ prefetch ...` * Unparse the prefetch directive * Add required tests Details on the prefetch directive: `!dir$ prefetch designator[, designator]...`, where the designator list can be a variable or an array reference. This directive is used to insert a hint to the code generator to prefetch instructions for memory references.
81 lines
3.2 KiB
Fortran
81 lines
3.2 KiB
Fortran
!RUN: %flang_fc1 -fdebug-unparse-no-sema %s 2>&1 | FileCheck %s -check-prefix=UNPARSE
|
|
!RUN: %flang_fc1 -fdebug-dump-parse-tree-no-sema %s 2>&1 | FileCheck %s -check-prefix=TREE
|
|
|
|
subroutine test_prefetch_01(a, b)
|
|
integer, intent(in) :: a
|
|
integer, intent(inout) :: b(5)
|
|
integer :: i = 2
|
|
integer :: res
|
|
|
|
!TREE: | | DeclarationConstruct -> SpecificationConstruct -> CompilerDirective -> Prefetch -> Designator -> DataRef -> Name = 'a'
|
|
|
|
!UNPARSE: !DIR$ PREFETCH a
|
|
!dir$ prefetch a
|
|
b(1) = a
|
|
|
|
!TREE: | | ExecutionPartConstruct -> ExecutableConstruct -> CompilerDirective -> Prefetch -> Designator -> DataRef -> Name = 'b'
|
|
|
|
!UNPARSE: !DIR$ PREFETCH b
|
|
!dir$ prefetch b
|
|
res = sum(b)
|
|
|
|
!TREE: | | ExecutionPartConstruct -> ExecutableConstruct -> CompilerDirective -> Prefetch -> Designator -> DataRef -> Name = 'a'
|
|
!TREE: | | Designator -> DataRef -> ArrayElement
|
|
!TREE: | | | DataRef -> Name = 'b'
|
|
!TREE: | | | SectionSubscript -> SubscriptTriplet
|
|
!TREE: | | | | Scalar -> Integer -> Expr -> LiteralConstant -> IntLiteralConstant = '3'
|
|
!TREE: | | | | Scalar -> Integer -> Expr -> LiteralConstant -> IntLiteralConstant = '5'
|
|
|
|
!UNPARSE: !DIR$ PREFETCH a, b(3:5)
|
|
!dir$ prefetch a, b(3:5)
|
|
res = a + b(4)
|
|
|
|
!TREE: | | ExecutionPartConstruct -> ExecutableConstruct -> CompilerDirective -> Prefetch -> Designator -> DataRef -> Name = 'res'
|
|
!TREE: | | Designator -> DataRef -> ArrayElement
|
|
!TREE: | | | DataRef -> Name = 'b'
|
|
!TREE: | | | SectionSubscript -> Integer -> Expr -> Add
|
|
!TREE: | | | | Expr -> Designator -> DataRef -> Name = 'i'
|
|
!TREE: | | | | Expr -> LiteralConstant -> IntLiteralConstant = '2'
|
|
|
|
!UNPARSE: !DIR$ PREFETCH res, b(i+2)
|
|
!dir$ prefetch res, b(i+2)
|
|
res = res + b(i+2)
|
|
end subroutine
|
|
|
|
subroutine test_prefetch_02(n, a)
|
|
integer, intent(in) :: n
|
|
integer, intent(in) :: a(n)
|
|
type :: t
|
|
real, allocatable :: x(:, :)
|
|
end type t
|
|
type(t) :: p
|
|
|
|
do i = 1, n
|
|
!TREE: | | | | ExecutionPartConstruct -> ExecutableConstruct -> CompilerDirective -> Prefetch -> Designator -> DataRef -> ArrayElement
|
|
!TREE: | | | | | DataRef -> StructureComponent
|
|
!TREE: | | | | | | DataRef -> Name = 'p'
|
|
!TREE: | | | | | | Name = 'x'
|
|
!TREE: | | | | | SectionSubscript -> Integer -> Expr -> Designator -> DataRef -> Name = 'i'
|
|
!TREE: | | | | | SectionSubscript -> SubscriptTriplet
|
|
!TREE: | | | | Designator -> DataRef -> Name = 'a'
|
|
|
|
!UNPARSE: !DIR$ PREFETCH p%x(i,:), a
|
|
!dir$ prefetch p%x(i, :), a
|
|
do j = 1, n
|
|
!TREE: | | | | | | ExecutionPartConstruct -> ExecutableConstruct -> CompilerDirective -> Prefetch -> Designator -> DataRef -> ArrayElement
|
|
!TREE: | | | | | | | DataRef -> StructureComponent
|
|
!TREE: | | | | | | | | DataRef -> Name = 'p'
|
|
!TREE: | | | | | | | | Name = 'x'
|
|
!TREE: | | | | | | | SectionSubscript -> Integer -> Expr -> Designator -> DataRef -> Name = 'i'
|
|
!TREE: | | | | | | | SectionSubscript -> Integer -> Expr -> Designator -> DataRef -> Name = 'j'
|
|
!TREE: | | | | | | Designator -> DataRef -> ArrayElement
|
|
!TREE: | | | | | | | DataRef -> Name = 'a'
|
|
!TREE: | | | | | | | SectionSubscript -> Integer -> Expr -> Designator -> DataRef -> Name = 'i'
|
|
|
|
!UNPARSE: !DIR$ PREFETCH p%x(i,j), a(i)
|
|
!dir$ prefetch p%x(i, j), a(i)
|
|
p%x(i, j) = p%x(i, j) ** a(j)
|
|
end do
|
|
end do
|
|
end subroutine
|