152
|
1 #include <omp.h>
|
|
2 #include <stdint.h>
|
|
3 #include <stdlib.h>
|
|
4
|
|
5 int
|
|
6 main ()
|
|
7 {
|
|
8 omp_alloctrait_t traits[3]
|
|
9 = { { omp_atk_alignment, 64 },
|
|
10 { omp_atk_fallback, omp_atv_null_fb },
|
|
11 { omp_atk_pool_size, 4096 } };
|
|
12 omp_allocator_handle_t a
|
|
13 = omp_init_allocator (omp_default_mem_space, 3, traits);
|
|
14 if (a == omp_null_allocator)
|
|
15 abort ();
|
|
16
|
|
17 #pragma omp parallel num_threads(4)
|
|
18 {
|
|
19 int n = omp_get_thread_num ();
|
|
20 double *volatile p, *volatile q;
|
|
21 omp_set_default_allocator ((n & 1) ? a : omp_default_mem_alloc);
|
|
22 p = (double *) omp_alloc (1696, omp_null_allocator);
|
|
23 if (p == NULL)
|
|
24 abort ();
|
|
25 p[0] = 1.0;
|
|
26 p[1695 / sizeof (double *)] = 2.0;
|
|
27 #pragma omp barrier
|
|
28 omp_set_default_allocator ((n & 1) ? omp_default_mem_alloc : a);
|
|
29 q = (double *) omp_alloc (1696, omp_null_allocator);
|
|
30 if (n & 1)
|
|
31 {
|
|
32 if (q == NULL)
|
|
33 abort ();
|
|
34 q[0] = 3.0;
|
|
35 q[1695 / sizeof (double *)] = 4.0;
|
|
36 }
|
|
37 else if (q != NULL)
|
|
38 abort ();
|
|
39 #pragma omp barrier
|
|
40 omp_free (p, omp_null_allocator);
|
|
41 omp_free (q, omp_null_allocator);
|
|
42 omp_set_default_allocator (omp_default_mem_alloc);
|
|
43 }
|
|
44 omp_destroy_allocator (a);
|
|
45 return 0;
|
|
46 }
|