66 lines
2.1 KiB
Fortran
66 lines
2.1 KiB
Fortran
use omp_lib
|
|
implicit none (type, external)
|
|
if (.not. env_exists ("OMP_NUM_TEAMS") &
|
|
.and. omp_get_max_teams () /= 0) &
|
|
error stop 1
|
|
call omp_set_num_teams (7)
|
|
if (omp_get_max_teams () /= 7) &
|
|
error stop 2
|
|
if (.not. env_exists ("OMP_TEAMS_THREAD_LIMIT") &
|
|
.and. omp_get_teams_thread_limit () /= 0) &
|
|
error stop 3
|
|
call omp_set_teams_thread_limit (15)
|
|
if (omp_get_teams_thread_limit () /= 15) &
|
|
error stop 4
|
|
!$omp teams
|
|
!$omp parallel if(.false.)
|
|
if (omp_get_max_teams () /= 7 &
|
|
.or. omp_get_teams_thread_limit () /= 15 &
|
|
.or. omp_get_num_teams () < 1 &
|
|
.or. omp_get_num_teams () > 7 &
|
|
.or. omp_get_team_num () < 0 &
|
|
.or. omp_get_team_num () >= omp_get_num_teams () &
|
|
.or. omp_get_thread_limit () < 1 &
|
|
.or. omp_get_thread_limit () > 15) &
|
|
error stop 5
|
|
!$omp end parallel
|
|
!$omp end teams
|
|
!$omp teams num_teams(5) thread_limit (13)
|
|
!$omp parallel if(.false.)
|
|
if (omp_get_max_teams () /= 7 &
|
|
.or. omp_get_teams_thread_limit () /= 15 &
|
|
.or. omp_get_num_teams () /= 5 &
|
|
.or. omp_get_team_num () < 0 &
|
|
.or. omp_get_team_num () >= omp_get_num_teams () &
|
|
.or. omp_get_thread_limit () < 1 &
|
|
.or. omp_get_thread_limit () > 13) &
|
|
error stop 6
|
|
!$omp end parallel
|
|
!$omp end teams
|
|
!$omp teams num_teams(8) thread_limit (16)
|
|
!$omp parallel if(.false.)
|
|
if (omp_get_max_teams () /= 7 &
|
|
.or. omp_get_teams_thread_limit () /= 15 &
|
|
.or. omp_get_num_teams () /= 8 &
|
|
.or. omp_get_team_num () < 0 &
|
|
.or. omp_get_team_num () >= omp_get_num_teams () &
|
|
.or. omp_get_thread_limit () < 1 &
|
|
.or. omp_get_thread_limit () > 16) &
|
|
error stop 7
|
|
!$omp end parallel
|
|
!$omp end teams
|
|
contains
|
|
logical function env_exists (name)
|
|
character(len=*) :: name
|
|
character(len=40) :: val
|
|
integer :: stat
|
|
call get_environment_variable (name, val, status=stat)
|
|
if (stat == 0) then
|
|
env_exists = .true.
|
|
else if (stat == 1) then
|
|
env_exists = .false.
|
|
else
|
|
error stop 10
|
|
endif
|
|
end
|
|
end
|