[gcc r12-7980] tree-optimization/105132 - add missing checking in vectorizable_operation

[gcc r12-7980] tree-optimization/105132 - add missing checking in vectorizable_operation

[Richard Biener]

https://gcc.gnu.org/g:3eda57b4ae50b5fe57c174556f1daf3bd9a46b09

commit r12-7980-g3eda57b4ae50b5fe57c174556f1daf3bd9a46b09
Author: Richard Biener <rguenther@suse.de>
Date:   Mon Apr 4 10:31:15 2022 +0200

    tree-optimization/105132 - add missing checking in vectorizable_operation
    
    The following adds missing verification that the input vectors
    have the same number of elements for vectorizable_operation.
    
    2022-04-04  Richard Biener  <rguenther@suse.de>
    
            PR tree-optimization/105132
            * tree-vect-stmts.cc (vectorizable_operation): Check that
            the input vectors have the same number of elements.
    
            * gcc.dg/torture/pr105132.c: New testcase.

Diff:
---
 gcc/testsuite/gcc.dg/torture/pr105132.c | 12 ++++++++++++
 gcc/tree-vect-stmts.cc                  |  6 ++++++
 2 files changed, 18 insertions(+)

diff --git a/gcc/testsuite/gcc.dg/torture/pr105132.c b/gcc/testsuite/gcc.dg/torture/pr105132.c
new file mode 100644
index 00000000000..f8f0b16ec56
--- /dev/null
+++ b/gcc/testsuite/gcc.dg/torture/pr105132.c
@@ -0,0 +1,12 @@
+/* { dg-do compile } */
+/* { dg-additional-options "-march=skylake-avx512" { target x86_64-*-* i?86-*-* } } */
+
+short a;
+extern int b[];
+int c;
+void d(long f[][5][5][17], int g[][5][5][17]) {
+  for (short e = 0; e < 17; e++) {
+    a = g[19][2][3][e];
+    b[e] = c & (f[3][2][3][e] && g[19][2][3][e]);
+  }
+}
diff --git a/gcc/tree-vect-stmts.cc b/gcc/tree-vect-stmts.cc
index f7449a79d1c..f6fc7e1fcdd 100644
--- a/gcc/tree-vect-stmts.cc
+++ b/gcc/tree-vect-stmts.cc
@@ -6133,6 +6133,9 @@ vectorizable_operation (vec_info *vinfo,
                              "use not simple.\n");
 	  return false;
 	}
+      if (vectype2
+	  && maybe_ne (nunits_out, TYPE_VECTOR_SUBPARTS (vectype2)))
+	return false;
     }
   if (op_type == ternary_op)
     {
@@ -6144,6 +6147,9 @@ vectorizable_operation (vec_info *vinfo,
                              "use not simple.\n");
 	  return false;
 	}
+      if (vectype3
+	  && maybe_ne (nunits_out, TYPE_VECTOR_SUBPARTS (vectype3)))
+	return false;
     }
 
   /* Multiple types in SLP are handled by creating the appropriate number of