| // { dg-require-effective-target size32plus } |
| // { dg-additional-options "-fopenmp-simd" } |
| // { dg-additional-options "-mavx" { target avx_runtime } } |
| // { dg-final { scan-tree-dump-times "vectorized \[1-3] loops" 2 "vect" { target i?86-*-* x86_64-*-* } } } */ |
| |
| #include "../../gcc.dg/vect/tree-vect.h" |
| |
| int r, a[1024], b[1024], q; |
| |
| __attribute__((noipa)) void |
| foo (int *a, int *b, int &r) |
| { |
| #pragma omp simd reduction (inscan, +:r) |
| for (int i = 0; i < 1024; i++) |
| { |
| r += a[i]; |
| #pragma omp scan inclusive(r) |
| b[i] = r; |
| } |
| } |
| |
| __attribute__((noipa)) int |
| bar (void) |
| { |
| int &s = q; |
| q = 0; |
| #pragma omp simd reduction (inscan, +:s) |
| for (int i = 0; i < 1024; i++) |
| { |
| s += 2 * a[i]; |
| #pragma omp scan inclusive(s) |
| b[i] = s; |
| } |
| return s; |
| } |
| |
| __attribute__((noipa)) void |
| baz (int *a, int *b, int &r) |
| { |
| #pragma omp simd reduction (inscan, +:r) if (simd: 0) |
| for (int i = 0; i < 1024; i++) |
| { |
| r += a[i]; |
| #pragma omp scan inclusive(r) |
| b[i] = r; |
| } |
| } |
| |
| __attribute__((noipa)) int |
| qux (void) |
| { |
| int &s = q; |
| q = 0; |
| #pragma omp simd reduction (inscan, +:s) simdlen (1) |
| for (int i = 0; i < 1024; i++) |
| { |
| s += 2 * a[i]; |
| #pragma omp scan inclusive(s) |
| b[i] = s; |
| } |
| return s; |
| } |
| |
| int |
| main () |
| { |
| int s = 0; |
| check_vect (); |
| for (int i = 0; i < 1024; ++i) |
| { |
| a[i] = i; |
| b[i] = -1; |
| asm ("" : "+g" (i)); |
| } |
| foo (a, b, r); |
| if (r != 1024 * 1023 / 2) |
| abort (); |
| for (int i = 0; i < 1024; ++i) |
| { |
| s += i; |
| if (b[i] != s) |
| abort (); |
| else |
| b[i] = 25; |
| } |
| if (bar () != 1024 * 1023) |
| abort (); |
| s = 0; |
| for (int i = 0; i < 1024; ++i) |
| { |
| s += 2 * i; |
| if (b[i] != s) |
| abort (); |
| else |
| b[i] = -1; |
| } |
| r = 0; |
| baz (a, b, r); |
| if (r != 1024 * 1023 / 2) |
| abort (); |
| s = 0; |
| for (int i = 0; i < 1024; ++i) |
| { |
| s += i; |
| if (b[i] != s) |
| abort (); |
| else |
| b[i] = -25; |
| } |
| if (qux () != 1024 * 1023) |
| abort (); |
| s = 0; |
| for (int i = 0; i < 1024; ++i) |
| { |
| s += 2 * i; |
| if (b[i] != s) |
| abort (); |
| } |
| return 0; |
| } |