mirror of
https://github.com/autc04/Retro68.git
synced 2024-12-01 11:52:47 +00:00
154 lines
2.8 KiB
C
154 lines
2.8 KiB
C
extern "C" void abort ();
|
|
|
|
struct S { S (); S (unsigned long long int, int); ~S (); static int cnt1, cnt2, cnt3; unsigned long long int s; int t; };
|
|
|
|
int S::cnt1;
|
|
int S::cnt2;
|
|
int S::cnt3;
|
|
|
|
S::S ()
|
|
{
|
|
#pragma omp atomic
|
|
cnt1++;
|
|
}
|
|
|
|
S::S (unsigned long long int x, int y) : s (x), t (y)
|
|
{
|
|
#pragma omp atomic update
|
|
++cnt2;
|
|
}
|
|
|
|
S::~S ()
|
|
{
|
|
#pragma omp atomic
|
|
cnt3 = cnt3 + 1;
|
|
if (t < 3 || t > 9 || (t & 1) == 0)
|
|
abort ();
|
|
}
|
|
|
|
void
|
|
rbar (S *p, S *o)
|
|
{
|
|
p->s = 1;
|
|
if (o->t != 5)
|
|
abort ();
|
|
p->t = 9;
|
|
}
|
|
|
|
static inline void
|
|
rbaz (S *o, S *i)
|
|
{
|
|
if (o->t != 5 || i->t != 9)
|
|
abort ();
|
|
o->s *= i->s;
|
|
}
|
|
|
|
#pragma omp declare reduction (+: S : omp_out.s += omp_in.s) \
|
|
initializer (omp_priv (0, 3))
|
|
#pragma omp declare reduction (*: S : rbaz (&omp_out, &omp_in)) \
|
|
initializer (rbar (&omp_priv, &omp_orig))
|
|
|
|
S gs = { 0, 7 };
|
|
S &g = gs;
|
|
S hs (1, 5);
|
|
S &h = hs;
|
|
|
|
int
|
|
foo (int *a, int &b)
|
|
{
|
|
int xs = 0;
|
|
int &x = xs;
|
|
#pragma omp taskloop reduction (+:x) in_reduction (+:b)
|
|
for (int i = 0; i < 64; i++)
|
|
{
|
|
x += a[i];
|
|
b += a[i] * 2;
|
|
}
|
|
return x;
|
|
}
|
|
|
|
unsigned long long int
|
|
bar (int *a, unsigned long long int &b)
|
|
{
|
|
unsigned long long int xs = 1;
|
|
unsigned long long int &x = xs;
|
|
#pragma omp taskloop reduction (*:x) in_reduction (*:b)
|
|
for (int i = 0; i < 64; i++)
|
|
{
|
|
#pragma omp task in_reduction (*:x)
|
|
x *= a[i];
|
|
#pragma omp task in_reduction (*:b)
|
|
b *= (3 - a[i]);
|
|
}
|
|
return x;
|
|
}
|
|
|
|
void
|
|
baz (int i, int *a, int *c)
|
|
{
|
|
#pragma omp task in_reduction (*:h) in_reduction (+:g)
|
|
{
|
|
g.s += 7 * a[i];
|
|
h.s *= (3 - c[i]);
|
|
if ((g.t != 7 && g.t != 3) || (h.t != 5 && h.t != 9))
|
|
abort ();
|
|
}
|
|
}
|
|
|
|
void
|
|
test ()
|
|
{
|
|
int i, j, a[64], b = 0, c[64];
|
|
unsigned long long int d = 1, e;
|
|
S ms (0, 7);
|
|
for (i = 0; i < 64; i++)
|
|
{
|
|
a[i] = 2 * i;
|
|
c[i] = 1 + ((i % 3) != 1);
|
|
}
|
|
#pragma omp parallel
|
|
#pragma omp master
|
|
{
|
|
S ns = { 1, 5 };
|
|
S &m = ms;
|
|
S &n = ns;
|
|
#pragma omp taskgroup task_reduction (+:b)
|
|
j = foo (a, b);
|
|
#pragma omp taskgroup task_reduction (*:d)
|
|
e = bar (c, d);
|
|
#pragma omp taskloop reduction (+: g, m) reduction (*: h, n)
|
|
for (i = 0; i < 64; ++i)
|
|
{
|
|
g.s += 3 * a[i];
|
|
h.s *= (3 - c[i]);
|
|
m.s += 4 * a[i];
|
|
n.s *= c[i];
|
|
if ((g.t != 7 && g.t != 3) || (h.t != 5 && h.t != 9)
|
|
|| (m.t != 7 && m.t != 3) || (n.t != 5 && n.t != 9))
|
|
abort ();
|
|
baz (i, a, c);
|
|
}
|
|
if (n.s != (1ULL << 43) || n.t != 5)
|
|
abort ();
|
|
}
|
|
if (j != 63 * 64 || b != 63 * 64 * 2)
|
|
abort ();
|
|
if (e != (1ULL << 43) || d != (1ULL << 21))
|
|
abort ();
|
|
if (g.s != 63 * 64 * 10 || g.t != 7)
|
|
abort ();
|
|
if (h.s != (1ULL << 42) || h.t != 5)
|
|
abort ();
|
|
if (ms.s != 63 * 64 * 4 || ms.t != 7)
|
|
abort ();
|
|
}
|
|
|
|
int
|
|
main ()
|
|
{
|
|
int c1 = S::cnt1, c2 = S::cnt2, c3 = S::cnt3;
|
|
test ();
|
|
if (S::cnt1 + S::cnt2 - c1 - c2 != S::cnt3 - c3)
|
|
abort ();
|
|
}
|