gcc/testsuite/gcc.target/aarch64/mls_intrinsic_1.c

   1 /* { dg-do run } */
   2 /* { dg-options "-O3 --save-temps" } */
   3
   4 #include <arm_neon.h>
   5
   6 extern void abort (void);
   7
   8 #define MAPs(size, xx) int##size##xx##_t
   9 #define MAPu(size, xx) uint##size##xx##_t
  10
  11
  12 #define TEST_VMLS(q, su, size, in1_lanes, in2_lanes)            \
  13 static void                                                     \
  14 test_vmlsq_lane##q##_##su##size (MAP##su (size, ) * res,        \
  15                                  const MAP##su(size, ) *in1,    \
  16                                  const MAP##su(size, ) *in2)    \
  17 {                                                               \
  18   MAP##su (size, x##in1_lanes) a = vld1q_##su##size (res);      \
  19   MAP##su (size, x##in1_lanes) b = vld1q_##su##size (in1);      \
  20   MAP##su (size, x##in2_lanes) c = vld1##q##_##su##size (in2);  \
  21   a = vmlsq_lane##q##_##su##size (a, b, c, 1);                  \
  22   vst1q_##su##size (res, a);                                    \
  23 }
  24
  25 #define BUILD_VARS(width, n_lanes, n_half_lanes)                \
  26 TEST_VMLS (, s, width, n_lanes, n_half_lanes)                   \
  27 TEST_VMLS (q, s, width, n_lanes, n_lanes)                       \
  28 TEST_VMLS (, u, width, n_lanes, n_half_lanes)                   \
  29 TEST_VMLS (q, u, width, n_lanes, n_lanes)                       \
  30
  31 BUILD_VARS (32, 4, 2)
  32 BUILD_VARS (16, 8, 4)
  33
  34 #define MAP_OPs +
  35 #define MAP_OPu -
  36
  37 #define POOL4 {0, 1, 2, 3}
  38 #define POOL8 {0, 1, 2, 3, 4, 5, 6, 7}
  39 #define EMPTY4s {0, 0, 0, 0}
  40 #define EMPTY8s {0, 0, 0, 0, 0, 0, 0, 0}
  41 #define EMPTY4u {0, 2, 4, 6}
  42 #define EMPTY8u {0, 2, 4, 6, 8, 10, 12, 14}
  43
  44 #define BUILD_TEST(su, size, lanes)                             \
  45 static void                                                     \
  46 test_##su##size (void)                                          \
  47 {                                                               \
  48   int i;                                                        \
  49   MAP##su (size,) pool[lanes] = POOL##lanes;                    \
  50   MAP##su (size,) res[lanes] = EMPTY##lanes##su;                \
  51   MAP##su (size,) res2[lanes] = EMPTY##lanes##su;               \
  52                                                                 \
  53   /* Forecfully avoid optimization.  */                         \
  54   asm volatile ("" : : : "memory");                             \
  55   test_vmlsq_lane_##su##size (res, pool, pool);                 \
  56   for (i = 0; i < lanes; i++)                                   \
  57     if (res[i] MAP_OP##su pool[i] != 0)                         \
  58       abort ();                                                 \
  59                                                                 \
  60   /* Forecfully avoid optimization.  */                         \
  61   asm volatile ("" : : : "memory");                             \
  62   test_vmlsq_laneq_##su##size (res2, pool, pool);               \
  63   for (i = 0; i < lanes; i++)                                   \
  64     if (res2[i] MAP_OP##su pool[i] != 0)                        \
  65       abort ();                                                 \
  66 }
  67
  68 #undef BUILD_VARS
  69 #define BUILD_VARS(size, lanes)                                 \
  70 BUILD_TEST (s, size, lanes)                                     \
  71 BUILD_TEST (u, size, lanes)
  72
  73 BUILD_VARS (32, 4)
  74 BUILD_VARS (16, 8)
  75
  76 int
  77 main (int argc, char **argv)
  78 {
  79   test_s32 ();
  80   test_u32 ();
  81   test_s16 ();
  82   test_u16 ();
  83   return 0;
  84 }
  85
  86 /* { dg-final { scan-assembler-times "mls\\tv\[0-9\]+\.4s, v\[0-9\]+\.4s, v\[0-9\]+\.4s\\\[\[0-9\]+\\\]" 4 } } */
  87 /* { dg-final { scan-assembler-times "mls\\tv\[0-9\]+\.8h, v\[0-9\]+\.8h, v\[0-9\]+\.8h\\\[\[0-9\]+\\\]" 4 } } */
  88