aboutsummaryrefslogtreecommitdiff
path: root/gcc/testsuite/gcc.dg/vect/slp-mask-store-1.c
blob: 50b7066778e84b91266465236994fc3014c4c88a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
/* { dg-do run } */
/* { dg-additional-options "-mavx2" { target avx2 } } */

#include "tree-vect.h"

void __attribute__((noipa))
foo (unsigned * __restrict x, int * __restrict flag)
{
  for (int i = 0; i < 32; ++i)
    {
      if (flag[2*i+0])
        x[2*i+0] = x[2*i+0] + 3;
      if (flag[2*i+1])
        x[2*i+1] = x[2*i+1] + 177;
    }
}

unsigned x[16];
int flag[32] = { 1, 1, 0, 0, 0, 1, 1, 0, 1, 1, 0, 0, 0, 1, 1, 0,
                 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 };
unsigned res[16] = { 3, 177, 0, 0, 0, 177, 3, 0, 3, 177, 0, 0, 0, 177, 3, 0 };

int
main ()
{
  check_vect ();

  foo (x, flag);

  if (__builtin_memcmp (x, res, sizeof (x)) != 0)
    abort ();
  for (int i = 0; i < 32; ++i)
    if (flag[i] != 0 && flag[i] != 1)
      abort ();

  return 0;
}

/* { dg-final { scan-tree-dump-times "LOOP VECTORIZED" 1 "vect" { target { vect_masked_store && vect_masked_load } } } } */