gcc/testsuite/gcc.dg/vect/vect-conditional_store_4.c

   1 /* { dg-do compile } */
   2 /* { dg-require-effective-target vect_int } */
   3 /* { dg-require-effective-target vect_masked_store } */
   4
   5 /* { dg-additional-options "-mavx2" { target avx2 } } */
   6 /* { dg-additional-options "-march=armv9-a" { target aarch64-*-* } } */
   7
   8 void foo4 (signed char *restrict a, int *restrict b, int *restrict c, int *restrict d, int n, int stride)
   9 {
  10   if (stride <= 1)
  11     return;
  12
  13   for (int i = 0; i < n; i++)
  14     {
  15       int res1 = c[i];
  16       int res2 = d[i];
  17       int t = b[i+stride];
  18       if (a[i] > 0)
  19         t = res1;
  20       else if (a[i] < 0)
  21         t = res2 * 2;
  22
  23       c[i] = t;
  24     }
  25 }
  26
  27 /* { dg-final { scan-tree-dump "LOOP VECTORIZED" "vect" } } */
  28 /* { dg-final { scan-tree-dump-times "VEC_COND_EXPR " "vect" 1 { target aarch64-*-* } } } */