1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
// RUN: %libomp-compile && env OMP_DYNAMIC=true KMP_DYNAMIC_MODE=random %libomp-run
// gcc/icc target offloading is incompatible with libomp
// UNSUPPORTED: icc, gcc
// This is a super simple unit test to see that teams behave properly when
// parallel regions inside the teams construct cannot allocate teams of
// thread_limit size.
#include <stdio.h>
#include <stdlib.h>
#include <omp.h>
#define NUM_TIMES 10
int main(int argc, char **argv) {
int num_procs = omp_get_max_threads();
int num_teams, thread_limit, i;
num_teams = 2;
thread_limit = num_procs / num_teams;
for (i = 0; i < NUM_TIMES; ++i) {
#pragma omp target teams num_teams(num_teams) thread_limit(thread_limit)
{
#pragma omp parallel num_threads(thread_limit)
{
int my_num_threads = omp_get_num_threads();
int my_num_teams = omp_get_num_teams();
int my_team_id = omp_get_team_num();
int my_thread_id = omp_get_thread_num();
if (my_num_teams < 0 || my_num_teams > num_teams) {
fprintf(stderr, "error: my_num_teams (%d) invalid\n", my_num_teams);
exit(1);
}
if (my_team_id < 0 || my_team_id >= my_num_teams) {
fprintf(stderr, "error: my_team_id (%d) invalid (nteams = %d)\n",
my_team_id, my_num_teams);
exit(1);
}
if (my_thread_id < 0 || my_thread_id >= my_num_threads) {
fprintf(stderr,
"error: my_thread_id (%d) invalid (my_num_threads = %d)\n",
my_thread_id, my_num_threads);
exit(1);
}
}
}
}
return 0;
}
|
