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