1 // RUN: %libomp-compile -D_GNU_SOURCE
2 // RUN: env OMP_NUM_THREADS=2,2 KMP_AFFINITY=reset,granularity=thread,compact %libomp-run
3 // REQUIRES: linux
4
5 #include <stdio.h>
6 #include <stdlib.h>
7 #include <string.h>
8 #include <omp.h>
9 #include "libomp_test_affinity.h"
10
11 #define CHECK_EQUAL 0
12 #define CHECK_NOT_EQUAL 1
13
check_primary_thread_affinity(int line,affinity_mask_t * other_aff,int type)14 void check_primary_thread_affinity(int line, affinity_mask_t *other_aff,
15 int type) {
16 #pragma omp master
17 {
18 affinity_mask_t *primary_aff = affinity_mask_alloc();
19 get_thread_affinity(primary_aff);
20 if (type == CHECK_EQUAL && !affinity_mask_equal(primary_aff, other_aff)) {
21 fprintf(stderr, "error: line %d: primary affinity was not equal\n", line);
22 exit(EXIT_FAILURE);
23 } else if (type == CHECK_NOT_EQUAL &&
24 affinity_mask_equal(primary_aff, other_aff)) {
25 fprintf(stderr, "error: line %d: primary affinity was equal\n", line);
26 exit(EXIT_FAILURE);
27 }
28 affinity_mask_free(primary_aff);
29 }
30 }
31
32 #define CHECK_PRIMARY_THREAD_AFFINITY_EQUAL(other_aff) \
33 check_primary_thread_affinity(__LINE__, other_aff, CHECK_EQUAL)
34 #define CHECK_PRIMARY_THREAD_AFFINITY_NOT_EQUAL(other_aff) \
35 check_primary_thread_affinity(__LINE__, other_aff, CHECK_NOT_EQUAL)
36
main()37 int main() {
38 int i;
39 affinity_mask_t *initial_mask = affinity_mask_alloc();
40 get_thread_affinity(initial_mask);
41
42 for (i = 0; i < 10; ++i) {
43 #pragma omp parallel
44 {
45 CHECK_PRIMARY_THREAD_AFFINITY_NOT_EQUAL(initial_mask);
46 }
47 CHECK_PRIMARY_THREAD_AFFINITY_EQUAL(initial_mask);
48 }
49
50 omp_set_max_active_levels(2);
51 for (i = 0; i < 10; ++i) {
52 #pragma omp parallel
53 {
54 CHECK_PRIMARY_THREAD_AFFINITY_NOT_EQUAL(initial_mask);
55
56 #pragma omp parallel
57 CHECK_PRIMARY_THREAD_AFFINITY_NOT_EQUAL(initial_mask);
58
59 CHECK_PRIMARY_THREAD_AFFINITY_NOT_EQUAL(initial_mask);
60 }
61 CHECK_PRIMARY_THREAD_AFFINITY_EQUAL(initial_mask);
62 }
63
64 affinity_mask_free(initial_mask);
65 return EXIT_SUCCESS;
66 }
67