1 #include <omp.h>
2 #include <stdlib.h>
3 #include <stdint.h>
4
5 struct S { int a, b; };
6
7 void
8 foo (int x, int *p, int *q, int px, omp_allocator_handle_t h, int fl)
9 {
10 int y = 0, r = 0, i, i1, l, l2[4], l3, n = 8;
11 int i2, j2, n2 = 9, l4;
12 int i3, j3, n3 = 10, l5;
13 int i4, j4, n4 = 11, l6;
14 int i5, n5;
15 int v[x], w[x];
16 int r2[4] = { 0, 0, 0, 0 };
17 int xo = x;
18 struct S s = { 27, 29 };
19 for (i = 0; i < 4; i++)
20 p[i] = 0;
21 for (i = 0; i < 3; i++)
22 q[i] = 0;
23 for (i = 0; i < x; i++)
24 w[i] = i;
25 #pragma omp parallel private (y, v) firstprivate (x) allocate (x, y, v)
26 {
27 int *volatile p1 = &x;
28 int *volatile p2 = &y;
29 if (x != 42)
30 abort ();
31 #pragma omp barrier
32 *p2 = 1;
33 p1[0]++;
34 v[0] = 7;
35 v[41] = 8;
36 #pragma omp barrier
37 if (x != 43 || y != 1)
38 abort ();
39 if (v[0] != 7 || v[41] != 8)
40 abort ();
41 if ((fl & 2) && (((uintptr_t) p1 | (uintptr_t) p2
42 | (uintptr_t) &v[0]) & 63) != 0)
43 abort ();
44 }
45 x = xo;
46 #pragma omp teams
47 #pragma omp parallel private (y) firstprivate (x, w) allocate (h: x, y, w)
48 {
49 int *volatile p1 = &x;
50 int *volatile p2 = &y;
51 if (x != 42 || w[17] != 17 || w[41] != 41)
52 abort ();
53 #pragma omp barrier
54 *p2 = 1;
55 p1[0]++;
56 w[19]++;
57 #pragma omp barrier
58 if (x != 43 || y != 1 || w[19] != 20)
59 abort ();
60 if ((fl & 1) && (((uintptr_t) p1 | (uintptr_t) p2
61 | (uintptr_t) &w[0]) & 63) != 0)
62 abort ();
63 }
64 x = xo;
65 #pragma omp parallel for private (y) firstprivate (x) allocate (h: x, y, r, l, n) reduction(+: r) lastprivate (l) linear (n: 16)
66 for (i = 0; i < 64; i++)
67 {
68 if (x != 42)
69 abort ();
70 y = 1;
71 l = i;
72 n += y + 15;
73 r += i;
74 if ((fl & 1) && (((uintptr_t) &x | (uintptr_t) &y | (uintptr_t) &r
75 | (uintptr_t) &l | (uintptr_t) &n) & 63) != 0)
76 abort ();
77 }
78 x = xo;
79 #pragma omp parallel
80 {
81 #pragma omp for lastprivate (l2) private (i1) allocate (h: l2, l3, i1) lastprivate (conditional: l3)
82 for (i1 = 0; i1 < 64; i1++)
83 {
84 l2[0] = i1;
85 l2[1] = i1 + 1;
86 l2[2] = i1 + 2;
87 l2[3] = i1 + 3;
88 if (i1 < 37)
89 l3 = i1;
90 if ((fl & 1) && (((uintptr_t) &l2[0] | (uintptr_t) &l3 | (uintptr_t) &i1) & 63) != 0)
91 abort ();
92 }
93 #pragma omp for collapse(2) lastprivate(l4, i2, j2) linear (n2:17) allocate (h: n2, l4, i2, j2)
94 for (i2 = 3; i2 < 5; i2++)
95 for (j2 = 17; j2 < 22; j2 += 2)
96 {
97 n2 += 17;
98 l4 = i2 * 31 + j2;
99 if ((fl & 1) && (((uintptr_t) &l4 | (uintptr_t) &n2
100 | (uintptr_t) &i2 | (uintptr_t) &j2) & 63) != 0)
101 abort ();
102 }
103 #pragma omp for collapse(2) lastprivate(l5, i3, j3) linear (n3:17) schedule (static, 3) allocate (n3, l5, i3, j3)
104 for (i3 = 3; i3 < 5; i3++)
105 for (j3 = 17; j3 < 23; j3 += 2)
106 {
107 n3 += 17;
108 l5 = i3 * 31 + j3;
109 if ((fl & 2) && (((uintptr_t) &l5 | (uintptr_t) &n3
110 | (uintptr_t) &i3 | (uintptr_t) &j3) & 63) != 0)
111 abort ();
112 }
113 #pragma omp for collapse(2) lastprivate(l6, i4, j4) linear (n4:17) schedule (dynamic) allocate (h: n4, l6, i4, j4)
114 for (i4 = 3; i4 < 5; i4++)
115 for (j4 = 17; j4 < 22; j4 += 2)
116 {
117 n4 += 17;
118 l6 = i4 * 31 + j4;
119 if ((fl & 1) && (((uintptr_t) &l6 | (uintptr_t) &n4
120 | (uintptr_t) &i4 | (uintptr_t) &j4) & 63) != 0)
121 abort ();
122 }
123 #pragma omp for lastprivate (i5) allocate (i5)
124 for (i5 = 1; i5 < 17; i5 += 3)
125 {
126 if ((fl & 2) && (((uintptr_t) &i5) & 63) != 0)
127 abort ();
128 }
129 #pragma omp for reduction(+:p[2:px], q[:3], r2) allocate(h: p, q, r2)
130 for (i = 0; i < 32; i++)
131 {
132 p[2] += i;
133 p[3] += 2 * i;
134 q[0] += 3 * i;
135 q[2] += 4 * i;
136 r2[0] += 5 * i;
137 r2[3] += 6 * i;
138 /* Can't really rely on alignment of &p[0], the implementation could
139 allocate the whole array or do what GCC does and allocate only part
140 of it. */
141 if ((fl & 1) && (((uintptr_t) &q[0] | (uintptr_t) &r2[0]) & 63) != 0)
142 abort ();
143 }
144 #pragma omp task private(y) firstprivate(x) allocate(x, y)
145 {
146 int *volatile p1 = &x;
147 int *volatile p2 = &y;
148 if (x != 42)
149 abort ();
150 p1[0]++;
151 p2[0] = 21;
152 if (x != 43 || y != 21)
153 abort ();
154 if ((fl & 2) && (((uintptr_t) p1 | (uintptr_t) p2) & 63) != 0)
155 abort ();
156 }
157 #pragma omp task private(y) firstprivate(x) allocate(h: x, y)
158 {
159 int *volatile p1 = &x;
160 int *volatile p2 = &y;
161 if (x != 42)
162 abort ();
163 p1[0]++;
164 p2[0] = 21;
165 if (x != 43 || y != 21)
166 abort ();
167 if ((fl & 1) && (((uintptr_t) p1 | (uintptr_t) p2) & 63) != 0)
168 abort ();
169 }
170 #pragma omp task private(y) firstprivate(s) allocate(s, y)
171 {
172 int *volatile p1 = &s.a;
173 int *volatile p2 = &s.b;
174 int *volatile p3 = &y;
175 if (s.a != 27 || s.b != 29)
176 abort ();
177 p1[0]++;
178 p2[0]++;
179 p3[0] = 21;
180 if (s.a != 28 || s.b != 30 || y != 21)
181 abort ();
182 if ((fl & 2) && (((uintptr_t) p1 | (uintptr_t) p3) & 63) != 0)
183 abort ();
184 }
185 #pragma omp task private(y) firstprivate(s) allocate(h: s, y)
186 {
187 int *volatile p1 = &s.a;
188 int *volatile p2 = &s.b;
189 int *volatile p3 = &y;
190 if (s.a != 27 || s.b != 29)
191 abort ();
192 p1[0]++;
193 p2[0]++;
194 p3[0] = 21;
195 if (s.a != 28 || s.b != 30 || y != 21)
196 abort ();
197 if ((fl & 1) && (((uintptr_t) p1 | (uintptr_t) p3) & 63) != 0)
198 abort ();
199 }
200 }
201 if (r != 64 * 63 / 2 || l != 63 || n != 8 + 16 * 64)
202 abort ();
203 if (l2[0] != 63 || l2[1] != 63 + 1 || l2[2] != 63 + 2 || l2[3] != 63 + 3 || l3 != 36)
204 abort ();
205 if (i2 != 5 || j2 != 23 || n2 != 9 + 6 * 17 || l4 != 4 * 31 + 21)
206 abort ();
207 if (i3 != 5 || j3 != 23 || n3 != 10 + 6 * 17 || l5 != 4 * 31 + 21)
208 abort ();
209 if (i4 != 5 || j4 != 23 || n4 != 11 + 6 * 17 || l6 != 4 * 31 + 21)
210 abort ();
211 if (i5 != 19)
212 abort ();
213 if (p[2] != (32 * 31) / 2 || p[3] != 2 * (32 * 31) / 2
214 || q[0] != 3 * (32 * 31) / 2 || q[2] != 4 * (32 * 31) / 2
215 || r2[0] != 5 * (32 * 31) / 2 || r2[3] != 6 * (32 * 31) / 2)
216 abort ();
217 r = 0;
218 x = xo;
219 #pragma omp parallel shared (x, y, r, n5) firstprivate (h)
220 {
221 #pragma omp masked
222 n5 = omp_get_num_threads ();
223 #pragma omp scope private (y) firstprivate (x) reduction(+:r) \
224 allocate (h: x, y, r)
225 {
226 int *volatile p1 = &x;
227 int *volatile p2 = &y;
228 int *volatile p3 = &r;
229 if (x != 42)
230 abort ();
231 #pragma omp barrier
232 *p2 = 1;
233 p1[0]++;
234 p3[0]++;
235 #pragma omp barrier
236 if (x != 43 || y != 1 || r != 1)
237 abort ();
238 if ((fl & 1) && (((uintptr_t) p1 | (uintptr_t) p2
239 | (uintptr_t) p3) & 63) != 0)
240 abort ();
241 }
242 }
243 if (x != 42 || r != n5)
244 abort ();
245 }
246
247 void
248 bar (int x, omp_allocator_handle_t h)
249 {
250 int y = 0, r = 0, i, i1, l, l2[4], l3, n = 8;
251 int i2, j2, n2 = 9, l4;
252 int i3, j3, n3 = 10, l5;
253 int i4, j4, n4 = 11, l6;
254 int i5;
255 struct S s = { 27, 29 };
256 int xo = x;
257 #pragma omp parallel private (y) firstprivate (x) allocate (x, y)
258 {
259 if (x != 42)
260 abort ();
261 #pragma omp barrier
262 y = 1;
263 x++;
264 #pragma omp barrier
265 if (x != 43 || y != 1)
266 abort ();
267 }
268 x = xo;
269 #pragma omp teams
270 #pragma omp parallel private (y) firstprivate (x) allocate (h: x, y)
271 {
272 if (x != 42)
273 abort ();
274 #pragma omp barrier
275 y = 1;
276 x++;
277 #pragma omp barrier
278 if (x != 43 || y != 1)
279 abort ();
280 }
281 x = xo;
282 #pragma omp parallel for private (y) firstprivate (x) allocate (h: x, y, r, l, n) reduction(+: r) lastprivate (l) linear (n: 16)
283 for (i = 0; i < 64; i++)
284 {
285 if (x != 42)
286 abort ();
287 y = 1;
288 l = i;
289 n += y + 15;
290 r += i;
291 }
292 x = xo;
293 #pragma omp parallel
294 {
295 #pragma omp for lastprivate (l2) private (i1) allocate (h: l2, l3, i1) lastprivate (conditional: l3)
296 for (i1 = 0; i1 < 64; i1++)
297 {
298 l2[0] = i1;
299 l2[1] = i1 + 1;
300 l2[2] = i1 + 2;
301 l2[3] = i1 + 3;
302 if (i1 < 37)
303 l3 = i1;
304 }
305 #pragma omp for collapse(2) lastprivate(l4, i2, j2) linear (n2:17) allocate (h: n2, l4, i2, j2)
306 for (i2 = 3; i2 < 5; i2++)
307 for (j2 = 17; j2 < 22; j2 += 2)
308 {
309 n2 += 17;
310 l4 = i2 * 31 + j2;
311 }
312 #pragma omp for collapse(2) lastprivate(l5, i3, j3) linear (n3:17) schedule (static, 3) allocate (n3, l5, i3, j3)
313 for (i3 = 3; i3 < 5; i3++)
314 for (j3 = 17; j3 < 23; j3 += 2)
315 {
316 n3 += 17;
317 l5 = i3 * 31 + j3;
318 }
319 #pragma omp for collapse(2) lastprivate(l6, i4, j4) linear (n4:17) schedule (dynamic) allocate (h: n4, l6, i4, j4)
320 for (i4 = 3; i4 < 5; i4++)
321 for (j4 = 17; j4 < 22; j4 += 2)
322 {
323 n4 += 17;
324 l6 = i4 * 31 + j4;
325 }
326 #pragma omp for lastprivate (i5) allocate (i5)
327 for (i5 = 1; i5 < 17; i5 += 3)
328 ;
329 #pragma omp task private(y) firstprivate(x) allocate(x, y)
330 {
331 if (x != 42)
332 abort ();
333 x++;
334 y = 21;
335 if (x != 43 || y != 21)
336 abort ();
337 }
338 #pragma omp task private(y) firstprivate(x) allocate(h: x, y)
339 {
340 if (x != 42)
341 abort ();
342 x++;
343 y = 21;
344 if (x != 43 || y != 21)
345 abort ();
346 }
347 #pragma omp task private(y) firstprivate(s) allocate(s, y)
348 {
349 if (s.a != 27 || s.b != 29)
350 abort ();
351 s.a++;
352 s.b++;
353 y = 21;
354 if (s.a != 28 || s.b != 30 || y != 21)
355 abort ();
356 }
357 #pragma omp task private(y) firstprivate(s) allocate(h: s, y)
358 {
359 if (s.a != 27 || s.b != 29)
360 abort ();
361 s.a++;
362 s.b++;
363 y = 21;
364 if (s.a != 28 || s.b != 30 || y != 21)
365 abort ();
366 }
367 }
368 if (r != 64 * 63 / 2 || l != 63 || n != 8 + 16 * 64)
369 abort ();
370 if (l2[0] != 63 || l2[1] != 63 + 1 || l2[2] != 63 + 2 || l2[3] != 63 + 3 || l3 != 36)
371 abort ();
372 if (i2 != 5 || j2 != 23 || n2 != 9 + 6 * 17 || l4 != 4 * 31 + 21)
373 abort ();
374 if (i3 != 5 || j3 != 23 || n3 != 10 + 6 * 17 || l5 != 4 * 31 + 21)
375 abort ();
376 if (i4 != 5 || j4 != 23 || n4 != 11 + 6 * 17 || l6 != 4 * 31 + 21)
377 abort ();
378 if (i5 != 19)
379 abort ();
380 }
381
382 int
383 main ()
384 {
385 omp_alloctrait_t traits[3]
386 = { { omp_atk_alignment, 64 },
387 { omp_atk_fallback, omp_atv_null_fb } };
388 omp_allocator_handle_t a
389 = omp_init_allocator (omp_default_mem_space, 2, traits);
390 int p[4], q[3];
391 if (a == omp_null_allocator)
392 abort ();
393 omp_set_default_allocator (omp_default_mem_alloc);
394 foo (42, p, q, 2, omp_null_allocator, 0);
395 foo (42, p, q, 2, omp_default_mem_alloc, 0);
396 foo (42, p, q, 2, a, 1);
397 omp_set_default_allocator (a);
398 foo (42, p, q, 2, omp_null_allocator, 3);
399 foo (42, p, q, 2, omp_default_mem_alloc, 2);
400 bar (42, a);
401 omp_destroy_allocator (a);
402 return 0;
403 }