123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182 |
- struct S { int a, b, c[2]; };
- #pragma omp declare reduction (+: S : (omp_out.a += omp_in.a, omp_out.b += omp_in.b)) \
- initializer (omp_priv = { 0, 0, { 0, 0 } })
- void
- foo (S &x, S *&y, int n, int v)
- {
- S zu[3] = { { 45, 47, {} }, { 46, 48, {} }, { 47, 49, {} } };
- S uu[n], wu[n];
- S (&z)[3] = zu;
- S (&u)[n] = uu;
- S (&w)[n] = wu;
- int i;
- for (i = 0; i < n; i++)
- {
- w[i].a = u[i].a = n + i;
- w[i].b = u[i].b = n - i;
- w[i].c[0] = u[i].c[0] = 0;
- w[i].c[1] = u[i].c[1] = 0;
- }
- #pragma omp taskgroup task_reduction (+: x, y[:2], z[1:2], u, w[1:v])
- {
- #pragma omp task in_reduction (+: x, y[:2], z[1:2], u, w[1:v])
- {
- x.a++;
- x.b++;
- y[0].a += 2;
- y[0].b += 12;
- y[1].a += 3;
- y[1].b += 13;
- z[1].a += 4;
- z[1].b += 14;
- u[0].a += 5;
- u[0].b += 15;
- w[1].a += 6;
- w[1].b += 16;
- }
- #pragma omp target in_reduction (+: x, y[:2], z[1:2], u, w[1:v])
- {
- x.a += 4;
- x.b += 14;
- y[0].a += 5;
- y[0].b += 15;
- y[1].a += 6;
- y[1].b += 16;
- z[2].a += 7;
- z[2].b += 17;
- u[1].a += 8;
- u[1].b += 18;
- w[2].a += 7;
- w[2].b += 17;
- }
- #pragma omp target in_reduction (+: x, y[:v], z[1:v], u, w[1:2])
- {
- x.a += 9;
- x.b += 19;
- y[0].a += 10;
- y[0].b += 20;
- y[1].a += 11;
- y[1].b += 21;
- z[1].a += 12;
- z[1].b += 22;
- u[2].a += 13;
- u[2].b += 23;
- w[1].a += 14;
- w[1].b += 24;
- }
- }
- if (x.a != 56 || y[0].a != 60 || y[1].a != 64)
- __builtin_abort ();
- if (x.b != 86 || y[0].b != 100 || y[1].b != 104)
- __builtin_abort ();
- if (z[0].a != 45 || z[1].a != 62 || z[2].a != 54)
- __builtin_abort ();
- if (z[0].b != 47 || z[1].b != 84 || z[2].b != 66)
- __builtin_abort ();
- if (u[0].a != 8 || u[1].a != 12 || u[2].a != 18)
- __builtin_abort ();
- if (u[0].b != 18 || u[1].b != 20 || u[2].b != 24)
- __builtin_abort ();
- if (w[0].a != 3 || w[1].a != 24 || w[2].a != 12)
- __builtin_abort ();
- if (w[0].b != 3 || w[1].b != 42 || w[2].b != 18)
- __builtin_abort ();
- }
- void
- bar (S &x, S *&y, int n, int v)
- {
- S zu[3] = { { 45, 47, {} }, { 46, 48, {} }, { 47, 49, {} } };
- S uu[n], wu[n];
- S (&z)[3] = zu;
- S (&u)[n] = uu;
- S (&w)[n] = wu;
- int i;
- for (i = 0; i < n; i++)
- {
- w[i].a = u[i].a = n + i;
- w[i].b = u[i].b = n - i;
- w[i].c[0] = u[i].c[0] = 0;
- w[i].c[1] = u[i].c[1] = 0;
- }
- #pragma omp parallel master
- #pragma omp taskgroup task_reduction (+: x, y[:2], z[1:2], u, w[1:v])
- {
- #pragma omp task in_reduction (+: x, y[:2], z[1:2], u, w[1:v])
- {
- x.a++;
- x.b++;
- y[0].a += 2;
- y[0].b += 12;
- y[1].a += 3;
- y[1].b += 13;
- z[1].a += 4;
- z[1].b += 14;
- u[0].a += 5;
- u[0].b += 15;
- w[1].a += 6;
- w[1].b += 16;
- }
- #pragma omp target in_reduction (+: x, y[:2], z[1:2], u, w[1:v])
- {
- x.a += 4;
- x.b += 14;
- y[0].a += 5;
- y[0].b += 15;
- y[1].a += 6;
- y[1].b += 16;
- z[2].a += 7;
- z[2].b += 17;
- u[1].a += 8;
- u[1].b += 18;
- w[2].a += 7;
- w[2].b += 17;
- }
- #pragma omp target in_reduction (+: x, y[:v], z[1:v], u, w[1:2])
- {
- x.a += 9;
- x.b += 19;
- y[0].a += 10;
- y[0].b += 20;
- y[1].a += 11;
- y[1].b += 21;
- z[1].a += 12;
- z[1].b += 22;
- u[2].a += 13;
- u[2].b += 23;
- w[1].a += 14;
- w[1].b += 24;
- }
- }
- if (x.a != 56 || y[0].a != 77 || y[1].a != 84)
- __builtin_abort ();
- if (x.b != 86 || y[0].b != 147 || y[1].b != 154)
- __builtin_abort ();
- if (z[0].a != 45 || z[1].a != 62 || z[2].a != 54)
- __builtin_abort ();
- if (z[0].b != 47 || z[1].b != 84 || z[2].b != 66)
- __builtin_abort ();
- if (u[0].a != 8 || u[1].a != 12 || u[2].a != 18)
- __builtin_abort ();
- if (u[0].b != 18 || u[1].b != 20 || u[2].b != 24)
- __builtin_abort ();
- if (w[0].a != 3 || w[1].a != 24 || w[2].a != 12)
- __builtin_abort ();
- if (w[0].b != 3 || w[1].b != 42 || w[2].b != 18)
- __builtin_abort ();
- }
- int
- main ()
- {
- S x = { 42, 52 };
- S yu[2] = { { 43, 53 }, { 44, 54 } };
- S *y = yu;
- #pragma omp parallel master
- foo (x, y, 3, 2);
- x.a = 42;
- x.b = 52;
- bar (x, y, 3, 2);
- return 0;
- }
|