fold-const.c (fold_binary_loc): Fix copy-and-pasto.
[gcc.git] / gcc / tree-vect-stmts.c
1 /* Statement Analysis and Transformation for Vectorization
2 Copyright (C) 2003-2014 Free Software Foundation, Inc.
3 Contributed by Dorit Naishlos <dorit@il.ibm.com>
4 and Ira Rosen <irar@il.ibm.com>
5
6 This file is part of GCC.
7
8 GCC is free software; you can redistribute it and/or modify it under
9 the terms of the GNU General Public License as published by the Free
10 Software Foundation; either version 3, or (at your option) any later
11 version.
12
13 GCC is distributed in the hope that it will be useful, but WITHOUT ANY
14 WARRANTY; without even the implied warranty of MERCHANTABILITY or
15 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
16 for more details.
17
18 You should have received a copy of the GNU General Public License
19 along with GCC; see the file COPYING3. If not see
20 <http://www.gnu.org/licenses/>. */
21
22 #include "config.h"
23 #include "system.h"
24 #include "coretypes.h"
25 #include "dumpfile.h"
26 #include "tm.h"
27 #include "tree.h"
28 #include "stor-layout.h"
29 #include "target.h"
30 #include "basic-block.h"
31 #include "gimple-pretty-print.h"
32 #include "tree-ssa-alias.h"
33 #include "internal-fn.h"
34 #include "tree-eh.h"
35 #include "gimple-expr.h"
36 #include "is-a.h"
37 #include "gimple.h"
38 #include "gimplify.h"
39 #include "gimple-iterator.h"
40 #include "gimplify-me.h"
41 #include "gimple-ssa.h"
42 #include "tree-cfg.h"
43 #include "tree-phinodes.h"
44 #include "ssa-iterators.h"
45 #include "stringpool.h"
46 #include "tree-ssanames.h"
47 #include "tree-ssa-loop-manip.h"
48 #include "cfgloop.h"
49 #include "tree-ssa-loop.h"
50 #include "tree-scalar-evolution.h"
51 #include "expr.h"
52 #include "recog.h" /* FIXME: for insn_data */
53 #include "optabs.h"
54 #include "diagnostic-core.h"
55 #include "tree-vectorizer.h"
56 #include "dumpfile.h"
57 #include "cgraph.h"
58 #include "builtins.h"
59
60 /* For lang_hooks.types.type_for_mode. */
61 #include "langhooks.h"
62
63 /* Return the vectorized type for the given statement. */
64
65 tree
66 stmt_vectype (struct _stmt_vec_info *stmt_info)
67 {
68 return STMT_VINFO_VECTYPE (stmt_info);
69 }
70
71 /* Return TRUE iff the given statement is in an inner loop relative to
72 the loop being vectorized. */
73 bool
74 stmt_in_inner_loop_p (struct _stmt_vec_info *stmt_info)
75 {
76 gimple stmt = STMT_VINFO_STMT (stmt_info);
77 basic_block bb = gimple_bb (stmt);
78 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
79 struct loop* loop;
80
81 if (!loop_vinfo)
82 return false;
83
84 loop = LOOP_VINFO_LOOP (loop_vinfo);
85
86 return (bb->loop_father == loop->inner);
87 }
88
89 /* Record the cost of a statement, either by directly informing the
90 target model or by saving it in a vector for later processing.
91 Return a preliminary estimate of the statement's cost. */
92
93 unsigned
94 record_stmt_cost (stmt_vector_for_cost *body_cost_vec, int count,
95 enum vect_cost_for_stmt kind, stmt_vec_info stmt_info,
96 int misalign, enum vect_cost_model_location where)
97 {
98 if (body_cost_vec)
99 {
100 tree vectype = stmt_info ? stmt_vectype (stmt_info) : NULL_TREE;
101 add_stmt_info_to_vec (body_cost_vec, count, kind,
102 stmt_info ? STMT_VINFO_STMT (stmt_info) : NULL,
103 misalign);
104 return (unsigned)
105 (builtin_vectorization_cost (kind, vectype, misalign) * count);
106
107 }
108 else
109 {
110 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
111 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
112 void *target_cost_data;
113
114 if (loop_vinfo)
115 target_cost_data = LOOP_VINFO_TARGET_COST_DATA (loop_vinfo);
116 else
117 target_cost_data = BB_VINFO_TARGET_COST_DATA (bb_vinfo);
118
119 return add_stmt_cost (target_cost_data, count, kind, stmt_info,
120 misalign, where);
121 }
122 }
123
124 /* Return a variable of type ELEM_TYPE[NELEMS]. */
125
126 static tree
127 create_vector_array (tree elem_type, unsigned HOST_WIDE_INT nelems)
128 {
129 return create_tmp_var (build_array_type_nelts (elem_type, nelems),
130 "vect_array");
131 }
132
133 /* ARRAY is an array of vectors created by create_vector_array.
134 Return an SSA_NAME for the vector in index N. The reference
135 is part of the vectorization of STMT and the vector is associated
136 with scalar destination SCALAR_DEST. */
137
138 static tree
139 read_vector_array (gimple stmt, gimple_stmt_iterator *gsi, tree scalar_dest,
140 tree array, unsigned HOST_WIDE_INT n)
141 {
142 tree vect_type, vect, vect_name, array_ref;
143 gimple new_stmt;
144
145 gcc_assert (TREE_CODE (TREE_TYPE (array)) == ARRAY_TYPE);
146 vect_type = TREE_TYPE (TREE_TYPE (array));
147 vect = vect_create_destination_var (scalar_dest, vect_type);
148 array_ref = build4 (ARRAY_REF, vect_type, array,
149 build_int_cst (size_type_node, n),
150 NULL_TREE, NULL_TREE);
151
152 new_stmt = gimple_build_assign (vect, array_ref);
153 vect_name = make_ssa_name (vect, new_stmt);
154 gimple_assign_set_lhs (new_stmt, vect_name);
155 vect_finish_stmt_generation (stmt, new_stmt, gsi);
156
157 return vect_name;
158 }
159
160 /* ARRAY is an array of vectors created by create_vector_array.
161 Emit code to store SSA_NAME VECT in index N of the array.
162 The store is part of the vectorization of STMT. */
163
164 static void
165 write_vector_array (gimple stmt, gimple_stmt_iterator *gsi, tree vect,
166 tree array, unsigned HOST_WIDE_INT n)
167 {
168 tree array_ref;
169 gimple new_stmt;
170
171 array_ref = build4 (ARRAY_REF, TREE_TYPE (vect), array,
172 build_int_cst (size_type_node, n),
173 NULL_TREE, NULL_TREE);
174
175 new_stmt = gimple_build_assign (array_ref, vect);
176 vect_finish_stmt_generation (stmt, new_stmt, gsi);
177 }
178
179 /* PTR is a pointer to an array of type TYPE. Return a representation
180 of *PTR. The memory reference replaces those in FIRST_DR
181 (and its group). */
182
183 static tree
184 create_array_ref (tree type, tree ptr, struct data_reference *first_dr)
185 {
186 tree mem_ref, alias_ptr_type;
187
188 alias_ptr_type = reference_alias_ptr_type (DR_REF (first_dr));
189 mem_ref = build2 (MEM_REF, type, ptr, build_int_cst (alias_ptr_type, 0));
190 /* Arrays have the same alignment as their type. */
191 set_ptr_info_alignment (get_ptr_info (ptr), TYPE_ALIGN_UNIT (type), 0);
192 return mem_ref;
193 }
194
195 /* Utility functions used by vect_mark_stmts_to_be_vectorized. */
196
197 /* Function vect_mark_relevant.
198
199 Mark STMT as "relevant for vectorization" and add it to WORKLIST. */
200
201 static void
202 vect_mark_relevant (vec<gimple> *worklist, gimple stmt,
203 enum vect_relevant relevant, bool live_p,
204 bool used_in_pattern)
205 {
206 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
207 enum vect_relevant save_relevant = STMT_VINFO_RELEVANT (stmt_info);
208 bool save_live_p = STMT_VINFO_LIVE_P (stmt_info);
209 gimple pattern_stmt;
210
211 if (dump_enabled_p ())
212 dump_printf_loc (MSG_NOTE, vect_location,
213 "mark relevant %d, live %d.\n", relevant, live_p);
214
215 /* If this stmt is an original stmt in a pattern, we might need to mark its
216 related pattern stmt instead of the original stmt. However, such stmts
217 may have their own uses that are not in any pattern, in such cases the
218 stmt itself should be marked. */
219 if (STMT_VINFO_IN_PATTERN_P (stmt_info))
220 {
221 bool found = false;
222 if (!used_in_pattern)
223 {
224 imm_use_iterator imm_iter;
225 use_operand_p use_p;
226 gimple use_stmt;
227 tree lhs;
228 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
229 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
230
231 if (is_gimple_assign (stmt))
232 lhs = gimple_assign_lhs (stmt);
233 else
234 lhs = gimple_call_lhs (stmt);
235
236 /* This use is out of pattern use, if LHS has other uses that are
237 pattern uses, we should mark the stmt itself, and not the pattern
238 stmt. */
239 if (lhs && TREE_CODE (lhs) == SSA_NAME)
240 FOR_EACH_IMM_USE_FAST (use_p, imm_iter, lhs)
241 {
242 if (is_gimple_debug (USE_STMT (use_p)))
243 continue;
244 use_stmt = USE_STMT (use_p);
245
246 if (!flow_bb_inside_loop_p (loop, gimple_bb (use_stmt)))
247 continue;
248
249 if (vinfo_for_stmt (use_stmt)
250 && STMT_VINFO_IN_PATTERN_P (vinfo_for_stmt (use_stmt)))
251 {
252 found = true;
253 break;
254 }
255 }
256 }
257
258 if (!found)
259 {
260 /* This is the last stmt in a sequence that was detected as a
261 pattern that can potentially be vectorized. Don't mark the stmt
262 as relevant/live because it's not going to be vectorized.
263 Instead mark the pattern-stmt that replaces it. */
264
265 pattern_stmt = STMT_VINFO_RELATED_STMT (stmt_info);
266
267 if (dump_enabled_p ())
268 dump_printf_loc (MSG_NOTE, vect_location,
269 "last stmt in pattern. don't mark"
270 " relevant/live.\n");
271 stmt_info = vinfo_for_stmt (pattern_stmt);
272 gcc_assert (STMT_VINFO_RELATED_STMT (stmt_info) == stmt);
273 save_relevant = STMT_VINFO_RELEVANT (stmt_info);
274 save_live_p = STMT_VINFO_LIVE_P (stmt_info);
275 stmt = pattern_stmt;
276 }
277 }
278
279 STMT_VINFO_LIVE_P (stmt_info) |= live_p;
280 if (relevant > STMT_VINFO_RELEVANT (stmt_info))
281 STMT_VINFO_RELEVANT (stmt_info) = relevant;
282
283 if (STMT_VINFO_RELEVANT (stmt_info) == save_relevant
284 && STMT_VINFO_LIVE_P (stmt_info) == save_live_p)
285 {
286 if (dump_enabled_p ())
287 dump_printf_loc (MSG_NOTE, vect_location,
288 "already marked relevant/live.\n");
289 return;
290 }
291
292 worklist->safe_push (stmt);
293 }
294
295
296 /* Function vect_stmt_relevant_p.
297
298 Return true if STMT in loop that is represented by LOOP_VINFO is
299 "relevant for vectorization".
300
301 A stmt is considered "relevant for vectorization" if:
302 - it has uses outside the loop.
303 - it has vdefs (it alters memory).
304 - control stmts in the loop (except for the exit condition).
305
306 CHECKME: what other side effects would the vectorizer allow? */
307
308 static bool
309 vect_stmt_relevant_p (gimple stmt, loop_vec_info loop_vinfo,
310 enum vect_relevant *relevant, bool *live_p)
311 {
312 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
313 ssa_op_iter op_iter;
314 imm_use_iterator imm_iter;
315 use_operand_p use_p;
316 def_operand_p def_p;
317
318 *relevant = vect_unused_in_scope;
319 *live_p = false;
320
321 /* cond stmt other than loop exit cond. */
322 if (is_ctrl_stmt (stmt)
323 && STMT_VINFO_TYPE (vinfo_for_stmt (stmt))
324 != loop_exit_ctrl_vec_info_type)
325 *relevant = vect_used_in_scope;
326
327 /* changing memory. */
328 if (gimple_code (stmt) != GIMPLE_PHI)
329 if (gimple_vdef (stmt))
330 {
331 if (dump_enabled_p ())
332 dump_printf_loc (MSG_NOTE, vect_location,
333 "vec_stmt_relevant_p: stmt has vdefs.\n");
334 *relevant = vect_used_in_scope;
335 }
336
337 /* uses outside the loop. */
338 FOR_EACH_PHI_OR_STMT_DEF (def_p, stmt, op_iter, SSA_OP_DEF)
339 {
340 FOR_EACH_IMM_USE_FAST (use_p, imm_iter, DEF_FROM_PTR (def_p))
341 {
342 basic_block bb = gimple_bb (USE_STMT (use_p));
343 if (!flow_bb_inside_loop_p (loop, bb))
344 {
345 if (dump_enabled_p ())
346 dump_printf_loc (MSG_NOTE, vect_location,
347 "vec_stmt_relevant_p: used out of loop.\n");
348
349 if (is_gimple_debug (USE_STMT (use_p)))
350 continue;
351
352 /* We expect all such uses to be in the loop exit phis
353 (because of loop closed form) */
354 gcc_assert (gimple_code (USE_STMT (use_p)) == GIMPLE_PHI);
355 gcc_assert (bb == single_exit (loop)->dest);
356
357 *live_p = true;
358 }
359 }
360 }
361
362 return (*live_p || *relevant);
363 }
364
365
366 /* Function exist_non_indexing_operands_for_use_p
367
368 USE is one of the uses attached to STMT. Check if USE is
369 used in STMT for anything other than indexing an array. */
370
371 static bool
372 exist_non_indexing_operands_for_use_p (tree use, gimple stmt)
373 {
374 tree operand;
375 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
376
377 /* USE corresponds to some operand in STMT. If there is no data
378 reference in STMT, then any operand that corresponds to USE
379 is not indexing an array. */
380 if (!STMT_VINFO_DATA_REF (stmt_info))
381 return true;
382
383 /* STMT has a data_ref. FORNOW this means that its of one of
384 the following forms:
385 -1- ARRAY_REF = var
386 -2- var = ARRAY_REF
387 (This should have been verified in analyze_data_refs).
388
389 'var' in the second case corresponds to a def, not a use,
390 so USE cannot correspond to any operands that are not used
391 for array indexing.
392
393 Therefore, all we need to check is if STMT falls into the
394 first case, and whether var corresponds to USE. */
395
396 if (!gimple_assign_copy_p (stmt))
397 {
398 if (is_gimple_call (stmt)
399 && gimple_call_internal_p (stmt))
400 switch (gimple_call_internal_fn (stmt))
401 {
402 case IFN_MASK_STORE:
403 operand = gimple_call_arg (stmt, 3);
404 if (operand == use)
405 return true;
406 /* FALLTHRU */
407 case IFN_MASK_LOAD:
408 operand = gimple_call_arg (stmt, 2);
409 if (operand == use)
410 return true;
411 break;
412 default:
413 break;
414 }
415 return false;
416 }
417
418 if (TREE_CODE (gimple_assign_lhs (stmt)) == SSA_NAME)
419 return false;
420 operand = gimple_assign_rhs1 (stmt);
421 if (TREE_CODE (operand) != SSA_NAME)
422 return false;
423
424 if (operand == use)
425 return true;
426
427 return false;
428 }
429
430
431 /*
432 Function process_use.
433
434 Inputs:
435 - a USE in STMT in a loop represented by LOOP_VINFO
436 - LIVE_P, RELEVANT - enum values to be set in the STMT_VINFO of the stmt
437 that defined USE. This is done by calling mark_relevant and passing it
438 the WORKLIST (to add DEF_STMT to the WORKLIST in case it is relevant).
439 - FORCE is true if exist_non_indexing_operands_for_use_p check shouldn't
440 be performed.
441
442 Outputs:
443 Generally, LIVE_P and RELEVANT are used to define the liveness and
444 relevance info of the DEF_STMT of this USE:
445 STMT_VINFO_LIVE_P (DEF_STMT_info) <-- live_p
446 STMT_VINFO_RELEVANT (DEF_STMT_info) <-- relevant
447 Exceptions:
448 - case 1: If USE is used only for address computations (e.g. array indexing),
449 which does not need to be directly vectorized, then the liveness/relevance
450 of the respective DEF_STMT is left unchanged.
451 - case 2: If STMT is a reduction phi and DEF_STMT is a reduction stmt, we
452 skip DEF_STMT cause it had already been processed.
453 - case 3: If DEF_STMT and STMT are in different nests, then "relevant" will
454 be modified accordingly.
455
456 Return true if everything is as expected. Return false otherwise. */
457
458 static bool
459 process_use (gimple stmt, tree use, loop_vec_info loop_vinfo, bool live_p,
460 enum vect_relevant relevant, vec<gimple> *worklist,
461 bool force)
462 {
463 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
464 stmt_vec_info stmt_vinfo = vinfo_for_stmt (stmt);
465 stmt_vec_info dstmt_vinfo;
466 basic_block bb, def_bb;
467 tree def;
468 gimple def_stmt;
469 enum vect_def_type dt;
470
471 /* case 1: we are only interested in uses that need to be vectorized. Uses
472 that are used for address computation are not considered relevant. */
473 if (!force && !exist_non_indexing_operands_for_use_p (use, stmt))
474 return true;
475
476 if (!vect_is_simple_use (use, stmt, loop_vinfo, NULL, &def_stmt, &def, &dt))
477 {
478 if (dump_enabled_p ())
479 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
480 "not vectorized: unsupported use in stmt.\n");
481 return false;
482 }
483
484 if (!def_stmt || gimple_nop_p (def_stmt))
485 return true;
486
487 def_bb = gimple_bb (def_stmt);
488 if (!flow_bb_inside_loop_p (loop, def_bb))
489 {
490 if (dump_enabled_p ())
491 dump_printf_loc (MSG_NOTE, vect_location, "def_stmt is out of loop.\n");
492 return true;
493 }
494
495 /* case 2: A reduction phi (STMT) defined by a reduction stmt (DEF_STMT).
496 DEF_STMT must have already been processed, because this should be the
497 only way that STMT, which is a reduction-phi, was put in the worklist,
498 as there should be no other uses for DEF_STMT in the loop. So we just
499 check that everything is as expected, and we are done. */
500 dstmt_vinfo = vinfo_for_stmt (def_stmt);
501 bb = gimple_bb (stmt);
502 if (gimple_code (stmt) == GIMPLE_PHI
503 && STMT_VINFO_DEF_TYPE (stmt_vinfo) == vect_reduction_def
504 && gimple_code (def_stmt) != GIMPLE_PHI
505 && STMT_VINFO_DEF_TYPE (dstmt_vinfo) == vect_reduction_def
506 && bb->loop_father == def_bb->loop_father)
507 {
508 if (dump_enabled_p ())
509 dump_printf_loc (MSG_NOTE, vect_location,
510 "reduc-stmt defining reduc-phi in the same nest.\n");
511 if (STMT_VINFO_IN_PATTERN_P (dstmt_vinfo))
512 dstmt_vinfo = vinfo_for_stmt (STMT_VINFO_RELATED_STMT (dstmt_vinfo));
513 gcc_assert (STMT_VINFO_RELEVANT (dstmt_vinfo) < vect_used_by_reduction);
514 gcc_assert (STMT_VINFO_LIVE_P (dstmt_vinfo)
515 || STMT_VINFO_RELEVANT (dstmt_vinfo) > vect_unused_in_scope);
516 return true;
517 }
518
519 /* case 3a: outer-loop stmt defining an inner-loop stmt:
520 outer-loop-header-bb:
521 d = def_stmt
522 inner-loop:
523 stmt # use (d)
524 outer-loop-tail-bb:
525 ... */
526 if (flow_loop_nested_p (def_bb->loop_father, bb->loop_father))
527 {
528 if (dump_enabled_p ())
529 dump_printf_loc (MSG_NOTE, vect_location,
530 "outer-loop def-stmt defining inner-loop stmt.\n");
531
532 switch (relevant)
533 {
534 case vect_unused_in_scope:
535 relevant = (STMT_VINFO_DEF_TYPE (stmt_vinfo) == vect_nested_cycle) ?
536 vect_used_in_scope : vect_unused_in_scope;
537 break;
538
539 case vect_used_in_outer_by_reduction:
540 gcc_assert (STMT_VINFO_DEF_TYPE (stmt_vinfo) != vect_reduction_def);
541 relevant = vect_used_by_reduction;
542 break;
543
544 case vect_used_in_outer:
545 gcc_assert (STMT_VINFO_DEF_TYPE (stmt_vinfo) != vect_reduction_def);
546 relevant = vect_used_in_scope;
547 break;
548
549 case vect_used_in_scope:
550 break;
551
552 default:
553 gcc_unreachable ();
554 }
555 }
556
557 /* case 3b: inner-loop stmt defining an outer-loop stmt:
558 outer-loop-header-bb:
559 ...
560 inner-loop:
561 d = def_stmt
562 outer-loop-tail-bb (or outer-loop-exit-bb in double reduction):
563 stmt # use (d) */
564 else if (flow_loop_nested_p (bb->loop_father, def_bb->loop_father))
565 {
566 if (dump_enabled_p ())
567 dump_printf_loc (MSG_NOTE, vect_location,
568 "inner-loop def-stmt defining outer-loop stmt.\n");
569
570 switch (relevant)
571 {
572 case vect_unused_in_scope:
573 relevant = (STMT_VINFO_DEF_TYPE (stmt_vinfo) == vect_reduction_def
574 || STMT_VINFO_DEF_TYPE (stmt_vinfo) == vect_double_reduction_def) ?
575 vect_used_in_outer_by_reduction : vect_unused_in_scope;
576 break;
577
578 case vect_used_by_reduction:
579 relevant = vect_used_in_outer_by_reduction;
580 break;
581
582 case vect_used_in_scope:
583 relevant = vect_used_in_outer;
584 break;
585
586 default:
587 gcc_unreachable ();
588 }
589 }
590
591 vect_mark_relevant (worklist, def_stmt, relevant, live_p,
592 is_pattern_stmt_p (stmt_vinfo));
593 return true;
594 }
595
596
597 /* Function vect_mark_stmts_to_be_vectorized.
598
599 Not all stmts in the loop need to be vectorized. For example:
600
601 for i...
602 for j...
603 1. T0 = i + j
604 2. T1 = a[T0]
605
606 3. j = j + 1
607
608 Stmt 1 and 3 do not need to be vectorized, because loop control and
609 addressing of vectorized data-refs are handled differently.
610
611 This pass detects such stmts. */
612
613 bool
614 vect_mark_stmts_to_be_vectorized (loop_vec_info loop_vinfo)
615 {
616 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
617 basic_block *bbs = LOOP_VINFO_BBS (loop_vinfo);
618 unsigned int nbbs = loop->num_nodes;
619 gimple_stmt_iterator si;
620 gimple stmt;
621 unsigned int i;
622 stmt_vec_info stmt_vinfo;
623 basic_block bb;
624 gimple phi;
625 bool live_p;
626 enum vect_relevant relevant, tmp_relevant;
627 enum vect_def_type def_type;
628
629 if (dump_enabled_p ())
630 dump_printf_loc (MSG_NOTE, vect_location,
631 "=== vect_mark_stmts_to_be_vectorized ===\n");
632
633 auto_vec<gimple, 64> worklist;
634
635 /* 1. Init worklist. */
636 for (i = 0; i < nbbs; i++)
637 {
638 bb = bbs[i];
639 for (si = gsi_start_phis (bb); !gsi_end_p (si); gsi_next (&si))
640 {
641 phi = gsi_stmt (si);
642 if (dump_enabled_p ())
643 {
644 dump_printf_loc (MSG_NOTE, vect_location, "init: phi relevant? ");
645 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, phi, 0);
646 dump_printf (MSG_NOTE, "\n");
647 }
648
649 if (vect_stmt_relevant_p (phi, loop_vinfo, &relevant, &live_p))
650 vect_mark_relevant (&worklist, phi, relevant, live_p, false);
651 }
652 for (si = gsi_start_bb (bb); !gsi_end_p (si); gsi_next (&si))
653 {
654 stmt = gsi_stmt (si);
655 if (dump_enabled_p ())
656 {
657 dump_printf_loc (MSG_NOTE, vect_location, "init: stmt relevant? ");
658 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
659 dump_printf (MSG_NOTE, "\n");
660 }
661
662 if (vect_stmt_relevant_p (stmt, loop_vinfo, &relevant, &live_p))
663 vect_mark_relevant (&worklist, stmt, relevant, live_p, false);
664 }
665 }
666
667 /* 2. Process_worklist */
668 while (worklist.length () > 0)
669 {
670 use_operand_p use_p;
671 ssa_op_iter iter;
672
673 stmt = worklist.pop ();
674 if (dump_enabled_p ())
675 {
676 dump_printf_loc (MSG_NOTE, vect_location, "worklist: examine stmt: ");
677 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
678 dump_printf (MSG_NOTE, "\n");
679 }
680
681 /* Examine the USEs of STMT. For each USE, mark the stmt that defines it
682 (DEF_STMT) as relevant/irrelevant and live/dead according to the
683 liveness and relevance properties of STMT. */
684 stmt_vinfo = vinfo_for_stmt (stmt);
685 relevant = STMT_VINFO_RELEVANT (stmt_vinfo);
686 live_p = STMT_VINFO_LIVE_P (stmt_vinfo);
687
688 /* Generally, the liveness and relevance properties of STMT are
689 propagated as is to the DEF_STMTs of its USEs:
690 live_p <-- STMT_VINFO_LIVE_P (STMT_VINFO)
691 relevant <-- STMT_VINFO_RELEVANT (STMT_VINFO)
692
693 One exception is when STMT has been identified as defining a reduction
694 variable; in this case we set the liveness/relevance as follows:
695 live_p = false
696 relevant = vect_used_by_reduction
697 This is because we distinguish between two kinds of relevant stmts -
698 those that are used by a reduction computation, and those that are
699 (also) used by a regular computation. This allows us later on to
700 identify stmts that are used solely by a reduction, and therefore the
701 order of the results that they produce does not have to be kept. */
702
703 def_type = STMT_VINFO_DEF_TYPE (stmt_vinfo);
704 tmp_relevant = relevant;
705 switch (def_type)
706 {
707 case vect_reduction_def:
708 switch (tmp_relevant)
709 {
710 case vect_unused_in_scope:
711 relevant = vect_used_by_reduction;
712 break;
713
714 case vect_used_by_reduction:
715 if (gimple_code (stmt) == GIMPLE_PHI)
716 break;
717 /* fall through */
718
719 default:
720 if (dump_enabled_p ())
721 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
722 "unsupported use of reduction.\n");
723 return false;
724 }
725
726 live_p = false;
727 break;
728
729 case vect_nested_cycle:
730 if (tmp_relevant != vect_unused_in_scope
731 && tmp_relevant != vect_used_in_outer_by_reduction
732 && tmp_relevant != vect_used_in_outer)
733 {
734 if (dump_enabled_p ())
735 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
736 "unsupported use of nested cycle.\n");
737
738 return false;
739 }
740
741 live_p = false;
742 break;
743
744 case vect_double_reduction_def:
745 if (tmp_relevant != vect_unused_in_scope
746 && tmp_relevant != vect_used_by_reduction)
747 {
748 if (dump_enabled_p ())
749 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
750 "unsupported use of double reduction.\n");
751
752 return false;
753 }
754
755 live_p = false;
756 break;
757
758 default:
759 break;
760 }
761
762 if (is_pattern_stmt_p (stmt_vinfo))
763 {
764 /* Pattern statements are not inserted into the code, so
765 FOR_EACH_PHI_OR_STMT_USE optimizes their operands out, and we
766 have to scan the RHS or function arguments instead. */
767 if (is_gimple_assign (stmt))
768 {
769 enum tree_code rhs_code = gimple_assign_rhs_code (stmt);
770 tree op = gimple_assign_rhs1 (stmt);
771
772 i = 1;
773 if (rhs_code == COND_EXPR && COMPARISON_CLASS_P (op))
774 {
775 if (!process_use (stmt, TREE_OPERAND (op, 0), loop_vinfo,
776 live_p, relevant, &worklist, false)
777 || !process_use (stmt, TREE_OPERAND (op, 1), loop_vinfo,
778 live_p, relevant, &worklist, false))
779 return false;
780 i = 2;
781 }
782 for (; i < gimple_num_ops (stmt); i++)
783 {
784 op = gimple_op (stmt, i);
785 if (!process_use (stmt, op, loop_vinfo, live_p, relevant,
786 &worklist, false))
787 return false;
788 }
789 }
790 else if (is_gimple_call (stmt))
791 {
792 for (i = 0; i < gimple_call_num_args (stmt); i++)
793 {
794 tree arg = gimple_call_arg (stmt, i);
795 if (!process_use (stmt, arg, loop_vinfo, live_p, relevant,
796 &worklist, false))
797 return false;
798 }
799 }
800 }
801 else
802 FOR_EACH_PHI_OR_STMT_USE (use_p, stmt, iter, SSA_OP_USE)
803 {
804 tree op = USE_FROM_PTR (use_p);
805 if (!process_use (stmt, op, loop_vinfo, live_p, relevant,
806 &worklist, false))
807 return false;
808 }
809
810 if (STMT_VINFO_GATHER_P (stmt_vinfo))
811 {
812 tree off;
813 tree decl = vect_check_gather (stmt, loop_vinfo, NULL, &off, NULL);
814 gcc_assert (decl);
815 if (!process_use (stmt, off, loop_vinfo, live_p, relevant,
816 &worklist, true))
817 return false;
818 }
819 } /* while worklist */
820
821 return true;
822 }
823
824
825 /* Function vect_model_simple_cost.
826
827 Models cost for simple operations, i.e. those that only emit ncopies of a
828 single op. Right now, this does not account for multiple insns that could
829 be generated for the single vector op. We will handle that shortly. */
830
831 void
832 vect_model_simple_cost (stmt_vec_info stmt_info, int ncopies,
833 enum vect_def_type *dt,
834 stmt_vector_for_cost *prologue_cost_vec,
835 stmt_vector_for_cost *body_cost_vec)
836 {
837 int i;
838 int inside_cost = 0, prologue_cost = 0;
839
840 /* The SLP costs were already calculated during SLP tree build. */
841 if (PURE_SLP_STMT (stmt_info))
842 return;
843
844 /* FORNOW: Assuming maximum 2 args per stmts. */
845 for (i = 0; i < 2; i++)
846 if (dt[i] == vect_constant_def || dt[i] == vect_external_def)
847 prologue_cost += record_stmt_cost (prologue_cost_vec, 1, vector_stmt,
848 stmt_info, 0, vect_prologue);
849
850 /* Pass the inside-of-loop statements to the target-specific cost model. */
851 inside_cost = record_stmt_cost (body_cost_vec, ncopies, vector_stmt,
852 stmt_info, 0, vect_body);
853
854 if (dump_enabled_p ())
855 dump_printf_loc (MSG_NOTE, vect_location,
856 "vect_model_simple_cost: inside_cost = %d, "
857 "prologue_cost = %d .\n", inside_cost, prologue_cost);
858 }
859
860
861 /* Model cost for type demotion and promotion operations. PWR is normally
862 zero for single-step promotions and demotions. It will be one if
863 two-step promotion/demotion is required, and so on. Each additional
864 step doubles the number of instructions required. */
865
866 static void
867 vect_model_promotion_demotion_cost (stmt_vec_info stmt_info,
868 enum vect_def_type *dt, int pwr)
869 {
870 int i, tmp;
871 int inside_cost = 0, prologue_cost = 0;
872 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
873 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
874 void *target_cost_data;
875
876 /* The SLP costs were already calculated during SLP tree build. */
877 if (PURE_SLP_STMT (stmt_info))
878 return;
879
880 if (loop_vinfo)
881 target_cost_data = LOOP_VINFO_TARGET_COST_DATA (loop_vinfo);
882 else
883 target_cost_data = BB_VINFO_TARGET_COST_DATA (bb_vinfo);
884
885 for (i = 0; i < pwr + 1; i++)
886 {
887 tmp = (STMT_VINFO_TYPE (stmt_info) == type_promotion_vec_info_type) ?
888 (i + 1) : i;
889 inside_cost += add_stmt_cost (target_cost_data, vect_pow2 (tmp),
890 vec_promote_demote, stmt_info, 0,
891 vect_body);
892 }
893
894 /* FORNOW: Assuming maximum 2 args per stmts. */
895 for (i = 0; i < 2; i++)
896 if (dt[i] == vect_constant_def || dt[i] == vect_external_def)
897 prologue_cost += add_stmt_cost (target_cost_data, 1, vector_stmt,
898 stmt_info, 0, vect_prologue);
899
900 if (dump_enabled_p ())
901 dump_printf_loc (MSG_NOTE, vect_location,
902 "vect_model_promotion_demotion_cost: inside_cost = %d, "
903 "prologue_cost = %d .\n", inside_cost, prologue_cost);
904 }
905
906 /* Function vect_cost_group_size
907
908 For grouped load or store, return the group_size only if it is the first
909 load or store of a group, else return 1. This ensures that group size is
910 only returned once per group. */
911
912 static int
913 vect_cost_group_size (stmt_vec_info stmt_info)
914 {
915 gimple first_stmt = GROUP_FIRST_ELEMENT (stmt_info);
916
917 if (first_stmt == STMT_VINFO_STMT (stmt_info))
918 return GROUP_SIZE (stmt_info);
919
920 return 1;
921 }
922
923
924 /* Function vect_model_store_cost
925
926 Models cost for stores. In the case of grouped accesses, one access
927 has the overhead of the grouped access attributed to it. */
928
929 void
930 vect_model_store_cost (stmt_vec_info stmt_info, int ncopies,
931 bool store_lanes_p, enum vect_def_type dt,
932 slp_tree slp_node,
933 stmt_vector_for_cost *prologue_cost_vec,
934 stmt_vector_for_cost *body_cost_vec)
935 {
936 int group_size;
937 unsigned int inside_cost = 0, prologue_cost = 0;
938 struct data_reference *first_dr;
939 gimple first_stmt;
940
941 /* The SLP costs were already calculated during SLP tree build. */
942 if (PURE_SLP_STMT (stmt_info))
943 return;
944
945 if (dt == vect_constant_def || dt == vect_external_def)
946 prologue_cost += record_stmt_cost (prologue_cost_vec, 1, scalar_to_vec,
947 stmt_info, 0, vect_prologue);
948
949 /* Grouped access? */
950 if (STMT_VINFO_GROUPED_ACCESS (stmt_info))
951 {
952 if (slp_node)
953 {
954 first_stmt = SLP_TREE_SCALAR_STMTS (slp_node)[0];
955 group_size = 1;
956 }
957 else
958 {
959 first_stmt = GROUP_FIRST_ELEMENT (stmt_info);
960 group_size = vect_cost_group_size (stmt_info);
961 }
962
963 first_dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (first_stmt));
964 }
965 /* Not a grouped access. */
966 else
967 {
968 group_size = 1;
969 first_dr = STMT_VINFO_DATA_REF (stmt_info);
970 }
971
972 /* We assume that the cost of a single store-lanes instruction is
973 equivalent to the cost of GROUP_SIZE separate stores. If a grouped
974 access is instead being provided by a permute-and-store operation,
975 include the cost of the permutes. */
976 if (!store_lanes_p && group_size > 1)
977 {
978 /* Uses a high and low interleave or shuffle operations for each
979 needed permute. */
980 int nstmts = ncopies * ceil_log2 (group_size) * group_size;
981 inside_cost = record_stmt_cost (body_cost_vec, nstmts, vec_perm,
982 stmt_info, 0, vect_body);
983
984 if (dump_enabled_p ())
985 dump_printf_loc (MSG_NOTE, vect_location,
986 "vect_model_store_cost: strided group_size = %d .\n",
987 group_size);
988 }
989
990 /* Costs of the stores. */
991 vect_get_store_cost (first_dr, ncopies, &inside_cost, body_cost_vec);
992
993 if (dump_enabled_p ())
994 dump_printf_loc (MSG_NOTE, vect_location,
995 "vect_model_store_cost: inside_cost = %d, "
996 "prologue_cost = %d .\n", inside_cost, prologue_cost);
997 }
998
999
1000 /* Calculate cost of DR's memory access. */
1001 void
1002 vect_get_store_cost (struct data_reference *dr, int ncopies,
1003 unsigned int *inside_cost,
1004 stmt_vector_for_cost *body_cost_vec)
1005 {
1006 int alignment_support_scheme = vect_supportable_dr_alignment (dr, false);
1007 gimple stmt = DR_STMT (dr);
1008 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
1009
1010 switch (alignment_support_scheme)
1011 {
1012 case dr_aligned:
1013 {
1014 *inside_cost += record_stmt_cost (body_cost_vec, ncopies,
1015 vector_store, stmt_info, 0,
1016 vect_body);
1017
1018 if (dump_enabled_p ())
1019 dump_printf_loc (MSG_NOTE, vect_location,
1020 "vect_model_store_cost: aligned.\n");
1021 break;
1022 }
1023
1024 case dr_unaligned_supported:
1025 {
1026 /* Here, we assign an additional cost for the unaligned store. */
1027 *inside_cost += record_stmt_cost (body_cost_vec, ncopies,
1028 unaligned_store, stmt_info,
1029 DR_MISALIGNMENT (dr), vect_body);
1030 if (dump_enabled_p ())
1031 dump_printf_loc (MSG_NOTE, vect_location,
1032 "vect_model_store_cost: unaligned supported by "
1033 "hardware.\n");
1034 break;
1035 }
1036
1037 case dr_unaligned_unsupported:
1038 {
1039 *inside_cost = VECT_MAX_COST;
1040
1041 if (dump_enabled_p ())
1042 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
1043 "vect_model_store_cost: unsupported access.\n");
1044 break;
1045 }
1046
1047 default:
1048 gcc_unreachable ();
1049 }
1050 }
1051
1052
1053 /* Function vect_model_load_cost
1054
1055 Models cost for loads. In the case of grouped accesses, the last access
1056 has the overhead of the grouped access attributed to it. Since unaligned
1057 accesses are supported for loads, we also account for the costs of the
1058 access scheme chosen. */
1059
1060 void
1061 vect_model_load_cost (stmt_vec_info stmt_info, int ncopies,
1062 bool load_lanes_p, slp_tree slp_node,
1063 stmt_vector_for_cost *prologue_cost_vec,
1064 stmt_vector_for_cost *body_cost_vec)
1065 {
1066 int group_size;
1067 gimple first_stmt;
1068 struct data_reference *dr = STMT_VINFO_DATA_REF (stmt_info), *first_dr;
1069 unsigned int inside_cost = 0, prologue_cost = 0;
1070
1071 /* The SLP costs were already calculated during SLP tree build. */
1072 if (PURE_SLP_STMT (stmt_info))
1073 return;
1074
1075 /* Grouped accesses? */
1076 first_stmt = GROUP_FIRST_ELEMENT (stmt_info);
1077 if (STMT_VINFO_GROUPED_ACCESS (stmt_info) && first_stmt && !slp_node)
1078 {
1079 group_size = vect_cost_group_size (stmt_info);
1080 first_dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (first_stmt));
1081 }
1082 /* Not a grouped access. */
1083 else
1084 {
1085 group_size = 1;
1086 first_dr = dr;
1087 }
1088
1089 /* We assume that the cost of a single load-lanes instruction is
1090 equivalent to the cost of GROUP_SIZE separate loads. If a grouped
1091 access is instead being provided by a load-and-permute operation,
1092 include the cost of the permutes. */
1093 if (!load_lanes_p && group_size > 1)
1094 {
1095 /* Uses an even and odd extract operations or shuffle operations
1096 for each needed permute. */
1097 int nstmts = ncopies * ceil_log2 (group_size) * group_size;
1098 inside_cost = record_stmt_cost (body_cost_vec, nstmts, vec_perm,
1099 stmt_info, 0, vect_body);
1100
1101 if (dump_enabled_p ())
1102 dump_printf_loc (MSG_NOTE, vect_location,
1103 "vect_model_load_cost: strided group_size = %d .\n",
1104 group_size);
1105 }
1106
1107 /* The loads themselves. */
1108 if (STMT_VINFO_STRIDE_LOAD_P (stmt_info))
1109 {
1110 /* N scalar loads plus gathering them into a vector. */
1111 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
1112 inside_cost += record_stmt_cost (body_cost_vec,
1113 ncopies * TYPE_VECTOR_SUBPARTS (vectype),
1114 scalar_load, stmt_info, 0, vect_body);
1115 inside_cost += record_stmt_cost (body_cost_vec, ncopies, vec_construct,
1116 stmt_info, 0, vect_body);
1117 }
1118 else
1119 vect_get_load_cost (first_dr, ncopies,
1120 ((!STMT_VINFO_GROUPED_ACCESS (stmt_info))
1121 || group_size > 1 || slp_node),
1122 &inside_cost, &prologue_cost,
1123 prologue_cost_vec, body_cost_vec, true);
1124
1125 if (dump_enabled_p ())
1126 dump_printf_loc (MSG_NOTE, vect_location,
1127 "vect_model_load_cost: inside_cost = %d, "
1128 "prologue_cost = %d .\n", inside_cost, prologue_cost);
1129 }
1130
1131
1132 /* Calculate cost of DR's memory access. */
1133 void
1134 vect_get_load_cost (struct data_reference *dr, int ncopies,
1135 bool add_realign_cost, unsigned int *inside_cost,
1136 unsigned int *prologue_cost,
1137 stmt_vector_for_cost *prologue_cost_vec,
1138 stmt_vector_for_cost *body_cost_vec,
1139 bool record_prologue_costs)
1140 {
1141 int alignment_support_scheme = vect_supportable_dr_alignment (dr, false);
1142 gimple stmt = DR_STMT (dr);
1143 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
1144
1145 switch (alignment_support_scheme)
1146 {
1147 case dr_aligned:
1148 {
1149 *inside_cost += record_stmt_cost (body_cost_vec, ncopies, vector_load,
1150 stmt_info, 0, vect_body);
1151
1152 if (dump_enabled_p ())
1153 dump_printf_loc (MSG_NOTE, vect_location,
1154 "vect_model_load_cost: aligned.\n");
1155
1156 break;
1157 }
1158 case dr_unaligned_supported:
1159 {
1160 /* Here, we assign an additional cost for the unaligned load. */
1161 *inside_cost += record_stmt_cost (body_cost_vec, ncopies,
1162 unaligned_load, stmt_info,
1163 DR_MISALIGNMENT (dr), vect_body);
1164
1165 if (dump_enabled_p ())
1166 dump_printf_loc (MSG_NOTE, vect_location,
1167 "vect_model_load_cost: unaligned supported by "
1168 "hardware.\n");
1169
1170 break;
1171 }
1172 case dr_explicit_realign:
1173 {
1174 *inside_cost += record_stmt_cost (body_cost_vec, ncopies * 2,
1175 vector_load, stmt_info, 0, vect_body);
1176 *inside_cost += record_stmt_cost (body_cost_vec, ncopies,
1177 vec_perm, stmt_info, 0, vect_body);
1178
1179 /* FIXME: If the misalignment remains fixed across the iterations of
1180 the containing loop, the following cost should be added to the
1181 prologue costs. */
1182 if (targetm.vectorize.builtin_mask_for_load)
1183 *inside_cost += record_stmt_cost (body_cost_vec, 1, vector_stmt,
1184 stmt_info, 0, vect_body);
1185
1186 if (dump_enabled_p ())
1187 dump_printf_loc (MSG_NOTE, vect_location,
1188 "vect_model_load_cost: explicit realign\n");
1189
1190 break;
1191 }
1192 case dr_explicit_realign_optimized:
1193 {
1194 if (dump_enabled_p ())
1195 dump_printf_loc (MSG_NOTE, vect_location,
1196 "vect_model_load_cost: unaligned software "
1197 "pipelined.\n");
1198
1199 /* Unaligned software pipeline has a load of an address, an initial
1200 load, and possibly a mask operation to "prime" the loop. However,
1201 if this is an access in a group of loads, which provide grouped
1202 access, then the above cost should only be considered for one
1203 access in the group. Inside the loop, there is a load op
1204 and a realignment op. */
1205
1206 if (add_realign_cost && record_prologue_costs)
1207 {
1208 *prologue_cost += record_stmt_cost (prologue_cost_vec, 2,
1209 vector_stmt, stmt_info,
1210 0, vect_prologue);
1211 if (targetm.vectorize.builtin_mask_for_load)
1212 *prologue_cost += record_stmt_cost (prologue_cost_vec, 1,
1213 vector_stmt, stmt_info,
1214 0, vect_prologue);
1215 }
1216
1217 *inside_cost += record_stmt_cost (body_cost_vec, ncopies, vector_load,
1218 stmt_info, 0, vect_body);
1219 *inside_cost += record_stmt_cost (body_cost_vec, ncopies, vec_perm,
1220 stmt_info, 0, vect_body);
1221
1222 if (dump_enabled_p ())
1223 dump_printf_loc (MSG_NOTE, vect_location,
1224 "vect_model_load_cost: explicit realign optimized"
1225 "\n");
1226
1227 break;
1228 }
1229
1230 case dr_unaligned_unsupported:
1231 {
1232 *inside_cost = VECT_MAX_COST;
1233
1234 if (dump_enabled_p ())
1235 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
1236 "vect_model_load_cost: unsupported access.\n");
1237 break;
1238 }
1239
1240 default:
1241 gcc_unreachable ();
1242 }
1243 }
1244
1245 /* Insert the new stmt NEW_STMT at *GSI or at the appropriate place in
1246 the loop preheader for the vectorized stmt STMT. */
1247
1248 static void
1249 vect_init_vector_1 (gimple stmt, gimple new_stmt, gimple_stmt_iterator *gsi)
1250 {
1251 if (gsi)
1252 vect_finish_stmt_generation (stmt, new_stmt, gsi);
1253 else
1254 {
1255 stmt_vec_info stmt_vinfo = vinfo_for_stmt (stmt);
1256 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_vinfo);
1257
1258 if (loop_vinfo)
1259 {
1260 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
1261 basic_block new_bb;
1262 edge pe;
1263
1264 if (nested_in_vect_loop_p (loop, stmt))
1265 loop = loop->inner;
1266
1267 pe = loop_preheader_edge (loop);
1268 new_bb = gsi_insert_on_edge_immediate (pe, new_stmt);
1269 gcc_assert (!new_bb);
1270 }
1271 else
1272 {
1273 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_vinfo);
1274 basic_block bb;
1275 gimple_stmt_iterator gsi_bb_start;
1276
1277 gcc_assert (bb_vinfo);
1278 bb = BB_VINFO_BB (bb_vinfo);
1279 gsi_bb_start = gsi_after_labels (bb);
1280 gsi_insert_before (&gsi_bb_start, new_stmt, GSI_SAME_STMT);
1281 }
1282 }
1283
1284 if (dump_enabled_p ())
1285 {
1286 dump_printf_loc (MSG_NOTE, vect_location,
1287 "created new init_stmt: ");
1288 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, new_stmt, 0);
1289 dump_printf (MSG_NOTE, "\n");
1290 }
1291 }
1292
1293 /* Function vect_init_vector.
1294
1295 Insert a new stmt (INIT_STMT) that initializes a new variable of type
1296 TYPE with the value VAL. If TYPE is a vector type and VAL does not have
1297 vector type a vector with all elements equal to VAL is created first.
1298 Place the initialization at BSI if it is not NULL. Otherwise, place the
1299 initialization at the loop preheader.
1300 Return the DEF of INIT_STMT.
1301 It will be used in the vectorization of STMT. */
1302
1303 tree
1304 vect_init_vector (gimple stmt, tree val, tree type, gimple_stmt_iterator *gsi)
1305 {
1306 tree new_var;
1307 gimple init_stmt;
1308 tree vec_oprnd;
1309 tree new_temp;
1310
1311 if (TREE_CODE (type) == VECTOR_TYPE
1312 && TREE_CODE (TREE_TYPE (val)) != VECTOR_TYPE)
1313 {
1314 if (!types_compatible_p (TREE_TYPE (type), TREE_TYPE (val)))
1315 {
1316 if (CONSTANT_CLASS_P (val))
1317 val = fold_unary (VIEW_CONVERT_EXPR, TREE_TYPE (type), val);
1318 else
1319 {
1320 new_temp = make_ssa_name (TREE_TYPE (type), NULL);
1321 init_stmt = gimple_build_assign_with_ops (NOP_EXPR,
1322 new_temp, val,
1323 NULL_TREE);
1324 vect_init_vector_1 (stmt, init_stmt, gsi);
1325 val = new_temp;
1326 }
1327 }
1328 val = build_vector_from_val (type, val);
1329 }
1330
1331 new_var = vect_get_new_vect_var (type, vect_simple_var, "cst_");
1332 init_stmt = gimple_build_assign (new_var, val);
1333 new_temp = make_ssa_name (new_var, init_stmt);
1334 gimple_assign_set_lhs (init_stmt, new_temp);
1335 vect_init_vector_1 (stmt, init_stmt, gsi);
1336 vec_oprnd = gimple_assign_lhs (init_stmt);
1337 return vec_oprnd;
1338 }
1339
1340
1341 /* Function vect_get_vec_def_for_operand.
1342
1343 OP is an operand in STMT. This function returns a (vector) def that will be
1344 used in the vectorized stmt for STMT.
1345
1346 In the case that OP is an SSA_NAME which is defined in the loop, then
1347 STMT_VINFO_VEC_STMT of the defining stmt holds the relevant def.
1348
1349 In case OP is an invariant or constant, a new stmt that creates a vector def
1350 needs to be introduced. */
1351
1352 tree
1353 vect_get_vec_def_for_operand (tree op, gimple stmt, tree *scalar_def)
1354 {
1355 tree vec_oprnd;
1356 gimple vec_stmt;
1357 gimple def_stmt;
1358 stmt_vec_info def_stmt_info = NULL;
1359 stmt_vec_info stmt_vinfo = vinfo_for_stmt (stmt);
1360 unsigned int nunits;
1361 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_vinfo);
1362 tree def;
1363 enum vect_def_type dt;
1364 bool is_simple_use;
1365 tree vector_type;
1366
1367 if (dump_enabled_p ())
1368 {
1369 dump_printf_loc (MSG_NOTE, vect_location,
1370 "vect_get_vec_def_for_operand: ");
1371 dump_generic_expr (MSG_NOTE, TDF_SLIM, op);
1372 dump_printf (MSG_NOTE, "\n");
1373 }
1374
1375 is_simple_use = vect_is_simple_use (op, stmt, loop_vinfo, NULL,
1376 &def_stmt, &def, &dt);
1377 gcc_assert (is_simple_use);
1378 if (dump_enabled_p ())
1379 {
1380 int loc_printed = 0;
1381 if (def)
1382 {
1383 dump_printf_loc (MSG_NOTE, vect_location, "def = ");
1384 loc_printed = 1;
1385 dump_generic_expr (MSG_NOTE, TDF_SLIM, def);
1386 dump_printf (MSG_NOTE, "\n");
1387 }
1388 if (def_stmt)
1389 {
1390 if (loc_printed)
1391 dump_printf (MSG_NOTE, " def_stmt = ");
1392 else
1393 dump_printf_loc (MSG_NOTE, vect_location, " def_stmt = ");
1394 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, def_stmt, 0);
1395 dump_printf (MSG_NOTE, "\n");
1396 }
1397 }
1398
1399 switch (dt)
1400 {
1401 /* Case 1: operand is a constant. */
1402 case vect_constant_def:
1403 {
1404 vector_type = get_vectype_for_scalar_type (TREE_TYPE (op));
1405 gcc_assert (vector_type);
1406 nunits = TYPE_VECTOR_SUBPARTS (vector_type);
1407
1408 if (scalar_def)
1409 *scalar_def = op;
1410
1411 /* Create 'vect_cst_ = {cst,cst,...,cst}' */
1412 if (dump_enabled_p ())
1413 dump_printf_loc (MSG_NOTE, vect_location,
1414 "Create vector_cst. nunits = %d\n", nunits);
1415
1416 return vect_init_vector (stmt, op, vector_type, NULL);
1417 }
1418
1419 /* Case 2: operand is defined outside the loop - loop invariant. */
1420 case vect_external_def:
1421 {
1422 vector_type = get_vectype_for_scalar_type (TREE_TYPE (def));
1423 gcc_assert (vector_type);
1424
1425 if (scalar_def)
1426 *scalar_def = def;
1427
1428 /* Create 'vec_inv = {inv,inv,..,inv}' */
1429 if (dump_enabled_p ())
1430 dump_printf_loc (MSG_NOTE, vect_location, "Create vector_inv.\n");
1431
1432 return vect_init_vector (stmt, def, vector_type, NULL);
1433 }
1434
1435 /* Case 3: operand is defined inside the loop. */
1436 case vect_internal_def:
1437 {
1438 if (scalar_def)
1439 *scalar_def = NULL/* FIXME tuples: def_stmt*/;
1440
1441 /* Get the def from the vectorized stmt. */
1442 def_stmt_info = vinfo_for_stmt (def_stmt);
1443
1444 vec_stmt = STMT_VINFO_VEC_STMT (def_stmt_info);
1445 /* Get vectorized pattern statement. */
1446 if (!vec_stmt
1447 && STMT_VINFO_IN_PATTERN_P (def_stmt_info)
1448 && !STMT_VINFO_RELEVANT (def_stmt_info))
1449 vec_stmt = STMT_VINFO_VEC_STMT (vinfo_for_stmt (
1450 STMT_VINFO_RELATED_STMT (def_stmt_info)));
1451 gcc_assert (vec_stmt);
1452 if (gimple_code (vec_stmt) == GIMPLE_PHI)
1453 vec_oprnd = PHI_RESULT (vec_stmt);
1454 else if (is_gimple_call (vec_stmt))
1455 vec_oprnd = gimple_call_lhs (vec_stmt);
1456 else
1457 vec_oprnd = gimple_assign_lhs (vec_stmt);
1458 return vec_oprnd;
1459 }
1460
1461 /* Case 4: operand is defined by a loop header phi - reduction */
1462 case vect_reduction_def:
1463 case vect_double_reduction_def:
1464 case vect_nested_cycle:
1465 {
1466 struct loop *loop;
1467
1468 gcc_assert (gimple_code (def_stmt) == GIMPLE_PHI);
1469 loop = (gimple_bb (def_stmt))->loop_father;
1470
1471 /* Get the def before the loop */
1472 op = PHI_ARG_DEF_FROM_EDGE (def_stmt, loop_preheader_edge (loop));
1473 return get_initial_def_for_reduction (stmt, op, scalar_def);
1474 }
1475
1476 /* Case 5: operand is defined by loop-header phi - induction. */
1477 case vect_induction_def:
1478 {
1479 gcc_assert (gimple_code (def_stmt) == GIMPLE_PHI);
1480
1481 /* Get the def from the vectorized stmt. */
1482 def_stmt_info = vinfo_for_stmt (def_stmt);
1483 vec_stmt = STMT_VINFO_VEC_STMT (def_stmt_info);
1484 if (gimple_code (vec_stmt) == GIMPLE_PHI)
1485 vec_oprnd = PHI_RESULT (vec_stmt);
1486 else
1487 vec_oprnd = gimple_get_lhs (vec_stmt);
1488 return vec_oprnd;
1489 }
1490
1491 default:
1492 gcc_unreachable ();
1493 }
1494 }
1495
1496
1497 /* Function vect_get_vec_def_for_stmt_copy
1498
1499 Return a vector-def for an operand. This function is used when the
1500 vectorized stmt to be created (by the caller to this function) is a "copy"
1501 created in case the vectorized result cannot fit in one vector, and several
1502 copies of the vector-stmt are required. In this case the vector-def is
1503 retrieved from the vector stmt recorded in the STMT_VINFO_RELATED_STMT field
1504 of the stmt that defines VEC_OPRND.
1505 DT is the type of the vector def VEC_OPRND.
1506
1507 Context:
1508 In case the vectorization factor (VF) is bigger than the number
1509 of elements that can fit in a vectype (nunits), we have to generate
1510 more than one vector stmt to vectorize the scalar stmt. This situation
1511 arises when there are multiple data-types operated upon in the loop; the
1512 smallest data-type determines the VF, and as a result, when vectorizing
1513 stmts operating on wider types we need to create 'VF/nunits' "copies" of the
1514 vector stmt (each computing a vector of 'nunits' results, and together
1515 computing 'VF' results in each iteration). This function is called when
1516 vectorizing such a stmt (e.g. vectorizing S2 in the illustration below, in
1517 which VF=16 and nunits=4, so the number of copies required is 4):
1518
1519 scalar stmt: vectorized into: STMT_VINFO_RELATED_STMT
1520
1521 S1: x = load VS1.0: vx.0 = memref0 VS1.1
1522 VS1.1: vx.1 = memref1 VS1.2
1523 VS1.2: vx.2 = memref2 VS1.3
1524 VS1.3: vx.3 = memref3
1525
1526 S2: z = x + ... VSnew.0: vz0 = vx.0 + ... VSnew.1
1527 VSnew.1: vz1 = vx.1 + ... VSnew.2
1528 VSnew.2: vz2 = vx.2 + ... VSnew.3
1529 VSnew.3: vz3 = vx.3 + ...
1530
1531 The vectorization of S1 is explained in vectorizable_load.
1532 The vectorization of S2:
1533 To create the first vector-stmt out of the 4 copies - VSnew.0 -
1534 the function 'vect_get_vec_def_for_operand' is called to
1535 get the relevant vector-def for each operand of S2. For operand x it
1536 returns the vector-def 'vx.0'.
1537
1538 To create the remaining copies of the vector-stmt (VSnew.j), this
1539 function is called to get the relevant vector-def for each operand. It is
1540 obtained from the respective VS1.j stmt, which is recorded in the
1541 STMT_VINFO_RELATED_STMT field of the stmt that defines VEC_OPRND.
1542
1543 For example, to obtain the vector-def 'vx.1' in order to create the
1544 vector stmt 'VSnew.1', this function is called with VEC_OPRND='vx.0'.
1545 Given 'vx0' we obtain the stmt that defines it ('VS1.0'); from the
1546 STMT_VINFO_RELATED_STMT field of 'VS1.0' we obtain the next copy - 'VS1.1',
1547 and return its def ('vx.1').
1548 Overall, to create the above sequence this function will be called 3 times:
1549 vx.1 = vect_get_vec_def_for_stmt_copy (dt, vx.0);
1550 vx.2 = vect_get_vec_def_for_stmt_copy (dt, vx.1);
1551 vx.3 = vect_get_vec_def_for_stmt_copy (dt, vx.2); */
1552
1553 tree
1554 vect_get_vec_def_for_stmt_copy (enum vect_def_type dt, tree vec_oprnd)
1555 {
1556 gimple vec_stmt_for_operand;
1557 stmt_vec_info def_stmt_info;
1558
1559 /* Do nothing; can reuse same def. */
1560 if (dt == vect_external_def || dt == vect_constant_def )
1561 return vec_oprnd;
1562
1563 vec_stmt_for_operand = SSA_NAME_DEF_STMT (vec_oprnd);
1564 def_stmt_info = vinfo_for_stmt (vec_stmt_for_operand);
1565 gcc_assert (def_stmt_info);
1566 vec_stmt_for_operand = STMT_VINFO_RELATED_STMT (def_stmt_info);
1567 gcc_assert (vec_stmt_for_operand);
1568 vec_oprnd = gimple_get_lhs (vec_stmt_for_operand);
1569 if (gimple_code (vec_stmt_for_operand) == GIMPLE_PHI)
1570 vec_oprnd = PHI_RESULT (vec_stmt_for_operand);
1571 else
1572 vec_oprnd = gimple_get_lhs (vec_stmt_for_operand);
1573 return vec_oprnd;
1574 }
1575
1576
1577 /* Get vectorized definitions for the operands to create a copy of an original
1578 stmt. See vect_get_vec_def_for_stmt_copy () for details. */
1579
1580 static void
1581 vect_get_vec_defs_for_stmt_copy (enum vect_def_type *dt,
1582 vec<tree> *vec_oprnds0,
1583 vec<tree> *vec_oprnds1)
1584 {
1585 tree vec_oprnd = vec_oprnds0->pop ();
1586
1587 vec_oprnd = vect_get_vec_def_for_stmt_copy (dt[0], vec_oprnd);
1588 vec_oprnds0->quick_push (vec_oprnd);
1589
1590 if (vec_oprnds1 && vec_oprnds1->length ())
1591 {
1592 vec_oprnd = vec_oprnds1->pop ();
1593 vec_oprnd = vect_get_vec_def_for_stmt_copy (dt[1], vec_oprnd);
1594 vec_oprnds1->quick_push (vec_oprnd);
1595 }
1596 }
1597
1598
1599 /* Get vectorized definitions for OP0 and OP1.
1600 REDUC_INDEX is the index of reduction operand in case of reduction,
1601 and -1 otherwise. */
1602
1603 void
1604 vect_get_vec_defs (tree op0, tree op1, gimple stmt,
1605 vec<tree> *vec_oprnds0,
1606 vec<tree> *vec_oprnds1,
1607 slp_tree slp_node, int reduc_index)
1608 {
1609 if (slp_node)
1610 {
1611 int nops = (op1 == NULL_TREE) ? 1 : 2;
1612 auto_vec<tree> ops (nops);
1613 auto_vec<vec<tree> > vec_defs (nops);
1614
1615 ops.quick_push (op0);
1616 if (op1)
1617 ops.quick_push (op1);
1618
1619 vect_get_slp_defs (ops, slp_node, &vec_defs, reduc_index);
1620
1621 *vec_oprnds0 = vec_defs[0];
1622 if (op1)
1623 *vec_oprnds1 = vec_defs[1];
1624 }
1625 else
1626 {
1627 tree vec_oprnd;
1628
1629 vec_oprnds0->create (1);
1630 vec_oprnd = vect_get_vec_def_for_operand (op0, stmt, NULL);
1631 vec_oprnds0->quick_push (vec_oprnd);
1632
1633 if (op1)
1634 {
1635 vec_oprnds1->create (1);
1636 vec_oprnd = vect_get_vec_def_for_operand (op1, stmt, NULL);
1637 vec_oprnds1->quick_push (vec_oprnd);
1638 }
1639 }
1640 }
1641
1642
1643 /* Function vect_finish_stmt_generation.
1644
1645 Insert a new stmt. */
1646
1647 void
1648 vect_finish_stmt_generation (gimple stmt, gimple vec_stmt,
1649 gimple_stmt_iterator *gsi)
1650 {
1651 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
1652 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
1653 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
1654
1655 gcc_assert (gimple_code (stmt) != GIMPLE_LABEL);
1656
1657 if (!gsi_end_p (*gsi)
1658 && gimple_has_mem_ops (vec_stmt))
1659 {
1660 gimple at_stmt = gsi_stmt (*gsi);
1661 tree vuse = gimple_vuse (at_stmt);
1662 if (vuse && TREE_CODE (vuse) == SSA_NAME)
1663 {
1664 tree vdef = gimple_vdef (at_stmt);
1665 gimple_set_vuse (vec_stmt, gimple_vuse (at_stmt));
1666 /* If we have an SSA vuse and insert a store, update virtual
1667 SSA form to avoid triggering the renamer. Do so only
1668 if we can easily see all uses - which is what almost always
1669 happens with the way vectorized stmts are inserted. */
1670 if ((vdef && TREE_CODE (vdef) == SSA_NAME)
1671 && ((is_gimple_assign (vec_stmt)
1672 && !is_gimple_reg (gimple_assign_lhs (vec_stmt)))
1673 || (is_gimple_call (vec_stmt)
1674 && !(gimple_call_flags (vec_stmt)
1675 & (ECF_CONST|ECF_PURE|ECF_NOVOPS)))))
1676 {
1677 tree new_vdef = copy_ssa_name (vuse, vec_stmt);
1678 gimple_set_vdef (vec_stmt, new_vdef);
1679 SET_USE (gimple_vuse_op (at_stmt), new_vdef);
1680 }
1681 }
1682 }
1683 gsi_insert_before (gsi, vec_stmt, GSI_SAME_STMT);
1684
1685 set_vinfo_for_stmt (vec_stmt, new_stmt_vec_info (vec_stmt, loop_vinfo,
1686 bb_vinfo));
1687
1688 if (dump_enabled_p ())
1689 {
1690 dump_printf_loc (MSG_NOTE, vect_location, "add new stmt: ");
1691 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, vec_stmt, 0);
1692 dump_printf (MSG_NOTE, "\n");
1693 }
1694
1695 gimple_set_location (vec_stmt, gimple_location (stmt));
1696
1697 /* While EH edges will generally prevent vectorization, stmt might
1698 e.g. be in a must-not-throw region. Ensure newly created stmts
1699 that could throw are part of the same region. */
1700 int lp_nr = lookup_stmt_eh_lp (stmt);
1701 if (lp_nr != 0 && stmt_could_throw_p (vec_stmt))
1702 add_stmt_to_eh_lp (vec_stmt, lp_nr);
1703 }
1704
1705 /* Checks if CALL can be vectorized in type VECTYPE. Returns
1706 a function declaration if the target has a vectorized version
1707 of the function, or NULL_TREE if the function cannot be vectorized. */
1708
1709 tree
1710 vectorizable_function (gimple call, tree vectype_out, tree vectype_in)
1711 {
1712 tree fndecl = gimple_call_fndecl (call);
1713
1714 /* We only handle functions that do not read or clobber memory -- i.e.
1715 const or novops ones. */
1716 if (!(gimple_call_flags (call) & (ECF_CONST | ECF_NOVOPS)))
1717 return NULL_TREE;
1718
1719 if (!fndecl
1720 || TREE_CODE (fndecl) != FUNCTION_DECL
1721 || !DECL_BUILT_IN (fndecl))
1722 return NULL_TREE;
1723
1724 return targetm.vectorize.builtin_vectorized_function (fndecl, vectype_out,
1725 vectype_in);
1726 }
1727
1728
1729 static tree permute_vec_elements (tree, tree, tree, gimple,
1730 gimple_stmt_iterator *);
1731
1732
1733 /* Function vectorizable_mask_load_store.
1734
1735 Check if STMT performs a conditional load or store that can be vectorized.
1736 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
1737 stmt to replace it, put it in VEC_STMT, and insert it at GSI.
1738 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
1739
1740 static bool
1741 vectorizable_mask_load_store (gimple stmt, gimple_stmt_iterator *gsi,
1742 gimple *vec_stmt, slp_tree slp_node)
1743 {
1744 tree vec_dest = NULL;
1745 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
1746 stmt_vec_info prev_stmt_info;
1747 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
1748 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
1749 bool nested_in_vect_loop = nested_in_vect_loop_p (loop, stmt);
1750 struct data_reference *dr = STMT_VINFO_DATA_REF (stmt_info);
1751 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
1752 tree elem_type;
1753 gimple new_stmt;
1754 tree dummy;
1755 tree dataref_ptr = NULL_TREE;
1756 gimple ptr_incr;
1757 int nunits = TYPE_VECTOR_SUBPARTS (vectype);
1758 int ncopies;
1759 int i, j;
1760 bool inv_p;
1761 tree gather_base = NULL_TREE, gather_off = NULL_TREE;
1762 tree gather_off_vectype = NULL_TREE, gather_decl = NULL_TREE;
1763 int gather_scale = 1;
1764 enum vect_def_type gather_dt = vect_unknown_def_type;
1765 bool is_store;
1766 tree mask;
1767 gimple def_stmt;
1768 tree def;
1769 enum vect_def_type dt;
1770
1771 if (slp_node != NULL)
1772 return false;
1773
1774 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits;
1775 gcc_assert (ncopies >= 1);
1776
1777 is_store = gimple_call_internal_fn (stmt) == IFN_MASK_STORE;
1778 mask = gimple_call_arg (stmt, 2);
1779 if (TYPE_PRECISION (TREE_TYPE (mask))
1780 != GET_MODE_BITSIZE (TYPE_MODE (TREE_TYPE (vectype))))
1781 return false;
1782
1783 /* FORNOW. This restriction should be relaxed. */
1784 if (nested_in_vect_loop && ncopies > 1)
1785 {
1786 if (dump_enabled_p ())
1787 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
1788 "multiple types in nested loop.");
1789 return false;
1790 }
1791
1792 if (!STMT_VINFO_RELEVANT_P (stmt_info))
1793 return false;
1794
1795 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
1796 return false;
1797
1798 if (!STMT_VINFO_DATA_REF (stmt_info))
1799 return false;
1800
1801 elem_type = TREE_TYPE (vectype);
1802
1803 if (STMT_VINFO_GROUPED_ACCESS (stmt_info))
1804 return false;
1805
1806 if (STMT_VINFO_STRIDE_LOAD_P (stmt_info))
1807 return false;
1808
1809 if (STMT_VINFO_GATHER_P (stmt_info))
1810 {
1811 gimple def_stmt;
1812 tree def;
1813 gather_decl = vect_check_gather (stmt, loop_vinfo, &gather_base,
1814 &gather_off, &gather_scale);
1815 gcc_assert (gather_decl);
1816 if (!vect_is_simple_use_1 (gather_off, NULL, loop_vinfo, NULL,
1817 &def_stmt, &def, &gather_dt,
1818 &gather_off_vectype))
1819 {
1820 if (dump_enabled_p ())
1821 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
1822 "gather index use not simple.");
1823 return false;
1824 }
1825
1826 tree arglist = TYPE_ARG_TYPES (TREE_TYPE (gather_decl));
1827 tree masktype
1828 = TREE_VALUE (TREE_CHAIN (TREE_CHAIN (TREE_CHAIN (arglist))));
1829 if (TREE_CODE (masktype) == INTEGER_TYPE)
1830 {
1831 if (dump_enabled_p ())
1832 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
1833 "masked gather with integer mask not supported.");
1834 return false;
1835 }
1836 }
1837 else if (tree_int_cst_compare (nested_in_vect_loop
1838 ? STMT_VINFO_DR_STEP (stmt_info)
1839 : DR_STEP (dr), size_zero_node) <= 0)
1840 return false;
1841 else if (!VECTOR_MODE_P (TYPE_MODE (vectype))
1842 || !can_vec_mask_load_store_p (TYPE_MODE (vectype), !is_store))
1843 return false;
1844
1845 if (TREE_CODE (mask) != SSA_NAME)
1846 return false;
1847
1848 if (!vect_is_simple_use (mask, stmt, loop_vinfo, NULL,
1849 &def_stmt, &def, &dt))
1850 return false;
1851
1852 if (is_store)
1853 {
1854 tree rhs = gimple_call_arg (stmt, 3);
1855 if (!vect_is_simple_use (rhs, stmt, loop_vinfo, NULL,
1856 &def_stmt, &def, &dt))
1857 return false;
1858 }
1859
1860 if (!vec_stmt) /* transformation not required. */
1861 {
1862 STMT_VINFO_TYPE (stmt_info) = call_vec_info_type;
1863 if (is_store)
1864 vect_model_store_cost (stmt_info, ncopies, false, dt,
1865 NULL, NULL, NULL);
1866 else
1867 vect_model_load_cost (stmt_info, ncopies, false, NULL, NULL, NULL);
1868 return true;
1869 }
1870
1871 /** Transform. **/
1872
1873 if (STMT_VINFO_GATHER_P (stmt_info))
1874 {
1875 tree vec_oprnd0 = NULL_TREE, op;
1876 tree arglist = TYPE_ARG_TYPES (TREE_TYPE (gather_decl));
1877 tree rettype, srctype, ptrtype, idxtype, masktype, scaletype;
1878 tree ptr, vec_mask = NULL_TREE, mask_op = NULL_TREE, var, scale;
1879 tree perm_mask = NULL_TREE, prev_res = NULL_TREE;
1880 tree mask_perm_mask = NULL_TREE;
1881 edge pe = loop_preheader_edge (loop);
1882 gimple_seq seq;
1883 basic_block new_bb;
1884 enum { NARROW, NONE, WIDEN } modifier;
1885 int gather_off_nunits = TYPE_VECTOR_SUBPARTS (gather_off_vectype);
1886
1887 rettype = TREE_TYPE (TREE_TYPE (gather_decl));
1888 srctype = TREE_VALUE (arglist); arglist = TREE_CHAIN (arglist);
1889 ptrtype = TREE_VALUE (arglist); arglist = TREE_CHAIN (arglist);
1890 idxtype = TREE_VALUE (arglist); arglist = TREE_CHAIN (arglist);
1891 masktype = TREE_VALUE (arglist); arglist = TREE_CHAIN (arglist);
1892 scaletype = TREE_VALUE (arglist);
1893 gcc_checking_assert (types_compatible_p (srctype, rettype)
1894 && types_compatible_p (srctype, masktype));
1895
1896 if (nunits == gather_off_nunits)
1897 modifier = NONE;
1898 else if (nunits == gather_off_nunits / 2)
1899 {
1900 unsigned char *sel = XALLOCAVEC (unsigned char, gather_off_nunits);
1901 modifier = WIDEN;
1902
1903 for (i = 0; i < gather_off_nunits; ++i)
1904 sel[i] = i | nunits;
1905
1906 perm_mask = vect_gen_perm_mask (gather_off_vectype, sel);
1907 gcc_assert (perm_mask != NULL_TREE);
1908 }
1909 else if (nunits == gather_off_nunits * 2)
1910 {
1911 unsigned char *sel = XALLOCAVEC (unsigned char, nunits);
1912 modifier = NARROW;
1913
1914 for (i = 0; i < nunits; ++i)
1915 sel[i] = i < gather_off_nunits
1916 ? i : i + nunits - gather_off_nunits;
1917
1918 perm_mask = vect_gen_perm_mask (vectype, sel);
1919 gcc_assert (perm_mask != NULL_TREE);
1920 ncopies *= 2;
1921 for (i = 0; i < nunits; ++i)
1922 sel[i] = i | gather_off_nunits;
1923 mask_perm_mask = vect_gen_perm_mask (masktype, sel);
1924 gcc_assert (mask_perm_mask != NULL_TREE);
1925 }
1926 else
1927 gcc_unreachable ();
1928
1929 vec_dest = vect_create_destination_var (gimple_call_lhs (stmt), vectype);
1930
1931 ptr = fold_convert (ptrtype, gather_base);
1932 if (!is_gimple_min_invariant (ptr))
1933 {
1934 ptr = force_gimple_operand (ptr, &seq, true, NULL_TREE);
1935 new_bb = gsi_insert_seq_on_edge_immediate (pe, seq);
1936 gcc_assert (!new_bb);
1937 }
1938
1939 scale = build_int_cst (scaletype, gather_scale);
1940
1941 prev_stmt_info = NULL;
1942 for (j = 0; j < ncopies; ++j)
1943 {
1944 if (modifier == WIDEN && (j & 1))
1945 op = permute_vec_elements (vec_oprnd0, vec_oprnd0,
1946 perm_mask, stmt, gsi);
1947 else if (j == 0)
1948 op = vec_oprnd0
1949 = vect_get_vec_def_for_operand (gather_off, stmt, NULL);
1950 else
1951 op = vec_oprnd0
1952 = vect_get_vec_def_for_stmt_copy (gather_dt, vec_oprnd0);
1953
1954 if (!useless_type_conversion_p (idxtype, TREE_TYPE (op)))
1955 {
1956 gcc_assert (TYPE_VECTOR_SUBPARTS (TREE_TYPE (op))
1957 == TYPE_VECTOR_SUBPARTS (idxtype));
1958 var = vect_get_new_vect_var (idxtype, vect_simple_var, NULL);
1959 var = make_ssa_name (var, NULL);
1960 op = build1 (VIEW_CONVERT_EXPR, idxtype, op);
1961 new_stmt
1962 = gimple_build_assign_with_ops (VIEW_CONVERT_EXPR, var,
1963 op, NULL_TREE);
1964 vect_finish_stmt_generation (stmt, new_stmt, gsi);
1965 op = var;
1966 }
1967
1968 if (mask_perm_mask && (j & 1))
1969 mask_op = permute_vec_elements (mask_op, mask_op,
1970 mask_perm_mask, stmt, gsi);
1971 else
1972 {
1973 if (j == 0)
1974 vec_mask = vect_get_vec_def_for_operand (mask, stmt, NULL);
1975 else
1976 {
1977 vect_is_simple_use (vec_mask, NULL, loop_vinfo, NULL,
1978 &def_stmt, &def, &dt);
1979 vec_mask = vect_get_vec_def_for_stmt_copy (dt, vec_mask);
1980 }
1981
1982 mask_op = vec_mask;
1983 if (!useless_type_conversion_p (masktype, TREE_TYPE (vec_mask)))
1984 {
1985 gcc_assert (TYPE_VECTOR_SUBPARTS (TREE_TYPE (mask_op))
1986 == TYPE_VECTOR_SUBPARTS (masktype));
1987 var = vect_get_new_vect_var (masktype, vect_simple_var,
1988 NULL);
1989 var = make_ssa_name (var, NULL);
1990 mask_op = build1 (VIEW_CONVERT_EXPR, masktype, mask_op);
1991 new_stmt
1992 = gimple_build_assign_with_ops (VIEW_CONVERT_EXPR, var,
1993 mask_op, NULL_TREE);
1994 vect_finish_stmt_generation (stmt, new_stmt, gsi);
1995 mask_op = var;
1996 }
1997 }
1998
1999 new_stmt
2000 = gimple_build_call (gather_decl, 5, mask_op, ptr, op, mask_op,
2001 scale);
2002
2003 if (!useless_type_conversion_p (vectype, rettype))
2004 {
2005 gcc_assert (TYPE_VECTOR_SUBPARTS (vectype)
2006 == TYPE_VECTOR_SUBPARTS (rettype));
2007 var = vect_get_new_vect_var (rettype, vect_simple_var, NULL);
2008 op = make_ssa_name (var, new_stmt);
2009 gimple_call_set_lhs (new_stmt, op);
2010 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2011 var = make_ssa_name (vec_dest, NULL);
2012 op = build1 (VIEW_CONVERT_EXPR, vectype, op);
2013 new_stmt
2014 = gimple_build_assign_with_ops (VIEW_CONVERT_EXPR, var, op,
2015 NULL_TREE);
2016 }
2017 else
2018 {
2019 var = make_ssa_name (vec_dest, new_stmt);
2020 gimple_call_set_lhs (new_stmt, var);
2021 }
2022
2023 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2024
2025 if (modifier == NARROW)
2026 {
2027 if ((j & 1) == 0)
2028 {
2029 prev_res = var;
2030 continue;
2031 }
2032 var = permute_vec_elements (prev_res, var,
2033 perm_mask, stmt, gsi);
2034 new_stmt = SSA_NAME_DEF_STMT (var);
2035 }
2036
2037 if (prev_stmt_info == NULL)
2038 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
2039 else
2040 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
2041 prev_stmt_info = vinfo_for_stmt (new_stmt);
2042 }
2043
2044 /* Ensure that even with -fno-tree-dce the scalar MASK_LOAD is removed
2045 from the IL. */
2046 tree lhs = gimple_call_lhs (stmt);
2047 new_stmt = gimple_build_assign (lhs, build_zero_cst (TREE_TYPE (lhs)));
2048 set_vinfo_for_stmt (new_stmt, stmt_info);
2049 set_vinfo_for_stmt (stmt, NULL);
2050 STMT_VINFO_STMT (stmt_info) = new_stmt;
2051 gsi_replace (gsi, new_stmt, true);
2052 return true;
2053 }
2054 else if (is_store)
2055 {
2056 tree vec_rhs = NULL_TREE, vec_mask = NULL_TREE;
2057 prev_stmt_info = NULL;
2058 for (i = 0; i < ncopies; i++)
2059 {
2060 unsigned align, misalign;
2061
2062 if (i == 0)
2063 {
2064 tree rhs = gimple_call_arg (stmt, 3);
2065 vec_rhs = vect_get_vec_def_for_operand (rhs, stmt, NULL);
2066 vec_mask = vect_get_vec_def_for_operand (mask, stmt, NULL);
2067 /* We should have catched mismatched types earlier. */
2068 gcc_assert (useless_type_conversion_p (vectype,
2069 TREE_TYPE (vec_rhs)));
2070 dataref_ptr = vect_create_data_ref_ptr (stmt, vectype, NULL,
2071 NULL_TREE, &dummy, gsi,
2072 &ptr_incr, false, &inv_p);
2073 gcc_assert (!inv_p);
2074 }
2075 else
2076 {
2077 vect_is_simple_use (vec_rhs, NULL, loop_vinfo, NULL, &def_stmt,
2078 &def, &dt);
2079 vec_rhs = vect_get_vec_def_for_stmt_copy (dt, vec_rhs);
2080 vect_is_simple_use (vec_mask, NULL, loop_vinfo, NULL, &def_stmt,
2081 &def, &dt);
2082 vec_mask = vect_get_vec_def_for_stmt_copy (dt, vec_mask);
2083 dataref_ptr = bump_vector_ptr (dataref_ptr, ptr_incr, gsi, stmt,
2084 TYPE_SIZE_UNIT (vectype));
2085 }
2086
2087 align = TYPE_ALIGN_UNIT (vectype);
2088 if (aligned_access_p (dr))
2089 misalign = 0;
2090 else if (DR_MISALIGNMENT (dr) == -1)
2091 {
2092 align = TYPE_ALIGN_UNIT (elem_type);
2093 misalign = 0;
2094 }
2095 else
2096 misalign = DR_MISALIGNMENT (dr);
2097 set_ptr_info_alignment (get_ptr_info (dataref_ptr), align,
2098 misalign);
2099 new_stmt
2100 = gimple_build_call_internal (IFN_MASK_STORE, 4, dataref_ptr,
2101 gimple_call_arg (stmt, 1),
2102 vec_mask, vec_rhs);
2103 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2104 if (i == 0)
2105 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
2106 else
2107 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
2108 prev_stmt_info = vinfo_for_stmt (new_stmt);
2109 }
2110 }
2111 else
2112 {
2113 tree vec_mask = NULL_TREE;
2114 prev_stmt_info = NULL;
2115 vec_dest = vect_create_destination_var (gimple_call_lhs (stmt), vectype);
2116 for (i = 0; i < ncopies; i++)
2117 {
2118 unsigned align, misalign;
2119
2120 if (i == 0)
2121 {
2122 vec_mask = vect_get_vec_def_for_operand (mask, stmt, NULL);
2123 dataref_ptr = vect_create_data_ref_ptr (stmt, vectype, NULL,
2124 NULL_TREE, &dummy, gsi,
2125 &ptr_incr, false, &inv_p);
2126 gcc_assert (!inv_p);
2127 }
2128 else
2129 {
2130 vect_is_simple_use (vec_mask, NULL, loop_vinfo, NULL, &def_stmt,
2131 &def, &dt);
2132 vec_mask = vect_get_vec_def_for_stmt_copy (dt, vec_mask);
2133 dataref_ptr = bump_vector_ptr (dataref_ptr, ptr_incr, gsi, stmt,
2134 TYPE_SIZE_UNIT (vectype));
2135 }
2136
2137 align = TYPE_ALIGN_UNIT (vectype);
2138 if (aligned_access_p (dr))
2139 misalign = 0;
2140 else if (DR_MISALIGNMENT (dr) == -1)
2141 {
2142 align = TYPE_ALIGN_UNIT (elem_type);
2143 misalign = 0;
2144 }
2145 else
2146 misalign = DR_MISALIGNMENT (dr);
2147 set_ptr_info_alignment (get_ptr_info (dataref_ptr), align,
2148 misalign);
2149 new_stmt
2150 = gimple_build_call_internal (IFN_MASK_LOAD, 3, dataref_ptr,
2151 gimple_call_arg (stmt, 1),
2152 vec_mask);
2153 gimple_call_set_lhs (new_stmt, make_ssa_name (vec_dest, NULL));
2154 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2155 if (i == 0)
2156 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
2157 else
2158 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
2159 prev_stmt_info = vinfo_for_stmt (new_stmt);
2160 }
2161 }
2162
2163 if (!is_store)
2164 {
2165 /* Ensure that even with -fno-tree-dce the scalar MASK_LOAD is removed
2166 from the IL. */
2167 tree lhs = gimple_call_lhs (stmt);
2168 new_stmt = gimple_build_assign (lhs, build_zero_cst (TREE_TYPE (lhs)));
2169 set_vinfo_for_stmt (new_stmt, stmt_info);
2170 set_vinfo_for_stmt (stmt, NULL);
2171 STMT_VINFO_STMT (stmt_info) = new_stmt;
2172 gsi_replace (gsi, new_stmt, true);
2173 }
2174
2175 return true;
2176 }
2177
2178
2179 /* Function vectorizable_call.
2180
2181 Check if STMT performs a function call that can be vectorized.
2182 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
2183 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
2184 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
2185
2186 static bool
2187 vectorizable_call (gimple stmt, gimple_stmt_iterator *gsi, gimple *vec_stmt,
2188 slp_tree slp_node)
2189 {
2190 tree vec_dest;
2191 tree scalar_dest;
2192 tree op, type;
2193 tree vec_oprnd0 = NULL_TREE, vec_oprnd1 = NULL_TREE;
2194 stmt_vec_info stmt_info = vinfo_for_stmt (stmt), prev_stmt_info;
2195 tree vectype_out, vectype_in;
2196 int nunits_in;
2197 int nunits_out;
2198 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
2199 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
2200 tree fndecl, new_temp, def, rhs_type;
2201 gimple def_stmt;
2202 enum vect_def_type dt[3]
2203 = {vect_unknown_def_type, vect_unknown_def_type, vect_unknown_def_type};
2204 gimple new_stmt = NULL;
2205 int ncopies, j;
2206 vec<tree> vargs = vNULL;
2207 enum { NARROW, NONE, WIDEN } modifier;
2208 size_t i, nargs;
2209 tree lhs;
2210
2211 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
2212 return false;
2213
2214 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
2215 return false;
2216
2217 /* Is STMT a vectorizable call? */
2218 if (!is_gimple_call (stmt))
2219 return false;
2220
2221 if (gimple_call_internal_p (stmt)
2222 && (gimple_call_internal_fn (stmt) == IFN_MASK_LOAD
2223 || gimple_call_internal_fn (stmt) == IFN_MASK_STORE))
2224 return vectorizable_mask_load_store (stmt, gsi, vec_stmt,
2225 slp_node);
2226
2227 if (gimple_call_lhs (stmt) == NULL_TREE
2228 || TREE_CODE (gimple_call_lhs (stmt)) != SSA_NAME)
2229 return false;
2230
2231 gcc_checking_assert (!stmt_can_throw_internal (stmt));
2232
2233 vectype_out = STMT_VINFO_VECTYPE (stmt_info);
2234
2235 /* Process function arguments. */
2236 rhs_type = NULL_TREE;
2237 vectype_in = NULL_TREE;
2238 nargs = gimple_call_num_args (stmt);
2239
2240 /* Bail out if the function has more than three arguments, we do not have
2241 interesting builtin functions to vectorize with more than two arguments
2242 except for fma. No arguments is also not good. */
2243 if (nargs == 0 || nargs > 3)
2244 return false;
2245
2246 /* Ignore the argument of IFN_GOMP_SIMD_LANE, it is magic. */
2247 if (gimple_call_internal_p (stmt)
2248 && gimple_call_internal_fn (stmt) == IFN_GOMP_SIMD_LANE)
2249 {
2250 nargs = 0;
2251 rhs_type = unsigned_type_node;
2252 }
2253
2254 for (i = 0; i < nargs; i++)
2255 {
2256 tree opvectype;
2257
2258 op = gimple_call_arg (stmt, i);
2259
2260 /* We can only handle calls with arguments of the same type. */
2261 if (rhs_type
2262 && !types_compatible_p (rhs_type, TREE_TYPE (op)))
2263 {
2264 if (dump_enabled_p ())
2265 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2266 "argument types differ.\n");
2267 return false;
2268 }
2269 if (!rhs_type)
2270 rhs_type = TREE_TYPE (op);
2271
2272 if (!vect_is_simple_use_1 (op, stmt, loop_vinfo, bb_vinfo,
2273 &def_stmt, &def, &dt[i], &opvectype))
2274 {
2275 if (dump_enabled_p ())
2276 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2277 "use not simple.\n");
2278 return false;
2279 }
2280
2281 if (!vectype_in)
2282 vectype_in = opvectype;
2283 else if (opvectype
2284 && opvectype != vectype_in)
2285 {
2286 if (dump_enabled_p ())
2287 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2288 "argument vector types differ.\n");
2289 return false;
2290 }
2291 }
2292 /* If all arguments are external or constant defs use a vector type with
2293 the same size as the output vector type. */
2294 if (!vectype_in)
2295 vectype_in = get_same_sized_vectype (rhs_type, vectype_out);
2296 if (vec_stmt)
2297 gcc_assert (vectype_in);
2298 if (!vectype_in)
2299 {
2300 if (dump_enabled_p ())
2301 {
2302 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2303 "no vectype for scalar type ");
2304 dump_generic_expr (MSG_MISSED_OPTIMIZATION, TDF_SLIM, rhs_type);
2305 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
2306 }
2307
2308 return false;
2309 }
2310
2311 /* FORNOW */
2312 nunits_in = TYPE_VECTOR_SUBPARTS (vectype_in);
2313 nunits_out = TYPE_VECTOR_SUBPARTS (vectype_out);
2314 if (nunits_in == nunits_out / 2)
2315 modifier = NARROW;
2316 else if (nunits_out == nunits_in)
2317 modifier = NONE;
2318 else if (nunits_out == nunits_in / 2)
2319 modifier = WIDEN;
2320 else
2321 return false;
2322
2323 /* For now, we only vectorize functions if a target specific builtin
2324 is available. TODO -- in some cases, it might be profitable to
2325 insert the calls for pieces of the vector, in order to be able
2326 to vectorize other operations in the loop. */
2327 fndecl = vectorizable_function (stmt, vectype_out, vectype_in);
2328 if (fndecl == NULL_TREE)
2329 {
2330 if (gimple_call_internal_p (stmt)
2331 && gimple_call_internal_fn (stmt) == IFN_GOMP_SIMD_LANE
2332 && !slp_node
2333 && loop_vinfo
2334 && LOOP_VINFO_LOOP (loop_vinfo)->simduid
2335 && TREE_CODE (gimple_call_arg (stmt, 0)) == SSA_NAME
2336 && LOOP_VINFO_LOOP (loop_vinfo)->simduid
2337 == SSA_NAME_VAR (gimple_call_arg (stmt, 0)))
2338 {
2339 /* We can handle IFN_GOMP_SIMD_LANE by returning a
2340 { 0, 1, 2, ... vf - 1 } vector. */
2341 gcc_assert (nargs == 0);
2342 }
2343 else
2344 {
2345 if (dump_enabled_p ())
2346 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2347 "function is not vectorizable.\n");
2348 return false;
2349 }
2350 }
2351
2352 gcc_assert (!gimple_vuse (stmt));
2353
2354 if (slp_node || PURE_SLP_STMT (stmt_info))
2355 ncopies = 1;
2356 else if (modifier == NARROW)
2357 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_out;
2358 else
2359 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_in;
2360
2361 /* Sanity check: make sure that at least one copy of the vectorized stmt
2362 needs to be generated. */
2363 gcc_assert (ncopies >= 1);
2364
2365 if (!vec_stmt) /* transformation not required. */
2366 {
2367 STMT_VINFO_TYPE (stmt_info) = call_vec_info_type;
2368 if (dump_enabled_p ())
2369 dump_printf_loc (MSG_NOTE, vect_location, "=== vectorizable_call ==="
2370 "\n");
2371 vect_model_simple_cost (stmt_info, ncopies, dt, NULL, NULL);
2372 return true;
2373 }
2374
2375 /** Transform. **/
2376
2377 if (dump_enabled_p ())
2378 dump_printf_loc (MSG_NOTE, vect_location, "transform call.\n");
2379
2380 /* Handle def. */
2381 scalar_dest = gimple_call_lhs (stmt);
2382 vec_dest = vect_create_destination_var (scalar_dest, vectype_out);
2383
2384 prev_stmt_info = NULL;
2385 switch (modifier)
2386 {
2387 case NONE:
2388 for (j = 0; j < ncopies; ++j)
2389 {
2390 /* Build argument list for the vectorized call. */
2391 if (j == 0)
2392 vargs.create (nargs);
2393 else
2394 vargs.truncate (0);
2395
2396 if (slp_node)
2397 {
2398 auto_vec<vec<tree> > vec_defs (nargs);
2399 vec<tree> vec_oprnds0;
2400
2401 for (i = 0; i < nargs; i++)
2402 vargs.quick_push (gimple_call_arg (stmt, i));
2403 vect_get_slp_defs (vargs, slp_node, &vec_defs, -1);
2404 vec_oprnds0 = vec_defs[0];
2405
2406 /* Arguments are ready. Create the new vector stmt. */
2407 FOR_EACH_VEC_ELT (vec_oprnds0, i, vec_oprnd0)
2408 {
2409 size_t k;
2410 for (k = 0; k < nargs; k++)
2411 {
2412 vec<tree> vec_oprndsk = vec_defs[k];
2413 vargs[k] = vec_oprndsk[i];
2414 }
2415 new_stmt = gimple_build_call_vec (fndecl, vargs);
2416 new_temp = make_ssa_name (vec_dest, new_stmt);
2417 gimple_call_set_lhs (new_stmt, new_temp);
2418 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2419 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
2420 }
2421
2422 for (i = 0; i < nargs; i++)
2423 {
2424 vec<tree> vec_oprndsi = vec_defs[i];
2425 vec_oprndsi.release ();
2426 }
2427 continue;
2428 }
2429
2430 for (i = 0; i < nargs; i++)
2431 {
2432 op = gimple_call_arg (stmt, i);
2433 if (j == 0)
2434 vec_oprnd0
2435 = vect_get_vec_def_for_operand (op, stmt, NULL);
2436 else
2437 {
2438 vec_oprnd0 = gimple_call_arg (new_stmt, i);
2439 vec_oprnd0
2440 = vect_get_vec_def_for_stmt_copy (dt[i], vec_oprnd0);
2441 }
2442
2443 vargs.quick_push (vec_oprnd0);
2444 }
2445
2446 if (gimple_call_internal_p (stmt)
2447 && gimple_call_internal_fn (stmt) == IFN_GOMP_SIMD_LANE)
2448 {
2449 tree *v = XALLOCAVEC (tree, nunits_out);
2450 int k;
2451 for (k = 0; k < nunits_out; ++k)
2452 v[k] = build_int_cst (unsigned_type_node, j * nunits_out + k);
2453 tree cst = build_vector (vectype_out, v);
2454 tree new_var
2455 = vect_get_new_vect_var (vectype_out, vect_simple_var, "cst_");
2456 gimple init_stmt = gimple_build_assign (new_var, cst);
2457 new_temp = make_ssa_name (new_var, init_stmt);
2458 gimple_assign_set_lhs (init_stmt, new_temp);
2459 vect_init_vector_1 (stmt, init_stmt, NULL);
2460 new_temp = make_ssa_name (vec_dest, NULL);
2461 new_stmt = gimple_build_assign (new_temp,
2462 gimple_assign_lhs (init_stmt));
2463 }
2464 else
2465 {
2466 new_stmt = gimple_build_call_vec (fndecl, vargs);
2467 new_temp = make_ssa_name (vec_dest, new_stmt);
2468 gimple_call_set_lhs (new_stmt, new_temp);
2469 }
2470 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2471
2472 if (j == 0)
2473 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
2474 else
2475 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
2476
2477 prev_stmt_info = vinfo_for_stmt (new_stmt);
2478 }
2479
2480 break;
2481
2482 case NARROW:
2483 for (j = 0; j < ncopies; ++j)
2484 {
2485 /* Build argument list for the vectorized call. */
2486 if (j == 0)
2487 vargs.create (nargs * 2);
2488 else
2489 vargs.truncate (0);
2490
2491 if (slp_node)
2492 {
2493 auto_vec<vec<tree> > vec_defs (nargs);
2494 vec<tree> vec_oprnds0;
2495
2496 for (i = 0; i < nargs; i++)
2497 vargs.quick_push (gimple_call_arg (stmt, i));
2498 vect_get_slp_defs (vargs, slp_node, &vec_defs, -1);
2499 vec_oprnds0 = vec_defs[0];
2500
2501 /* Arguments are ready. Create the new vector stmt. */
2502 for (i = 0; vec_oprnds0.iterate (i, &vec_oprnd0); i += 2)
2503 {
2504 size_t k;
2505 vargs.truncate (0);
2506 for (k = 0; k < nargs; k++)
2507 {
2508 vec<tree> vec_oprndsk = vec_defs[k];
2509 vargs.quick_push (vec_oprndsk[i]);
2510 vargs.quick_push (vec_oprndsk[i + 1]);
2511 }
2512 new_stmt = gimple_build_call_vec (fndecl, vargs);
2513 new_temp = make_ssa_name (vec_dest, new_stmt);
2514 gimple_call_set_lhs (new_stmt, new_temp);
2515 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2516 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
2517 }
2518
2519 for (i = 0; i < nargs; i++)
2520 {
2521 vec<tree> vec_oprndsi = vec_defs[i];
2522 vec_oprndsi.release ();
2523 }
2524 continue;
2525 }
2526
2527 for (i = 0; i < nargs; i++)
2528 {
2529 op = gimple_call_arg (stmt, i);
2530 if (j == 0)
2531 {
2532 vec_oprnd0
2533 = vect_get_vec_def_for_operand (op, stmt, NULL);
2534 vec_oprnd1
2535 = vect_get_vec_def_for_stmt_copy (dt[i], vec_oprnd0);
2536 }
2537 else
2538 {
2539 vec_oprnd1 = gimple_call_arg (new_stmt, 2*i + 1);
2540 vec_oprnd0
2541 = vect_get_vec_def_for_stmt_copy (dt[i], vec_oprnd1);
2542 vec_oprnd1
2543 = vect_get_vec_def_for_stmt_copy (dt[i], vec_oprnd0);
2544 }
2545
2546 vargs.quick_push (vec_oprnd0);
2547 vargs.quick_push (vec_oprnd1);
2548 }
2549
2550 new_stmt = gimple_build_call_vec (fndecl, vargs);
2551 new_temp = make_ssa_name (vec_dest, new_stmt);
2552 gimple_call_set_lhs (new_stmt, new_temp);
2553 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2554
2555 if (j == 0)
2556 STMT_VINFO_VEC_STMT (stmt_info) = new_stmt;
2557 else
2558 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
2559
2560 prev_stmt_info = vinfo_for_stmt (new_stmt);
2561 }
2562
2563 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
2564
2565 break;
2566
2567 case WIDEN:
2568 /* No current target implements this case. */
2569 return false;
2570 }
2571
2572 vargs.release ();
2573
2574 /* The call in STMT might prevent it from being removed in dce.
2575 We however cannot remove it here, due to the way the ssa name
2576 it defines is mapped to the new definition. So just replace
2577 rhs of the statement with something harmless. */
2578
2579 if (slp_node)
2580 return true;
2581
2582 type = TREE_TYPE (scalar_dest);
2583 if (is_pattern_stmt_p (stmt_info))
2584 lhs = gimple_call_lhs (STMT_VINFO_RELATED_STMT (stmt_info));
2585 else
2586 lhs = gimple_call_lhs (stmt);
2587 new_stmt = gimple_build_assign (lhs, build_zero_cst (type));
2588 set_vinfo_for_stmt (new_stmt, stmt_info);
2589 set_vinfo_for_stmt (stmt, NULL);
2590 STMT_VINFO_STMT (stmt_info) = new_stmt;
2591 gsi_replace (gsi, new_stmt, false);
2592
2593 return true;
2594 }
2595
2596
2597 struct simd_call_arg_info
2598 {
2599 tree vectype;
2600 tree op;
2601 enum vect_def_type dt;
2602 HOST_WIDE_INT linear_step;
2603 unsigned int align;
2604 };
2605
2606 /* Function vectorizable_simd_clone_call.
2607
2608 Check if STMT performs a function call that can be vectorized
2609 by calling a simd clone of the function.
2610 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
2611 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
2612 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
2613
2614 static bool
2615 vectorizable_simd_clone_call (gimple stmt, gimple_stmt_iterator *gsi,
2616 gimple *vec_stmt, slp_tree slp_node)
2617 {
2618 tree vec_dest;
2619 tree scalar_dest;
2620 tree op, type;
2621 tree vec_oprnd0 = NULL_TREE;
2622 stmt_vec_info stmt_info = vinfo_for_stmt (stmt), prev_stmt_info;
2623 tree vectype;
2624 unsigned int nunits;
2625 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
2626 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
2627 struct loop *loop = loop_vinfo ? LOOP_VINFO_LOOP (loop_vinfo) : NULL;
2628 tree fndecl, new_temp, def;
2629 gimple def_stmt;
2630 gimple new_stmt = NULL;
2631 int ncopies, j;
2632 vec<simd_call_arg_info> arginfo = vNULL;
2633 vec<tree> vargs = vNULL;
2634 size_t i, nargs;
2635 tree lhs, rtype, ratype;
2636 vec<constructor_elt, va_gc> *ret_ctor_elts;
2637
2638 /* Is STMT a vectorizable call? */
2639 if (!is_gimple_call (stmt))
2640 return false;
2641
2642 fndecl = gimple_call_fndecl (stmt);
2643 if (fndecl == NULL_TREE)
2644 return false;
2645
2646 struct cgraph_node *node = cgraph_node::get (fndecl);
2647 if (node == NULL || node->simd_clones == NULL)
2648 return false;
2649
2650 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
2651 return false;
2652
2653 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
2654 return false;
2655
2656 if (gimple_call_lhs (stmt)
2657 && TREE_CODE (gimple_call_lhs (stmt)) != SSA_NAME)
2658 return false;
2659
2660 gcc_checking_assert (!stmt_can_throw_internal (stmt));
2661
2662 vectype = STMT_VINFO_VECTYPE (stmt_info);
2663
2664 if (loop_vinfo && nested_in_vect_loop_p (loop, stmt))
2665 return false;
2666
2667 /* FORNOW */
2668 if (slp_node || PURE_SLP_STMT (stmt_info))
2669 return false;
2670
2671 /* Process function arguments. */
2672 nargs = gimple_call_num_args (stmt);
2673
2674 /* Bail out if the function has zero arguments. */
2675 if (nargs == 0)
2676 return false;
2677
2678 arginfo.create (nargs);
2679
2680 for (i = 0; i < nargs; i++)
2681 {
2682 simd_call_arg_info thisarginfo;
2683 affine_iv iv;
2684
2685 thisarginfo.linear_step = 0;
2686 thisarginfo.align = 0;
2687 thisarginfo.op = NULL_TREE;
2688
2689 op = gimple_call_arg (stmt, i);
2690 if (!vect_is_simple_use_1 (op, stmt, loop_vinfo, bb_vinfo,
2691 &def_stmt, &def, &thisarginfo.dt,
2692 &thisarginfo.vectype)
2693 || thisarginfo.dt == vect_uninitialized_def)
2694 {
2695 if (dump_enabled_p ())
2696 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2697 "use not simple.\n");
2698 arginfo.release ();
2699 return false;
2700 }
2701
2702 if (thisarginfo.dt == vect_constant_def
2703 || thisarginfo.dt == vect_external_def)
2704 gcc_assert (thisarginfo.vectype == NULL_TREE);
2705 else
2706 gcc_assert (thisarginfo.vectype != NULL_TREE);
2707
2708 if (thisarginfo.dt != vect_constant_def
2709 && thisarginfo.dt != vect_external_def
2710 && loop_vinfo
2711 && TREE_CODE (op) == SSA_NAME
2712 && simple_iv (loop, loop_containing_stmt (stmt), op, &iv, false)
2713 && tree_fits_shwi_p (iv.step))
2714 {
2715 thisarginfo.linear_step = tree_to_shwi (iv.step);
2716 thisarginfo.op = iv.base;
2717 }
2718 else if ((thisarginfo.dt == vect_constant_def
2719 || thisarginfo.dt == vect_external_def)
2720 && POINTER_TYPE_P (TREE_TYPE (op)))
2721 thisarginfo.align = get_pointer_alignment (op) / BITS_PER_UNIT;
2722
2723 arginfo.quick_push (thisarginfo);
2724 }
2725
2726 unsigned int badness = 0;
2727 struct cgraph_node *bestn = NULL;
2728 if (STMT_VINFO_SIMD_CLONE_FNDECL (stmt_info))
2729 bestn = cgraph_node::get (STMT_VINFO_SIMD_CLONE_FNDECL (stmt_info));
2730 else
2731 for (struct cgraph_node *n = node->simd_clones; n != NULL;
2732 n = n->simdclone->next_clone)
2733 {
2734 unsigned int this_badness = 0;
2735 if (n->simdclone->simdlen
2736 > (unsigned) LOOP_VINFO_VECT_FACTOR (loop_vinfo)
2737 || n->simdclone->nargs != nargs)
2738 continue;
2739 if (n->simdclone->simdlen
2740 < (unsigned) LOOP_VINFO_VECT_FACTOR (loop_vinfo))
2741 this_badness += (exact_log2 (LOOP_VINFO_VECT_FACTOR (loop_vinfo))
2742 - exact_log2 (n->simdclone->simdlen)) * 1024;
2743 if (n->simdclone->inbranch)
2744 this_badness += 2048;
2745 int target_badness = targetm.simd_clone.usable (n);
2746 if (target_badness < 0)
2747 continue;
2748 this_badness += target_badness * 512;
2749 /* FORNOW: Have to add code to add the mask argument. */
2750 if (n->simdclone->inbranch)
2751 continue;
2752 for (i = 0; i < nargs; i++)
2753 {
2754 switch (n->simdclone->args[i].arg_type)
2755 {
2756 case SIMD_CLONE_ARG_TYPE_VECTOR:
2757 if (!useless_type_conversion_p
2758 (n->simdclone->args[i].orig_type,
2759 TREE_TYPE (gimple_call_arg (stmt, i))))
2760 i = -1;
2761 else if (arginfo[i].dt == vect_constant_def
2762 || arginfo[i].dt == vect_external_def
2763 || arginfo[i].linear_step)
2764 this_badness += 64;
2765 break;
2766 case SIMD_CLONE_ARG_TYPE_UNIFORM:
2767 if (arginfo[i].dt != vect_constant_def
2768 && arginfo[i].dt != vect_external_def)
2769 i = -1;
2770 break;
2771 case SIMD_CLONE_ARG_TYPE_LINEAR_CONSTANT_STEP:
2772 if (arginfo[i].dt == vect_constant_def
2773 || arginfo[i].dt == vect_external_def
2774 || (arginfo[i].linear_step
2775 != n->simdclone->args[i].linear_step))
2776 i = -1;
2777 break;
2778 case SIMD_CLONE_ARG_TYPE_LINEAR_VARIABLE_STEP:
2779 /* FORNOW */
2780 i = -1;
2781 break;
2782 case SIMD_CLONE_ARG_TYPE_MASK:
2783 gcc_unreachable ();
2784 }
2785 if (i == (size_t) -1)
2786 break;
2787 if (n->simdclone->args[i].alignment > arginfo[i].align)
2788 {
2789 i = -1;
2790 break;
2791 }
2792 if (arginfo[i].align)
2793 this_badness += (exact_log2 (arginfo[i].align)
2794 - exact_log2 (n->simdclone->args[i].alignment));
2795 }
2796 if (i == (size_t) -1)
2797 continue;
2798 if (bestn == NULL || this_badness < badness)
2799 {
2800 bestn = n;
2801 badness = this_badness;
2802 }
2803 }
2804
2805 if (bestn == NULL)
2806 {
2807 arginfo.release ();
2808 return false;
2809 }
2810
2811 for (i = 0; i < nargs; i++)
2812 if ((arginfo[i].dt == vect_constant_def
2813 || arginfo[i].dt == vect_external_def)
2814 && bestn->simdclone->args[i].arg_type == SIMD_CLONE_ARG_TYPE_VECTOR)
2815 {
2816 arginfo[i].vectype
2817 = get_vectype_for_scalar_type (TREE_TYPE (gimple_call_arg (stmt,
2818 i)));
2819 if (arginfo[i].vectype == NULL
2820 || (TYPE_VECTOR_SUBPARTS (arginfo[i].vectype)
2821 > bestn->simdclone->simdlen))
2822 {
2823 arginfo.release ();
2824 return false;
2825 }
2826 }
2827
2828 fndecl = bestn->decl;
2829 nunits = bestn->simdclone->simdlen;
2830 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits;
2831
2832 /* If the function isn't const, only allow it in simd loops where user
2833 has asserted that at least nunits consecutive iterations can be
2834 performed using SIMD instructions. */
2835 if ((loop == NULL || (unsigned) loop->safelen < nunits)
2836 && gimple_vuse (stmt))
2837 {
2838 arginfo.release ();
2839 return false;
2840 }
2841
2842 /* Sanity check: make sure that at least one copy of the vectorized stmt
2843 needs to be generated. */
2844 gcc_assert (ncopies >= 1);
2845
2846 if (!vec_stmt) /* transformation not required. */
2847 {
2848 STMT_VINFO_SIMD_CLONE_FNDECL (stmt_info) = bestn->decl;
2849 STMT_VINFO_TYPE (stmt_info) = call_simd_clone_vec_info_type;
2850 if (dump_enabled_p ())
2851 dump_printf_loc (MSG_NOTE, vect_location,
2852 "=== vectorizable_simd_clone_call ===\n");
2853 /* vect_model_simple_cost (stmt_info, ncopies, dt, NULL, NULL); */
2854 arginfo.release ();
2855 return true;
2856 }
2857
2858 /** Transform. **/
2859
2860 if (dump_enabled_p ())
2861 dump_printf_loc (MSG_NOTE, vect_location, "transform call.\n");
2862
2863 /* Handle def. */
2864 scalar_dest = gimple_call_lhs (stmt);
2865 vec_dest = NULL_TREE;
2866 rtype = NULL_TREE;
2867 ratype = NULL_TREE;
2868 if (scalar_dest)
2869 {
2870 vec_dest = vect_create_destination_var (scalar_dest, vectype);
2871 rtype = TREE_TYPE (TREE_TYPE (fndecl));
2872 if (TREE_CODE (rtype) == ARRAY_TYPE)
2873 {
2874 ratype = rtype;
2875 rtype = TREE_TYPE (ratype);
2876 }
2877 }
2878
2879 prev_stmt_info = NULL;
2880 for (j = 0; j < ncopies; ++j)
2881 {
2882 /* Build argument list for the vectorized call. */
2883 if (j == 0)
2884 vargs.create (nargs);
2885 else
2886 vargs.truncate (0);
2887
2888 for (i = 0; i < nargs; i++)
2889 {
2890 unsigned int k, l, m, o;
2891 tree atype;
2892 op = gimple_call_arg (stmt, i);
2893 switch (bestn->simdclone->args[i].arg_type)
2894 {
2895 case SIMD_CLONE_ARG_TYPE_VECTOR:
2896 atype = bestn->simdclone->args[i].vector_type;
2897 o = nunits / TYPE_VECTOR_SUBPARTS (atype);
2898 for (m = j * o; m < (j + 1) * o; m++)
2899 {
2900 if (TYPE_VECTOR_SUBPARTS (atype)
2901 < TYPE_VECTOR_SUBPARTS (arginfo[i].vectype))
2902 {
2903 unsigned int prec = GET_MODE_BITSIZE (TYPE_MODE (atype));
2904 k = (TYPE_VECTOR_SUBPARTS (arginfo[i].vectype)
2905 / TYPE_VECTOR_SUBPARTS (atype));
2906 gcc_assert ((k & (k - 1)) == 0);
2907 if (m == 0)
2908 vec_oprnd0
2909 = vect_get_vec_def_for_operand (op, stmt, NULL);
2910 else
2911 {
2912 vec_oprnd0 = arginfo[i].op;
2913 if ((m & (k - 1)) == 0)
2914 vec_oprnd0
2915 = vect_get_vec_def_for_stmt_copy (arginfo[i].dt,
2916 vec_oprnd0);
2917 }
2918 arginfo[i].op = vec_oprnd0;
2919 vec_oprnd0
2920 = build3 (BIT_FIELD_REF, atype, vec_oprnd0,
2921 size_int (prec),
2922 bitsize_int ((m & (k - 1)) * prec));
2923 new_stmt
2924 = gimple_build_assign (make_ssa_name (atype, NULL),
2925 vec_oprnd0);
2926 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2927 vargs.safe_push (gimple_assign_lhs (new_stmt));
2928 }
2929 else
2930 {
2931 k = (TYPE_VECTOR_SUBPARTS (atype)
2932 / TYPE_VECTOR_SUBPARTS (arginfo[i].vectype));
2933 gcc_assert ((k & (k - 1)) == 0);
2934 vec<constructor_elt, va_gc> *ctor_elts;
2935 if (k != 1)
2936 vec_alloc (ctor_elts, k);
2937 else
2938 ctor_elts = NULL;
2939 for (l = 0; l < k; l++)
2940 {
2941 if (m == 0 && l == 0)
2942 vec_oprnd0
2943 = vect_get_vec_def_for_operand (op, stmt, NULL);
2944 else
2945 vec_oprnd0
2946 = vect_get_vec_def_for_stmt_copy (arginfo[i].dt,
2947 arginfo[i].op);
2948 arginfo[i].op = vec_oprnd0;
2949 if (k == 1)
2950 break;
2951 CONSTRUCTOR_APPEND_ELT (ctor_elts, NULL_TREE,
2952 vec_oprnd0);
2953 }
2954 if (k == 1)
2955 vargs.safe_push (vec_oprnd0);
2956 else
2957 {
2958 vec_oprnd0 = build_constructor (atype, ctor_elts);
2959 new_stmt
2960 = gimple_build_assign (make_ssa_name (atype, NULL),
2961 vec_oprnd0);
2962 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2963 vargs.safe_push (gimple_assign_lhs (new_stmt));
2964 }
2965 }
2966 }
2967 break;
2968 case SIMD_CLONE_ARG_TYPE_UNIFORM:
2969 vargs.safe_push (op);
2970 break;
2971 case SIMD_CLONE_ARG_TYPE_LINEAR_CONSTANT_STEP:
2972 if (j == 0)
2973 {
2974 gimple_seq stmts;
2975 arginfo[i].op
2976 = force_gimple_operand (arginfo[i].op, &stmts, true,
2977 NULL_TREE);
2978 if (stmts != NULL)
2979 {
2980 basic_block new_bb;
2981 edge pe = loop_preheader_edge (loop);
2982 new_bb = gsi_insert_seq_on_edge_immediate (pe, stmts);
2983 gcc_assert (!new_bb);
2984 }
2985 tree phi_res = copy_ssa_name (op, NULL);
2986 gimple new_phi = create_phi_node (phi_res, loop->header);
2987 set_vinfo_for_stmt (new_phi,
2988 new_stmt_vec_info (new_phi, loop_vinfo,
2989 NULL));
2990 add_phi_arg (new_phi, arginfo[i].op,
2991 loop_preheader_edge (loop), UNKNOWN_LOCATION);
2992 enum tree_code code
2993 = POINTER_TYPE_P (TREE_TYPE (op))
2994 ? POINTER_PLUS_EXPR : PLUS_EXPR;
2995 tree type = POINTER_TYPE_P (TREE_TYPE (op))
2996 ? sizetype : TREE_TYPE (op);
2997 widest_int cst
2998 = wi::mul (bestn->simdclone->args[i].linear_step,
2999 ncopies * nunits);
3000 tree tcst = wide_int_to_tree (type, cst);
3001 tree phi_arg = copy_ssa_name (op, NULL);
3002 new_stmt = gimple_build_assign_with_ops (code, phi_arg,
3003 phi_res, tcst);
3004 gimple_stmt_iterator si = gsi_after_labels (loop->header);
3005 gsi_insert_after (&si, new_stmt, GSI_NEW_STMT);
3006 set_vinfo_for_stmt (new_stmt,
3007 new_stmt_vec_info (new_stmt, loop_vinfo,
3008 NULL));
3009 add_phi_arg (new_phi, phi_arg, loop_latch_edge (loop),
3010 UNKNOWN_LOCATION);
3011 arginfo[i].op = phi_res;
3012 vargs.safe_push (phi_res);
3013 }
3014 else
3015 {
3016 enum tree_code code
3017 = POINTER_TYPE_P (TREE_TYPE (op))
3018 ? POINTER_PLUS_EXPR : PLUS_EXPR;
3019 tree type = POINTER_TYPE_P (TREE_TYPE (op))
3020 ? sizetype : TREE_TYPE (op);
3021 widest_int cst
3022 = wi::mul (bestn->simdclone->args[i].linear_step,
3023 j * nunits);
3024 tree tcst = wide_int_to_tree (type, cst);
3025 new_temp = make_ssa_name (TREE_TYPE (op), NULL);
3026 new_stmt
3027 = gimple_build_assign_with_ops (code, new_temp,
3028 arginfo[i].op, tcst);
3029 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3030 vargs.safe_push (new_temp);
3031 }
3032 break;
3033 case SIMD_CLONE_ARG_TYPE_LINEAR_VARIABLE_STEP:
3034 default:
3035 gcc_unreachable ();
3036 }
3037 }
3038
3039 new_stmt = gimple_build_call_vec (fndecl, vargs);
3040 if (vec_dest)
3041 {
3042 gcc_assert (ratype || TYPE_VECTOR_SUBPARTS (rtype) == nunits);
3043 if (ratype)
3044 new_temp = create_tmp_var (ratype, NULL);
3045 else if (TYPE_VECTOR_SUBPARTS (vectype)
3046 == TYPE_VECTOR_SUBPARTS (rtype))
3047 new_temp = make_ssa_name (vec_dest, new_stmt);
3048 else
3049 new_temp = make_ssa_name (rtype, new_stmt);
3050 gimple_call_set_lhs (new_stmt, new_temp);
3051 }
3052 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3053
3054 if (vec_dest)
3055 {
3056 if (TYPE_VECTOR_SUBPARTS (vectype) < nunits)
3057 {
3058 unsigned int k, l;
3059 unsigned int prec = GET_MODE_BITSIZE (TYPE_MODE (vectype));
3060 k = nunits / TYPE_VECTOR_SUBPARTS (vectype);
3061 gcc_assert ((k & (k - 1)) == 0);
3062 for (l = 0; l < k; l++)
3063 {
3064 tree t;
3065 if (ratype)
3066 {
3067 t = build_fold_addr_expr (new_temp);
3068 t = build2 (MEM_REF, vectype, t,
3069 build_int_cst (TREE_TYPE (t),
3070 l * prec / BITS_PER_UNIT));
3071 }
3072 else
3073 t = build3 (BIT_FIELD_REF, vectype, new_temp,
3074 size_int (prec), bitsize_int (l * prec));
3075 new_stmt
3076 = gimple_build_assign (make_ssa_name (vectype, NULL), t);
3077 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3078 if (j == 0 && l == 0)
3079 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
3080 else
3081 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
3082
3083 prev_stmt_info = vinfo_for_stmt (new_stmt);
3084 }
3085
3086 if (ratype)
3087 {
3088 tree clobber = build_constructor (ratype, NULL);
3089 TREE_THIS_VOLATILE (clobber) = 1;
3090 new_stmt = gimple_build_assign (new_temp, clobber);
3091 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3092 }
3093 continue;
3094 }
3095 else if (TYPE_VECTOR_SUBPARTS (vectype) > nunits)
3096 {
3097 unsigned int k = (TYPE_VECTOR_SUBPARTS (vectype)
3098 / TYPE_VECTOR_SUBPARTS (rtype));
3099 gcc_assert ((k & (k - 1)) == 0);
3100 if ((j & (k - 1)) == 0)
3101 vec_alloc (ret_ctor_elts, k);
3102 if (ratype)
3103 {
3104 unsigned int m, o = nunits / TYPE_VECTOR_SUBPARTS (rtype);
3105 for (m = 0; m < o; m++)
3106 {
3107 tree tem = build4 (ARRAY_REF, rtype, new_temp,
3108 size_int (m), NULL_TREE, NULL_TREE);
3109 new_stmt
3110 = gimple_build_assign (make_ssa_name (rtype, NULL),
3111 tem);
3112 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3113 CONSTRUCTOR_APPEND_ELT (ret_ctor_elts, NULL_TREE,
3114 gimple_assign_lhs (new_stmt));
3115 }
3116 tree clobber = build_constructor (ratype, NULL);
3117 TREE_THIS_VOLATILE (clobber) = 1;
3118 new_stmt = gimple_build_assign (new_temp, clobber);
3119 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3120 }
3121 else
3122 CONSTRUCTOR_APPEND_ELT (ret_ctor_elts, NULL_TREE, new_temp);
3123 if ((j & (k - 1)) != k - 1)
3124 continue;
3125 vec_oprnd0 = build_constructor (vectype, ret_ctor_elts);
3126 new_stmt
3127 = gimple_build_assign (make_ssa_name (vec_dest, NULL),
3128 vec_oprnd0);
3129 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3130
3131 if ((unsigned) j == k - 1)
3132 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
3133 else
3134 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
3135
3136 prev_stmt_info = vinfo_for_stmt (new_stmt);
3137 continue;
3138 }
3139 else if (ratype)
3140 {
3141 tree t = build_fold_addr_expr (new_temp);
3142 t = build2 (MEM_REF, vectype, t,
3143 build_int_cst (TREE_TYPE (t), 0));
3144 new_stmt
3145 = gimple_build_assign (make_ssa_name (vec_dest, NULL), t);
3146 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3147 tree clobber = build_constructor (ratype, NULL);
3148 TREE_THIS_VOLATILE (clobber) = 1;
3149 vect_finish_stmt_generation (stmt,
3150 gimple_build_assign (new_temp,
3151 clobber), gsi);
3152 }
3153 }
3154
3155 if (j == 0)
3156 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
3157 else
3158 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
3159
3160 prev_stmt_info = vinfo_for_stmt (new_stmt);
3161 }
3162
3163 vargs.release ();
3164
3165 /* The call in STMT might prevent it from being removed in dce.
3166 We however cannot remove it here, due to the way the ssa name
3167 it defines is mapped to the new definition. So just replace
3168 rhs of the statement with something harmless. */
3169
3170 if (slp_node)
3171 return true;
3172
3173 if (scalar_dest)
3174 {
3175 type = TREE_TYPE (scalar_dest);
3176 if (is_pattern_stmt_p (stmt_info))
3177 lhs = gimple_call_lhs (STMT_VINFO_RELATED_STMT (stmt_info));
3178 else
3179 lhs = gimple_call_lhs (stmt);
3180 new_stmt = gimple_build_assign (lhs, build_zero_cst (type));
3181 }
3182 else
3183 new_stmt = gimple_build_nop ();
3184 set_vinfo_for_stmt (new_stmt, stmt_info);
3185 set_vinfo_for_stmt (stmt, NULL);
3186 STMT_VINFO_STMT (stmt_info) = new_stmt;
3187 gsi_replace (gsi, new_stmt, false);
3188 unlink_stmt_vdef (stmt);
3189
3190 return true;
3191 }
3192
3193
3194 /* Function vect_gen_widened_results_half
3195
3196 Create a vector stmt whose code, type, number of arguments, and result
3197 variable are CODE, OP_TYPE, and VEC_DEST, and its arguments are
3198 VEC_OPRND0 and VEC_OPRND1. The new vector stmt is to be inserted at BSI.
3199 In the case that CODE is a CALL_EXPR, this means that a call to DECL
3200 needs to be created (DECL is a function-decl of a target-builtin).
3201 STMT is the original scalar stmt that we are vectorizing. */
3202
3203 static gimple
3204 vect_gen_widened_results_half (enum tree_code code,
3205 tree decl,
3206 tree vec_oprnd0, tree vec_oprnd1, int op_type,
3207 tree vec_dest, gimple_stmt_iterator *gsi,
3208 gimple stmt)
3209 {
3210 gimple new_stmt;
3211 tree new_temp;
3212
3213 /* Generate half of the widened result: */
3214 if (code == CALL_EXPR)
3215 {
3216 /* Target specific support */
3217 if (op_type == binary_op)
3218 new_stmt = gimple_build_call (decl, 2, vec_oprnd0, vec_oprnd1);
3219 else
3220 new_stmt = gimple_build_call (decl, 1, vec_oprnd0);
3221 new_temp = make_ssa_name (vec_dest, new_stmt);
3222 gimple_call_set_lhs (new_stmt, new_temp);
3223 }
3224 else
3225 {
3226 /* Generic support */
3227 gcc_assert (op_type == TREE_CODE_LENGTH (code));
3228 if (op_type != binary_op)
3229 vec_oprnd1 = NULL;
3230 new_stmt = gimple_build_assign_with_ops (code, vec_dest, vec_oprnd0,
3231 vec_oprnd1);
3232 new_temp = make_ssa_name (vec_dest, new_stmt);
3233 gimple_assign_set_lhs (new_stmt, new_temp);
3234 }
3235 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3236
3237 return new_stmt;
3238 }
3239
3240
3241 /* Get vectorized definitions for loop-based vectorization. For the first
3242 operand we call vect_get_vec_def_for_operand() (with OPRND containing
3243 scalar operand), and for the rest we get a copy with
3244 vect_get_vec_def_for_stmt_copy() using the previous vector definition
3245 (stored in OPRND). See vect_get_vec_def_for_stmt_copy() for details.
3246 The vectors are collected into VEC_OPRNDS. */
3247
3248 static void
3249 vect_get_loop_based_defs (tree *oprnd, gimple stmt, enum vect_def_type dt,
3250 vec<tree> *vec_oprnds, int multi_step_cvt)
3251 {
3252 tree vec_oprnd;
3253
3254 /* Get first vector operand. */
3255 /* All the vector operands except the very first one (that is scalar oprnd)
3256 are stmt copies. */
3257 if (TREE_CODE (TREE_TYPE (*oprnd)) != VECTOR_TYPE)
3258 vec_oprnd = vect_get_vec_def_for_operand (*oprnd, stmt, NULL);
3259 else
3260 vec_oprnd = vect_get_vec_def_for_stmt_copy (dt, *oprnd);
3261
3262 vec_oprnds->quick_push (vec_oprnd);
3263
3264 /* Get second vector operand. */
3265 vec_oprnd = vect_get_vec_def_for_stmt_copy (dt, vec_oprnd);
3266 vec_oprnds->quick_push (vec_oprnd);
3267
3268 *oprnd = vec_oprnd;
3269
3270 /* For conversion in multiple steps, continue to get operands
3271 recursively. */
3272 if (multi_step_cvt)
3273 vect_get_loop_based_defs (oprnd, stmt, dt, vec_oprnds, multi_step_cvt - 1);
3274 }
3275
3276
3277 /* Create vectorized demotion statements for vector operands from VEC_OPRNDS.
3278 For multi-step conversions store the resulting vectors and call the function
3279 recursively. */
3280
3281 static void
3282 vect_create_vectorized_demotion_stmts (vec<tree> *vec_oprnds,
3283 int multi_step_cvt, gimple stmt,
3284 vec<tree> vec_dsts,
3285 gimple_stmt_iterator *gsi,
3286 slp_tree slp_node, enum tree_code code,
3287 stmt_vec_info *prev_stmt_info)
3288 {
3289 unsigned int i;
3290 tree vop0, vop1, new_tmp, vec_dest;
3291 gimple new_stmt;
3292 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
3293
3294 vec_dest = vec_dsts.pop ();
3295
3296 for (i = 0; i < vec_oprnds->length (); i += 2)
3297 {
3298 /* Create demotion operation. */
3299 vop0 = (*vec_oprnds)[i];
3300 vop1 = (*vec_oprnds)[i + 1];
3301 new_stmt = gimple_build_assign_with_ops (code, vec_dest, vop0, vop1);
3302 new_tmp = make_ssa_name (vec_dest, new_stmt);
3303 gimple_assign_set_lhs (new_stmt, new_tmp);
3304 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3305
3306 if (multi_step_cvt)
3307 /* Store the resulting vector for next recursive call. */
3308 (*vec_oprnds)[i/2] = new_tmp;
3309 else
3310 {
3311 /* This is the last step of the conversion sequence. Store the
3312 vectors in SLP_NODE or in vector info of the scalar statement
3313 (or in STMT_VINFO_RELATED_STMT chain). */
3314 if (slp_node)
3315 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
3316 else
3317 {
3318 if (!*prev_stmt_info)
3319 STMT_VINFO_VEC_STMT (stmt_info) = new_stmt;
3320 else
3321 STMT_VINFO_RELATED_STMT (*prev_stmt_info) = new_stmt;
3322
3323 *prev_stmt_info = vinfo_for_stmt (new_stmt);
3324 }
3325 }
3326 }
3327
3328 /* For multi-step demotion operations we first generate demotion operations
3329 from the source type to the intermediate types, and then combine the
3330 results (stored in VEC_OPRNDS) in demotion operation to the destination
3331 type. */
3332 if (multi_step_cvt)
3333 {
3334 /* At each level of recursion we have half of the operands we had at the
3335 previous level. */
3336 vec_oprnds->truncate ((i+1)/2);
3337 vect_create_vectorized_demotion_stmts (vec_oprnds, multi_step_cvt - 1,
3338 stmt, vec_dsts, gsi, slp_node,
3339 VEC_PACK_TRUNC_EXPR,
3340 prev_stmt_info);
3341 }
3342
3343 vec_dsts.quick_push (vec_dest);
3344 }
3345
3346
3347 /* Create vectorized promotion statements for vector operands from VEC_OPRNDS0
3348 and VEC_OPRNDS1 (for binary operations). For multi-step conversions store
3349 the resulting vectors and call the function recursively. */
3350
3351 static void
3352 vect_create_vectorized_promotion_stmts (vec<tree> *vec_oprnds0,
3353 vec<tree> *vec_oprnds1,
3354 gimple stmt, tree vec_dest,
3355 gimple_stmt_iterator *gsi,
3356 enum tree_code code1,
3357 enum tree_code code2, tree decl1,
3358 tree decl2, int op_type)
3359 {
3360 int i;
3361 tree vop0, vop1, new_tmp1, new_tmp2;
3362 gimple new_stmt1, new_stmt2;
3363 vec<tree> vec_tmp = vNULL;
3364
3365 vec_tmp.create (vec_oprnds0->length () * 2);
3366 FOR_EACH_VEC_ELT (*vec_oprnds0, i, vop0)
3367 {
3368 if (op_type == binary_op)
3369 vop1 = (*vec_oprnds1)[i];
3370 else
3371 vop1 = NULL_TREE;
3372
3373 /* Generate the two halves of promotion operation. */
3374 new_stmt1 = vect_gen_widened_results_half (code1, decl1, vop0, vop1,
3375 op_type, vec_dest, gsi, stmt);
3376 new_stmt2 = vect_gen_widened_results_half (code2, decl2, vop0, vop1,
3377 op_type, vec_dest, gsi, stmt);
3378 if (is_gimple_call (new_stmt1))
3379 {
3380 new_tmp1 = gimple_call_lhs (new_stmt1);
3381 new_tmp2 = gimple_call_lhs (new_stmt2);
3382 }
3383 else
3384 {
3385 new_tmp1 = gimple_assign_lhs (new_stmt1);
3386 new_tmp2 = gimple_assign_lhs (new_stmt2);
3387 }
3388
3389 /* Store the results for the next step. */
3390 vec_tmp.quick_push (new_tmp1);
3391 vec_tmp.quick_push (new_tmp2);
3392 }
3393
3394 vec_oprnds0->release ();
3395 *vec_oprnds0 = vec_tmp;
3396 }
3397
3398
3399 /* Check if STMT performs a conversion operation, that can be vectorized.
3400 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
3401 stmt to replace it, put it in VEC_STMT, and insert it at GSI.
3402 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
3403
3404 static bool
3405 vectorizable_conversion (gimple stmt, gimple_stmt_iterator *gsi,
3406 gimple *vec_stmt, slp_tree slp_node)
3407 {
3408 tree vec_dest;
3409 tree scalar_dest;
3410 tree op0, op1 = NULL_TREE;
3411 tree vec_oprnd0 = NULL_TREE, vec_oprnd1 = NULL_TREE;
3412 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
3413 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
3414 enum tree_code code, code1 = ERROR_MARK, code2 = ERROR_MARK;
3415 enum tree_code codecvt1 = ERROR_MARK, codecvt2 = ERROR_MARK;
3416 tree decl1 = NULL_TREE, decl2 = NULL_TREE;
3417 tree new_temp;
3418 tree def;
3419 gimple def_stmt;
3420 enum vect_def_type dt[2] = {vect_unknown_def_type, vect_unknown_def_type};
3421 gimple new_stmt = NULL;
3422 stmt_vec_info prev_stmt_info;
3423 int nunits_in;
3424 int nunits_out;
3425 tree vectype_out, vectype_in;
3426 int ncopies, i, j;
3427 tree lhs_type, rhs_type;
3428 enum { NARROW, NONE, WIDEN } modifier;
3429 vec<tree> vec_oprnds0 = vNULL;
3430 vec<tree> vec_oprnds1 = vNULL;
3431 tree vop0;
3432 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
3433 int multi_step_cvt = 0;
3434 vec<tree> vec_dsts = vNULL;
3435 vec<tree> interm_types = vNULL;
3436 tree last_oprnd, intermediate_type, cvt_type = NULL_TREE;
3437 int op_type;
3438 enum machine_mode rhs_mode;
3439 unsigned short fltsz;
3440
3441 /* Is STMT a vectorizable conversion? */
3442
3443 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
3444 return false;
3445
3446 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
3447 return false;
3448
3449 if (!is_gimple_assign (stmt))
3450 return false;
3451
3452 if (TREE_CODE (gimple_assign_lhs (stmt)) != SSA_NAME)
3453 return false;
3454
3455 code = gimple_assign_rhs_code (stmt);
3456 if (!CONVERT_EXPR_CODE_P (code)
3457 && code != FIX_TRUNC_EXPR
3458 && code != FLOAT_EXPR
3459 && code != WIDEN_MULT_EXPR
3460 && code != WIDEN_LSHIFT_EXPR)
3461 return false;
3462
3463 op_type = TREE_CODE_LENGTH (code);
3464
3465 /* Check types of lhs and rhs. */
3466 scalar_dest = gimple_assign_lhs (stmt);
3467 lhs_type = TREE_TYPE (scalar_dest);
3468 vectype_out = STMT_VINFO_VECTYPE (stmt_info);
3469
3470 op0 = gimple_assign_rhs1 (stmt);
3471 rhs_type = TREE_TYPE (op0);
3472
3473 if ((code != FIX_TRUNC_EXPR && code != FLOAT_EXPR)
3474 && !((INTEGRAL_TYPE_P (lhs_type)
3475 && INTEGRAL_TYPE_P (rhs_type))
3476 || (SCALAR_FLOAT_TYPE_P (lhs_type)
3477 && SCALAR_FLOAT_TYPE_P (rhs_type))))
3478 return false;
3479
3480 if ((INTEGRAL_TYPE_P (lhs_type)
3481 && (TYPE_PRECISION (lhs_type)
3482 != GET_MODE_PRECISION (TYPE_MODE (lhs_type))))
3483 || (INTEGRAL_TYPE_P (rhs_type)
3484 && (TYPE_PRECISION (rhs_type)
3485 != GET_MODE_PRECISION (TYPE_MODE (rhs_type)))))
3486 {
3487 if (dump_enabled_p ())
3488 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
3489 "type conversion to/from bit-precision unsupported."
3490 "\n");
3491 return false;
3492 }
3493
3494 /* Check the operands of the operation. */
3495 if (!vect_is_simple_use_1 (op0, stmt, loop_vinfo, bb_vinfo,
3496 &def_stmt, &def, &dt[0], &vectype_in))
3497 {
3498 if (dump_enabled_p ())
3499 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
3500 "use not simple.\n");
3501 return false;
3502 }
3503 if (op_type == binary_op)
3504 {
3505 bool ok;
3506
3507 op1 = gimple_assign_rhs2 (stmt);
3508 gcc_assert (code == WIDEN_MULT_EXPR || code == WIDEN_LSHIFT_EXPR);
3509 /* For WIDEN_MULT_EXPR, if OP0 is a constant, use the type of
3510 OP1. */
3511 if (CONSTANT_CLASS_P (op0))
3512 ok = vect_is_simple_use_1 (op1, stmt, loop_vinfo, bb_vinfo,
3513 &def_stmt, &def, &dt[1], &vectype_in);
3514 else
3515 ok = vect_is_simple_use (op1, stmt, loop_vinfo, bb_vinfo, &def_stmt,
3516 &def, &dt[1]);
3517
3518 if (!ok)
3519 {
3520 if (dump_enabled_p ())
3521 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
3522 "use not simple.\n");
3523 return false;
3524 }
3525 }
3526
3527 /* If op0 is an external or constant defs use a vector type of
3528 the same size as the output vector type. */
3529 if (!vectype_in)
3530 vectype_in = get_same_sized_vectype (rhs_type, vectype_out);
3531 if (vec_stmt)
3532 gcc_assert (vectype_in);
3533 if (!vectype_in)
3534 {
3535 if (dump_enabled_p ())
3536 {
3537 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
3538 "no vectype for scalar type ");
3539 dump_generic_expr (MSG_MISSED_OPTIMIZATION, TDF_SLIM, rhs_type);
3540 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
3541 }
3542
3543 return false;
3544 }
3545
3546 nunits_in = TYPE_VECTOR_SUBPARTS (vectype_in);
3547 nunits_out = TYPE_VECTOR_SUBPARTS (vectype_out);
3548 if (nunits_in < nunits_out)
3549 modifier = NARROW;
3550 else if (nunits_out == nunits_in)
3551 modifier = NONE;
3552 else
3553 modifier = WIDEN;
3554
3555 /* Multiple types in SLP are handled by creating the appropriate number of
3556 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
3557 case of SLP. */
3558 if (slp_node || PURE_SLP_STMT (stmt_info))
3559 ncopies = 1;
3560 else if (modifier == NARROW)
3561 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_out;
3562 else
3563 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_in;
3564
3565 /* Sanity check: make sure that at least one copy of the vectorized stmt
3566 needs to be generated. */
3567 gcc_assert (ncopies >= 1);
3568
3569 /* Supportable by target? */
3570 switch (modifier)
3571 {
3572 case NONE:
3573 if (code != FIX_TRUNC_EXPR && code != FLOAT_EXPR)
3574 return false;
3575 if (supportable_convert_operation (code, vectype_out, vectype_in,
3576 &decl1, &code1))
3577 break;
3578 /* FALLTHRU */
3579 unsupported:
3580 if (dump_enabled_p ())
3581 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
3582 "conversion not supported by target.\n");
3583 return false;
3584
3585 case WIDEN:
3586 if (supportable_widening_operation (code, stmt, vectype_out, vectype_in,
3587 &code1, &code2, &multi_step_cvt,
3588 &interm_types))
3589 {
3590 /* Binary widening operation can only be supported directly by the
3591 architecture. */
3592 gcc_assert (!(multi_step_cvt && op_type == binary_op));
3593 break;
3594 }
3595
3596 if (code != FLOAT_EXPR
3597 || (GET_MODE_SIZE (TYPE_MODE (lhs_type))
3598 <= GET_MODE_SIZE (TYPE_MODE (rhs_type))))
3599 goto unsupported;
3600
3601 rhs_mode = TYPE_MODE (rhs_type);
3602 fltsz = GET_MODE_SIZE (TYPE_MODE (lhs_type));
3603 for (rhs_mode = GET_MODE_2XWIDER_MODE (TYPE_MODE (rhs_type));
3604 rhs_mode != VOIDmode && GET_MODE_SIZE (rhs_mode) <= fltsz;
3605 rhs_mode = GET_MODE_2XWIDER_MODE (rhs_mode))
3606 {
3607 cvt_type
3608 = build_nonstandard_integer_type (GET_MODE_BITSIZE (rhs_mode), 0);
3609 cvt_type = get_same_sized_vectype (cvt_type, vectype_in);
3610 if (cvt_type == NULL_TREE)
3611 goto unsupported;
3612
3613 if (GET_MODE_SIZE (rhs_mode) == fltsz)
3614 {
3615 if (!supportable_convert_operation (code, vectype_out,
3616 cvt_type, &decl1, &codecvt1))
3617 goto unsupported;
3618 }
3619 else if (!supportable_widening_operation (code, stmt, vectype_out,
3620 cvt_type, &codecvt1,
3621 &codecvt2, &multi_step_cvt,
3622 &interm_types))
3623 continue;
3624 else
3625 gcc_assert (multi_step_cvt == 0);
3626
3627 if (supportable_widening_operation (NOP_EXPR, stmt, cvt_type,
3628 vectype_in, &code1, &code2,
3629 &multi_step_cvt, &interm_types))
3630 break;
3631 }
3632
3633 if (rhs_mode == VOIDmode || GET_MODE_SIZE (rhs_mode) > fltsz)
3634 goto unsupported;
3635
3636 if (GET_MODE_SIZE (rhs_mode) == fltsz)
3637 codecvt2 = ERROR_MARK;
3638 else
3639 {
3640 multi_step_cvt++;
3641 interm_types.safe_push (cvt_type);
3642 cvt_type = NULL_TREE;
3643 }
3644 break;
3645
3646 case NARROW:
3647 gcc_assert (op_type == unary_op);
3648 if (supportable_narrowing_operation (code, vectype_out, vectype_in,
3649 &code1, &multi_step_cvt,
3650 &interm_types))
3651 break;
3652
3653 if (code != FIX_TRUNC_EXPR
3654 || (GET_MODE_SIZE (TYPE_MODE (lhs_type))
3655 >= GET_MODE_SIZE (TYPE_MODE (rhs_type))))
3656 goto unsupported;
3657
3658 rhs_mode = TYPE_MODE (rhs_type);
3659 cvt_type
3660 = build_nonstandard_integer_type (GET_MODE_BITSIZE (rhs_mode), 0);
3661 cvt_type = get_same_sized_vectype (cvt_type, vectype_in);
3662 if (cvt_type == NULL_TREE)
3663 goto unsupported;
3664 if (!supportable_convert_operation (code, cvt_type, vectype_in,
3665 &decl1, &codecvt1))
3666 goto unsupported;
3667 if (supportable_narrowing_operation (NOP_EXPR, vectype_out, cvt_type,
3668 &code1, &multi_step_cvt,
3669 &interm_types))
3670 break;
3671 goto unsupported;
3672
3673 default:
3674 gcc_unreachable ();
3675 }
3676
3677 if (!vec_stmt) /* transformation not required. */
3678 {
3679 if (dump_enabled_p ())
3680 dump_printf_loc (MSG_NOTE, vect_location,
3681 "=== vectorizable_conversion ===\n");
3682 if (code == FIX_TRUNC_EXPR || code == FLOAT_EXPR)
3683 {
3684 STMT_VINFO_TYPE (stmt_info) = type_conversion_vec_info_type;
3685 vect_model_simple_cost (stmt_info, ncopies, dt, NULL, NULL);
3686 }
3687 else if (modifier == NARROW)
3688 {
3689 STMT_VINFO_TYPE (stmt_info) = type_demotion_vec_info_type;
3690 vect_model_promotion_demotion_cost (stmt_info, dt, multi_step_cvt);
3691 }
3692 else
3693 {
3694 STMT_VINFO_TYPE (stmt_info) = type_promotion_vec_info_type;
3695 vect_model_promotion_demotion_cost (stmt_info, dt, multi_step_cvt);
3696 }
3697 interm_types.release ();
3698 return true;
3699 }
3700
3701 /** Transform. **/
3702 if (dump_enabled_p ())
3703 dump_printf_loc (MSG_NOTE, vect_location,
3704 "transform conversion. ncopies = %d.\n", ncopies);
3705
3706 if (op_type == binary_op)
3707 {
3708 if (CONSTANT_CLASS_P (op0))
3709 op0 = fold_convert (TREE_TYPE (op1), op0);
3710 else if (CONSTANT_CLASS_P (op1))
3711 op1 = fold_convert (TREE_TYPE (op0), op1);
3712 }
3713
3714 /* In case of multi-step conversion, we first generate conversion operations
3715 to the intermediate types, and then from that types to the final one.
3716 We create vector destinations for the intermediate type (TYPES) received
3717 from supportable_*_operation, and store them in the correct order
3718 for future use in vect_create_vectorized_*_stmts (). */
3719 vec_dsts.create (multi_step_cvt + 1);
3720 vec_dest = vect_create_destination_var (scalar_dest,
3721 (cvt_type && modifier == WIDEN)
3722 ? cvt_type : vectype_out);
3723 vec_dsts.quick_push (vec_dest);
3724
3725 if (multi_step_cvt)
3726 {
3727 for (i = interm_types.length () - 1;
3728 interm_types.iterate (i, &intermediate_type); i--)
3729 {
3730 vec_dest = vect_create_destination_var (scalar_dest,
3731 intermediate_type);
3732 vec_dsts.quick_push (vec_dest);
3733 }
3734 }
3735
3736 if (cvt_type)
3737 vec_dest = vect_create_destination_var (scalar_dest,
3738 modifier == WIDEN
3739 ? vectype_out : cvt_type);
3740
3741 if (!slp_node)
3742 {
3743 if (modifier == WIDEN)
3744 {
3745 vec_oprnds0.create (multi_step_cvt ? vect_pow2 (multi_step_cvt) : 1);
3746 if (op_type == binary_op)
3747 vec_oprnds1.create (1);
3748 }
3749 else if (modifier == NARROW)
3750 vec_oprnds0.create (
3751 2 * (multi_step_cvt ? vect_pow2 (multi_step_cvt) : 1));
3752 }
3753 else if (code == WIDEN_LSHIFT_EXPR)
3754 vec_oprnds1.create (slp_node->vec_stmts_size);
3755
3756 last_oprnd = op0;
3757 prev_stmt_info = NULL;
3758 switch (modifier)
3759 {
3760 case NONE:
3761 for (j = 0; j < ncopies; j++)
3762 {
3763 if (j == 0)
3764 vect_get_vec_defs (op0, NULL, stmt, &vec_oprnds0, NULL, slp_node,
3765 -1);
3766 else
3767 vect_get_vec_defs_for_stmt_copy (dt, &vec_oprnds0, NULL);
3768
3769 FOR_EACH_VEC_ELT (vec_oprnds0, i, vop0)
3770 {
3771 /* Arguments are ready, create the new vector stmt. */
3772 if (code1 == CALL_EXPR)
3773 {
3774 new_stmt = gimple_build_call (decl1, 1, vop0);
3775 new_temp = make_ssa_name (vec_dest, new_stmt);
3776 gimple_call_set_lhs (new_stmt, new_temp);
3777 }
3778 else
3779 {
3780 gcc_assert (TREE_CODE_LENGTH (code1) == unary_op);
3781 new_stmt = gimple_build_assign_with_ops (code1, vec_dest,
3782 vop0, NULL);
3783 new_temp = make_ssa_name (vec_dest, new_stmt);
3784 gimple_assign_set_lhs (new_stmt, new_temp);
3785 }
3786
3787 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3788 if (slp_node)
3789 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
3790 }
3791
3792 if (j == 0)
3793 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
3794 else
3795 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
3796 prev_stmt_info = vinfo_for_stmt (new_stmt);
3797 }
3798 break;
3799
3800 case WIDEN:
3801 /* In case the vectorization factor (VF) is bigger than the number
3802 of elements that we can fit in a vectype (nunits), we have to
3803 generate more than one vector stmt - i.e - we need to "unroll"
3804 the vector stmt by a factor VF/nunits. */
3805 for (j = 0; j < ncopies; j++)
3806 {
3807 /* Handle uses. */
3808 if (j == 0)
3809 {
3810 if (slp_node)
3811 {
3812 if (code == WIDEN_LSHIFT_EXPR)
3813 {
3814 unsigned int k;
3815
3816 vec_oprnd1 = op1;
3817 /* Store vec_oprnd1 for every vector stmt to be created
3818 for SLP_NODE. We check during the analysis that all
3819 the shift arguments are the same. */
3820 for (k = 0; k < slp_node->vec_stmts_size - 1; k++)
3821 vec_oprnds1.quick_push (vec_oprnd1);
3822
3823 vect_get_vec_defs (op0, NULL_TREE, stmt, &vec_oprnds0, NULL,
3824 slp_node, -1);
3825 }
3826 else
3827 vect_get_vec_defs (op0, op1, stmt, &vec_oprnds0,
3828 &vec_oprnds1, slp_node, -1);
3829 }
3830 else
3831 {
3832 vec_oprnd0 = vect_get_vec_def_for_operand (op0, stmt, NULL);
3833 vec_oprnds0.quick_push (vec_oprnd0);
3834 if (op_type == binary_op)
3835 {
3836 if (code == WIDEN_LSHIFT_EXPR)
3837 vec_oprnd1 = op1;
3838 else
3839 vec_oprnd1 = vect_get_vec_def_for_operand (op1, stmt,
3840 NULL);
3841 vec_oprnds1.quick_push (vec_oprnd1);
3842 }
3843 }
3844 }
3845 else
3846 {
3847 vec_oprnd0 = vect_get_vec_def_for_stmt_copy (dt[0], vec_oprnd0);
3848 vec_oprnds0.truncate (0);
3849 vec_oprnds0.quick_push (vec_oprnd0);
3850 if (op_type == binary_op)
3851 {
3852 if (code == WIDEN_LSHIFT_EXPR)
3853 vec_oprnd1 = op1;
3854 else
3855 vec_oprnd1 = vect_get_vec_def_for_stmt_copy (dt[1],
3856 vec_oprnd1);
3857 vec_oprnds1.truncate (0);
3858 vec_oprnds1.quick_push (vec_oprnd1);
3859 }
3860 }
3861
3862 /* Arguments are ready. Create the new vector stmts. */
3863 for (i = multi_step_cvt; i >= 0; i--)
3864 {
3865 tree this_dest = vec_dsts[i];
3866 enum tree_code c1 = code1, c2 = code2;
3867 if (i == 0 && codecvt2 != ERROR_MARK)
3868 {
3869 c1 = codecvt1;
3870 c2 = codecvt2;
3871 }
3872 vect_create_vectorized_promotion_stmts (&vec_oprnds0,
3873 &vec_oprnds1,
3874 stmt, this_dest, gsi,
3875 c1, c2, decl1, decl2,
3876 op_type);
3877 }
3878
3879 FOR_EACH_VEC_ELT (vec_oprnds0, i, vop0)
3880 {
3881 if (cvt_type)
3882 {
3883 if (codecvt1 == CALL_EXPR)
3884 {
3885 new_stmt = gimple_build_call (decl1, 1, vop0);
3886 new_temp = make_ssa_name (vec_dest, new_stmt);
3887 gimple_call_set_lhs (new_stmt, new_temp);
3888 }
3889 else
3890 {
3891 gcc_assert (TREE_CODE_LENGTH (codecvt1) == unary_op);
3892 new_temp = make_ssa_name (vec_dest, NULL);
3893 new_stmt = gimple_build_assign_with_ops (codecvt1,
3894 new_temp,
3895 vop0, NULL);
3896 }
3897
3898 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3899 }
3900 else
3901 new_stmt = SSA_NAME_DEF_STMT (vop0);
3902
3903 if (slp_node)
3904 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
3905 else
3906 {
3907 if (!prev_stmt_info)
3908 STMT_VINFO_VEC_STMT (stmt_info) = new_stmt;
3909 else
3910 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
3911 prev_stmt_info = vinfo_for_stmt (new_stmt);
3912 }
3913 }
3914 }
3915
3916 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
3917 break;
3918
3919 case NARROW:
3920 /* In case the vectorization factor (VF) is bigger than the number
3921 of elements that we can fit in a vectype (nunits), we have to
3922 generate more than one vector stmt - i.e - we need to "unroll"
3923 the vector stmt by a factor VF/nunits. */
3924 for (j = 0; j < ncopies; j++)
3925 {
3926 /* Handle uses. */
3927 if (slp_node)
3928 vect_get_vec_defs (op0, NULL_TREE, stmt, &vec_oprnds0, NULL,
3929 slp_node, -1);
3930 else
3931 {
3932 vec_oprnds0.truncate (0);
3933 vect_get_loop_based_defs (&last_oprnd, stmt, dt[0], &vec_oprnds0,
3934 vect_pow2 (multi_step_cvt) - 1);
3935 }
3936
3937 /* Arguments are ready. Create the new vector stmts. */
3938 if (cvt_type)
3939 FOR_EACH_VEC_ELT (vec_oprnds0, i, vop0)
3940 {
3941 if (codecvt1 == CALL_EXPR)
3942 {
3943 new_stmt = gimple_build_call (decl1, 1, vop0);
3944 new_temp = make_ssa_name (vec_dest, new_stmt);
3945 gimple_call_set_lhs (new_stmt, new_temp);
3946 }
3947 else
3948 {
3949 gcc_assert (TREE_CODE_LENGTH (codecvt1) == unary_op);
3950 new_temp = make_ssa_name (vec_dest, NULL);
3951 new_stmt = gimple_build_assign_with_ops (codecvt1, new_temp,
3952 vop0, NULL);
3953 }
3954
3955 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3956 vec_oprnds0[i] = new_temp;
3957 }
3958
3959 vect_create_vectorized_demotion_stmts (&vec_oprnds0, multi_step_cvt,
3960 stmt, vec_dsts, gsi,
3961 slp_node, code1,
3962 &prev_stmt_info);
3963 }
3964
3965 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
3966 break;
3967 }
3968
3969 vec_oprnds0.release ();
3970 vec_oprnds1.release ();
3971 vec_dsts.release ();
3972 interm_types.release ();
3973
3974 return true;
3975 }
3976
3977
3978 /* Function vectorizable_assignment.
3979
3980 Check if STMT performs an assignment (copy) that can be vectorized.
3981 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
3982 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
3983 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
3984
3985 static bool
3986 vectorizable_assignment (gimple stmt, gimple_stmt_iterator *gsi,
3987 gimple *vec_stmt, slp_tree slp_node)
3988 {
3989 tree vec_dest;
3990 tree scalar_dest;
3991 tree op;
3992 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
3993 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
3994 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
3995 tree new_temp;
3996 tree def;
3997 gimple def_stmt;
3998 enum vect_def_type dt[2] = {vect_unknown_def_type, vect_unknown_def_type};
3999 unsigned int nunits = TYPE_VECTOR_SUBPARTS (vectype);
4000 int ncopies;
4001 int i, j;
4002 vec<tree> vec_oprnds = vNULL;
4003 tree vop;
4004 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
4005 gimple new_stmt = NULL;
4006 stmt_vec_info prev_stmt_info = NULL;
4007 enum tree_code code;
4008 tree vectype_in;
4009
4010 /* Multiple types in SLP are handled by creating the appropriate number of
4011 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
4012 case of SLP. */
4013 if (slp_node || PURE_SLP_STMT (stmt_info))
4014 ncopies = 1;
4015 else
4016 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits;
4017
4018 gcc_assert (ncopies >= 1);
4019
4020 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
4021 return false;
4022
4023 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
4024 return false;
4025
4026 /* Is vectorizable assignment? */
4027 if (!is_gimple_assign (stmt))
4028 return false;
4029
4030 scalar_dest = gimple_assign_lhs (stmt);
4031 if (TREE_CODE (scalar_dest) != SSA_NAME)
4032 return false;
4033
4034 code = gimple_assign_rhs_code (stmt);
4035 if (gimple_assign_single_p (stmt)
4036 || code == PAREN_EXPR
4037 || CONVERT_EXPR_CODE_P (code))
4038 op = gimple_assign_rhs1 (stmt);
4039 else
4040 return false;
4041
4042 if (code == VIEW_CONVERT_EXPR)
4043 op = TREE_OPERAND (op, 0);
4044
4045 if (!vect_is_simple_use_1 (op, stmt, loop_vinfo, bb_vinfo,
4046 &def_stmt, &def, &dt[0], &vectype_in))
4047 {
4048 if (dump_enabled_p ())
4049 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4050 "use not simple.\n");
4051 return false;
4052 }
4053
4054 /* We can handle NOP_EXPR conversions that do not change the number
4055 of elements or the vector size. */
4056 if ((CONVERT_EXPR_CODE_P (code)
4057 || code == VIEW_CONVERT_EXPR)
4058 && (!vectype_in
4059 || TYPE_VECTOR_SUBPARTS (vectype_in) != nunits
4060 || (GET_MODE_SIZE (TYPE_MODE (vectype))
4061 != GET_MODE_SIZE (TYPE_MODE (vectype_in)))))
4062 return false;
4063
4064 /* We do not handle bit-precision changes. */
4065 if ((CONVERT_EXPR_CODE_P (code)
4066 || code == VIEW_CONVERT_EXPR)
4067 && INTEGRAL_TYPE_P (TREE_TYPE (scalar_dest))
4068 && ((TYPE_PRECISION (TREE_TYPE (scalar_dest))
4069 != GET_MODE_PRECISION (TYPE_MODE (TREE_TYPE (scalar_dest))))
4070 || ((TYPE_PRECISION (TREE_TYPE (op))
4071 != GET_MODE_PRECISION (TYPE_MODE (TREE_TYPE (op))))))
4072 /* But a conversion that does not change the bit-pattern is ok. */
4073 && !((TYPE_PRECISION (TREE_TYPE (scalar_dest))
4074 > TYPE_PRECISION (TREE_TYPE (op)))
4075 && TYPE_UNSIGNED (TREE_TYPE (op))))
4076 {
4077 if (dump_enabled_p ())
4078 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4079 "type conversion to/from bit-precision "
4080 "unsupported.\n");
4081 return false;
4082 }
4083
4084 if (!vec_stmt) /* transformation not required. */
4085 {
4086 STMT_VINFO_TYPE (stmt_info) = assignment_vec_info_type;
4087 if (dump_enabled_p ())
4088 dump_printf_loc (MSG_NOTE, vect_location,
4089 "=== vectorizable_assignment ===\n");
4090 vect_model_simple_cost (stmt_info, ncopies, dt, NULL, NULL);
4091 return true;
4092 }
4093
4094 /** Transform. **/
4095 if (dump_enabled_p ())
4096 dump_printf_loc (MSG_NOTE, vect_location, "transform assignment.\n");
4097
4098 /* Handle def. */
4099 vec_dest = vect_create_destination_var (scalar_dest, vectype);
4100
4101 /* Handle use. */
4102 for (j = 0; j < ncopies; j++)
4103 {
4104 /* Handle uses. */
4105 if (j == 0)
4106 vect_get_vec_defs (op, NULL, stmt, &vec_oprnds, NULL, slp_node, -1);
4107 else
4108 vect_get_vec_defs_for_stmt_copy (dt, &vec_oprnds, NULL);
4109
4110 /* Arguments are ready. create the new vector stmt. */
4111 FOR_EACH_VEC_ELT (vec_oprnds, i, vop)
4112 {
4113 if (CONVERT_EXPR_CODE_P (code)
4114 || code == VIEW_CONVERT_EXPR)
4115 vop = build1 (VIEW_CONVERT_EXPR, vectype, vop);
4116 new_stmt = gimple_build_assign (vec_dest, vop);
4117 new_temp = make_ssa_name (vec_dest, new_stmt);
4118 gimple_assign_set_lhs (new_stmt, new_temp);
4119 vect_finish_stmt_generation (stmt, new_stmt, gsi);
4120 if (slp_node)
4121 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
4122 }
4123
4124 if (slp_node)
4125 continue;
4126
4127 if (j == 0)
4128 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
4129 else
4130 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
4131
4132 prev_stmt_info = vinfo_for_stmt (new_stmt);
4133 }
4134
4135 vec_oprnds.release ();
4136 return true;
4137 }
4138
4139
4140 /* Return TRUE if CODE (a shift operation) is supported for SCALAR_TYPE
4141 either as shift by a scalar or by a vector. */
4142
4143 bool
4144 vect_supportable_shift (enum tree_code code, tree scalar_type)
4145 {
4146
4147 enum machine_mode vec_mode;
4148 optab optab;
4149 int icode;
4150 tree vectype;
4151
4152 vectype = get_vectype_for_scalar_type (scalar_type);
4153 if (!vectype)
4154 return false;
4155
4156 optab = optab_for_tree_code (code, vectype, optab_scalar);
4157 if (!optab
4158 || optab_handler (optab, TYPE_MODE (vectype)) == CODE_FOR_nothing)
4159 {
4160 optab = optab_for_tree_code (code, vectype, optab_vector);
4161 if (!optab
4162 || (optab_handler (optab, TYPE_MODE (vectype))
4163 == CODE_FOR_nothing))
4164 return false;
4165 }
4166
4167 vec_mode = TYPE_MODE (vectype);
4168 icode = (int) optab_handler (optab, vec_mode);
4169 if (icode == CODE_FOR_nothing)
4170 return false;
4171
4172 return true;
4173 }
4174
4175
4176 /* Function vectorizable_shift.
4177
4178 Check if STMT performs a shift operation that can be vectorized.
4179 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
4180 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
4181 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
4182
4183 static bool
4184 vectorizable_shift (gimple stmt, gimple_stmt_iterator *gsi,
4185 gimple *vec_stmt, slp_tree slp_node)
4186 {
4187 tree vec_dest;
4188 tree scalar_dest;
4189 tree op0, op1 = NULL;
4190 tree vec_oprnd1 = NULL_TREE;
4191 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
4192 tree vectype;
4193 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
4194 enum tree_code code;
4195 enum machine_mode vec_mode;
4196 tree new_temp;
4197 optab optab;
4198 int icode;
4199 enum machine_mode optab_op2_mode;
4200 tree def;
4201 gimple def_stmt;
4202 enum vect_def_type dt[2] = {vect_unknown_def_type, vect_unknown_def_type};
4203 gimple new_stmt = NULL;
4204 stmt_vec_info prev_stmt_info;
4205 int nunits_in;
4206 int nunits_out;
4207 tree vectype_out;
4208 tree op1_vectype;
4209 int ncopies;
4210 int j, i;
4211 vec<tree> vec_oprnds0 = vNULL;
4212 vec<tree> vec_oprnds1 = vNULL;
4213 tree vop0, vop1;
4214 unsigned int k;
4215 bool scalar_shift_arg = true;
4216 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
4217 int vf;
4218
4219 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
4220 return false;
4221
4222 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
4223 return false;
4224
4225 /* Is STMT a vectorizable binary/unary operation? */
4226 if (!is_gimple_assign (stmt))
4227 return false;
4228
4229 if (TREE_CODE (gimple_assign_lhs (stmt)) != SSA_NAME)
4230 return false;
4231
4232 code = gimple_assign_rhs_code (stmt);
4233
4234 if (!(code == LSHIFT_EXPR || code == RSHIFT_EXPR || code == LROTATE_EXPR
4235 || code == RROTATE_EXPR))
4236 return false;
4237
4238 scalar_dest = gimple_assign_lhs (stmt);
4239 vectype_out = STMT_VINFO_VECTYPE (stmt_info);
4240 if (TYPE_PRECISION (TREE_TYPE (scalar_dest))
4241 != GET_MODE_PRECISION (TYPE_MODE (TREE_TYPE (scalar_dest))))
4242 {
4243 if (dump_enabled_p ())
4244 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4245 "bit-precision shifts not supported.\n");
4246 return false;
4247 }
4248
4249 op0 = gimple_assign_rhs1 (stmt);
4250 if (!vect_is_simple_use_1 (op0, stmt, loop_vinfo, bb_vinfo,
4251 &def_stmt, &def, &dt[0], &vectype))
4252 {
4253 if (dump_enabled_p ())
4254 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4255 "use not simple.\n");
4256 return false;
4257 }
4258 /* If op0 is an external or constant def use a vector type with
4259 the same size as the output vector type. */
4260 if (!vectype)
4261 vectype = get_same_sized_vectype (TREE_TYPE (op0), vectype_out);
4262 if (vec_stmt)
4263 gcc_assert (vectype);
4264 if (!vectype)
4265 {
4266 if (dump_enabled_p ())
4267 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4268 "no vectype for scalar type\n");
4269 return false;
4270 }
4271
4272 nunits_out = TYPE_VECTOR_SUBPARTS (vectype_out);
4273 nunits_in = TYPE_VECTOR_SUBPARTS (vectype);
4274 if (nunits_out != nunits_in)
4275 return false;
4276
4277 op1 = gimple_assign_rhs2 (stmt);
4278 if (!vect_is_simple_use_1 (op1, stmt, loop_vinfo, bb_vinfo, &def_stmt,
4279 &def, &dt[1], &op1_vectype))
4280 {
4281 if (dump_enabled_p ())
4282 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4283 "use not simple.\n");
4284 return false;
4285 }
4286
4287 if (loop_vinfo)
4288 vf = LOOP_VINFO_VECT_FACTOR (loop_vinfo);
4289 else
4290 vf = 1;
4291
4292 /* Multiple types in SLP are handled by creating the appropriate number of
4293 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
4294 case of SLP. */
4295 if (slp_node || PURE_SLP_STMT (stmt_info))
4296 ncopies = 1;
4297 else
4298 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_in;
4299
4300 gcc_assert (ncopies >= 1);
4301
4302 /* Determine whether the shift amount is a vector, or scalar. If the
4303 shift/rotate amount is a vector, use the vector/vector shift optabs. */
4304
4305 if (dt[1] == vect_internal_def && !slp_node)
4306 scalar_shift_arg = false;
4307 else if (dt[1] == vect_constant_def
4308 || dt[1] == vect_external_def
4309 || dt[1] == vect_internal_def)
4310 {
4311 /* In SLP, need to check whether the shift count is the same,
4312 in loops if it is a constant or invariant, it is always
4313 a scalar shift. */
4314 if (slp_node)
4315 {
4316 vec<gimple> stmts = SLP_TREE_SCALAR_STMTS (slp_node);
4317 gimple slpstmt;
4318
4319 FOR_EACH_VEC_ELT (stmts, k, slpstmt)
4320 if (!operand_equal_p (gimple_assign_rhs2 (slpstmt), op1, 0))
4321 scalar_shift_arg = false;
4322 }
4323 }
4324 else
4325 {
4326 if (dump_enabled_p ())
4327 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4328 "operand mode requires invariant argument.\n");
4329 return false;
4330 }
4331
4332 /* Vector shifted by vector. */
4333 if (!scalar_shift_arg)
4334 {
4335 optab = optab_for_tree_code (code, vectype, optab_vector);
4336 if (dump_enabled_p ())
4337 dump_printf_loc (MSG_NOTE, vect_location,
4338 "vector/vector shift/rotate found.\n");
4339
4340 if (!op1_vectype)
4341 op1_vectype = get_same_sized_vectype (TREE_TYPE (op1), vectype_out);
4342 if (op1_vectype == NULL_TREE
4343 || TYPE_MODE (op1_vectype) != TYPE_MODE (vectype))
4344 {
4345 if (dump_enabled_p ())
4346 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4347 "unusable type for last operand in"
4348 " vector/vector shift/rotate.\n");
4349 return false;
4350 }
4351 }
4352 /* See if the machine has a vector shifted by scalar insn and if not
4353 then see if it has a vector shifted by vector insn. */
4354 else
4355 {
4356 optab = optab_for_tree_code (code, vectype, optab_scalar);
4357 if (optab
4358 && optab_handler (optab, TYPE_MODE (vectype)) != CODE_FOR_nothing)
4359 {
4360 if (dump_enabled_p ())
4361 dump_printf_loc (MSG_NOTE, vect_location,
4362 "vector/scalar shift/rotate found.\n");
4363 }
4364 else
4365 {
4366 optab = optab_for_tree_code (code, vectype, optab_vector);
4367 if (optab
4368 && (optab_handler (optab, TYPE_MODE (vectype))
4369 != CODE_FOR_nothing))
4370 {
4371 scalar_shift_arg = false;
4372
4373 if (dump_enabled_p ())
4374 dump_printf_loc (MSG_NOTE, vect_location,
4375 "vector/vector shift/rotate found.\n");
4376
4377 /* Unlike the other binary operators, shifts/rotates have
4378 the rhs being int, instead of the same type as the lhs,
4379 so make sure the scalar is the right type if we are
4380 dealing with vectors of long long/long/short/char. */
4381 if (dt[1] == vect_constant_def)
4382 op1 = fold_convert (TREE_TYPE (vectype), op1);
4383 else if (!useless_type_conversion_p (TREE_TYPE (vectype),
4384 TREE_TYPE (op1)))
4385 {
4386 if (slp_node
4387 && TYPE_MODE (TREE_TYPE (vectype))
4388 != TYPE_MODE (TREE_TYPE (op1)))
4389 {
4390 if (dump_enabled_p ())
4391 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4392 "unusable type for last operand in"
4393 " vector/vector shift/rotate.\n");
4394 return false;
4395 }
4396 if (vec_stmt && !slp_node)
4397 {
4398 op1 = fold_convert (TREE_TYPE (vectype), op1);
4399 op1 = vect_init_vector (stmt, op1,
4400 TREE_TYPE (vectype), NULL);
4401 }
4402 }
4403 }
4404 }
4405 }
4406
4407 /* Supportable by target? */
4408 if (!optab)
4409 {
4410 if (dump_enabled_p ())
4411 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4412 "no optab.\n");
4413 return false;
4414 }
4415 vec_mode = TYPE_MODE (vectype);
4416 icode = (int) optab_handler (optab, vec_mode);
4417 if (icode == CODE_FOR_nothing)
4418 {
4419 if (dump_enabled_p ())
4420 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4421 "op not supported by target.\n");
4422 /* Check only during analysis. */
4423 if (GET_MODE_SIZE (vec_mode) != UNITS_PER_WORD
4424 || (vf < vect_min_worthwhile_factor (code)
4425 && !vec_stmt))
4426 return false;
4427 if (dump_enabled_p ())
4428 dump_printf_loc (MSG_NOTE, vect_location,
4429 "proceeding using word mode.\n");
4430 }
4431
4432 /* Worthwhile without SIMD support? Check only during analysis. */
4433 if (!VECTOR_MODE_P (TYPE_MODE (vectype))
4434 && vf < vect_min_worthwhile_factor (code)
4435 && !vec_stmt)
4436 {
4437 if (dump_enabled_p ())
4438 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4439 "not worthwhile without SIMD support.\n");
4440 return false;
4441 }
4442
4443 if (!vec_stmt) /* transformation not required. */
4444 {
4445 STMT_VINFO_TYPE (stmt_info) = shift_vec_info_type;
4446 if (dump_enabled_p ())
4447 dump_printf_loc (MSG_NOTE, vect_location,
4448 "=== vectorizable_shift ===\n");
4449 vect_model_simple_cost (stmt_info, ncopies, dt, NULL, NULL);
4450 return true;
4451 }
4452
4453 /** Transform. **/
4454
4455 if (dump_enabled_p ())
4456 dump_printf_loc (MSG_NOTE, vect_location,
4457 "transform binary/unary operation.\n");
4458
4459 /* Handle def. */
4460 vec_dest = vect_create_destination_var (scalar_dest, vectype);
4461
4462 prev_stmt_info = NULL;
4463 for (j = 0; j < ncopies; j++)
4464 {
4465 /* Handle uses. */
4466 if (j == 0)
4467 {
4468 if (scalar_shift_arg)
4469 {
4470 /* Vector shl and shr insn patterns can be defined with scalar
4471 operand 2 (shift operand). In this case, use constant or loop
4472 invariant op1 directly, without extending it to vector mode
4473 first. */
4474 optab_op2_mode = insn_data[icode].operand[2].mode;
4475 if (!VECTOR_MODE_P (optab_op2_mode))
4476 {
4477 if (dump_enabled_p ())
4478 dump_printf_loc (MSG_NOTE, vect_location,
4479 "operand 1 using scalar mode.\n");
4480 vec_oprnd1 = op1;
4481 vec_oprnds1.create (slp_node ? slp_node->vec_stmts_size : 1);
4482 vec_oprnds1.quick_push (vec_oprnd1);
4483 if (slp_node)
4484 {
4485 /* Store vec_oprnd1 for every vector stmt to be created
4486 for SLP_NODE. We check during the analysis that all
4487 the shift arguments are the same.
4488 TODO: Allow different constants for different vector
4489 stmts generated for an SLP instance. */
4490 for (k = 0; k < slp_node->vec_stmts_size - 1; k++)
4491 vec_oprnds1.quick_push (vec_oprnd1);
4492 }
4493 }
4494 }
4495
4496 /* vec_oprnd1 is available if operand 1 should be of a scalar-type
4497 (a special case for certain kind of vector shifts); otherwise,
4498 operand 1 should be of a vector type (the usual case). */
4499 if (vec_oprnd1)
4500 vect_get_vec_defs (op0, NULL_TREE, stmt, &vec_oprnds0, NULL,
4501 slp_node, -1);
4502 else
4503 vect_get_vec_defs (op0, op1, stmt, &vec_oprnds0, &vec_oprnds1,
4504 slp_node, -1);
4505 }
4506 else
4507 vect_get_vec_defs_for_stmt_copy (dt, &vec_oprnds0, &vec_oprnds1);
4508
4509 /* Arguments are ready. Create the new vector stmt. */
4510 FOR_EACH_VEC_ELT (vec_oprnds0, i, vop0)
4511 {
4512 vop1 = vec_oprnds1[i];
4513 new_stmt = gimple_build_assign_with_ops (code, vec_dest, vop0, vop1);
4514 new_temp = make_ssa_name (vec_dest, new_stmt);
4515 gimple_assign_set_lhs (new_stmt, new_temp);
4516 vect_finish_stmt_generation (stmt, new_stmt, gsi);
4517 if (slp_node)
4518 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
4519 }
4520
4521 if (slp_node)
4522 continue;
4523
4524 if (j == 0)
4525 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
4526 else
4527 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
4528 prev_stmt_info = vinfo_for_stmt (new_stmt);
4529 }
4530
4531 vec_oprnds0.release ();
4532 vec_oprnds1.release ();
4533
4534 return true;
4535 }
4536
4537
4538 /* Function vectorizable_operation.
4539
4540 Check if STMT performs a binary, unary or ternary operation that can
4541 be vectorized.
4542 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
4543 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
4544 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
4545
4546 static bool
4547 vectorizable_operation (gimple stmt, gimple_stmt_iterator *gsi,
4548 gimple *vec_stmt, slp_tree slp_node)
4549 {
4550 tree vec_dest;
4551 tree scalar_dest;
4552 tree op0, op1 = NULL_TREE, op2 = NULL_TREE;
4553 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
4554 tree vectype;
4555 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
4556 enum tree_code code;
4557 enum machine_mode vec_mode;
4558 tree new_temp;
4559 int op_type;
4560 optab optab;
4561 int icode;
4562 tree def;
4563 gimple def_stmt;
4564 enum vect_def_type dt[3]
4565 = {vect_unknown_def_type, vect_unknown_def_type, vect_unknown_def_type};
4566 gimple new_stmt = NULL;
4567 stmt_vec_info prev_stmt_info;
4568 int nunits_in;
4569 int nunits_out;
4570 tree vectype_out;
4571 int ncopies;
4572 int j, i;
4573 vec<tree> vec_oprnds0 = vNULL;
4574 vec<tree> vec_oprnds1 = vNULL;
4575 vec<tree> vec_oprnds2 = vNULL;
4576 tree vop0, vop1, vop2;
4577 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
4578 int vf;
4579
4580 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
4581 return false;
4582
4583 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
4584 return false;
4585
4586 /* Is STMT a vectorizable binary/unary operation? */
4587 if (!is_gimple_assign (stmt))
4588 return false;
4589
4590 if (TREE_CODE (gimple_assign_lhs (stmt)) != SSA_NAME)
4591 return false;
4592
4593 code = gimple_assign_rhs_code (stmt);
4594
4595 /* For pointer addition, we should use the normal plus for
4596 the vector addition. */
4597 if (code == POINTER_PLUS_EXPR)
4598 code = PLUS_EXPR;
4599
4600 /* Support only unary or binary operations. */
4601 op_type = TREE_CODE_LENGTH (code);
4602 if (op_type != unary_op && op_type != binary_op && op_type != ternary_op)
4603 {
4604 if (dump_enabled_p ())
4605 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4606 "num. args = %d (not unary/binary/ternary op).\n",
4607 op_type);
4608 return false;
4609 }
4610
4611 scalar_dest = gimple_assign_lhs (stmt);
4612 vectype_out = STMT_VINFO_VECTYPE (stmt_info);
4613
4614 /* Most operations cannot handle bit-precision types without extra
4615 truncations. */
4616 if ((TYPE_PRECISION (TREE_TYPE (scalar_dest))
4617 != GET_MODE_PRECISION (TYPE_MODE (TREE_TYPE (scalar_dest))))
4618 /* Exception are bitwise binary operations. */
4619 && code != BIT_IOR_EXPR
4620 && code != BIT_XOR_EXPR
4621 && code != BIT_AND_EXPR)
4622 {
4623 if (dump_enabled_p ())
4624 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4625 "bit-precision arithmetic not supported.\n");
4626 return false;
4627 }
4628
4629 op0 = gimple_assign_rhs1 (stmt);
4630 if (!vect_is_simple_use_1 (op0, stmt, loop_vinfo, bb_vinfo,
4631 &def_stmt, &def, &dt[0], &vectype))
4632 {
4633 if (dump_enabled_p ())
4634 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4635 "use not simple.\n");
4636 return false;
4637 }
4638 /* If op0 is an external or constant def use a vector type with
4639 the same size as the output vector type. */
4640 if (!vectype)
4641 vectype = get_same_sized_vectype (TREE_TYPE (op0), vectype_out);
4642 if (vec_stmt)
4643 gcc_assert (vectype);
4644 if (!vectype)
4645 {
4646 if (dump_enabled_p ())
4647 {
4648 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4649 "no vectype for scalar type ");
4650 dump_generic_expr (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
4651 TREE_TYPE (op0));
4652 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
4653 }
4654
4655 return false;
4656 }
4657
4658 nunits_out = TYPE_VECTOR_SUBPARTS (vectype_out);
4659 nunits_in = TYPE_VECTOR_SUBPARTS (vectype);
4660 if (nunits_out != nunits_in)
4661 return false;
4662
4663 if (op_type == binary_op || op_type == ternary_op)
4664 {
4665 op1 = gimple_assign_rhs2 (stmt);
4666 if (!vect_is_simple_use (op1, stmt, loop_vinfo, bb_vinfo, &def_stmt,
4667 &def, &dt[1]))
4668 {
4669 if (dump_enabled_p ())
4670 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4671 "use not simple.\n");
4672 return false;
4673 }
4674 }
4675 if (op_type == ternary_op)
4676 {
4677 op2 = gimple_assign_rhs3 (stmt);
4678 if (!vect_is_simple_use (op2, stmt, loop_vinfo, bb_vinfo, &def_stmt,
4679 &def, &dt[2]))
4680 {
4681 if (dump_enabled_p ())
4682 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4683 "use not simple.\n");
4684 return false;
4685 }
4686 }
4687
4688 if (loop_vinfo)
4689 vf = LOOP_VINFO_VECT_FACTOR (loop_vinfo);
4690 else
4691 vf = 1;
4692
4693 /* Multiple types in SLP are handled by creating the appropriate number of
4694 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
4695 case of SLP. */
4696 if (slp_node || PURE_SLP_STMT (stmt_info))
4697 ncopies = 1;
4698 else
4699 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_in;
4700
4701 gcc_assert (ncopies >= 1);
4702
4703 /* Shifts are handled in vectorizable_shift (). */
4704 if (code == LSHIFT_EXPR || code == RSHIFT_EXPR || code == LROTATE_EXPR
4705 || code == RROTATE_EXPR)
4706 return false;
4707
4708 /* Supportable by target? */
4709
4710 vec_mode = TYPE_MODE (vectype);
4711 if (code == MULT_HIGHPART_EXPR)
4712 {
4713 if (can_mult_highpart_p (vec_mode, TYPE_UNSIGNED (vectype)))
4714 icode = LAST_INSN_CODE;
4715 else
4716 icode = CODE_FOR_nothing;
4717 }
4718 else
4719 {
4720 optab = optab_for_tree_code (code, vectype, optab_default);
4721 if (!optab)
4722 {
4723 if (dump_enabled_p ())
4724 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4725 "no optab.\n");
4726 return false;
4727 }
4728 icode = (int) optab_handler (optab, vec_mode);
4729 }
4730
4731 if (icode == CODE_FOR_nothing)
4732 {
4733 if (dump_enabled_p ())
4734 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4735 "op not supported by target.\n");
4736 /* Check only during analysis. */
4737 if (GET_MODE_SIZE (vec_mode) != UNITS_PER_WORD
4738 || (!vec_stmt && vf < vect_min_worthwhile_factor (code)))
4739 return false;
4740 if (dump_enabled_p ())
4741 dump_printf_loc (MSG_NOTE, vect_location,
4742 "proceeding using word mode.\n");
4743 }
4744
4745 /* Worthwhile without SIMD support? Check only during analysis. */
4746 if (!VECTOR_MODE_P (vec_mode)
4747 && !vec_stmt
4748 && vf < vect_min_worthwhile_factor (code))
4749 {
4750 if (dump_enabled_p ())
4751 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4752 "not worthwhile without SIMD support.\n");
4753 return false;
4754 }
4755
4756 if (!vec_stmt) /* transformation not required. */
4757 {
4758 STMT_VINFO_TYPE (stmt_info) = op_vec_info_type;
4759 if (dump_enabled_p ())
4760 dump_printf_loc (MSG_NOTE, vect_location,
4761 "=== vectorizable_operation ===\n");
4762 vect_model_simple_cost (stmt_info, ncopies, dt, NULL, NULL);
4763 return true;
4764 }
4765
4766 /** Transform. **/
4767
4768 if (dump_enabled_p ())
4769 dump_printf_loc (MSG_NOTE, vect_location,
4770 "transform binary/unary operation.\n");
4771
4772 /* Handle def. */
4773 vec_dest = vect_create_destination_var (scalar_dest, vectype);
4774
4775 /* In case the vectorization factor (VF) is bigger than the number
4776 of elements that we can fit in a vectype (nunits), we have to generate
4777 more than one vector stmt - i.e - we need to "unroll" the
4778 vector stmt by a factor VF/nunits. In doing so, we record a pointer
4779 from one copy of the vector stmt to the next, in the field
4780 STMT_VINFO_RELATED_STMT. This is necessary in order to allow following
4781 stages to find the correct vector defs to be used when vectorizing
4782 stmts that use the defs of the current stmt. The example below
4783 illustrates the vectorization process when VF=16 and nunits=4 (i.e.,
4784 we need to create 4 vectorized stmts):
4785
4786 before vectorization:
4787 RELATED_STMT VEC_STMT
4788 S1: x = memref - -
4789 S2: z = x + 1 - -
4790
4791 step 1: vectorize stmt S1 (done in vectorizable_load. See more details
4792 there):
4793 RELATED_STMT VEC_STMT
4794 VS1_0: vx0 = memref0 VS1_1 -
4795 VS1_1: vx1 = memref1 VS1_2 -
4796 VS1_2: vx2 = memref2 VS1_3 -
4797 VS1_3: vx3 = memref3 - -
4798 S1: x = load - VS1_0
4799 S2: z = x + 1 - -
4800
4801 step2: vectorize stmt S2 (done here):
4802 To vectorize stmt S2 we first need to find the relevant vector
4803 def for the first operand 'x'. This is, as usual, obtained from
4804 the vector stmt recorded in the STMT_VINFO_VEC_STMT of the stmt
4805 that defines 'x' (S1). This way we find the stmt VS1_0, and the
4806 relevant vector def 'vx0'. Having found 'vx0' we can generate
4807 the vector stmt VS2_0, and as usual, record it in the
4808 STMT_VINFO_VEC_STMT of stmt S2.
4809 When creating the second copy (VS2_1), we obtain the relevant vector
4810 def from the vector stmt recorded in the STMT_VINFO_RELATED_STMT of
4811 stmt VS1_0. This way we find the stmt VS1_1 and the relevant
4812 vector def 'vx1'. Using 'vx1' we create stmt VS2_1 and record a
4813 pointer to it in the STMT_VINFO_RELATED_STMT of the vector stmt VS2_0.
4814 Similarly when creating stmts VS2_2 and VS2_3. This is the resulting
4815 chain of stmts and pointers:
4816 RELATED_STMT VEC_STMT
4817 VS1_0: vx0 = memref0 VS1_1 -
4818 VS1_1: vx1 = memref1 VS1_2 -
4819 VS1_2: vx2 = memref2 VS1_3 -
4820 VS1_3: vx3 = memref3 - -
4821 S1: x = load - VS1_0
4822 VS2_0: vz0 = vx0 + v1 VS2_1 -
4823 VS2_1: vz1 = vx1 + v1 VS2_2 -
4824 VS2_2: vz2 = vx2 + v1 VS2_3 -
4825 VS2_3: vz3 = vx3 + v1 - -
4826 S2: z = x + 1 - VS2_0 */
4827
4828 prev_stmt_info = NULL;
4829 for (j = 0; j < ncopies; j++)
4830 {
4831 /* Handle uses. */
4832 if (j == 0)
4833 {
4834 if (op_type == binary_op || op_type == ternary_op)
4835 vect_get_vec_defs (op0, op1, stmt, &vec_oprnds0, &vec_oprnds1,
4836 slp_node, -1);
4837 else
4838 vect_get_vec_defs (op0, NULL_TREE, stmt, &vec_oprnds0, NULL,
4839 slp_node, -1);
4840 if (op_type == ternary_op)
4841 {
4842 vec_oprnds2.create (1);
4843 vec_oprnds2.quick_push (vect_get_vec_def_for_operand (op2,
4844 stmt,
4845 NULL));
4846 }
4847 }
4848 else
4849 {
4850 vect_get_vec_defs_for_stmt_copy (dt, &vec_oprnds0, &vec_oprnds1);
4851 if (op_type == ternary_op)
4852 {
4853 tree vec_oprnd = vec_oprnds2.pop ();
4854 vec_oprnds2.quick_push (vect_get_vec_def_for_stmt_copy (dt[2],
4855 vec_oprnd));
4856 }
4857 }
4858
4859 /* Arguments are ready. Create the new vector stmt. */
4860 FOR_EACH_VEC_ELT (vec_oprnds0, i, vop0)
4861 {
4862 vop1 = ((op_type == binary_op || op_type == ternary_op)
4863 ? vec_oprnds1[i] : NULL_TREE);
4864 vop2 = ((op_type == ternary_op)
4865 ? vec_oprnds2[i] : NULL_TREE);
4866 new_stmt = gimple_build_assign_with_ops (code, vec_dest,
4867 vop0, vop1, vop2);
4868 new_temp = make_ssa_name (vec_dest, new_stmt);
4869 gimple_assign_set_lhs (new_stmt, new_temp);
4870 vect_finish_stmt_generation (stmt, new_stmt, gsi);
4871 if (slp_node)
4872 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
4873 }
4874
4875 if (slp_node)
4876 continue;
4877
4878 if (j == 0)
4879 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
4880 else
4881 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
4882 prev_stmt_info = vinfo_for_stmt (new_stmt);
4883 }
4884
4885 vec_oprnds0.release ();
4886 vec_oprnds1.release ();
4887 vec_oprnds2.release ();
4888
4889 return true;
4890 }
4891
4892 /* A helper function to ensure data reference DR's base alignment
4893 for STMT_INFO. */
4894
4895 static void
4896 ensure_base_align (stmt_vec_info stmt_info, struct data_reference *dr)
4897 {
4898 if (!dr->aux)
4899 return;
4900
4901 if (((dataref_aux *)dr->aux)->base_misaligned)
4902 {
4903 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
4904 tree base_decl = ((dataref_aux *)dr->aux)->base_decl;
4905
4906 DECL_ALIGN (base_decl) = TYPE_ALIGN (vectype);
4907 DECL_USER_ALIGN (base_decl) = 1;
4908 ((dataref_aux *)dr->aux)->base_misaligned = false;
4909 }
4910 }
4911
4912
4913 /* Given a vector type VECTYPE returns the VECTOR_CST mask that implements
4914 reversal of the vector elements. If that is impossible to do,
4915 returns NULL. */
4916
4917 static tree
4918 perm_mask_for_reverse (tree vectype)
4919 {
4920 int i, nunits;
4921 unsigned char *sel;
4922
4923 nunits = TYPE_VECTOR_SUBPARTS (vectype);
4924 sel = XALLOCAVEC (unsigned char, nunits);
4925
4926 for (i = 0; i < nunits; ++i)
4927 sel[i] = nunits - 1 - i;
4928
4929 return vect_gen_perm_mask (vectype, sel);
4930 }
4931
4932 /* Function vectorizable_store.
4933
4934 Check if STMT defines a non scalar data-ref (array/pointer/structure) that
4935 can be vectorized.
4936 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
4937 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
4938 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
4939
4940 static bool
4941 vectorizable_store (gimple stmt, gimple_stmt_iterator *gsi, gimple *vec_stmt,
4942 slp_tree slp_node)
4943 {
4944 tree scalar_dest;
4945 tree data_ref;
4946 tree op;
4947 tree vec_oprnd = NULL_TREE;
4948 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
4949 struct data_reference *dr = STMT_VINFO_DATA_REF (stmt_info), *first_dr = NULL;
4950 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
4951 tree elem_type;
4952 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
4953 struct loop *loop = NULL;
4954 enum machine_mode vec_mode;
4955 tree dummy;
4956 enum dr_alignment_support alignment_support_scheme;
4957 tree def;
4958 gimple def_stmt;
4959 enum vect_def_type dt;
4960 stmt_vec_info prev_stmt_info = NULL;
4961 tree dataref_ptr = NULL_TREE;
4962 tree dataref_offset = NULL_TREE;
4963 gimple ptr_incr = NULL;
4964 int nunits = TYPE_VECTOR_SUBPARTS (vectype);
4965 int ncopies;
4966 int j;
4967 gimple next_stmt, first_stmt = NULL;
4968 bool grouped_store = false;
4969 bool store_lanes_p = false;
4970 unsigned int group_size, i;
4971 vec<tree> dr_chain = vNULL;
4972 vec<tree> oprnds = vNULL;
4973 vec<tree> result_chain = vNULL;
4974 bool inv_p;
4975 bool negative = false;
4976 tree offset = NULL_TREE;
4977 vec<tree> vec_oprnds = vNULL;
4978 bool slp = (slp_node != NULL);
4979 unsigned int vec_num;
4980 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
4981 tree aggr_type;
4982
4983 if (loop_vinfo)
4984 loop = LOOP_VINFO_LOOP (loop_vinfo);
4985
4986 /* Multiple types in SLP are handled by creating the appropriate number of
4987 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
4988 case of SLP. */
4989 if (slp || PURE_SLP_STMT (stmt_info))
4990 ncopies = 1;
4991 else
4992 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits;
4993
4994 gcc_assert (ncopies >= 1);
4995
4996 /* FORNOW. This restriction should be relaxed. */
4997 if (loop && nested_in_vect_loop_p (loop, stmt) && ncopies > 1)
4998 {
4999 if (dump_enabled_p ())
5000 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5001 "multiple types in nested loop.\n");
5002 return false;
5003 }
5004
5005 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
5006 return false;
5007
5008 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
5009 return false;
5010
5011 /* Is vectorizable store? */
5012
5013 if (!is_gimple_assign (stmt))
5014 return false;
5015
5016 scalar_dest = gimple_assign_lhs (stmt);
5017 if (TREE_CODE (scalar_dest) == VIEW_CONVERT_EXPR
5018 && is_pattern_stmt_p (stmt_info))
5019 scalar_dest = TREE_OPERAND (scalar_dest, 0);
5020 if (TREE_CODE (scalar_dest) != ARRAY_REF
5021 && TREE_CODE (scalar_dest) != BIT_FIELD_REF
5022 && TREE_CODE (scalar_dest) != INDIRECT_REF
5023 && TREE_CODE (scalar_dest) != COMPONENT_REF
5024 && TREE_CODE (scalar_dest) != IMAGPART_EXPR
5025 && TREE_CODE (scalar_dest) != REALPART_EXPR
5026 && TREE_CODE (scalar_dest) != MEM_REF)
5027 return false;
5028
5029 gcc_assert (gimple_assign_single_p (stmt));
5030 op = gimple_assign_rhs1 (stmt);
5031 if (!vect_is_simple_use (op, stmt, loop_vinfo, bb_vinfo, &def_stmt,
5032 &def, &dt))
5033 {
5034 if (dump_enabled_p ())
5035 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5036 "use not simple.\n");
5037 return false;
5038 }
5039
5040 elem_type = TREE_TYPE (vectype);
5041 vec_mode = TYPE_MODE (vectype);
5042
5043 /* FORNOW. In some cases can vectorize even if data-type not supported
5044 (e.g. - array initialization with 0). */
5045 if (optab_handler (mov_optab, vec_mode) == CODE_FOR_nothing)
5046 return false;
5047
5048 if (!STMT_VINFO_DATA_REF (stmt_info))
5049 return false;
5050
5051 negative =
5052 tree_int_cst_compare (loop && nested_in_vect_loop_p (loop, stmt)
5053 ? STMT_VINFO_DR_STEP (stmt_info) : DR_STEP (dr),
5054 size_zero_node) < 0;
5055 if (negative && ncopies > 1)
5056 {
5057 if (dump_enabled_p ())
5058 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5059 "multiple types with negative step.\n");
5060 return false;
5061 }
5062
5063 if (negative)
5064 {
5065 gcc_assert (!grouped_store);
5066 alignment_support_scheme = vect_supportable_dr_alignment (dr, false);
5067 if (alignment_support_scheme != dr_aligned
5068 && alignment_support_scheme != dr_unaligned_supported)
5069 {
5070 if (dump_enabled_p ())
5071 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5072 "negative step but alignment required.\n");
5073 return false;
5074 }
5075 if (dt != vect_constant_def
5076 && dt != vect_external_def
5077 && !perm_mask_for_reverse (vectype))
5078 {
5079 if (dump_enabled_p ())
5080 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5081 "negative step and reversing not supported.\n");
5082 return false;
5083 }
5084 }
5085
5086 if (STMT_VINFO_GROUPED_ACCESS (stmt_info))
5087 {
5088 grouped_store = true;
5089 first_stmt = GROUP_FIRST_ELEMENT (stmt_info);
5090 if (!slp && !PURE_SLP_STMT (stmt_info))
5091 {
5092 group_size = GROUP_SIZE (vinfo_for_stmt (first_stmt));
5093 if (vect_store_lanes_supported (vectype, group_size))
5094 store_lanes_p = true;
5095 else if (!vect_grouped_store_supported (vectype, group_size))
5096 return false;
5097 }
5098
5099 if (first_stmt == stmt)
5100 {
5101 /* STMT is the leader of the group. Check the operands of all the
5102 stmts of the group. */
5103 next_stmt = GROUP_NEXT_ELEMENT (stmt_info);
5104 while (next_stmt)
5105 {
5106 gcc_assert (gimple_assign_single_p (next_stmt));
5107 op = gimple_assign_rhs1 (next_stmt);
5108 if (!vect_is_simple_use (op, next_stmt, loop_vinfo, bb_vinfo,
5109 &def_stmt, &def, &dt))
5110 {
5111 if (dump_enabled_p ())
5112 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5113 "use not simple.\n");
5114 return false;
5115 }
5116 next_stmt = GROUP_NEXT_ELEMENT (vinfo_for_stmt (next_stmt));
5117 }
5118 }
5119 }
5120
5121 if (!vec_stmt) /* transformation not required. */
5122 {
5123 STMT_VINFO_TYPE (stmt_info) = store_vec_info_type;
5124 vect_model_store_cost (stmt_info, ncopies, store_lanes_p, dt,
5125 NULL, NULL, NULL);
5126 return true;
5127 }
5128
5129 /** Transform. **/
5130
5131 ensure_base_align (stmt_info, dr);
5132
5133 if (grouped_store)
5134 {
5135 first_dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (first_stmt));
5136 group_size = GROUP_SIZE (vinfo_for_stmt (first_stmt));
5137
5138 GROUP_STORE_COUNT (vinfo_for_stmt (first_stmt))++;
5139
5140 /* FORNOW */
5141 gcc_assert (!loop || !nested_in_vect_loop_p (loop, stmt));
5142
5143 /* We vectorize all the stmts of the interleaving group when we
5144 reach the last stmt in the group. */
5145 if (GROUP_STORE_COUNT (vinfo_for_stmt (first_stmt))
5146 < GROUP_SIZE (vinfo_for_stmt (first_stmt))
5147 && !slp)
5148 {
5149 *vec_stmt = NULL;
5150 return true;
5151 }
5152
5153 if (slp)
5154 {
5155 grouped_store = false;
5156 /* VEC_NUM is the number of vect stmts to be created for this
5157 group. */
5158 vec_num = SLP_TREE_NUMBER_OF_VEC_STMTS (slp_node);
5159 first_stmt = SLP_TREE_SCALAR_STMTS (slp_node)[0];
5160 first_dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (first_stmt));
5161 op = gimple_assign_rhs1 (first_stmt);
5162 }
5163 else
5164 /* VEC_NUM is the number of vect stmts to be created for this
5165 group. */
5166 vec_num = group_size;
5167 }
5168 else
5169 {
5170 first_stmt = stmt;
5171 first_dr = dr;
5172 group_size = vec_num = 1;
5173 }
5174
5175 if (dump_enabled_p ())
5176 dump_printf_loc (MSG_NOTE, vect_location,
5177 "transform store. ncopies = %d\n", ncopies);
5178
5179 dr_chain.create (group_size);
5180 oprnds.create (group_size);
5181
5182 alignment_support_scheme = vect_supportable_dr_alignment (first_dr, false);
5183 gcc_assert (alignment_support_scheme);
5184 /* Targets with store-lane instructions must not require explicit
5185 realignment. */
5186 gcc_assert (!store_lanes_p
5187 || alignment_support_scheme == dr_aligned
5188 || alignment_support_scheme == dr_unaligned_supported);
5189
5190 if (negative)
5191 offset = size_int (-TYPE_VECTOR_SUBPARTS (vectype) + 1);
5192
5193 if (store_lanes_p)
5194 aggr_type = build_array_type_nelts (elem_type, vec_num * nunits);
5195 else
5196 aggr_type = vectype;
5197
5198 /* In case the vectorization factor (VF) is bigger than the number
5199 of elements that we can fit in a vectype (nunits), we have to generate
5200 more than one vector stmt - i.e - we need to "unroll" the
5201 vector stmt by a factor VF/nunits. For more details see documentation in
5202 vect_get_vec_def_for_copy_stmt. */
5203
5204 /* In case of interleaving (non-unit grouped access):
5205
5206 S1: &base + 2 = x2
5207 S2: &base = x0
5208 S3: &base + 1 = x1
5209 S4: &base + 3 = x3
5210
5211 We create vectorized stores starting from base address (the access of the
5212 first stmt in the chain (S2 in the above example), when the last store stmt
5213 of the chain (S4) is reached:
5214
5215 VS1: &base = vx2
5216 VS2: &base + vec_size*1 = vx0
5217 VS3: &base + vec_size*2 = vx1
5218 VS4: &base + vec_size*3 = vx3
5219
5220 Then permutation statements are generated:
5221
5222 VS5: vx5 = VEC_PERM_EXPR < vx0, vx3, {0, 8, 1, 9, 2, 10, 3, 11} >
5223 VS6: vx6 = VEC_PERM_EXPR < vx0, vx3, {4, 12, 5, 13, 6, 14, 7, 15} >
5224 ...
5225
5226 And they are put in STMT_VINFO_VEC_STMT of the corresponding scalar stmts
5227 (the order of the data-refs in the output of vect_permute_store_chain
5228 corresponds to the order of scalar stmts in the interleaving chain - see
5229 the documentation of vect_permute_store_chain()).
5230
5231 In case of both multiple types and interleaving, above vector stores and
5232 permutation stmts are created for every copy. The result vector stmts are
5233 put in STMT_VINFO_VEC_STMT for the first copy and in the corresponding
5234 STMT_VINFO_RELATED_STMT for the next copies.
5235 */
5236
5237 prev_stmt_info = NULL;
5238 for (j = 0; j < ncopies; j++)
5239 {
5240 gimple new_stmt;
5241
5242 if (j == 0)
5243 {
5244 if (slp)
5245 {
5246 /* Get vectorized arguments for SLP_NODE. */
5247 vect_get_vec_defs (op, NULL_TREE, stmt, &vec_oprnds,
5248 NULL, slp_node, -1);
5249
5250 vec_oprnd = vec_oprnds[0];
5251 }
5252 else
5253 {
5254 /* For interleaved stores we collect vectorized defs for all the
5255 stores in the group in DR_CHAIN and OPRNDS. DR_CHAIN is then
5256 used as an input to vect_permute_store_chain(), and OPRNDS as
5257 an input to vect_get_vec_def_for_stmt_copy() for the next copy.
5258
5259 If the store is not grouped, GROUP_SIZE is 1, and DR_CHAIN and
5260 OPRNDS are of size 1. */
5261 next_stmt = first_stmt;
5262 for (i = 0; i < group_size; i++)
5263 {
5264 /* Since gaps are not supported for interleaved stores,
5265 GROUP_SIZE is the exact number of stmts in the chain.
5266 Therefore, NEXT_STMT can't be NULL_TREE. In case that
5267 there is no interleaving, GROUP_SIZE is 1, and only one
5268 iteration of the loop will be executed. */
5269 gcc_assert (next_stmt
5270 && gimple_assign_single_p (next_stmt));
5271 op = gimple_assign_rhs1 (next_stmt);
5272
5273 vec_oprnd = vect_get_vec_def_for_operand (op, next_stmt,
5274 NULL);
5275 dr_chain.quick_push (vec_oprnd);
5276 oprnds.quick_push (vec_oprnd);
5277 next_stmt = GROUP_NEXT_ELEMENT (vinfo_for_stmt (next_stmt));
5278 }
5279 }
5280
5281 /* We should have catched mismatched types earlier. */
5282 gcc_assert (useless_type_conversion_p (vectype,
5283 TREE_TYPE (vec_oprnd)));
5284 bool simd_lane_access_p
5285 = STMT_VINFO_SIMD_LANE_ACCESS_P (stmt_info);
5286 if (simd_lane_access_p
5287 && TREE_CODE (DR_BASE_ADDRESS (first_dr)) == ADDR_EXPR
5288 && VAR_P (TREE_OPERAND (DR_BASE_ADDRESS (first_dr), 0))
5289 && integer_zerop (DR_OFFSET (first_dr))
5290 && integer_zerop (DR_INIT (first_dr))
5291 && alias_sets_conflict_p (get_alias_set (aggr_type),
5292 get_alias_set (DR_REF (first_dr))))
5293 {
5294 dataref_ptr = unshare_expr (DR_BASE_ADDRESS (first_dr));
5295 dataref_offset = build_int_cst (reference_alias_ptr_type
5296 (DR_REF (first_dr)), 0);
5297 inv_p = false;
5298 }
5299 else
5300 dataref_ptr
5301 = vect_create_data_ref_ptr (first_stmt, aggr_type,
5302 simd_lane_access_p ? loop : NULL,
5303 offset, &dummy, gsi, &ptr_incr,
5304 simd_lane_access_p, &inv_p);
5305 gcc_assert (bb_vinfo || !inv_p);
5306 }
5307 else
5308 {
5309 /* For interleaved stores we created vectorized defs for all the
5310 defs stored in OPRNDS in the previous iteration (previous copy).
5311 DR_CHAIN is then used as an input to vect_permute_store_chain(),
5312 and OPRNDS as an input to vect_get_vec_def_for_stmt_copy() for the
5313 next copy.
5314 If the store is not grouped, GROUP_SIZE is 1, and DR_CHAIN and
5315 OPRNDS are of size 1. */
5316 for (i = 0; i < group_size; i++)
5317 {
5318 op = oprnds[i];
5319 vect_is_simple_use (op, NULL, loop_vinfo, bb_vinfo, &def_stmt,
5320 &def, &dt);
5321 vec_oprnd = vect_get_vec_def_for_stmt_copy (dt, op);
5322 dr_chain[i] = vec_oprnd;
5323 oprnds[i] = vec_oprnd;
5324 }
5325 if (dataref_offset)
5326 dataref_offset
5327 = int_const_binop (PLUS_EXPR, dataref_offset,
5328 TYPE_SIZE_UNIT (aggr_type));
5329 else
5330 dataref_ptr = bump_vector_ptr (dataref_ptr, ptr_incr, gsi, stmt,
5331 TYPE_SIZE_UNIT (aggr_type));
5332 }
5333
5334 if (store_lanes_p)
5335 {
5336 tree vec_array;
5337
5338 /* Combine all the vectors into an array. */
5339 vec_array = create_vector_array (vectype, vec_num);
5340 for (i = 0; i < vec_num; i++)
5341 {
5342 vec_oprnd = dr_chain[i];
5343 write_vector_array (stmt, gsi, vec_oprnd, vec_array, i);
5344 }
5345
5346 /* Emit:
5347 MEM_REF[...all elements...] = STORE_LANES (VEC_ARRAY). */
5348 data_ref = create_array_ref (aggr_type, dataref_ptr, first_dr);
5349 new_stmt = gimple_build_call_internal (IFN_STORE_LANES, 1, vec_array);
5350 gimple_call_set_lhs (new_stmt, data_ref);
5351 vect_finish_stmt_generation (stmt, new_stmt, gsi);
5352 }
5353 else
5354 {
5355 new_stmt = NULL;
5356 if (grouped_store)
5357 {
5358 if (j == 0)
5359 result_chain.create (group_size);
5360 /* Permute. */
5361 vect_permute_store_chain (dr_chain, group_size, stmt, gsi,
5362 &result_chain);
5363 }
5364
5365 next_stmt = first_stmt;
5366 for (i = 0; i < vec_num; i++)
5367 {
5368 unsigned align, misalign;
5369
5370 if (i > 0)
5371 /* Bump the vector pointer. */
5372 dataref_ptr = bump_vector_ptr (dataref_ptr, ptr_incr, gsi,
5373 stmt, NULL_TREE);
5374
5375 if (slp)
5376 vec_oprnd = vec_oprnds[i];
5377 else if (grouped_store)
5378 /* For grouped stores vectorized defs are interleaved in
5379 vect_permute_store_chain(). */
5380 vec_oprnd = result_chain[i];
5381
5382 data_ref = build2 (MEM_REF, TREE_TYPE (vec_oprnd), dataref_ptr,
5383 dataref_offset
5384 ? dataref_offset
5385 : build_int_cst (reference_alias_ptr_type
5386 (DR_REF (first_dr)), 0));
5387 align = TYPE_ALIGN_UNIT (vectype);
5388 if (aligned_access_p (first_dr))
5389 misalign = 0;
5390 else if (DR_MISALIGNMENT (first_dr) == -1)
5391 {
5392 TREE_TYPE (data_ref)
5393 = build_aligned_type (TREE_TYPE (data_ref),
5394 TYPE_ALIGN (elem_type));
5395 align = TYPE_ALIGN_UNIT (elem_type);
5396 misalign = 0;
5397 }
5398 else
5399 {
5400 TREE_TYPE (data_ref)
5401 = build_aligned_type (TREE_TYPE (data_ref),
5402 TYPE_ALIGN (elem_type));
5403 misalign = DR_MISALIGNMENT (first_dr);
5404 }
5405 if (dataref_offset == NULL_TREE)
5406 set_ptr_info_alignment (get_ptr_info (dataref_ptr), align,
5407 misalign);
5408
5409 if (negative
5410 && dt != vect_constant_def
5411 && dt != vect_external_def)
5412 {
5413 tree perm_mask = perm_mask_for_reverse (vectype);
5414 tree perm_dest
5415 = vect_create_destination_var (gimple_assign_rhs1 (stmt),
5416 vectype);
5417 tree new_temp = make_ssa_name (perm_dest, NULL);
5418
5419 /* Generate the permute statement. */
5420 gimple perm_stmt
5421 = gimple_build_assign_with_ops (VEC_PERM_EXPR, new_temp,
5422 vec_oprnd, vec_oprnd,
5423 perm_mask);
5424 vect_finish_stmt_generation (stmt, perm_stmt, gsi);
5425
5426 perm_stmt = SSA_NAME_DEF_STMT (new_temp);
5427 vec_oprnd = new_temp;
5428 }
5429
5430 /* Arguments are ready. Create the new vector stmt. */
5431 new_stmt = gimple_build_assign (data_ref, vec_oprnd);
5432 vect_finish_stmt_generation (stmt, new_stmt, gsi);
5433
5434 if (slp)
5435 continue;
5436
5437 next_stmt = GROUP_NEXT_ELEMENT (vinfo_for_stmt (next_stmt));
5438 if (!next_stmt)
5439 break;
5440 }
5441 }
5442 if (!slp)
5443 {
5444 if (j == 0)
5445 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
5446 else
5447 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
5448 prev_stmt_info = vinfo_for_stmt (new_stmt);
5449 }
5450 }
5451
5452 dr_chain.release ();
5453 oprnds.release ();
5454 result_chain.release ();
5455 vec_oprnds.release ();
5456
5457 return true;
5458 }
5459
5460 /* Given a vector type VECTYPE and permutation SEL returns
5461 the VECTOR_CST mask that implements the permutation of the
5462 vector elements. If that is impossible to do, returns NULL. */
5463
5464 tree
5465 vect_gen_perm_mask (tree vectype, unsigned char *sel)
5466 {
5467 tree mask_elt_type, mask_type, mask_vec, *mask_elts;
5468 int i, nunits;
5469
5470 nunits = TYPE_VECTOR_SUBPARTS (vectype);
5471
5472 if (!can_vec_perm_p (TYPE_MODE (vectype), false, sel))
5473 return NULL;
5474
5475 mask_elt_type = lang_hooks.types.type_for_mode
5476 (int_mode_for_mode (TYPE_MODE (TREE_TYPE (vectype))), 1);
5477 mask_type = get_vectype_for_scalar_type (mask_elt_type);
5478
5479 mask_elts = XALLOCAVEC (tree, nunits);
5480 for (i = nunits - 1; i >= 0; i--)
5481 mask_elts[i] = build_int_cst (mask_elt_type, sel[i]);
5482 mask_vec = build_vector (mask_type, mask_elts);
5483
5484 return mask_vec;
5485 }
5486
5487 /* Given a vector variable X and Y, that was generated for the scalar
5488 STMT, generate instructions to permute the vector elements of X and Y
5489 using permutation mask MASK_VEC, insert them at *GSI and return the
5490 permuted vector variable. */
5491
5492 static tree
5493 permute_vec_elements (tree x, tree y, tree mask_vec, gimple stmt,
5494 gimple_stmt_iterator *gsi)
5495 {
5496 tree vectype = TREE_TYPE (x);
5497 tree perm_dest, data_ref;
5498 gimple perm_stmt;
5499
5500 perm_dest = vect_create_destination_var (gimple_get_lhs (stmt), vectype);
5501 data_ref = make_ssa_name (perm_dest, NULL);
5502
5503 /* Generate the permute statement. */
5504 perm_stmt = gimple_build_assign_with_ops (VEC_PERM_EXPR, data_ref,
5505 x, y, mask_vec);
5506 vect_finish_stmt_generation (stmt, perm_stmt, gsi);
5507
5508 return data_ref;
5509 }
5510
5511 /* Hoist the definitions of all SSA uses on STMT out of the loop LOOP,
5512 inserting them on the loops preheader edge. Returns true if we
5513 were successful in doing so (and thus STMT can be moved then),
5514 otherwise returns false. */
5515
5516 static bool
5517 hoist_defs_of_uses (gimple stmt, struct loop *loop)
5518 {
5519 ssa_op_iter i;
5520 tree op;
5521 bool any = false;
5522
5523 FOR_EACH_SSA_TREE_OPERAND (op, stmt, i, SSA_OP_USE)
5524 {
5525 gimple def_stmt = SSA_NAME_DEF_STMT (op);
5526 if (!gimple_nop_p (def_stmt)
5527 && flow_bb_inside_loop_p (loop, gimple_bb (def_stmt)))
5528 {
5529 /* Make sure we don't need to recurse. While we could do
5530 so in simple cases when there are more complex use webs
5531 we don't have an easy way to preserve stmt order to fulfil
5532 dependencies within them. */
5533 tree op2;
5534 ssa_op_iter i2;
5535 if (gimple_code (def_stmt) == GIMPLE_PHI)
5536 return false;
5537 FOR_EACH_SSA_TREE_OPERAND (op2, def_stmt, i2, SSA_OP_USE)
5538 {
5539 gimple def_stmt2 = SSA_NAME_DEF_STMT (op2);
5540 if (!gimple_nop_p (def_stmt2)
5541 && flow_bb_inside_loop_p (loop, gimple_bb (def_stmt2)))
5542 return false;
5543 }
5544 any = true;
5545 }
5546 }
5547
5548 if (!any)
5549 return true;
5550
5551 FOR_EACH_SSA_TREE_OPERAND (op, stmt, i, SSA_OP_USE)
5552 {
5553 gimple def_stmt = SSA_NAME_DEF_STMT (op);
5554 if (!gimple_nop_p (def_stmt)
5555 && flow_bb_inside_loop_p (loop, gimple_bb (def_stmt)))
5556 {
5557 gimple_stmt_iterator gsi = gsi_for_stmt (def_stmt);
5558 gsi_remove (&gsi, false);
5559 gsi_insert_on_edge_immediate (loop_preheader_edge (loop), def_stmt);
5560 }
5561 }
5562
5563 return true;
5564 }
5565
5566 /* vectorizable_load.
5567
5568 Check if STMT reads a non scalar data-ref (array/pointer/structure) that
5569 can be vectorized.
5570 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
5571 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
5572 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
5573
5574 static bool
5575 vectorizable_load (gimple stmt, gimple_stmt_iterator *gsi, gimple *vec_stmt,
5576 slp_tree slp_node, slp_instance slp_node_instance)
5577 {
5578 tree scalar_dest;
5579 tree vec_dest = NULL;
5580 tree data_ref = NULL;
5581 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
5582 stmt_vec_info prev_stmt_info;
5583 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
5584 struct loop *loop = NULL;
5585 struct loop *containing_loop = (gimple_bb (stmt))->loop_father;
5586 bool nested_in_vect_loop = false;
5587 struct data_reference *dr = STMT_VINFO_DATA_REF (stmt_info), *first_dr = NULL;
5588 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
5589 tree elem_type;
5590 tree new_temp;
5591 enum machine_mode mode;
5592 gimple new_stmt = NULL;
5593 tree dummy;
5594 enum dr_alignment_support alignment_support_scheme;
5595 tree dataref_ptr = NULL_TREE;
5596 tree dataref_offset = NULL_TREE;
5597 gimple ptr_incr = NULL;
5598 int nunits = TYPE_VECTOR_SUBPARTS (vectype);
5599 int ncopies;
5600 int i, j, group_size, group_gap;
5601 tree msq = NULL_TREE, lsq;
5602 tree offset = NULL_TREE;
5603 tree byte_offset = NULL_TREE;
5604 tree realignment_token = NULL_TREE;
5605 gimple phi = NULL;
5606 vec<tree> dr_chain = vNULL;
5607 bool grouped_load = false;
5608 bool load_lanes_p = false;
5609 gimple first_stmt;
5610 bool inv_p;
5611 bool negative = false;
5612 bool compute_in_loop = false;
5613 struct loop *at_loop;
5614 int vec_num;
5615 bool slp = (slp_node != NULL);
5616 bool slp_perm = false;
5617 enum tree_code code;
5618 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
5619 int vf;
5620 tree aggr_type;
5621 tree gather_base = NULL_TREE, gather_off = NULL_TREE;
5622 tree gather_off_vectype = NULL_TREE, gather_decl = NULL_TREE;
5623 int gather_scale = 1;
5624 enum vect_def_type gather_dt = vect_unknown_def_type;
5625
5626 if (loop_vinfo)
5627 {
5628 loop = LOOP_VINFO_LOOP (loop_vinfo);
5629 nested_in_vect_loop = nested_in_vect_loop_p (loop, stmt);
5630 vf = LOOP_VINFO_VECT_FACTOR (loop_vinfo);
5631 }
5632 else
5633 vf = 1;
5634
5635 /* Multiple types in SLP are handled by creating the appropriate number of
5636 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
5637 case of SLP. */
5638 if (slp || PURE_SLP_STMT (stmt_info))
5639 ncopies = 1;
5640 else
5641 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits;
5642
5643 gcc_assert (ncopies >= 1);
5644
5645 /* FORNOW. This restriction should be relaxed. */
5646 if (nested_in_vect_loop && ncopies > 1)
5647 {
5648 if (dump_enabled_p ())
5649 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5650 "multiple types in nested loop.\n");
5651 return false;
5652 }
5653
5654 /* Invalidate assumptions made by dependence analysis when vectorization
5655 on the unrolled body effectively re-orders stmts. */
5656 if (ncopies > 1
5657 && STMT_VINFO_MIN_NEG_DIST (stmt_info) != 0
5658 && ((unsigned)LOOP_VINFO_VECT_FACTOR (loop_vinfo)
5659 > STMT_VINFO_MIN_NEG_DIST (stmt_info)))
5660 {
5661 if (dump_enabled_p ())
5662 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5663 "cannot perform implicit CSE when unrolling "
5664 "with negative dependence distance\n");
5665 return false;
5666 }
5667
5668 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
5669 return false;
5670
5671 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
5672 return false;
5673
5674 /* Is vectorizable load? */
5675 if (!is_gimple_assign (stmt))
5676 return false;
5677
5678 scalar_dest = gimple_assign_lhs (stmt);
5679 if (TREE_CODE (scalar_dest) != SSA_NAME)
5680 return false;
5681
5682 code = gimple_assign_rhs_code (stmt);
5683 if (code != ARRAY_REF
5684 && code != BIT_FIELD_REF
5685 && code != INDIRECT_REF
5686 && code != COMPONENT_REF
5687 && code != IMAGPART_EXPR
5688 && code != REALPART_EXPR
5689 && code != MEM_REF
5690 && TREE_CODE_CLASS (code) != tcc_declaration)
5691 return false;
5692
5693 if (!STMT_VINFO_DATA_REF (stmt_info))
5694 return false;
5695
5696 elem_type = TREE_TYPE (vectype);
5697 mode = TYPE_MODE (vectype);
5698
5699 /* FORNOW. In some cases can vectorize even if data-type not supported
5700 (e.g. - data copies). */
5701 if (optab_handler (mov_optab, mode) == CODE_FOR_nothing)
5702 {
5703 if (dump_enabled_p ())
5704 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5705 "Aligned load, but unsupported type.\n");
5706 return false;
5707 }
5708
5709 /* Check if the load is a part of an interleaving chain. */
5710 if (STMT_VINFO_GROUPED_ACCESS (stmt_info))
5711 {
5712 grouped_load = true;
5713 /* FORNOW */
5714 gcc_assert (! nested_in_vect_loop && !STMT_VINFO_GATHER_P (stmt_info));
5715
5716 first_stmt = GROUP_FIRST_ELEMENT (stmt_info);
5717 if (!slp && !PURE_SLP_STMT (stmt_info))
5718 {
5719 group_size = GROUP_SIZE (vinfo_for_stmt (first_stmt));
5720 if (vect_load_lanes_supported (vectype, group_size))
5721 load_lanes_p = true;
5722 else if (!vect_grouped_load_supported (vectype, group_size))
5723 return false;
5724 }
5725
5726 /* Invalidate assumptions made by dependence analysis when vectorization
5727 on the unrolled body effectively re-orders stmts. */
5728 if (!PURE_SLP_STMT (stmt_info)
5729 && STMT_VINFO_MIN_NEG_DIST (stmt_info) != 0
5730 && ((unsigned)LOOP_VINFO_VECT_FACTOR (loop_vinfo)
5731 > STMT_VINFO_MIN_NEG_DIST (stmt_info)))
5732 {
5733 if (dump_enabled_p ())
5734 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5735 "cannot perform implicit CSE when performing "
5736 "group loads with negative dependence distance\n");
5737 return false;
5738 }
5739 }
5740
5741
5742 if (STMT_VINFO_GATHER_P (stmt_info))
5743 {
5744 gimple def_stmt;
5745 tree def;
5746 gather_decl = vect_check_gather (stmt, loop_vinfo, &gather_base,
5747 &gather_off, &gather_scale);
5748 gcc_assert (gather_decl);
5749 if (!vect_is_simple_use_1 (gather_off, NULL, loop_vinfo, bb_vinfo,
5750 &def_stmt, &def, &gather_dt,
5751 &gather_off_vectype))
5752 {
5753 if (dump_enabled_p ())
5754 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5755 "gather index use not simple.\n");
5756 return false;
5757 }
5758 }
5759 else if (STMT_VINFO_STRIDE_LOAD_P (stmt_info))
5760 ;
5761 else
5762 {
5763 negative = tree_int_cst_compare (nested_in_vect_loop
5764 ? STMT_VINFO_DR_STEP (stmt_info)
5765 : DR_STEP (dr),
5766 size_zero_node) < 0;
5767 if (negative && ncopies > 1)
5768 {
5769 if (dump_enabled_p ())
5770 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5771 "multiple types with negative step.\n");
5772 return false;
5773 }
5774
5775 if (negative)
5776 {
5777 if (grouped_load)
5778 {
5779 if (dump_enabled_p ())
5780 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5781 "negative step for group load not supported"
5782 "\n");
5783 return false;
5784 }
5785 alignment_support_scheme = vect_supportable_dr_alignment (dr, false);
5786 if (alignment_support_scheme != dr_aligned
5787 && alignment_support_scheme != dr_unaligned_supported)
5788 {
5789 if (dump_enabled_p ())
5790 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5791 "negative step but alignment required.\n");
5792 return false;
5793 }
5794 if (!perm_mask_for_reverse (vectype))
5795 {
5796 if (dump_enabled_p ())
5797 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5798 "negative step and reversing not supported."
5799 "\n");
5800 return false;
5801 }
5802 }
5803 }
5804
5805 if (!vec_stmt) /* transformation not required. */
5806 {
5807 STMT_VINFO_TYPE (stmt_info) = load_vec_info_type;
5808 vect_model_load_cost (stmt_info, ncopies, load_lanes_p, NULL, NULL, NULL);
5809 return true;
5810 }
5811
5812 if (dump_enabled_p ())
5813 dump_printf_loc (MSG_NOTE, vect_location,
5814 "transform load. ncopies = %d\n", ncopies);
5815
5816 /** Transform. **/
5817
5818 ensure_base_align (stmt_info, dr);
5819
5820 if (STMT_VINFO_GATHER_P (stmt_info))
5821 {
5822 tree vec_oprnd0 = NULL_TREE, op;
5823 tree arglist = TYPE_ARG_TYPES (TREE_TYPE (gather_decl));
5824 tree rettype, srctype, ptrtype, idxtype, masktype, scaletype;
5825 tree ptr, mask, var, scale, merge, perm_mask = NULL_TREE, prev_res = NULL_TREE;
5826 edge pe = loop_preheader_edge (loop);
5827 gimple_seq seq;
5828 basic_block new_bb;
5829 enum { NARROW, NONE, WIDEN } modifier;
5830 int gather_off_nunits = TYPE_VECTOR_SUBPARTS (gather_off_vectype);
5831
5832 if (nunits == gather_off_nunits)
5833 modifier = NONE;
5834 else if (nunits == gather_off_nunits / 2)
5835 {
5836 unsigned char *sel = XALLOCAVEC (unsigned char, gather_off_nunits);
5837 modifier = WIDEN;
5838
5839 for (i = 0; i < gather_off_nunits; ++i)
5840 sel[i] = i | nunits;
5841
5842 perm_mask = vect_gen_perm_mask (gather_off_vectype, sel);
5843 gcc_assert (perm_mask != NULL_TREE);
5844 }
5845 else if (nunits == gather_off_nunits * 2)
5846 {
5847 unsigned char *sel = XALLOCAVEC (unsigned char, nunits);
5848 modifier = NARROW;
5849
5850 for (i = 0; i < nunits; ++i)
5851 sel[i] = i < gather_off_nunits
5852 ? i : i + nunits - gather_off_nunits;
5853
5854 perm_mask = vect_gen_perm_mask (vectype, sel);
5855 gcc_assert (perm_mask != NULL_TREE);
5856 ncopies *= 2;
5857 }
5858 else
5859 gcc_unreachable ();
5860
5861 rettype = TREE_TYPE (TREE_TYPE (gather_decl));
5862 srctype = TREE_VALUE (arglist); arglist = TREE_CHAIN (arglist);
5863 ptrtype = TREE_VALUE (arglist); arglist = TREE_CHAIN (arglist);
5864 idxtype = TREE_VALUE (arglist); arglist = TREE_CHAIN (arglist);
5865 masktype = TREE_VALUE (arglist); arglist = TREE_CHAIN (arglist);
5866 scaletype = TREE_VALUE (arglist);
5867 gcc_checking_assert (types_compatible_p (srctype, rettype));
5868
5869 vec_dest = vect_create_destination_var (scalar_dest, vectype);
5870
5871 ptr = fold_convert (ptrtype, gather_base);
5872 if (!is_gimple_min_invariant (ptr))
5873 {
5874 ptr = force_gimple_operand (ptr, &seq, true, NULL_TREE);
5875 new_bb = gsi_insert_seq_on_edge_immediate (pe, seq);
5876 gcc_assert (!new_bb);
5877 }
5878
5879 /* Currently we support only unconditional gather loads,
5880 so mask should be all ones. */
5881 if (TREE_CODE (masktype) == INTEGER_TYPE)
5882 mask = build_int_cst (masktype, -1);
5883 else if (TREE_CODE (TREE_TYPE (masktype)) == INTEGER_TYPE)
5884 {
5885 mask = build_int_cst (TREE_TYPE (masktype), -1);
5886 mask = build_vector_from_val (masktype, mask);
5887 mask = vect_init_vector (stmt, mask, masktype, NULL);
5888 }
5889 else if (SCALAR_FLOAT_TYPE_P (TREE_TYPE (masktype)))
5890 {
5891 REAL_VALUE_TYPE r;
5892 long tmp[6];
5893 for (j = 0; j < 6; ++j)
5894 tmp[j] = -1;
5895 real_from_target (&r, tmp, TYPE_MODE (TREE_TYPE (masktype)));
5896 mask = build_real (TREE_TYPE (masktype), r);
5897 mask = build_vector_from_val (masktype, mask);
5898 mask = vect_init_vector (stmt, mask, masktype, NULL);
5899 }
5900 else
5901 gcc_unreachable ();
5902
5903 scale = build_int_cst (scaletype, gather_scale);
5904
5905 if (TREE_CODE (TREE_TYPE (rettype)) == INTEGER_TYPE)
5906 merge = build_int_cst (TREE_TYPE (rettype), 0);
5907 else if (SCALAR_FLOAT_TYPE_P (TREE_TYPE (rettype)))
5908 {
5909 REAL_VALUE_TYPE r;
5910 long tmp[6];
5911 for (j = 0; j < 6; ++j)
5912 tmp[j] = 0;
5913 real_from_target (&r, tmp, TYPE_MODE (TREE_TYPE (rettype)));
5914 merge = build_real (TREE_TYPE (rettype), r);
5915 }
5916 else
5917 gcc_unreachable ();
5918 merge = build_vector_from_val (rettype, merge);
5919 merge = vect_init_vector (stmt, merge, rettype, NULL);
5920
5921 prev_stmt_info = NULL;
5922 for (j = 0; j < ncopies; ++j)
5923 {
5924 if (modifier == WIDEN && (j & 1))
5925 op = permute_vec_elements (vec_oprnd0, vec_oprnd0,
5926 perm_mask, stmt, gsi);
5927 else if (j == 0)
5928 op = vec_oprnd0
5929 = vect_get_vec_def_for_operand (gather_off, stmt, NULL);
5930 else
5931 op = vec_oprnd0
5932 = vect_get_vec_def_for_stmt_copy (gather_dt, vec_oprnd0);
5933
5934 if (!useless_type_conversion_p (idxtype, TREE_TYPE (op)))
5935 {
5936 gcc_assert (TYPE_VECTOR_SUBPARTS (TREE_TYPE (op))
5937 == TYPE_VECTOR_SUBPARTS (idxtype));
5938 var = vect_get_new_vect_var (idxtype, vect_simple_var, NULL);
5939 var = make_ssa_name (var, NULL);
5940 op = build1 (VIEW_CONVERT_EXPR, idxtype, op);
5941 new_stmt
5942 = gimple_build_assign_with_ops (VIEW_CONVERT_EXPR, var,
5943 op, NULL_TREE);
5944 vect_finish_stmt_generation (stmt, new_stmt, gsi);
5945 op = var;
5946 }
5947
5948 new_stmt
5949 = gimple_build_call (gather_decl, 5, merge, ptr, op, mask, scale);
5950
5951 if (!useless_type_conversion_p (vectype, rettype))
5952 {
5953 gcc_assert (TYPE_VECTOR_SUBPARTS (vectype)
5954 == TYPE_VECTOR_SUBPARTS (rettype));
5955 var = vect_get_new_vect_var (rettype, vect_simple_var, NULL);
5956 op = make_ssa_name (var, new_stmt);
5957 gimple_call_set_lhs (new_stmt, op);
5958 vect_finish_stmt_generation (stmt, new_stmt, gsi);
5959 var = make_ssa_name (vec_dest, NULL);
5960 op = build1 (VIEW_CONVERT_EXPR, vectype, op);
5961 new_stmt
5962 = gimple_build_assign_with_ops (VIEW_CONVERT_EXPR, var, op,
5963 NULL_TREE);
5964 }
5965 else
5966 {
5967 var = make_ssa_name (vec_dest, new_stmt);
5968 gimple_call_set_lhs (new_stmt, var);
5969 }
5970
5971 vect_finish_stmt_generation (stmt, new_stmt, gsi);
5972
5973 if (modifier == NARROW)
5974 {
5975 if ((j & 1) == 0)
5976 {
5977 prev_res = var;
5978 continue;
5979 }
5980 var = permute_vec_elements (prev_res, var,
5981 perm_mask, stmt, gsi);
5982 new_stmt = SSA_NAME_DEF_STMT (var);
5983 }
5984
5985 if (prev_stmt_info == NULL)
5986 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
5987 else
5988 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
5989 prev_stmt_info = vinfo_for_stmt (new_stmt);
5990 }
5991 return true;
5992 }
5993 else if (STMT_VINFO_STRIDE_LOAD_P (stmt_info))
5994 {
5995 gimple_stmt_iterator incr_gsi;
5996 bool insert_after;
5997 gimple incr;
5998 tree offvar;
5999 tree ivstep;
6000 tree running_off;
6001 vec<constructor_elt, va_gc> *v = NULL;
6002 gimple_seq stmts = NULL;
6003 tree stride_base, stride_step, alias_off;
6004
6005 gcc_assert (!nested_in_vect_loop);
6006
6007 stride_base
6008 = fold_build_pointer_plus
6009 (unshare_expr (DR_BASE_ADDRESS (dr)),
6010 size_binop (PLUS_EXPR,
6011 convert_to_ptrofftype (unshare_expr (DR_OFFSET (dr))),
6012 convert_to_ptrofftype (DR_INIT (dr))));
6013 stride_step = fold_convert (sizetype, unshare_expr (DR_STEP (dr)));
6014
6015 /* For a load with loop-invariant (but other than power-of-2)
6016 stride (i.e. not a grouped access) like so:
6017
6018 for (i = 0; i < n; i += stride)
6019 ... = array[i];
6020
6021 we generate a new induction variable and new accesses to
6022 form a new vector (or vectors, depending on ncopies):
6023
6024 for (j = 0; ; j += VF*stride)
6025 tmp1 = array[j];
6026 tmp2 = array[j + stride];
6027 ...
6028 vectemp = {tmp1, tmp2, ...}
6029 */
6030
6031 ivstep = stride_step;
6032 ivstep = fold_build2 (MULT_EXPR, TREE_TYPE (ivstep), ivstep,
6033 build_int_cst (TREE_TYPE (ivstep), vf));
6034
6035 standard_iv_increment_position (loop, &incr_gsi, &insert_after);
6036
6037 create_iv (stride_base, ivstep, NULL,
6038 loop, &incr_gsi, insert_after,
6039 &offvar, NULL);
6040 incr = gsi_stmt (incr_gsi);
6041 set_vinfo_for_stmt (incr, new_stmt_vec_info (incr, loop_vinfo, NULL));
6042
6043 stride_step = force_gimple_operand (stride_step, &stmts, true, NULL_TREE);
6044 if (stmts)
6045 gsi_insert_seq_on_edge_immediate (loop_preheader_edge (loop), stmts);
6046
6047 prev_stmt_info = NULL;
6048 running_off = offvar;
6049 alias_off = build_int_cst (reference_alias_ptr_type (DR_REF (dr)), 0);
6050 for (j = 0; j < ncopies; j++)
6051 {
6052 tree vec_inv;
6053
6054 vec_alloc (v, nunits);
6055 for (i = 0; i < nunits; i++)
6056 {
6057 tree newref, newoff;
6058 gimple incr;
6059 newref = build2 (MEM_REF, TREE_TYPE (vectype),
6060 running_off, alias_off);
6061
6062 newref = force_gimple_operand_gsi (gsi, newref, true,
6063 NULL_TREE, true,
6064 GSI_SAME_STMT);
6065 CONSTRUCTOR_APPEND_ELT (v, NULL_TREE, newref);
6066 newoff = copy_ssa_name (running_off, NULL);
6067 incr = gimple_build_assign_with_ops (POINTER_PLUS_EXPR, newoff,
6068 running_off, stride_step);
6069 vect_finish_stmt_generation (stmt, incr, gsi);
6070
6071 running_off = newoff;
6072 }
6073
6074 vec_inv = build_constructor (vectype, v);
6075 new_temp = vect_init_vector (stmt, vec_inv, vectype, gsi);
6076 new_stmt = SSA_NAME_DEF_STMT (new_temp);
6077
6078 if (j == 0)
6079 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
6080 else
6081 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
6082 prev_stmt_info = vinfo_for_stmt (new_stmt);
6083 }
6084 return true;
6085 }
6086
6087 if (grouped_load)
6088 {
6089 first_stmt = GROUP_FIRST_ELEMENT (stmt_info);
6090 if (slp
6091 && !SLP_TREE_LOAD_PERMUTATION (slp_node).exists ()
6092 && first_stmt != SLP_TREE_SCALAR_STMTS (slp_node)[0])
6093 first_stmt = SLP_TREE_SCALAR_STMTS (slp_node)[0];
6094
6095 /* Check if the chain of loads is already vectorized. */
6096 if (STMT_VINFO_VEC_STMT (vinfo_for_stmt (first_stmt))
6097 /* For SLP we would need to copy over SLP_TREE_VEC_STMTS.
6098 ??? But we can only do so if there is exactly one
6099 as we have no way to get at the rest. Leave the CSE
6100 opportunity alone.
6101 ??? With the group load eventually participating
6102 in multiple different permutations (having multiple
6103 slp nodes which refer to the same group) the CSE
6104 is even wrong code. See PR56270. */
6105 && !slp)
6106 {
6107 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
6108 return true;
6109 }
6110 first_dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (first_stmt));
6111 group_size = GROUP_SIZE (vinfo_for_stmt (first_stmt));
6112
6113 /* VEC_NUM is the number of vect stmts to be created for this group. */
6114 if (slp)
6115 {
6116 grouped_load = false;
6117 vec_num = SLP_TREE_NUMBER_OF_VEC_STMTS (slp_node);
6118 if (SLP_TREE_LOAD_PERMUTATION (slp_node).exists ())
6119 slp_perm = true;
6120 group_gap = GROUP_GAP (vinfo_for_stmt (first_stmt));
6121 }
6122 else
6123 {
6124 vec_num = group_size;
6125 group_gap = 0;
6126 }
6127 }
6128 else
6129 {
6130 first_stmt = stmt;
6131 first_dr = dr;
6132 group_size = vec_num = 1;
6133 group_gap = 0;
6134 }
6135
6136 alignment_support_scheme = vect_supportable_dr_alignment (first_dr, false);
6137 gcc_assert (alignment_support_scheme);
6138 /* Targets with load-lane instructions must not require explicit
6139 realignment. */
6140 gcc_assert (!load_lanes_p
6141 || alignment_support_scheme == dr_aligned
6142 || alignment_support_scheme == dr_unaligned_supported);
6143
6144 /* In case the vectorization factor (VF) is bigger than the number
6145 of elements that we can fit in a vectype (nunits), we have to generate
6146 more than one vector stmt - i.e - we need to "unroll" the
6147 vector stmt by a factor VF/nunits. In doing so, we record a pointer
6148 from one copy of the vector stmt to the next, in the field
6149 STMT_VINFO_RELATED_STMT. This is necessary in order to allow following
6150 stages to find the correct vector defs to be used when vectorizing
6151 stmts that use the defs of the current stmt. The example below
6152 illustrates the vectorization process when VF=16 and nunits=4 (i.e., we
6153 need to create 4 vectorized stmts):
6154
6155 before vectorization:
6156 RELATED_STMT VEC_STMT
6157 S1: x = memref - -
6158 S2: z = x + 1 - -
6159
6160 step 1: vectorize stmt S1:
6161 We first create the vector stmt VS1_0, and, as usual, record a
6162 pointer to it in the STMT_VINFO_VEC_STMT of the scalar stmt S1.
6163 Next, we create the vector stmt VS1_1, and record a pointer to
6164 it in the STMT_VINFO_RELATED_STMT of the vector stmt VS1_0.
6165 Similarly, for VS1_2 and VS1_3. This is the resulting chain of
6166 stmts and pointers:
6167 RELATED_STMT VEC_STMT
6168 VS1_0: vx0 = memref0 VS1_1 -
6169 VS1_1: vx1 = memref1 VS1_2 -
6170 VS1_2: vx2 = memref2 VS1_3 -
6171 VS1_3: vx3 = memref3 - -
6172 S1: x = load - VS1_0
6173 S2: z = x + 1 - -
6174
6175 See in documentation in vect_get_vec_def_for_stmt_copy for how the
6176 information we recorded in RELATED_STMT field is used to vectorize
6177 stmt S2. */
6178
6179 /* In case of interleaving (non-unit grouped access):
6180
6181 S1: x2 = &base + 2
6182 S2: x0 = &base
6183 S3: x1 = &base + 1
6184 S4: x3 = &base + 3
6185
6186 Vectorized loads are created in the order of memory accesses
6187 starting from the access of the first stmt of the chain:
6188
6189 VS1: vx0 = &base
6190 VS2: vx1 = &base + vec_size*1
6191 VS3: vx3 = &base + vec_size*2
6192 VS4: vx4 = &base + vec_size*3
6193
6194 Then permutation statements are generated:
6195
6196 VS5: vx5 = VEC_PERM_EXPR < vx0, vx1, { 0, 2, ..., i*2 } >
6197 VS6: vx6 = VEC_PERM_EXPR < vx0, vx1, { 1, 3, ..., i*2+1 } >
6198 ...
6199
6200 And they are put in STMT_VINFO_VEC_STMT of the corresponding scalar stmts
6201 (the order of the data-refs in the output of vect_permute_load_chain
6202 corresponds to the order of scalar stmts in the interleaving chain - see
6203 the documentation of vect_permute_load_chain()).
6204 The generation of permutation stmts and recording them in
6205 STMT_VINFO_VEC_STMT is done in vect_transform_grouped_load().
6206
6207 In case of both multiple types and interleaving, the vector loads and
6208 permutation stmts above are created for every copy. The result vector
6209 stmts are put in STMT_VINFO_VEC_STMT for the first copy and in the
6210 corresponding STMT_VINFO_RELATED_STMT for the next copies. */
6211
6212 /* If the data reference is aligned (dr_aligned) or potentially unaligned
6213 on a target that supports unaligned accesses (dr_unaligned_supported)
6214 we generate the following code:
6215 p = initial_addr;
6216 indx = 0;
6217 loop {
6218 p = p + indx * vectype_size;
6219 vec_dest = *(p);
6220 indx = indx + 1;
6221 }
6222
6223 Otherwise, the data reference is potentially unaligned on a target that
6224 does not support unaligned accesses (dr_explicit_realign_optimized) -
6225 then generate the following code, in which the data in each iteration is
6226 obtained by two vector loads, one from the previous iteration, and one
6227 from the current iteration:
6228 p1 = initial_addr;
6229 msq_init = *(floor(p1))
6230 p2 = initial_addr + VS - 1;
6231 realignment_token = call target_builtin;
6232 indx = 0;
6233 loop {
6234 p2 = p2 + indx * vectype_size
6235 lsq = *(floor(p2))
6236 vec_dest = realign_load (msq, lsq, realignment_token)
6237 indx = indx + 1;
6238 msq = lsq;
6239 } */
6240
6241 /* If the misalignment remains the same throughout the execution of the
6242 loop, we can create the init_addr and permutation mask at the loop
6243 preheader. Otherwise, it needs to be created inside the loop.
6244 This can only occur when vectorizing memory accesses in the inner-loop
6245 nested within an outer-loop that is being vectorized. */
6246
6247 if (nested_in_vect_loop
6248 && (TREE_INT_CST_LOW (DR_STEP (dr))
6249 % GET_MODE_SIZE (TYPE_MODE (vectype)) != 0))
6250 {
6251 gcc_assert (alignment_support_scheme != dr_explicit_realign_optimized);
6252 compute_in_loop = true;
6253 }
6254
6255 if ((alignment_support_scheme == dr_explicit_realign_optimized
6256 || alignment_support_scheme == dr_explicit_realign)
6257 && !compute_in_loop)
6258 {
6259 msq = vect_setup_realignment (first_stmt, gsi, &realignment_token,
6260 alignment_support_scheme, NULL_TREE,
6261 &at_loop);
6262 if (alignment_support_scheme == dr_explicit_realign_optimized)
6263 {
6264 phi = SSA_NAME_DEF_STMT (msq);
6265 byte_offset = size_binop (MINUS_EXPR, TYPE_SIZE_UNIT (vectype),
6266 size_one_node);
6267 }
6268 }
6269 else
6270 at_loop = loop;
6271
6272 if (negative)
6273 offset = size_int (-TYPE_VECTOR_SUBPARTS (vectype) + 1);
6274
6275 if (load_lanes_p)
6276 aggr_type = build_array_type_nelts (elem_type, vec_num * nunits);
6277 else
6278 aggr_type = vectype;
6279
6280 prev_stmt_info = NULL;
6281 for (j = 0; j < ncopies; j++)
6282 {
6283 /* 1. Create the vector or array pointer update chain. */
6284 if (j == 0)
6285 {
6286 bool simd_lane_access_p
6287 = STMT_VINFO_SIMD_LANE_ACCESS_P (stmt_info);
6288 if (simd_lane_access_p
6289 && TREE_CODE (DR_BASE_ADDRESS (first_dr)) == ADDR_EXPR
6290 && VAR_P (TREE_OPERAND (DR_BASE_ADDRESS (first_dr), 0))
6291 && integer_zerop (DR_OFFSET (first_dr))
6292 && integer_zerop (DR_INIT (first_dr))
6293 && alias_sets_conflict_p (get_alias_set (aggr_type),
6294 get_alias_set (DR_REF (first_dr)))
6295 && (alignment_support_scheme == dr_aligned
6296 || alignment_support_scheme == dr_unaligned_supported))
6297 {
6298 dataref_ptr = unshare_expr (DR_BASE_ADDRESS (first_dr));
6299 dataref_offset = build_int_cst (reference_alias_ptr_type
6300 (DR_REF (first_dr)), 0);
6301 inv_p = false;
6302 }
6303 else
6304 dataref_ptr
6305 = vect_create_data_ref_ptr (first_stmt, aggr_type, at_loop,
6306 offset, &dummy, gsi, &ptr_incr,
6307 simd_lane_access_p, &inv_p,
6308 byte_offset);
6309 }
6310 else if (dataref_offset)
6311 dataref_offset = int_const_binop (PLUS_EXPR, dataref_offset,
6312 TYPE_SIZE_UNIT (aggr_type));
6313 else
6314 dataref_ptr = bump_vector_ptr (dataref_ptr, ptr_incr, gsi, stmt,
6315 TYPE_SIZE_UNIT (aggr_type));
6316
6317 if (grouped_load || slp_perm)
6318 dr_chain.create (vec_num);
6319
6320 if (load_lanes_p)
6321 {
6322 tree vec_array;
6323
6324 vec_array = create_vector_array (vectype, vec_num);
6325
6326 /* Emit:
6327 VEC_ARRAY = LOAD_LANES (MEM_REF[...all elements...]). */
6328 data_ref = create_array_ref (aggr_type, dataref_ptr, first_dr);
6329 new_stmt = gimple_build_call_internal (IFN_LOAD_LANES, 1, data_ref);
6330 gimple_call_set_lhs (new_stmt, vec_array);
6331 vect_finish_stmt_generation (stmt, new_stmt, gsi);
6332
6333 /* Extract each vector into an SSA_NAME. */
6334 for (i = 0; i < vec_num; i++)
6335 {
6336 new_temp = read_vector_array (stmt, gsi, scalar_dest,
6337 vec_array, i);
6338 dr_chain.quick_push (new_temp);
6339 }
6340
6341 /* Record the mapping between SSA_NAMEs and statements. */
6342 vect_record_grouped_load_vectors (stmt, dr_chain);
6343 }
6344 else
6345 {
6346 for (i = 0; i < vec_num; i++)
6347 {
6348 if (i > 0)
6349 dataref_ptr = bump_vector_ptr (dataref_ptr, ptr_incr, gsi,
6350 stmt, NULL_TREE);
6351
6352 /* 2. Create the vector-load in the loop. */
6353 switch (alignment_support_scheme)
6354 {
6355 case dr_aligned:
6356 case dr_unaligned_supported:
6357 {
6358 unsigned int align, misalign;
6359
6360 data_ref
6361 = build2 (MEM_REF, vectype, dataref_ptr,
6362 dataref_offset
6363 ? dataref_offset
6364 : build_int_cst (reference_alias_ptr_type
6365 (DR_REF (first_dr)), 0));
6366 align = TYPE_ALIGN_UNIT (vectype);
6367 if (alignment_support_scheme == dr_aligned)
6368 {
6369 gcc_assert (aligned_access_p (first_dr));
6370 misalign = 0;
6371 }
6372 else if (DR_MISALIGNMENT (first_dr) == -1)
6373 {
6374 TREE_TYPE (data_ref)
6375 = build_aligned_type (TREE_TYPE (data_ref),
6376 TYPE_ALIGN (elem_type));
6377 align = TYPE_ALIGN_UNIT (elem_type);
6378 misalign = 0;
6379 }
6380 else
6381 {
6382 TREE_TYPE (data_ref)
6383 = build_aligned_type (TREE_TYPE (data_ref),
6384 TYPE_ALIGN (elem_type));
6385 misalign = DR_MISALIGNMENT (first_dr);
6386 }
6387 if (dataref_offset == NULL_TREE)
6388 set_ptr_info_alignment (get_ptr_info (dataref_ptr),
6389 align, misalign);
6390 break;
6391 }
6392 case dr_explicit_realign:
6393 {
6394 tree ptr, bump;
6395 tree vs_minus_1;
6396
6397 vs_minus_1 = size_int (TYPE_VECTOR_SUBPARTS (vectype) - 1);
6398
6399 if (compute_in_loop)
6400 msq = vect_setup_realignment (first_stmt, gsi,
6401 &realignment_token,
6402 dr_explicit_realign,
6403 dataref_ptr, NULL);
6404
6405 ptr = copy_ssa_name (dataref_ptr, NULL);
6406 new_stmt = gimple_build_assign_with_ops
6407 (BIT_AND_EXPR, ptr, dataref_ptr,
6408 build_int_cst
6409 (TREE_TYPE (dataref_ptr),
6410 -(HOST_WIDE_INT)TYPE_ALIGN_UNIT (vectype)));
6411 vect_finish_stmt_generation (stmt, new_stmt, gsi);
6412 data_ref
6413 = build2 (MEM_REF, vectype, ptr,
6414 build_int_cst (reference_alias_ptr_type
6415 (DR_REF (first_dr)), 0));
6416 vec_dest = vect_create_destination_var (scalar_dest,
6417 vectype);
6418 new_stmt = gimple_build_assign (vec_dest, data_ref);
6419 new_temp = make_ssa_name (vec_dest, new_stmt);
6420 gimple_assign_set_lhs (new_stmt, new_temp);
6421 gimple_set_vdef (new_stmt, gimple_vdef (stmt));
6422 gimple_set_vuse (new_stmt, gimple_vuse (stmt));
6423 vect_finish_stmt_generation (stmt, new_stmt, gsi);
6424 msq = new_temp;
6425
6426 bump = size_binop (MULT_EXPR, vs_minus_1,
6427 TYPE_SIZE_UNIT (elem_type));
6428 ptr = bump_vector_ptr (dataref_ptr, NULL, gsi, stmt, bump);
6429 new_stmt = gimple_build_assign_with_ops
6430 (BIT_AND_EXPR, NULL_TREE, ptr,
6431 build_int_cst
6432 (TREE_TYPE (ptr),
6433 -(HOST_WIDE_INT)TYPE_ALIGN_UNIT (vectype)));
6434 ptr = copy_ssa_name (dataref_ptr, new_stmt);
6435 gimple_assign_set_lhs (new_stmt, ptr);
6436 vect_finish_stmt_generation (stmt, new_stmt, gsi);
6437 data_ref
6438 = build2 (MEM_REF, vectype, ptr,
6439 build_int_cst (reference_alias_ptr_type
6440 (DR_REF (first_dr)), 0));
6441 break;
6442 }
6443 case dr_explicit_realign_optimized:
6444 new_temp = copy_ssa_name (dataref_ptr, NULL);
6445 new_stmt = gimple_build_assign_with_ops
6446 (BIT_AND_EXPR, new_temp, dataref_ptr,
6447 build_int_cst
6448 (TREE_TYPE (dataref_ptr),
6449 -(HOST_WIDE_INT)TYPE_ALIGN_UNIT (vectype)));
6450 vect_finish_stmt_generation (stmt, new_stmt, gsi);
6451 data_ref
6452 = build2 (MEM_REF, vectype, new_temp,
6453 build_int_cst (reference_alias_ptr_type
6454 (DR_REF (first_dr)), 0));
6455 break;
6456 default:
6457 gcc_unreachable ();
6458 }
6459 vec_dest = vect_create_destination_var (scalar_dest, vectype);
6460 new_stmt = gimple_build_assign (vec_dest, data_ref);
6461 new_temp = make_ssa_name (vec_dest, new_stmt);
6462 gimple_assign_set_lhs (new_stmt, new_temp);
6463 vect_finish_stmt_generation (stmt, new_stmt, gsi);
6464
6465 /* 3. Handle explicit realignment if necessary/supported.
6466 Create in loop:
6467 vec_dest = realign_load (msq, lsq, realignment_token) */
6468 if (alignment_support_scheme == dr_explicit_realign_optimized
6469 || alignment_support_scheme == dr_explicit_realign)
6470 {
6471 lsq = gimple_assign_lhs (new_stmt);
6472 if (!realignment_token)
6473 realignment_token = dataref_ptr;
6474 vec_dest = vect_create_destination_var (scalar_dest, vectype);
6475 new_stmt
6476 = gimple_build_assign_with_ops (REALIGN_LOAD_EXPR,
6477 vec_dest, msq, lsq,
6478 realignment_token);
6479 new_temp = make_ssa_name (vec_dest, new_stmt);
6480 gimple_assign_set_lhs (new_stmt, new_temp);
6481 vect_finish_stmt_generation (stmt, new_stmt, gsi);
6482
6483 if (alignment_support_scheme == dr_explicit_realign_optimized)
6484 {
6485 gcc_assert (phi);
6486 if (i == vec_num - 1 && j == ncopies - 1)
6487 add_phi_arg (phi, lsq,
6488 loop_latch_edge (containing_loop),
6489 UNKNOWN_LOCATION);
6490 msq = lsq;
6491 }
6492 }
6493
6494 /* 4. Handle invariant-load. */
6495 if (inv_p && !bb_vinfo)
6496 {
6497 gcc_assert (!grouped_load);
6498 /* If we have versioned for aliasing or the loop doesn't
6499 have any data dependencies that would preclude this,
6500 then we are sure this is a loop invariant load and
6501 thus we can insert it on the preheader edge. */
6502 if (LOOP_VINFO_NO_DATA_DEPENDENCIES (loop_vinfo)
6503 && !nested_in_vect_loop
6504 && hoist_defs_of_uses (stmt, loop))
6505 {
6506 if (dump_enabled_p ())
6507 {
6508 dump_printf_loc (MSG_NOTE, vect_location,
6509 "hoisting out of the vectorized "
6510 "loop: ");
6511 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
6512 dump_printf (MSG_NOTE, "\n");
6513 }
6514 tree tem = copy_ssa_name (scalar_dest, NULL);
6515 gsi_insert_on_edge_immediate
6516 (loop_preheader_edge (loop),
6517 gimple_build_assign (tem,
6518 unshare_expr
6519 (gimple_assign_rhs1 (stmt))));
6520 new_temp = vect_init_vector (stmt, tem, vectype, NULL);
6521 }
6522 else
6523 {
6524 gimple_stmt_iterator gsi2 = *gsi;
6525 gsi_next (&gsi2);
6526 new_temp = vect_init_vector (stmt, scalar_dest,
6527 vectype, &gsi2);
6528 }
6529 new_stmt = SSA_NAME_DEF_STMT (new_temp);
6530 set_vinfo_for_stmt (new_stmt,
6531 new_stmt_vec_info (new_stmt, loop_vinfo,
6532 bb_vinfo));
6533 }
6534
6535 if (negative)
6536 {
6537 tree perm_mask = perm_mask_for_reverse (vectype);
6538 new_temp = permute_vec_elements (new_temp, new_temp,
6539 perm_mask, stmt, gsi);
6540 new_stmt = SSA_NAME_DEF_STMT (new_temp);
6541 }
6542
6543 /* Collect vector loads and later create their permutation in
6544 vect_transform_grouped_load (). */
6545 if (grouped_load || slp_perm)
6546 dr_chain.quick_push (new_temp);
6547
6548 /* Store vector loads in the corresponding SLP_NODE. */
6549 if (slp && !slp_perm)
6550 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
6551 }
6552 /* Bump the vector pointer to account for a gap. */
6553 if (slp && group_gap != 0)
6554 {
6555 tree bump = size_binop (MULT_EXPR,
6556 TYPE_SIZE_UNIT (elem_type),
6557 size_int (group_gap));
6558 dataref_ptr = bump_vector_ptr (dataref_ptr, ptr_incr, gsi,
6559 stmt, bump);
6560 }
6561 }
6562
6563 if (slp && !slp_perm)
6564 continue;
6565
6566 if (slp_perm)
6567 {
6568 if (!vect_transform_slp_perm_load (slp_node, dr_chain, gsi, vf,
6569 slp_node_instance, false))
6570 {
6571 dr_chain.release ();
6572 return false;
6573 }
6574 }
6575 else
6576 {
6577 if (grouped_load)
6578 {
6579 if (!load_lanes_p)
6580 vect_transform_grouped_load (stmt, dr_chain, group_size, gsi);
6581 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
6582 }
6583 else
6584 {
6585 if (j == 0)
6586 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
6587 else
6588 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
6589 prev_stmt_info = vinfo_for_stmt (new_stmt);
6590 }
6591 }
6592 dr_chain.release ();
6593 }
6594
6595 return true;
6596 }
6597
6598 /* Function vect_is_simple_cond.
6599
6600 Input:
6601 LOOP - the loop that is being vectorized.
6602 COND - Condition that is checked for simple use.
6603
6604 Output:
6605 *COMP_VECTYPE - the vector type for the comparison.
6606
6607 Returns whether a COND can be vectorized. Checks whether
6608 condition operands are supportable using vec_is_simple_use. */
6609
6610 static bool
6611 vect_is_simple_cond (tree cond, gimple stmt, loop_vec_info loop_vinfo,
6612 bb_vec_info bb_vinfo, tree *comp_vectype)
6613 {
6614 tree lhs, rhs;
6615 tree def;
6616 enum vect_def_type dt;
6617 tree vectype1 = NULL_TREE, vectype2 = NULL_TREE;
6618
6619 if (!COMPARISON_CLASS_P (cond))
6620 return false;
6621
6622 lhs = TREE_OPERAND (cond, 0);
6623 rhs = TREE_OPERAND (cond, 1);
6624
6625 if (TREE_CODE (lhs) == SSA_NAME)
6626 {
6627 gimple lhs_def_stmt = SSA_NAME_DEF_STMT (lhs);
6628 if (!vect_is_simple_use_1 (lhs, stmt, loop_vinfo, bb_vinfo,
6629 &lhs_def_stmt, &def, &dt, &vectype1))
6630 return false;
6631 }
6632 else if (TREE_CODE (lhs) != INTEGER_CST && TREE_CODE (lhs) != REAL_CST
6633 && TREE_CODE (lhs) != FIXED_CST)
6634 return false;
6635
6636 if (TREE_CODE (rhs) == SSA_NAME)
6637 {
6638 gimple rhs_def_stmt = SSA_NAME_DEF_STMT (rhs);
6639 if (!vect_is_simple_use_1 (rhs, stmt, loop_vinfo, bb_vinfo,
6640 &rhs_def_stmt, &def, &dt, &vectype2))
6641 return false;
6642 }
6643 else if (TREE_CODE (rhs) != INTEGER_CST && TREE_CODE (rhs) != REAL_CST
6644 && TREE_CODE (rhs) != FIXED_CST)
6645 return false;
6646
6647 *comp_vectype = vectype1 ? vectype1 : vectype2;
6648 return true;
6649 }
6650
6651 /* vectorizable_condition.
6652
6653 Check if STMT is conditional modify expression that can be vectorized.
6654 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
6655 stmt using VEC_COND_EXPR to replace it, put it in VEC_STMT, and insert it
6656 at GSI.
6657
6658 When STMT is vectorized as nested cycle, REDUC_DEF is the vector variable
6659 to be used at REDUC_INDEX (in then clause if REDUC_INDEX is 1, and in
6660 else caluse if it is 2).
6661
6662 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
6663
6664 bool
6665 vectorizable_condition (gimple stmt, gimple_stmt_iterator *gsi,
6666 gimple *vec_stmt, tree reduc_def, int reduc_index,
6667 slp_tree slp_node)
6668 {
6669 tree scalar_dest = NULL_TREE;
6670 tree vec_dest = NULL_TREE;
6671 tree cond_expr, then_clause, else_clause;
6672 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
6673 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
6674 tree comp_vectype = NULL_TREE;
6675 tree vec_cond_lhs = NULL_TREE, vec_cond_rhs = NULL_TREE;
6676 tree vec_then_clause = NULL_TREE, vec_else_clause = NULL_TREE;
6677 tree vec_compare, vec_cond_expr;
6678 tree new_temp;
6679 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
6680 tree def;
6681 enum vect_def_type dt, dts[4];
6682 int nunits = TYPE_VECTOR_SUBPARTS (vectype);
6683 int ncopies;
6684 enum tree_code code;
6685 stmt_vec_info prev_stmt_info = NULL;
6686 int i, j;
6687 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
6688 vec<tree> vec_oprnds0 = vNULL;
6689 vec<tree> vec_oprnds1 = vNULL;
6690 vec<tree> vec_oprnds2 = vNULL;
6691 vec<tree> vec_oprnds3 = vNULL;
6692 tree vec_cmp_type;
6693
6694 if (slp_node || PURE_SLP_STMT (stmt_info))
6695 ncopies = 1;
6696 else
6697 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits;
6698
6699 gcc_assert (ncopies >= 1);
6700 if (reduc_index && ncopies > 1)
6701 return false; /* FORNOW */
6702
6703 if (reduc_index && STMT_SLP_TYPE (stmt_info))
6704 return false;
6705
6706 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
6707 return false;
6708
6709 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def
6710 && !(STMT_VINFO_DEF_TYPE (stmt_info) == vect_nested_cycle
6711 && reduc_def))
6712 return false;
6713
6714 /* FORNOW: not yet supported. */
6715 if (STMT_VINFO_LIVE_P (stmt_info))
6716 {
6717 if (dump_enabled_p ())
6718 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
6719 "value used after loop.\n");
6720 return false;
6721 }
6722
6723 /* Is vectorizable conditional operation? */
6724 if (!is_gimple_assign (stmt))
6725 return false;
6726
6727 code = gimple_assign_rhs_code (stmt);
6728
6729 if (code != COND_EXPR)
6730 return false;
6731
6732 cond_expr = gimple_assign_rhs1 (stmt);
6733 then_clause = gimple_assign_rhs2 (stmt);
6734 else_clause = gimple_assign_rhs3 (stmt);
6735
6736 if (!vect_is_simple_cond (cond_expr, stmt, loop_vinfo, bb_vinfo,
6737 &comp_vectype)
6738 || !comp_vectype)
6739 return false;
6740
6741 if (TREE_CODE (then_clause) == SSA_NAME)
6742 {
6743 gimple then_def_stmt = SSA_NAME_DEF_STMT (then_clause);
6744 if (!vect_is_simple_use (then_clause, stmt, loop_vinfo, bb_vinfo,
6745 &then_def_stmt, &def, &dt))
6746 return false;
6747 }
6748 else if (TREE_CODE (then_clause) != INTEGER_CST
6749 && TREE_CODE (then_clause) != REAL_CST
6750 && TREE_CODE (then_clause) != FIXED_CST)
6751 return false;
6752
6753 if (TREE_CODE (else_clause) == SSA_NAME)
6754 {
6755 gimple else_def_stmt = SSA_NAME_DEF_STMT (else_clause);
6756 if (!vect_is_simple_use (else_clause, stmt, loop_vinfo, bb_vinfo,
6757 &else_def_stmt, &def, &dt))
6758 return false;
6759 }
6760 else if (TREE_CODE (else_clause) != INTEGER_CST
6761 && TREE_CODE (else_clause) != REAL_CST
6762 && TREE_CODE (else_clause) != FIXED_CST)
6763 return false;
6764
6765 unsigned int prec = GET_MODE_BITSIZE (TYPE_MODE (TREE_TYPE (vectype)));
6766 /* The result of a vector comparison should be signed type. */
6767 tree cmp_type = build_nonstandard_integer_type (prec, 0);
6768 vec_cmp_type = get_same_sized_vectype (cmp_type, vectype);
6769 if (vec_cmp_type == NULL_TREE)
6770 return false;
6771
6772 if (!vec_stmt)
6773 {
6774 STMT_VINFO_TYPE (stmt_info) = condition_vec_info_type;
6775 return expand_vec_cond_expr_p (vectype, comp_vectype);
6776 }
6777
6778 /* Transform. */
6779
6780 if (!slp_node)
6781 {
6782 vec_oprnds0.create (1);
6783 vec_oprnds1.create (1);
6784 vec_oprnds2.create (1);
6785 vec_oprnds3.create (1);
6786 }
6787
6788 /* Handle def. */
6789 scalar_dest = gimple_assign_lhs (stmt);
6790 vec_dest = vect_create_destination_var (scalar_dest, vectype);
6791
6792 /* Handle cond expr. */
6793 for (j = 0; j < ncopies; j++)
6794 {
6795 gimple new_stmt = NULL;
6796 if (j == 0)
6797 {
6798 if (slp_node)
6799 {
6800 auto_vec<tree, 4> ops;
6801 auto_vec<vec<tree>, 4> vec_defs;
6802
6803 ops.safe_push (TREE_OPERAND (cond_expr, 0));
6804 ops.safe_push (TREE_OPERAND (cond_expr, 1));
6805 ops.safe_push (then_clause);
6806 ops.safe_push (else_clause);
6807 vect_get_slp_defs (ops, slp_node, &vec_defs, -1);
6808 vec_oprnds3 = vec_defs.pop ();
6809 vec_oprnds2 = vec_defs.pop ();
6810 vec_oprnds1 = vec_defs.pop ();
6811 vec_oprnds0 = vec_defs.pop ();
6812
6813 ops.release ();
6814 vec_defs.release ();
6815 }
6816 else
6817 {
6818 gimple gtemp;
6819 vec_cond_lhs =
6820 vect_get_vec_def_for_operand (TREE_OPERAND (cond_expr, 0),
6821 stmt, NULL);
6822 vect_is_simple_use (TREE_OPERAND (cond_expr, 0), stmt,
6823 loop_vinfo, NULL, &gtemp, &def, &dts[0]);
6824
6825 vec_cond_rhs =
6826 vect_get_vec_def_for_operand (TREE_OPERAND (cond_expr, 1),
6827 stmt, NULL);
6828 vect_is_simple_use (TREE_OPERAND (cond_expr, 1), stmt,
6829 loop_vinfo, NULL, &gtemp, &def, &dts[1]);
6830 if (reduc_index == 1)
6831 vec_then_clause = reduc_def;
6832 else
6833 {
6834 vec_then_clause = vect_get_vec_def_for_operand (then_clause,
6835 stmt, NULL);
6836 vect_is_simple_use (then_clause, stmt, loop_vinfo,
6837 NULL, &gtemp, &def, &dts[2]);
6838 }
6839 if (reduc_index == 2)
6840 vec_else_clause = reduc_def;
6841 else
6842 {
6843 vec_else_clause = vect_get_vec_def_for_operand (else_clause,
6844 stmt, NULL);
6845 vect_is_simple_use (else_clause, stmt, loop_vinfo,
6846 NULL, &gtemp, &def, &dts[3]);
6847 }
6848 }
6849 }
6850 else
6851 {
6852 vec_cond_lhs = vect_get_vec_def_for_stmt_copy (dts[0],
6853 vec_oprnds0.pop ());
6854 vec_cond_rhs = vect_get_vec_def_for_stmt_copy (dts[1],
6855 vec_oprnds1.pop ());
6856 vec_then_clause = vect_get_vec_def_for_stmt_copy (dts[2],
6857 vec_oprnds2.pop ());
6858 vec_else_clause = vect_get_vec_def_for_stmt_copy (dts[3],
6859 vec_oprnds3.pop ());
6860 }
6861
6862 if (!slp_node)
6863 {
6864 vec_oprnds0.quick_push (vec_cond_lhs);
6865 vec_oprnds1.quick_push (vec_cond_rhs);
6866 vec_oprnds2.quick_push (vec_then_clause);
6867 vec_oprnds3.quick_push (vec_else_clause);
6868 }
6869
6870 /* Arguments are ready. Create the new vector stmt. */
6871 FOR_EACH_VEC_ELT (vec_oprnds0, i, vec_cond_lhs)
6872 {
6873 vec_cond_rhs = vec_oprnds1[i];
6874 vec_then_clause = vec_oprnds2[i];
6875 vec_else_clause = vec_oprnds3[i];
6876
6877 vec_compare = build2 (TREE_CODE (cond_expr), vec_cmp_type,
6878 vec_cond_lhs, vec_cond_rhs);
6879 vec_cond_expr = build3 (VEC_COND_EXPR, vectype,
6880 vec_compare, vec_then_clause, vec_else_clause);
6881
6882 new_stmt = gimple_build_assign (vec_dest, vec_cond_expr);
6883 new_temp = make_ssa_name (vec_dest, new_stmt);
6884 gimple_assign_set_lhs (new_stmt, new_temp);
6885 vect_finish_stmt_generation (stmt, new_stmt, gsi);
6886 if (slp_node)
6887 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
6888 }
6889
6890 if (slp_node)
6891 continue;
6892
6893 if (j == 0)
6894 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
6895 else
6896 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
6897
6898 prev_stmt_info = vinfo_for_stmt (new_stmt);
6899 }
6900
6901 vec_oprnds0.release ();
6902 vec_oprnds1.release ();
6903 vec_oprnds2.release ();
6904 vec_oprnds3.release ();
6905
6906 return true;
6907 }
6908
6909
6910 /* Make sure the statement is vectorizable. */
6911
6912 bool
6913 vect_analyze_stmt (gimple stmt, bool *need_to_vectorize, slp_tree node)
6914 {
6915 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
6916 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
6917 enum vect_relevant relevance = STMT_VINFO_RELEVANT (stmt_info);
6918 bool ok;
6919 tree scalar_type, vectype;
6920 gimple pattern_stmt;
6921 gimple_seq pattern_def_seq;
6922
6923 if (dump_enabled_p ())
6924 {
6925 dump_printf_loc (MSG_NOTE, vect_location, "==> examining statement: ");
6926 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
6927 dump_printf (MSG_NOTE, "\n");
6928 }
6929
6930 if (gimple_has_volatile_ops (stmt))
6931 {
6932 if (dump_enabled_p ())
6933 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
6934 "not vectorized: stmt has volatile operands\n");
6935
6936 return false;
6937 }
6938
6939 /* Skip stmts that do not need to be vectorized. In loops this is expected
6940 to include:
6941 - the COND_EXPR which is the loop exit condition
6942 - any LABEL_EXPRs in the loop
6943 - computations that are used only for array indexing or loop control.
6944 In basic blocks we only analyze statements that are a part of some SLP
6945 instance, therefore, all the statements are relevant.
6946
6947 Pattern statement needs to be analyzed instead of the original statement
6948 if the original statement is not relevant. Otherwise, we analyze both
6949 statements. In basic blocks we are called from some SLP instance
6950 traversal, don't analyze pattern stmts instead, the pattern stmts
6951 already will be part of SLP instance. */
6952
6953 pattern_stmt = STMT_VINFO_RELATED_STMT (stmt_info);
6954 if (!STMT_VINFO_RELEVANT_P (stmt_info)
6955 && !STMT_VINFO_LIVE_P (stmt_info))
6956 {
6957 if (STMT_VINFO_IN_PATTERN_P (stmt_info)
6958 && pattern_stmt
6959 && (STMT_VINFO_RELEVANT_P (vinfo_for_stmt (pattern_stmt))
6960 || STMT_VINFO_LIVE_P (vinfo_for_stmt (pattern_stmt))))
6961 {
6962 /* Analyze PATTERN_STMT instead of the original stmt. */
6963 stmt = pattern_stmt;
6964 stmt_info = vinfo_for_stmt (pattern_stmt);
6965 if (dump_enabled_p ())
6966 {
6967 dump_printf_loc (MSG_NOTE, vect_location,
6968 "==> examining pattern statement: ");
6969 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
6970 dump_printf (MSG_NOTE, "\n");
6971 }
6972 }
6973 else
6974 {
6975 if (dump_enabled_p ())
6976 dump_printf_loc (MSG_NOTE, vect_location, "irrelevant.\n");
6977
6978 return true;
6979 }
6980 }
6981 else if (STMT_VINFO_IN_PATTERN_P (stmt_info)
6982 && node == NULL
6983 && pattern_stmt
6984 && (STMT_VINFO_RELEVANT_P (vinfo_for_stmt (pattern_stmt))
6985 || STMT_VINFO_LIVE_P (vinfo_for_stmt (pattern_stmt))))
6986 {
6987 /* Analyze PATTERN_STMT too. */
6988 if (dump_enabled_p ())
6989 {
6990 dump_printf_loc (MSG_NOTE, vect_location,
6991 "==> examining pattern statement: ");
6992 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
6993 dump_printf (MSG_NOTE, "\n");
6994 }
6995
6996 if (!vect_analyze_stmt (pattern_stmt, need_to_vectorize, node))
6997 return false;
6998 }
6999
7000 if (is_pattern_stmt_p (stmt_info)
7001 && node == NULL
7002 && (pattern_def_seq = STMT_VINFO_PATTERN_DEF_SEQ (stmt_info)))
7003 {
7004 gimple_stmt_iterator si;
7005
7006 for (si = gsi_start (pattern_def_seq); !gsi_end_p (si); gsi_next (&si))
7007 {
7008 gimple pattern_def_stmt = gsi_stmt (si);
7009 if (STMT_VINFO_RELEVANT_P (vinfo_for_stmt (pattern_def_stmt))
7010 || STMT_VINFO_LIVE_P (vinfo_for_stmt (pattern_def_stmt)))
7011 {
7012 /* Analyze def stmt of STMT if it's a pattern stmt. */
7013 if (dump_enabled_p ())
7014 {
7015 dump_printf_loc (MSG_NOTE, vect_location,
7016 "==> examining pattern def statement: ");
7017 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, pattern_def_stmt, 0);
7018 dump_printf (MSG_NOTE, "\n");
7019 }
7020
7021 if (!vect_analyze_stmt (pattern_def_stmt,
7022 need_to_vectorize, node))
7023 return false;
7024 }
7025 }
7026 }
7027
7028 switch (STMT_VINFO_DEF_TYPE (stmt_info))
7029 {
7030 case vect_internal_def:
7031 break;
7032
7033 case vect_reduction_def:
7034 case vect_nested_cycle:
7035 gcc_assert (!bb_vinfo && (relevance == vect_used_in_outer
7036 || relevance == vect_used_in_outer_by_reduction
7037 || relevance == vect_unused_in_scope));
7038 break;
7039
7040 case vect_induction_def:
7041 case vect_constant_def:
7042 case vect_external_def:
7043 case vect_unknown_def_type:
7044 default:
7045 gcc_unreachable ();
7046 }
7047
7048 if (bb_vinfo)
7049 {
7050 gcc_assert (PURE_SLP_STMT (stmt_info));
7051
7052 scalar_type = TREE_TYPE (gimple_get_lhs (stmt));
7053 if (dump_enabled_p ())
7054 {
7055 dump_printf_loc (MSG_NOTE, vect_location,
7056 "get vectype for scalar type: ");
7057 dump_generic_expr (MSG_NOTE, TDF_SLIM, scalar_type);
7058 dump_printf (MSG_NOTE, "\n");
7059 }
7060
7061 vectype = get_vectype_for_scalar_type (scalar_type);
7062 if (!vectype)
7063 {
7064 if (dump_enabled_p ())
7065 {
7066 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
7067 "not SLPed: unsupported data-type ");
7068 dump_generic_expr (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
7069 scalar_type);
7070 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
7071 }
7072 return false;
7073 }
7074
7075 if (dump_enabled_p ())
7076 {
7077 dump_printf_loc (MSG_NOTE, vect_location, "vectype: ");
7078 dump_generic_expr (MSG_NOTE, TDF_SLIM, vectype);
7079 dump_printf (MSG_NOTE, "\n");
7080 }
7081
7082 STMT_VINFO_VECTYPE (stmt_info) = vectype;
7083 }
7084
7085 if (STMT_VINFO_RELEVANT_P (stmt_info))
7086 {
7087 gcc_assert (!VECTOR_MODE_P (TYPE_MODE (gimple_expr_type (stmt))));
7088 gcc_assert (STMT_VINFO_VECTYPE (stmt_info)
7089 || (is_gimple_call (stmt)
7090 && gimple_call_lhs (stmt) == NULL_TREE));
7091 *need_to_vectorize = true;
7092 }
7093
7094 ok = true;
7095 if (!bb_vinfo
7096 && (STMT_VINFO_RELEVANT_P (stmt_info)
7097 || STMT_VINFO_DEF_TYPE (stmt_info) == vect_reduction_def))
7098 ok = (vectorizable_simd_clone_call (stmt, NULL, NULL, NULL)
7099 || vectorizable_conversion (stmt, NULL, NULL, NULL)
7100 || vectorizable_shift (stmt, NULL, NULL, NULL)
7101 || vectorizable_operation (stmt, NULL, NULL, NULL)
7102 || vectorizable_assignment (stmt, NULL, NULL, NULL)
7103 || vectorizable_load (stmt, NULL, NULL, NULL, NULL)
7104 || vectorizable_call (stmt, NULL, NULL, NULL)
7105 || vectorizable_store (stmt, NULL, NULL, NULL)
7106 || vectorizable_reduction (stmt, NULL, NULL, NULL)
7107 || vectorizable_condition (stmt, NULL, NULL, NULL, 0, NULL));
7108 else
7109 {
7110 if (bb_vinfo)
7111 ok = (vectorizable_simd_clone_call (stmt, NULL, NULL, node)
7112 || vectorizable_conversion (stmt, NULL, NULL, node)
7113 || vectorizable_shift (stmt, NULL, NULL, node)
7114 || vectorizable_operation (stmt, NULL, NULL, node)
7115 || vectorizable_assignment (stmt, NULL, NULL, node)
7116 || vectorizable_load (stmt, NULL, NULL, node, NULL)
7117 || vectorizable_call (stmt, NULL, NULL, node)
7118 || vectorizable_store (stmt, NULL, NULL, node)
7119 || vectorizable_condition (stmt, NULL, NULL, NULL, 0, node));
7120 }
7121
7122 if (!ok)
7123 {
7124 if (dump_enabled_p ())
7125 {
7126 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
7127 "not vectorized: relevant stmt not ");
7128 dump_printf (MSG_MISSED_OPTIMIZATION, "supported: ");
7129 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM, stmt, 0);
7130 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
7131 }
7132
7133 return false;
7134 }
7135
7136 if (bb_vinfo)
7137 return true;
7138
7139 /* Stmts that are (also) "live" (i.e. - that are used out of the loop)
7140 need extra handling, except for vectorizable reductions. */
7141 if (STMT_VINFO_LIVE_P (stmt_info)
7142 && STMT_VINFO_TYPE (stmt_info) != reduc_vec_info_type)
7143 ok = vectorizable_live_operation (stmt, NULL, NULL);
7144
7145 if (!ok)
7146 {
7147 if (dump_enabled_p ())
7148 {
7149 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
7150 "not vectorized: live stmt not ");
7151 dump_printf (MSG_MISSED_OPTIMIZATION, "supported: ");
7152 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM, stmt, 0);
7153 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
7154 }
7155
7156 return false;
7157 }
7158
7159 return true;
7160 }
7161
7162
7163 /* Function vect_transform_stmt.
7164
7165 Create a vectorized stmt to replace STMT, and insert it at BSI. */
7166
7167 bool
7168 vect_transform_stmt (gimple stmt, gimple_stmt_iterator *gsi,
7169 bool *grouped_store, slp_tree slp_node,
7170 slp_instance slp_node_instance)
7171 {
7172 bool is_store = false;
7173 gimple vec_stmt = NULL;
7174 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
7175 bool done;
7176
7177 switch (STMT_VINFO_TYPE (stmt_info))
7178 {
7179 case type_demotion_vec_info_type:
7180 case type_promotion_vec_info_type:
7181 case type_conversion_vec_info_type:
7182 done = vectorizable_conversion (stmt, gsi, &vec_stmt, slp_node);
7183 gcc_assert (done);
7184 break;
7185
7186 case induc_vec_info_type:
7187 gcc_assert (!slp_node);
7188 done = vectorizable_induction (stmt, gsi, &vec_stmt);
7189 gcc_assert (done);
7190 break;
7191
7192 case shift_vec_info_type:
7193 done = vectorizable_shift (stmt, gsi, &vec_stmt, slp_node);
7194 gcc_assert (done);
7195 break;
7196
7197 case op_vec_info_type:
7198 done = vectorizable_operation (stmt, gsi, &vec_stmt, slp_node);
7199 gcc_assert (done);
7200 break;
7201
7202 case assignment_vec_info_type:
7203 done = vectorizable_assignment (stmt, gsi, &vec_stmt, slp_node);
7204 gcc_assert (done);
7205 break;
7206
7207 case load_vec_info_type:
7208 done = vectorizable_load (stmt, gsi, &vec_stmt, slp_node,
7209 slp_node_instance);
7210 gcc_assert (done);
7211 break;
7212
7213 case store_vec_info_type:
7214 done = vectorizable_store (stmt, gsi, &vec_stmt, slp_node);
7215 gcc_assert (done);
7216 if (STMT_VINFO_GROUPED_ACCESS (stmt_info) && !slp_node)
7217 {
7218 /* In case of interleaving, the whole chain is vectorized when the
7219 last store in the chain is reached. Store stmts before the last
7220 one are skipped, and there vec_stmt_info shouldn't be freed
7221 meanwhile. */
7222 *grouped_store = true;
7223 if (STMT_VINFO_VEC_STMT (stmt_info))
7224 is_store = true;
7225 }
7226 else
7227 is_store = true;
7228 break;
7229
7230 case condition_vec_info_type:
7231 done = vectorizable_condition (stmt, gsi, &vec_stmt, NULL, 0, slp_node);
7232 gcc_assert (done);
7233 break;
7234
7235 case call_vec_info_type:
7236 done = vectorizable_call (stmt, gsi, &vec_stmt, slp_node);
7237 stmt = gsi_stmt (*gsi);
7238 if (is_gimple_call (stmt)
7239 && gimple_call_internal_p (stmt)
7240 && gimple_call_internal_fn (stmt) == IFN_MASK_STORE)
7241 is_store = true;
7242 break;
7243
7244 case call_simd_clone_vec_info_type:
7245 done = vectorizable_simd_clone_call (stmt, gsi, &vec_stmt, slp_node);
7246 stmt = gsi_stmt (*gsi);
7247 break;
7248
7249 case reduc_vec_info_type:
7250 done = vectorizable_reduction (stmt, gsi, &vec_stmt, slp_node);
7251 gcc_assert (done);
7252 break;
7253
7254 default:
7255 if (!STMT_VINFO_LIVE_P (stmt_info))
7256 {
7257 if (dump_enabled_p ())
7258 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
7259 "stmt not supported.\n");
7260 gcc_unreachable ();
7261 }
7262 }
7263
7264 /* Handle inner-loop stmts whose DEF is used in the loop-nest that
7265 is being vectorized, but outside the immediately enclosing loop. */
7266 if (vec_stmt
7267 && STMT_VINFO_LOOP_VINFO (stmt_info)
7268 && nested_in_vect_loop_p (LOOP_VINFO_LOOP (
7269 STMT_VINFO_LOOP_VINFO (stmt_info)), stmt)
7270 && STMT_VINFO_TYPE (stmt_info) != reduc_vec_info_type
7271 && (STMT_VINFO_RELEVANT (stmt_info) == vect_used_in_outer
7272 || STMT_VINFO_RELEVANT (stmt_info) ==
7273 vect_used_in_outer_by_reduction))
7274 {
7275 struct loop *innerloop = LOOP_VINFO_LOOP (
7276 STMT_VINFO_LOOP_VINFO (stmt_info))->inner;
7277 imm_use_iterator imm_iter;
7278 use_operand_p use_p;
7279 tree scalar_dest;
7280 gimple exit_phi;
7281
7282 if (dump_enabled_p ())
7283 dump_printf_loc (MSG_NOTE, vect_location,
7284 "Record the vdef for outer-loop vectorization.\n");
7285
7286 /* Find the relevant loop-exit phi-node, and reord the vec_stmt there
7287 (to be used when vectorizing outer-loop stmts that use the DEF of
7288 STMT). */
7289 if (gimple_code (stmt) == GIMPLE_PHI)
7290 scalar_dest = PHI_RESULT (stmt);
7291 else
7292 scalar_dest = gimple_assign_lhs (stmt);
7293
7294 FOR_EACH_IMM_USE_FAST (use_p, imm_iter, scalar_dest)
7295 {
7296 if (!flow_bb_inside_loop_p (innerloop, gimple_bb (USE_STMT (use_p))))
7297 {
7298 exit_phi = USE_STMT (use_p);
7299 STMT_VINFO_VEC_STMT (vinfo_for_stmt (exit_phi)) = vec_stmt;
7300 }
7301 }
7302 }
7303
7304 /* Handle stmts whose DEF is used outside the loop-nest that is
7305 being vectorized. */
7306 if (STMT_VINFO_LIVE_P (stmt_info)
7307 && STMT_VINFO_TYPE (stmt_info) != reduc_vec_info_type)
7308 {
7309 done = vectorizable_live_operation (stmt, gsi, &vec_stmt);
7310 gcc_assert (done);
7311 }
7312
7313 if (vec_stmt)
7314 STMT_VINFO_VEC_STMT (stmt_info) = vec_stmt;
7315
7316 return is_store;
7317 }
7318
7319
7320 /* Remove a group of stores (for SLP or interleaving), free their
7321 stmt_vec_info. */
7322
7323 void
7324 vect_remove_stores (gimple first_stmt)
7325 {
7326 gimple next = first_stmt;
7327 gimple tmp;
7328 gimple_stmt_iterator next_si;
7329
7330 while (next)
7331 {
7332 stmt_vec_info stmt_info = vinfo_for_stmt (next);
7333
7334 tmp = GROUP_NEXT_ELEMENT (stmt_info);
7335 if (is_pattern_stmt_p (stmt_info))
7336 next = STMT_VINFO_RELATED_STMT (stmt_info);
7337 /* Free the attached stmt_vec_info and remove the stmt. */
7338 next_si = gsi_for_stmt (next);
7339 unlink_stmt_vdef (next);
7340 gsi_remove (&next_si, true);
7341 release_defs (next);
7342 free_stmt_vec_info (next);
7343 next = tmp;
7344 }
7345 }
7346
7347
7348 /* Function new_stmt_vec_info.
7349
7350 Create and initialize a new stmt_vec_info struct for STMT. */
7351
7352 stmt_vec_info
7353 new_stmt_vec_info (gimple stmt, loop_vec_info loop_vinfo,
7354 bb_vec_info bb_vinfo)
7355 {
7356 stmt_vec_info res;
7357 res = (stmt_vec_info) xcalloc (1, sizeof (struct _stmt_vec_info));
7358
7359 STMT_VINFO_TYPE (res) = undef_vec_info_type;
7360 STMT_VINFO_STMT (res) = stmt;
7361 STMT_VINFO_LOOP_VINFO (res) = loop_vinfo;
7362 STMT_VINFO_BB_VINFO (res) = bb_vinfo;
7363 STMT_VINFO_RELEVANT (res) = vect_unused_in_scope;
7364 STMT_VINFO_LIVE_P (res) = false;
7365 STMT_VINFO_VECTYPE (res) = NULL;
7366 STMT_VINFO_VEC_STMT (res) = NULL;
7367 STMT_VINFO_VECTORIZABLE (res) = true;
7368 STMT_VINFO_IN_PATTERN_P (res) = false;
7369 STMT_VINFO_RELATED_STMT (res) = NULL;
7370 STMT_VINFO_PATTERN_DEF_SEQ (res) = NULL;
7371 STMT_VINFO_DATA_REF (res) = NULL;
7372
7373 STMT_VINFO_DR_BASE_ADDRESS (res) = NULL;
7374 STMT_VINFO_DR_OFFSET (res) = NULL;
7375 STMT_VINFO_DR_INIT (res) = NULL;
7376 STMT_VINFO_DR_STEP (res) = NULL;
7377 STMT_VINFO_DR_ALIGNED_TO (res) = NULL;
7378
7379 if (gimple_code (stmt) == GIMPLE_PHI
7380 && is_loop_header_bb_p (gimple_bb (stmt)))
7381 STMT_VINFO_DEF_TYPE (res) = vect_unknown_def_type;
7382 else
7383 STMT_VINFO_DEF_TYPE (res) = vect_internal_def;
7384
7385 STMT_VINFO_SAME_ALIGN_REFS (res).create (0);
7386 STMT_SLP_TYPE (res) = loop_vect;
7387 GROUP_FIRST_ELEMENT (res) = NULL;
7388 GROUP_NEXT_ELEMENT (res) = NULL;
7389 GROUP_SIZE (res) = 0;
7390 GROUP_STORE_COUNT (res) = 0;
7391 GROUP_GAP (res) = 0;
7392 GROUP_SAME_DR_STMT (res) = NULL;
7393
7394 return res;
7395 }
7396
7397
7398 /* Create a hash table for stmt_vec_info. */
7399
7400 void
7401 init_stmt_vec_info_vec (void)
7402 {
7403 gcc_assert (!stmt_vec_info_vec.exists ());
7404 stmt_vec_info_vec.create (50);
7405 }
7406
7407
7408 /* Free hash table for stmt_vec_info. */
7409
7410 void
7411 free_stmt_vec_info_vec (void)
7412 {
7413 unsigned int i;
7414 vec_void_p info;
7415 FOR_EACH_VEC_ELT (stmt_vec_info_vec, i, info)
7416 if (info != NULL)
7417 free_stmt_vec_info (STMT_VINFO_STMT ((stmt_vec_info) info));
7418 gcc_assert (stmt_vec_info_vec.exists ());
7419 stmt_vec_info_vec.release ();
7420 }
7421
7422
7423 /* Free stmt vectorization related info. */
7424
7425 void
7426 free_stmt_vec_info (gimple stmt)
7427 {
7428 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
7429
7430 if (!stmt_info)
7431 return;
7432
7433 /* Check if this statement has a related "pattern stmt"
7434 (introduced by the vectorizer during the pattern recognition
7435 pass). Free pattern's stmt_vec_info and def stmt's stmt_vec_info
7436 too. */
7437 if (STMT_VINFO_IN_PATTERN_P (stmt_info))
7438 {
7439 stmt_vec_info patt_info
7440 = vinfo_for_stmt (STMT_VINFO_RELATED_STMT (stmt_info));
7441 if (patt_info)
7442 {
7443 gimple_seq seq = STMT_VINFO_PATTERN_DEF_SEQ (patt_info);
7444 gimple patt_stmt = STMT_VINFO_STMT (patt_info);
7445 gimple_set_bb (patt_stmt, NULL);
7446 tree lhs = gimple_get_lhs (patt_stmt);
7447 if (TREE_CODE (lhs) == SSA_NAME)
7448 release_ssa_name (lhs);
7449 if (seq)
7450 {
7451 gimple_stmt_iterator si;
7452 for (si = gsi_start (seq); !gsi_end_p (si); gsi_next (&si))
7453 {
7454 gimple seq_stmt = gsi_stmt (si);
7455 gimple_set_bb (seq_stmt, NULL);
7456 lhs = gimple_get_lhs (patt_stmt);
7457 if (TREE_CODE (lhs) == SSA_NAME)
7458 release_ssa_name (lhs);
7459 free_stmt_vec_info (seq_stmt);
7460 }
7461 }
7462 free_stmt_vec_info (patt_stmt);
7463 }
7464 }
7465
7466 STMT_VINFO_SAME_ALIGN_REFS (stmt_info).release ();
7467 set_vinfo_for_stmt (stmt, NULL);
7468 free (stmt_info);
7469 }
7470
7471
7472 /* Function get_vectype_for_scalar_type_and_size.
7473
7474 Returns the vector type corresponding to SCALAR_TYPE and SIZE as supported
7475 by the target. */
7476
7477 static tree
7478 get_vectype_for_scalar_type_and_size (tree scalar_type, unsigned size)
7479 {
7480 enum machine_mode inner_mode = TYPE_MODE (scalar_type);
7481 enum machine_mode simd_mode;
7482 unsigned int nbytes = GET_MODE_SIZE (inner_mode);
7483 int nunits;
7484 tree vectype;
7485
7486 if (nbytes == 0)
7487 return NULL_TREE;
7488
7489 if (GET_MODE_CLASS (inner_mode) != MODE_INT
7490 && GET_MODE_CLASS (inner_mode) != MODE_FLOAT)
7491 return NULL_TREE;
7492
7493 /* For vector types of elements whose mode precision doesn't
7494 match their types precision we use a element type of mode
7495 precision. The vectorization routines will have to make sure
7496 they support the proper result truncation/extension.
7497 We also make sure to build vector types with INTEGER_TYPE
7498 component type only. */
7499 if (INTEGRAL_TYPE_P (scalar_type)
7500 && (GET_MODE_BITSIZE (inner_mode) != TYPE_PRECISION (scalar_type)
7501 || TREE_CODE (scalar_type) != INTEGER_TYPE))
7502 scalar_type = build_nonstandard_integer_type (GET_MODE_BITSIZE (inner_mode),
7503 TYPE_UNSIGNED (scalar_type));
7504
7505 /* We shouldn't end up building VECTOR_TYPEs of non-scalar components.
7506 When the component mode passes the above test simply use a type
7507 corresponding to that mode. The theory is that any use that
7508 would cause problems with this will disable vectorization anyway. */
7509 else if (!SCALAR_FLOAT_TYPE_P (scalar_type)
7510 && !INTEGRAL_TYPE_P (scalar_type))
7511 scalar_type = lang_hooks.types.type_for_mode (inner_mode, 1);
7512
7513 /* We can't build a vector type of elements with alignment bigger than
7514 their size. */
7515 else if (nbytes < TYPE_ALIGN_UNIT (scalar_type))
7516 scalar_type = lang_hooks.types.type_for_mode (inner_mode,
7517 TYPE_UNSIGNED (scalar_type));
7518
7519 /* If we felt back to using the mode fail if there was
7520 no scalar type for it. */
7521 if (scalar_type == NULL_TREE)
7522 return NULL_TREE;
7523
7524 /* If no size was supplied use the mode the target prefers. Otherwise
7525 lookup a vector mode of the specified size. */
7526 if (size == 0)
7527 simd_mode = targetm.vectorize.preferred_simd_mode (inner_mode);
7528 else
7529 simd_mode = mode_for_vector (inner_mode, size / nbytes);
7530 nunits = GET_MODE_SIZE (simd_mode) / nbytes;
7531 if (nunits <= 1)
7532 return NULL_TREE;
7533
7534 vectype = build_vector_type (scalar_type, nunits);
7535
7536 if (!VECTOR_MODE_P (TYPE_MODE (vectype))
7537 && !INTEGRAL_MODE_P (TYPE_MODE (vectype)))
7538 return NULL_TREE;
7539
7540 return vectype;
7541 }
7542
7543 unsigned int current_vector_size;
7544
7545 /* Function get_vectype_for_scalar_type.
7546
7547 Returns the vector type corresponding to SCALAR_TYPE as supported
7548 by the target. */
7549
7550 tree
7551 get_vectype_for_scalar_type (tree scalar_type)
7552 {
7553 tree vectype;
7554 vectype = get_vectype_for_scalar_type_and_size (scalar_type,
7555 current_vector_size);
7556 if (vectype
7557 && current_vector_size == 0)
7558 current_vector_size = GET_MODE_SIZE (TYPE_MODE (vectype));
7559 return vectype;
7560 }
7561
7562 /* Function get_same_sized_vectype
7563
7564 Returns a vector type corresponding to SCALAR_TYPE of size
7565 VECTOR_TYPE if supported by the target. */
7566
7567 tree
7568 get_same_sized_vectype (tree scalar_type, tree vector_type)
7569 {
7570 return get_vectype_for_scalar_type_and_size
7571 (scalar_type, GET_MODE_SIZE (TYPE_MODE (vector_type)));
7572 }
7573
7574 /* Function vect_is_simple_use.
7575
7576 Input:
7577 LOOP_VINFO - the vect info of the loop that is being vectorized.
7578 BB_VINFO - the vect info of the basic block that is being vectorized.
7579 OPERAND - operand of STMT in the loop or bb.
7580 DEF - the defining stmt in case OPERAND is an SSA_NAME.
7581
7582 Returns whether a stmt with OPERAND can be vectorized.
7583 For loops, supportable operands are constants, loop invariants, and operands
7584 that are defined by the current iteration of the loop. Unsupportable
7585 operands are those that are defined by a previous iteration of the loop (as
7586 is the case in reduction/induction computations).
7587 For basic blocks, supportable operands are constants and bb invariants.
7588 For now, operands defined outside the basic block are not supported. */
7589
7590 bool
7591 vect_is_simple_use (tree operand, gimple stmt, loop_vec_info loop_vinfo,
7592 bb_vec_info bb_vinfo, gimple *def_stmt,
7593 tree *def, enum vect_def_type *dt)
7594 {
7595 basic_block bb;
7596 stmt_vec_info stmt_vinfo;
7597 struct loop *loop = NULL;
7598
7599 if (loop_vinfo)
7600 loop = LOOP_VINFO_LOOP (loop_vinfo);
7601
7602 *def_stmt = NULL;
7603 *def = NULL_TREE;
7604
7605 if (dump_enabled_p ())
7606 {
7607 dump_printf_loc (MSG_NOTE, vect_location,
7608 "vect_is_simple_use: operand ");
7609 dump_generic_expr (MSG_NOTE, TDF_SLIM, operand);
7610 dump_printf (MSG_NOTE, "\n");
7611 }
7612
7613 if (CONSTANT_CLASS_P (operand))
7614 {
7615 *dt = vect_constant_def;
7616 return true;
7617 }
7618
7619 if (is_gimple_min_invariant (operand))
7620 {
7621 *def = operand;
7622 *dt = vect_external_def;
7623 return true;
7624 }
7625
7626 if (TREE_CODE (operand) == PAREN_EXPR)
7627 {
7628 if (dump_enabled_p ())
7629 dump_printf_loc (MSG_NOTE, vect_location, "non-associatable copy.\n");
7630 operand = TREE_OPERAND (operand, 0);
7631 }
7632
7633 if (TREE_CODE (operand) != SSA_NAME)
7634 {
7635 if (dump_enabled_p ())
7636 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
7637 "not ssa-name.\n");
7638 return false;
7639 }
7640
7641 *def_stmt = SSA_NAME_DEF_STMT (operand);
7642 if (*def_stmt == NULL)
7643 {
7644 if (dump_enabled_p ())
7645 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
7646 "no def_stmt.\n");
7647 return false;
7648 }
7649
7650 if (dump_enabled_p ())
7651 {
7652 dump_printf_loc (MSG_NOTE, vect_location, "def_stmt: ");
7653 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, *def_stmt, 0);
7654 dump_printf (MSG_NOTE, "\n");
7655 }
7656
7657 /* Empty stmt is expected only in case of a function argument.
7658 (Otherwise - we expect a phi_node or a GIMPLE_ASSIGN). */
7659 if (gimple_nop_p (*def_stmt))
7660 {
7661 *def = operand;
7662 *dt = vect_external_def;
7663 return true;
7664 }
7665
7666 bb = gimple_bb (*def_stmt);
7667
7668 if ((loop && !flow_bb_inside_loop_p (loop, bb))
7669 || (!loop && bb != BB_VINFO_BB (bb_vinfo))
7670 || (!loop && gimple_code (*def_stmt) == GIMPLE_PHI))
7671 *dt = vect_external_def;
7672 else
7673 {
7674 stmt_vinfo = vinfo_for_stmt (*def_stmt);
7675 *dt = STMT_VINFO_DEF_TYPE (stmt_vinfo);
7676 }
7677
7678 if (*dt == vect_unknown_def_type
7679 || (stmt
7680 && *dt == vect_double_reduction_def
7681 && gimple_code (stmt) != GIMPLE_PHI))
7682 {
7683 if (dump_enabled_p ())
7684 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
7685 "Unsupported pattern.\n");
7686 return false;
7687 }
7688
7689 if (dump_enabled_p ())
7690 dump_printf_loc (MSG_NOTE, vect_location, "type of def: %d.\n", *dt);
7691
7692 switch (gimple_code (*def_stmt))
7693 {
7694 case GIMPLE_PHI:
7695 *def = gimple_phi_result (*def_stmt);
7696 break;
7697
7698 case GIMPLE_ASSIGN:
7699 *def = gimple_assign_lhs (*def_stmt);
7700 break;
7701
7702 case GIMPLE_CALL:
7703 *def = gimple_call_lhs (*def_stmt);
7704 if (*def != NULL)
7705 break;
7706 /* FALLTHRU */
7707 default:
7708 if (dump_enabled_p ())
7709 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
7710 "unsupported defining stmt:\n");
7711 return false;
7712 }
7713
7714 return true;
7715 }
7716
7717 /* Function vect_is_simple_use_1.
7718
7719 Same as vect_is_simple_use_1 but also determines the vector operand
7720 type of OPERAND and stores it to *VECTYPE. If the definition of
7721 OPERAND is vect_uninitialized_def, vect_constant_def or
7722 vect_external_def *VECTYPE will be set to NULL_TREE and the caller
7723 is responsible to compute the best suited vector type for the
7724 scalar operand. */
7725
7726 bool
7727 vect_is_simple_use_1 (tree operand, gimple stmt, loop_vec_info loop_vinfo,
7728 bb_vec_info bb_vinfo, gimple *def_stmt,
7729 tree *def, enum vect_def_type *dt, tree *vectype)
7730 {
7731 if (!vect_is_simple_use (operand, stmt, loop_vinfo, bb_vinfo, def_stmt,
7732 def, dt))
7733 return false;
7734
7735 /* Now get a vector type if the def is internal, otherwise supply
7736 NULL_TREE and leave it up to the caller to figure out a proper
7737 type for the use stmt. */
7738 if (*dt == vect_internal_def
7739 || *dt == vect_induction_def
7740 || *dt == vect_reduction_def
7741 || *dt == vect_double_reduction_def
7742 || *dt == vect_nested_cycle)
7743 {
7744 stmt_vec_info stmt_info = vinfo_for_stmt (*def_stmt);
7745
7746 if (STMT_VINFO_IN_PATTERN_P (stmt_info)
7747 && !STMT_VINFO_RELEVANT (stmt_info)
7748 && !STMT_VINFO_LIVE_P (stmt_info))
7749 stmt_info = vinfo_for_stmt (STMT_VINFO_RELATED_STMT (stmt_info));
7750
7751 *vectype = STMT_VINFO_VECTYPE (stmt_info);
7752 gcc_assert (*vectype != NULL_TREE);
7753 }
7754 else if (*dt == vect_uninitialized_def
7755 || *dt == vect_constant_def
7756 || *dt == vect_external_def)
7757 *vectype = NULL_TREE;
7758 else
7759 gcc_unreachable ();
7760
7761 return true;
7762 }
7763
7764
7765 /* Function supportable_widening_operation
7766
7767 Check whether an operation represented by the code CODE is a
7768 widening operation that is supported by the target platform in
7769 vector form (i.e., when operating on arguments of type VECTYPE_IN
7770 producing a result of type VECTYPE_OUT).
7771
7772 Widening operations we currently support are NOP (CONVERT), FLOAT
7773 and WIDEN_MULT. This function checks if these operations are supported
7774 by the target platform either directly (via vector tree-codes), or via
7775 target builtins.
7776
7777 Output:
7778 - CODE1 and CODE2 are codes of vector operations to be used when
7779 vectorizing the operation, if available.
7780 - MULTI_STEP_CVT determines the number of required intermediate steps in
7781 case of multi-step conversion (like char->short->int - in that case
7782 MULTI_STEP_CVT will be 1).
7783 - INTERM_TYPES contains the intermediate type required to perform the
7784 widening operation (short in the above example). */
7785
7786 bool
7787 supportable_widening_operation (enum tree_code code, gimple stmt,
7788 tree vectype_out, tree vectype_in,
7789 enum tree_code *code1, enum tree_code *code2,
7790 int *multi_step_cvt,
7791 vec<tree> *interm_types)
7792 {
7793 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
7794 loop_vec_info loop_info = STMT_VINFO_LOOP_VINFO (stmt_info);
7795 struct loop *vect_loop = NULL;
7796 enum machine_mode vec_mode;
7797 enum insn_code icode1, icode2;
7798 optab optab1, optab2;
7799 tree vectype = vectype_in;
7800 tree wide_vectype = vectype_out;
7801 enum tree_code c1, c2;
7802 int i;
7803 tree prev_type, intermediate_type;
7804 enum machine_mode intermediate_mode, prev_mode;
7805 optab optab3, optab4;
7806
7807 *multi_step_cvt = 0;
7808 if (loop_info)
7809 vect_loop = LOOP_VINFO_LOOP (loop_info);
7810
7811 switch (code)
7812 {
7813 case WIDEN_MULT_EXPR:
7814 /* The result of a vectorized widening operation usually requires
7815 two vectors (because the widened results do not fit into one vector).
7816 The generated vector results would normally be expected to be
7817 generated in the same order as in the original scalar computation,
7818 i.e. if 8 results are generated in each vector iteration, they are
7819 to be organized as follows:
7820 vect1: [res1,res2,res3,res4],
7821 vect2: [res5,res6,res7,res8].
7822
7823 However, in the special case that the result of the widening
7824 operation is used in a reduction computation only, the order doesn't
7825 matter (because when vectorizing a reduction we change the order of
7826 the computation). Some targets can take advantage of this and
7827 generate more efficient code. For example, targets like Altivec,
7828 that support widen_mult using a sequence of {mult_even,mult_odd}
7829 generate the following vectors:
7830 vect1: [res1,res3,res5,res7],
7831 vect2: [res2,res4,res6,res8].
7832
7833 When vectorizing outer-loops, we execute the inner-loop sequentially
7834 (each vectorized inner-loop iteration contributes to VF outer-loop
7835 iterations in parallel). We therefore don't allow to change the
7836 order of the computation in the inner-loop during outer-loop
7837 vectorization. */
7838 /* TODO: Another case in which order doesn't *really* matter is when we
7839 widen and then contract again, e.g. (short)((int)x * y >> 8).
7840 Normally, pack_trunc performs an even/odd permute, whereas the
7841 repack from an even/odd expansion would be an interleave, which
7842 would be significantly simpler for e.g. AVX2. */
7843 /* In any case, in order to avoid duplicating the code below, recurse
7844 on VEC_WIDEN_MULT_EVEN_EXPR. If it succeeds, all the return values
7845 are properly set up for the caller. If we fail, we'll continue with
7846 a VEC_WIDEN_MULT_LO/HI_EXPR check. */
7847 if (vect_loop
7848 && STMT_VINFO_RELEVANT (stmt_info) == vect_used_by_reduction
7849 && !nested_in_vect_loop_p (vect_loop, stmt)
7850 && supportable_widening_operation (VEC_WIDEN_MULT_EVEN_EXPR,
7851 stmt, vectype_out, vectype_in,
7852 code1, code2, multi_step_cvt,
7853 interm_types))
7854 {
7855 /* Elements in a vector with vect_used_by_reduction property cannot
7856 be reordered if the use chain with this property does not have the
7857 same operation. One such an example is s += a * b, where elements
7858 in a and b cannot be reordered. Here we check if the vector defined
7859 by STMT is only directly used in the reduction statement. */
7860 tree lhs = gimple_assign_lhs (stmt);
7861 use_operand_p dummy;
7862 gimple use_stmt;
7863 stmt_vec_info use_stmt_info = NULL;
7864 if (single_imm_use (lhs, &dummy, &use_stmt)
7865 && (use_stmt_info = vinfo_for_stmt (use_stmt))
7866 && STMT_VINFO_DEF_TYPE (use_stmt_info) == vect_reduction_def)
7867 return true;
7868 }
7869 c1 = VEC_WIDEN_MULT_LO_EXPR;
7870 c2 = VEC_WIDEN_MULT_HI_EXPR;
7871 break;
7872
7873 case VEC_WIDEN_MULT_EVEN_EXPR:
7874 /* Support the recursion induced just above. */
7875 c1 = VEC_WIDEN_MULT_EVEN_EXPR;
7876 c2 = VEC_WIDEN_MULT_ODD_EXPR;
7877 break;
7878
7879 case WIDEN_LSHIFT_EXPR:
7880 c1 = VEC_WIDEN_LSHIFT_LO_EXPR;
7881 c2 = VEC_WIDEN_LSHIFT_HI_EXPR;
7882 break;
7883
7884 CASE_CONVERT:
7885 c1 = VEC_UNPACK_LO_EXPR;
7886 c2 = VEC_UNPACK_HI_EXPR;
7887 break;
7888
7889 case FLOAT_EXPR:
7890 c1 = VEC_UNPACK_FLOAT_LO_EXPR;
7891 c2 = VEC_UNPACK_FLOAT_HI_EXPR;
7892 break;
7893
7894 case FIX_TRUNC_EXPR:
7895 /* ??? Not yet implemented due to missing VEC_UNPACK_FIX_TRUNC_HI_EXPR/
7896 VEC_UNPACK_FIX_TRUNC_LO_EXPR tree codes and optabs used for
7897 computing the operation. */
7898 return false;
7899
7900 default:
7901 gcc_unreachable ();
7902 }
7903
7904 if (BYTES_BIG_ENDIAN && c1 != VEC_WIDEN_MULT_EVEN_EXPR)
7905 {
7906 enum tree_code ctmp = c1;
7907 c1 = c2;
7908 c2 = ctmp;
7909 }
7910
7911 if (code == FIX_TRUNC_EXPR)
7912 {
7913 /* The signedness is determined from output operand. */
7914 optab1 = optab_for_tree_code (c1, vectype_out, optab_default);
7915 optab2 = optab_for_tree_code (c2, vectype_out, optab_default);
7916 }
7917 else
7918 {
7919 optab1 = optab_for_tree_code (c1, vectype, optab_default);
7920 optab2 = optab_for_tree_code (c2, vectype, optab_default);
7921 }
7922
7923 if (!optab1 || !optab2)
7924 return false;
7925
7926 vec_mode = TYPE_MODE (vectype);
7927 if ((icode1 = optab_handler (optab1, vec_mode)) == CODE_FOR_nothing
7928 || (icode2 = optab_handler (optab2, vec_mode)) == CODE_FOR_nothing)
7929 return false;
7930
7931 *code1 = c1;
7932 *code2 = c2;
7933
7934 if (insn_data[icode1].operand[0].mode == TYPE_MODE (wide_vectype)
7935 && insn_data[icode2].operand[0].mode == TYPE_MODE (wide_vectype))
7936 return true;
7937
7938 /* Check if it's a multi-step conversion that can be done using intermediate
7939 types. */
7940
7941 prev_type = vectype;
7942 prev_mode = vec_mode;
7943
7944 if (!CONVERT_EXPR_CODE_P (code))
7945 return false;
7946
7947 /* We assume here that there will not be more than MAX_INTERM_CVT_STEPS
7948 intermediate steps in promotion sequence. We try
7949 MAX_INTERM_CVT_STEPS to get to NARROW_VECTYPE, and fail if we do
7950 not. */
7951 interm_types->create (MAX_INTERM_CVT_STEPS);
7952 for (i = 0; i < MAX_INTERM_CVT_STEPS; i++)
7953 {
7954 intermediate_mode = insn_data[icode1].operand[0].mode;
7955 intermediate_type
7956 = lang_hooks.types.type_for_mode (intermediate_mode,
7957 TYPE_UNSIGNED (prev_type));
7958 optab3 = optab_for_tree_code (c1, intermediate_type, optab_default);
7959 optab4 = optab_for_tree_code (c2, intermediate_type, optab_default);
7960
7961 if (!optab3 || !optab4
7962 || (icode1 = optab_handler (optab1, prev_mode)) == CODE_FOR_nothing
7963 || insn_data[icode1].operand[0].mode != intermediate_mode
7964 || (icode2 = optab_handler (optab2, prev_mode)) == CODE_FOR_nothing
7965 || insn_data[icode2].operand[0].mode != intermediate_mode
7966 || ((icode1 = optab_handler (optab3, intermediate_mode))
7967 == CODE_FOR_nothing)
7968 || ((icode2 = optab_handler (optab4, intermediate_mode))
7969 == CODE_FOR_nothing))
7970 break;
7971
7972 interm_types->quick_push (intermediate_type);
7973 (*multi_step_cvt)++;
7974
7975 if (insn_data[icode1].operand[0].mode == TYPE_MODE (wide_vectype)
7976 && insn_data[icode2].operand[0].mode == TYPE_MODE (wide_vectype))
7977 return true;
7978
7979 prev_type = intermediate_type;
7980 prev_mode = intermediate_mode;
7981 }
7982
7983 interm_types->release ();
7984 return false;
7985 }
7986
7987
7988 /* Function supportable_narrowing_operation
7989
7990 Check whether an operation represented by the code CODE is a
7991 narrowing operation that is supported by the target platform in
7992 vector form (i.e., when operating on arguments of type VECTYPE_IN
7993 and producing a result of type VECTYPE_OUT).
7994
7995 Narrowing operations we currently support are NOP (CONVERT) and
7996 FIX_TRUNC. This function checks if these operations are supported by
7997 the target platform directly via vector tree-codes.
7998
7999 Output:
8000 - CODE1 is the code of a vector operation to be used when
8001 vectorizing the operation, if available.
8002 - MULTI_STEP_CVT determines the number of required intermediate steps in
8003 case of multi-step conversion (like int->short->char - in that case
8004 MULTI_STEP_CVT will be 1).
8005 - INTERM_TYPES contains the intermediate type required to perform the
8006 narrowing operation (short in the above example). */
8007
8008 bool
8009 supportable_narrowing_operation (enum tree_code code,
8010 tree vectype_out, tree vectype_in,
8011 enum tree_code *code1, int *multi_step_cvt,
8012 vec<tree> *interm_types)
8013 {
8014 enum machine_mode vec_mode;
8015 enum insn_code icode1;
8016 optab optab1, interm_optab;
8017 tree vectype = vectype_in;
8018 tree narrow_vectype = vectype_out;
8019 enum tree_code c1;
8020 tree intermediate_type;
8021 enum machine_mode intermediate_mode, prev_mode;
8022 int i;
8023 bool uns;
8024
8025 *multi_step_cvt = 0;
8026 switch (code)
8027 {
8028 CASE_CONVERT:
8029 c1 = VEC_PACK_TRUNC_EXPR;
8030 break;
8031
8032 case FIX_TRUNC_EXPR:
8033 c1 = VEC_PACK_FIX_TRUNC_EXPR;
8034 break;
8035
8036 case FLOAT_EXPR:
8037 /* ??? Not yet implemented due to missing VEC_PACK_FLOAT_EXPR
8038 tree code and optabs used for computing the operation. */
8039 return false;
8040
8041 default:
8042 gcc_unreachable ();
8043 }
8044
8045 if (code == FIX_TRUNC_EXPR)
8046 /* The signedness is determined from output operand. */
8047 optab1 = optab_for_tree_code (c1, vectype_out, optab_default);
8048 else
8049 optab1 = optab_for_tree_code (c1, vectype, optab_default);
8050
8051 if (!optab1)
8052 return false;
8053
8054 vec_mode = TYPE_MODE (vectype);
8055 if ((icode1 = optab_handler (optab1, vec_mode)) == CODE_FOR_nothing)
8056 return false;
8057
8058 *code1 = c1;
8059
8060 if (insn_data[icode1].operand[0].mode == TYPE_MODE (narrow_vectype))
8061 return true;
8062
8063 /* Check if it's a multi-step conversion that can be done using intermediate
8064 types. */
8065 prev_mode = vec_mode;
8066 if (code == FIX_TRUNC_EXPR)
8067 uns = TYPE_UNSIGNED (vectype_out);
8068 else
8069 uns = TYPE_UNSIGNED (vectype);
8070
8071 /* For multi-step FIX_TRUNC_EXPR prefer signed floating to integer
8072 conversion over unsigned, as unsigned FIX_TRUNC_EXPR is often more
8073 costly than signed. */
8074 if (code == FIX_TRUNC_EXPR && uns)
8075 {
8076 enum insn_code icode2;
8077
8078 intermediate_type
8079 = lang_hooks.types.type_for_mode (TYPE_MODE (vectype_out), 0);
8080 interm_optab
8081 = optab_for_tree_code (c1, intermediate_type, optab_default);
8082 if (interm_optab != unknown_optab
8083 && (icode2 = optab_handler (optab1, vec_mode)) != CODE_FOR_nothing
8084 && insn_data[icode1].operand[0].mode
8085 == insn_data[icode2].operand[0].mode)
8086 {
8087 uns = false;
8088 optab1 = interm_optab;
8089 icode1 = icode2;
8090 }
8091 }
8092
8093 /* We assume here that there will not be more than MAX_INTERM_CVT_STEPS
8094 intermediate steps in promotion sequence. We try
8095 MAX_INTERM_CVT_STEPS to get to NARROW_VECTYPE, and fail if we do not. */
8096 interm_types->create (MAX_INTERM_CVT_STEPS);
8097 for (i = 0; i < MAX_INTERM_CVT_STEPS; i++)
8098 {
8099 intermediate_mode = insn_data[icode1].operand[0].mode;
8100 intermediate_type
8101 = lang_hooks.types.type_for_mode (intermediate_mode, uns);
8102 interm_optab
8103 = optab_for_tree_code (VEC_PACK_TRUNC_EXPR, intermediate_type,
8104 optab_default);
8105 if (!interm_optab
8106 || ((icode1 = optab_handler (optab1, prev_mode)) == CODE_FOR_nothing)
8107 || insn_data[icode1].operand[0].mode != intermediate_mode
8108 || ((icode1 = optab_handler (interm_optab, intermediate_mode))
8109 == CODE_FOR_nothing))
8110 break;
8111
8112 interm_types->quick_push (intermediate_type);
8113 (*multi_step_cvt)++;
8114
8115 if (insn_data[icode1].operand[0].mode == TYPE_MODE (narrow_vectype))
8116 return true;
8117
8118 prev_mode = intermediate_mode;
8119 optab1 = interm_optab;
8120 }
8121
8122 interm_types->release ();
8123 return false;
8124 }