gcc/testsuite/gcc.target/aarch64/sve/dup_lane_1.c

   1 /* { dg-do compile } */
   2 /* { dg-options "-O -msve-vector-bits=256" } */
   3
   4 #include <stdint.h>
   5
   6 typedef int64_t vnx2di __attribute__((vector_size (32)));
   7 typedef int32_t vnx4si __attribute__((vector_size (32)));
   8 typedef int16_t vnx8hi __attribute__((vector_size (32)));
   9 typedef int8_t vnx16qi __attribute__((vector_size (32)));
  10 typedef double vnx2df __attribute__((vector_size (32)));
  11 typedef float vnx4sf __attribute__((vector_size (32)));
  12 typedef _Float16 vnx8hf __attribute__((vector_size (32)));
  13
  14 #define MASK_2(X) X, X
  15 #define MASK_4(X) MASK_2 (X), MASK_2 (X)
  16 #define MASK_8(X) MASK_4 (X), MASK_4 (X)
  17 #define MASK_16(X) MASK_8 (X), MASK_8 (X)
  18 #define MASK_32(X) MASK_16 (X), MASK_16 (X)
  19
  20 #define INDEX_4 vnx2di
  21 #define INDEX_8 vnx4si
  22 #define INDEX_16 vnx8hi
  23 #define INDEX_32 vnx16qi
  24
  25 #define DUP_LANE(TYPE, NUNITS, INDEX)                                        \
  26   TYPE dup_##INDEX##_##TYPE (TYPE values1, TYPE values2)                     \
  27   {                                                                          \
  28     return __builtin_shuffle (values1, values2,                              \
  29                               ((INDEX_##NUNITS) { MASK_##NUNITS (INDEX) })); \
  30   }
  31
  32 #define TEST_ALL(T)                             \
  33   T (vnx2di, 4, 0)                              \
  34   T (vnx2di, 4, 2)                              \
  35   T (vnx2di, 4, 3)                              \
  36   T (vnx4si, 8, 0)                              \
  37   T (vnx4si, 8, 5)                              \
  38   T (vnx4si, 8, 7)                              \
  39   T (vnx8hi, 16, 0)                             \
  40   T (vnx8hi, 16, 6)                             \
  41   T (vnx8hi, 16, 15)                            \
  42   T (vnx16qi, 32, 0)                            \
  43   T (vnx16qi, 32, 19)                           \
  44   T (vnx16qi, 32, 31)                           \
  45   T (vnx2df, 4, 0)                              \
  46   T (vnx2df, 4, 2)                              \
  47   T (vnx2df, 4, 3)                              \
  48   T (vnx4sf, 8, 0)                              \
  49   T (vnx4sf, 8, 5)                              \
  50   T (vnx4sf, 8, 7)                              \
  51   T (vnx8hf, 16, 0)                             \
  52   T (vnx8hf, 16, 6)                             \
  53   T (vnx8hf, 16, 15)                            \
  54
  55 TEST_ALL (DUP_LANE)
  56
  57 /* { dg-final { scan-assembler-not {\ttbl\t} } } */
  58
  59 /* { dg-final { scan-assembler-times {\tdup\tz[0-9]+\.d, z[0-9]+\.d\[0\]} 2 } } */
  60 /* { dg-final { scan-assembler-times {\tdup\tz[0-9]+\.d, z[0-9]+\.d\[2\]} 2 } } */
  61 /* { dg-final { scan-assembler-times {\tdup\tz[0-9]+\.d, z[0-9]+\.d\[3\]} 2 } } */
  62 /* { dg-final { scan-assembler-times {\tdup\tz[0-9]+\.s, z[0-9]+\.s\[0\]} 2 } } */
  63 /* { dg-final { scan-assembler-times {\tdup\tz[0-9]+\.s, z[0-9]+\.s\[5\]} 2 } } */
  64 /* { dg-final { scan-assembler-times {\tdup\tz[0-9]+\.s, z[0-9]+\.s\[7\]} 2 } } */
  65 /* { dg-final { scan-assembler-times {\tdup\tz[0-9]+\.h, z[0-9]+\.h\[0\]} 2 } } */
  66 /* { dg-final { scan-assembler-times {\tdup\tz[0-9]+\.h, z[0-9]+\.h\[6\]} 2 } } */
  67 /* { dg-final { scan-assembler-times {\tdup\tz[0-9]+\.h, z[0-9]+\.h\[15\]} 2 } } */
  68 /* { dg-final { scan-assembler-times {\tdup\tz[0-9]+\.b, z[0-9]+\.b\[0\]} 1 } } */
  69 /* { dg-final { scan-assembler-times {\tdup\tz[0-9]+\.b, z[0-9]+\.b\[19\]} 1 } } */
  70 /* { dg-final { scan-assembler-times {\tdup\tz[0-9]+\.b, z[0-9]+\.b\[31\]} 1 } } */