The following fixes the vector type registered for external defs
in call arguments when vectorizing with SLP.  We assumed uniform
vectype_in types here but with calls like .COND_MUL we also have
mask arguments which, when invariant or external, need to have
a proper mask vector type.

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

2022-01-28  Richard Biener  <rguent...@suse.de>

        PR tree-optimization/104267
        * tree-vect-stmts.c (vectorizable_call): Properly use the
        per-argument determined vector type for externals and
        invariants.
---
 gcc/tree-vect-stmts.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/gcc/tree-vect-stmts.cc b/gcc/tree-vect-stmts.cc
index 824ebb6354b..000a0f4b47e 100644
--- a/gcc/tree-vect-stmts.cc
+++ b/gcc/tree-vect-stmts.cc
@@ -3444,7 +3444,9 @@ vectorizable_call (vec_info *vinfo,
     {
       if (slp_node)
        for (i = 0; i < nargs; ++i)
-         if (!vect_maybe_update_slp_op_vectype (slp_op[i], vectype_in))
+         if (!vect_maybe_update_slp_op_vectype (slp_op[i],
+                                                vectypes[i]
+                                                ? vectypes[i] : vectype_in))
            {
              if (dump_enabled_p ())
                dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
-- 
2.31.1

Reply via email to