[gcc(refs/users/marxin/heads/PR97079-fix-SLP-and-SVE)] SLP: fix SVE issues

[gcc(refs/users/marxin/heads/PR97079-fix-SLP-and-SVE)] SLP: fix SVE issues

[Martin Liska]

https://gcc.gnu.org/g:ce243245707e1d5e23f3f564b6987b82adfb67fd

commit ce243245707e1d5e23f3f564b6987b82adfb67fd
Author: Martin Liska <mliska@suse.cz>
Date:   Mon Oct 12 12:54:31 2020 +0200

    SLP: fix SVE issues
    
    The patch fixes the following 2 issues:
    
      .MASK_STORE_LANES (&a, 4B, max_mask_34, vect_array.12);
    
    here we miss to return the last argument as stored value.
    
    ivtmp_32 = ivtmp_31 + POLY_INT_CST [4, 4];
    
    here we miss a bail out in vect_recog_over_widening_pattern.
    
    gcc/ChangeLog:
    
            PR tree-optimization/97079
            * internal-fn.c (internal_fn_stored_value_index): Handle also
            .MASK_STORE_LANES.
            * tree-vect-patterns.c (vect_recog_over_widening_pattern): Bail
            out for POLY_INT_CST.
    
    gcc/testsuite/ChangeLog:
    
            PR tree-optimization/97079
            * gcc.target/aarch64/sve/pr97079.c: New test.

Diff:
---
 gcc/internal-fn.c                              |  1 +
 gcc/testsuite/gcc.target/aarch64/sve/pr97079.c | 22 ++++++++++++++++++++++
 gcc/tree-vect-patterns.c                       |  2 ++
 3 files changed, 25 insertions(+)

diff --git a/gcc/internal-fn.c b/gcc/internal-fn.c
index 92cb3cd845a..792d2ca568a 100644
--- a/gcc/internal-fn.c
+++ b/gcc/internal-fn.c
@@ -3795,6 +3795,7 @@ internal_fn_stored_value_index (internal_fn fn)
   switch (fn)
     {
     case IFN_MASK_STORE:
+    case IFN_MASK_STORE_LANES:
     case IFN_SCATTER_STORE:
     case IFN_MASK_SCATTER_STORE:
     case IFN_LEN_STORE:
diff --git a/gcc/testsuite/gcc.target/aarch64/sve/pr97079.c b/gcc/testsuite/gcc.target/aarch64/sve/pr97079.c
new file mode 100644
index 00000000000..06e6a7cde94
--- /dev/null
+++ b/gcc/testsuite/gcc.target/aarch64/sve/pr97079.c
@@ -0,0 +1,22 @@
+/* { dg-do compile } */
+/* { dg-options "-O3 -march=armv8.2-a+sve" } */
+
+void g(void);
+
+int a[8][3];
+int b;
+void c(void)
+{
+  int d[] = {7, 3};
+  int *e = a[0];
+  int f;
+  b = 0;
+  for (; b < 8; b++)
+    {
+      f = 0;
+      for (; f < 3; f++)
+	a[b][f] = 0;
+    }
+  g();
+  *e = (long)d;
+}
diff --git a/gcc/tree-vect-patterns.c b/gcc/tree-vect-patterns.c
index 71e4e106202..6302bc42f46 100644
--- a/gcc/tree-vect-patterns.c
+++ b/gcc/tree-vect-patterns.c
@@ -1571,6 +1571,8 @@ vect_recog_over_widening_pattern (vec_info *vinfo,
       tree op = gimple_op (last_stmt, first_op + i);
       if (TREE_CODE (op) == INTEGER_CST)
 	unprom[i].set_op (op, vect_constant_def);
+      else if (TREE_CODE (op) == POLY_INT_CST)
+	return NULL;
       else if (TREE_CODE (op) == SSA_NAME)
 	{
 	  bool op_single_use_p = true;