]> git.ipfire.org Git - thirdparty/gcc.git/blame - gcc/tree-vect-slp.c
Support SHF_EXCLUDE on non-x86 and with Solaris as
[thirdparty/gcc.git] / gcc / tree-vect-slp.c
CommitLineData
ebfd146a 1/* SLP - Basic Block Vectorization
85ec4feb 2 Copyright (C) 2007-2018 Free Software Foundation, Inc.
b8698a0f 3 Contributed by Dorit Naishlos <dorit@il.ibm.com>
ebfd146a
IR
4 and Ira Rosen <irar@il.ibm.com>
5
6This file is part of GCC.
7
8GCC is free software; you can redistribute it and/or modify it under
9the terms of the GNU General Public License as published by the Free
10Software Foundation; either version 3, or (at your option) any later
11version.
12
13GCC is distributed in the hope that it will be useful, but WITHOUT ANY
14WARRANTY; without even the implied warranty of MERCHANTABILITY or
15FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
16for more details.
17
18You should have received a copy of the GNU General Public License
19along with GCC; see the file COPYING3. If not see
20<http://www.gnu.org/licenses/>. */
21
22#include "config.h"
23#include "system.h"
24#include "coretypes.h"
c7131fb2 25#include "backend.h"
957060b5
AM
26#include "target.h"
27#include "rtl.h"
ebfd146a 28#include "tree.h"
c7131fb2 29#include "gimple.h"
957060b5 30#include "tree-pass.h"
c7131fb2 31#include "ssa.h"
957060b5
AM
32#include "optabs-tree.h"
33#include "insn-config.h"
34#include "recog.h" /* FIXME: for insn_data */
957060b5 35#include "params.h"
40e23961 36#include "fold-const.h"
d8a2d370 37#include "stor-layout.h"
5be5c238 38#include "gimple-iterator.h"
ebfd146a 39#include "cfgloop.h"
ebfd146a 40#include "tree-vectorizer.h"
2635892a 41#include "langhooks.h"
642fce57 42#include "gimple-walk.h"
428db0ba 43#include "dbgcnt.h"
5ebaa477 44#include "tree-vector-builder.h"
f151c9e1 45#include "vec-perm-indices.h"
018b2744
RS
46#include "gimple-fold.h"
47#include "internal-fn.h"
a70d6342
IR
48
49
ebfd146a
IR
50/* Recursively free the memory allocated for the SLP tree rooted at NODE. */
51
52static void
53vect_free_slp_tree (slp_tree node)
54{
d092494c 55 int i;
d755c7ef 56 slp_tree child;
d092494c 57
9771b263 58 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
d755c7ef 59 vect_free_slp_tree (child);
b8698a0f 60
78810bd3
RB
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))
65 {
66 gcc_assert (STMT_VINFO_NUM_SLP_USES (vinfo_for_stmt (stmt)) > 0);
67 STMT_VINFO_NUM_SLP_USES (vinfo_for_stmt (stmt))--;
68 }
69
9771b263
DN
70 SLP_TREE_CHILDREN (node).release ();
71 SLP_TREE_SCALAR_STMTS (node).release ();
72 SLP_TREE_VEC_STMTS (node).release ();
01d8bf07 73 SLP_TREE_LOAD_PERMUTATION (node).release ();
ebfd146a
IR
74
75 free (node);
76}
77
78
79/* Free the memory allocated for the SLP instance. */
80
81void
82vect_free_slp_instance (slp_instance instance)
83{
84 vect_free_slp_tree (SLP_INSTANCE_TREE (instance));
9771b263 85 SLP_INSTANCE_LOADS (instance).release ();
c7e62a26 86 free (instance);
ebfd146a
IR
87}
88
89
d092494c
IR
90/* Create an SLP node for SCALAR_STMTS. */
91
92static slp_tree
355fe088 93vect_create_new_slp_node (vec<gimple *> scalar_stmts)
d092494c 94{
d3cfd39e 95 slp_tree node;
355fe088 96 gimple *stmt = scalar_stmts[0];
d092494c
IR
97 unsigned int nops;
98
99 if (is_gimple_call (stmt))
100 nops = gimple_call_num_args (stmt);
101 else if (is_gimple_assign (stmt))
f7e531cf
IR
102 {
103 nops = gimple_num_ops (stmt) - 1;
104 if (gimple_assign_rhs_code (stmt) == COND_EXPR)
105 nops++;
106 }
e7baeb39
RB
107 else if (gimple_code (stmt) == GIMPLE_PHI)
108 nops = 0;
d092494c
IR
109 else
110 return NULL;
111
d3cfd39e 112 node = XNEW (struct _slp_tree);
d092494c 113 SLP_TREE_SCALAR_STMTS (node) = scalar_stmts;
9771b263 114 SLP_TREE_VEC_STMTS (node).create (0);
68435eb2 115 SLP_TREE_NUMBER_OF_VEC_STMTS (node) = 0;
9771b263 116 SLP_TREE_CHILDREN (node).create (nops);
01d8bf07 117 SLP_TREE_LOAD_PERMUTATION (node) = vNULL;
6876e5bc 118 SLP_TREE_TWO_OPERATORS (node) = false;
603cca93 119 SLP_TREE_DEF_TYPE (node) = vect_internal_def;
d092494c 120
78810bd3
RB
121 unsigned i;
122 FOR_EACH_VEC_ELT (scalar_stmts, i, stmt)
123 STMT_VINFO_NUM_SLP_USES (vinfo_for_stmt (stmt))++;
124
d092494c
IR
125 return node;
126}
127
128
ddf56386
RB
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. */
132typedef struct _slp_oprnd_info
133{
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. */
ddf56386 139 tree first_op_type;
34e82342 140 enum vect_def_type first_dt;
ddf56386
RB
141 bool first_pattern;
142 bool second_pattern;
143} *slp_oprnd_info;
144
145
d092494c
IR
146/* Allocate operands info for NOPS operands, and GROUP_SIZE def-stmts for each
147 operand. */
9771b263 148static vec<slp_oprnd_info>
d092494c
IR
149vect_create_oprnd_info (int nops, int group_size)
150{
151 int i;
152 slp_oprnd_info oprnd_info;
9771b263 153 vec<slp_oprnd_info> oprnds_info;
d092494c 154
9771b263 155 oprnds_info.create (nops);
d092494c
IR
156 for (i = 0; i < nops; i++)
157 {
158 oprnd_info = XNEW (struct _slp_oprnd_info);
9771b263 159 oprnd_info->def_stmts.create (group_size);
d092494c 160 oprnd_info->first_dt = vect_uninitialized_def;
793d9a16 161 oprnd_info->first_op_type = NULL_TREE;
d092494c 162 oprnd_info->first_pattern = false;
effb52da 163 oprnd_info->second_pattern = false;
9771b263 164 oprnds_info.quick_push (oprnd_info);
d092494c
IR
165 }
166
167 return oprnds_info;
168}
169
170
d3cfd39e
JJ
171/* Free operands info. */
172
d092494c 173static void
9771b263 174vect_free_oprnd_info (vec<slp_oprnd_info> &oprnds_info)
d092494c
IR
175{
176 int i;
177 slp_oprnd_info oprnd_info;
178
9771b263 179 FOR_EACH_VEC_ELT (oprnds_info, i, oprnd_info)
d3cfd39e 180 {
9771b263 181 oprnd_info->def_stmts.release ();
d3cfd39e
JJ
182 XDELETE (oprnd_info);
183 }
d092494c 184
9771b263 185 oprnds_info.release ();
d092494c
IR
186}
187
188
d755c7ef
RB
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. */
191
b210f45f 192int
355fe088 193vect_get_place_in_interleaving_chain (gimple *stmt, gimple *first_stmt)
d755c7ef 194{
355fe088 195 gimple *next_stmt = first_stmt;
d755c7ef
RB
196 int result = 0;
197
198 if (first_stmt != GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt)))
199 return -1;
200
201 do
202 {
203 if (next_stmt == stmt)
204 return result;
d755c7ef 205 next_stmt = GROUP_NEXT_ELEMENT (vinfo_for_stmt (next_stmt));
c8047699
RB
206 if (next_stmt)
207 result += GROUP_GAP (vinfo_for_stmt (next_stmt));
d755c7ef
RB
208 }
209 while (next_stmt);
210
211 return -1;
212}
213
018b2744
RS
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). */
219
f1739b48 220bool
018b2744 221can_duplicate_and_interleave_p (unsigned int count, machine_mode elt_mode,
f1739b48
RS
222 unsigned int *nvectors_out,
223 tree *vector_type_out,
224 tree *permutes)
018b2744
RS
225{
226 poly_int64 elt_bytes = count * GET_MODE_SIZE (elt_mode);
227 poly_int64 nelts;
228 unsigned int nvectors = 1;
229 for (;;)
230 {
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))
235 {
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)))
240 {
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)
245 {
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);
250 }
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))
255 {
256 if (nvectors_out)
257 *nvectors_out = nvectors;
258 if (vector_type_out)
259 *vector_type_out = vector_type;
260 if (permutes)
261 {
262 permutes[0] = vect_gen_perm_mask_checked (vector_type,
263 indices1);
264 permutes[1] = vect_gen_perm_mask_checked (vector_type,
265 indices2);
266 }
267 return true;
268 }
269 }
270 }
271 if (!multiple_p (elt_bytes, 2, &elt_bytes))
272 return false;
273 nvectors *= 2;
274 }
275}
d755c7ef 276
d092494c
IR
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
4cecd659 279 the SLP group (stored in OPRNDS_INFO). This function tries to match stmts
018b2744
RS
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.
4cecd659
BC
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. */
4cecd659
BC
289static int
290vect_get_and_check_slp_defs (vec_info *vinfo, unsigned char *swap,
018b2744 291 vec<gimple *> stmts, unsigned stmt_num,
4cecd659 292 vec<slp_oprnd_info> *oprnds_info)
ebfd146a 293{
018b2744 294 gimple *stmt = stmts[stmt_num];
ebfd146a
IR
295 tree oprnd;
296 unsigned int i, number_of_oprnds;
355fe088 297 gimple *def_stmt;
d092494c 298 enum vect_def_type dt = vect_uninitialized_def;
d092494c 299 bool pattern = false;
abf9bfbc 300 slp_oprnd_info oprnd_info;
b0b4483e
RB
301 int first_op_idx = 1;
302 bool commutative = false;
303 bool first_op_cond = false;
effb52da
RB
304 bool first = stmt_num == 0;
305 bool second = stmt_num == 1;
b8698a0f 306
d092494c 307 if (is_gimple_call (stmt))
190c2236
JJ
308 {
309 number_of_oprnds = gimple_call_num_args (stmt);
b0b4483e 310 first_op_idx = 3;
190c2236 311 }
f7e531cf
IR
312 else if (is_gimple_assign (stmt))
313 {
b0b4483e 314 enum tree_code code = gimple_assign_rhs_code (stmt);
f7e531cf 315 number_of_oprnds = gimple_num_ops (stmt) - 1;
4cecd659
BC
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. */
a414c77f
IE
318 if (gimple_assign_rhs_code (stmt) == COND_EXPR
319 && COMPARISON_CLASS_P (gimple_assign_rhs1 (stmt)))
b0b4483e
RB
320 {
321 first_op_cond = true;
b0b4483e
RB
322 number_of_oprnds++;
323 }
324 else
325 commutative = commutative_tree_code (code);
f7e531cf 326 }
d092494c 327 else
b0b4483e 328 return -1;
ebfd146a 329
4cecd659
BC
330 bool swapped = (*swap != 0);
331 gcc_assert (!swapped || first_op_cond);
ebfd146a
IR
332 for (i = 0; i < number_of_oprnds; i++)
333 {
b0b4483e
RB
334again:
335 if (first_op_cond)
f7e531cf 336 {
4cecd659
BC
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];
340
341 if (i < 2)
342 oprnd = TREE_OPERAND (gimple_op (stmt, first_op_idx), map[i]);
b0b4483e 343 else
4cecd659 344 oprnd = gimple_op (stmt, map[i]);
f7e531cf
IR
345 }
346 else
4cecd659 347 oprnd = gimple_op (stmt, first_op_idx + (swapped ? !i : i));
f7e531cf 348
9771b263 349 oprnd_info = (*oprnds_info)[i];
ebfd146a 350
81c40241 351 if (!vect_is_simple_use (oprnd, vinfo, &def_stmt, &dt))
ebfd146a 352 {
73fbfcad 353 if (dump_enabled_p ())
ebfd146a 354 {
78c60e3d 355 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
3fc356dc 356 "Build SLP failed: can't analyze def for ");
78c60e3d 357 dump_generic_expr (MSG_MISSED_OPTIMIZATION, TDF_SLIM, oprnd);
e645e942 358 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
ebfd146a
IR
359 }
360
b0b4483e 361 return -1;
ebfd146a
IR
362 }
363
a70d6342 364 /* Check if DEF_STMT is a part of a pattern in LOOP and get the def stmt
ff802fa1 365 from the pattern. Check that all the stmts of the node are in the
ebfd146a 366 pattern. */
f5709183 367 if (def_stmt && gimple_bb (def_stmt)
61d371eb 368 && vect_stmt_in_region_p (vinfo, def_stmt)
ebfd146a 369 && vinfo_for_stmt (def_stmt)
83197f37 370 && STMT_VINFO_IN_PATTERN_P (vinfo_for_stmt (def_stmt))
f5709183
IR
371 && !STMT_VINFO_RELEVANT (vinfo_for_stmt (def_stmt))
372 && !STMT_VINFO_LIVE_P (vinfo_for_stmt (def_stmt)))
ebfd146a 373 {
d092494c 374 pattern = true;
effb52da
RB
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)))
d092494c 380 {
b0b4483e
RB
381 if (i == 0
382 && !swapped
383 && commutative)
384 {
385 swapped = true;
386 goto again;
387 }
388
73fbfcad 389 if (dump_enabled_p ())
d092494c 390 {
78c60e3d
SS
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);
e645e942 395 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
d092494c 396 }
ebfd146a 397
b0b4483e 398 return 1;
ebfd146a
IR
399 }
400
401 def_stmt = STMT_VINFO_RELATED_STMT (vinfo_for_stmt (def_stmt));
d092494c 402 dt = STMT_VINFO_DEF_TYPE (vinfo_for_stmt (def_stmt));
ebfd146a 403
f7e531cf 404 if (dt == vect_unknown_def_type)
ebfd146a 405 {
73fbfcad 406 if (dump_enabled_p ())
78c60e3d 407 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
e645e942 408 "Unsupported pattern.\n");
b0b4483e 409 return -1;
ebfd146a
IR
410 }
411
412 switch (gimple_code (def_stmt))
413 {
81c40241
RB
414 case GIMPLE_PHI:
415 case GIMPLE_ASSIGN:
416 break;
417
418 default:
419 if (dump_enabled_p ())
420 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
421 "unsupported defining stmt:\n");
422 return -1;
ebfd146a
IR
423 }
424 }
425
effb52da
RB
426 if (second)
427 oprnd_info->second_pattern = pattern;
428
d092494c 429 if (first)
ebfd146a 430 {
d092494c
IR
431 oprnd_info->first_dt = dt;
432 oprnd_info->first_pattern = pattern;
793d9a16 433 oprnd_info->first_op_type = TREE_TYPE (oprnd);
ebfd146a 434 }
ebfd146a
IR
435 else
436 {
d092494c
IR
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. */
018b2744
RS
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))
ebfd146a 451 {
b0b4483e
RB
452 /* Try swapping operands if we got a mismatch. */
453 if (i == 0
454 && !swapped
455 && commutative)
456 {
457 swapped = true;
458 goto again;
459 }
460
abf9bfbc
RB
461 if (dump_enabled_p ())
462 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
e645e942 463 "Build SLP failed: different types\n");
d092494c 464
b0b4483e 465 return 1;
ebfd146a 466 }
018b2744
RS
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))))
a23644f2
RS
473 {
474 if (dump_enabled_p ())
475 {
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");
481 }
482 return -1;
483 }
018b2744
RS
484 }
485
486 /* Check the types of the definitions. */
487 switch (dt)
488 {
489 case vect_constant_def:
490 case vect_external_def:
ebfd146a 491 break;
b8698a0f 492
c78e3652 493 case vect_reduction_def:
e7baeb39 494 case vect_induction_def:
8644a673 495 case vect_internal_def:
abf9bfbc 496 oprnd_info->def_stmts.quick_push (def_stmt);
ebfd146a
IR
497 break;
498
499 default:
500 /* FORNOW: Not supported. */
73fbfcad 501 if (dump_enabled_p ())
ebfd146a 502 {
78c60e3d
SS
503 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
504 "Build SLP failed: illegal type of def ");
81c40241 505 dump_generic_expr (MSG_MISSED_OPTIMIZATION, TDF_SLIM, oprnd);
e645e942 506 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
ebfd146a
IR
507 }
508
b0b4483e 509 return -1;
ebfd146a
IR
510 }
511 }
512
b0b4483e
RB
513 /* Swap operands. */
514 if (swapped)
515 {
78810bd3
RB
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)
519 {
520 if (dump_enabled_p ())
521 {
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);
526 }
527 return -1;
528 }
529
b0b4483e
RB
530 if (first_op_cond)
531 {
532 tree cond = gimple_assign_rhs1 (stmt);
4cecd659
BC
533 enum tree_code code = TREE_CODE (cond);
534
535 /* Swap. */
536 if (*swap == 1)
537 {
538 swap_ssa_operands (stmt, &TREE_OPERAND (cond, 0),
539 &TREE_OPERAND (cond, 1));
540 TREE_SET_CODE (cond, swap_tree_comparison (code));
541 }
542 /* Invert. */
543 else
544 {
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);
551 }
b0b4483e
RB
552 }
553 else
554 swap_ssa_operands (stmt, gimple_assign_rhs1_ptr (stmt),
555 gimple_assign_rhs2_ptr (stmt));
78810bd3
RB
556 if (dump_enabled_p ())
557 {
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);
561 }
b0b4483e
RB
562 }
563
4cecd659 564 *swap = swapped;
b0b4483e 565 return 0;
ebfd146a
IR
566}
567
b161f2c9
RS
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. */
574
575static bool
576vect_record_max_nunits (vec_info *vinfo, gimple *stmt, unsigned int group_size,
4b6068ea 577 tree vectype, poly_uint64 *max_nunits)
b161f2c9
RS
578{
579 if (!vectype)
580 {
581 if (dump_enabled_p ())
582 {
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");
587 }
588 /* Fatal mismatch. */
589 return false;
590 }
591
592 /* If populating the vector type requires unrolling then fail
593 before adjusting *max_nunits for basic-block vectorization. */
4b6068ea
RS
594 poly_uint64 nunits = TYPE_VECTOR_SUBPARTS (vectype);
595 unsigned HOST_WIDE_INT const_nunits;
b161f2c9 596 if (is_a <bb_vec_info> (vinfo)
4b6068ea
RS
597 && (!nunits.is_constant (&const_nunits)
598 || const_nunits > group_size))
b161f2c9
RS
599 {
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;
605 }
606
607 /* In case of multiple types we need to detect the smallest type. */
4b6068ea 608 vect_update_max_nunits (max_nunits, vectype);
b161f2c9
RS
609 return true;
610}
ebfd146a 611
1f3cb663
RS
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. */
618
619static bool
620vect_two_operations_perm_ok_p (vec<gimple *> stmts, unsigned int group_size,
621 tree vectype, tree_code alt_stmt_code)
622{
623 unsigned HOST_WIDE_INT count;
624 if (!TYPE_VECTOR_SUBPARTS (vectype).is_constant (&count))
625 return false;
626
627 vec_perm_builder sel (count, count, 1);
628 for (unsigned int i = 0; i < count; ++i)
629 {
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);
634 }
635 vec_perm_indices indices (sel, 2, count);
636 return can_vec_perm_const_p (TYPE_MODE (vectype), indices);
637}
638
6983e6b5
RB
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
4cecd659
BC
644 carried out or the stmts will never be vectorized by SLP.
645
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. */
ebfd146a
IR
652
653static bool
4cecd659 654vect_build_slp_tree_1 (vec_info *vinfo, unsigned char *swap,
355fe088 655 vec<gimple *> stmts, unsigned int group_size,
4b6068ea 656 unsigned nops, poly_uint64 *max_nunits,
97a1a642 657 bool *matches, bool *two_operators)
ebfd146a 658{
ebfd146a 659 unsigned int i;
355fe088 660 gimple *first_stmt = stmts[0], *stmt = stmts[0];
6876e5bc
RB
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;
f7e531cf 664 enum tree_code first_cond_code = ERROR_MARK;
ebfd146a 665 tree lhs;
6983e6b5 666 bool need_same_oprnds = false;
1f3cb663 667 tree vectype = NULL_TREE, first_op1 = NULL_TREE;
ebfd146a
IR
668 optab optab;
669 int icode;
ef4bddc2
RS
670 machine_mode optab_op2_mode;
671 machine_mode vec_mode;
355fe088 672 gimple *first_load = NULL, *prev_first_load = NULL;
d092494c 673
ebfd146a 674 /* For every stmt in NODE find its def stmt/s. */
9771b263 675 FOR_EACH_VEC_ELT (stmts, i, stmt)
ebfd146a 676 {
1f3cb663 677 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
4cecd659 678 swap[i] = 0;
6983e6b5
RB
679 matches[i] = false;
680
73fbfcad 681 if (dump_enabled_p ())
ebfd146a 682 {
78c60e3d
SS
683 dump_printf_loc (MSG_NOTE, vect_location, "Build SLP for ");
684 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
ebfd146a
IR
685 }
686
4b5caab7
IR
687 /* Fail to vectorize statements marked as unvectorizable. */
688 if (!STMT_VINFO_VECTORIZABLE (vinfo_for_stmt (stmt)))
689 {
73fbfcad 690 if (dump_enabled_p ())
4b5caab7 691 {
78c60e3d
SS
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);
4b5caab7 695 }
6983e6b5
RB
696 /* Fatal mismatch. */
697 matches[0] = false;
4b5caab7
IR
698 return false;
699 }
700
ebfd146a
IR
701 lhs = gimple_get_lhs (stmt);
702 if (lhs == NULL_TREE)
703 {
73fbfcad 704 if (dump_enabled_p ())
ebfd146a 705 {
78c60e3d
SS
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);
ebfd146a 710 }
6983e6b5
RB
711 /* Fatal mismatch. */
712 matches[0] = false;
ebfd146a
IR
713 return false;
714 }
715
1f3cb663
RS
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)))
b161f2c9 722 {
6983e6b5
RB
723 /* Fatal mismatch. */
724 matches[0] = false;
1f3cb663
RS
725 return false;
726 }
727
728 gcc_assert (vectype);
b8698a0f 729
538dd0b7 730 if (gcall *call_stmt = dyn_cast <gcall *> (stmt))
190c2236
JJ
731 {
732 rhs_code = CALL_EXPR;
538dd0b7
DM
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))
190c2236 738 {
73fbfcad 739 if (dump_enabled_p ())
190c2236 740 {
78c60e3d
SS
741 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
742 "Build SLP failed: unsupported call type ");
538dd0b7
DM
743 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
744 call_stmt, 0);
190c2236 745 }
6983e6b5
RB
746 /* Fatal mismatch. */
747 matches[0] = false;
190c2236
JJ
748 return false;
749 }
750 }
ebfd146a
IR
751 else
752 rhs_code = gimple_assign_rhs_code (stmt);
753
754 /* Check the operation. */
755 if (i == 0)
756 {
757 first_stmt_code = rhs_code;
758
b8698a0f 759 /* Shift arguments should be equal in all the packed stmts for a
ebfd146a
IR
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)
764 {
1f3cb663
RS
765 if (vectype == boolean_type_node)
766 {
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;
774 }
775
ebfd146a
IR
776 vec_mode = TYPE_MODE (vectype);
777
778 /* First see if we have a vector/vector shift. */
779 optab = optab_for_tree_code (rhs_code, vectype,
780 optab_vector);
781
782 if (!optab
947131ba 783 || optab_handler (optab, vec_mode) == CODE_FOR_nothing)
ebfd146a
IR
784 {
785 /* No vector/vector shift, try for a vector/scalar shift. */
786 optab = optab_for_tree_code (rhs_code, vectype,
787 optab_scalar);
788
789 if (!optab)
790 {
73fbfcad 791 if (dump_enabled_p ())
78c60e3d 792 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
e645e942 793 "Build SLP failed: no optab.\n");
6983e6b5
RB
794 /* Fatal mismatch. */
795 matches[0] = false;
ebfd146a
IR
796 return false;
797 }
947131ba 798 icode = (int) optab_handler (optab, vec_mode);
ebfd146a
IR
799 if (icode == CODE_FOR_nothing)
800 {
73fbfcad 801 if (dump_enabled_p ())
78c60e3d
SS
802 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
803 "Build SLP failed: "
e645e942 804 "op not supported by target.\n");
6983e6b5
RB
805 /* Fatal mismatch. */
806 matches[0] = false;
ebfd146a
IR
807 return false;
808 }
809 optab_op2_mode = insn_data[icode].operand[2].mode;
810 if (!VECTOR_MODE_P (optab_op2_mode))
811 {
812 need_same_oprnds = true;
813 first_op1 = gimple_assign_rhs2 (stmt);
814 }
815 }
816 }
36ba4aae
IR
817 else if (rhs_code == WIDEN_LSHIFT_EXPR)
818 {
819 need_same_oprnds = true;
820 first_op1 = gimple_assign_rhs2 (stmt);
821 }
ebfd146a
IR
822 }
823 else
824 {
6876e5bc
RB
825 if (first_stmt_code != rhs_code
826 && alt_stmt_code == ERROR_MARK)
827 alt_stmt_code = rhs_code;
ebfd146a
IR
828 if (first_stmt_code != rhs_code
829 && (first_stmt_code != IMAGPART_EXPR
830 || rhs_code != REALPART_EXPR)
831 && (first_stmt_code != REALPART_EXPR
69f11a13 832 || rhs_code != IMAGPART_EXPR)
6876e5bc
RB
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)
0d0293ac 840 && !(STMT_VINFO_GROUPED_ACCESS (vinfo_for_stmt (stmt))
69f11a13 841 && (first_stmt_code == ARRAY_REF
38000232 842 || first_stmt_code == BIT_FIELD_REF
69f11a13
IR
843 || first_stmt_code == INDIRECT_REF
844 || first_stmt_code == COMPONENT_REF
845 || first_stmt_code == MEM_REF)))
ebfd146a 846 {
73fbfcad 847 if (dump_enabled_p ())
ebfd146a 848 {
78c60e3d
SS
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);
6876e5bc
RB
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);
ebfd146a 857 }
6983e6b5
RB
858 /* Mismatch. */
859 continue;
ebfd146a 860 }
b8698a0f
L
861
862 if (need_same_oprnds
ebfd146a
IR
863 && !operand_equal_p (first_op1, gimple_assign_rhs2 (stmt), 0))
864 {
73fbfcad 865 if (dump_enabled_p ())
ebfd146a 866 {
78c60e3d
SS
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);
ebfd146a 871 }
6983e6b5
RB
872 /* Mismatch. */
873 continue;
ebfd146a 874 }
190c2236
JJ
875
876 if (rhs_code == CALL_EXPR)
877 {
355fe088 878 gimple *first_stmt = stmts[0];
190c2236
JJ
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))
884 {
73fbfcad 885 if (dump_enabled_p ())
190c2236 886 {
78c60e3d
SS
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);
190c2236 891 }
6983e6b5
RB
892 /* Mismatch. */
893 continue;
190c2236
JJ
894 }
895 }
ebfd146a
IR
896 }
897
0d0293ac
MM
898 /* Grouped store or load. */
899 if (STMT_VINFO_GROUPED_ACCESS (vinfo_for_stmt (stmt)))
ebfd146a
IR
900 {
901 if (REFERENCE_CLASS_P (lhs))
902 {
903 /* Store. */
6983e6b5 904 ;
ebfd146a 905 }
b5aeb3bb
IR
906 else
907 {
908 /* Load. */
e14c1050 909 first_load = GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt));
b5aeb3bb
IR
910 if (prev_first_load)
911 {
912 /* Check that there are no loads from different interleaving
6983e6b5
RB
913 chains in the same node. */
914 if (prev_first_load != first_load)
78c60e3d 915 {
73fbfcad 916 if (dump_enabled_p ())
b5aeb3bb 917 {
78c60e3d
SS
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);
b5aeb3bb 924 }
6983e6b5
RB
925 /* Mismatch. */
926 continue;
b5aeb3bb
IR
927 }
928 }
929 else
930 prev_first_load = first_load;
ebfd146a 931 }
0d0293ac 932 } /* Grouped access. */
ebfd146a
IR
933 else
934 {
935 if (TREE_CODE_CLASS (rhs_code) == tcc_reference)
936 {
0d0293ac 937 /* Not grouped load. */
73fbfcad 938 if (dump_enabled_p ())
ebfd146a 939 {
78c60e3d
SS
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);
ebfd146a
IR
943 }
944
0d0293ac 945 /* FORNOW: Not grouped loads are not supported. */
6983e6b5
RB
946 /* Fatal mismatch. */
947 matches[0] = false;
ebfd146a
IR
948 return false;
949 }
950
951 /* Not memory operation. */
952 if (TREE_CODE_CLASS (rhs_code) != tcc_binary
f7e531cf 953 && TREE_CODE_CLASS (rhs_code) != tcc_unary
effb52da 954 && TREE_CODE_CLASS (rhs_code) != tcc_expression
42fd8198 955 && TREE_CODE_CLASS (rhs_code) != tcc_comparison
190c2236 956 && rhs_code != CALL_EXPR)
ebfd146a 957 {
73fbfcad 958 if (dump_enabled_p ())
ebfd146a 959 {
78c60e3d
SS
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);
ebfd146a 964 }
6983e6b5
RB
965 /* Fatal mismatch. */
966 matches[0] = false;
ebfd146a
IR
967 return false;
968 }
969
4cecd659
BC
970 if (rhs_code == COND_EXPR)
971 {
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;
f7e531cf
IR
976
977 if (i == 0)
978 first_cond_code = TREE_CODE (cond_expr);
4cecd659
BC
979 else if (TREE_CODE_CLASS (cond_code) == tcc_comparison)
980 {
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);
984 }
985
986 if (first_cond_code == cond_code)
987 ;
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
995 {
996 if (dump_enabled_p ())
997 {
998 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
78c60e3d
SS
999 "Build SLP failed: different"
1000 " operation");
4cecd659 1001 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
78c60e3d 1002 stmt, 0);
4cecd659 1003 }
6983e6b5
RB
1004 /* Mismatch. */
1005 continue;
f7e531cf 1006 }
4cecd659 1007 }
ebfd146a 1008 }
6983e6b5
RB
1009
1010 matches[i] = true;
1011 }
1012
1013 for (i = 0; i < group_size; ++i)
1014 if (!matches[i])
1015 return false;
1016
6876e5bc
RB
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)
1021 {
1f3cb663
RS
1022 if (vectype == boolean_type_node
1023 || !vect_two_operations_perm_ok_p (stmts, group_size,
1024 vectype, alt_stmt_code))
6876e5bc
RB
1025 {
1026 for (i = 0; i < group_size; ++i)
1027 if (gimple_assign_rhs_code (stmts[i]) == alt_stmt_code)
1028 {
1029 matches[i] = false;
1030 if (dump_enabled_p ())
1031 {
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);
1041 }
1042 }
1043 return false;
1044 }
1045 *two_operators = true;
1046 }
1047
6983e6b5
RB
1048 return true;
1049}
1050
26d66f28
RB
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. */
1054struct bst_traits
1055{
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 (); }
1065};
1066inline hashval_t
1067bst_traits::hash (value_type x)
1068{
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 ();
1073}
1074inline bool
1075bst_traits::equal (value_type existing, value_type candidate)
1076{
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;
1083}
1084
f7300fff
RB
1085typedef hash_set <vec <gimple *>, bst_traits> scalar_stmts_set_t;
1086static scalar_stmts_set_t *bst_fail;
26d66f28 1087
68435eb2
RB
1088typedef hash_map <vec <gimple *>, slp_tree,
1089 simple_hashmap_traits <bst_traits, slp_tree> >
1090 scalar_stmts_to_slp_tree_map_t;
1091
26d66f28
RB
1092static slp_tree
1093vect_build_slp_tree_2 (vec_info *vinfo,
1094 vec<gimple *> stmts, unsigned int group_size,
4b6068ea 1095 poly_uint64 *max_nunits,
26d66f28
RB
1096 vec<slp_tree> *loads,
1097 bool *matches, unsigned *npermutes, unsigned *tree_size,
1098 unsigned max_tree_size);
6983e6b5 1099
e403d17e 1100static slp_tree
310213d4 1101vect_build_slp_tree (vec_info *vinfo,
4b6068ea
RS
1102 vec<gimple *> stmts, unsigned int group_size,
1103 poly_uint64 *max_nunits, vec<slp_tree> *loads,
1428105c
RB
1104 bool *matches, unsigned *npermutes, unsigned *tree_size,
1105 unsigned max_tree_size)
26d66f28
RB
1106{
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)
1116 {
1117 vec <gimple *> x;
1118 x.create (stmts.length ());
1119 x.splice (stmts);
1120 bst_fail->add (x);
1121 }
1122 return res;
1123}
1124
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. */
1131
1132static slp_tree
1133vect_build_slp_tree_2 (vec_info *vinfo,
1134 vec<gimple *> stmts, unsigned int group_size,
4b6068ea 1135 poly_uint64 *max_nunits,
26d66f28
RB
1136 vec<slp_tree> *loads,
1137 bool *matches, unsigned *npermutes, unsigned *tree_size,
1138 unsigned max_tree_size)
6983e6b5 1139{
4b6068ea
RS
1140 unsigned nops, i, this_tree_size = 0;
1141 poly_uint64 this_max_nunits = *max_nunits;
355fe088 1142 gimple *stmt;
e403d17e 1143 slp_tree node;
6983e6b5 1144
6983e6b5
RB
1145 matches[0] = false;
1146
e403d17e 1147 stmt = stmts[0];
6983e6b5
RB
1148 if (is_gimple_call (stmt))
1149 nops = gimple_call_num_args (stmt);
1150 else if (is_gimple_assign (stmt))
1151 {
1152 nops = gimple_num_ops (stmt) - 1;
1153 if (gimple_assign_rhs_code (stmt) == COND_EXPR)
1154 nops++;
ebfd146a 1155 }
e7baeb39
RB
1156 else if (gimple_code (stmt) == GIMPLE_PHI)
1157 nops = 0;
6983e6b5 1158 else
e403d17e 1159 return NULL;
6983e6b5 1160
c78e3652
RB
1161 /* If the SLP node is a PHI (induction or reduction), terminate
1162 the recursion. */
e7baeb39
RB
1163 if (gimple_code (stmt) == GIMPLE_PHI)
1164 {
b161f2c9
RS
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;
1170
719488f8 1171 vect_def_type def_type = STMT_VINFO_DEF_TYPE (vinfo_for_stmt (stmt));
c78e3652 1172 /* Induction from different IVs is not supported. */
719488f8
RB
1173 if (def_type == vect_induction_def)
1174 {
1175 FOR_EACH_VEC_ELT (stmts, i, stmt)
1176 if (stmt != stmts[0])
1177 return NULL;
1178 }
1179 else
1180 {
1181 /* Else def types have to match. */
1182 FOR_EACH_VEC_ELT (stmts, i, stmt)
1183 {
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;
1190 }
1191 }
e7baeb39
RB
1192 node = vect_create_new_slp_node (stmts);
1193 return node;
1194 }
1195
1196
6876e5bc 1197 bool two_operators = false;
4cecd659
BC
1198 unsigned char *swap = XALLOCAVEC (unsigned char, group_size);
1199 if (!vect_build_slp_tree_1 (vinfo, swap,
e403d17e
RB
1200 stmts, group_size, nops,
1201 &this_max_nunits, matches, &two_operators))
1202 return NULL;
ebfd146a 1203
6983e6b5
RB
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))))
ebfd146a 1207 {
e403d17e
RB
1208 *max_nunits = this_max_nunits;
1209 node = vect_create_new_slp_node (stmts);
1210 loads->safe_push (node);
1211 return node;
ebfd146a
IR
1212 }
1213
6983e6b5
RB
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;
e403d17e 1217 FOR_EACH_VEC_ELT (stmts, i, stmt)
6983e6b5 1218 {
4cecd659 1219 int res = vect_get_and_check_slp_defs (vinfo, &swap[i],
018b2744 1220 stmts, i, &oprnds_info);
4cecd659
BC
1221 if (res != 0)
1222 matches[(res == -1) ? 0 : i] = false;
1223 if (!matches[0])
1224 break;
6983e6b5 1225 }
b0b4483e
RB
1226 for (i = 0; i < group_size; ++i)
1227 if (!matches[i])
1228 {
1229 vect_free_oprnd_info (oprnds_info);
e403d17e 1230 return NULL;
b0b4483e 1231 }
6983e6b5 1232
e403d17e
RB
1233 auto_vec<slp_tree, 4> children;
1234 auto_vec<slp_tree> this_loads;
1235
1236 stmt = stmts[0];
6983e6b5 1237
26d66f28
RB
1238 if (tree_size)
1239 max_tree_size -= *tree_size;
1240
b8698a0f 1241 /* Create SLP_TREE nodes for the definition node/s. */
9771b263 1242 FOR_EACH_VEC_ELT (oprnds_info, i, oprnd_info)
ebfd146a 1243 {
d092494c 1244 slp_tree child;
e403d17e
RB
1245 unsigned old_nloads = this_loads.length ();
1246 unsigned old_tree_size = this_tree_size;
1247 unsigned int j;
b8698a0f 1248
e7baeb39 1249 if (oprnd_info->first_dt != vect_internal_def
c78e3652 1250 && oprnd_info->first_dt != vect_reduction_def
e7baeb39 1251 && oprnd_info->first_dt != vect_induction_def)
d092494c 1252 continue;
ebfd146a 1253
1428105c
RB
1254 if (++this_tree_size > max_tree_size)
1255 {
e403d17e
RB
1256 FOR_EACH_VEC_ELT (children, j, child)
1257 vect_free_slp_tree (child);
1428105c 1258 vect_free_oprnd_info (oprnds_info);
e403d17e 1259 return NULL;
1428105c
RB
1260 }
1261
e403d17e
RB
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)
6983e6b5 1267 {
3fc356dc
RB
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. */
995b6fe0
RB
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])))
3fc356dc 1276 {
3fc356dc
RB
1277 slp_tree grandchild;
1278
1279 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (child), j, grandchild)
603cca93 1280 if (SLP_TREE_DEF_TYPE (grandchild) == vect_internal_def)
3fc356dc
RB
1281 break;
1282 if (!grandchild)
1283 {
1284 /* Roll back. */
e403d17e
RB
1285 this_loads.truncate (old_nloads);
1286 this_tree_size = old_tree_size;
3fc356dc 1287 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (child), j, grandchild)
603cca93 1288 vect_free_slp_tree (grandchild);
3fc356dc
RB
1289 SLP_TREE_CHILDREN (child).truncate (0);
1290
1291 dump_printf_loc (MSG_NOTE, vect_location,
1292 "Building parent vector operands from "
1293 "scalars instead\n");
1294 oprnd_info->def_stmts = vNULL;
603cca93 1295 SLP_TREE_DEF_TYPE (child) = vect_external_def;
e403d17e 1296 children.safe_push (child);
3fc356dc
RB
1297 continue;
1298 }
1299 }
1300
6983e6b5 1301 oprnd_info->def_stmts = vNULL;
e403d17e 1302 children.safe_push (child);
6983e6b5
RB
1303 continue;
1304 }
1305
90dd6e3d
RB
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... */
310213d4 1314 if (is_a <bb_vec_info> (vinfo)
90dd6e3d
RB
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)))
1320 {
1321 dump_printf_loc (MSG_NOTE, vect_location,
1322 "Building vector operands from scalars\n");
e403d17e 1323 child = vect_create_new_slp_node (oprnd_info->def_stmts);
603cca93 1324 SLP_TREE_DEF_TYPE (child) = vect_external_def;
e403d17e
RB
1325 children.safe_push (child);
1326 oprnd_info->def_stmts = vNULL;
90dd6e3d
RB
1327 continue;
1328 }
1329
6983e6b5
RB
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)
6983e6b5
RB
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)
1346 {
85c5e2f5
RB
1347 /* See whether we can swap the matching or the non-matching
1348 stmt operands. */
1349 bool swap_not_matching = true;
1350 do
1351 {
1352 for (j = 0; j < group_size; ++j)
1353 {
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)))
1360 {
1361 if (!swap_not_matching)
1362 goto fail;
1363 swap_not_matching = false;
1364 break;
1365 }
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)))
1374 {
1375 if (!swap_not_matching)
1376 {
1377 if (dump_enabled_p ())
1378 {
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);
1385 }
1386 goto fail;
1387 }
1388 swap_not_matching = false;
1389 break;
1390 }
1391 }
1392 }
1393 while (j != group_size);
78810bd3 1394
6983e6b5 1395 /* Swap mismatched definition stmts. */
b0b4483e
RB
1396 dump_printf_loc (MSG_NOTE, vect_location,
1397 "Re-trying with swapped operands of stmts ");
e72baed7 1398 for (j = 0; j < group_size; ++j)
85c5e2f5 1399 if (matches[j] == !swap_not_matching)
6983e6b5 1400 {
6b4db501
MM
1401 std::swap (oprnds_info[0]->def_stmts[j],
1402 oprnds_info[1]->def_stmts[j]);
b0b4483e 1403 dump_printf (MSG_NOTE, "%d ", j);
6983e6b5 1404 }
b0b4483e 1405 dump_printf (MSG_NOTE, "\n");
74574669
RB
1406 /* And try again with scratch 'matches' ... */
1407 bool *tem = XALLOCAVEC (bool, group_size);
e403d17e
RB
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)
6983e6b5 1413 {
60f2b864
RB
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)
f47cda24 1421 {
e403d17e 1422 gimple *stmt = stmts[j];
f47cda24
RB
1423 gimple_set_plf (stmt, GF_PLF_1, false);
1424 }
1425 for (j = 0; j < group_size; ++j)
1426 {
e403d17e 1427 gimple *stmt = stmts[j];
85c5e2f5 1428 if (matches[j] == !swap_not_matching)
f47cda24
RB
1429 {
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);
1436 }
1437 }
1438 /* Verify we swap all duplicates or none. */
1439 if (flag_checking)
1440 for (j = 0; j < group_size; ++j)
60f2b864 1441 {
e403d17e 1442 gimple *stmt = stmts[j];
85c5e2f5
RB
1443 gcc_assert (gimple_plf (stmt, GF_PLF_1)
1444 == (matches[j] == !swap_not_matching));
60f2b864
RB
1445 }
1446
85c69b0b
RB
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. */
995b6fe0
RB
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])))
85c69b0b
RB
1455 {
1456 unsigned int j;
1457 slp_tree grandchild;
1458
1459 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (child), j, grandchild)
603cca93 1460 if (SLP_TREE_DEF_TYPE (grandchild) == vect_internal_def)
85c69b0b
RB
1461 break;
1462 if (!grandchild)
1463 {
1464 /* Roll back. */
e403d17e
RB
1465 this_loads.truncate (old_nloads);
1466 this_tree_size = old_tree_size;
85c69b0b
RB
1467 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (child), j, grandchild)
1468 vect_free_slp_tree (grandchild);
1469 SLP_TREE_CHILDREN (child).truncate (0);
1470
1471 dump_printf_loc (MSG_NOTE, vect_location,
1472 "Building parent vector operands from "
1473 "scalars instead\n");
1474 oprnd_info->def_stmts = vNULL;
603cca93 1475 SLP_TREE_DEF_TYPE (child) = vect_external_def;
e403d17e 1476 children.safe_push (child);
85c69b0b
RB
1477 continue;
1478 }
1479 }
1480
6983e6b5 1481 oprnd_info->def_stmts = vNULL;
e403d17e 1482 children.safe_push (child);
6983e6b5
RB
1483 continue;
1484 }
1485
1486 ++*npermutes;
1487 }
1488
78810bd3 1489fail:
e403d17e
RB
1490 gcc_assert (child == NULL);
1491 FOR_EACH_VEC_ELT (children, j, child)
1492 vect_free_slp_tree (child);
6983e6b5 1493 vect_free_oprnd_info (oprnds_info);
e403d17e 1494 return NULL;
ebfd146a
IR
1495 }
1496
e403d17e
RB
1497 vect_free_oprnd_info (oprnds_info);
1498
1428105c
RB
1499 if (tree_size)
1500 *tree_size += this_tree_size;
e403d17e
RB
1501 *max_nunits = this_max_nunits;
1502 loads->safe_splice (this_loads);
1428105c 1503
e403d17e
RB
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;
ebfd146a
IR
1508}
1509
78c60e3d 1510/* Dump a slp tree NODE using flags specified in DUMP_KIND. */
ebfd146a
IR
1511
1512static void
1a817418 1513vect_print_slp_tree (dump_flags_t dump_kind, location_t loc, slp_tree node)
ebfd146a
IR
1514{
1515 int i;
355fe088 1516 gimple *stmt;
d755c7ef 1517 slp_tree child;
ebfd146a 1518
603cca93
RB
1519 dump_printf_loc (dump_kind, loc, "node%s\n",
1520 SLP_TREE_DEF_TYPE (node) != vect_internal_def
1521 ? " (external)" : "");
9771b263 1522 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, stmt)
ebfd146a 1523 {
c2a12ca0 1524 dump_printf_loc (dump_kind, loc, "\tstmt %d ", i);
78c60e3d 1525 dump_gimple_stmt (dump_kind, TDF_SLIM, stmt, 0);
ebfd146a 1526 }
9771b263 1527 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
c2a12ca0 1528 vect_print_slp_tree (dump_kind, loc, child);
ebfd146a
IR
1529}
1530
1531
b8698a0f
L
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
ff802fa1 1534 J). Otherwise, MARK is PURE_SLP and J is -1, which indicates that all the
ebfd146a
IR
1535 stmts in NODE are to be marked. */
1536
1537static void
1538vect_mark_slp_stmts (slp_tree node, enum slp_vect_type mark, int j)
1539{
1540 int i;
355fe088 1541 gimple *stmt;
d755c7ef 1542 slp_tree child;
ebfd146a 1543
603cca93 1544 if (SLP_TREE_DEF_TYPE (node) != vect_internal_def)
ebfd146a
IR
1545 return;
1546
9771b263 1547 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, stmt)
ebfd146a
IR
1548 if (j < 0 || i == j)
1549 STMT_SLP_TYPE (vinfo_for_stmt (stmt)) = mark;
1550
9771b263 1551 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
d755c7ef 1552 vect_mark_slp_stmts (child, mark, j);
ebfd146a
IR
1553}
1554
1555
a70d6342
IR
1556/* Mark the statements of the tree rooted at NODE as relevant (vect_used). */
1557
1558static void
1559vect_mark_slp_stmts_relevant (slp_tree node)
1560{
1561 int i;
355fe088 1562 gimple *stmt;
a70d6342 1563 stmt_vec_info stmt_info;
d755c7ef 1564 slp_tree child;
a70d6342 1565
603cca93 1566 if (SLP_TREE_DEF_TYPE (node) != vect_internal_def)
a70d6342
IR
1567 return;
1568
9771b263 1569 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, stmt)
a70d6342
IR
1570 {
1571 stmt_info = vinfo_for_stmt (stmt);
b8698a0f 1572 gcc_assert (!STMT_VINFO_RELEVANT (stmt_info)
a70d6342
IR
1573 || STMT_VINFO_RELEVANT (stmt_info) == vect_used_in_scope);
1574 STMT_VINFO_RELEVANT (stmt_info) = vect_used_in_scope;
1575 }
1576
9771b263 1577 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
d755c7ef 1578 vect_mark_slp_stmts_relevant (child);
a70d6342
IR
1579}
1580
1581
b5aeb3bb
IR
1582/* Rearrange the statements of NODE according to PERMUTATION. */
1583
1584static void
1585vect_slp_rearrange_stmts (slp_tree node, unsigned int group_size,
01d8bf07 1586 vec<unsigned> permutation)
b5aeb3bb 1587{
355fe088
TS
1588 gimple *stmt;
1589 vec<gimple *> tmp_stmts;
d755c7ef
RB
1590 unsigned int i;
1591 slp_tree child;
b5aeb3bb 1592
9771b263 1593 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
d755c7ef 1594 vect_slp_rearrange_stmts (child, group_size, permutation);
b5aeb3bb 1595
9771b263
DN
1596 gcc_assert (group_size == SLP_TREE_SCALAR_STMTS (node).length ());
1597 tmp_stmts.create (group_size);
d755c7ef 1598 tmp_stmts.quick_grow_cleared (group_size);
b5aeb3bb 1599
9771b263 1600 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, stmt)
d755c7ef 1601 tmp_stmts[permutation[i]] = stmt;
b5aeb3bb 1602
9771b263 1603 SLP_TREE_SCALAR_STMTS (node).release ();
b5aeb3bb
IR
1604 SLP_TREE_SCALAR_STMTS (node) = tmp_stmts;
1605}
1606
1607
b266b968
RB
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. */
1611
1612static bool
1613vect_attempt_slp_rearrange_stmts (slp_instance slp_instn)
1614{
1615 unsigned int group_size = SLP_INSTANCE_GROUP_SIZE (slp_instn);
1616 unsigned int i, j;
b266b968
RB
1617 unsigned int lidx;
1618 slp_tree node, load;
1619
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)
1626 {
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;
1632 }
1633
1634 /* Check that the loads in the first sequence are different and there
1635 are no gaps between them. */
7ba9e72d 1636 auto_sbitmap load_index (group_size);
b266b968
RB
1637 bitmap_clear (load_index);
1638 FOR_EACH_VEC_ELT (node->load_permutation, i, lidx)
1639 {
41eefe13 1640 if (lidx >= group_size)
7ba9e72d 1641 return false;
b266b968 1642 if (bitmap_bit_p (load_index, lidx))
7ba9e72d
TS
1643 return false;
1644
b266b968
RB
1645 bitmap_set_bit (load_index, lidx);
1646 }
1647 for (i = 0; i < group_size; i++)
1648 if (!bitmap_bit_p (load_index, i))
7ba9e72d 1649 return false;
b266b968
RB
1650
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);
1657
1658 /* We are done, no actual permutations need to be generated. */
d9f21f6a 1659 poly_uint64 unrolling_factor = SLP_INSTANCE_UNROLLING_FACTOR (slp_instn);
b266b968 1660 FOR_EACH_VEC_ELT (SLP_INSTANCE_LOADS (slp_instn), i, node)
c4e360f4
RB
1661 {
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. */
d9f21f6a 1666 if (known_eq (unrolling_factor, 1U)
c4e360f4
RB
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 ();
cbd400b4
RB
1670 else
1671 for (j = 0; j < SLP_TREE_LOAD_PERMUTATION (node).length (); ++j)
1672 SLP_TREE_LOAD_PERMUTATION (node)[j] = j;
c4e360f4
RB
1673 }
1674
b266b968
RB
1675 return true;
1676}
1677
01d8bf07
RB
1678/* Check if the required load permutations in the SLP instance
1679 SLP_INSTN are supported. */
ebfd146a
IR
1680
1681static bool
01d8bf07 1682vect_supported_load_permutation_p (slp_instance slp_instn)
ebfd146a 1683{
01d8bf07
RB
1684 unsigned int group_size = SLP_INSTANCE_GROUP_SIZE (slp_instn);
1685 unsigned int i, j, k, next;
6983e6b5 1686 slp_tree node;
a5b50aa1 1687 gimple *stmt, *load, *next_load;
ebfd146a 1688
73fbfcad 1689 if (dump_enabled_p ())
ebfd146a 1690 {
78c60e3d 1691 dump_printf_loc (MSG_NOTE, vect_location, "Load permutation ");
01d8bf07
RB
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
bddc974e
TJ
1697 for (k = 0; k < group_size; ++k)
1698 dump_printf (MSG_NOTE, "%d ", k);
e645e942 1699 dump_printf (MSG_NOTE, "\n");
ebfd146a
IR
1700 }
1701
b5aeb3bb
IR
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
0d0293ac 1704 grouped stores). The only condition we need to check is that all the
b5aeb3bb
IR
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). */
1708
1709 /* Check that all the load nodes are of the same size. */
01d8bf07 1710 /* ??? Can't we assert this? */
9771b263 1711 FOR_EACH_VEC_ELT (SLP_INSTANCE_LOADS (slp_instn), i, node)
6983e6b5
RB
1712 if (SLP_TREE_SCALAR_STMTS (node).length () != (unsigned) group_size)
1713 return false;
2200fc49 1714
b5aeb3bb 1715 node = SLP_INSTANCE_TREE (slp_instn);
9771b263 1716 stmt = SLP_TREE_SCALAR_STMTS (node)[0];
b5aeb3bb 1717
b010117a 1718 /* Reduction (there are no data-refs in the root).
b266b968 1719 In reduction chain the order of the loads is not important. */
b010117a
IR
1720 if (!STMT_VINFO_DATA_REF (vinfo_for_stmt (stmt))
1721 && !GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt)))
c4e360f4 1722 vect_attempt_slp_rearrange_stmts (slp_instn);
b5aeb3bb 1723
6aa904c4
IR
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. */
01d8bf07 1727 if (STMT_VINFO_BB_VINFO (vinfo_for_stmt (stmt)))
6aa904c4 1728 {
240a94da
RB
1729 /* Check whether the loads in an instance form a subchain and thus
1730 no permutation is necessary. */
9771b263 1731 FOR_EACH_VEC_ELT (SLP_INSTANCE_LOADS (slp_instn), i, node)
6aa904c4 1732 {
9626d143
RB
1733 if (!SLP_TREE_LOAD_PERMUTATION (node).exists ())
1734 continue;
240a94da 1735 bool subchain_p = true;
6aa904c4 1736 next_load = NULL;
9771b263 1737 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), j, load)
6aa904c4 1738 {
5b5826c4
RB
1739 if (j != 0
1740 && (next_load != load
1741 || GROUP_GAP (vinfo_for_stmt (load)) != 1))
240a94da
RB
1742 {
1743 subchain_p = false;
1744 break;
1745 }
6aa904c4
IR
1746 next_load = GROUP_NEXT_ELEMENT (vinfo_for_stmt (load));
1747 }
240a94da
RB
1748 if (subchain_p)
1749 SLP_TREE_LOAD_PERMUTATION (node).release ();
1750 else
1751 {
fe73a332
RB
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));
928686b1 1755 unsigned HOST_WIDE_INT nunits;
fe73a332
RB
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. */
928686b1
RS
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)))
fe73a332
RB
1765 {
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;
1770 }
1771
240a94da
RB
1772 /* Verify the permutation can be generated. */
1773 vec<tree> tem;
29afecdf 1774 unsigned n_perms;
240a94da 1775 if (!vect_transform_slp_perm_load (node, tem, NULL,
29afecdf 1776 1, slp_instn, true, &n_perms))
240a94da
RB
1777 {
1778 dump_printf_loc (MSG_MISSED_OPTIMIZATION,
1779 vect_location,
1780 "unsupported load permutation\n");
1781 return false;
1782 }
1783 }
6aa904c4 1784 }
01d8bf07 1785 return true;
6aa904c4
IR
1786 }
1787
31bee964
RB
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. */
29afecdf 1793 unsigned n_perms;
d9f21f6a
RS
1794 poly_uint64 test_vf
1795 = force_common_multiple (SLP_INSTANCE_UNROLLING_FACTOR (slp_instn),
31bee964 1796 LOOP_VINFO_VECT_FACTOR
d9f21f6a 1797 (STMT_VINFO_LOOP_VINFO (vinfo_for_stmt (stmt))));
01d8bf07
RB
1798 FOR_EACH_VEC_ELT (SLP_INSTANCE_LOADS (slp_instn), i, node)
1799 if (node->load_permutation.exists ()
31bee964
RB
1800 && !vect_transform_slp_perm_load (node, vNULL, NULL, test_vf,
1801 slp_instn, true, &n_perms))
01d8bf07 1802 return false;
9b999e8c 1803
01d8bf07 1804 return true;
ebfd146a
IR
1805}
1806
1807
e4a707c4 1808/* Find the last store in SLP INSTANCE. */
ff802fa1 1809
64900538 1810gimple *
2e8ab70c 1811vect_find_last_scalar_stmt_in_slp (slp_tree node)
e4a707c4 1812{
355fe088 1813 gimple *last = NULL, *stmt;
e4a707c4 1814
2e8ab70c
RB
1815 for (int i = 0; SLP_TREE_SCALAR_STMTS (node).iterate (i, &stmt); i++)
1816 {
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);
1822 }
e4a707c4 1823
2e8ab70c 1824 return last;
e4a707c4
IR
1825}
1826
1ba91a49
AL
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. */
1832
1833static gimple *
1834vect_split_slp_store_group (gimple *first_stmt, unsigned group1_size)
1835{
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;
1842
1843 gimple *stmt = first_stmt;
1844 for (unsigned i = group1_size; i > 1; i--)
1845 {
1846 stmt = GROUP_NEXT_ELEMENT (vinfo_for_stmt (stmt));
1847 gcc_assert (GROUP_GAP (vinfo_for_stmt (stmt)) == 1);
1848 }
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;
1852
1853 GROUP_SIZE (vinfo_for_stmt (group2)) = group2_size;
1854 for (stmt = group2; stmt; stmt = GROUP_NEXT_ELEMENT (vinfo_for_stmt (stmt)))
1855 {
1856 GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt)) = group2;
1857 gcc_assert (GROUP_GAP (vinfo_for_stmt (stmt)) == 1);
1858 }
1859
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;
1864
1865 /* GROUP_GAP of the first group now has to skip over the second group too. */
1866 GROUP_GAP (first_vinfo) += group2_size;
1867
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);
1871
1872 return group2;
1873}
1874
4b6068ea
RS
1875/* Calculate the unrolling factor for an SLP instance with GROUP_SIZE
1876 statements and a vector of NUNITS elements. */
1877
1878static poly_uint64
1879calculate_unrolling_factor (poly_uint64 nunits, unsigned int group_size)
1880{
1881 return exact_div (common_multiple (nunits, group_size), group_size);
1882}
1883
0d0293ac 1884/* Analyze an SLP instance starting from a group of grouped stores. Call
b8698a0f 1885 vect_build_slp_tree to build a tree of packed stmts if possible.
ebfd146a
IR
1886 Return FALSE if it's impossible to SLP any stmt in the loop. */
1887
1888static bool
310213d4 1889vect_analyze_slp_instance (vec_info *vinfo,
355fe088 1890 gimple *stmt, unsigned max_tree_size)
ebfd146a
IR
1891{
1892 slp_instance new_instance;
d092494c 1893 slp_tree node;
e14c1050 1894 unsigned int group_size = GROUP_SIZE (vinfo_for_stmt (stmt));
b5aeb3bb 1895 tree vectype, scalar_type = NULL_TREE;
355fe088 1896 gimple *next;
1ba91a49 1897 unsigned int i;
9771b263 1898 vec<slp_tree> loads;
b5aeb3bb 1899 struct data_reference *dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (stmt));
355fe088 1900 vec<gimple *> scalar_stmts;
b5aeb3bb 1901
b010117a 1902 if (GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt)))
b5aeb3bb 1903 {
b010117a
IR
1904 if (dr)
1905 {
1906 scalar_type = TREE_TYPE (DR_REF (dr));
1907 vectype = get_vectype_for_scalar_type (scalar_type);
1908 }
1909 else
1910 {
310213d4 1911 gcc_assert (is_a <loop_vec_info> (vinfo));
b010117a
IR
1912 vectype = STMT_VINFO_VECTYPE (vinfo_for_stmt (stmt));
1913 }
1914
e14c1050 1915 group_size = GROUP_SIZE (vinfo_for_stmt (stmt));
b5aeb3bb
IR
1916 }
1917 else
1918 {
310213d4 1919 gcc_assert (is_a <loop_vec_info> (vinfo));
b5aeb3bb 1920 vectype = STMT_VINFO_VECTYPE (vinfo_for_stmt (stmt));
310213d4 1921 group_size = as_a <loop_vec_info> (vinfo)->reductions.length ();
b5aeb3bb 1922 }
b8698a0f 1923
ebfd146a
IR
1924 if (!vectype)
1925 {
73fbfcad 1926 if (dump_enabled_p ())
ebfd146a 1927 {
78c60e3d
SS
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);
e645e942 1931 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
ebfd146a 1932 }
b5aeb3bb 1933
ebfd146a
IR
1934 return false;
1935 }
4b6068ea 1936 poly_uint64 nunits = TYPE_VECTOR_SUBPARTS (vectype);
a70d6342 1937
0d0293ac 1938 /* Create a node (a root of the SLP tree) for the packed grouped stores. */
9771b263 1939 scalar_stmts.create (group_size);
ebfd146a 1940 next = stmt;
b010117a 1941 if (GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt)))
ebfd146a 1942 {
b5aeb3bb
IR
1943 /* Collect the stores and store them in SLP_TREE_SCALAR_STMTS. */
1944 while (next)
1945 {
f7e531cf
IR
1946 if (STMT_VINFO_IN_PATTERN_P (vinfo_for_stmt (next))
1947 && STMT_VINFO_RELATED_STMT (vinfo_for_stmt (next)))
9771b263
DN
1948 scalar_stmts.safe_push (
1949 STMT_VINFO_RELATED_STMT (vinfo_for_stmt (next)));
f7e531cf 1950 else
9771b263 1951 scalar_stmts.safe_push (next);
e14c1050 1952 next = GROUP_NEXT_ELEMENT (vinfo_for_stmt (next));
b5aeb3bb 1953 }
14a61437
RB
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;
b5aeb3bb
IR
1959 }
1960 else
1961 {
1962 /* Collect reduction statements. */
310213d4 1963 vec<gimple *> reductions = as_a <loop_vec_info> (vinfo)->reductions;
9771b263
DN
1964 for (i = 0; reductions.iterate (i, &next); i++)
1965 scalar_stmts.safe_push (next);
ebfd146a
IR
1966 }
1967
9771b263 1968 loads.create (group_size);
ebfd146a
IR
1969
1970 /* Build the tree for the SLP instance. */
89d390e5
RB
1971 bool *matches = XALLOCAVEC (bool, group_size);
1972 unsigned npermutes = 0;
f7300fff 1973 bst_fail = new scalar_stmts_set_t ();
4b6068ea 1974 poly_uint64 max_nunits = nunits;
e569db5f 1975 node = vect_build_slp_tree (vinfo, scalar_stmts, group_size,
4b6068ea 1976 &max_nunits, &loads, matches, &npermutes,
e569db5f 1977 NULL, max_tree_size);
26d66f28 1978 delete bst_fail;
e569db5f 1979 if (node != NULL)
ebfd146a 1980 {
4ef69dfc 1981 /* Calculate the unrolling factor based on the smallest type. */
d9f21f6a 1982 poly_uint64 unrolling_factor
4b6068ea 1983 = calculate_unrolling_factor (max_nunits, group_size);
b8698a0f 1984
d9f21f6a 1985 if (maybe_ne (unrolling_factor, 1U)
e569db5f
VK
1986 && is_a <bb_vec_info> (vinfo))
1987 {
4b6068ea
RS
1988 unsigned HOST_WIDE_INT const_max_nunits;
1989 if (!max_nunits.is_constant (&const_max_nunits)
1990 || const_max_nunits > group_size)
1991 {
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;
2000 }
e569db5f 2001 /* Fatal mismatch. */
4b6068ea 2002 matches[group_size / const_max_nunits * const_max_nunits] = false;
e569db5f
VK
2003 vect_free_slp_tree (node);
2004 loads.release ();
2005 }
2006 else
2007 {
4ef69dfc
IR
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;
ebfd146a 2012 SLP_INSTANCE_UNROLLING_FACTOR (new_instance) = unrolling_factor;
ebfd146a 2013 SLP_INSTANCE_LOADS (new_instance) = loads;
abf9bfbc
RB
2014
2015 /* Compute the load permutation. */
2016 slp_tree load_node;
2017 bool loads_permuted = false;
abf9bfbc
RB
2018 FOR_EACH_VEC_ELT (loads, i, load_node)
2019 {
01d8bf07 2020 vec<unsigned> load_permutation;
abf9bfbc 2021 int j;
355fe088 2022 gimple *load, *first_stmt;
01d8bf07
RB
2023 bool this_load_permuted = false;
2024 load_permutation.create (group_size);
6983e6b5
RB
2025 first_stmt = GROUP_FIRST_ELEMENT
2026 (vinfo_for_stmt (SLP_TREE_SCALAR_STMTS (load_node)[0]));
abf9bfbc
RB
2027 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (load_node), j, load)
2028 {
e569db5f
VK
2029 int load_place = vect_get_place_in_interleaving_chain
2030 (load, first_stmt);
6983e6b5
RB
2031 gcc_assert (load_place != -1);
2032 if (load_place != j)
01d8bf07 2033 this_load_permuted = true;
abf9bfbc
RB
2034 load_permutation.safe_push (load_place);
2035 }
fe2bef71
RB
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. */
d9f21f6a 2040 && (known_eq (unrolling_factor, 1U)
fe2bef71
RB
2041 || (group_size == GROUP_SIZE (vinfo_for_stmt (first_stmt))
2042 && GROUP_GAP (vinfo_for_stmt (first_stmt)) == 0)))
01d8bf07
RB
2043 {
2044 load_permutation.release ();
2045 continue;
2046 }
2047 SLP_TREE_LOAD_PERMUTATION (load_node) = load_permutation;
2048 loads_permuted = true;
abf9bfbc 2049 }
6aa904c4
IR
2050
2051 if (loads_permuted)
ebfd146a 2052 {
01d8bf07 2053 if (!vect_supported_load_permutation_p (new_instance))
ebfd146a 2054 {
73fbfcad 2055 if (dump_enabled_p ())
ebfd146a 2056 {
e645e942 2057 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
78c60e3d
SS
2058 "Build SLP failed: unsupported load "
2059 "permutation ");
e569db5f
VK
2060 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION,
2061 TDF_SLIM, stmt, 0);
ebfd146a 2062 }
ebfd146a
IR
2063 vect_free_slp_instance (new_instance);
2064 return false;
2065 }
ebfd146a 2066 }
ebfd146a 2067
e569db5f 2068 /* If the loads and stores can be handled with load/store-lan
bb0f5ca7
AL
2069 instructions do not generate this SLP instance. */
2070 if (is_a <loop_vec_info> (vinfo)
2071 && loads_permuted
7e11fc7f 2072 && dr && vect_store_lanes_supported (vectype, group_size, false))
bb0f5ca7
AL
2073 {
2074 slp_tree load_node;
2075 FOR_EACH_VEC_ELT (loads, i, load_node)
2076 {
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);
e569db5f
VK
2080 /* Use SLP for strided accesses (or if we
2081 can't load-lanes). */
bb0f5ca7
AL
2082 if (STMT_VINFO_STRIDED_P (stmt_vinfo)
2083 || ! vect_load_lanes_supported
2084 (STMT_VINFO_VECTYPE (stmt_vinfo),
7e11fc7f 2085 GROUP_SIZE (stmt_vinfo), false))
bb0f5ca7
AL
2086 break;
2087 }
2088 if (i == loads.length ())
2089 {
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;
2096 }
2097 }
2098
310213d4 2099 vinfo->slp_instances.safe_push (new_instance);
b8698a0f 2100
73fbfcad 2101 if (dump_enabled_p ())
c2a12ca0
RB
2102 {
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);
2106 }
ebfd146a
IR
2107
2108 return true;
2109 }
e569db5f
VK
2110 }
2111 else
2112 {
ebfd146a
IR
2113 /* Failed to SLP. */
2114 /* Free the allocated memory. */
e403d17e 2115 scalar_stmts.release ();
9771b263 2116 loads.release ();
e569db5f 2117 }
b8698a0f 2118
1ba91a49 2119 /* For basic block SLP, try to break the group up into multiples of the
97a1a642 2120 vector size. */
4b6068ea 2121 unsigned HOST_WIDE_INT const_nunits;
1ba91a49
AL
2122 if (is_a <bb_vec_info> (vinfo)
2123 && GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt))
4b6068ea
RS
2124 && STMT_VINFO_GROUPED_ACCESS (vinfo_for_stmt (stmt))
2125 && nunits.is_constant (&const_nunits))
1ba91a49
AL
2126 {
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;
2131
4b6068ea 2132 if (i >= const_nunits && i < group_size)
1ba91a49
AL
2133 {
2134 /* Split into two groups at the first vector boundary before i. */
4b6068ea
RS
2135 gcc_assert ((const_nunits & (const_nunits - 1)) == 0);
2136 unsigned group1_size = i & ~(const_nunits - 1);
1ba91a49
AL
2137
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)
2143 {
4b6068ea 2144 i = group1_size + const_nunits;
1ba91a49 2145 if (i < group_size)
4b6068ea 2146 rest = vect_split_slp_store_group (rest, const_nunits);
1ba91a49
AL
2147 }
2148 if (i < group_size)
2149 res |= vect_analyze_slp_instance (vinfo, rest, max_tree_size);
2150 return res;
2151 }
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. */
2154 }
2155
a70d6342 2156 return false;
ebfd146a
IR
2157}
2158
2159
ff802fa1 2160/* Check if there are stmts in the loop can be vectorized using SLP. Build SLP
ebfd146a
IR
2161 trees of packed scalar stmts if SLP is possible. */
2162
2163bool
310213d4 2164vect_analyze_slp (vec_info *vinfo, unsigned max_tree_size)
ebfd146a
IR
2165{
2166 unsigned int i;
355fe088 2167 gimple *first_element;
ebfd146a 2168
73fbfcad 2169 if (dump_enabled_p ())
e645e942 2170 dump_printf_loc (MSG_NOTE, vect_location, "=== vect_analyze_slp ===\n");
ebfd146a 2171
0d0293ac 2172 /* Find SLP sequences starting from groups of grouped stores. */
310213d4 2173 FOR_EACH_VEC_ELT (vinfo->grouped_stores, i, first_element)
0630a4ec 2174 vect_analyze_slp_instance (vinfo, first_element, max_tree_size);
ebfd146a 2175
310213d4 2176 if (loop_vec_info loop_vinfo = dyn_cast <loop_vec_info> (vinfo))
b010117a 2177 {
310213d4
RB
2178 if (loop_vinfo->reduction_chains.length () > 0)
2179 {
2180 /* Find SLP sequences starting from reduction chains. */
2181 FOR_EACH_VEC_ELT (loop_vinfo->reduction_chains, i, first_element)
0630a4ec 2182 if (! vect_analyze_slp_instance (vinfo, first_element,
310213d4 2183 max_tree_size))
6b5e165b
RB
2184 {
2185 /* Dissolve reduction chain group. */
2186 gimple *next, *stmt = first_element;
2187 while (stmt)
2188 {
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;
2194 }
2195 STMT_VINFO_DEF_TYPE (vinfo_for_stmt (first_element))
2196 = vect_internal_def;
2197 }
310213d4 2198 }
b010117a 2199
310213d4 2200 /* Find SLP sequences starting from groups of reductions. */
0630a4ec
RB
2201 if (loop_vinfo->reductions.length () > 1)
2202 vect_analyze_slp_instance (vinfo, loop_vinfo->reductions[0],
2203 max_tree_size);
310213d4 2204 }
b5aeb3bb 2205
ebfd146a
IR
2206 return true;
2207}
2208
2209
2210/* For each possible SLP instance decide whether to SLP it and calculate overall
437f4a00
IR
2211 unrolling factor needed to SLP the loop. Return TRUE if decided to SLP at
2212 least one instance. */
ebfd146a 2213
437f4a00 2214bool
ebfd146a
IR
2215vect_make_slp_decision (loop_vec_info loop_vinfo)
2216{
d9f21f6a
RS
2217 unsigned int i;
2218 poly_uint64 unrolling_factor = 1;
9771b263 2219 vec<slp_instance> slp_instances = LOOP_VINFO_SLP_INSTANCES (loop_vinfo);
ebfd146a
IR
2220 slp_instance instance;
2221 int decided_to_slp = 0;
2222
73fbfcad 2223 if (dump_enabled_p ())
e645e942
TJ
2224 dump_printf_loc (MSG_NOTE, vect_location, "=== vect_make_slp_decision ==="
2225 "\n");
ebfd146a 2226
9771b263 2227 FOR_EACH_VEC_ELT (slp_instances, i, instance)
ebfd146a
IR
2228 {
2229 /* FORNOW: SLP if you can. */
d9f21f6a
RS
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));
ebfd146a 2235
ff802fa1 2236 /* Mark all the stmts that belong to INSTANCE as PURE_SLP stmts. Later we
b8698a0f 2237 call vect_detect_hybrid_slp () to find stmts that need hybrid SLP and
ff802fa1 2238 loop-based vectorization. Such stmts will be marked as HYBRID. */
ebfd146a
IR
2239 vect_mark_slp_stmts (SLP_INSTANCE_TREE (instance), pure_slp, -1);
2240 decided_to_slp++;
2241 }
2242
2243 LOOP_VINFO_SLP_UNROLLING_FACTOR (loop_vinfo) = unrolling_factor;
2244
73fbfcad 2245 if (decided_to_slp && dump_enabled_p ())
d9f21f6a
RS
2246 {
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");
2252 }
437f4a00
IR
2253
2254 return (decided_to_slp > 0);
ebfd146a
IR
2255}
2256
2257
2258/* Find stmts that must be both vectorized and SLPed (since they feed stmts that
ff802fa1 2259 can't be SLPed) in the tree rooted at NODE. Mark such stmts as HYBRID. */
ebfd146a
IR
2260
2261static void
642fce57 2262vect_detect_hybrid_slp_stmts (slp_tree node, unsigned i, slp_vect_type stype)
ebfd146a 2263{
355fe088 2264 gimple *stmt = SLP_TREE_SCALAR_STMTS (node)[i];
ebfd146a 2265 imm_use_iterator imm_iter;
355fe088 2266 gimple *use_stmt;
642fce57 2267 stmt_vec_info use_vinfo, stmt_vinfo = vinfo_for_stmt (stmt);
d755c7ef 2268 slp_tree child;
f2c74cc4 2269 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_vinfo);
642fce57
RB
2270 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
2271 int j;
2272
2273 /* Propagate hybrid down the SLP tree. */
2274 if (stype == hybrid)
2275 ;
2276 else if (HYBRID_SLP_STMT (stmt_vinfo))
2277 stype = hybrid;
2278 else
2279 {
2280 /* Check if a pure SLP stmt has uses in non-SLP stmts. */
2281 gcc_checking_assert (PURE_SLP_STMT (stmt_vinfo));
2935d994
RB
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))
29764870 2286 stmt = STMT_VINFO_RELATED_STMT (stmt_vinfo);
e7baeb39
RB
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)
29764870
RB
2294 {
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))
502f0263
RB
2306 {
2307 if (dump_enabled_p ())
2308 {
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);
2312 }
2313 stype = hybrid;
2314 }
29764870 2315 }
642fce57 2316 }
ebfd146a 2317
502f0263
RB
2318 if (stype == hybrid
2319 && !HYBRID_SLP_STMT (stmt_vinfo))
b1af7da6
RB
2320 {
2321 if (dump_enabled_p ())
2322 {
2323 dump_printf_loc (MSG_NOTE, vect_location, "marking hybrid: ");
2324 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
2325 }
2326 STMT_SLP_TYPE (stmt_vinfo) = hybrid;
2327 }
ebfd146a 2328
642fce57 2329 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), j, child)
603cca93 2330 if (SLP_TREE_DEF_TYPE (child) != vect_external_def)
90dd6e3d 2331 vect_detect_hybrid_slp_stmts (child, i, stype);
642fce57 2332}
f2c74cc4 2333
642fce57 2334/* Helpers for vect_detect_hybrid_slp walking pattern stmt uses. */
ebfd146a 2335
642fce57
RB
2336static tree
2337vect_detect_hybrid_slp_1 (tree *tp, int *, void *data)
2338{
2339 walk_stmt_info *wi = (walk_stmt_info *)data;
2340 struct loop *loopp = (struct loop *)wi->info;
2341
2342 if (wi->is_lhs)
2343 return NULL_TREE;
2344
2345 if (TREE_CODE (*tp) == SSA_NAME
2346 && !SSA_NAME_IS_DEFAULT_DEF (*tp))
2347 {
355fe088 2348 gimple *def_stmt = SSA_NAME_DEF_STMT (*tp);
642fce57
RB
2349 if (flow_bb_inside_loop_p (loopp, gimple_bb (def_stmt))
2350 && PURE_SLP_STMT (vinfo_for_stmt (def_stmt)))
b1af7da6
RB
2351 {
2352 if (dump_enabled_p ())
2353 {
2354 dump_printf_loc (MSG_NOTE, vect_location, "marking hybrid: ");
2355 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, def_stmt, 0);
2356 }
2357 STMT_SLP_TYPE (vinfo_for_stmt (def_stmt)) = hybrid;
2358 }
642fce57
RB
2359 }
2360
2361 return NULL_TREE;
ebfd146a
IR
2362}
2363
642fce57
RB
2364static tree
2365vect_detect_hybrid_slp_2 (gimple_stmt_iterator *gsi, bool *handled,
2366 walk_stmt_info *)
2367{
9186a9d3 2368 stmt_vec_info use_vinfo = vinfo_for_stmt (gsi_stmt (*gsi));
642fce57
RB
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. */
9186a9d3
RB
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))
2376 ;
2377 else
642fce57
RB
2378 *handled = true;
2379 return NULL_TREE;
2380}
ebfd146a
IR
2381
2382/* Find stmts that must be both vectorized and SLPed. */
2383
2384void
2385vect_detect_hybrid_slp (loop_vec_info loop_vinfo)
2386{
2387 unsigned int i;
9771b263 2388 vec<slp_instance> slp_instances = LOOP_VINFO_SLP_INSTANCES (loop_vinfo);
ebfd146a
IR
2389 slp_instance instance;
2390
73fbfcad 2391 if (dump_enabled_p ())
e645e942
TJ
2392 dump_printf_loc (MSG_NOTE, vect_location, "=== vect_detect_hybrid_slp ==="
2393 "\n");
ebfd146a 2394
642fce57
RB
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)
2398 {
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))
2402 {
355fe088 2403 gimple *stmt = gsi_stmt (gsi);
642fce57
RB
2404 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
2405 if (STMT_VINFO_IN_PATTERN_P (stmt_info))
2406 {
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);
2417 }
2418 }
2419 }
2420
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. */
9771b263 2424 FOR_EACH_VEC_ELT (slp_instances, i, instance)
642fce57
RB
2425 {
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);
2429 }
ebfd146a
IR
2430}
2431
a70d6342 2432
2c515559
RS
2433/* Initialize a bb_vec_info struct for the statements between
2434 REGION_BEGIN_IN (inclusive) and REGION_END_IN (exclusive). */
2435
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)
a70d6342 2442{
a70d6342
IR
2443 gimple_stmt_iterator gsi;
2444
61d371eb
RB
2445 for (gsi = region_begin; gsi_stmt (gsi) != gsi_stmt (region_end);
2446 gsi_next (&gsi))
a70d6342 2447 {
355fe088 2448 gimple *stmt = gsi_stmt (gsi);
a70d6342 2449 gimple_set_uid (stmt, 0);
2c515559 2450 set_vinfo_for_stmt (stmt, new_stmt_vec_info (stmt, this));
a70d6342
IR
2451 }
2452
2c515559 2453 bb->aux = this;
a70d6342
IR
2454}
2455
2456
2457/* Free BB_VINFO struct, as well as all the stmt_vec_info structs of all the
2458 stmts in the basic block. */
2459
2c515559 2460_bb_vec_info::~_bb_vec_info ()
a70d6342 2461{
2c515559
RS
2462 for (gimple_stmt_iterator si = region_begin;
2463 gsi_stmt (si) != gsi_stmt (region_end); gsi_next (&si))
a70d6342 2464 {
355fe088 2465 gimple *stmt = gsi_stmt (si);
a70d6342
IR
2466 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
2467
2468 if (stmt_info)
2469 /* Free stmt_vec_info. */
2470 free_stmt_vec_info (stmt);
61d371eb
RB
2471
2472 /* Reset region marker. */
2473 gimple_set_uid (stmt, -1);
a70d6342
IR
2474 }
2475
2c515559 2476 bb->aux = NULL;
a70d6342
IR
2477}
2478
15944069
RS
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. */
a70d6342
IR
2482
2483static bool
15944069
RS
2484vect_slp_analyze_node_operations_1 (vec_info *vinfo, slp_tree node,
2485 slp_instance node_instance,
2486 stmt_vector_for_cost *cost_vec)
a70d6342 2487{
15944069 2488 gimple *stmt = SLP_TREE_SCALAR_STMTS (node)[0];
bd2f172f
RB
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);
2492
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))
a70d6342 2499 {
1f3cb663
RS
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)
bd2f172f 2506 {
1f3cb663
RS
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;
bd2f172f
RB
2512 }
2513
2514 gimple *sstmt;
15944069 2515 unsigned int i;
bd2f172f
RB
2516 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, sstmt)
2517 STMT_VINFO_VECTYPE (vinfo_for_stmt (sstmt)) = vectype;
a70d6342
IR
2518 }
2519
8b7e9dba
RS
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
2531 {
d9f21f6a 2532 poly_uint64 vf;
8b7e9dba
RS
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)
d9f21f6a 2540 = vect_get_num_vectors (vf * group_size, vectype);
8b7e9dba
RS
2541 }
2542
15944069
RS
2543 bool dummy;
2544 return vect_analyze_stmt (stmt, &dummy, node, node_instance, cost_vec);
2545}
2546
2547/* Analyze statements contained in SLP tree NODE after recursively analyzing
2548 the subtree. NODE_INSTANCE contains NODE and VINFO contains INSTANCE.
2549
2550 Return true if the operations are supported. */
2551
2552static bool
2553vect_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)
2558{
2559 int i, j;
2560 slp_tree child;
2561
2562 if (SLP_TREE_DEF_TYPE (node) != vect_internal_def)
2563 return true;
2564
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))))
2570 {
2571 SLP_TREE_NUMBER_OF_VEC_STMTS (node)
2572 = SLP_TREE_NUMBER_OF_VEC_STMTS (*leader);
2573 return true;
2574 }
2575
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);
2580
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;
2585
bd2f172f
RB
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);
15944069
RS
2591 bool res = vect_slp_analyze_node_operations_1 (vinfo, node, node_instance,
2592 cost_vec);
bd2f172f
RB
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;
2600
2601 return true;
a70d6342
IR
2602}
2603
2604
8b7e9dba 2605/* Analyze statements in SLP instances of VINFO. Return true if the
a70d6342
IR
2606 operations are supported. */
2607
a12e42fc 2608bool
8b7e9dba 2609vect_slp_analyze_operations (vec_info *vinfo)
a70d6342 2610{
a70d6342
IR
2611 slp_instance instance;
2612 int i;
2613
a12e42fc
RB
2614 if (dump_enabled_p ())
2615 dump_printf_loc (MSG_NOTE, vect_location,
2616 "=== vect_slp_analyze_operations ===\n");
2617
68435eb2
RB
2618 scalar_stmts_to_slp_tree_map_t *visited
2619 = new scalar_stmts_to_slp_tree_map_t ();
8b7e9dba 2620 for (i = 0; vinfo->slp_instances.iterate (i, &instance); )
a70d6342 2621 {
68435eb2
RB
2622 scalar_stmts_to_slp_tree_map_t lvisited;
2623 stmt_vector_for_cost cost_vec;
2624 cost_vec.create (2);
8b7e9dba
RS
2625 if (!vect_slp_analyze_node_operations (vinfo,
2626 SLP_INSTANCE_TREE (instance),
68435eb2
RB
2627 instance, visited, &lvisited,
2628 &cost_vec))
a70d6342 2629 {
a12e42fc
RB
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);
8b7e9dba 2636 vinfo->slp_instances.ordered_remove (i);
68435eb2 2637 cost_vec.release ();
a70d6342
IR
2638 }
2639 else
68435eb2
RB
2640 {
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++;
78604de0 2645
68435eb2
RB
2646 add_stmt_costs (vinfo->target_cost_data, &cost_vec);
2647 cost_vec.release ();
2648 }
2649 }
78604de0
RB
2650 delete visited;
2651
8b7e9dba 2652 return !vinfo->slp_instances.is_empty ();
a70d6342
IR
2653}
2654
6eddf228
RB
2655
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. */
2659
a296d6d3 2660static void
292cba13 2661vect_bb_slp_scalar_cost (basic_block bb,
a296d6d3
RB
2662 slp_tree node, vec<bool, va_heap> *life,
2663 stmt_vector_for_cost *cost_vec)
6eddf228 2664{
6eddf228 2665 unsigned i;
355fe088 2666 gimple *stmt;
6eddf228
RB
2667 slp_tree child;
2668
2669 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, stmt)
2670 {
6eddf228
RB
2671 ssa_op_iter op_iter;
2672 def_operand_p def_p;
2673 stmt_vec_info stmt_info;
2674
ff4c81cc 2675 if ((*life)[i])
6eddf228
RB
2676 continue;
2677
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)
2684 {
2685 imm_use_iterator use_iter;
355fe088 2686 gimple *use_stmt;
6eddf228 2687 FOR_EACH_IMM_USE_STMT (use_stmt, use_iter, DEF_FROM_PTR (def_p))
f30a0ba5 2688 if (!is_gimple_debug (use_stmt)
61d371eb
RB
2689 && (! vect_stmt_in_region_p (vinfo_for_stmt (stmt)->vinfo,
2690 use_stmt)
603cca93 2691 || ! PURE_SLP_STMT (vinfo_for_stmt (use_stmt))))
6eddf228 2692 {
ff4c81cc 2693 (*life)[i] = true;
6eddf228
RB
2694 BREAK_FROM_IMM_USE_STMT (use_iter);
2695 }
2696 }
ff4c81cc 2697 if ((*life)[i])
6eddf228
RB
2698 continue;
2699
b555a2e4
RB
2700 /* Count scalar stmts only once. */
2701 if (gimple_visited_p (stmt))
2702 continue;
2703 gimple_set_visited (stmt, true);
2704
6eddf228 2705 stmt_info = vinfo_for_stmt (stmt);
a296d6d3 2706 vect_cost_for_stmt kind;
6eddf228
RB
2707 if (STMT_VINFO_DATA_REF (stmt_info))
2708 {
2709 if (DR_IS_READ (STMT_VINFO_DATA_REF (stmt_info)))
a296d6d3 2710 kind = scalar_load;
6eddf228 2711 else
a296d6d3 2712 kind = scalar_store;
6eddf228
RB
2713 }
2714 else
a296d6d3
RB
2715 kind = scalar_stmt;
2716 record_stmt_cost (cost_vec, 1, kind, stmt_info, 0, vect_body);
6eddf228
RB
2717 }
2718
faa5399b 2719 auto_vec<bool, 20> subtree_life;
6eddf228 2720 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
faa5399b
RB
2721 {
2722 if (SLP_TREE_DEF_TYPE (child) == vect_internal_def)
2723 {
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);
a296d6d3 2727 vect_bb_slp_scalar_cost (bb, child, &subtree_life, cost_vec);
faa5399b
RB
2728 subtree_life.truncate (0);
2729 }
2730 }
6eddf228
RB
2731}
2732
69f11a13
IR
2733/* Check if vectorization of the basic block is profitable. */
2734
2735static bool
2736vect_bb_vectorization_profitable_p (bb_vec_info bb_vinfo)
2737{
9771b263 2738 vec<slp_instance> slp_instances = BB_VINFO_SLP_INSTANCES (bb_vinfo);
69f11a13 2739 slp_instance instance;
1a4b99c1 2740 int i;
c3e7ee41 2741 unsigned int vec_inside_cost = 0, vec_outside_cost = 0, scalar_cost = 0;
92345349 2742 unsigned int vec_prologue_cost = 0, vec_epilogue_cost = 0;
69f11a13
IR
2743
2744 /* Calculate scalar cost. */
a296d6d3
RB
2745 stmt_vector_for_cost scalar_costs;
2746 scalar_costs.create (0);
6eddf228 2747 FOR_EACH_VEC_ELT (slp_instances, i, instance)
69f11a13 2748 {
00f96dc9 2749 auto_vec<bool, 20> life;
ff4c81cc 2750 life.safe_grow_cleared (SLP_INSTANCE_GROUP_SIZE (instance));
a296d6d3
RB
2751 vect_bb_slp_scalar_cost (BB_VINFO_BB (bb_vinfo),
2752 SLP_INSTANCE_TREE (instance),
2753 &life, &scalar_costs);
2754 }
2755 void *target_cost_data = init_cost (NULL);
68435eb2 2756 add_stmt_costs (target_cost_data, &scalar_costs);
a296d6d3
RB
2757 scalar_costs.release ();
2758 unsigned dummy;
2759 finish_cost (target_cost_data, &dummy, &scalar_cost, &dummy);
2760 destroy_cost_data (target_cost_data);
69f11a13 2761
b555a2e4
RB
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);
2766
c3e7ee41 2767 /* Complete the target-specific cost calculation. */
92345349
BS
2768 finish_cost (BB_VINFO_TARGET_COST_DATA (bb_vinfo), &vec_prologue_cost,
2769 &vec_inside_cost, &vec_epilogue_cost);
2770
2771 vec_outside_cost = vec_prologue_cost + vec_epilogue_cost;
c3e7ee41 2772
73fbfcad 2773 if (dump_enabled_p ())
69f11a13 2774 {
78c60e3d
SS
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);
e645e942 2780 dump_printf (MSG_NOTE, " Scalar cost of basic block: %d\n", scalar_cost);
69f11a13
IR
2781 }
2782
a6524bba
RB
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)
69f11a13
IR
2789 return false;
2790
2791 return true;
2792}
2793
a5b50aa1
RB
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. */
a70d6342 2797
8e19f5a1 2798static bb_vec_info
61d371eb
RB
2799vect_slp_analyze_bb_1 (gimple_stmt_iterator region_begin,
2800 gimple_stmt_iterator region_end,
a5b50aa1
RB
2801 vec<data_reference_p> datarefs, int n_stmts,
2802 bool &fatal)
a70d6342
IR
2803{
2804 bb_vec_info bb_vinfo;
a70d6342 2805 slp_instance instance;
8e19f5a1 2806 int i;
d9f21f6a 2807 poly_uint64 min_vf = 2;
e4a707c4 2808
a5b50aa1
RB
2809 /* The first group of checks is independent of the vector size. */
2810 fatal = true;
2811
61d371eb
RB
2812 if (n_stmts > PARAM_VALUE (PARAM_SLP_MAX_INSNS_IN_BB))
2813 {
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;
2820 }
2821
2c515559 2822 bb_vinfo = new _bb_vec_info (region_begin, region_end);
a70d6342
IR
2823 if (!bb_vinfo)
2824 return NULL;
2825
61d371eb 2826 BB_VINFO_DATAREFS (bb_vinfo) = datarefs;
428db0ba
RB
2827
2828 /* Analyze the data references. */
2829
2830 if (!vect_analyze_data_refs (bb_vinfo, &min_vf))
a70d6342 2831 {
73fbfcad 2832 if (dump_enabled_p ())
78c60e3d
SS
2833 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2834 "not vectorized: unhandled data-ref in basic "
2835 "block.\n");
b8698a0f 2836
2c515559 2837 delete bb_vinfo;
a70d6342
IR
2838 return NULL;
2839 }
2840
fcac74a1 2841 if (BB_VINFO_DATAREFS (bb_vinfo).length () < 2)
a70d6342 2842 {
73fbfcad 2843 if (dump_enabled_p ())
78c60e3d
SS
2844 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2845 "not vectorized: not enough data-refs in "
2846 "basic block.\n");
a70d6342 2847
2c515559 2848 delete bb_vinfo;
a70d6342
IR
2849 return NULL;
2850 }
2851
310213d4 2852 if (!vect_analyze_data_ref_accesses (bb_vinfo))
5abe1e05
RB
2853 {
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");
2858
2c515559 2859 delete bb_vinfo;
5abe1e05
RB
2860 return NULL;
2861 }
2862
a5b50aa1
RB
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 ())
a70d6342 2867 {
73fbfcad 2868 if (dump_enabled_p ())
a5b50aa1
RB
2869 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2870 "not vectorized: no grouped stores in "
2871 "basic block.\n");
b8698a0f 2872
2c515559 2873 delete bb_vinfo;
a70d6342
IR
2874 return NULL;
2875 }
b8698a0f 2876
a5b50aa1
RB
2877 /* While the rest of the analysis below depends on it in some way. */
2878 fatal = false;
2879
2880 vect_pattern_recog (bb_vinfo);
2881
a70d6342
IR
2882 /* Check the SLP opportunities in the basic block, analyze and build SLP
2883 trees. */
310213d4 2884 if (!vect_analyze_slp (bb_vinfo, n_stmts))
a70d6342 2885 {
73fbfcad 2886 if (dump_enabled_p ())
effb52da
RB
2887 {
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");
2893 }
a70d6342 2894
2c515559 2895 delete bb_vinfo;
a70d6342
IR
2896 return NULL;
2897 }
b8698a0f 2898
62c8a2cf
RS
2899 vect_record_base_alignments (bb_vinfo);
2900
c2a12ca0
RB
2901 /* Analyze and verify the alignment of data references and the
2902 dependence in the SLP instances. */
a5b50aa1
RB
2903 for (i = 0; BB_VINFO_SLP_INSTANCES (bb_vinfo).iterate (i, &instance); )
2904 {
c2a12ca0
RB
2905 if (! vect_slp_analyze_and_verify_instance_alignment (instance)
2906 || ! vect_slp_analyze_instance_dependence (instance))
a5b50aa1
RB
2907 {
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;
2916 }
c2a12ca0
RB
2917
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));
2922
a5b50aa1
RB
2923 i++;
2924 }
a5b50aa1
RB
2925 if (! BB_VINFO_SLP_INSTANCES (bb_vinfo).length ())
2926 {
2c515559 2927 delete bb_vinfo;
a5b50aa1
RB
2928 return NULL;
2929 }
2930
8b7e9dba 2931 if (!vect_slp_analyze_operations (bb_vinfo))
a70d6342 2932 {
73fbfcad 2933 if (dump_enabled_p ())
e645e942 2934 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
78c60e3d 2935 "not vectorized: bad operation in basic block.\n");
a70d6342 2936
2c515559 2937 delete bb_vinfo;
a70d6342
IR
2938 return NULL;
2939 }
2940
69f11a13 2941 /* Cost model: check if the vectorization is worthwhile. */
8b5e1202 2942 if (!unlimited_cost_model (NULL)
69f11a13
IR
2943 && !vect_bb_vectorization_profitable_p (bb_vinfo))
2944 {
73fbfcad 2945 if (dump_enabled_p ())
78c60e3d
SS
2946 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2947 "not vectorized: vectorization is not "
2948 "profitable.\n");
69f11a13 2949
2c515559 2950 delete bb_vinfo;
69f11a13
IR
2951 return NULL;
2952 }
2953
73fbfcad 2954 if (dump_enabled_p ())
78c60e3d
SS
2955 dump_printf_loc (MSG_NOTE, vect_location,
2956 "Basic block will be vectorized using SLP\n");
a70d6342
IR
2957
2958 return bb_vinfo;
2959}
2960
2961
428db0ba
RB
2962/* Main entry for the BB vectorizer. Analyze and transform BB, returns
2963 true if anything in the basic-block was vectorized. */
2964
2965bool
2966vect_slp_bb (basic_block bb)
8e19f5a1
IR
2967{
2968 bb_vec_info bb_vinfo;
8e19f5a1 2969 gimple_stmt_iterator gsi;
61d371eb 2970 bool any_vectorized = false;
86e36728 2971 auto_vector_sizes vector_sizes;
8e19f5a1 2972
73fbfcad 2973 if (dump_enabled_p ())
78c60e3d 2974 dump_printf_loc (MSG_NOTE, vect_location, "===vect_slp_analyze_bb===\n");
8e19f5a1 2975
8e19f5a1
IR
2976 /* Autodetect first vector size we try. */
2977 current_vector_size = 0;
86e36728
RS
2978 targetm.vectorize.autovectorize_vector_sizes (&vector_sizes);
2979 unsigned int next_size = 0;
8e19f5a1 2980
61d371eb
RB
2981 gsi = gsi_start_bb (bb);
2982
86e36728 2983 poly_uint64 autodetected_vector_size = 0;
8e19f5a1
IR
2984 while (1)
2985 {
61d371eb
RB
2986 if (gsi_end_p (gsi))
2987 break;
2988
2989 gimple_stmt_iterator region_begin = gsi;
2990 vec<data_reference_p> datarefs = vNULL;
2991 int insns = 0;
2992
2993 for (; !gsi_end_p (gsi); gsi_next (&gsi))
428db0ba 2994 {
61d371eb
RB
2995 gimple *stmt = gsi_stmt (gsi);
2996 if (is_gimple_debug (stmt))
2997 continue;
2998 insns++;
2999
3000 if (gimple_location (stmt) != UNKNOWN_LOCATION)
3001 vect_location = gimple_location (stmt);
3002
3003 if (!find_data_references_in_stmt (NULL, stmt, &datarefs))
3004 break;
3005 }
3006
3007 /* Skip leading unhandled stmts. */
3008 if (gsi_stmt (region_begin) == gsi_stmt (gsi))
3009 {
3010 gsi_next (&gsi);
3011 continue;
3012 }
428db0ba 3013
61d371eb
RB
3014 gimple_stmt_iterator region_end = gsi;
3015
3016 bool vectorized = false;
a5b50aa1 3017 bool fatal = false;
61d371eb 3018 bb_vinfo = vect_slp_analyze_bb_1 (region_begin, region_end,
a5b50aa1 3019 datarefs, insns, fatal);
61d371eb
RB
3020 if (bb_vinfo
3021 && dbg_cnt (vect_slp))
3022 {
428db0ba 3023 if (dump_enabled_p ())
61d371eb 3024 dump_printf_loc (MSG_NOTE, vect_location, "SLPing BB part\n");
428db0ba
RB
3025
3026 vect_schedule_slp (bb_vinfo);
3027
3028 if (dump_enabled_p ())
3029 dump_printf_loc (MSG_NOTE, vect_location,
61d371eb 3030 "basic block part vectorized\n");
428db0ba 3031
61d371eb 3032 vectorized = true;
428db0ba 3033 }
2c515559 3034 delete bb_vinfo;
8e19f5a1 3035
61d371eb 3036 any_vectorized |= vectorized;
8e19f5a1 3037
86e36728
RS
3038 if (next_size == 0)
3039 autodetected_vector_size = current_vector_size;
3040
3041 if (next_size < vector_sizes.length ()
3042 && known_eq (vector_sizes[next_size], autodetected_vector_size))
3043 next_size += 1;
3044
61d371eb 3045 if (vectorized
86e36728
RS
3046 || next_size == vector_sizes.length ()
3047 || known_eq (current_vector_size, 0U)
a5b50aa1
RB
3048 /* If vect_slp_analyze_bb_1 signaled that analysis for all
3049 vector sizes will fail do not bother iterating. */
3050 || fatal)
61d371eb
RB
3051 {
3052 if (gsi_end_p (region_end))
3053 break;
8e19f5a1 3054
61d371eb
RB
3055 /* Skip the unhandled stmt. */
3056 gsi_next (&gsi);
3057
3058 /* And reset vector sizes. */
3059 current_vector_size = 0;
86e36728 3060 next_size = 0;
61d371eb
RB
3061 }
3062 else
3063 {
3064 /* Try the next biggest vector size. */
86e36728 3065 current_vector_size = vector_sizes[next_size++];
61d371eb 3066 if (dump_enabled_p ())
86e36728
RS
3067 {
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");
3073 }
61d371eb
RB
3074
3075 /* Start over. */
3076 gsi = region_begin;
3077 }
8e19f5a1 3078 }
61d371eb
RB
3079
3080 return any_vectorized;
8e19f5a1
IR
3081}
3082
3083
e4af0bc4
IE
3084/* Return 1 if vector type of boolean constant which is OPNUM
3085 operand in statement STMT is a boolean vector. */
3086
3087static bool
3088vect_mask_constant_operand_p (gimple *stmt, int opnum)
3089{
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;
3095
3096 /* For comparison and COND_EXPR type is chosen depending
3097 on the other comparison operand. */
3098 if (TREE_CODE_CLASS (code) == tcc_comparison)
3099 {
3100 if (opnum)
3101 op = gimple_assign_rhs1 (stmt);
3102 else
3103 op = gimple_assign_rhs2 (stmt);
3104
3105 if (!vect_is_simple_use (op, stmt_vinfo->vinfo, &def_stmt,
3106 &dt, &vectype))
3107 gcc_unreachable ();
3108
3109 return !vectype || VECTOR_BOOLEAN_TYPE_P (vectype);
3110 }
3111
3112 if (code == COND_EXPR)
3113 {
3114 tree cond = gimple_assign_rhs1 (stmt);
3115
3116 if (TREE_CODE (cond) == SSA_NAME)
7b1b0cc1
RB
3117 op = cond;
3118 else if (opnum)
e4af0bc4
IE
3119 op = TREE_OPERAND (cond, 1);
3120 else
3121 op = TREE_OPERAND (cond, 0);
3122
3123 if (!vect_is_simple_use (op, stmt_vinfo->vinfo, &def_stmt,
3124 &dt, &vectype))
3125 gcc_unreachable ();
3126
3127 return !vectype || VECTOR_BOOLEAN_TYPE_P (vectype);
3128 }
3129
3130 return VECTOR_BOOLEAN_TYPE_P (STMT_VINFO_VECTYPE (stmt_vinfo));
3131}
3132
018b2744
RS
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.
3136
3137 The approach we use is:
3138
3139 (1) Find a vector mode VM with integer elements of mode IM.
3140
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.
3144
3145 (3) Duplicate each ELTS'[I] into a vector of mode VM.
3146
3147 (4) Use a tree of interleaving VEC_PERM_EXPRs to create VMs with the
3148 correct byte contents.
3149
3150 (5) Use VIEW_CONVERT_EXPR to cast the final VMs to the required type.
3151
3152 We try to find the largest IM for which this sequence works, in order
3153 to cut down on the number of interleaves. */
3154
f1739b48 3155void
018b2744
RS
3156duplicate_and_interleave (gimple_seq *seq, tree vector_type, vec<tree> elts,
3157 unsigned int nresults, vec<tree> &results)
3158{
3159 unsigned int nelts = elts.length ();
3160 tree element_type = TREE_TYPE (vector_type);
3161
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 ();
3170
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);
3174
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)
3179 {
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);
3188
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);
3191 }
3192
3193 /* (4) Use a tree of VEC_PERM_EXPRs to create a single VM with the
3194 correct byte contents.
3195
3196 We need to repeat the following operation log2(nvectors) times:
3197
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);
3200
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)
3210 {
3211 noutputs_bound /= 2;
3212 unsigned int limit = MIN (noutputs_bound, nvectors);
3213 for (unsigned int i = 0; i < limit; ++i)
3214 {
3215 if ((i & 1) != 0
3216 && multiple_p (TYPE_VECTOR_SUBPARTS (new_vector_type),
3217 2 * in_repeat))
3218 {
3219 pieces[out_start + i] = pieces[out_start + i - 1];
3220 continue;
3221 }
3222
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;
3231 }
3232 std::swap (in_start, out_start);
3233 }
3234
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]);
3243}
3244
e4af0bc4 3245
b8698a0f
L
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.
d59dc888
IR
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.
b5aeb3bb
IR
3250 REDUC_INDEX is the index of the reduction operand in the statements, unless
3251 it is -1. */
ebfd146a
IR
3252
3253static void
9dc3f7de 3254vect_get_constant_vectors (tree op, slp_tree slp_node,
9771b263 3255 vec<tree> *vec_oprnds,
306b0c92 3256 unsigned int op_num, unsigned int number_of_vectors)
ebfd146a 3257{
355fe088
TS
3258 vec<gimple *> stmts = SLP_TREE_SCALAR_STMTS (slp_node);
3259 gimple *stmt = stmts[0];
ebfd146a 3260 stmt_vec_info stmt_vinfo = vinfo_for_stmt (stmt);
018b2744 3261 unsigned HOST_WIDE_INT nunits;
ebfd146a 3262 tree vec_cst;
d2a12ae7 3263 unsigned j, number_of_places_left_in_vector;
ebfd146a 3264 tree vector_type;
9dc3f7de 3265 tree vop;
9771b263 3266 int group_size = stmts.length ();
ebfd146a 3267 unsigned int vec_num, i;
d2a12ae7 3268 unsigned number_of_copies = 1;
9771b263
DN
3269 vec<tree> voprnds;
3270 voprnds.create (number_of_vectors);
ebfd146a 3271 bool constant_p, is_store;
b5aeb3bb 3272 tree neutral_op = NULL;
bac430c9 3273 enum tree_code code = gimple_expr_code (stmt);
13396b6e 3274 gimple_seq ctor_seq = NULL;
018b2744 3275 auto_vec<tree, 16> permute_results;
b5aeb3bb 3276
42fd8198 3277 /* Check if vector type is a boolean vector. */
2568d8a1 3278 if (VECT_SCALAR_BOOLEAN_TYPE_P (TREE_TYPE (op))
e4af0bc4 3279 && vect_mask_constant_operand_p (stmt, op_num))
42fd8198
IE
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));
afbe6325 3284
ebfd146a
IR
3285 if (STMT_VINFO_DATA_REF (stmt_vinfo))
3286 {
3287 is_store = true;
3288 op = gimple_assign_rhs1 (stmt);
3289 }
3290 else
9dc3f7de
IR
3291 is_store = false;
3292
3293 gcc_assert (op);
ebfd146a 3294
ebfd146a 3295 /* NUMBER_OF_COPIES is the number of times we need to use the same values in
b8698a0f 3296 created vectors. It is greater than 1 if unrolling is performed.
ebfd146a
IR
3297
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
f7e531cf
IR
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
ebfd146a
IR
3302 will be 2).
3303
b8698a0f 3304 If GROUP_SIZE > NUNITS, the scalars will be split into several vectors
ebfd146a
IR
3305 containing the operands.
3306
3307 For example, NUNITS is four as before, and the group size is 8
f7e531cf 3308 (s1, s2, ..., s8). We will create two vectors {s1, s2, s3, s4} and
ebfd146a 3309 {s5, s6, s7, s8}. */
b8698a0f 3310
018b2744
RS
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;
3315
14a61437 3316 number_of_copies = nunits * number_of_vectors / group_size;
ebfd146a
IR
3317
3318 number_of_places_left_in_vector = nunits;
62cf7335 3319 constant_p = true;
5ebaa477 3320 tree_vector_builder elts (vector_type, nunits, 1);
794e3180 3321 elts.quick_grow (nunits);
90dd6e3d 3322 bool place_after_defs = false;
ebfd146a
IR
3323 for (j = 0; j < number_of_copies; j++)
3324 {
9771b263 3325 for (i = group_size - 1; stmts.iterate (i, &stmt); i--)
ebfd146a
IR
3326 {
3327 if (is_store)
3328 op = gimple_assign_rhs1 (stmt);
bac430c9 3329 else
f7e531cf 3330 {
bac430c9 3331 switch (code)
f7e531cf 3332 {
bac430c9 3333 case COND_EXPR:
a989bcc3
IE
3334 {
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)
bac430c9 3339 op = TREE_OPERAND (cond, op_num);
a989bcc3
IE
3340 else
3341 {
3342 if (op_num == 2)
3343 op = gimple_assign_rhs2 (stmt);
3344 else
3345 op = gimple_assign_rhs3 (stmt);
3346 }
3347 }
bac430c9
IR
3348 break;
3349
3350 case CALL_EXPR:
3351 op = gimple_call_arg (stmt, op_num);
3352 break;
3353
b84b294a
JJ
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. */
793d9a16 3364 if (op_num == 1 && TREE_CODE (op) == INTEGER_CST)
b84b294a
JJ
3365 op = fold_convert (TREE_TYPE (vector_type), op);
3366 break;
3367
bac430c9
IR
3368 default:
3369 op = gimple_op (stmt, op_num + 1);
b84b294a 3370 break;
f7e531cf
IR
3371 }
3372 }
b8698a0f 3373
ebfd146a 3374 /* Create 'vect_ = {op0,op1,...,opn}'. */
ebfd146a 3375 number_of_places_left_in_vector--;
90dd6e3d 3376 tree orig_op = op;
13396b6e 3377 if (!types_compatible_p (TREE_TYPE (vector_type), TREE_TYPE (op)))
50eeef09 3378 {
793d9a16 3379 if (CONSTANT_CLASS_P (op))
13396b6e 3380 {
42fd8198
IE
3381 if (VECTOR_BOOLEAN_TYPE_P (vector_type))
3382 {
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))
158beb4a 3389 op = build_all_ones_cst (TREE_TYPE (vector_type));
42fd8198
IE
3390 else
3391 gcc_unreachable ();
3392 }
3393 else
3394 op = fold_unary (VIEW_CONVERT_EXPR,
3395 TREE_TYPE (vector_type), op);
13396b6e
JJ
3396 gcc_assert (op && CONSTANT_CLASS_P (op));
3397 }
3398 else
3399 {
b731b390 3400 tree new_temp = make_ssa_name (TREE_TYPE (vector_type));
355fe088 3401 gimple *init_stmt;
262a363f
JJ
3402 if (VECTOR_BOOLEAN_TYPE_P (vector_type))
3403 {
158beb4a
JJ
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));
7c285ab9 3408 gcc_assert (INTEGRAL_TYPE_P (TREE_TYPE (op)));
158beb4a
JJ
3409 init_stmt = gimple_build_assign (new_temp, COND_EXPR,
3410 op, true_val,
3411 false_val);
262a363f 3412 }
262a363f
JJ
3413 else
3414 {
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);
3420 }
13396b6e
JJ
3421 gimple_seq_add_stmt (&ctor_seq, init_stmt);
3422 op = new_temp;
3423 }
50eeef09 3424 }
d2a12ae7 3425 elts[number_of_places_left_in_vector] = op;
793d9a16
RB
3426 if (!CONSTANT_CLASS_P (op))
3427 constant_p = false;
90dd6e3d
RB
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;
ebfd146a
IR
3434
3435 if (number_of_places_left_in_vector == 0)
3436 {
018b2744
RS
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);
ebfd146a 3441 else
d2a12ae7 3442 {
018b2744
RS
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];
d2a12ae7 3448 }
90dd6e3d
RB
3449 tree init;
3450 gimple_stmt_iterator gsi;
3451 if (place_after_defs)
3452 {
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);
3456 }
3457 else
3458 init = vect_init_vector (stmt, vec_cst, vector_type, NULL);
13396b6e
JJ
3459 if (ctor_seq != NULL)
3460 {
90dd6e3d 3461 gsi = gsi_for_stmt (SSA_NAME_DEF_STMT (init));
018b2744 3462 gsi_insert_seq_before (&gsi, ctor_seq, GSI_SAME_STMT);
13396b6e
JJ
3463 ctor_seq = NULL;
3464 }
90dd6e3d
RB
3465 voprnds.quick_push (init);
3466 place_after_defs = false;
62cf7335
RB
3467 number_of_places_left_in_vector = nunits;
3468 constant_p = true;
5ebaa477
RS
3469 elts.new_vector (vector_type, nunits, 1);
3470 elts.quick_grow (nunits);
ebfd146a
IR
3471 }
3472 }
3473 }
3474
b8698a0f 3475 /* Since the vectors are created in the reverse order, we should invert
ebfd146a 3476 them. */
9771b263 3477 vec_num = voprnds.length ();
d2a12ae7 3478 for (j = vec_num; j != 0; j--)
ebfd146a 3479 {
9771b263
DN
3480 vop = voprnds[j - 1];
3481 vec_oprnds->quick_push (vop);
ebfd146a
IR
3482 }
3483
9771b263 3484 voprnds.release ();
ebfd146a
IR
3485
3486 /* In case that VF is greater than the unrolling factor needed for the SLP
b8698a0f
L
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
ebfd146a 3489 to replicate the vectors. */
9771b263 3490 while (number_of_vectors > vec_oprnds->length ())
ebfd146a 3491 {
b5aeb3bb
IR
3492 tree neutral_vec = NULL;
3493
3494 if (neutral_op)
3495 {
3496 if (!neutral_vec)
b9acc9f1 3497 neutral_vec = build_vector_from_val (vector_type, neutral_op);
b5aeb3bb 3498
9771b263 3499 vec_oprnds->quick_push (neutral_vec);
b5aeb3bb
IR
3500 }
3501 else
3502 {
9771b263
DN
3503 for (i = 0; vec_oprnds->iterate (i, &vop) && i < vec_num; i++)
3504 vec_oprnds->quick_push (vop);
b5aeb3bb 3505 }
ebfd146a
IR
3506 }
3507}
3508
3509
3510/* Get vectorized definitions from SLP_NODE that contains corresponding
3511 vectorized def-stmts. */
3512
3513static void
9771b263 3514vect_get_slp_vect_defs (slp_tree slp_node, vec<tree> *vec_oprnds)
ebfd146a
IR
3515{
3516 tree vec_oprnd;
355fe088 3517 gimple *vec_def_stmt;
ebfd146a
IR
3518 unsigned int i;
3519
9771b263 3520 gcc_assert (SLP_TREE_VEC_STMTS (slp_node).exists ());
ebfd146a 3521
9771b263 3522 FOR_EACH_VEC_ELT (SLP_TREE_VEC_STMTS (slp_node), i, vec_def_stmt)
ebfd146a
IR
3523 {
3524 gcc_assert (vec_def_stmt);
e7baeb39
RB
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);
9771b263 3529 vec_oprnds->quick_push (vec_oprnd);
ebfd146a
IR
3530 }
3531}
3532
3533
b8698a0f
L
3534/* Get vectorized definitions for SLP_NODE.
3535 If the scalar definitions are loop invariants or constants, collect them and
ebfd146a
IR
3536 call vect_get_constant_vectors() to create vector stmts.
3537 Otherwise, the def-stmts must be already vectorized and the vectorized stmts
d092494c
IR
3538 must be stored in the corresponding child of SLP_NODE, and we call
3539 vect_get_slp_vect_defs () to retrieve them. */
b8698a0f 3540
ebfd146a 3541void
9771b263 3542vect_get_slp_defs (vec<tree> ops, slp_tree slp_node,
306b0c92 3543 vec<vec<tree> > *vec_oprnds)
ebfd146a 3544{
355fe088 3545 gimple *first_stmt;
d092494c 3546 int number_of_vects = 0, i;
77eefb71 3547 unsigned int child_index = 0;
b8698a0f 3548 HOST_WIDE_INT lhs_size_unit, rhs_size_unit;
d092494c 3549 slp_tree child = NULL;
37b5ec8f 3550 vec<tree> vec_defs;
e44978dc 3551 tree oprnd;
77eefb71 3552 bool vectorized_defs;
ebfd146a 3553
9771b263
DN
3554 first_stmt = SLP_TREE_SCALAR_STMTS (slp_node)[0];
3555 FOR_EACH_VEC_ELT (ops, i, oprnd)
ebfd146a 3556 {
d092494c
IR
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
77eefb71
RB
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)
ebfd146a 3565 {
01d8bf07 3566 child = SLP_TREE_CHILDREN (slp_node)[child_index];
d092494c 3567
e44978dc 3568 /* We have to check both pattern and original def, if available. */
603cca93 3569 if (SLP_TREE_DEF_TYPE (child) == vect_internal_def)
e44978dc 3570 {
355fe088
TS
3571 gimple *first_def = SLP_TREE_SCALAR_STMTS (child)[0];
3572 gimple *related
90dd6e3d 3573 = STMT_VINFO_RELATED_STMT (vinfo_for_stmt (first_def));
e7baeb39 3574 tree first_def_op;
90dd6e3d 3575
e7baeb39
RB
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)
90dd6e3d
RB
3581 || (related
3582 && operand_equal_p (oprnd, gimple_get_lhs (related), 0)))
3583 {
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;
77eefb71 3589 child_index++;
90dd6e3d 3590 }
e44978dc 3591 }
77eefb71
RB
3592 else
3593 child_index++;
d092494c 3594 }
ebfd146a 3595
77eefb71
RB
3596 if (!vectorized_defs)
3597 {
3598 if (i == 0)
3599 {
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)
3608 {
3609 number_of_vects *= rhs_size_unit;
3610 number_of_vects /= lhs_size_unit;
3611 }
3612 }
d092494c 3613 }
b5aeb3bb 3614
d092494c 3615 /* Allocate memory for vectorized defs. */
37b5ec8f
JJ
3616 vec_defs = vNULL;
3617 vec_defs.create (number_of_vects);
ebfd146a 3618
d092494c
IR
3619 /* For reduction defs we call vect_get_constant_vectors (), since we are
3620 looking for initial loop invariant values. */
306b0c92 3621 if (vectorized_defs)
d092494c 3622 /* The defs are already vectorized. */
37b5ec8f 3623 vect_get_slp_vect_defs (child, &vec_defs);
d092494c 3624 else
e7baeb39 3625 /* Build vectors from scalar defs. */
37b5ec8f 3626 vect_get_constant_vectors (oprnd, slp_node, &vec_defs, i,
306b0c92 3627 number_of_vects);
ebfd146a 3628
37b5ec8f 3629 vec_oprnds->quick_push (vec_defs);
d092494c 3630 }
ebfd146a
IR
3631}
3632
ebfd146a
IR
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
01d8bf07
RB
3635 permute statements for the SLP node NODE of the SLP instance
3636 SLP_NODE_INSTANCE. */
3637
ebfd146a 3638bool
01d8bf07 3639vect_transform_slp_perm_load (slp_tree node, vec<tree> dr_chain,
d9f21f6a
RS
3640 gimple_stmt_iterator *gsi, poly_uint64 vf,
3641 slp_instance slp_node_instance, bool analyze_only,
29afecdf 3642 unsigned *n_perms)
ebfd146a 3643{
355fe088 3644 gimple *stmt = SLP_TREE_SCALAR_STMTS (node)[0];
ebfd146a
IR
3645 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
3646 tree mask_element_type = NULL_TREE, mask_type;
928686b1 3647 int vec_index = 0;
2635892a 3648 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
ebfd146a 3649 int group_size = SLP_INSTANCE_GROUP_SIZE (slp_node_instance);
928686b1 3650 unsigned int mask_element;
ef4bddc2 3651 machine_mode mode;
928686b1 3652 unsigned HOST_WIDE_INT nunits, const_vf;
ebfd146a 3653
91ff1504
RB
3654 if (!STMT_VINFO_GROUPED_ACCESS (stmt_info))
3655 return false;
3656
3657 stmt_info = vinfo_for_stmt (GROUP_FIRST_ELEMENT (stmt_info));
3658
22e4dee7
RH
3659 mode = TYPE_MODE (vectype);
3660
d9f21f6a 3661 /* At the moment, all permutations are represented using per-element
928686b1
RS
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))
d9f21f6a
RS
3666 return false;
3667
2635892a
RH
3668 /* The generic VEC_PERM_EXPR code always uses an integral type of the
3669 same size as the vector element being permuted. */
96f9265a 3670 mask_element_type = lang_hooks.types.type_for_mode
304b9962 3671 (int_mode_for_mode (TYPE_MODE (TREE_TYPE (vectype))).require (), 1);
ebfd146a 3672 mask_type = get_vectype_for_scalar_type (mask_element_type);
e3342de4 3673 vec_perm_builder mask (nunits, nunits, 1);
908a1a16 3674 mask.quick_grow (nunits);
e3342de4 3675 vec_perm_indices indices;
ebfd146a 3676
61fdfd8c
RB
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);
ebfd146a 3683
b8698a0f
L
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:
ebfd146a
IR
3689 for a's: a0a0a0a1 a1a1a2a2 a2a3a3a3
3690 for b's: b0b0b0b1 b1b1b2b2 b2b3b3b3
3691 ...
3692
2635892a 3693 The masks for a's should be: {0,0,0,3} {3,3,6,6} {6,9,9,9}.
b8698a0f 3694 The last mask is illegal since we assume two operands for permute
ff802fa1
IR
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}.
b8698a0f 3697 For the first two permutations we need the first and the second input
ebfd146a 3698 vectors: {a0,b0,c0,a1} and {b1,c1,a2,b2}, and for the last permutation
b8698a0f 3699 we need the second and the third vectors: {b1,c1,a2,b2} and
ebfd146a
IR
3700 {c2,a3,b3,c3}. */
3701
2ce27200 3702 int vect_stmts_counter = 0;
928686b1 3703 unsigned int index = 0;
2ce27200
RB
3704 int first_vec_index = -1;
3705 int second_vec_index = -1;
be377c80 3706 bool noop_p = true;
29afecdf 3707 *n_perms = 0;
ebfd146a 3708
d9f21f6a 3709 for (unsigned int j = 0; j < const_vf; j++)
2ce27200
RB
3710 {
3711 for (int k = 0; k < group_size; k++)
3712 {
928686b1
RS
3713 unsigned int i = (SLP_TREE_LOAD_PERMUTATION (node)[k]
3714 + j * STMT_VINFO_GROUP_SIZE (stmt_info));
2ce27200
RB
3715 vec_index = i / nunits;
3716 mask_element = i % nunits;
3717 if (vec_index == first_vec_index
3718 || first_vec_index == -1)
3719 {
3720 first_vec_index = vec_index;
3721 }
3722 else if (vec_index == second_vec_index
3723 || second_vec_index == -1)
3724 {
3725 second_vec_index = vec_index;
3726 mask_element += nunits;
3727 }
3728 else
3729 {
3730 if (dump_enabled_p ())
3731 {
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);
2ce27200 3737 }
31bee964 3738 gcc_assert (analyze_only);
2ce27200
RB
3739 return false;
3740 }
ebfd146a 3741
928686b1 3742 gcc_assert (mask_element < 2 * nunits);
be377c80
RB
3743 if (mask_element != index)
3744 noop_p = false;
2ce27200
RB
3745 mask[index++] = mask_element;
3746
e3342de4 3747 if (index == nunits && !noop_p)
2ce27200 3748 {
e3342de4
RS
3749 indices.new_vector (mask, 2, nunits);
3750 if (!can_vec_perm_const_p (mode, indices))
2ce27200
RB
3751 {
3752 if (dump_enabled_p ())
22e4dee7 3753 {
2ce27200
RB
3754 dump_printf_loc (MSG_MISSED_OPTIMIZATION,
3755 vect_location,
3756 "unsupported vect permute { ");
3757 for (i = 0; i < nunits; ++i)
6b0630fb
RS
3758 {
3759 dump_dec (MSG_MISSED_OPTIMIZATION, mask[i]);
3760 dump_printf (MSG_MISSED_OPTIMIZATION, " ");
3761 }
2ce27200 3762 dump_printf (MSG_MISSED_OPTIMIZATION, "}\n");
22e4dee7 3763 }
31bee964 3764 gcc_assert (analyze_only);
2ce27200
RB
3765 return false;
3766 }
22e4dee7 3767
e3342de4
RS
3768 ++*n_perms;
3769 }
29afecdf 3770
e3342de4
RS
3771 if (index == nunits)
3772 {
2ce27200
RB
3773 if (!analyze_only)
3774 {
be377c80
RB
3775 tree mask_vec = NULL_TREE;
3776
3777 if (! noop_p)
736d0f28 3778 mask_vec = vec_perm_indices_to_tree (mask_type, indices);
2ce27200
RB
3779
3780 if (second_vec_index == -1)
3781 second_vec_index = first_vec_index;
61fdfd8c
RB
3782
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)
3788 {
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);
3798 }
3799 else
3800 /* If mask was NULL_TREE generate the requested
3801 identity transform. */
3802 perm_stmt = SSA_NAME_DEF_STMT (first_vec);
3803
3804 /* Store the vector statement in NODE. */
3805 SLP_TREE_VEC_STMTS (node)[vect_stmts_counter++] = perm_stmt;
2ce27200 3806 }
ebfd146a 3807
2ce27200
RB
3808 index = 0;
3809 first_vec_index = -1;
3810 second_vec_index = -1;
be377c80 3811 noop_p = true;
2ce27200
RB
3812 }
3813 }
b8698a0f 3814 }
ebfd146a 3815
ebfd146a
IR
3816 return true;
3817}
3818
ebfd146a
IR
3819/* Vectorize SLP instance tree in postorder. */
3820
3821static bool
f7300fff
RB
3822vect_schedule_slp_instance (slp_tree node, slp_instance instance,
3823 scalar_stmts_to_slp_tree_map_t *bst_map)
ebfd146a 3824{
355fe088 3825 gimple *stmt;
0d0293ac 3826 bool grouped_store, is_store;
ebfd146a
IR
3827 gimple_stmt_iterator si;
3828 stmt_vec_info stmt_info;
8b7e9dba 3829 unsigned int group_size;
ebfd146a 3830 tree vectype;
603cca93 3831 int i, j;
d755c7ef 3832 slp_tree child;
ebfd146a 3833
603cca93 3834 if (SLP_TREE_DEF_TYPE (node) != vect_internal_def)
ebfd146a
IR
3835 return false;
3836
f7300fff
RB
3837 /* See if we have already vectorized the same set of stmts and reuse their
3838 vectorized stmts. */
dd172744 3839 if (slp_tree *leader = bst_map->get (SLP_TREE_SCALAR_STMTS (node)))
f7300fff 3840 {
dd172744 3841 SLP_TREE_VEC_STMTS (node).safe_splice (SLP_TREE_VEC_STMTS (*leader));
f7300fff
RB
3842 return false;
3843 }
3844
dd172744 3845 bst_map->put (SLP_TREE_SCALAR_STMTS (node).copy (), node);
9771b263 3846 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
f7300fff 3847 vect_schedule_slp_instance (child, instance, bst_map);
b8698a0f 3848
603cca93
RB
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);
3854
9771b263 3855 stmt = SLP_TREE_SCALAR_STMTS (node)[0];
ebfd146a
IR
3856 stmt_info = vinfo_for_stmt (stmt);
3857
3858 /* VECTYPE is the type of the destination. */
b690cc0f 3859 vectype = STMT_VINFO_VECTYPE (stmt_info);
dad55d70 3860 poly_uint64 nunits = TYPE_VECTOR_SUBPARTS (vectype);
ebfd146a
IR
3861 group_size = SLP_INSTANCE_GROUP_SIZE (instance);
3862
68435eb2 3863 gcc_assert (SLP_TREE_NUMBER_OF_VEC_STMTS (node) != 0);
9771b263 3864 if (!SLP_TREE_VEC_STMTS (node).exists ())
8b7e9dba 3865 SLP_TREE_VEC_STMTS (node).create (SLP_TREE_NUMBER_OF_VEC_STMTS (node));
ebfd146a 3866
73fbfcad 3867 if (dump_enabled_p ())
ebfd146a 3868 {
78c60e3d
SS
3869 dump_printf_loc (MSG_NOTE,vect_location,
3870 "------>vectorizing SLP node starting from: ");
3871 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
b8698a0f 3872 }
ebfd146a 3873
2e8ab70c
RB
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));
e4a707c4 3877
b010117a
IR
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. */
0d0293ac 3881 if (GROUP_FIRST_ELEMENT (stmt_info) && !STMT_VINFO_GROUPED_ACCESS (stmt_info)
b010117a
IR
3882 && GROUP_FIRST_ELEMENT (stmt_info) == stmt)
3883 {
3884 STMT_VINFO_DEF_TYPE (stmt_info) = vect_reduction_def;
3885 STMT_VINFO_TYPE (stmt_info) = reduc_vec_info_type;
3886 }
3887
6876e5bc
RB
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))
3891 {
3892 enum tree_code code0 = gimple_assign_rhs_code (stmt);
567a3691 3893 enum tree_code ocode = ERROR_MARK;
355fe088 3894 gimple *ostmt;
e3342de4 3895 vec_perm_builder mask (group_size, group_size, 1);
6876e5bc
RB
3896 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, ostmt)
3897 if (gimple_assign_rhs_code (ostmt) != code0)
3898 {
908a1a16 3899 mask.quick_push (1);
6876e5bc
RB
3900 ocode = gimple_assign_rhs_code (ostmt);
3901 }
3902 else
908a1a16 3903 mask.quick_push (0);
567a3691 3904 if (ocode != ERROR_MARK)
6876e5bc 3905 {
355fe088
TS
3906 vec<gimple *> v0;
3907 vec<gimple *> v1;
6876e5bc
RB
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
b397965c 3919 (GET_MODE_BITSIZE (SCALAR_TYPE_MODE (TREE_TYPE (vectype))), 1);
6876e5bc
RB
3920 tree mvectype = get_same_sized_vectype (meltype, vectype);
3921 unsigned k = 0, l;
3922 for (j = 0; j < v0.length (); ++j)
3923 {
dad55d70
RS
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)
6876e5bc 3929 {
1ece8d4c 3930 if (k >= group_size)
6876e5bc 3931 k = 0;
dad55d70
RS
3932 tree t = build_int_cst (meltype,
3933 mask[k++] * const_nunits + l);
794e3180 3934 melts.quick_push (t);
6876e5bc 3935 }
5ebaa477 3936 tmask = melts.build ();
6876e5bc
RB
3937
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. */
355fe088 3945 gimple *vstmt;
6876e5bc
RB
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);
3952 }
3953 v0.release ();
3954 v1.release ();
3955 return false;
3956 }
3957 }
0d0293ac 3958 is_store = vect_transform_stmt (stmt, &si, &grouped_store, node, instance);
603cca93
RB
3959
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;
3965
b5aeb3bb 3966 return is_store;
ebfd146a
IR
3967}
3968
dd34c087
JJ
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. */
3973
3974static void
3975vect_remove_slp_scalar_calls (slp_tree node)
3976{
355fe088 3977 gimple *stmt, *new_stmt;
dd34c087
JJ
3978 gimple_stmt_iterator gsi;
3979 int i;
d755c7ef 3980 slp_tree child;
dd34c087
JJ
3981 tree lhs;
3982 stmt_vec_info stmt_info;
3983
603cca93 3984 if (SLP_TREE_DEF_TYPE (node) != vect_internal_def)
dd34c087
JJ
3985 return;
3986
9771b263 3987 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
d755c7ef 3988 vect_remove_slp_scalar_calls (child);
dd34c087 3989
9771b263 3990 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, stmt)
dd34c087
JJ
3991 {
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;
4007 }
4008}
ebfd146a 4009
ff802fa1
IR
4010/* Generate vector code for all SLP instances in the loop/basic block. */
4011
ebfd146a 4012bool
310213d4 4013vect_schedule_slp (vec_info *vinfo)
ebfd146a 4014{
9771b263 4015 vec<slp_instance> slp_instances;
ebfd146a 4016 slp_instance instance;
8b7e9dba 4017 unsigned int i;
ebfd146a
IR
4018 bool is_store = false;
4019
78604de0
RB
4020
4021 scalar_stmts_to_slp_tree_map_t *bst_map
4022 = new scalar_stmts_to_slp_tree_map_t ();
310213d4 4023 slp_instances = vinfo->slp_instances;
9771b263 4024 FOR_EACH_VEC_ELT (slp_instances, i, instance)
ebfd146a
IR
4025 {
4026 /* Schedule the tree of INSTANCE. */
4027 is_store = vect_schedule_slp_instance (SLP_INSTANCE_TREE (instance),
f7300fff 4028 instance, bst_map);
73fbfcad 4029 if (dump_enabled_p ())
78c60e3d 4030 dump_printf_loc (MSG_NOTE, vect_location,
e645e942 4031 "vectorizing stmts using SLP.\n");
ebfd146a 4032 }
78604de0 4033 delete bst_map;
ebfd146a 4034
9771b263 4035 FOR_EACH_VEC_ELT (slp_instances, i, instance)
b5aeb3bb
IR
4036 {
4037 slp_tree root = SLP_INSTANCE_TREE (instance);
355fe088 4038 gimple *store;
b5aeb3bb
IR
4039 unsigned int j;
4040 gimple_stmt_iterator gsi;
4041
c40eced0
RB
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. */
310213d4 4049 if (is_a <loop_vec_info> (vinfo))
c40eced0 4050 vect_remove_slp_scalar_calls (root);
dd34c087 4051
9771b263 4052 for (j = 0; SLP_TREE_SCALAR_STMTS (root).iterate (j, &store)
b5aeb3bb
IR
4053 && j < SLP_INSTANCE_GROUP_SIZE (instance); j++)
4054 {
4055 if (!STMT_VINFO_DATA_REF (vinfo_for_stmt (store)))
4056 break;
4057
a024e70e
IR
4058 if (is_pattern_stmt_p (vinfo_for_stmt (store)))
4059 store = STMT_VINFO_RELATED_STMT (vinfo_for_stmt (store));
b5aeb3bb
IR
4060 /* Free the attached stmt_vec_info and remove the stmt. */
4061 gsi = gsi_for_stmt (store);
3d3f2249 4062 unlink_stmt_vdef (store);
b5aeb3bb 4063 gsi_remove (&gsi, true);
3d3f2249 4064 release_defs (store);
b5aeb3bb
IR
4065 free_stmt_vec_info (store);
4066 }
4067 }
4068
ebfd146a
IR
4069 return is_store;
4070}