123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122 |
- // { dg-require-effective-target size32plus }
- // { dg-additional-options "-O2 -fopenmp -fdump-tree-vect-details" }
- // { dg-additional-options "-msse2" { target sse2_runtime } }
- // { dg-additional-options "-mavx" { target avx_runtime } }
- // { dg-final { scan-tree-dump-times "vectorized \[2-6] loops" 2 "vect" { target sse2_runtime } } }
- extern "C" void abort ();
- int r, a[1024], b[1024], q;
- #pragma omp declare reduction (foo: int: omp_out += omp_in) initializer (omp_priv = 0)
- __attribute__((noipa)) void
- foo (int *a, int *b, int &r)
- {
- #pragma omp for simd reduction (inscan, foo:r)
- for (int i = 0; i < 1024; i++)
- {
- b[i] = r;
- #pragma omp scan exclusive(r)
- r += a[i];
- }
- }
- __attribute__((noipa)) int
- bar (void)
- {
- int &s = q;
- q = 0;
- #pragma omp parallel
- #pragma omp for simd reduction (inscan, foo:s) nowait
- for (int i = 0; i < 1024; i++)
- {
- b[i] = s;
- #pragma omp scan exclusive(s)
- s += 2 * a[i];
- }
- return s;
- }
- __attribute__((noipa)) void
- baz (int *a, int *b, int &r)
- {
- #pragma omp parallel for simd reduction (inscan, foo:r) if (simd: 0)
- for (int i = 0; i < 1024; i++)
- {
- b[i] = r;
- #pragma omp scan exclusive(r)
- r += a[i];
- }
- }
- __attribute__((noipa)) int
- qux (void)
- {
- int &s = q;
- q = 0;
- #pragma omp parallel for simd reduction (inscan, foo:s)simdlen(1)
- for (int i = 0; i < 1024; i++)
- {
- b[i] = s;
- #pragma omp scan exclusive(s)
- s += 2 * a[i];
- }
- return s;
- }
- int
- main ()
- {
- int s = 0;
- for (int i = 0; i < 1024; ++i)
- {
- a[i] = i;
- b[i] = -1;
- asm ("" : "+g" (i));
- }
- #pragma omp parallel
- foo (a, b, r);
- if (r != 1024 * 1023 / 2)
- abort ();
- for (int i = 0; i < 1024; ++i)
- {
- if (b[i] != s)
- abort ();
- else
- b[i] = 25;
- s += i;
- }
- if (bar () != 1024 * 1023)
- abort ();
- s = 0;
- for (int i = 0; i < 1024; ++i)
- {
- if (b[i] != s)
- abort ();
- else
- b[i] = -1;
- s += 2 * i;
- }
- r = 0;
- baz (a, b, r);
- if (r != 1024 * 1023 / 2)
- abort ();
- s = 0;
- for (int i = 0; i < 1024; ++i)
- {
- if (b[i] != s)
- abort ();
- else
- b[i] = -25;
- s += i;
- }
- if (qux () != 1024 * 1023)
- abort ();
- s = 0;
- for (int i = 0; i < 1024; ++i)
- {
- if (b[i] != s)
- abort ();
- s += 2 * i;
- }
- }
|