diff mbox series

tree-optimization/105132 - add missing checking in vectorizable_operation

Message ID 20220404093409.75DB713216@imap2.suse-dmz.suse.de
State Committed
Commit 3eda57b4ae50b5fe57c174556f1daf3bd9a46b09
Headers
Series tree-optimization/105132 - add missing checking in vectorizable_operation |

Commit Message

Richard Biener April 4, 2022, 9:34 a.m. UTC 
  The following adds missing verification that the input vectors
have the same number of elements for vectorizable_operation.

Bootstrapped and tested on x86_64-unknown-linux-gnu, pushed.

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.
---
 gcc/testsuite/gcc.dg/torture/pr105132.c | 12 ++++++++++++
 gcc/tree-vect-stmts.cc                  |  6 ++++++
 2 files changed, 18 insertions(+)
 create mode 100644 gcc/testsuite/gcc.dg/torture/pr105132.c
diff mbox series

Patch

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