1 | // RUN: %libomp-compile-and-run |
2 | // UNSUPPORTED: gcc-4, gcc-5, gcc-6, gcc-7, gcc-8 |
3 | // UNSUPPORTED: icc, clang |
4 | |
5 | #include <stdio.h> |
6 | #include <stdlib.h> |
7 | #include <omp.h> |
8 | |
9 | #define NUM_TEAMS 2 |
10 | #define NUM_THREADS_PER_TEAM 3 |
11 | |
12 | int main(int argc, char** argv) { |
13 | #pragma omp teams num_teams(NUM_TEAMS) |
14 | { |
15 | int i; |
16 | int members[NUM_THREADS_PER_TEAM]; |
17 | // Only an upper bound is guaranteed for number of teams |
18 | int nteams = omp_get_num_teams(); |
19 | if (nteams > NUM_TEAMS) { |
20 | fprintf(stderr, format: "error: too many teams: %d\n" , nteams); |
21 | exit(status: 1); |
22 | } |
23 | for (i = 0; i < NUM_THREADS_PER_TEAM; ++i) |
24 | members[i] = -1; |
25 | #pragma omp parallel num_threads(NUM_THREADS_PER_TEAM) private(i) |
26 | { |
27 | int tid = omp_get_thread_num(); |
28 | int team_id = omp_get_team_num(); |
29 | int nthreads = omp_get_num_threads(); |
30 | if (nthreads != NUM_THREADS_PER_TEAM) { |
31 | fprintf(stderr, format: "error: detected number of threads (%d) is not %d\n" , |
32 | nthreads, NUM_THREADS_PER_TEAM); |
33 | exit(status: 1); |
34 | } |
35 | if (tid < 0 || tid >= nthreads) { |
36 | fprintf(stderr, format: "error: thread id is out of range: %d\n" , tid); |
37 | exit(status: 1); |
38 | } |
39 | if (team_id < 0 || team_id > omp_get_num_teams()) { |
40 | fprintf(stderr, format: "error: team id is out of range: %d\n" , team_id); |
41 | exit(status: 1); |
42 | } |
43 | members[omp_get_thread_num()] = 1; |
44 | #pragma omp barrier |
45 | #pragma omp single |
46 | { |
47 | for (i = 0; i < NUM_THREADS_PER_TEAM; ++i) { |
48 | if (members[i] != 1) { |
49 | fprintf(stderr, format: "error: worker %d not flagged\n" , i); |
50 | exit(status: 1); |
51 | } |
52 | } |
53 | } |
54 | } |
55 | } |
56 | return 0; |
57 | } |
58 | |