1  /* { dg-do run } */
       2  /* { dg-require-effective-target s390_vxe2 } */
       3  /* { dg-options "-O3 -mzarch -march=arch13 -mzvector --save-temps" } */
       4  
       5  #include <string.h>
       6  #include <vecintrin.h>
       7  
       8  typedef vector unsigned char uv16qi;
       9  
      10  const unsigned char test_vec[16] = { 1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16 };
      11  
      12  #define NUM_TEST_LENGTHS 3
      13  
      14  unsigned int test_len[NUM_TEST_LENGTHS] = { 0, 12, 18 };
      15  
      16  
      17  /* Proceeding from left to right, the specified number (LEN+1) of
      18     bytes from SOURCE are stored right-justified in TARGET.  */
      19  void __attribute__((noinline, noclone, target ("arch=zEC12")))
      20  emul (const unsigned char *source, unsigned char *target, unsigned int len)
      21  {
      22    int start = 15 - len;
      23    if (start < 0)
      24      start = 0;
      25    for (int s = 0, t = start; t < 16; s++, t++)
      26      target[t] = source[s];
      27  }
      28  
      29  uv16qi __attribute__((noinline, noclone))
      30  vec_load_len_r_reg (const unsigned char *s, unsigned int len)
      31  {
      32    return vec_load_len_r (s, len);
      33  }
      34  
      35  void __attribute__((noinline, noclone))
      36  vec_load_len_r_mem (const unsigned char *s, uv16qi *t, unsigned int *len)
      37  {
      38    *t = vec_load_len_r (s, *len);
      39  }
      40  
      41  #define GEN_CONST_FUNC(CONST)				\
      42    static uv16qi inline						\
      43    vec_load_len_r_const##CONST (const unsigned char *s)	\
      44    {							\
      45      return vec_load_len_r (s, CONST);			\
      46    }
      47  
      48  #define GEN_CONST_TEST(CONST)				\
      49    memset (exp_result, 0, 16);				\
      50    emul (test_vec, exp_result, CONST);			\
      51    result = (uv16qi) { 0 };				\
      52    result = vec_load_len_r_const##CONST (test_vec);	\
      53    if (memcmp ((char*)&result, exp_result, 16) != 0)	\
      54      __builtin_abort ();
      55  
      56  GEN_CONST_FUNC(0)
      57  GEN_CONST_FUNC(12)
      58  GEN_CONST_FUNC(18)
      59  
      60  int
      61  main ()
      62  {
      63    unsigned char exp_result[16];
      64    uv16qi result;
      65  
      66    for (int i = 0; i < NUM_TEST_LENGTHS; i++)
      67      {
      68        memset (exp_result, 0, 16);
      69  
      70        emul (test_vec, exp_result, test_len[i]);
      71  
      72        result = (uv16qi) { 0 };
      73        result = vec_load_len_r_reg (test_vec, test_len[i]);
      74        if (memcmp ((char*)&result, exp_result, 16) != 0)
      75  	__builtin_abort ();
      76  
      77        result = (uv16qi) { 0 };
      78        vec_load_len_r_mem (test_vec, &result, &test_len[i]);
      79        if (memcmp ((char*)&result, exp_result, 16) != 0)
      80  	__builtin_abort ();
      81      }
      82  
      83    GEN_CONST_TEST(0)
      84    GEN_CONST_TEST(12)
      85    GEN_CONST_TEST(18)
      86  
      87    return 0;
      88  }
      89  
      90  /* vec_load_len_r_reg and vec_load_len_r_mem */
      91  /* { dg-final { scan-assembler-times "vlrlr\t" 2 } } */
      92  
      93  /* For the 2 constants.  The 3. should be implemented with vl.  */
      94  /* { dg-final { scan-assembler-times "vlrl\t" 2 } } */