1  /* { dg-do assemble { target aarch64_asm_sve_ok } } */
       2  /* { dg-options "-O2 -ftree-vectorize -ffast-math --save-temps" } */
       3  
       4  #include <stdint.h>
       5  
       6  #ifndef INDEX32
       7  #define INDEX32 int32_t
       8  #define INDEX64 int64_t
       9  #endif
      10  
      11  #define TEST_LOOP(DATA_TYPE, CMP_TYPE)					\
      12    void									\
      13    f_##DATA_TYPE##_##CMP_TYPE						\
      14      (DATA_TYPE *restrict dest, DATA_TYPE *restrict *restrict src,	\
      15       CMP_TYPE *cmp1, CMP_TYPE *cmp2, int n)				\
      16    {									\
      17      for (int i = 0; i < n; ++i)						\
      18        if (cmp1[i] == cmp2[i])						\
      19  	dest[i] += *src[i];						\
      20    }
      21  
      22  #define TEST_TYPE(T, DATA_TYPE)		\
      23    T (DATA_TYPE, int64_t)		\
      24    T (DATA_TYPE, uint64_t)		\
      25    T (DATA_TYPE, double)
      26  
      27  #define TEST_ALL(T)			\
      28    TEST_TYPE (T, int64_t)		\
      29    TEST_TYPE (T, uint64_t)		\
      30    TEST_TYPE (T, double)
      31  
      32  TEST_ALL (TEST_LOOP)
      33  
      34  /* { dg-final { scan-assembler-times {\tld1d\tz[0-9]+\.d, p[0-7]/z, \[x[0-9]+, x[0-9]+, lsl 3\]\n} 36 } } */
      35  /* { dg-final { scan-assembler-times {\tcmpeq\tp[0-7]\.d, p[0-7]/z, z[0-9]+\.d, z[0-9]+\.d\n} 6 } } */
      36  /* { dg-final { scan-assembler-times {\tfcmeq\tp[0-7]\.d, p[0-7]/z, z[0-9]+\.d, z[0-9]+\.d\n} 3 } } */
      37  /* { dg-final { scan-assembler-times {\tld1d\tz[0-9]+\.d, p[0-7]/z, \[z[0-9]+\.d\]\n} 9 } } */
      38  /* { dg-final { scan-assembler-times {\tst1d\tz[0-9]+\.d, p[0-7], \[x[0-9]+, x[0-9]+, lsl 3\]\n} 9 } } */