CWG 616, 1213 - value category of subobject references.
[official-gcc.git] / gcc / tree-vect-slp.c
blob508cd9d2fe796a82cae1cd6275ac06de2ded9986
1 /* SLP - Basic Block Vectorization
2 Copyright (C) 2007-2018 Free Software Foundation, Inc.
3 Contributed by Dorit Naishlos <dorit@il.ibm.com>
4 and Ira Rosen <irar@il.ibm.com>
6 This file is part of GCC.
8 GCC is free software; you can redistribute it and/or modify it under
9 the terms of the GNU General Public License as published by the Free
10 Software Foundation; either version 3, or (at your option) any later
11 version.
13 GCC is distributed in the hope that it will be useful, but WITHOUT ANY
14 WARRANTY; without even the implied warranty of MERCHANTABILITY or
15 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
16 for more details.
18 You should have received a copy of the GNU General Public License
19 along with GCC; see the file COPYING3. If not see
20 <http://www.gnu.org/licenses/>. */
22 #include "config.h"
23 #include "system.h"
24 #include "coretypes.h"
25 #include "backend.h"
26 #include "target.h"
27 #include "rtl.h"
28 #include "tree.h"
29 #include "gimple.h"
30 #include "tree-pass.h"
31 #include "ssa.h"
32 #include "optabs-tree.h"
33 #include "insn-config.h"
34 #include "recog.h" /* FIXME: for insn_data */
35 #include "params.h"
36 #include "fold-const.h"
37 #include "stor-layout.h"
38 #include "gimple-iterator.h"
39 #include "cfgloop.h"
40 #include "tree-vectorizer.h"
41 #include "langhooks.h"
42 #include "gimple-walk.h"
43 #include "dbgcnt.h"
44 #include "tree-vector-builder.h"
45 #include "vec-perm-indices.h"
46 #include "gimple-fold.h"
47 #include "internal-fn.h"
50 /* Recursively free the memory allocated for the SLP tree rooted at NODE. */
52 static void
53 vect_free_slp_tree (slp_tree node)
55 int i;
56 slp_tree child;
58 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
59 vect_free_slp_tree (child);
61 gimple *stmt;
62 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, stmt)
63 /* After transform some stmts are removed and thus their vinfo is gone. */
64 if (vinfo_for_stmt (stmt))
66 gcc_assert (STMT_VINFO_NUM_SLP_USES (vinfo_for_stmt (stmt)) > 0);
67 STMT_VINFO_NUM_SLP_USES (vinfo_for_stmt (stmt))--;
70 SLP_TREE_CHILDREN (node).release ();
71 SLP_TREE_SCALAR_STMTS (node).release ();
72 SLP_TREE_VEC_STMTS (node).release ();
73 SLP_TREE_LOAD_PERMUTATION (node).release ();
75 free (node);
79 /* Free the memory allocated for the SLP instance. */
81 void
82 vect_free_slp_instance (slp_instance instance)
84 vect_free_slp_tree (SLP_INSTANCE_TREE (instance));
85 SLP_INSTANCE_LOADS (instance).release ();
86 free (instance);
90 /* Create an SLP node for SCALAR_STMTS. */
92 static slp_tree
93 vect_create_new_slp_node (vec<gimple *> scalar_stmts)
95 slp_tree node;
96 gimple *stmt = scalar_stmts[0];
97 unsigned int nops;
99 if (is_gimple_call (stmt))
100 nops = gimple_call_num_args (stmt);
101 else if (is_gimple_assign (stmt))
103 nops = gimple_num_ops (stmt) - 1;
104 if (gimple_assign_rhs_code (stmt) == COND_EXPR)
105 nops++;
107 else if (gimple_code (stmt) == GIMPLE_PHI)
108 nops = 0;
109 else
110 return NULL;
112 node = XNEW (struct _slp_tree);
113 SLP_TREE_SCALAR_STMTS (node) = scalar_stmts;
114 SLP_TREE_VEC_STMTS (node).create (0);
115 SLP_TREE_NUMBER_OF_VEC_STMTS (node) = 0;
116 SLP_TREE_CHILDREN (node).create (nops);
117 SLP_TREE_LOAD_PERMUTATION (node) = vNULL;
118 SLP_TREE_TWO_OPERATORS (node) = false;
119 SLP_TREE_DEF_TYPE (node) = vect_internal_def;
121 unsigned i;
122 FOR_EACH_VEC_ELT (scalar_stmts, i, stmt)
123 STMT_VINFO_NUM_SLP_USES (vinfo_for_stmt (stmt))++;
125 return node;
129 /* This structure is used in creation of an SLP tree. Each instance
130 corresponds to the same operand in a group of scalar stmts in an SLP
131 node. */
132 typedef struct _slp_oprnd_info
134 /* Def-stmts for the operands. */
135 vec<gimple *> def_stmts;
136 /* Information about the first statement, its vector def-type, type, the
137 operand itself in case it's constant, and an indication if it's a pattern
138 stmt. */
139 tree first_op_type;
140 enum vect_def_type first_dt;
141 bool first_pattern;
142 bool second_pattern;
143 } *slp_oprnd_info;
146 /* Allocate operands info for NOPS operands, and GROUP_SIZE def-stmts for each
147 operand. */
148 static vec<slp_oprnd_info>
149 vect_create_oprnd_info (int nops, int group_size)
151 int i;
152 slp_oprnd_info oprnd_info;
153 vec<slp_oprnd_info> oprnds_info;
155 oprnds_info.create (nops);
156 for (i = 0; i < nops; i++)
158 oprnd_info = XNEW (struct _slp_oprnd_info);
159 oprnd_info->def_stmts.create (group_size);
160 oprnd_info->first_dt = vect_uninitialized_def;
161 oprnd_info->first_op_type = NULL_TREE;
162 oprnd_info->first_pattern = false;
163 oprnd_info->second_pattern = false;
164 oprnds_info.quick_push (oprnd_info);
167 return oprnds_info;
171 /* Free operands info. */
173 static void
174 vect_free_oprnd_info (vec<slp_oprnd_info> &oprnds_info)
176 int i;
177 slp_oprnd_info oprnd_info;
179 FOR_EACH_VEC_ELT (oprnds_info, i, oprnd_info)
181 oprnd_info->def_stmts.release ();
182 XDELETE (oprnd_info);
185 oprnds_info.release ();
189 /* Find the place of the data-ref in STMT in the interleaving chain that starts
190 from FIRST_STMT. Return -1 if the data-ref is not a part of the chain. */
193 vect_get_place_in_interleaving_chain (gimple *stmt, gimple *first_stmt)
195 gimple *next_stmt = first_stmt;
196 int result = 0;
198 if (first_stmt != GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt)))
199 return -1;
203 if (next_stmt == stmt)
204 return result;
205 next_stmt = GROUP_NEXT_ELEMENT (vinfo_for_stmt (next_stmt));
206 if (next_stmt)
207 result += GROUP_GAP (vinfo_for_stmt (next_stmt));
209 while (next_stmt);
211 return -1;
214 /* Check whether it is possible to load COUNT elements of type ELT_MODE
215 using the method implemented by duplicate_and_interleave. Return true
216 if so, returning the number of intermediate vectors in *NVECTORS_OUT
217 (if nonnull) and the type of each intermediate vector in *VECTOR_TYPE_OUT
218 (if nonnull). */
220 bool
221 can_duplicate_and_interleave_p (unsigned int count, machine_mode elt_mode,
222 unsigned int *nvectors_out,
223 tree *vector_type_out,
224 tree *permutes)
226 poly_int64 elt_bytes = count * GET_MODE_SIZE (elt_mode);
227 poly_int64 nelts;
228 unsigned int nvectors = 1;
229 for (;;)
231 scalar_int_mode int_mode;
232 poly_int64 elt_bits = elt_bytes * BITS_PER_UNIT;
233 if (multiple_p (current_vector_size, elt_bytes, &nelts)
234 && int_mode_for_size (elt_bits, 0).exists (&int_mode))
236 tree int_type = build_nonstandard_integer_type
237 (GET_MODE_BITSIZE (int_mode), 1);
238 tree vector_type = build_vector_type (int_type, nelts);
239 if (VECTOR_MODE_P (TYPE_MODE (vector_type)))
241 vec_perm_builder sel1 (nelts, 2, 3);
242 vec_perm_builder sel2 (nelts, 2, 3);
243 poly_int64 half_nelts = exact_div (nelts, 2);
244 for (unsigned int i = 0; i < 3; ++i)
246 sel1.quick_push (i);
247 sel1.quick_push (i + nelts);
248 sel2.quick_push (half_nelts + i);
249 sel2.quick_push (half_nelts + i + nelts);
251 vec_perm_indices indices1 (sel1, 2, nelts);
252 vec_perm_indices indices2 (sel2, 2, nelts);
253 if (can_vec_perm_const_p (TYPE_MODE (vector_type), indices1)
254 && can_vec_perm_const_p (TYPE_MODE (vector_type), indices2))
256 if (nvectors_out)
257 *nvectors_out = nvectors;
258 if (vector_type_out)
259 *vector_type_out = vector_type;
260 if (permutes)
262 permutes[0] = vect_gen_perm_mask_checked (vector_type,
263 indices1);
264 permutes[1] = vect_gen_perm_mask_checked (vector_type,
265 indices2);
267 return true;
271 if (!multiple_p (elt_bytes, 2, &elt_bytes))
272 return false;
273 nvectors *= 2;
277 /* Get the defs for the rhs of STMT (collect them in OPRNDS_INFO), check that
278 they are of a valid type and that they match the defs of the first stmt of
279 the SLP group (stored in OPRNDS_INFO). This function tries to match stmts
280 by swapping operands of STMTS[STMT_NUM] when possible. Non-zero *SWAP
281 indicates swap is required for cond_expr stmts. Specifically, *SWAP
282 is 1 if STMT is cond and operands of comparison need to be swapped;
283 *SWAP is 2 if STMT is cond and code of comparison needs to be inverted.
284 If there is any operand swap in this function, *SWAP is set to non-zero
285 value.
286 If there was a fatal error return -1; if the error could be corrected by
287 swapping operands of father node of this one, return 1; if everything is
288 ok return 0. */
289 static int
290 vect_get_and_check_slp_defs (vec_info *vinfo, unsigned char *swap,
291 vec<gimple *> stmts, unsigned stmt_num,
292 vec<slp_oprnd_info> *oprnds_info)
294 gimple *stmt = stmts[stmt_num];
295 tree oprnd;
296 unsigned int i, number_of_oprnds;
297 gimple *def_stmt;
298 enum vect_def_type dt = vect_uninitialized_def;
299 bool pattern = false;
300 slp_oprnd_info oprnd_info;
301 int first_op_idx = 1;
302 bool commutative = false;
303 bool first_op_cond = false;
304 bool first = stmt_num == 0;
305 bool second = stmt_num == 1;
307 if (is_gimple_call (stmt))
309 number_of_oprnds = gimple_call_num_args (stmt);
310 first_op_idx = 3;
312 else if (is_gimple_assign (stmt))
314 enum tree_code code = gimple_assign_rhs_code (stmt);
315 number_of_oprnds = gimple_num_ops (stmt) - 1;
316 /* Swap can only be done for cond_expr if asked to, otherwise we
317 could result in different comparison code to the first stmt. */
318 if (gimple_assign_rhs_code (stmt) == COND_EXPR
319 && COMPARISON_CLASS_P (gimple_assign_rhs1 (stmt)))
321 first_op_cond = true;
322 number_of_oprnds++;
324 else
325 commutative = commutative_tree_code (code);
327 else
328 return -1;
330 bool swapped = (*swap != 0);
331 gcc_assert (!swapped || first_op_cond);
332 for (i = 0; i < number_of_oprnds; i++)
334 again:
335 if (first_op_cond)
337 /* Map indicating how operands of cond_expr should be swapped. */
338 int maps[3][4] = {{0, 1, 2, 3}, {1, 0, 2, 3}, {0, 1, 3, 2}};
339 int *map = maps[*swap];
341 if (i < 2)
342 oprnd = TREE_OPERAND (gimple_op (stmt, first_op_idx), map[i]);
343 else
344 oprnd = gimple_op (stmt, map[i]);
346 else
347 oprnd = gimple_op (stmt, first_op_idx + (swapped ? !i : i));
349 oprnd_info = (*oprnds_info)[i];
351 if (!vect_is_simple_use (oprnd, vinfo, &def_stmt, &dt))
353 if (dump_enabled_p ())
355 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
356 "Build SLP failed: can't analyze def for ");
357 dump_generic_expr (MSG_MISSED_OPTIMIZATION, TDF_SLIM, oprnd);
358 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
361 return -1;
364 /* Check if DEF_STMT is a part of a pattern in LOOP and get the def stmt
365 from the pattern. Check that all the stmts of the node are in the
366 pattern. */
367 if (def_stmt && gimple_bb (def_stmt)
368 && vect_stmt_in_region_p (vinfo, def_stmt)
369 && vinfo_for_stmt (def_stmt)
370 && STMT_VINFO_IN_PATTERN_P (vinfo_for_stmt (def_stmt))
371 && !STMT_VINFO_RELEVANT (vinfo_for_stmt (def_stmt))
372 && !STMT_VINFO_LIVE_P (vinfo_for_stmt (def_stmt)))
374 pattern = true;
375 if (!first && !oprnd_info->first_pattern
376 /* Allow different pattern state for the defs of the
377 first stmt in reduction chains. */
378 && (oprnd_info->first_dt != vect_reduction_def
379 || (!second && !oprnd_info->second_pattern)))
381 if (i == 0
382 && !swapped
383 && commutative)
385 swapped = true;
386 goto again;
389 if (dump_enabled_p ())
391 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
392 "Build SLP failed: some of the stmts"
393 " are in a pattern, and others are not ");
394 dump_generic_expr (MSG_MISSED_OPTIMIZATION, TDF_SLIM, oprnd);
395 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
398 return 1;
401 def_stmt = STMT_VINFO_RELATED_STMT (vinfo_for_stmt (def_stmt));
402 dt = STMT_VINFO_DEF_TYPE (vinfo_for_stmt (def_stmt));
404 if (dt == vect_unknown_def_type)
406 if (dump_enabled_p ())
407 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
408 "Unsupported pattern.\n");
409 return -1;
412 switch (gimple_code (def_stmt))
414 case GIMPLE_PHI:
415 case GIMPLE_ASSIGN:
416 break;
418 default:
419 if (dump_enabled_p ())
420 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
421 "unsupported defining stmt:\n");
422 return -1;
426 if (second)
427 oprnd_info->second_pattern = pattern;
429 if (first)
431 oprnd_info->first_dt = dt;
432 oprnd_info->first_pattern = pattern;
433 oprnd_info->first_op_type = TREE_TYPE (oprnd);
435 else
437 /* Not first stmt of the group, check that the def-stmt/s match
438 the def-stmt/s of the first stmt. Allow different definition
439 types for reduction chains: the first stmt must be a
440 vect_reduction_def (a phi node), and the rest
441 vect_internal_def. */
442 tree type = TREE_TYPE (oprnd);
443 if ((oprnd_info->first_dt != dt
444 && !(oprnd_info->first_dt == vect_reduction_def
445 && dt == vect_internal_def)
446 && !((oprnd_info->first_dt == vect_external_def
447 || oprnd_info->first_dt == vect_constant_def)
448 && (dt == vect_external_def
449 || dt == vect_constant_def)))
450 || !types_compatible_p (oprnd_info->first_op_type, type))
452 /* Try swapping operands if we got a mismatch. */
453 if (i == 0
454 && !swapped
455 && commutative)
457 swapped = true;
458 goto again;
461 if (dump_enabled_p ())
462 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
463 "Build SLP failed: different types\n");
465 return 1;
467 if ((dt == vect_constant_def
468 || dt == vect_external_def)
469 && !current_vector_size.is_constant ()
470 && (TREE_CODE (type) == BOOLEAN_TYPE
471 || !can_duplicate_and_interleave_p (stmts.length (),
472 TYPE_MODE (type))))
474 if (dump_enabled_p ())
476 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
477 "Build SLP failed: invalid type of def "
478 "for variable-length SLP ");
479 dump_generic_expr (MSG_MISSED_OPTIMIZATION, TDF_SLIM, oprnd);
480 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
482 return -1;
486 /* Check the types of the definitions. */
487 switch (dt)
489 case vect_constant_def:
490 case vect_external_def:
491 break;
493 case vect_reduction_def:
494 case vect_induction_def:
495 case vect_internal_def:
496 oprnd_info->def_stmts.quick_push (def_stmt);
497 break;
499 default:
500 /* FORNOW: Not supported. */
501 if (dump_enabled_p ())
503 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
504 "Build SLP failed: illegal type of def ");
505 dump_generic_expr (MSG_MISSED_OPTIMIZATION, TDF_SLIM, oprnd);
506 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
509 return -1;
513 /* Swap operands. */
514 if (swapped)
516 /* If there are already uses of this stmt in a SLP instance then
517 we've committed to the operand order and can't swap it. */
518 if (STMT_VINFO_NUM_SLP_USES (vinfo_for_stmt (stmt)) != 0)
520 if (dump_enabled_p ())
522 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
523 "Build SLP failed: cannot swap operands of "
524 "shared stmt ");
525 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM, stmt, 0);
527 return -1;
530 if (first_op_cond)
532 tree cond = gimple_assign_rhs1 (stmt);
533 enum tree_code code = TREE_CODE (cond);
535 /* Swap. */
536 if (*swap == 1)
538 swap_ssa_operands (stmt, &TREE_OPERAND (cond, 0),
539 &TREE_OPERAND (cond, 1));
540 TREE_SET_CODE (cond, swap_tree_comparison (code));
542 /* Invert. */
543 else
545 swap_ssa_operands (stmt, gimple_assign_rhs2_ptr (stmt),
546 gimple_assign_rhs3_ptr (stmt));
547 bool honor_nans = HONOR_NANS (TREE_OPERAND (cond, 0));
548 code = invert_tree_comparison (TREE_CODE (cond), honor_nans);
549 gcc_assert (code != ERROR_MARK);
550 TREE_SET_CODE (cond, code);
553 else
554 swap_ssa_operands (stmt, gimple_assign_rhs1_ptr (stmt),
555 gimple_assign_rhs2_ptr (stmt));
556 if (dump_enabled_p ())
558 dump_printf_loc (MSG_NOTE, vect_location,
559 "swapped operands to match def types in ");
560 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
564 *swap = swapped;
565 return 0;
568 /* A subroutine of vect_build_slp_tree for checking VECTYPE, which is the
569 caller's attempt to find the vector type in STMT with the narrowest
570 element type. Return true if VECTYPE is nonnull and if it is valid
571 for VINFO. When returning true, update MAX_NUNITS to reflect the
572 number of units in VECTYPE. VINFO, GORUP_SIZE and MAX_NUNITS are
573 as for vect_build_slp_tree. */
575 static bool
576 vect_record_max_nunits (vec_info *vinfo, gimple *stmt, unsigned int group_size,
577 tree vectype, poly_uint64 *max_nunits)
579 if (!vectype)
581 if (dump_enabled_p ())
583 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
584 "Build SLP failed: unsupported data-type in ");
585 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM, stmt, 0);
586 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
588 /* Fatal mismatch. */
589 return false;
592 /* If populating the vector type requires unrolling then fail
593 before adjusting *max_nunits for basic-block vectorization. */
594 poly_uint64 nunits = TYPE_VECTOR_SUBPARTS (vectype);
595 unsigned HOST_WIDE_INT const_nunits;
596 if (is_a <bb_vec_info> (vinfo)
597 && (!nunits.is_constant (&const_nunits)
598 || const_nunits > group_size))
600 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
601 "Build SLP failed: unrolling required "
602 "in basic block SLP\n");
603 /* Fatal mismatch. */
604 return false;
607 /* In case of multiple types we need to detect the smallest type. */
608 vect_update_max_nunits (max_nunits, vectype);
609 return true;
612 /* STMTS is a group of GROUP_SIZE SLP statements in which some
613 statements do the same operation as the first statement and in which
614 the others do ALT_STMT_CODE. Return true if we can take one vector
615 of the first operation and one vector of the second and permute them
616 to get the required result. VECTYPE is the type of the vector that
617 would be permuted. */
619 static bool
620 vect_two_operations_perm_ok_p (vec<gimple *> stmts, unsigned int group_size,
621 tree vectype, tree_code alt_stmt_code)
623 unsigned HOST_WIDE_INT count;
624 if (!TYPE_VECTOR_SUBPARTS (vectype).is_constant (&count))
625 return false;
627 vec_perm_builder sel (count, count, 1);
628 for (unsigned int i = 0; i < count; ++i)
630 unsigned int elt = i;
631 if (gimple_assign_rhs_code (stmts[i % group_size]) == alt_stmt_code)
632 elt += count;
633 sel.quick_push (elt);
635 vec_perm_indices indices (sel, 2, count);
636 return can_vec_perm_const_p (TYPE_MODE (vectype), indices);
639 /* Verify if the scalar stmts STMTS are isomorphic, require data
640 permutation or are of unsupported types of operation. Return
641 true if they are, otherwise return false and indicate in *MATCHES
642 which stmts are not isomorphic to the first one. If MATCHES[0]
643 is false then this indicates the comparison could not be
644 carried out or the stmts will never be vectorized by SLP.
646 Note COND_EXPR is possibly ismorphic to another one after swapping its
647 operands. Set SWAP[i] to 1 if stmt I is COND_EXPR and isomorphic to
648 the first stmt by swapping the two operands of comparison; set SWAP[i]
649 to 2 if stmt I is isormorphic to the first stmt by inverting the code
650 of comparison. Take A1 >= B1 ? X1 : Y1 as an exmple, it can be swapped
651 to (B1 <= A1 ? X1 : Y1); or be inverted to (A1 < B1) ? Y1 : X1. */
653 static bool
654 vect_build_slp_tree_1 (vec_info *vinfo, unsigned char *swap,
655 vec<gimple *> stmts, unsigned int group_size,
656 unsigned nops, poly_uint64 *max_nunits,
657 bool *matches, bool *two_operators)
659 unsigned int i;
660 gimple *first_stmt = stmts[0], *stmt = stmts[0];
661 enum tree_code first_stmt_code = ERROR_MARK;
662 enum tree_code alt_stmt_code = ERROR_MARK;
663 enum tree_code rhs_code = ERROR_MARK;
664 enum tree_code first_cond_code = ERROR_MARK;
665 tree lhs;
666 bool need_same_oprnds = false;
667 tree vectype = NULL_TREE, first_op1 = NULL_TREE;
668 optab optab;
669 int icode;
670 machine_mode optab_op2_mode;
671 machine_mode vec_mode;
672 gimple *first_load = NULL, *prev_first_load = NULL;
674 /* For every stmt in NODE find its def stmt/s. */
675 FOR_EACH_VEC_ELT (stmts, i, stmt)
677 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
678 swap[i] = 0;
679 matches[i] = false;
681 if (dump_enabled_p ())
683 dump_printf_loc (MSG_NOTE, vect_location, "Build SLP for ");
684 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
687 /* Fail to vectorize statements marked as unvectorizable. */
688 if (!STMT_VINFO_VECTORIZABLE (vinfo_for_stmt (stmt)))
690 if (dump_enabled_p ())
692 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
693 "Build SLP failed: unvectorizable statement ");
694 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM, stmt, 0);
696 /* Fatal mismatch. */
697 matches[0] = false;
698 return false;
701 lhs = gimple_get_lhs (stmt);
702 if (lhs == NULL_TREE)
704 if (dump_enabled_p ())
706 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
707 "Build SLP failed: not GIMPLE_ASSIGN nor "
708 "GIMPLE_CALL ");
709 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM, stmt, 0);
711 /* Fatal mismatch. */
712 matches[0] = false;
713 return false;
716 tree nunits_vectype;
717 if (!vect_get_vector_types_for_stmt (stmt_info, &vectype,
718 &nunits_vectype)
719 || (nunits_vectype
720 && !vect_record_max_nunits (vinfo, stmt, group_size,
721 nunits_vectype, max_nunits)))
723 /* Fatal mismatch. */
724 matches[0] = false;
725 return false;
728 gcc_assert (vectype);
730 if (gcall *call_stmt = dyn_cast <gcall *> (stmt))
732 rhs_code = CALL_EXPR;
733 if (gimple_call_internal_p (call_stmt)
734 || gimple_call_tail_p (call_stmt)
735 || gimple_call_noreturn_p (call_stmt)
736 || !gimple_call_nothrow_p (call_stmt)
737 || gimple_call_chain (call_stmt))
739 if (dump_enabled_p ())
741 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
742 "Build SLP failed: unsupported call type ");
743 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
744 call_stmt, 0);
746 /* Fatal mismatch. */
747 matches[0] = false;
748 return false;
751 else
752 rhs_code = gimple_assign_rhs_code (stmt);
754 /* Check the operation. */
755 if (i == 0)
757 first_stmt_code = rhs_code;
759 /* Shift arguments should be equal in all the packed stmts for a
760 vector shift with scalar shift operand. */
761 if (rhs_code == LSHIFT_EXPR || rhs_code == RSHIFT_EXPR
762 || rhs_code == LROTATE_EXPR
763 || rhs_code == RROTATE_EXPR)
765 if (vectype == boolean_type_node)
767 if (dump_enabled_p ())
768 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
769 "Build SLP failed: shift of a"
770 " boolean.\n");
771 /* Fatal mismatch. */
772 matches[0] = false;
773 return false;
776 vec_mode = TYPE_MODE (vectype);
778 /* First see if we have a vector/vector shift. */
779 optab = optab_for_tree_code (rhs_code, vectype,
780 optab_vector);
782 if (!optab
783 || optab_handler (optab, vec_mode) == CODE_FOR_nothing)
785 /* No vector/vector shift, try for a vector/scalar shift. */
786 optab = optab_for_tree_code (rhs_code, vectype,
787 optab_scalar);
789 if (!optab)
791 if (dump_enabled_p ())
792 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
793 "Build SLP failed: no optab.\n");
794 /* Fatal mismatch. */
795 matches[0] = false;
796 return false;
798 icode = (int) optab_handler (optab, vec_mode);
799 if (icode == CODE_FOR_nothing)
801 if (dump_enabled_p ())
802 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
803 "Build SLP failed: "
804 "op not supported by target.\n");
805 /* Fatal mismatch. */
806 matches[0] = false;
807 return false;
809 optab_op2_mode = insn_data[icode].operand[2].mode;
810 if (!VECTOR_MODE_P (optab_op2_mode))
812 need_same_oprnds = true;
813 first_op1 = gimple_assign_rhs2 (stmt);
817 else if (rhs_code == WIDEN_LSHIFT_EXPR)
819 need_same_oprnds = true;
820 first_op1 = gimple_assign_rhs2 (stmt);
823 else
825 if (first_stmt_code != rhs_code
826 && alt_stmt_code == ERROR_MARK)
827 alt_stmt_code = rhs_code;
828 if (first_stmt_code != rhs_code
829 && (first_stmt_code != IMAGPART_EXPR
830 || rhs_code != REALPART_EXPR)
831 && (first_stmt_code != REALPART_EXPR
832 || rhs_code != IMAGPART_EXPR)
833 /* Handle mismatches in plus/minus by computing both
834 and merging the results. */
835 && !((first_stmt_code == PLUS_EXPR
836 || first_stmt_code == MINUS_EXPR)
837 && (alt_stmt_code == PLUS_EXPR
838 || alt_stmt_code == MINUS_EXPR)
839 && rhs_code == alt_stmt_code)
840 && !(STMT_VINFO_GROUPED_ACCESS (vinfo_for_stmt (stmt))
841 && (first_stmt_code == ARRAY_REF
842 || first_stmt_code == BIT_FIELD_REF
843 || first_stmt_code == INDIRECT_REF
844 || first_stmt_code == COMPONENT_REF
845 || first_stmt_code == MEM_REF)))
847 if (dump_enabled_p ())
849 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
850 "Build SLP failed: different operation "
851 "in stmt ");
852 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM, stmt, 0);
853 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
854 "original stmt ");
855 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
856 first_stmt, 0);
858 /* Mismatch. */
859 continue;
862 if (need_same_oprnds
863 && !operand_equal_p (first_op1, gimple_assign_rhs2 (stmt), 0))
865 if (dump_enabled_p ())
867 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
868 "Build SLP failed: different shift "
869 "arguments in ");
870 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM, stmt, 0);
872 /* Mismatch. */
873 continue;
876 if (rhs_code == CALL_EXPR)
878 gimple *first_stmt = stmts[0];
879 if (gimple_call_num_args (stmt) != nops
880 || !operand_equal_p (gimple_call_fn (first_stmt),
881 gimple_call_fn (stmt), 0)
882 || gimple_call_fntype (first_stmt)
883 != gimple_call_fntype (stmt))
885 if (dump_enabled_p ())
887 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
888 "Build SLP failed: different calls in ");
889 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
890 stmt, 0);
892 /* Mismatch. */
893 continue;
898 /* Grouped store or load. */
899 if (STMT_VINFO_GROUPED_ACCESS (vinfo_for_stmt (stmt)))
901 if (REFERENCE_CLASS_P (lhs))
903 /* Store. */
906 else
908 /* Load. */
909 first_load = GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt));
910 if (prev_first_load)
912 /* Check that there are no loads from different interleaving
913 chains in the same node. */
914 if (prev_first_load != first_load)
916 if (dump_enabled_p ())
918 dump_printf_loc (MSG_MISSED_OPTIMIZATION,
919 vect_location,
920 "Build SLP failed: different "
921 "interleaving chains in one node ");
922 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
923 stmt, 0);
925 /* Mismatch. */
926 continue;
929 else
930 prev_first_load = first_load;
932 } /* Grouped access. */
933 else
935 if (TREE_CODE_CLASS (rhs_code) == tcc_reference)
937 /* Not grouped load. */
938 if (dump_enabled_p ())
940 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
941 "Build SLP failed: not grouped load ");
942 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM, stmt, 0);
945 /* FORNOW: Not grouped loads are not supported. */
946 /* Fatal mismatch. */
947 matches[0] = false;
948 return false;
951 /* Not memory operation. */
952 if (TREE_CODE_CLASS (rhs_code) != tcc_binary
953 && TREE_CODE_CLASS (rhs_code) != tcc_unary
954 && TREE_CODE_CLASS (rhs_code) != tcc_expression
955 && TREE_CODE_CLASS (rhs_code) != tcc_comparison
956 && rhs_code != CALL_EXPR)
958 if (dump_enabled_p ())
960 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
961 "Build SLP failed: operation");
962 dump_printf (MSG_MISSED_OPTIMIZATION, " unsupported ");
963 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM, stmt, 0);
965 /* Fatal mismatch. */
966 matches[0] = false;
967 return false;
970 if (rhs_code == COND_EXPR)
972 tree cond_expr = gimple_assign_rhs1 (stmt);
973 enum tree_code cond_code = TREE_CODE (cond_expr);
974 enum tree_code swap_code = ERROR_MARK;
975 enum tree_code invert_code = ERROR_MARK;
977 if (i == 0)
978 first_cond_code = TREE_CODE (cond_expr);
979 else if (TREE_CODE_CLASS (cond_code) == tcc_comparison)
981 bool honor_nans = HONOR_NANS (TREE_OPERAND (cond_expr, 0));
982 swap_code = swap_tree_comparison (cond_code);
983 invert_code = invert_tree_comparison (cond_code, honor_nans);
986 if (first_cond_code == cond_code)
988 /* Isomorphic can be achieved by swapping. */
989 else if (first_cond_code == swap_code)
990 swap[i] = 1;
991 /* Isomorphic can be achieved by inverting. */
992 else if (first_cond_code == invert_code)
993 swap[i] = 2;
994 else
996 if (dump_enabled_p ())
998 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
999 "Build SLP failed: different"
1000 " operation");
1001 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
1002 stmt, 0);
1004 /* Mismatch. */
1005 continue;
1010 matches[i] = true;
1013 for (i = 0; i < group_size; ++i)
1014 if (!matches[i])
1015 return false;
1017 /* If we allowed a two-operation SLP node verify the target can cope
1018 with the permute we are going to use. */
1019 if (alt_stmt_code != ERROR_MARK
1020 && TREE_CODE_CLASS (alt_stmt_code) != tcc_reference)
1022 if (vectype == boolean_type_node
1023 || !vect_two_operations_perm_ok_p (stmts, group_size,
1024 vectype, alt_stmt_code))
1026 for (i = 0; i < group_size; ++i)
1027 if (gimple_assign_rhs_code (stmts[i]) == alt_stmt_code)
1029 matches[i] = false;
1030 if (dump_enabled_p ())
1032 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
1033 "Build SLP failed: different operation "
1034 "in stmt ");
1035 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
1036 stmts[i], 0);
1037 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
1038 "original stmt ");
1039 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
1040 first_stmt, 0);
1043 return false;
1045 *two_operators = true;
1048 return true;
1051 /* Traits for the hash_set to record failed SLP builds for a stmt set.
1052 Note we never remove apart from at destruction time so we do not
1053 need a special value for deleted that differs from empty. */
1054 struct bst_traits
1056 typedef vec <gimple *> value_type;
1057 typedef vec <gimple *> compare_type;
1058 static inline hashval_t hash (value_type);
1059 static inline bool equal (value_type existing, value_type candidate);
1060 static inline bool is_empty (value_type x) { return !x.exists (); }
1061 static inline bool is_deleted (value_type x) { return !x.exists (); }
1062 static inline void mark_empty (value_type &x) { x.release (); }
1063 static inline void mark_deleted (value_type &x) { x.release (); }
1064 static inline void remove (value_type &x) { x.release (); }
1066 inline hashval_t
1067 bst_traits::hash (value_type x)
1069 inchash::hash h;
1070 for (unsigned i = 0; i < x.length (); ++i)
1071 h.add_int (gimple_uid (x[i]));
1072 return h.end ();
1074 inline bool
1075 bst_traits::equal (value_type existing, value_type candidate)
1077 if (existing.length () != candidate.length ())
1078 return false;
1079 for (unsigned i = 0; i < existing.length (); ++i)
1080 if (existing[i] != candidate[i])
1081 return false;
1082 return true;
1085 typedef hash_set <vec <gimple *>, bst_traits> scalar_stmts_set_t;
1086 static scalar_stmts_set_t *bst_fail;
1088 typedef hash_map <vec <gimple *>, slp_tree,
1089 simple_hashmap_traits <bst_traits, slp_tree> >
1090 scalar_stmts_to_slp_tree_map_t;
1092 static slp_tree
1093 vect_build_slp_tree_2 (vec_info *vinfo,
1094 vec<gimple *> stmts, unsigned int group_size,
1095 poly_uint64 *max_nunits,
1096 vec<slp_tree> *loads,
1097 bool *matches, unsigned *npermutes, unsigned *tree_size,
1098 unsigned max_tree_size);
1100 static slp_tree
1101 vect_build_slp_tree (vec_info *vinfo,
1102 vec<gimple *> stmts, unsigned int group_size,
1103 poly_uint64 *max_nunits, vec<slp_tree> *loads,
1104 bool *matches, unsigned *npermutes, unsigned *tree_size,
1105 unsigned max_tree_size)
1107 if (bst_fail->contains (stmts))
1108 return NULL;
1109 slp_tree res = vect_build_slp_tree_2 (vinfo, stmts, group_size, max_nunits,
1110 loads, matches, npermutes, tree_size,
1111 max_tree_size);
1112 /* When SLP build fails for stmts record this, otherwise SLP build
1113 can be exponential in time when we allow to construct parts from
1114 scalars, see PR81723. */
1115 if (! res)
1117 vec <gimple *> x;
1118 x.create (stmts.length ());
1119 x.splice (stmts);
1120 bst_fail->add (x);
1122 return res;
1125 /* Recursively build an SLP tree starting from NODE.
1126 Fail (and return a value not equal to zero) if def-stmts are not
1127 isomorphic, require data permutation or are of unsupported types of
1128 operation. Otherwise, return 0.
1129 The value returned is the depth in the SLP tree where a mismatch
1130 was found. */
1132 static slp_tree
1133 vect_build_slp_tree_2 (vec_info *vinfo,
1134 vec<gimple *> stmts, unsigned int group_size,
1135 poly_uint64 *max_nunits,
1136 vec<slp_tree> *loads,
1137 bool *matches, unsigned *npermutes, unsigned *tree_size,
1138 unsigned max_tree_size)
1140 unsigned nops, i, this_tree_size = 0;
1141 poly_uint64 this_max_nunits = *max_nunits;
1142 gimple *stmt;
1143 slp_tree node;
1145 matches[0] = false;
1147 stmt = stmts[0];
1148 if (is_gimple_call (stmt))
1149 nops = gimple_call_num_args (stmt);
1150 else if (is_gimple_assign (stmt))
1152 nops = gimple_num_ops (stmt) - 1;
1153 if (gimple_assign_rhs_code (stmt) == COND_EXPR)
1154 nops++;
1156 else if (gimple_code (stmt) == GIMPLE_PHI)
1157 nops = 0;
1158 else
1159 return NULL;
1161 /* If the SLP node is a PHI (induction or reduction), terminate
1162 the recursion. */
1163 if (gimple_code (stmt) == GIMPLE_PHI)
1165 tree scalar_type = TREE_TYPE (PHI_RESULT (stmt));
1166 tree vectype = get_vectype_for_scalar_type (scalar_type);
1167 if (!vect_record_max_nunits (vinfo, stmt, group_size, vectype,
1168 max_nunits))
1169 return NULL;
1171 vect_def_type def_type = STMT_VINFO_DEF_TYPE (vinfo_for_stmt (stmt));
1172 /* Induction from different IVs is not supported. */
1173 if (def_type == vect_induction_def)
1175 FOR_EACH_VEC_ELT (stmts, i, stmt)
1176 if (stmt != stmts[0])
1177 return NULL;
1179 else
1181 /* Else def types have to match. */
1182 FOR_EACH_VEC_ELT (stmts, i, stmt)
1184 /* But for reduction chains only check on the first stmt. */
1185 if (GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt))
1186 && GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt)) != stmt)
1187 continue;
1188 if (STMT_VINFO_DEF_TYPE (vinfo_for_stmt (stmt)) != def_type)
1189 return NULL;
1192 node = vect_create_new_slp_node (stmts);
1193 return node;
1197 bool two_operators = false;
1198 unsigned char *swap = XALLOCAVEC (unsigned char, group_size);
1199 if (!vect_build_slp_tree_1 (vinfo, swap,
1200 stmts, group_size, nops,
1201 &this_max_nunits, matches, &two_operators))
1202 return NULL;
1204 /* If the SLP node is a load, terminate the recursion. */
1205 if (STMT_VINFO_GROUPED_ACCESS (vinfo_for_stmt (stmt))
1206 && DR_IS_READ (STMT_VINFO_DATA_REF (vinfo_for_stmt (stmt))))
1208 *max_nunits = this_max_nunits;
1209 node = vect_create_new_slp_node (stmts);
1210 loads->safe_push (node);
1211 return node;
1214 /* Get at the operands, verifying they are compatible. */
1215 vec<slp_oprnd_info> oprnds_info = vect_create_oprnd_info (nops, group_size);
1216 slp_oprnd_info oprnd_info;
1217 FOR_EACH_VEC_ELT (stmts, i, stmt)
1219 int res = vect_get_and_check_slp_defs (vinfo, &swap[i],
1220 stmts, i, &oprnds_info);
1221 if (res != 0)
1222 matches[(res == -1) ? 0 : i] = false;
1223 if (!matches[0])
1224 break;
1226 for (i = 0; i < group_size; ++i)
1227 if (!matches[i])
1229 vect_free_oprnd_info (oprnds_info);
1230 return NULL;
1233 auto_vec<slp_tree, 4> children;
1234 auto_vec<slp_tree> this_loads;
1236 stmt = stmts[0];
1238 if (tree_size)
1239 max_tree_size -= *tree_size;
1241 /* Create SLP_TREE nodes for the definition node/s. */
1242 FOR_EACH_VEC_ELT (oprnds_info, i, oprnd_info)
1244 slp_tree child;
1245 unsigned old_nloads = this_loads.length ();
1246 unsigned old_tree_size = this_tree_size;
1247 unsigned int j;
1249 if (oprnd_info->first_dt != vect_internal_def
1250 && oprnd_info->first_dt != vect_reduction_def
1251 && oprnd_info->first_dt != vect_induction_def)
1252 continue;
1254 if (++this_tree_size > max_tree_size)
1256 FOR_EACH_VEC_ELT (children, j, child)
1257 vect_free_slp_tree (child);
1258 vect_free_oprnd_info (oprnds_info);
1259 return NULL;
1262 if ((child = vect_build_slp_tree (vinfo, oprnd_info->def_stmts,
1263 group_size, &this_max_nunits,
1264 &this_loads, matches, npermutes,
1265 &this_tree_size,
1266 max_tree_size)) != NULL)
1268 /* If we have all children of child built up from scalars then just
1269 throw that away and build it up this node from scalars. */
1270 if (!SLP_TREE_CHILDREN (child).is_empty ()
1271 /* ??? Rejecting patterns this way doesn't work. We'd have to
1272 do extra work to cancel the pattern so the uses see the
1273 scalar version. */
1274 && !is_pattern_stmt_p
1275 (vinfo_for_stmt (SLP_TREE_SCALAR_STMTS (child)[0])))
1277 slp_tree grandchild;
1279 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (child), j, grandchild)
1280 if (SLP_TREE_DEF_TYPE (grandchild) == vect_internal_def)
1281 break;
1282 if (!grandchild)
1284 /* Roll back. */
1285 this_loads.truncate (old_nloads);
1286 this_tree_size = old_tree_size;
1287 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (child), j, grandchild)
1288 vect_free_slp_tree (grandchild);
1289 SLP_TREE_CHILDREN (child).truncate (0);
1291 dump_printf_loc (MSG_NOTE, vect_location,
1292 "Building parent vector operands from "
1293 "scalars instead\n");
1294 oprnd_info->def_stmts = vNULL;
1295 SLP_TREE_DEF_TYPE (child) = vect_external_def;
1296 children.safe_push (child);
1297 continue;
1301 oprnd_info->def_stmts = vNULL;
1302 children.safe_push (child);
1303 continue;
1306 /* If the SLP build failed fatally and we analyze a basic-block
1307 simply treat nodes we fail to build as externally defined
1308 (and thus build vectors from the scalar defs).
1309 The cost model will reject outright expensive cases.
1310 ??? This doesn't treat cases where permutation ultimatively
1311 fails (or we don't try permutation below). Ideally we'd
1312 even compute a permutation that will end up with the maximum
1313 SLP tree size... */
1314 if (is_a <bb_vec_info> (vinfo)
1315 && !matches[0]
1316 /* ??? Rejecting patterns this way doesn't work. We'd have to
1317 do extra work to cancel the pattern so the uses see the
1318 scalar version. */
1319 && !is_pattern_stmt_p (vinfo_for_stmt (stmt)))
1321 dump_printf_loc (MSG_NOTE, vect_location,
1322 "Building vector operands from scalars\n");
1323 child = vect_create_new_slp_node (oprnd_info->def_stmts);
1324 SLP_TREE_DEF_TYPE (child) = vect_external_def;
1325 children.safe_push (child);
1326 oprnd_info->def_stmts = vNULL;
1327 continue;
1330 /* If the SLP build for operand zero failed and operand zero
1331 and one can be commutated try that for the scalar stmts
1332 that failed the match. */
1333 if (i == 0
1334 /* A first scalar stmt mismatch signals a fatal mismatch. */
1335 && matches[0]
1336 /* ??? For COND_EXPRs we can swap the comparison operands
1337 as well as the arms under some constraints. */
1338 && nops == 2
1339 && oprnds_info[1]->first_dt == vect_internal_def
1340 && is_gimple_assign (stmt)
1341 /* Do so only if the number of not successful permutes was nor more
1342 than a cut-ff as re-trying the recursive match on
1343 possibly each level of the tree would expose exponential
1344 behavior. */
1345 && *npermutes < 4)
1347 /* See whether we can swap the matching or the non-matching
1348 stmt operands. */
1349 bool swap_not_matching = true;
1352 for (j = 0; j < group_size; ++j)
1354 if (matches[j] != !swap_not_matching)
1355 continue;
1356 gimple *stmt = stmts[j];
1357 /* Verify if we can swap operands of this stmt. */
1358 if (!is_gimple_assign (stmt)
1359 || !commutative_tree_code (gimple_assign_rhs_code (stmt)))
1361 if (!swap_not_matching)
1362 goto fail;
1363 swap_not_matching = false;
1364 break;
1366 /* Verify if we can safely swap or if we committed to a
1367 specific operand order already.
1368 ??? Instead of modifying GIMPLE stmts here we could
1369 record whether we want to swap operands in the SLP
1370 node and temporarily do that when processing it
1371 (or wrap operand accessors in a helper). */
1372 else if (swap[j] != 0
1373 || STMT_VINFO_NUM_SLP_USES (vinfo_for_stmt (stmt)))
1375 if (!swap_not_matching)
1377 if (dump_enabled_p ())
1379 dump_printf_loc (MSG_MISSED_OPTIMIZATION,
1380 vect_location,
1381 "Build SLP failed: cannot swap "
1382 "operands of shared stmt ");
1383 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION,
1384 TDF_SLIM, stmts[j], 0);
1386 goto fail;
1388 swap_not_matching = false;
1389 break;
1393 while (j != group_size);
1395 /* Swap mismatched definition stmts. */
1396 dump_printf_loc (MSG_NOTE, vect_location,
1397 "Re-trying with swapped operands of stmts ");
1398 for (j = 0; j < group_size; ++j)
1399 if (matches[j] == !swap_not_matching)
1401 std::swap (oprnds_info[0]->def_stmts[j],
1402 oprnds_info[1]->def_stmts[j]);
1403 dump_printf (MSG_NOTE, "%d ", j);
1405 dump_printf (MSG_NOTE, "\n");
1406 /* And try again with scratch 'matches' ... */
1407 bool *tem = XALLOCAVEC (bool, group_size);
1408 if ((child = vect_build_slp_tree (vinfo, oprnd_info->def_stmts,
1409 group_size, &this_max_nunits,
1410 &this_loads, tem, npermutes,
1411 &this_tree_size,
1412 max_tree_size)) != NULL)
1414 /* ... so if successful we can apply the operand swapping
1415 to the GIMPLE IL. This is necessary because for example
1416 vect_get_slp_defs uses operand indexes and thus expects
1417 canonical operand order. This is also necessary even
1418 if we end up building the operand from scalars as
1419 we'll continue to process swapped operand two. */
1420 for (j = 0; j < group_size; ++j)
1422 gimple *stmt = stmts[j];
1423 gimple_set_plf (stmt, GF_PLF_1, false);
1425 for (j = 0; j < group_size; ++j)
1427 gimple *stmt = stmts[j];
1428 if (matches[j] == !swap_not_matching)
1430 /* Avoid swapping operands twice. */
1431 if (gimple_plf (stmt, GF_PLF_1))
1432 continue;
1433 swap_ssa_operands (stmt, gimple_assign_rhs1_ptr (stmt),
1434 gimple_assign_rhs2_ptr (stmt));
1435 gimple_set_plf (stmt, GF_PLF_1, true);
1438 /* Verify we swap all duplicates or none. */
1439 if (flag_checking)
1440 for (j = 0; j < group_size; ++j)
1442 gimple *stmt = stmts[j];
1443 gcc_assert (gimple_plf (stmt, GF_PLF_1)
1444 == (matches[j] == !swap_not_matching));
1447 /* If we have all children of child built up from scalars then
1448 just throw that away and build it up this node from scalars. */
1449 if (!SLP_TREE_CHILDREN (child).is_empty ()
1450 /* ??? Rejecting patterns this way doesn't work. We'd have
1451 to do extra work to cancel the pattern so the uses see the
1452 scalar version. */
1453 && !is_pattern_stmt_p
1454 (vinfo_for_stmt (SLP_TREE_SCALAR_STMTS (child)[0])))
1456 unsigned int j;
1457 slp_tree grandchild;
1459 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (child), j, grandchild)
1460 if (SLP_TREE_DEF_TYPE (grandchild) == vect_internal_def)
1461 break;
1462 if (!grandchild)
1464 /* Roll back. */
1465 this_loads.truncate (old_nloads);
1466 this_tree_size = old_tree_size;
1467 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (child), j, grandchild)
1468 vect_free_slp_tree (grandchild);
1469 SLP_TREE_CHILDREN (child).truncate (0);
1471 dump_printf_loc (MSG_NOTE, vect_location,
1472 "Building parent vector operands from "
1473 "scalars instead\n");
1474 oprnd_info->def_stmts = vNULL;
1475 SLP_TREE_DEF_TYPE (child) = vect_external_def;
1476 children.safe_push (child);
1477 continue;
1481 oprnd_info->def_stmts = vNULL;
1482 children.safe_push (child);
1483 continue;
1486 ++*npermutes;
1489 fail:
1490 gcc_assert (child == NULL);
1491 FOR_EACH_VEC_ELT (children, j, child)
1492 vect_free_slp_tree (child);
1493 vect_free_oprnd_info (oprnds_info);
1494 return NULL;
1497 vect_free_oprnd_info (oprnds_info);
1499 if (tree_size)
1500 *tree_size += this_tree_size;
1501 *max_nunits = this_max_nunits;
1502 loads->safe_splice (this_loads);
1504 node = vect_create_new_slp_node (stmts);
1505 SLP_TREE_TWO_OPERATORS (node) = two_operators;
1506 SLP_TREE_CHILDREN (node).splice (children);
1507 return node;
1510 /* Dump a slp tree NODE using flags specified in DUMP_KIND. */
1512 static void
1513 vect_print_slp_tree (dump_flags_t dump_kind, location_t loc, slp_tree node)
1515 int i;
1516 gimple *stmt;
1517 slp_tree child;
1519 dump_printf_loc (dump_kind, loc, "node%s\n",
1520 SLP_TREE_DEF_TYPE (node) != vect_internal_def
1521 ? " (external)" : "");
1522 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, stmt)
1524 dump_printf_loc (dump_kind, loc, "\tstmt %d ", i);
1525 dump_gimple_stmt (dump_kind, TDF_SLIM, stmt, 0);
1527 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
1528 vect_print_slp_tree (dump_kind, loc, child);
1532 /* Mark the tree rooted at NODE with MARK (PURE_SLP or HYBRID).
1533 If MARK is HYBRID, it refers to a specific stmt in NODE (the stmt at index
1534 J). Otherwise, MARK is PURE_SLP and J is -1, which indicates that all the
1535 stmts in NODE are to be marked. */
1537 static void
1538 vect_mark_slp_stmts (slp_tree node, enum slp_vect_type mark, int j)
1540 int i;
1541 gimple *stmt;
1542 slp_tree child;
1544 if (SLP_TREE_DEF_TYPE (node) != vect_internal_def)
1545 return;
1547 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, stmt)
1548 if (j < 0 || i == j)
1549 STMT_SLP_TYPE (vinfo_for_stmt (stmt)) = mark;
1551 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
1552 vect_mark_slp_stmts (child, mark, j);
1556 /* Mark the statements of the tree rooted at NODE as relevant (vect_used). */
1558 static void
1559 vect_mark_slp_stmts_relevant (slp_tree node)
1561 int i;
1562 gimple *stmt;
1563 stmt_vec_info stmt_info;
1564 slp_tree child;
1566 if (SLP_TREE_DEF_TYPE (node) != vect_internal_def)
1567 return;
1569 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, stmt)
1571 stmt_info = vinfo_for_stmt (stmt);
1572 gcc_assert (!STMT_VINFO_RELEVANT (stmt_info)
1573 || STMT_VINFO_RELEVANT (stmt_info) == vect_used_in_scope);
1574 STMT_VINFO_RELEVANT (stmt_info) = vect_used_in_scope;
1577 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
1578 vect_mark_slp_stmts_relevant (child);
1582 /* Rearrange the statements of NODE according to PERMUTATION. */
1584 static void
1585 vect_slp_rearrange_stmts (slp_tree node, unsigned int group_size,
1586 vec<unsigned> permutation)
1588 gimple *stmt;
1589 vec<gimple *> tmp_stmts;
1590 unsigned int i;
1591 slp_tree child;
1593 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
1594 vect_slp_rearrange_stmts (child, group_size, permutation);
1596 gcc_assert (group_size == SLP_TREE_SCALAR_STMTS (node).length ());
1597 tmp_stmts.create (group_size);
1598 tmp_stmts.quick_grow_cleared (group_size);
1600 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, stmt)
1601 tmp_stmts[permutation[i]] = stmt;
1603 SLP_TREE_SCALAR_STMTS (node).release ();
1604 SLP_TREE_SCALAR_STMTS (node) = tmp_stmts;
1608 /* Attempt to reorder stmts in a reduction chain so that we don't
1609 require any load permutation. Return true if that was possible,
1610 otherwise return false. */
1612 static bool
1613 vect_attempt_slp_rearrange_stmts (slp_instance slp_instn)
1615 unsigned int group_size = SLP_INSTANCE_GROUP_SIZE (slp_instn);
1616 unsigned int i, j;
1617 unsigned int lidx;
1618 slp_tree node, load;
1620 /* Compare all the permutation sequences to the first one. We know
1621 that at least one load is permuted. */
1622 node = SLP_INSTANCE_LOADS (slp_instn)[0];
1623 if (!node->load_permutation.exists ())
1624 return false;
1625 for (i = 1; SLP_INSTANCE_LOADS (slp_instn).iterate (i, &load); ++i)
1627 if (!load->load_permutation.exists ())
1628 return false;
1629 FOR_EACH_VEC_ELT (load->load_permutation, j, lidx)
1630 if (lidx != node->load_permutation[j])
1631 return false;
1634 /* Check that the loads in the first sequence are different and there
1635 are no gaps between them. */
1636 auto_sbitmap load_index (group_size);
1637 bitmap_clear (load_index);
1638 FOR_EACH_VEC_ELT (node->load_permutation, i, lidx)
1640 if (lidx >= group_size)
1641 return false;
1642 if (bitmap_bit_p (load_index, lidx))
1643 return false;
1645 bitmap_set_bit (load_index, lidx);
1647 for (i = 0; i < group_size; i++)
1648 if (!bitmap_bit_p (load_index, i))
1649 return false;
1651 /* This permutation is valid for reduction. Since the order of the
1652 statements in the nodes is not important unless they are memory
1653 accesses, we can rearrange the statements in all the nodes
1654 according to the order of the loads. */
1655 vect_slp_rearrange_stmts (SLP_INSTANCE_TREE (slp_instn), group_size,
1656 node->load_permutation);
1658 /* We are done, no actual permutations need to be generated. */
1659 poly_uint64 unrolling_factor = SLP_INSTANCE_UNROLLING_FACTOR (slp_instn);
1660 FOR_EACH_VEC_ELT (SLP_INSTANCE_LOADS (slp_instn), i, node)
1662 gimple *first_stmt = SLP_TREE_SCALAR_STMTS (node)[0];
1663 first_stmt = GROUP_FIRST_ELEMENT (vinfo_for_stmt (first_stmt));
1664 /* But we have to keep those permutations that are required because
1665 of handling of gaps. */
1666 if (known_eq (unrolling_factor, 1U)
1667 || (group_size == GROUP_SIZE (vinfo_for_stmt (first_stmt))
1668 && GROUP_GAP (vinfo_for_stmt (first_stmt)) == 0))
1669 SLP_TREE_LOAD_PERMUTATION (node).release ();
1670 else
1671 for (j = 0; j < SLP_TREE_LOAD_PERMUTATION (node).length (); ++j)
1672 SLP_TREE_LOAD_PERMUTATION (node)[j] = j;
1675 return true;
1678 /* Check if the required load permutations in the SLP instance
1679 SLP_INSTN are supported. */
1681 static bool
1682 vect_supported_load_permutation_p (slp_instance slp_instn)
1684 unsigned int group_size = SLP_INSTANCE_GROUP_SIZE (slp_instn);
1685 unsigned int i, j, k, next;
1686 slp_tree node;
1687 gimple *stmt, *load, *next_load;
1689 if (dump_enabled_p ())
1691 dump_printf_loc (MSG_NOTE, vect_location, "Load permutation ");
1692 FOR_EACH_VEC_ELT (SLP_INSTANCE_LOADS (slp_instn), i, node)
1693 if (node->load_permutation.exists ())
1694 FOR_EACH_VEC_ELT (node->load_permutation, j, next)
1695 dump_printf (MSG_NOTE, "%d ", next);
1696 else
1697 for (k = 0; k < group_size; ++k)
1698 dump_printf (MSG_NOTE, "%d ", k);
1699 dump_printf (MSG_NOTE, "\n");
1702 /* In case of reduction every load permutation is allowed, since the order
1703 of the reduction statements is not important (as opposed to the case of
1704 grouped stores). The only condition we need to check is that all the
1705 load nodes are of the same size and have the same permutation (and then
1706 rearrange all the nodes of the SLP instance according to this
1707 permutation). */
1709 /* Check that all the load nodes are of the same size. */
1710 /* ??? Can't we assert this? */
1711 FOR_EACH_VEC_ELT (SLP_INSTANCE_LOADS (slp_instn), i, node)
1712 if (SLP_TREE_SCALAR_STMTS (node).length () != (unsigned) group_size)
1713 return false;
1715 node = SLP_INSTANCE_TREE (slp_instn);
1716 stmt = SLP_TREE_SCALAR_STMTS (node)[0];
1718 /* Reduction (there are no data-refs in the root).
1719 In reduction chain the order of the loads is not important. */
1720 if (!STMT_VINFO_DATA_REF (vinfo_for_stmt (stmt))
1721 && !GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt)))
1722 vect_attempt_slp_rearrange_stmts (slp_instn);
1724 /* In basic block vectorization we allow any subchain of an interleaving
1725 chain.
1726 FORNOW: not supported in loop SLP because of realignment compications. */
1727 if (STMT_VINFO_BB_VINFO (vinfo_for_stmt (stmt)))
1729 /* Check whether the loads in an instance form a subchain and thus
1730 no permutation is necessary. */
1731 FOR_EACH_VEC_ELT (SLP_INSTANCE_LOADS (slp_instn), i, node)
1733 if (!SLP_TREE_LOAD_PERMUTATION (node).exists ())
1734 continue;
1735 bool subchain_p = true;
1736 next_load = NULL;
1737 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), j, load)
1739 if (j != 0
1740 && (next_load != load
1741 || GROUP_GAP (vinfo_for_stmt (load)) != 1))
1743 subchain_p = false;
1744 break;
1746 next_load = GROUP_NEXT_ELEMENT (vinfo_for_stmt (load));
1748 if (subchain_p)
1749 SLP_TREE_LOAD_PERMUTATION (node).release ();
1750 else
1752 stmt_vec_info group_info
1753 = vinfo_for_stmt (SLP_TREE_SCALAR_STMTS (node)[0]);
1754 group_info = vinfo_for_stmt (GROUP_FIRST_ELEMENT (group_info));
1755 unsigned HOST_WIDE_INT nunits;
1756 unsigned k, maxk = 0;
1757 FOR_EACH_VEC_ELT (SLP_TREE_LOAD_PERMUTATION (node), j, k)
1758 if (k > maxk)
1759 maxk = k;
1760 /* In BB vectorization we may not actually use a loaded vector
1761 accessing elements in excess of GROUP_SIZE. */
1762 tree vectype = STMT_VINFO_VECTYPE (group_info);
1763 if (!TYPE_VECTOR_SUBPARTS (vectype).is_constant (&nunits)
1764 || maxk >= (GROUP_SIZE (group_info) & ~(nunits - 1)))
1766 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
1767 "BB vectorization with gaps at the end of "
1768 "a load is not supported\n");
1769 return false;
1772 /* Verify the permutation can be generated. */
1773 vec<tree> tem;
1774 unsigned n_perms;
1775 if (!vect_transform_slp_perm_load (node, tem, NULL,
1776 1, slp_instn, true, &n_perms))
1778 dump_printf_loc (MSG_MISSED_OPTIMIZATION,
1779 vect_location,
1780 "unsupported load permutation\n");
1781 return false;
1785 return true;
1788 /* For loop vectorization verify we can generate the permutation. Be
1789 conservative about the vectorization factor, there are permutations
1790 that will use three vector inputs only starting from a specific factor
1791 and the vectorization factor is not yet final.
1792 ??? The SLP instance unrolling factor might not be the maximum one. */
1793 unsigned n_perms;
1794 poly_uint64 test_vf
1795 = force_common_multiple (SLP_INSTANCE_UNROLLING_FACTOR (slp_instn),
1796 LOOP_VINFO_VECT_FACTOR
1797 (STMT_VINFO_LOOP_VINFO (vinfo_for_stmt (stmt))));
1798 FOR_EACH_VEC_ELT (SLP_INSTANCE_LOADS (slp_instn), i, node)
1799 if (node->load_permutation.exists ()
1800 && !vect_transform_slp_perm_load (node, vNULL, NULL, test_vf,
1801 slp_instn, true, &n_perms))
1802 return false;
1804 return true;
1808 /* Find the last store in SLP INSTANCE. */
1810 gimple *
1811 vect_find_last_scalar_stmt_in_slp (slp_tree node)
1813 gimple *last = NULL, *stmt;
1815 for (int i = 0; SLP_TREE_SCALAR_STMTS (node).iterate (i, &stmt); i++)
1817 stmt_vec_info stmt_vinfo = vinfo_for_stmt (stmt);
1818 if (is_pattern_stmt_p (stmt_vinfo))
1819 last = get_later_stmt (STMT_VINFO_RELATED_STMT (stmt_vinfo), last);
1820 else
1821 last = get_later_stmt (stmt, last);
1824 return last;
1827 /* Splits a group of stores, currently beginning at FIRST_STMT, into two groups:
1828 one (still beginning at FIRST_STMT) of size GROUP1_SIZE (also containing
1829 the first GROUP1_SIZE stmts, since stores are consecutive), the second
1830 containing the remainder.
1831 Return the first stmt in the second group. */
1833 static gimple *
1834 vect_split_slp_store_group (gimple *first_stmt, unsigned group1_size)
1836 stmt_vec_info first_vinfo = vinfo_for_stmt (first_stmt);
1837 gcc_assert (GROUP_FIRST_ELEMENT (first_vinfo) == first_stmt);
1838 gcc_assert (group1_size > 0);
1839 int group2_size = GROUP_SIZE (first_vinfo) - group1_size;
1840 gcc_assert (group2_size > 0);
1841 GROUP_SIZE (first_vinfo) = group1_size;
1843 gimple *stmt = first_stmt;
1844 for (unsigned i = group1_size; i > 1; i--)
1846 stmt = GROUP_NEXT_ELEMENT (vinfo_for_stmt (stmt));
1847 gcc_assert (GROUP_GAP (vinfo_for_stmt (stmt)) == 1);
1849 /* STMT is now the last element of the first group. */
1850 gimple *group2 = GROUP_NEXT_ELEMENT (vinfo_for_stmt (stmt));
1851 GROUP_NEXT_ELEMENT (vinfo_for_stmt (stmt)) = 0;
1853 GROUP_SIZE (vinfo_for_stmt (group2)) = group2_size;
1854 for (stmt = group2; stmt; stmt = GROUP_NEXT_ELEMENT (vinfo_for_stmt (stmt)))
1856 GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt)) = group2;
1857 gcc_assert (GROUP_GAP (vinfo_for_stmt (stmt)) == 1);
1860 /* For the second group, the GROUP_GAP is that before the original group,
1861 plus skipping over the first vector. */
1862 GROUP_GAP (vinfo_for_stmt (group2)) =
1863 GROUP_GAP (first_vinfo) + group1_size;
1865 /* GROUP_GAP of the first group now has to skip over the second group too. */
1866 GROUP_GAP (first_vinfo) += group2_size;
1868 if (dump_enabled_p ())
1869 dump_printf_loc (MSG_NOTE, vect_location, "Split group into %d and %d\n",
1870 group1_size, group2_size);
1872 return group2;
1875 /* Calculate the unrolling factor for an SLP instance with GROUP_SIZE
1876 statements and a vector of NUNITS elements. */
1878 static poly_uint64
1879 calculate_unrolling_factor (poly_uint64 nunits, unsigned int group_size)
1881 return exact_div (common_multiple (nunits, group_size), group_size);
1884 /* Analyze an SLP instance starting from a group of grouped stores. Call
1885 vect_build_slp_tree to build a tree of packed stmts if possible.
1886 Return FALSE if it's impossible to SLP any stmt in the loop. */
1888 static bool
1889 vect_analyze_slp_instance (vec_info *vinfo,
1890 gimple *stmt, unsigned max_tree_size)
1892 slp_instance new_instance;
1893 slp_tree node;
1894 unsigned int group_size = GROUP_SIZE (vinfo_for_stmt (stmt));
1895 tree vectype, scalar_type = NULL_TREE;
1896 gimple *next;
1897 unsigned int i;
1898 vec<slp_tree> loads;
1899 struct data_reference *dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (stmt));
1900 vec<gimple *> scalar_stmts;
1902 if (GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt)))
1904 if (dr)
1906 scalar_type = TREE_TYPE (DR_REF (dr));
1907 vectype = get_vectype_for_scalar_type (scalar_type);
1909 else
1911 gcc_assert (is_a <loop_vec_info> (vinfo));
1912 vectype = STMT_VINFO_VECTYPE (vinfo_for_stmt (stmt));
1915 group_size = GROUP_SIZE (vinfo_for_stmt (stmt));
1917 else
1919 gcc_assert (is_a <loop_vec_info> (vinfo));
1920 vectype = STMT_VINFO_VECTYPE (vinfo_for_stmt (stmt));
1921 group_size = as_a <loop_vec_info> (vinfo)->reductions.length ();
1924 if (!vectype)
1926 if (dump_enabled_p ())
1928 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
1929 "Build SLP failed: unsupported data-type ");
1930 dump_generic_expr (MSG_MISSED_OPTIMIZATION, TDF_SLIM, scalar_type);
1931 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
1934 return false;
1936 poly_uint64 nunits = TYPE_VECTOR_SUBPARTS (vectype);
1938 /* Create a node (a root of the SLP tree) for the packed grouped stores. */
1939 scalar_stmts.create (group_size);
1940 next = stmt;
1941 if (GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt)))
1943 /* Collect the stores and store them in SLP_TREE_SCALAR_STMTS. */
1944 while (next)
1946 if (STMT_VINFO_IN_PATTERN_P (vinfo_for_stmt (next))
1947 && STMT_VINFO_RELATED_STMT (vinfo_for_stmt (next)))
1948 scalar_stmts.safe_push (
1949 STMT_VINFO_RELATED_STMT (vinfo_for_stmt (next)));
1950 else
1951 scalar_stmts.safe_push (next);
1952 next = GROUP_NEXT_ELEMENT (vinfo_for_stmt (next));
1954 /* Mark the first element of the reduction chain as reduction to properly
1955 transform the node. In the reduction analysis phase only the last
1956 element of the chain is marked as reduction. */
1957 if (!STMT_VINFO_GROUPED_ACCESS (vinfo_for_stmt (stmt)))
1958 STMT_VINFO_DEF_TYPE (vinfo_for_stmt (stmt)) = vect_reduction_def;
1960 else
1962 /* Collect reduction statements. */
1963 vec<gimple *> reductions = as_a <loop_vec_info> (vinfo)->reductions;
1964 for (i = 0; reductions.iterate (i, &next); i++)
1965 scalar_stmts.safe_push (next);
1968 loads.create (group_size);
1970 /* Build the tree for the SLP instance. */
1971 bool *matches = XALLOCAVEC (bool, group_size);
1972 unsigned npermutes = 0;
1973 bst_fail = new scalar_stmts_set_t ();
1974 poly_uint64 max_nunits = nunits;
1975 node = vect_build_slp_tree (vinfo, scalar_stmts, group_size,
1976 &max_nunits, &loads, matches, &npermutes,
1977 NULL, max_tree_size);
1978 delete bst_fail;
1979 if (node != NULL)
1981 /* Calculate the unrolling factor based on the smallest type. */
1982 poly_uint64 unrolling_factor
1983 = calculate_unrolling_factor (max_nunits, group_size);
1985 if (maybe_ne (unrolling_factor, 1U)
1986 && is_a <bb_vec_info> (vinfo))
1988 unsigned HOST_WIDE_INT const_max_nunits;
1989 if (!max_nunits.is_constant (&const_max_nunits)
1990 || const_max_nunits > group_size)
1992 if (dump_enabled_p ())
1993 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
1994 "Build SLP failed: store group "
1995 "size not a multiple of the vector size "
1996 "in basic block SLP\n");
1997 vect_free_slp_tree (node);
1998 loads.release ();
1999 return false;
2001 /* Fatal mismatch. */
2002 matches[group_size / const_max_nunits * const_max_nunits] = false;
2003 vect_free_slp_tree (node);
2004 loads.release ();
2006 else
2008 /* Create a new SLP instance. */
2009 new_instance = XNEW (struct _slp_instance);
2010 SLP_INSTANCE_TREE (new_instance) = node;
2011 SLP_INSTANCE_GROUP_SIZE (new_instance) = group_size;
2012 SLP_INSTANCE_UNROLLING_FACTOR (new_instance) = unrolling_factor;
2013 SLP_INSTANCE_LOADS (new_instance) = loads;
2015 /* Compute the load permutation. */
2016 slp_tree load_node;
2017 bool loads_permuted = false;
2018 FOR_EACH_VEC_ELT (loads, i, load_node)
2020 vec<unsigned> load_permutation;
2021 int j;
2022 gimple *load, *first_stmt;
2023 bool this_load_permuted = false;
2024 load_permutation.create (group_size);
2025 first_stmt = GROUP_FIRST_ELEMENT
2026 (vinfo_for_stmt (SLP_TREE_SCALAR_STMTS (load_node)[0]));
2027 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (load_node), j, load)
2029 int load_place = vect_get_place_in_interleaving_chain
2030 (load, first_stmt);
2031 gcc_assert (load_place != -1);
2032 if (load_place != j)
2033 this_load_permuted = true;
2034 load_permutation.safe_push (load_place);
2036 if (!this_load_permuted
2037 /* The load requires permutation when unrolling exposes
2038 a gap either because the group is larger than the SLP
2039 group-size or because there is a gap between the groups. */
2040 && (known_eq (unrolling_factor, 1U)
2041 || (group_size == GROUP_SIZE (vinfo_for_stmt (first_stmt))
2042 && GROUP_GAP (vinfo_for_stmt (first_stmt)) == 0)))
2044 load_permutation.release ();
2045 continue;
2047 SLP_TREE_LOAD_PERMUTATION (load_node) = load_permutation;
2048 loads_permuted = true;
2051 if (loads_permuted)
2053 if (!vect_supported_load_permutation_p (new_instance))
2055 if (dump_enabled_p ())
2057 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2058 "Build SLP failed: unsupported load "
2059 "permutation ");
2060 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION,
2061 TDF_SLIM, stmt, 0);
2063 vect_free_slp_instance (new_instance);
2064 return false;
2068 /* If the loads and stores can be handled with load/store-lan
2069 instructions do not generate this SLP instance. */
2070 if (is_a <loop_vec_info> (vinfo)
2071 && loads_permuted
2072 && dr && vect_store_lanes_supported (vectype, group_size, false))
2074 slp_tree load_node;
2075 FOR_EACH_VEC_ELT (loads, i, load_node)
2077 gimple *first_stmt = GROUP_FIRST_ELEMENT
2078 (vinfo_for_stmt (SLP_TREE_SCALAR_STMTS (load_node)[0]));
2079 stmt_vec_info stmt_vinfo = vinfo_for_stmt (first_stmt);
2080 /* Use SLP for strided accesses (or if we
2081 can't load-lanes). */
2082 if (STMT_VINFO_STRIDED_P (stmt_vinfo)
2083 || ! vect_load_lanes_supported
2084 (STMT_VINFO_VECTYPE (stmt_vinfo),
2085 GROUP_SIZE (stmt_vinfo), false))
2086 break;
2088 if (i == loads.length ())
2090 if (dump_enabled_p ())
2091 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2092 "Built SLP cancelled: can use "
2093 "load/store-lanes\n");
2094 vect_free_slp_instance (new_instance);
2095 return false;
2099 vinfo->slp_instances.safe_push (new_instance);
2101 if (dump_enabled_p ())
2103 dump_printf_loc (MSG_NOTE, vect_location,
2104 "Final SLP tree for instance:\n");
2105 vect_print_slp_tree (MSG_NOTE, vect_location, node);
2108 return true;
2111 else
2113 /* Failed to SLP. */
2114 /* Free the allocated memory. */
2115 scalar_stmts.release ();
2116 loads.release ();
2119 /* For basic block SLP, try to break the group up into multiples of the
2120 vector size. */
2121 unsigned HOST_WIDE_INT const_nunits;
2122 if (is_a <bb_vec_info> (vinfo)
2123 && GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt))
2124 && STMT_VINFO_GROUPED_ACCESS (vinfo_for_stmt (stmt))
2125 && nunits.is_constant (&const_nunits))
2127 /* We consider breaking the group only on VF boundaries from the existing
2128 start. */
2129 for (i = 0; i < group_size; i++)
2130 if (!matches[i]) break;
2132 if (i >= const_nunits && i < group_size)
2134 /* Split into two groups at the first vector boundary before i. */
2135 gcc_assert ((const_nunits & (const_nunits - 1)) == 0);
2136 unsigned group1_size = i & ~(const_nunits - 1);
2138 gimple *rest = vect_split_slp_store_group (stmt, group1_size);
2139 bool res = vect_analyze_slp_instance (vinfo, stmt, max_tree_size);
2140 /* If the first non-match was in the middle of a vector,
2141 skip the rest of that vector. */
2142 if (group1_size < i)
2144 i = group1_size + const_nunits;
2145 if (i < group_size)
2146 rest = vect_split_slp_store_group (rest, const_nunits);
2148 if (i < group_size)
2149 res |= vect_analyze_slp_instance (vinfo, rest, max_tree_size);
2150 return res;
2152 /* Even though the first vector did not all match, we might be able to SLP
2153 (some) of the remainder. FORNOW ignore this possibility. */
2156 return false;
2160 /* Check if there are stmts in the loop can be vectorized using SLP. Build SLP
2161 trees of packed scalar stmts if SLP is possible. */
2163 bool
2164 vect_analyze_slp (vec_info *vinfo, unsigned max_tree_size)
2166 unsigned int i;
2167 gimple *first_element;
2169 if (dump_enabled_p ())
2170 dump_printf_loc (MSG_NOTE, vect_location, "=== vect_analyze_slp ===\n");
2172 /* Find SLP sequences starting from groups of grouped stores. */
2173 FOR_EACH_VEC_ELT (vinfo->grouped_stores, i, first_element)
2174 vect_analyze_slp_instance (vinfo, first_element, max_tree_size);
2176 if (loop_vec_info loop_vinfo = dyn_cast <loop_vec_info> (vinfo))
2178 if (loop_vinfo->reduction_chains.length () > 0)
2180 /* Find SLP sequences starting from reduction chains. */
2181 FOR_EACH_VEC_ELT (loop_vinfo->reduction_chains, i, first_element)
2182 if (! vect_analyze_slp_instance (vinfo, first_element,
2183 max_tree_size))
2185 /* Dissolve reduction chain group. */
2186 gimple *next, *stmt = first_element;
2187 while (stmt)
2189 stmt_vec_info vinfo = vinfo_for_stmt (stmt);
2190 next = GROUP_NEXT_ELEMENT (vinfo);
2191 GROUP_FIRST_ELEMENT (vinfo) = NULL;
2192 GROUP_NEXT_ELEMENT (vinfo) = NULL;
2193 stmt = next;
2195 STMT_VINFO_DEF_TYPE (vinfo_for_stmt (first_element))
2196 = vect_internal_def;
2200 /* Find SLP sequences starting from groups of reductions. */
2201 if (loop_vinfo->reductions.length () > 1)
2202 vect_analyze_slp_instance (vinfo, loop_vinfo->reductions[0],
2203 max_tree_size);
2206 return true;
2210 /* For each possible SLP instance decide whether to SLP it and calculate overall
2211 unrolling factor needed to SLP the loop. Return TRUE if decided to SLP at
2212 least one instance. */
2214 bool
2215 vect_make_slp_decision (loop_vec_info loop_vinfo)
2217 unsigned int i;
2218 poly_uint64 unrolling_factor = 1;
2219 vec<slp_instance> slp_instances = LOOP_VINFO_SLP_INSTANCES (loop_vinfo);
2220 slp_instance instance;
2221 int decided_to_slp = 0;
2223 if (dump_enabled_p ())
2224 dump_printf_loc (MSG_NOTE, vect_location, "=== vect_make_slp_decision ==="
2225 "\n");
2227 FOR_EACH_VEC_ELT (slp_instances, i, instance)
2229 /* FORNOW: SLP if you can. */
2230 /* All unroll factors have the form current_vector_size * X for some
2231 rational X, so they must have a common multiple. */
2232 unrolling_factor
2233 = force_common_multiple (unrolling_factor,
2234 SLP_INSTANCE_UNROLLING_FACTOR (instance));
2236 /* Mark all the stmts that belong to INSTANCE as PURE_SLP stmts. Later we
2237 call vect_detect_hybrid_slp () to find stmts that need hybrid SLP and
2238 loop-based vectorization. Such stmts will be marked as HYBRID. */
2239 vect_mark_slp_stmts (SLP_INSTANCE_TREE (instance), pure_slp, -1);
2240 decided_to_slp++;
2243 LOOP_VINFO_SLP_UNROLLING_FACTOR (loop_vinfo) = unrolling_factor;
2245 if (decided_to_slp && dump_enabled_p ())
2247 dump_printf_loc (MSG_NOTE, vect_location,
2248 "Decided to SLP %d instances. Unrolling factor ",
2249 decided_to_slp);
2250 dump_dec (MSG_NOTE, unrolling_factor);
2251 dump_printf (MSG_NOTE, "\n");
2254 return (decided_to_slp > 0);
2258 /* Find stmts that must be both vectorized and SLPed (since they feed stmts that
2259 can't be SLPed) in the tree rooted at NODE. Mark such stmts as HYBRID. */
2261 static void
2262 vect_detect_hybrid_slp_stmts (slp_tree node, unsigned i, slp_vect_type stype)
2264 gimple *stmt = SLP_TREE_SCALAR_STMTS (node)[i];
2265 imm_use_iterator imm_iter;
2266 gimple *use_stmt;
2267 stmt_vec_info use_vinfo, stmt_vinfo = vinfo_for_stmt (stmt);
2268 slp_tree child;
2269 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_vinfo);
2270 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
2271 int j;
2273 /* Propagate hybrid down the SLP tree. */
2274 if (stype == hybrid)
2276 else if (HYBRID_SLP_STMT (stmt_vinfo))
2277 stype = hybrid;
2278 else
2280 /* Check if a pure SLP stmt has uses in non-SLP stmts. */
2281 gcc_checking_assert (PURE_SLP_STMT (stmt_vinfo));
2282 /* If we get a pattern stmt here we have to use the LHS of the
2283 original stmt for immediate uses. */
2284 if (! STMT_VINFO_IN_PATTERN_P (stmt_vinfo)
2285 && STMT_VINFO_RELATED_STMT (stmt_vinfo))
2286 stmt = STMT_VINFO_RELATED_STMT (stmt_vinfo);
2287 tree def;
2288 if (gimple_code (stmt) == GIMPLE_PHI)
2289 def = gimple_phi_result (stmt);
2290 else
2291 def = SINGLE_SSA_TREE_OPERAND (stmt, SSA_OP_DEF);
2292 if (def)
2293 FOR_EACH_IMM_USE_STMT (use_stmt, imm_iter, def)
2295 if (!flow_bb_inside_loop_p (loop, gimple_bb (use_stmt)))
2296 continue;
2297 use_vinfo = vinfo_for_stmt (use_stmt);
2298 if (STMT_VINFO_IN_PATTERN_P (use_vinfo)
2299 && STMT_VINFO_RELATED_STMT (use_vinfo))
2300 use_vinfo = vinfo_for_stmt (STMT_VINFO_RELATED_STMT (use_vinfo));
2301 if (!STMT_SLP_TYPE (use_vinfo)
2302 && (STMT_VINFO_RELEVANT (use_vinfo)
2303 || VECTORIZABLE_CYCLE_DEF (STMT_VINFO_DEF_TYPE (use_vinfo)))
2304 && !(gimple_code (use_stmt) == GIMPLE_PHI
2305 && STMT_VINFO_DEF_TYPE (use_vinfo) == vect_reduction_def))
2307 if (dump_enabled_p ())
2309 dump_printf_loc (MSG_NOTE, vect_location, "use of SLP "
2310 "def in non-SLP stmt: ");
2311 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, use_stmt, 0);
2313 stype = hybrid;
2318 if (stype == hybrid
2319 && !HYBRID_SLP_STMT (stmt_vinfo))
2321 if (dump_enabled_p ())
2323 dump_printf_loc (MSG_NOTE, vect_location, "marking hybrid: ");
2324 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
2326 STMT_SLP_TYPE (stmt_vinfo) = hybrid;
2329 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), j, child)
2330 if (SLP_TREE_DEF_TYPE (child) != vect_external_def)
2331 vect_detect_hybrid_slp_stmts (child, i, stype);
2334 /* Helpers for vect_detect_hybrid_slp walking pattern stmt uses. */
2336 static tree
2337 vect_detect_hybrid_slp_1 (tree *tp, int *, void *data)
2339 walk_stmt_info *wi = (walk_stmt_info *)data;
2340 struct loop *loopp = (struct loop *)wi->info;
2342 if (wi->is_lhs)
2343 return NULL_TREE;
2345 if (TREE_CODE (*tp) == SSA_NAME
2346 && !SSA_NAME_IS_DEFAULT_DEF (*tp))
2348 gimple *def_stmt = SSA_NAME_DEF_STMT (*tp);
2349 if (flow_bb_inside_loop_p (loopp, gimple_bb (def_stmt))
2350 && PURE_SLP_STMT (vinfo_for_stmt (def_stmt)))
2352 if (dump_enabled_p ())
2354 dump_printf_loc (MSG_NOTE, vect_location, "marking hybrid: ");
2355 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, def_stmt, 0);
2357 STMT_SLP_TYPE (vinfo_for_stmt (def_stmt)) = hybrid;
2361 return NULL_TREE;
2364 static tree
2365 vect_detect_hybrid_slp_2 (gimple_stmt_iterator *gsi, bool *handled,
2366 walk_stmt_info *)
2368 stmt_vec_info use_vinfo = vinfo_for_stmt (gsi_stmt (*gsi));
2369 /* If the stmt is in a SLP instance then this isn't a reason
2370 to mark use definitions in other SLP instances as hybrid. */
2371 if (! STMT_SLP_TYPE (use_vinfo)
2372 && (STMT_VINFO_RELEVANT (use_vinfo)
2373 || VECTORIZABLE_CYCLE_DEF (STMT_VINFO_DEF_TYPE (use_vinfo)))
2374 && ! (gimple_code (gsi_stmt (*gsi)) == GIMPLE_PHI
2375 && STMT_VINFO_DEF_TYPE (use_vinfo) == vect_reduction_def))
2377 else
2378 *handled = true;
2379 return NULL_TREE;
2382 /* Find stmts that must be both vectorized and SLPed. */
2384 void
2385 vect_detect_hybrid_slp (loop_vec_info loop_vinfo)
2387 unsigned int i;
2388 vec<slp_instance> slp_instances = LOOP_VINFO_SLP_INSTANCES (loop_vinfo);
2389 slp_instance instance;
2391 if (dump_enabled_p ())
2392 dump_printf_loc (MSG_NOTE, vect_location, "=== vect_detect_hybrid_slp ==="
2393 "\n");
2395 /* First walk all pattern stmt in the loop and mark defs of uses as
2396 hybrid because immediate uses in them are not recorded. */
2397 for (i = 0; i < LOOP_VINFO_LOOP (loop_vinfo)->num_nodes; ++i)
2399 basic_block bb = LOOP_VINFO_BBS (loop_vinfo)[i];
2400 for (gimple_stmt_iterator gsi = gsi_start_bb (bb); !gsi_end_p (gsi);
2401 gsi_next (&gsi))
2403 gimple *stmt = gsi_stmt (gsi);
2404 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
2405 if (STMT_VINFO_IN_PATTERN_P (stmt_info))
2407 walk_stmt_info wi;
2408 memset (&wi, 0, sizeof (wi));
2409 wi.info = LOOP_VINFO_LOOP (loop_vinfo);
2410 gimple_stmt_iterator gsi2
2411 = gsi_for_stmt (STMT_VINFO_RELATED_STMT (stmt_info));
2412 walk_gimple_stmt (&gsi2, vect_detect_hybrid_slp_2,
2413 vect_detect_hybrid_slp_1, &wi);
2414 walk_gimple_seq (STMT_VINFO_PATTERN_DEF_SEQ (stmt_info),
2415 vect_detect_hybrid_slp_2,
2416 vect_detect_hybrid_slp_1, &wi);
2421 /* Then walk the SLP instance trees marking stmts with uses in
2422 non-SLP stmts as hybrid, also propagating hybrid down the
2423 SLP tree, collecting the above info on-the-fly. */
2424 FOR_EACH_VEC_ELT (slp_instances, i, instance)
2426 for (unsigned i = 0; i < SLP_INSTANCE_GROUP_SIZE (instance); ++i)
2427 vect_detect_hybrid_slp_stmts (SLP_INSTANCE_TREE (instance),
2428 i, pure_slp);
2433 /* Initialize a bb_vec_info struct for the statements between
2434 REGION_BEGIN_IN (inclusive) and REGION_END_IN (exclusive). */
2436 _bb_vec_info::_bb_vec_info (gimple_stmt_iterator region_begin_in,
2437 gimple_stmt_iterator region_end_in)
2438 : vec_info (vec_info::bb, init_cost (NULL)),
2439 bb (gsi_bb (region_begin_in)),
2440 region_begin (region_begin_in),
2441 region_end (region_end_in)
2443 gimple_stmt_iterator gsi;
2445 for (gsi = region_begin; gsi_stmt (gsi) != gsi_stmt (region_end);
2446 gsi_next (&gsi))
2448 gimple *stmt = gsi_stmt (gsi);
2449 gimple_set_uid (stmt, 0);
2450 set_vinfo_for_stmt (stmt, new_stmt_vec_info (stmt, this));
2453 bb->aux = this;
2457 /* Free BB_VINFO struct, as well as all the stmt_vec_info structs of all the
2458 stmts in the basic block. */
2460 _bb_vec_info::~_bb_vec_info ()
2462 for (gimple_stmt_iterator si = region_begin;
2463 gsi_stmt (si) != gsi_stmt (region_end); gsi_next (&si))
2465 gimple *stmt = gsi_stmt (si);
2466 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
2468 if (stmt_info)
2469 /* Free stmt_vec_info. */
2470 free_stmt_vec_info (stmt);
2472 /* Reset region marker. */
2473 gimple_set_uid (stmt, -1);
2476 bb->aux = NULL;
2479 /* Subroutine of vect_slp_analyze_node_operations. Handle the root of NODE,
2480 given then that child nodes have already been processed, and that
2481 their def types currently match their SLP node's def type. */
2483 static bool
2484 vect_slp_analyze_node_operations_1 (vec_info *vinfo, slp_tree node,
2485 slp_instance node_instance,
2486 stmt_vector_for_cost *cost_vec)
2488 gimple *stmt = SLP_TREE_SCALAR_STMTS (node)[0];
2489 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
2490 gcc_assert (stmt_info);
2491 gcc_assert (STMT_SLP_TYPE (stmt_info) != loop_vect);
2493 /* For BB vectorization vector types are assigned here.
2494 Memory accesses already got their vector type assigned
2495 in vect_analyze_data_refs. */
2496 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
2497 if (bb_vinfo
2498 && ! STMT_VINFO_DATA_REF (stmt_info))
2500 tree vectype, nunits_vectype;
2501 if (!vect_get_vector_types_for_stmt (stmt_info, &vectype,
2502 &nunits_vectype))
2503 /* We checked this when building the node. */
2504 gcc_unreachable ();
2505 if (vectype == boolean_type_node)
2507 vectype = vect_get_mask_type_for_stmt (stmt_info);
2508 if (!vectype)
2509 /* vect_get_mask_type_for_stmt has already explained the
2510 failure. */
2511 return false;
2514 gimple *sstmt;
2515 unsigned int i;
2516 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, sstmt)
2517 STMT_VINFO_VECTYPE (vinfo_for_stmt (sstmt)) = vectype;
2520 /* Calculate the number of vector statements to be created for the
2521 scalar stmts in this node. For SLP reductions it is equal to the
2522 number of vector statements in the children (which has already been
2523 calculated by the recursive call). Otherwise it is the number of
2524 scalar elements in one scalar iteration (GROUP_SIZE) multiplied by
2525 VF divided by the number of elements in a vector. */
2526 if (GROUP_FIRST_ELEMENT (stmt_info)
2527 && !STMT_VINFO_GROUPED_ACCESS (stmt_info))
2528 SLP_TREE_NUMBER_OF_VEC_STMTS (node)
2529 = SLP_TREE_NUMBER_OF_VEC_STMTS (SLP_TREE_CHILDREN (node)[0]);
2530 else
2532 poly_uint64 vf;
2533 if (loop_vec_info loop_vinfo = dyn_cast <loop_vec_info> (vinfo))
2534 vf = loop_vinfo->vectorization_factor;
2535 else
2536 vf = 1;
2537 unsigned int group_size = SLP_INSTANCE_GROUP_SIZE (node_instance);
2538 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
2539 SLP_TREE_NUMBER_OF_VEC_STMTS (node)
2540 = vect_get_num_vectors (vf * group_size, vectype);
2543 bool dummy;
2544 return vect_analyze_stmt (stmt, &dummy, node, node_instance, cost_vec);
2547 /* Analyze statements contained in SLP tree NODE after recursively analyzing
2548 the subtree. NODE_INSTANCE contains NODE and VINFO contains INSTANCE.
2550 Return true if the operations are supported. */
2552 static bool
2553 vect_slp_analyze_node_operations (vec_info *vinfo, slp_tree node,
2554 slp_instance node_instance,
2555 scalar_stmts_to_slp_tree_map_t *visited,
2556 scalar_stmts_to_slp_tree_map_t *lvisited,
2557 stmt_vector_for_cost *cost_vec)
2559 int i, j;
2560 slp_tree child;
2562 if (SLP_TREE_DEF_TYPE (node) != vect_internal_def)
2563 return true;
2565 /* If we already analyzed the exact same set of scalar stmts we're done.
2566 We share the generated vector stmts for those. */
2567 slp_tree *leader;
2568 if ((leader = visited->get (SLP_TREE_SCALAR_STMTS (node)))
2569 || (leader = lvisited->get (SLP_TREE_SCALAR_STMTS (node))))
2571 SLP_TREE_NUMBER_OF_VEC_STMTS (node)
2572 = SLP_TREE_NUMBER_OF_VEC_STMTS (*leader);
2573 return true;
2576 /* The SLP graph is acyclic so not caching whether we failed or succeeded
2577 doesn't result in any issue since we throw away the lvisited set
2578 when we fail. */
2579 lvisited->put (SLP_TREE_SCALAR_STMTS (node).copy (), node);
2581 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
2582 if (!vect_slp_analyze_node_operations (vinfo, child, node_instance,
2583 visited, lvisited, cost_vec))
2584 return false;
2586 /* Push SLP node def-type to stmt operands. */
2587 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), j, child)
2588 if (SLP_TREE_DEF_TYPE (child) != vect_internal_def)
2589 STMT_VINFO_DEF_TYPE (vinfo_for_stmt (SLP_TREE_SCALAR_STMTS (child)[0]))
2590 = SLP_TREE_DEF_TYPE (child);
2591 bool res = vect_slp_analyze_node_operations_1 (vinfo, node, node_instance,
2592 cost_vec);
2593 /* Restore def-types. */
2594 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), j, child)
2595 if (SLP_TREE_DEF_TYPE (child) != vect_internal_def)
2596 STMT_VINFO_DEF_TYPE (vinfo_for_stmt (SLP_TREE_SCALAR_STMTS (child)[0]))
2597 = vect_internal_def;
2598 if (! res)
2599 return false;
2601 return true;
2605 /* Analyze statements in SLP instances of VINFO. Return true if the
2606 operations are supported. */
2608 bool
2609 vect_slp_analyze_operations (vec_info *vinfo)
2611 slp_instance instance;
2612 int i;
2614 if (dump_enabled_p ())
2615 dump_printf_loc (MSG_NOTE, vect_location,
2616 "=== vect_slp_analyze_operations ===\n");
2618 scalar_stmts_to_slp_tree_map_t *visited
2619 = new scalar_stmts_to_slp_tree_map_t ();
2620 for (i = 0; vinfo->slp_instances.iterate (i, &instance); )
2622 scalar_stmts_to_slp_tree_map_t lvisited;
2623 stmt_vector_for_cost cost_vec;
2624 cost_vec.create (2);
2625 if (!vect_slp_analyze_node_operations (vinfo,
2626 SLP_INSTANCE_TREE (instance),
2627 instance, visited, &lvisited,
2628 &cost_vec))
2630 dump_printf_loc (MSG_NOTE, vect_location,
2631 "removing SLP instance operations starting from: ");
2632 dump_gimple_stmt (MSG_NOTE, TDF_SLIM,
2633 SLP_TREE_SCALAR_STMTS
2634 (SLP_INSTANCE_TREE (instance))[0], 0);
2635 vect_free_slp_instance (instance);
2636 vinfo->slp_instances.ordered_remove (i);
2637 cost_vec.release ();
2639 else
2641 for (scalar_stmts_to_slp_tree_map_t::iterator x = lvisited.begin();
2642 x != lvisited.end(); ++x)
2643 visited->put ((*x).first.copy (), (*x).second);
2644 i++;
2646 add_stmt_costs (vinfo->target_cost_data, &cost_vec);
2647 cost_vec.release ();
2650 delete visited;
2652 return !vinfo->slp_instances.is_empty ();
2656 /* Compute the scalar cost of the SLP node NODE and its children
2657 and return it. Do not account defs that are marked in LIFE and
2658 update LIFE according to uses of NODE. */
2660 static void
2661 vect_bb_slp_scalar_cost (basic_block bb,
2662 slp_tree node, vec<bool, va_heap> *life,
2663 stmt_vector_for_cost *cost_vec)
2665 unsigned i;
2666 gimple *stmt;
2667 slp_tree child;
2669 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, stmt)
2671 ssa_op_iter op_iter;
2672 def_operand_p def_p;
2673 stmt_vec_info stmt_info;
2675 if ((*life)[i])
2676 continue;
2678 /* If there is a non-vectorized use of the defs then the scalar
2679 stmt is kept live in which case we do not account it or any
2680 required defs in the SLP children in the scalar cost. This
2681 way we make the vectorization more costly when compared to
2682 the scalar cost. */
2683 FOR_EACH_SSA_DEF_OPERAND (def_p, stmt, op_iter, SSA_OP_DEF)
2685 imm_use_iterator use_iter;
2686 gimple *use_stmt;
2687 FOR_EACH_IMM_USE_STMT (use_stmt, use_iter, DEF_FROM_PTR (def_p))
2688 if (!is_gimple_debug (use_stmt)
2689 && (! vect_stmt_in_region_p (vinfo_for_stmt (stmt)->vinfo,
2690 use_stmt)
2691 || ! PURE_SLP_STMT (vinfo_for_stmt (use_stmt))))
2693 (*life)[i] = true;
2694 BREAK_FROM_IMM_USE_STMT (use_iter);
2697 if ((*life)[i])
2698 continue;
2700 /* Count scalar stmts only once. */
2701 if (gimple_visited_p (stmt))
2702 continue;
2703 gimple_set_visited (stmt, true);
2705 stmt_info = vinfo_for_stmt (stmt);
2706 vect_cost_for_stmt kind;
2707 if (STMT_VINFO_DATA_REF (stmt_info))
2709 if (DR_IS_READ (STMT_VINFO_DATA_REF (stmt_info)))
2710 kind = scalar_load;
2711 else
2712 kind = scalar_store;
2714 else
2715 kind = scalar_stmt;
2716 record_stmt_cost (cost_vec, 1, kind, stmt_info, 0, vect_body);
2719 auto_vec<bool, 20> subtree_life;
2720 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
2722 if (SLP_TREE_DEF_TYPE (child) == vect_internal_def)
2724 /* Do not directly pass LIFE to the recursive call, copy it to
2725 confine changes in the callee to the current child/subtree. */
2726 subtree_life.safe_splice (*life);
2727 vect_bb_slp_scalar_cost (bb, child, &subtree_life, cost_vec);
2728 subtree_life.truncate (0);
2733 /* Check if vectorization of the basic block is profitable. */
2735 static bool
2736 vect_bb_vectorization_profitable_p (bb_vec_info bb_vinfo)
2738 vec<slp_instance> slp_instances = BB_VINFO_SLP_INSTANCES (bb_vinfo);
2739 slp_instance instance;
2740 int i;
2741 unsigned int vec_inside_cost = 0, vec_outside_cost = 0, scalar_cost = 0;
2742 unsigned int vec_prologue_cost = 0, vec_epilogue_cost = 0;
2744 /* Calculate scalar cost. */
2745 stmt_vector_for_cost scalar_costs;
2746 scalar_costs.create (0);
2747 FOR_EACH_VEC_ELT (slp_instances, i, instance)
2749 auto_vec<bool, 20> life;
2750 life.safe_grow_cleared (SLP_INSTANCE_GROUP_SIZE (instance));
2751 vect_bb_slp_scalar_cost (BB_VINFO_BB (bb_vinfo),
2752 SLP_INSTANCE_TREE (instance),
2753 &life, &scalar_costs);
2755 void *target_cost_data = init_cost (NULL);
2756 add_stmt_costs (target_cost_data, &scalar_costs);
2757 scalar_costs.release ();
2758 unsigned dummy;
2759 finish_cost (target_cost_data, &dummy, &scalar_cost, &dummy);
2760 destroy_cost_data (target_cost_data);
2762 /* Unset visited flag. */
2763 for (gimple_stmt_iterator gsi = bb_vinfo->region_begin;
2764 gsi_stmt (gsi) != gsi_stmt (bb_vinfo->region_end); gsi_next (&gsi))
2765 gimple_set_visited (gsi_stmt (gsi), false);
2767 /* Complete the target-specific cost calculation. */
2768 finish_cost (BB_VINFO_TARGET_COST_DATA (bb_vinfo), &vec_prologue_cost,
2769 &vec_inside_cost, &vec_epilogue_cost);
2771 vec_outside_cost = vec_prologue_cost + vec_epilogue_cost;
2773 if (dump_enabled_p ())
2775 dump_printf_loc (MSG_NOTE, vect_location, "Cost model analysis: \n");
2776 dump_printf (MSG_NOTE, " Vector inside of basic block cost: %d\n",
2777 vec_inside_cost);
2778 dump_printf (MSG_NOTE, " Vector prologue cost: %d\n", vec_prologue_cost);
2779 dump_printf (MSG_NOTE, " Vector epilogue cost: %d\n", vec_epilogue_cost);
2780 dump_printf (MSG_NOTE, " Scalar cost of basic block: %d\n", scalar_cost);
2783 /* Vectorization is profitable if its cost is more than the cost of scalar
2784 version. Note that we err on the vector side for equal cost because
2785 the cost estimate is otherwise quite pessimistic (constant uses are
2786 free on the scalar side but cost a load on the vector side for
2787 example). */
2788 if (vec_outside_cost + vec_inside_cost > scalar_cost)
2789 return false;
2791 return true;
2794 /* Check if the basic block can be vectorized. Returns a bb_vec_info
2795 if so and sets fatal to true if failure is independent of
2796 current_vector_size. */
2798 static bb_vec_info
2799 vect_slp_analyze_bb_1 (gimple_stmt_iterator region_begin,
2800 gimple_stmt_iterator region_end,
2801 vec<data_reference_p> datarefs, int n_stmts,
2802 bool &fatal)
2804 bb_vec_info bb_vinfo;
2805 slp_instance instance;
2806 int i;
2807 poly_uint64 min_vf = 2;
2809 /* The first group of checks is independent of the vector size. */
2810 fatal = true;
2812 if (n_stmts > PARAM_VALUE (PARAM_SLP_MAX_INSNS_IN_BB))
2814 if (dump_enabled_p ())
2815 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2816 "not vectorized: too many instructions in "
2817 "basic block.\n");
2818 free_data_refs (datarefs);
2819 return NULL;
2822 bb_vinfo = new _bb_vec_info (region_begin, region_end);
2823 if (!bb_vinfo)
2824 return NULL;
2826 BB_VINFO_DATAREFS (bb_vinfo) = datarefs;
2828 /* Analyze the data references. */
2830 if (!vect_analyze_data_refs (bb_vinfo, &min_vf))
2832 if (dump_enabled_p ())
2833 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2834 "not vectorized: unhandled data-ref in basic "
2835 "block.\n");
2837 delete bb_vinfo;
2838 return NULL;
2841 if (BB_VINFO_DATAREFS (bb_vinfo).length () < 2)
2843 if (dump_enabled_p ())
2844 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2845 "not vectorized: not enough data-refs in "
2846 "basic block.\n");
2848 delete bb_vinfo;
2849 return NULL;
2852 if (!vect_analyze_data_ref_accesses (bb_vinfo))
2854 if (dump_enabled_p ())
2855 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2856 "not vectorized: unhandled data access in "
2857 "basic block.\n");
2859 delete bb_vinfo;
2860 return NULL;
2863 /* If there are no grouped stores in the region there is no need
2864 to continue with pattern recog as vect_analyze_slp will fail
2865 anyway. */
2866 if (bb_vinfo->grouped_stores.is_empty ())
2868 if (dump_enabled_p ())
2869 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2870 "not vectorized: no grouped stores in "
2871 "basic block.\n");
2873 delete bb_vinfo;
2874 return NULL;
2877 /* While the rest of the analysis below depends on it in some way. */
2878 fatal = false;
2880 vect_pattern_recog (bb_vinfo);
2882 /* Check the SLP opportunities in the basic block, analyze and build SLP
2883 trees. */
2884 if (!vect_analyze_slp (bb_vinfo, n_stmts))
2886 if (dump_enabled_p ())
2888 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2889 "Failed to SLP the basic block.\n");
2890 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2891 "not vectorized: failed to find SLP opportunities "
2892 "in basic block.\n");
2895 delete bb_vinfo;
2896 return NULL;
2899 vect_record_base_alignments (bb_vinfo);
2901 /* Analyze and verify the alignment of data references and the
2902 dependence in the SLP instances. */
2903 for (i = 0; BB_VINFO_SLP_INSTANCES (bb_vinfo).iterate (i, &instance); )
2905 if (! vect_slp_analyze_and_verify_instance_alignment (instance)
2906 || ! vect_slp_analyze_instance_dependence (instance))
2908 dump_printf_loc (MSG_NOTE, vect_location,
2909 "removing SLP instance operations starting from: ");
2910 dump_gimple_stmt (MSG_NOTE, TDF_SLIM,
2911 SLP_TREE_SCALAR_STMTS
2912 (SLP_INSTANCE_TREE (instance))[0], 0);
2913 vect_free_slp_instance (instance);
2914 BB_VINFO_SLP_INSTANCES (bb_vinfo).ordered_remove (i);
2915 continue;
2918 /* Mark all the statements that we want to vectorize as pure SLP and
2919 relevant. */
2920 vect_mark_slp_stmts (SLP_INSTANCE_TREE (instance), pure_slp, -1);
2921 vect_mark_slp_stmts_relevant (SLP_INSTANCE_TREE (instance));
2923 i++;
2925 if (! BB_VINFO_SLP_INSTANCES (bb_vinfo).length ())
2927 delete bb_vinfo;
2928 return NULL;
2931 if (!vect_slp_analyze_operations (bb_vinfo))
2933 if (dump_enabled_p ())
2934 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2935 "not vectorized: bad operation in basic block.\n");
2937 delete bb_vinfo;
2938 return NULL;
2941 /* Cost model: check if the vectorization is worthwhile. */
2942 if (!unlimited_cost_model (NULL)
2943 && !vect_bb_vectorization_profitable_p (bb_vinfo))
2945 if (dump_enabled_p ())
2946 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2947 "not vectorized: vectorization is not "
2948 "profitable.\n");
2950 delete bb_vinfo;
2951 return NULL;
2954 if (dump_enabled_p ())
2955 dump_printf_loc (MSG_NOTE, vect_location,
2956 "Basic block will be vectorized using SLP\n");
2958 return bb_vinfo;
2962 /* Main entry for the BB vectorizer. Analyze and transform BB, returns
2963 true if anything in the basic-block was vectorized. */
2965 bool
2966 vect_slp_bb (basic_block bb)
2968 bb_vec_info bb_vinfo;
2969 gimple_stmt_iterator gsi;
2970 bool any_vectorized = false;
2971 auto_vector_sizes vector_sizes;
2973 if (dump_enabled_p ())
2974 dump_printf_loc (MSG_NOTE, vect_location, "===vect_slp_analyze_bb===\n");
2976 /* Autodetect first vector size we try. */
2977 current_vector_size = 0;
2978 targetm.vectorize.autovectorize_vector_sizes (&vector_sizes);
2979 unsigned int next_size = 0;
2981 gsi = gsi_start_bb (bb);
2983 poly_uint64 autodetected_vector_size = 0;
2984 while (1)
2986 if (gsi_end_p (gsi))
2987 break;
2989 gimple_stmt_iterator region_begin = gsi;
2990 vec<data_reference_p> datarefs = vNULL;
2991 int insns = 0;
2993 for (; !gsi_end_p (gsi); gsi_next (&gsi))
2995 gimple *stmt = gsi_stmt (gsi);
2996 if (is_gimple_debug (stmt))
2997 continue;
2998 insns++;
3000 if (gimple_location (stmt) != UNKNOWN_LOCATION)
3001 vect_location = gimple_location (stmt);
3003 if (!find_data_references_in_stmt (NULL, stmt, &datarefs))
3004 break;
3007 /* Skip leading unhandled stmts. */
3008 if (gsi_stmt (region_begin) == gsi_stmt (gsi))
3010 gsi_next (&gsi);
3011 continue;
3014 gimple_stmt_iterator region_end = gsi;
3016 bool vectorized = false;
3017 bool fatal = false;
3018 bb_vinfo = vect_slp_analyze_bb_1 (region_begin, region_end,
3019 datarefs, insns, fatal);
3020 if (bb_vinfo
3021 && dbg_cnt (vect_slp))
3023 if (dump_enabled_p ())
3024 dump_printf_loc (MSG_NOTE, vect_location, "SLPing BB part\n");
3026 vect_schedule_slp (bb_vinfo);
3028 if (dump_enabled_p ())
3029 dump_printf_loc (MSG_NOTE, vect_location,
3030 "basic block part vectorized\n");
3032 vectorized = true;
3034 delete bb_vinfo;
3036 any_vectorized |= vectorized;
3038 if (next_size == 0)
3039 autodetected_vector_size = current_vector_size;
3041 if (next_size < vector_sizes.length ()
3042 && known_eq (vector_sizes[next_size], autodetected_vector_size))
3043 next_size += 1;
3045 if (vectorized
3046 || next_size == vector_sizes.length ()
3047 || known_eq (current_vector_size, 0U)
3048 /* If vect_slp_analyze_bb_1 signaled that analysis for all
3049 vector sizes will fail do not bother iterating. */
3050 || fatal)
3052 if (gsi_end_p (region_end))
3053 break;
3055 /* Skip the unhandled stmt. */
3056 gsi_next (&gsi);
3058 /* And reset vector sizes. */
3059 current_vector_size = 0;
3060 next_size = 0;
3062 else
3064 /* Try the next biggest vector size. */
3065 current_vector_size = vector_sizes[next_size++];
3066 if (dump_enabled_p ())
3068 dump_printf_loc (MSG_NOTE, vect_location,
3069 "***** Re-trying analysis with "
3070 "vector size ");
3071 dump_dec (MSG_NOTE, current_vector_size);
3072 dump_printf (MSG_NOTE, "\n");
3075 /* Start over. */
3076 gsi = region_begin;
3080 return any_vectorized;
3084 /* Return 1 if vector type of boolean constant which is OPNUM
3085 operand in statement STMT is a boolean vector. */
3087 static bool
3088 vect_mask_constant_operand_p (gimple *stmt, int opnum)
3090 stmt_vec_info stmt_vinfo = vinfo_for_stmt (stmt);
3091 enum tree_code code = gimple_expr_code (stmt);
3092 tree op, vectype;
3093 gimple *def_stmt;
3094 enum vect_def_type dt;
3096 /* For comparison and COND_EXPR type is chosen depending
3097 on the other comparison operand. */
3098 if (TREE_CODE_CLASS (code) == tcc_comparison)
3100 if (opnum)
3101 op = gimple_assign_rhs1 (stmt);
3102 else
3103 op = gimple_assign_rhs2 (stmt);
3105 if (!vect_is_simple_use (op, stmt_vinfo->vinfo, &def_stmt,
3106 &dt, &vectype))
3107 gcc_unreachable ();
3109 return !vectype || VECTOR_BOOLEAN_TYPE_P (vectype);
3112 if (code == COND_EXPR)
3114 tree cond = gimple_assign_rhs1 (stmt);
3116 if (TREE_CODE (cond) == SSA_NAME)
3117 op = cond;
3118 else if (opnum)
3119 op = TREE_OPERAND (cond, 1);
3120 else
3121 op = TREE_OPERAND (cond, 0);
3123 if (!vect_is_simple_use (op, stmt_vinfo->vinfo, &def_stmt,
3124 &dt, &vectype))
3125 gcc_unreachable ();
3127 return !vectype || VECTOR_BOOLEAN_TYPE_P (vectype);
3130 return VECTOR_BOOLEAN_TYPE_P (STMT_VINFO_VECTYPE (stmt_vinfo));
3133 /* Build a variable-length vector in which the elements in ELTS are repeated
3134 to a fill NRESULTS vectors of type VECTOR_TYPE. Store the vectors in
3135 RESULTS and add any new instructions to SEQ.
3137 The approach we use is:
3139 (1) Find a vector mode VM with integer elements of mode IM.
3141 (2) Replace ELTS[0:NELTS] with ELTS'[0:NELTS'], where each element of
3142 ELTS' has mode IM. This involves creating NELTS' VIEW_CONVERT_EXPRs
3143 from small vectors to IM.
3145 (3) Duplicate each ELTS'[I] into a vector of mode VM.
3147 (4) Use a tree of interleaving VEC_PERM_EXPRs to create VMs with the
3148 correct byte contents.
3150 (5) Use VIEW_CONVERT_EXPR to cast the final VMs to the required type.
3152 We try to find the largest IM for which this sequence works, in order
3153 to cut down on the number of interleaves. */
3155 void
3156 duplicate_and_interleave (gimple_seq *seq, tree vector_type, vec<tree> elts,
3157 unsigned int nresults, vec<tree> &results)
3159 unsigned int nelts = elts.length ();
3160 tree element_type = TREE_TYPE (vector_type);
3162 /* (1) Find a vector mode VM with integer elements of mode IM. */
3163 unsigned int nvectors = 1;
3164 tree new_vector_type;
3165 tree permutes[2];
3166 if (!can_duplicate_and_interleave_p (nelts, TYPE_MODE (element_type),
3167 &nvectors, &new_vector_type,
3168 permutes))
3169 gcc_unreachable ();
3171 /* Get a vector type that holds ELTS[0:NELTS/NELTS']. */
3172 unsigned int partial_nelts = nelts / nvectors;
3173 tree partial_vector_type = build_vector_type (element_type, partial_nelts);
3175 tree_vector_builder partial_elts;
3176 auto_vec<tree, 32> pieces (nvectors * 2);
3177 pieces.quick_grow (nvectors * 2);
3178 for (unsigned int i = 0; i < nvectors; ++i)
3180 /* (2) Replace ELTS[0:NELTS] with ELTS'[0:NELTS'], where each element of
3181 ELTS' has mode IM. */
3182 partial_elts.new_vector (partial_vector_type, partial_nelts, 1);
3183 for (unsigned int j = 0; j < partial_nelts; ++j)
3184 partial_elts.quick_push (elts[i * partial_nelts + j]);
3185 tree t = gimple_build_vector (seq, &partial_elts);
3186 t = gimple_build (seq, VIEW_CONVERT_EXPR,
3187 TREE_TYPE (new_vector_type), t);
3189 /* (3) Duplicate each ELTS'[I] into a vector of mode VM. */
3190 pieces[i] = gimple_build_vector_from_val (seq, new_vector_type, t);
3193 /* (4) Use a tree of VEC_PERM_EXPRs to create a single VM with the
3194 correct byte contents.
3196 We need to repeat the following operation log2(nvectors) times:
3198 out[i * 2] = VEC_PERM_EXPR (in[i], in[i + hi_start], lo_permute);
3199 out[i * 2 + 1] = VEC_PERM_EXPR (in[i], in[i + hi_start], hi_permute);
3201 However, if each input repeats every N elements and the VF is
3202 a multiple of N * 2, the HI result is the same as the LO. */
3203 unsigned int in_start = 0;
3204 unsigned int out_start = nvectors;
3205 unsigned int hi_start = nvectors / 2;
3206 /* A bound on the number of outputs needed to produce NRESULTS results
3207 in the final iteration. */
3208 unsigned int noutputs_bound = nvectors * nresults;
3209 for (unsigned int in_repeat = 1; in_repeat < nvectors; in_repeat *= 2)
3211 noutputs_bound /= 2;
3212 unsigned int limit = MIN (noutputs_bound, nvectors);
3213 for (unsigned int i = 0; i < limit; ++i)
3215 if ((i & 1) != 0
3216 && multiple_p (TYPE_VECTOR_SUBPARTS (new_vector_type),
3217 2 * in_repeat))
3219 pieces[out_start + i] = pieces[out_start + i - 1];
3220 continue;
3223 tree output = make_ssa_name (new_vector_type);
3224 tree input1 = pieces[in_start + (i / 2)];
3225 tree input2 = pieces[in_start + (i / 2) + hi_start];
3226 gassign *stmt = gimple_build_assign (output, VEC_PERM_EXPR,
3227 input1, input2,
3228 permutes[i & 1]);
3229 gimple_seq_add_stmt (seq, stmt);
3230 pieces[out_start + i] = output;
3232 std::swap (in_start, out_start);
3235 /* (5) Use VIEW_CONVERT_EXPR to cast the final VM to the required type. */
3236 results.reserve (nresults);
3237 for (unsigned int i = 0; i < nresults; ++i)
3238 if (i < nvectors)
3239 results.quick_push (gimple_build (seq, VIEW_CONVERT_EXPR, vector_type,
3240 pieces[in_start + i]));
3241 else
3242 results.quick_push (results[i - nvectors]);
3246 /* For constant and loop invariant defs of SLP_NODE this function returns
3247 (vector) defs (VEC_OPRNDS) that will be used in the vectorized stmts.
3248 OP_NUM determines if we gather defs for operand 0 or operand 1 of the RHS of
3249 scalar stmts. NUMBER_OF_VECTORS is the number of vector defs to create.
3250 REDUC_INDEX is the index of the reduction operand in the statements, unless
3251 it is -1. */
3253 static void
3254 vect_get_constant_vectors (tree op, slp_tree slp_node,
3255 vec<tree> *vec_oprnds,
3256 unsigned int op_num, unsigned int number_of_vectors)
3258 vec<gimple *> stmts = SLP_TREE_SCALAR_STMTS (slp_node);
3259 gimple *stmt = stmts[0];
3260 stmt_vec_info stmt_vinfo = vinfo_for_stmt (stmt);
3261 unsigned HOST_WIDE_INT nunits;
3262 tree vec_cst;
3263 unsigned j, number_of_places_left_in_vector;
3264 tree vector_type;
3265 tree vop;
3266 int group_size = stmts.length ();
3267 unsigned int vec_num, i;
3268 unsigned number_of_copies = 1;
3269 vec<tree> voprnds;
3270 voprnds.create (number_of_vectors);
3271 bool constant_p, is_store;
3272 tree neutral_op = NULL;
3273 enum tree_code code = gimple_expr_code (stmt);
3274 gimple_seq ctor_seq = NULL;
3275 auto_vec<tree, 16> permute_results;
3277 /* Check if vector type is a boolean vector. */
3278 if (VECT_SCALAR_BOOLEAN_TYPE_P (TREE_TYPE (op))
3279 && vect_mask_constant_operand_p (stmt, op_num))
3280 vector_type
3281 = build_same_sized_truth_vector_type (STMT_VINFO_VECTYPE (stmt_vinfo));
3282 else
3283 vector_type = get_vectype_for_scalar_type (TREE_TYPE (op));
3285 if (STMT_VINFO_DATA_REF (stmt_vinfo))
3287 is_store = true;
3288 op = gimple_assign_rhs1 (stmt);
3290 else
3291 is_store = false;
3293 gcc_assert (op);
3295 /* NUMBER_OF_COPIES is the number of times we need to use the same values in
3296 created vectors. It is greater than 1 if unrolling is performed.
3298 For example, we have two scalar operands, s1 and s2 (e.g., group of
3299 strided accesses of size two), while NUNITS is four (i.e., four scalars
3300 of this type can be packed in a vector). The output vector will contain
3301 two copies of each scalar operand: {s1, s2, s1, s2}. (NUMBER_OF_COPIES
3302 will be 2).
3304 If GROUP_SIZE > NUNITS, the scalars will be split into several vectors
3305 containing the operands.
3307 For example, NUNITS is four as before, and the group size is 8
3308 (s1, s2, ..., s8). We will create two vectors {s1, s2, s3, s4} and
3309 {s5, s6, s7, s8}. */
3311 /* When using duplicate_and_interleave, we just need one element for
3312 each scalar statement. */
3313 if (!TYPE_VECTOR_SUBPARTS (vector_type).is_constant (&nunits))
3314 nunits = group_size;
3316 number_of_copies = nunits * number_of_vectors / group_size;
3318 number_of_places_left_in_vector = nunits;
3319 constant_p = true;
3320 tree_vector_builder elts (vector_type, nunits, 1);
3321 elts.quick_grow (nunits);
3322 bool place_after_defs = false;
3323 for (j = 0; j < number_of_copies; j++)
3325 for (i = group_size - 1; stmts.iterate (i, &stmt); i--)
3327 if (is_store)
3328 op = gimple_assign_rhs1 (stmt);
3329 else
3331 switch (code)
3333 case COND_EXPR:
3335 tree cond = gimple_assign_rhs1 (stmt);
3336 if (TREE_CODE (cond) == SSA_NAME)
3337 op = gimple_op (stmt, op_num + 1);
3338 else if (op_num == 0 || op_num == 1)
3339 op = TREE_OPERAND (cond, op_num);
3340 else
3342 if (op_num == 2)
3343 op = gimple_assign_rhs2 (stmt);
3344 else
3345 op = gimple_assign_rhs3 (stmt);
3348 break;
3350 case CALL_EXPR:
3351 op = gimple_call_arg (stmt, op_num);
3352 break;
3354 case LSHIFT_EXPR:
3355 case RSHIFT_EXPR:
3356 case LROTATE_EXPR:
3357 case RROTATE_EXPR:
3358 op = gimple_op (stmt, op_num + 1);
3359 /* Unlike the other binary operators, shifts/rotates have
3360 the shift count being int, instead of the same type as
3361 the lhs, so make sure the scalar is the right type if
3362 we are dealing with vectors of
3363 long long/long/short/char. */
3364 if (op_num == 1 && TREE_CODE (op) == INTEGER_CST)
3365 op = fold_convert (TREE_TYPE (vector_type), op);
3366 break;
3368 default:
3369 op = gimple_op (stmt, op_num + 1);
3370 break;
3374 /* Create 'vect_ = {op0,op1,...,opn}'. */
3375 number_of_places_left_in_vector--;
3376 tree orig_op = op;
3377 if (!types_compatible_p (TREE_TYPE (vector_type), TREE_TYPE (op)))
3379 if (CONSTANT_CLASS_P (op))
3381 if (VECTOR_BOOLEAN_TYPE_P (vector_type))
3383 /* Can't use VIEW_CONVERT_EXPR for booleans because
3384 of possibly different sizes of scalar value and
3385 vector element. */
3386 if (integer_zerop (op))
3387 op = build_int_cst (TREE_TYPE (vector_type), 0);
3388 else if (integer_onep (op))
3389 op = build_all_ones_cst (TREE_TYPE (vector_type));
3390 else
3391 gcc_unreachable ();
3393 else
3394 op = fold_unary (VIEW_CONVERT_EXPR,
3395 TREE_TYPE (vector_type), op);
3396 gcc_assert (op && CONSTANT_CLASS_P (op));
3398 else
3400 tree new_temp = make_ssa_name (TREE_TYPE (vector_type));
3401 gimple *init_stmt;
3402 if (VECTOR_BOOLEAN_TYPE_P (vector_type))
3404 tree true_val
3405 = build_all_ones_cst (TREE_TYPE (vector_type));
3406 tree false_val
3407 = build_zero_cst (TREE_TYPE (vector_type));
3408 gcc_assert (INTEGRAL_TYPE_P (TREE_TYPE (op)));
3409 init_stmt = gimple_build_assign (new_temp, COND_EXPR,
3410 op, true_val,
3411 false_val);
3413 else
3415 op = build1 (VIEW_CONVERT_EXPR, TREE_TYPE (vector_type),
3416 op);
3417 init_stmt
3418 = gimple_build_assign (new_temp, VIEW_CONVERT_EXPR,
3419 op);
3421 gimple_seq_add_stmt (&ctor_seq, init_stmt);
3422 op = new_temp;
3425 elts[number_of_places_left_in_vector] = op;
3426 if (!CONSTANT_CLASS_P (op))
3427 constant_p = false;
3428 if (TREE_CODE (orig_op) == SSA_NAME
3429 && !SSA_NAME_IS_DEFAULT_DEF (orig_op)
3430 && STMT_VINFO_BB_VINFO (stmt_vinfo)
3431 && (STMT_VINFO_BB_VINFO (stmt_vinfo)->bb
3432 == gimple_bb (SSA_NAME_DEF_STMT (orig_op))))
3433 place_after_defs = true;
3435 if (number_of_places_left_in_vector == 0)
3437 if (constant_p
3438 ? multiple_p (TYPE_VECTOR_SUBPARTS (vector_type), nunits)
3439 : known_eq (TYPE_VECTOR_SUBPARTS (vector_type), nunits))
3440 vec_cst = gimple_build_vector (&ctor_seq, &elts);
3441 else
3443 if (vec_oprnds->is_empty ())
3444 duplicate_and_interleave (&ctor_seq, vector_type, elts,
3445 number_of_vectors,
3446 permute_results);
3447 vec_cst = permute_results[number_of_vectors - j - 1];
3449 tree init;
3450 gimple_stmt_iterator gsi;
3451 if (place_after_defs)
3453 gsi = gsi_for_stmt
3454 (vect_find_last_scalar_stmt_in_slp (slp_node));
3455 init = vect_init_vector (stmt, vec_cst, vector_type, &gsi);
3457 else
3458 init = vect_init_vector (stmt, vec_cst, vector_type, NULL);
3459 if (ctor_seq != NULL)
3461 gsi = gsi_for_stmt (SSA_NAME_DEF_STMT (init));
3462 gsi_insert_seq_before (&gsi, ctor_seq, GSI_SAME_STMT);
3463 ctor_seq = NULL;
3465 voprnds.quick_push (init);
3466 place_after_defs = false;
3467 number_of_places_left_in_vector = nunits;
3468 constant_p = true;
3469 elts.new_vector (vector_type, nunits, 1);
3470 elts.quick_grow (nunits);
3475 /* Since the vectors are created in the reverse order, we should invert
3476 them. */
3477 vec_num = voprnds.length ();
3478 for (j = vec_num; j != 0; j--)
3480 vop = voprnds[j - 1];
3481 vec_oprnds->quick_push (vop);
3484 voprnds.release ();
3486 /* In case that VF is greater than the unrolling factor needed for the SLP
3487 group of stmts, NUMBER_OF_VECTORS to be created is greater than
3488 NUMBER_OF_SCALARS/NUNITS or NUNITS/NUMBER_OF_SCALARS, and hence we have
3489 to replicate the vectors. */
3490 while (number_of_vectors > vec_oprnds->length ())
3492 tree neutral_vec = NULL;
3494 if (neutral_op)
3496 if (!neutral_vec)
3497 neutral_vec = build_vector_from_val (vector_type, neutral_op);
3499 vec_oprnds->quick_push (neutral_vec);
3501 else
3503 for (i = 0; vec_oprnds->iterate (i, &vop) && i < vec_num; i++)
3504 vec_oprnds->quick_push (vop);
3510 /* Get vectorized definitions from SLP_NODE that contains corresponding
3511 vectorized def-stmts. */
3513 static void
3514 vect_get_slp_vect_defs (slp_tree slp_node, vec<tree> *vec_oprnds)
3516 tree vec_oprnd;
3517 gimple *vec_def_stmt;
3518 unsigned int i;
3520 gcc_assert (SLP_TREE_VEC_STMTS (slp_node).exists ());
3522 FOR_EACH_VEC_ELT (SLP_TREE_VEC_STMTS (slp_node), i, vec_def_stmt)
3524 gcc_assert (vec_def_stmt);
3525 if (gimple_code (vec_def_stmt) == GIMPLE_PHI)
3526 vec_oprnd = gimple_phi_result (vec_def_stmt);
3527 else
3528 vec_oprnd = gimple_get_lhs (vec_def_stmt);
3529 vec_oprnds->quick_push (vec_oprnd);
3534 /* Get vectorized definitions for SLP_NODE.
3535 If the scalar definitions are loop invariants or constants, collect them and
3536 call vect_get_constant_vectors() to create vector stmts.
3537 Otherwise, the def-stmts must be already vectorized and the vectorized stmts
3538 must be stored in the corresponding child of SLP_NODE, and we call
3539 vect_get_slp_vect_defs () to retrieve them. */
3541 void
3542 vect_get_slp_defs (vec<tree> ops, slp_tree slp_node,
3543 vec<vec<tree> > *vec_oprnds)
3545 gimple *first_stmt;
3546 int number_of_vects = 0, i;
3547 unsigned int child_index = 0;
3548 HOST_WIDE_INT lhs_size_unit, rhs_size_unit;
3549 slp_tree child = NULL;
3550 vec<tree> vec_defs;
3551 tree oprnd;
3552 bool vectorized_defs;
3554 first_stmt = SLP_TREE_SCALAR_STMTS (slp_node)[0];
3555 FOR_EACH_VEC_ELT (ops, i, oprnd)
3557 /* For each operand we check if it has vectorized definitions in a child
3558 node or we need to create them (for invariants and constants). We
3559 check if the LHS of the first stmt of the next child matches OPRND.
3560 If it does, we found the correct child. Otherwise, we call
3561 vect_get_constant_vectors (), and not advance CHILD_INDEX in order
3562 to check this child node for the next operand. */
3563 vectorized_defs = false;
3564 if (SLP_TREE_CHILDREN (slp_node).length () > child_index)
3566 child = SLP_TREE_CHILDREN (slp_node)[child_index];
3568 /* We have to check both pattern and original def, if available. */
3569 if (SLP_TREE_DEF_TYPE (child) == vect_internal_def)
3571 gimple *first_def = SLP_TREE_SCALAR_STMTS (child)[0];
3572 gimple *related
3573 = STMT_VINFO_RELATED_STMT (vinfo_for_stmt (first_def));
3574 tree first_def_op;
3576 if (gimple_code (first_def) == GIMPLE_PHI)
3577 first_def_op = gimple_phi_result (first_def);
3578 else
3579 first_def_op = gimple_get_lhs (first_def);
3580 if (operand_equal_p (oprnd, first_def_op, 0)
3581 || (related
3582 && operand_equal_p (oprnd, gimple_get_lhs (related), 0)))
3584 /* The number of vector defs is determined by the number of
3585 vector statements in the node from which we get those
3586 statements. */
3587 number_of_vects = SLP_TREE_NUMBER_OF_VEC_STMTS (child);
3588 vectorized_defs = true;
3589 child_index++;
3592 else
3593 child_index++;
3596 if (!vectorized_defs)
3598 if (i == 0)
3600 number_of_vects = SLP_TREE_NUMBER_OF_VEC_STMTS (slp_node);
3601 /* Number of vector stmts was calculated according to LHS in
3602 vect_schedule_slp_instance (), fix it by replacing LHS with
3603 RHS, if necessary. See vect_get_smallest_scalar_type () for
3604 details. */
3605 vect_get_smallest_scalar_type (first_stmt, &lhs_size_unit,
3606 &rhs_size_unit);
3607 if (rhs_size_unit != lhs_size_unit)
3609 number_of_vects *= rhs_size_unit;
3610 number_of_vects /= lhs_size_unit;
3615 /* Allocate memory for vectorized defs. */
3616 vec_defs = vNULL;
3617 vec_defs.create (number_of_vects);
3619 /* For reduction defs we call vect_get_constant_vectors (), since we are
3620 looking for initial loop invariant values. */
3621 if (vectorized_defs)
3622 /* The defs are already vectorized. */
3623 vect_get_slp_vect_defs (child, &vec_defs);
3624 else
3625 /* Build vectors from scalar defs. */
3626 vect_get_constant_vectors (oprnd, slp_node, &vec_defs, i,
3627 number_of_vects);
3629 vec_oprnds->quick_push (vec_defs);
3633 /* Generate vector permute statements from a list of loads in DR_CHAIN.
3634 If ANALYZE_ONLY is TRUE, only check that it is possible to create valid
3635 permute statements for the SLP node NODE of the SLP instance
3636 SLP_NODE_INSTANCE. */
3638 bool
3639 vect_transform_slp_perm_load (slp_tree node, vec<tree> dr_chain,
3640 gimple_stmt_iterator *gsi, poly_uint64 vf,
3641 slp_instance slp_node_instance, bool analyze_only,
3642 unsigned *n_perms)
3644 gimple *stmt = SLP_TREE_SCALAR_STMTS (node)[0];
3645 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
3646 tree mask_element_type = NULL_TREE, mask_type;
3647 int vec_index = 0;
3648 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
3649 int group_size = SLP_INSTANCE_GROUP_SIZE (slp_node_instance);
3650 unsigned int mask_element;
3651 machine_mode mode;
3652 unsigned HOST_WIDE_INT nunits, const_vf;
3654 if (!STMT_VINFO_GROUPED_ACCESS (stmt_info))
3655 return false;
3657 stmt_info = vinfo_for_stmt (GROUP_FIRST_ELEMENT (stmt_info));
3659 mode = TYPE_MODE (vectype);
3661 /* At the moment, all permutations are represented using per-element
3662 indices, so we can't cope with variable vector lengths or
3663 vectorization factors. */
3664 if (!TYPE_VECTOR_SUBPARTS (vectype).is_constant (&nunits)
3665 || !vf.is_constant (&const_vf))
3666 return false;
3668 /* The generic VEC_PERM_EXPR code always uses an integral type of the
3669 same size as the vector element being permuted. */
3670 mask_element_type = lang_hooks.types.type_for_mode
3671 (int_mode_for_mode (TYPE_MODE (TREE_TYPE (vectype))).require (), 1);
3672 mask_type = get_vectype_for_scalar_type (mask_element_type);
3673 vec_perm_builder mask (nunits, nunits, 1);
3674 mask.quick_grow (nunits);
3675 vec_perm_indices indices;
3677 /* Initialize the vect stmts of NODE to properly insert the generated
3678 stmts later. */
3679 if (! analyze_only)
3680 for (unsigned i = SLP_TREE_VEC_STMTS (node).length ();
3681 i < SLP_TREE_NUMBER_OF_VEC_STMTS (node); i++)
3682 SLP_TREE_VEC_STMTS (node).quick_push (NULL);
3684 /* Generate permutation masks for every NODE. Number of masks for each NODE
3685 is equal to GROUP_SIZE.
3686 E.g., we have a group of three nodes with three loads from the same
3687 location in each node, and the vector size is 4. I.e., we have a
3688 a0b0c0a1b1c1... sequence and we need to create the following vectors:
3689 for a's: a0a0a0a1 a1a1a2a2 a2a3a3a3
3690 for b's: b0b0b0b1 b1b1b2b2 b2b3b3b3
3693 The masks for a's should be: {0,0,0,3} {3,3,6,6} {6,9,9,9}.
3694 The last mask is illegal since we assume two operands for permute
3695 operation, and the mask element values can't be outside that range.
3696 Hence, the last mask must be converted into {2,5,5,5}.
3697 For the first two permutations we need the first and the second input
3698 vectors: {a0,b0,c0,a1} and {b1,c1,a2,b2}, and for the last permutation
3699 we need the second and the third vectors: {b1,c1,a2,b2} and
3700 {c2,a3,b3,c3}. */
3702 int vect_stmts_counter = 0;
3703 unsigned int index = 0;
3704 int first_vec_index = -1;
3705 int second_vec_index = -1;
3706 bool noop_p = true;
3707 *n_perms = 0;
3709 for (unsigned int j = 0; j < const_vf; j++)
3711 for (int k = 0; k < group_size; k++)
3713 unsigned int i = (SLP_TREE_LOAD_PERMUTATION (node)[k]
3714 + j * STMT_VINFO_GROUP_SIZE (stmt_info));
3715 vec_index = i / nunits;
3716 mask_element = i % nunits;
3717 if (vec_index == first_vec_index
3718 || first_vec_index == -1)
3720 first_vec_index = vec_index;
3722 else if (vec_index == second_vec_index
3723 || second_vec_index == -1)
3725 second_vec_index = vec_index;
3726 mask_element += nunits;
3728 else
3730 if (dump_enabled_p ())
3732 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
3733 "permutation requires at "
3734 "least three vectors ");
3735 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
3736 stmt, 0);
3738 gcc_assert (analyze_only);
3739 return false;
3742 gcc_assert (mask_element < 2 * nunits);
3743 if (mask_element != index)
3744 noop_p = false;
3745 mask[index++] = mask_element;
3747 if (index == nunits && !noop_p)
3749 indices.new_vector (mask, 2, nunits);
3750 if (!can_vec_perm_const_p (mode, indices))
3752 if (dump_enabled_p ())
3754 dump_printf_loc (MSG_MISSED_OPTIMIZATION,
3755 vect_location,
3756 "unsupported vect permute { ");
3757 for (i = 0; i < nunits; ++i)
3759 dump_dec (MSG_MISSED_OPTIMIZATION, mask[i]);
3760 dump_printf (MSG_MISSED_OPTIMIZATION, " ");
3762 dump_printf (MSG_MISSED_OPTIMIZATION, "}\n");
3764 gcc_assert (analyze_only);
3765 return false;
3768 ++*n_perms;
3771 if (index == nunits)
3773 if (!analyze_only)
3775 tree mask_vec = NULL_TREE;
3777 if (! noop_p)
3778 mask_vec = vec_perm_indices_to_tree (mask_type, indices);
3780 if (second_vec_index == -1)
3781 second_vec_index = first_vec_index;
3783 /* Generate the permute statement if necessary. */
3784 tree first_vec = dr_chain[first_vec_index];
3785 tree second_vec = dr_chain[second_vec_index];
3786 gimple *perm_stmt;
3787 if (! noop_p)
3789 tree perm_dest
3790 = vect_create_destination_var (gimple_assign_lhs (stmt),
3791 vectype);
3792 perm_dest = make_ssa_name (perm_dest);
3793 perm_stmt = gimple_build_assign (perm_dest,
3794 VEC_PERM_EXPR,
3795 first_vec, second_vec,
3796 mask_vec);
3797 vect_finish_stmt_generation (stmt, perm_stmt, gsi);
3799 else
3800 /* If mask was NULL_TREE generate the requested
3801 identity transform. */
3802 perm_stmt = SSA_NAME_DEF_STMT (first_vec);
3804 /* Store the vector statement in NODE. */
3805 SLP_TREE_VEC_STMTS (node)[vect_stmts_counter++] = perm_stmt;
3808 index = 0;
3809 first_vec_index = -1;
3810 second_vec_index = -1;
3811 noop_p = true;
3816 return true;
3819 /* Vectorize SLP instance tree in postorder. */
3821 static bool
3822 vect_schedule_slp_instance (slp_tree node, slp_instance instance,
3823 scalar_stmts_to_slp_tree_map_t *bst_map)
3825 gimple *stmt;
3826 bool grouped_store, is_store;
3827 gimple_stmt_iterator si;
3828 stmt_vec_info stmt_info;
3829 unsigned int group_size;
3830 tree vectype;
3831 int i, j;
3832 slp_tree child;
3834 if (SLP_TREE_DEF_TYPE (node) != vect_internal_def)
3835 return false;
3837 /* See if we have already vectorized the same set of stmts and reuse their
3838 vectorized stmts. */
3839 if (slp_tree *leader = bst_map->get (SLP_TREE_SCALAR_STMTS (node)))
3841 SLP_TREE_VEC_STMTS (node).safe_splice (SLP_TREE_VEC_STMTS (*leader));
3842 return false;
3845 bst_map->put (SLP_TREE_SCALAR_STMTS (node).copy (), node);
3846 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
3847 vect_schedule_slp_instance (child, instance, bst_map);
3849 /* Push SLP node def-type to stmts. */
3850 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
3851 if (SLP_TREE_DEF_TYPE (child) != vect_internal_def)
3852 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (child), j, stmt)
3853 STMT_VINFO_DEF_TYPE (vinfo_for_stmt (stmt)) = SLP_TREE_DEF_TYPE (child);
3855 stmt = SLP_TREE_SCALAR_STMTS (node)[0];
3856 stmt_info = vinfo_for_stmt (stmt);
3858 /* VECTYPE is the type of the destination. */
3859 vectype = STMT_VINFO_VECTYPE (stmt_info);
3860 poly_uint64 nunits = TYPE_VECTOR_SUBPARTS (vectype);
3861 group_size = SLP_INSTANCE_GROUP_SIZE (instance);
3863 gcc_assert (SLP_TREE_NUMBER_OF_VEC_STMTS (node) != 0);
3864 if (!SLP_TREE_VEC_STMTS (node).exists ())
3865 SLP_TREE_VEC_STMTS (node).create (SLP_TREE_NUMBER_OF_VEC_STMTS (node));
3867 if (dump_enabled_p ())
3869 dump_printf_loc (MSG_NOTE,vect_location,
3870 "------>vectorizing SLP node starting from: ");
3871 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
3874 /* Vectorized stmts go before the last scalar stmt which is where
3875 all uses are ready. */
3876 si = gsi_for_stmt (vect_find_last_scalar_stmt_in_slp (node));
3878 /* Mark the first element of the reduction chain as reduction to properly
3879 transform the node. In the analysis phase only the last element of the
3880 chain is marked as reduction. */
3881 if (GROUP_FIRST_ELEMENT (stmt_info) && !STMT_VINFO_GROUPED_ACCESS (stmt_info)
3882 && GROUP_FIRST_ELEMENT (stmt_info) == stmt)
3884 STMT_VINFO_DEF_TYPE (stmt_info) = vect_reduction_def;
3885 STMT_VINFO_TYPE (stmt_info) = reduc_vec_info_type;
3888 /* Handle two-operation SLP nodes by vectorizing the group with
3889 both operations and then performing a merge. */
3890 if (SLP_TREE_TWO_OPERATORS (node))
3892 enum tree_code code0 = gimple_assign_rhs_code (stmt);
3893 enum tree_code ocode = ERROR_MARK;
3894 gimple *ostmt;
3895 vec_perm_builder mask (group_size, group_size, 1);
3896 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, ostmt)
3897 if (gimple_assign_rhs_code (ostmt) != code0)
3899 mask.quick_push (1);
3900 ocode = gimple_assign_rhs_code (ostmt);
3902 else
3903 mask.quick_push (0);
3904 if (ocode != ERROR_MARK)
3906 vec<gimple *> v0;
3907 vec<gimple *> v1;
3908 unsigned j;
3909 tree tmask = NULL_TREE;
3910 vect_transform_stmt (stmt, &si, &grouped_store, node, instance);
3911 v0 = SLP_TREE_VEC_STMTS (node).copy ();
3912 SLP_TREE_VEC_STMTS (node).truncate (0);
3913 gimple_assign_set_rhs_code (stmt, ocode);
3914 vect_transform_stmt (stmt, &si, &grouped_store, node, instance);
3915 gimple_assign_set_rhs_code (stmt, code0);
3916 v1 = SLP_TREE_VEC_STMTS (node).copy ();
3917 SLP_TREE_VEC_STMTS (node).truncate (0);
3918 tree meltype = build_nonstandard_integer_type
3919 (GET_MODE_BITSIZE (SCALAR_TYPE_MODE (TREE_TYPE (vectype))), 1);
3920 tree mvectype = get_same_sized_vectype (meltype, vectype);
3921 unsigned k = 0, l;
3922 for (j = 0; j < v0.length (); ++j)
3924 /* Enforced by vect_build_slp_tree, which rejects variable-length
3925 vectors for SLP_TREE_TWO_OPERATORS. */
3926 unsigned int const_nunits = nunits.to_constant ();
3927 tree_vector_builder melts (mvectype, const_nunits, 1);
3928 for (l = 0; l < const_nunits; ++l)
3930 if (k >= group_size)
3931 k = 0;
3932 tree t = build_int_cst (meltype,
3933 mask[k++] * const_nunits + l);
3934 melts.quick_push (t);
3936 tmask = melts.build ();
3938 /* ??? Not all targets support a VEC_PERM_EXPR with a
3939 constant mask that would translate to a vec_merge RTX
3940 (with their vec_perm_const_ok). We can either not
3941 vectorize in that case or let veclower do its job.
3942 Unfortunately that isn't too great and at least for
3943 plus/minus we'd eventually like to match targets
3944 vector addsub instructions. */
3945 gimple *vstmt;
3946 vstmt = gimple_build_assign (make_ssa_name (vectype),
3947 VEC_PERM_EXPR,
3948 gimple_assign_lhs (v0[j]),
3949 gimple_assign_lhs (v1[j]), tmask);
3950 vect_finish_stmt_generation (stmt, vstmt, &si);
3951 SLP_TREE_VEC_STMTS (node).quick_push (vstmt);
3953 v0.release ();
3954 v1.release ();
3955 return false;
3958 is_store = vect_transform_stmt (stmt, &si, &grouped_store, node, instance);
3960 /* Restore stmt def-types. */
3961 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
3962 if (SLP_TREE_DEF_TYPE (child) != vect_internal_def)
3963 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (child), j, stmt)
3964 STMT_VINFO_DEF_TYPE (vinfo_for_stmt (stmt)) = vect_internal_def;
3966 return is_store;
3969 /* Replace scalar calls from SLP node NODE with setting of their lhs to zero.
3970 For loop vectorization this is done in vectorizable_call, but for SLP
3971 it needs to be deferred until end of vect_schedule_slp, because multiple
3972 SLP instances may refer to the same scalar stmt. */
3974 static void
3975 vect_remove_slp_scalar_calls (slp_tree node)
3977 gimple *stmt, *new_stmt;
3978 gimple_stmt_iterator gsi;
3979 int i;
3980 slp_tree child;
3981 tree lhs;
3982 stmt_vec_info stmt_info;
3984 if (SLP_TREE_DEF_TYPE (node) != vect_internal_def)
3985 return;
3987 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
3988 vect_remove_slp_scalar_calls (child);
3990 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, stmt)
3992 if (!is_gimple_call (stmt) || gimple_bb (stmt) == NULL)
3993 continue;
3994 stmt_info = vinfo_for_stmt (stmt);
3995 if (stmt_info == NULL
3996 || is_pattern_stmt_p (stmt_info)
3997 || !PURE_SLP_STMT (stmt_info))
3998 continue;
3999 lhs = gimple_call_lhs (stmt);
4000 new_stmt = gimple_build_assign (lhs, build_zero_cst (TREE_TYPE (lhs)));
4001 set_vinfo_for_stmt (new_stmt, stmt_info);
4002 set_vinfo_for_stmt (stmt, NULL);
4003 STMT_VINFO_STMT (stmt_info) = new_stmt;
4004 gsi = gsi_for_stmt (stmt);
4005 gsi_replace (&gsi, new_stmt, false);
4006 SSA_NAME_DEF_STMT (gimple_assign_lhs (new_stmt)) = new_stmt;
4010 /* Generate vector code for all SLP instances in the loop/basic block. */
4012 bool
4013 vect_schedule_slp (vec_info *vinfo)
4015 vec<slp_instance> slp_instances;
4016 slp_instance instance;
4017 unsigned int i;
4018 bool is_store = false;
4021 scalar_stmts_to_slp_tree_map_t *bst_map
4022 = new scalar_stmts_to_slp_tree_map_t ();
4023 slp_instances = vinfo->slp_instances;
4024 FOR_EACH_VEC_ELT (slp_instances, i, instance)
4026 /* Schedule the tree of INSTANCE. */
4027 is_store = vect_schedule_slp_instance (SLP_INSTANCE_TREE (instance),
4028 instance, bst_map);
4029 if (dump_enabled_p ())
4030 dump_printf_loc (MSG_NOTE, vect_location,
4031 "vectorizing stmts using SLP.\n");
4033 delete bst_map;
4035 FOR_EACH_VEC_ELT (slp_instances, i, instance)
4037 slp_tree root = SLP_INSTANCE_TREE (instance);
4038 gimple *store;
4039 unsigned int j;
4040 gimple_stmt_iterator gsi;
4042 /* Remove scalar call stmts. Do not do this for basic-block
4043 vectorization as not all uses may be vectorized.
4044 ??? Why should this be necessary? DCE should be able to
4045 remove the stmts itself.
4046 ??? For BB vectorization we can as well remove scalar
4047 stmts starting from the SLP tree root if they have no
4048 uses. */
4049 if (is_a <loop_vec_info> (vinfo))
4050 vect_remove_slp_scalar_calls (root);
4052 for (j = 0; SLP_TREE_SCALAR_STMTS (root).iterate (j, &store)
4053 && j < SLP_INSTANCE_GROUP_SIZE (instance); j++)
4055 if (!STMT_VINFO_DATA_REF (vinfo_for_stmt (store)))
4056 break;
4058 if (is_pattern_stmt_p (vinfo_for_stmt (store)))
4059 store = STMT_VINFO_RELATED_STMT (vinfo_for_stmt (store));
4060 /* Free the attached stmt_vec_info and remove the stmt. */
4061 gsi = gsi_for_stmt (store);
4062 unlink_stmt_vdef (store);
4063 gsi_remove (&gsi, true);
4064 release_defs (store);
4065 free_stmt_vec_info (store);
4069 return is_store;