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 --git a/gcc/testsuite/gcc.dg/torture/pr105132.c b/gcc/testsuite/gcc.dg/torture/pr105132.c
new file mode 100644 (file)
index 0000000..f8f0b16
--- /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 f7449a79d1cae1e7d67d49acb7778225e14f5428..f6fc7e1fcddafa92af4cf4cb761d58f39e854fea 100644 (file)
--- 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