115 lines
2.9 KiB
Fortran
115 lines
2.9 KiB
Fortran
|
! RUN: %flang_fc1 -emit-hlfir -fopenmp %s -o - | FileCheck %s
|
||
|
|
|
||
|
|
! CHECK-LABEL: func @_QPteams_simple
|
||
|
|
subroutine teams_simple()
|
||
|
|
! CHECK: omp.teams
|
||
|
|
!$omp teams
|
||
|
|
! CHECK: fir.call
|
||
|
|
call f1()
|
||
|
|
! CHECK: omp.terminator
|
||
|
|
!$omp end teams
|
||
|
|
end subroutine teams_simple
|
||
|
|
|
||
|
|
!===============================================================================
|
||
|
|
! `num_teams` clause
|
||
|
|
!===============================================================================
|
||
|
|
|
||
|
|
! CHECK-LABEL: func @_QPteams_numteams
|
||
|
|
subroutine teams_numteams(num_teams)
|
||
|
|
integer, intent(inout) :: num_teams
|
||
|
|
|
||
|
|
! CHECK: omp.teams
|
||
|
|
! CHECK-SAME: num_teams( to %{{.*}}: i32)
|
||
|
|
!$omp teams num_teams(4)
|
||
|
|
! CHECK: fir.call
|
||
|
|
call f1()
|
||
|
|
! CHECK: omp.terminator
|
||
|
|
!$omp end teams
|
||
|
|
|
||
|
|
! CHECK: omp.teams
|
||
|
|
! CHECK-SAME: num_teams( to %{{.*}}: i32)
|
||
|
|
!$omp teams num_teams(num_teams)
|
||
|
|
! CHECK: fir.call
|
||
|
|
call f2()
|
||
|
|
! CHECK: omp.terminator
|
||
|
|
!$omp end teams
|
||
|
|
|
||
|
|
end subroutine teams_numteams
|
||
|
|
|
||
|
|
!===============================================================================
|
||
|
|
! `if` clause
|
||
|
|
!===============================================================================
|
||
|
|
|
||
|
|
! CHECK-LABEL: func @_QPteams_if
|
||
|
|
subroutine teams_if(alpha)
|
||
|
|
integer, intent(in) :: alpha
|
||
|
|
logical :: condition
|
||
|
|
|
||
|
|
! CHECK: omp.teams
|
||
|
|
! CHECK-SAME: if(%{{.*}})
|
||
|
|
!$omp teams if(.false.)
|
||
|
|
! CHECK: fir.call
|
||
|
|
call f1()
|
||
|
|
! CHECK: omp.terminator
|
||
|
|
!$omp end teams
|
||
|
|
|
||
|
|
! CHECK: omp.teams
|
||
|
|
! CHECK-SAME: if(%{{.*}})
|
||
|
|
!$omp teams if(alpha .le. 0)
|
||
|
|
! CHECK: fir.call
|
||
|
|
call f2()
|
||
|
|
! CHECK: omp.terminator
|
||
|
|
!$omp end teams
|
||
|
|
|
||
|
|
! CHECK: omp.teams
|
||
|
|
! CHECK-SAME: if(%{{.*}})
|
||
|
|
!$omp teams if(condition)
|
||
|
|
! CHECK: fir.call
|
||
|
|
call f3()
|
||
|
|
! CHECK: omp.terminator
|
||
|
|
!$omp end teams
|
||
|
|
end subroutine teams_if
|
||
|
|
|
||
|
|
!===============================================================================
|
||
|
|
! `thread_limit` clause
|
||
|
|
!===============================================================================
|
||
|
|
|
||
|
|
! CHECK-LABEL: func @_QPteams_threadlimit
|
||
|
|
subroutine teams_threadlimit(thread_limit)
|
||
|
|
integer, intent(inout) :: thread_limit
|
||
|
|
|
||
|
|
! CHECK: omp.teams
|
||
|
|
! CHECK-SAME: thread_limit(%{{.*}}: i32)
|
||
|
|
!$omp teams thread_limit(4)
|
||
|
|
! CHECK: fir.call
|
||
|
|
call f1()
|
||
|
|
! CHECK: omp.terminator
|
||
|
|
!$omp end teams
|
||
|
|
|
||
|
|
! CHECK: omp.teams
|
||
|
|
! CHECK-SAME: thread_limit(%{{.*}}: i32)
|
||
|
|
!$omp teams thread_limit(thread_limit)
|
||
|
|
! CHECK: fir.call
|
||
|
|
call f2()
|
||
|
|
! CHECK: omp.terminator
|
||
|
|
!$omp end teams
|
||
|
|
|
||
|
|
end subroutine teams_threadlimit
|
||
|
|
|
||
|
|
!===============================================================================
|
||
|
|
! `allocate` clause
|
||
|
|
!===============================================================================
|
||
|
|
|
||
|
|
! CHECK-LABEL: func @_QPteams_allocate
|
||
|
|
subroutine teams_allocate()
|
||
|
|
use omp_lib
|
||
|
|
integer :: x
|
||
|
|
! CHECK: omp.teams
|
||
|
|
! CHECK-SAME: allocate(%{{.+}} : i32 -> %{{.+}} : !fir.ref<i32>)
|
||
|
|
!$omp teams allocate(omp_high_bw_mem_alloc: x) private(x)
|
||
|
|
! CHECK: arith.addi
|
||
|
|
x = x + 12
|
||
|
|
! CHECK: omp.terminator
|
||
|
|
!$omp end teams
|
||
|
|
end subroutine teams_allocate
|