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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
// RUN: %libomp-compile -D_GNU_SOURCE
// RUN: env OMP_NUM_THREADS=2,2 KMP_AFFINITY=reset,granularity=thread,compact %libomp-run
// REQUIRES: linux
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <omp.h>
#include "libomp_test_affinity.h"
#define CHECK_EQUAL 0
#define CHECK_NOT_EQUAL 1
void check_primary_thread_affinity(int line, affinity_mask_t *other_aff,
int type) {
#pragma omp master
{
affinity_mask_t *primary_aff = affinity_mask_alloc();
get_thread_affinity(primary_aff);
if (type == CHECK_EQUAL && !affinity_mask_equal(primary_aff, other_aff)) {
fprintf(stderr, "error: line %d: primary affinity was not equal\n", line);
exit(EXIT_FAILURE);
} else if (type == CHECK_NOT_EQUAL &&
affinity_mask_equal(primary_aff, other_aff)) {
fprintf(stderr, "error: line %d: primary affinity was equal\n", line);
exit(EXIT_FAILURE);
}
affinity_mask_free(primary_aff);
}
}
#define CHECK_PRIMARY_THREAD_AFFINITY_EQUAL(other_aff) \
check_primary_thread_affinity(__LINE__, other_aff, CHECK_EQUAL)
#define CHECK_PRIMARY_THREAD_AFFINITY_NOT_EQUAL(other_aff) \
check_primary_thread_affinity(__LINE__, other_aff, CHECK_NOT_EQUAL)
int main() {
int i;
affinity_mask_t *initial_mask = affinity_mask_alloc();
get_thread_affinity(initial_mask);
for (i = 0; i < 10; ++i) {
#pragma omp parallel
{
CHECK_PRIMARY_THREAD_AFFINITY_NOT_EQUAL(initial_mask);
}
CHECK_PRIMARY_THREAD_AFFINITY_EQUAL(initial_mask);
}
omp_set_max_active_levels(2);
for (i = 0; i < 10; ++i) {
#pragma omp parallel
{
CHECK_PRIMARY_THREAD_AFFINITY_NOT_EQUAL(initial_mask);
#pragma omp parallel
CHECK_PRIMARY_THREAD_AFFINITY_NOT_EQUAL(initial_mask);
CHECK_PRIMARY_THREAD_AFFINITY_NOT_EQUAL(initial_mask);
}
CHECK_PRIMARY_THREAD_AFFINITY_EQUAL(initial_mask);
}
affinity_mask_free(initial_mask);
return EXIT_SUCCESS;
}
|
