1  #include <arm_neon.h>
       2  #include "arm-neon-ref.h"
       3  #include "compute-ref-data.h"
       4  
       5  /* Expected results.  */
       6  VECT_VAR_DECL(expected,int,8,8) [] = { 0xf8, 0xf8, 0xf9, 0xf9,
       7  				       0xfa, 0xfa, 0xfb, 0xfb };
       8  VECT_VAR_DECL(expected,int,16,4) [] = { 0xffff, 0xffff, 0xffff, 0xffff };
       9  VECT_VAR_DECL(expected,int,32,2) [] = { 0xfffffffc, 0xfffffffc };
      10  VECT_VAR_DECL(expected,int,64,1) [] = { 0xffffffffffffffff };
      11  VECT_VAR_DECL(expected,uint,8,8) [] = { 0x3c, 0x3c, 0x3c, 0x3c,
      12  					0x3d, 0x3d, 0x3d, 0x3d };
      13  VECT_VAR_DECL(expected,uint,16,4) [] = { 0x1ffe, 0x1ffe, 0x1ffe, 0x1ffe };
      14  VECT_VAR_DECL(expected,uint,32,2) [] = { 0x7ffffff, 0x7ffffff };
      15  VECT_VAR_DECL(expected,uint,64,1) [] = { 0x7fffffff };
      16  VECT_VAR_DECL(expected,int,8,16) [] = { 0xf8, 0xf8, 0xf9, 0xf9,
      17  					0xfa, 0xfa, 0xfb, 0xfb,
      18  					0xfc, 0xfc, 0xfd, 0xfd,
      19  					0xfe, 0xfe, 0xff, 0xff };
      20  VECT_VAR_DECL(expected,int,16,8) [] = { 0xffff, 0xffff, 0xffff, 0xffff,
      21  					0xffff, 0xffff, 0xffff, 0xffff };
      22  VECT_VAR_DECL(expected,int,32,4) [] = { 0xfffffffc, 0xfffffffc,
      23  					0xfffffffc, 0xfffffffc };
      24  VECT_VAR_DECL(expected,int,64,2) [] = { 0xffffffffffffffff, 0xffffffffffffffff };
      25  VECT_VAR_DECL(expected,uint,8,16) [] = { 0x3c, 0x3c, 0x3c, 0x3c,
      26  					 0x3d, 0x3d, 0x3d, 0x3d,
      27  					 0x3e, 0x3e, 0x3e, 0x3e,
      28  					 0x3f, 0x3f, 0x3f, 0x3f };
      29  VECT_VAR_DECL(expected,uint,16,8) [] = { 0x1ffe, 0x1ffe, 0x1ffe, 0x1ffe,
      30  					 0x1ffe, 0x1ffe, 0x1ffe, 0x1ffe };
      31  VECT_VAR_DECL(expected,uint,32,4) [] = { 0x7ffffff, 0x7ffffff,
      32  					 0x7ffffff, 0x7ffffff };
      33  VECT_VAR_DECL(expected,uint,64,2) [] = { 0x7fffffff, 0x7fffffff };
      34  
      35  #define TEST_MSG "VSHR_N"
      36  void exec_vshr_n (void)
      37  {
      38    /* Basic test: y=vshr_n(x,v), then store the result.  */
      39  #define TEST_VSHR_N(Q, T1, T2, W, N, V)					\
      40    VECT_VAR(vector_res, T1, W, N) =					\
      41      vshr##Q##_n_##T2##W(VECT_VAR(vector, T1, W, N),			\
      42  			V);						\
      43    vst1##Q##_##T2##W(VECT_VAR(result, T1, W, N), VECT_VAR(vector_res, T1, W, N))
      44  
      45    DECL_VARIABLE_ALL_VARIANTS(vector);
      46    DECL_VARIABLE_ALL_VARIANTS(vector_res);
      47  
      48    clean_results ();
      49  
      50    /* Initialize input "vector" from "buffer".  */
      51    TEST_MACRO_ALL_VARIANTS_2_5(VLOAD, vector, buffer);
      52  
      53    /* Choose shift amount arbitrarily.  */
      54    TEST_VSHR_N(, int, s, 8, 8, 1);
      55    TEST_VSHR_N(, int, s, 16, 4, 12);
      56    TEST_VSHR_N(, int, s, 32, 2, 2);
      57    TEST_VSHR_N(, int, s, 64, 1, 32);
      58    TEST_VSHR_N(, uint, u, 8, 8, 2);
      59    TEST_VSHR_N(, uint, u, 16, 4, 3);
      60    TEST_VSHR_N(, uint, u, 32, 2, 5);
      61    TEST_VSHR_N(, uint, u, 64, 1, 33);
      62  
      63    TEST_VSHR_N(q, int, s, 8, 16, 1);
      64    TEST_VSHR_N(q, int, s, 16, 8, 12);
      65    TEST_VSHR_N(q, int, s, 32, 4, 2);
      66    TEST_VSHR_N(q, int, s, 64, 2, 32);
      67    TEST_VSHR_N(q, uint, u, 8, 16, 2);
      68    TEST_VSHR_N(q, uint, u, 16, 8, 3);
      69    TEST_VSHR_N(q, uint, u, 32, 4, 5);
      70    TEST_VSHR_N(q, uint, u, 64, 2, 33);
      71  
      72  #define CMT ""
      73    CHECK(TEST_MSG, int, 8, 8, PRIx8, expected, CMT);
      74    CHECK(TEST_MSG, int, 16, 4, PRIx16, expected, CMT);
      75    CHECK(TEST_MSG, int, 32, 2, PRIx32, expected, CMT);
      76    CHECK(TEST_MSG, int, 64, 1, PRIx64, expected, CMT);
      77    CHECK(TEST_MSG, uint, 8, 8, PRIx8, expected, CMT);
      78    CHECK(TEST_MSG, uint, 16, 4, PRIx16, expected, CMT);
      79    CHECK(TEST_MSG, uint, 32, 2, PRIx32, expected, CMT);
      80    CHECK(TEST_MSG, uint, 64, 1, PRIx64, expected, CMT);
      81    CHECK(TEST_MSG, int, 8, 16, PRIx8, expected, CMT);
      82    CHECK(TEST_MSG, int, 16, 8, PRIx16, expected, CMT);
      83    CHECK(TEST_MSG, int, 32, 4, PRIx32, expected, CMT);
      84    CHECK(TEST_MSG, int, 64, 2, PRIx64, expected, CMT);
      85    CHECK(TEST_MSG, uint, 8, 16, PRIx8, expected, CMT);
      86    CHECK(TEST_MSG, uint, 16, 8, PRIx16, expected, CMT);
      87    CHECK(TEST_MSG, uint, 32, 4, PRIx32, expected, CMT);
      88    CHECK(TEST_MSG, uint, 64, 2, PRIx64, expected, CMT);
      89  }
      90  
      91  int main (void)
      92  {
      93    exec_vshr_n ();
      94    return 0;
      95  }