1// RUN: %libomp-compile-and-run
2// RUN: %libomp-run | %python %S/check.py -c 'CHECK' %s
3
4#include <stdio.h>
5#include <stdlib.h>
6#include <string.h>
7#include <omp.h>
8
9#define XSTR(x) #x
10#define STR(x) XSTR(x)
11
12#define streqls(s1, s2) (!strcmp(s1, s2))
13
14#define check(condition) \
15 if (!(condition)) { \
16 fprintf(stderr, "error: %s: %d: " STR(condition) "\n", __FILE__, \
17 __LINE__); \
18 exit(1); \
19 }
20
21#define BUFFER_SIZE 1024
22
23int main(int argc, char** argv) {
24 char buf[BUFFER_SIZE];
25 size_t needed;
26
27 omp_set_affinity_format("0123456789");
28
29 needed = omp_get_affinity_format(buf, BUFFER_SIZE);
30 check(streqls(buf, "0123456789"));
31 check(needed == 10)
32
33 // Check that it is truncated properly
34 omp_get_affinity_format(buf, 5);
35 check(streqls(buf, "0123"));
36
37 #pragma omp parallel
38 {
39 char my_buf[512];
40 size_t needed = omp_capture_affinity(my_buf, 512, NULL);
41 check(streqls(my_buf, "0123456789"));
42 check(needed == 10);
43 // Check that it is truncated properly
44 omp_capture_affinity(my_buf, 5, NULL);
45 check(streqls(my_buf, "0123"));
46 }
47
48 #pragma omp parallel num_threads(4)
49 {
50 omp_display_affinity(NULL);
51 }
52
53 return 0;
54}
55
56// CHECK: num_threads=4 0123456789
57

source code of openmp/runtime/test/affinity/format/api.c