| ! { dg-set-target-env-var OMP_NUM_TEAMS_ALL "3" } |
| ! { dg-set-target-env-var OMP_NUM_TEAMS_DEV "4" } |
| ! { dg-set-target-env-var OMP_NUM_TEAMS "5" } |
| ! { dg-set-target-env-var OMP_NUM_TEAMS_DEV_0 "6" } |
| ! { dg-set-target-env-var OMP_NUM_TEAMS_DEV_1 "7" } |
| ! { dg-set-target-env-var OMP_NUM_TEAMS_DEV_2 "8" } |
| ! { dg-set-target-env-var OMP_TEAMS_THREAD_LIMIT_ALL "2" } |
| ! { dg-set-target-env-var OMP_TEAMS_THREAD_LIMIT_DEV "3" } |
| ! { dg-set-target-env-var OMP_TEAMS_THREAD_LIMIT "4" } |
| ! { dg-set-target-env-var OMP_TEAMS_THREAD_LIMIT_DEV_0 "5" } |
| ! { dg-set-target-env-var OMP_TEAMS_THREAD_LIMIT_DEV_1 "6" } |
| ! { dg-set-target-env-var OMP_TEAMS_THREAD_LIMIT_DEV_2 "7" } |
| |
| use omp_lib |
| implicit none (type, external) |
| integer :: num_devices, i, large_num_teams, large_threads_limit |
| logical :: err |
| |
| if (omp_get_num_devices () > 3) then |
| num_devices = 3 |
| else |
| num_devices = omp_get_num_devices () |
| end if |
| |
| do i=0,num_devices-1 |
| |
| ! Testing NUM_TEAMS. |
| if (env_is_set_dev ("OMP_NUM_TEAMS_DEV_", i, 6 + i)) then |
| err = .false. |
| !$omp target device(i) map(tofrom: err) |
| if (omp_get_max_teams () /= 6 + i) err = .true. |
| !$omp end target |
| if (err) stop 1 |
| |
| err = .false. |
| !$omp target device(i) map(tofrom: err) |
| !$omp teams |
| if (omp_get_num_teams () > 6 + i .or. omp_get_team_num () >= 6 + i) & |
| err = .true. |
| !$omp end teams |
| !$omp end target |
| if (err) stop 2 |
| |
| err = .false. |
| !$omp target device(i) map(tofrom: err) |
| call omp_set_num_teams (5 + i) |
| if (omp_get_max_teams () /= 5 + i) err = .true. |
| !$omp end target |
| if (err) stop 3 |
| |
| err = .false. |
| !$omp target device(i) map(tofrom: err) |
| if (omp_get_max_teams () /= 5 + i) err = .true. |
| !$omp end target |
| if (err) stop 4 |
| |
| err = .false. |
| !$omp target device(i) map(tofrom: err) |
| !$omp teams |
| if (omp_get_num_teams () > 5 + i .or. omp_get_team_num () >= 5 + i) & |
| err = .true. |
| !$omp end teams |
| !$omp end target |
| if (err) stop 5 |
| |
| err = .false. |
| !$omp target device(i) map(tofrom: err) |
| !$omp teams num_teams(6 + i) |
| if (omp_get_num_teams () > 6 + i .or. omp_get_team_num () >= 6 + i) & |
| err = .true. |
| !$omp end teams |
| !$omp end target |
| if (err) stop 6 |
| |
| err = .false. |
| !$omp target device(i) map(tofrom: err) |
| !$omp teams num_teams(4 + i) |
| if (omp_get_num_teams () > 4 + i .or. omp_get_team_num () >= 4 + i) & |
| err = .true. |
| !$omp end teams |
| !$omp end target |
| if (err) stop 7 |
| |
| large_num_teams = 66000 |
| err = .false. |
| !$omp target device(i) map(tofrom: err) |
| call omp_set_num_teams (large_num_teams + i) |
| if (omp_get_max_teams () /= large_num_teams + i) err = .true. |
| !$omp end target |
| if (err) stop 8 |
| |
| err = .false. |
| !$omp target device(i) map(tofrom: err) |
| if (omp_get_max_teams () /= large_num_teams + i) err = .true. |
| !$omp end target |
| if (err) stop 9 |
| |
| err = .false. |
| !$omp target device(i) map(tofrom: err) |
| !$omp teams |
| if (omp_get_num_teams () > large_num_teams + i & |
| .or. omp_get_team_num () >= large_num_teams + i) err = .true. |
| !$omp end teams |
| !$omp end target |
| if (err) stop 10 |
| end if |
| |
| ! Testing TEAMS-THREAD-LIMIT |
| if (env_is_set_dev ("OMP_TEAMS_THREAD_LIMIT_DEV_", i, 5 + i)) then |
| err = .false. |
| !$omp target device(i) map(tofrom: err) |
| if (omp_get_teams_thread_limit () /= 5 + i) err = .true. |
| !$omp end target |
| if (err) stop 11 |
| |
| err = .false. |
| !$omp target device(i) map(tofrom: err) |
| !$omp teams |
| !$omp parallel |
| if (omp_get_thread_limit () > 5 + i .or. omp_get_thread_num () >= 5 + i) & |
| err = .true. |
| !$omp end parallel |
| !$omp end teams |
| !$omp end target |
| if (err) stop 12 |
| |
| err = .false. |
| !$omp target device(i) map(tofrom: err) |
| call omp_set_teams_thread_limit (4 + i) |
| if (omp_get_teams_thread_limit () /= 4 + i) err = .true. |
| !$omp end target |
| if (err) stop 13 |
| |
| err = .false. |
| !$omp target device(i) map(tofrom: err) |
| if (omp_get_teams_thread_limit () /= 4 + i) err = .true. |
| !$omp end target |
| if (err) stop 14 |
| |
| err = .false. |
| !$omp target device(i) map(tofrom: err) |
| !$omp teams |
| !$omp parallel |
| if (omp_get_thread_limit () > 4 + i .or. omp_get_thread_num () >= 4 + i) & |
| err = .true. |
| !$omp end parallel |
| !$omp end teams |
| !$omp end target |
| if (err) stop 15 |
| |
| err = .false. |
| !$omp target device(i) map(tofrom: err) |
| !$omp teams thread_limit(5 + i) |
| !$omp parallel |
| if (omp_get_thread_limit () > 5 + i .or. omp_get_thread_num () >= 5 + i) & |
| err = .true. |
| !$omp end parallel |
| !$omp end teams |
| !$omp end target |
| if (err) stop 16 |
| |
| err = .false. |
| !$omp target device(i) map(tofrom: err) |
| !$omp teams thread_limit(3 + i) |
| !$omp parallel |
| if (omp_get_thread_limit () > 3 + i .or. omp_get_thread_num () >= 3 + i) & |
| err = .true. |
| !$omp end parallel |
| !$omp end teams |
| !$omp end target |
| if (err) stop 17 |
| |
| large_threads_limit = 67000 |
| err = .false. |
| !$omp target device(i) map(tofrom: err) |
| call omp_set_teams_thread_limit (large_threads_limit + i) |
| if (omp_get_teams_thread_limit () /= large_threads_limit + i) err = .true. |
| !$omp end target |
| if (err) stop 18 |
| |
| err = .false. |
| !$omp target device(i) map(tofrom: err) |
| if (omp_get_teams_thread_limit () /= large_threads_limit + i) err = .true. |
| !$omp end target |
| if (err) stop 19 |
| |
| err = .false. |
| !$omp target device(i) map(tofrom: err) |
| !$omp teams |
| !$omp parallel |
| if (omp_get_thread_limit () > large_threads_limit + i & |
| .or. omp_get_thread_num () >= large_threads_limit + i) err = .true. |
| !$omp end parallel |
| !$omp end teams |
| !$omp end target |
| if (err) stop 20 |
| end if |
| |
| end do |
| |
| contains |
| logical function env_is_set (name, val) |
| character(len=*) :: name, val |
| character(len=40) :: val2 |
| integer :: stat |
| call get_environment_variable (name, val2, status=stat) |
| if (stat == 0) then |
| if (val == val2) then |
| env_is_set = .true. |
| return |
| end if |
| else if (stat /= 1) then |
| error stop 30 |
| endif |
| env_is_set = .false. |
| end |
| logical function env_is_set_dev (name, dev_num, val) |
| character(len=*) :: name |
| integer :: dev_num, val |
| character(len=64) :: dev_num_str, env_var, val_str |
| dev_num_str = ADJUSTL(dev_num_str) |
| env_var = name // dev_num_str |
| val_str = ADJUSTL(val_str) |
| env_is_set_dev = env_is_set (TRIM(env_var), TRIM(val_str)) |
| end |
| end |
