]> git.ipfire.org Git - thirdparty/gcc.git/blame - gcc/tree-vect-slp.c
builtins.c (fold_builtin_FUNCTION): Use lang_hooks.decl_printable_name.
[thirdparty/gcc.git] / gcc / tree-vect-slp.c
CommitLineData
ebfd146a 1/* SLP - Basic Block Vectorization
cbe34bb5 2 Copyright (C) 2007-2017 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"
a70d6342
IR
44
45
ebfd146a
IR
46/* Recursively free the memory allocated for the SLP tree rooted at NODE. */
47
48static void
49vect_free_slp_tree (slp_tree node)
50{
d092494c 51 int i;
d755c7ef 52 slp_tree child;
d092494c 53
9771b263 54 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
d755c7ef 55 vect_free_slp_tree (child);
b8698a0f 56
78810bd3
RB
57 gimple *stmt;
58 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, stmt)
59 /* After transform some stmts are removed and thus their vinfo is gone. */
60 if (vinfo_for_stmt (stmt))
61 {
62 gcc_assert (STMT_VINFO_NUM_SLP_USES (vinfo_for_stmt (stmt)) > 0);
63 STMT_VINFO_NUM_SLP_USES (vinfo_for_stmt (stmt))--;
64 }
65
9771b263
DN
66 SLP_TREE_CHILDREN (node).release ();
67 SLP_TREE_SCALAR_STMTS (node).release ();
68 SLP_TREE_VEC_STMTS (node).release ();
01d8bf07 69 SLP_TREE_LOAD_PERMUTATION (node).release ();
ebfd146a
IR
70
71 free (node);
72}
73
74
75/* Free the memory allocated for the SLP instance. */
76
77void
78vect_free_slp_instance (slp_instance instance)
79{
80 vect_free_slp_tree (SLP_INSTANCE_TREE (instance));
9771b263 81 SLP_INSTANCE_LOADS (instance).release ();
c7e62a26 82 free (instance);
ebfd146a
IR
83}
84
85
d092494c
IR
86/* Create an SLP node for SCALAR_STMTS. */
87
88static slp_tree
355fe088 89vect_create_new_slp_node (vec<gimple *> scalar_stmts)
d092494c 90{
d3cfd39e 91 slp_tree node;
355fe088 92 gimple *stmt = scalar_stmts[0];
d092494c
IR
93 unsigned int nops;
94
95 if (is_gimple_call (stmt))
96 nops = gimple_call_num_args (stmt);
97 else if (is_gimple_assign (stmt))
f7e531cf
IR
98 {
99 nops = gimple_num_ops (stmt) - 1;
100 if (gimple_assign_rhs_code (stmt) == COND_EXPR)
101 nops++;
102 }
e7baeb39
RB
103 else if (gimple_code (stmt) == GIMPLE_PHI)
104 nops = 0;
d092494c
IR
105 else
106 return NULL;
107
d3cfd39e 108 node = XNEW (struct _slp_tree);
d092494c 109 SLP_TREE_SCALAR_STMTS (node) = scalar_stmts;
9771b263
DN
110 SLP_TREE_VEC_STMTS (node).create (0);
111 SLP_TREE_CHILDREN (node).create (nops);
01d8bf07 112 SLP_TREE_LOAD_PERMUTATION (node) = vNULL;
6876e5bc 113 SLP_TREE_TWO_OPERATORS (node) = false;
603cca93 114 SLP_TREE_DEF_TYPE (node) = vect_internal_def;
d092494c 115
78810bd3
RB
116 unsigned i;
117 FOR_EACH_VEC_ELT (scalar_stmts, i, stmt)
118 STMT_VINFO_NUM_SLP_USES (vinfo_for_stmt (stmt))++;
119
d092494c
IR
120 return node;
121}
122
123
ddf56386
RB
124/* This structure is used in creation of an SLP tree. Each instance
125 corresponds to the same operand in a group of scalar stmts in an SLP
126 node. */
127typedef struct _slp_oprnd_info
128{
129 /* Def-stmts for the operands. */
130 vec<gimple *> def_stmts;
131 /* Information about the first statement, its vector def-type, type, the
132 operand itself in case it's constant, and an indication if it's a pattern
133 stmt. */
ddf56386 134 tree first_op_type;
34e82342 135 enum vect_def_type first_dt;
ddf56386
RB
136 bool first_pattern;
137 bool second_pattern;
138} *slp_oprnd_info;
139
140
d092494c
IR
141/* Allocate operands info for NOPS operands, and GROUP_SIZE def-stmts for each
142 operand. */
9771b263 143static vec<slp_oprnd_info>
d092494c
IR
144vect_create_oprnd_info (int nops, int group_size)
145{
146 int i;
147 slp_oprnd_info oprnd_info;
9771b263 148 vec<slp_oprnd_info> oprnds_info;
d092494c 149
9771b263 150 oprnds_info.create (nops);
d092494c
IR
151 for (i = 0; i < nops; i++)
152 {
153 oprnd_info = XNEW (struct _slp_oprnd_info);
9771b263 154 oprnd_info->def_stmts.create (group_size);
d092494c 155 oprnd_info->first_dt = vect_uninitialized_def;
793d9a16 156 oprnd_info->first_op_type = NULL_TREE;
d092494c 157 oprnd_info->first_pattern = false;
effb52da 158 oprnd_info->second_pattern = false;
9771b263 159 oprnds_info.quick_push (oprnd_info);
d092494c
IR
160 }
161
162 return oprnds_info;
163}
164
165
d3cfd39e
JJ
166/* Free operands info. */
167
d092494c 168static void
9771b263 169vect_free_oprnd_info (vec<slp_oprnd_info> &oprnds_info)
d092494c
IR
170{
171 int i;
172 slp_oprnd_info oprnd_info;
173
9771b263 174 FOR_EACH_VEC_ELT (oprnds_info, i, oprnd_info)
d3cfd39e 175 {
9771b263 176 oprnd_info->def_stmts.release ();
d3cfd39e
JJ
177 XDELETE (oprnd_info);
178 }
d092494c 179
9771b263 180 oprnds_info.release ();
d092494c
IR
181}
182
183
d755c7ef
RB
184/* Find the place of the data-ref in STMT in the interleaving chain that starts
185 from FIRST_STMT. Return -1 if the data-ref is not a part of the chain. */
186
187static int
355fe088 188vect_get_place_in_interleaving_chain (gimple *stmt, gimple *first_stmt)
d755c7ef 189{
355fe088 190 gimple *next_stmt = first_stmt;
d755c7ef
RB
191 int result = 0;
192
193 if (first_stmt != GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt)))
194 return -1;
195
196 do
197 {
198 if (next_stmt == stmt)
199 return result;
d755c7ef 200 next_stmt = GROUP_NEXT_ELEMENT (vinfo_for_stmt (next_stmt));
c8047699
RB
201 if (next_stmt)
202 result += GROUP_GAP (vinfo_for_stmt (next_stmt));
d755c7ef
RB
203 }
204 while (next_stmt);
205
206 return -1;
207}
208
209
d092494c
IR
210/* Get the defs for the rhs of STMT (collect them in OPRNDS_INFO), check that
211 they are of a valid type and that they match the defs of the first stmt of
4cecd659
BC
212 the SLP group (stored in OPRNDS_INFO). This function tries to match stmts
213 by swapping operands of STMT when possible. Non-zero *SWAP indicates swap
214 is required for cond_expr stmts. Specifically, *SWAP is 1 if STMT is cond
215 and operands of comparison need to be swapped; *SWAP is 2 if STMT is cond
216 and code of comparison needs to be inverted. If there is any operand swap
217 in this function, *SWAP is set to non-zero value.
218 If there was a fatal error return -1; if the error could be corrected by
219 swapping operands of father node of this one, return 1; if everything is
220 ok return 0. */
ebfd146a 221
4cecd659
BC
222static int
223vect_get_and_check_slp_defs (vec_info *vinfo, unsigned char *swap,
355fe088 224 gimple *stmt, unsigned stmt_num,
4cecd659 225 vec<slp_oprnd_info> *oprnds_info)
ebfd146a
IR
226{
227 tree oprnd;
228 unsigned int i, number_of_oprnds;
355fe088 229 gimple *def_stmt;
d092494c 230 enum vect_def_type dt = vect_uninitialized_def;
d092494c 231 bool pattern = false;
abf9bfbc 232 slp_oprnd_info oprnd_info;
b0b4483e
RB
233 int first_op_idx = 1;
234 bool commutative = false;
235 bool first_op_cond = false;
effb52da
RB
236 bool first = stmt_num == 0;
237 bool second = stmt_num == 1;
b8698a0f 238
d092494c 239 if (is_gimple_call (stmt))
190c2236
JJ
240 {
241 number_of_oprnds = gimple_call_num_args (stmt);
b0b4483e 242 first_op_idx = 3;
190c2236 243 }
f7e531cf
IR
244 else if (is_gimple_assign (stmt))
245 {
b0b4483e 246 enum tree_code code = gimple_assign_rhs_code (stmt);
f7e531cf 247 number_of_oprnds = gimple_num_ops (stmt) - 1;
4cecd659
BC
248 /* Swap can only be done for cond_expr if asked to, otherwise we
249 could result in different comparison code to the first stmt. */
a414c77f
IE
250 if (gimple_assign_rhs_code (stmt) == COND_EXPR
251 && COMPARISON_CLASS_P (gimple_assign_rhs1 (stmt)))
b0b4483e
RB
252 {
253 first_op_cond = true;
b0b4483e
RB
254 number_of_oprnds++;
255 }
256 else
257 commutative = commutative_tree_code (code);
f7e531cf 258 }
d092494c 259 else
b0b4483e 260 return -1;
ebfd146a 261
4cecd659
BC
262 bool swapped = (*swap != 0);
263 gcc_assert (!swapped || first_op_cond);
ebfd146a
IR
264 for (i = 0; i < number_of_oprnds; i++)
265 {
b0b4483e
RB
266again:
267 if (first_op_cond)
f7e531cf 268 {
4cecd659
BC
269 /* Map indicating how operands of cond_expr should be swapped. */
270 int maps[3][4] = {{0, 1, 2, 3}, {1, 0, 2, 3}, {0, 1, 3, 2}};
271 int *map = maps[*swap];
272
273 if (i < 2)
274 oprnd = TREE_OPERAND (gimple_op (stmt, first_op_idx), map[i]);
b0b4483e 275 else
4cecd659 276 oprnd = gimple_op (stmt, map[i]);
f7e531cf
IR
277 }
278 else
4cecd659 279 oprnd = gimple_op (stmt, first_op_idx + (swapped ? !i : i));
f7e531cf 280
9771b263 281 oprnd_info = (*oprnds_info)[i];
ebfd146a 282
81c40241 283 if (!vect_is_simple_use (oprnd, vinfo, &def_stmt, &dt))
ebfd146a 284 {
73fbfcad 285 if (dump_enabled_p ())
ebfd146a 286 {
78c60e3d 287 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
3fc356dc 288 "Build SLP failed: can't analyze def for ");
78c60e3d 289 dump_generic_expr (MSG_MISSED_OPTIMIZATION, TDF_SLIM, oprnd);
e645e942 290 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
ebfd146a
IR
291 }
292
b0b4483e 293 return -1;
ebfd146a
IR
294 }
295
a70d6342 296 /* Check if DEF_STMT is a part of a pattern in LOOP and get the def stmt
ff802fa1 297 from the pattern. Check that all the stmts of the node are in the
ebfd146a 298 pattern. */
f5709183 299 if (def_stmt && gimple_bb (def_stmt)
61d371eb 300 && vect_stmt_in_region_p (vinfo, def_stmt)
ebfd146a 301 && vinfo_for_stmt (def_stmt)
83197f37 302 && STMT_VINFO_IN_PATTERN_P (vinfo_for_stmt (def_stmt))
f5709183
IR
303 && !STMT_VINFO_RELEVANT (vinfo_for_stmt (def_stmt))
304 && !STMT_VINFO_LIVE_P (vinfo_for_stmt (def_stmt)))
ebfd146a 305 {
d092494c 306 pattern = true;
effb52da
RB
307 if (!first && !oprnd_info->first_pattern
308 /* Allow different pattern state for the defs of the
309 first stmt in reduction chains. */
310 && (oprnd_info->first_dt != vect_reduction_def
311 || (!second && !oprnd_info->second_pattern)))
d092494c 312 {
b0b4483e
RB
313 if (i == 0
314 && !swapped
315 && commutative)
316 {
317 swapped = true;
318 goto again;
319 }
320
73fbfcad 321 if (dump_enabled_p ())
d092494c 322 {
78c60e3d
SS
323 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
324 "Build SLP failed: some of the stmts"
325 " are in a pattern, and others are not ");
326 dump_generic_expr (MSG_MISSED_OPTIMIZATION, TDF_SLIM, oprnd);
e645e942 327 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
d092494c 328 }
ebfd146a 329
b0b4483e 330 return 1;
ebfd146a
IR
331 }
332
333 def_stmt = STMT_VINFO_RELATED_STMT (vinfo_for_stmt (def_stmt));
d092494c 334 dt = STMT_VINFO_DEF_TYPE (vinfo_for_stmt (def_stmt));
ebfd146a 335
f7e531cf 336 if (dt == vect_unknown_def_type)
ebfd146a 337 {
73fbfcad 338 if (dump_enabled_p ())
78c60e3d 339 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
e645e942 340 "Unsupported pattern.\n");
b0b4483e 341 return -1;
ebfd146a
IR
342 }
343
344 switch (gimple_code (def_stmt))
345 {
81c40241
RB
346 case GIMPLE_PHI:
347 case GIMPLE_ASSIGN:
348 break;
349
350 default:
351 if (dump_enabled_p ())
352 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
353 "unsupported defining stmt:\n");
354 return -1;
ebfd146a
IR
355 }
356 }
357
effb52da
RB
358 if (second)
359 oprnd_info->second_pattern = pattern;
360
d092494c 361 if (first)
ebfd146a 362 {
d092494c
IR
363 oprnd_info->first_dt = dt;
364 oprnd_info->first_pattern = pattern;
793d9a16 365 oprnd_info->first_op_type = TREE_TYPE (oprnd);
ebfd146a 366 }
ebfd146a
IR
367 else
368 {
d092494c
IR
369 /* Not first stmt of the group, check that the def-stmt/s match
370 the def-stmt/s of the first stmt. Allow different definition
371 types for reduction chains: the first stmt must be a
372 vect_reduction_def (a phi node), and the rest
373 vect_internal_def. */
374 if (((oprnd_info->first_dt != dt
375 && !(oprnd_info->first_dt == vect_reduction_def
793d9a16
RB
376 && dt == vect_internal_def)
377 && !((oprnd_info->first_dt == vect_external_def
378 || oprnd_info->first_dt == vect_constant_def)
379 && (dt == vect_external_def
380 || dt == vect_constant_def)))
381 || !types_compatible_p (oprnd_info->first_op_type,
382 TREE_TYPE (oprnd))))
ebfd146a 383 {
b0b4483e
RB
384 /* Try swapping operands if we got a mismatch. */
385 if (i == 0
386 && !swapped
387 && commutative)
388 {
389 swapped = true;
390 goto again;
391 }
392
abf9bfbc
RB
393 if (dump_enabled_p ())
394 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
e645e942 395 "Build SLP failed: different types\n");
d092494c 396
b0b4483e 397 return 1;
ebfd146a
IR
398 }
399 }
400
401 /* Check the types of the definitions. */
d092494c 402 switch (dt)
ebfd146a
IR
403 {
404 case vect_constant_def:
8644a673 405 case vect_external_def:
e7baeb39 406 case vect_reduction_def:
ebfd146a 407 break;
b8698a0f 408
e7baeb39 409 case vect_induction_def:
8644a673 410 case vect_internal_def:
abf9bfbc 411 oprnd_info->def_stmts.quick_push (def_stmt);
ebfd146a
IR
412 break;
413
414 default:
415 /* FORNOW: Not supported. */
73fbfcad 416 if (dump_enabled_p ())
ebfd146a 417 {
78c60e3d
SS
418 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
419 "Build SLP failed: illegal type of def ");
81c40241 420 dump_generic_expr (MSG_MISSED_OPTIMIZATION, TDF_SLIM, oprnd);
e645e942 421 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
ebfd146a
IR
422 }
423
b0b4483e 424 return -1;
ebfd146a
IR
425 }
426 }
427
b0b4483e
RB
428 /* Swap operands. */
429 if (swapped)
430 {
78810bd3
RB
431 /* If there are already uses of this stmt in a SLP instance then
432 we've committed to the operand order and can't swap it. */
433 if (STMT_VINFO_NUM_SLP_USES (vinfo_for_stmt (stmt)) != 0)
434 {
435 if (dump_enabled_p ())
436 {
437 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
438 "Build SLP failed: cannot swap operands of "
439 "shared stmt ");
440 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM, stmt, 0);
441 }
442 return -1;
443 }
444
b0b4483e
RB
445 if (first_op_cond)
446 {
447 tree cond = gimple_assign_rhs1 (stmt);
4cecd659
BC
448 enum tree_code code = TREE_CODE (cond);
449
450 /* Swap. */
451 if (*swap == 1)
452 {
453 swap_ssa_operands (stmt, &TREE_OPERAND (cond, 0),
454 &TREE_OPERAND (cond, 1));
455 TREE_SET_CODE (cond, swap_tree_comparison (code));
456 }
457 /* Invert. */
458 else
459 {
460 swap_ssa_operands (stmt, gimple_assign_rhs2_ptr (stmt),
461 gimple_assign_rhs3_ptr (stmt));
462 bool honor_nans = HONOR_NANS (TREE_OPERAND (cond, 0));
463 code = invert_tree_comparison (TREE_CODE (cond), honor_nans);
464 gcc_assert (code != ERROR_MARK);
465 TREE_SET_CODE (cond, code);
466 }
b0b4483e
RB
467 }
468 else
469 swap_ssa_operands (stmt, gimple_assign_rhs1_ptr (stmt),
470 gimple_assign_rhs2_ptr (stmt));
78810bd3
RB
471 if (dump_enabled_p ())
472 {
473 dump_printf_loc (MSG_NOTE, vect_location,
474 "swapped operands to match def types in ");
475 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
476 }
b0b4483e
RB
477 }
478
4cecd659 479 *swap = swapped;
b0b4483e 480 return 0;
ebfd146a
IR
481}
482
483
6983e6b5
RB
484/* Verify if the scalar stmts STMTS are isomorphic, require data
485 permutation or are of unsupported types of operation. Return
486 true if they are, otherwise return false and indicate in *MATCHES
487 which stmts are not isomorphic to the first one. If MATCHES[0]
488 is false then this indicates the comparison could not be
4cecd659
BC
489 carried out or the stmts will never be vectorized by SLP.
490
491 Note COND_EXPR is possibly ismorphic to another one after swapping its
492 operands. Set SWAP[i] to 1 if stmt I is COND_EXPR and isomorphic to
493 the first stmt by swapping the two operands of comparison; set SWAP[i]
494 to 2 if stmt I is isormorphic to the first stmt by inverting the code
495 of comparison. Take A1 >= B1 ? X1 : Y1 as an exmple, it can be swapped
496 to (B1 <= A1 ? X1 : Y1); or be inverted to (A1 < B1) ? Y1 : X1. */
ebfd146a
IR
497
498static bool
4cecd659 499vect_build_slp_tree_1 (vec_info *vinfo, unsigned char *swap,
355fe088 500 vec<gimple *> stmts, unsigned int group_size,
6983e6b5 501 unsigned nops, unsigned int *max_nunits,
97a1a642 502 bool *matches, bool *two_operators)
ebfd146a 503{
ebfd146a 504 unsigned int i;
355fe088 505 gimple *first_stmt = stmts[0], *stmt = stmts[0];
6876e5bc
RB
506 enum tree_code first_stmt_code = ERROR_MARK;
507 enum tree_code alt_stmt_code = ERROR_MARK;
508 enum tree_code rhs_code = ERROR_MARK;
f7e531cf 509 enum tree_code first_cond_code = ERROR_MARK;
ebfd146a 510 tree lhs;
6983e6b5 511 bool need_same_oprnds = false;
e00cdb8a 512 tree vectype = NULL_TREE, scalar_type, first_op1 = NULL_TREE;
ebfd146a
IR
513 optab optab;
514 int icode;
ef4bddc2
RS
515 machine_mode optab_op2_mode;
516 machine_mode vec_mode;
ebfd146a 517 HOST_WIDE_INT dummy;
355fe088 518 gimple *first_load = NULL, *prev_first_load = NULL;
d092494c 519
ebfd146a 520 /* For every stmt in NODE find its def stmt/s. */
9771b263 521 FOR_EACH_VEC_ELT (stmts, i, stmt)
ebfd146a 522 {
4cecd659 523 swap[i] = 0;
6983e6b5
RB
524 matches[i] = false;
525
73fbfcad 526 if (dump_enabled_p ())
ebfd146a 527 {
78c60e3d
SS
528 dump_printf_loc (MSG_NOTE, vect_location, "Build SLP for ");
529 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
ebfd146a
IR
530 }
531
4b5caab7
IR
532 /* Fail to vectorize statements marked as unvectorizable. */
533 if (!STMT_VINFO_VECTORIZABLE (vinfo_for_stmt (stmt)))
534 {
73fbfcad 535 if (dump_enabled_p ())
4b5caab7 536 {
78c60e3d
SS
537 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
538 "Build SLP failed: unvectorizable statement ");
539 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM, stmt, 0);
4b5caab7 540 }
6983e6b5
RB
541 /* Fatal mismatch. */
542 matches[0] = false;
4b5caab7
IR
543 return false;
544 }
545
ebfd146a
IR
546 lhs = gimple_get_lhs (stmt);
547 if (lhs == NULL_TREE)
548 {
73fbfcad 549 if (dump_enabled_p ())
ebfd146a 550 {
78c60e3d
SS
551 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
552 "Build SLP failed: not GIMPLE_ASSIGN nor "
553 "GIMPLE_CALL ");
554 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM, stmt, 0);
ebfd146a 555 }
6983e6b5
RB
556 /* Fatal mismatch. */
557 matches[0] = false;
ebfd146a
IR
558 return false;
559 }
560
b8698a0f 561 scalar_type = vect_get_smallest_scalar_type (stmt, &dummy, &dummy);
ebfd146a
IR
562 vectype = get_vectype_for_scalar_type (scalar_type);
563 if (!vectype)
564 {
73fbfcad 565 if (dump_enabled_p ())
ebfd146a 566 {
78c60e3d
SS
567 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
568 "Build SLP failed: unsupported data-type ");
569 dump_generic_expr (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
570 scalar_type);
e645e942 571 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
ebfd146a 572 }
6983e6b5
RB
573 /* Fatal mismatch. */
574 matches[0] = false;
ebfd146a
IR
575 return false;
576 }
b8698a0f 577
dfc55d30
RB
578 /* If populating the vector type requires unrolling then fail
579 before adjusting *max_nunits for basic-block vectorization. */
310213d4 580 if (is_a <bb_vec_info> (vinfo)
dfc55d30
RB
581 && TYPE_VECTOR_SUBPARTS (vectype) > group_size)
582 {
583 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
584 "Build SLP failed: unrolling required "
585 "in basic block SLP\n");
586 /* Fatal mismatch. */
587 matches[0] = false;
588 return false;
589 }
590
4ef69dfc
IR
591 /* In case of multiple types we need to detect the smallest type. */
592 if (*max_nunits < TYPE_VECTOR_SUBPARTS (vectype))
97a1a642 593 *max_nunits = TYPE_VECTOR_SUBPARTS (vectype);
b8698a0f 594
538dd0b7 595 if (gcall *call_stmt = dyn_cast <gcall *> (stmt))
190c2236
JJ
596 {
597 rhs_code = CALL_EXPR;
538dd0b7
DM
598 if (gimple_call_internal_p (call_stmt)
599 || gimple_call_tail_p (call_stmt)
600 || gimple_call_noreturn_p (call_stmt)
601 || !gimple_call_nothrow_p (call_stmt)
602 || gimple_call_chain (call_stmt))
190c2236 603 {
73fbfcad 604 if (dump_enabled_p ())
190c2236 605 {
78c60e3d
SS
606 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
607 "Build SLP failed: unsupported call type ");
538dd0b7
DM
608 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
609 call_stmt, 0);
190c2236 610 }
6983e6b5
RB
611 /* Fatal mismatch. */
612 matches[0] = false;
190c2236
JJ
613 return false;
614 }
615 }
ebfd146a
IR
616 else
617 rhs_code = gimple_assign_rhs_code (stmt);
618
619 /* Check the operation. */
620 if (i == 0)
621 {
622 first_stmt_code = rhs_code;
623
b8698a0f 624 /* Shift arguments should be equal in all the packed stmts for a
ebfd146a
IR
625 vector shift with scalar shift operand. */
626 if (rhs_code == LSHIFT_EXPR || rhs_code == RSHIFT_EXPR
627 || rhs_code == LROTATE_EXPR
628 || rhs_code == RROTATE_EXPR)
629 {
630 vec_mode = TYPE_MODE (vectype);
631
632 /* First see if we have a vector/vector shift. */
633 optab = optab_for_tree_code (rhs_code, vectype,
634 optab_vector);
635
636 if (!optab
947131ba 637 || optab_handler (optab, vec_mode) == CODE_FOR_nothing)
ebfd146a
IR
638 {
639 /* No vector/vector shift, try for a vector/scalar shift. */
640 optab = optab_for_tree_code (rhs_code, vectype,
641 optab_scalar);
642
643 if (!optab)
644 {
73fbfcad 645 if (dump_enabled_p ())
78c60e3d 646 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
e645e942 647 "Build SLP failed: no optab.\n");
6983e6b5
RB
648 /* Fatal mismatch. */
649 matches[0] = false;
ebfd146a
IR
650 return false;
651 }
947131ba 652 icode = (int) optab_handler (optab, vec_mode);
ebfd146a
IR
653 if (icode == CODE_FOR_nothing)
654 {
73fbfcad 655 if (dump_enabled_p ())
78c60e3d
SS
656 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
657 "Build SLP failed: "
e645e942 658 "op not supported by target.\n");
6983e6b5
RB
659 /* Fatal mismatch. */
660 matches[0] = false;
ebfd146a
IR
661 return false;
662 }
663 optab_op2_mode = insn_data[icode].operand[2].mode;
664 if (!VECTOR_MODE_P (optab_op2_mode))
665 {
666 need_same_oprnds = true;
667 first_op1 = gimple_assign_rhs2 (stmt);
668 }
669 }
670 }
36ba4aae
IR
671 else if (rhs_code == WIDEN_LSHIFT_EXPR)
672 {
673 need_same_oprnds = true;
674 first_op1 = gimple_assign_rhs2 (stmt);
675 }
ebfd146a
IR
676 }
677 else
678 {
6876e5bc
RB
679 if (first_stmt_code != rhs_code
680 && alt_stmt_code == ERROR_MARK)
681 alt_stmt_code = rhs_code;
ebfd146a
IR
682 if (first_stmt_code != rhs_code
683 && (first_stmt_code != IMAGPART_EXPR
684 || rhs_code != REALPART_EXPR)
685 && (first_stmt_code != REALPART_EXPR
69f11a13 686 || rhs_code != IMAGPART_EXPR)
6876e5bc
RB
687 /* Handle mismatches in plus/minus by computing both
688 and merging the results. */
689 && !((first_stmt_code == PLUS_EXPR
690 || first_stmt_code == MINUS_EXPR)
691 && (alt_stmt_code == PLUS_EXPR
692 || alt_stmt_code == MINUS_EXPR)
693 && rhs_code == alt_stmt_code)
0d0293ac 694 && !(STMT_VINFO_GROUPED_ACCESS (vinfo_for_stmt (stmt))
69f11a13 695 && (first_stmt_code == ARRAY_REF
38000232 696 || first_stmt_code == BIT_FIELD_REF
69f11a13
IR
697 || first_stmt_code == INDIRECT_REF
698 || first_stmt_code == COMPONENT_REF
699 || first_stmt_code == MEM_REF)))
ebfd146a 700 {
73fbfcad 701 if (dump_enabled_p ())
ebfd146a 702 {
78c60e3d
SS
703 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
704 "Build SLP failed: different operation "
705 "in stmt ");
706 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM, stmt, 0);
6876e5bc
RB
707 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
708 "original stmt ");
709 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
710 first_stmt, 0);
ebfd146a 711 }
6983e6b5
RB
712 /* Mismatch. */
713 continue;
ebfd146a 714 }
b8698a0f
L
715
716 if (need_same_oprnds
ebfd146a
IR
717 && !operand_equal_p (first_op1, gimple_assign_rhs2 (stmt), 0))
718 {
73fbfcad 719 if (dump_enabled_p ())
ebfd146a 720 {
78c60e3d
SS
721 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
722 "Build SLP failed: different shift "
723 "arguments in ");
724 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM, stmt, 0);
ebfd146a 725 }
6983e6b5
RB
726 /* Mismatch. */
727 continue;
ebfd146a 728 }
190c2236
JJ
729
730 if (rhs_code == CALL_EXPR)
731 {
355fe088 732 gimple *first_stmt = stmts[0];
190c2236
JJ
733 if (gimple_call_num_args (stmt) != nops
734 || !operand_equal_p (gimple_call_fn (first_stmt),
735 gimple_call_fn (stmt), 0)
736 || gimple_call_fntype (first_stmt)
737 != gimple_call_fntype (stmt))
738 {
73fbfcad 739 if (dump_enabled_p ())
190c2236 740 {
78c60e3d
SS
741 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
742 "Build SLP failed: different calls in ");
743 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
744 stmt, 0);
190c2236 745 }
6983e6b5
RB
746 /* Mismatch. */
747 continue;
190c2236
JJ
748 }
749 }
ebfd146a
IR
750 }
751
0d0293ac
MM
752 /* Grouped store or load. */
753 if (STMT_VINFO_GROUPED_ACCESS (vinfo_for_stmt (stmt)))
ebfd146a
IR
754 {
755 if (REFERENCE_CLASS_P (lhs))
756 {
757 /* Store. */
6983e6b5 758 ;
ebfd146a 759 }
b5aeb3bb
IR
760 else
761 {
762 /* Load. */
e14c1050 763 first_load = GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt));
b5aeb3bb
IR
764 if (prev_first_load)
765 {
766 /* Check that there are no loads from different interleaving
6983e6b5
RB
767 chains in the same node. */
768 if (prev_first_load != first_load)
78c60e3d 769 {
73fbfcad 770 if (dump_enabled_p ())
b5aeb3bb 771 {
78c60e3d
SS
772 dump_printf_loc (MSG_MISSED_OPTIMIZATION,
773 vect_location,
774 "Build SLP failed: different "
775 "interleaving chains in one node ");
776 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
777 stmt, 0);
b5aeb3bb 778 }
6983e6b5
RB
779 /* Mismatch. */
780 continue;
b5aeb3bb
IR
781 }
782 }
783 else
784 prev_first_load = first_load;
ebfd146a 785 }
0d0293ac 786 } /* Grouped access. */
ebfd146a
IR
787 else
788 {
789 if (TREE_CODE_CLASS (rhs_code) == tcc_reference)
790 {
0d0293ac 791 /* Not grouped load. */
73fbfcad 792 if (dump_enabled_p ())
ebfd146a 793 {
78c60e3d
SS
794 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
795 "Build SLP failed: not grouped load ");
796 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM, stmt, 0);
ebfd146a
IR
797 }
798
0d0293ac 799 /* FORNOW: Not grouped loads are not supported. */
6983e6b5
RB
800 /* Fatal mismatch. */
801 matches[0] = false;
ebfd146a
IR
802 return false;
803 }
804
805 /* Not memory operation. */
806 if (TREE_CODE_CLASS (rhs_code) != tcc_binary
f7e531cf 807 && TREE_CODE_CLASS (rhs_code) != tcc_unary
effb52da 808 && TREE_CODE_CLASS (rhs_code) != tcc_expression
42fd8198 809 && TREE_CODE_CLASS (rhs_code) != tcc_comparison
190c2236 810 && rhs_code != CALL_EXPR)
ebfd146a 811 {
73fbfcad 812 if (dump_enabled_p ())
ebfd146a 813 {
78c60e3d
SS
814 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
815 "Build SLP failed: operation");
816 dump_printf (MSG_MISSED_OPTIMIZATION, " unsupported ");
817 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM, stmt, 0);
ebfd146a 818 }
6983e6b5
RB
819 /* Fatal mismatch. */
820 matches[0] = false;
ebfd146a
IR
821 return false;
822 }
823
4cecd659
BC
824 if (rhs_code == COND_EXPR)
825 {
826 tree cond_expr = gimple_assign_rhs1 (stmt);
827 enum tree_code cond_code = TREE_CODE (cond_expr);
828 enum tree_code swap_code = ERROR_MARK;
829 enum tree_code invert_code = ERROR_MARK;
f7e531cf
IR
830
831 if (i == 0)
832 first_cond_code = TREE_CODE (cond_expr);
4cecd659
BC
833 else if (TREE_CODE_CLASS (cond_code) == tcc_comparison)
834 {
835 bool honor_nans = HONOR_NANS (TREE_OPERAND (cond_expr, 0));
836 swap_code = swap_tree_comparison (cond_code);
837 invert_code = invert_tree_comparison (cond_code, honor_nans);
838 }
839
840 if (first_cond_code == cond_code)
841 ;
842 /* Isomorphic can be achieved by swapping. */
843 else if (first_cond_code == swap_code)
844 swap[i] = 1;
845 /* Isomorphic can be achieved by inverting. */
846 else if (first_cond_code == invert_code)
847 swap[i] = 2;
848 else
849 {
850 if (dump_enabled_p ())
851 {
852 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
78c60e3d
SS
853 "Build SLP failed: different"
854 " operation");
4cecd659 855 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
78c60e3d 856 stmt, 0);
4cecd659 857 }
6983e6b5
RB
858 /* Mismatch. */
859 continue;
f7e531cf 860 }
4cecd659 861 }
ebfd146a 862 }
6983e6b5
RB
863
864 matches[i] = true;
865 }
866
867 for (i = 0; i < group_size; ++i)
868 if (!matches[i])
869 return false;
870
6876e5bc
RB
871 /* If we allowed a two-operation SLP node verify the target can cope
872 with the permute we are going to use. */
873 if (alt_stmt_code != ERROR_MARK
874 && TREE_CODE_CLASS (alt_stmt_code) != tcc_reference)
875 {
876 unsigned char *sel
877 = XALLOCAVEC (unsigned char, TYPE_VECTOR_SUBPARTS (vectype));
878 for (i = 0; i < TYPE_VECTOR_SUBPARTS (vectype); ++i)
879 {
880 sel[i] = i;
881 if (gimple_assign_rhs_code (stmts[i % group_size]) == alt_stmt_code)
882 sel[i] += TYPE_VECTOR_SUBPARTS (vectype);
883 }
884 if (!can_vec_perm_p (TYPE_MODE (vectype), false, sel))
885 {
886 for (i = 0; i < group_size; ++i)
887 if (gimple_assign_rhs_code (stmts[i]) == alt_stmt_code)
888 {
889 matches[i] = false;
890 if (dump_enabled_p ())
891 {
892 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
893 "Build SLP failed: different operation "
894 "in stmt ");
895 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
896 stmts[i], 0);
897 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
898 "original stmt ");
899 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
900 first_stmt, 0);
901 }
902 }
903 return false;
904 }
905 *two_operators = true;
906 }
907
6983e6b5
RB
908 return true;
909}
910
911/* Recursively build an SLP tree starting from NODE.
912 Fail (and return a value not equal to zero) if def-stmts are not
913 isomorphic, require data permutation or are of unsupported types of
914 operation. Otherwise, return 0.
915 The value returned is the depth in the SLP tree where a mismatch
916 was found. */
917
e403d17e 918static slp_tree
310213d4 919vect_build_slp_tree (vec_info *vinfo,
e403d17e 920 vec<gimple *> stmts, unsigned int group_size,
6983e6b5
RB
921 unsigned int *max_nunits,
922 vec<slp_tree> *loads,
1428105c
RB
923 bool *matches, unsigned *npermutes, unsigned *tree_size,
924 unsigned max_tree_size)
6983e6b5 925{
e403d17e 926 unsigned nops, i, this_tree_size = 0, this_max_nunits = *max_nunits;
355fe088 927 gimple *stmt;
e403d17e 928 slp_tree node;
6983e6b5 929
6983e6b5
RB
930 matches[0] = false;
931
e403d17e 932 stmt = stmts[0];
6983e6b5
RB
933 if (is_gimple_call (stmt))
934 nops = gimple_call_num_args (stmt);
935 else if (is_gimple_assign (stmt))
936 {
937 nops = gimple_num_ops (stmt) - 1;
938 if (gimple_assign_rhs_code (stmt) == COND_EXPR)
939 nops++;
ebfd146a 940 }
e7baeb39
RB
941 else if (gimple_code (stmt) == GIMPLE_PHI)
942 nops = 0;
6983e6b5 943 else
e403d17e 944 return NULL;
6983e6b5 945
e7baeb39
RB
946 /* If the SLP node is a PHI (induction), terminate the recursion. */
947 if (gimple_code (stmt) == GIMPLE_PHI)
948 {
949 FOR_EACH_VEC_ELT (stmts, i, stmt)
950 if (stmt != stmts[0])
951 /* Induction from different IVs is not supported. */
952 return NULL;
953 node = vect_create_new_slp_node (stmts);
954 return node;
955 }
956
957
6876e5bc 958 bool two_operators = false;
4cecd659
BC
959 unsigned char *swap = XALLOCAVEC (unsigned char, group_size);
960 if (!vect_build_slp_tree_1 (vinfo, swap,
e403d17e
RB
961 stmts, group_size, nops,
962 &this_max_nunits, matches, &two_operators))
963 return NULL;
ebfd146a 964
6983e6b5
RB
965 /* If the SLP node is a load, terminate the recursion. */
966 if (STMT_VINFO_GROUPED_ACCESS (vinfo_for_stmt (stmt))
967 && DR_IS_READ (STMT_VINFO_DATA_REF (vinfo_for_stmt (stmt))))
ebfd146a 968 {
e403d17e
RB
969 *max_nunits = this_max_nunits;
970 node = vect_create_new_slp_node (stmts);
971 loads->safe_push (node);
972 return node;
ebfd146a
IR
973 }
974
6983e6b5
RB
975 /* Get at the operands, verifying they are compatible. */
976 vec<slp_oprnd_info> oprnds_info = vect_create_oprnd_info (nops, group_size);
977 slp_oprnd_info oprnd_info;
e403d17e 978 FOR_EACH_VEC_ELT (stmts, i, stmt)
6983e6b5 979 {
4cecd659
BC
980 int res = vect_get_and_check_slp_defs (vinfo, &swap[i],
981 stmt, i, &oprnds_info);
982 if (res != 0)
983 matches[(res == -1) ? 0 : i] = false;
984 if (!matches[0])
985 break;
6983e6b5 986 }
b0b4483e
RB
987 for (i = 0; i < group_size; ++i)
988 if (!matches[i])
989 {
990 vect_free_oprnd_info (oprnds_info);
e403d17e 991 return NULL;
b0b4483e 992 }
6983e6b5 993
e403d17e
RB
994 auto_vec<slp_tree, 4> children;
995 auto_vec<slp_tree> this_loads;
996
997 stmt = stmts[0];
6983e6b5 998
b8698a0f 999 /* Create SLP_TREE nodes for the definition node/s. */
9771b263 1000 FOR_EACH_VEC_ELT (oprnds_info, i, oprnd_info)
ebfd146a 1001 {
d092494c 1002 slp_tree child;
e403d17e
RB
1003 unsigned old_nloads = this_loads.length ();
1004 unsigned old_tree_size = this_tree_size;
1005 unsigned int j;
b8698a0f 1006
e7baeb39
RB
1007 if (oprnd_info->first_dt != vect_internal_def
1008 && oprnd_info->first_dt != vect_induction_def)
d092494c 1009 continue;
ebfd146a 1010
1428105c
RB
1011 if (++this_tree_size > max_tree_size)
1012 {
e403d17e
RB
1013 FOR_EACH_VEC_ELT (children, j, child)
1014 vect_free_slp_tree (child);
1428105c 1015 vect_free_oprnd_info (oprnds_info);
e403d17e 1016 return NULL;
1428105c
RB
1017 }
1018
e403d17e
RB
1019 if ((child = vect_build_slp_tree (vinfo, oprnd_info->def_stmts,
1020 group_size, &this_max_nunits,
1021 &this_loads, matches, npermutes,
1022 &this_tree_size,
1023 max_tree_size)) != NULL)
6983e6b5 1024 {
3fc356dc
RB
1025 /* If we have all children of child built up from scalars then just
1026 throw that away and build it up this node from scalars. */
995b6fe0
RB
1027 if (!SLP_TREE_CHILDREN (child).is_empty ()
1028 /* ??? Rejecting patterns this way doesn't work. We'd have to
1029 do extra work to cancel the pattern so the uses see the
1030 scalar version. */
1031 && !is_pattern_stmt_p
1032 (vinfo_for_stmt (SLP_TREE_SCALAR_STMTS (child)[0])))
3fc356dc 1033 {
3fc356dc
RB
1034 slp_tree grandchild;
1035
1036 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (child), j, grandchild)
603cca93 1037 if (SLP_TREE_DEF_TYPE (grandchild) == vect_internal_def)
3fc356dc
RB
1038 break;
1039 if (!grandchild)
1040 {
1041 /* Roll back. */
e403d17e
RB
1042 this_loads.truncate (old_nloads);
1043 this_tree_size = old_tree_size;
3fc356dc 1044 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (child), j, grandchild)
603cca93 1045 vect_free_slp_tree (grandchild);
3fc356dc
RB
1046 SLP_TREE_CHILDREN (child).truncate (0);
1047
1048 dump_printf_loc (MSG_NOTE, vect_location,
1049 "Building parent vector operands from "
1050 "scalars instead\n");
1051 oprnd_info->def_stmts = vNULL;
603cca93 1052 SLP_TREE_DEF_TYPE (child) = vect_external_def;
e403d17e 1053 children.safe_push (child);
3fc356dc
RB
1054 continue;
1055 }
1056 }
1057
6983e6b5 1058 oprnd_info->def_stmts = vNULL;
e403d17e 1059 children.safe_push (child);
6983e6b5
RB
1060 continue;
1061 }
1062
90dd6e3d
RB
1063 /* If the SLP build failed fatally and we analyze a basic-block
1064 simply treat nodes we fail to build as externally defined
1065 (and thus build vectors from the scalar defs).
1066 The cost model will reject outright expensive cases.
1067 ??? This doesn't treat cases where permutation ultimatively
1068 fails (or we don't try permutation below). Ideally we'd
1069 even compute a permutation that will end up with the maximum
1070 SLP tree size... */
310213d4 1071 if (is_a <bb_vec_info> (vinfo)
90dd6e3d
RB
1072 && !matches[0]
1073 /* ??? Rejecting patterns this way doesn't work. We'd have to
1074 do extra work to cancel the pattern so the uses see the
1075 scalar version. */
1076 && !is_pattern_stmt_p (vinfo_for_stmt (stmt)))
1077 {
1078 dump_printf_loc (MSG_NOTE, vect_location,
1079 "Building vector operands from scalars\n");
e403d17e 1080 child = vect_create_new_slp_node (oprnd_info->def_stmts);
603cca93 1081 SLP_TREE_DEF_TYPE (child) = vect_external_def;
e403d17e
RB
1082 children.safe_push (child);
1083 oprnd_info->def_stmts = vNULL;
90dd6e3d
RB
1084 continue;
1085 }
1086
6983e6b5
RB
1087 /* If the SLP build for operand zero failed and operand zero
1088 and one can be commutated try that for the scalar stmts
1089 that failed the match. */
1090 if (i == 0
1091 /* A first scalar stmt mismatch signals a fatal mismatch. */
1092 && matches[0]
1093 /* ??? For COND_EXPRs we can swap the comparison operands
1094 as well as the arms under some constraints. */
1095 && nops == 2
1096 && oprnds_info[1]->first_dt == vect_internal_def
1097 && is_gimple_assign (stmt)
1098 && commutative_tree_code (gimple_assign_rhs_code (stmt))
e403d17e 1099 && ! two_operators
6983e6b5
RB
1100 /* Do so only if the number of not successful permutes was nor more
1101 than a cut-ff as re-trying the recursive match on
1102 possibly each level of the tree would expose exponential
1103 behavior. */
1104 && *npermutes < 4)
1105 {
78810bd3
RB
1106 /* Verify if we can safely swap or if we committed to a specific
1107 operand order already. */
1108 for (j = 0; j < group_size; ++j)
1109 if (!matches[j]
4cecd659
BC
1110 && (swap[j] != 0
1111 || STMT_VINFO_NUM_SLP_USES (vinfo_for_stmt (stmts[j]))))
78810bd3
RB
1112 {
1113 if (dump_enabled_p ())
1114 {
1115 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
1116 "Build SLP failed: cannot swap operands "
1117 "of shared stmt ");
1118 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
1119 stmts[j], 0);
1120 }
1121 goto fail;
1122 }
1123
6983e6b5 1124 /* Swap mismatched definition stmts. */
b0b4483e
RB
1125 dump_printf_loc (MSG_NOTE, vect_location,
1126 "Re-trying with swapped operands of stmts ");
e72baed7 1127 for (j = 0; j < group_size; ++j)
6983e6b5
RB
1128 if (!matches[j])
1129 {
6b4db501
MM
1130 std::swap (oprnds_info[0]->def_stmts[j],
1131 oprnds_info[1]->def_stmts[j]);
b0b4483e 1132 dump_printf (MSG_NOTE, "%d ", j);
6983e6b5 1133 }
b0b4483e 1134 dump_printf (MSG_NOTE, "\n");
74574669
RB
1135 /* And try again with scratch 'matches' ... */
1136 bool *tem = XALLOCAVEC (bool, group_size);
e403d17e
RB
1137 if ((child = vect_build_slp_tree (vinfo, oprnd_info->def_stmts,
1138 group_size, &this_max_nunits,
1139 &this_loads, tem, npermutes,
1140 &this_tree_size,
1141 max_tree_size)) != NULL)
6983e6b5 1142 {
60f2b864
RB
1143 /* ... so if successful we can apply the operand swapping
1144 to the GIMPLE IL. This is necessary because for example
1145 vect_get_slp_defs uses operand indexes and thus expects
1146 canonical operand order. This is also necessary even
1147 if we end up building the operand from scalars as
1148 we'll continue to process swapped operand two. */
1149 for (j = 0; j < group_size; ++j)
f47cda24 1150 {
e403d17e 1151 gimple *stmt = stmts[j];
f47cda24
RB
1152 gimple_set_plf (stmt, GF_PLF_1, false);
1153 }
1154 for (j = 0; j < group_size; ++j)
1155 {
e403d17e 1156 gimple *stmt = stmts[j];
f47cda24
RB
1157 if (!matches[j])
1158 {
1159 /* Avoid swapping operands twice. */
1160 if (gimple_plf (stmt, GF_PLF_1))
1161 continue;
1162 swap_ssa_operands (stmt, gimple_assign_rhs1_ptr (stmt),
1163 gimple_assign_rhs2_ptr (stmt));
1164 gimple_set_plf (stmt, GF_PLF_1, true);
1165 }
1166 }
1167 /* Verify we swap all duplicates or none. */
1168 if (flag_checking)
1169 for (j = 0; j < group_size; ++j)
60f2b864 1170 {
e403d17e 1171 gimple *stmt = stmts[j];
f47cda24 1172 gcc_assert (gimple_plf (stmt, GF_PLF_1) == ! matches[j]);
60f2b864
RB
1173 }
1174
85c69b0b
RB
1175 /* If we have all children of child built up from scalars then
1176 just throw that away and build it up this node from scalars. */
995b6fe0
RB
1177 if (!SLP_TREE_CHILDREN (child).is_empty ()
1178 /* ??? Rejecting patterns this way doesn't work. We'd have
1179 to do extra work to cancel the pattern so the uses see the
1180 scalar version. */
1181 && !is_pattern_stmt_p
1182 (vinfo_for_stmt (SLP_TREE_SCALAR_STMTS (child)[0])))
85c69b0b
RB
1183 {
1184 unsigned int j;
1185 slp_tree grandchild;
1186
1187 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (child), j, grandchild)
603cca93 1188 if (SLP_TREE_DEF_TYPE (grandchild) == vect_internal_def)
85c69b0b
RB
1189 break;
1190 if (!grandchild)
1191 {
1192 /* Roll back. */
e403d17e
RB
1193 this_loads.truncate (old_nloads);
1194 this_tree_size = old_tree_size;
85c69b0b
RB
1195 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (child), j, grandchild)
1196 vect_free_slp_tree (grandchild);
1197 SLP_TREE_CHILDREN (child).truncate (0);
1198
1199 dump_printf_loc (MSG_NOTE, vect_location,
1200 "Building parent vector operands from "
1201 "scalars instead\n");
1202 oprnd_info->def_stmts = vNULL;
603cca93 1203 SLP_TREE_DEF_TYPE (child) = vect_external_def;
e403d17e 1204 children.safe_push (child);
85c69b0b
RB
1205 continue;
1206 }
1207 }
1208
6983e6b5 1209 oprnd_info->def_stmts = vNULL;
e403d17e 1210 children.safe_push (child);
6983e6b5
RB
1211 continue;
1212 }
1213
1214 ++*npermutes;
1215 }
1216
78810bd3 1217fail:
e403d17e
RB
1218 gcc_assert (child == NULL);
1219 FOR_EACH_VEC_ELT (children, j, child)
1220 vect_free_slp_tree (child);
6983e6b5 1221 vect_free_oprnd_info (oprnds_info);
e403d17e 1222 return NULL;
ebfd146a
IR
1223 }
1224
e403d17e
RB
1225 vect_free_oprnd_info (oprnds_info);
1226
1428105c
RB
1227 if (tree_size)
1228 *tree_size += this_tree_size;
e403d17e
RB
1229 *max_nunits = this_max_nunits;
1230 loads->safe_splice (this_loads);
1428105c 1231
e403d17e
RB
1232 node = vect_create_new_slp_node (stmts);
1233 SLP_TREE_TWO_OPERATORS (node) = two_operators;
1234 SLP_TREE_CHILDREN (node).splice (children);
1235 return node;
ebfd146a
IR
1236}
1237
78c60e3d 1238/* Dump a slp tree NODE using flags specified in DUMP_KIND. */
ebfd146a
IR
1239
1240static void
1a817418 1241vect_print_slp_tree (dump_flags_t dump_kind, location_t loc, slp_tree node)
ebfd146a
IR
1242{
1243 int i;
355fe088 1244 gimple *stmt;
d755c7ef 1245 slp_tree child;
ebfd146a 1246
603cca93
RB
1247 dump_printf_loc (dump_kind, loc, "node%s\n",
1248 SLP_TREE_DEF_TYPE (node) != vect_internal_def
1249 ? " (external)" : "");
9771b263 1250 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, stmt)
ebfd146a 1251 {
c2a12ca0 1252 dump_printf_loc (dump_kind, loc, "\tstmt %d ", i);
78c60e3d 1253 dump_gimple_stmt (dump_kind, TDF_SLIM, stmt, 0);
ebfd146a 1254 }
9771b263 1255 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
c2a12ca0 1256 vect_print_slp_tree (dump_kind, loc, child);
ebfd146a
IR
1257}
1258
1259
b8698a0f
L
1260/* Mark the tree rooted at NODE with MARK (PURE_SLP or HYBRID).
1261 If MARK is HYBRID, it refers to a specific stmt in NODE (the stmt at index
ff802fa1 1262 J). Otherwise, MARK is PURE_SLP and J is -1, which indicates that all the
ebfd146a
IR
1263 stmts in NODE are to be marked. */
1264
1265static void
1266vect_mark_slp_stmts (slp_tree node, enum slp_vect_type mark, int j)
1267{
1268 int i;
355fe088 1269 gimple *stmt;
d755c7ef 1270 slp_tree child;
ebfd146a 1271
603cca93 1272 if (SLP_TREE_DEF_TYPE (node) != vect_internal_def)
ebfd146a
IR
1273 return;
1274
9771b263 1275 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, stmt)
ebfd146a
IR
1276 if (j < 0 || i == j)
1277 STMT_SLP_TYPE (vinfo_for_stmt (stmt)) = mark;
1278
9771b263 1279 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
d755c7ef 1280 vect_mark_slp_stmts (child, mark, j);
ebfd146a
IR
1281}
1282
1283
a70d6342
IR
1284/* Mark the statements of the tree rooted at NODE as relevant (vect_used). */
1285
1286static void
1287vect_mark_slp_stmts_relevant (slp_tree node)
1288{
1289 int i;
355fe088 1290 gimple *stmt;
a70d6342 1291 stmt_vec_info stmt_info;
d755c7ef 1292 slp_tree child;
a70d6342 1293
603cca93 1294 if (SLP_TREE_DEF_TYPE (node) != vect_internal_def)
a70d6342
IR
1295 return;
1296
9771b263 1297 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, stmt)
a70d6342
IR
1298 {
1299 stmt_info = vinfo_for_stmt (stmt);
b8698a0f 1300 gcc_assert (!STMT_VINFO_RELEVANT (stmt_info)
a70d6342
IR
1301 || STMT_VINFO_RELEVANT (stmt_info) == vect_used_in_scope);
1302 STMT_VINFO_RELEVANT (stmt_info) = vect_used_in_scope;
1303 }
1304
9771b263 1305 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
d755c7ef 1306 vect_mark_slp_stmts_relevant (child);
a70d6342
IR
1307}
1308
1309
b5aeb3bb
IR
1310/* Rearrange the statements of NODE according to PERMUTATION. */
1311
1312static void
1313vect_slp_rearrange_stmts (slp_tree node, unsigned int group_size,
01d8bf07 1314 vec<unsigned> permutation)
b5aeb3bb 1315{
355fe088
TS
1316 gimple *stmt;
1317 vec<gimple *> tmp_stmts;
d755c7ef
RB
1318 unsigned int i;
1319 slp_tree child;
b5aeb3bb 1320
9771b263 1321 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
d755c7ef 1322 vect_slp_rearrange_stmts (child, group_size, permutation);
b5aeb3bb 1323
9771b263
DN
1324 gcc_assert (group_size == SLP_TREE_SCALAR_STMTS (node).length ());
1325 tmp_stmts.create (group_size);
d755c7ef 1326 tmp_stmts.quick_grow_cleared (group_size);
b5aeb3bb 1327
9771b263 1328 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, stmt)
d755c7ef 1329 tmp_stmts[permutation[i]] = stmt;
b5aeb3bb 1330
9771b263 1331 SLP_TREE_SCALAR_STMTS (node).release ();
b5aeb3bb
IR
1332 SLP_TREE_SCALAR_STMTS (node) = tmp_stmts;
1333}
1334
1335
b266b968
RB
1336/* Attempt to reorder stmts in a reduction chain so that we don't
1337 require any load permutation. Return true if that was possible,
1338 otherwise return false. */
1339
1340static bool
1341vect_attempt_slp_rearrange_stmts (slp_instance slp_instn)
1342{
1343 unsigned int group_size = SLP_INSTANCE_GROUP_SIZE (slp_instn);
1344 unsigned int i, j;
b266b968
RB
1345 unsigned int lidx;
1346 slp_tree node, load;
1347
1348 /* Compare all the permutation sequences to the first one. We know
1349 that at least one load is permuted. */
1350 node = SLP_INSTANCE_LOADS (slp_instn)[0];
1351 if (!node->load_permutation.exists ())
1352 return false;
1353 for (i = 1; SLP_INSTANCE_LOADS (slp_instn).iterate (i, &load); ++i)
1354 {
1355 if (!load->load_permutation.exists ())
1356 return false;
1357 FOR_EACH_VEC_ELT (load->load_permutation, j, lidx)
1358 if (lidx != node->load_permutation[j])
1359 return false;
1360 }
1361
1362 /* Check that the loads in the first sequence are different and there
1363 are no gaps between them. */
7ba9e72d 1364 auto_sbitmap load_index (group_size);
b266b968
RB
1365 bitmap_clear (load_index);
1366 FOR_EACH_VEC_ELT (node->load_permutation, i, lidx)
1367 {
41eefe13 1368 if (lidx >= group_size)
7ba9e72d 1369 return false;
b266b968 1370 if (bitmap_bit_p (load_index, lidx))
7ba9e72d
TS
1371 return false;
1372
b266b968
RB
1373 bitmap_set_bit (load_index, lidx);
1374 }
1375 for (i = 0; i < group_size; i++)
1376 if (!bitmap_bit_p (load_index, i))
7ba9e72d 1377 return false;
b266b968
RB
1378
1379 /* This permutation is valid for reduction. Since the order of the
1380 statements in the nodes is not important unless they are memory
1381 accesses, we can rearrange the statements in all the nodes
1382 according to the order of the loads. */
1383 vect_slp_rearrange_stmts (SLP_INSTANCE_TREE (slp_instn), group_size,
1384 node->load_permutation);
1385
1386 /* We are done, no actual permutations need to be generated. */
c4e360f4 1387 unsigned int unrolling_factor = SLP_INSTANCE_UNROLLING_FACTOR (slp_instn);
b266b968 1388 FOR_EACH_VEC_ELT (SLP_INSTANCE_LOADS (slp_instn), i, node)
c4e360f4
RB
1389 {
1390 gimple *first_stmt = SLP_TREE_SCALAR_STMTS (node)[0];
1391 first_stmt = GROUP_FIRST_ELEMENT (vinfo_for_stmt (first_stmt));
1392 /* But we have to keep those permutations that are required because
1393 of handling of gaps. */
1394 if (unrolling_factor == 1
1395 || (group_size == GROUP_SIZE (vinfo_for_stmt (first_stmt))
1396 && GROUP_GAP (vinfo_for_stmt (first_stmt)) == 0))
1397 SLP_TREE_LOAD_PERMUTATION (node).release ();
cbd400b4
RB
1398 else
1399 for (j = 0; j < SLP_TREE_LOAD_PERMUTATION (node).length (); ++j)
1400 SLP_TREE_LOAD_PERMUTATION (node)[j] = j;
c4e360f4
RB
1401 }
1402
b266b968
RB
1403 return true;
1404}
1405
01d8bf07
RB
1406/* Check if the required load permutations in the SLP instance
1407 SLP_INSTN are supported. */
ebfd146a
IR
1408
1409static bool
01d8bf07 1410vect_supported_load_permutation_p (slp_instance slp_instn)
ebfd146a 1411{
01d8bf07
RB
1412 unsigned int group_size = SLP_INSTANCE_GROUP_SIZE (slp_instn);
1413 unsigned int i, j, k, next;
6983e6b5 1414 slp_tree node;
a5b50aa1 1415 gimple *stmt, *load, *next_load;
ebfd146a 1416
73fbfcad 1417 if (dump_enabled_p ())
ebfd146a 1418 {
78c60e3d 1419 dump_printf_loc (MSG_NOTE, vect_location, "Load permutation ");
01d8bf07
RB
1420 FOR_EACH_VEC_ELT (SLP_INSTANCE_LOADS (slp_instn), i, node)
1421 if (node->load_permutation.exists ())
1422 FOR_EACH_VEC_ELT (node->load_permutation, j, next)
1423 dump_printf (MSG_NOTE, "%d ", next);
1424 else
bddc974e
TJ
1425 for (k = 0; k < group_size; ++k)
1426 dump_printf (MSG_NOTE, "%d ", k);
e645e942 1427 dump_printf (MSG_NOTE, "\n");
ebfd146a
IR
1428 }
1429
b5aeb3bb
IR
1430 /* In case of reduction every load permutation is allowed, since the order
1431 of the reduction statements is not important (as opposed to the case of
0d0293ac 1432 grouped stores). The only condition we need to check is that all the
b5aeb3bb
IR
1433 load nodes are of the same size and have the same permutation (and then
1434 rearrange all the nodes of the SLP instance according to this
1435 permutation). */
1436
1437 /* Check that all the load nodes are of the same size. */
01d8bf07 1438 /* ??? Can't we assert this? */
9771b263 1439 FOR_EACH_VEC_ELT (SLP_INSTANCE_LOADS (slp_instn), i, node)
6983e6b5
RB
1440 if (SLP_TREE_SCALAR_STMTS (node).length () != (unsigned) group_size)
1441 return false;
2200fc49 1442
b5aeb3bb 1443 node = SLP_INSTANCE_TREE (slp_instn);
9771b263 1444 stmt = SLP_TREE_SCALAR_STMTS (node)[0];
b5aeb3bb 1445
b010117a 1446 /* Reduction (there are no data-refs in the root).
b266b968 1447 In reduction chain the order of the loads is not important. */
b010117a
IR
1448 if (!STMT_VINFO_DATA_REF (vinfo_for_stmt (stmt))
1449 && !GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt)))
c4e360f4 1450 vect_attempt_slp_rearrange_stmts (slp_instn);
b5aeb3bb 1451
6aa904c4
IR
1452 /* In basic block vectorization we allow any subchain of an interleaving
1453 chain.
1454 FORNOW: not supported in loop SLP because of realignment compications. */
01d8bf07 1455 if (STMT_VINFO_BB_VINFO (vinfo_for_stmt (stmt)))
6aa904c4 1456 {
240a94da
RB
1457 /* Check whether the loads in an instance form a subchain and thus
1458 no permutation is necessary. */
9771b263 1459 FOR_EACH_VEC_ELT (SLP_INSTANCE_LOADS (slp_instn), i, node)
6aa904c4 1460 {
9626d143
RB
1461 if (!SLP_TREE_LOAD_PERMUTATION (node).exists ())
1462 continue;
240a94da 1463 bool subchain_p = true;
6aa904c4 1464 next_load = NULL;
9771b263 1465 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), j, load)
6aa904c4 1466 {
5b5826c4
RB
1467 if (j != 0
1468 && (next_load != load
1469 || GROUP_GAP (vinfo_for_stmt (load)) != 1))
240a94da
RB
1470 {
1471 subchain_p = false;
1472 break;
1473 }
6aa904c4
IR
1474 next_load = GROUP_NEXT_ELEMENT (vinfo_for_stmt (load));
1475 }
240a94da
RB
1476 if (subchain_p)
1477 SLP_TREE_LOAD_PERMUTATION (node).release ();
1478 else
1479 {
fe73a332
RB
1480 stmt_vec_info group_info
1481 = vinfo_for_stmt (SLP_TREE_SCALAR_STMTS (node)[0]);
1482 group_info = vinfo_for_stmt (GROUP_FIRST_ELEMENT (group_info));
1483 unsigned nunits
1484 = TYPE_VECTOR_SUBPARTS (STMT_VINFO_VECTYPE (group_info));
1485 unsigned k, maxk = 0;
1486 FOR_EACH_VEC_ELT (SLP_TREE_LOAD_PERMUTATION (node), j, k)
1487 if (k > maxk)
1488 maxk = k;
1489 /* In BB vectorization we may not actually use a loaded vector
1490 accessing elements in excess of GROUP_SIZE. */
1491 if (maxk >= (GROUP_SIZE (group_info) & ~(nunits - 1)))
1492 {
1493 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
1494 "BB vectorization with gaps at the end of "
1495 "a load is not supported\n");
1496 return false;
1497 }
1498
240a94da
RB
1499 /* Verify the permutation can be generated. */
1500 vec<tree> tem;
29afecdf 1501 unsigned n_perms;
240a94da 1502 if (!vect_transform_slp_perm_load (node, tem, NULL,
29afecdf 1503 1, slp_instn, true, &n_perms))
240a94da
RB
1504 {
1505 dump_printf_loc (MSG_MISSED_OPTIMIZATION,
1506 vect_location,
1507 "unsupported load permutation\n");
1508 return false;
1509 }
1510 }
6aa904c4 1511 }
01d8bf07 1512 return true;
6aa904c4
IR
1513 }
1514
9b999e8c 1515 /* For loop vectorization verify we can generate the permutation. */
29afecdf 1516 unsigned n_perms;
01d8bf07
RB
1517 FOR_EACH_VEC_ELT (SLP_INSTANCE_LOADS (slp_instn), i, node)
1518 if (node->load_permutation.exists ()
1519 && !vect_transform_slp_perm_load
1520 (node, vNULL, NULL,
29afecdf
RB
1521 SLP_INSTANCE_UNROLLING_FACTOR (slp_instn), slp_instn, true,
1522 &n_perms))
01d8bf07 1523 return false;
9b999e8c 1524
01d8bf07 1525 return true;
ebfd146a
IR
1526}
1527
1528
e4a707c4 1529/* Find the last store in SLP INSTANCE. */
ff802fa1 1530
64900538 1531gimple *
2e8ab70c 1532vect_find_last_scalar_stmt_in_slp (slp_tree node)
e4a707c4 1533{
355fe088 1534 gimple *last = NULL, *stmt;
e4a707c4 1535
2e8ab70c
RB
1536 for (int i = 0; SLP_TREE_SCALAR_STMTS (node).iterate (i, &stmt); i++)
1537 {
1538 stmt_vec_info stmt_vinfo = vinfo_for_stmt (stmt);
1539 if (is_pattern_stmt_p (stmt_vinfo))
1540 last = get_later_stmt (STMT_VINFO_RELATED_STMT (stmt_vinfo), last);
1541 else
1542 last = get_later_stmt (stmt, last);
1543 }
e4a707c4 1544
2e8ab70c 1545 return last;
e4a707c4
IR
1546}
1547
23847df4
RB
1548/* Compute the cost for the SLP node NODE in the SLP instance INSTANCE. */
1549
1550static void
1a4b99c1 1551vect_analyze_slp_cost_1 (slp_instance instance, slp_tree node,
23847df4 1552 stmt_vector_for_cost *prologue_cost_vec,
1a4b99c1 1553 stmt_vector_for_cost *body_cost_vec,
23847df4
RB
1554 unsigned ncopies_for_cost)
1555{
603cca93 1556 unsigned i, j;
23847df4 1557 slp_tree child;
8155f4d8 1558 gimple *stmt;
23847df4
RB
1559 stmt_vec_info stmt_info;
1560 tree lhs;
23847df4
RB
1561
1562 /* Recurse down the SLP tree. */
1563 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
603cca93 1564 if (SLP_TREE_DEF_TYPE (child) == vect_internal_def)
1a4b99c1
RB
1565 vect_analyze_slp_cost_1 (instance, child, prologue_cost_vec,
1566 body_cost_vec, ncopies_for_cost);
23847df4
RB
1567
1568 /* Look at the first scalar stmt to determine the cost. */
1569 stmt = SLP_TREE_SCALAR_STMTS (node)[0];
1570 stmt_info = vinfo_for_stmt (stmt);
1571 if (STMT_VINFO_GROUPED_ACCESS (stmt_info))
1572 {
2de001ee
RS
1573 vect_memory_access_type memory_access_type
1574 = (STMT_VINFO_STRIDED_P (stmt_info)
1575 ? VMAT_STRIDED_SLP
1576 : VMAT_CONTIGUOUS);
23847df4 1577 if (DR_IS_WRITE (STMT_VINFO_DATA_REF (stmt_info)))
2de001ee
RS
1578 vect_model_store_cost (stmt_info, ncopies_for_cost,
1579 memory_access_type, vect_uninitialized_def,
23847df4
RB
1580 node, prologue_cost_vec, body_cost_vec);
1581 else
1582 {
23847df4 1583 gcc_checking_assert (DR_IS_READ (STMT_VINFO_DATA_REF (stmt_info)));
52eab378
RB
1584 if (SLP_TREE_LOAD_PERMUTATION (node).exists ())
1585 {
8155f4d8
RB
1586 /* If the load is permuted then the alignment is determined by
1587 the first group element not by the first scalar stmt DR. */
52eab378
RB
1588 stmt = GROUP_FIRST_ELEMENT (stmt_info);
1589 stmt_info = vinfo_for_stmt (stmt);
8155f4d8 1590 /* Record the cost for the permutation. */
29afecdf
RB
1591 unsigned n_perms;
1592 vect_transform_slp_perm_load (node, vNULL, NULL,
1593 ncopies_for_cost, instance, true,
1594 &n_perms);
1595 record_stmt_cost (body_cost_vec, n_perms, vec_perm,
8155f4d8 1596 stmt_info, 0, vect_body);
8155f4d8
RB
1597 unsigned nunits
1598 = TYPE_VECTOR_SUBPARTS (STMT_VINFO_VECTYPE (stmt_info));
29afecdf
RB
1599 /* And adjust the number of loads performed. This handles
1600 redundancies as well as loads that are later dead. */
1601 auto_sbitmap perm (GROUP_SIZE (stmt_info));
1602 bitmap_clear (perm);
1603 for (i = 0; i < SLP_TREE_LOAD_PERMUTATION (node).length (); ++i)
1604 bitmap_set_bit (perm, SLP_TREE_LOAD_PERMUTATION (node)[i]);
1605 ncopies_for_cost = 0;
1606 bool load_seen = false;
1607 for (i = 0; i < GROUP_SIZE (stmt_info); ++i)
1608 {
1609 if (i % nunits == 0)
1610 {
1611 if (load_seen)
1612 ncopies_for_cost++;
1613 load_seen = false;
1614 }
1615 if (bitmap_bit_p (perm, i))
1616 load_seen = true;
1617 }
1618 if (load_seen)
1619 ncopies_for_cost++;
1620 gcc_assert (ncopies_for_cost
1621 <= (GROUP_SIZE (stmt_info) - GROUP_GAP (stmt_info)
1622 + nunits - 1) / nunits);
8155f4d8 1623 ncopies_for_cost *= SLP_INSTANCE_UNROLLING_FACTOR (instance);
52eab378 1624 }
8155f4d8 1625 /* Record the cost for the vector loads. */
2de001ee
RS
1626 vect_model_load_cost (stmt_info, ncopies_for_cost,
1627 memory_access_type, node, prologue_cost_vec,
1628 body_cost_vec);
89483f99 1629 return;
23847df4
RB
1630 }
1631 }
e7baeb39
RB
1632 else if (STMT_VINFO_TYPE (stmt_info) == induc_vec_info_type)
1633 {
1634 /* ncopies_for_cost is the number of IVs we generate. */
1635 record_stmt_cost (body_cost_vec, ncopies_for_cost, vector_stmt,
1636 stmt_info, 0, vect_body);
1637
1638 /* Prologue cost for the initial values and step vector. */
1639 record_stmt_cost (prologue_cost_vec, ncopies_for_cost,
1640 CONSTANT_CLASS_P
1641 (STMT_VINFO_LOOP_PHI_EVOLUTION_BASE_UNCHANGED
1642 (stmt_info))
1643 ? vector_load : vec_construct,
1644 stmt_info, 0, vect_prologue);
1645 record_stmt_cost (prologue_cost_vec, 1,
1646 CONSTANT_CLASS_P
1647 (STMT_VINFO_LOOP_PHI_EVOLUTION_PART (stmt_info))
1648 ? vector_load : vec_construct,
1649 stmt_info, 0, vect_prologue);
1650
1651 /* ??? No easy way to get at the actual number of vector stmts
1652 to be geneated and thus the derived IVs. */
1653 }
89483f99 1654 else
6876e5bc
RB
1655 {
1656 record_stmt_cost (body_cost_vec, ncopies_for_cost, vector_stmt,
1657 stmt_info, 0, vect_body);
89483f99
RB
1658 if (SLP_TREE_TWO_OPERATORS (node))
1659 {
1660 record_stmt_cost (body_cost_vec, ncopies_for_cost, vector_stmt,
1661 stmt_info, 0, vect_body);
1662 record_stmt_cost (body_cost_vec, ncopies_for_cost, vec_perm,
1663 stmt_info, 0, vect_body);
1664 }
6876e5bc 1665 }
23847df4 1666
603cca93
RB
1667 /* Push SLP node def-type to stmts. */
1668 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
1669 if (SLP_TREE_DEF_TYPE (child) != vect_internal_def)
1670 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (child), j, stmt)
1671 STMT_VINFO_DEF_TYPE (vinfo_for_stmt (stmt)) = SLP_TREE_DEF_TYPE (child);
1672
23847df4
RB
1673 /* Scan operands and account for prologue cost of constants/externals.
1674 ??? This over-estimates cost for multiple uses and should be
1675 re-engineered. */
603cca93 1676 stmt = SLP_TREE_SCALAR_STMTS (node)[0];
23847df4
RB
1677 lhs = gimple_get_lhs (stmt);
1678 for (i = 0; i < gimple_num_ops (stmt); ++i)
1679 {
81c40241 1680 tree op = gimple_op (stmt, i);
355fe088 1681 gimple *def_stmt;
23847df4
RB
1682 enum vect_def_type dt;
1683 if (!op || op == lhs)
1684 continue;
81c40241 1685 if (vect_is_simple_use (op, stmt_info->vinfo, &def_stmt, &dt))
2e8ab70c
RB
1686 {
1687 /* Without looking at the actual initializer a vector of
1688 constants can be implemented as load from the constant pool.
1689 ??? We need to pass down stmt_info for a vector type
1690 even if it points to the wrong stmt. */
1691 if (dt == vect_constant_def)
1692 record_stmt_cost (prologue_cost_vec, 1, vector_load,
1693 stmt_info, 0, vect_prologue);
1694 else if (dt == vect_external_def)
1695 record_stmt_cost (prologue_cost_vec, 1, vec_construct,
1696 stmt_info, 0, vect_prologue);
1697 }
23847df4 1698 }
603cca93
RB
1699
1700 /* Restore stmt def-types. */
1701 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
1702 if (SLP_TREE_DEF_TYPE (child) != vect_internal_def)
1703 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (child), j, stmt)
1704 STMT_VINFO_DEF_TYPE (vinfo_for_stmt (stmt)) = vect_internal_def;
23847df4
RB
1705}
1706
1707/* Compute the cost for the SLP instance INSTANCE. */
1708
1709static void
1a4b99c1 1710vect_analyze_slp_cost (slp_instance instance, void *data)
23847df4
RB
1711{
1712 stmt_vector_for_cost body_cost_vec, prologue_cost_vec;
1713 unsigned ncopies_for_cost;
1714 stmt_info_for_cost *si;
1715 unsigned i;
1716
b939ea86
RB
1717 if (dump_enabled_p ())
1718 dump_printf_loc (MSG_NOTE, vect_location,
1719 "=== vect_analyze_slp_cost ===\n");
1720
23847df4
RB
1721 /* Calculate the number of vector stmts to create based on the unrolling
1722 factor (number of vectors is 1 if NUNITS >= GROUP_SIZE, and is
1723 GROUP_SIZE / NUNITS otherwise. */
1724 unsigned group_size = SLP_INSTANCE_GROUP_SIZE (instance);
1a4b99c1
RB
1725 slp_tree node = SLP_INSTANCE_TREE (instance);
1726 stmt_vec_info stmt_info = vinfo_for_stmt (SLP_TREE_SCALAR_STMTS (node)[0]);
1727 /* Adjust the group_size by the vectorization factor which is always one
1728 for basic-block vectorization. */
1729 if (STMT_VINFO_LOOP_VINFO (stmt_info))
1730 group_size *= LOOP_VINFO_VECT_FACTOR (STMT_VINFO_LOOP_VINFO (stmt_info));
1731 unsigned nunits = TYPE_VECTOR_SUBPARTS (STMT_VINFO_VECTYPE (stmt_info));
1732 /* For reductions look at a reduction operand in case the reduction
1733 operation is widening like DOT_PROD or SAD. */
1734 if (!STMT_VINFO_GROUPED_ACCESS (stmt_info))
1735 {
355fe088 1736 gimple *stmt = SLP_TREE_SCALAR_STMTS (node)[0];
1a4b99c1
RB
1737 switch (gimple_assign_rhs_code (stmt))
1738 {
1739 case DOT_PROD_EXPR:
1740 case SAD_EXPR:
1741 nunits = TYPE_VECTOR_SUBPARTS (get_vectype_for_scalar_type
1742 (TREE_TYPE (gimple_assign_rhs1 (stmt))));
1743 break;
1744 default:;
1745 }
1746 }
23847df4
RB
1747 ncopies_for_cost = least_common_multiple (nunits, group_size) / nunits;
1748
1749 prologue_cost_vec.create (10);
1750 body_cost_vec.create (10);
1a4b99c1
RB
1751 vect_analyze_slp_cost_1 (instance, SLP_INSTANCE_TREE (instance),
1752 &prologue_cost_vec, &body_cost_vec,
1753 ncopies_for_cost);
23847df4
RB
1754
1755 /* Record the prologue costs, which were delayed until we were
1a4b99c1 1756 sure that SLP was successful. */
23847df4
RB
1757 FOR_EACH_VEC_ELT (prologue_cost_vec, i, si)
1758 {
1759 struct _stmt_vec_info *stmt_info
1760 = si->stmt ? vinfo_for_stmt (si->stmt) : NULL;
1761 (void) add_stmt_cost (data, si->count, si->kind, stmt_info,
1762 si->misalign, vect_prologue);
1763 }
1764
1a4b99c1
RB
1765 /* Record the instance's instructions in the target cost model. */
1766 FOR_EACH_VEC_ELT (body_cost_vec, i, si)
1767 {
1768 struct _stmt_vec_info *stmt_info
1769 = si->stmt ? vinfo_for_stmt (si->stmt) : NULL;
1770 (void) add_stmt_cost (data, si->count, si->kind, stmt_info,
1771 si->misalign, vect_body);
1772 }
1773
23847df4 1774 prologue_cost_vec.release ();
1a4b99c1 1775 body_cost_vec.release ();
23847df4 1776}
e4a707c4 1777
1ba91a49
AL
1778/* Splits a group of stores, currently beginning at FIRST_STMT, into two groups:
1779 one (still beginning at FIRST_STMT) of size GROUP1_SIZE (also containing
1780 the first GROUP1_SIZE stmts, since stores are consecutive), the second
1781 containing the remainder.
1782 Return the first stmt in the second group. */
1783
1784static gimple *
1785vect_split_slp_store_group (gimple *first_stmt, unsigned group1_size)
1786{
1787 stmt_vec_info first_vinfo = vinfo_for_stmt (first_stmt);
1788 gcc_assert (GROUP_FIRST_ELEMENT (first_vinfo) == first_stmt);
1789 gcc_assert (group1_size > 0);
1790 int group2_size = GROUP_SIZE (first_vinfo) - group1_size;
1791 gcc_assert (group2_size > 0);
1792 GROUP_SIZE (first_vinfo) = group1_size;
1793
1794 gimple *stmt = first_stmt;
1795 for (unsigned i = group1_size; i > 1; i--)
1796 {
1797 stmt = GROUP_NEXT_ELEMENT (vinfo_for_stmt (stmt));
1798 gcc_assert (GROUP_GAP (vinfo_for_stmt (stmt)) == 1);
1799 }
1800 /* STMT is now the last element of the first group. */
1801 gimple *group2 = GROUP_NEXT_ELEMENT (vinfo_for_stmt (stmt));
1802 GROUP_NEXT_ELEMENT (vinfo_for_stmt (stmt)) = 0;
1803
1804 GROUP_SIZE (vinfo_for_stmt (group2)) = group2_size;
1805 for (stmt = group2; stmt; stmt = GROUP_NEXT_ELEMENT (vinfo_for_stmt (stmt)))
1806 {
1807 GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt)) = group2;
1808 gcc_assert (GROUP_GAP (vinfo_for_stmt (stmt)) == 1);
1809 }
1810
1811 /* For the second group, the GROUP_GAP is that before the original group,
1812 plus skipping over the first vector. */
1813 GROUP_GAP (vinfo_for_stmt (group2)) =
1814 GROUP_GAP (first_vinfo) + group1_size;
1815
1816 /* GROUP_GAP of the first group now has to skip over the second group too. */
1817 GROUP_GAP (first_vinfo) += group2_size;
1818
1819 if (dump_enabled_p ())
1820 dump_printf_loc (MSG_NOTE, vect_location, "Split group into %d and %d\n",
1821 group1_size, group2_size);
1822
1823 return group2;
1824}
1825
0d0293ac 1826/* Analyze an SLP instance starting from a group of grouped stores. Call
b8698a0f 1827 vect_build_slp_tree to build a tree of packed stmts if possible.
ebfd146a
IR
1828 Return FALSE if it's impossible to SLP any stmt in the loop. */
1829
1830static bool
310213d4 1831vect_analyze_slp_instance (vec_info *vinfo,
355fe088 1832 gimple *stmt, unsigned max_tree_size)
ebfd146a
IR
1833{
1834 slp_instance new_instance;
d092494c 1835 slp_tree node;
e14c1050 1836 unsigned int group_size = GROUP_SIZE (vinfo_for_stmt (stmt));
ebfd146a 1837 unsigned int unrolling_factor = 1, nunits;
b5aeb3bb 1838 tree vectype, scalar_type = NULL_TREE;
355fe088 1839 gimple *next;
1ba91a49 1840 unsigned int i;
ebfd146a 1841 unsigned int max_nunits = 0;
9771b263 1842 vec<slp_tree> loads;
b5aeb3bb 1843 struct data_reference *dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (stmt));
355fe088 1844 vec<gimple *> scalar_stmts;
b5aeb3bb 1845
b010117a 1846 if (GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt)))
b5aeb3bb 1847 {
b010117a
IR
1848 if (dr)
1849 {
1850 scalar_type = TREE_TYPE (DR_REF (dr));
1851 vectype = get_vectype_for_scalar_type (scalar_type);
1852 }
1853 else
1854 {
310213d4 1855 gcc_assert (is_a <loop_vec_info> (vinfo));
b010117a
IR
1856 vectype = STMT_VINFO_VECTYPE (vinfo_for_stmt (stmt));
1857 }
1858
e14c1050 1859 group_size = GROUP_SIZE (vinfo_for_stmt (stmt));
b5aeb3bb
IR
1860 }
1861 else
1862 {
310213d4 1863 gcc_assert (is_a <loop_vec_info> (vinfo));
b5aeb3bb 1864 vectype = STMT_VINFO_VECTYPE (vinfo_for_stmt (stmt));
310213d4 1865 group_size = as_a <loop_vec_info> (vinfo)->reductions.length ();
b5aeb3bb 1866 }
b8698a0f 1867
ebfd146a
IR
1868 if (!vectype)
1869 {
73fbfcad 1870 if (dump_enabled_p ())
ebfd146a 1871 {
78c60e3d
SS
1872 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
1873 "Build SLP failed: unsupported data-type ");
1874 dump_generic_expr (MSG_MISSED_OPTIMIZATION, TDF_SLIM, scalar_type);
e645e942 1875 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
ebfd146a 1876 }
b5aeb3bb 1877
ebfd146a
IR
1878 return false;
1879 }
ebfd146a 1880 nunits = TYPE_VECTOR_SUBPARTS (vectype);
a70d6342 1881
0d0293ac 1882 /* Create a node (a root of the SLP tree) for the packed grouped stores. */
9771b263 1883 scalar_stmts.create (group_size);
ebfd146a 1884 next = stmt;
b010117a 1885 if (GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt)))
ebfd146a 1886 {
b5aeb3bb
IR
1887 /* Collect the stores and store them in SLP_TREE_SCALAR_STMTS. */
1888 while (next)
1889 {
f7e531cf
IR
1890 if (STMT_VINFO_IN_PATTERN_P (vinfo_for_stmt (next))
1891 && STMT_VINFO_RELATED_STMT (vinfo_for_stmt (next)))
9771b263
DN
1892 scalar_stmts.safe_push (
1893 STMT_VINFO_RELATED_STMT (vinfo_for_stmt (next)));
f7e531cf 1894 else
9771b263 1895 scalar_stmts.safe_push (next);
e14c1050 1896 next = GROUP_NEXT_ELEMENT (vinfo_for_stmt (next));
b5aeb3bb 1897 }
14a61437
RB
1898 /* Mark the first element of the reduction chain as reduction to properly
1899 transform the node. In the reduction analysis phase only the last
1900 element of the chain is marked as reduction. */
1901 if (!STMT_VINFO_GROUPED_ACCESS (vinfo_for_stmt (stmt)))
1902 STMT_VINFO_DEF_TYPE (vinfo_for_stmt (stmt)) = vect_reduction_def;
b5aeb3bb
IR
1903 }
1904 else
1905 {
1906 /* Collect reduction statements. */
310213d4 1907 vec<gimple *> reductions = as_a <loop_vec_info> (vinfo)->reductions;
9771b263
DN
1908 for (i = 0; reductions.iterate (i, &next); i++)
1909 scalar_stmts.safe_push (next);
ebfd146a
IR
1910 }
1911
9771b263 1912 loads.create (group_size);
ebfd146a
IR
1913
1914 /* Build the tree for the SLP instance. */
89d390e5
RB
1915 bool *matches = XALLOCAVEC (bool, group_size);
1916 unsigned npermutes = 0;
e569db5f 1917 node = vect_build_slp_tree (vinfo, scalar_stmts, group_size,
e403d17e 1918 &max_nunits, &loads, matches, &npermutes,
e569db5f
VK
1919 NULL, max_tree_size);
1920 if (node != NULL)
ebfd146a 1921 {
4ef69dfc 1922 /* Calculate the unrolling factor based on the smallest type. */
e569db5f
VK
1923 unrolling_factor
1924 = least_common_multiple (max_nunits, group_size) / group_size;
b8698a0f 1925
e569db5f
VK
1926 if (unrolling_factor != 1
1927 && is_a <bb_vec_info> (vinfo))
1928 {
1929
1930 if (max_nunits > group_size)
4ef69dfc 1931 {
e645e942 1932 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
e569db5f
VK
1933 "Build SLP failed: store group "
1934 "size not a multiple of the vector size "
1935 "in basic block SLP\n");
c7e62a26 1936 vect_free_slp_tree (node);
9771b263 1937 loads.release ();
4ef69dfc
IR
1938 return false;
1939 }
e569db5f
VK
1940 /* Fatal mismatch. */
1941 matches[group_size/max_nunits * max_nunits] = false;
1942 vect_free_slp_tree (node);
1943 loads.release ();
1944 }
1945 else
1946 {
4ef69dfc
IR
1947 /* Create a new SLP instance. */
1948 new_instance = XNEW (struct _slp_instance);
1949 SLP_INSTANCE_TREE (new_instance) = node;
1950 SLP_INSTANCE_GROUP_SIZE (new_instance) = group_size;
ebfd146a 1951 SLP_INSTANCE_UNROLLING_FACTOR (new_instance) = unrolling_factor;
ebfd146a 1952 SLP_INSTANCE_LOADS (new_instance) = loads;
abf9bfbc
RB
1953
1954 /* Compute the load permutation. */
1955 slp_tree load_node;
1956 bool loads_permuted = false;
abf9bfbc
RB
1957 FOR_EACH_VEC_ELT (loads, i, load_node)
1958 {
01d8bf07 1959 vec<unsigned> load_permutation;
abf9bfbc 1960 int j;
355fe088 1961 gimple *load, *first_stmt;
01d8bf07
RB
1962 bool this_load_permuted = false;
1963 load_permutation.create (group_size);
6983e6b5
RB
1964 first_stmt = GROUP_FIRST_ELEMENT
1965 (vinfo_for_stmt (SLP_TREE_SCALAR_STMTS (load_node)[0]));
abf9bfbc
RB
1966 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (load_node), j, load)
1967 {
e569db5f
VK
1968 int load_place = vect_get_place_in_interleaving_chain
1969 (load, first_stmt);
6983e6b5
RB
1970 gcc_assert (load_place != -1);
1971 if (load_place != j)
01d8bf07 1972 this_load_permuted = true;
abf9bfbc
RB
1973 load_permutation.safe_push (load_place);
1974 }
fe2bef71
RB
1975 if (!this_load_permuted
1976 /* The load requires permutation when unrolling exposes
1977 a gap either because the group is larger than the SLP
1978 group-size or because there is a gap between the groups. */
1979 && (unrolling_factor == 1
1980 || (group_size == GROUP_SIZE (vinfo_for_stmt (first_stmt))
1981 && GROUP_GAP (vinfo_for_stmt (first_stmt)) == 0)))
01d8bf07
RB
1982 {
1983 load_permutation.release ();
1984 continue;
1985 }
1986 SLP_TREE_LOAD_PERMUTATION (load_node) = load_permutation;
1987 loads_permuted = true;
abf9bfbc 1988 }
6aa904c4
IR
1989
1990 if (loads_permuted)
ebfd146a 1991 {
01d8bf07 1992 if (!vect_supported_load_permutation_p (new_instance))
ebfd146a 1993 {
73fbfcad 1994 if (dump_enabled_p ())
ebfd146a 1995 {
e645e942 1996 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
78c60e3d
SS
1997 "Build SLP failed: unsupported load "
1998 "permutation ");
e569db5f
VK
1999 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION,
2000 TDF_SLIM, stmt, 0);
ebfd146a 2001 }
ebfd146a
IR
2002 vect_free_slp_instance (new_instance);
2003 return false;
2004 }
ebfd146a 2005 }
ebfd146a 2006
e569db5f 2007 /* If the loads and stores can be handled with load/store-lan
bb0f5ca7
AL
2008 instructions do not generate this SLP instance. */
2009 if (is_a <loop_vec_info> (vinfo)
2010 && loads_permuted
2011 && dr && vect_store_lanes_supported (vectype, group_size))
2012 {
2013 slp_tree load_node;
2014 FOR_EACH_VEC_ELT (loads, i, load_node)
2015 {
2016 gimple *first_stmt = GROUP_FIRST_ELEMENT
2017 (vinfo_for_stmt (SLP_TREE_SCALAR_STMTS (load_node)[0]));
2018 stmt_vec_info stmt_vinfo = vinfo_for_stmt (first_stmt);
e569db5f
VK
2019 /* Use SLP for strided accesses (or if we
2020 can't load-lanes). */
bb0f5ca7
AL
2021 if (STMT_VINFO_STRIDED_P (stmt_vinfo)
2022 || ! vect_load_lanes_supported
2023 (STMT_VINFO_VECTYPE (stmt_vinfo),
2024 GROUP_SIZE (stmt_vinfo)))
2025 break;
2026 }
2027 if (i == loads.length ())
2028 {
2029 if (dump_enabled_p ())
2030 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2031 "Built SLP cancelled: can use "
2032 "load/store-lanes\n");
2033 vect_free_slp_instance (new_instance);
2034 return false;
2035 }
2036 }
2037
310213d4 2038 vinfo->slp_instances.safe_push (new_instance);
b8698a0f 2039
73fbfcad 2040 if (dump_enabled_p ())
c2a12ca0
RB
2041 {
2042 dump_printf_loc (MSG_NOTE, vect_location,
2043 "Final SLP tree for instance:\n");
2044 vect_print_slp_tree (MSG_NOTE, vect_location, node);
2045 }
ebfd146a
IR
2046
2047 return true;
2048 }
e569db5f
VK
2049 }
2050 else
2051 {
ebfd146a
IR
2052 /* Failed to SLP. */
2053 /* Free the allocated memory. */
e403d17e 2054 scalar_stmts.release ();
9771b263 2055 loads.release ();
e569db5f 2056 }
b8698a0f 2057
1ba91a49 2058 /* For basic block SLP, try to break the group up into multiples of the
97a1a642 2059 vector size. */
1ba91a49
AL
2060 if (is_a <bb_vec_info> (vinfo)
2061 && GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt))
2062 && STMT_VINFO_GROUPED_ACCESS (vinfo_for_stmt (stmt)))
2063 {
2064 /* We consider breaking the group only on VF boundaries from the existing
2065 start. */
2066 for (i = 0; i < group_size; i++)
2067 if (!matches[i]) break;
2068
97a1a642 2069 if (i >= nunits && i < group_size)
1ba91a49
AL
2070 {
2071 /* Split into two groups at the first vector boundary before i. */
97a1a642
RB
2072 gcc_assert ((nunits & (nunits - 1)) == 0);
2073 unsigned group1_size = i & ~(nunits - 1);
1ba91a49
AL
2074
2075 gimple *rest = vect_split_slp_store_group (stmt, group1_size);
2076 bool res = vect_analyze_slp_instance (vinfo, stmt, max_tree_size);
2077 /* If the first non-match was in the middle of a vector,
2078 skip the rest of that vector. */
2079 if (group1_size < i)
2080 {
97a1a642 2081 i = group1_size + nunits;
1ba91a49 2082 if (i < group_size)
97a1a642 2083 rest = vect_split_slp_store_group (rest, nunits);
1ba91a49
AL
2084 }
2085 if (i < group_size)
2086 res |= vect_analyze_slp_instance (vinfo, rest, max_tree_size);
2087 return res;
2088 }
2089 /* Even though the first vector did not all match, we might be able to SLP
2090 (some) of the remainder. FORNOW ignore this possibility. */
2091 }
2092
a70d6342 2093 return false;
ebfd146a
IR
2094}
2095
2096
ff802fa1 2097/* Check if there are stmts in the loop can be vectorized using SLP. Build SLP
ebfd146a
IR
2098 trees of packed scalar stmts if SLP is possible. */
2099
2100bool
310213d4 2101vect_analyze_slp (vec_info *vinfo, unsigned max_tree_size)
ebfd146a
IR
2102{
2103 unsigned int i;
355fe088 2104 gimple *first_element;
a70d6342 2105 bool ok = false;
ebfd146a 2106
73fbfcad 2107 if (dump_enabled_p ())
e645e942 2108 dump_printf_loc (MSG_NOTE, vect_location, "=== vect_analyze_slp ===\n");
ebfd146a 2109
0d0293ac 2110 /* Find SLP sequences starting from groups of grouped stores. */
310213d4
RB
2111 FOR_EACH_VEC_ELT (vinfo->grouped_stores, i, first_element)
2112 if (vect_analyze_slp_instance (vinfo, first_element, max_tree_size))
a70d6342 2113 ok = true;
ebfd146a 2114
310213d4 2115 if (loop_vec_info loop_vinfo = dyn_cast <loop_vec_info> (vinfo))
b010117a 2116 {
310213d4
RB
2117 if (loop_vinfo->reduction_chains.length () > 0)
2118 {
2119 /* Find SLP sequences starting from reduction chains. */
2120 FOR_EACH_VEC_ELT (loop_vinfo->reduction_chains, i, first_element)
2121 if (vect_analyze_slp_instance (vinfo, first_element,
2122 max_tree_size))
2123 ok = true;
2124 else
2125 return false;
b010117a 2126
310213d4
RB
2127 /* Don't try to vectorize SLP reductions if reduction chain was
2128 detected. */
2129 return ok;
2130 }
b010117a 2131
310213d4
RB
2132 /* Find SLP sequences starting from groups of reductions. */
2133 if (loop_vinfo->reductions.length () > 1
2134 && vect_analyze_slp_instance (vinfo, loop_vinfo->reductions[0],
2135 max_tree_size))
2136 ok = true;
2137 }
b5aeb3bb 2138
ebfd146a
IR
2139 return true;
2140}
2141
2142
2143/* For each possible SLP instance decide whether to SLP it and calculate overall
437f4a00
IR
2144 unrolling factor needed to SLP the loop. Return TRUE if decided to SLP at
2145 least one instance. */
ebfd146a 2146
437f4a00 2147bool
ebfd146a
IR
2148vect_make_slp_decision (loop_vec_info loop_vinfo)
2149{
2150 unsigned int i, unrolling_factor = 1;
9771b263 2151 vec<slp_instance> slp_instances = LOOP_VINFO_SLP_INSTANCES (loop_vinfo);
ebfd146a
IR
2152 slp_instance instance;
2153 int decided_to_slp = 0;
2154
73fbfcad 2155 if (dump_enabled_p ())
e645e942
TJ
2156 dump_printf_loc (MSG_NOTE, vect_location, "=== vect_make_slp_decision ==="
2157 "\n");
ebfd146a 2158
9771b263 2159 FOR_EACH_VEC_ELT (slp_instances, i, instance)
ebfd146a
IR
2160 {
2161 /* FORNOW: SLP if you can. */
2162 if (unrolling_factor < SLP_INSTANCE_UNROLLING_FACTOR (instance))
2163 unrolling_factor = SLP_INSTANCE_UNROLLING_FACTOR (instance);
2164
ff802fa1 2165 /* Mark all the stmts that belong to INSTANCE as PURE_SLP stmts. Later we
b8698a0f 2166 call vect_detect_hybrid_slp () to find stmts that need hybrid SLP and
ff802fa1 2167 loop-based vectorization. Such stmts will be marked as HYBRID. */
ebfd146a
IR
2168 vect_mark_slp_stmts (SLP_INSTANCE_TREE (instance), pure_slp, -1);
2169 decided_to_slp++;
2170 }
2171
2172 LOOP_VINFO_SLP_UNROLLING_FACTOR (loop_vinfo) = unrolling_factor;
2173
73fbfcad 2174 if (decided_to_slp && dump_enabled_p ())
ccb3ad87 2175 dump_printf_loc (MSG_NOTE, vect_location,
e645e942 2176 "Decided to SLP %d instances. Unrolling factor %d\n",
78c60e3d 2177 decided_to_slp, unrolling_factor);
437f4a00
IR
2178
2179 return (decided_to_slp > 0);
ebfd146a
IR
2180}
2181
2182
2183/* Find stmts that must be both vectorized and SLPed (since they feed stmts that
ff802fa1 2184 can't be SLPed) in the tree rooted at NODE. Mark such stmts as HYBRID. */
ebfd146a
IR
2185
2186static void
642fce57 2187vect_detect_hybrid_slp_stmts (slp_tree node, unsigned i, slp_vect_type stype)
ebfd146a 2188{
355fe088 2189 gimple *stmt = SLP_TREE_SCALAR_STMTS (node)[i];
ebfd146a 2190 imm_use_iterator imm_iter;
355fe088 2191 gimple *use_stmt;
642fce57 2192 stmt_vec_info use_vinfo, stmt_vinfo = vinfo_for_stmt (stmt);
d755c7ef 2193 slp_tree child;
f2c74cc4 2194 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_vinfo);
642fce57
RB
2195 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
2196 int j;
2197
2198 /* Propagate hybrid down the SLP tree. */
2199 if (stype == hybrid)
2200 ;
2201 else if (HYBRID_SLP_STMT (stmt_vinfo))
2202 stype = hybrid;
2203 else
2204 {
2205 /* Check if a pure SLP stmt has uses in non-SLP stmts. */
2206 gcc_checking_assert (PURE_SLP_STMT (stmt_vinfo));
2935d994
RB
2207 /* If we get a pattern stmt here we have to use the LHS of the
2208 original stmt for immediate uses. */
2209 if (! STMT_VINFO_IN_PATTERN_P (stmt_vinfo)
2210 && STMT_VINFO_RELATED_STMT (stmt_vinfo))
29764870 2211 stmt = STMT_VINFO_RELATED_STMT (stmt_vinfo);
e7baeb39
RB
2212 tree def;
2213 if (gimple_code (stmt) == GIMPLE_PHI)
2214 def = gimple_phi_result (stmt);
2215 else
2216 def = SINGLE_SSA_TREE_OPERAND (stmt, SSA_OP_DEF);
2217 if (def)
2218 FOR_EACH_IMM_USE_STMT (use_stmt, imm_iter, def)
29764870
RB
2219 {
2220 if (!flow_bb_inside_loop_p (loop, gimple_bb (use_stmt)))
2221 continue;
2222 use_vinfo = vinfo_for_stmt (use_stmt);
2223 if (STMT_VINFO_IN_PATTERN_P (use_vinfo)
2224 && STMT_VINFO_RELATED_STMT (use_vinfo))
2225 use_vinfo = vinfo_for_stmt (STMT_VINFO_RELATED_STMT (use_vinfo));
2226 if (!STMT_SLP_TYPE (use_vinfo)
2227 && (STMT_VINFO_RELEVANT (use_vinfo)
2228 || VECTORIZABLE_CYCLE_DEF (STMT_VINFO_DEF_TYPE (use_vinfo)))
2229 && !(gimple_code (use_stmt) == GIMPLE_PHI
2230 && STMT_VINFO_DEF_TYPE (use_vinfo) == vect_reduction_def))
502f0263
RB
2231 {
2232 if (dump_enabled_p ())
2233 {
2234 dump_printf_loc (MSG_NOTE, vect_location, "use of SLP "
2235 "def in non-SLP stmt: ");
2236 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, use_stmt, 0);
2237 }
2238 stype = hybrid;
2239 }
29764870 2240 }
642fce57 2241 }
ebfd146a 2242
502f0263
RB
2243 if (stype == hybrid
2244 && !HYBRID_SLP_STMT (stmt_vinfo))
b1af7da6
RB
2245 {
2246 if (dump_enabled_p ())
2247 {
2248 dump_printf_loc (MSG_NOTE, vect_location, "marking hybrid: ");
2249 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
2250 }
2251 STMT_SLP_TYPE (stmt_vinfo) = hybrid;
2252 }
ebfd146a 2253
642fce57 2254 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), j, child)
603cca93 2255 if (SLP_TREE_DEF_TYPE (child) != vect_external_def)
90dd6e3d 2256 vect_detect_hybrid_slp_stmts (child, i, stype);
642fce57 2257}
f2c74cc4 2258
642fce57 2259/* Helpers for vect_detect_hybrid_slp walking pattern stmt uses. */
ebfd146a 2260
642fce57
RB
2261static tree
2262vect_detect_hybrid_slp_1 (tree *tp, int *, void *data)
2263{
2264 walk_stmt_info *wi = (walk_stmt_info *)data;
2265 struct loop *loopp = (struct loop *)wi->info;
2266
2267 if (wi->is_lhs)
2268 return NULL_TREE;
2269
2270 if (TREE_CODE (*tp) == SSA_NAME
2271 && !SSA_NAME_IS_DEFAULT_DEF (*tp))
2272 {
355fe088 2273 gimple *def_stmt = SSA_NAME_DEF_STMT (*tp);
642fce57
RB
2274 if (flow_bb_inside_loop_p (loopp, gimple_bb (def_stmt))
2275 && PURE_SLP_STMT (vinfo_for_stmt (def_stmt)))
b1af7da6
RB
2276 {
2277 if (dump_enabled_p ())
2278 {
2279 dump_printf_loc (MSG_NOTE, vect_location, "marking hybrid: ");
2280 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, def_stmt, 0);
2281 }
2282 STMT_SLP_TYPE (vinfo_for_stmt (def_stmt)) = hybrid;
2283 }
642fce57
RB
2284 }
2285
2286 return NULL_TREE;
ebfd146a
IR
2287}
2288
642fce57
RB
2289static tree
2290vect_detect_hybrid_slp_2 (gimple_stmt_iterator *gsi, bool *handled,
2291 walk_stmt_info *)
2292{
9186a9d3 2293 stmt_vec_info use_vinfo = vinfo_for_stmt (gsi_stmt (*gsi));
642fce57
RB
2294 /* If the stmt is in a SLP instance then this isn't a reason
2295 to mark use definitions in other SLP instances as hybrid. */
9186a9d3
RB
2296 if (! STMT_SLP_TYPE (use_vinfo)
2297 && (STMT_VINFO_RELEVANT (use_vinfo)
2298 || VECTORIZABLE_CYCLE_DEF (STMT_VINFO_DEF_TYPE (use_vinfo)))
2299 && ! (gimple_code (gsi_stmt (*gsi)) == GIMPLE_PHI
2300 && STMT_VINFO_DEF_TYPE (use_vinfo) == vect_reduction_def))
2301 ;
2302 else
642fce57
RB
2303 *handled = true;
2304 return NULL_TREE;
2305}
ebfd146a
IR
2306
2307/* Find stmts that must be both vectorized and SLPed. */
2308
2309void
2310vect_detect_hybrid_slp (loop_vec_info loop_vinfo)
2311{
2312 unsigned int i;
9771b263 2313 vec<slp_instance> slp_instances = LOOP_VINFO_SLP_INSTANCES (loop_vinfo);
ebfd146a
IR
2314 slp_instance instance;
2315
73fbfcad 2316 if (dump_enabled_p ())
e645e942
TJ
2317 dump_printf_loc (MSG_NOTE, vect_location, "=== vect_detect_hybrid_slp ==="
2318 "\n");
ebfd146a 2319
642fce57
RB
2320 /* First walk all pattern stmt in the loop and mark defs of uses as
2321 hybrid because immediate uses in them are not recorded. */
2322 for (i = 0; i < LOOP_VINFO_LOOP (loop_vinfo)->num_nodes; ++i)
2323 {
2324 basic_block bb = LOOP_VINFO_BBS (loop_vinfo)[i];
2325 for (gimple_stmt_iterator gsi = gsi_start_bb (bb); !gsi_end_p (gsi);
2326 gsi_next (&gsi))
2327 {
355fe088 2328 gimple *stmt = gsi_stmt (gsi);
642fce57
RB
2329 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
2330 if (STMT_VINFO_IN_PATTERN_P (stmt_info))
2331 {
2332 walk_stmt_info wi;
2333 memset (&wi, 0, sizeof (wi));
2334 wi.info = LOOP_VINFO_LOOP (loop_vinfo);
2335 gimple_stmt_iterator gsi2
2336 = gsi_for_stmt (STMT_VINFO_RELATED_STMT (stmt_info));
2337 walk_gimple_stmt (&gsi2, vect_detect_hybrid_slp_2,
2338 vect_detect_hybrid_slp_1, &wi);
2339 walk_gimple_seq (STMT_VINFO_PATTERN_DEF_SEQ (stmt_info),
2340 vect_detect_hybrid_slp_2,
2341 vect_detect_hybrid_slp_1, &wi);
2342 }
2343 }
2344 }
2345
2346 /* Then walk the SLP instance trees marking stmts with uses in
2347 non-SLP stmts as hybrid, also propagating hybrid down the
2348 SLP tree, collecting the above info on-the-fly. */
9771b263 2349 FOR_EACH_VEC_ELT (slp_instances, i, instance)
642fce57
RB
2350 {
2351 for (unsigned i = 0; i < SLP_INSTANCE_GROUP_SIZE (instance); ++i)
2352 vect_detect_hybrid_slp_stmts (SLP_INSTANCE_TREE (instance),
2353 i, pure_slp);
2354 }
ebfd146a
IR
2355}
2356
a70d6342
IR
2357
2358/* Create and initialize a new bb_vec_info struct for BB, as well as
2359 stmt_vec_info structs for all the stmts in it. */
b8698a0f 2360
a70d6342 2361static bb_vec_info
61d371eb
RB
2362new_bb_vec_info (gimple_stmt_iterator region_begin,
2363 gimple_stmt_iterator region_end)
a70d6342 2364{
61d371eb 2365 basic_block bb = gsi_bb (region_begin);
a70d6342
IR
2366 bb_vec_info res = NULL;
2367 gimple_stmt_iterator gsi;
2368
2369 res = (bb_vec_info) xcalloc (1, sizeof (struct _bb_vec_info));
310213d4 2370 res->kind = vec_info::bb;
a70d6342 2371 BB_VINFO_BB (res) = bb;
61d371eb
RB
2372 res->region_begin = region_begin;
2373 res->region_end = region_end;
a70d6342 2374
61d371eb
RB
2375 for (gsi = region_begin; gsi_stmt (gsi) != gsi_stmt (region_end);
2376 gsi_next (&gsi))
a70d6342 2377 {
355fe088 2378 gimple *stmt = gsi_stmt (gsi);
a70d6342 2379 gimple_set_uid (stmt, 0);
310213d4 2380 set_vinfo_for_stmt (stmt, new_stmt_vec_info (stmt, res));
a70d6342
IR
2381 }
2382
9771b263
DN
2383 BB_VINFO_GROUPED_STORES (res).create (10);
2384 BB_VINFO_SLP_INSTANCES (res).create (2);
c3e7ee41 2385 BB_VINFO_TARGET_COST_DATA (res) = init_cost (NULL);
a70d6342
IR
2386
2387 bb->aux = res;
2388 return res;
2389}
2390
2391
2392/* Free BB_VINFO struct, as well as all the stmt_vec_info structs of all the
2393 stmts in the basic block. */
2394
2395static void
2396destroy_bb_vec_info (bb_vec_info bb_vinfo)
2397{
c7e62a26 2398 slp_instance instance;
c7e62a26 2399 unsigned i;
a70d6342
IR
2400
2401 if (!bb_vinfo)
2402 return;
2403
78810bd3
RB
2404 vect_destroy_datarefs (bb_vinfo);
2405 free_dependence_relations (BB_VINFO_DDRS (bb_vinfo));
2406 BB_VINFO_GROUPED_STORES (bb_vinfo).release ();
2407 FOR_EACH_VEC_ELT (BB_VINFO_SLP_INSTANCES (bb_vinfo), i, instance)
2408 vect_free_slp_instance (instance);
2409 BB_VINFO_SLP_INSTANCES (bb_vinfo).release ();
2410 destroy_cost_data (BB_VINFO_TARGET_COST_DATA (bb_vinfo));
a70d6342 2411
78810bd3 2412 for (gimple_stmt_iterator si = bb_vinfo->region_begin;
61d371eb 2413 gsi_stmt (si) != gsi_stmt (bb_vinfo->region_end); gsi_next (&si))
a70d6342 2414 {
355fe088 2415 gimple *stmt = gsi_stmt (si);
a70d6342
IR
2416 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
2417
2418 if (stmt_info)
2419 /* Free stmt_vec_info. */
2420 free_stmt_vec_info (stmt);
61d371eb
RB
2421
2422 /* Reset region marker. */
2423 gimple_set_uid (stmt, -1);
a70d6342
IR
2424 }
2425
78810bd3 2426 BB_VINFO_BB (bb_vinfo)->aux = NULL;
a70d6342 2427 free (bb_vinfo);
a70d6342
IR
2428}
2429
2430
2431/* Analyze statements contained in SLP tree node after recursively analyzing
2432 the subtree. Return TRUE if the operations are supported. */
2433
2434static bool
a12e42fc 2435vect_slp_analyze_node_operations (slp_tree node)
a70d6342
IR
2436{
2437 bool dummy;
603cca93 2438 int i, j;
355fe088 2439 gimple *stmt;
d755c7ef 2440 slp_tree child;
a70d6342 2441
603cca93 2442 if (SLP_TREE_DEF_TYPE (node) != vect_internal_def)
a70d6342
IR
2443 return true;
2444
9771b263 2445 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
a12e42fc 2446 if (!vect_slp_analyze_node_operations (child))
d092494c 2447 return false;
a70d6342 2448
603cca93 2449 bool res = true;
9771b263 2450 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, stmt)
a70d6342
IR
2451 {
2452 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
2453 gcc_assert (stmt_info);
a12e42fc 2454 gcc_assert (STMT_SLP_TYPE (stmt_info) != loop_vect);
a70d6342 2455
6379dfb5
RB
2456 /* Push SLP node def-type to stmt operands. */
2457 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), j, child)
2458 if (SLP_TREE_DEF_TYPE (child) != vect_internal_def)
2459 STMT_VINFO_DEF_TYPE (vinfo_for_stmt (SLP_TREE_SCALAR_STMTS (child)[i]))
2460 = SLP_TREE_DEF_TYPE (child);
2461 res = vect_analyze_stmt (stmt, &dummy, node);
2462 /* Restore def-types. */
2463 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), j, child)
2464 if (SLP_TREE_DEF_TYPE (child) != vect_internal_def)
2465 STMT_VINFO_DEF_TYPE (vinfo_for_stmt (SLP_TREE_SCALAR_STMTS (child)[i]))
2466 = vect_internal_def;
2467 if (! res)
2468 break;
a70d6342
IR
2469 }
2470
603cca93 2471 return res;
a70d6342
IR
2472}
2473
2474
ff802fa1 2475/* Analyze statements in SLP instances of the basic block. Return TRUE if the
a70d6342
IR
2476 operations are supported. */
2477
a12e42fc 2478bool
1a4b99c1 2479vect_slp_analyze_operations (vec<slp_instance> slp_instances, void *data)
a70d6342 2480{
a70d6342
IR
2481 slp_instance instance;
2482 int i;
2483
a12e42fc
RB
2484 if (dump_enabled_p ())
2485 dump_printf_loc (MSG_NOTE, vect_location,
2486 "=== vect_slp_analyze_operations ===\n");
2487
9771b263 2488 for (i = 0; slp_instances.iterate (i, &instance); )
a70d6342 2489 {
a12e42fc 2490 if (!vect_slp_analyze_node_operations (SLP_INSTANCE_TREE (instance)))
a70d6342 2491 {
a12e42fc
RB
2492 dump_printf_loc (MSG_NOTE, vect_location,
2493 "removing SLP instance operations starting from: ");
2494 dump_gimple_stmt (MSG_NOTE, TDF_SLIM,
2495 SLP_TREE_SCALAR_STMTS
2496 (SLP_INSTANCE_TREE (instance))[0], 0);
2497 vect_free_slp_instance (instance);
9771b263 2498 slp_instances.ordered_remove (i);
a70d6342
IR
2499 }
2500 else
1a4b99c1
RB
2501 {
2502 /* Compute the costs of the SLP instance. */
2503 vect_analyze_slp_cost (instance, data);
2504 i++;
2505 }
b8698a0f
L
2506 }
2507
9771b263 2508 if (!slp_instances.length ())
a70d6342
IR
2509 return false;
2510
2511 return true;
2512}
2513
6eddf228
RB
2514
2515/* Compute the scalar cost of the SLP node NODE and its children
2516 and return it. Do not account defs that are marked in LIFE and
2517 update LIFE according to uses of NODE. */
2518
2519static unsigned
292cba13 2520vect_bb_slp_scalar_cost (basic_block bb,
ff4c81cc 2521 slp_tree node, vec<bool, va_heap> *life)
6eddf228
RB
2522{
2523 unsigned scalar_cost = 0;
2524 unsigned i;
355fe088 2525 gimple *stmt;
6eddf228
RB
2526 slp_tree child;
2527
2528 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, stmt)
2529 {
2530 unsigned stmt_cost;
2531 ssa_op_iter op_iter;
2532 def_operand_p def_p;
2533 stmt_vec_info stmt_info;
2534
ff4c81cc 2535 if ((*life)[i])
6eddf228
RB
2536 continue;
2537
2538 /* If there is a non-vectorized use of the defs then the scalar
2539 stmt is kept live in which case we do not account it or any
2540 required defs in the SLP children in the scalar cost. This
2541 way we make the vectorization more costly when compared to
2542 the scalar cost. */
2543 FOR_EACH_SSA_DEF_OPERAND (def_p, stmt, op_iter, SSA_OP_DEF)
2544 {
2545 imm_use_iterator use_iter;
355fe088 2546 gimple *use_stmt;
6eddf228 2547 FOR_EACH_IMM_USE_STMT (use_stmt, use_iter, DEF_FROM_PTR (def_p))
f30a0ba5 2548 if (!is_gimple_debug (use_stmt)
61d371eb
RB
2549 && (! vect_stmt_in_region_p (vinfo_for_stmt (stmt)->vinfo,
2550 use_stmt)
603cca93 2551 || ! PURE_SLP_STMT (vinfo_for_stmt (use_stmt))))
6eddf228 2552 {
ff4c81cc 2553 (*life)[i] = true;
6eddf228
RB
2554 BREAK_FROM_IMM_USE_STMT (use_iter);
2555 }
2556 }
ff4c81cc 2557 if ((*life)[i])
6eddf228
RB
2558 continue;
2559
b555a2e4
RB
2560 /* Count scalar stmts only once. */
2561 if (gimple_visited_p (stmt))
2562 continue;
2563 gimple_set_visited (stmt, true);
2564
6eddf228
RB
2565 stmt_info = vinfo_for_stmt (stmt);
2566 if (STMT_VINFO_DATA_REF (stmt_info))
2567 {
2568 if (DR_IS_READ (STMT_VINFO_DATA_REF (stmt_info)))
2569 stmt_cost = vect_get_stmt_cost (scalar_load);
2570 else
2571 stmt_cost = vect_get_stmt_cost (scalar_store);
2572 }
2573 else
2574 stmt_cost = vect_get_stmt_cost (scalar_stmt);
2575
2576 scalar_cost += stmt_cost;
2577 }
2578
2579 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
603cca93 2580 if (SLP_TREE_DEF_TYPE (child) == vect_internal_def)
90dd6e3d 2581 scalar_cost += vect_bb_slp_scalar_cost (bb, child, life);
6eddf228
RB
2582
2583 return scalar_cost;
2584}
2585
69f11a13
IR
2586/* Check if vectorization of the basic block is profitable. */
2587
2588static bool
2589vect_bb_vectorization_profitable_p (bb_vec_info bb_vinfo)
2590{
9771b263 2591 vec<slp_instance> slp_instances = BB_VINFO_SLP_INSTANCES (bb_vinfo);
69f11a13 2592 slp_instance instance;
1a4b99c1 2593 int i;
c3e7ee41 2594 unsigned int vec_inside_cost = 0, vec_outside_cost = 0, scalar_cost = 0;
92345349 2595 unsigned int vec_prologue_cost = 0, vec_epilogue_cost = 0;
69f11a13
IR
2596
2597 /* Calculate scalar cost. */
6eddf228 2598 FOR_EACH_VEC_ELT (slp_instances, i, instance)
69f11a13 2599 {
00f96dc9 2600 auto_vec<bool, 20> life;
ff4c81cc 2601 life.safe_grow_cleared (SLP_INSTANCE_GROUP_SIZE (instance));
292cba13
RB
2602 scalar_cost += vect_bb_slp_scalar_cost (BB_VINFO_BB (bb_vinfo),
2603 SLP_INSTANCE_TREE (instance),
ff4c81cc 2604 &life);
69f11a13
IR
2605 }
2606
b555a2e4
RB
2607 /* Unset visited flag. */
2608 for (gimple_stmt_iterator gsi = bb_vinfo->region_begin;
2609 gsi_stmt (gsi) != gsi_stmt (bb_vinfo->region_end); gsi_next (&gsi))
2610 gimple_set_visited (gsi_stmt (gsi), false);
2611
c3e7ee41 2612 /* Complete the target-specific cost calculation. */
92345349
BS
2613 finish_cost (BB_VINFO_TARGET_COST_DATA (bb_vinfo), &vec_prologue_cost,
2614 &vec_inside_cost, &vec_epilogue_cost);
2615
2616 vec_outside_cost = vec_prologue_cost + vec_epilogue_cost;
c3e7ee41 2617
73fbfcad 2618 if (dump_enabled_p ())
69f11a13 2619 {
78c60e3d
SS
2620 dump_printf_loc (MSG_NOTE, vect_location, "Cost model analysis: \n");
2621 dump_printf (MSG_NOTE, " Vector inside of basic block cost: %d\n",
2622 vec_inside_cost);
2623 dump_printf (MSG_NOTE, " Vector prologue cost: %d\n", vec_prologue_cost);
2624 dump_printf (MSG_NOTE, " Vector epilogue cost: %d\n", vec_epilogue_cost);
e645e942 2625 dump_printf (MSG_NOTE, " Scalar cost of basic block: %d\n", scalar_cost);
69f11a13
IR
2626 }
2627
a6524bba
RB
2628 /* Vectorization is profitable if its cost is more than the cost of scalar
2629 version. Note that we err on the vector side for equal cost because
2630 the cost estimate is otherwise quite pessimistic (constant uses are
2631 free on the scalar side but cost a load on the vector side for
2632 example). */
2633 if (vec_outside_cost + vec_inside_cost > scalar_cost)
69f11a13
IR
2634 return false;
2635
2636 return true;
2637}
2638
a5b50aa1
RB
2639/* Check if the basic block can be vectorized. Returns a bb_vec_info
2640 if so and sets fatal to true if failure is independent of
2641 current_vector_size. */
a70d6342 2642
8e19f5a1 2643static bb_vec_info
61d371eb
RB
2644vect_slp_analyze_bb_1 (gimple_stmt_iterator region_begin,
2645 gimple_stmt_iterator region_end,
a5b50aa1
RB
2646 vec<data_reference_p> datarefs, int n_stmts,
2647 bool &fatal)
a70d6342
IR
2648{
2649 bb_vec_info bb_vinfo;
a70d6342 2650 slp_instance instance;
8e19f5a1 2651 int i;
777e1f09 2652 int min_vf = 2;
e4a707c4 2653
a5b50aa1
RB
2654 /* The first group of checks is independent of the vector size. */
2655 fatal = true;
2656
61d371eb
RB
2657 if (n_stmts > PARAM_VALUE (PARAM_SLP_MAX_INSNS_IN_BB))
2658 {
2659 if (dump_enabled_p ())
2660 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2661 "not vectorized: too many instructions in "
2662 "basic block.\n");
2663 free_data_refs (datarefs);
2664 return NULL;
2665 }
2666
2667 bb_vinfo = new_bb_vec_info (region_begin, region_end);
a70d6342
IR
2668 if (!bb_vinfo)
2669 return NULL;
2670
61d371eb 2671 BB_VINFO_DATAREFS (bb_vinfo) = datarefs;
428db0ba
RB
2672
2673 /* Analyze the data references. */
2674
2675 if (!vect_analyze_data_refs (bb_vinfo, &min_vf))
a70d6342 2676 {
73fbfcad 2677 if (dump_enabled_p ())
78c60e3d
SS
2678 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2679 "not vectorized: unhandled data-ref in basic "
2680 "block.\n");
b8698a0f 2681
a70d6342
IR
2682 destroy_bb_vec_info (bb_vinfo);
2683 return NULL;
2684 }
2685
fcac74a1 2686 if (BB_VINFO_DATAREFS (bb_vinfo).length () < 2)
a70d6342 2687 {
73fbfcad 2688 if (dump_enabled_p ())
78c60e3d
SS
2689 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2690 "not vectorized: not enough data-refs in "
2691 "basic block.\n");
a70d6342
IR
2692
2693 destroy_bb_vec_info (bb_vinfo);
2694 return NULL;
2695 }
2696
310213d4 2697 if (!vect_analyze_data_ref_accesses (bb_vinfo))
5abe1e05
RB
2698 {
2699 if (dump_enabled_p ())
2700 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2701 "not vectorized: unhandled data access in "
2702 "basic block.\n");
2703
2704 destroy_bb_vec_info (bb_vinfo);
2705 return NULL;
2706 }
2707
a5b50aa1
RB
2708 /* If there are no grouped stores in the region there is no need
2709 to continue with pattern recog as vect_analyze_slp will fail
2710 anyway. */
2711 if (bb_vinfo->grouped_stores.is_empty ())
a70d6342 2712 {
73fbfcad 2713 if (dump_enabled_p ())
a5b50aa1
RB
2714 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2715 "not vectorized: no grouped stores in "
2716 "basic block.\n");
b8698a0f 2717
a70d6342
IR
2718 destroy_bb_vec_info (bb_vinfo);
2719 return NULL;
2720 }
b8698a0f 2721
a5b50aa1
RB
2722 /* While the rest of the analysis below depends on it in some way. */
2723 fatal = false;
2724
2725 vect_pattern_recog (bb_vinfo);
2726
a70d6342
IR
2727 /* Check the SLP opportunities in the basic block, analyze and build SLP
2728 trees. */
310213d4 2729 if (!vect_analyze_slp (bb_vinfo, n_stmts))
a70d6342 2730 {
73fbfcad 2731 if (dump_enabled_p ())
effb52da
RB
2732 {
2733 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2734 "Failed to SLP the basic block.\n");
2735 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2736 "not vectorized: failed to find SLP opportunities "
2737 "in basic block.\n");
2738 }
a70d6342
IR
2739
2740 destroy_bb_vec_info (bb_vinfo);
2741 return NULL;
2742 }
b8698a0f 2743
c2a12ca0
RB
2744 /* Analyze and verify the alignment of data references and the
2745 dependence in the SLP instances. */
a5b50aa1
RB
2746 for (i = 0; BB_VINFO_SLP_INSTANCES (bb_vinfo).iterate (i, &instance); )
2747 {
c2a12ca0
RB
2748 if (! vect_slp_analyze_and_verify_instance_alignment (instance)
2749 || ! vect_slp_analyze_instance_dependence (instance))
a5b50aa1
RB
2750 {
2751 dump_printf_loc (MSG_NOTE, vect_location,
2752 "removing SLP instance operations starting from: ");
2753 dump_gimple_stmt (MSG_NOTE, TDF_SLIM,
2754 SLP_TREE_SCALAR_STMTS
2755 (SLP_INSTANCE_TREE (instance))[0], 0);
2756 vect_free_slp_instance (instance);
2757 BB_VINFO_SLP_INSTANCES (bb_vinfo).ordered_remove (i);
2758 continue;
2759 }
c2a12ca0
RB
2760
2761 /* Mark all the statements that we want to vectorize as pure SLP and
2762 relevant. */
2763 vect_mark_slp_stmts (SLP_INSTANCE_TREE (instance), pure_slp, -1);
2764 vect_mark_slp_stmts_relevant (SLP_INSTANCE_TREE (instance));
2765
a5b50aa1
RB
2766 i++;
2767 }
a5b50aa1
RB
2768 if (! BB_VINFO_SLP_INSTANCES (bb_vinfo).length ())
2769 {
2770 destroy_bb_vec_info (bb_vinfo);
2771 return NULL;
2772 }
2773
1a4b99c1
RB
2774 if (!vect_slp_analyze_operations (BB_VINFO_SLP_INSTANCES (bb_vinfo),
2775 BB_VINFO_TARGET_COST_DATA (bb_vinfo)))
a70d6342 2776 {
73fbfcad 2777 if (dump_enabled_p ())
e645e942 2778 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
78c60e3d 2779 "not vectorized: bad operation in basic block.\n");
a70d6342
IR
2780
2781 destroy_bb_vec_info (bb_vinfo);
2782 return NULL;
2783 }
2784
69f11a13 2785 /* Cost model: check if the vectorization is worthwhile. */
8b5e1202 2786 if (!unlimited_cost_model (NULL)
69f11a13
IR
2787 && !vect_bb_vectorization_profitable_p (bb_vinfo))
2788 {
73fbfcad 2789 if (dump_enabled_p ())
78c60e3d
SS
2790 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2791 "not vectorized: vectorization is not "
2792 "profitable.\n");
69f11a13
IR
2793
2794 destroy_bb_vec_info (bb_vinfo);
2795 return NULL;
2796 }
2797
73fbfcad 2798 if (dump_enabled_p ())
78c60e3d
SS
2799 dump_printf_loc (MSG_NOTE, vect_location,
2800 "Basic block will be vectorized using SLP\n");
a70d6342
IR
2801
2802 return bb_vinfo;
2803}
2804
2805
428db0ba
RB
2806/* Main entry for the BB vectorizer. Analyze and transform BB, returns
2807 true if anything in the basic-block was vectorized. */
2808
2809bool
2810vect_slp_bb (basic_block bb)
8e19f5a1
IR
2811{
2812 bb_vec_info bb_vinfo;
8e19f5a1
IR
2813 gimple_stmt_iterator gsi;
2814 unsigned int vector_sizes;
61d371eb 2815 bool any_vectorized = false;
8e19f5a1 2816
73fbfcad 2817 if (dump_enabled_p ())
78c60e3d 2818 dump_printf_loc (MSG_NOTE, vect_location, "===vect_slp_analyze_bb===\n");
8e19f5a1 2819
8e19f5a1
IR
2820 /* Autodetect first vector size we try. */
2821 current_vector_size = 0;
2822 vector_sizes = targetm.vectorize.autovectorize_vector_sizes ();
2823
61d371eb
RB
2824 gsi = gsi_start_bb (bb);
2825
8e19f5a1
IR
2826 while (1)
2827 {
61d371eb
RB
2828 if (gsi_end_p (gsi))
2829 break;
2830
2831 gimple_stmt_iterator region_begin = gsi;
2832 vec<data_reference_p> datarefs = vNULL;
2833 int insns = 0;
2834
2835 for (; !gsi_end_p (gsi); gsi_next (&gsi))
428db0ba 2836 {
61d371eb
RB
2837 gimple *stmt = gsi_stmt (gsi);
2838 if (is_gimple_debug (stmt))
2839 continue;
2840 insns++;
2841
2842 if (gimple_location (stmt) != UNKNOWN_LOCATION)
2843 vect_location = gimple_location (stmt);
2844
2845 if (!find_data_references_in_stmt (NULL, stmt, &datarefs))
2846 break;
2847 }
2848
2849 /* Skip leading unhandled stmts. */
2850 if (gsi_stmt (region_begin) == gsi_stmt (gsi))
2851 {
2852 gsi_next (&gsi);
2853 continue;
2854 }
428db0ba 2855
61d371eb
RB
2856 gimple_stmt_iterator region_end = gsi;
2857
2858 bool vectorized = false;
a5b50aa1 2859 bool fatal = false;
61d371eb 2860 bb_vinfo = vect_slp_analyze_bb_1 (region_begin, region_end,
a5b50aa1 2861 datarefs, insns, fatal);
61d371eb
RB
2862 if (bb_vinfo
2863 && dbg_cnt (vect_slp))
2864 {
428db0ba 2865 if (dump_enabled_p ())
61d371eb 2866 dump_printf_loc (MSG_NOTE, vect_location, "SLPing BB part\n");
428db0ba
RB
2867
2868 vect_schedule_slp (bb_vinfo);
2869
2870 if (dump_enabled_p ())
2871 dump_printf_loc (MSG_NOTE, vect_location,
61d371eb 2872 "basic block part vectorized\n");
428db0ba
RB
2873
2874 destroy_bb_vec_info (bb_vinfo);
2875
61d371eb 2876 vectorized = true;
428db0ba 2877 }
61d371eb
RB
2878 else
2879 destroy_bb_vec_info (bb_vinfo);
8e19f5a1 2880
61d371eb 2881 any_vectorized |= vectorized;
8e19f5a1
IR
2882
2883 vector_sizes &= ~current_vector_size;
61d371eb
RB
2884 if (vectorized
2885 || vector_sizes == 0
a5b50aa1
RB
2886 || current_vector_size == 0
2887 /* If vect_slp_analyze_bb_1 signaled that analysis for all
2888 vector sizes will fail do not bother iterating. */
2889 || fatal)
61d371eb
RB
2890 {
2891 if (gsi_end_p (region_end))
2892 break;
8e19f5a1 2893
61d371eb
RB
2894 /* Skip the unhandled stmt. */
2895 gsi_next (&gsi);
2896
2897 /* And reset vector sizes. */
2898 current_vector_size = 0;
2899 vector_sizes = targetm.vectorize.autovectorize_vector_sizes ();
2900 }
2901 else
2902 {
2903 /* Try the next biggest vector size. */
2904 current_vector_size = 1 << floor_log2 (vector_sizes);
2905 if (dump_enabled_p ())
2906 dump_printf_loc (MSG_NOTE, vect_location,
2907 "***** Re-trying analysis with "
2908 "vector size %d\n", current_vector_size);
2909
2910 /* Start over. */
2911 gsi = region_begin;
2912 }
8e19f5a1 2913 }
61d371eb
RB
2914
2915 return any_vectorized;
8e19f5a1
IR
2916}
2917
2918
e4af0bc4
IE
2919/* Return 1 if vector type of boolean constant which is OPNUM
2920 operand in statement STMT is a boolean vector. */
2921
2922static bool
2923vect_mask_constant_operand_p (gimple *stmt, int opnum)
2924{
2925 stmt_vec_info stmt_vinfo = vinfo_for_stmt (stmt);
2926 enum tree_code code = gimple_expr_code (stmt);
2927 tree op, vectype;
2928 gimple *def_stmt;
2929 enum vect_def_type dt;
2930
2931 /* For comparison and COND_EXPR type is chosen depending
2932 on the other comparison operand. */
2933 if (TREE_CODE_CLASS (code) == tcc_comparison)
2934 {
2935 if (opnum)
2936 op = gimple_assign_rhs1 (stmt);
2937 else
2938 op = gimple_assign_rhs2 (stmt);
2939
2940 if (!vect_is_simple_use (op, stmt_vinfo->vinfo, &def_stmt,
2941 &dt, &vectype))
2942 gcc_unreachable ();
2943
2944 return !vectype || VECTOR_BOOLEAN_TYPE_P (vectype);
2945 }
2946
2947 if (code == COND_EXPR)
2948 {
2949 tree cond = gimple_assign_rhs1 (stmt);
2950
2951 if (TREE_CODE (cond) == SSA_NAME)
7b1b0cc1
RB
2952 op = cond;
2953 else if (opnum)
e4af0bc4
IE
2954 op = TREE_OPERAND (cond, 1);
2955 else
2956 op = TREE_OPERAND (cond, 0);
2957
2958 if (!vect_is_simple_use (op, stmt_vinfo->vinfo, &def_stmt,
2959 &dt, &vectype))
2960 gcc_unreachable ();
2961
2962 return !vectype || VECTOR_BOOLEAN_TYPE_P (vectype);
2963 }
2964
2965 return VECTOR_BOOLEAN_TYPE_P (STMT_VINFO_VECTYPE (stmt_vinfo));
2966}
2967
2968
b8698a0f
L
2969/* For constant and loop invariant defs of SLP_NODE this function returns
2970 (vector) defs (VEC_OPRNDS) that will be used in the vectorized stmts.
d59dc888
IR
2971 OP_NUM determines if we gather defs for operand 0 or operand 1 of the RHS of
2972 scalar stmts. NUMBER_OF_VECTORS is the number of vector defs to create.
b5aeb3bb
IR
2973 REDUC_INDEX is the index of the reduction operand in the statements, unless
2974 it is -1. */
ebfd146a
IR
2975
2976static void
9dc3f7de 2977vect_get_constant_vectors (tree op, slp_tree slp_node,
9771b263 2978 vec<tree> *vec_oprnds,
306b0c92 2979 unsigned int op_num, unsigned int number_of_vectors)
ebfd146a 2980{
355fe088
TS
2981 vec<gimple *> stmts = SLP_TREE_SCALAR_STMTS (slp_node);
2982 gimple *stmt = stmts[0];
ebfd146a 2983 stmt_vec_info stmt_vinfo = vinfo_for_stmt (stmt);
d2a12ae7 2984 unsigned nunits;
ebfd146a 2985 tree vec_cst;
d2a12ae7
RG
2986 tree *elts;
2987 unsigned j, number_of_places_left_in_vector;
ebfd146a 2988 tree vector_type;
9dc3f7de 2989 tree vop;
9771b263 2990 int group_size = stmts.length ();
ebfd146a 2991 unsigned int vec_num, i;
d2a12ae7 2992 unsigned number_of_copies = 1;
9771b263
DN
2993 vec<tree> voprnds;
2994 voprnds.create (number_of_vectors);
ebfd146a 2995 bool constant_p, is_store;
b5aeb3bb 2996 tree neutral_op = NULL;
bac430c9 2997 enum tree_code code = gimple_expr_code (stmt);
13396b6e 2998 gimple_seq ctor_seq = NULL;
b5aeb3bb 2999
42fd8198 3000 /* Check if vector type is a boolean vector. */
2568d8a1 3001 if (VECT_SCALAR_BOOLEAN_TYPE_P (TREE_TYPE (op))
e4af0bc4 3002 && vect_mask_constant_operand_p (stmt, op_num))
42fd8198
IE
3003 vector_type
3004 = build_same_sized_truth_vector_type (STMT_VINFO_VECTYPE (stmt_vinfo));
3005 else
3006 vector_type = get_vectype_for_scalar_type (TREE_TYPE (op));
afbe6325
RB
3007 nunits = TYPE_VECTOR_SUBPARTS (vector_type);
3008
ebfd146a
IR
3009 if (STMT_VINFO_DATA_REF (stmt_vinfo))
3010 {
3011 is_store = true;
3012 op = gimple_assign_rhs1 (stmt);
3013 }
3014 else
9dc3f7de
IR
3015 is_store = false;
3016
3017 gcc_assert (op);
ebfd146a 3018
ebfd146a 3019 /* NUMBER_OF_COPIES is the number of times we need to use the same values in
b8698a0f 3020 created vectors. It is greater than 1 if unrolling is performed.
ebfd146a
IR
3021
3022 For example, we have two scalar operands, s1 and s2 (e.g., group of
3023 strided accesses of size two), while NUNITS is four (i.e., four scalars
f7e531cf
IR
3024 of this type can be packed in a vector). The output vector will contain
3025 two copies of each scalar operand: {s1, s2, s1, s2}. (NUMBER_OF_COPIES
ebfd146a
IR
3026 will be 2).
3027
b8698a0f 3028 If GROUP_SIZE > NUNITS, the scalars will be split into several vectors
ebfd146a
IR
3029 containing the operands.
3030
3031 For example, NUNITS is four as before, and the group size is 8
f7e531cf 3032 (s1, s2, ..., s8). We will create two vectors {s1, s2, s3, s4} and
ebfd146a 3033 {s5, s6, s7, s8}. */
b8698a0f 3034
14a61437 3035 number_of_copies = nunits * number_of_vectors / group_size;
ebfd146a
IR
3036
3037 number_of_places_left_in_vector = nunits;
62cf7335 3038 constant_p = true;
d2a12ae7 3039 elts = XALLOCAVEC (tree, nunits);
90dd6e3d 3040 bool place_after_defs = false;
ebfd146a
IR
3041 for (j = 0; j < number_of_copies; j++)
3042 {
9771b263 3043 for (i = group_size - 1; stmts.iterate (i, &stmt); i--)
ebfd146a
IR
3044 {
3045 if (is_store)
3046 op = gimple_assign_rhs1 (stmt);
bac430c9 3047 else
f7e531cf 3048 {
bac430c9 3049 switch (code)
f7e531cf 3050 {
bac430c9 3051 case COND_EXPR:
a989bcc3
IE
3052 {
3053 tree cond = gimple_assign_rhs1 (stmt);
3054 if (TREE_CODE (cond) == SSA_NAME)
3055 op = gimple_op (stmt, op_num + 1);
3056 else if (op_num == 0 || op_num == 1)
bac430c9 3057 op = TREE_OPERAND (cond, op_num);
a989bcc3
IE
3058 else
3059 {
3060 if (op_num == 2)
3061 op = gimple_assign_rhs2 (stmt);
3062 else
3063 op = gimple_assign_rhs3 (stmt);
3064 }
3065 }
bac430c9
IR
3066 break;
3067
3068 case CALL_EXPR:
3069 op = gimple_call_arg (stmt, op_num);
3070 break;
3071
b84b294a
JJ
3072 case LSHIFT_EXPR:
3073 case RSHIFT_EXPR:
3074 case LROTATE_EXPR:
3075 case RROTATE_EXPR:
3076 op = gimple_op (stmt, op_num + 1);
3077 /* Unlike the other binary operators, shifts/rotates have
3078 the shift count being int, instead of the same type as
3079 the lhs, so make sure the scalar is the right type if
3080 we are dealing with vectors of
3081 long long/long/short/char. */
793d9a16 3082 if (op_num == 1 && TREE_CODE (op) == INTEGER_CST)
b84b294a
JJ
3083 op = fold_convert (TREE_TYPE (vector_type), op);
3084 break;
3085
bac430c9
IR
3086 default:
3087 op = gimple_op (stmt, op_num + 1);
b84b294a 3088 break;
f7e531cf
IR
3089 }
3090 }
b8698a0f 3091
ebfd146a 3092 /* Create 'vect_ = {op0,op1,...,opn}'. */
ebfd146a 3093 number_of_places_left_in_vector--;
90dd6e3d 3094 tree orig_op = op;
13396b6e 3095 if (!types_compatible_p (TREE_TYPE (vector_type), TREE_TYPE (op)))
50eeef09 3096 {
793d9a16 3097 if (CONSTANT_CLASS_P (op))
13396b6e 3098 {
42fd8198
IE
3099 if (VECTOR_BOOLEAN_TYPE_P (vector_type))
3100 {
3101 /* Can't use VIEW_CONVERT_EXPR for booleans because
3102 of possibly different sizes of scalar value and
3103 vector element. */
3104 if (integer_zerop (op))
3105 op = build_int_cst (TREE_TYPE (vector_type), 0);
3106 else if (integer_onep (op))
158beb4a 3107 op = build_all_ones_cst (TREE_TYPE (vector_type));
42fd8198
IE
3108 else
3109 gcc_unreachable ();
3110 }
3111 else
3112 op = fold_unary (VIEW_CONVERT_EXPR,
3113 TREE_TYPE (vector_type), op);
13396b6e
JJ
3114 gcc_assert (op && CONSTANT_CLASS_P (op));
3115 }
3116 else
3117 {
b731b390 3118 tree new_temp = make_ssa_name (TREE_TYPE (vector_type));
355fe088 3119 gimple *init_stmt;
262a363f
JJ
3120 if (VECTOR_BOOLEAN_TYPE_P (vector_type))
3121 {
158beb4a
JJ
3122 tree true_val
3123 = build_all_ones_cst (TREE_TYPE (vector_type));
3124 tree false_val
3125 = build_zero_cst (TREE_TYPE (vector_type));
7c285ab9 3126 gcc_assert (INTEGRAL_TYPE_P (TREE_TYPE (op)));
158beb4a
JJ
3127 init_stmt = gimple_build_assign (new_temp, COND_EXPR,
3128 op, true_val,
3129 false_val);
262a363f 3130 }
262a363f
JJ
3131 else
3132 {
3133 op = build1 (VIEW_CONVERT_EXPR, TREE_TYPE (vector_type),
3134 op);
3135 init_stmt
3136 = gimple_build_assign (new_temp, VIEW_CONVERT_EXPR,
3137 op);
3138 }
13396b6e
JJ
3139 gimple_seq_add_stmt (&ctor_seq, init_stmt);
3140 op = new_temp;
3141 }
50eeef09 3142 }
d2a12ae7 3143 elts[number_of_places_left_in_vector] = op;
793d9a16
RB
3144 if (!CONSTANT_CLASS_P (op))
3145 constant_p = false;
90dd6e3d
RB
3146 if (TREE_CODE (orig_op) == SSA_NAME
3147 && !SSA_NAME_IS_DEFAULT_DEF (orig_op)
3148 && STMT_VINFO_BB_VINFO (stmt_vinfo)
3149 && (STMT_VINFO_BB_VINFO (stmt_vinfo)->bb
3150 == gimple_bb (SSA_NAME_DEF_STMT (orig_op))))
3151 place_after_defs = true;
ebfd146a
IR
3152
3153 if (number_of_places_left_in_vector == 0)
3154 {
ebfd146a 3155 if (constant_p)
d2a12ae7 3156 vec_cst = build_vector (vector_type, elts);
ebfd146a 3157 else
d2a12ae7 3158 {
9771b263 3159 vec<constructor_elt, va_gc> *v;
d2a12ae7 3160 unsigned k;
9771b263 3161 vec_alloc (v, nunits);
d2a12ae7
RG
3162 for (k = 0; k < nunits; ++k)
3163 CONSTRUCTOR_APPEND_ELT (v, NULL_TREE, elts[k]);
3164 vec_cst = build_constructor (vector_type, v);
3165 }
90dd6e3d
RB
3166 tree init;
3167 gimple_stmt_iterator gsi;
3168 if (place_after_defs)
3169 {
3170 gsi = gsi_for_stmt
3171 (vect_find_last_scalar_stmt_in_slp (slp_node));
3172 init = vect_init_vector (stmt, vec_cst, vector_type, &gsi);
3173 }
3174 else
3175 init = vect_init_vector (stmt, vec_cst, vector_type, NULL);
13396b6e
JJ
3176 if (ctor_seq != NULL)
3177 {
90dd6e3d 3178 gsi = gsi_for_stmt (SSA_NAME_DEF_STMT (init));
13396b6e
JJ
3179 gsi_insert_seq_before_without_update (&gsi, ctor_seq,
3180 GSI_SAME_STMT);
3181 ctor_seq = NULL;
3182 }
90dd6e3d
RB
3183 voprnds.quick_push (init);
3184 place_after_defs = false;
62cf7335
RB
3185 number_of_places_left_in_vector = nunits;
3186 constant_p = true;
ebfd146a
IR
3187 }
3188 }
3189 }
3190
b8698a0f 3191 /* Since the vectors are created in the reverse order, we should invert
ebfd146a 3192 them. */
9771b263 3193 vec_num = voprnds.length ();
d2a12ae7 3194 for (j = vec_num; j != 0; j--)
ebfd146a 3195 {
9771b263
DN
3196 vop = voprnds[j - 1];
3197 vec_oprnds->quick_push (vop);
ebfd146a
IR
3198 }
3199
9771b263 3200 voprnds.release ();
ebfd146a
IR
3201
3202 /* In case that VF is greater than the unrolling factor needed for the SLP
b8698a0f
L
3203 group of stmts, NUMBER_OF_VECTORS to be created is greater than
3204 NUMBER_OF_SCALARS/NUNITS or NUNITS/NUMBER_OF_SCALARS, and hence we have
ebfd146a 3205 to replicate the vectors. */
9771b263 3206 while (number_of_vectors > vec_oprnds->length ())
ebfd146a 3207 {
b5aeb3bb
IR
3208 tree neutral_vec = NULL;
3209
3210 if (neutral_op)
3211 {
3212 if (!neutral_vec)
b9acc9f1 3213 neutral_vec = build_vector_from_val (vector_type, neutral_op);
b5aeb3bb 3214
9771b263 3215 vec_oprnds->quick_push (neutral_vec);
b5aeb3bb
IR
3216 }
3217 else
3218 {
9771b263
DN
3219 for (i = 0; vec_oprnds->iterate (i, &vop) && i < vec_num; i++)
3220 vec_oprnds->quick_push (vop);
b5aeb3bb 3221 }
ebfd146a
IR
3222 }
3223}
3224
3225
3226/* Get vectorized definitions from SLP_NODE that contains corresponding
3227 vectorized def-stmts. */
3228
3229static void
9771b263 3230vect_get_slp_vect_defs (slp_tree slp_node, vec<tree> *vec_oprnds)
ebfd146a
IR
3231{
3232 tree vec_oprnd;
355fe088 3233 gimple *vec_def_stmt;
ebfd146a
IR
3234 unsigned int i;
3235
9771b263 3236 gcc_assert (SLP_TREE_VEC_STMTS (slp_node).exists ());
ebfd146a 3237
9771b263 3238 FOR_EACH_VEC_ELT (SLP_TREE_VEC_STMTS (slp_node), i, vec_def_stmt)
ebfd146a
IR
3239 {
3240 gcc_assert (vec_def_stmt);
e7baeb39
RB
3241 if (gimple_code (vec_def_stmt) == GIMPLE_PHI)
3242 vec_oprnd = gimple_phi_result (vec_def_stmt);
3243 else
3244 vec_oprnd = gimple_get_lhs (vec_def_stmt);
9771b263 3245 vec_oprnds->quick_push (vec_oprnd);
ebfd146a
IR
3246 }
3247}
3248
3249
b8698a0f
L
3250/* Get vectorized definitions for SLP_NODE.
3251 If the scalar definitions are loop invariants or constants, collect them and
ebfd146a
IR
3252 call vect_get_constant_vectors() to create vector stmts.
3253 Otherwise, the def-stmts must be already vectorized and the vectorized stmts
d092494c
IR
3254 must be stored in the corresponding child of SLP_NODE, and we call
3255 vect_get_slp_vect_defs () to retrieve them. */
b8698a0f 3256
ebfd146a 3257void
9771b263 3258vect_get_slp_defs (vec<tree> ops, slp_tree slp_node,
306b0c92 3259 vec<vec<tree> > *vec_oprnds)
ebfd146a 3260{
355fe088 3261 gimple *first_stmt;
d092494c 3262 int number_of_vects = 0, i;
b8698a0f 3263 HOST_WIDE_INT lhs_size_unit, rhs_size_unit;
d092494c 3264 slp_tree child = NULL;
37b5ec8f 3265 vec<tree> vec_defs;
e44978dc 3266 tree oprnd;
038000bd 3267 bool first_iteration = true;
ebfd146a 3268
9771b263
DN
3269 first_stmt = SLP_TREE_SCALAR_STMTS (slp_node)[0];
3270 FOR_EACH_VEC_ELT (ops, i, oprnd)
ebfd146a 3271 {
038000bd
AH
3272 bool vectorized_defs = false;
3273
3274 if (oprnd == NULL)
3275 {
3276 vec_defs = vNULL;
3277 vec_defs.create (0);
3278 vec_oprnds->quick_push (vec_defs);
3279 continue;
3280 }
3281
d092494c
IR
3282 /* For each operand we check if it has vectorized definitions in a child
3283 node or we need to create them (for invariants and constants). We
3284 check if the LHS of the first stmt of the next child matches OPRND.
3285 If it does, we found the correct child. Otherwise, we call
038000bd
AH
3286 vect_get_constant_vectors (). */
3287 for (unsigned int child_index = 0;
3288 child_index < SLP_TREE_CHILDREN (slp_node).length (); child_index++)
ebfd146a 3289 {
01d8bf07 3290 child = SLP_TREE_CHILDREN (slp_node)[child_index];
d092494c 3291
e44978dc 3292 /* We have to check both pattern and original def, if available. */
603cca93 3293 if (SLP_TREE_DEF_TYPE (child) == vect_internal_def)
e44978dc 3294 {
355fe088
TS
3295 gimple *first_def = SLP_TREE_SCALAR_STMTS (child)[0];
3296 gimple *related
90dd6e3d 3297 = STMT_VINFO_RELATED_STMT (vinfo_for_stmt (first_def));
e7baeb39 3298 tree first_def_op;
90dd6e3d 3299
e7baeb39
RB
3300 if (gimple_code (first_def) == GIMPLE_PHI)
3301 first_def_op = gimple_phi_result (first_def);
3302 else
3303 first_def_op = gimple_get_lhs (first_def);
3304 if (operand_equal_p (oprnd, first_def_op, 0)
90dd6e3d
RB
3305 || (related
3306 && operand_equal_p (oprnd, gimple_get_lhs (related), 0)))
3307 {
3308 /* The number of vector defs is determined by the number of
3309 vector statements in the node from which we get those
3310 statements. */
3311 number_of_vects = SLP_TREE_NUMBER_OF_VEC_STMTS (child);
3312 vectorized_defs = true;
038000bd 3313 break;
90dd6e3d 3314 }
e44978dc 3315 }
d092494c 3316 }
ebfd146a 3317
038000bd
AH
3318 if (!vectorized_defs && first_iteration)
3319 {
3320 number_of_vects = SLP_TREE_NUMBER_OF_VEC_STMTS (slp_node);
3321 /* Number of vector stmts was calculated according to LHS in
3322 vect_schedule_slp_instance (), fix it by replacing LHS with
3323 RHS, if necessary. See vect_get_smallest_scalar_type () for
3324 details. */
3325 vect_get_smallest_scalar_type (first_stmt, &lhs_size_unit,
3326 &rhs_size_unit);
3327 if (rhs_size_unit != lhs_size_unit)
3328 {
3329 number_of_vects *= rhs_size_unit;
3330 number_of_vects /= lhs_size_unit;
3331 }
d092494c 3332 }
b5aeb3bb 3333
d092494c 3334 /* Allocate memory for vectorized defs. */
37b5ec8f
JJ
3335 vec_defs = vNULL;
3336 vec_defs.create (number_of_vects);
ebfd146a 3337
d092494c
IR
3338 /* For reduction defs we call vect_get_constant_vectors (), since we are
3339 looking for initial loop invariant values. */
306b0c92 3340 if (vectorized_defs)
d092494c 3341 /* The defs are already vectorized. */
37b5ec8f 3342 vect_get_slp_vect_defs (child, &vec_defs);
d092494c 3343 else
e7baeb39 3344 /* Build vectors from scalar defs. */
37b5ec8f 3345 vect_get_constant_vectors (oprnd, slp_node, &vec_defs, i,
306b0c92 3346 number_of_vects);
ebfd146a 3347
37b5ec8f 3348 vec_oprnds->quick_push (vec_defs);
ebfd146a 3349
038000bd 3350 first_iteration = false;
d092494c 3351 }
ebfd146a
IR
3352}
3353
ebfd146a
IR
3354/* Generate vector permute statements from a list of loads in DR_CHAIN.
3355 If ANALYZE_ONLY is TRUE, only check that it is possible to create valid
01d8bf07
RB
3356 permute statements for the SLP node NODE of the SLP instance
3357 SLP_NODE_INSTANCE. */
3358
ebfd146a 3359bool
01d8bf07 3360vect_transform_slp_perm_load (slp_tree node, vec<tree> dr_chain,
ebfd146a 3361 gimple_stmt_iterator *gsi, int vf,
29afecdf
RB
3362 slp_instance slp_node_instance, bool analyze_only,
3363 unsigned *n_perms)
ebfd146a 3364{
355fe088 3365 gimple *stmt = SLP_TREE_SCALAR_STMTS (node)[0];
ebfd146a
IR
3366 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
3367 tree mask_element_type = NULL_TREE, mask_type;
2ce27200 3368 int nunits, vec_index = 0;
2635892a 3369 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
ebfd146a 3370 int group_size = SLP_INSTANCE_GROUP_SIZE (slp_node_instance);
61fdfd8c 3371 int mask_element;
22e4dee7 3372 unsigned char *mask;
ef4bddc2 3373 machine_mode mode;
ebfd146a 3374
91ff1504
RB
3375 if (!STMT_VINFO_GROUPED_ACCESS (stmt_info))
3376 return false;
3377
3378 stmt_info = vinfo_for_stmt (GROUP_FIRST_ELEMENT (stmt_info));
3379
22e4dee7
RH
3380 mode = TYPE_MODE (vectype);
3381
2635892a
RH
3382 /* The generic VEC_PERM_EXPR code always uses an integral type of the
3383 same size as the vector element being permuted. */
96f9265a
RG
3384 mask_element_type = lang_hooks.types.type_for_mode
3385 (int_mode_for_mode (TYPE_MODE (TREE_TYPE (vectype))), 1);
ebfd146a 3386 mask_type = get_vectype_for_scalar_type (mask_element_type);
ebfd146a 3387 nunits = TYPE_VECTOR_SUBPARTS (vectype);
22e4dee7 3388 mask = XALLOCAVEC (unsigned char, nunits);
ebfd146a 3389
61fdfd8c
RB
3390 /* Initialize the vect stmts of NODE to properly insert the generated
3391 stmts later. */
3392 if (! analyze_only)
3393 for (unsigned i = SLP_TREE_VEC_STMTS (node).length ();
3394 i < SLP_TREE_NUMBER_OF_VEC_STMTS (node); i++)
3395 SLP_TREE_VEC_STMTS (node).quick_push (NULL);
ebfd146a 3396
b8698a0f
L
3397 /* Generate permutation masks for every NODE. Number of masks for each NODE
3398 is equal to GROUP_SIZE.
3399 E.g., we have a group of three nodes with three loads from the same
3400 location in each node, and the vector size is 4. I.e., we have a
3401 a0b0c0a1b1c1... sequence and we need to create the following vectors:
ebfd146a
IR
3402 for a's: a0a0a0a1 a1a1a2a2 a2a3a3a3
3403 for b's: b0b0b0b1 b1b1b2b2 b2b3b3b3
3404 ...
3405
2635892a 3406 The masks for a's should be: {0,0,0,3} {3,3,6,6} {6,9,9,9}.
b8698a0f 3407 The last mask is illegal since we assume two operands for permute
ff802fa1
IR
3408 operation, and the mask element values can't be outside that range.
3409 Hence, the last mask must be converted into {2,5,5,5}.
b8698a0f 3410 For the first two permutations we need the first and the second input
ebfd146a 3411 vectors: {a0,b0,c0,a1} and {b1,c1,a2,b2}, and for the last permutation
b8698a0f 3412 we need the second and the third vectors: {b1,c1,a2,b2} and
ebfd146a
IR
3413 {c2,a3,b3,c3}. */
3414
2ce27200
RB
3415 int vect_stmts_counter = 0;
3416 int index = 0;
3417 int first_vec_index = -1;
3418 int second_vec_index = -1;
be377c80 3419 bool noop_p = true;
29afecdf 3420 *n_perms = 0;
ebfd146a 3421
61fdfd8c 3422 for (int j = 0; j < vf; j++)
2ce27200
RB
3423 {
3424 for (int k = 0; k < group_size; k++)
3425 {
3426 int i = (SLP_TREE_LOAD_PERMUTATION (node)[k]
3427 + j * STMT_VINFO_GROUP_SIZE (stmt_info));
3428 vec_index = i / nunits;
3429 mask_element = i % nunits;
3430 if (vec_index == first_vec_index
3431 || first_vec_index == -1)
3432 {
3433 first_vec_index = vec_index;
3434 }
3435 else if (vec_index == second_vec_index
3436 || second_vec_index == -1)
3437 {
3438 second_vec_index = vec_index;
3439 mask_element += nunits;
3440 }
3441 else
3442 {
3443 if (dump_enabled_p ())
3444 {
3445 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
3446 "permutation requires at "
3447 "least three vectors ");
3448 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
3449 stmt, 0);
2ce27200
RB
3450 }
3451 return false;
3452 }
ebfd146a 3453
2ce27200
RB
3454 gcc_assert (mask_element >= 0
3455 && mask_element < 2 * nunits);
be377c80
RB
3456 if (mask_element != index)
3457 noop_p = false;
2ce27200
RB
3458 mask[index++] = mask_element;
3459
3460 if (index == nunits)
3461 {
be377c80
RB
3462 if (! noop_p
3463 && ! can_vec_perm_p (mode, false, mask))
2ce27200
RB
3464 {
3465 if (dump_enabled_p ())
22e4dee7 3466 {
2ce27200
RB
3467 dump_printf_loc (MSG_MISSED_OPTIMIZATION,
3468 vect_location,
3469 "unsupported vect permute { ");
3470 for (i = 0; i < nunits; ++i)
3471 dump_printf (MSG_MISSED_OPTIMIZATION, "%d ", mask[i]);
3472 dump_printf (MSG_MISSED_OPTIMIZATION, "}\n");
22e4dee7 3473 }
2ce27200
RB
3474 return false;
3475 }
22e4dee7 3476
29afecdf
RB
3477 if (! noop_p)
3478 ++*n_perms;
3479
2ce27200
RB
3480 if (!analyze_only)
3481 {
be377c80
RB
3482 tree mask_vec = NULL_TREE;
3483
3484 if (! noop_p)
3485 {
3486 tree *mask_elts = XALLOCAVEC (tree, nunits);
3487 for (int l = 0; l < nunits; ++l)
3488 mask_elts[l] = build_int_cst (mask_element_type,
3489 mask[l]);
3490 mask_vec = build_vector (mask_type, mask_elts);
3491 }
2ce27200
RB
3492
3493 if (second_vec_index == -1)
3494 second_vec_index = first_vec_index;
61fdfd8c
RB
3495
3496 /* Generate the permute statement if necessary. */
3497 tree first_vec = dr_chain[first_vec_index];
3498 tree second_vec = dr_chain[second_vec_index];
3499 gimple *perm_stmt;
3500 if (! noop_p)
3501 {
3502 tree perm_dest
3503 = vect_create_destination_var (gimple_assign_lhs (stmt),
3504 vectype);
3505 perm_dest = make_ssa_name (perm_dest);
3506 perm_stmt = gimple_build_assign (perm_dest,
3507 VEC_PERM_EXPR,
3508 first_vec, second_vec,
3509 mask_vec);
3510 vect_finish_stmt_generation (stmt, perm_stmt, gsi);
3511 }
3512 else
3513 /* If mask was NULL_TREE generate the requested
3514 identity transform. */
3515 perm_stmt = SSA_NAME_DEF_STMT (first_vec);
3516
3517 /* Store the vector statement in NODE. */
3518 SLP_TREE_VEC_STMTS (node)[vect_stmts_counter++] = perm_stmt;
2ce27200 3519 }
ebfd146a 3520
2ce27200
RB
3521 index = 0;
3522 first_vec_index = -1;
3523 second_vec_index = -1;
be377c80 3524 noop_p = true;
2ce27200
RB
3525 }
3526 }
b8698a0f 3527 }
ebfd146a 3528
ebfd146a
IR
3529 return true;
3530}
3531
3532
3533
3534/* Vectorize SLP instance tree in postorder. */
3535
3536static bool
3537vect_schedule_slp_instance (slp_tree node, slp_instance instance,
a70d6342 3538 unsigned int vectorization_factor)
ebfd146a 3539{
355fe088 3540 gimple *stmt;
0d0293ac 3541 bool grouped_store, is_store;
ebfd146a
IR
3542 gimple_stmt_iterator si;
3543 stmt_vec_info stmt_info;
3544 unsigned int vec_stmts_size, nunits, group_size;
3545 tree vectype;
603cca93 3546 int i, j;
d755c7ef 3547 slp_tree child;
ebfd146a 3548
603cca93 3549 if (SLP_TREE_DEF_TYPE (node) != vect_internal_def)
ebfd146a
IR
3550 return false;
3551
9771b263 3552 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
d755c7ef 3553 vect_schedule_slp_instance (child, instance, vectorization_factor);
b8698a0f 3554
603cca93
RB
3555 /* Push SLP node def-type to stmts. */
3556 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
3557 if (SLP_TREE_DEF_TYPE (child) != vect_internal_def)
3558 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (child), j, stmt)
3559 STMT_VINFO_DEF_TYPE (vinfo_for_stmt (stmt)) = SLP_TREE_DEF_TYPE (child);
3560
9771b263 3561 stmt = SLP_TREE_SCALAR_STMTS (node)[0];
ebfd146a
IR
3562 stmt_info = vinfo_for_stmt (stmt);
3563
3564 /* VECTYPE is the type of the destination. */
b690cc0f 3565 vectype = STMT_VINFO_VECTYPE (stmt_info);
ebfd146a
IR
3566 nunits = (unsigned int) TYPE_VECTOR_SUBPARTS (vectype);
3567 group_size = SLP_INSTANCE_GROUP_SIZE (instance);
3568
3569 /* For each SLP instance calculate number of vector stmts to be created
ff802fa1 3570 for the scalar stmts in each node of the SLP tree. Number of vector
ebfd146a
IR
3571 elements in one vector iteration is the number of scalar elements in
3572 one scalar iteration (GROUP_SIZE) multiplied by VF divided by vector
14a61437
RB
3573 size.
3574 Unless this is a SLP reduction in which case the number of vector
3575 stmts is equal to the number of vector stmts of the children. */
3576 if (GROUP_FIRST_ELEMENT (stmt_info)
3577 && !STMT_VINFO_GROUPED_ACCESS (stmt_info))
3578 vec_stmts_size = SLP_TREE_NUMBER_OF_VEC_STMTS (SLP_TREE_CHILDREN (node)[0]);
3579 else
3580 vec_stmts_size = (vectorization_factor * group_size) / nunits;
ebfd146a 3581
9771b263 3582 if (!SLP_TREE_VEC_STMTS (node).exists ())
ebfd146a 3583 {
9771b263 3584 SLP_TREE_VEC_STMTS (node).create (vec_stmts_size);
ebfd146a
IR
3585 SLP_TREE_NUMBER_OF_VEC_STMTS (node) = vec_stmts_size;
3586 }
3587
73fbfcad 3588 if (dump_enabled_p ())
ebfd146a 3589 {
78c60e3d
SS
3590 dump_printf_loc (MSG_NOTE,vect_location,
3591 "------>vectorizing SLP node starting from: ");
3592 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
b8698a0f 3593 }
ebfd146a 3594
2e8ab70c
RB
3595 /* Vectorized stmts go before the last scalar stmt which is where
3596 all uses are ready. */
3597 si = gsi_for_stmt (vect_find_last_scalar_stmt_in_slp (node));
e4a707c4 3598
b010117a
IR
3599 /* Mark the first element of the reduction chain as reduction to properly
3600 transform the node. In the analysis phase only the last element of the
3601 chain is marked as reduction. */
0d0293ac 3602 if (GROUP_FIRST_ELEMENT (stmt_info) && !STMT_VINFO_GROUPED_ACCESS (stmt_info)
b010117a
IR
3603 && GROUP_FIRST_ELEMENT (stmt_info) == stmt)
3604 {
3605 STMT_VINFO_DEF_TYPE (stmt_info) = vect_reduction_def;
3606 STMT_VINFO_TYPE (stmt_info) = reduc_vec_info_type;
3607 }
3608
6876e5bc
RB
3609 /* Handle two-operation SLP nodes by vectorizing the group with
3610 both operations and then performing a merge. */
3611 if (SLP_TREE_TWO_OPERATORS (node))
3612 {
3613 enum tree_code code0 = gimple_assign_rhs_code (stmt);
567a3691 3614 enum tree_code ocode = ERROR_MARK;
355fe088 3615 gimple *ostmt;
6876e5bc 3616 unsigned char *mask = XALLOCAVEC (unsigned char, group_size);
6876e5bc
RB
3617 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, ostmt)
3618 if (gimple_assign_rhs_code (ostmt) != code0)
3619 {
3620 mask[i] = 1;
6876e5bc
RB
3621 ocode = gimple_assign_rhs_code (ostmt);
3622 }
3623 else
3624 mask[i] = 0;
567a3691 3625 if (ocode != ERROR_MARK)
6876e5bc 3626 {
355fe088
TS
3627 vec<gimple *> v0;
3628 vec<gimple *> v1;
6876e5bc
RB
3629 unsigned j;
3630 tree tmask = NULL_TREE;
3631 vect_transform_stmt (stmt, &si, &grouped_store, node, instance);
3632 v0 = SLP_TREE_VEC_STMTS (node).copy ();
3633 SLP_TREE_VEC_STMTS (node).truncate (0);
3634 gimple_assign_set_rhs_code (stmt, ocode);
3635 vect_transform_stmt (stmt, &si, &grouped_store, node, instance);
3636 gimple_assign_set_rhs_code (stmt, code0);
3637 v1 = SLP_TREE_VEC_STMTS (node).copy ();
3638 SLP_TREE_VEC_STMTS (node).truncate (0);
3639 tree meltype = build_nonstandard_integer_type
3640 (GET_MODE_BITSIZE (TYPE_MODE (TREE_TYPE (vectype))), 1);
3641 tree mvectype = get_same_sized_vectype (meltype, vectype);
3642 unsigned k = 0, l;
3643 for (j = 0; j < v0.length (); ++j)
3644 {
3645 tree *melts = XALLOCAVEC (tree, TYPE_VECTOR_SUBPARTS (vectype));
3646 for (l = 0; l < TYPE_VECTOR_SUBPARTS (vectype); ++l)
3647 {
1ece8d4c 3648 if (k >= group_size)
6876e5bc
RB
3649 k = 0;
3650 melts[l] = build_int_cst
3651 (meltype, mask[k++] * TYPE_VECTOR_SUBPARTS (vectype) + l);
3652 }
3653 tmask = build_vector (mvectype, melts);
3654
3655 /* ??? Not all targets support a VEC_PERM_EXPR with a
3656 constant mask that would translate to a vec_merge RTX
3657 (with their vec_perm_const_ok). We can either not
3658 vectorize in that case or let veclower do its job.
3659 Unfortunately that isn't too great and at least for
3660 plus/minus we'd eventually like to match targets
3661 vector addsub instructions. */
355fe088 3662 gimple *vstmt;
6876e5bc
RB
3663 vstmt = gimple_build_assign (make_ssa_name (vectype),
3664 VEC_PERM_EXPR,
3665 gimple_assign_lhs (v0[j]),
3666 gimple_assign_lhs (v1[j]), tmask);
3667 vect_finish_stmt_generation (stmt, vstmt, &si);
3668 SLP_TREE_VEC_STMTS (node).quick_push (vstmt);
3669 }
3670 v0.release ();
3671 v1.release ();
3672 return false;
3673 }
3674 }
0d0293ac 3675 is_store = vect_transform_stmt (stmt, &si, &grouped_store, node, instance);
603cca93
RB
3676
3677 /* Restore stmt def-types. */
3678 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
3679 if (SLP_TREE_DEF_TYPE (child) != vect_internal_def)
3680 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (child), j, stmt)
3681 STMT_VINFO_DEF_TYPE (vinfo_for_stmt (stmt)) = vect_internal_def;
3682
b5aeb3bb 3683 return is_store;
ebfd146a
IR
3684}
3685
dd34c087
JJ
3686/* Replace scalar calls from SLP node NODE with setting of their lhs to zero.
3687 For loop vectorization this is done in vectorizable_call, but for SLP
3688 it needs to be deferred until end of vect_schedule_slp, because multiple
3689 SLP instances may refer to the same scalar stmt. */
3690
3691static void
3692vect_remove_slp_scalar_calls (slp_tree node)
3693{
355fe088 3694 gimple *stmt, *new_stmt;
dd34c087
JJ
3695 gimple_stmt_iterator gsi;
3696 int i;
d755c7ef 3697 slp_tree child;
dd34c087
JJ
3698 tree lhs;
3699 stmt_vec_info stmt_info;
3700
603cca93 3701 if (SLP_TREE_DEF_TYPE (node) != vect_internal_def)
dd34c087
JJ
3702 return;
3703
9771b263 3704 FOR_EACH_VEC_ELT (SLP_TREE_CHILDREN (node), i, child)
d755c7ef 3705 vect_remove_slp_scalar_calls (child);
dd34c087 3706
9771b263 3707 FOR_EACH_VEC_ELT (SLP_TREE_SCALAR_STMTS (node), i, stmt)
dd34c087
JJ
3708 {
3709 if (!is_gimple_call (stmt) || gimple_bb (stmt) == NULL)
3710 continue;
3711 stmt_info = vinfo_for_stmt (stmt);
3712 if (stmt_info == NULL
3713 || is_pattern_stmt_p (stmt_info)
3714 || !PURE_SLP_STMT (stmt_info))
3715 continue;
3716 lhs = gimple_call_lhs (stmt);
3717 new_stmt = gimple_build_assign (lhs, build_zero_cst (TREE_TYPE (lhs)));
3718 set_vinfo_for_stmt (new_stmt, stmt_info);
3719 set_vinfo_for_stmt (stmt, NULL);
3720 STMT_VINFO_STMT (stmt_info) = new_stmt;
3721 gsi = gsi_for_stmt (stmt);
3722 gsi_replace (&gsi, new_stmt, false);
3723 SSA_NAME_DEF_STMT (gimple_assign_lhs (new_stmt)) = new_stmt;
3724 }
3725}
ebfd146a 3726
ff802fa1
IR
3727/* Generate vector code for all SLP instances in the loop/basic block. */
3728
ebfd146a 3729bool
310213d4 3730vect_schedule_slp (vec_info *vinfo)
ebfd146a 3731{
9771b263 3732 vec<slp_instance> slp_instances;
ebfd146a 3733 slp_instance instance;
01d8bf07 3734 unsigned int i, vf;
ebfd146a
IR
3735 bool is_store = false;
3736
310213d4
RB
3737 slp_instances = vinfo->slp_instances;
3738 if (is_a <loop_vec_info> (vinfo))
3739 vf = as_a <loop_vec_info> (vinfo)->vectorization_factor;
a70d6342 3740 else
310213d4 3741 vf = 1;
a70d6342 3742
9771b263 3743 FOR_EACH_VEC_ELT (slp_instances, i, instance)
ebfd146a
IR
3744 {
3745 /* Schedule the tree of INSTANCE. */
3746 is_store = vect_schedule_slp_instance (SLP_INSTANCE_TREE (instance),
a70d6342 3747 instance, vf);
73fbfcad 3748 if (dump_enabled_p ())
78c60e3d 3749 dump_printf_loc (MSG_NOTE, vect_location,
e645e942 3750 "vectorizing stmts using SLP.\n");
ebfd146a
IR
3751 }
3752
9771b263 3753 FOR_EACH_VEC_ELT (slp_instances, i, instance)
b5aeb3bb
IR
3754 {
3755 slp_tree root = SLP_INSTANCE_TREE (instance);
355fe088 3756 gimple *store;
b5aeb3bb
IR
3757 unsigned int j;
3758 gimple_stmt_iterator gsi;
3759
c40eced0
RB
3760 /* Remove scalar call stmts. Do not do this for basic-block
3761 vectorization as not all uses may be vectorized.
3762 ??? Why should this be necessary? DCE should be able to
3763 remove the stmts itself.
3764 ??? For BB vectorization we can as well remove scalar
3765 stmts starting from the SLP tree root if they have no
3766 uses. */
310213d4 3767 if (is_a <loop_vec_info> (vinfo))
c40eced0 3768 vect_remove_slp_scalar_calls (root);
dd34c087 3769
9771b263 3770 for (j = 0; SLP_TREE_SCALAR_STMTS (root).iterate (j, &store)
b5aeb3bb
IR
3771 && j < SLP_INSTANCE_GROUP_SIZE (instance); j++)
3772 {
3773 if (!STMT_VINFO_DATA_REF (vinfo_for_stmt (store)))
3774 break;
3775
a024e70e
IR
3776 if (is_pattern_stmt_p (vinfo_for_stmt (store)))
3777 store = STMT_VINFO_RELATED_STMT (vinfo_for_stmt (store));
b5aeb3bb
IR
3778 /* Free the attached stmt_vec_info and remove the stmt. */
3779 gsi = gsi_for_stmt (store);
3d3f2249 3780 unlink_stmt_vdef (store);
b5aeb3bb 3781 gsi_remove (&gsi, true);
3d3f2249 3782 release_defs (store);
b5aeb3bb
IR
3783 free_stmt_vec_info (store);
3784 }
3785 }
3786
ebfd146a
IR
3787 return is_store;
3788}