1  /* Disabling epilogues until we find a better way to deal with scans.  */
       2  /* { dg-additional-options "--param vect-epilogues-nomask=0" } */
       3  /* { dg-require-effective-target vect_int } */
       4  /* { dg-require-effective-target arm_v8_2a_dotprod_neon_hw { target { aarch64*-*-* || arm*-*-* } } } */
       5  /* { dg-additional-options "-march=armv8.2-a+dotprod" { target { aarch64*-*-* } } } */
       6  /* { dg-add-options arm_v8_2a_dotprod_neon }  */
       7  
       8  #include <stdarg.h>
       9  #include "tree-vect.h"
      10  
      11  #define N 64
      12  
      13  #define DOT 43680
      14  
      15  unsigned char X[N] __attribute__ ((__aligned__(__BIGGEST_ALIGNMENT__))) = {0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,34,35,36,37,38,39,40,41,42,43,44,45,46,47,48,49,50,51,52,53,54,55,56,57,58,59,60,61,62,63};
      16  unsigned char Y[N] __attribute__ ((__aligned__(__BIGGEST_ALIGNMENT__))) = {64,63,62,61,60,59,58,57,56,55,54,53,52,51,50,49,48,47,46,45,44,43,42,41,40,39,38,37,36,35,34,33,32,31,30,29,28,27,26,25,24,23,22,21,20,19,18,17,16,15,14,13,12,11,10,9,8,7,6,5,4,3,2,1};
      17  
      18  /* char->short->int dot product. 
      19     Detected as a dot-product pattern.
      20     Should be vectorized on targets that support dot-product for unsigned chars
      21     (vect_udot_qi),
      22     and on targets that support widening-multiplication and widening-summation
      23     (vect_widen_mult_qi && vec_widen_sum_qi_to_si).
      24     Widening-multiplication can also be supported by type promotion and non-widening 
      25     multiplication (vect_unpack && vect_short_mult);
      26     Widening summation can also be supported by type promotion and non-widening 
      27     summation (vect_unpack).
      28     */
      29  __attribute__ ((noinline)) unsigned int
      30  foo (int len) {
      31    int i;
      32    unsigned int result = 0;
      33    unsigned short prod;
      34  
      35    for (i=0; i<len; i++) {
      36      prod = X[i] * Y[i];
      37      result += prod;
      38    }
      39    return result;
      40  }
      41  
      42  int main (void)
      43  {
      44    unsigned int dot;
      45    int i;
      46  
      47    check_vect ();
      48  
      49    dot = foo (N);
      50    if (dot != DOT)
      51      abort ();
      52  
      53    return 0;
      54  }
      55  
      56  /* { dg-final { scan-tree-dump-times "vect_recog_dot_prod_pattern: detected" 1 "vect" } } */
      57  /* { dg-final { scan-tree-dump-times "vectorized 1 loops" 1 "vect" { target vect_udot_qi } } } */
      58  /* { dg-final { scan-tree-dump-times "vectorized 1 loops" 1 "vect" { target { vect_widen_mult_qi_to_hi && vect_widen_sum_qi_to_si } } } } */
      59  
      60