1 #include <omp.h>
2 #include <stdint.h>
3 #include <stdlib.h>
4
5 const omp_alloctrait_t traits2[]
6 = { { omp_atk_alignment, 16 },
7 { omp_atk_sync_hint, omp_atv_default },
8 { omp_atk_access, omp_atv_default },
9 { omp_atk_pool_size, 1024 },
10 { omp_atk_fallback, omp_atv_default_mem_fb },
11 { omp_atk_partition, omp_atv_environment } };
12 omp_alloctrait_t traits3[]
13 = { { omp_atk_sync_hint, omp_atv_uncontended },
14 { omp_atk_alignment, 32 },
15 { omp_atk_access, omp_atv_all },
16 { omp_atk_pool_size, 512 },
17 { omp_atk_fallback, omp_atv_allocator_fb },
18 { omp_atk_fb_data, 0 },
19 { omp_atk_partition, omp_atv_default } };
20 const omp_alloctrait_t traits4[]
21 = { { omp_atk_alignment, 128 },
22 { omp_atk_pool_size, 1024 },
23 { omp_atk_fallback, omp_atv_null_fb } };
24
25 int
26 main ()
27 {
28 int *volatile p = (int *) omp_aligned_alloc (sizeof (int), 3 * sizeof (int), omp_default_mem_alloc);
29 int *volatile q;
30 int *volatile r;
31 omp_alloctrait_t traits[3]
32 = { { omp_atk_alignment, 64 },
33 { omp_atk_fallback, omp_atv_null_fb },
34 { omp_atk_pool_size, 4096 } };
35 omp_allocator_handle_t a, a2;
36
37 if ((((uintptr_t) p) % __alignof (int)) != 0)
38 abort ();
39 p[0] = 1;
40 p[1] = 2;
41 p[2] = 3;
42 omp_free (p, omp_default_mem_alloc);
43 p = (int *) omp_aligned_alloc (2 * sizeof (int), 2 * sizeof (int), omp_default_mem_alloc);
44 if ((((uintptr_t) p) % (2 * sizeof (int))) != 0)
45 abort ();
46 p[0] = 1;
47 p[1] = 2;
48 omp_free (p, omp_null_allocator);
49 omp_set_default_allocator (omp_default_mem_alloc);
50 p = (int *) omp_aligned_alloc (1, sizeof (int), omp_null_allocator);
51 if ((((uintptr_t) p) % __alignof (int)) != 0)
52 abort ();
53 p[0] = 3;
54 omp_free (p, omp_get_default_allocator ());
55
56 a = omp_init_allocator (omp_default_mem_space, 3, traits);
57 if (a == omp_null_allocator)
58 abort ();
59 p = (int *) omp_aligned_alloc (32, 3072, a);
60 if ((((uintptr_t) p) % 64) != 0)
61 abort ();
62 p[0] = 1;
63 p[3071 / sizeof (int)] = 2;
64 if (omp_aligned_alloc (8, 3072, a) != NULL)
65 abort ();
66 omp_free (p, a);
67 p = (int *) omp_aligned_alloc (128, 3072, a);
68 if ((((uintptr_t) p) % 128) != 0)
69 abort ();
70 p[0] = 3;
71 p[3071 / sizeof (int)] = 4;
72 omp_free (p, omp_null_allocator);
73 omp_set_default_allocator (a);
74 if (omp_get_default_allocator () != a)
75 abort ();
76 p = (int *) omp_aligned_alloc (64, 3072, omp_null_allocator);
77 if (omp_aligned_alloc (8, 3072, omp_null_allocator) != NULL)
78 abort ();
79 omp_free (p, a);
80 omp_destroy_allocator (a);
81
82 a = omp_init_allocator (omp_default_mem_space,
83 sizeof (traits2) / sizeof (traits2[0]),
84 traits2);
85 if (a == omp_null_allocator)
86 abort ();
87 if (traits3[5].key != omp_atk_fb_data)
88 abort ();
89 traits3[5].value = (uintptr_t) a;
90 a2 = omp_init_allocator (omp_default_mem_space,
91 sizeof (traits3) / sizeof (traits3[0]),
92 traits3);
93 if (a2 == omp_null_allocator)
94 abort ();
95 p = (int *) omp_aligned_alloc (4, 420, a2);
96 if ((((uintptr_t) p) % 32) != 0)
97 abort ();
98 p[0] = 5;
99 p[419 / sizeof (int)] = 6;
100 q = (int *) omp_aligned_alloc (8, 768, a2);
101 if ((((uintptr_t) q) % 16) != 0)
102 abort ();
103 q[0] = 7;
104 q[767 / sizeof (int)] = 8;
105 r = (int *) omp_aligned_alloc (8, 512, a2);
106 if ((((uintptr_t) r) % 8) != 0)
107 abort ();
108 r[0] = 9;
109 r[511 / sizeof (int)] = 10;
110 omp_free (p, omp_null_allocator);
111 omp_free (q, a2);
112 omp_free (r, omp_null_allocator);
113 omp_destroy_allocator (a2);
114 omp_destroy_allocator (a);
115
116 a = omp_init_allocator (omp_default_mem_space,
117 sizeof (traits4) / sizeof (traits4[0]),
118 traits4);
119 if (a == omp_null_allocator)
120 abort ();
121 if (traits3[5].key != omp_atk_fb_data)
122 abort ();
123 traits3[5].value = (uintptr_t) a;
124 a2 = omp_init_allocator (omp_default_mem_space,
125 sizeof (traits3) / sizeof (traits3[0]),
126 traits3);
127 if (a2 == omp_null_allocator)
128 abort ();
129 omp_set_default_allocator (a2);
130 #ifdef __cplusplus
131 p = static_cast <int *> (omp_aligned_alloc (4, 420));
132 #else
133 p = (int *) omp_aligned_alloc (4, 420, omp_null_allocator);
134 #endif
135 if ((((uintptr_t) p) % 32) != 0)
136 abort ();
137 p[0] = 5;
138 p[419 / sizeof (int)] = 6;
139 q = (int *) omp_aligned_alloc (64, 768, omp_null_allocator);
140 if ((((uintptr_t) q) % 128) != 0)
141 abort ();
142 q[0] = 7;
143 q[767 / sizeof (int)] = 8;
144 if (omp_aligned_alloc (8, 768, omp_null_allocator) != NULL)
145 abort ();
146 #ifdef __cplusplus
147 omp_free (p);
148 omp_free (q);
149 omp_free (NULL);
150 #else
151 omp_free (p, omp_null_allocator);
152 omp_free (q, omp_null_allocator);
153 omp_free (NULL, omp_null_allocator);
154 #endif
155 omp_free (NULL, omp_null_allocator);
156 omp_destroy_allocator (a2);
157 omp_destroy_allocator (a);
158 return 0;
159 }