match.pd: Relax some tree_nop_conversion_p
[official-gcc.git] / gcc / tree-vect-stmts.c
blob3bcd0ce1946490105f1b7ea84e2d7da5b9843462
1 /* Statement Analysis and Transformation for Vectorization
2 Copyright (C) 2003-2016 Free Software Foundation, Inc.
3 Contributed by Dorit Naishlos <dorit@il.ibm.com>
4 and Ira Rosen <irar@il.ibm.com>
6 This file is part of GCC.
8 GCC is free software; you can redistribute it and/or modify it under
9 the terms of the GNU General Public License as published by the Free
10 Software Foundation; either version 3, or (at your option) any later
11 version.
13 GCC is distributed in the hope that it will be useful, but WITHOUT ANY
14 WARRANTY; without even the implied warranty of MERCHANTABILITY or
15 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
16 for more details.
18 You should have received a copy of the GNU General Public License
19 along with GCC; see the file COPYING3. If not see
20 <http://www.gnu.org/licenses/>. */
22 #include "config.h"
23 #include "system.h"
24 #include "coretypes.h"
25 #include "backend.h"
26 #include "target.h"
27 #include "rtl.h"
28 #include "tree.h"
29 #include "gimple.h"
30 #include "ssa.h"
31 #include "optabs-tree.h"
32 #include "insn-config.h"
33 #include "recog.h" /* FIXME: for insn_data */
34 #include "cgraph.h"
35 #include "dumpfile.h"
36 #include "alias.h"
37 #include "fold-const.h"
38 #include "stor-layout.h"
39 #include "tree-eh.h"
40 #include "gimplify.h"
41 #include "gimple-iterator.h"
42 #include "gimplify-me.h"
43 #include "tree-cfg.h"
44 #include "tree-ssa-loop-manip.h"
45 #include "cfgloop.h"
46 #include "tree-ssa-loop.h"
47 #include "tree-scalar-evolution.h"
48 #include "tree-vectorizer.h"
49 #include "builtins.h"
50 #include "internal-fn.h"
52 /* For lang_hooks.types.type_for_mode. */
53 #include "langhooks.h"
55 /* Return the vectorized type for the given statement. */
57 tree
58 stmt_vectype (struct _stmt_vec_info *stmt_info)
60 return STMT_VINFO_VECTYPE (stmt_info);
63 /* Return TRUE iff the given statement is in an inner loop relative to
64 the loop being vectorized. */
65 bool
66 stmt_in_inner_loop_p (struct _stmt_vec_info *stmt_info)
68 gimple *stmt = STMT_VINFO_STMT (stmt_info);
69 basic_block bb = gimple_bb (stmt);
70 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
71 struct loop* loop;
73 if (!loop_vinfo)
74 return false;
76 loop = LOOP_VINFO_LOOP (loop_vinfo);
78 return (bb->loop_father == loop->inner);
81 /* Record the cost of a statement, either by directly informing the
82 target model or by saving it in a vector for later processing.
83 Return a preliminary estimate of the statement's cost. */
85 unsigned
86 record_stmt_cost (stmt_vector_for_cost *body_cost_vec, int count,
87 enum vect_cost_for_stmt kind, stmt_vec_info stmt_info,
88 int misalign, enum vect_cost_model_location where)
90 if (body_cost_vec)
92 tree vectype = stmt_info ? stmt_vectype (stmt_info) : NULL_TREE;
93 stmt_info_for_cost si = { count, kind,
94 stmt_info ? STMT_VINFO_STMT (stmt_info) : NULL,
95 misalign };
96 body_cost_vec->safe_push (si);
97 return (unsigned)
98 (builtin_vectorization_cost (kind, vectype, misalign) * count);
100 else
101 return add_stmt_cost (stmt_info->vinfo->target_cost_data,
102 count, kind, stmt_info, misalign, where);
105 /* Return a variable of type ELEM_TYPE[NELEMS]. */
107 static tree
108 create_vector_array (tree elem_type, unsigned HOST_WIDE_INT nelems)
110 return create_tmp_var (build_array_type_nelts (elem_type, nelems),
111 "vect_array");
114 /* ARRAY is an array of vectors created by create_vector_array.
115 Return an SSA_NAME for the vector in index N. The reference
116 is part of the vectorization of STMT and the vector is associated
117 with scalar destination SCALAR_DEST. */
119 static tree
120 read_vector_array (gimple *stmt, gimple_stmt_iterator *gsi, tree scalar_dest,
121 tree array, unsigned HOST_WIDE_INT n)
123 tree vect_type, vect, vect_name, array_ref;
124 gimple *new_stmt;
126 gcc_assert (TREE_CODE (TREE_TYPE (array)) == ARRAY_TYPE);
127 vect_type = TREE_TYPE (TREE_TYPE (array));
128 vect = vect_create_destination_var (scalar_dest, vect_type);
129 array_ref = build4 (ARRAY_REF, vect_type, array,
130 build_int_cst (size_type_node, n),
131 NULL_TREE, NULL_TREE);
133 new_stmt = gimple_build_assign (vect, array_ref);
134 vect_name = make_ssa_name (vect, new_stmt);
135 gimple_assign_set_lhs (new_stmt, vect_name);
136 vect_finish_stmt_generation (stmt, new_stmt, gsi);
138 return vect_name;
141 /* ARRAY is an array of vectors created by create_vector_array.
142 Emit code to store SSA_NAME VECT in index N of the array.
143 The store is part of the vectorization of STMT. */
145 static void
146 write_vector_array (gimple *stmt, gimple_stmt_iterator *gsi, tree vect,
147 tree array, unsigned HOST_WIDE_INT n)
149 tree array_ref;
150 gimple *new_stmt;
152 array_ref = build4 (ARRAY_REF, TREE_TYPE (vect), array,
153 build_int_cst (size_type_node, n),
154 NULL_TREE, NULL_TREE);
156 new_stmt = gimple_build_assign (array_ref, vect);
157 vect_finish_stmt_generation (stmt, new_stmt, gsi);
160 /* PTR is a pointer to an array of type TYPE. Return a representation
161 of *PTR. The memory reference replaces those in FIRST_DR
162 (and its group). */
164 static tree
165 create_array_ref (tree type, tree ptr, struct data_reference *first_dr)
167 tree mem_ref, alias_ptr_type;
169 alias_ptr_type = reference_alias_ptr_type (DR_REF (first_dr));
170 mem_ref = build2 (MEM_REF, type, ptr, build_int_cst (alias_ptr_type, 0));
171 /* Arrays have the same alignment as their type. */
172 set_ptr_info_alignment (get_ptr_info (ptr), TYPE_ALIGN_UNIT (type), 0);
173 return mem_ref;
176 /* Utility functions used by vect_mark_stmts_to_be_vectorized. */
178 /* Function vect_mark_relevant.
180 Mark STMT as "relevant for vectorization" and add it to WORKLIST. */
182 static void
183 vect_mark_relevant (vec<gimple *> *worklist, gimple *stmt,
184 enum vect_relevant relevant, bool live_p)
186 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
187 enum vect_relevant save_relevant = STMT_VINFO_RELEVANT (stmt_info);
188 bool save_live_p = STMT_VINFO_LIVE_P (stmt_info);
189 gimple *pattern_stmt;
191 if (dump_enabled_p ())
193 dump_printf_loc (MSG_NOTE, vect_location,
194 "mark relevant %d, live %d: ", relevant, live_p);
195 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
198 /* If this stmt is an original stmt in a pattern, we might need to mark its
199 related pattern stmt instead of the original stmt. However, such stmts
200 may have their own uses that are not in any pattern, in such cases the
201 stmt itself should be marked. */
202 if (STMT_VINFO_IN_PATTERN_P (stmt_info))
204 /* This is the last stmt in a sequence that was detected as a
205 pattern that can potentially be vectorized. Don't mark the stmt
206 as relevant/live because it's not going to be vectorized.
207 Instead mark the pattern-stmt that replaces it. */
209 pattern_stmt = STMT_VINFO_RELATED_STMT (stmt_info);
211 if (dump_enabled_p ())
212 dump_printf_loc (MSG_NOTE, vect_location,
213 "last stmt in pattern. don't mark"
214 " relevant/live.\n");
215 stmt_info = vinfo_for_stmt (pattern_stmt);
216 gcc_assert (STMT_VINFO_RELATED_STMT (stmt_info) == stmt);
217 save_relevant = STMT_VINFO_RELEVANT (stmt_info);
218 save_live_p = STMT_VINFO_LIVE_P (stmt_info);
219 stmt = pattern_stmt;
222 STMT_VINFO_LIVE_P (stmt_info) |= live_p;
223 if (relevant > STMT_VINFO_RELEVANT (stmt_info))
224 STMT_VINFO_RELEVANT (stmt_info) = relevant;
226 if (STMT_VINFO_RELEVANT (stmt_info) == save_relevant
227 && STMT_VINFO_LIVE_P (stmt_info) == save_live_p)
229 if (dump_enabled_p ())
230 dump_printf_loc (MSG_NOTE, vect_location,
231 "already marked relevant/live.\n");
232 return;
235 worklist->safe_push (stmt);
239 /* Function vect_stmt_relevant_p.
241 Return true if STMT in loop that is represented by LOOP_VINFO is
242 "relevant for vectorization".
244 A stmt is considered "relevant for vectorization" if:
245 - it has uses outside the loop.
246 - it has vdefs (it alters memory).
247 - control stmts in the loop (except for the exit condition).
249 CHECKME: what other side effects would the vectorizer allow? */
251 static bool
252 vect_stmt_relevant_p (gimple *stmt, loop_vec_info loop_vinfo,
253 enum vect_relevant *relevant, bool *live_p)
255 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
256 ssa_op_iter op_iter;
257 imm_use_iterator imm_iter;
258 use_operand_p use_p;
259 def_operand_p def_p;
261 *relevant = vect_unused_in_scope;
262 *live_p = false;
264 /* cond stmt other than loop exit cond. */
265 if (is_ctrl_stmt (stmt)
266 && STMT_VINFO_TYPE (vinfo_for_stmt (stmt))
267 != loop_exit_ctrl_vec_info_type)
268 *relevant = vect_used_in_scope;
270 /* changing memory. */
271 if (gimple_code (stmt) != GIMPLE_PHI)
272 if (gimple_vdef (stmt)
273 && !gimple_clobber_p (stmt))
275 if (dump_enabled_p ())
276 dump_printf_loc (MSG_NOTE, vect_location,
277 "vec_stmt_relevant_p: stmt has vdefs.\n");
278 *relevant = vect_used_in_scope;
281 /* uses outside the loop. */
282 FOR_EACH_PHI_OR_STMT_DEF (def_p, stmt, op_iter, SSA_OP_DEF)
284 FOR_EACH_IMM_USE_FAST (use_p, imm_iter, DEF_FROM_PTR (def_p))
286 basic_block bb = gimple_bb (USE_STMT (use_p));
287 if (!flow_bb_inside_loop_p (loop, bb))
289 if (dump_enabled_p ())
290 dump_printf_loc (MSG_NOTE, vect_location,
291 "vec_stmt_relevant_p: used out of loop.\n");
293 if (is_gimple_debug (USE_STMT (use_p)))
294 continue;
296 /* We expect all such uses to be in the loop exit phis
297 (because of loop closed form) */
298 gcc_assert (gimple_code (USE_STMT (use_p)) == GIMPLE_PHI);
299 gcc_assert (bb == single_exit (loop)->dest);
301 *live_p = true;
306 return (*live_p || *relevant);
310 /* Function exist_non_indexing_operands_for_use_p
312 USE is one of the uses attached to STMT. Check if USE is
313 used in STMT for anything other than indexing an array. */
315 static bool
316 exist_non_indexing_operands_for_use_p (tree use, gimple *stmt)
318 tree operand;
319 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
321 /* USE corresponds to some operand in STMT. If there is no data
322 reference in STMT, then any operand that corresponds to USE
323 is not indexing an array. */
324 if (!STMT_VINFO_DATA_REF (stmt_info))
325 return true;
327 /* STMT has a data_ref. FORNOW this means that its of one of
328 the following forms:
329 -1- ARRAY_REF = var
330 -2- var = ARRAY_REF
331 (This should have been verified in analyze_data_refs).
333 'var' in the second case corresponds to a def, not a use,
334 so USE cannot correspond to any operands that are not used
335 for array indexing.
337 Therefore, all we need to check is if STMT falls into the
338 first case, and whether var corresponds to USE. */
340 if (!gimple_assign_copy_p (stmt))
342 if (is_gimple_call (stmt)
343 && gimple_call_internal_p (stmt))
344 switch (gimple_call_internal_fn (stmt))
346 case IFN_MASK_STORE:
347 operand = gimple_call_arg (stmt, 3);
348 if (operand == use)
349 return true;
350 /* FALLTHRU */
351 case IFN_MASK_LOAD:
352 operand = gimple_call_arg (stmt, 2);
353 if (operand == use)
354 return true;
355 break;
356 default:
357 break;
359 return false;
362 if (TREE_CODE (gimple_assign_lhs (stmt)) == SSA_NAME)
363 return false;
364 operand = gimple_assign_rhs1 (stmt);
365 if (TREE_CODE (operand) != SSA_NAME)
366 return false;
368 if (operand == use)
369 return true;
371 return false;
376 Function process_use.
378 Inputs:
379 - a USE in STMT in a loop represented by LOOP_VINFO
380 - LIVE_P, RELEVANT - enum values to be set in the STMT_VINFO of the stmt
381 that defined USE. This is done by calling mark_relevant and passing it
382 the WORKLIST (to add DEF_STMT to the WORKLIST in case it is relevant).
383 - FORCE is true if exist_non_indexing_operands_for_use_p check shouldn't
384 be performed.
386 Outputs:
387 Generally, LIVE_P and RELEVANT are used to define the liveness and
388 relevance info of the DEF_STMT of this USE:
389 STMT_VINFO_LIVE_P (DEF_STMT_info) <-- live_p
390 STMT_VINFO_RELEVANT (DEF_STMT_info) <-- relevant
391 Exceptions:
392 - case 1: If USE is used only for address computations (e.g. array indexing),
393 which does not need to be directly vectorized, then the liveness/relevance
394 of the respective DEF_STMT is left unchanged.
395 - case 2: If STMT is a reduction phi and DEF_STMT is a reduction stmt, we
396 skip DEF_STMT cause it had already been processed.
397 - case 3: If DEF_STMT and STMT are in different nests, then "relevant" will
398 be modified accordingly.
400 Return true if everything is as expected. Return false otherwise. */
402 static bool
403 process_use (gimple *stmt, tree use, loop_vec_info loop_vinfo, bool live_p,
404 enum vect_relevant relevant, vec<gimple *> *worklist,
405 bool force)
407 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
408 stmt_vec_info stmt_vinfo = vinfo_for_stmt (stmt);
409 stmt_vec_info dstmt_vinfo;
410 basic_block bb, def_bb;
411 gimple *def_stmt;
412 enum vect_def_type dt;
414 /* case 1: we are only interested in uses that need to be vectorized. Uses
415 that are used for address computation are not considered relevant. */
416 if (!force && !exist_non_indexing_operands_for_use_p (use, stmt))
417 return true;
419 if (!vect_is_simple_use (use, loop_vinfo, &def_stmt, &dt))
421 if (dump_enabled_p ())
422 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
423 "not vectorized: unsupported use in stmt.\n");
424 return false;
427 if (!def_stmt || gimple_nop_p (def_stmt))
428 return true;
430 def_bb = gimple_bb (def_stmt);
431 if (!flow_bb_inside_loop_p (loop, def_bb))
433 if (dump_enabled_p ())
434 dump_printf_loc (MSG_NOTE, vect_location, "def_stmt is out of loop.\n");
435 return true;
438 /* case 2: A reduction phi (STMT) defined by a reduction stmt (DEF_STMT).
439 DEF_STMT must have already been processed, because this should be the
440 only way that STMT, which is a reduction-phi, was put in the worklist,
441 as there should be no other uses for DEF_STMT in the loop. So we just
442 check that everything is as expected, and we are done. */
443 dstmt_vinfo = vinfo_for_stmt (def_stmt);
444 bb = gimple_bb (stmt);
445 if (gimple_code (stmt) == GIMPLE_PHI
446 && STMT_VINFO_DEF_TYPE (stmt_vinfo) == vect_reduction_def
447 && gimple_code (def_stmt) != GIMPLE_PHI
448 && STMT_VINFO_DEF_TYPE (dstmt_vinfo) == vect_reduction_def
449 && bb->loop_father == def_bb->loop_father)
451 if (dump_enabled_p ())
452 dump_printf_loc (MSG_NOTE, vect_location,
453 "reduc-stmt defining reduc-phi in the same nest.\n");
454 if (STMT_VINFO_IN_PATTERN_P (dstmt_vinfo))
455 dstmt_vinfo = vinfo_for_stmt (STMT_VINFO_RELATED_STMT (dstmt_vinfo));
456 gcc_assert (STMT_VINFO_RELEVANT (dstmt_vinfo) < vect_used_by_reduction);
457 gcc_assert (STMT_VINFO_LIVE_P (dstmt_vinfo)
458 || STMT_VINFO_RELEVANT (dstmt_vinfo) > vect_unused_in_scope);
459 return true;
462 /* case 3a: outer-loop stmt defining an inner-loop stmt:
463 outer-loop-header-bb:
464 d = def_stmt
465 inner-loop:
466 stmt # use (d)
467 outer-loop-tail-bb:
468 ... */
469 if (flow_loop_nested_p (def_bb->loop_father, bb->loop_father))
471 if (dump_enabled_p ())
472 dump_printf_loc (MSG_NOTE, vect_location,
473 "outer-loop def-stmt defining inner-loop stmt.\n");
475 switch (relevant)
477 case vect_unused_in_scope:
478 relevant = (STMT_VINFO_DEF_TYPE (stmt_vinfo) == vect_nested_cycle) ?
479 vect_used_in_scope : vect_unused_in_scope;
480 break;
482 case vect_used_in_outer_by_reduction:
483 gcc_assert (STMT_VINFO_DEF_TYPE (stmt_vinfo) != vect_reduction_def);
484 relevant = vect_used_by_reduction;
485 break;
487 case vect_used_in_outer:
488 gcc_assert (STMT_VINFO_DEF_TYPE (stmt_vinfo) != vect_reduction_def);
489 relevant = vect_used_in_scope;
490 break;
492 case vect_used_in_scope:
493 break;
495 default:
496 gcc_unreachable ();
500 /* case 3b: inner-loop stmt defining an outer-loop stmt:
501 outer-loop-header-bb:
503 inner-loop:
504 d = def_stmt
505 outer-loop-tail-bb (or outer-loop-exit-bb in double reduction):
506 stmt # use (d) */
507 else if (flow_loop_nested_p (bb->loop_father, def_bb->loop_father))
509 if (dump_enabled_p ())
510 dump_printf_loc (MSG_NOTE, vect_location,
511 "inner-loop def-stmt defining outer-loop stmt.\n");
513 switch (relevant)
515 case vect_unused_in_scope:
516 relevant = (STMT_VINFO_DEF_TYPE (stmt_vinfo) == vect_reduction_def
517 || STMT_VINFO_DEF_TYPE (stmt_vinfo) == vect_double_reduction_def) ?
518 vect_used_in_outer_by_reduction : vect_unused_in_scope;
519 break;
521 case vect_used_by_reduction:
522 relevant = vect_used_in_outer_by_reduction;
523 break;
525 case vect_used_in_scope:
526 relevant = vect_used_in_outer;
527 break;
529 default:
530 gcc_unreachable ();
534 vect_mark_relevant (worklist, def_stmt, relevant, live_p);
535 return true;
539 /* Function vect_mark_stmts_to_be_vectorized.
541 Not all stmts in the loop need to be vectorized. For example:
543 for i...
544 for j...
545 1. T0 = i + j
546 2. T1 = a[T0]
548 3. j = j + 1
550 Stmt 1 and 3 do not need to be vectorized, because loop control and
551 addressing of vectorized data-refs are handled differently.
553 This pass detects such stmts. */
555 bool
556 vect_mark_stmts_to_be_vectorized (loop_vec_info loop_vinfo)
558 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
559 basic_block *bbs = LOOP_VINFO_BBS (loop_vinfo);
560 unsigned int nbbs = loop->num_nodes;
561 gimple_stmt_iterator si;
562 gimple *stmt;
563 unsigned int i;
564 stmt_vec_info stmt_vinfo;
565 basic_block bb;
566 gimple *phi;
567 bool live_p;
568 enum vect_relevant relevant, tmp_relevant;
569 enum vect_def_type def_type;
571 if (dump_enabled_p ())
572 dump_printf_loc (MSG_NOTE, vect_location,
573 "=== vect_mark_stmts_to_be_vectorized ===\n");
575 auto_vec<gimple *, 64> worklist;
577 /* 1. Init worklist. */
578 for (i = 0; i < nbbs; i++)
580 bb = bbs[i];
581 for (si = gsi_start_phis (bb); !gsi_end_p (si); gsi_next (&si))
583 phi = gsi_stmt (si);
584 if (dump_enabled_p ())
586 dump_printf_loc (MSG_NOTE, vect_location, "init: phi relevant? ");
587 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, phi, 0);
590 if (vect_stmt_relevant_p (phi, loop_vinfo, &relevant, &live_p))
591 vect_mark_relevant (&worklist, phi, relevant, live_p);
593 for (si = gsi_start_bb (bb); !gsi_end_p (si); gsi_next (&si))
595 stmt = gsi_stmt (si);
596 if (dump_enabled_p ())
598 dump_printf_loc (MSG_NOTE, vect_location, "init: stmt relevant? ");
599 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
602 if (vect_stmt_relevant_p (stmt, loop_vinfo, &relevant, &live_p))
603 vect_mark_relevant (&worklist, stmt, relevant, live_p);
607 /* 2. Process_worklist */
608 while (worklist.length () > 0)
610 use_operand_p use_p;
611 ssa_op_iter iter;
613 stmt = worklist.pop ();
614 if (dump_enabled_p ())
616 dump_printf_loc (MSG_NOTE, vect_location, "worklist: examine stmt: ");
617 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
620 /* Examine the USEs of STMT. For each USE, mark the stmt that defines it
621 (DEF_STMT) as relevant/irrelevant and live/dead according to the
622 liveness and relevance properties of STMT. */
623 stmt_vinfo = vinfo_for_stmt (stmt);
624 relevant = STMT_VINFO_RELEVANT (stmt_vinfo);
625 live_p = STMT_VINFO_LIVE_P (stmt_vinfo);
627 /* Generally, the liveness and relevance properties of STMT are
628 propagated as is to the DEF_STMTs of its USEs:
629 live_p <-- STMT_VINFO_LIVE_P (STMT_VINFO)
630 relevant <-- STMT_VINFO_RELEVANT (STMT_VINFO)
632 One exception is when STMT has been identified as defining a reduction
633 variable; in this case we set the liveness/relevance as follows:
634 live_p = false
635 relevant = vect_used_by_reduction
636 This is because we distinguish between two kinds of relevant stmts -
637 those that are used by a reduction computation, and those that are
638 (also) used by a regular computation. This allows us later on to
639 identify stmts that are used solely by a reduction, and therefore the
640 order of the results that they produce does not have to be kept. */
642 def_type = STMT_VINFO_DEF_TYPE (stmt_vinfo);
643 tmp_relevant = relevant;
644 switch (def_type)
646 case vect_reduction_def:
647 switch (tmp_relevant)
649 case vect_unused_in_scope:
650 relevant = vect_used_by_reduction;
651 break;
653 case vect_used_by_reduction:
654 if (gimple_code (stmt) == GIMPLE_PHI)
655 break;
656 /* fall through */
658 default:
659 if (dump_enabled_p ())
660 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
661 "unsupported use of reduction.\n");
662 return false;
665 live_p = false;
666 break;
668 case vect_nested_cycle:
669 if (tmp_relevant != vect_unused_in_scope
670 && tmp_relevant != vect_used_in_outer_by_reduction
671 && tmp_relevant != vect_used_in_outer)
673 if (dump_enabled_p ())
674 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
675 "unsupported use of nested cycle.\n");
677 return false;
680 live_p = false;
681 break;
683 case vect_double_reduction_def:
684 if (tmp_relevant != vect_unused_in_scope
685 && tmp_relevant != vect_used_by_reduction)
687 if (dump_enabled_p ())
688 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
689 "unsupported use of double reduction.\n");
691 return false;
694 live_p = false;
695 break;
697 default:
698 break;
701 if (is_pattern_stmt_p (stmt_vinfo))
703 /* Pattern statements are not inserted into the code, so
704 FOR_EACH_PHI_OR_STMT_USE optimizes their operands out, and we
705 have to scan the RHS or function arguments instead. */
706 if (is_gimple_assign (stmt))
708 enum tree_code rhs_code = gimple_assign_rhs_code (stmt);
709 tree op = gimple_assign_rhs1 (stmt);
711 i = 1;
712 if (rhs_code == COND_EXPR && COMPARISON_CLASS_P (op))
714 if (!process_use (stmt, TREE_OPERAND (op, 0), loop_vinfo,
715 live_p, relevant, &worklist, false)
716 || !process_use (stmt, TREE_OPERAND (op, 1), loop_vinfo,
717 live_p, relevant, &worklist, false))
718 return false;
719 i = 2;
721 for (; i < gimple_num_ops (stmt); i++)
723 op = gimple_op (stmt, i);
724 if (TREE_CODE (op) == SSA_NAME
725 && !process_use (stmt, op, loop_vinfo, live_p, relevant,
726 &worklist, false))
727 return false;
730 else if (is_gimple_call (stmt))
732 for (i = 0; i < gimple_call_num_args (stmt); i++)
734 tree arg = gimple_call_arg (stmt, i);
735 if (!process_use (stmt, arg, loop_vinfo, live_p, relevant,
736 &worklist, false))
737 return false;
741 else
742 FOR_EACH_PHI_OR_STMT_USE (use_p, stmt, iter, SSA_OP_USE)
744 tree op = USE_FROM_PTR (use_p);
745 if (!process_use (stmt, op, loop_vinfo, live_p, relevant,
746 &worklist, false))
747 return false;
750 if (STMT_VINFO_GATHER_SCATTER_P (stmt_vinfo))
752 tree off;
753 tree decl = vect_check_gather_scatter (stmt, loop_vinfo, NULL, &off, NULL);
754 gcc_assert (decl);
755 if (!process_use (stmt, off, loop_vinfo, live_p, relevant,
756 &worklist, true))
757 return false;
759 } /* while worklist */
761 return true;
765 /* Function vect_model_simple_cost.
767 Models cost for simple operations, i.e. those that only emit ncopies of a
768 single op. Right now, this does not account for multiple insns that could
769 be generated for the single vector op. We will handle that shortly. */
771 void
772 vect_model_simple_cost (stmt_vec_info stmt_info, int ncopies,
773 enum vect_def_type *dt,
774 stmt_vector_for_cost *prologue_cost_vec,
775 stmt_vector_for_cost *body_cost_vec)
777 int i;
778 int inside_cost = 0, prologue_cost = 0;
780 /* The SLP costs were already calculated during SLP tree build. */
781 if (PURE_SLP_STMT (stmt_info))
782 return;
784 /* FORNOW: Assuming maximum 2 args per stmts. */
785 for (i = 0; i < 2; i++)
786 if (dt[i] == vect_constant_def || dt[i] == vect_external_def)
787 prologue_cost += record_stmt_cost (prologue_cost_vec, 1, vector_stmt,
788 stmt_info, 0, vect_prologue);
790 /* Pass the inside-of-loop statements to the target-specific cost model. */
791 inside_cost = record_stmt_cost (body_cost_vec, ncopies, vector_stmt,
792 stmt_info, 0, vect_body);
794 if (dump_enabled_p ())
795 dump_printf_loc (MSG_NOTE, vect_location,
796 "vect_model_simple_cost: inside_cost = %d, "
797 "prologue_cost = %d .\n", inside_cost, prologue_cost);
801 /* Model cost for type demotion and promotion operations. PWR is normally
802 zero for single-step promotions and demotions. It will be one if
803 two-step promotion/demotion is required, and so on. Each additional
804 step doubles the number of instructions required. */
806 static void
807 vect_model_promotion_demotion_cost (stmt_vec_info stmt_info,
808 enum vect_def_type *dt, int pwr)
810 int i, tmp;
811 int inside_cost = 0, prologue_cost = 0;
812 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
813 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
814 void *target_cost_data;
816 /* The SLP costs were already calculated during SLP tree build. */
817 if (PURE_SLP_STMT (stmt_info))
818 return;
820 if (loop_vinfo)
821 target_cost_data = LOOP_VINFO_TARGET_COST_DATA (loop_vinfo);
822 else
823 target_cost_data = BB_VINFO_TARGET_COST_DATA (bb_vinfo);
825 for (i = 0; i < pwr + 1; i++)
827 tmp = (STMT_VINFO_TYPE (stmt_info) == type_promotion_vec_info_type) ?
828 (i + 1) : i;
829 inside_cost += add_stmt_cost (target_cost_data, vect_pow2 (tmp),
830 vec_promote_demote, stmt_info, 0,
831 vect_body);
834 /* FORNOW: Assuming maximum 2 args per stmts. */
835 for (i = 0; i < 2; i++)
836 if (dt[i] == vect_constant_def || dt[i] == vect_external_def)
837 prologue_cost += add_stmt_cost (target_cost_data, 1, vector_stmt,
838 stmt_info, 0, vect_prologue);
840 if (dump_enabled_p ())
841 dump_printf_loc (MSG_NOTE, vect_location,
842 "vect_model_promotion_demotion_cost: inside_cost = %d, "
843 "prologue_cost = %d .\n", inside_cost, prologue_cost);
846 /* Function vect_cost_group_size
848 For grouped load or store, return the group_size only if it is the first
849 load or store of a group, else return 1. This ensures that group size is
850 only returned once per group. */
852 static int
853 vect_cost_group_size (stmt_vec_info stmt_info)
855 gimple *first_stmt = GROUP_FIRST_ELEMENT (stmt_info);
857 if (first_stmt == STMT_VINFO_STMT (stmt_info))
858 return GROUP_SIZE (stmt_info);
860 return 1;
864 /* Function vect_model_store_cost
866 Models cost for stores. In the case of grouped accesses, one access
867 has the overhead of the grouped access attributed to it. */
869 void
870 vect_model_store_cost (stmt_vec_info stmt_info, int ncopies,
871 bool store_lanes_p, enum vect_def_type dt,
872 slp_tree slp_node,
873 stmt_vector_for_cost *prologue_cost_vec,
874 stmt_vector_for_cost *body_cost_vec)
876 int group_size;
877 unsigned int inside_cost = 0, prologue_cost = 0;
878 struct data_reference *first_dr;
879 gimple *first_stmt;
881 if (dt == vect_constant_def || dt == vect_external_def)
882 prologue_cost += record_stmt_cost (prologue_cost_vec, 1, scalar_to_vec,
883 stmt_info, 0, vect_prologue);
885 /* Grouped access? */
886 if (STMT_VINFO_GROUPED_ACCESS (stmt_info))
888 if (slp_node)
890 first_stmt = SLP_TREE_SCALAR_STMTS (slp_node)[0];
891 group_size = 1;
893 else
895 first_stmt = GROUP_FIRST_ELEMENT (stmt_info);
896 group_size = vect_cost_group_size (stmt_info);
899 first_dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (first_stmt));
901 /* Not a grouped access. */
902 else
904 group_size = 1;
905 first_dr = STMT_VINFO_DATA_REF (stmt_info);
908 /* We assume that the cost of a single store-lanes instruction is
909 equivalent to the cost of GROUP_SIZE separate stores. If a grouped
910 access is instead being provided by a permute-and-store operation,
911 include the cost of the permutes. */
912 if (!store_lanes_p && group_size > 1
913 && !STMT_VINFO_STRIDED_P (stmt_info))
915 /* Uses a high and low interleave or shuffle operations for each
916 needed permute. */
917 int nstmts = ncopies * ceil_log2 (group_size) * group_size;
918 inside_cost = record_stmt_cost (body_cost_vec, nstmts, vec_perm,
919 stmt_info, 0, vect_body);
921 if (dump_enabled_p ())
922 dump_printf_loc (MSG_NOTE, vect_location,
923 "vect_model_store_cost: strided group_size = %d .\n",
924 group_size);
927 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
928 /* Costs of the stores. */
929 if (STMT_VINFO_STRIDED_P (stmt_info)
930 && !STMT_VINFO_GROUPED_ACCESS (stmt_info))
932 /* N scalar stores plus extracting the elements. */
933 inside_cost += record_stmt_cost (body_cost_vec,
934 ncopies * TYPE_VECTOR_SUBPARTS (vectype),
935 scalar_store, stmt_info, 0, vect_body);
937 else
938 vect_get_store_cost (first_dr, ncopies, &inside_cost, body_cost_vec);
940 if (STMT_VINFO_STRIDED_P (stmt_info))
941 inside_cost += record_stmt_cost (body_cost_vec,
942 ncopies * TYPE_VECTOR_SUBPARTS (vectype),
943 vec_to_scalar, stmt_info, 0, vect_body);
945 if (dump_enabled_p ())
946 dump_printf_loc (MSG_NOTE, vect_location,
947 "vect_model_store_cost: inside_cost = %d, "
948 "prologue_cost = %d .\n", inside_cost, prologue_cost);
952 /* Calculate cost of DR's memory access. */
953 void
954 vect_get_store_cost (struct data_reference *dr, int ncopies,
955 unsigned int *inside_cost,
956 stmt_vector_for_cost *body_cost_vec)
958 int alignment_support_scheme = vect_supportable_dr_alignment (dr, false);
959 gimple *stmt = DR_STMT (dr);
960 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
962 switch (alignment_support_scheme)
964 case dr_aligned:
966 *inside_cost += record_stmt_cost (body_cost_vec, ncopies,
967 vector_store, stmt_info, 0,
968 vect_body);
970 if (dump_enabled_p ())
971 dump_printf_loc (MSG_NOTE, vect_location,
972 "vect_model_store_cost: aligned.\n");
973 break;
976 case dr_unaligned_supported:
978 /* Here, we assign an additional cost for the unaligned store. */
979 *inside_cost += record_stmt_cost (body_cost_vec, ncopies,
980 unaligned_store, stmt_info,
981 DR_MISALIGNMENT (dr), vect_body);
982 if (dump_enabled_p ())
983 dump_printf_loc (MSG_NOTE, vect_location,
984 "vect_model_store_cost: unaligned supported by "
985 "hardware.\n");
986 break;
989 case dr_unaligned_unsupported:
991 *inside_cost = VECT_MAX_COST;
993 if (dump_enabled_p ())
994 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
995 "vect_model_store_cost: unsupported access.\n");
996 break;
999 default:
1000 gcc_unreachable ();
1005 /* Function vect_model_load_cost
1007 Models cost for loads. In the case of grouped accesses, the last access
1008 has the overhead of the grouped access attributed to it. Since unaligned
1009 accesses are supported for loads, we also account for the costs of the
1010 access scheme chosen. */
1012 void
1013 vect_model_load_cost (stmt_vec_info stmt_info, int ncopies,
1014 bool load_lanes_p, slp_tree slp_node,
1015 stmt_vector_for_cost *prologue_cost_vec,
1016 stmt_vector_for_cost *body_cost_vec)
1018 int group_size;
1019 gimple *first_stmt;
1020 struct data_reference *dr = STMT_VINFO_DATA_REF (stmt_info), *first_dr;
1021 unsigned int inside_cost = 0, prologue_cost = 0;
1023 /* Grouped accesses? */
1024 first_stmt = GROUP_FIRST_ELEMENT (stmt_info);
1025 if (STMT_VINFO_GROUPED_ACCESS (stmt_info) && first_stmt && !slp_node)
1027 group_size = vect_cost_group_size (stmt_info);
1028 first_dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (first_stmt));
1030 /* Not a grouped access. */
1031 else
1033 group_size = 1;
1034 first_dr = dr;
1037 /* We assume that the cost of a single load-lanes instruction is
1038 equivalent to the cost of GROUP_SIZE separate loads. If a grouped
1039 access is instead being provided by a load-and-permute operation,
1040 include the cost of the permutes. */
1041 if (!load_lanes_p && group_size > 1
1042 && !STMT_VINFO_STRIDED_P (stmt_info))
1044 /* Uses an even and odd extract operations or shuffle operations
1045 for each needed permute. */
1046 int nstmts = ncopies * ceil_log2 (group_size) * group_size;
1047 inside_cost = record_stmt_cost (body_cost_vec, nstmts, vec_perm,
1048 stmt_info, 0, vect_body);
1050 if (dump_enabled_p ())
1051 dump_printf_loc (MSG_NOTE, vect_location,
1052 "vect_model_load_cost: strided group_size = %d .\n",
1053 group_size);
1056 /* The loads themselves. */
1057 if (STMT_VINFO_STRIDED_P (stmt_info)
1058 && !STMT_VINFO_GROUPED_ACCESS (stmt_info))
1060 /* N scalar loads plus gathering them into a vector. */
1061 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
1062 inside_cost += record_stmt_cost (body_cost_vec,
1063 ncopies * TYPE_VECTOR_SUBPARTS (vectype),
1064 scalar_load, stmt_info, 0, vect_body);
1066 else
1067 vect_get_load_cost (first_dr, ncopies,
1068 ((!STMT_VINFO_GROUPED_ACCESS (stmt_info))
1069 || group_size > 1 || slp_node),
1070 &inside_cost, &prologue_cost,
1071 prologue_cost_vec, body_cost_vec, true);
1072 if (STMT_VINFO_STRIDED_P (stmt_info))
1073 inside_cost += record_stmt_cost (body_cost_vec, ncopies, vec_construct,
1074 stmt_info, 0, vect_body);
1076 if (dump_enabled_p ())
1077 dump_printf_loc (MSG_NOTE, vect_location,
1078 "vect_model_load_cost: inside_cost = %d, "
1079 "prologue_cost = %d .\n", inside_cost, prologue_cost);
1083 /* Calculate cost of DR's memory access. */
1084 void
1085 vect_get_load_cost (struct data_reference *dr, int ncopies,
1086 bool add_realign_cost, unsigned int *inside_cost,
1087 unsigned int *prologue_cost,
1088 stmt_vector_for_cost *prologue_cost_vec,
1089 stmt_vector_for_cost *body_cost_vec,
1090 bool record_prologue_costs)
1092 int alignment_support_scheme = vect_supportable_dr_alignment (dr, false);
1093 gimple *stmt = DR_STMT (dr);
1094 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
1096 switch (alignment_support_scheme)
1098 case dr_aligned:
1100 *inside_cost += record_stmt_cost (body_cost_vec, ncopies, vector_load,
1101 stmt_info, 0, vect_body);
1103 if (dump_enabled_p ())
1104 dump_printf_loc (MSG_NOTE, vect_location,
1105 "vect_model_load_cost: aligned.\n");
1107 break;
1109 case dr_unaligned_supported:
1111 /* Here, we assign an additional cost for the unaligned load. */
1112 *inside_cost += record_stmt_cost (body_cost_vec, ncopies,
1113 unaligned_load, stmt_info,
1114 DR_MISALIGNMENT (dr), vect_body);
1116 if (dump_enabled_p ())
1117 dump_printf_loc (MSG_NOTE, vect_location,
1118 "vect_model_load_cost: unaligned supported by "
1119 "hardware.\n");
1121 break;
1123 case dr_explicit_realign:
1125 *inside_cost += record_stmt_cost (body_cost_vec, ncopies * 2,
1126 vector_load, stmt_info, 0, vect_body);
1127 *inside_cost += record_stmt_cost (body_cost_vec, ncopies,
1128 vec_perm, stmt_info, 0, vect_body);
1130 /* FIXME: If the misalignment remains fixed across the iterations of
1131 the containing loop, the following cost should be added to the
1132 prologue costs. */
1133 if (targetm.vectorize.builtin_mask_for_load)
1134 *inside_cost += record_stmt_cost (body_cost_vec, 1, vector_stmt,
1135 stmt_info, 0, vect_body);
1137 if (dump_enabled_p ())
1138 dump_printf_loc (MSG_NOTE, vect_location,
1139 "vect_model_load_cost: explicit realign\n");
1141 break;
1143 case dr_explicit_realign_optimized:
1145 if (dump_enabled_p ())
1146 dump_printf_loc (MSG_NOTE, vect_location,
1147 "vect_model_load_cost: unaligned software "
1148 "pipelined.\n");
1150 /* Unaligned software pipeline has a load of an address, an initial
1151 load, and possibly a mask operation to "prime" the loop. However,
1152 if this is an access in a group of loads, which provide grouped
1153 access, then the above cost should only be considered for one
1154 access in the group. Inside the loop, there is a load op
1155 and a realignment op. */
1157 if (add_realign_cost && record_prologue_costs)
1159 *prologue_cost += record_stmt_cost (prologue_cost_vec, 2,
1160 vector_stmt, stmt_info,
1161 0, vect_prologue);
1162 if (targetm.vectorize.builtin_mask_for_load)
1163 *prologue_cost += record_stmt_cost (prologue_cost_vec, 1,
1164 vector_stmt, stmt_info,
1165 0, vect_prologue);
1168 *inside_cost += record_stmt_cost (body_cost_vec, ncopies, vector_load,
1169 stmt_info, 0, vect_body);
1170 *inside_cost += record_stmt_cost (body_cost_vec, ncopies, vec_perm,
1171 stmt_info, 0, vect_body);
1173 if (dump_enabled_p ())
1174 dump_printf_loc (MSG_NOTE, vect_location,
1175 "vect_model_load_cost: explicit realign optimized"
1176 "\n");
1178 break;
1181 case dr_unaligned_unsupported:
1183 *inside_cost = VECT_MAX_COST;
1185 if (dump_enabled_p ())
1186 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
1187 "vect_model_load_cost: unsupported access.\n");
1188 break;
1191 default:
1192 gcc_unreachable ();
1196 /* Insert the new stmt NEW_STMT at *GSI or at the appropriate place in
1197 the loop preheader for the vectorized stmt STMT. */
1199 static void
1200 vect_init_vector_1 (gimple *stmt, gimple *new_stmt, gimple_stmt_iterator *gsi)
1202 if (gsi)
1203 vect_finish_stmt_generation (stmt, new_stmt, gsi);
1204 else
1206 stmt_vec_info stmt_vinfo = vinfo_for_stmt (stmt);
1207 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_vinfo);
1209 if (loop_vinfo)
1211 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
1212 basic_block new_bb;
1213 edge pe;
1215 if (nested_in_vect_loop_p (loop, stmt))
1216 loop = loop->inner;
1218 pe = loop_preheader_edge (loop);
1219 new_bb = gsi_insert_on_edge_immediate (pe, new_stmt);
1220 gcc_assert (!new_bb);
1222 else
1224 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_vinfo);
1225 basic_block bb;
1226 gimple_stmt_iterator gsi_bb_start;
1228 gcc_assert (bb_vinfo);
1229 bb = BB_VINFO_BB (bb_vinfo);
1230 gsi_bb_start = gsi_after_labels (bb);
1231 gsi_insert_before (&gsi_bb_start, new_stmt, GSI_SAME_STMT);
1235 if (dump_enabled_p ())
1237 dump_printf_loc (MSG_NOTE, vect_location,
1238 "created new init_stmt: ");
1239 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, new_stmt, 0);
1243 /* Function vect_init_vector.
1245 Insert a new stmt (INIT_STMT) that initializes a new variable of type
1246 TYPE with the value VAL. If TYPE is a vector type and VAL does not have
1247 vector type a vector with all elements equal to VAL is created first.
1248 Place the initialization at BSI if it is not NULL. Otherwise, place the
1249 initialization at the loop preheader.
1250 Return the DEF of INIT_STMT.
1251 It will be used in the vectorization of STMT. */
1253 tree
1254 vect_init_vector (gimple *stmt, tree val, tree type, gimple_stmt_iterator *gsi)
1256 gimple *init_stmt;
1257 tree new_temp;
1259 if (TREE_CODE (type) == VECTOR_TYPE
1260 && TREE_CODE (TREE_TYPE (val)) != VECTOR_TYPE)
1262 if (!types_compatible_p (TREE_TYPE (type), TREE_TYPE (val)))
1264 /* Scalar boolean value should be transformed into
1265 all zeros or all ones value before building a vector. */
1266 if (VECTOR_BOOLEAN_TYPE_P (type))
1268 tree true_val = build_all_ones_cst (TREE_TYPE (type));
1269 tree false_val = build_zero_cst (TREE_TYPE (type));
1271 if (CONSTANT_CLASS_P (val))
1272 val = integer_zerop (val) ? false_val : true_val;
1273 else
1275 new_temp = make_ssa_name (TREE_TYPE (type));
1276 init_stmt = gimple_build_assign (new_temp, COND_EXPR,
1277 val, true_val, false_val);
1278 vect_init_vector_1 (stmt, init_stmt, gsi);
1279 val = new_temp;
1282 else if (CONSTANT_CLASS_P (val))
1283 val = fold_convert (TREE_TYPE (type), val);
1284 else
1286 new_temp = make_ssa_name (TREE_TYPE (type));
1287 init_stmt = gimple_build_assign (new_temp, NOP_EXPR, val);
1288 vect_init_vector_1 (stmt, init_stmt, gsi);
1289 val = new_temp;
1292 val = build_vector_from_val (type, val);
1295 new_temp = vect_get_new_ssa_name (type, vect_simple_var, "cst_");
1296 init_stmt = gimple_build_assign (new_temp, val);
1297 vect_init_vector_1 (stmt, init_stmt, gsi);
1298 return new_temp;
1302 /* Function vect_get_vec_def_for_operand.
1304 OP is an operand in STMT. This function returns a (vector) def that will be
1305 used in the vectorized stmt for STMT.
1307 In the case that OP is an SSA_NAME which is defined in the loop, then
1308 STMT_VINFO_VEC_STMT of the defining stmt holds the relevant def.
1310 In case OP is an invariant or constant, a new stmt that creates a vector def
1311 needs to be introduced. VECTYPE may be used to specify a required type for
1312 vector invariant. */
1314 tree
1315 vect_get_vec_def_for_operand (tree op, gimple *stmt, tree vectype)
1317 tree vec_oprnd;
1318 gimple *vec_stmt;
1319 gimple *def_stmt;
1320 stmt_vec_info def_stmt_info = NULL;
1321 stmt_vec_info stmt_vinfo = vinfo_for_stmt (stmt);
1322 tree stmt_vectype = STMT_VINFO_VECTYPE (stmt_vinfo);
1323 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_vinfo);
1324 enum vect_def_type dt;
1325 bool is_simple_use;
1326 tree vector_type;
1328 if (dump_enabled_p ())
1330 dump_printf_loc (MSG_NOTE, vect_location,
1331 "vect_get_vec_def_for_operand: ");
1332 dump_generic_expr (MSG_NOTE, TDF_SLIM, op);
1333 dump_printf (MSG_NOTE, "\n");
1336 is_simple_use = vect_is_simple_use (op, loop_vinfo, &def_stmt, &dt);
1337 gcc_assert (is_simple_use);
1338 if (dump_enabled_p ())
1340 int loc_printed = 0;
1341 if (def_stmt)
1343 if (loc_printed)
1344 dump_printf (MSG_NOTE, " def_stmt = ");
1345 else
1346 dump_printf_loc (MSG_NOTE, vect_location, " def_stmt = ");
1347 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, def_stmt, 0);
1351 switch (dt)
1353 /* operand is a constant or a loop invariant. */
1354 case vect_constant_def:
1355 case vect_external_def:
1357 if (vectype)
1358 vector_type = vectype;
1359 else if (TREE_CODE (TREE_TYPE (op)) == BOOLEAN_TYPE
1360 && VECTOR_BOOLEAN_TYPE_P (stmt_vectype))
1361 vector_type = build_same_sized_truth_vector_type (stmt_vectype);
1362 else
1363 vector_type = get_vectype_for_scalar_type (TREE_TYPE (op));
1365 gcc_assert (vector_type);
1366 return vect_init_vector (stmt, op, vector_type, NULL);
1369 /* operand is defined inside the loop. */
1370 case vect_internal_def:
1372 /* Get the def from the vectorized stmt. */
1373 def_stmt_info = vinfo_for_stmt (def_stmt);
1375 vec_stmt = STMT_VINFO_VEC_STMT (def_stmt_info);
1376 /* Get vectorized pattern statement. */
1377 if (!vec_stmt
1378 && STMT_VINFO_IN_PATTERN_P (def_stmt_info)
1379 && !STMT_VINFO_RELEVANT (def_stmt_info))
1380 vec_stmt = STMT_VINFO_VEC_STMT (vinfo_for_stmt (
1381 STMT_VINFO_RELATED_STMT (def_stmt_info)));
1382 gcc_assert (vec_stmt);
1383 if (gimple_code (vec_stmt) == GIMPLE_PHI)
1384 vec_oprnd = PHI_RESULT (vec_stmt);
1385 else if (is_gimple_call (vec_stmt))
1386 vec_oprnd = gimple_call_lhs (vec_stmt);
1387 else
1388 vec_oprnd = gimple_assign_lhs (vec_stmt);
1389 return vec_oprnd;
1392 /* operand is defined by a loop header phi - reduction */
1393 case vect_reduction_def:
1394 case vect_double_reduction_def:
1395 case vect_nested_cycle:
1396 /* Code should use get_initial_def_for_reduction. */
1397 gcc_unreachable ();
1399 /* operand is defined by loop-header phi - induction. */
1400 case vect_induction_def:
1402 gcc_assert (gimple_code (def_stmt) == GIMPLE_PHI);
1404 /* Get the def from the vectorized stmt. */
1405 def_stmt_info = vinfo_for_stmt (def_stmt);
1406 vec_stmt = STMT_VINFO_VEC_STMT (def_stmt_info);
1407 if (gimple_code (vec_stmt) == GIMPLE_PHI)
1408 vec_oprnd = PHI_RESULT (vec_stmt);
1409 else
1410 vec_oprnd = gimple_get_lhs (vec_stmt);
1411 return vec_oprnd;
1414 default:
1415 gcc_unreachable ();
1420 /* Function vect_get_vec_def_for_stmt_copy
1422 Return a vector-def for an operand. This function is used when the
1423 vectorized stmt to be created (by the caller to this function) is a "copy"
1424 created in case the vectorized result cannot fit in one vector, and several
1425 copies of the vector-stmt are required. In this case the vector-def is
1426 retrieved from the vector stmt recorded in the STMT_VINFO_RELATED_STMT field
1427 of the stmt that defines VEC_OPRND.
1428 DT is the type of the vector def VEC_OPRND.
1430 Context:
1431 In case the vectorization factor (VF) is bigger than the number
1432 of elements that can fit in a vectype (nunits), we have to generate
1433 more than one vector stmt to vectorize the scalar stmt. This situation
1434 arises when there are multiple data-types operated upon in the loop; the
1435 smallest data-type determines the VF, and as a result, when vectorizing
1436 stmts operating on wider types we need to create 'VF/nunits' "copies" of the
1437 vector stmt (each computing a vector of 'nunits' results, and together
1438 computing 'VF' results in each iteration). This function is called when
1439 vectorizing such a stmt (e.g. vectorizing S2 in the illustration below, in
1440 which VF=16 and nunits=4, so the number of copies required is 4):
1442 scalar stmt: vectorized into: STMT_VINFO_RELATED_STMT
1444 S1: x = load VS1.0: vx.0 = memref0 VS1.1
1445 VS1.1: vx.1 = memref1 VS1.2
1446 VS1.2: vx.2 = memref2 VS1.3
1447 VS1.3: vx.3 = memref3
1449 S2: z = x + ... VSnew.0: vz0 = vx.0 + ... VSnew.1
1450 VSnew.1: vz1 = vx.1 + ... VSnew.2
1451 VSnew.2: vz2 = vx.2 + ... VSnew.3
1452 VSnew.3: vz3 = vx.3 + ...
1454 The vectorization of S1 is explained in vectorizable_load.
1455 The vectorization of S2:
1456 To create the first vector-stmt out of the 4 copies - VSnew.0 -
1457 the function 'vect_get_vec_def_for_operand' is called to
1458 get the relevant vector-def for each operand of S2. For operand x it
1459 returns the vector-def 'vx.0'.
1461 To create the remaining copies of the vector-stmt (VSnew.j), this
1462 function is called to get the relevant vector-def for each operand. It is
1463 obtained from the respective VS1.j stmt, which is recorded in the
1464 STMT_VINFO_RELATED_STMT field of the stmt that defines VEC_OPRND.
1466 For example, to obtain the vector-def 'vx.1' in order to create the
1467 vector stmt 'VSnew.1', this function is called with VEC_OPRND='vx.0'.
1468 Given 'vx0' we obtain the stmt that defines it ('VS1.0'); from the
1469 STMT_VINFO_RELATED_STMT field of 'VS1.0' we obtain the next copy - 'VS1.1',
1470 and return its def ('vx.1').
1471 Overall, to create the above sequence this function will be called 3 times:
1472 vx.1 = vect_get_vec_def_for_stmt_copy (dt, vx.0);
1473 vx.2 = vect_get_vec_def_for_stmt_copy (dt, vx.1);
1474 vx.3 = vect_get_vec_def_for_stmt_copy (dt, vx.2); */
1476 tree
1477 vect_get_vec_def_for_stmt_copy (enum vect_def_type dt, tree vec_oprnd)
1479 gimple *vec_stmt_for_operand;
1480 stmt_vec_info def_stmt_info;
1482 /* Do nothing; can reuse same def. */
1483 if (dt == vect_external_def || dt == vect_constant_def )
1484 return vec_oprnd;
1486 vec_stmt_for_operand = SSA_NAME_DEF_STMT (vec_oprnd);
1487 def_stmt_info = vinfo_for_stmt (vec_stmt_for_operand);
1488 gcc_assert (def_stmt_info);
1489 vec_stmt_for_operand = STMT_VINFO_RELATED_STMT (def_stmt_info);
1490 gcc_assert (vec_stmt_for_operand);
1491 if (gimple_code (vec_stmt_for_operand) == GIMPLE_PHI)
1492 vec_oprnd = PHI_RESULT (vec_stmt_for_operand);
1493 else
1494 vec_oprnd = gimple_get_lhs (vec_stmt_for_operand);
1495 return vec_oprnd;
1499 /* Get vectorized definitions for the operands to create a copy of an original
1500 stmt. See vect_get_vec_def_for_stmt_copy () for details. */
1502 static void
1503 vect_get_vec_defs_for_stmt_copy (enum vect_def_type *dt,
1504 vec<tree> *vec_oprnds0,
1505 vec<tree> *vec_oprnds1)
1507 tree vec_oprnd = vec_oprnds0->pop ();
1509 vec_oprnd = vect_get_vec_def_for_stmt_copy (dt[0], vec_oprnd);
1510 vec_oprnds0->quick_push (vec_oprnd);
1512 if (vec_oprnds1 && vec_oprnds1->length ())
1514 vec_oprnd = vec_oprnds1->pop ();
1515 vec_oprnd = vect_get_vec_def_for_stmt_copy (dt[1], vec_oprnd);
1516 vec_oprnds1->quick_push (vec_oprnd);
1521 /* Get vectorized definitions for OP0 and OP1.
1522 REDUC_INDEX is the index of reduction operand in case of reduction,
1523 and -1 otherwise. */
1525 void
1526 vect_get_vec_defs (tree op0, tree op1, gimple *stmt,
1527 vec<tree> *vec_oprnds0,
1528 vec<tree> *vec_oprnds1,
1529 slp_tree slp_node, int reduc_index)
1531 if (slp_node)
1533 int nops = (op1 == NULL_TREE) ? 1 : 2;
1534 auto_vec<tree> ops (nops);
1535 auto_vec<vec<tree> > vec_defs (nops);
1537 ops.quick_push (op0);
1538 if (op1)
1539 ops.quick_push (op1);
1541 vect_get_slp_defs (ops, slp_node, &vec_defs, reduc_index);
1543 *vec_oprnds0 = vec_defs[0];
1544 if (op1)
1545 *vec_oprnds1 = vec_defs[1];
1547 else
1549 tree vec_oprnd;
1551 vec_oprnds0->create (1);
1552 vec_oprnd = vect_get_vec_def_for_operand (op0, stmt);
1553 vec_oprnds0->quick_push (vec_oprnd);
1555 if (op1)
1557 vec_oprnds1->create (1);
1558 vec_oprnd = vect_get_vec_def_for_operand (op1, stmt);
1559 vec_oprnds1->quick_push (vec_oprnd);
1565 /* Function vect_finish_stmt_generation.
1567 Insert a new stmt. */
1569 void
1570 vect_finish_stmt_generation (gimple *stmt, gimple *vec_stmt,
1571 gimple_stmt_iterator *gsi)
1573 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
1574 vec_info *vinfo = stmt_info->vinfo;
1576 gcc_assert (gimple_code (stmt) != GIMPLE_LABEL);
1578 if (!gsi_end_p (*gsi)
1579 && gimple_has_mem_ops (vec_stmt))
1581 gimple *at_stmt = gsi_stmt (*gsi);
1582 tree vuse = gimple_vuse (at_stmt);
1583 if (vuse && TREE_CODE (vuse) == SSA_NAME)
1585 tree vdef = gimple_vdef (at_stmt);
1586 gimple_set_vuse (vec_stmt, gimple_vuse (at_stmt));
1587 /* If we have an SSA vuse and insert a store, update virtual
1588 SSA form to avoid triggering the renamer. Do so only
1589 if we can easily see all uses - which is what almost always
1590 happens with the way vectorized stmts are inserted. */
1591 if ((vdef && TREE_CODE (vdef) == SSA_NAME)
1592 && ((is_gimple_assign (vec_stmt)
1593 && !is_gimple_reg (gimple_assign_lhs (vec_stmt)))
1594 || (is_gimple_call (vec_stmt)
1595 && !(gimple_call_flags (vec_stmt)
1596 & (ECF_CONST|ECF_PURE|ECF_NOVOPS)))))
1598 tree new_vdef = copy_ssa_name (vuse, vec_stmt);
1599 gimple_set_vdef (vec_stmt, new_vdef);
1600 SET_USE (gimple_vuse_op (at_stmt), new_vdef);
1604 gsi_insert_before (gsi, vec_stmt, GSI_SAME_STMT);
1606 set_vinfo_for_stmt (vec_stmt, new_stmt_vec_info (vec_stmt, vinfo));
1608 if (dump_enabled_p ())
1610 dump_printf_loc (MSG_NOTE, vect_location, "add new stmt: ");
1611 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, vec_stmt, 0);
1614 gimple_set_location (vec_stmt, gimple_location (stmt));
1616 /* While EH edges will generally prevent vectorization, stmt might
1617 e.g. be in a must-not-throw region. Ensure newly created stmts
1618 that could throw are part of the same region. */
1619 int lp_nr = lookup_stmt_eh_lp (stmt);
1620 if (lp_nr != 0 && stmt_could_throw_p (vec_stmt))
1621 add_stmt_to_eh_lp (vec_stmt, lp_nr);
1624 /* We want to vectorize a call to combined function CFN with function
1625 decl FNDECL, using VECTYPE_OUT as the type of the output and VECTYPE_IN
1626 as the types of all inputs. Check whether this is possible using
1627 an internal function, returning its code if so or IFN_LAST if not. */
1629 static internal_fn
1630 vectorizable_internal_function (combined_fn cfn, tree fndecl,
1631 tree vectype_out, tree vectype_in)
1633 internal_fn ifn;
1634 if (internal_fn_p (cfn))
1635 ifn = as_internal_fn (cfn);
1636 else
1637 ifn = associated_internal_fn (fndecl);
1638 if (ifn != IFN_LAST && direct_internal_fn_p (ifn))
1640 const direct_internal_fn_info &info = direct_internal_fn (ifn);
1641 if (info.vectorizable)
1643 tree type0 = (info.type0 < 0 ? vectype_out : vectype_in);
1644 tree type1 = (info.type1 < 0 ? vectype_out : vectype_in);
1645 if (direct_internal_fn_supported_p (ifn, tree_pair (type0, type1),
1646 OPTIMIZE_FOR_SPEED))
1647 return ifn;
1650 return IFN_LAST;
1654 static tree permute_vec_elements (tree, tree, tree, gimple *,
1655 gimple_stmt_iterator *);
1658 /* Function vectorizable_mask_load_store.
1660 Check if STMT performs a conditional load or store that can be vectorized.
1661 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
1662 stmt to replace it, put it in VEC_STMT, and insert it at GSI.
1663 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
1665 static bool
1666 vectorizable_mask_load_store (gimple *stmt, gimple_stmt_iterator *gsi,
1667 gimple **vec_stmt, slp_tree slp_node)
1669 tree vec_dest = NULL;
1670 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
1671 stmt_vec_info prev_stmt_info;
1672 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
1673 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
1674 bool nested_in_vect_loop = nested_in_vect_loop_p (loop, stmt);
1675 struct data_reference *dr = STMT_VINFO_DATA_REF (stmt_info);
1676 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
1677 tree rhs_vectype = NULL_TREE;
1678 tree mask_vectype;
1679 tree elem_type;
1680 gimple *new_stmt;
1681 tree dummy;
1682 tree dataref_ptr = NULL_TREE;
1683 gimple *ptr_incr;
1684 int nunits = TYPE_VECTOR_SUBPARTS (vectype);
1685 int ncopies;
1686 int i, j;
1687 bool inv_p;
1688 tree gather_base = NULL_TREE, gather_off = NULL_TREE;
1689 tree gather_off_vectype = NULL_TREE, gather_decl = NULL_TREE;
1690 int gather_scale = 1;
1691 enum vect_def_type gather_dt = vect_unknown_def_type;
1692 bool is_store;
1693 tree mask;
1694 gimple *def_stmt;
1695 enum vect_def_type dt;
1697 if (slp_node != NULL)
1698 return false;
1700 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits;
1701 gcc_assert (ncopies >= 1);
1703 is_store = gimple_call_internal_fn (stmt) == IFN_MASK_STORE;
1704 mask = gimple_call_arg (stmt, 2);
1706 if (TREE_CODE (TREE_TYPE (mask)) != BOOLEAN_TYPE)
1707 return false;
1709 /* FORNOW. This restriction should be relaxed. */
1710 if (nested_in_vect_loop && ncopies > 1)
1712 if (dump_enabled_p ())
1713 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
1714 "multiple types in nested loop.");
1715 return false;
1718 if (!STMT_VINFO_RELEVANT_P (stmt_info))
1719 return false;
1721 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def
1722 && ! vec_stmt)
1723 return false;
1725 if (!STMT_VINFO_DATA_REF (stmt_info))
1726 return false;
1728 elem_type = TREE_TYPE (vectype);
1730 if (STMT_VINFO_GROUPED_ACCESS (stmt_info))
1731 return false;
1733 if (STMT_VINFO_STRIDED_P (stmt_info))
1734 return false;
1736 if (TREE_CODE (mask) != SSA_NAME)
1737 return false;
1739 if (!vect_is_simple_use (mask, loop_vinfo, &def_stmt, &dt, &mask_vectype))
1740 return false;
1742 if (!mask_vectype)
1743 mask_vectype = get_mask_type_for_scalar_type (TREE_TYPE (vectype));
1745 if (!mask_vectype || !VECTOR_BOOLEAN_TYPE_P (mask_vectype)
1746 || TYPE_VECTOR_SUBPARTS (mask_vectype) != TYPE_VECTOR_SUBPARTS (vectype))
1747 return false;
1749 if (is_store)
1751 tree rhs = gimple_call_arg (stmt, 3);
1752 if (!vect_is_simple_use (rhs, loop_vinfo, &def_stmt, &dt, &rhs_vectype))
1753 return false;
1756 if (STMT_VINFO_GATHER_SCATTER_P (stmt_info))
1758 gimple *def_stmt;
1759 gather_decl = vect_check_gather_scatter (stmt, loop_vinfo, &gather_base,
1760 &gather_off, &gather_scale);
1761 gcc_assert (gather_decl);
1762 if (!vect_is_simple_use (gather_off, loop_vinfo, &def_stmt, &gather_dt,
1763 &gather_off_vectype))
1765 if (dump_enabled_p ())
1766 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
1767 "gather index use not simple.");
1768 return false;
1771 tree arglist = TYPE_ARG_TYPES (TREE_TYPE (gather_decl));
1772 tree masktype
1773 = TREE_VALUE (TREE_CHAIN (TREE_CHAIN (TREE_CHAIN (arglist))));
1774 if (TREE_CODE (masktype) == INTEGER_TYPE)
1776 if (dump_enabled_p ())
1777 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
1778 "masked gather with integer mask not supported.");
1779 return false;
1782 else if (tree_int_cst_compare (nested_in_vect_loop
1783 ? STMT_VINFO_DR_STEP (stmt_info)
1784 : DR_STEP (dr), size_zero_node) <= 0)
1785 return false;
1786 else if (!VECTOR_MODE_P (TYPE_MODE (vectype))
1787 || !can_vec_mask_load_store_p (TYPE_MODE (vectype),
1788 TYPE_MODE (mask_vectype),
1789 !is_store)
1790 || (rhs_vectype
1791 && !useless_type_conversion_p (vectype, rhs_vectype)))
1792 return false;
1794 if (!vec_stmt) /* transformation not required. */
1796 STMT_VINFO_TYPE (stmt_info) = call_vec_info_type;
1797 if (is_store)
1798 vect_model_store_cost (stmt_info, ncopies, false, dt,
1799 NULL, NULL, NULL);
1800 else
1801 vect_model_load_cost (stmt_info, ncopies, false, NULL, NULL, NULL);
1802 return true;
1805 /** Transform. **/
1807 if (STMT_VINFO_GATHER_SCATTER_P (stmt_info))
1809 tree vec_oprnd0 = NULL_TREE, op;
1810 tree arglist = TYPE_ARG_TYPES (TREE_TYPE (gather_decl));
1811 tree rettype, srctype, ptrtype, idxtype, masktype, scaletype;
1812 tree ptr, vec_mask = NULL_TREE, mask_op = NULL_TREE, var, scale;
1813 tree perm_mask = NULL_TREE, prev_res = NULL_TREE;
1814 tree mask_perm_mask = NULL_TREE;
1815 edge pe = loop_preheader_edge (loop);
1816 gimple_seq seq;
1817 basic_block new_bb;
1818 enum { NARROW, NONE, WIDEN } modifier;
1819 int gather_off_nunits = TYPE_VECTOR_SUBPARTS (gather_off_vectype);
1821 rettype = TREE_TYPE (TREE_TYPE (gather_decl));
1822 srctype = TREE_VALUE (arglist); arglist = TREE_CHAIN (arglist);
1823 ptrtype = TREE_VALUE (arglist); arglist = TREE_CHAIN (arglist);
1824 idxtype = TREE_VALUE (arglist); arglist = TREE_CHAIN (arglist);
1825 masktype = TREE_VALUE (arglist); arglist = TREE_CHAIN (arglist);
1826 scaletype = TREE_VALUE (arglist);
1827 gcc_checking_assert (types_compatible_p (srctype, rettype)
1828 && types_compatible_p (srctype, masktype));
1830 if (nunits == gather_off_nunits)
1831 modifier = NONE;
1832 else if (nunits == gather_off_nunits / 2)
1834 unsigned char *sel = XALLOCAVEC (unsigned char, gather_off_nunits);
1835 modifier = WIDEN;
1837 for (i = 0; i < gather_off_nunits; ++i)
1838 sel[i] = i | nunits;
1840 perm_mask = vect_gen_perm_mask_checked (gather_off_vectype, sel);
1842 else if (nunits == gather_off_nunits * 2)
1844 unsigned char *sel = XALLOCAVEC (unsigned char, nunits);
1845 modifier = NARROW;
1847 for (i = 0; i < nunits; ++i)
1848 sel[i] = i < gather_off_nunits
1849 ? i : i + nunits - gather_off_nunits;
1851 perm_mask = vect_gen_perm_mask_checked (vectype, sel);
1852 ncopies *= 2;
1853 for (i = 0; i < nunits; ++i)
1854 sel[i] = i | gather_off_nunits;
1855 mask_perm_mask = vect_gen_perm_mask_checked (masktype, sel);
1857 else
1858 gcc_unreachable ();
1860 vec_dest = vect_create_destination_var (gimple_call_lhs (stmt), vectype);
1862 ptr = fold_convert (ptrtype, gather_base);
1863 if (!is_gimple_min_invariant (ptr))
1865 ptr = force_gimple_operand (ptr, &seq, true, NULL_TREE);
1866 new_bb = gsi_insert_seq_on_edge_immediate (pe, seq);
1867 gcc_assert (!new_bb);
1870 scale = build_int_cst (scaletype, gather_scale);
1872 prev_stmt_info = NULL;
1873 for (j = 0; j < ncopies; ++j)
1875 if (modifier == WIDEN && (j & 1))
1876 op = permute_vec_elements (vec_oprnd0, vec_oprnd0,
1877 perm_mask, stmt, gsi);
1878 else if (j == 0)
1879 op = vec_oprnd0
1880 = vect_get_vec_def_for_operand (gather_off, stmt);
1881 else
1882 op = vec_oprnd0
1883 = vect_get_vec_def_for_stmt_copy (gather_dt, vec_oprnd0);
1885 if (!useless_type_conversion_p (idxtype, TREE_TYPE (op)))
1887 gcc_assert (TYPE_VECTOR_SUBPARTS (TREE_TYPE (op))
1888 == TYPE_VECTOR_SUBPARTS (idxtype));
1889 var = vect_get_new_ssa_name (idxtype, vect_simple_var);
1890 op = build1 (VIEW_CONVERT_EXPR, idxtype, op);
1891 new_stmt
1892 = gimple_build_assign (var, VIEW_CONVERT_EXPR, op);
1893 vect_finish_stmt_generation (stmt, new_stmt, gsi);
1894 op = var;
1897 if (mask_perm_mask && (j & 1))
1898 mask_op = permute_vec_elements (mask_op, mask_op,
1899 mask_perm_mask, stmt, gsi);
1900 else
1902 if (j == 0)
1903 vec_mask = vect_get_vec_def_for_operand (mask, stmt);
1904 else
1906 vect_is_simple_use (vec_mask, loop_vinfo, &def_stmt, &dt);
1907 vec_mask = vect_get_vec_def_for_stmt_copy (dt, vec_mask);
1910 mask_op = vec_mask;
1911 if (!useless_type_conversion_p (masktype, TREE_TYPE (vec_mask)))
1913 gcc_assert (TYPE_VECTOR_SUBPARTS (TREE_TYPE (mask_op))
1914 == TYPE_VECTOR_SUBPARTS (masktype));
1915 var = vect_get_new_ssa_name (masktype, vect_simple_var);
1916 mask_op = build1 (VIEW_CONVERT_EXPR, masktype, mask_op);
1917 new_stmt
1918 = gimple_build_assign (var, VIEW_CONVERT_EXPR, mask_op);
1919 vect_finish_stmt_generation (stmt, new_stmt, gsi);
1920 mask_op = var;
1924 new_stmt
1925 = gimple_build_call (gather_decl, 5, mask_op, ptr, op, mask_op,
1926 scale);
1928 if (!useless_type_conversion_p (vectype, rettype))
1930 gcc_assert (TYPE_VECTOR_SUBPARTS (vectype)
1931 == TYPE_VECTOR_SUBPARTS (rettype));
1932 op = vect_get_new_ssa_name (rettype, vect_simple_var);
1933 gimple_call_set_lhs (new_stmt, op);
1934 vect_finish_stmt_generation (stmt, new_stmt, gsi);
1935 var = make_ssa_name (vec_dest);
1936 op = build1 (VIEW_CONVERT_EXPR, vectype, op);
1937 new_stmt = gimple_build_assign (var, VIEW_CONVERT_EXPR, op);
1939 else
1941 var = make_ssa_name (vec_dest, new_stmt);
1942 gimple_call_set_lhs (new_stmt, var);
1945 vect_finish_stmt_generation (stmt, new_stmt, gsi);
1947 if (modifier == NARROW)
1949 if ((j & 1) == 0)
1951 prev_res = var;
1952 continue;
1954 var = permute_vec_elements (prev_res, var,
1955 perm_mask, stmt, gsi);
1956 new_stmt = SSA_NAME_DEF_STMT (var);
1959 if (prev_stmt_info == NULL)
1960 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
1961 else
1962 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
1963 prev_stmt_info = vinfo_for_stmt (new_stmt);
1966 /* Ensure that even with -fno-tree-dce the scalar MASK_LOAD is removed
1967 from the IL. */
1968 if (STMT_VINFO_RELATED_STMT (stmt_info))
1970 stmt = STMT_VINFO_RELATED_STMT (stmt_info);
1971 stmt_info = vinfo_for_stmt (stmt);
1973 tree lhs = gimple_call_lhs (stmt);
1974 new_stmt = gimple_build_assign (lhs, build_zero_cst (TREE_TYPE (lhs)));
1975 set_vinfo_for_stmt (new_stmt, stmt_info);
1976 set_vinfo_for_stmt (stmt, NULL);
1977 STMT_VINFO_STMT (stmt_info) = new_stmt;
1978 gsi_replace (gsi, new_stmt, true);
1979 return true;
1981 else if (is_store)
1983 tree vec_rhs = NULL_TREE, vec_mask = NULL_TREE;
1984 prev_stmt_info = NULL;
1985 LOOP_VINFO_HAS_MASK_STORE (loop_vinfo) = true;
1986 for (i = 0; i < ncopies; i++)
1988 unsigned align, misalign;
1990 if (i == 0)
1992 tree rhs = gimple_call_arg (stmt, 3);
1993 vec_rhs = vect_get_vec_def_for_operand (rhs, stmt);
1994 vec_mask = vect_get_vec_def_for_operand (mask, stmt);
1995 /* We should have catched mismatched types earlier. */
1996 gcc_assert (useless_type_conversion_p (vectype,
1997 TREE_TYPE (vec_rhs)));
1998 dataref_ptr = vect_create_data_ref_ptr (stmt, vectype, NULL,
1999 NULL_TREE, &dummy, gsi,
2000 &ptr_incr, false, &inv_p);
2001 gcc_assert (!inv_p);
2003 else
2005 vect_is_simple_use (vec_rhs, loop_vinfo, &def_stmt, &dt);
2006 vec_rhs = vect_get_vec_def_for_stmt_copy (dt, vec_rhs);
2007 vect_is_simple_use (vec_mask, loop_vinfo, &def_stmt, &dt);
2008 vec_mask = vect_get_vec_def_for_stmt_copy (dt, vec_mask);
2009 dataref_ptr = bump_vector_ptr (dataref_ptr, ptr_incr, gsi, stmt,
2010 TYPE_SIZE_UNIT (vectype));
2013 align = TYPE_ALIGN_UNIT (vectype);
2014 if (aligned_access_p (dr))
2015 misalign = 0;
2016 else if (DR_MISALIGNMENT (dr) == -1)
2018 align = TYPE_ALIGN_UNIT (elem_type);
2019 misalign = 0;
2021 else
2022 misalign = DR_MISALIGNMENT (dr);
2023 set_ptr_info_alignment (get_ptr_info (dataref_ptr), align,
2024 misalign);
2025 tree ptr = build_int_cst (TREE_TYPE (gimple_call_arg (stmt, 1)),
2026 misalign ? misalign & -misalign : align);
2027 new_stmt
2028 = gimple_build_call_internal (IFN_MASK_STORE, 4, dataref_ptr,
2029 ptr, vec_mask, vec_rhs);
2030 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2031 if (i == 0)
2032 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
2033 else
2034 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
2035 prev_stmt_info = vinfo_for_stmt (new_stmt);
2038 else
2040 tree vec_mask = NULL_TREE;
2041 prev_stmt_info = NULL;
2042 vec_dest = vect_create_destination_var (gimple_call_lhs (stmt), vectype);
2043 for (i = 0; i < ncopies; i++)
2045 unsigned align, misalign;
2047 if (i == 0)
2049 vec_mask = vect_get_vec_def_for_operand (mask, stmt);
2050 dataref_ptr = vect_create_data_ref_ptr (stmt, vectype, NULL,
2051 NULL_TREE, &dummy, gsi,
2052 &ptr_incr, false, &inv_p);
2053 gcc_assert (!inv_p);
2055 else
2057 vect_is_simple_use (vec_mask, loop_vinfo, &def_stmt, &dt);
2058 vec_mask = vect_get_vec_def_for_stmt_copy (dt, vec_mask);
2059 dataref_ptr = bump_vector_ptr (dataref_ptr, ptr_incr, gsi, stmt,
2060 TYPE_SIZE_UNIT (vectype));
2063 align = TYPE_ALIGN_UNIT (vectype);
2064 if (aligned_access_p (dr))
2065 misalign = 0;
2066 else if (DR_MISALIGNMENT (dr) == -1)
2068 align = TYPE_ALIGN_UNIT (elem_type);
2069 misalign = 0;
2071 else
2072 misalign = DR_MISALIGNMENT (dr);
2073 set_ptr_info_alignment (get_ptr_info (dataref_ptr), align,
2074 misalign);
2075 tree ptr = build_int_cst (TREE_TYPE (gimple_call_arg (stmt, 1)),
2076 misalign ? misalign & -misalign : align);
2077 new_stmt
2078 = gimple_build_call_internal (IFN_MASK_LOAD, 3, dataref_ptr,
2079 ptr, vec_mask);
2080 gimple_call_set_lhs (new_stmt, make_ssa_name (vec_dest));
2081 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2082 if (i == 0)
2083 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
2084 else
2085 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
2086 prev_stmt_info = vinfo_for_stmt (new_stmt);
2090 if (!is_store)
2092 /* Ensure that even with -fno-tree-dce the scalar MASK_LOAD is removed
2093 from the IL. */
2094 if (STMT_VINFO_RELATED_STMT (stmt_info))
2096 stmt = STMT_VINFO_RELATED_STMT (stmt_info);
2097 stmt_info = vinfo_for_stmt (stmt);
2099 tree lhs = gimple_call_lhs (stmt);
2100 new_stmt = gimple_build_assign (lhs, build_zero_cst (TREE_TYPE (lhs)));
2101 set_vinfo_for_stmt (new_stmt, stmt_info);
2102 set_vinfo_for_stmt (stmt, NULL);
2103 STMT_VINFO_STMT (stmt_info) = new_stmt;
2104 gsi_replace (gsi, new_stmt, true);
2107 return true;
2110 /* Return true if vector types VECTYPE_IN and VECTYPE_OUT have
2111 integer elements and if we can narrow VECTYPE_IN to VECTYPE_OUT
2112 in a single step. On success, store the binary pack code in
2113 *CONVERT_CODE. */
2115 static bool
2116 simple_integer_narrowing (tree vectype_out, tree vectype_in,
2117 tree_code *convert_code)
2119 if (!INTEGRAL_TYPE_P (TREE_TYPE (vectype_out))
2120 || !INTEGRAL_TYPE_P (TREE_TYPE (vectype_in)))
2121 return false;
2123 tree_code code;
2124 int multi_step_cvt = 0;
2125 auto_vec <tree, 8> interm_types;
2126 if (!supportable_narrowing_operation (NOP_EXPR, vectype_out, vectype_in,
2127 &code, &multi_step_cvt,
2128 &interm_types)
2129 || multi_step_cvt)
2130 return false;
2132 *convert_code = code;
2133 return true;
2136 /* Function vectorizable_call.
2138 Check if GS performs a function call that can be vectorized.
2139 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
2140 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
2141 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
2143 static bool
2144 vectorizable_call (gimple *gs, gimple_stmt_iterator *gsi, gimple **vec_stmt,
2145 slp_tree slp_node)
2147 gcall *stmt;
2148 tree vec_dest;
2149 tree scalar_dest;
2150 tree op, type;
2151 tree vec_oprnd0 = NULL_TREE, vec_oprnd1 = NULL_TREE;
2152 stmt_vec_info stmt_info = vinfo_for_stmt (gs), prev_stmt_info;
2153 tree vectype_out, vectype_in;
2154 int nunits_in;
2155 int nunits_out;
2156 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
2157 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
2158 vec_info *vinfo = stmt_info->vinfo;
2159 tree fndecl, new_temp, rhs_type;
2160 gimple *def_stmt;
2161 enum vect_def_type dt[3]
2162 = {vect_unknown_def_type, vect_unknown_def_type, vect_unknown_def_type};
2163 gimple *new_stmt = NULL;
2164 int ncopies, j;
2165 vec<tree> vargs = vNULL;
2166 enum { NARROW, NONE, WIDEN } modifier;
2167 size_t i, nargs;
2168 tree lhs;
2170 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
2171 return false;
2173 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def
2174 && ! vec_stmt)
2175 return false;
2177 /* Is GS a vectorizable call? */
2178 stmt = dyn_cast <gcall *> (gs);
2179 if (!stmt)
2180 return false;
2182 if (gimple_call_internal_p (stmt)
2183 && (gimple_call_internal_fn (stmt) == IFN_MASK_LOAD
2184 || gimple_call_internal_fn (stmt) == IFN_MASK_STORE))
2185 return vectorizable_mask_load_store (stmt, gsi, vec_stmt,
2186 slp_node);
2188 if (gimple_call_lhs (stmt) == NULL_TREE
2189 || TREE_CODE (gimple_call_lhs (stmt)) != SSA_NAME)
2190 return false;
2192 gcc_checking_assert (!stmt_can_throw_internal (stmt));
2194 vectype_out = STMT_VINFO_VECTYPE (stmt_info);
2196 /* Process function arguments. */
2197 rhs_type = NULL_TREE;
2198 vectype_in = NULL_TREE;
2199 nargs = gimple_call_num_args (stmt);
2201 /* Bail out if the function has more than three arguments, we do not have
2202 interesting builtin functions to vectorize with more than two arguments
2203 except for fma. No arguments is also not good. */
2204 if (nargs == 0 || nargs > 3)
2205 return false;
2207 /* Ignore the argument of IFN_GOMP_SIMD_LANE, it is magic. */
2208 if (gimple_call_internal_p (stmt)
2209 && gimple_call_internal_fn (stmt) == IFN_GOMP_SIMD_LANE)
2211 nargs = 0;
2212 rhs_type = unsigned_type_node;
2215 for (i = 0; i < nargs; i++)
2217 tree opvectype;
2219 op = gimple_call_arg (stmt, i);
2221 /* We can only handle calls with arguments of the same type. */
2222 if (rhs_type
2223 && !types_compatible_p (rhs_type, TREE_TYPE (op)))
2225 if (dump_enabled_p ())
2226 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2227 "argument types differ.\n");
2228 return false;
2230 if (!rhs_type)
2231 rhs_type = TREE_TYPE (op);
2233 if (!vect_is_simple_use (op, vinfo, &def_stmt, &dt[i], &opvectype))
2235 if (dump_enabled_p ())
2236 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2237 "use not simple.\n");
2238 return false;
2241 if (!vectype_in)
2242 vectype_in = opvectype;
2243 else if (opvectype
2244 && opvectype != vectype_in)
2246 if (dump_enabled_p ())
2247 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2248 "argument vector types differ.\n");
2249 return false;
2252 /* If all arguments are external or constant defs use a vector type with
2253 the same size as the output vector type. */
2254 if (!vectype_in)
2255 vectype_in = get_same_sized_vectype (rhs_type, vectype_out);
2256 if (vec_stmt)
2257 gcc_assert (vectype_in);
2258 if (!vectype_in)
2260 if (dump_enabled_p ())
2262 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2263 "no vectype for scalar type ");
2264 dump_generic_expr (MSG_MISSED_OPTIMIZATION, TDF_SLIM, rhs_type);
2265 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
2268 return false;
2271 /* FORNOW */
2272 nunits_in = TYPE_VECTOR_SUBPARTS (vectype_in);
2273 nunits_out = TYPE_VECTOR_SUBPARTS (vectype_out);
2274 if (nunits_in == nunits_out / 2)
2275 modifier = NARROW;
2276 else if (nunits_out == nunits_in)
2277 modifier = NONE;
2278 else if (nunits_out == nunits_in / 2)
2279 modifier = WIDEN;
2280 else
2281 return false;
2283 /* We only handle functions that do not read or clobber memory. */
2284 if (gimple_vuse (stmt))
2286 if (dump_enabled_p ())
2287 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2288 "function reads from or writes to memory.\n");
2289 return false;
2292 /* For now, we only vectorize functions if a target specific builtin
2293 is available. TODO -- in some cases, it might be profitable to
2294 insert the calls for pieces of the vector, in order to be able
2295 to vectorize other operations in the loop. */
2296 fndecl = NULL_TREE;
2297 internal_fn ifn = IFN_LAST;
2298 combined_fn cfn = gimple_call_combined_fn (stmt);
2299 tree callee = gimple_call_fndecl (stmt);
2301 /* First try using an internal function. */
2302 tree_code convert_code = ERROR_MARK;
2303 if (cfn != CFN_LAST
2304 && (modifier == NONE
2305 || (modifier == NARROW
2306 && simple_integer_narrowing (vectype_out, vectype_in,
2307 &convert_code))))
2308 ifn = vectorizable_internal_function (cfn, callee, vectype_out,
2309 vectype_in);
2311 /* If that fails, try asking for a target-specific built-in function. */
2312 if (ifn == IFN_LAST)
2314 if (cfn != CFN_LAST)
2315 fndecl = targetm.vectorize.builtin_vectorized_function
2316 (cfn, vectype_out, vectype_in);
2317 else
2318 fndecl = targetm.vectorize.builtin_md_vectorized_function
2319 (callee, vectype_out, vectype_in);
2322 if (ifn == IFN_LAST && !fndecl)
2324 if (cfn == CFN_GOMP_SIMD_LANE
2325 && !slp_node
2326 && loop_vinfo
2327 && LOOP_VINFO_LOOP (loop_vinfo)->simduid
2328 && TREE_CODE (gimple_call_arg (stmt, 0)) == SSA_NAME
2329 && LOOP_VINFO_LOOP (loop_vinfo)->simduid
2330 == SSA_NAME_VAR (gimple_call_arg (stmt, 0)))
2332 /* We can handle IFN_GOMP_SIMD_LANE by returning a
2333 { 0, 1, 2, ... vf - 1 } vector. */
2334 gcc_assert (nargs == 0);
2336 else
2338 if (dump_enabled_p ())
2339 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2340 "function is not vectorizable.\n");
2341 return false;
2345 if (slp_node || PURE_SLP_STMT (stmt_info))
2346 ncopies = 1;
2347 else if (modifier == NARROW && ifn == IFN_LAST)
2348 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_out;
2349 else
2350 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_in;
2352 /* Sanity check: make sure that at least one copy of the vectorized stmt
2353 needs to be generated. */
2354 gcc_assert (ncopies >= 1);
2356 if (!vec_stmt) /* transformation not required. */
2358 STMT_VINFO_TYPE (stmt_info) = call_vec_info_type;
2359 if (dump_enabled_p ())
2360 dump_printf_loc (MSG_NOTE, vect_location, "=== vectorizable_call ==="
2361 "\n");
2362 vect_model_simple_cost (stmt_info, ncopies, dt, NULL, NULL);
2363 if (ifn != IFN_LAST && modifier == NARROW && !slp_node)
2364 add_stmt_cost (stmt_info->vinfo->target_cost_data, ncopies / 2,
2365 vec_promote_demote, stmt_info, 0, vect_body);
2367 return true;
2370 /** Transform. **/
2372 if (dump_enabled_p ())
2373 dump_printf_loc (MSG_NOTE, vect_location, "transform call.\n");
2375 /* Handle def. */
2376 scalar_dest = gimple_call_lhs (stmt);
2377 vec_dest = vect_create_destination_var (scalar_dest, vectype_out);
2379 prev_stmt_info = NULL;
2380 if (modifier == NONE || ifn != IFN_LAST)
2382 tree prev_res = NULL_TREE;
2383 for (j = 0; j < ncopies; ++j)
2385 /* Build argument list for the vectorized call. */
2386 if (j == 0)
2387 vargs.create (nargs);
2388 else
2389 vargs.truncate (0);
2391 if (slp_node)
2393 auto_vec<vec<tree> > vec_defs (nargs);
2394 vec<tree> vec_oprnds0;
2396 for (i = 0; i < nargs; i++)
2397 vargs.quick_push (gimple_call_arg (stmt, i));
2398 vect_get_slp_defs (vargs, slp_node, &vec_defs, -1);
2399 vec_oprnds0 = vec_defs[0];
2401 /* Arguments are ready. Create the new vector stmt. */
2402 FOR_EACH_VEC_ELT (vec_oprnds0, i, vec_oprnd0)
2404 size_t k;
2405 for (k = 0; k < nargs; k++)
2407 vec<tree> vec_oprndsk = vec_defs[k];
2408 vargs[k] = vec_oprndsk[i];
2410 if (modifier == NARROW)
2412 tree half_res = make_ssa_name (vectype_in);
2413 new_stmt = gimple_build_call_internal_vec (ifn, vargs);
2414 gimple_call_set_lhs (new_stmt, half_res);
2415 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2416 if ((i & 1) == 0)
2418 prev_res = half_res;
2419 continue;
2421 new_temp = make_ssa_name (vec_dest);
2422 new_stmt = gimple_build_assign (new_temp, convert_code,
2423 prev_res, half_res);
2425 else
2427 if (ifn != IFN_LAST)
2428 new_stmt = gimple_build_call_internal_vec (ifn, vargs);
2429 else
2430 new_stmt = gimple_build_call_vec (fndecl, vargs);
2431 new_temp = make_ssa_name (vec_dest, new_stmt);
2432 gimple_call_set_lhs (new_stmt, new_temp);
2434 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2435 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
2438 for (i = 0; i < nargs; i++)
2440 vec<tree> vec_oprndsi = vec_defs[i];
2441 vec_oprndsi.release ();
2443 continue;
2446 for (i = 0; i < nargs; i++)
2448 op = gimple_call_arg (stmt, i);
2449 if (j == 0)
2450 vec_oprnd0
2451 = vect_get_vec_def_for_operand (op, stmt);
2452 else
2454 vec_oprnd0 = gimple_call_arg (new_stmt, i);
2455 vec_oprnd0
2456 = vect_get_vec_def_for_stmt_copy (dt[i], vec_oprnd0);
2459 vargs.quick_push (vec_oprnd0);
2462 if (gimple_call_internal_p (stmt)
2463 && gimple_call_internal_fn (stmt) == IFN_GOMP_SIMD_LANE)
2465 tree *v = XALLOCAVEC (tree, nunits_out);
2466 int k;
2467 for (k = 0; k < nunits_out; ++k)
2468 v[k] = build_int_cst (unsigned_type_node, j * nunits_out + k);
2469 tree cst = build_vector (vectype_out, v);
2470 tree new_var
2471 = vect_get_new_ssa_name (vectype_out, vect_simple_var, "cst_");
2472 gimple *init_stmt = gimple_build_assign (new_var, cst);
2473 vect_init_vector_1 (stmt, init_stmt, NULL);
2474 new_temp = make_ssa_name (vec_dest);
2475 new_stmt = gimple_build_assign (new_temp, new_var);
2477 else if (modifier == NARROW)
2479 tree half_res = make_ssa_name (vectype_in);
2480 new_stmt = gimple_build_call_internal_vec (ifn, vargs);
2481 gimple_call_set_lhs (new_stmt, half_res);
2482 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2483 if ((j & 1) == 0)
2485 prev_res = half_res;
2486 continue;
2488 new_temp = make_ssa_name (vec_dest);
2489 new_stmt = gimple_build_assign (new_temp, convert_code,
2490 prev_res, half_res);
2492 else
2494 if (ifn != IFN_LAST)
2495 new_stmt = gimple_build_call_internal_vec (ifn, vargs);
2496 else
2497 new_stmt = gimple_build_call_vec (fndecl, vargs);
2498 new_temp = make_ssa_name (vec_dest, new_stmt);
2499 gimple_call_set_lhs (new_stmt, new_temp);
2501 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2503 if (j == (modifier == NARROW ? 1 : 0))
2504 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
2505 else
2506 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
2508 prev_stmt_info = vinfo_for_stmt (new_stmt);
2511 else if (modifier == NARROW)
2513 for (j = 0; j < ncopies; ++j)
2515 /* Build argument list for the vectorized call. */
2516 if (j == 0)
2517 vargs.create (nargs * 2);
2518 else
2519 vargs.truncate (0);
2521 if (slp_node)
2523 auto_vec<vec<tree> > vec_defs (nargs);
2524 vec<tree> vec_oprnds0;
2526 for (i = 0; i < nargs; i++)
2527 vargs.quick_push (gimple_call_arg (stmt, i));
2528 vect_get_slp_defs (vargs, slp_node, &vec_defs, -1);
2529 vec_oprnds0 = vec_defs[0];
2531 /* Arguments are ready. Create the new vector stmt. */
2532 for (i = 0; vec_oprnds0.iterate (i, &vec_oprnd0); i += 2)
2534 size_t k;
2535 vargs.truncate (0);
2536 for (k = 0; k < nargs; k++)
2538 vec<tree> vec_oprndsk = vec_defs[k];
2539 vargs.quick_push (vec_oprndsk[i]);
2540 vargs.quick_push (vec_oprndsk[i + 1]);
2542 if (ifn != IFN_LAST)
2543 new_stmt = gimple_build_call_internal_vec (ifn, vargs);
2544 else
2545 new_stmt = gimple_build_call_vec (fndecl, vargs);
2546 new_temp = make_ssa_name (vec_dest, new_stmt);
2547 gimple_call_set_lhs (new_stmt, new_temp);
2548 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2549 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
2552 for (i = 0; i < nargs; i++)
2554 vec<tree> vec_oprndsi = vec_defs[i];
2555 vec_oprndsi.release ();
2557 continue;
2560 for (i = 0; i < nargs; i++)
2562 op = gimple_call_arg (stmt, i);
2563 if (j == 0)
2565 vec_oprnd0
2566 = vect_get_vec_def_for_operand (op, stmt);
2567 vec_oprnd1
2568 = vect_get_vec_def_for_stmt_copy (dt[i], vec_oprnd0);
2570 else
2572 vec_oprnd1 = gimple_call_arg (new_stmt, 2*i + 1);
2573 vec_oprnd0
2574 = vect_get_vec_def_for_stmt_copy (dt[i], vec_oprnd1);
2575 vec_oprnd1
2576 = vect_get_vec_def_for_stmt_copy (dt[i], vec_oprnd0);
2579 vargs.quick_push (vec_oprnd0);
2580 vargs.quick_push (vec_oprnd1);
2583 new_stmt = gimple_build_call_vec (fndecl, vargs);
2584 new_temp = make_ssa_name (vec_dest, new_stmt);
2585 gimple_call_set_lhs (new_stmt, new_temp);
2586 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2588 if (j == 0)
2589 STMT_VINFO_VEC_STMT (stmt_info) = new_stmt;
2590 else
2591 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
2593 prev_stmt_info = vinfo_for_stmt (new_stmt);
2596 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
2598 else
2599 /* No current target implements this case. */
2600 return false;
2602 vargs.release ();
2604 /* The call in STMT might prevent it from being removed in dce.
2605 We however cannot remove it here, due to the way the ssa name
2606 it defines is mapped to the new definition. So just replace
2607 rhs of the statement with something harmless. */
2609 if (slp_node)
2610 return true;
2612 type = TREE_TYPE (scalar_dest);
2613 if (is_pattern_stmt_p (stmt_info))
2614 lhs = gimple_call_lhs (STMT_VINFO_RELATED_STMT (stmt_info));
2615 else
2616 lhs = gimple_call_lhs (stmt);
2618 if (gimple_call_internal_p (stmt)
2619 && gimple_call_internal_fn (stmt) == IFN_GOMP_SIMD_LANE)
2621 /* Replace uses of the lhs of GOMP_SIMD_LANE call outside the loop
2622 with vf - 1 rather than 0, that is the last iteration of the
2623 vectorized loop. */
2624 imm_use_iterator iter;
2625 use_operand_p use_p;
2626 gimple *use_stmt;
2627 FOR_EACH_IMM_USE_STMT (use_stmt, iter, lhs)
2629 basic_block use_bb = gimple_bb (use_stmt);
2630 if (use_bb
2631 && !flow_bb_inside_loop_p (LOOP_VINFO_LOOP (loop_vinfo), use_bb))
2633 FOR_EACH_IMM_USE_ON_STMT (use_p, iter)
2634 SET_USE (use_p, build_int_cst (TREE_TYPE (lhs),
2635 ncopies * nunits_out - 1));
2636 update_stmt (use_stmt);
2641 new_stmt = gimple_build_assign (lhs, build_zero_cst (type));
2642 set_vinfo_for_stmt (new_stmt, stmt_info);
2643 set_vinfo_for_stmt (stmt, NULL);
2644 STMT_VINFO_STMT (stmt_info) = new_stmt;
2645 gsi_replace (gsi, new_stmt, false);
2647 return true;
2651 struct simd_call_arg_info
2653 tree vectype;
2654 tree op;
2655 enum vect_def_type dt;
2656 HOST_WIDE_INT linear_step;
2657 unsigned int align;
2658 bool simd_lane_linear;
2661 /* Helper function of vectorizable_simd_clone_call. If OP, an SSA_NAME,
2662 is linear within simd lane (but not within whole loop), note it in
2663 *ARGINFO. */
2665 static void
2666 vect_simd_lane_linear (tree op, struct loop *loop,
2667 struct simd_call_arg_info *arginfo)
2669 gimple *def_stmt = SSA_NAME_DEF_STMT (op);
2671 if (!is_gimple_assign (def_stmt)
2672 || gimple_assign_rhs_code (def_stmt) != POINTER_PLUS_EXPR
2673 || !is_gimple_min_invariant (gimple_assign_rhs1 (def_stmt)))
2674 return;
2676 tree base = gimple_assign_rhs1 (def_stmt);
2677 HOST_WIDE_INT linear_step = 0;
2678 tree v = gimple_assign_rhs2 (def_stmt);
2679 while (TREE_CODE (v) == SSA_NAME)
2681 tree t;
2682 def_stmt = SSA_NAME_DEF_STMT (v);
2683 if (is_gimple_assign (def_stmt))
2684 switch (gimple_assign_rhs_code (def_stmt))
2686 case PLUS_EXPR:
2687 t = gimple_assign_rhs2 (def_stmt);
2688 if (linear_step || TREE_CODE (t) != INTEGER_CST)
2689 return;
2690 base = fold_build2 (POINTER_PLUS_EXPR, TREE_TYPE (base), base, t);
2691 v = gimple_assign_rhs1 (def_stmt);
2692 continue;
2693 case MULT_EXPR:
2694 t = gimple_assign_rhs2 (def_stmt);
2695 if (linear_step || !tree_fits_shwi_p (t) || integer_zerop (t))
2696 return;
2697 linear_step = tree_to_shwi (t);
2698 v = gimple_assign_rhs1 (def_stmt);
2699 continue;
2700 CASE_CONVERT:
2701 t = gimple_assign_rhs1 (def_stmt);
2702 if (TREE_CODE (TREE_TYPE (t)) != INTEGER_TYPE
2703 || (TYPE_PRECISION (TREE_TYPE (v))
2704 < TYPE_PRECISION (TREE_TYPE (t))))
2705 return;
2706 if (!linear_step)
2707 linear_step = 1;
2708 v = t;
2709 continue;
2710 default:
2711 return;
2713 else if (is_gimple_call (def_stmt)
2714 && gimple_call_internal_p (def_stmt)
2715 && gimple_call_internal_fn (def_stmt) == IFN_GOMP_SIMD_LANE
2716 && loop->simduid
2717 && TREE_CODE (gimple_call_arg (def_stmt, 0)) == SSA_NAME
2718 && (SSA_NAME_VAR (gimple_call_arg (def_stmt, 0))
2719 == loop->simduid))
2721 if (!linear_step)
2722 linear_step = 1;
2723 arginfo->linear_step = linear_step;
2724 arginfo->op = base;
2725 arginfo->simd_lane_linear = true;
2726 return;
2731 /* Function vectorizable_simd_clone_call.
2733 Check if STMT performs a function call that can be vectorized
2734 by calling a simd clone of the function.
2735 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
2736 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
2737 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
2739 static bool
2740 vectorizable_simd_clone_call (gimple *stmt, gimple_stmt_iterator *gsi,
2741 gimple **vec_stmt, slp_tree slp_node)
2743 tree vec_dest;
2744 tree scalar_dest;
2745 tree op, type;
2746 tree vec_oprnd0 = NULL_TREE;
2747 stmt_vec_info stmt_info = vinfo_for_stmt (stmt), prev_stmt_info;
2748 tree vectype;
2749 unsigned int nunits;
2750 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
2751 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
2752 vec_info *vinfo = stmt_info->vinfo;
2753 struct loop *loop = loop_vinfo ? LOOP_VINFO_LOOP (loop_vinfo) : NULL;
2754 tree fndecl, new_temp;
2755 gimple *def_stmt;
2756 gimple *new_stmt = NULL;
2757 int ncopies, j;
2758 auto_vec<simd_call_arg_info> arginfo;
2759 vec<tree> vargs = vNULL;
2760 size_t i, nargs;
2761 tree lhs, rtype, ratype;
2762 vec<constructor_elt, va_gc> *ret_ctor_elts;
2764 /* Is STMT a vectorizable call? */
2765 if (!is_gimple_call (stmt))
2766 return false;
2768 fndecl = gimple_call_fndecl (stmt);
2769 if (fndecl == NULL_TREE)
2770 return false;
2772 struct cgraph_node *node = cgraph_node::get (fndecl);
2773 if (node == NULL || node->simd_clones == NULL)
2774 return false;
2776 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
2777 return false;
2779 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def
2780 && ! vec_stmt)
2781 return false;
2783 if (gimple_call_lhs (stmt)
2784 && TREE_CODE (gimple_call_lhs (stmt)) != SSA_NAME)
2785 return false;
2787 gcc_checking_assert (!stmt_can_throw_internal (stmt));
2789 vectype = STMT_VINFO_VECTYPE (stmt_info);
2791 if (loop_vinfo && nested_in_vect_loop_p (loop, stmt))
2792 return false;
2794 /* FORNOW */
2795 if (slp_node || PURE_SLP_STMT (stmt_info))
2796 return false;
2798 /* Process function arguments. */
2799 nargs = gimple_call_num_args (stmt);
2801 /* Bail out if the function has zero arguments. */
2802 if (nargs == 0)
2803 return false;
2805 arginfo.reserve (nargs, true);
2807 for (i = 0; i < nargs; i++)
2809 simd_call_arg_info thisarginfo;
2810 affine_iv iv;
2812 thisarginfo.linear_step = 0;
2813 thisarginfo.align = 0;
2814 thisarginfo.op = NULL_TREE;
2815 thisarginfo.simd_lane_linear = false;
2817 op = gimple_call_arg (stmt, i);
2818 if (!vect_is_simple_use (op, vinfo, &def_stmt, &thisarginfo.dt,
2819 &thisarginfo.vectype)
2820 || thisarginfo.dt == vect_uninitialized_def)
2822 if (dump_enabled_p ())
2823 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
2824 "use not simple.\n");
2825 return false;
2828 if (thisarginfo.dt == vect_constant_def
2829 || thisarginfo.dt == vect_external_def)
2830 gcc_assert (thisarginfo.vectype == NULL_TREE);
2831 else
2832 gcc_assert (thisarginfo.vectype != NULL_TREE);
2834 /* For linear arguments, the analyze phase should have saved
2835 the base and step in STMT_VINFO_SIMD_CLONE_INFO. */
2836 if (i * 3 + 4 <= STMT_VINFO_SIMD_CLONE_INFO (stmt_info).length ()
2837 && STMT_VINFO_SIMD_CLONE_INFO (stmt_info)[i * 3 + 2])
2839 gcc_assert (vec_stmt);
2840 thisarginfo.linear_step
2841 = tree_to_shwi (STMT_VINFO_SIMD_CLONE_INFO (stmt_info)[i * 3 + 2]);
2842 thisarginfo.op
2843 = STMT_VINFO_SIMD_CLONE_INFO (stmt_info)[i * 3 + 1];
2844 thisarginfo.simd_lane_linear
2845 = (STMT_VINFO_SIMD_CLONE_INFO (stmt_info)[i * 3 + 3]
2846 == boolean_true_node);
2847 /* If loop has been peeled for alignment, we need to adjust it. */
2848 tree n1 = LOOP_VINFO_NITERS_UNCHANGED (loop_vinfo);
2849 tree n2 = LOOP_VINFO_NITERS (loop_vinfo);
2850 if (n1 != n2 && !thisarginfo.simd_lane_linear)
2852 tree bias = fold_build2 (MINUS_EXPR, TREE_TYPE (n1), n1, n2);
2853 tree step = STMT_VINFO_SIMD_CLONE_INFO (stmt_info)[i * 3 + 2];
2854 tree opt = TREE_TYPE (thisarginfo.op);
2855 bias = fold_convert (TREE_TYPE (step), bias);
2856 bias = fold_build2 (MULT_EXPR, TREE_TYPE (step), bias, step);
2857 thisarginfo.op
2858 = fold_build2 (POINTER_TYPE_P (opt)
2859 ? POINTER_PLUS_EXPR : PLUS_EXPR, opt,
2860 thisarginfo.op, bias);
2863 else if (!vec_stmt
2864 && thisarginfo.dt != vect_constant_def
2865 && thisarginfo.dt != vect_external_def
2866 && loop_vinfo
2867 && TREE_CODE (op) == SSA_NAME
2868 && simple_iv (loop, loop_containing_stmt (stmt), op,
2869 &iv, false)
2870 && tree_fits_shwi_p (iv.step))
2872 thisarginfo.linear_step = tree_to_shwi (iv.step);
2873 thisarginfo.op = iv.base;
2875 else if ((thisarginfo.dt == vect_constant_def
2876 || thisarginfo.dt == vect_external_def)
2877 && POINTER_TYPE_P (TREE_TYPE (op)))
2878 thisarginfo.align = get_pointer_alignment (op) / BITS_PER_UNIT;
2879 /* Addresses of array elements indexed by GOMP_SIMD_LANE are
2880 linear too. */
2881 if (POINTER_TYPE_P (TREE_TYPE (op))
2882 && !thisarginfo.linear_step
2883 && !vec_stmt
2884 && thisarginfo.dt != vect_constant_def
2885 && thisarginfo.dt != vect_external_def
2886 && loop_vinfo
2887 && !slp_node
2888 && TREE_CODE (op) == SSA_NAME)
2889 vect_simd_lane_linear (op, loop, &thisarginfo);
2891 arginfo.quick_push (thisarginfo);
2894 unsigned int badness = 0;
2895 struct cgraph_node *bestn = NULL;
2896 if (STMT_VINFO_SIMD_CLONE_INFO (stmt_info).exists ())
2897 bestn = cgraph_node::get (STMT_VINFO_SIMD_CLONE_INFO (stmt_info)[0]);
2898 else
2899 for (struct cgraph_node *n = node->simd_clones; n != NULL;
2900 n = n->simdclone->next_clone)
2902 unsigned int this_badness = 0;
2903 if (n->simdclone->simdlen
2904 > (unsigned) LOOP_VINFO_VECT_FACTOR (loop_vinfo)
2905 || n->simdclone->nargs != nargs)
2906 continue;
2907 if (n->simdclone->simdlen
2908 < (unsigned) LOOP_VINFO_VECT_FACTOR (loop_vinfo))
2909 this_badness += (exact_log2 (LOOP_VINFO_VECT_FACTOR (loop_vinfo))
2910 - exact_log2 (n->simdclone->simdlen)) * 1024;
2911 if (n->simdclone->inbranch)
2912 this_badness += 2048;
2913 int target_badness = targetm.simd_clone.usable (n);
2914 if (target_badness < 0)
2915 continue;
2916 this_badness += target_badness * 512;
2917 /* FORNOW: Have to add code to add the mask argument. */
2918 if (n->simdclone->inbranch)
2919 continue;
2920 for (i = 0; i < nargs; i++)
2922 switch (n->simdclone->args[i].arg_type)
2924 case SIMD_CLONE_ARG_TYPE_VECTOR:
2925 if (!useless_type_conversion_p
2926 (n->simdclone->args[i].orig_type,
2927 TREE_TYPE (gimple_call_arg (stmt, i))))
2928 i = -1;
2929 else if (arginfo[i].dt == vect_constant_def
2930 || arginfo[i].dt == vect_external_def
2931 || arginfo[i].linear_step)
2932 this_badness += 64;
2933 break;
2934 case SIMD_CLONE_ARG_TYPE_UNIFORM:
2935 if (arginfo[i].dt != vect_constant_def
2936 && arginfo[i].dt != vect_external_def)
2937 i = -1;
2938 break;
2939 case SIMD_CLONE_ARG_TYPE_LINEAR_CONSTANT_STEP:
2940 case SIMD_CLONE_ARG_TYPE_LINEAR_REF_CONSTANT_STEP:
2941 if (arginfo[i].dt == vect_constant_def
2942 || arginfo[i].dt == vect_external_def
2943 || (arginfo[i].linear_step
2944 != n->simdclone->args[i].linear_step))
2945 i = -1;
2946 break;
2947 case SIMD_CLONE_ARG_TYPE_LINEAR_VARIABLE_STEP:
2948 case SIMD_CLONE_ARG_TYPE_LINEAR_VAL_CONSTANT_STEP:
2949 case SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_CONSTANT_STEP:
2950 case SIMD_CLONE_ARG_TYPE_LINEAR_REF_VARIABLE_STEP:
2951 case SIMD_CLONE_ARG_TYPE_LINEAR_VAL_VARIABLE_STEP:
2952 case SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_VARIABLE_STEP:
2953 /* FORNOW */
2954 i = -1;
2955 break;
2956 case SIMD_CLONE_ARG_TYPE_MASK:
2957 gcc_unreachable ();
2959 if (i == (size_t) -1)
2960 break;
2961 if (n->simdclone->args[i].alignment > arginfo[i].align)
2963 i = -1;
2964 break;
2966 if (arginfo[i].align)
2967 this_badness += (exact_log2 (arginfo[i].align)
2968 - exact_log2 (n->simdclone->args[i].alignment));
2970 if (i == (size_t) -1)
2971 continue;
2972 if (bestn == NULL || this_badness < badness)
2974 bestn = n;
2975 badness = this_badness;
2979 if (bestn == NULL)
2980 return false;
2982 for (i = 0; i < nargs; i++)
2983 if ((arginfo[i].dt == vect_constant_def
2984 || arginfo[i].dt == vect_external_def)
2985 && bestn->simdclone->args[i].arg_type == SIMD_CLONE_ARG_TYPE_VECTOR)
2987 arginfo[i].vectype
2988 = get_vectype_for_scalar_type (TREE_TYPE (gimple_call_arg (stmt,
2989 i)));
2990 if (arginfo[i].vectype == NULL
2991 || (TYPE_VECTOR_SUBPARTS (arginfo[i].vectype)
2992 > bestn->simdclone->simdlen))
2993 return false;
2996 fndecl = bestn->decl;
2997 nunits = bestn->simdclone->simdlen;
2998 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits;
3000 /* If the function isn't const, only allow it in simd loops where user
3001 has asserted that at least nunits consecutive iterations can be
3002 performed using SIMD instructions. */
3003 if ((loop == NULL || (unsigned) loop->safelen < nunits)
3004 && gimple_vuse (stmt))
3005 return false;
3007 /* Sanity check: make sure that at least one copy of the vectorized stmt
3008 needs to be generated. */
3009 gcc_assert (ncopies >= 1);
3011 if (!vec_stmt) /* transformation not required. */
3013 STMT_VINFO_SIMD_CLONE_INFO (stmt_info).safe_push (bestn->decl);
3014 for (i = 0; i < nargs; i++)
3015 if (bestn->simdclone->args[i].arg_type
3016 == SIMD_CLONE_ARG_TYPE_LINEAR_CONSTANT_STEP)
3018 STMT_VINFO_SIMD_CLONE_INFO (stmt_info).safe_grow_cleared (i * 3
3019 + 1);
3020 STMT_VINFO_SIMD_CLONE_INFO (stmt_info).safe_push (arginfo[i].op);
3021 tree lst = POINTER_TYPE_P (TREE_TYPE (arginfo[i].op))
3022 ? size_type_node : TREE_TYPE (arginfo[i].op);
3023 tree ls = build_int_cst (lst, arginfo[i].linear_step);
3024 STMT_VINFO_SIMD_CLONE_INFO (stmt_info).safe_push (ls);
3025 tree sll = arginfo[i].simd_lane_linear
3026 ? boolean_true_node : boolean_false_node;
3027 STMT_VINFO_SIMD_CLONE_INFO (stmt_info).safe_push (sll);
3029 STMT_VINFO_TYPE (stmt_info) = call_simd_clone_vec_info_type;
3030 if (dump_enabled_p ())
3031 dump_printf_loc (MSG_NOTE, vect_location,
3032 "=== vectorizable_simd_clone_call ===\n");
3033 /* vect_model_simple_cost (stmt_info, ncopies, dt, NULL, NULL); */
3034 return true;
3037 /** Transform. **/
3039 if (dump_enabled_p ())
3040 dump_printf_loc (MSG_NOTE, vect_location, "transform call.\n");
3042 /* Handle def. */
3043 scalar_dest = gimple_call_lhs (stmt);
3044 vec_dest = NULL_TREE;
3045 rtype = NULL_TREE;
3046 ratype = NULL_TREE;
3047 if (scalar_dest)
3049 vec_dest = vect_create_destination_var (scalar_dest, vectype);
3050 rtype = TREE_TYPE (TREE_TYPE (fndecl));
3051 if (TREE_CODE (rtype) == ARRAY_TYPE)
3053 ratype = rtype;
3054 rtype = TREE_TYPE (ratype);
3058 prev_stmt_info = NULL;
3059 for (j = 0; j < ncopies; ++j)
3061 /* Build argument list for the vectorized call. */
3062 if (j == 0)
3063 vargs.create (nargs);
3064 else
3065 vargs.truncate (0);
3067 for (i = 0; i < nargs; i++)
3069 unsigned int k, l, m, o;
3070 tree atype;
3071 op = gimple_call_arg (stmt, i);
3072 switch (bestn->simdclone->args[i].arg_type)
3074 case SIMD_CLONE_ARG_TYPE_VECTOR:
3075 atype = bestn->simdclone->args[i].vector_type;
3076 o = nunits / TYPE_VECTOR_SUBPARTS (atype);
3077 for (m = j * o; m < (j + 1) * o; m++)
3079 if (TYPE_VECTOR_SUBPARTS (atype)
3080 < TYPE_VECTOR_SUBPARTS (arginfo[i].vectype))
3082 unsigned int prec = GET_MODE_BITSIZE (TYPE_MODE (atype));
3083 k = (TYPE_VECTOR_SUBPARTS (arginfo[i].vectype)
3084 / TYPE_VECTOR_SUBPARTS (atype));
3085 gcc_assert ((k & (k - 1)) == 0);
3086 if (m == 0)
3087 vec_oprnd0
3088 = vect_get_vec_def_for_operand (op, stmt);
3089 else
3091 vec_oprnd0 = arginfo[i].op;
3092 if ((m & (k - 1)) == 0)
3093 vec_oprnd0
3094 = vect_get_vec_def_for_stmt_copy (arginfo[i].dt,
3095 vec_oprnd0);
3097 arginfo[i].op = vec_oprnd0;
3098 vec_oprnd0
3099 = build3 (BIT_FIELD_REF, atype, vec_oprnd0,
3100 size_int (prec),
3101 bitsize_int ((m & (k - 1)) * prec));
3102 new_stmt
3103 = gimple_build_assign (make_ssa_name (atype),
3104 vec_oprnd0);
3105 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3106 vargs.safe_push (gimple_assign_lhs (new_stmt));
3108 else
3110 k = (TYPE_VECTOR_SUBPARTS (atype)
3111 / TYPE_VECTOR_SUBPARTS (arginfo[i].vectype));
3112 gcc_assert ((k & (k - 1)) == 0);
3113 vec<constructor_elt, va_gc> *ctor_elts;
3114 if (k != 1)
3115 vec_alloc (ctor_elts, k);
3116 else
3117 ctor_elts = NULL;
3118 for (l = 0; l < k; l++)
3120 if (m == 0 && l == 0)
3121 vec_oprnd0
3122 = vect_get_vec_def_for_operand (op, stmt);
3123 else
3124 vec_oprnd0
3125 = vect_get_vec_def_for_stmt_copy (arginfo[i].dt,
3126 arginfo[i].op);
3127 arginfo[i].op = vec_oprnd0;
3128 if (k == 1)
3129 break;
3130 CONSTRUCTOR_APPEND_ELT (ctor_elts, NULL_TREE,
3131 vec_oprnd0);
3133 if (k == 1)
3134 vargs.safe_push (vec_oprnd0);
3135 else
3137 vec_oprnd0 = build_constructor (atype, ctor_elts);
3138 new_stmt
3139 = gimple_build_assign (make_ssa_name (atype),
3140 vec_oprnd0);
3141 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3142 vargs.safe_push (gimple_assign_lhs (new_stmt));
3146 break;
3147 case SIMD_CLONE_ARG_TYPE_UNIFORM:
3148 vargs.safe_push (op);
3149 break;
3150 case SIMD_CLONE_ARG_TYPE_LINEAR_CONSTANT_STEP:
3151 if (j == 0)
3153 gimple_seq stmts;
3154 arginfo[i].op
3155 = force_gimple_operand (arginfo[i].op, &stmts, true,
3156 NULL_TREE);
3157 if (stmts != NULL)
3159 basic_block new_bb;
3160 edge pe = loop_preheader_edge (loop);
3161 new_bb = gsi_insert_seq_on_edge_immediate (pe, stmts);
3162 gcc_assert (!new_bb);
3164 if (arginfo[i].simd_lane_linear)
3166 vargs.safe_push (arginfo[i].op);
3167 break;
3169 tree phi_res = copy_ssa_name (op);
3170 gphi *new_phi = create_phi_node (phi_res, loop->header);
3171 set_vinfo_for_stmt (new_phi,
3172 new_stmt_vec_info (new_phi, loop_vinfo));
3173 add_phi_arg (new_phi, arginfo[i].op,
3174 loop_preheader_edge (loop), UNKNOWN_LOCATION);
3175 enum tree_code code
3176 = POINTER_TYPE_P (TREE_TYPE (op))
3177 ? POINTER_PLUS_EXPR : PLUS_EXPR;
3178 tree type = POINTER_TYPE_P (TREE_TYPE (op))
3179 ? sizetype : TREE_TYPE (op);
3180 widest_int cst
3181 = wi::mul (bestn->simdclone->args[i].linear_step,
3182 ncopies * nunits);
3183 tree tcst = wide_int_to_tree (type, cst);
3184 tree phi_arg = copy_ssa_name (op);
3185 new_stmt
3186 = gimple_build_assign (phi_arg, code, phi_res, tcst);
3187 gimple_stmt_iterator si = gsi_after_labels (loop->header);
3188 gsi_insert_after (&si, new_stmt, GSI_NEW_STMT);
3189 set_vinfo_for_stmt (new_stmt,
3190 new_stmt_vec_info (new_stmt, loop_vinfo));
3191 add_phi_arg (new_phi, phi_arg, loop_latch_edge (loop),
3192 UNKNOWN_LOCATION);
3193 arginfo[i].op = phi_res;
3194 vargs.safe_push (phi_res);
3196 else
3198 enum tree_code code
3199 = POINTER_TYPE_P (TREE_TYPE (op))
3200 ? POINTER_PLUS_EXPR : PLUS_EXPR;
3201 tree type = POINTER_TYPE_P (TREE_TYPE (op))
3202 ? sizetype : TREE_TYPE (op);
3203 widest_int cst
3204 = wi::mul (bestn->simdclone->args[i].linear_step,
3205 j * nunits);
3206 tree tcst = wide_int_to_tree (type, cst);
3207 new_temp = make_ssa_name (TREE_TYPE (op));
3208 new_stmt = gimple_build_assign (new_temp, code,
3209 arginfo[i].op, tcst);
3210 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3211 vargs.safe_push (new_temp);
3213 break;
3214 case SIMD_CLONE_ARG_TYPE_LINEAR_VARIABLE_STEP:
3215 case SIMD_CLONE_ARG_TYPE_LINEAR_REF_VARIABLE_STEP:
3216 case SIMD_CLONE_ARG_TYPE_LINEAR_VAL_VARIABLE_STEP:
3217 case SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_VARIABLE_STEP:
3218 default:
3219 gcc_unreachable ();
3223 new_stmt = gimple_build_call_vec (fndecl, vargs);
3224 if (vec_dest)
3226 gcc_assert (ratype || TYPE_VECTOR_SUBPARTS (rtype) == nunits);
3227 if (ratype)
3228 new_temp = create_tmp_var (ratype);
3229 else if (TYPE_VECTOR_SUBPARTS (vectype)
3230 == TYPE_VECTOR_SUBPARTS (rtype))
3231 new_temp = make_ssa_name (vec_dest, new_stmt);
3232 else
3233 new_temp = make_ssa_name (rtype, new_stmt);
3234 gimple_call_set_lhs (new_stmt, new_temp);
3236 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3238 if (vec_dest)
3240 if (TYPE_VECTOR_SUBPARTS (vectype) < nunits)
3242 unsigned int k, l;
3243 unsigned int prec = GET_MODE_BITSIZE (TYPE_MODE (vectype));
3244 k = nunits / TYPE_VECTOR_SUBPARTS (vectype);
3245 gcc_assert ((k & (k - 1)) == 0);
3246 for (l = 0; l < k; l++)
3248 tree t;
3249 if (ratype)
3251 t = build_fold_addr_expr (new_temp);
3252 t = build2 (MEM_REF, vectype, t,
3253 build_int_cst (TREE_TYPE (t),
3254 l * prec / BITS_PER_UNIT));
3256 else
3257 t = build3 (BIT_FIELD_REF, vectype, new_temp,
3258 size_int (prec), bitsize_int (l * prec));
3259 new_stmt
3260 = gimple_build_assign (make_ssa_name (vectype), t);
3261 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3262 if (j == 0 && l == 0)
3263 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
3264 else
3265 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
3267 prev_stmt_info = vinfo_for_stmt (new_stmt);
3270 if (ratype)
3272 tree clobber = build_constructor (ratype, NULL);
3273 TREE_THIS_VOLATILE (clobber) = 1;
3274 new_stmt = gimple_build_assign (new_temp, clobber);
3275 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3277 continue;
3279 else if (TYPE_VECTOR_SUBPARTS (vectype) > nunits)
3281 unsigned int k = (TYPE_VECTOR_SUBPARTS (vectype)
3282 / TYPE_VECTOR_SUBPARTS (rtype));
3283 gcc_assert ((k & (k - 1)) == 0);
3284 if ((j & (k - 1)) == 0)
3285 vec_alloc (ret_ctor_elts, k);
3286 if (ratype)
3288 unsigned int m, o = nunits / TYPE_VECTOR_SUBPARTS (rtype);
3289 for (m = 0; m < o; m++)
3291 tree tem = build4 (ARRAY_REF, rtype, new_temp,
3292 size_int (m), NULL_TREE, NULL_TREE);
3293 new_stmt
3294 = gimple_build_assign (make_ssa_name (rtype), tem);
3295 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3296 CONSTRUCTOR_APPEND_ELT (ret_ctor_elts, NULL_TREE,
3297 gimple_assign_lhs (new_stmt));
3299 tree clobber = build_constructor (ratype, NULL);
3300 TREE_THIS_VOLATILE (clobber) = 1;
3301 new_stmt = gimple_build_assign (new_temp, clobber);
3302 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3304 else
3305 CONSTRUCTOR_APPEND_ELT (ret_ctor_elts, NULL_TREE, new_temp);
3306 if ((j & (k - 1)) != k - 1)
3307 continue;
3308 vec_oprnd0 = build_constructor (vectype, ret_ctor_elts);
3309 new_stmt
3310 = gimple_build_assign (make_ssa_name (vec_dest), vec_oprnd0);
3311 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3313 if ((unsigned) j == k - 1)
3314 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
3315 else
3316 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
3318 prev_stmt_info = vinfo_for_stmt (new_stmt);
3319 continue;
3321 else if (ratype)
3323 tree t = build_fold_addr_expr (new_temp);
3324 t = build2 (MEM_REF, vectype, t,
3325 build_int_cst (TREE_TYPE (t), 0));
3326 new_stmt
3327 = gimple_build_assign (make_ssa_name (vec_dest), t);
3328 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3329 tree clobber = build_constructor (ratype, NULL);
3330 TREE_THIS_VOLATILE (clobber) = 1;
3331 vect_finish_stmt_generation (stmt,
3332 gimple_build_assign (new_temp,
3333 clobber), gsi);
3337 if (j == 0)
3338 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
3339 else
3340 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
3342 prev_stmt_info = vinfo_for_stmt (new_stmt);
3345 vargs.release ();
3347 /* The call in STMT might prevent it from being removed in dce.
3348 We however cannot remove it here, due to the way the ssa name
3349 it defines is mapped to the new definition. So just replace
3350 rhs of the statement with something harmless. */
3352 if (slp_node)
3353 return true;
3355 if (scalar_dest)
3357 type = TREE_TYPE (scalar_dest);
3358 if (is_pattern_stmt_p (stmt_info))
3359 lhs = gimple_call_lhs (STMT_VINFO_RELATED_STMT (stmt_info));
3360 else
3361 lhs = gimple_call_lhs (stmt);
3362 new_stmt = gimple_build_assign (lhs, build_zero_cst (type));
3364 else
3365 new_stmt = gimple_build_nop ();
3366 set_vinfo_for_stmt (new_stmt, stmt_info);
3367 set_vinfo_for_stmt (stmt, NULL);
3368 STMT_VINFO_STMT (stmt_info) = new_stmt;
3369 gsi_replace (gsi, new_stmt, true);
3370 unlink_stmt_vdef (stmt);
3372 return true;
3376 /* Function vect_gen_widened_results_half
3378 Create a vector stmt whose code, type, number of arguments, and result
3379 variable are CODE, OP_TYPE, and VEC_DEST, and its arguments are
3380 VEC_OPRND0 and VEC_OPRND1. The new vector stmt is to be inserted at BSI.
3381 In the case that CODE is a CALL_EXPR, this means that a call to DECL
3382 needs to be created (DECL is a function-decl of a target-builtin).
3383 STMT is the original scalar stmt that we are vectorizing. */
3385 static gimple *
3386 vect_gen_widened_results_half (enum tree_code code,
3387 tree decl,
3388 tree vec_oprnd0, tree vec_oprnd1, int op_type,
3389 tree vec_dest, gimple_stmt_iterator *gsi,
3390 gimple *stmt)
3392 gimple *new_stmt;
3393 tree new_temp;
3395 /* Generate half of the widened result: */
3396 if (code == CALL_EXPR)
3398 /* Target specific support */
3399 if (op_type == binary_op)
3400 new_stmt = gimple_build_call (decl, 2, vec_oprnd0, vec_oprnd1);
3401 else
3402 new_stmt = gimple_build_call (decl, 1, vec_oprnd0);
3403 new_temp = make_ssa_name (vec_dest, new_stmt);
3404 gimple_call_set_lhs (new_stmt, new_temp);
3406 else
3408 /* Generic support */
3409 gcc_assert (op_type == TREE_CODE_LENGTH (code));
3410 if (op_type != binary_op)
3411 vec_oprnd1 = NULL;
3412 new_stmt = gimple_build_assign (vec_dest, code, vec_oprnd0, vec_oprnd1);
3413 new_temp = make_ssa_name (vec_dest, new_stmt);
3414 gimple_assign_set_lhs (new_stmt, new_temp);
3416 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3418 return new_stmt;
3422 /* Get vectorized definitions for loop-based vectorization. For the first
3423 operand we call vect_get_vec_def_for_operand() (with OPRND containing
3424 scalar operand), and for the rest we get a copy with
3425 vect_get_vec_def_for_stmt_copy() using the previous vector definition
3426 (stored in OPRND). See vect_get_vec_def_for_stmt_copy() for details.
3427 The vectors are collected into VEC_OPRNDS. */
3429 static void
3430 vect_get_loop_based_defs (tree *oprnd, gimple *stmt, enum vect_def_type dt,
3431 vec<tree> *vec_oprnds, int multi_step_cvt)
3433 tree vec_oprnd;
3435 /* Get first vector operand. */
3436 /* All the vector operands except the very first one (that is scalar oprnd)
3437 are stmt copies. */
3438 if (TREE_CODE (TREE_TYPE (*oprnd)) != VECTOR_TYPE)
3439 vec_oprnd = vect_get_vec_def_for_operand (*oprnd, stmt);
3440 else
3441 vec_oprnd = vect_get_vec_def_for_stmt_copy (dt, *oprnd);
3443 vec_oprnds->quick_push (vec_oprnd);
3445 /* Get second vector operand. */
3446 vec_oprnd = vect_get_vec_def_for_stmt_copy (dt, vec_oprnd);
3447 vec_oprnds->quick_push (vec_oprnd);
3449 *oprnd = vec_oprnd;
3451 /* For conversion in multiple steps, continue to get operands
3452 recursively. */
3453 if (multi_step_cvt)
3454 vect_get_loop_based_defs (oprnd, stmt, dt, vec_oprnds, multi_step_cvt - 1);
3458 /* Create vectorized demotion statements for vector operands from VEC_OPRNDS.
3459 For multi-step conversions store the resulting vectors and call the function
3460 recursively. */
3462 static void
3463 vect_create_vectorized_demotion_stmts (vec<tree> *vec_oprnds,
3464 int multi_step_cvt, gimple *stmt,
3465 vec<tree> vec_dsts,
3466 gimple_stmt_iterator *gsi,
3467 slp_tree slp_node, enum tree_code code,
3468 stmt_vec_info *prev_stmt_info)
3470 unsigned int i;
3471 tree vop0, vop1, new_tmp, vec_dest;
3472 gimple *new_stmt;
3473 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
3475 vec_dest = vec_dsts.pop ();
3477 for (i = 0; i < vec_oprnds->length (); i += 2)
3479 /* Create demotion operation. */
3480 vop0 = (*vec_oprnds)[i];
3481 vop1 = (*vec_oprnds)[i + 1];
3482 new_stmt = gimple_build_assign (vec_dest, code, vop0, vop1);
3483 new_tmp = make_ssa_name (vec_dest, new_stmt);
3484 gimple_assign_set_lhs (new_stmt, new_tmp);
3485 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3487 if (multi_step_cvt)
3488 /* Store the resulting vector for next recursive call. */
3489 (*vec_oprnds)[i/2] = new_tmp;
3490 else
3492 /* This is the last step of the conversion sequence. Store the
3493 vectors in SLP_NODE or in vector info of the scalar statement
3494 (or in STMT_VINFO_RELATED_STMT chain). */
3495 if (slp_node)
3496 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
3497 else
3499 if (!*prev_stmt_info)
3500 STMT_VINFO_VEC_STMT (stmt_info) = new_stmt;
3501 else
3502 STMT_VINFO_RELATED_STMT (*prev_stmt_info) = new_stmt;
3504 *prev_stmt_info = vinfo_for_stmt (new_stmt);
3509 /* For multi-step demotion operations we first generate demotion operations
3510 from the source type to the intermediate types, and then combine the
3511 results (stored in VEC_OPRNDS) in demotion operation to the destination
3512 type. */
3513 if (multi_step_cvt)
3515 /* At each level of recursion we have half of the operands we had at the
3516 previous level. */
3517 vec_oprnds->truncate ((i+1)/2);
3518 vect_create_vectorized_demotion_stmts (vec_oprnds, multi_step_cvt - 1,
3519 stmt, vec_dsts, gsi, slp_node,
3520 VEC_PACK_TRUNC_EXPR,
3521 prev_stmt_info);
3524 vec_dsts.quick_push (vec_dest);
3528 /* Create vectorized promotion statements for vector operands from VEC_OPRNDS0
3529 and VEC_OPRNDS1 (for binary operations). For multi-step conversions store
3530 the resulting vectors and call the function recursively. */
3532 static void
3533 vect_create_vectorized_promotion_stmts (vec<tree> *vec_oprnds0,
3534 vec<tree> *vec_oprnds1,
3535 gimple *stmt, tree vec_dest,
3536 gimple_stmt_iterator *gsi,
3537 enum tree_code code1,
3538 enum tree_code code2, tree decl1,
3539 tree decl2, int op_type)
3541 int i;
3542 tree vop0, vop1, new_tmp1, new_tmp2;
3543 gimple *new_stmt1, *new_stmt2;
3544 vec<tree> vec_tmp = vNULL;
3546 vec_tmp.create (vec_oprnds0->length () * 2);
3547 FOR_EACH_VEC_ELT (*vec_oprnds0, i, vop0)
3549 if (op_type == binary_op)
3550 vop1 = (*vec_oprnds1)[i];
3551 else
3552 vop1 = NULL_TREE;
3554 /* Generate the two halves of promotion operation. */
3555 new_stmt1 = vect_gen_widened_results_half (code1, decl1, vop0, vop1,
3556 op_type, vec_dest, gsi, stmt);
3557 new_stmt2 = vect_gen_widened_results_half (code2, decl2, vop0, vop1,
3558 op_type, vec_dest, gsi, stmt);
3559 if (is_gimple_call (new_stmt1))
3561 new_tmp1 = gimple_call_lhs (new_stmt1);
3562 new_tmp2 = gimple_call_lhs (new_stmt2);
3564 else
3566 new_tmp1 = gimple_assign_lhs (new_stmt1);
3567 new_tmp2 = gimple_assign_lhs (new_stmt2);
3570 /* Store the results for the next step. */
3571 vec_tmp.quick_push (new_tmp1);
3572 vec_tmp.quick_push (new_tmp2);
3575 vec_oprnds0->release ();
3576 *vec_oprnds0 = vec_tmp;
3580 /* Check if STMT performs a conversion operation, that can be vectorized.
3581 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
3582 stmt to replace it, put it in VEC_STMT, and insert it at GSI.
3583 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
3585 static bool
3586 vectorizable_conversion (gimple *stmt, gimple_stmt_iterator *gsi,
3587 gimple **vec_stmt, slp_tree slp_node)
3589 tree vec_dest;
3590 tree scalar_dest;
3591 tree op0, op1 = NULL_TREE;
3592 tree vec_oprnd0 = NULL_TREE, vec_oprnd1 = NULL_TREE;
3593 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
3594 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
3595 enum tree_code code, code1 = ERROR_MARK, code2 = ERROR_MARK;
3596 enum tree_code codecvt1 = ERROR_MARK, codecvt2 = ERROR_MARK;
3597 tree decl1 = NULL_TREE, decl2 = NULL_TREE;
3598 tree new_temp;
3599 gimple *def_stmt;
3600 enum vect_def_type dt[2] = {vect_unknown_def_type, vect_unknown_def_type};
3601 gimple *new_stmt = NULL;
3602 stmt_vec_info prev_stmt_info;
3603 int nunits_in;
3604 int nunits_out;
3605 tree vectype_out, vectype_in;
3606 int ncopies, i, j;
3607 tree lhs_type, rhs_type;
3608 enum { NARROW, NONE, WIDEN } modifier;
3609 vec<tree> vec_oprnds0 = vNULL;
3610 vec<tree> vec_oprnds1 = vNULL;
3611 tree vop0;
3612 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
3613 vec_info *vinfo = stmt_info->vinfo;
3614 int multi_step_cvt = 0;
3615 vec<tree> vec_dsts = vNULL;
3616 vec<tree> interm_types = vNULL;
3617 tree last_oprnd, intermediate_type, cvt_type = NULL_TREE;
3618 int op_type;
3619 machine_mode rhs_mode;
3620 unsigned short fltsz;
3622 /* Is STMT a vectorizable conversion? */
3624 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
3625 return false;
3627 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def
3628 && ! vec_stmt)
3629 return false;
3631 if (!is_gimple_assign (stmt))
3632 return false;
3634 if (TREE_CODE (gimple_assign_lhs (stmt)) != SSA_NAME)
3635 return false;
3637 code = gimple_assign_rhs_code (stmt);
3638 if (!CONVERT_EXPR_CODE_P (code)
3639 && code != FIX_TRUNC_EXPR
3640 && code != FLOAT_EXPR
3641 && code != WIDEN_MULT_EXPR
3642 && code != WIDEN_LSHIFT_EXPR)
3643 return false;
3645 op_type = TREE_CODE_LENGTH (code);
3647 /* Check types of lhs and rhs. */
3648 scalar_dest = gimple_assign_lhs (stmt);
3649 lhs_type = TREE_TYPE (scalar_dest);
3650 vectype_out = STMT_VINFO_VECTYPE (stmt_info);
3652 op0 = gimple_assign_rhs1 (stmt);
3653 rhs_type = TREE_TYPE (op0);
3655 if ((code != FIX_TRUNC_EXPR && code != FLOAT_EXPR)
3656 && !((INTEGRAL_TYPE_P (lhs_type)
3657 && INTEGRAL_TYPE_P (rhs_type))
3658 || (SCALAR_FLOAT_TYPE_P (lhs_type)
3659 && SCALAR_FLOAT_TYPE_P (rhs_type))))
3660 return false;
3662 if (!VECTOR_BOOLEAN_TYPE_P (vectype_out)
3663 && ((INTEGRAL_TYPE_P (lhs_type)
3664 && (TYPE_PRECISION (lhs_type)
3665 != GET_MODE_PRECISION (TYPE_MODE (lhs_type))))
3666 || (INTEGRAL_TYPE_P (rhs_type)
3667 && (TYPE_PRECISION (rhs_type)
3668 != GET_MODE_PRECISION (TYPE_MODE (rhs_type))))))
3670 if (dump_enabled_p ())
3671 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
3672 "type conversion to/from bit-precision unsupported."
3673 "\n");
3674 return false;
3677 /* Check the operands of the operation. */
3678 if (!vect_is_simple_use (op0, vinfo, &def_stmt, &dt[0], &vectype_in))
3680 if (dump_enabled_p ())
3681 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
3682 "use not simple.\n");
3683 return false;
3685 if (op_type == binary_op)
3687 bool ok;
3689 op1 = gimple_assign_rhs2 (stmt);
3690 gcc_assert (code == WIDEN_MULT_EXPR || code == WIDEN_LSHIFT_EXPR);
3691 /* For WIDEN_MULT_EXPR, if OP0 is a constant, use the type of
3692 OP1. */
3693 if (CONSTANT_CLASS_P (op0))
3694 ok = vect_is_simple_use (op1, vinfo, &def_stmt, &dt[1], &vectype_in);
3695 else
3696 ok = vect_is_simple_use (op1, vinfo, &def_stmt, &dt[1]);
3698 if (!ok)
3700 if (dump_enabled_p ())
3701 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
3702 "use not simple.\n");
3703 return false;
3707 /* If op0 is an external or constant defs use a vector type of
3708 the same size as the output vector type. */
3709 if (!vectype_in)
3710 vectype_in = get_same_sized_vectype (rhs_type, vectype_out);
3711 if (vec_stmt)
3712 gcc_assert (vectype_in);
3713 if (!vectype_in)
3715 if (dump_enabled_p ())
3717 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
3718 "no vectype for scalar type ");
3719 dump_generic_expr (MSG_MISSED_OPTIMIZATION, TDF_SLIM, rhs_type);
3720 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
3723 return false;
3726 if (VECTOR_BOOLEAN_TYPE_P (vectype_out)
3727 && !VECTOR_BOOLEAN_TYPE_P (vectype_in))
3729 if (dump_enabled_p ())
3731 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
3732 "can't convert between boolean and non "
3733 "boolean vectors");
3734 dump_generic_expr (MSG_MISSED_OPTIMIZATION, TDF_SLIM, rhs_type);
3735 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
3738 return false;
3741 nunits_in = TYPE_VECTOR_SUBPARTS (vectype_in);
3742 nunits_out = TYPE_VECTOR_SUBPARTS (vectype_out);
3743 if (nunits_in < nunits_out)
3744 modifier = NARROW;
3745 else if (nunits_out == nunits_in)
3746 modifier = NONE;
3747 else
3748 modifier = WIDEN;
3750 /* Multiple types in SLP are handled by creating the appropriate number of
3751 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
3752 case of SLP. */
3753 if (slp_node || PURE_SLP_STMT (stmt_info))
3754 ncopies = 1;
3755 else if (modifier == NARROW)
3756 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_out;
3757 else
3758 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_in;
3760 /* Sanity check: make sure that at least one copy of the vectorized stmt
3761 needs to be generated. */
3762 gcc_assert (ncopies >= 1);
3764 /* Supportable by target? */
3765 switch (modifier)
3767 case NONE:
3768 if (code != FIX_TRUNC_EXPR && code != FLOAT_EXPR)
3769 return false;
3770 if (supportable_convert_operation (code, vectype_out, vectype_in,
3771 &decl1, &code1))
3772 break;
3773 /* FALLTHRU */
3774 unsupported:
3775 if (dump_enabled_p ())
3776 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
3777 "conversion not supported by target.\n");
3778 return false;
3780 case WIDEN:
3781 if (supportable_widening_operation (code, stmt, vectype_out, vectype_in,
3782 &code1, &code2, &multi_step_cvt,
3783 &interm_types))
3785 /* Binary widening operation can only be supported directly by the
3786 architecture. */
3787 gcc_assert (!(multi_step_cvt && op_type == binary_op));
3788 break;
3791 if (code != FLOAT_EXPR
3792 || (GET_MODE_SIZE (TYPE_MODE (lhs_type))
3793 <= GET_MODE_SIZE (TYPE_MODE (rhs_type))))
3794 goto unsupported;
3796 rhs_mode = TYPE_MODE (rhs_type);
3797 fltsz = GET_MODE_SIZE (TYPE_MODE (lhs_type));
3798 for (rhs_mode = GET_MODE_2XWIDER_MODE (TYPE_MODE (rhs_type));
3799 rhs_mode != VOIDmode && GET_MODE_SIZE (rhs_mode) <= fltsz;
3800 rhs_mode = GET_MODE_2XWIDER_MODE (rhs_mode))
3802 cvt_type
3803 = build_nonstandard_integer_type (GET_MODE_BITSIZE (rhs_mode), 0);
3804 cvt_type = get_same_sized_vectype (cvt_type, vectype_in);
3805 if (cvt_type == NULL_TREE)
3806 goto unsupported;
3808 if (GET_MODE_SIZE (rhs_mode) == fltsz)
3810 if (!supportable_convert_operation (code, vectype_out,
3811 cvt_type, &decl1, &codecvt1))
3812 goto unsupported;
3814 else if (!supportable_widening_operation (code, stmt, vectype_out,
3815 cvt_type, &codecvt1,
3816 &codecvt2, &multi_step_cvt,
3817 &interm_types))
3818 continue;
3819 else
3820 gcc_assert (multi_step_cvt == 0);
3822 if (supportable_widening_operation (NOP_EXPR, stmt, cvt_type,
3823 vectype_in, &code1, &code2,
3824 &multi_step_cvt, &interm_types))
3825 break;
3828 if (rhs_mode == VOIDmode || GET_MODE_SIZE (rhs_mode) > fltsz)
3829 goto unsupported;
3831 if (GET_MODE_SIZE (rhs_mode) == fltsz)
3832 codecvt2 = ERROR_MARK;
3833 else
3835 multi_step_cvt++;
3836 interm_types.safe_push (cvt_type);
3837 cvt_type = NULL_TREE;
3839 break;
3841 case NARROW:
3842 gcc_assert (op_type == unary_op);
3843 if (supportable_narrowing_operation (code, vectype_out, vectype_in,
3844 &code1, &multi_step_cvt,
3845 &interm_types))
3846 break;
3848 if (code != FIX_TRUNC_EXPR
3849 || (GET_MODE_SIZE (TYPE_MODE (lhs_type))
3850 >= GET_MODE_SIZE (TYPE_MODE (rhs_type))))
3851 goto unsupported;
3853 rhs_mode = TYPE_MODE (rhs_type);
3854 cvt_type
3855 = build_nonstandard_integer_type (GET_MODE_BITSIZE (rhs_mode), 0);
3856 cvt_type = get_same_sized_vectype (cvt_type, vectype_in);
3857 if (cvt_type == NULL_TREE)
3858 goto unsupported;
3859 if (!supportable_convert_operation (code, cvt_type, vectype_in,
3860 &decl1, &codecvt1))
3861 goto unsupported;
3862 if (supportable_narrowing_operation (NOP_EXPR, vectype_out, cvt_type,
3863 &code1, &multi_step_cvt,
3864 &interm_types))
3865 break;
3866 goto unsupported;
3868 default:
3869 gcc_unreachable ();
3872 if (!vec_stmt) /* transformation not required. */
3874 if (dump_enabled_p ())
3875 dump_printf_loc (MSG_NOTE, vect_location,
3876 "=== vectorizable_conversion ===\n");
3877 if (code == FIX_TRUNC_EXPR || code == FLOAT_EXPR)
3879 STMT_VINFO_TYPE (stmt_info) = type_conversion_vec_info_type;
3880 vect_model_simple_cost (stmt_info, ncopies, dt, NULL, NULL);
3882 else if (modifier == NARROW)
3884 STMT_VINFO_TYPE (stmt_info) = type_demotion_vec_info_type;
3885 vect_model_promotion_demotion_cost (stmt_info, dt, multi_step_cvt);
3887 else
3889 STMT_VINFO_TYPE (stmt_info) = type_promotion_vec_info_type;
3890 vect_model_promotion_demotion_cost (stmt_info, dt, multi_step_cvt);
3892 interm_types.release ();
3893 return true;
3896 /** Transform. **/
3897 if (dump_enabled_p ())
3898 dump_printf_loc (MSG_NOTE, vect_location,
3899 "transform conversion. ncopies = %d.\n", ncopies);
3901 if (op_type == binary_op)
3903 if (CONSTANT_CLASS_P (op0))
3904 op0 = fold_convert (TREE_TYPE (op1), op0);
3905 else if (CONSTANT_CLASS_P (op1))
3906 op1 = fold_convert (TREE_TYPE (op0), op1);
3909 /* In case of multi-step conversion, we first generate conversion operations
3910 to the intermediate types, and then from that types to the final one.
3911 We create vector destinations for the intermediate type (TYPES) received
3912 from supportable_*_operation, and store them in the correct order
3913 for future use in vect_create_vectorized_*_stmts (). */
3914 vec_dsts.create (multi_step_cvt + 1);
3915 vec_dest = vect_create_destination_var (scalar_dest,
3916 (cvt_type && modifier == WIDEN)
3917 ? cvt_type : vectype_out);
3918 vec_dsts.quick_push (vec_dest);
3920 if (multi_step_cvt)
3922 for (i = interm_types.length () - 1;
3923 interm_types.iterate (i, &intermediate_type); i--)
3925 vec_dest = vect_create_destination_var (scalar_dest,
3926 intermediate_type);
3927 vec_dsts.quick_push (vec_dest);
3931 if (cvt_type)
3932 vec_dest = vect_create_destination_var (scalar_dest,
3933 modifier == WIDEN
3934 ? vectype_out : cvt_type);
3936 if (!slp_node)
3938 if (modifier == WIDEN)
3940 vec_oprnds0.create (multi_step_cvt ? vect_pow2 (multi_step_cvt) : 1);
3941 if (op_type == binary_op)
3942 vec_oprnds1.create (1);
3944 else if (modifier == NARROW)
3945 vec_oprnds0.create (
3946 2 * (multi_step_cvt ? vect_pow2 (multi_step_cvt) : 1));
3948 else if (code == WIDEN_LSHIFT_EXPR)
3949 vec_oprnds1.create (slp_node->vec_stmts_size);
3951 last_oprnd = op0;
3952 prev_stmt_info = NULL;
3953 switch (modifier)
3955 case NONE:
3956 for (j = 0; j < ncopies; j++)
3958 if (j == 0)
3959 vect_get_vec_defs (op0, NULL, stmt, &vec_oprnds0, NULL, slp_node,
3960 -1);
3961 else
3962 vect_get_vec_defs_for_stmt_copy (dt, &vec_oprnds0, NULL);
3964 FOR_EACH_VEC_ELT (vec_oprnds0, i, vop0)
3966 /* Arguments are ready, create the new vector stmt. */
3967 if (code1 == CALL_EXPR)
3969 new_stmt = gimple_build_call (decl1, 1, vop0);
3970 new_temp = make_ssa_name (vec_dest, new_stmt);
3971 gimple_call_set_lhs (new_stmt, new_temp);
3973 else
3975 gcc_assert (TREE_CODE_LENGTH (code1) == unary_op);
3976 new_stmt = gimple_build_assign (vec_dest, code1, vop0);
3977 new_temp = make_ssa_name (vec_dest, new_stmt);
3978 gimple_assign_set_lhs (new_stmt, new_temp);
3981 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3982 if (slp_node)
3983 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
3984 else
3986 if (!prev_stmt_info)
3987 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
3988 else
3989 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
3990 prev_stmt_info = vinfo_for_stmt (new_stmt);
3994 break;
3996 case WIDEN:
3997 /* In case the vectorization factor (VF) is bigger than the number
3998 of elements that we can fit in a vectype (nunits), we have to
3999 generate more than one vector stmt - i.e - we need to "unroll"
4000 the vector stmt by a factor VF/nunits. */
4001 for (j = 0; j < ncopies; j++)
4003 /* Handle uses. */
4004 if (j == 0)
4006 if (slp_node)
4008 if (code == WIDEN_LSHIFT_EXPR)
4010 unsigned int k;
4012 vec_oprnd1 = op1;
4013 /* Store vec_oprnd1 for every vector stmt to be created
4014 for SLP_NODE. We check during the analysis that all
4015 the shift arguments are the same. */
4016 for (k = 0; k < slp_node->vec_stmts_size - 1; k++)
4017 vec_oprnds1.quick_push (vec_oprnd1);
4019 vect_get_vec_defs (op0, NULL_TREE, stmt, &vec_oprnds0, NULL,
4020 slp_node, -1);
4022 else
4023 vect_get_vec_defs (op0, op1, stmt, &vec_oprnds0,
4024 &vec_oprnds1, slp_node, -1);
4026 else
4028 vec_oprnd0 = vect_get_vec_def_for_operand (op0, stmt);
4029 vec_oprnds0.quick_push (vec_oprnd0);
4030 if (op_type == binary_op)
4032 if (code == WIDEN_LSHIFT_EXPR)
4033 vec_oprnd1 = op1;
4034 else
4035 vec_oprnd1 = vect_get_vec_def_for_operand (op1, stmt);
4036 vec_oprnds1.quick_push (vec_oprnd1);
4040 else
4042 vec_oprnd0 = vect_get_vec_def_for_stmt_copy (dt[0], vec_oprnd0);
4043 vec_oprnds0.truncate (0);
4044 vec_oprnds0.quick_push (vec_oprnd0);
4045 if (op_type == binary_op)
4047 if (code == WIDEN_LSHIFT_EXPR)
4048 vec_oprnd1 = op1;
4049 else
4050 vec_oprnd1 = vect_get_vec_def_for_stmt_copy (dt[1],
4051 vec_oprnd1);
4052 vec_oprnds1.truncate (0);
4053 vec_oprnds1.quick_push (vec_oprnd1);
4057 /* Arguments are ready. Create the new vector stmts. */
4058 for (i = multi_step_cvt; i >= 0; i--)
4060 tree this_dest = vec_dsts[i];
4061 enum tree_code c1 = code1, c2 = code2;
4062 if (i == 0 && codecvt2 != ERROR_MARK)
4064 c1 = codecvt1;
4065 c2 = codecvt2;
4067 vect_create_vectorized_promotion_stmts (&vec_oprnds0,
4068 &vec_oprnds1,
4069 stmt, this_dest, gsi,
4070 c1, c2, decl1, decl2,
4071 op_type);
4074 FOR_EACH_VEC_ELT (vec_oprnds0, i, vop0)
4076 if (cvt_type)
4078 if (codecvt1 == CALL_EXPR)
4080 new_stmt = gimple_build_call (decl1, 1, vop0);
4081 new_temp = make_ssa_name (vec_dest, new_stmt);
4082 gimple_call_set_lhs (new_stmt, new_temp);
4084 else
4086 gcc_assert (TREE_CODE_LENGTH (codecvt1) == unary_op);
4087 new_temp = make_ssa_name (vec_dest);
4088 new_stmt = gimple_build_assign (new_temp, codecvt1,
4089 vop0);
4092 vect_finish_stmt_generation (stmt, new_stmt, gsi);
4094 else
4095 new_stmt = SSA_NAME_DEF_STMT (vop0);
4097 if (slp_node)
4098 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
4099 else
4101 if (!prev_stmt_info)
4102 STMT_VINFO_VEC_STMT (stmt_info) = new_stmt;
4103 else
4104 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
4105 prev_stmt_info = vinfo_for_stmt (new_stmt);
4110 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
4111 break;
4113 case NARROW:
4114 /* In case the vectorization factor (VF) is bigger than the number
4115 of elements that we can fit in a vectype (nunits), we have to
4116 generate more than one vector stmt - i.e - we need to "unroll"
4117 the vector stmt by a factor VF/nunits. */
4118 for (j = 0; j < ncopies; j++)
4120 /* Handle uses. */
4121 if (slp_node)
4122 vect_get_vec_defs (op0, NULL_TREE, stmt, &vec_oprnds0, NULL,
4123 slp_node, -1);
4124 else
4126 vec_oprnds0.truncate (0);
4127 vect_get_loop_based_defs (&last_oprnd, stmt, dt[0], &vec_oprnds0,
4128 vect_pow2 (multi_step_cvt) - 1);
4131 /* Arguments are ready. Create the new vector stmts. */
4132 if (cvt_type)
4133 FOR_EACH_VEC_ELT (vec_oprnds0, i, vop0)
4135 if (codecvt1 == CALL_EXPR)
4137 new_stmt = gimple_build_call (decl1, 1, vop0);
4138 new_temp = make_ssa_name (vec_dest, new_stmt);
4139 gimple_call_set_lhs (new_stmt, new_temp);
4141 else
4143 gcc_assert (TREE_CODE_LENGTH (codecvt1) == unary_op);
4144 new_temp = make_ssa_name (vec_dest);
4145 new_stmt = gimple_build_assign (new_temp, codecvt1,
4146 vop0);
4149 vect_finish_stmt_generation (stmt, new_stmt, gsi);
4150 vec_oprnds0[i] = new_temp;
4153 vect_create_vectorized_demotion_stmts (&vec_oprnds0, multi_step_cvt,
4154 stmt, vec_dsts, gsi,
4155 slp_node, code1,
4156 &prev_stmt_info);
4159 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
4160 break;
4163 vec_oprnds0.release ();
4164 vec_oprnds1.release ();
4165 vec_dsts.release ();
4166 interm_types.release ();
4168 return true;
4172 /* Function vectorizable_assignment.
4174 Check if STMT performs an assignment (copy) that can be vectorized.
4175 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
4176 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
4177 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
4179 static bool
4180 vectorizable_assignment (gimple *stmt, gimple_stmt_iterator *gsi,
4181 gimple **vec_stmt, slp_tree slp_node)
4183 tree vec_dest;
4184 tree scalar_dest;
4185 tree op;
4186 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
4187 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
4188 tree new_temp;
4189 gimple *def_stmt;
4190 enum vect_def_type dt[2] = {vect_unknown_def_type, vect_unknown_def_type};
4191 int ncopies;
4192 int i, j;
4193 vec<tree> vec_oprnds = vNULL;
4194 tree vop;
4195 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
4196 vec_info *vinfo = stmt_info->vinfo;
4197 gimple *new_stmt = NULL;
4198 stmt_vec_info prev_stmt_info = NULL;
4199 enum tree_code code;
4200 tree vectype_in;
4202 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
4203 return false;
4205 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def
4206 && ! vec_stmt)
4207 return false;
4209 /* Is vectorizable assignment? */
4210 if (!is_gimple_assign (stmt))
4211 return false;
4213 scalar_dest = gimple_assign_lhs (stmt);
4214 if (TREE_CODE (scalar_dest) != SSA_NAME)
4215 return false;
4217 code = gimple_assign_rhs_code (stmt);
4218 if (gimple_assign_single_p (stmt)
4219 || code == PAREN_EXPR
4220 || CONVERT_EXPR_CODE_P (code))
4221 op = gimple_assign_rhs1 (stmt);
4222 else
4223 return false;
4225 if (code == VIEW_CONVERT_EXPR)
4226 op = TREE_OPERAND (op, 0);
4228 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
4229 unsigned int nunits = TYPE_VECTOR_SUBPARTS (vectype);
4231 /* Multiple types in SLP are handled by creating the appropriate number of
4232 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
4233 case of SLP. */
4234 if (slp_node || PURE_SLP_STMT (stmt_info))
4235 ncopies = 1;
4236 else
4237 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits;
4239 gcc_assert (ncopies >= 1);
4241 if (!vect_is_simple_use (op, vinfo, &def_stmt, &dt[0], &vectype_in))
4243 if (dump_enabled_p ())
4244 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4245 "use not simple.\n");
4246 return false;
4249 /* We can handle NOP_EXPR conversions that do not change the number
4250 of elements or the vector size. */
4251 if ((CONVERT_EXPR_CODE_P (code)
4252 || code == VIEW_CONVERT_EXPR)
4253 && (!vectype_in
4254 || TYPE_VECTOR_SUBPARTS (vectype_in) != nunits
4255 || (GET_MODE_SIZE (TYPE_MODE (vectype))
4256 != GET_MODE_SIZE (TYPE_MODE (vectype_in)))))
4257 return false;
4259 /* We do not handle bit-precision changes. */
4260 if ((CONVERT_EXPR_CODE_P (code)
4261 || code == VIEW_CONVERT_EXPR)
4262 && INTEGRAL_TYPE_P (TREE_TYPE (scalar_dest))
4263 && ((TYPE_PRECISION (TREE_TYPE (scalar_dest))
4264 != GET_MODE_PRECISION (TYPE_MODE (TREE_TYPE (scalar_dest))))
4265 || ((TYPE_PRECISION (TREE_TYPE (op))
4266 != GET_MODE_PRECISION (TYPE_MODE (TREE_TYPE (op))))))
4267 /* But a conversion that does not change the bit-pattern is ok. */
4268 && !((TYPE_PRECISION (TREE_TYPE (scalar_dest))
4269 > TYPE_PRECISION (TREE_TYPE (op)))
4270 && TYPE_UNSIGNED (TREE_TYPE (op)))
4271 /* Conversion between boolean types of different sizes is
4272 a simple assignment in case their vectypes are same
4273 boolean vectors. */
4274 && (!VECTOR_BOOLEAN_TYPE_P (vectype)
4275 || !VECTOR_BOOLEAN_TYPE_P (vectype_in)))
4277 if (dump_enabled_p ())
4278 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4279 "type conversion to/from bit-precision "
4280 "unsupported.\n");
4281 return false;
4284 if (!vec_stmt) /* transformation not required. */
4286 STMT_VINFO_TYPE (stmt_info) = assignment_vec_info_type;
4287 if (dump_enabled_p ())
4288 dump_printf_loc (MSG_NOTE, vect_location,
4289 "=== vectorizable_assignment ===\n");
4290 vect_model_simple_cost (stmt_info, ncopies, dt, NULL, NULL);
4291 return true;
4294 /** Transform. **/
4295 if (dump_enabled_p ())
4296 dump_printf_loc (MSG_NOTE, vect_location, "transform assignment.\n");
4298 /* Handle def. */
4299 vec_dest = vect_create_destination_var (scalar_dest, vectype);
4301 /* Handle use. */
4302 for (j = 0; j < ncopies; j++)
4304 /* Handle uses. */
4305 if (j == 0)
4306 vect_get_vec_defs (op, NULL, stmt, &vec_oprnds, NULL, slp_node, -1);
4307 else
4308 vect_get_vec_defs_for_stmt_copy (dt, &vec_oprnds, NULL);
4310 /* Arguments are ready. create the new vector stmt. */
4311 FOR_EACH_VEC_ELT (vec_oprnds, i, vop)
4313 if (CONVERT_EXPR_CODE_P (code)
4314 || code == VIEW_CONVERT_EXPR)
4315 vop = build1 (VIEW_CONVERT_EXPR, vectype, vop);
4316 new_stmt = gimple_build_assign (vec_dest, vop);
4317 new_temp = make_ssa_name (vec_dest, new_stmt);
4318 gimple_assign_set_lhs (new_stmt, new_temp);
4319 vect_finish_stmt_generation (stmt, new_stmt, gsi);
4320 if (slp_node)
4321 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
4324 if (slp_node)
4325 continue;
4327 if (j == 0)
4328 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
4329 else
4330 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
4332 prev_stmt_info = vinfo_for_stmt (new_stmt);
4335 vec_oprnds.release ();
4336 return true;
4340 /* Return TRUE if CODE (a shift operation) is supported for SCALAR_TYPE
4341 either as shift by a scalar or by a vector. */
4343 bool
4344 vect_supportable_shift (enum tree_code code, tree scalar_type)
4347 machine_mode vec_mode;
4348 optab optab;
4349 int icode;
4350 tree vectype;
4352 vectype = get_vectype_for_scalar_type (scalar_type);
4353 if (!vectype)
4354 return false;
4356 optab = optab_for_tree_code (code, vectype, optab_scalar);
4357 if (!optab
4358 || optab_handler (optab, TYPE_MODE (vectype)) == CODE_FOR_nothing)
4360 optab = optab_for_tree_code (code, vectype, optab_vector);
4361 if (!optab
4362 || (optab_handler (optab, TYPE_MODE (vectype))
4363 == CODE_FOR_nothing))
4364 return false;
4367 vec_mode = TYPE_MODE (vectype);
4368 icode = (int) optab_handler (optab, vec_mode);
4369 if (icode == CODE_FOR_nothing)
4370 return false;
4372 return true;
4376 /* Function vectorizable_shift.
4378 Check if STMT performs a shift operation that can be vectorized.
4379 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
4380 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
4381 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
4383 static bool
4384 vectorizable_shift (gimple *stmt, gimple_stmt_iterator *gsi,
4385 gimple **vec_stmt, slp_tree slp_node)
4387 tree vec_dest;
4388 tree scalar_dest;
4389 tree op0, op1 = NULL;
4390 tree vec_oprnd1 = NULL_TREE;
4391 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
4392 tree vectype;
4393 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
4394 enum tree_code code;
4395 machine_mode vec_mode;
4396 tree new_temp;
4397 optab optab;
4398 int icode;
4399 machine_mode optab_op2_mode;
4400 gimple *def_stmt;
4401 enum vect_def_type dt[2] = {vect_unknown_def_type, vect_unknown_def_type};
4402 gimple *new_stmt = NULL;
4403 stmt_vec_info prev_stmt_info;
4404 int nunits_in;
4405 int nunits_out;
4406 tree vectype_out;
4407 tree op1_vectype;
4408 int ncopies;
4409 int j, i;
4410 vec<tree> vec_oprnds0 = vNULL;
4411 vec<tree> vec_oprnds1 = vNULL;
4412 tree vop0, vop1;
4413 unsigned int k;
4414 bool scalar_shift_arg = true;
4415 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
4416 vec_info *vinfo = stmt_info->vinfo;
4417 int vf;
4419 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
4420 return false;
4422 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def
4423 && ! vec_stmt)
4424 return false;
4426 /* Is STMT a vectorizable binary/unary operation? */
4427 if (!is_gimple_assign (stmt))
4428 return false;
4430 if (TREE_CODE (gimple_assign_lhs (stmt)) != SSA_NAME)
4431 return false;
4433 code = gimple_assign_rhs_code (stmt);
4435 if (!(code == LSHIFT_EXPR || code == RSHIFT_EXPR || code == LROTATE_EXPR
4436 || code == RROTATE_EXPR))
4437 return false;
4439 scalar_dest = gimple_assign_lhs (stmt);
4440 vectype_out = STMT_VINFO_VECTYPE (stmt_info);
4441 if (TYPE_PRECISION (TREE_TYPE (scalar_dest))
4442 != GET_MODE_PRECISION (TYPE_MODE (TREE_TYPE (scalar_dest))))
4444 if (dump_enabled_p ())
4445 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4446 "bit-precision shifts not supported.\n");
4447 return false;
4450 op0 = gimple_assign_rhs1 (stmt);
4451 if (!vect_is_simple_use (op0, vinfo, &def_stmt, &dt[0], &vectype))
4453 if (dump_enabled_p ())
4454 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4455 "use not simple.\n");
4456 return false;
4458 /* If op0 is an external or constant def use a vector type with
4459 the same size as the output vector type. */
4460 if (!vectype)
4461 vectype = get_same_sized_vectype (TREE_TYPE (op0), vectype_out);
4462 if (vec_stmt)
4463 gcc_assert (vectype);
4464 if (!vectype)
4466 if (dump_enabled_p ())
4467 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4468 "no vectype for scalar type\n");
4469 return false;
4472 nunits_out = TYPE_VECTOR_SUBPARTS (vectype_out);
4473 nunits_in = TYPE_VECTOR_SUBPARTS (vectype);
4474 if (nunits_out != nunits_in)
4475 return false;
4477 op1 = gimple_assign_rhs2 (stmt);
4478 if (!vect_is_simple_use (op1, vinfo, &def_stmt, &dt[1], &op1_vectype))
4480 if (dump_enabled_p ())
4481 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4482 "use not simple.\n");
4483 return false;
4486 if (loop_vinfo)
4487 vf = LOOP_VINFO_VECT_FACTOR (loop_vinfo);
4488 else
4489 vf = 1;
4491 /* Multiple types in SLP are handled by creating the appropriate number of
4492 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
4493 case of SLP. */
4494 if (slp_node || PURE_SLP_STMT (stmt_info))
4495 ncopies = 1;
4496 else
4497 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_in;
4499 gcc_assert (ncopies >= 1);
4501 /* Determine whether the shift amount is a vector, or scalar. If the
4502 shift/rotate amount is a vector, use the vector/vector shift optabs. */
4504 if ((dt[1] == vect_internal_def
4505 || dt[1] == vect_induction_def)
4506 && !slp_node)
4507 scalar_shift_arg = false;
4508 else if (dt[1] == vect_constant_def
4509 || dt[1] == vect_external_def
4510 || dt[1] == vect_internal_def)
4512 /* In SLP, need to check whether the shift count is the same,
4513 in loops if it is a constant or invariant, it is always
4514 a scalar shift. */
4515 if (slp_node)
4517 vec<gimple *> stmts = SLP_TREE_SCALAR_STMTS (slp_node);
4518 gimple *slpstmt;
4520 FOR_EACH_VEC_ELT (stmts, k, slpstmt)
4521 if (!operand_equal_p (gimple_assign_rhs2 (slpstmt), op1, 0))
4522 scalar_shift_arg = false;
4525 /* If the shift amount is computed by a pattern stmt we cannot
4526 use the scalar amount directly thus give up and use a vector
4527 shift. */
4528 if (dt[1] == vect_internal_def)
4530 gimple *def = SSA_NAME_DEF_STMT (op1);
4531 if (is_pattern_stmt_p (vinfo_for_stmt (def)))
4532 scalar_shift_arg = false;
4535 else
4537 if (dump_enabled_p ())
4538 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4539 "operand mode requires invariant argument.\n");
4540 return false;
4543 /* Vector shifted by vector. */
4544 if (!scalar_shift_arg)
4546 optab = optab_for_tree_code (code, vectype, optab_vector);
4547 if (dump_enabled_p ())
4548 dump_printf_loc (MSG_NOTE, vect_location,
4549 "vector/vector shift/rotate found.\n");
4551 if (!op1_vectype)
4552 op1_vectype = get_same_sized_vectype (TREE_TYPE (op1), vectype_out);
4553 if (op1_vectype == NULL_TREE
4554 || TYPE_MODE (op1_vectype) != TYPE_MODE (vectype))
4556 if (dump_enabled_p ())
4557 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4558 "unusable type for last operand in"
4559 " vector/vector shift/rotate.\n");
4560 return false;
4563 /* See if the machine has a vector shifted by scalar insn and if not
4564 then see if it has a vector shifted by vector insn. */
4565 else
4567 optab = optab_for_tree_code (code, vectype, optab_scalar);
4568 if (optab
4569 && optab_handler (optab, TYPE_MODE (vectype)) != CODE_FOR_nothing)
4571 if (dump_enabled_p ())
4572 dump_printf_loc (MSG_NOTE, vect_location,
4573 "vector/scalar shift/rotate found.\n");
4575 else
4577 optab = optab_for_tree_code (code, vectype, optab_vector);
4578 if (optab
4579 && (optab_handler (optab, TYPE_MODE (vectype))
4580 != CODE_FOR_nothing))
4582 scalar_shift_arg = false;
4584 if (dump_enabled_p ())
4585 dump_printf_loc (MSG_NOTE, vect_location,
4586 "vector/vector shift/rotate found.\n");
4588 /* Unlike the other binary operators, shifts/rotates have
4589 the rhs being int, instead of the same type as the lhs,
4590 so make sure the scalar is the right type if we are
4591 dealing with vectors of long long/long/short/char. */
4592 if (dt[1] == vect_constant_def)
4593 op1 = fold_convert (TREE_TYPE (vectype), op1);
4594 else if (!useless_type_conversion_p (TREE_TYPE (vectype),
4595 TREE_TYPE (op1)))
4597 if (slp_node
4598 && TYPE_MODE (TREE_TYPE (vectype))
4599 != TYPE_MODE (TREE_TYPE (op1)))
4601 if (dump_enabled_p ())
4602 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4603 "unusable type for last operand in"
4604 " vector/vector shift/rotate.\n");
4605 return false;
4607 if (vec_stmt && !slp_node)
4609 op1 = fold_convert (TREE_TYPE (vectype), op1);
4610 op1 = vect_init_vector (stmt, op1,
4611 TREE_TYPE (vectype), NULL);
4618 /* Supportable by target? */
4619 if (!optab)
4621 if (dump_enabled_p ())
4622 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4623 "no optab.\n");
4624 return false;
4626 vec_mode = TYPE_MODE (vectype);
4627 icode = (int) optab_handler (optab, vec_mode);
4628 if (icode == CODE_FOR_nothing)
4630 if (dump_enabled_p ())
4631 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4632 "op not supported by target.\n");
4633 /* Check only during analysis. */
4634 if (GET_MODE_SIZE (vec_mode) != UNITS_PER_WORD
4635 || (vf < vect_min_worthwhile_factor (code)
4636 && !vec_stmt))
4637 return false;
4638 if (dump_enabled_p ())
4639 dump_printf_loc (MSG_NOTE, vect_location,
4640 "proceeding using word mode.\n");
4643 /* Worthwhile without SIMD support? Check only during analysis. */
4644 if (!VECTOR_MODE_P (TYPE_MODE (vectype))
4645 && vf < vect_min_worthwhile_factor (code)
4646 && !vec_stmt)
4648 if (dump_enabled_p ())
4649 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4650 "not worthwhile without SIMD support.\n");
4651 return false;
4654 if (!vec_stmt) /* transformation not required. */
4656 STMT_VINFO_TYPE (stmt_info) = shift_vec_info_type;
4657 if (dump_enabled_p ())
4658 dump_printf_loc (MSG_NOTE, vect_location,
4659 "=== vectorizable_shift ===\n");
4660 vect_model_simple_cost (stmt_info, ncopies, dt, NULL, NULL);
4661 return true;
4664 /** Transform. **/
4666 if (dump_enabled_p ())
4667 dump_printf_loc (MSG_NOTE, vect_location,
4668 "transform binary/unary operation.\n");
4670 /* Handle def. */
4671 vec_dest = vect_create_destination_var (scalar_dest, vectype);
4673 prev_stmt_info = NULL;
4674 for (j = 0; j < ncopies; j++)
4676 /* Handle uses. */
4677 if (j == 0)
4679 if (scalar_shift_arg)
4681 /* Vector shl and shr insn patterns can be defined with scalar
4682 operand 2 (shift operand). In this case, use constant or loop
4683 invariant op1 directly, without extending it to vector mode
4684 first. */
4685 optab_op2_mode = insn_data[icode].operand[2].mode;
4686 if (!VECTOR_MODE_P (optab_op2_mode))
4688 if (dump_enabled_p ())
4689 dump_printf_loc (MSG_NOTE, vect_location,
4690 "operand 1 using scalar mode.\n");
4691 vec_oprnd1 = op1;
4692 vec_oprnds1.create (slp_node ? slp_node->vec_stmts_size : 1);
4693 vec_oprnds1.quick_push (vec_oprnd1);
4694 if (slp_node)
4696 /* Store vec_oprnd1 for every vector stmt to be created
4697 for SLP_NODE. We check during the analysis that all
4698 the shift arguments are the same.
4699 TODO: Allow different constants for different vector
4700 stmts generated for an SLP instance. */
4701 for (k = 0; k < slp_node->vec_stmts_size - 1; k++)
4702 vec_oprnds1.quick_push (vec_oprnd1);
4707 /* vec_oprnd1 is available if operand 1 should be of a scalar-type
4708 (a special case for certain kind of vector shifts); otherwise,
4709 operand 1 should be of a vector type (the usual case). */
4710 if (vec_oprnd1)
4711 vect_get_vec_defs (op0, NULL_TREE, stmt, &vec_oprnds0, NULL,
4712 slp_node, -1);
4713 else
4714 vect_get_vec_defs (op0, op1, stmt, &vec_oprnds0, &vec_oprnds1,
4715 slp_node, -1);
4717 else
4718 vect_get_vec_defs_for_stmt_copy (dt, &vec_oprnds0, &vec_oprnds1);
4720 /* Arguments are ready. Create the new vector stmt. */
4721 FOR_EACH_VEC_ELT (vec_oprnds0, i, vop0)
4723 vop1 = vec_oprnds1[i];
4724 new_stmt = gimple_build_assign (vec_dest, code, vop0, vop1);
4725 new_temp = make_ssa_name (vec_dest, new_stmt);
4726 gimple_assign_set_lhs (new_stmt, new_temp);
4727 vect_finish_stmt_generation (stmt, new_stmt, gsi);
4728 if (slp_node)
4729 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
4732 if (slp_node)
4733 continue;
4735 if (j == 0)
4736 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
4737 else
4738 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
4739 prev_stmt_info = vinfo_for_stmt (new_stmt);
4742 vec_oprnds0.release ();
4743 vec_oprnds1.release ();
4745 return true;
4749 /* Function vectorizable_operation.
4751 Check if STMT performs a binary, unary or ternary operation that can
4752 be vectorized.
4753 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
4754 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
4755 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
4757 static bool
4758 vectorizable_operation (gimple *stmt, gimple_stmt_iterator *gsi,
4759 gimple **vec_stmt, slp_tree slp_node)
4761 tree vec_dest;
4762 tree scalar_dest;
4763 tree op0, op1 = NULL_TREE, op2 = NULL_TREE;
4764 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
4765 tree vectype;
4766 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
4767 enum tree_code code;
4768 machine_mode vec_mode;
4769 tree new_temp;
4770 int op_type;
4771 optab optab;
4772 bool target_support_p;
4773 gimple *def_stmt;
4774 enum vect_def_type dt[3]
4775 = {vect_unknown_def_type, vect_unknown_def_type, vect_unknown_def_type};
4776 gimple *new_stmt = NULL;
4777 stmt_vec_info prev_stmt_info;
4778 int nunits_in;
4779 int nunits_out;
4780 tree vectype_out;
4781 int ncopies;
4782 int j, i;
4783 vec<tree> vec_oprnds0 = vNULL;
4784 vec<tree> vec_oprnds1 = vNULL;
4785 vec<tree> vec_oprnds2 = vNULL;
4786 tree vop0, vop1, vop2;
4787 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
4788 vec_info *vinfo = stmt_info->vinfo;
4789 int vf;
4791 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
4792 return false;
4794 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def
4795 && ! vec_stmt)
4796 return false;
4798 /* Is STMT a vectorizable binary/unary operation? */
4799 if (!is_gimple_assign (stmt))
4800 return false;
4802 if (TREE_CODE (gimple_assign_lhs (stmt)) != SSA_NAME)
4803 return false;
4805 code = gimple_assign_rhs_code (stmt);
4807 /* For pointer addition, we should use the normal plus for
4808 the vector addition. */
4809 if (code == POINTER_PLUS_EXPR)
4810 code = PLUS_EXPR;
4812 /* Support only unary or binary operations. */
4813 op_type = TREE_CODE_LENGTH (code);
4814 if (op_type != unary_op && op_type != binary_op && op_type != ternary_op)
4816 if (dump_enabled_p ())
4817 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4818 "num. args = %d (not unary/binary/ternary op).\n",
4819 op_type);
4820 return false;
4823 scalar_dest = gimple_assign_lhs (stmt);
4824 vectype_out = STMT_VINFO_VECTYPE (stmt_info);
4826 /* Most operations cannot handle bit-precision types without extra
4827 truncations. */
4828 if (!VECTOR_BOOLEAN_TYPE_P (vectype_out)
4829 && (TYPE_PRECISION (TREE_TYPE (scalar_dest))
4830 != GET_MODE_PRECISION (TYPE_MODE (TREE_TYPE (scalar_dest))))
4831 /* Exception are bitwise binary operations. */
4832 && code != BIT_IOR_EXPR
4833 && code != BIT_XOR_EXPR
4834 && code != BIT_AND_EXPR)
4836 if (dump_enabled_p ())
4837 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4838 "bit-precision arithmetic not supported.\n");
4839 return false;
4842 op0 = gimple_assign_rhs1 (stmt);
4843 if (!vect_is_simple_use (op0, vinfo, &def_stmt, &dt[0], &vectype))
4845 if (dump_enabled_p ())
4846 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4847 "use not simple.\n");
4848 return false;
4850 /* If op0 is an external or constant def use a vector type with
4851 the same size as the output vector type. */
4852 if (!vectype)
4854 /* For boolean type we cannot determine vectype by
4855 invariant value (don't know whether it is a vector
4856 of booleans or vector of integers). We use output
4857 vectype because operations on boolean don't change
4858 type. */
4859 if (TREE_CODE (TREE_TYPE (op0)) == BOOLEAN_TYPE)
4861 if (TREE_CODE (TREE_TYPE (scalar_dest)) != BOOLEAN_TYPE)
4863 if (dump_enabled_p ())
4864 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4865 "not supported operation on bool value.\n");
4866 return false;
4868 vectype = vectype_out;
4870 else
4871 vectype = get_same_sized_vectype (TREE_TYPE (op0), vectype_out);
4873 if (vec_stmt)
4874 gcc_assert (vectype);
4875 if (!vectype)
4877 if (dump_enabled_p ())
4879 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4880 "no vectype for scalar type ");
4881 dump_generic_expr (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
4882 TREE_TYPE (op0));
4883 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
4886 return false;
4889 nunits_out = TYPE_VECTOR_SUBPARTS (vectype_out);
4890 nunits_in = TYPE_VECTOR_SUBPARTS (vectype);
4891 if (nunits_out != nunits_in)
4892 return false;
4894 if (op_type == binary_op || op_type == ternary_op)
4896 op1 = gimple_assign_rhs2 (stmt);
4897 if (!vect_is_simple_use (op1, vinfo, &def_stmt, &dt[1]))
4899 if (dump_enabled_p ())
4900 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4901 "use not simple.\n");
4902 return false;
4905 if (op_type == ternary_op)
4907 op2 = gimple_assign_rhs3 (stmt);
4908 if (!vect_is_simple_use (op2, vinfo, &def_stmt, &dt[2]))
4910 if (dump_enabled_p ())
4911 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4912 "use not simple.\n");
4913 return false;
4917 if (loop_vinfo)
4918 vf = LOOP_VINFO_VECT_FACTOR (loop_vinfo);
4919 else
4920 vf = 1;
4922 /* Multiple types in SLP are handled by creating the appropriate number of
4923 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
4924 case of SLP. */
4925 if (slp_node || PURE_SLP_STMT (stmt_info))
4926 ncopies = 1;
4927 else
4928 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_in;
4930 gcc_assert (ncopies >= 1);
4932 /* Shifts are handled in vectorizable_shift (). */
4933 if (code == LSHIFT_EXPR || code == RSHIFT_EXPR || code == LROTATE_EXPR
4934 || code == RROTATE_EXPR)
4935 return false;
4937 /* Supportable by target? */
4939 vec_mode = TYPE_MODE (vectype);
4940 if (code == MULT_HIGHPART_EXPR)
4941 target_support_p = can_mult_highpart_p (vec_mode, TYPE_UNSIGNED (vectype));
4942 else
4944 optab = optab_for_tree_code (code, vectype, optab_default);
4945 if (!optab)
4947 if (dump_enabled_p ())
4948 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4949 "no optab.\n");
4950 return false;
4952 target_support_p = (optab_handler (optab, vec_mode)
4953 != CODE_FOR_nothing);
4956 if (!target_support_p)
4958 if (dump_enabled_p ())
4959 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4960 "op not supported by target.\n");
4961 /* Check only during analysis. */
4962 if (GET_MODE_SIZE (vec_mode) != UNITS_PER_WORD
4963 || (!vec_stmt && vf < vect_min_worthwhile_factor (code)))
4964 return false;
4965 if (dump_enabled_p ())
4966 dump_printf_loc (MSG_NOTE, vect_location,
4967 "proceeding using word mode.\n");
4970 /* Worthwhile without SIMD support? Check only during analysis. */
4971 if (!VECTOR_MODE_P (vec_mode)
4972 && !vec_stmt
4973 && vf < vect_min_worthwhile_factor (code))
4975 if (dump_enabled_p ())
4976 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
4977 "not worthwhile without SIMD support.\n");
4978 return false;
4981 if (!vec_stmt) /* transformation not required. */
4983 STMT_VINFO_TYPE (stmt_info) = op_vec_info_type;
4984 if (dump_enabled_p ())
4985 dump_printf_loc (MSG_NOTE, vect_location,
4986 "=== vectorizable_operation ===\n");
4987 vect_model_simple_cost (stmt_info, ncopies, dt, NULL, NULL);
4988 return true;
4991 /** Transform. **/
4993 if (dump_enabled_p ())
4994 dump_printf_loc (MSG_NOTE, vect_location,
4995 "transform binary/unary operation.\n");
4997 /* Handle def. */
4998 vec_dest = vect_create_destination_var (scalar_dest, vectype);
5000 /* In case the vectorization factor (VF) is bigger than the number
5001 of elements that we can fit in a vectype (nunits), we have to generate
5002 more than one vector stmt - i.e - we need to "unroll" the
5003 vector stmt by a factor VF/nunits. In doing so, we record a pointer
5004 from one copy of the vector stmt to the next, in the field
5005 STMT_VINFO_RELATED_STMT. This is necessary in order to allow following
5006 stages to find the correct vector defs to be used when vectorizing
5007 stmts that use the defs of the current stmt. The example below
5008 illustrates the vectorization process when VF=16 and nunits=4 (i.e.,
5009 we need to create 4 vectorized stmts):
5011 before vectorization:
5012 RELATED_STMT VEC_STMT
5013 S1: x = memref - -
5014 S2: z = x + 1 - -
5016 step 1: vectorize stmt S1 (done in vectorizable_load. See more details
5017 there):
5018 RELATED_STMT VEC_STMT
5019 VS1_0: vx0 = memref0 VS1_1 -
5020 VS1_1: vx1 = memref1 VS1_2 -
5021 VS1_2: vx2 = memref2 VS1_3 -
5022 VS1_3: vx3 = memref3 - -
5023 S1: x = load - VS1_0
5024 S2: z = x + 1 - -
5026 step2: vectorize stmt S2 (done here):
5027 To vectorize stmt S2 we first need to find the relevant vector
5028 def for the first operand 'x'. This is, as usual, obtained from
5029 the vector stmt recorded in the STMT_VINFO_VEC_STMT of the stmt
5030 that defines 'x' (S1). This way we find the stmt VS1_0, and the
5031 relevant vector def 'vx0'. Having found 'vx0' we can generate
5032 the vector stmt VS2_0, and as usual, record it in the
5033 STMT_VINFO_VEC_STMT of stmt S2.
5034 When creating the second copy (VS2_1), we obtain the relevant vector
5035 def from the vector stmt recorded in the STMT_VINFO_RELATED_STMT of
5036 stmt VS1_0. This way we find the stmt VS1_1 and the relevant
5037 vector def 'vx1'. Using 'vx1' we create stmt VS2_1 and record a
5038 pointer to it in the STMT_VINFO_RELATED_STMT of the vector stmt VS2_0.
5039 Similarly when creating stmts VS2_2 and VS2_3. This is the resulting
5040 chain of stmts and pointers:
5041 RELATED_STMT VEC_STMT
5042 VS1_0: vx0 = memref0 VS1_1 -
5043 VS1_1: vx1 = memref1 VS1_2 -
5044 VS1_2: vx2 = memref2 VS1_3 -
5045 VS1_3: vx3 = memref3 - -
5046 S1: x = load - VS1_0
5047 VS2_0: vz0 = vx0 + v1 VS2_1 -
5048 VS2_1: vz1 = vx1 + v1 VS2_2 -
5049 VS2_2: vz2 = vx2 + v1 VS2_3 -
5050 VS2_3: vz3 = vx3 + v1 - -
5051 S2: z = x + 1 - VS2_0 */
5053 prev_stmt_info = NULL;
5054 for (j = 0; j < ncopies; j++)
5056 /* Handle uses. */
5057 if (j == 0)
5059 if (op_type == binary_op || op_type == ternary_op)
5060 vect_get_vec_defs (op0, op1, stmt, &vec_oprnds0, &vec_oprnds1,
5061 slp_node, -1);
5062 else
5063 vect_get_vec_defs (op0, NULL_TREE, stmt, &vec_oprnds0, NULL,
5064 slp_node, -1);
5065 if (op_type == ternary_op)
5067 vec_oprnds2.create (1);
5068 vec_oprnds2.quick_push (vect_get_vec_def_for_operand (op2,
5069 stmt));
5072 else
5074 vect_get_vec_defs_for_stmt_copy (dt, &vec_oprnds0, &vec_oprnds1);
5075 if (op_type == ternary_op)
5077 tree vec_oprnd = vec_oprnds2.pop ();
5078 vec_oprnds2.quick_push (vect_get_vec_def_for_stmt_copy (dt[2],
5079 vec_oprnd));
5083 /* Arguments are ready. Create the new vector stmt. */
5084 FOR_EACH_VEC_ELT (vec_oprnds0, i, vop0)
5086 vop1 = ((op_type == binary_op || op_type == ternary_op)
5087 ? vec_oprnds1[i] : NULL_TREE);
5088 vop2 = ((op_type == ternary_op)
5089 ? vec_oprnds2[i] : NULL_TREE);
5090 new_stmt = gimple_build_assign (vec_dest, code, vop0, vop1, vop2);
5091 new_temp = make_ssa_name (vec_dest, new_stmt);
5092 gimple_assign_set_lhs (new_stmt, new_temp);
5093 vect_finish_stmt_generation (stmt, new_stmt, gsi);
5094 if (slp_node)
5095 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
5098 if (slp_node)
5099 continue;
5101 if (j == 0)
5102 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
5103 else
5104 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
5105 prev_stmt_info = vinfo_for_stmt (new_stmt);
5108 vec_oprnds0.release ();
5109 vec_oprnds1.release ();
5110 vec_oprnds2.release ();
5112 return true;
5115 /* A helper function to ensure data reference DR's base alignment
5116 for STMT_INFO. */
5118 static void
5119 ensure_base_align (stmt_vec_info stmt_info, struct data_reference *dr)
5121 if (!dr->aux)
5122 return;
5124 if (DR_VECT_AUX (dr)->base_misaligned)
5126 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
5127 tree base_decl = DR_VECT_AUX (dr)->base_decl;
5129 if (decl_in_symtab_p (base_decl))
5130 symtab_node::get (base_decl)->increase_alignment (TYPE_ALIGN (vectype));
5131 else
5133 SET_DECL_ALIGN (base_decl, TYPE_ALIGN (vectype));
5134 DECL_USER_ALIGN (base_decl) = 1;
5136 DR_VECT_AUX (dr)->base_misaligned = false;
5141 /* Given a vector type VECTYPE returns the VECTOR_CST mask that implements
5142 reversal of the vector elements. If that is impossible to do,
5143 returns NULL. */
5145 static tree
5146 perm_mask_for_reverse (tree vectype)
5148 int i, nunits;
5149 unsigned char *sel;
5151 nunits = TYPE_VECTOR_SUBPARTS (vectype);
5152 sel = XALLOCAVEC (unsigned char, nunits);
5154 for (i = 0; i < nunits; ++i)
5155 sel[i] = nunits - 1 - i;
5157 if (!can_vec_perm_p (TYPE_MODE (vectype), false, sel))
5158 return NULL_TREE;
5159 return vect_gen_perm_mask_checked (vectype, sel);
5162 /* Function vectorizable_store.
5164 Check if STMT defines a non scalar data-ref (array/pointer/structure) that
5165 can be vectorized.
5166 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
5167 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
5168 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
5170 static bool
5171 vectorizable_store (gimple *stmt, gimple_stmt_iterator *gsi, gimple **vec_stmt,
5172 slp_tree slp_node)
5174 tree scalar_dest;
5175 tree data_ref;
5176 tree op;
5177 tree vec_oprnd = NULL_TREE;
5178 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
5179 struct data_reference *dr = STMT_VINFO_DATA_REF (stmt_info), *first_dr = NULL;
5180 tree elem_type;
5181 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
5182 struct loop *loop = NULL;
5183 machine_mode vec_mode;
5184 tree dummy;
5185 enum dr_alignment_support alignment_support_scheme;
5186 gimple *def_stmt;
5187 enum vect_def_type dt;
5188 stmt_vec_info prev_stmt_info = NULL;
5189 tree dataref_ptr = NULL_TREE;
5190 tree dataref_offset = NULL_TREE;
5191 gimple *ptr_incr = NULL;
5192 int ncopies;
5193 int j;
5194 gimple *next_stmt, *first_stmt = NULL;
5195 bool grouped_store = false;
5196 bool store_lanes_p = false;
5197 unsigned int group_size, i;
5198 vec<tree> dr_chain = vNULL;
5199 vec<tree> oprnds = vNULL;
5200 vec<tree> result_chain = vNULL;
5201 bool inv_p;
5202 bool negative = false;
5203 tree offset = NULL_TREE;
5204 vec<tree> vec_oprnds = vNULL;
5205 bool slp = (slp_node != NULL);
5206 unsigned int vec_num;
5207 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
5208 vec_info *vinfo = stmt_info->vinfo;
5209 tree aggr_type;
5210 tree scatter_base = NULL_TREE, scatter_off = NULL_TREE;
5211 tree scatter_off_vectype = NULL_TREE, scatter_decl = NULL_TREE;
5212 int scatter_scale = 1;
5213 enum vect_def_type scatter_idx_dt = vect_unknown_def_type;
5214 enum vect_def_type scatter_src_dt = vect_unknown_def_type;
5215 gimple *new_stmt;
5217 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
5218 return false;
5220 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def
5221 && ! vec_stmt)
5222 return false;
5224 /* Is vectorizable store? */
5226 if (!is_gimple_assign (stmt))
5227 return false;
5229 scalar_dest = gimple_assign_lhs (stmt);
5230 if (TREE_CODE (scalar_dest) == VIEW_CONVERT_EXPR
5231 && is_pattern_stmt_p (stmt_info))
5232 scalar_dest = TREE_OPERAND (scalar_dest, 0);
5233 if (TREE_CODE (scalar_dest) != ARRAY_REF
5234 && TREE_CODE (scalar_dest) != BIT_FIELD_REF
5235 && TREE_CODE (scalar_dest) != INDIRECT_REF
5236 && TREE_CODE (scalar_dest) != COMPONENT_REF
5237 && TREE_CODE (scalar_dest) != IMAGPART_EXPR
5238 && TREE_CODE (scalar_dest) != REALPART_EXPR
5239 && TREE_CODE (scalar_dest) != MEM_REF)
5240 return false;
5242 gcc_assert (gimple_assign_single_p (stmt));
5244 tree vectype = STMT_VINFO_VECTYPE (stmt_info), rhs_vectype = NULL_TREE;
5245 unsigned int nunits = TYPE_VECTOR_SUBPARTS (vectype);
5247 if (loop_vinfo)
5248 loop = LOOP_VINFO_LOOP (loop_vinfo);
5250 /* Multiple types in SLP are handled by creating the appropriate number of
5251 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
5252 case of SLP. */
5253 if (slp || PURE_SLP_STMT (stmt_info))
5254 ncopies = 1;
5255 else
5256 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits;
5258 gcc_assert (ncopies >= 1);
5260 /* FORNOW. This restriction should be relaxed. */
5261 if (loop && nested_in_vect_loop_p (loop, stmt) && ncopies > 1)
5263 if (dump_enabled_p ())
5264 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5265 "multiple types in nested loop.\n");
5266 return false;
5269 op = gimple_assign_rhs1 (stmt);
5271 if (!vect_is_simple_use (op, vinfo, &def_stmt, &dt, &rhs_vectype))
5273 if (dump_enabled_p ())
5274 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5275 "use not simple.\n");
5276 return false;
5279 if (rhs_vectype && !useless_type_conversion_p (vectype, rhs_vectype))
5280 return false;
5282 elem_type = TREE_TYPE (vectype);
5283 vec_mode = TYPE_MODE (vectype);
5285 /* FORNOW. In some cases can vectorize even if data-type not supported
5286 (e.g. - array initialization with 0). */
5287 if (optab_handler (mov_optab, vec_mode) == CODE_FOR_nothing)
5288 return false;
5290 if (!STMT_VINFO_DATA_REF (stmt_info))
5291 return false;
5293 if (!STMT_VINFO_STRIDED_P (stmt_info))
5295 negative =
5296 tree_int_cst_compare (loop && nested_in_vect_loop_p (loop, stmt)
5297 ? STMT_VINFO_DR_STEP (stmt_info) : DR_STEP (dr),
5298 size_zero_node) < 0;
5299 if (negative && ncopies > 1)
5301 if (dump_enabled_p ())
5302 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5303 "multiple types with negative step.\n");
5304 return false;
5306 if (negative)
5308 gcc_assert (!grouped_store);
5309 alignment_support_scheme = vect_supportable_dr_alignment (dr, false);
5310 if (alignment_support_scheme != dr_aligned
5311 && alignment_support_scheme != dr_unaligned_supported)
5313 if (dump_enabled_p ())
5314 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5315 "negative step but alignment required.\n");
5316 return false;
5318 if (dt != vect_constant_def
5319 && dt != vect_external_def
5320 && !perm_mask_for_reverse (vectype))
5322 if (dump_enabled_p ())
5323 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5324 "negative step and reversing not supported.\n");
5325 return false;
5330 if (STMT_VINFO_GROUPED_ACCESS (stmt_info))
5332 grouped_store = true;
5333 first_stmt = GROUP_FIRST_ELEMENT (stmt_info);
5334 group_size = GROUP_SIZE (vinfo_for_stmt (first_stmt));
5335 if (!slp
5336 && !PURE_SLP_STMT (stmt_info)
5337 && !STMT_VINFO_STRIDED_P (stmt_info))
5339 if (vect_store_lanes_supported (vectype, group_size))
5340 store_lanes_p = true;
5341 else if (!vect_grouped_store_supported (vectype, group_size))
5342 return false;
5345 if (STMT_VINFO_STRIDED_P (stmt_info)
5346 && (slp || PURE_SLP_STMT (stmt_info))
5347 && (group_size > nunits
5348 || nunits % group_size != 0))
5350 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5351 "unhandled strided group store\n");
5352 return false;
5355 if (first_stmt == stmt)
5357 /* STMT is the leader of the group. Check the operands of all the
5358 stmts of the group. */
5359 next_stmt = GROUP_NEXT_ELEMENT (stmt_info);
5360 while (next_stmt)
5362 gcc_assert (gimple_assign_single_p (next_stmt));
5363 op = gimple_assign_rhs1 (next_stmt);
5364 if (!vect_is_simple_use (op, vinfo, &def_stmt, &dt))
5366 if (dump_enabled_p ())
5367 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5368 "use not simple.\n");
5369 return false;
5371 next_stmt = GROUP_NEXT_ELEMENT (vinfo_for_stmt (next_stmt));
5376 if (STMT_VINFO_GATHER_SCATTER_P (stmt_info))
5378 gimple *def_stmt;
5379 scatter_decl = vect_check_gather_scatter (stmt, loop_vinfo, &scatter_base,
5380 &scatter_off, &scatter_scale);
5381 gcc_assert (scatter_decl);
5382 if (!vect_is_simple_use (scatter_off, vinfo, &def_stmt, &scatter_idx_dt,
5383 &scatter_off_vectype))
5385 if (dump_enabled_p ())
5386 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
5387 "scatter index use not simple.");
5388 return false;
5392 if (!vec_stmt) /* transformation not required. */
5394 STMT_VINFO_TYPE (stmt_info) = store_vec_info_type;
5395 /* The SLP costs are calculated during SLP analysis. */
5396 if (!PURE_SLP_STMT (stmt_info))
5397 vect_model_store_cost (stmt_info, ncopies, store_lanes_p, dt,
5398 NULL, NULL, NULL);
5399 return true;
5402 /** Transform. **/
5404 ensure_base_align (stmt_info, dr);
5406 if (STMT_VINFO_GATHER_SCATTER_P (stmt_info))
5408 tree vec_oprnd0 = NULL_TREE, vec_oprnd1 = NULL_TREE, op, src;
5409 tree arglist = TYPE_ARG_TYPES (TREE_TYPE (scatter_decl));
5410 tree rettype, srctype, ptrtype, idxtype, masktype, scaletype;
5411 tree ptr, mask, var, scale, perm_mask = NULL_TREE;
5412 edge pe = loop_preheader_edge (loop);
5413 gimple_seq seq;
5414 basic_block new_bb;
5415 enum { NARROW, NONE, WIDEN } modifier;
5416 int scatter_off_nunits = TYPE_VECTOR_SUBPARTS (scatter_off_vectype);
5418 if (nunits == (unsigned int) scatter_off_nunits)
5419 modifier = NONE;
5420 else if (nunits == (unsigned int) scatter_off_nunits / 2)
5422 unsigned char *sel = XALLOCAVEC (unsigned char, scatter_off_nunits);
5423 modifier = WIDEN;
5425 for (i = 0; i < (unsigned int) scatter_off_nunits; ++i)
5426 sel[i] = i | nunits;
5428 perm_mask = vect_gen_perm_mask_checked (scatter_off_vectype, sel);
5429 gcc_assert (perm_mask != NULL_TREE);
5431 else if (nunits == (unsigned int) scatter_off_nunits * 2)
5433 unsigned char *sel = XALLOCAVEC (unsigned char, nunits);
5434 modifier = NARROW;
5436 for (i = 0; i < (unsigned int) nunits; ++i)
5437 sel[i] = i | scatter_off_nunits;
5439 perm_mask = vect_gen_perm_mask_checked (vectype, sel);
5440 gcc_assert (perm_mask != NULL_TREE);
5441 ncopies *= 2;
5443 else
5444 gcc_unreachable ();
5446 rettype = TREE_TYPE (TREE_TYPE (scatter_decl));
5447 ptrtype = TREE_VALUE (arglist); arglist = TREE_CHAIN (arglist);
5448 masktype = TREE_VALUE (arglist); arglist = TREE_CHAIN (arglist);
5449 idxtype = TREE_VALUE (arglist); arglist = TREE_CHAIN (arglist);
5450 srctype = TREE_VALUE (arglist); arglist = TREE_CHAIN (arglist);
5451 scaletype = TREE_VALUE (arglist);
5453 gcc_checking_assert (TREE_CODE (masktype) == INTEGER_TYPE
5454 && TREE_CODE (rettype) == VOID_TYPE);
5456 ptr = fold_convert (ptrtype, scatter_base);
5457 if (!is_gimple_min_invariant (ptr))
5459 ptr = force_gimple_operand (ptr, &seq, true, NULL_TREE);
5460 new_bb = gsi_insert_seq_on_edge_immediate (pe, seq);
5461 gcc_assert (!new_bb);
5464 /* Currently we support only unconditional scatter stores,
5465 so mask should be all ones. */
5466 mask = build_int_cst (masktype, -1);
5467 mask = vect_init_vector (stmt, mask, masktype, NULL);
5469 scale = build_int_cst (scaletype, scatter_scale);
5471 prev_stmt_info = NULL;
5472 for (j = 0; j < ncopies; ++j)
5474 if (j == 0)
5476 src = vec_oprnd1
5477 = vect_get_vec_def_for_operand (gimple_assign_rhs1 (stmt), stmt);
5478 op = vec_oprnd0
5479 = vect_get_vec_def_for_operand (scatter_off, stmt);
5481 else if (modifier != NONE && (j & 1))
5483 if (modifier == WIDEN)
5485 src = vec_oprnd1
5486 = vect_get_vec_def_for_stmt_copy (scatter_src_dt, vec_oprnd1);
5487 op = permute_vec_elements (vec_oprnd0, vec_oprnd0, perm_mask,
5488 stmt, gsi);
5490 else if (modifier == NARROW)
5492 src = permute_vec_elements (vec_oprnd1, vec_oprnd1, perm_mask,
5493 stmt, gsi);
5494 op = vec_oprnd0
5495 = vect_get_vec_def_for_stmt_copy (scatter_idx_dt, vec_oprnd0);
5497 else
5498 gcc_unreachable ();
5500 else
5502 src = vec_oprnd1
5503 = vect_get_vec_def_for_stmt_copy (scatter_src_dt, vec_oprnd1);
5504 op = vec_oprnd0
5505 = vect_get_vec_def_for_stmt_copy (scatter_idx_dt, vec_oprnd0);
5508 if (!useless_type_conversion_p (srctype, TREE_TYPE (src)))
5510 gcc_assert (TYPE_VECTOR_SUBPARTS (TREE_TYPE (src))
5511 == TYPE_VECTOR_SUBPARTS (srctype));
5512 var = vect_get_new_ssa_name (srctype, vect_simple_var);
5513 src = build1 (VIEW_CONVERT_EXPR, srctype, src);
5514 new_stmt = gimple_build_assign (var, VIEW_CONVERT_EXPR, src);
5515 vect_finish_stmt_generation (stmt, new_stmt, gsi);
5516 src = var;
5519 if (!useless_type_conversion_p (idxtype, TREE_TYPE (op)))
5521 gcc_assert (TYPE_VECTOR_SUBPARTS (TREE_TYPE (op))
5522 == TYPE_VECTOR_SUBPARTS (idxtype));
5523 var = vect_get_new_ssa_name (idxtype, vect_simple_var);
5524 op = build1 (VIEW_CONVERT_EXPR, idxtype, op);
5525 new_stmt = gimple_build_assign (var, VIEW_CONVERT_EXPR, op);
5526 vect_finish_stmt_generation (stmt, new_stmt, gsi);
5527 op = var;
5530 new_stmt
5531 = gimple_build_call (scatter_decl, 5, ptr, mask, op, src, scale);
5533 vect_finish_stmt_generation (stmt, new_stmt, gsi);
5535 if (prev_stmt_info == NULL)
5536 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
5537 else
5538 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
5539 prev_stmt_info = vinfo_for_stmt (new_stmt);
5541 return true;
5544 if (grouped_store)
5546 first_dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (first_stmt));
5547 group_size = GROUP_SIZE (vinfo_for_stmt (first_stmt));
5549 GROUP_STORE_COUNT (vinfo_for_stmt (first_stmt))++;
5551 /* FORNOW */
5552 gcc_assert (!loop || !nested_in_vect_loop_p (loop, stmt));
5554 /* We vectorize all the stmts of the interleaving group when we
5555 reach the last stmt in the group. */
5556 if (GROUP_STORE_COUNT (vinfo_for_stmt (first_stmt))
5557 < GROUP_SIZE (vinfo_for_stmt (first_stmt))
5558 && !slp)
5560 *vec_stmt = NULL;
5561 return true;
5564 if (slp)
5566 grouped_store = false;
5567 /* VEC_NUM is the number of vect stmts to be created for this
5568 group. */
5569 vec_num = SLP_TREE_NUMBER_OF_VEC_STMTS (slp_node);
5570 first_stmt = SLP_TREE_SCALAR_STMTS (slp_node)[0];
5571 gcc_assert (GROUP_FIRST_ELEMENT (vinfo_for_stmt (first_stmt)) == first_stmt);
5572 first_dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (first_stmt));
5573 op = gimple_assign_rhs1 (first_stmt);
5575 else
5576 /* VEC_NUM is the number of vect stmts to be created for this
5577 group. */
5578 vec_num = group_size;
5580 else
5582 first_stmt = stmt;
5583 first_dr = dr;
5584 group_size = vec_num = 1;
5587 if (dump_enabled_p ())
5588 dump_printf_loc (MSG_NOTE, vect_location,
5589 "transform store. ncopies = %d\n", ncopies);
5591 if (STMT_VINFO_STRIDED_P (stmt_info))
5593 gimple_stmt_iterator incr_gsi;
5594 bool insert_after;
5595 gimple *incr;
5596 tree offvar;
5597 tree ivstep;
5598 tree running_off;
5599 gimple_seq stmts = NULL;
5600 tree stride_base, stride_step, alias_off;
5601 tree vec_oprnd;
5602 unsigned int g;
5604 gcc_assert (!nested_in_vect_loop_p (loop, stmt));
5606 stride_base
5607 = fold_build_pointer_plus
5608 (unshare_expr (DR_BASE_ADDRESS (first_dr)),
5609 size_binop (PLUS_EXPR,
5610 convert_to_ptrofftype (unshare_expr (DR_OFFSET (first_dr))),
5611 convert_to_ptrofftype (DR_INIT(first_dr))));
5612 stride_step = fold_convert (sizetype, unshare_expr (DR_STEP (first_dr)));
5614 /* For a store with loop-invariant (but other than power-of-2)
5615 stride (i.e. not a grouped access) like so:
5617 for (i = 0; i < n; i += stride)
5618 array[i] = ...;
5620 we generate a new induction variable and new stores from
5621 the components of the (vectorized) rhs:
5623 for (j = 0; ; j += VF*stride)
5624 vectemp = ...;
5625 tmp1 = vectemp[0];
5626 array[j] = tmp1;
5627 tmp2 = vectemp[1];
5628 array[j + stride] = tmp2;
5632 unsigned nstores = nunits;
5633 tree ltype = elem_type;
5634 if (slp)
5636 nstores = nunits / group_size;
5637 if (group_size < nunits)
5638 ltype = build_vector_type (elem_type, group_size);
5639 else
5640 ltype = vectype;
5641 ltype = build_aligned_type (ltype, TYPE_ALIGN (elem_type));
5642 ncopies = SLP_TREE_NUMBER_OF_VEC_STMTS (slp_node);
5643 group_size = 1;
5646 ivstep = stride_step;
5647 ivstep = fold_build2 (MULT_EXPR, TREE_TYPE (ivstep), ivstep,
5648 build_int_cst (TREE_TYPE (ivstep),
5649 ncopies * nstores));
5651 standard_iv_increment_position (loop, &incr_gsi, &insert_after);
5653 create_iv (stride_base, ivstep, NULL,
5654 loop, &incr_gsi, insert_after,
5655 &offvar, NULL);
5656 incr = gsi_stmt (incr_gsi);
5657 set_vinfo_for_stmt (incr, new_stmt_vec_info (incr, loop_vinfo));
5659 stride_step = force_gimple_operand (stride_step, &stmts, true, NULL_TREE);
5660 if (stmts)
5661 gsi_insert_seq_on_edge_immediate (loop_preheader_edge (loop), stmts);
5663 prev_stmt_info = NULL;
5664 alias_off = build_int_cst (reference_alias_ptr_type (DR_REF (first_dr)), 0);
5665 next_stmt = first_stmt;
5666 for (g = 0; g < group_size; g++)
5668 running_off = offvar;
5669 if (g)
5671 tree size = TYPE_SIZE_UNIT (ltype);
5672 tree pos = fold_build2 (MULT_EXPR, sizetype, size_int (g),
5673 size);
5674 tree newoff = copy_ssa_name (running_off, NULL);
5675 incr = gimple_build_assign (newoff, POINTER_PLUS_EXPR,
5676 running_off, pos);
5677 vect_finish_stmt_generation (stmt, incr, gsi);
5678 running_off = newoff;
5680 for (j = 0; j < ncopies; j++)
5682 /* We've set op and dt above, from gimple_assign_rhs1(stmt),
5683 and first_stmt == stmt. */
5684 if (j == 0)
5686 if (slp)
5688 vect_get_vec_defs (op, NULL_TREE, stmt, &vec_oprnds, NULL,
5689 slp_node, -1);
5690 vec_oprnd = vec_oprnds[0];
5692 else
5694 gcc_assert (gimple_assign_single_p (next_stmt));
5695 op = gimple_assign_rhs1 (next_stmt);
5696 vec_oprnd = vect_get_vec_def_for_operand (op, next_stmt);
5699 else
5701 if (slp)
5702 vec_oprnd = vec_oprnds[j];
5703 else
5705 vect_is_simple_use (vec_oprnd, vinfo, &def_stmt, &dt);
5706 vec_oprnd = vect_get_vec_def_for_stmt_copy (dt, vec_oprnd);
5710 for (i = 0; i < nstores; i++)
5712 tree newref, newoff;
5713 gimple *incr, *assign;
5714 tree size = TYPE_SIZE (ltype);
5715 /* Extract the i'th component. */
5716 tree pos = fold_build2 (MULT_EXPR, bitsizetype,
5717 bitsize_int (i), size);
5718 tree elem = fold_build3 (BIT_FIELD_REF, ltype, vec_oprnd,
5719 size, pos);
5721 elem = force_gimple_operand_gsi (gsi, elem, true,
5722 NULL_TREE, true,
5723 GSI_SAME_STMT);
5725 newref = build2 (MEM_REF, ltype,
5726 running_off, alias_off);
5728 /* And store it to *running_off. */
5729 assign = gimple_build_assign (newref, elem);
5730 vect_finish_stmt_generation (stmt, assign, gsi);
5732 newoff = copy_ssa_name (running_off, NULL);
5733 incr = gimple_build_assign (newoff, POINTER_PLUS_EXPR,
5734 running_off, stride_step);
5735 vect_finish_stmt_generation (stmt, incr, gsi);
5737 running_off = newoff;
5738 if (g == group_size - 1
5739 && !slp)
5741 if (j == 0 && i == 0)
5742 STMT_VINFO_VEC_STMT (stmt_info)
5743 = *vec_stmt = assign;
5744 else
5745 STMT_VINFO_RELATED_STMT (prev_stmt_info) = assign;
5746 prev_stmt_info = vinfo_for_stmt (assign);
5750 next_stmt = GROUP_NEXT_ELEMENT (vinfo_for_stmt (next_stmt));
5752 return true;
5755 dr_chain.create (group_size);
5756 oprnds.create (group_size);
5758 alignment_support_scheme = vect_supportable_dr_alignment (first_dr, false);
5759 gcc_assert (alignment_support_scheme);
5760 /* Targets with store-lane instructions must not require explicit
5761 realignment. */
5762 gcc_assert (!store_lanes_p
5763 || alignment_support_scheme == dr_aligned
5764 || alignment_support_scheme == dr_unaligned_supported);
5766 if (negative)
5767 offset = size_int (-TYPE_VECTOR_SUBPARTS (vectype) + 1);
5769 if (store_lanes_p)
5770 aggr_type = build_array_type_nelts (elem_type, vec_num * nunits);
5771 else
5772 aggr_type = vectype;
5774 /* In case the vectorization factor (VF) is bigger than the number
5775 of elements that we can fit in a vectype (nunits), we have to generate
5776 more than one vector stmt - i.e - we need to "unroll" the
5777 vector stmt by a factor VF/nunits. For more details see documentation in
5778 vect_get_vec_def_for_copy_stmt. */
5780 /* In case of interleaving (non-unit grouped access):
5782 S1: &base + 2 = x2
5783 S2: &base = x0
5784 S3: &base + 1 = x1
5785 S4: &base + 3 = x3
5787 We create vectorized stores starting from base address (the access of the
5788 first stmt in the chain (S2 in the above example), when the last store stmt
5789 of the chain (S4) is reached:
5791 VS1: &base = vx2
5792 VS2: &base + vec_size*1 = vx0
5793 VS3: &base + vec_size*2 = vx1
5794 VS4: &base + vec_size*3 = vx3
5796 Then permutation statements are generated:
5798 VS5: vx5 = VEC_PERM_EXPR < vx0, vx3, {0, 8, 1, 9, 2, 10, 3, 11} >
5799 VS6: vx6 = VEC_PERM_EXPR < vx0, vx3, {4, 12, 5, 13, 6, 14, 7, 15} >
5802 And they are put in STMT_VINFO_VEC_STMT of the corresponding scalar stmts
5803 (the order of the data-refs in the output of vect_permute_store_chain
5804 corresponds to the order of scalar stmts in the interleaving chain - see
5805 the documentation of vect_permute_store_chain()).
5807 In case of both multiple types and interleaving, above vector stores and
5808 permutation stmts are created for every copy. The result vector stmts are
5809 put in STMT_VINFO_VEC_STMT for the first copy and in the corresponding
5810 STMT_VINFO_RELATED_STMT for the next copies.
5813 prev_stmt_info = NULL;
5814 for (j = 0; j < ncopies; j++)
5817 if (j == 0)
5819 if (slp)
5821 /* Get vectorized arguments for SLP_NODE. */
5822 vect_get_vec_defs (op, NULL_TREE, stmt, &vec_oprnds,
5823 NULL, slp_node, -1);
5825 vec_oprnd = vec_oprnds[0];
5827 else
5829 /* For interleaved stores we collect vectorized defs for all the
5830 stores in the group in DR_CHAIN and OPRNDS. DR_CHAIN is then
5831 used as an input to vect_permute_store_chain(), and OPRNDS as
5832 an input to vect_get_vec_def_for_stmt_copy() for the next copy.
5834 If the store is not grouped, GROUP_SIZE is 1, and DR_CHAIN and
5835 OPRNDS are of size 1. */
5836 next_stmt = first_stmt;
5837 for (i = 0; i < group_size; i++)
5839 /* Since gaps are not supported for interleaved stores,
5840 GROUP_SIZE is the exact number of stmts in the chain.
5841 Therefore, NEXT_STMT can't be NULL_TREE. In case that
5842 there is no interleaving, GROUP_SIZE is 1, and only one
5843 iteration of the loop will be executed. */
5844 gcc_assert (next_stmt
5845 && gimple_assign_single_p (next_stmt));
5846 op = gimple_assign_rhs1 (next_stmt);
5848 vec_oprnd = vect_get_vec_def_for_operand (op, next_stmt);
5849 dr_chain.quick_push (vec_oprnd);
5850 oprnds.quick_push (vec_oprnd);
5851 next_stmt = GROUP_NEXT_ELEMENT (vinfo_for_stmt (next_stmt));
5855 /* We should have catched mismatched types earlier. */
5856 gcc_assert (useless_type_conversion_p (vectype,
5857 TREE_TYPE (vec_oprnd)));
5858 bool simd_lane_access_p
5859 = STMT_VINFO_SIMD_LANE_ACCESS_P (stmt_info);
5860 if (simd_lane_access_p
5861 && TREE_CODE (DR_BASE_ADDRESS (first_dr)) == ADDR_EXPR
5862 && VAR_P (TREE_OPERAND (DR_BASE_ADDRESS (first_dr), 0))
5863 && integer_zerop (DR_OFFSET (first_dr))
5864 && integer_zerop (DR_INIT (first_dr))
5865 && alias_sets_conflict_p (get_alias_set (aggr_type),
5866 get_alias_set (DR_REF (first_dr))))
5868 dataref_ptr = unshare_expr (DR_BASE_ADDRESS (first_dr));
5869 dataref_offset = build_int_cst (reference_alias_ptr_type
5870 (DR_REF (first_dr)), 0);
5871 inv_p = false;
5873 else
5874 dataref_ptr
5875 = vect_create_data_ref_ptr (first_stmt, aggr_type,
5876 simd_lane_access_p ? loop : NULL,
5877 offset, &dummy, gsi, &ptr_incr,
5878 simd_lane_access_p, &inv_p);
5879 gcc_assert (bb_vinfo || !inv_p);
5881 else
5883 /* For interleaved stores we created vectorized defs for all the
5884 defs stored in OPRNDS in the previous iteration (previous copy).
5885 DR_CHAIN is then used as an input to vect_permute_store_chain(),
5886 and OPRNDS as an input to vect_get_vec_def_for_stmt_copy() for the
5887 next copy.
5888 If the store is not grouped, GROUP_SIZE is 1, and DR_CHAIN and
5889 OPRNDS are of size 1. */
5890 for (i = 0; i < group_size; i++)
5892 op = oprnds[i];
5893 vect_is_simple_use (op, vinfo, &def_stmt, &dt);
5894 vec_oprnd = vect_get_vec_def_for_stmt_copy (dt, op);
5895 dr_chain[i] = vec_oprnd;
5896 oprnds[i] = vec_oprnd;
5898 if (dataref_offset)
5899 dataref_offset
5900 = int_const_binop (PLUS_EXPR, dataref_offset,
5901 TYPE_SIZE_UNIT (aggr_type));
5902 else
5903 dataref_ptr = bump_vector_ptr (dataref_ptr, ptr_incr, gsi, stmt,
5904 TYPE_SIZE_UNIT (aggr_type));
5907 if (store_lanes_p)
5909 tree vec_array;
5911 /* Combine all the vectors into an array. */
5912 vec_array = create_vector_array (vectype, vec_num);
5913 for (i = 0; i < vec_num; i++)
5915 vec_oprnd = dr_chain[i];
5916 write_vector_array (stmt, gsi, vec_oprnd, vec_array, i);
5919 /* Emit:
5920 MEM_REF[...all elements...] = STORE_LANES (VEC_ARRAY). */
5921 data_ref = create_array_ref (aggr_type, dataref_ptr, first_dr);
5922 new_stmt = gimple_build_call_internal (IFN_STORE_LANES, 1, vec_array);
5923 gimple_call_set_lhs (new_stmt, data_ref);
5924 vect_finish_stmt_generation (stmt, new_stmt, gsi);
5926 else
5928 new_stmt = NULL;
5929 if (grouped_store)
5931 if (j == 0)
5932 result_chain.create (group_size);
5933 /* Permute. */
5934 vect_permute_store_chain (dr_chain, group_size, stmt, gsi,
5935 &result_chain);
5938 next_stmt = first_stmt;
5939 for (i = 0; i < vec_num; i++)
5941 unsigned align, misalign;
5943 if (i > 0)
5944 /* Bump the vector pointer. */
5945 dataref_ptr = bump_vector_ptr (dataref_ptr, ptr_incr, gsi,
5946 stmt, NULL_TREE);
5948 if (slp)
5949 vec_oprnd = vec_oprnds[i];
5950 else if (grouped_store)
5951 /* For grouped stores vectorized defs are interleaved in
5952 vect_permute_store_chain(). */
5953 vec_oprnd = result_chain[i];
5955 data_ref = fold_build2 (MEM_REF, TREE_TYPE (vec_oprnd),
5956 dataref_ptr,
5957 dataref_offset
5958 ? dataref_offset
5959 : build_int_cst (reference_alias_ptr_type
5960 (DR_REF (first_dr)), 0));
5961 align = TYPE_ALIGN_UNIT (vectype);
5962 if (aligned_access_p (first_dr))
5963 misalign = 0;
5964 else if (DR_MISALIGNMENT (first_dr) == -1)
5966 if (DR_VECT_AUX (first_dr)->base_element_aligned)
5967 align = TYPE_ALIGN_UNIT (elem_type);
5968 else
5969 align = get_object_alignment (DR_REF (first_dr))
5970 / BITS_PER_UNIT;
5971 misalign = 0;
5972 TREE_TYPE (data_ref)
5973 = build_aligned_type (TREE_TYPE (data_ref),
5974 align * BITS_PER_UNIT);
5976 else
5978 TREE_TYPE (data_ref)
5979 = build_aligned_type (TREE_TYPE (data_ref),
5980 TYPE_ALIGN (elem_type));
5981 misalign = DR_MISALIGNMENT (first_dr);
5983 if (dataref_offset == NULL_TREE
5984 && TREE_CODE (dataref_ptr) == SSA_NAME)
5985 set_ptr_info_alignment (get_ptr_info (dataref_ptr), align,
5986 misalign);
5988 if (negative
5989 && dt != vect_constant_def
5990 && dt != vect_external_def)
5992 tree perm_mask = perm_mask_for_reverse (vectype);
5993 tree perm_dest
5994 = vect_create_destination_var (gimple_assign_rhs1 (stmt),
5995 vectype);
5996 tree new_temp = make_ssa_name (perm_dest);
5998 /* Generate the permute statement. */
5999 gimple *perm_stmt
6000 = gimple_build_assign (new_temp, VEC_PERM_EXPR, vec_oprnd,
6001 vec_oprnd, perm_mask);
6002 vect_finish_stmt_generation (stmt, perm_stmt, gsi);
6004 perm_stmt = SSA_NAME_DEF_STMT (new_temp);
6005 vec_oprnd = new_temp;
6008 /* Arguments are ready. Create the new vector stmt. */
6009 new_stmt = gimple_build_assign (data_ref, vec_oprnd);
6010 vect_finish_stmt_generation (stmt, new_stmt, gsi);
6012 if (slp)
6013 continue;
6015 next_stmt = GROUP_NEXT_ELEMENT (vinfo_for_stmt (next_stmt));
6016 if (!next_stmt)
6017 break;
6020 if (!slp)
6022 if (j == 0)
6023 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
6024 else
6025 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
6026 prev_stmt_info = vinfo_for_stmt (new_stmt);
6030 dr_chain.release ();
6031 oprnds.release ();
6032 result_chain.release ();
6033 vec_oprnds.release ();
6035 return true;
6038 /* Given a vector type VECTYPE, turns permutation SEL into the equivalent
6039 VECTOR_CST mask. No checks are made that the target platform supports the
6040 mask, so callers may wish to test can_vec_perm_p separately, or use
6041 vect_gen_perm_mask_checked. */
6043 tree
6044 vect_gen_perm_mask_any (tree vectype, const unsigned char *sel)
6046 tree mask_elt_type, mask_type, mask_vec, *mask_elts;
6047 int i, nunits;
6049 nunits = TYPE_VECTOR_SUBPARTS (vectype);
6051 mask_elt_type = lang_hooks.types.type_for_mode
6052 (int_mode_for_mode (TYPE_MODE (TREE_TYPE (vectype))), 1);
6053 mask_type = get_vectype_for_scalar_type (mask_elt_type);
6055 mask_elts = XALLOCAVEC (tree, nunits);
6056 for (i = nunits - 1; i >= 0; i--)
6057 mask_elts[i] = build_int_cst (mask_elt_type, sel[i]);
6058 mask_vec = build_vector (mask_type, mask_elts);
6060 return mask_vec;
6063 /* Checked version of vect_gen_perm_mask_any. Asserts can_vec_perm_p,
6064 i.e. that the target supports the pattern _for arbitrary input vectors_. */
6066 tree
6067 vect_gen_perm_mask_checked (tree vectype, const unsigned char *sel)
6069 gcc_assert (can_vec_perm_p (TYPE_MODE (vectype), false, sel));
6070 return vect_gen_perm_mask_any (vectype, sel);
6073 /* Given a vector variable X and Y, that was generated for the scalar
6074 STMT, generate instructions to permute the vector elements of X and Y
6075 using permutation mask MASK_VEC, insert them at *GSI and return the
6076 permuted vector variable. */
6078 static tree
6079 permute_vec_elements (tree x, tree y, tree mask_vec, gimple *stmt,
6080 gimple_stmt_iterator *gsi)
6082 tree vectype = TREE_TYPE (x);
6083 tree perm_dest, data_ref;
6084 gimple *perm_stmt;
6086 perm_dest = vect_create_destination_var (gimple_get_lhs (stmt), vectype);
6087 data_ref = make_ssa_name (perm_dest);
6089 /* Generate the permute statement. */
6090 perm_stmt = gimple_build_assign (data_ref, VEC_PERM_EXPR, x, y, mask_vec);
6091 vect_finish_stmt_generation (stmt, perm_stmt, gsi);
6093 return data_ref;
6096 /* Hoist the definitions of all SSA uses on STMT out of the loop LOOP,
6097 inserting them on the loops preheader edge. Returns true if we
6098 were successful in doing so (and thus STMT can be moved then),
6099 otherwise returns false. */
6101 static bool
6102 hoist_defs_of_uses (gimple *stmt, struct loop *loop)
6104 ssa_op_iter i;
6105 tree op;
6106 bool any = false;
6108 FOR_EACH_SSA_TREE_OPERAND (op, stmt, i, SSA_OP_USE)
6110 gimple *def_stmt = SSA_NAME_DEF_STMT (op);
6111 if (!gimple_nop_p (def_stmt)
6112 && flow_bb_inside_loop_p (loop, gimple_bb (def_stmt)))
6114 /* Make sure we don't need to recurse. While we could do
6115 so in simple cases when there are more complex use webs
6116 we don't have an easy way to preserve stmt order to fulfil
6117 dependencies within them. */
6118 tree op2;
6119 ssa_op_iter i2;
6120 if (gimple_code (def_stmt) == GIMPLE_PHI)
6121 return false;
6122 FOR_EACH_SSA_TREE_OPERAND (op2, def_stmt, i2, SSA_OP_USE)
6124 gimple *def_stmt2 = SSA_NAME_DEF_STMT (op2);
6125 if (!gimple_nop_p (def_stmt2)
6126 && flow_bb_inside_loop_p (loop, gimple_bb (def_stmt2)))
6127 return false;
6129 any = true;
6133 if (!any)
6134 return true;
6136 FOR_EACH_SSA_TREE_OPERAND (op, stmt, i, SSA_OP_USE)
6138 gimple *def_stmt = SSA_NAME_DEF_STMT (op);
6139 if (!gimple_nop_p (def_stmt)
6140 && flow_bb_inside_loop_p (loop, gimple_bb (def_stmt)))
6142 gimple_stmt_iterator gsi = gsi_for_stmt (def_stmt);
6143 gsi_remove (&gsi, false);
6144 gsi_insert_on_edge_immediate (loop_preheader_edge (loop), def_stmt);
6148 return true;
6151 /* vectorizable_load.
6153 Check if STMT reads a non scalar data-ref (array/pointer/structure) that
6154 can be vectorized.
6155 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
6156 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
6157 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
6159 static bool
6160 vectorizable_load (gimple *stmt, gimple_stmt_iterator *gsi, gimple **vec_stmt,
6161 slp_tree slp_node, slp_instance slp_node_instance)
6163 tree scalar_dest;
6164 tree vec_dest = NULL;
6165 tree data_ref = NULL;
6166 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
6167 stmt_vec_info prev_stmt_info;
6168 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
6169 struct loop *loop = NULL;
6170 struct loop *containing_loop = (gimple_bb (stmt))->loop_father;
6171 bool nested_in_vect_loop = false;
6172 struct data_reference *dr = STMT_VINFO_DATA_REF (stmt_info), *first_dr = NULL;
6173 tree elem_type;
6174 tree new_temp;
6175 machine_mode mode;
6176 gimple *new_stmt = NULL;
6177 tree dummy;
6178 enum dr_alignment_support alignment_support_scheme;
6179 tree dataref_ptr = NULL_TREE;
6180 tree dataref_offset = NULL_TREE;
6181 gimple *ptr_incr = NULL;
6182 int ncopies;
6183 int i, j, group_size = -1, group_gap_adj;
6184 tree msq = NULL_TREE, lsq;
6185 tree offset = NULL_TREE;
6186 tree byte_offset = NULL_TREE;
6187 tree realignment_token = NULL_TREE;
6188 gphi *phi = NULL;
6189 vec<tree> dr_chain = vNULL;
6190 bool grouped_load = false;
6191 bool load_lanes_p = false;
6192 gimple *first_stmt;
6193 gimple *first_stmt_for_drptr = NULL;
6194 bool inv_p;
6195 bool negative = false;
6196 bool compute_in_loop = false;
6197 struct loop *at_loop;
6198 int vec_num;
6199 bool slp = (slp_node != NULL);
6200 bool slp_perm = false;
6201 enum tree_code code;
6202 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
6203 int vf;
6204 tree aggr_type;
6205 tree gather_base = NULL_TREE, gather_off = NULL_TREE;
6206 tree gather_off_vectype = NULL_TREE, gather_decl = NULL_TREE;
6207 int gather_scale = 1;
6208 enum vect_def_type gather_dt = vect_unknown_def_type;
6209 vec_info *vinfo = stmt_info->vinfo;
6211 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
6212 return false;
6214 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def
6215 && ! vec_stmt)
6216 return false;
6218 /* Is vectorizable load? */
6219 if (!is_gimple_assign (stmt))
6220 return false;
6222 scalar_dest = gimple_assign_lhs (stmt);
6223 if (TREE_CODE (scalar_dest) != SSA_NAME)
6224 return false;
6226 code = gimple_assign_rhs_code (stmt);
6227 if (code != ARRAY_REF
6228 && code != BIT_FIELD_REF
6229 && code != INDIRECT_REF
6230 && code != COMPONENT_REF
6231 && code != IMAGPART_EXPR
6232 && code != REALPART_EXPR
6233 && code != MEM_REF
6234 && TREE_CODE_CLASS (code) != tcc_declaration)
6235 return false;
6237 if (!STMT_VINFO_DATA_REF (stmt_info))
6238 return false;
6240 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
6241 int nunits = TYPE_VECTOR_SUBPARTS (vectype);
6243 if (loop_vinfo)
6245 loop = LOOP_VINFO_LOOP (loop_vinfo);
6246 nested_in_vect_loop = nested_in_vect_loop_p (loop, stmt);
6247 vf = LOOP_VINFO_VECT_FACTOR (loop_vinfo);
6249 else
6250 vf = 1;
6252 /* Multiple types in SLP are handled by creating the appropriate number of
6253 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
6254 case of SLP. */
6255 if (slp || PURE_SLP_STMT (stmt_info))
6256 ncopies = 1;
6257 else
6258 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits;
6260 gcc_assert (ncopies >= 1);
6262 /* FORNOW. This restriction should be relaxed. */
6263 if (nested_in_vect_loop && ncopies > 1)
6265 if (dump_enabled_p ())
6266 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
6267 "multiple types in nested loop.\n");
6268 return false;
6271 /* Invalidate assumptions made by dependence analysis when vectorization
6272 on the unrolled body effectively re-orders stmts. */
6273 if (ncopies > 1
6274 && STMT_VINFO_MIN_NEG_DIST (stmt_info) != 0
6275 && ((unsigned)LOOP_VINFO_VECT_FACTOR (loop_vinfo)
6276 > STMT_VINFO_MIN_NEG_DIST (stmt_info)))
6278 if (dump_enabled_p ())
6279 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
6280 "cannot perform implicit CSE when unrolling "
6281 "with negative dependence distance\n");
6282 return false;
6285 elem_type = TREE_TYPE (vectype);
6286 mode = TYPE_MODE (vectype);
6288 /* FORNOW. In some cases can vectorize even if data-type not supported
6289 (e.g. - data copies). */
6290 if (optab_handler (mov_optab, mode) == CODE_FOR_nothing)
6292 if (dump_enabled_p ())
6293 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
6294 "Aligned load, but unsupported type.\n");
6295 return false;
6298 /* Check if the load is a part of an interleaving chain. */
6299 if (STMT_VINFO_GROUPED_ACCESS (stmt_info))
6301 grouped_load = true;
6302 /* FORNOW */
6303 gcc_assert (!nested_in_vect_loop && !STMT_VINFO_GATHER_SCATTER_P (stmt_info));
6305 first_stmt = GROUP_FIRST_ELEMENT (stmt_info);
6307 /* If this is single-element interleaving with an element distance
6308 that leaves unused vector loads around punt - we at least create
6309 very sub-optimal code in that case (and blow up memory,
6310 see PR65518). */
6311 bool force_peeling = false;
6312 if (first_stmt == stmt
6313 && !GROUP_NEXT_ELEMENT (stmt_info))
6315 if (GROUP_SIZE (stmt_info) > TYPE_VECTOR_SUBPARTS (vectype))
6317 if (dump_enabled_p ())
6318 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
6319 "single-element interleaving not supported "
6320 "for not adjacent vector loads\n");
6321 return false;
6324 /* Single-element interleaving requires peeling for gaps. */
6325 force_peeling = true;
6328 /* If there is a gap in the end of the group or the group size cannot
6329 be made a multiple of the vector element count then we access excess
6330 elements in the last iteration and thus need to peel that off. */
6331 if (loop_vinfo
6332 && ! STMT_VINFO_STRIDED_P (stmt_info)
6333 && (force_peeling
6334 || GROUP_GAP (vinfo_for_stmt (first_stmt)) != 0
6335 || (!slp && vf % GROUP_SIZE (vinfo_for_stmt (first_stmt)) != 0)))
6337 if (dump_enabled_p ())
6338 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
6339 "Data access with gaps requires scalar "
6340 "epilogue loop\n");
6341 if (loop->inner)
6343 if (dump_enabled_p ())
6344 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
6345 "Peeling for outer loop is not supported\n");
6346 return false;
6349 LOOP_VINFO_PEELING_FOR_GAPS (loop_vinfo) = true;
6352 if (slp && SLP_TREE_LOAD_PERMUTATION (slp_node).exists ())
6353 slp_perm = true;
6355 group_size = GROUP_SIZE (vinfo_for_stmt (first_stmt));
6357 /* ??? The following is overly pessimistic (as well as the loop
6358 case above) in the case we can statically determine the excess
6359 elements loaded are within the bounds of a decl that is accessed.
6360 Likewise for BB vectorizations using masked loads is a possibility. */
6361 if (bb_vinfo && slp_perm && group_size % nunits != 0)
6363 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
6364 "BB vectorization with gaps at the end of a load "
6365 "is not supported\n");
6366 return false;
6369 if (!slp
6370 && !PURE_SLP_STMT (stmt_info)
6371 && !STMT_VINFO_STRIDED_P (stmt_info))
6373 if (vect_load_lanes_supported (vectype, group_size))
6374 load_lanes_p = true;
6375 else if (!vect_grouped_load_supported (vectype, group_size))
6376 return false;
6379 /* Invalidate assumptions made by dependence analysis when vectorization
6380 on the unrolled body effectively re-orders stmts. */
6381 if (!PURE_SLP_STMT (stmt_info)
6382 && STMT_VINFO_MIN_NEG_DIST (stmt_info) != 0
6383 && ((unsigned)LOOP_VINFO_VECT_FACTOR (loop_vinfo)
6384 > STMT_VINFO_MIN_NEG_DIST (stmt_info)))
6386 if (dump_enabled_p ())
6387 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
6388 "cannot perform implicit CSE when performing "
6389 "group loads with negative dependence distance\n");
6390 return false;
6393 /* Similarly when the stmt is a load that is both part of a SLP
6394 instance and a loop vectorized stmt via the same-dr mechanism
6395 we have to give up. */
6396 if (STMT_VINFO_GROUP_SAME_DR_STMT (stmt_info)
6397 && (STMT_SLP_TYPE (stmt_info)
6398 != STMT_SLP_TYPE (vinfo_for_stmt
6399 (STMT_VINFO_GROUP_SAME_DR_STMT (stmt_info)))))
6401 if (dump_enabled_p ())
6402 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
6403 "conflicting SLP types for CSEd load\n");
6404 return false;
6409 if (STMT_VINFO_GATHER_SCATTER_P (stmt_info))
6411 gimple *def_stmt;
6412 gather_decl = vect_check_gather_scatter (stmt, loop_vinfo, &gather_base,
6413 &gather_off, &gather_scale);
6414 gcc_assert (gather_decl);
6415 if (!vect_is_simple_use (gather_off, vinfo, &def_stmt, &gather_dt,
6416 &gather_off_vectype))
6418 if (dump_enabled_p ())
6419 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
6420 "gather index use not simple.\n");
6421 return false;
6424 else if (STMT_VINFO_STRIDED_P (stmt_info))
6426 if ((grouped_load
6427 && (slp || PURE_SLP_STMT (stmt_info)))
6428 && (group_size > nunits
6429 || nunits % group_size != 0))
6431 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
6432 "unhandled strided group load\n");
6433 return false;
6436 else
6438 negative = tree_int_cst_compare (nested_in_vect_loop
6439 ? STMT_VINFO_DR_STEP (stmt_info)
6440 : DR_STEP (dr),
6441 size_zero_node) < 0;
6442 if (negative && ncopies > 1)
6444 if (dump_enabled_p ())
6445 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
6446 "multiple types with negative step.\n");
6447 return false;
6450 if (negative)
6452 if (grouped_load)
6454 if (dump_enabled_p ())
6455 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
6456 "negative step for group load not supported"
6457 "\n");
6458 return false;
6460 alignment_support_scheme = vect_supportable_dr_alignment (dr, false);
6461 if (alignment_support_scheme != dr_aligned
6462 && alignment_support_scheme != dr_unaligned_supported)
6464 if (dump_enabled_p ())
6465 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
6466 "negative step but alignment required.\n");
6467 return false;
6469 if (!perm_mask_for_reverse (vectype))
6471 if (dump_enabled_p ())
6472 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
6473 "negative step and reversing not supported."
6474 "\n");
6475 return false;
6480 if (!vec_stmt) /* transformation not required. */
6482 STMT_VINFO_TYPE (stmt_info) = load_vec_info_type;
6483 /* The SLP costs are calculated during SLP analysis. */
6484 if (!PURE_SLP_STMT (stmt_info))
6485 vect_model_load_cost (stmt_info, ncopies, load_lanes_p,
6486 NULL, NULL, NULL);
6487 return true;
6490 if (dump_enabled_p ())
6491 dump_printf_loc (MSG_NOTE, vect_location,
6492 "transform load. ncopies = %d\n", ncopies);
6494 /** Transform. **/
6496 ensure_base_align (stmt_info, dr);
6498 if (STMT_VINFO_GATHER_SCATTER_P (stmt_info))
6500 tree vec_oprnd0 = NULL_TREE, op;
6501 tree arglist = TYPE_ARG_TYPES (TREE_TYPE (gather_decl));
6502 tree rettype, srctype, ptrtype, idxtype, masktype, scaletype;
6503 tree ptr, mask, var, scale, merge, perm_mask = NULL_TREE, prev_res = NULL_TREE;
6504 edge pe = loop_preheader_edge (loop);
6505 gimple_seq seq;
6506 basic_block new_bb;
6507 enum { NARROW, NONE, WIDEN } modifier;
6508 int gather_off_nunits = TYPE_VECTOR_SUBPARTS (gather_off_vectype);
6510 if (nunits == gather_off_nunits)
6511 modifier = NONE;
6512 else if (nunits == gather_off_nunits / 2)
6514 unsigned char *sel = XALLOCAVEC (unsigned char, gather_off_nunits);
6515 modifier = WIDEN;
6517 for (i = 0; i < gather_off_nunits; ++i)
6518 sel[i] = i | nunits;
6520 perm_mask = vect_gen_perm_mask_checked (gather_off_vectype, sel);
6522 else if (nunits == gather_off_nunits * 2)
6524 unsigned char *sel = XALLOCAVEC (unsigned char, nunits);
6525 modifier = NARROW;
6527 for (i = 0; i < nunits; ++i)
6528 sel[i] = i < gather_off_nunits
6529 ? i : i + nunits - gather_off_nunits;
6531 perm_mask = vect_gen_perm_mask_checked (vectype, sel);
6532 ncopies *= 2;
6534 else
6535 gcc_unreachable ();
6537 rettype = TREE_TYPE (TREE_TYPE (gather_decl));
6538 srctype = TREE_VALUE (arglist); arglist = TREE_CHAIN (arglist);
6539 ptrtype = TREE_VALUE (arglist); arglist = TREE_CHAIN (arglist);
6540 idxtype = TREE_VALUE (arglist); arglist = TREE_CHAIN (arglist);
6541 masktype = TREE_VALUE (arglist); arglist = TREE_CHAIN (arglist);
6542 scaletype = TREE_VALUE (arglist);
6543 gcc_checking_assert (types_compatible_p (srctype, rettype));
6545 vec_dest = vect_create_destination_var (scalar_dest, vectype);
6547 ptr = fold_convert (ptrtype, gather_base);
6548 if (!is_gimple_min_invariant (ptr))
6550 ptr = force_gimple_operand (ptr, &seq, true, NULL_TREE);
6551 new_bb = gsi_insert_seq_on_edge_immediate (pe, seq);
6552 gcc_assert (!new_bb);
6555 /* Currently we support only unconditional gather loads,
6556 so mask should be all ones. */
6557 if (TREE_CODE (masktype) == INTEGER_TYPE)
6558 mask = build_int_cst (masktype, -1);
6559 else if (TREE_CODE (TREE_TYPE (masktype)) == INTEGER_TYPE)
6561 mask = build_int_cst (TREE_TYPE (masktype), -1);
6562 mask = build_vector_from_val (masktype, mask);
6563 mask = vect_init_vector (stmt, mask, masktype, NULL);
6565 else if (SCALAR_FLOAT_TYPE_P (TREE_TYPE (masktype)))
6567 REAL_VALUE_TYPE r;
6568 long tmp[6];
6569 for (j = 0; j < 6; ++j)
6570 tmp[j] = -1;
6571 real_from_target (&r, tmp, TYPE_MODE (TREE_TYPE (masktype)));
6572 mask = build_real (TREE_TYPE (masktype), r);
6573 mask = build_vector_from_val (masktype, mask);
6574 mask = vect_init_vector (stmt, mask, masktype, NULL);
6576 else
6577 gcc_unreachable ();
6579 scale = build_int_cst (scaletype, gather_scale);
6581 if (TREE_CODE (TREE_TYPE (rettype)) == INTEGER_TYPE)
6582 merge = build_int_cst (TREE_TYPE (rettype), 0);
6583 else if (SCALAR_FLOAT_TYPE_P (TREE_TYPE (rettype)))
6585 REAL_VALUE_TYPE r;
6586 long tmp[6];
6587 for (j = 0; j < 6; ++j)
6588 tmp[j] = 0;
6589 real_from_target (&r, tmp, TYPE_MODE (TREE_TYPE (rettype)));
6590 merge = build_real (TREE_TYPE (rettype), r);
6592 else
6593 gcc_unreachable ();
6594 merge = build_vector_from_val (rettype, merge);
6595 merge = vect_init_vector (stmt, merge, rettype, NULL);
6597 prev_stmt_info = NULL;
6598 for (j = 0; j < ncopies; ++j)
6600 if (modifier == WIDEN && (j & 1))
6601 op = permute_vec_elements (vec_oprnd0, vec_oprnd0,
6602 perm_mask, stmt, gsi);
6603 else if (j == 0)
6604 op = vec_oprnd0
6605 = vect_get_vec_def_for_operand (gather_off, stmt);
6606 else
6607 op = vec_oprnd0
6608 = vect_get_vec_def_for_stmt_copy (gather_dt, vec_oprnd0);
6610 if (!useless_type_conversion_p (idxtype, TREE_TYPE (op)))
6612 gcc_assert (TYPE_VECTOR_SUBPARTS (TREE_TYPE (op))
6613 == TYPE_VECTOR_SUBPARTS (idxtype));
6614 var = vect_get_new_ssa_name (idxtype, vect_simple_var);
6615 op = build1 (VIEW_CONVERT_EXPR, idxtype, op);
6616 new_stmt
6617 = gimple_build_assign (var, VIEW_CONVERT_EXPR, op);
6618 vect_finish_stmt_generation (stmt, new_stmt, gsi);
6619 op = var;
6622 new_stmt
6623 = gimple_build_call (gather_decl, 5, merge, ptr, op, mask, scale);
6625 if (!useless_type_conversion_p (vectype, rettype))
6627 gcc_assert (TYPE_VECTOR_SUBPARTS (vectype)
6628 == TYPE_VECTOR_SUBPARTS (rettype));
6629 op = vect_get_new_ssa_name (rettype, vect_simple_var);
6630 gimple_call_set_lhs (new_stmt, op);
6631 vect_finish_stmt_generation (stmt, new_stmt, gsi);
6632 var = make_ssa_name (vec_dest);
6633 op = build1 (VIEW_CONVERT_EXPR, vectype, op);
6634 new_stmt
6635 = gimple_build_assign (var, VIEW_CONVERT_EXPR, op);
6637 else
6639 var = make_ssa_name (vec_dest, new_stmt);
6640 gimple_call_set_lhs (new_stmt, var);
6643 vect_finish_stmt_generation (stmt, new_stmt, gsi);
6645 if (modifier == NARROW)
6647 if ((j & 1) == 0)
6649 prev_res = var;
6650 continue;
6652 var = permute_vec_elements (prev_res, var,
6653 perm_mask, stmt, gsi);
6654 new_stmt = SSA_NAME_DEF_STMT (var);
6657 if (prev_stmt_info == NULL)
6658 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
6659 else
6660 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
6661 prev_stmt_info = vinfo_for_stmt (new_stmt);
6663 return true;
6665 else if (STMT_VINFO_STRIDED_P (stmt_info))
6667 gimple_stmt_iterator incr_gsi;
6668 bool insert_after;
6669 gimple *incr;
6670 tree offvar;
6671 tree ivstep;
6672 tree running_off;
6673 vec<constructor_elt, va_gc> *v = NULL;
6674 gimple_seq stmts = NULL;
6675 tree stride_base, stride_step, alias_off;
6677 gcc_assert (!nested_in_vect_loop);
6679 if (slp && grouped_load)
6680 first_dr = STMT_VINFO_DATA_REF
6681 (vinfo_for_stmt (GROUP_FIRST_ELEMENT (stmt_info)));
6682 else
6683 first_dr = dr;
6685 stride_base
6686 = fold_build_pointer_plus
6687 (DR_BASE_ADDRESS (first_dr),
6688 size_binop (PLUS_EXPR,
6689 convert_to_ptrofftype (DR_OFFSET (first_dr)),
6690 convert_to_ptrofftype (DR_INIT (first_dr))));
6691 stride_step = fold_convert (sizetype, DR_STEP (first_dr));
6693 /* For a load with loop-invariant (but other than power-of-2)
6694 stride (i.e. not a grouped access) like so:
6696 for (i = 0; i < n; i += stride)
6697 ... = array[i];
6699 we generate a new induction variable and new accesses to
6700 form a new vector (or vectors, depending on ncopies):
6702 for (j = 0; ; j += VF*stride)
6703 tmp1 = array[j];
6704 tmp2 = array[j + stride];
6706 vectemp = {tmp1, tmp2, ...}
6709 ivstep = fold_build2 (MULT_EXPR, TREE_TYPE (stride_step), stride_step,
6710 build_int_cst (TREE_TYPE (stride_step), vf));
6712 standard_iv_increment_position (loop, &incr_gsi, &insert_after);
6714 create_iv (unshare_expr (stride_base), unshare_expr (ivstep), NULL,
6715 loop, &incr_gsi, insert_after,
6716 &offvar, NULL);
6717 incr = gsi_stmt (incr_gsi);
6718 set_vinfo_for_stmt (incr, new_stmt_vec_info (incr, loop_vinfo));
6720 stride_step = force_gimple_operand (unshare_expr (stride_step),
6721 &stmts, true, NULL_TREE);
6722 if (stmts)
6723 gsi_insert_seq_on_edge_immediate (loop_preheader_edge (loop), stmts);
6725 prev_stmt_info = NULL;
6726 running_off = offvar;
6727 alias_off = build_int_cst (reference_alias_ptr_type (DR_REF (first_dr)), 0);
6728 int nloads = nunits;
6729 tree ltype = TREE_TYPE (vectype);
6730 auto_vec<tree> dr_chain;
6731 if (slp)
6733 nloads = nunits / group_size;
6734 if (group_size < nunits)
6735 ltype = build_vector_type (TREE_TYPE (vectype), group_size);
6736 else
6737 ltype = vectype;
6738 ltype = build_aligned_type (ltype, TYPE_ALIGN (TREE_TYPE (vectype)));
6739 /* For SLP permutation support we need to load the whole group,
6740 not only the number of vector stmts the permutation result
6741 fits in. */
6742 if (slp_perm)
6744 ncopies = (group_size * vf + nunits - 1) / nunits;
6745 dr_chain.create (ncopies);
6747 else
6748 ncopies = SLP_TREE_NUMBER_OF_VEC_STMTS (slp_node);
6750 for (j = 0; j < ncopies; j++)
6752 tree vec_inv;
6754 if (nloads > 1)
6756 vec_alloc (v, nloads);
6757 for (i = 0; i < nloads; i++)
6759 tree newref, newoff;
6760 gimple *incr;
6761 newref = build2 (MEM_REF, ltype, running_off, alias_off);
6763 newref = force_gimple_operand_gsi (gsi, newref, true,
6764 NULL_TREE, true,
6765 GSI_SAME_STMT);
6766 CONSTRUCTOR_APPEND_ELT (v, NULL_TREE, newref);
6767 newoff = copy_ssa_name (running_off);
6768 incr = gimple_build_assign (newoff, POINTER_PLUS_EXPR,
6769 running_off, stride_step);
6770 vect_finish_stmt_generation (stmt, incr, gsi);
6772 running_off = newoff;
6775 vec_inv = build_constructor (vectype, v);
6776 new_temp = vect_init_vector (stmt, vec_inv, vectype, gsi);
6777 new_stmt = SSA_NAME_DEF_STMT (new_temp);
6779 else
6781 new_stmt = gimple_build_assign (make_ssa_name (ltype),
6782 build2 (MEM_REF, ltype,
6783 running_off, alias_off));
6784 vect_finish_stmt_generation (stmt, new_stmt, gsi);
6786 tree newoff = copy_ssa_name (running_off);
6787 gimple *incr = gimple_build_assign (newoff, POINTER_PLUS_EXPR,
6788 running_off, stride_step);
6789 vect_finish_stmt_generation (stmt, incr, gsi);
6791 running_off = newoff;
6794 if (slp)
6796 if (slp_perm)
6797 dr_chain.quick_push (gimple_assign_lhs (new_stmt));
6798 else
6799 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
6801 else
6803 if (j == 0)
6804 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
6805 else
6806 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
6807 prev_stmt_info = vinfo_for_stmt (new_stmt);
6810 if (slp_perm)
6811 vect_transform_slp_perm_load (slp_node, dr_chain, gsi, vf,
6812 slp_node_instance, false);
6813 return true;
6816 if (grouped_load)
6818 first_stmt = GROUP_FIRST_ELEMENT (stmt_info);
6819 /* For SLP vectorization we directly vectorize a subchain
6820 without permutation. */
6821 if (slp && ! SLP_TREE_LOAD_PERMUTATION (slp_node).exists ())
6822 first_stmt = SLP_TREE_SCALAR_STMTS (slp_node)[0];
6823 /* For BB vectorization always use the first stmt to base
6824 the data ref pointer on. */
6825 if (bb_vinfo)
6826 first_stmt_for_drptr = SLP_TREE_SCALAR_STMTS (slp_node)[0];
6828 /* Check if the chain of loads is already vectorized. */
6829 if (STMT_VINFO_VEC_STMT (vinfo_for_stmt (first_stmt))
6830 /* For SLP we would need to copy over SLP_TREE_VEC_STMTS.
6831 ??? But we can only do so if there is exactly one
6832 as we have no way to get at the rest. Leave the CSE
6833 opportunity alone.
6834 ??? With the group load eventually participating
6835 in multiple different permutations (having multiple
6836 slp nodes which refer to the same group) the CSE
6837 is even wrong code. See PR56270. */
6838 && !slp)
6840 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
6841 return true;
6843 first_dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (first_stmt));
6844 group_size = GROUP_SIZE (vinfo_for_stmt (first_stmt));
6845 group_gap_adj = 0;
6847 /* VEC_NUM is the number of vect stmts to be created for this group. */
6848 if (slp)
6850 grouped_load = false;
6851 /* For SLP permutation support we need to load the whole group,
6852 not only the number of vector stmts the permutation result
6853 fits in. */
6854 if (slp_perm)
6855 vec_num = (group_size * vf + nunits - 1) / nunits;
6856 else
6857 vec_num = SLP_TREE_NUMBER_OF_VEC_STMTS (slp_node);
6858 group_gap_adj = vf * group_size - nunits * vec_num;
6860 else
6861 vec_num = group_size;
6863 else
6865 first_stmt = stmt;
6866 first_dr = dr;
6867 group_size = vec_num = 1;
6868 group_gap_adj = 0;
6871 alignment_support_scheme = vect_supportable_dr_alignment (first_dr, false);
6872 gcc_assert (alignment_support_scheme);
6873 /* Targets with load-lane instructions must not require explicit
6874 realignment. */
6875 gcc_assert (!load_lanes_p
6876 || alignment_support_scheme == dr_aligned
6877 || alignment_support_scheme == dr_unaligned_supported);
6879 /* In case the vectorization factor (VF) is bigger than the number
6880 of elements that we can fit in a vectype (nunits), we have to generate
6881 more than one vector stmt - i.e - we need to "unroll" the
6882 vector stmt by a factor VF/nunits. In doing so, we record a pointer
6883 from one copy of the vector stmt to the next, in the field
6884 STMT_VINFO_RELATED_STMT. This is necessary in order to allow following
6885 stages to find the correct vector defs to be used when vectorizing
6886 stmts that use the defs of the current stmt. The example below
6887 illustrates the vectorization process when VF=16 and nunits=4 (i.e., we
6888 need to create 4 vectorized stmts):
6890 before vectorization:
6891 RELATED_STMT VEC_STMT
6892 S1: x = memref - -
6893 S2: z = x + 1 - -
6895 step 1: vectorize stmt S1:
6896 We first create the vector stmt VS1_0, and, as usual, record a
6897 pointer to it in the STMT_VINFO_VEC_STMT of the scalar stmt S1.
6898 Next, we create the vector stmt VS1_1, and record a pointer to
6899 it in the STMT_VINFO_RELATED_STMT of the vector stmt VS1_0.
6900 Similarly, for VS1_2 and VS1_3. This is the resulting chain of
6901 stmts and pointers:
6902 RELATED_STMT VEC_STMT
6903 VS1_0: vx0 = memref0 VS1_1 -
6904 VS1_1: vx1 = memref1 VS1_2 -
6905 VS1_2: vx2 = memref2 VS1_3 -
6906 VS1_3: vx3 = memref3 - -
6907 S1: x = load - VS1_0
6908 S2: z = x + 1 - -
6910 See in documentation in vect_get_vec_def_for_stmt_copy for how the
6911 information we recorded in RELATED_STMT field is used to vectorize
6912 stmt S2. */
6914 /* In case of interleaving (non-unit grouped access):
6916 S1: x2 = &base + 2
6917 S2: x0 = &base
6918 S3: x1 = &base + 1
6919 S4: x3 = &base + 3
6921 Vectorized loads are created in the order of memory accesses
6922 starting from the access of the first stmt of the chain:
6924 VS1: vx0 = &base
6925 VS2: vx1 = &base + vec_size*1
6926 VS3: vx3 = &base + vec_size*2
6927 VS4: vx4 = &base + vec_size*3
6929 Then permutation statements are generated:
6931 VS5: vx5 = VEC_PERM_EXPR < vx0, vx1, { 0, 2, ..., i*2 } >
6932 VS6: vx6 = VEC_PERM_EXPR < vx0, vx1, { 1, 3, ..., i*2+1 } >
6935 And they are put in STMT_VINFO_VEC_STMT of the corresponding scalar stmts
6936 (the order of the data-refs in the output of vect_permute_load_chain
6937 corresponds to the order of scalar stmts in the interleaving chain - see
6938 the documentation of vect_permute_load_chain()).
6939 The generation of permutation stmts and recording them in
6940 STMT_VINFO_VEC_STMT is done in vect_transform_grouped_load().
6942 In case of both multiple types and interleaving, the vector loads and
6943 permutation stmts above are created for every copy. The result vector
6944 stmts are put in STMT_VINFO_VEC_STMT for the first copy and in the
6945 corresponding STMT_VINFO_RELATED_STMT for the next copies. */
6947 /* If the data reference is aligned (dr_aligned) or potentially unaligned
6948 on a target that supports unaligned accesses (dr_unaligned_supported)
6949 we generate the following code:
6950 p = initial_addr;
6951 indx = 0;
6952 loop {
6953 p = p + indx * vectype_size;
6954 vec_dest = *(p);
6955 indx = indx + 1;
6958 Otherwise, the data reference is potentially unaligned on a target that
6959 does not support unaligned accesses (dr_explicit_realign_optimized) -
6960 then generate the following code, in which the data in each iteration is
6961 obtained by two vector loads, one from the previous iteration, and one
6962 from the current iteration:
6963 p1 = initial_addr;
6964 msq_init = *(floor(p1))
6965 p2 = initial_addr + VS - 1;
6966 realignment_token = call target_builtin;
6967 indx = 0;
6968 loop {
6969 p2 = p2 + indx * vectype_size
6970 lsq = *(floor(p2))
6971 vec_dest = realign_load (msq, lsq, realignment_token)
6972 indx = indx + 1;
6973 msq = lsq;
6974 } */
6976 /* If the misalignment remains the same throughout the execution of the
6977 loop, we can create the init_addr and permutation mask at the loop
6978 preheader. Otherwise, it needs to be created inside the loop.
6979 This can only occur when vectorizing memory accesses in the inner-loop
6980 nested within an outer-loop that is being vectorized. */
6982 if (nested_in_vect_loop
6983 && (TREE_INT_CST_LOW (DR_STEP (dr))
6984 % GET_MODE_SIZE (TYPE_MODE (vectype)) != 0))
6986 gcc_assert (alignment_support_scheme != dr_explicit_realign_optimized);
6987 compute_in_loop = true;
6990 if ((alignment_support_scheme == dr_explicit_realign_optimized
6991 || alignment_support_scheme == dr_explicit_realign)
6992 && !compute_in_loop)
6994 msq = vect_setup_realignment (first_stmt, gsi, &realignment_token,
6995 alignment_support_scheme, NULL_TREE,
6996 &at_loop);
6997 if (alignment_support_scheme == dr_explicit_realign_optimized)
6999 phi = as_a <gphi *> (SSA_NAME_DEF_STMT (msq));
7000 byte_offset = size_binop (MINUS_EXPR, TYPE_SIZE_UNIT (vectype),
7001 size_one_node);
7004 else
7005 at_loop = loop;
7007 if (negative)
7008 offset = size_int (-TYPE_VECTOR_SUBPARTS (vectype) + 1);
7010 if (load_lanes_p)
7011 aggr_type = build_array_type_nelts (elem_type, vec_num * nunits);
7012 else
7013 aggr_type = vectype;
7015 prev_stmt_info = NULL;
7016 for (j = 0; j < ncopies; j++)
7018 /* 1. Create the vector or array pointer update chain. */
7019 if (j == 0)
7021 bool simd_lane_access_p
7022 = STMT_VINFO_SIMD_LANE_ACCESS_P (stmt_info);
7023 if (simd_lane_access_p
7024 && TREE_CODE (DR_BASE_ADDRESS (first_dr)) == ADDR_EXPR
7025 && VAR_P (TREE_OPERAND (DR_BASE_ADDRESS (first_dr), 0))
7026 && integer_zerop (DR_OFFSET (first_dr))
7027 && integer_zerop (DR_INIT (first_dr))
7028 && alias_sets_conflict_p (get_alias_set (aggr_type),
7029 get_alias_set (DR_REF (first_dr)))
7030 && (alignment_support_scheme == dr_aligned
7031 || alignment_support_scheme == dr_unaligned_supported))
7033 dataref_ptr = unshare_expr (DR_BASE_ADDRESS (first_dr));
7034 dataref_offset = build_int_cst (reference_alias_ptr_type
7035 (DR_REF (first_dr)), 0);
7036 inv_p = false;
7038 else if (first_stmt_for_drptr
7039 && first_stmt != first_stmt_for_drptr)
7041 dataref_ptr
7042 = vect_create_data_ref_ptr (first_stmt_for_drptr, aggr_type,
7043 at_loop, offset, &dummy, gsi,
7044 &ptr_incr, simd_lane_access_p,
7045 &inv_p, byte_offset);
7046 /* Adjust the pointer by the difference to first_stmt. */
7047 data_reference_p ptrdr
7048 = STMT_VINFO_DATA_REF (vinfo_for_stmt (first_stmt_for_drptr));
7049 tree diff = fold_convert (sizetype,
7050 size_binop (MINUS_EXPR,
7051 DR_INIT (first_dr),
7052 DR_INIT (ptrdr)));
7053 dataref_ptr = bump_vector_ptr (dataref_ptr, ptr_incr, gsi,
7054 stmt, diff);
7056 else
7057 dataref_ptr
7058 = vect_create_data_ref_ptr (first_stmt, aggr_type, at_loop,
7059 offset, &dummy, gsi, &ptr_incr,
7060 simd_lane_access_p, &inv_p,
7061 byte_offset);
7063 else if (dataref_offset)
7064 dataref_offset = int_const_binop (PLUS_EXPR, dataref_offset,
7065 TYPE_SIZE_UNIT (aggr_type));
7066 else
7067 dataref_ptr = bump_vector_ptr (dataref_ptr, ptr_incr, gsi, stmt,
7068 TYPE_SIZE_UNIT (aggr_type));
7070 if (grouped_load || slp_perm)
7071 dr_chain.create (vec_num);
7073 if (load_lanes_p)
7075 tree vec_array;
7077 vec_array = create_vector_array (vectype, vec_num);
7079 /* Emit:
7080 VEC_ARRAY = LOAD_LANES (MEM_REF[...all elements...]). */
7081 data_ref = create_array_ref (aggr_type, dataref_ptr, first_dr);
7082 new_stmt = gimple_build_call_internal (IFN_LOAD_LANES, 1, data_ref);
7083 gimple_call_set_lhs (new_stmt, vec_array);
7084 vect_finish_stmt_generation (stmt, new_stmt, gsi);
7086 /* Extract each vector into an SSA_NAME. */
7087 for (i = 0; i < vec_num; i++)
7089 new_temp = read_vector_array (stmt, gsi, scalar_dest,
7090 vec_array, i);
7091 dr_chain.quick_push (new_temp);
7094 /* Record the mapping between SSA_NAMEs and statements. */
7095 vect_record_grouped_load_vectors (stmt, dr_chain);
7097 else
7099 for (i = 0; i < vec_num; i++)
7101 if (i > 0)
7102 dataref_ptr = bump_vector_ptr (dataref_ptr, ptr_incr, gsi,
7103 stmt, NULL_TREE);
7105 /* 2. Create the vector-load in the loop. */
7106 switch (alignment_support_scheme)
7108 case dr_aligned:
7109 case dr_unaligned_supported:
7111 unsigned int align, misalign;
7113 data_ref
7114 = fold_build2 (MEM_REF, vectype, dataref_ptr,
7115 dataref_offset
7116 ? dataref_offset
7117 : build_int_cst (reference_alias_ptr_type
7118 (DR_REF (first_dr)), 0));
7119 align = TYPE_ALIGN_UNIT (vectype);
7120 if (alignment_support_scheme == dr_aligned)
7122 gcc_assert (aligned_access_p (first_dr));
7123 misalign = 0;
7125 else if (DR_MISALIGNMENT (first_dr) == -1)
7127 if (DR_VECT_AUX (first_dr)->base_element_aligned)
7128 align = TYPE_ALIGN_UNIT (elem_type);
7129 else
7130 align = (get_object_alignment (DR_REF (first_dr))
7131 / BITS_PER_UNIT);
7132 misalign = 0;
7133 TREE_TYPE (data_ref)
7134 = build_aligned_type (TREE_TYPE (data_ref),
7135 align * BITS_PER_UNIT);
7137 else
7139 TREE_TYPE (data_ref)
7140 = build_aligned_type (TREE_TYPE (data_ref),
7141 TYPE_ALIGN (elem_type));
7142 misalign = DR_MISALIGNMENT (first_dr);
7144 if (dataref_offset == NULL_TREE
7145 && TREE_CODE (dataref_ptr) == SSA_NAME)
7146 set_ptr_info_alignment (get_ptr_info (dataref_ptr),
7147 align, misalign);
7148 break;
7150 case dr_explicit_realign:
7152 tree ptr, bump;
7154 tree vs = size_int (TYPE_VECTOR_SUBPARTS (vectype));
7156 if (compute_in_loop)
7157 msq = vect_setup_realignment (first_stmt, gsi,
7158 &realignment_token,
7159 dr_explicit_realign,
7160 dataref_ptr, NULL);
7162 if (TREE_CODE (dataref_ptr) == SSA_NAME)
7163 ptr = copy_ssa_name (dataref_ptr);
7164 else
7165 ptr = make_ssa_name (TREE_TYPE (dataref_ptr));
7166 new_stmt = gimple_build_assign
7167 (ptr, BIT_AND_EXPR, dataref_ptr,
7168 build_int_cst
7169 (TREE_TYPE (dataref_ptr),
7170 -(HOST_WIDE_INT)TYPE_ALIGN_UNIT (vectype)));
7171 vect_finish_stmt_generation (stmt, new_stmt, gsi);
7172 data_ref
7173 = build2 (MEM_REF, vectype, ptr,
7174 build_int_cst (reference_alias_ptr_type
7175 (DR_REF (first_dr)), 0));
7176 vec_dest = vect_create_destination_var (scalar_dest,
7177 vectype);
7178 new_stmt = gimple_build_assign (vec_dest, data_ref);
7179 new_temp = make_ssa_name (vec_dest, new_stmt);
7180 gimple_assign_set_lhs (new_stmt, new_temp);
7181 gimple_set_vdef (new_stmt, gimple_vdef (stmt));
7182 gimple_set_vuse (new_stmt, gimple_vuse (stmt));
7183 vect_finish_stmt_generation (stmt, new_stmt, gsi);
7184 msq = new_temp;
7186 bump = size_binop (MULT_EXPR, vs,
7187 TYPE_SIZE_UNIT (elem_type));
7188 bump = size_binop (MINUS_EXPR, bump, size_one_node);
7189 ptr = bump_vector_ptr (dataref_ptr, NULL, gsi, stmt, bump);
7190 new_stmt = gimple_build_assign
7191 (NULL_TREE, BIT_AND_EXPR, ptr,
7192 build_int_cst
7193 (TREE_TYPE (ptr),
7194 -(HOST_WIDE_INT)TYPE_ALIGN_UNIT (vectype)));
7195 ptr = copy_ssa_name (ptr, new_stmt);
7196 gimple_assign_set_lhs (new_stmt, ptr);
7197 vect_finish_stmt_generation (stmt, new_stmt, gsi);
7198 data_ref
7199 = build2 (MEM_REF, vectype, ptr,
7200 build_int_cst (reference_alias_ptr_type
7201 (DR_REF (first_dr)), 0));
7202 break;
7204 case dr_explicit_realign_optimized:
7205 if (TREE_CODE (dataref_ptr) == SSA_NAME)
7206 new_temp = copy_ssa_name (dataref_ptr);
7207 else
7208 new_temp = make_ssa_name (TREE_TYPE (dataref_ptr));
7209 new_stmt = gimple_build_assign
7210 (new_temp, BIT_AND_EXPR, dataref_ptr,
7211 build_int_cst
7212 (TREE_TYPE (dataref_ptr),
7213 -(HOST_WIDE_INT)TYPE_ALIGN_UNIT (vectype)));
7214 vect_finish_stmt_generation (stmt, new_stmt, gsi);
7215 data_ref
7216 = build2 (MEM_REF, vectype, new_temp,
7217 build_int_cst (reference_alias_ptr_type
7218 (DR_REF (first_dr)), 0));
7219 break;
7220 default:
7221 gcc_unreachable ();
7223 vec_dest = vect_create_destination_var (scalar_dest, vectype);
7224 new_stmt = gimple_build_assign (vec_dest, data_ref);
7225 new_temp = make_ssa_name (vec_dest, new_stmt);
7226 gimple_assign_set_lhs (new_stmt, new_temp);
7227 vect_finish_stmt_generation (stmt, new_stmt, gsi);
7229 /* 3. Handle explicit realignment if necessary/supported.
7230 Create in loop:
7231 vec_dest = realign_load (msq, lsq, realignment_token) */
7232 if (alignment_support_scheme == dr_explicit_realign_optimized
7233 || alignment_support_scheme == dr_explicit_realign)
7235 lsq = gimple_assign_lhs (new_stmt);
7236 if (!realignment_token)
7237 realignment_token = dataref_ptr;
7238 vec_dest = vect_create_destination_var (scalar_dest, vectype);
7239 new_stmt = gimple_build_assign (vec_dest, REALIGN_LOAD_EXPR,
7240 msq, lsq, realignment_token);
7241 new_temp = make_ssa_name (vec_dest, new_stmt);
7242 gimple_assign_set_lhs (new_stmt, new_temp);
7243 vect_finish_stmt_generation (stmt, new_stmt, gsi);
7245 if (alignment_support_scheme == dr_explicit_realign_optimized)
7247 gcc_assert (phi);
7248 if (i == vec_num - 1 && j == ncopies - 1)
7249 add_phi_arg (phi, lsq,
7250 loop_latch_edge (containing_loop),
7251 UNKNOWN_LOCATION);
7252 msq = lsq;
7256 /* 4. Handle invariant-load. */
7257 if (inv_p && !bb_vinfo)
7259 gcc_assert (!grouped_load);
7260 /* If we have versioned for aliasing or the loop doesn't
7261 have any data dependencies that would preclude this,
7262 then we are sure this is a loop invariant load and
7263 thus we can insert it on the preheader edge. */
7264 if (LOOP_VINFO_NO_DATA_DEPENDENCIES (loop_vinfo)
7265 && !nested_in_vect_loop
7266 && hoist_defs_of_uses (stmt, loop))
7268 if (dump_enabled_p ())
7270 dump_printf_loc (MSG_NOTE, vect_location,
7271 "hoisting out of the vectorized "
7272 "loop: ");
7273 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
7275 tree tem = copy_ssa_name (scalar_dest);
7276 gsi_insert_on_edge_immediate
7277 (loop_preheader_edge (loop),
7278 gimple_build_assign (tem,
7279 unshare_expr
7280 (gimple_assign_rhs1 (stmt))));
7281 new_temp = vect_init_vector (stmt, tem, vectype, NULL);
7282 new_stmt = SSA_NAME_DEF_STMT (new_temp);
7283 set_vinfo_for_stmt (new_stmt,
7284 new_stmt_vec_info (new_stmt, vinfo));
7286 else
7288 gimple_stmt_iterator gsi2 = *gsi;
7289 gsi_next (&gsi2);
7290 new_temp = vect_init_vector (stmt, scalar_dest,
7291 vectype, &gsi2);
7292 new_stmt = SSA_NAME_DEF_STMT (new_temp);
7296 if (negative)
7298 tree perm_mask = perm_mask_for_reverse (vectype);
7299 new_temp = permute_vec_elements (new_temp, new_temp,
7300 perm_mask, stmt, gsi);
7301 new_stmt = SSA_NAME_DEF_STMT (new_temp);
7304 /* Collect vector loads and later create their permutation in
7305 vect_transform_grouped_load (). */
7306 if (grouped_load || slp_perm)
7307 dr_chain.quick_push (new_temp);
7309 /* Store vector loads in the corresponding SLP_NODE. */
7310 if (slp && !slp_perm)
7311 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
7313 /* Bump the vector pointer to account for a gap or for excess
7314 elements loaded for a permuted SLP load. */
7315 if (group_gap_adj != 0)
7317 bool ovf;
7318 tree bump
7319 = wide_int_to_tree (sizetype,
7320 wi::smul (TYPE_SIZE_UNIT (elem_type),
7321 group_gap_adj, &ovf));
7322 dataref_ptr = bump_vector_ptr (dataref_ptr, ptr_incr, gsi,
7323 stmt, bump);
7327 if (slp && !slp_perm)
7328 continue;
7330 if (slp_perm)
7332 if (!vect_transform_slp_perm_load (slp_node, dr_chain, gsi, vf,
7333 slp_node_instance, false))
7335 dr_chain.release ();
7336 return false;
7339 else
7341 if (grouped_load)
7343 if (!load_lanes_p)
7344 vect_transform_grouped_load (stmt, dr_chain, group_size, gsi);
7345 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
7347 else
7349 if (j == 0)
7350 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
7351 else
7352 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
7353 prev_stmt_info = vinfo_for_stmt (new_stmt);
7356 dr_chain.release ();
7359 return true;
7362 /* Function vect_is_simple_cond.
7364 Input:
7365 LOOP - the loop that is being vectorized.
7366 COND - Condition that is checked for simple use.
7368 Output:
7369 *COMP_VECTYPE - the vector type for the comparison.
7371 Returns whether a COND can be vectorized. Checks whether
7372 condition operands are supportable using vec_is_simple_use. */
7374 static bool
7375 vect_is_simple_cond (tree cond, vec_info *vinfo, tree *comp_vectype)
7377 tree lhs, rhs;
7378 enum vect_def_type dt;
7379 tree vectype1 = NULL_TREE, vectype2 = NULL_TREE;
7381 /* Mask case. */
7382 if (TREE_CODE (cond) == SSA_NAME
7383 && TREE_CODE (TREE_TYPE (cond)) == BOOLEAN_TYPE)
7385 gimple *lhs_def_stmt = SSA_NAME_DEF_STMT (cond);
7386 if (!vect_is_simple_use (cond, vinfo, &lhs_def_stmt,
7387 &dt, comp_vectype)
7388 || !*comp_vectype
7389 || !VECTOR_BOOLEAN_TYPE_P (*comp_vectype))
7390 return false;
7391 return true;
7394 if (!COMPARISON_CLASS_P (cond))
7395 return false;
7397 lhs = TREE_OPERAND (cond, 0);
7398 rhs = TREE_OPERAND (cond, 1);
7400 if (TREE_CODE (lhs) == SSA_NAME)
7402 gimple *lhs_def_stmt = SSA_NAME_DEF_STMT (lhs);
7403 if (!vect_is_simple_use (lhs, vinfo, &lhs_def_stmt, &dt, &vectype1))
7404 return false;
7406 else if (TREE_CODE (lhs) != INTEGER_CST && TREE_CODE (lhs) != REAL_CST
7407 && TREE_CODE (lhs) != FIXED_CST)
7408 return false;
7410 if (TREE_CODE (rhs) == SSA_NAME)
7412 gimple *rhs_def_stmt = SSA_NAME_DEF_STMT (rhs);
7413 if (!vect_is_simple_use (rhs, vinfo, &rhs_def_stmt, &dt, &vectype2))
7414 return false;
7416 else if (TREE_CODE (rhs) != INTEGER_CST && TREE_CODE (rhs) != REAL_CST
7417 && TREE_CODE (rhs) != FIXED_CST)
7418 return false;
7420 if (vectype1 && vectype2
7421 && TYPE_VECTOR_SUBPARTS (vectype1) != TYPE_VECTOR_SUBPARTS (vectype2))
7422 return false;
7424 *comp_vectype = vectype1 ? vectype1 : vectype2;
7425 return true;
7428 /* vectorizable_condition.
7430 Check if STMT is conditional modify expression that can be vectorized.
7431 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
7432 stmt using VEC_COND_EXPR to replace it, put it in VEC_STMT, and insert it
7433 at GSI.
7435 When STMT is vectorized as nested cycle, REDUC_DEF is the vector variable
7436 to be used at REDUC_INDEX (in then clause if REDUC_INDEX is 1, and in
7437 else clause if it is 2).
7439 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
7441 bool
7442 vectorizable_condition (gimple *stmt, gimple_stmt_iterator *gsi,
7443 gimple **vec_stmt, tree reduc_def, int reduc_index,
7444 slp_tree slp_node)
7446 tree scalar_dest = NULL_TREE;
7447 tree vec_dest = NULL_TREE;
7448 tree cond_expr, then_clause, else_clause;
7449 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
7450 tree comp_vectype = NULL_TREE;
7451 tree vec_cond_lhs = NULL_TREE, vec_cond_rhs = NULL_TREE;
7452 tree vec_then_clause = NULL_TREE, vec_else_clause = NULL_TREE;
7453 tree vec_compare;
7454 tree new_temp;
7455 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
7456 enum vect_def_type dt, dts[4];
7457 int ncopies;
7458 enum tree_code code;
7459 stmt_vec_info prev_stmt_info = NULL;
7460 int i, j;
7461 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
7462 vec<tree> vec_oprnds0 = vNULL;
7463 vec<tree> vec_oprnds1 = vNULL;
7464 vec<tree> vec_oprnds2 = vNULL;
7465 vec<tree> vec_oprnds3 = vNULL;
7466 tree vec_cmp_type;
7467 bool masked = false;
7469 if (reduc_index && STMT_SLP_TYPE (stmt_info))
7470 return false;
7472 if (STMT_VINFO_VEC_REDUCTION_TYPE (stmt_info) == TREE_CODE_REDUCTION)
7474 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
7475 return false;
7477 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def
7478 && !(STMT_VINFO_DEF_TYPE (stmt_info) == vect_nested_cycle
7479 && reduc_def))
7480 return false;
7482 /* FORNOW: not yet supported. */
7483 if (STMT_VINFO_LIVE_P (stmt_info))
7485 if (dump_enabled_p ())
7486 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
7487 "value used after loop.\n");
7488 return false;
7492 /* Is vectorizable conditional operation? */
7493 if (!is_gimple_assign (stmt))
7494 return false;
7496 code = gimple_assign_rhs_code (stmt);
7498 if (code != COND_EXPR)
7499 return false;
7501 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
7502 int nunits = TYPE_VECTOR_SUBPARTS (vectype);
7503 tree vectype1 = NULL_TREE, vectype2 = NULL_TREE;
7505 if (slp_node || PURE_SLP_STMT (stmt_info))
7506 ncopies = 1;
7507 else
7508 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits;
7510 gcc_assert (ncopies >= 1);
7511 if (reduc_index && ncopies > 1)
7512 return false; /* FORNOW */
7514 cond_expr = gimple_assign_rhs1 (stmt);
7515 then_clause = gimple_assign_rhs2 (stmt);
7516 else_clause = gimple_assign_rhs3 (stmt);
7518 if (!vect_is_simple_cond (cond_expr, stmt_info->vinfo, &comp_vectype)
7519 || !comp_vectype)
7520 return false;
7522 gimple *def_stmt;
7523 if (!vect_is_simple_use (then_clause, stmt_info->vinfo, &def_stmt, &dt,
7524 &vectype1))
7525 return false;
7526 if (!vect_is_simple_use (else_clause, stmt_info->vinfo, &def_stmt, &dt,
7527 &vectype2))
7528 return false;
7530 if (vectype1 && !useless_type_conversion_p (vectype, vectype1))
7531 return false;
7533 if (vectype2 && !useless_type_conversion_p (vectype, vectype2))
7534 return false;
7536 masked = !COMPARISON_CLASS_P (cond_expr);
7537 vec_cmp_type = build_same_sized_truth_vector_type (comp_vectype);
7539 if (vec_cmp_type == NULL_TREE)
7540 return false;
7542 if (!vec_stmt)
7544 STMT_VINFO_TYPE (stmt_info) = condition_vec_info_type;
7545 return expand_vec_cond_expr_p (vectype, comp_vectype);
7548 /* Transform. */
7550 if (!slp_node)
7552 vec_oprnds0.create (1);
7553 vec_oprnds1.create (1);
7554 vec_oprnds2.create (1);
7555 vec_oprnds3.create (1);
7558 /* Handle def. */
7559 scalar_dest = gimple_assign_lhs (stmt);
7560 vec_dest = vect_create_destination_var (scalar_dest, vectype);
7562 /* Handle cond expr. */
7563 for (j = 0; j < ncopies; j++)
7565 gassign *new_stmt = NULL;
7566 if (j == 0)
7568 if (slp_node)
7570 auto_vec<tree, 4> ops;
7571 auto_vec<vec<tree>, 4> vec_defs;
7573 if (masked)
7574 ops.safe_push (cond_expr);
7575 else
7577 ops.safe_push (TREE_OPERAND (cond_expr, 0));
7578 ops.safe_push (TREE_OPERAND (cond_expr, 1));
7580 ops.safe_push (then_clause);
7581 ops.safe_push (else_clause);
7582 vect_get_slp_defs (ops, slp_node, &vec_defs, -1);
7583 vec_oprnds3 = vec_defs.pop ();
7584 vec_oprnds2 = vec_defs.pop ();
7585 if (!masked)
7586 vec_oprnds1 = vec_defs.pop ();
7587 vec_oprnds0 = vec_defs.pop ();
7589 ops.release ();
7590 vec_defs.release ();
7592 else
7594 gimple *gtemp;
7595 if (masked)
7597 vec_cond_lhs
7598 = vect_get_vec_def_for_operand (cond_expr, stmt,
7599 comp_vectype);
7600 vect_is_simple_use (cond_expr, stmt_info->vinfo,
7601 &gtemp, &dts[0]);
7603 else
7605 vec_cond_lhs =
7606 vect_get_vec_def_for_operand (TREE_OPERAND (cond_expr, 0),
7607 stmt, comp_vectype);
7608 vect_is_simple_use (TREE_OPERAND (cond_expr, 0),
7609 loop_vinfo, &gtemp, &dts[0]);
7611 vec_cond_rhs =
7612 vect_get_vec_def_for_operand (TREE_OPERAND (cond_expr, 1),
7613 stmt, comp_vectype);
7614 vect_is_simple_use (TREE_OPERAND (cond_expr, 1),
7615 loop_vinfo, &gtemp, &dts[1]);
7617 if (reduc_index == 1)
7618 vec_then_clause = reduc_def;
7619 else
7621 vec_then_clause = vect_get_vec_def_for_operand (then_clause,
7622 stmt);
7623 vect_is_simple_use (then_clause, loop_vinfo,
7624 &gtemp, &dts[2]);
7626 if (reduc_index == 2)
7627 vec_else_clause = reduc_def;
7628 else
7630 vec_else_clause = vect_get_vec_def_for_operand (else_clause,
7631 stmt);
7632 vect_is_simple_use (else_clause, loop_vinfo, &gtemp, &dts[3]);
7636 else
7638 vec_cond_lhs
7639 = vect_get_vec_def_for_stmt_copy (dts[0],
7640 vec_oprnds0.pop ());
7641 if (!masked)
7642 vec_cond_rhs
7643 = vect_get_vec_def_for_stmt_copy (dts[1],
7644 vec_oprnds1.pop ());
7646 vec_then_clause = vect_get_vec_def_for_stmt_copy (dts[2],
7647 vec_oprnds2.pop ());
7648 vec_else_clause = vect_get_vec_def_for_stmt_copy (dts[3],
7649 vec_oprnds3.pop ());
7652 if (!slp_node)
7654 vec_oprnds0.quick_push (vec_cond_lhs);
7655 if (!masked)
7656 vec_oprnds1.quick_push (vec_cond_rhs);
7657 vec_oprnds2.quick_push (vec_then_clause);
7658 vec_oprnds3.quick_push (vec_else_clause);
7661 /* Arguments are ready. Create the new vector stmt. */
7662 FOR_EACH_VEC_ELT (vec_oprnds0, i, vec_cond_lhs)
7664 vec_then_clause = vec_oprnds2[i];
7665 vec_else_clause = vec_oprnds3[i];
7667 if (masked)
7668 vec_compare = vec_cond_lhs;
7669 else
7671 vec_cond_rhs = vec_oprnds1[i];
7672 vec_compare = build2 (TREE_CODE (cond_expr), vec_cmp_type,
7673 vec_cond_lhs, vec_cond_rhs);
7675 new_temp = make_ssa_name (vec_dest);
7676 new_stmt = gimple_build_assign (new_temp, VEC_COND_EXPR,
7677 vec_compare, vec_then_clause,
7678 vec_else_clause);
7679 vect_finish_stmt_generation (stmt, new_stmt, gsi);
7680 if (slp_node)
7681 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
7684 if (slp_node)
7685 continue;
7687 if (j == 0)
7688 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
7689 else
7690 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
7692 prev_stmt_info = vinfo_for_stmt (new_stmt);
7695 vec_oprnds0.release ();
7696 vec_oprnds1.release ();
7697 vec_oprnds2.release ();
7698 vec_oprnds3.release ();
7700 return true;
7703 /* vectorizable_comparison.
7705 Check if STMT is comparison expression that can be vectorized.
7706 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
7707 comparison, put it in VEC_STMT, and insert it at GSI.
7709 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
7711 bool
7712 vectorizable_comparison (gimple *stmt, gimple_stmt_iterator *gsi,
7713 gimple **vec_stmt, tree reduc_def,
7714 slp_tree slp_node)
7716 tree lhs, rhs1, rhs2;
7717 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
7718 tree vectype1 = NULL_TREE, vectype2 = NULL_TREE;
7719 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
7720 tree vec_rhs1 = NULL_TREE, vec_rhs2 = NULL_TREE;
7721 tree new_temp;
7722 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
7723 enum vect_def_type dts[2] = {vect_unknown_def_type, vect_unknown_def_type};
7724 unsigned nunits;
7725 int ncopies;
7726 enum tree_code code;
7727 stmt_vec_info prev_stmt_info = NULL;
7728 int i, j;
7729 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
7730 vec<tree> vec_oprnds0 = vNULL;
7731 vec<tree> vec_oprnds1 = vNULL;
7732 gimple *def_stmt;
7733 tree mask_type;
7734 tree mask;
7736 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
7737 return false;
7739 if (!vectype || !VECTOR_BOOLEAN_TYPE_P (vectype))
7740 return false;
7742 mask_type = vectype;
7743 nunits = TYPE_VECTOR_SUBPARTS (vectype);
7745 if (slp_node || PURE_SLP_STMT (stmt_info))
7746 ncopies = 1;
7747 else
7748 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits;
7750 gcc_assert (ncopies >= 1);
7751 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def
7752 && !(STMT_VINFO_DEF_TYPE (stmt_info) == vect_nested_cycle
7753 && reduc_def))
7754 return false;
7756 if (STMT_VINFO_LIVE_P (stmt_info))
7758 if (dump_enabled_p ())
7759 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
7760 "value used after loop.\n");
7761 return false;
7764 if (!is_gimple_assign (stmt))
7765 return false;
7767 code = gimple_assign_rhs_code (stmt);
7769 if (TREE_CODE_CLASS (code) != tcc_comparison)
7770 return false;
7772 rhs1 = gimple_assign_rhs1 (stmt);
7773 rhs2 = gimple_assign_rhs2 (stmt);
7775 if (!vect_is_simple_use (rhs1, stmt_info->vinfo, &def_stmt,
7776 &dts[0], &vectype1))
7777 return false;
7779 if (!vect_is_simple_use (rhs2, stmt_info->vinfo, &def_stmt,
7780 &dts[1], &vectype2))
7781 return false;
7783 if (vectype1 && vectype2
7784 && TYPE_VECTOR_SUBPARTS (vectype1) != TYPE_VECTOR_SUBPARTS (vectype2))
7785 return false;
7787 vectype = vectype1 ? vectype1 : vectype2;
7789 /* Invariant comparison. */
7790 if (!vectype)
7792 vectype = get_vectype_for_scalar_type (TREE_TYPE (rhs1));
7793 if (TYPE_VECTOR_SUBPARTS (vectype) != nunits)
7794 return false;
7796 else if (nunits != TYPE_VECTOR_SUBPARTS (vectype))
7797 return false;
7799 if (!vec_stmt)
7801 STMT_VINFO_TYPE (stmt_info) = comparison_vec_info_type;
7802 vect_model_simple_cost (stmt_info, ncopies, dts, NULL, NULL);
7803 return expand_vec_cmp_expr_p (vectype, mask_type);
7806 /* Transform. */
7807 if (!slp_node)
7809 vec_oprnds0.create (1);
7810 vec_oprnds1.create (1);
7813 /* Handle def. */
7814 lhs = gimple_assign_lhs (stmt);
7815 mask = vect_create_destination_var (lhs, mask_type);
7817 /* Handle cmp expr. */
7818 for (j = 0; j < ncopies; j++)
7820 gassign *new_stmt = NULL;
7821 if (j == 0)
7823 if (slp_node)
7825 auto_vec<tree, 2> ops;
7826 auto_vec<vec<tree>, 2> vec_defs;
7828 ops.safe_push (rhs1);
7829 ops.safe_push (rhs2);
7830 vect_get_slp_defs (ops, slp_node, &vec_defs, -1);
7831 vec_oprnds1 = vec_defs.pop ();
7832 vec_oprnds0 = vec_defs.pop ();
7834 else
7836 vec_rhs1 = vect_get_vec_def_for_operand (rhs1, stmt, vectype);
7837 vec_rhs2 = vect_get_vec_def_for_operand (rhs2, stmt, vectype);
7840 else
7842 vec_rhs1 = vect_get_vec_def_for_stmt_copy (dts[0],
7843 vec_oprnds0.pop ());
7844 vec_rhs2 = vect_get_vec_def_for_stmt_copy (dts[1],
7845 vec_oprnds1.pop ());
7848 if (!slp_node)
7850 vec_oprnds0.quick_push (vec_rhs1);
7851 vec_oprnds1.quick_push (vec_rhs2);
7854 /* Arguments are ready. Create the new vector stmt. */
7855 FOR_EACH_VEC_ELT (vec_oprnds0, i, vec_rhs1)
7857 vec_rhs2 = vec_oprnds1[i];
7859 new_temp = make_ssa_name (mask);
7860 new_stmt = gimple_build_assign (new_temp, code, vec_rhs1, vec_rhs2);
7861 vect_finish_stmt_generation (stmt, new_stmt, gsi);
7862 if (slp_node)
7863 SLP_TREE_VEC_STMTS (slp_node).quick_push (new_stmt);
7866 if (slp_node)
7867 continue;
7869 if (j == 0)
7870 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
7871 else
7872 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
7874 prev_stmt_info = vinfo_for_stmt (new_stmt);
7877 vec_oprnds0.release ();
7878 vec_oprnds1.release ();
7880 return true;
7883 /* Make sure the statement is vectorizable. */
7885 bool
7886 vect_analyze_stmt (gimple *stmt, bool *need_to_vectorize, slp_tree node)
7888 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
7889 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
7890 enum vect_relevant relevance = STMT_VINFO_RELEVANT (stmt_info);
7891 bool ok;
7892 tree scalar_type, vectype;
7893 gimple *pattern_stmt;
7894 gimple_seq pattern_def_seq;
7896 if (dump_enabled_p ())
7898 dump_printf_loc (MSG_NOTE, vect_location, "==> examining statement: ");
7899 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
7902 if (gimple_has_volatile_ops (stmt))
7904 if (dump_enabled_p ())
7905 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
7906 "not vectorized: stmt has volatile operands\n");
7908 return false;
7911 /* Skip stmts that do not need to be vectorized. In loops this is expected
7912 to include:
7913 - the COND_EXPR which is the loop exit condition
7914 - any LABEL_EXPRs in the loop
7915 - computations that are used only for array indexing or loop control.
7916 In basic blocks we only analyze statements that are a part of some SLP
7917 instance, therefore, all the statements are relevant.
7919 Pattern statement needs to be analyzed instead of the original statement
7920 if the original statement is not relevant. Otherwise, we analyze both
7921 statements. In basic blocks we are called from some SLP instance
7922 traversal, don't analyze pattern stmts instead, the pattern stmts
7923 already will be part of SLP instance. */
7925 pattern_stmt = STMT_VINFO_RELATED_STMT (stmt_info);
7926 if (!STMT_VINFO_RELEVANT_P (stmt_info)
7927 && !STMT_VINFO_LIVE_P (stmt_info))
7929 if (STMT_VINFO_IN_PATTERN_P (stmt_info)
7930 && pattern_stmt
7931 && (STMT_VINFO_RELEVANT_P (vinfo_for_stmt (pattern_stmt))
7932 || STMT_VINFO_LIVE_P (vinfo_for_stmt (pattern_stmt))))
7934 /* Analyze PATTERN_STMT instead of the original stmt. */
7935 stmt = pattern_stmt;
7936 stmt_info = vinfo_for_stmt (pattern_stmt);
7937 if (dump_enabled_p ())
7939 dump_printf_loc (MSG_NOTE, vect_location,
7940 "==> examining pattern statement: ");
7941 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
7944 else
7946 if (dump_enabled_p ())
7947 dump_printf_loc (MSG_NOTE, vect_location, "irrelevant.\n");
7949 return true;
7952 else if (STMT_VINFO_IN_PATTERN_P (stmt_info)
7953 && node == NULL
7954 && pattern_stmt
7955 && (STMT_VINFO_RELEVANT_P (vinfo_for_stmt (pattern_stmt))
7956 || STMT_VINFO_LIVE_P (vinfo_for_stmt (pattern_stmt))))
7958 /* Analyze PATTERN_STMT too. */
7959 if (dump_enabled_p ())
7961 dump_printf_loc (MSG_NOTE, vect_location,
7962 "==> examining pattern statement: ");
7963 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, stmt, 0);
7966 if (!vect_analyze_stmt (pattern_stmt, need_to_vectorize, node))
7967 return false;
7970 if (is_pattern_stmt_p (stmt_info)
7971 && node == NULL
7972 && (pattern_def_seq = STMT_VINFO_PATTERN_DEF_SEQ (stmt_info)))
7974 gimple_stmt_iterator si;
7976 for (si = gsi_start (pattern_def_seq); !gsi_end_p (si); gsi_next (&si))
7978 gimple *pattern_def_stmt = gsi_stmt (si);
7979 if (STMT_VINFO_RELEVANT_P (vinfo_for_stmt (pattern_def_stmt))
7980 || STMT_VINFO_LIVE_P (vinfo_for_stmt (pattern_def_stmt)))
7982 /* Analyze def stmt of STMT if it's a pattern stmt. */
7983 if (dump_enabled_p ())
7985 dump_printf_loc (MSG_NOTE, vect_location,
7986 "==> examining pattern def statement: ");
7987 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, pattern_def_stmt, 0);
7990 if (!vect_analyze_stmt (pattern_def_stmt,
7991 need_to_vectorize, node))
7992 return false;
7997 switch (STMT_VINFO_DEF_TYPE (stmt_info))
7999 case vect_internal_def:
8000 break;
8002 case vect_reduction_def:
8003 case vect_nested_cycle:
8004 gcc_assert (!bb_vinfo
8005 && (relevance == vect_used_in_outer
8006 || relevance == vect_used_in_outer_by_reduction
8007 || relevance == vect_used_by_reduction
8008 || relevance == vect_unused_in_scope));
8009 break;
8011 case vect_induction_def:
8012 case vect_constant_def:
8013 case vect_external_def:
8014 case vect_unknown_def_type:
8015 default:
8016 gcc_unreachable ();
8019 if (bb_vinfo)
8021 gcc_assert (PURE_SLP_STMT (stmt_info));
8023 scalar_type = TREE_TYPE (gimple_get_lhs (stmt));
8024 if (dump_enabled_p ())
8026 dump_printf_loc (MSG_NOTE, vect_location,
8027 "get vectype for scalar type: ");
8028 dump_generic_expr (MSG_NOTE, TDF_SLIM, scalar_type);
8029 dump_printf (MSG_NOTE, "\n");
8032 vectype = get_vectype_for_scalar_type (scalar_type);
8033 if (!vectype)
8035 if (dump_enabled_p ())
8037 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
8038 "not SLPed: unsupported data-type ");
8039 dump_generic_expr (MSG_MISSED_OPTIMIZATION, TDF_SLIM,
8040 scalar_type);
8041 dump_printf (MSG_MISSED_OPTIMIZATION, "\n");
8043 return false;
8046 if (dump_enabled_p ())
8048 dump_printf_loc (MSG_NOTE, vect_location, "vectype: ");
8049 dump_generic_expr (MSG_NOTE, TDF_SLIM, vectype);
8050 dump_printf (MSG_NOTE, "\n");
8053 STMT_VINFO_VECTYPE (stmt_info) = vectype;
8056 if (STMT_VINFO_RELEVANT_P (stmt_info))
8058 gcc_assert (!VECTOR_MODE_P (TYPE_MODE (gimple_expr_type (stmt))));
8059 gcc_assert (STMT_VINFO_VECTYPE (stmt_info)
8060 || (is_gimple_call (stmt)
8061 && gimple_call_lhs (stmt) == NULL_TREE));
8062 *need_to_vectorize = true;
8065 if (PURE_SLP_STMT (stmt_info) && !node)
8067 dump_printf_loc (MSG_NOTE, vect_location,
8068 "handled only by SLP analysis\n");
8069 return true;
8072 ok = true;
8073 if (!bb_vinfo
8074 && (STMT_VINFO_RELEVANT_P (stmt_info)
8075 || STMT_VINFO_DEF_TYPE (stmt_info) == vect_reduction_def))
8076 ok = (vectorizable_simd_clone_call (stmt, NULL, NULL, node)
8077 || vectorizable_conversion (stmt, NULL, NULL, node)
8078 || vectorizable_shift (stmt, NULL, NULL, node)
8079 || vectorizable_operation (stmt, NULL, NULL, node)
8080 || vectorizable_assignment (stmt, NULL, NULL, node)
8081 || vectorizable_load (stmt, NULL, NULL, node, NULL)
8082 || vectorizable_call (stmt, NULL, NULL, node)
8083 || vectorizable_store (stmt, NULL, NULL, node)
8084 || vectorizable_reduction (stmt, NULL, NULL, node)
8085 || vectorizable_condition (stmt, NULL, NULL, NULL, 0, node)
8086 || vectorizable_comparison (stmt, NULL, NULL, NULL, node));
8087 else
8089 if (bb_vinfo)
8090 ok = (vectorizable_simd_clone_call (stmt, NULL, NULL, node)
8091 || vectorizable_conversion (stmt, NULL, NULL, node)
8092 || vectorizable_shift (stmt, NULL, NULL, node)
8093 || vectorizable_operation (stmt, NULL, NULL, node)
8094 || vectorizable_assignment (stmt, NULL, NULL, node)
8095 || vectorizable_load (stmt, NULL, NULL, node, NULL)
8096 || vectorizable_call (stmt, NULL, NULL, node)
8097 || vectorizable_store (stmt, NULL, NULL, node)
8098 || vectorizable_condition (stmt, NULL, NULL, NULL, 0, node)
8099 || vectorizable_comparison (stmt, NULL, NULL, NULL, node));
8102 if (!ok)
8104 if (dump_enabled_p ())
8106 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
8107 "not vectorized: relevant stmt not ");
8108 dump_printf (MSG_MISSED_OPTIMIZATION, "supported: ");
8109 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM, stmt, 0);
8112 return false;
8115 if (bb_vinfo)
8116 return true;
8118 /* Stmts that are (also) "live" (i.e. - that are used out of the loop)
8119 need extra handling, except for vectorizable reductions. */
8120 if (STMT_VINFO_LIVE_P (stmt_info)
8121 && STMT_VINFO_TYPE (stmt_info) != reduc_vec_info_type)
8122 ok = vectorizable_live_operation (stmt, NULL, NULL);
8124 if (!ok)
8126 if (dump_enabled_p ())
8128 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
8129 "not vectorized: live stmt not ");
8130 dump_printf (MSG_MISSED_OPTIMIZATION, "supported: ");
8131 dump_gimple_stmt (MSG_MISSED_OPTIMIZATION, TDF_SLIM, stmt, 0);
8134 return false;
8137 return true;
8141 /* Function vect_transform_stmt.
8143 Create a vectorized stmt to replace STMT, and insert it at BSI. */
8145 bool
8146 vect_transform_stmt (gimple *stmt, gimple_stmt_iterator *gsi,
8147 bool *grouped_store, slp_tree slp_node,
8148 slp_instance slp_node_instance)
8150 bool is_store = false;
8151 gimple *vec_stmt = NULL;
8152 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
8153 bool done;
8155 gimple *old_vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
8157 switch (STMT_VINFO_TYPE (stmt_info))
8159 case type_demotion_vec_info_type:
8160 case type_promotion_vec_info_type:
8161 case type_conversion_vec_info_type:
8162 done = vectorizable_conversion (stmt, gsi, &vec_stmt, slp_node);
8163 gcc_assert (done);
8164 break;
8166 case induc_vec_info_type:
8167 gcc_assert (!slp_node);
8168 done = vectorizable_induction (stmt, gsi, &vec_stmt);
8169 gcc_assert (done);
8170 break;
8172 case shift_vec_info_type:
8173 done = vectorizable_shift (stmt, gsi, &vec_stmt, slp_node);
8174 gcc_assert (done);
8175 break;
8177 case op_vec_info_type:
8178 done = vectorizable_operation (stmt, gsi, &vec_stmt, slp_node);
8179 gcc_assert (done);
8180 break;
8182 case assignment_vec_info_type:
8183 done = vectorizable_assignment (stmt, gsi, &vec_stmt, slp_node);
8184 gcc_assert (done);
8185 break;
8187 case load_vec_info_type:
8188 done = vectorizable_load (stmt, gsi, &vec_stmt, slp_node,
8189 slp_node_instance);
8190 gcc_assert (done);
8191 break;
8193 case store_vec_info_type:
8194 done = vectorizable_store (stmt, gsi, &vec_stmt, slp_node);
8195 gcc_assert (done);
8196 if (STMT_VINFO_GROUPED_ACCESS (stmt_info) && !slp_node)
8198 /* In case of interleaving, the whole chain is vectorized when the
8199 last store in the chain is reached. Store stmts before the last
8200 one are skipped, and there vec_stmt_info shouldn't be freed
8201 meanwhile. */
8202 *grouped_store = true;
8203 if (STMT_VINFO_VEC_STMT (stmt_info))
8204 is_store = true;
8206 else
8207 is_store = true;
8208 break;
8210 case condition_vec_info_type:
8211 done = vectorizable_condition (stmt, gsi, &vec_stmt, NULL, 0, slp_node);
8212 gcc_assert (done);
8213 break;
8215 case comparison_vec_info_type:
8216 done = vectorizable_comparison (stmt, gsi, &vec_stmt, NULL, slp_node);
8217 gcc_assert (done);
8218 break;
8220 case call_vec_info_type:
8221 done = vectorizable_call (stmt, gsi, &vec_stmt, slp_node);
8222 stmt = gsi_stmt (*gsi);
8223 if (is_gimple_call (stmt)
8224 && gimple_call_internal_p (stmt)
8225 && gimple_call_internal_fn (stmt) == IFN_MASK_STORE)
8226 is_store = true;
8227 break;
8229 case call_simd_clone_vec_info_type:
8230 done = vectorizable_simd_clone_call (stmt, gsi, &vec_stmt, slp_node);
8231 stmt = gsi_stmt (*gsi);
8232 break;
8234 case reduc_vec_info_type:
8235 done = vectorizable_reduction (stmt, gsi, &vec_stmt, slp_node);
8236 gcc_assert (done);
8237 break;
8239 default:
8240 if (!STMT_VINFO_LIVE_P (stmt_info))
8242 if (dump_enabled_p ())
8243 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
8244 "stmt not supported.\n");
8245 gcc_unreachable ();
8249 /* Verify SLP vectorization doesn't mess with STMT_VINFO_VEC_STMT.
8250 This would break hybrid SLP vectorization. */
8251 if (slp_node)
8252 gcc_assert (!vec_stmt
8253 && STMT_VINFO_VEC_STMT (stmt_info) == old_vec_stmt);
8255 /* Handle inner-loop stmts whose DEF is used in the loop-nest that
8256 is being vectorized, but outside the immediately enclosing loop. */
8257 if (vec_stmt
8258 && STMT_VINFO_LOOP_VINFO (stmt_info)
8259 && nested_in_vect_loop_p (LOOP_VINFO_LOOP (
8260 STMT_VINFO_LOOP_VINFO (stmt_info)), stmt)
8261 && STMT_VINFO_TYPE (stmt_info) != reduc_vec_info_type
8262 && (STMT_VINFO_RELEVANT (stmt_info) == vect_used_in_outer
8263 || STMT_VINFO_RELEVANT (stmt_info) ==
8264 vect_used_in_outer_by_reduction))
8266 struct loop *innerloop = LOOP_VINFO_LOOP (
8267 STMT_VINFO_LOOP_VINFO (stmt_info))->inner;
8268 imm_use_iterator imm_iter;
8269 use_operand_p use_p;
8270 tree scalar_dest;
8271 gimple *exit_phi;
8273 if (dump_enabled_p ())
8274 dump_printf_loc (MSG_NOTE, vect_location,
8275 "Record the vdef for outer-loop vectorization.\n");
8277 /* Find the relevant loop-exit phi-node, and reord the vec_stmt there
8278 (to be used when vectorizing outer-loop stmts that use the DEF of
8279 STMT). */
8280 if (gimple_code (stmt) == GIMPLE_PHI)
8281 scalar_dest = PHI_RESULT (stmt);
8282 else
8283 scalar_dest = gimple_assign_lhs (stmt);
8285 FOR_EACH_IMM_USE_FAST (use_p, imm_iter, scalar_dest)
8287 if (!flow_bb_inside_loop_p (innerloop, gimple_bb (USE_STMT (use_p))))
8289 exit_phi = USE_STMT (use_p);
8290 STMT_VINFO_VEC_STMT (vinfo_for_stmt (exit_phi)) = vec_stmt;
8295 /* Handle stmts whose DEF is used outside the loop-nest that is
8296 being vectorized. */
8297 if (STMT_VINFO_LIVE_P (stmt_info)
8298 && STMT_VINFO_TYPE (stmt_info) != reduc_vec_info_type)
8300 done = vectorizable_live_operation (stmt, gsi, &vec_stmt);
8301 gcc_assert (done);
8304 if (vec_stmt)
8305 STMT_VINFO_VEC_STMT (stmt_info) = vec_stmt;
8307 return is_store;
8311 /* Remove a group of stores (for SLP or interleaving), free their
8312 stmt_vec_info. */
8314 void
8315 vect_remove_stores (gimple *first_stmt)
8317 gimple *next = first_stmt;
8318 gimple *tmp;
8319 gimple_stmt_iterator next_si;
8321 while (next)
8323 stmt_vec_info stmt_info = vinfo_for_stmt (next);
8325 tmp = GROUP_NEXT_ELEMENT (stmt_info);
8326 if (is_pattern_stmt_p (stmt_info))
8327 next = STMT_VINFO_RELATED_STMT (stmt_info);
8328 /* Free the attached stmt_vec_info and remove the stmt. */
8329 next_si = gsi_for_stmt (next);
8330 unlink_stmt_vdef (next);
8331 gsi_remove (&next_si, true);
8332 release_defs (next);
8333 free_stmt_vec_info (next);
8334 next = tmp;
8339 /* Function new_stmt_vec_info.
8341 Create and initialize a new stmt_vec_info struct for STMT. */
8343 stmt_vec_info
8344 new_stmt_vec_info (gimple *stmt, vec_info *vinfo)
8346 stmt_vec_info res;
8347 res = (stmt_vec_info) xcalloc (1, sizeof (struct _stmt_vec_info));
8349 STMT_VINFO_TYPE (res) = undef_vec_info_type;
8350 STMT_VINFO_STMT (res) = stmt;
8351 res->vinfo = vinfo;
8352 STMT_VINFO_RELEVANT (res) = vect_unused_in_scope;
8353 STMT_VINFO_LIVE_P (res) = false;
8354 STMT_VINFO_VECTYPE (res) = NULL;
8355 STMT_VINFO_VEC_STMT (res) = NULL;
8356 STMT_VINFO_VECTORIZABLE (res) = true;
8357 STMT_VINFO_IN_PATTERN_P (res) = false;
8358 STMT_VINFO_RELATED_STMT (res) = NULL;
8359 STMT_VINFO_PATTERN_DEF_SEQ (res) = NULL;
8360 STMT_VINFO_DATA_REF (res) = NULL;
8361 STMT_VINFO_VEC_REDUCTION_TYPE (res) = TREE_CODE_REDUCTION;
8363 STMT_VINFO_DR_BASE_ADDRESS (res) = NULL;
8364 STMT_VINFO_DR_OFFSET (res) = NULL;
8365 STMT_VINFO_DR_INIT (res) = NULL;
8366 STMT_VINFO_DR_STEP (res) = NULL;
8367 STMT_VINFO_DR_ALIGNED_TO (res) = NULL;
8369 if (gimple_code (stmt) == GIMPLE_PHI
8370 && is_loop_header_bb_p (gimple_bb (stmt)))
8371 STMT_VINFO_DEF_TYPE (res) = vect_unknown_def_type;
8372 else
8373 STMT_VINFO_DEF_TYPE (res) = vect_internal_def;
8375 STMT_VINFO_SAME_ALIGN_REFS (res).create (0);
8376 STMT_SLP_TYPE (res) = loop_vect;
8377 STMT_VINFO_NUM_SLP_USES (res) = 0;
8379 GROUP_FIRST_ELEMENT (res) = NULL;
8380 GROUP_NEXT_ELEMENT (res) = NULL;
8381 GROUP_SIZE (res) = 0;
8382 GROUP_STORE_COUNT (res) = 0;
8383 GROUP_GAP (res) = 0;
8384 GROUP_SAME_DR_STMT (res) = NULL;
8386 return res;
8390 /* Create a hash table for stmt_vec_info. */
8392 void
8393 init_stmt_vec_info_vec (void)
8395 gcc_assert (!stmt_vec_info_vec.exists ());
8396 stmt_vec_info_vec.create (50);
8400 /* Free hash table for stmt_vec_info. */
8402 void
8403 free_stmt_vec_info_vec (void)
8405 unsigned int i;
8406 stmt_vec_info info;
8407 FOR_EACH_VEC_ELT (stmt_vec_info_vec, i, info)
8408 if (info != NULL)
8409 free_stmt_vec_info (STMT_VINFO_STMT (info));
8410 gcc_assert (stmt_vec_info_vec.exists ());
8411 stmt_vec_info_vec.release ();
8415 /* Free stmt vectorization related info. */
8417 void
8418 free_stmt_vec_info (gimple *stmt)
8420 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
8422 if (!stmt_info)
8423 return;
8425 /* Check if this statement has a related "pattern stmt"
8426 (introduced by the vectorizer during the pattern recognition
8427 pass). Free pattern's stmt_vec_info and def stmt's stmt_vec_info
8428 too. */
8429 if (STMT_VINFO_IN_PATTERN_P (stmt_info))
8431 stmt_vec_info patt_info
8432 = vinfo_for_stmt (STMT_VINFO_RELATED_STMT (stmt_info));
8433 if (patt_info)
8435 gimple_seq seq = STMT_VINFO_PATTERN_DEF_SEQ (patt_info);
8436 gimple *patt_stmt = STMT_VINFO_STMT (patt_info);
8437 gimple_set_bb (patt_stmt, NULL);
8438 tree lhs = gimple_get_lhs (patt_stmt);
8439 if (lhs && TREE_CODE (lhs) == SSA_NAME)
8440 release_ssa_name (lhs);
8441 if (seq)
8443 gimple_stmt_iterator si;
8444 for (si = gsi_start (seq); !gsi_end_p (si); gsi_next (&si))
8446 gimple *seq_stmt = gsi_stmt (si);
8447 gimple_set_bb (seq_stmt, NULL);
8448 lhs = gimple_get_lhs (seq_stmt);
8449 if (lhs && TREE_CODE (lhs) == SSA_NAME)
8450 release_ssa_name (lhs);
8451 free_stmt_vec_info (seq_stmt);
8454 free_stmt_vec_info (patt_stmt);
8458 STMT_VINFO_SAME_ALIGN_REFS (stmt_info).release ();
8459 STMT_VINFO_SIMD_CLONE_INFO (stmt_info).release ();
8460 set_vinfo_for_stmt (stmt, NULL);
8461 free (stmt_info);
8465 /* Function get_vectype_for_scalar_type_and_size.
8467 Returns the vector type corresponding to SCALAR_TYPE and SIZE as supported
8468 by the target. */
8470 static tree
8471 get_vectype_for_scalar_type_and_size (tree scalar_type, unsigned size)
8473 machine_mode inner_mode = TYPE_MODE (scalar_type);
8474 machine_mode simd_mode;
8475 unsigned int nbytes = GET_MODE_SIZE (inner_mode);
8476 int nunits;
8477 tree vectype;
8479 if (nbytes == 0)
8480 return NULL_TREE;
8482 if (GET_MODE_CLASS (inner_mode) != MODE_INT
8483 && GET_MODE_CLASS (inner_mode) != MODE_FLOAT)
8484 return NULL_TREE;
8486 /* For vector types of elements whose mode precision doesn't
8487 match their types precision we use a element type of mode
8488 precision. The vectorization routines will have to make sure
8489 they support the proper result truncation/extension.
8490 We also make sure to build vector types with INTEGER_TYPE
8491 component type only. */
8492 if (INTEGRAL_TYPE_P (scalar_type)
8493 && (GET_MODE_BITSIZE (inner_mode) != TYPE_PRECISION (scalar_type)
8494 || TREE_CODE (scalar_type) != INTEGER_TYPE))
8495 scalar_type = build_nonstandard_integer_type (GET_MODE_BITSIZE (inner_mode),
8496 TYPE_UNSIGNED (scalar_type));
8498 /* We shouldn't end up building VECTOR_TYPEs of non-scalar components.
8499 When the component mode passes the above test simply use a type
8500 corresponding to that mode. The theory is that any use that
8501 would cause problems with this will disable vectorization anyway. */
8502 else if (!SCALAR_FLOAT_TYPE_P (scalar_type)
8503 && !INTEGRAL_TYPE_P (scalar_type))
8504 scalar_type = lang_hooks.types.type_for_mode (inner_mode, 1);
8506 /* We can't build a vector type of elements with alignment bigger than
8507 their size. */
8508 else if (nbytes < TYPE_ALIGN_UNIT (scalar_type))
8509 scalar_type = lang_hooks.types.type_for_mode (inner_mode,
8510 TYPE_UNSIGNED (scalar_type));
8512 /* If we felt back to using the mode fail if there was
8513 no scalar type for it. */
8514 if (scalar_type == NULL_TREE)
8515 return NULL_TREE;
8517 /* If no size was supplied use the mode the target prefers. Otherwise
8518 lookup a vector mode of the specified size. */
8519 if (size == 0)
8520 simd_mode = targetm.vectorize.preferred_simd_mode (inner_mode);
8521 else
8522 simd_mode = mode_for_vector (inner_mode, size / nbytes);
8523 nunits = GET_MODE_SIZE (simd_mode) / nbytes;
8524 if (nunits <= 1)
8525 return NULL_TREE;
8527 vectype = build_vector_type (scalar_type, nunits);
8529 if (!VECTOR_MODE_P (TYPE_MODE (vectype))
8530 && !INTEGRAL_MODE_P (TYPE_MODE (vectype)))
8531 return NULL_TREE;
8533 return vectype;
8536 unsigned int current_vector_size;
8538 /* Function get_vectype_for_scalar_type.
8540 Returns the vector type corresponding to SCALAR_TYPE as supported
8541 by the target. */
8543 tree
8544 get_vectype_for_scalar_type (tree scalar_type)
8546 tree vectype;
8547 vectype = get_vectype_for_scalar_type_and_size (scalar_type,
8548 current_vector_size);
8549 if (vectype
8550 && current_vector_size == 0)
8551 current_vector_size = GET_MODE_SIZE (TYPE_MODE (vectype));
8552 return vectype;
8555 /* Function get_mask_type_for_scalar_type.
8557 Returns the mask type corresponding to a result of comparison
8558 of vectors of specified SCALAR_TYPE as supported by target. */
8560 tree
8561 get_mask_type_for_scalar_type (tree scalar_type)
8563 tree vectype = get_vectype_for_scalar_type (scalar_type);
8565 if (!vectype)
8566 return NULL;
8568 return build_truth_vector_type (TYPE_VECTOR_SUBPARTS (vectype),
8569 current_vector_size);
8572 /* Function get_same_sized_vectype
8574 Returns a vector type corresponding to SCALAR_TYPE of size
8575 VECTOR_TYPE if supported by the target. */
8577 tree
8578 get_same_sized_vectype (tree scalar_type, tree vector_type)
8580 if (TREE_CODE (scalar_type) == BOOLEAN_TYPE)
8581 return build_same_sized_truth_vector_type (vector_type);
8583 return get_vectype_for_scalar_type_and_size
8584 (scalar_type, GET_MODE_SIZE (TYPE_MODE (vector_type)));
8587 /* Function vect_is_simple_use.
8589 Input:
8590 VINFO - the vect info of the loop or basic block that is being vectorized.
8591 OPERAND - operand in the loop or bb.
8592 Output:
8593 DEF_STMT - the defining stmt in case OPERAND is an SSA_NAME.
8594 DT - the type of definition
8596 Returns whether a stmt with OPERAND can be vectorized.
8597 For loops, supportable operands are constants, loop invariants, and operands
8598 that are defined by the current iteration of the loop. Unsupportable
8599 operands are those that are defined by a previous iteration of the loop (as
8600 is the case in reduction/induction computations).
8601 For basic blocks, supportable operands are constants and bb invariants.
8602 For now, operands defined outside the basic block are not supported. */
8604 bool
8605 vect_is_simple_use (tree operand, vec_info *vinfo,
8606 gimple **def_stmt, enum vect_def_type *dt)
8608 *def_stmt = NULL;
8609 *dt = vect_unknown_def_type;
8611 if (dump_enabled_p ())
8613 dump_printf_loc (MSG_NOTE, vect_location,
8614 "vect_is_simple_use: operand ");
8615 dump_generic_expr (MSG_NOTE, TDF_SLIM, operand);
8616 dump_printf (MSG_NOTE, "\n");
8619 if (CONSTANT_CLASS_P (operand))
8621 *dt = vect_constant_def;
8622 return true;
8625 if (is_gimple_min_invariant (operand))
8627 *dt = vect_external_def;
8628 return true;
8631 if (TREE_CODE (operand) != SSA_NAME)
8633 if (dump_enabled_p ())
8634 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
8635 "not ssa-name.\n");
8636 return false;
8639 if (SSA_NAME_IS_DEFAULT_DEF (operand))
8641 *dt = vect_external_def;
8642 return true;
8645 *def_stmt = SSA_NAME_DEF_STMT (operand);
8646 if (dump_enabled_p ())
8648 dump_printf_loc (MSG_NOTE, vect_location, "def_stmt: ");
8649 dump_gimple_stmt (MSG_NOTE, TDF_SLIM, *def_stmt, 0);
8652 if (! vect_stmt_in_region_p (vinfo, *def_stmt))
8653 *dt = vect_external_def;
8654 else
8656 stmt_vec_info stmt_vinfo = vinfo_for_stmt (*def_stmt);
8657 *dt = STMT_VINFO_DEF_TYPE (stmt_vinfo);
8660 if (dump_enabled_p ())
8662 dump_printf_loc (MSG_NOTE, vect_location, "type of def: ");
8663 switch (*dt)
8665 case vect_uninitialized_def:
8666 dump_printf (MSG_NOTE, "uninitialized\n");
8667 break;
8668 case vect_constant_def:
8669 dump_printf (MSG_NOTE, "constant\n");
8670 break;
8671 case vect_external_def:
8672 dump_printf (MSG_NOTE, "external\n");
8673 break;
8674 case vect_internal_def:
8675 dump_printf (MSG_NOTE, "internal\n");
8676 break;
8677 case vect_induction_def:
8678 dump_printf (MSG_NOTE, "induction\n");
8679 break;
8680 case vect_reduction_def:
8681 dump_printf (MSG_NOTE, "reduction\n");
8682 break;
8683 case vect_double_reduction_def:
8684 dump_printf (MSG_NOTE, "double reduction\n");
8685 break;
8686 case vect_nested_cycle:
8687 dump_printf (MSG_NOTE, "nested cycle\n");
8688 break;
8689 case vect_unknown_def_type:
8690 dump_printf (MSG_NOTE, "unknown\n");
8691 break;
8695 if (*dt == vect_unknown_def_type)
8697 if (dump_enabled_p ())
8698 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
8699 "Unsupported pattern.\n");
8700 return false;
8703 switch (gimple_code (*def_stmt))
8705 case GIMPLE_PHI:
8706 case GIMPLE_ASSIGN:
8707 case GIMPLE_CALL:
8708 break;
8709 default:
8710 if (dump_enabled_p ())
8711 dump_printf_loc (MSG_MISSED_OPTIMIZATION, vect_location,
8712 "unsupported defining stmt:\n");
8713 return false;
8716 return true;
8719 /* Function vect_is_simple_use.
8721 Same as vect_is_simple_use but also determines the vector operand
8722 type of OPERAND and stores it to *VECTYPE. If the definition of
8723 OPERAND is vect_uninitialized_def, vect_constant_def or
8724 vect_external_def *VECTYPE will be set to NULL_TREE and the caller
8725 is responsible to compute the best suited vector type for the
8726 scalar operand. */
8728 bool
8729 vect_is_simple_use (tree operand, vec_info *vinfo,
8730 gimple **def_stmt, enum vect_def_type *dt, tree *vectype)
8732 if (!vect_is_simple_use (operand, vinfo, def_stmt, dt))
8733 return false;
8735 /* Now get a vector type if the def is internal, otherwise supply
8736 NULL_TREE and leave it up to the caller to figure out a proper
8737 type for the use stmt. */
8738 if (*dt == vect_internal_def
8739 || *dt == vect_induction_def
8740 || *dt == vect_reduction_def
8741 || *dt == vect_double_reduction_def
8742 || *dt == vect_nested_cycle)
8744 stmt_vec_info stmt_info = vinfo_for_stmt (*def_stmt);
8746 if (STMT_VINFO_IN_PATTERN_P (stmt_info)
8747 && !STMT_VINFO_RELEVANT (stmt_info)
8748 && !STMT_VINFO_LIVE_P (stmt_info))
8749 stmt_info = vinfo_for_stmt (STMT_VINFO_RELATED_STMT (stmt_info));
8751 *vectype = STMT_VINFO_VECTYPE (stmt_info);
8752 gcc_assert (*vectype != NULL_TREE);
8754 else if (*dt == vect_uninitialized_def
8755 || *dt == vect_constant_def
8756 || *dt == vect_external_def)
8757 *vectype = NULL_TREE;
8758 else
8759 gcc_unreachable ();
8761 return true;
8765 /* Function supportable_widening_operation
8767 Check whether an operation represented by the code CODE is a
8768 widening operation that is supported by the target platform in
8769 vector form (i.e., when operating on arguments of type VECTYPE_IN
8770 producing a result of type VECTYPE_OUT).
8772 Widening operations we currently support are NOP (CONVERT), FLOAT
8773 and WIDEN_MULT. This function checks if these operations are supported
8774 by the target platform either directly (via vector tree-codes), or via
8775 target builtins.
8777 Output:
8778 - CODE1 and CODE2 are codes of vector operations to be used when
8779 vectorizing the operation, if available.
8780 - MULTI_STEP_CVT determines the number of required intermediate steps in
8781 case of multi-step conversion (like char->short->int - in that case
8782 MULTI_STEP_CVT will be 1).
8783 - INTERM_TYPES contains the intermediate type required to perform the
8784 widening operation (short in the above example). */
8786 bool
8787 supportable_widening_operation (enum tree_code code, gimple *stmt,
8788 tree vectype_out, tree vectype_in,
8789 enum tree_code *code1, enum tree_code *code2,
8790 int *multi_step_cvt,
8791 vec<tree> *interm_types)
8793 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
8794 loop_vec_info loop_info = STMT_VINFO_LOOP_VINFO (stmt_info);
8795 struct loop *vect_loop = NULL;
8796 machine_mode vec_mode;
8797 enum insn_code icode1, icode2;
8798 optab optab1, optab2;
8799 tree vectype = vectype_in;
8800 tree wide_vectype = vectype_out;
8801 enum tree_code c1, c2;
8802 int i;
8803 tree prev_type, intermediate_type;
8804 machine_mode intermediate_mode, prev_mode;
8805 optab optab3, optab4;
8807 *multi_step_cvt = 0;
8808 if (loop_info)
8809 vect_loop = LOOP_VINFO_LOOP (loop_info);
8811 switch (code)
8813 case WIDEN_MULT_EXPR:
8814 /* The result of a vectorized widening operation usually requires
8815 two vectors (because the widened results do not fit into one vector).
8816 The generated vector results would normally be expected to be
8817 generated in the same order as in the original scalar computation,
8818 i.e. if 8 results are generated in each vector iteration, they are
8819 to be organized as follows:
8820 vect1: [res1,res2,res3,res4],
8821 vect2: [res5,res6,res7,res8].
8823 However, in the special case that the result of the widening
8824 operation is used in a reduction computation only, the order doesn't
8825 matter (because when vectorizing a reduction we change the order of
8826 the computation). Some targets can take advantage of this and
8827 generate more efficient code. For example, targets like Altivec,
8828 that support widen_mult using a sequence of {mult_even,mult_odd}
8829 generate the following vectors:
8830 vect1: [res1,res3,res5,res7],
8831 vect2: [res2,res4,res6,res8].
8833 When vectorizing outer-loops, we execute the inner-loop sequentially
8834 (each vectorized inner-loop iteration contributes to VF outer-loop
8835 iterations in parallel). We therefore don't allow to change the
8836 order of the computation in the inner-loop during outer-loop
8837 vectorization. */
8838 /* TODO: Another case in which order doesn't *really* matter is when we
8839 widen and then contract again, e.g. (short)((int)x * y >> 8).
8840 Normally, pack_trunc performs an even/odd permute, whereas the
8841 repack from an even/odd expansion would be an interleave, which
8842 would be significantly simpler for e.g. AVX2. */
8843 /* In any case, in order to avoid duplicating the code below, recurse
8844 on VEC_WIDEN_MULT_EVEN_EXPR. If it succeeds, all the return values
8845 are properly set up for the caller. If we fail, we'll continue with
8846 a VEC_WIDEN_MULT_LO/HI_EXPR check. */
8847 if (vect_loop
8848 && STMT_VINFO_RELEVANT (stmt_info) == vect_used_by_reduction
8849 && !nested_in_vect_loop_p (vect_loop, stmt)
8850 && supportable_widening_operation (VEC_WIDEN_MULT_EVEN_EXPR,
8851 stmt, vectype_out, vectype_in,
8852 code1, code2, multi_step_cvt,
8853 interm_types))
8855 /* Elements in a vector with vect_used_by_reduction property cannot
8856 be reordered if the use chain with this property does not have the
8857 same operation. One such an example is s += a * b, where elements
8858 in a and b cannot be reordered. Here we check if the vector defined
8859 by STMT is only directly used in the reduction statement. */
8860 tree lhs = gimple_assign_lhs (stmt);
8861 use_operand_p dummy;
8862 gimple *use_stmt;
8863 stmt_vec_info use_stmt_info = NULL;
8864 if (single_imm_use (lhs, &dummy, &use_stmt)
8865 && (use_stmt_info = vinfo_for_stmt (use_stmt))
8866 && STMT_VINFO_DEF_TYPE (use_stmt_info) == vect_reduction_def)
8867 return true;
8869 c1 = VEC_WIDEN_MULT_LO_EXPR;
8870 c2 = VEC_WIDEN_MULT_HI_EXPR;
8871 break;
8873 case DOT_PROD_EXPR:
8874 c1 = DOT_PROD_EXPR;
8875 c2 = DOT_PROD_EXPR;
8876 break;
8878 case SAD_EXPR:
8879 c1 = SAD_EXPR;
8880 c2 = SAD_EXPR;
8881 break;
8883 case VEC_WIDEN_MULT_EVEN_EXPR:
8884 /* Support the recursion induced just above. */
8885 c1 = VEC_WIDEN_MULT_EVEN_EXPR;
8886 c2 = VEC_WIDEN_MULT_ODD_EXPR;
8887 break;
8889 case WIDEN_LSHIFT_EXPR:
8890 c1 = VEC_WIDEN_LSHIFT_LO_EXPR;
8891 c2 = VEC_WIDEN_LSHIFT_HI_EXPR;
8892 break;
8894 CASE_CONVERT:
8895 c1 = VEC_UNPACK_LO_EXPR;
8896 c2 = VEC_UNPACK_HI_EXPR;
8897 break;
8899 case FLOAT_EXPR:
8900 c1 = VEC_UNPACK_FLOAT_LO_EXPR;
8901 c2 = VEC_UNPACK_FLOAT_HI_EXPR;
8902 break;
8904 case FIX_TRUNC_EXPR:
8905 /* ??? Not yet implemented due to missing VEC_UNPACK_FIX_TRUNC_HI_EXPR/
8906 VEC_UNPACK_FIX_TRUNC_LO_EXPR tree codes and optabs used for
8907 computing the operation. */
8908 return false;
8910 default:
8911 gcc_unreachable ();
8914 if (BYTES_BIG_ENDIAN && c1 != VEC_WIDEN_MULT_EVEN_EXPR)
8915 std::swap (c1, c2);
8917 if (code == FIX_TRUNC_EXPR)
8919 /* The signedness is determined from output operand. */
8920 optab1 = optab_for_tree_code (c1, vectype_out, optab_default);
8921 optab2 = optab_for_tree_code (c2, vectype_out, optab_default);
8923 else
8925 optab1 = optab_for_tree_code (c1, vectype, optab_default);
8926 optab2 = optab_for_tree_code (c2, vectype, optab_default);
8929 if (!optab1 || !optab2)
8930 return false;
8932 vec_mode = TYPE_MODE (vectype);
8933 if ((icode1 = optab_handler (optab1, vec_mode)) == CODE_FOR_nothing
8934 || (icode2 = optab_handler (optab2, vec_mode)) == CODE_FOR_nothing)
8935 return false;
8937 *code1 = c1;
8938 *code2 = c2;
8940 if (insn_data[icode1].operand[0].mode == TYPE_MODE (wide_vectype)
8941 && insn_data[icode2].operand[0].mode == TYPE_MODE (wide_vectype))
8942 /* For scalar masks we may have different boolean
8943 vector types having the same QImode. Thus we
8944 add additional check for elements number. */
8945 return (!VECTOR_BOOLEAN_TYPE_P (vectype)
8946 || (TYPE_VECTOR_SUBPARTS (vectype) / 2
8947 == TYPE_VECTOR_SUBPARTS (wide_vectype)));
8949 /* Check if it's a multi-step conversion that can be done using intermediate
8950 types. */
8952 prev_type = vectype;
8953 prev_mode = vec_mode;
8955 if (!CONVERT_EXPR_CODE_P (code))
8956 return false;
8958 /* We assume here that there will not be more than MAX_INTERM_CVT_STEPS
8959 intermediate steps in promotion sequence. We try
8960 MAX_INTERM_CVT_STEPS to get to NARROW_VECTYPE, and fail if we do
8961 not. */
8962 interm_types->create (MAX_INTERM_CVT_STEPS);
8963 for (i = 0; i < MAX_INTERM_CVT_STEPS; i++)
8965 intermediate_mode = insn_data[icode1].operand[0].mode;
8966 if (VECTOR_BOOLEAN_TYPE_P (prev_type))
8968 intermediate_type
8969 = build_truth_vector_type (TYPE_VECTOR_SUBPARTS (prev_type) / 2,
8970 current_vector_size);
8971 if (intermediate_mode != TYPE_MODE (intermediate_type))
8972 return false;
8974 else
8975 intermediate_type
8976 = lang_hooks.types.type_for_mode (intermediate_mode,
8977 TYPE_UNSIGNED (prev_type));
8979 optab3 = optab_for_tree_code (c1, intermediate_type, optab_default);
8980 optab4 = optab_for_tree_code (c2, intermediate_type, optab_default);
8982 if (!optab3 || !optab4
8983 || (icode1 = optab_handler (optab1, prev_mode)) == CODE_FOR_nothing
8984 || insn_data[icode1].operand[0].mode != intermediate_mode
8985 || (icode2 = optab_handler (optab2, prev_mode)) == CODE_FOR_nothing
8986 || insn_data[icode2].operand[0].mode != intermediate_mode
8987 || ((icode1 = optab_handler (optab3, intermediate_mode))
8988 == CODE_FOR_nothing)
8989 || ((icode2 = optab_handler (optab4, intermediate_mode))
8990 == CODE_FOR_nothing))
8991 break;
8993 interm_types->quick_push (intermediate_type);
8994 (*multi_step_cvt)++;
8996 if (insn_data[icode1].operand[0].mode == TYPE_MODE (wide_vectype)
8997 && insn_data[icode2].operand[0].mode == TYPE_MODE (wide_vectype))
8998 return (!VECTOR_BOOLEAN_TYPE_P (vectype)
8999 || (TYPE_VECTOR_SUBPARTS (intermediate_type) / 2
9000 == TYPE_VECTOR_SUBPARTS (wide_vectype)));
9002 prev_type = intermediate_type;
9003 prev_mode = intermediate_mode;
9006 interm_types->release ();
9007 return false;
9011 /* Function supportable_narrowing_operation
9013 Check whether an operation represented by the code CODE is a
9014 narrowing operation that is supported by the target platform in
9015 vector form (i.e., when operating on arguments of type VECTYPE_IN
9016 and producing a result of type VECTYPE_OUT).
9018 Narrowing operations we currently support are NOP (CONVERT) and
9019 FIX_TRUNC. This function checks if these operations are supported by
9020 the target platform directly via vector tree-codes.
9022 Output:
9023 - CODE1 is the code of a vector operation to be used when
9024 vectorizing the operation, if available.
9025 - MULTI_STEP_CVT determines the number of required intermediate steps in
9026 case of multi-step conversion (like int->short->char - in that case
9027 MULTI_STEP_CVT will be 1).
9028 - INTERM_TYPES contains the intermediate type required to perform the
9029 narrowing operation (short in the above example). */
9031 bool
9032 supportable_narrowing_operation (enum tree_code code,
9033 tree vectype_out, tree vectype_in,
9034 enum tree_code *code1, int *multi_step_cvt,
9035 vec<tree> *interm_types)
9037 machine_mode vec_mode;
9038 enum insn_code icode1;
9039 optab optab1, interm_optab;
9040 tree vectype = vectype_in;
9041 tree narrow_vectype = vectype_out;
9042 enum tree_code c1;
9043 tree intermediate_type, prev_type;
9044 machine_mode intermediate_mode, prev_mode;
9045 int i;
9046 bool uns;
9048 *multi_step_cvt = 0;
9049 switch (code)
9051 CASE_CONVERT:
9052 c1 = VEC_PACK_TRUNC_EXPR;
9053 break;
9055 case FIX_TRUNC_EXPR:
9056 c1 = VEC_PACK_FIX_TRUNC_EXPR;
9057 break;
9059 case FLOAT_EXPR:
9060 /* ??? Not yet implemented due to missing VEC_PACK_FLOAT_EXPR
9061 tree code and optabs used for computing the operation. */
9062 return false;
9064 default:
9065 gcc_unreachable ();
9068 if (code == FIX_TRUNC_EXPR)
9069 /* The signedness is determined from output operand. */
9070 optab1 = optab_for_tree_code (c1, vectype_out, optab_default);
9071 else
9072 optab1 = optab_for_tree_code (c1, vectype, optab_default);
9074 if (!optab1)
9075 return false;
9077 vec_mode = TYPE_MODE (vectype);
9078 if ((icode1 = optab_handler (optab1, vec_mode)) == CODE_FOR_nothing)
9079 return false;
9081 *code1 = c1;
9083 if (insn_data[icode1].operand[0].mode == TYPE_MODE (narrow_vectype))
9084 /* For scalar masks we may have different boolean
9085 vector types having the same QImode. Thus we
9086 add additional check for elements number. */
9087 return (!VECTOR_BOOLEAN_TYPE_P (vectype)
9088 || (TYPE_VECTOR_SUBPARTS (vectype) * 2
9089 == TYPE_VECTOR_SUBPARTS (narrow_vectype)));
9091 /* Check if it's a multi-step conversion that can be done using intermediate
9092 types. */
9093 prev_mode = vec_mode;
9094 prev_type = vectype;
9095 if (code == FIX_TRUNC_EXPR)
9096 uns = TYPE_UNSIGNED (vectype_out);
9097 else
9098 uns = TYPE_UNSIGNED (vectype);
9100 /* For multi-step FIX_TRUNC_EXPR prefer signed floating to integer
9101 conversion over unsigned, as unsigned FIX_TRUNC_EXPR is often more
9102 costly than signed. */
9103 if (code == FIX_TRUNC_EXPR && uns)
9105 enum insn_code icode2;
9107 intermediate_type
9108 = lang_hooks.types.type_for_mode (TYPE_MODE (vectype_out), 0);
9109 interm_optab
9110 = optab_for_tree_code (c1, intermediate_type, optab_default);
9111 if (interm_optab != unknown_optab
9112 && (icode2 = optab_handler (optab1, vec_mode)) != CODE_FOR_nothing
9113 && insn_data[icode1].operand[0].mode
9114 == insn_data[icode2].operand[0].mode)
9116 uns = false;
9117 optab1 = interm_optab;
9118 icode1 = icode2;
9122 /* We assume here that there will not be more than MAX_INTERM_CVT_STEPS
9123 intermediate steps in promotion sequence. We try
9124 MAX_INTERM_CVT_STEPS to get to NARROW_VECTYPE, and fail if we do not. */
9125 interm_types->create (MAX_INTERM_CVT_STEPS);
9126 for (i = 0; i < MAX_INTERM_CVT_STEPS; i++)
9128 intermediate_mode = insn_data[icode1].operand[0].mode;
9129 if (VECTOR_BOOLEAN_TYPE_P (prev_type))
9131 intermediate_type
9132 = build_truth_vector_type (TYPE_VECTOR_SUBPARTS (prev_type) * 2,
9133 current_vector_size);
9134 if (intermediate_mode != TYPE_MODE (intermediate_type))
9135 return false;
9137 else
9138 intermediate_type
9139 = lang_hooks.types.type_for_mode (intermediate_mode, uns);
9140 interm_optab
9141 = optab_for_tree_code (VEC_PACK_TRUNC_EXPR, intermediate_type,
9142 optab_default);
9143 if (!interm_optab
9144 || ((icode1 = optab_handler (optab1, prev_mode)) == CODE_FOR_nothing)
9145 || insn_data[icode1].operand[0].mode != intermediate_mode
9146 || ((icode1 = optab_handler (interm_optab, intermediate_mode))
9147 == CODE_FOR_nothing))
9148 break;
9150 interm_types->quick_push (intermediate_type);
9151 (*multi_step_cvt)++;
9153 if (insn_data[icode1].operand[0].mode == TYPE_MODE (narrow_vectype))
9154 return (!VECTOR_BOOLEAN_TYPE_P (vectype)
9155 || (TYPE_VECTOR_SUBPARTS (intermediate_type) * 2
9156 == TYPE_VECTOR_SUBPARTS (narrow_vectype)));
9158 prev_mode = intermediate_mode;
9159 prev_type = intermediate_type;
9160 optab1 = interm_optab;
9163 interm_types->release ();
9164 return false;