1  /* { dg-do assemble { target aarch64_asm_sve_ok } } */
       2  /* { dg-options "-O2 -ftree-vectorize -fwrapv --save-temps" } */
       3  
       4  #include <stdint.h>
       5  
       6  #ifndef INDEX32
       7  #define INDEX16 int16_t
       8  #define INDEX32 int32_t
       9  #endif
      10  
      11  #define TEST_LOOP(DATA_TYPE, BITS)					\
      12    void __attribute__ ((noinline, noclone))				\
      13    f_##DATA_TYPE (DATA_TYPE *restrict dest, DATA_TYPE *restrict src,	\
      14  		 INDEX##BITS *indices, INDEX##BITS mask, int n)		\
      15    {									\
      16      for (int i = 0; i < n; ++i)						\
      17        dest[i] = src[(INDEX##BITS) (indices[i] + mask)];			\
      18    }
      19  
      20  #define TEST_ALL(T)				\
      21    T (int8_t, 16)				\
      22    T (uint8_t, 16)				\
      23    T (int16_t, 16)				\
      24    T (uint16_t, 16)				\
      25    T (_Float16, 16)				\
      26    T (int32_t, 16)				\
      27    T (uint32_t, 16)				\
      28    T (float, 16)					\
      29    T (int64_t, 32)				\
      30    T (uint64_t, 32)				\
      31    T (double, 32)
      32  
      33  TEST_ALL (TEST_LOOP)
      34  
      35  /* { dg-final { scan-assembler-times {\tld1b\tz[0-9]+\.s, p[0-7]/z, \[x[0-9]+, z[0-9]+.s, sxtw\]\n} 2 } } */
      36  /* { dg-final { scan-assembler-times {\tld1h\tz[0-9]+\.s, p[0-7]/z, \[x[0-9]+, z[0-9]+.s, sxtw 1\]\n} 3 } } */
      37  /* { dg-final { scan-assembler-times {\tld1w\tz[0-9]+\.s, p[0-7]/z, \[x[0-9]+, z[0-9]+.s, sxtw 2\]\n} 3 } } */
      38  /* { dg-final { scan-assembler-times {\tld1d\tz[0-9]+\.d, p[0-7]/z, \[x[0-9]+, z[0-9]+.d, sxtw 3\]\n} 3 } } */
      39  
      40  /* { dg-final { scan-assembler-times {\tsxt.\tz} 8 } } */
      41  /* { dg-final { scan-assembler-times {\tsxth\tz[0-9]+\.s,} 8 } } */
      42  
      43  /* { dg-final { scan-assembler-times {\tst1b\tz[0-9]+\.s,} 2 } } */
      44  /* { dg-final { scan-assembler-times {\tst1h\tz[0-9]+\.s,} 3 } } */
      45  /* { dg-final { scan-assembler-times {\tst1w\tz[0-9]+\.s,} 3 } } */
      46  /* { dg-final { scan-assembler-times {\tst1d\tz[0-9]+\.d,} 3 } } */