2011-02-06 Paul Thomas <pault@gcc.gnu.org>
[official-gcc.git] / gcc / tree-vect-stmts.c
blobe5bfcbe81acb2b9350fa5524bcbeb59a2ddad6dd
1 /* Statement Analysis and Transformation for Vectorization
2 Copyright (C) 2003, 2004, 2005, 2006, 2007, 2008, 2009, 2010
3 Free Software Foundation, Inc.
4 Contributed by Dorit Naishlos <dorit@il.ibm.com>
5 and Ira Rosen <irar@il.ibm.com>
7 This file is part of GCC.
9 GCC is free software; you can redistribute it and/or modify it under
10 the terms of the GNU General Public License as published by the Free
11 Software Foundation; either version 3, or (at your option) any later
12 version.
14 GCC is distributed in the hope that it will be useful, but WITHOUT ANY
15 WARRANTY; without even the implied warranty of MERCHANTABILITY or
16 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
17 for more details.
19 You should have received a copy of the GNU General Public License
20 along with GCC; see the file COPYING3. If not see
21 <http://www.gnu.org/licenses/>. */
23 #include "config.h"
24 #include "system.h"
25 #include "coretypes.h"
26 #include "tm.h"
27 #include "ggc.h"
28 #include "tree.h"
29 #include "target.h"
30 #include "basic-block.h"
31 #include "tree-pretty-print.h"
32 #include "gimple-pretty-print.h"
33 #include "tree-flow.h"
34 #include "tree-dump.h"
35 #include "cfgloop.h"
36 #include "cfglayout.h"
37 #include "expr.h"
38 #include "recog.h"
39 #include "optabs.h"
40 #include "diagnostic-core.h"
41 #include "tree-vectorizer.h"
42 #include "langhooks.h"
45 /* Utility functions used by vect_mark_stmts_to_be_vectorized. */
47 /* Function vect_mark_relevant.
49 Mark STMT as "relevant for vectorization" and add it to WORKLIST. */
51 static void
52 vect_mark_relevant (VEC(gimple,heap) **worklist, gimple stmt,
53 enum vect_relevant relevant, bool live_p)
55 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
56 enum vect_relevant save_relevant = STMT_VINFO_RELEVANT (stmt_info);
57 bool save_live_p = STMT_VINFO_LIVE_P (stmt_info);
59 if (vect_print_dump_info (REPORT_DETAILS))
60 fprintf (vect_dump, "mark relevant %d, live %d.", relevant, live_p);
62 if (STMT_VINFO_IN_PATTERN_P (stmt_info))
64 gimple pattern_stmt;
66 /* This is the last stmt in a sequence that was detected as a
67 pattern that can potentially be vectorized. Don't mark the stmt
68 as relevant/live because it's not going to be vectorized.
69 Instead mark the pattern-stmt that replaces it. */
71 pattern_stmt = STMT_VINFO_RELATED_STMT (stmt_info);
73 if (vect_print_dump_info (REPORT_DETAILS))
74 fprintf (vect_dump, "last stmt in pattern. don't mark relevant/live.");
75 stmt_info = vinfo_for_stmt (pattern_stmt);
76 gcc_assert (STMT_VINFO_RELATED_STMT (stmt_info) == stmt);
77 save_relevant = STMT_VINFO_RELEVANT (stmt_info);
78 save_live_p = STMT_VINFO_LIVE_P (stmt_info);
79 stmt = pattern_stmt;
82 STMT_VINFO_LIVE_P (stmt_info) |= live_p;
83 if (relevant > STMT_VINFO_RELEVANT (stmt_info))
84 STMT_VINFO_RELEVANT (stmt_info) = relevant;
86 if (STMT_VINFO_RELEVANT (stmt_info) == save_relevant
87 && STMT_VINFO_LIVE_P (stmt_info) == save_live_p)
89 if (vect_print_dump_info (REPORT_DETAILS))
90 fprintf (vect_dump, "already marked relevant/live.");
91 return;
94 VEC_safe_push (gimple, heap, *worklist, stmt);
98 /* Function vect_stmt_relevant_p.
100 Return true if STMT in loop that is represented by LOOP_VINFO is
101 "relevant for vectorization".
103 A stmt is considered "relevant for vectorization" if:
104 - it has uses outside the loop.
105 - it has vdefs (it alters memory).
106 - control stmts in the loop (except for the exit condition).
108 CHECKME: what other side effects would the vectorizer allow? */
110 static bool
111 vect_stmt_relevant_p (gimple stmt, loop_vec_info loop_vinfo,
112 enum vect_relevant *relevant, bool *live_p)
114 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
115 ssa_op_iter op_iter;
116 imm_use_iterator imm_iter;
117 use_operand_p use_p;
118 def_operand_p def_p;
120 *relevant = vect_unused_in_scope;
121 *live_p = false;
123 /* cond stmt other than loop exit cond. */
124 if (is_ctrl_stmt (stmt)
125 && STMT_VINFO_TYPE (vinfo_for_stmt (stmt))
126 != loop_exit_ctrl_vec_info_type)
127 *relevant = vect_used_in_scope;
129 /* changing memory. */
130 if (gimple_code (stmt) != GIMPLE_PHI)
131 if (gimple_vdef (stmt))
133 if (vect_print_dump_info (REPORT_DETAILS))
134 fprintf (vect_dump, "vec_stmt_relevant_p: stmt has vdefs.");
135 *relevant = vect_used_in_scope;
138 /* uses outside the loop. */
139 FOR_EACH_PHI_OR_STMT_DEF (def_p, stmt, op_iter, SSA_OP_DEF)
141 FOR_EACH_IMM_USE_FAST (use_p, imm_iter, DEF_FROM_PTR (def_p))
143 basic_block bb = gimple_bb (USE_STMT (use_p));
144 if (!flow_bb_inside_loop_p (loop, bb))
146 if (vect_print_dump_info (REPORT_DETAILS))
147 fprintf (vect_dump, "vec_stmt_relevant_p: used out of loop.");
149 if (is_gimple_debug (USE_STMT (use_p)))
150 continue;
152 /* We expect all such uses to be in the loop exit phis
153 (because of loop closed form) */
154 gcc_assert (gimple_code (USE_STMT (use_p)) == GIMPLE_PHI);
155 gcc_assert (bb == single_exit (loop)->dest);
157 *live_p = true;
162 return (*live_p || *relevant);
166 /* Function exist_non_indexing_operands_for_use_p
168 USE is one of the uses attached to STMT. Check if USE is
169 used in STMT for anything other than indexing an array. */
171 static bool
172 exist_non_indexing_operands_for_use_p (tree use, gimple stmt)
174 tree operand;
175 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
177 /* USE corresponds to some operand in STMT. If there is no data
178 reference in STMT, then any operand that corresponds to USE
179 is not indexing an array. */
180 if (!STMT_VINFO_DATA_REF (stmt_info))
181 return true;
183 /* STMT has a data_ref. FORNOW this means that its of one of
184 the following forms:
185 -1- ARRAY_REF = var
186 -2- var = ARRAY_REF
187 (This should have been verified in analyze_data_refs).
189 'var' in the second case corresponds to a def, not a use,
190 so USE cannot correspond to any operands that are not used
191 for array indexing.
193 Therefore, all we need to check is if STMT falls into the
194 first case, and whether var corresponds to USE. */
196 if (!gimple_assign_copy_p (stmt))
197 return false;
198 if (TREE_CODE (gimple_assign_lhs (stmt)) == SSA_NAME)
199 return false;
200 operand = gimple_assign_rhs1 (stmt);
201 if (TREE_CODE (operand) != SSA_NAME)
202 return false;
204 if (operand == use)
205 return true;
207 return false;
212 Function process_use.
214 Inputs:
215 - a USE in STMT in a loop represented by LOOP_VINFO
216 - LIVE_P, RELEVANT - enum values to be set in the STMT_VINFO of the stmt
217 that defined USE. This is done by calling mark_relevant and passing it
218 the WORKLIST (to add DEF_STMT to the WORKLIST in case it is relevant).
220 Outputs:
221 Generally, LIVE_P and RELEVANT are used to define the liveness and
222 relevance info of the DEF_STMT of this USE:
223 STMT_VINFO_LIVE_P (DEF_STMT_info) <-- live_p
224 STMT_VINFO_RELEVANT (DEF_STMT_info) <-- relevant
225 Exceptions:
226 - case 1: If USE is used only for address computations (e.g. array indexing),
227 which does not need to be directly vectorized, then the liveness/relevance
228 of the respective DEF_STMT is left unchanged.
229 - case 2: If STMT is a reduction phi and DEF_STMT is a reduction stmt, we
230 skip DEF_STMT cause it had already been processed.
231 - case 3: If DEF_STMT and STMT are in different nests, then "relevant" will
232 be modified accordingly.
234 Return true if everything is as expected. Return false otherwise. */
236 static bool
237 process_use (gimple stmt, tree use, loop_vec_info loop_vinfo, bool live_p,
238 enum vect_relevant relevant, VEC(gimple,heap) **worklist)
240 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
241 stmt_vec_info stmt_vinfo = vinfo_for_stmt (stmt);
242 stmt_vec_info dstmt_vinfo;
243 basic_block bb, def_bb;
244 tree def;
245 gimple def_stmt;
246 enum vect_def_type dt;
248 /* case 1: we are only interested in uses that need to be vectorized. Uses
249 that are used for address computation are not considered relevant. */
250 if (!exist_non_indexing_operands_for_use_p (use, stmt))
251 return true;
253 if (!vect_is_simple_use (use, loop_vinfo, NULL, &def_stmt, &def, &dt))
255 if (vect_print_dump_info (REPORT_UNVECTORIZED_LOCATIONS))
256 fprintf (vect_dump, "not vectorized: unsupported use in stmt.");
257 return false;
260 if (!def_stmt || gimple_nop_p (def_stmt))
261 return true;
263 def_bb = gimple_bb (def_stmt);
264 if (!flow_bb_inside_loop_p (loop, def_bb))
266 if (vect_print_dump_info (REPORT_DETAILS))
267 fprintf (vect_dump, "def_stmt is out of loop.");
268 return true;
271 /* case 2: A reduction phi (STMT) defined by a reduction stmt (DEF_STMT).
272 DEF_STMT must have already been processed, because this should be the
273 only way that STMT, which is a reduction-phi, was put in the worklist,
274 as there should be no other uses for DEF_STMT in the loop. So we just
275 check that everything is as expected, and we are done. */
276 dstmt_vinfo = vinfo_for_stmt (def_stmt);
277 bb = gimple_bb (stmt);
278 if (gimple_code (stmt) == GIMPLE_PHI
279 && STMT_VINFO_DEF_TYPE (stmt_vinfo) == vect_reduction_def
280 && gimple_code (def_stmt) != GIMPLE_PHI
281 && STMT_VINFO_DEF_TYPE (dstmt_vinfo) == vect_reduction_def
282 && bb->loop_father == def_bb->loop_father)
284 if (vect_print_dump_info (REPORT_DETAILS))
285 fprintf (vect_dump, "reduc-stmt defining reduc-phi in the same nest.");
286 if (STMT_VINFO_IN_PATTERN_P (dstmt_vinfo))
287 dstmt_vinfo = vinfo_for_stmt (STMT_VINFO_RELATED_STMT (dstmt_vinfo));
288 gcc_assert (STMT_VINFO_RELEVANT (dstmt_vinfo) < vect_used_by_reduction);
289 gcc_assert (STMT_VINFO_LIVE_P (dstmt_vinfo)
290 || STMT_VINFO_RELEVANT (dstmt_vinfo) > vect_unused_in_scope);
291 return true;
294 /* case 3a: outer-loop stmt defining an inner-loop stmt:
295 outer-loop-header-bb:
296 d = def_stmt
297 inner-loop:
298 stmt # use (d)
299 outer-loop-tail-bb:
300 ... */
301 if (flow_loop_nested_p (def_bb->loop_father, bb->loop_father))
303 if (vect_print_dump_info (REPORT_DETAILS))
304 fprintf (vect_dump, "outer-loop def-stmt defining inner-loop stmt.");
306 switch (relevant)
308 case vect_unused_in_scope:
309 relevant = (STMT_VINFO_DEF_TYPE (stmt_vinfo) == vect_nested_cycle) ?
310 vect_used_in_scope : vect_unused_in_scope;
311 break;
313 case vect_used_in_outer_by_reduction:
314 gcc_assert (STMT_VINFO_DEF_TYPE (stmt_vinfo) != vect_reduction_def);
315 relevant = vect_used_by_reduction;
316 break;
318 case vect_used_in_outer:
319 gcc_assert (STMT_VINFO_DEF_TYPE (stmt_vinfo) != vect_reduction_def);
320 relevant = vect_used_in_scope;
321 break;
323 case vect_used_in_scope:
324 break;
326 default:
327 gcc_unreachable ();
331 /* case 3b: inner-loop stmt defining an outer-loop stmt:
332 outer-loop-header-bb:
334 inner-loop:
335 d = def_stmt
336 outer-loop-tail-bb (or outer-loop-exit-bb in double reduction):
337 stmt # use (d) */
338 else if (flow_loop_nested_p (bb->loop_father, def_bb->loop_father))
340 if (vect_print_dump_info (REPORT_DETAILS))
341 fprintf (vect_dump, "inner-loop def-stmt defining outer-loop stmt.");
343 switch (relevant)
345 case vect_unused_in_scope:
346 relevant = (STMT_VINFO_DEF_TYPE (stmt_vinfo) == vect_reduction_def
347 || STMT_VINFO_DEF_TYPE (stmt_vinfo) == vect_double_reduction_def) ?
348 vect_used_in_outer_by_reduction : vect_unused_in_scope;
349 break;
351 case vect_used_by_reduction:
352 relevant = vect_used_in_outer_by_reduction;
353 break;
355 case vect_used_in_scope:
356 relevant = vect_used_in_outer;
357 break;
359 default:
360 gcc_unreachable ();
364 vect_mark_relevant (worklist, def_stmt, relevant, live_p);
365 return true;
369 /* Function vect_mark_stmts_to_be_vectorized.
371 Not all stmts in the loop need to be vectorized. For example:
373 for i...
374 for j...
375 1. T0 = i + j
376 2. T1 = a[T0]
378 3. j = j + 1
380 Stmt 1 and 3 do not need to be vectorized, because loop control and
381 addressing of vectorized data-refs are handled differently.
383 This pass detects such stmts. */
385 bool
386 vect_mark_stmts_to_be_vectorized (loop_vec_info loop_vinfo)
388 VEC(gimple,heap) *worklist;
389 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
390 basic_block *bbs = LOOP_VINFO_BBS (loop_vinfo);
391 unsigned int nbbs = loop->num_nodes;
392 gimple_stmt_iterator si;
393 gimple stmt;
394 unsigned int i;
395 stmt_vec_info stmt_vinfo;
396 basic_block bb;
397 gimple phi;
398 bool live_p;
399 enum vect_relevant relevant, tmp_relevant;
400 enum vect_def_type def_type;
402 if (vect_print_dump_info (REPORT_DETAILS))
403 fprintf (vect_dump, "=== vect_mark_stmts_to_be_vectorized ===");
405 worklist = VEC_alloc (gimple, heap, 64);
407 /* 1. Init worklist. */
408 for (i = 0; i < nbbs; i++)
410 bb = bbs[i];
411 for (si = gsi_start_phis (bb); !gsi_end_p (si); gsi_next (&si))
413 phi = gsi_stmt (si);
414 if (vect_print_dump_info (REPORT_DETAILS))
416 fprintf (vect_dump, "init: phi relevant? ");
417 print_gimple_stmt (vect_dump, phi, 0, TDF_SLIM);
420 if (vect_stmt_relevant_p (phi, loop_vinfo, &relevant, &live_p))
421 vect_mark_relevant (&worklist, phi, relevant, live_p);
423 for (si = gsi_start_bb (bb); !gsi_end_p (si); gsi_next (&si))
425 stmt = gsi_stmt (si);
426 if (vect_print_dump_info (REPORT_DETAILS))
428 fprintf (vect_dump, "init: stmt relevant? ");
429 print_gimple_stmt (vect_dump, stmt, 0, TDF_SLIM);
432 if (vect_stmt_relevant_p (stmt, loop_vinfo, &relevant, &live_p))
433 vect_mark_relevant (&worklist, stmt, relevant, live_p);
437 /* 2. Process_worklist */
438 while (VEC_length (gimple, worklist) > 0)
440 use_operand_p use_p;
441 ssa_op_iter iter;
443 stmt = VEC_pop (gimple, worklist);
444 if (vect_print_dump_info (REPORT_DETAILS))
446 fprintf (vect_dump, "worklist: examine stmt: ");
447 print_gimple_stmt (vect_dump, stmt, 0, TDF_SLIM);
450 /* Examine the USEs of STMT. For each USE, mark the stmt that defines it
451 (DEF_STMT) as relevant/irrelevant and live/dead according to the
452 liveness and relevance properties of STMT. */
453 stmt_vinfo = vinfo_for_stmt (stmt);
454 relevant = STMT_VINFO_RELEVANT (stmt_vinfo);
455 live_p = STMT_VINFO_LIVE_P (stmt_vinfo);
457 /* Generally, the liveness and relevance properties of STMT are
458 propagated as is to the DEF_STMTs of its USEs:
459 live_p <-- STMT_VINFO_LIVE_P (STMT_VINFO)
460 relevant <-- STMT_VINFO_RELEVANT (STMT_VINFO)
462 One exception is when STMT has been identified as defining a reduction
463 variable; in this case we set the liveness/relevance as follows:
464 live_p = false
465 relevant = vect_used_by_reduction
466 This is because we distinguish between two kinds of relevant stmts -
467 those that are used by a reduction computation, and those that are
468 (also) used by a regular computation. This allows us later on to
469 identify stmts that are used solely by a reduction, and therefore the
470 order of the results that they produce does not have to be kept. */
472 def_type = STMT_VINFO_DEF_TYPE (stmt_vinfo);
473 tmp_relevant = relevant;
474 switch (def_type)
476 case vect_reduction_def:
477 switch (tmp_relevant)
479 case vect_unused_in_scope:
480 relevant = vect_used_by_reduction;
481 break;
483 case vect_used_by_reduction:
484 if (gimple_code (stmt) == GIMPLE_PHI)
485 break;
486 /* fall through */
488 default:
489 if (vect_print_dump_info (REPORT_DETAILS))
490 fprintf (vect_dump, "unsupported use of reduction.");
492 VEC_free (gimple, heap, worklist);
493 return false;
496 live_p = false;
497 break;
499 case vect_nested_cycle:
500 if (tmp_relevant != vect_unused_in_scope
501 && tmp_relevant != vect_used_in_outer_by_reduction
502 && tmp_relevant != vect_used_in_outer)
504 if (vect_print_dump_info (REPORT_DETAILS))
505 fprintf (vect_dump, "unsupported use of nested cycle.");
507 VEC_free (gimple, heap, worklist);
508 return false;
511 live_p = false;
512 break;
514 case vect_double_reduction_def:
515 if (tmp_relevant != vect_unused_in_scope
516 && tmp_relevant != vect_used_by_reduction)
518 if (vect_print_dump_info (REPORT_DETAILS))
519 fprintf (vect_dump, "unsupported use of double reduction.");
521 VEC_free (gimple, heap, worklist);
522 return false;
525 live_p = false;
526 break;
528 default:
529 break;
532 FOR_EACH_PHI_OR_STMT_USE (use_p, stmt, iter, SSA_OP_USE)
534 tree op = USE_FROM_PTR (use_p);
535 if (!process_use (stmt, op, loop_vinfo, live_p, relevant, &worklist))
537 VEC_free (gimple, heap, worklist);
538 return false;
541 } /* while worklist */
543 VEC_free (gimple, heap, worklist);
544 return true;
548 /* Get cost by calling cost target builtin. */
550 static inline
551 int vect_get_stmt_cost (enum vect_cost_for_stmt type_of_cost)
553 tree dummy_type = NULL;
554 int dummy = 0;
556 return targetm.vectorize.builtin_vectorization_cost (type_of_cost,
557 dummy_type, dummy);
561 /* Get cost for STMT. */
564 cost_for_stmt (gimple stmt)
566 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
568 switch (STMT_VINFO_TYPE (stmt_info))
570 case load_vec_info_type:
571 return vect_get_stmt_cost (scalar_load);
572 case store_vec_info_type:
573 return vect_get_stmt_cost (scalar_store);
574 case op_vec_info_type:
575 case condition_vec_info_type:
576 case assignment_vec_info_type:
577 case reduc_vec_info_type:
578 case induc_vec_info_type:
579 case type_promotion_vec_info_type:
580 case type_demotion_vec_info_type:
581 case type_conversion_vec_info_type:
582 case call_vec_info_type:
583 return vect_get_stmt_cost (scalar_stmt);
584 case undef_vec_info_type:
585 default:
586 gcc_unreachable ();
590 /* Function vect_model_simple_cost.
592 Models cost for simple operations, i.e. those that only emit ncopies of a
593 single op. Right now, this does not account for multiple insns that could
594 be generated for the single vector op. We will handle that shortly. */
596 void
597 vect_model_simple_cost (stmt_vec_info stmt_info, int ncopies,
598 enum vect_def_type *dt, slp_tree slp_node)
600 int i;
601 int inside_cost = 0, outside_cost = 0;
603 /* The SLP costs were already calculated during SLP tree build. */
604 if (PURE_SLP_STMT (stmt_info))
605 return;
607 inside_cost = ncopies * vect_get_stmt_cost (vector_stmt);
609 /* FORNOW: Assuming maximum 2 args per stmts. */
610 for (i = 0; i < 2; i++)
612 if (dt[i] == vect_constant_def || dt[i] == vect_external_def)
613 outside_cost += vect_get_stmt_cost (vector_stmt);
616 if (vect_print_dump_info (REPORT_COST))
617 fprintf (vect_dump, "vect_model_simple_cost: inside_cost = %d, "
618 "outside_cost = %d .", inside_cost, outside_cost);
620 /* Set the costs either in STMT_INFO or SLP_NODE (if exists). */
621 stmt_vinfo_set_inside_of_loop_cost (stmt_info, slp_node, inside_cost);
622 stmt_vinfo_set_outside_of_loop_cost (stmt_info, slp_node, outside_cost);
626 /* Function vect_cost_strided_group_size
628 For strided load or store, return the group_size only if it is the first
629 load or store of a group, else return 1. This ensures that group size is
630 only returned once per group. */
632 static int
633 vect_cost_strided_group_size (stmt_vec_info stmt_info)
635 gimple first_stmt = DR_GROUP_FIRST_DR (stmt_info);
637 if (first_stmt == STMT_VINFO_STMT (stmt_info))
638 return DR_GROUP_SIZE (stmt_info);
640 return 1;
644 /* Function vect_model_store_cost
646 Models cost for stores. In the case of strided accesses, one access
647 has the overhead of the strided access attributed to it. */
649 void
650 vect_model_store_cost (stmt_vec_info stmt_info, int ncopies,
651 enum vect_def_type dt, slp_tree slp_node)
653 int group_size;
654 unsigned int inside_cost = 0, outside_cost = 0;
655 struct data_reference *first_dr;
656 gimple first_stmt;
658 /* The SLP costs were already calculated during SLP tree build. */
659 if (PURE_SLP_STMT (stmt_info))
660 return;
662 if (dt == vect_constant_def || dt == vect_external_def)
663 outside_cost = vect_get_stmt_cost (scalar_to_vec);
665 /* Strided access? */
666 if (DR_GROUP_FIRST_DR (stmt_info))
668 if (slp_node)
670 first_stmt = VEC_index (gimple, SLP_TREE_SCALAR_STMTS (slp_node), 0);
671 group_size = 1;
673 else
675 first_stmt = DR_GROUP_FIRST_DR (stmt_info);
676 group_size = vect_cost_strided_group_size (stmt_info);
679 first_dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (first_stmt));
681 /* Not a strided access. */
682 else
684 group_size = 1;
685 first_dr = STMT_VINFO_DATA_REF (stmt_info);
688 /* Is this an access in a group of stores, which provide strided access?
689 If so, add in the cost of the permutes. */
690 if (group_size > 1)
692 /* Uses a high and low interleave operation for each needed permute. */
693 inside_cost = ncopies * exact_log2(group_size) * group_size
694 * vect_get_stmt_cost (vector_stmt);
696 if (vect_print_dump_info (REPORT_COST))
697 fprintf (vect_dump, "vect_model_store_cost: strided group_size = %d .",
698 group_size);
702 /* Costs of the stores. */
703 vect_get_store_cost (first_dr, ncopies, &inside_cost);
705 if (vect_print_dump_info (REPORT_COST))
706 fprintf (vect_dump, "vect_model_store_cost: inside_cost = %d, "
707 "outside_cost = %d .", inside_cost, outside_cost);
709 /* Set the costs either in STMT_INFO or SLP_NODE (if exists). */
710 stmt_vinfo_set_inside_of_loop_cost (stmt_info, slp_node, inside_cost);
711 stmt_vinfo_set_outside_of_loop_cost (stmt_info, slp_node, outside_cost);
715 /* Calculate cost of DR's memory access. */
716 void
717 vect_get_store_cost (struct data_reference *dr, int ncopies,
718 unsigned int *inside_cost)
720 int alignment_support_scheme = vect_supportable_dr_alignment (dr, false);
722 switch (alignment_support_scheme)
724 case dr_aligned:
726 *inside_cost += ncopies * vect_get_stmt_cost (vector_store);
728 if (vect_print_dump_info (REPORT_COST))
729 fprintf (vect_dump, "vect_model_store_cost: aligned.");
731 break;
734 case dr_unaligned_supported:
736 gimple stmt = DR_STMT (dr);
737 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
738 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
740 /* Here, we assign an additional cost for the unaligned store. */
741 *inside_cost += ncopies
742 * targetm.vectorize.builtin_vectorization_cost (unaligned_store,
743 vectype, DR_MISALIGNMENT (dr));
745 if (vect_print_dump_info (REPORT_COST))
746 fprintf (vect_dump, "vect_model_store_cost: unaligned supported by "
747 "hardware.");
749 break;
752 default:
753 gcc_unreachable ();
758 /* Function vect_model_load_cost
760 Models cost for loads. In the case of strided accesses, the last access
761 has the overhead of the strided access attributed to it. Since unaligned
762 accesses are supported for loads, we also account for the costs of the
763 access scheme chosen. */
765 void
766 vect_model_load_cost (stmt_vec_info stmt_info, int ncopies, slp_tree slp_node)
769 int group_size;
770 gimple first_stmt;
771 struct data_reference *dr = STMT_VINFO_DATA_REF (stmt_info), *first_dr;
772 unsigned int inside_cost = 0, outside_cost = 0;
774 /* The SLP costs were already calculated during SLP tree build. */
775 if (PURE_SLP_STMT (stmt_info))
776 return;
778 /* Strided accesses? */
779 first_stmt = DR_GROUP_FIRST_DR (stmt_info);
780 if (first_stmt && !slp_node)
782 group_size = vect_cost_strided_group_size (stmt_info);
783 first_dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (first_stmt));
785 /* Not a strided access. */
786 else
788 group_size = 1;
789 first_dr = dr;
792 /* Is this an access in a group of loads providing strided access?
793 If so, add in the cost of the permutes. */
794 if (group_size > 1)
796 /* Uses an even and odd extract operations for each needed permute. */
797 inside_cost = ncopies * exact_log2(group_size) * group_size
798 * vect_get_stmt_cost (vector_stmt);
800 if (vect_print_dump_info (REPORT_COST))
801 fprintf (vect_dump, "vect_model_load_cost: strided group_size = %d .",
802 group_size);
805 /* The loads themselves. */
806 vect_get_load_cost (first_dr, ncopies,
807 ((!DR_GROUP_FIRST_DR (stmt_info)) || group_size > 1 || slp_node),
808 &inside_cost, &outside_cost);
810 if (vect_print_dump_info (REPORT_COST))
811 fprintf (vect_dump, "vect_model_load_cost: inside_cost = %d, "
812 "outside_cost = %d .", inside_cost, outside_cost);
814 /* Set the costs either in STMT_INFO or SLP_NODE (if exists). */
815 stmt_vinfo_set_inside_of_loop_cost (stmt_info, slp_node, inside_cost);
816 stmt_vinfo_set_outside_of_loop_cost (stmt_info, slp_node, outside_cost);
820 /* Calculate cost of DR's memory access. */
821 void
822 vect_get_load_cost (struct data_reference *dr, int ncopies,
823 bool add_realign_cost, unsigned int *inside_cost,
824 unsigned int *outside_cost)
826 int alignment_support_scheme = vect_supportable_dr_alignment (dr, false);
828 switch (alignment_support_scheme)
830 case dr_aligned:
832 *inside_cost += ncopies * vect_get_stmt_cost (vector_load);
834 if (vect_print_dump_info (REPORT_COST))
835 fprintf (vect_dump, "vect_model_load_cost: aligned.");
837 break;
839 case dr_unaligned_supported:
841 gimple stmt = DR_STMT (dr);
842 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
843 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
845 /* Here, we assign an additional cost for the unaligned load. */
846 *inside_cost += ncopies
847 * targetm.vectorize.builtin_vectorization_cost (unaligned_load,
848 vectype, DR_MISALIGNMENT (dr));
849 if (vect_print_dump_info (REPORT_COST))
850 fprintf (vect_dump, "vect_model_load_cost: unaligned supported by "
851 "hardware.");
853 break;
855 case dr_explicit_realign:
857 *inside_cost += ncopies * (2 * vect_get_stmt_cost (vector_load)
858 + vect_get_stmt_cost (vector_stmt));
860 /* FIXME: If the misalignment remains fixed across the iterations of
861 the containing loop, the following cost should be added to the
862 outside costs. */
863 if (targetm.vectorize.builtin_mask_for_load)
864 *inside_cost += vect_get_stmt_cost (vector_stmt);
866 break;
868 case dr_explicit_realign_optimized:
870 if (vect_print_dump_info (REPORT_COST))
871 fprintf (vect_dump, "vect_model_load_cost: unaligned software "
872 "pipelined.");
874 /* Unaligned software pipeline has a load of an address, an initial
875 load, and possibly a mask operation to "prime" the loop. However,
876 if this is an access in a group of loads, which provide strided
877 access, then the above cost should only be considered for one
878 access in the group. Inside the loop, there is a load op
879 and a realignment op. */
881 if (add_realign_cost)
883 *outside_cost = 2 * vect_get_stmt_cost (vector_stmt);
884 if (targetm.vectorize.builtin_mask_for_load)
885 *outside_cost += vect_get_stmt_cost (vector_stmt);
888 *inside_cost += ncopies * (vect_get_stmt_cost (vector_load)
889 + vect_get_stmt_cost (vector_stmt));
890 break;
893 default:
894 gcc_unreachable ();
899 /* Function vect_init_vector.
901 Insert a new stmt (INIT_STMT) that initializes a new vector variable with
902 the vector elements of VECTOR_VAR. Place the initialization at BSI if it
903 is not NULL. Otherwise, place the initialization at the loop preheader.
904 Return the DEF of INIT_STMT.
905 It will be used in the vectorization of STMT. */
907 tree
908 vect_init_vector (gimple stmt, tree vector_var, tree vector_type,
909 gimple_stmt_iterator *gsi)
911 stmt_vec_info stmt_vinfo = vinfo_for_stmt (stmt);
912 tree new_var;
913 gimple init_stmt;
914 tree vec_oprnd;
915 edge pe;
916 tree new_temp;
917 basic_block new_bb;
919 new_var = vect_get_new_vect_var (vector_type, vect_simple_var, "cst_");
920 add_referenced_var (new_var);
921 init_stmt = gimple_build_assign (new_var, vector_var);
922 new_temp = make_ssa_name (new_var, init_stmt);
923 gimple_assign_set_lhs (init_stmt, new_temp);
925 if (gsi)
926 vect_finish_stmt_generation (stmt, init_stmt, gsi);
927 else
929 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_vinfo);
931 if (loop_vinfo)
933 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
935 if (nested_in_vect_loop_p (loop, stmt))
936 loop = loop->inner;
938 pe = loop_preheader_edge (loop);
939 new_bb = gsi_insert_on_edge_immediate (pe, init_stmt);
940 gcc_assert (!new_bb);
942 else
944 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_vinfo);
945 basic_block bb;
946 gimple_stmt_iterator gsi_bb_start;
948 gcc_assert (bb_vinfo);
949 bb = BB_VINFO_BB (bb_vinfo);
950 gsi_bb_start = gsi_after_labels (bb);
951 gsi_insert_before (&gsi_bb_start, init_stmt, GSI_SAME_STMT);
955 if (vect_print_dump_info (REPORT_DETAILS))
957 fprintf (vect_dump, "created new init_stmt: ");
958 print_gimple_stmt (vect_dump, init_stmt, 0, TDF_SLIM);
961 vec_oprnd = gimple_assign_lhs (init_stmt);
962 return vec_oprnd;
966 /* Function vect_get_vec_def_for_operand.
968 OP is an operand in STMT. This function returns a (vector) def that will be
969 used in the vectorized stmt for STMT.
971 In the case that OP is an SSA_NAME which is defined in the loop, then
972 STMT_VINFO_VEC_STMT of the defining stmt holds the relevant def.
974 In case OP is an invariant or constant, a new stmt that creates a vector def
975 needs to be introduced. */
977 tree
978 vect_get_vec_def_for_operand (tree op, gimple stmt, tree *scalar_def)
980 tree vec_oprnd;
981 gimple vec_stmt;
982 gimple def_stmt;
983 stmt_vec_info def_stmt_info = NULL;
984 stmt_vec_info stmt_vinfo = vinfo_for_stmt (stmt);
985 unsigned int nunits;
986 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_vinfo);
987 tree vec_inv;
988 tree vec_cst;
989 tree t = NULL_TREE;
990 tree def;
991 int i;
992 enum vect_def_type dt;
993 bool is_simple_use;
994 tree vector_type;
996 if (vect_print_dump_info (REPORT_DETAILS))
998 fprintf (vect_dump, "vect_get_vec_def_for_operand: ");
999 print_generic_expr (vect_dump, op, TDF_SLIM);
1002 is_simple_use = vect_is_simple_use (op, loop_vinfo, NULL, &def_stmt, &def,
1003 &dt);
1004 gcc_assert (is_simple_use);
1005 if (vect_print_dump_info (REPORT_DETAILS))
1007 if (def)
1009 fprintf (vect_dump, "def = ");
1010 print_generic_expr (vect_dump, def, TDF_SLIM);
1012 if (def_stmt)
1014 fprintf (vect_dump, " def_stmt = ");
1015 print_gimple_stmt (vect_dump, def_stmt, 0, TDF_SLIM);
1019 switch (dt)
1021 /* Case 1: operand is a constant. */
1022 case vect_constant_def:
1024 vector_type = get_vectype_for_scalar_type (TREE_TYPE (op));
1025 gcc_assert (vector_type);
1026 nunits = TYPE_VECTOR_SUBPARTS (vector_type);
1028 if (scalar_def)
1029 *scalar_def = op;
1031 /* Create 'vect_cst_ = {cst,cst,...,cst}' */
1032 if (vect_print_dump_info (REPORT_DETAILS))
1033 fprintf (vect_dump, "Create vector_cst. nunits = %d", nunits);
1035 vec_cst = build_vector_from_val (vector_type, op);
1036 return vect_init_vector (stmt, vec_cst, vector_type, NULL);
1039 /* Case 2: operand is defined outside the loop - loop invariant. */
1040 case vect_external_def:
1042 vector_type = get_vectype_for_scalar_type (TREE_TYPE (def));
1043 gcc_assert (vector_type);
1044 nunits = TYPE_VECTOR_SUBPARTS (vector_type);
1046 if (scalar_def)
1047 *scalar_def = def;
1049 /* Create 'vec_inv = {inv,inv,..,inv}' */
1050 if (vect_print_dump_info (REPORT_DETAILS))
1051 fprintf (vect_dump, "Create vector_inv.");
1053 for (i = nunits - 1; i >= 0; --i)
1055 t = tree_cons (NULL_TREE, def, t);
1058 /* FIXME: use build_constructor directly. */
1059 vec_inv = build_constructor_from_list (vector_type, t);
1060 return vect_init_vector (stmt, vec_inv, vector_type, NULL);
1063 /* Case 3: operand is defined inside the loop. */
1064 case vect_internal_def:
1066 if (scalar_def)
1067 *scalar_def = NULL/* FIXME tuples: def_stmt*/;
1069 /* Get the def from the vectorized stmt. */
1070 def_stmt_info = vinfo_for_stmt (def_stmt);
1071 vec_stmt = STMT_VINFO_VEC_STMT (def_stmt_info);
1072 gcc_assert (vec_stmt);
1073 if (gimple_code (vec_stmt) == GIMPLE_PHI)
1074 vec_oprnd = PHI_RESULT (vec_stmt);
1075 else if (is_gimple_call (vec_stmt))
1076 vec_oprnd = gimple_call_lhs (vec_stmt);
1077 else
1078 vec_oprnd = gimple_assign_lhs (vec_stmt);
1079 return vec_oprnd;
1082 /* Case 4: operand is defined by a loop header phi - reduction */
1083 case vect_reduction_def:
1084 case vect_double_reduction_def:
1085 case vect_nested_cycle:
1087 struct loop *loop;
1089 gcc_assert (gimple_code (def_stmt) == GIMPLE_PHI);
1090 loop = (gimple_bb (def_stmt))->loop_father;
1092 /* Get the def before the loop */
1093 op = PHI_ARG_DEF_FROM_EDGE (def_stmt, loop_preheader_edge (loop));
1094 return get_initial_def_for_reduction (stmt, op, scalar_def);
1097 /* Case 5: operand is defined by loop-header phi - induction. */
1098 case vect_induction_def:
1100 gcc_assert (gimple_code (def_stmt) == GIMPLE_PHI);
1102 /* Get the def from the vectorized stmt. */
1103 def_stmt_info = vinfo_for_stmt (def_stmt);
1104 vec_stmt = STMT_VINFO_VEC_STMT (def_stmt_info);
1105 if (gimple_code (vec_stmt) == GIMPLE_PHI)
1106 vec_oprnd = PHI_RESULT (vec_stmt);
1107 else
1108 vec_oprnd = gimple_get_lhs (vec_stmt);
1109 return vec_oprnd;
1112 default:
1113 gcc_unreachable ();
1118 /* Function vect_get_vec_def_for_stmt_copy
1120 Return a vector-def for an operand. This function is used when the
1121 vectorized stmt to be created (by the caller to this function) is a "copy"
1122 created in case the vectorized result cannot fit in one vector, and several
1123 copies of the vector-stmt are required. In this case the vector-def is
1124 retrieved from the vector stmt recorded in the STMT_VINFO_RELATED_STMT field
1125 of the stmt that defines VEC_OPRND.
1126 DT is the type of the vector def VEC_OPRND.
1128 Context:
1129 In case the vectorization factor (VF) is bigger than the number
1130 of elements that can fit in a vectype (nunits), we have to generate
1131 more than one vector stmt to vectorize the scalar stmt. This situation
1132 arises when there are multiple data-types operated upon in the loop; the
1133 smallest data-type determines the VF, and as a result, when vectorizing
1134 stmts operating on wider types we need to create 'VF/nunits' "copies" of the
1135 vector stmt (each computing a vector of 'nunits' results, and together
1136 computing 'VF' results in each iteration). This function is called when
1137 vectorizing such a stmt (e.g. vectorizing S2 in the illustration below, in
1138 which VF=16 and nunits=4, so the number of copies required is 4):
1140 scalar stmt: vectorized into: STMT_VINFO_RELATED_STMT
1142 S1: x = load VS1.0: vx.0 = memref0 VS1.1
1143 VS1.1: vx.1 = memref1 VS1.2
1144 VS1.2: vx.2 = memref2 VS1.3
1145 VS1.3: vx.3 = memref3
1147 S2: z = x + ... VSnew.0: vz0 = vx.0 + ... VSnew.1
1148 VSnew.1: vz1 = vx.1 + ... VSnew.2
1149 VSnew.2: vz2 = vx.2 + ... VSnew.3
1150 VSnew.3: vz3 = vx.3 + ...
1152 The vectorization of S1 is explained in vectorizable_load.
1153 The vectorization of S2:
1154 To create the first vector-stmt out of the 4 copies - VSnew.0 -
1155 the function 'vect_get_vec_def_for_operand' is called to
1156 get the relevant vector-def for each operand of S2. For operand x it
1157 returns the vector-def 'vx.0'.
1159 To create the remaining copies of the vector-stmt (VSnew.j), this
1160 function is called to get the relevant vector-def for each operand. It is
1161 obtained from the respective VS1.j stmt, which is recorded in the
1162 STMT_VINFO_RELATED_STMT field of the stmt that defines VEC_OPRND.
1164 For example, to obtain the vector-def 'vx.1' in order to create the
1165 vector stmt 'VSnew.1', this function is called with VEC_OPRND='vx.0'.
1166 Given 'vx0' we obtain the stmt that defines it ('VS1.0'); from the
1167 STMT_VINFO_RELATED_STMT field of 'VS1.0' we obtain the next copy - 'VS1.1',
1168 and return its def ('vx.1').
1169 Overall, to create the above sequence this function will be called 3 times:
1170 vx.1 = vect_get_vec_def_for_stmt_copy (dt, vx.0);
1171 vx.2 = vect_get_vec_def_for_stmt_copy (dt, vx.1);
1172 vx.3 = vect_get_vec_def_for_stmt_copy (dt, vx.2); */
1174 tree
1175 vect_get_vec_def_for_stmt_copy (enum vect_def_type dt, tree vec_oprnd)
1177 gimple vec_stmt_for_operand;
1178 stmt_vec_info def_stmt_info;
1180 /* Do nothing; can reuse same def. */
1181 if (dt == vect_external_def || dt == vect_constant_def )
1182 return vec_oprnd;
1184 vec_stmt_for_operand = SSA_NAME_DEF_STMT (vec_oprnd);
1185 def_stmt_info = vinfo_for_stmt (vec_stmt_for_operand);
1186 gcc_assert (def_stmt_info);
1187 vec_stmt_for_operand = STMT_VINFO_RELATED_STMT (def_stmt_info);
1188 gcc_assert (vec_stmt_for_operand);
1189 vec_oprnd = gimple_get_lhs (vec_stmt_for_operand);
1190 if (gimple_code (vec_stmt_for_operand) == GIMPLE_PHI)
1191 vec_oprnd = PHI_RESULT (vec_stmt_for_operand);
1192 else
1193 vec_oprnd = gimple_get_lhs (vec_stmt_for_operand);
1194 return vec_oprnd;
1198 /* Get vectorized definitions for the operands to create a copy of an original
1199 stmt. See vect_get_vec_def_for_stmt_copy () for details. */
1201 static void
1202 vect_get_vec_defs_for_stmt_copy (enum vect_def_type *dt,
1203 VEC(tree,heap) **vec_oprnds0,
1204 VEC(tree,heap) **vec_oprnds1)
1206 tree vec_oprnd = VEC_pop (tree, *vec_oprnds0);
1208 vec_oprnd = vect_get_vec_def_for_stmt_copy (dt[0], vec_oprnd);
1209 VEC_quick_push (tree, *vec_oprnds0, vec_oprnd);
1211 if (vec_oprnds1 && *vec_oprnds1)
1213 vec_oprnd = VEC_pop (tree, *vec_oprnds1);
1214 vec_oprnd = vect_get_vec_def_for_stmt_copy (dt[1], vec_oprnd);
1215 VEC_quick_push (tree, *vec_oprnds1, vec_oprnd);
1220 /* Get vectorized definitions for OP0 and OP1, or SLP_NODE if it is not
1221 NULL. */
1223 static void
1224 vect_get_vec_defs (tree op0, tree op1, gimple stmt,
1225 VEC(tree,heap) **vec_oprnds0, VEC(tree,heap) **vec_oprnds1,
1226 slp_tree slp_node)
1228 if (slp_node)
1229 vect_get_slp_defs (op0, op1, slp_node, vec_oprnds0, vec_oprnds1, -1);
1230 else
1232 tree vec_oprnd;
1234 *vec_oprnds0 = VEC_alloc (tree, heap, 1);
1235 vec_oprnd = vect_get_vec_def_for_operand (op0, stmt, NULL);
1236 VEC_quick_push (tree, *vec_oprnds0, vec_oprnd);
1238 if (op1)
1240 *vec_oprnds1 = VEC_alloc (tree, heap, 1);
1241 vec_oprnd = vect_get_vec_def_for_operand (op1, stmt, NULL);
1242 VEC_quick_push (tree, *vec_oprnds1, vec_oprnd);
1248 /* Function vect_finish_stmt_generation.
1250 Insert a new stmt. */
1252 void
1253 vect_finish_stmt_generation (gimple stmt, gimple vec_stmt,
1254 gimple_stmt_iterator *gsi)
1256 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
1257 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
1258 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
1260 gcc_assert (gimple_code (stmt) != GIMPLE_LABEL);
1262 gsi_insert_before (gsi, vec_stmt, GSI_SAME_STMT);
1264 set_vinfo_for_stmt (vec_stmt, new_stmt_vec_info (vec_stmt, loop_vinfo,
1265 bb_vinfo));
1267 if (vect_print_dump_info (REPORT_DETAILS))
1269 fprintf (vect_dump, "add new stmt: ");
1270 print_gimple_stmt (vect_dump, vec_stmt, 0, TDF_SLIM);
1273 gimple_set_location (vec_stmt, gimple_location (gsi_stmt (*gsi)));
1276 /* Checks if CALL can be vectorized in type VECTYPE. Returns
1277 a function declaration if the target has a vectorized version
1278 of the function, or NULL_TREE if the function cannot be vectorized. */
1280 tree
1281 vectorizable_function (gimple call, tree vectype_out, tree vectype_in)
1283 tree fndecl = gimple_call_fndecl (call);
1285 /* We only handle functions that do not read or clobber memory -- i.e.
1286 const or novops ones. */
1287 if (!(gimple_call_flags (call) & (ECF_CONST | ECF_NOVOPS)))
1288 return NULL_TREE;
1290 if (!fndecl
1291 || TREE_CODE (fndecl) != FUNCTION_DECL
1292 || !DECL_BUILT_IN (fndecl))
1293 return NULL_TREE;
1295 return targetm.vectorize.builtin_vectorized_function (fndecl, vectype_out,
1296 vectype_in);
1299 /* Function vectorizable_call.
1301 Check if STMT performs a function call that can be vectorized.
1302 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
1303 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
1304 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
1306 static bool
1307 vectorizable_call (gimple stmt, gimple_stmt_iterator *gsi, gimple *vec_stmt)
1309 tree vec_dest;
1310 tree scalar_dest;
1311 tree op, type;
1312 tree vec_oprnd0 = NULL_TREE, vec_oprnd1 = NULL_TREE;
1313 stmt_vec_info stmt_info = vinfo_for_stmt (stmt), prev_stmt_info;
1314 tree vectype_out, vectype_in;
1315 int nunits_in;
1316 int nunits_out;
1317 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
1318 tree fndecl, new_temp, def, rhs_type;
1319 gimple def_stmt;
1320 enum vect_def_type dt[3]
1321 = {vect_unknown_def_type, vect_unknown_def_type, vect_unknown_def_type};
1322 gimple new_stmt = NULL;
1323 int ncopies, j;
1324 VEC(tree, heap) *vargs = NULL;
1325 enum { NARROW, NONE, WIDEN } modifier;
1326 size_t i, nargs;
1328 /* FORNOW: unsupported in basic block SLP. */
1329 gcc_assert (loop_vinfo);
1331 if (!STMT_VINFO_RELEVANT_P (stmt_info))
1332 return false;
1334 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
1335 return false;
1337 /* FORNOW: SLP not supported. */
1338 if (STMT_SLP_TYPE (stmt_info))
1339 return false;
1341 /* Is STMT a vectorizable call? */
1342 if (!is_gimple_call (stmt))
1343 return false;
1345 if (TREE_CODE (gimple_call_lhs (stmt)) != SSA_NAME)
1346 return false;
1348 if (stmt_could_throw_p (stmt))
1349 return false;
1351 vectype_out = STMT_VINFO_VECTYPE (stmt_info);
1353 /* Process function arguments. */
1354 rhs_type = NULL_TREE;
1355 vectype_in = NULL_TREE;
1356 nargs = gimple_call_num_args (stmt);
1358 /* Bail out if the function has more than three arguments, we do not have
1359 interesting builtin functions to vectorize with more than two arguments
1360 except for fma. No arguments is also not good. */
1361 if (nargs == 0 || nargs > 3)
1362 return false;
1364 for (i = 0; i < nargs; i++)
1366 tree opvectype;
1368 op = gimple_call_arg (stmt, i);
1370 /* We can only handle calls with arguments of the same type. */
1371 if (rhs_type
1372 && !types_compatible_p (rhs_type, TREE_TYPE (op)))
1374 if (vect_print_dump_info (REPORT_DETAILS))
1375 fprintf (vect_dump, "argument types differ.");
1376 return false;
1378 if (!rhs_type)
1379 rhs_type = TREE_TYPE (op);
1381 if (!vect_is_simple_use_1 (op, loop_vinfo, NULL,
1382 &def_stmt, &def, &dt[i], &opvectype))
1384 if (vect_print_dump_info (REPORT_DETAILS))
1385 fprintf (vect_dump, "use not simple.");
1386 return false;
1389 if (!vectype_in)
1390 vectype_in = opvectype;
1391 else if (opvectype
1392 && opvectype != vectype_in)
1394 if (vect_print_dump_info (REPORT_DETAILS))
1395 fprintf (vect_dump, "argument vector types differ.");
1396 return false;
1399 /* If all arguments are external or constant defs use a vector type with
1400 the same size as the output vector type. */
1401 if (!vectype_in)
1402 vectype_in = get_same_sized_vectype (rhs_type, vectype_out);
1403 if (vec_stmt)
1404 gcc_assert (vectype_in);
1405 if (!vectype_in)
1407 if (vect_print_dump_info (REPORT_DETAILS))
1409 fprintf (vect_dump, "no vectype for scalar type ");
1410 print_generic_expr (vect_dump, rhs_type, TDF_SLIM);
1413 return false;
1416 /* FORNOW */
1417 nunits_in = TYPE_VECTOR_SUBPARTS (vectype_in);
1418 nunits_out = TYPE_VECTOR_SUBPARTS (vectype_out);
1419 if (nunits_in == nunits_out / 2)
1420 modifier = NARROW;
1421 else if (nunits_out == nunits_in)
1422 modifier = NONE;
1423 else if (nunits_out == nunits_in / 2)
1424 modifier = WIDEN;
1425 else
1426 return false;
1428 /* For now, we only vectorize functions if a target specific builtin
1429 is available. TODO -- in some cases, it might be profitable to
1430 insert the calls for pieces of the vector, in order to be able
1431 to vectorize other operations in the loop. */
1432 fndecl = vectorizable_function (stmt, vectype_out, vectype_in);
1433 if (fndecl == NULL_TREE)
1435 if (vect_print_dump_info (REPORT_DETAILS))
1436 fprintf (vect_dump, "function is not vectorizable.");
1438 return false;
1441 gcc_assert (!gimple_vuse (stmt));
1443 if (modifier == NARROW)
1444 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_out;
1445 else
1446 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_in;
1448 /* Sanity check: make sure that at least one copy of the vectorized stmt
1449 needs to be generated. */
1450 gcc_assert (ncopies >= 1);
1452 if (!vec_stmt) /* transformation not required. */
1454 STMT_VINFO_TYPE (stmt_info) = call_vec_info_type;
1455 if (vect_print_dump_info (REPORT_DETAILS))
1456 fprintf (vect_dump, "=== vectorizable_call ===");
1457 vect_model_simple_cost (stmt_info, ncopies, dt, NULL);
1458 return true;
1461 /** Transform. **/
1463 if (vect_print_dump_info (REPORT_DETAILS))
1464 fprintf (vect_dump, "transform operation.");
1466 /* Handle def. */
1467 scalar_dest = gimple_call_lhs (stmt);
1468 vec_dest = vect_create_destination_var (scalar_dest, vectype_out);
1470 prev_stmt_info = NULL;
1471 switch (modifier)
1473 case NONE:
1474 for (j = 0; j < ncopies; ++j)
1476 /* Build argument list for the vectorized call. */
1477 if (j == 0)
1478 vargs = VEC_alloc (tree, heap, nargs);
1479 else
1480 VEC_truncate (tree, vargs, 0);
1482 for (i = 0; i < nargs; i++)
1484 op = gimple_call_arg (stmt, i);
1485 if (j == 0)
1486 vec_oprnd0
1487 = vect_get_vec_def_for_operand (op, stmt, NULL);
1488 else
1490 vec_oprnd0 = gimple_call_arg (new_stmt, i);
1491 vec_oprnd0
1492 = vect_get_vec_def_for_stmt_copy (dt[i], vec_oprnd0);
1495 VEC_quick_push (tree, vargs, vec_oprnd0);
1498 new_stmt = gimple_build_call_vec (fndecl, vargs);
1499 new_temp = make_ssa_name (vec_dest, new_stmt);
1500 gimple_call_set_lhs (new_stmt, new_temp);
1502 vect_finish_stmt_generation (stmt, new_stmt, gsi);
1503 mark_symbols_for_renaming (new_stmt);
1505 if (j == 0)
1506 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
1507 else
1508 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
1510 prev_stmt_info = vinfo_for_stmt (new_stmt);
1513 break;
1515 case NARROW:
1516 for (j = 0; j < ncopies; ++j)
1518 /* Build argument list for the vectorized call. */
1519 if (j == 0)
1520 vargs = VEC_alloc (tree, heap, nargs * 2);
1521 else
1522 VEC_truncate (tree, vargs, 0);
1524 for (i = 0; i < nargs; i++)
1526 op = gimple_call_arg (stmt, i);
1527 if (j == 0)
1529 vec_oprnd0
1530 = vect_get_vec_def_for_operand (op, stmt, NULL);
1531 vec_oprnd1
1532 = vect_get_vec_def_for_stmt_copy (dt[i], vec_oprnd0);
1534 else
1536 vec_oprnd1 = gimple_call_arg (new_stmt, 2*i);
1537 vec_oprnd0
1538 = vect_get_vec_def_for_stmt_copy (dt[i], vec_oprnd1);
1539 vec_oprnd1
1540 = vect_get_vec_def_for_stmt_copy (dt[i], vec_oprnd0);
1543 VEC_quick_push (tree, vargs, vec_oprnd0);
1544 VEC_quick_push (tree, vargs, vec_oprnd1);
1547 new_stmt = gimple_build_call_vec (fndecl, vargs);
1548 new_temp = make_ssa_name (vec_dest, new_stmt);
1549 gimple_call_set_lhs (new_stmt, new_temp);
1551 vect_finish_stmt_generation (stmt, new_stmt, gsi);
1552 mark_symbols_for_renaming (new_stmt);
1554 if (j == 0)
1555 STMT_VINFO_VEC_STMT (stmt_info) = new_stmt;
1556 else
1557 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
1559 prev_stmt_info = vinfo_for_stmt (new_stmt);
1562 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
1564 break;
1566 case WIDEN:
1567 /* No current target implements this case. */
1568 return false;
1571 VEC_free (tree, heap, vargs);
1573 /* Update the exception handling table with the vector stmt if necessary. */
1574 if (maybe_clean_or_replace_eh_stmt (stmt, *vec_stmt))
1575 gimple_purge_dead_eh_edges (gimple_bb (stmt));
1577 /* The call in STMT might prevent it from being removed in dce.
1578 We however cannot remove it here, due to the way the ssa name
1579 it defines is mapped to the new definition. So just replace
1580 rhs of the statement with something harmless. */
1582 type = TREE_TYPE (scalar_dest);
1583 new_stmt = gimple_build_assign (gimple_call_lhs (stmt),
1584 build_zero_cst (type));
1585 set_vinfo_for_stmt (new_stmt, stmt_info);
1586 set_vinfo_for_stmt (stmt, NULL);
1587 STMT_VINFO_STMT (stmt_info) = new_stmt;
1588 gsi_replace (gsi, new_stmt, false);
1589 SSA_NAME_DEF_STMT (gimple_assign_lhs (new_stmt)) = new_stmt;
1591 return true;
1595 /* Function vect_gen_widened_results_half
1597 Create a vector stmt whose code, type, number of arguments, and result
1598 variable are CODE, OP_TYPE, and VEC_DEST, and its arguments are
1599 VEC_OPRND0 and VEC_OPRND1. The new vector stmt is to be inserted at BSI.
1600 In the case that CODE is a CALL_EXPR, this means that a call to DECL
1601 needs to be created (DECL is a function-decl of a target-builtin).
1602 STMT is the original scalar stmt that we are vectorizing. */
1604 static gimple
1605 vect_gen_widened_results_half (enum tree_code code,
1606 tree decl,
1607 tree vec_oprnd0, tree vec_oprnd1, int op_type,
1608 tree vec_dest, gimple_stmt_iterator *gsi,
1609 gimple stmt)
1611 gimple new_stmt;
1612 tree new_temp;
1614 /* Generate half of the widened result: */
1615 if (code == CALL_EXPR)
1617 /* Target specific support */
1618 if (op_type == binary_op)
1619 new_stmt = gimple_build_call (decl, 2, vec_oprnd0, vec_oprnd1);
1620 else
1621 new_stmt = gimple_build_call (decl, 1, vec_oprnd0);
1622 new_temp = make_ssa_name (vec_dest, new_stmt);
1623 gimple_call_set_lhs (new_stmt, new_temp);
1625 else
1627 /* Generic support */
1628 gcc_assert (op_type == TREE_CODE_LENGTH (code));
1629 if (op_type != binary_op)
1630 vec_oprnd1 = NULL;
1631 new_stmt = gimple_build_assign_with_ops (code, vec_dest, vec_oprnd0,
1632 vec_oprnd1);
1633 new_temp = make_ssa_name (vec_dest, new_stmt);
1634 gimple_assign_set_lhs (new_stmt, new_temp);
1636 vect_finish_stmt_generation (stmt, new_stmt, gsi);
1638 return new_stmt;
1642 /* Check if STMT performs a conversion operation, that can be vectorized.
1643 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
1644 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
1645 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
1647 static bool
1648 vectorizable_conversion (gimple stmt, gimple_stmt_iterator *gsi,
1649 gimple *vec_stmt, slp_tree slp_node)
1651 tree vec_dest;
1652 tree scalar_dest;
1653 tree op0;
1654 tree vec_oprnd0 = NULL_TREE, vec_oprnd1 = NULL_TREE;
1655 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
1656 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
1657 enum tree_code code, code1 = ERROR_MARK, code2 = ERROR_MARK;
1658 tree decl1 = NULL_TREE, decl2 = NULL_TREE;
1659 tree new_temp;
1660 tree def;
1661 gimple def_stmt;
1662 enum vect_def_type dt[2] = {vect_unknown_def_type, vect_unknown_def_type};
1663 gimple new_stmt = NULL;
1664 stmt_vec_info prev_stmt_info;
1665 int nunits_in;
1666 int nunits_out;
1667 tree vectype_out, vectype_in;
1668 int ncopies, j;
1669 tree rhs_type;
1670 tree builtin_decl;
1671 enum { NARROW, NONE, WIDEN } modifier;
1672 int i;
1673 VEC(tree,heap) *vec_oprnds0 = NULL;
1674 tree vop0;
1675 VEC(tree,heap) *dummy = NULL;
1676 int dummy_int;
1678 /* Is STMT a vectorizable conversion? */
1680 /* FORNOW: unsupported in basic block SLP. */
1681 gcc_assert (loop_vinfo);
1683 if (!STMT_VINFO_RELEVANT_P (stmt_info))
1684 return false;
1686 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
1687 return false;
1689 if (!is_gimple_assign (stmt))
1690 return false;
1692 if (TREE_CODE (gimple_assign_lhs (stmt)) != SSA_NAME)
1693 return false;
1695 code = gimple_assign_rhs_code (stmt);
1696 if (code != FIX_TRUNC_EXPR && code != FLOAT_EXPR)
1697 return false;
1699 /* Check types of lhs and rhs. */
1700 scalar_dest = gimple_assign_lhs (stmt);
1701 vectype_out = STMT_VINFO_VECTYPE (stmt_info);
1703 op0 = gimple_assign_rhs1 (stmt);
1704 rhs_type = TREE_TYPE (op0);
1705 /* Check the operands of the operation. */
1706 if (!vect_is_simple_use_1 (op0, loop_vinfo, NULL,
1707 &def_stmt, &def, &dt[0], &vectype_in))
1709 if (vect_print_dump_info (REPORT_DETAILS))
1710 fprintf (vect_dump, "use not simple.");
1711 return false;
1713 /* If op0 is an external or constant defs use a vector type of
1714 the same size as the output vector type. */
1715 if (!vectype_in)
1716 vectype_in = get_same_sized_vectype (rhs_type, vectype_out);
1717 if (vec_stmt)
1718 gcc_assert (vectype_in);
1719 if (!vectype_in)
1721 if (vect_print_dump_info (REPORT_DETAILS))
1723 fprintf (vect_dump, "no vectype for scalar type ");
1724 print_generic_expr (vect_dump, rhs_type, TDF_SLIM);
1727 return false;
1730 /* FORNOW */
1731 nunits_in = TYPE_VECTOR_SUBPARTS (vectype_in);
1732 nunits_out = TYPE_VECTOR_SUBPARTS (vectype_out);
1733 if (nunits_in == nunits_out / 2)
1734 modifier = NARROW;
1735 else if (nunits_out == nunits_in)
1736 modifier = NONE;
1737 else if (nunits_out == nunits_in / 2)
1738 modifier = WIDEN;
1739 else
1740 return false;
1742 if (modifier == NARROW)
1743 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_out;
1744 else
1745 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_in;
1747 /* Multiple types in SLP are handled by creating the appropriate number of
1748 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
1749 case of SLP. */
1750 if (slp_node)
1751 ncopies = 1;
1753 /* Sanity check: make sure that at least one copy of the vectorized stmt
1754 needs to be generated. */
1755 gcc_assert (ncopies >= 1);
1757 /* Supportable by target? */
1758 if ((modifier == NONE
1759 && !targetm.vectorize.builtin_conversion (code, vectype_out, vectype_in))
1760 || (modifier == WIDEN
1761 && !supportable_widening_operation (code, stmt,
1762 vectype_out, vectype_in,
1763 &decl1, &decl2,
1764 &code1, &code2,
1765 &dummy_int, &dummy))
1766 || (modifier == NARROW
1767 && !supportable_narrowing_operation (code, vectype_out, vectype_in,
1768 &code1, &dummy_int, &dummy)))
1770 if (vect_print_dump_info (REPORT_DETAILS))
1771 fprintf (vect_dump, "conversion not supported by target.");
1772 return false;
1775 if (modifier != NONE)
1777 /* FORNOW: SLP not supported. */
1778 if (STMT_SLP_TYPE (stmt_info))
1779 return false;
1782 if (!vec_stmt) /* transformation not required. */
1784 STMT_VINFO_TYPE (stmt_info) = type_conversion_vec_info_type;
1785 return true;
1788 /** Transform. **/
1789 if (vect_print_dump_info (REPORT_DETAILS))
1790 fprintf (vect_dump, "transform conversion.");
1792 /* Handle def. */
1793 vec_dest = vect_create_destination_var (scalar_dest, vectype_out);
1795 if (modifier == NONE && !slp_node)
1796 vec_oprnds0 = VEC_alloc (tree, heap, 1);
1798 prev_stmt_info = NULL;
1799 switch (modifier)
1801 case NONE:
1802 for (j = 0; j < ncopies; j++)
1804 if (j == 0)
1805 vect_get_vec_defs (op0, NULL, stmt, &vec_oprnds0, NULL, slp_node);
1806 else
1807 vect_get_vec_defs_for_stmt_copy (dt, &vec_oprnds0, NULL);
1809 builtin_decl =
1810 targetm.vectorize.builtin_conversion (code,
1811 vectype_out, vectype_in);
1812 FOR_EACH_VEC_ELT (tree, vec_oprnds0, i, vop0)
1814 /* Arguments are ready. create the new vector stmt. */
1815 new_stmt = gimple_build_call (builtin_decl, 1, vop0);
1816 new_temp = make_ssa_name (vec_dest, new_stmt);
1817 gimple_call_set_lhs (new_stmt, new_temp);
1818 vect_finish_stmt_generation (stmt, new_stmt, gsi);
1819 if (slp_node)
1820 VEC_quick_push (gimple, SLP_TREE_VEC_STMTS (slp_node), new_stmt);
1823 if (j == 0)
1824 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
1825 else
1826 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
1827 prev_stmt_info = vinfo_for_stmt (new_stmt);
1829 break;
1831 case WIDEN:
1832 /* In case the vectorization factor (VF) is bigger than the number
1833 of elements that we can fit in a vectype (nunits), we have to
1834 generate more than one vector stmt - i.e - we need to "unroll"
1835 the vector stmt by a factor VF/nunits. */
1836 for (j = 0; j < ncopies; j++)
1838 if (j == 0)
1839 vec_oprnd0 = vect_get_vec_def_for_operand (op0, stmt, NULL);
1840 else
1841 vec_oprnd0 = vect_get_vec_def_for_stmt_copy (dt[0], vec_oprnd0);
1843 /* Generate first half of the widened result: */
1844 new_stmt
1845 = vect_gen_widened_results_half (code1, decl1,
1846 vec_oprnd0, vec_oprnd1,
1847 unary_op, vec_dest, gsi, stmt);
1848 if (j == 0)
1849 STMT_VINFO_VEC_STMT (stmt_info) = new_stmt;
1850 else
1851 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
1852 prev_stmt_info = vinfo_for_stmt (new_stmt);
1854 /* Generate second half of the widened result: */
1855 new_stmt
1856 = vect_gen_widened_results_half (code2, decl2,
1857 vec_oprnd0, vec_oprnd1,
1858 unary_op, vec_dest, gsi, stmt);
1859 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
1860 prev_stmt_info = vinfo_for_stmt (new_stmt);
1862 break;
1864 case NARROW:
1865 /* In case the vectorization factor (VF) is bigger than the number
1866 of elements that we can fit in a vectype (nunits), we have to
1867 generate more than one vector stmt - i.e - we need to "unroll"
1868 the vector stmt by a factor VF/nunits. */
1869 for (j = 0; j < ncopies; j++)
1871 /* Handle uses. */
1872 if (j == 0)
1874 vec_oprnd0 = vect_get_vec_def_for_operand (op0, stmt, NULL);
1875 vec_oprnd1 = vect_get_vec_def_for_stmt_copy (dt[0], vec_oprnd0);
1877 else
1879 vec_oprnd0 = vect_get_vec_def_for_stmt_copy (dt[0], vec_oprnd1);
1880 vec_oprnd1 = vect_get_vec_def_for_stmt_copy (dt[0], vec_oprnd0);
1883 /* Arguments are ready. Create the new vector stmt. */
1884 new_stmt = gimple_build_assign_with_ops (code1, vec_dest, vec_oprnd0,
1885 vec_oprnd1);
1886 new_temp = make_ssa_name (vec_dest, new_stmt);
1887 gimple_assign_set_lhs (new_stmt, new_temp);
1888 vect_finish_stmt_generation (stmt, new_stmt, gsi);
1890 if (j == 0)
1891 STMT_VINFO_VEC_STMT (stmt_info) = new_stmt;
1892 else
1893 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
1895 prev_stmt_info = vinfo_for_stmt (new_stmt);
1898 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
1901 if (vec_oprnds0)
1902 VEC_free (tree, heap, vec_oprnds0);
1904 return true;
1908 /* Function vectorizable_assignment.
1910 Check if STMT performs an assignment (copy) that can be vectorized.
1911 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
1912 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
1913 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
1915 static bool
1916 vectorizable_assignment (gimple stmt, gimple_stmt_iterator *gsi,
1917 gimple *vec_stmt, slp_tree slp_node)
1919 tree vec_dest;
1920 tree scalar_dest;
1921 tree op;
1922 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
1923 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
1924 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
1925 tree new_temp;
1926 tree def;
1927 gimple def_stmt;
1928 enum vect_def_type dt[2] = {vect_unknown_def_type, vect_unknown_def_type};
1929 unsigned int nunits = TYPE_VECTOR_SUBPARTS (vectype);
1930 int ncopies;
1931 int i, j;
1932 VEC(tree,heap) *vec_oprnds = NULL;
1933 tree vop;
1934 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
1935 gimple new_stmt = NULL;
1936 stmt_vec_info prev_stmt_info = NULL;
1937 enum tree_code code;
1938 tree vectype_in;
1940 /* Multiple types in SLP are handled by creating the appropriate number of
1941 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
1942 case of SLP. */
1943 if (slp_node)
1944 ncopies = 1;
1945 else
1946 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits;
1948 gcc_assert (ncopies >= 1);
1950 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
1951 return false;
1953 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
1954 return false;
1956 /* Is vectorizable assignment? */
1957 if (!is_gimple_assign (stmt))
1958 return false;
1960 scalar_dest = gimple_assign_lhs (stmt);
1961 if (TREE_CODE (scalar_dest) != SSA_NAME)
1962 return false;
1964 code = gimple_assign_rhs_code (stmt);
1965 if (gimple_assign_single_p (stmt)
1966 || code == PAREN_EXPR
1967 || CONVERT_EXPR_CODE_P (code))
1968 op = gimple_assign_rhs1 (stmt);
1969 else
1970 return false;
1972 if (!vect_is_simple_use_1 (op, loop_vinfo, bb_vinfo,
1973 &def_stmt, &def, &dt[0], &vectype_in))
1975 if (vect_print_dump_info (REPORT_DETAILS))
1976 fprintf (vect_dump, "use not simple.");
1977 return false;
1980 /* We can handle NOP_EXPR conversions that do not change the number
1981 of elements or the vector size. */
1982 if (CONVERT_EXPR_CODE_P (code)
1983 && (!vectype_in
1984 || TYPE_VECTOR_SUBPARTS (vectype_in) != nunits
1985 || (GET_MODE_SIZE (TYPE_MODE (vectype))
1986 != GET_MODE_SIZE (TYPE_MODE (vectype_in)))))
1987 return false;
1989 if (!vec_stmt) /* transformation not required. */
1991 STMT_VINFO_TYPE (stmt_info) = assignment_vec_info_type;
1992 if (vect_print_dump_info (REPORT_DETAILS))
1993 fprintf (vect_dump, "=== vectorizable_assignment ===");
1994 vect_model_simple_cost (stmt_info, ncopies, dt, NULL);
1995 return true;
1998 /** Transform. **/
1999 if (vect_print_dump_info (REPORT_DETAILS))
2000 fprintf (vect_dump, "transform assignment.");
2002 /* Handle def. */
2003 vec_dest = vect_create_destination_var (scalar_dest, vectype);
2005 /* Handle use. */
2006 for (j = 0; j < ncopies; j++)
2008 /* Handle uses. */
2009 if (j == 0)
2010 vect_get_vec_defs (op, NULL, stmt, &vec_oprnds, NULL, slp_node);
2011 else
2012 vect_get_vec_defs_for_stmt_copy (dt, &vec_oprnds, NULL);
2014 /* Arguments are ready. create the new vector stmt. */
2015 FOR_EACH_VEC_ELT (tree, vec_oprnds, i, vop)
2017 if (CONVERT_EXPR_CODE_P (code))
2018 vop = build1 (VIEW_CONVERT_EXPR, vectype, vop);
2019 new_stmt = gimple_build_assign (vec_dest, vop);
2020 new_temp = make_ssa_name (vec_dest, new_stmt);
2021 gimple_assign_set_lhs (new_stmt, new_temp);
2022 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2023 if (slp_node)
2024 VEC_quick_push (gimple, SLP_TREE_VEC_STMTS (slp_node), new_stmt);
2027 if (slp_node)
2028 continue;
2030 if (j == 0)
2031 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
2032 else
2033 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
2035 prev_stmt_info = vinfo_for_stmt (new_stmt);
2038 VEC_free (tree, heap, vec_oprnds);
2039 return true;
2043 /* Function vectorizable_shift.
2045 Check if STMT performs a shift operation that can be vectorized.
2046 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
2047 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
2048 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
2050 static bool
2051 vectorizable_shift (gimple stmt, gimple_stmt_iterator *gsi,
2052 gimple *vec_stmt, slp_tree slp_node)
2054 tree vec_dest;
2055 tree scalar_dest;
2056 tree op0, op1 = NULL;
2057 tree vec_oprnd1 = NULL_TREE;
2058 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
2059 tree vectype;
2060 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
2061 enum tree_code code;
2062 enum machine_mode vec_mode;
2063 tree new_temp;
2064 optab optab;
2065 int icode;
2066 enum machine_mode optab_op2_mode;
2067 tree def;
2068 gimple def_stmt;
2069 enum vect_def_type dt[2] = {vect_unknown_def_type, vect_unknown_def_type};
2070 gimple new_stmt = NULL;
2071 stmt_vec_info prev_stmt_info;
2072 int nunits_in;
2073 int nunits_out;
2074 tree vectype_out;
2075 int ncopies;
2076 int j, i;
2077 VEC (tree, heap) *vec_oprnds0 = NULL, *vec_oprnds1 = NULL;
2078 tree vop0, vop1;
2079 unsigned int k;
2080 bool scalar_shift_arg = false;
2081 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
2082 int vf;
2084 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
2085 return false;
2087 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
2088 return false;
2090 /* Is STMT a vectorizable binary/unary operation? */
2091 if (!is_gimple_assign (stmt))
2092 return false;
2094 if (TREE_CODE (gimple_assign_lhs (stmt)) != SSA_NAME)
2095 return false;
2097 code = gimple_assign_rhs_code (stmt);
2099 if (!(code == LSHIFT_EXPR || code == RSHIFT_EXPR || code == LROTATE_EXPR
2100 || code == RROTATE_EXPR))
2101 return false;
2103 scalar_dest = gimple_assign_lhs (stmt);
2104 vectype_out = STMT_VINFO_VECTYPE (stmt_info);
2106 op0 = gimple_assign_rhs1 (stmt);
2107 if (!vect_is_simple_use_1 (op0, loop_vinfo, bb_vinfo,
2108 &def_stmt, &def, &dt[0], &vectype))
2110 if (vect_print_dump_info (REPORT_DETAILS))
2111 fprintf (vect_dump, "use not simple.");
2112 return false;
2114 /* If op0 is an external or constant def use a vector type with
2115 the same size as the output vector type. */
2116 if (!vectype)
2117 vectype = get_same_sized_vectype (TREE_TYPE (op0), vectype_out);
2118 if (vec_stmt)
2119 gcc_assert (vectype);
2120 if (!vectype)
2122 if (vect_print_dump_info (REPORT_DETAILS))
2124 fprintf (vect_dump, "no vectype for scalar type ");
2125 print_generic_expr (vect_dump, TREE_TYPE (op0), TDF_SLIM);
2128 return false;
2131 nunits_out = TYPE_VECTOR_SUBPARTS (vectype_out);
2132 nunits_in = TYPE_VECTOR_SUBPARTS (vectype);
2133 if (nunits_out != nunits_in)
2134 return false;
2136 op1 = gimple_assign_rhs2 (stmt);
2137 if (!vect_is_simple_use (op1, loop_vinfo, bb_vinfo, &def_stmt, &def, &dt[1]))
2139 if (vect_print_dump_info (REPORT_DETAILS))
2140 fprintf (vect_dump, "use not simple.");
2141 return false;
2144 if (loop_vinfo)
2145 vf = LOOP_VINFO_VECT_FACTOR (loop_vinfo);
2146 else
2147 vf = 1;
2149 /* Multiple types in SLP are handled by creating the appropriate number of
2150 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
2151 case of SLP. */
2152 if (slp_node)
2153 ncopies = 1;
2154 else
2155 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_in;
2157 gcc_assert (ncopies >= 1);
2159 /* Determine whether the shift amount is a vector, or scalar. If the
2160 shift/rotate amount is a vector, use the vector/vector shift optabs. */
2162 /* Vector shifted by vector. */
2163 if (dt[1] == vect_internal_def)
2165 optab = optab_for_tree_code (code, vectype, optab_vector);
2166 if (vect_print_dump_info (REPORT_DETAILS))
2167 fprintf (vect_dump, "vector/vector shift/rotate found.");
2169 /* See if the machine has a vector shifted by scalar insn and if not
2170 then see if it has a vector shifted by vector insn. */
2171 else if (dt[1] == vect_constant_def || dt[1] == vect_external_def)
2173 optab = optab_for_tree_code (code, vectype, optab_scalar);
2174 if (optab
2175 && optab_handler (optab, TYPE_MODE (vectype)) != CODE_FOR_nothing)
2177 scalar_shift_arg = true;
2178 if (vect_print_dump_info (REPORT_DETAILS))
2179 fprintf (vect_dump, "vector/scalar shift/rotate found.");
2181 else
2183 optab = optab_for_tree_code (code, vectype, optab_vector);
2184 if (optab
2185 && (optab_handler (optab, TYPE_MODE (vectype))
2186 != CODE_FOR_nothing))
2188 if (vect_print_dump_info (REPORT_DETAILS))
2189 fprintf (vect_dump, "vector/vector shift/rotate found.");
2191 /* Unlike the other binary operators, shifts/rotates have
2192 the rhs being int, instead of the same type as the lhs,
2193 so make sure the scalar is the right type if we are
2194 dealing with vectors of short/char. */
2195 if (dt[1] == vect_constant_def)
2196 op1 = fold_convert (TREE_TYPE (vectype), op1);
2200 else
2202 if (vect_print_dump_info (REPORT_DETAILS))
2203 fprintf (vect_dump, "operand mode requires invariant argument.");
2204 return false;
2207 /* Supportable by target? */
2208 if (!optab)
2210 if (vect_print_dump_info (REPORT_DETAILS))
2211 fprintf (vect_dump, "no optab.");
2212 return false;
2214 vec_mode = TYPE_MODE (vectype);
2215 icode = (int) optab_handler (optab, vec_mode);
2216 if (icode == CODE_FOR_nothing)
2218 if (vect_print_dump_info (REPORT_DETAILS))
2219 fprintf (vect_dump, "op not supported by target.");
2220 /* Check only during analysis. */
2221 if (GET_MODE_SIZE (vec_mode) != UNITS_PER_WORD
2222 || (vf < vect_min_worthwhile_factor (code)
2223 && !vec_stmt))
2224 return false;
2225 if (vect_print_dump_info (REPORT_DETAILS))
2226 fprintf (vect_dump, "proceeding using word mode.");
2229 /* Worthwhile without SIMD support? Check only during analysis. */
2230 if (!VECTOR_MODE_P (TYPE_MODE (vectype))
2231 && vf < vect_min_worthwhile_factor (code)
2232 && !vec_stmt)
2234 if (vect_print_dump_info (REPORT_DETAILS))
2235 fprintf (vect_dump, "not worthwhile without SIMD support.");
2236 return false;
2239 if (!vec_stmt) /* transformation not required. */
2241 STMT_VINFO_TYPE (stmt_info) = shift_vec_info_type;
2242 if (vect_print_dump_info (REPORT_DETAILS))
2243 fprintf (vect_dump, "=== vectorizable_shift ===");
2244 vect_model_simple_cost (stmt_info, ncopies, dt, NULL);
2245 return true;
2248 /** Transform. **/
2250 if (vect_print_dump_info (REPORT_DETAILS))
2251 fprintf (vect_dump, "transform binary/unary operation.");
2253 /* Handle def. */
2254 vec_dest = vect_create_destination_var (scalar_dest, vectype);
2256 /* Allocate VECs for vector operands. In case of SLP, vector operands are
2257 created in the previous stages of the recursion, so no allocation is
2258 needed, except for the case of shift with scalar shift argument. In that
2259 case we store the scalar operand in VEC_OPRNDS1 for every vector stmt to
2260 be created to vectorize the SLP group, i.e., SLP_NODE->VEC_STMTS_SIZE.
2261 In case of loop-based vectorization we allocate VECs of size 1. We
2262 allocate VEC_OPRNDS1 only in case of binary operation. */
2263 if (!slp_node)
2265 vec_oprnds0 = VEC_alloc (tree, heap, 1);
2266 vec_oprnds1 = VEC_alloc (tree, heap, 1);
2268 else if (scalar_shift_arg)
2269 vec_oprnds1 = VEC_alloc (tree, heap, slp_node->vec_stmts_size);
2271 prev_stmt_info = NULL;
2272 for (j = 0; j < ncopies; j++)
2274 /* Handle uses. */
2275 if (j == 0)
2277 if (scalar_shift_arg)
2279 /* Vector shl and shr insn patterns can be defined with scalar
2280 operand 2 (shift operand). In this case, use constant or loop
2281 invariant op1 directly, without extending it to vector mode
2282 first. */
2283 optab_op2_mode = insn_data[icode].operand[2].mode;
2284 if (!VECTOR_MODE_P (optab_op2_mode))
2286 if (vect_print_dump_info (REPORT_DETAILS))
2287 fprintf (vect_dump, "operand 1 using scalar mode.");
2288 vec_oprnd1 = op1;
2289 VEC_quick_push (tree, vec_oprnds1, vec_oprnd1);
2290 if (slp_node)
2292 /* Store vec_oprnd1 for every vector stmt to be created
2293 for SLP_NODE. We check during the analysis that all
2294 the shift arguments are the same.
2295 TODO: Allow different constants for different vector
2296 stmts generated for an SLP instance. */
2297 for (k = 0; k < slp_node->vec_stmts_size - 1; k++)
2298 VEC_quick_push (tree, vec_oprnds1, vec_oprnd1);
2303 /* vec_oprnd1 is available if operand 1 should be of a scalar-type
2304 (a special case for certain kind of vector shifts); otherwise,
2305 operand 1 should be of a vector type (the usual case). */
2306 if (vec_oprnd1)
2307 vect_get_vec_defs (op0, NULL_TREE, stmt, &vec_oprnds0, NULL,
2308 slp_node);
2309 else
2310 vect_get_vec_defs (op0, op1, stmt, &vec_oprnds0, &vec_oprnds1,
2311 slp_node);
2313 else
2314 vect_get_vec_defs_for_stmt_copy (dt, &vec_oprnds0, &vec_oprnds1);
2316 /* Arguments are ready. Create the new vector stmt. */
2317 FOR_EACH_VEC_ELT (tree, vec_oprnds0, i, vop0)
2319 vop1 = VEC_index (tree, vec_oprnds1, i);
2320 new_stmt = gimple_build_assign_with_ops (code, vec_dest, vop0, vop1);
2321 new_temp = make_ssa_name (vec_dest, new_stmt);
2322 gimple_assign_set_lhs (new_stmt, new_temp);
2323 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2324 if (slp_node)
2325 VEC_quick_push (gimple, SLP_TREE_VEC_STMTS (slp_node), new_stmt);
2328 if (slp_node)
2329 continue;
2331 if (j == 0)
2332 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
2333 else
2334 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
2335 prev_stmt_info = vinfo_for_stmt (new_stmt);
2338 VEC_free (tree, heap, vec_oprnds0);
2339 VEC_free (tree, heap, vec_oprnds1);
2341 return true;
2345 /* Function vectorizable_operation.
2347 Check if STMT performs a binary, unary or ternary operation that can
2348 be vectorized.
2349 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
2350 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
2351 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
2353 static bool
2354 vectorizable_operation (gimple stmt, gimple_stmt_iterator *gsi,
2355 gimple *vec_stmt, slp_tree slp_node)
2357 tree vec_dest;
2358 tree scalar_dest;
2359 tree op0, op1 = NULL_TREE, op2 = NULL_TREE;
2360 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
2361 tree vectype;
2362 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
2363 enum tree_code code;
2364 enum machine_mode vec_mode;
2365 tree new_temp;
2366 int op_type;
2367 optab optab;
2368 int icode;
2369 tree def;
2370 gimple def_stmt;
2371 enum vect_def_type dt[3]
2372 = {vect_unknown_def_type, vect_unknown_def_type, vect_unknown_def_type};
2373 gimple new_stmt = NULL;
2374 stmt_vec_info prev_stmt_info;
2375 int nunits_in;
2376 int nunits_out;
2377 tree vectype_out;
2378 int ncopies;
2379 int j, i;
2380 VEC(tree,heap) *vec_oprnds0 = NULL, *vec_oprnds1 = NULL, *vec_oprnds2 = NULL;
2381 tree vop0, vop1, vop2;
2382 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
2383 int vf;
2385 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
2386 return false;
2388 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
2389 return false;
2391 /* Is STMT a vectorizable binary/unary operation? */
2392 if (!is_gimple_assign (stmt))
2393 return false;
2395 if (TREE_CODE (gimple_assign_lhs (stmt)) != SSA_NAME)
2396 return false;
2398 code = gimple_assign_rhs_code (stmt);
2400 /* For pointer addition, we should use the normal plus for
2401 the vector addition. */
2402 if (code == POINTER_PLUS_EXPR)
2403 code = PLUS_EXPR;
2405 /* Support only unary or binary operations. */
2406 op_type = TREE_CODE_LENGTH (code);
2407 if (op_type != unary_op && op_type != binary_op && op_type != ternary_op)
2409 if (vect_print_dump_info (REPORT_DETAILS))
2410 fprintf (vect_dump, "num. args = %d (not unary/binary/ternary op).",
2411 op_type);
2412 return false;
2415 scalar_dest = gimple_assign_lhs (stmt);
2416 vectype_out = STMT_VINFO_VECTYPE (stmt_info);
2418 op0 = gimple_assign_rhs1 (stmt);
2419 if (!vect_is_simple_use_1 (op0, loop_vinfo, bb_vinfo,
2420 &def_stmt, &def, &dt[0], &vectype))
2422 if (vect_print_dump_info (REPORT_DETAILS))
2423 fprintf (vect_dump, "use not simple.");
2424 return false;
2426 /* If op0 is an external or constant def use a vector type with
2427 the same size as the output vector type. */
2428 if (!vectype)
2429 vectype = get_same_sized_vectype (TREE_TYPE (op0), vectype_out);
2430 if (vec_stmt)
2431 gcc_assert (vectype);
2432 if (!vectype)
2434 if (vect_print_dump_info (REPORT_DETAILS))
2436 fprintf (vect_dump, "no vectype for scalar type ");
2437 print_generic_expr (vect_dump, TREE_TYPE (op0), TDF_SLIM);
2440 return false;
2443 nunits_out = TYPE_VECTOR_SUBPARTS (vectype_out);
2444 nunits_in = TYPE_VECTOR_SUBPARTS (vectype);
2445 if (nunits_out != nunits_in)
2446 return false;
2448 if (op_type == binary_op || op_type == ternary_op)
2450 op1 = gimple_assign_rhs2 (stmt);
2451 if (!vect_is_simple_use (op1, loop_vinfo, bb_vinfo, &def_stmt, &def,
2452 &dt[1]))
2454 if (vect_print_dump_info (REPORT_DETAILS))
2455 fprintf (vect_dump, "use not simple.");
2456 return false;
2459 if (op_type == ternary_op)
2461 op2 = gimple_assign_rhs3 (stmt);
2462 if (!vect_is_simple_use (op2, loop_vinfo, bb_vinfo, &def_stmt, &def,
2463 &dt[2]))
2465 if (vect_print_dump_info (REPORT_DETAILS))
2466 fprintf (vect_dump, "use not simple.");
2467 return false;
2471 if (loop_vinfo)
2472 vf = LOOP_VINFO_VECT_FACTOR (loop_vinfo);
2473 else
2474 vf = 1;
2476 /* Multiple types in SLP are handled by creating the appropriate number of
2477 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
2478 case of SLP. */
2479 if (slp_node)
2480 ncopies = 1;
2481 else
2482 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_in;
2484 gcc_assert (ncopies >= 1);
2486 /* Shifts are handled in vectorizable_shift (). */
2487 if (code == LSHIFT_EXPR || code == RSHIFT_EXPR || code == LROTATE_EXPR
2488 || code == RROTATE_EXPR)
2489 return false;
2491 optab = optab_for_tree_code (code, vectype, optab_default);
2493 /* Supportable by target? */
2494 if (!optab)
2496 if (vect_print_dump_info (REPORT_DETAILS))
2497 fprintf (vect_dump, "no optab.");
2498 return false;
2500 vec_mode = TYPE_MODE (vectype);
2501 icode = (int) optab_handler (optab, vec_mode);
2502 if (icode == CODE_FOR_nothing)
2504 if (vect_print_dump_info (REPORT_DETAILS))
2505 fprintf (vect_dump, "op not supported by target.");
2506 /* Check only during analysis. */
2507 if (GET_MODE_SIZE (vec_mode) != UNITS_PER_WORD
2508 || (vf < vect_min_worthwhile_factor (code)
2509 && !vec_stmt))
2510 return false;
2511 if (vect_print_dump_info (REPORT_DETAILS))
2512 fprintf (vect_dump, "proceeding using word mode.");
2515 /* Worthwhile without SIMD support? Check only during analysis. */
2516 if (!VECTOR_MODE_P (TYPE_MODE (vectype))
2517 && vf < vect_min_worthwhile_factor (code)
2518 && !vec_stmt)
2520 if (vect_print_dump_info (REPORT_DETAILS))
2521 fprintf (vect_dump, "not worthwhile without SIMD support.");
2522 return false;
2525 if (!vec_stmt) /* transformation not required. */
2527 STMT_VINFO_TYPE (stmt_info) = op_vec_info_type;
2528 if (vect_print_dump_info (REPORT_DETAILS))
2529 fprintf (vect_dump, "=== vectorizable_operation ===");
2530 vect_model_simple_cost (stmt_info, ncopies, dt, NULL);
2531 return true;
2534 /** Transform. **/
2536 if (vect_print_dump_info (REPORT_DETAILS))
2537 fprintf (vect_dump, "transform binary/unary operation.");
2539 /* Handle def. */
2540 vec_dest = vect_create_destination_var (scalar_dest, vectype);
2542 /* Allocate VECs for vector operands. In case of SLP, vector operands are
2543 created in the previous stages of the recursion, so no allocation is
2544 needed, except for the case of shift with scalar shift argument. In that
2545 case we store the scalar operand in VEC_OPRNDS1 for every vector stmt to
2546 be created to vectorize the SLP group, i.e., SLP_NODE->VEC_STMTS_SIZE.
2547 In case of loop-based vectorization we allocate VECs of size 1. We
2548 allocate VEC_OPRNDS1 only in case of binary operation. */
2549 if (!slp_node)
2551 vec_oprnds0 = VEC_alloc (tree, heap, 1);
2552 if (op_type == binary_op || op_type == ternary_op)
2553 vec_oprnds1 = VEC_alloc (tree, heap, 1);
2554 if (op_type == ternary_op)
2555 vec_oprnds2 = VEC_alloc (tree, heap, 1);
2558 /* In case the vectorization factor (VF) is bigger than the number
2559 of elements that we can fit in a vectype (nunits), we have to generate
2560 more than one vector stmt - i.e - we need to "unroll" the
2561 vector stmt by a factor VF/nunits. In doing so, we record a pointer
2562 from one copy of the vector stmt to the next, in the field
2563 STMT_VINFO_RELATED_STMT. This is necessary in order to allow following
2564 stages to find the correct vector defs to be used when vectorizing
2565 stmts that use the defs of the current stmt. The example below
2566 illustrates the vectorization process when VF=16 and nunits=4 (i.e.,
2567 we need to create 4 vectorized stmts):
2569 before vectorization:
2570 RELATED_STMT VEC_STMT
2571 S1: x = memref - -
2572 S2: z = x + 1 - -
2574 step 1: vectorize stmt S1 (done in vectorizable_load. See more details
2575 there):
2576 RELATED_STMT VEC_STMT
2577 VS1_0: vx0 = memref0 VS1_1 -
2578 VS1_1: vx1 = memref1 VS1_2 -
2579 VS1_2: vx2 = memref2 VS1_3 -
2580 VS1_3: vx3 = memref3 - -
2581 S1: x = load - VS1_0
2582 S2: z = x + 1 - -
2584 step2: vectorize stmt S2 (done here):
2585 To vectorize stmt S2 we first need to find the relevant vector
2586 def for the first operand 'x'. This is, as usual, obtained from
2587 the vector stmt recorded in the STMT_VINFO_VEC_STMT of the stmt
2588 that defines 'x' (S1). This way we find the stmt VS1_0, and the
2589 relevant vector def 'vx0'. Having found 'vx0' we can generate
2590 the vector stmt VS2_0, and as usual, record it in the
2591 STMT_VINFO_VEC_STMT of stmt S2.
2592 When creating the second copy (VS2_1), we obtain the relevant vector
2593 def from the vector stmt recorded in the STMT_VINFO_RELATED_STMT of
2594 stmt VS1_0. This way we find the stmt VS1_1 and the relevant
2595 vector def 'vx1'. Using 'vx1' we create stmt VS2_1 and record a
2596 pointer to it in the STMT_VINFO_RELATED_STMT of the vector stmt VS2_0.
2597 Similarly when creating stmts VS2_2 and VS2_3. This is the resulting
2598 chain of stmts and pointers:
2599 RELATED_STMT VEC_STMT
2600 VS1_0: vx0 = memref0 VS1_1 -
2601 VS1_1: vx1 = memref1 VS1_2 -
2602 VS1_2: vx2 = memref2 VS1_3 -
2603 VS1_3: vx3 = memref3 - -
2604 S1: x = load - VS1_0
2605 VS2_0: vz0 = vx0 + v1 VS2_1 -
2606 VS2_1: vz1 = vx1 + v1 VS2_2 -
2607 VS2_2: vz2 = vx2 + v1 VS2_3 -
2608 VS2_3: vz3 = vx3 + v1 - -
2609 S2: z = x + 1 - VS2_0 */
2611 prev_stmt_info = NULL;
2612 for (j = 0; j < ncopies; j++)
2614 /* Handle uses. */
2615 if (j == 0)
2617 if (op_type == binary_op || op_type == ternary_op)
2618 vect_get_vec_defs (op0, op1, stmt, &vec_oprnds0, &vec_oprnds1,
2619 slp_node);
2620 else
2621 vect_get_vec_defs (op0, NULL_TREE, stmt, &vec_oprnds0, NULL,
2622 slp_node);
2623 if (op_type == ternary_op)
2625 vec_oprnds2 = VEC_alloc (tree, heap, 1);
2626 VEC_quick_push (tree, vec_oprnds2,
2627 vect_get_vec_def_for_operand (op2, stmt, NULL));
2630 else
2632 vect_get_vec_defs_for_stmt_copy (dt, &vec_oprnds0, &vec_oprnds1);
2633 if (op_type == ternary_op)
2635 tree vec_oprnd = VEC_pop (tree, vec_oprnds2);
2636 VEC_quick_push (tree, vec_oprnds2,
2637 vect_get_vec_def_for_stmt_copy (dt[2],
2638 vec_oprnd));
2642 /* Arguments are ready. Create the new vector stmt. */
2643 FOR_EACH_VEC_ELT (tree, vec_oprnds0, i, vop0)
2645 vop1 = ((op_type == binary_op || op_type == ternary_op)
2646 ? VEC_index (tree, vec_oprnds1, i) : NULL_TREE);
2647 vop2 = ((op_type == ternary_op)
2648 ? VEC_index (tree, vec_oprnds2, i) : NULL_TREE);
2649 new_stmt = gimple_build_assign_with_ops3 (code, vec_dest,
2650 vop0, vop1, vop2);
2651 new_temp = make_ssa_name (vec_dest, new_stmt);
2652 gimple_assign_set_lhs (new_stmt, new_temp);
2653 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2654 if (slp_node)
2655 VEC_quick_push (gimple, SLP_TREE_VEC_STMTS (slp_node), new_stmt);
2658 if (slp_node)
2659 continue;
2661 if (j == 0)
2662 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
2663 else
2664 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
2665 prev_stmt_info = vinfo_for_stmt (new_stmt);
2668 VEC_free (tree, heap, vec_oprnds0);
2669 if (vec_oprnds1)
2670 VEC_free (tree, heap, vec_oprnds1);
2671 if (vec_oprnds2)
2672 VEC_free (tree, heap, vec_oprnds2);
2674 return true;
2678 /* Get vectorized definitions for loop-based vectorization. For the first
2679 operand we call vect_get_vec_def_for_operand() (with OPRND containing
2680 scalar operand), and for the rest we get a copy with
2681 vect_get_vec_def_for_stmt_copy() using the previous vector definition
2682 (stored in OPRND). See vect_get_vec_def_for_stmt_copy() for details.
2683 The vectors are collected into VEC_OPRNDS. */
2685 static void
2686 vect_get_loop_based_defs (tree *oprnd, gimple stmt, enum vect_def_type dt,
2687 VEC (tree, heap) **vec_oprnds, int multi_step_cvt)
2689 tree vec_oprnd;
2691 /* Get first vector operand. */
2692 /* All the vector operands except the very first one (that is scalar oprnd)
2693 are stmt copies. */
2694 if (TREE_CODE (TREE_TYPE (*oprnd)) != VECTOR_TYPE)
2695 vec_oprnd = vect_get_vec_def_for_operand (*oprnd, stmt, NULL);
2696 else
2697 vec_oprnd = vect_get_vec_def_for_stmt_copy (dt, *oprnd);
2699 VEC_quick_push (tree, *vec_oprnds, vec_oprnd);
2701 /* Get second vector operand. */
2702 vec_oprnd = vect_get_vec_def_for_stmt_copy (dt, vec_oprnd);
2703 VEC_quick_push (tree, *vec_oprnds, vec_oprnd);
2705 *oprnd = vec_oprnd;
2707 /* For conversion in multiple steps, continue to get operands
2708 recursively. */
2709 if (multi_step_cvt)
2710 vect_get_loop_based_defs (oprnd, stmt, dt, vec_oprnds, multi_step_cvt - 1);
2714 /* Create vectorized demotion statements for vector operands from VEC_OPRNDS.
2715 For multi-step conversions store the resulting vectors and call the function
2716 recursively. */
2718 static void
2719 vect_create_vectorized_demotion_stmts (VEC (tree, heap) **vec_oprnds,
2720 int multi_step_cvt, gimple stmt,
2721 VEC (tree, heap) *vec_dsts,
2722 gimple_stmt_iterator *gsi,
2723 slp_tree slp_node, enum tree_code code,
2724 stmt_vec_info *prev_stmt_info)
2726 unsigned int i;
2727 tree vop0, vop1, new_tmp, vec_dest;
2728 gimple new_stmt;
2729 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
2731 vec_dest = VEC_pop (tree, vec_dsts);
2733 for (i = 0; i < VEC_length (tree, *vec_oprnds); i += 2)
2735 /* Create demotion operation. */
2736 vop0 = VEC_index (tree, *vec_oprnds, i);
2737 vop1 = VEC_index (tree, *vec_oprnds, i + 1);
2738 new_stmt = gimple_build_assign_with_ops (code, vec_dest, vop0, vop1);
2739 new_tmp = make_ssa_name (vec_dest, new_stmt);
2740 gimple_assign_set_lhs (new_stmt, new_tmp);
2741 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2743 if (multi_step_cvt)
2744 /* Store the resulting vector for next recursive call. */
2745 VEC_replace (tree, *vec_oprnds, i/2, new_tmp);
2746 else
2748 /* This is the last step of the conversion sequence. Store the
2749 vectors in SLP_NODE or in vector info of the scalar statement
2750 (or in STMT_VINFO_RELATED_STMT chain). */
2751 if (slp_node)
2752 VEC_quick_push (gimple, SLP_TREE_VEC_STMTS (slp_node), new_stmt);
2753 else
2755 if (!*prev_stmt_info)
2756 STMT_VINFO_VEC_STMT (stmt_info) = new_stmt;
2757 else
2758 STMT_VINFO_RELATED_STMT (*prev_stmt_info) = new_stmt;
2760 *prev_stmt_info = vinfo_for_stmt (new_stmt);
2765 /* For multi-step demotion operations we first generate demotion operations
2766 from the source type to the intermediate types, and then combine the
2767 results (stored in VEC_OPRNDS) in demotion operation to the destination
2768 type. */
2769 if (multi_step_cvt)
2771 /* At each level of recursion we have have of the operands we had at the
2772 previous level. */
2773 VEC_truncate (tree, *vec_oprnds, (i+1)/2);
2774 vect_create_vectorized_demotion_stmts (vec_oprnds, multi_step_cvt - 1,
2775 stmt, vec_dsts, gsi, slp_node,
2776 code, prev_stmt_info);
2781 /* Function vectorizable_type_demotion
2783 Check if STMT performs a binary or unary operation that involves
2784 type demotion, and if it can be vectorized.
2785 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
2786 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
2787 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
2789 static bool
2790 vectorizable_type_demotion (gimple stmt, gimple_stmt_iterator *gsi,
2791 gimple *vec_stmt, slp_tree slp_node)
2793 tree vec_dest;
2794 tree scalar_dest;
2795 tree op0;
2796 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
2797 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
2798 enum tree_code code, code1 = ERROR_MARK;
2799 tree def;
2800 gimple def_stmt;
2801 enum vect_def_type dt[2] = {vect_unknown_def_type, vect_unknown_def_type};
2802 stmt_vec_info prev_stmt_info;
2803 int nunits_in;
2804 int nunits_out;
2805 tree vectype_out;
2806 int ncopies;
2807 int j, i;
2808 tree vectype_in;
2809 int multi_step_cvt = 0;
2810 VEC (tree, heap) *vec_oprnds0 = NULL;
2811 VEC (tree, heap) *vec_dsts = NULL, *interm_types = NULL, *tmp_vec_dsts = NULL;
2812 tree last_oprnd, intermediate_type;
2814 /* FORNOW: not supported by basic block SLP vectorization. */
2815 gcc_assert (loop_vinfo);
2817 if (!STMT_VINFO_RELEVANT_P (stmt_info))
2818 return false;
2820 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
2821 return false;
2823 /* Is STMT a vectorizable type-demotion operation? */
2824 if (!is_gimple_assign (stmt))
2825 return false;
2827 if (TREE_CODE (gimple_assign_lhs (stmt)) != SSA_NAME)
2828 return false;
2830 code = gimple_assign_rhs_code (stmt);
2831 if (!CONVERT_EXPR_CODE_P (code))
2832 return false;
2834 scalar_dest = gimple_assign_lhs (stmt);
2835 vectype_out = STMT_VINFO_VECTYPE (stmt_info);
2837 /* Check the operands of the operation. */
2838 op0 = gimple_assign_rhs1 (stmt);
2839 if (! ((INTEGRAL_TYPE_P (TREE_TYPE (scalar_dest))
2840 && INTEGRAL_TYPE_P (TREE_TYPE (op0)))
2841 || (SCALAR_FLOAT_TYPE_P (TREE_TYPE (scalar_dest))
2842 && SCALAR_FLOAT_TYPE_P (TREE_TYPE (op0))
2843 && CONVERT_EXPR_CODE_P (code))))
2844 return false;
2845 if (!vect_is_simple_use_1 (op0, loop_vinfo, NULL,
2846 &def_stmt, &def, &dt[0], &vectype_in))
2848 if (vect_print_dump_info (REPORT_DETAILS))
2849 fprintf (vect_dump, "use not simple.");
2850 return false;
2852 /* If op0 is an external def use a vector type with the
2853 same size as the output vector type if possible. */
2854 if (!vectype_in)
2855 vectype_in = get_same_sized_vectype (TREE_TYPE (op0), vectype_out);
2856 if (vec_stmt)
2857 gcc_assert (vectype_in);
2858 if (!vectype_in)
2860 if (vect_print_dump_info (REPORT_DETAILS))
2862 fprintf (vect_dump, "no vectype for scalar type ");
2863 print_generic_expr (vect_dump, TREE_TYPE (op0), TDF_SLIM);
2866 return false;
2869 nunits_in = TYPE_VECTOR_SUBPARTS (vectype_in);
2870 nunits_out = TYPE_VECTOR_SUBPARTS (vectype_out);
2871 if (nunits_in >= nunits_out)
2872 return false;
2874 /* Multiple types in SLP are handled by creating the appropriate number of
2875 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
2876 case of SLP. */
2877 if (slp_node)
2878 ncopies = 1;
2879 else
2880 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_out;
2881 gcc_assert (ncopies >= 1);
2883 /* Supportable by target? */
2884 if (!supportable_narrowing_operation (code, vectype_out, vectype_in,
2885 &code1, &multi_step_cvt, &interm_types))
2886 return false;
2888 if (!vec_stmt) /* transformation not required. */
2890 STMT_VINFO_TYPE (stmt_info) = type_demotion_vec_info_type;
2891 if (vect_print_dump_info (REPORT_DETAILS))
2892 fprintf (vect_dump, "=== vectorizable_demotion ===");
2893 vect_model_simple_cost (stmt_info, ncopies, dt, NULL);
2894 return true;
2897 /** Transform. **/
2898 if (vect_print_dump_info (REPORT_DETAILS))
2899 fprintf (vect_dump, "transform type demotion operation. ncopies = %d.",
2900 ncopies);
2902 /* In case of multi-step demotion, we first generate demotion operations to
2903 the intermediate types, and then from that types to the final one.
2904 We create vector destinations for the intermediate type (TYPES) received
2905 from supportable_narrowing_operation, and store them in the correct order
2906 for future use in vect_create_vectorized_demotion_stmts(). */
2907 if (multi_step_cvt)
2908 vec_dsts = VEC_alloc (tree, heap, multi_step_cvt + 1);
2909 else
2910 vec_dsts = VEC_alloc (tree, heap, 1);
2912 vec_dest = vect_create_destination_var (scalar_dest, vectype_out);
2913 VEC_quick_push (tree, vec_dsts, vec_dest);
2915 if (multi_step_cvt)
2917 for (i = VEC_length (tree, interm_types) - 1;
2918 VEC_iterate (tree, interm_types, i, intermediate_type); i--)
2920 vec_dest = vect_create_destination_var (scalar_dest,
2921 intermediate_type);
2922 VEC_quick_push (tree, vec_dsts, vec_dest);
2926 /* In case the vectorization factor (VF) is bigger than the number
2927 of elements that we can fit in a vectype (nunits), we have to generate
2928 more than one vector stmt - i.e - we need to "unroll" the
2929 vector stmt by a factor VF/nunits. */
2930 last_oprnd = op0;
2931 prev_stmt_info = NULL;
2932 for (j = 0; j < ncopies; j++)
2934 /* Handle uses. */
2935 if (slp_node)
2936 vect_get_slp_defs (op0, NULL_TREE, slp_node, &vec_oprnds0, NULL, -1);
2937 else
2939 VEC_free (tree, heap, vec_oprnds0);
2940 vec_oprnds0 = VEC_alloc (tree, heap,
2941 (multi_step_cvt ? vect_pow2 (multi_step_cvt) * 2 : 2));
2942 vect_get_loop_based_defs (&last_oprnd, stmt, dt[0], &vec_oprnds0,
2943 vect_pow2 (multi_step_cvt) - 1);
2946 /* Arguments are ready. Create the new vector stmts. */
2947 tmp_vec_dsts = VEC_copy (tree, heap, vec_dsts);
2948 vect_create_vectorized_demotion_stmts (&vec_oprnds0,
2949 multi_step_cvt, stmt, tmp_vec_dsts,
2950 gsi, slp_node, code1,
2951 &prev_stmt_info);
2954 VEC_free (tree, heap, vec_oprnds0);
2955 VEC_free (tree, heap, vec_dsts);
2956 VEC_free (tree, heap, tmp_vec_dsts);
2957 VEC_free (tree, heap, interm_types);
2959 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
2960 return true;
2964 /* Create vectorized promotion statements for vector operands from VEC_OPRNDS0
2965 and VEC_OPRNDS1 (for binary operations). For multi-step conversions store
2966 the resulting vectors and call the function recursively. */
2968 static void
2969 vect_create_vectorized_promotion_stmts (VEC (tree, heap) **vec_oprnds0,
2970 VEC (tree, heap) **vec_oprnds1,
2971 int multi_step_cvt, gimple stmt,
2972 VEC (tree, heap) *vec_dsts,
2973 gimple_stmt_iterator *gsi,
2974 slp_tree slp_node, enum tree_code code1,
2975 enum tree_code code2, tree decl1,
2976 tree decl2, int op_type,
2977 stmt_vec_info *prev_stmt_info)
2979 int i;
2980 tree vop0, vop1, new_tmp1, new_tmp2, vec_dest;
2981 gimple new_stmt1, new_stmt2;
2982 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
2983 VEC (tree, heap) *vec_tmp;
2985 vec_dest = VEC_pop (tree, vec_dsts);
2986 vec_tmp = VEC_alloc (tree, heap, VEC_length (tree, *vec_oprnds0) * 2);
2988 FOR_EACH_VEC_ELT (tree, *vec_oprnds0, i, vop0)
2990 if (op_type == binary_op)
2991 vop1 = VEC_index (tree, *vec_oprnds1, i);
2992 else
2993 vop1 = NULL_TREE;
2995 /* Generate the two halves of promotion operation. */
2996 new_stmt1 = vect_gen_widened_results_half (code1, decl1, vop0, vop1,
2997 op_type, vec_dest, gsi, stmt);
2998 new_stmt2 = vect_gen_widened_results_half (code2, decl2, vop0, vop1,
2999 op_type, vec_dest, gsi, stmt);
3000 if (is_gimple_call (new_stmt1))
3002 new_tmp1 = gimple_call_lhs (new_stmt1);
3003 new_tmp2 = gimple_call_lhs (new_stmt2);
3005 else
3007 new_tmp1 = gimple_assign_lhs (new_stmt1);
3008 new_tmp2 = gimple_assign_lhs (new_stmt2);
3011 if (multi_step_cvt)
3013 /* Store the results for the recursive call. */
3014 VEC_quick_push (tree, vec_tmp, new_tmp1);
3015 VEC_quick_push (tree, vec_tmp, new_tmp2);
3017 else
3019 /* Last step of promotion sequience - store the results. */
3020 if (slp_node)
3022 VEC_quick_push (gimple, SLP_TREE_VEC_STMTS (slp_node), new_stmt1);
3023 VEC_quick_push (gimple, SLP_TREE_VEC_STMTS (slp_node), new_stmt2);
3025 else
3027 if (!*prev_stmt_info)
3028 STMT_VINFO_VEC_STMT (stmt_info) = new_stmt1;
3029 else
3030 STMT_VINFO_RELATED_STMT (*prev_stmt_info) = new_stmt1;
3032 *prev_stmt_info = vinfo_for_stmt (new_stmt1);
3033 STMT_VINFO_RELATED_STMT (*prev_stmt_info) = new_stmt2;
3034 *prev_stmt_info = vinfo_for_stmt (new_stmt2);
3039 if (multi_step_cvt)
3041 /* For multi-step promotion operation we first generate we call the
3042 function recurcively for every stage. We start from the input type,
3043 create promotion operations to the intermediate types, and then
3044 create promotions to the output type. */
3045 *vec_oprnds0 = VEC_copy (tree, heap, vec_tmp);
3046 vect_create_vectorized_promotion_stmts (vec_oprnds0, vec_oprnds1,
3047 multi_step_cvt - 1, stmt,
3048 vec_dsts, gsi, slp_node, code1,
3049 code2, decl2, decl2, op_type,
3050 prev_stmt_info);
3053 VEC_free (tree, heap, vec_tmp);
3057 /* Function vectorizable_type_promotion
3059 Check if STMT performs a binary or unary operation that involves
3060 type promotion, and if it can be vectorized.
3061 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
3062 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
3063 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
3065 static bool
3066 vectorizable_type_promotion (gimple stmt, gimple_stmt_iterator *gsi,
3067 gimple *vec_stmt, slp_tree slp_node)
3069 tree vec_dest;
3070 tree scalar_dest;
3071 tree op0, op1 = NULL;
3072 tree vec_oprnd0=NULL, vec_oprnd1=NULL;
3073 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
3074 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
3075 enum tree_code code, code1 = ERROR_MARK, code2 = ERROR_MARK;
3076 tree decl1 = NULL_TREE, decl2 = NULL_TREE;
3077 int op_type;
3078 tree def;
3079 gimple def_stmt;
3080 enum vect_def_type dt[2] = {vect_unknown_def_type, vect_unknown_def_type};
3081 stmt_vec_info prev_stmt_info;
3082 int nunits_in;
3083 int nunits_out;
3084 tree vectype_out;
3085 int ncopies;
3086 int j, i;
3087 tree vectype_in;
3088 tree intermediate_type = NULL_TREE;
3089 int multi_step_cvt = 0;
3090 VEC (tree, heap) *vec_oprnds0 = NULL, *vec_oprnds1 = NULL;
3091 VEC (tree, heap) *vec_dsts = NULL, *interm_types = NULL, *tmp_vec_dsts = NULL;
3093 /* FORNOW: not supported by basic block SLP vectorization. */
3094 gcc_assert (loop_vinfo);
3096 if (!STMT_VINFO_RELEVANT_P (stmt_info))
3097 return false;
3099 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
3100 return false;
3102 /* Is STMT a vectorizable type-promotion operation? */
3103 if (!is_gimple_assign (stmt))
3104 return false;
3106 if (TREE_CODE (gimple_assign_lhs (stmt)) != SSA_NAME)
3107 return false;
3109 code = gimple_assign_rhs_code (stmt);
3110 if (!CONVERT_EXPR_CODE_P (code)
3111 && code != WIDEN_MULT_EXPR)
3112 return false;
3114 scalar_dest = gimple_assign_lhs (stmt);
3115 vectype_out = STMT_VINFO_VECTYPE (stmt_info);
3117 /* Check the operands of the operation. */
3118 op0 = gimple_assign_rhs1 (stmt);
3119 if (! ((INTEGRAL_TYPE_P (TREE_TYPE (scalar_dest))
3120 && INTEGRAL_TYPE_P (TREE_TYPE (op0)))
3121 || (SCALAR_FLOAT_TYPE_P (TREE_TYPE (scalar_dest))
3122 && SCALAR_FLOAT_TYPE_P (TREE_TYPE (op0))
3123 && CONVERT_EXPR_CODE_P (code))))
3124 return false;
3125 if (!vect_is_simple_use_1 (op0, loop_vinfo, NULL,
3126 &def_stmt, &def, &dt[0], &vectype_in))
3128 if (vect_print_dump_info (REPORT_DETAILS))
3129 fprintf (vect_dump, "use not simple.");
3130 return false;
3132 /* If op0 is an external or constant def use a vector type with
3133 the same size as the output vector type. */
3134 if (!vectype_in)
3135 vectype_in = get_same_sized_vectype (TREE_TYPE (op0), vectype_out);
3136 if (vec_stmt)
3137 gcc_assert (vectype_in);
3138 if (!vectype_in)
3140 if (vect_print_dump_info (REPORT_DETAILS))
3142 fprintf (vect_dump, "no vectype for scalar type ");
3143 print_generic_expr (vect_dump, TREE_TYPE (op0), TDF_SLIM);
3146 return false;
3149 nunits_in = TYPE_VECTOR_SUBPARTS (vectype_in);
3150 nunits_out = TYPE_VECTOR_SUBPARTS (vectype_out);
3151 if (nunits_in <= nunits_out)
3152 return false;
3154 /* Multiple types in SLP are handled by creating the appropriate number of
3155 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
3156 case of SLP. */
3157 if (slp_node)
3158 ncopies = 1;
3159 else
3160 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_in;
3162 gcc_assert (ncopies >= 1);
3164 op_type = TREE_CODE_LENGTH (code);
3165 if (op_type == binary_op)
3167 op1 = gimple_assign_rhs2 (stmt);
3168 if (!vect_is_simple_use (op1, loop_vinfo, NULL, &def_stmt, &def, &dt[1]))
3170 if (vect_print_dump_info (REPORT_DETAILS))
3171 fprintf (vect_dump, "use not simple.");
3172 return false;
3176 /* Supportable by target? */
3177 if (!supportable_widening_operation (code, stmt, vectype_out, vectype_in,
3178 &decl1, &decl2, &code1, &code2,
3179 &multi_step_cvt, &interm_types))
3180 return false;
3182 /* Binary widening operation can only be supported directly by the
3183 architecture. */
3184 gcc_assert (!(multi_step_cvt && op_type == binary_op));
3186 if (!vec_stmt) /* transformation not required. */
3188 STMT_VINFO_TYPE (stmt_info) = type_promotion_vec_info_type;
3189 if (vect_print_dump_info (REPORT_DETAILS))
3190 fprintf (vect_dump, "=== vectorizable_promotion ===");
3191 vect_model_simple_cost (stmt_info, 2*ncopies, dt, NULL);
3192 return true;
3195 /** Transform. **/
3197 if (vect_print_dump_info (REPORT_DETAILS))
3198 fprintf (vect_dump, "transform type promotion operation. ncopies = %d.",
3199 ncopies);
3201 /* Handle def. */
3202 /* In case of multi-step promotion, we first generate promotion operations
3203 to the intermediate types, and then from that types to the final one.
3204 We store vector destination in VEC_DSTS in the correct order for
3205 recursive creation of promotion operations in
3206 vect_create_vectorized_promotion_stmts(). Vector destinations are created
3207 according to TYPES recieved from supportable_widening_operation(). */
3208 if (multi_step_cvt)
3209 vec_dsts = VEC_alloc (tree, heap, multi_step_cvt + 1);
3210 else
3211 vec_dsts = VEC_alloc (tree, heap, 1);
3213 vec_dest = vect_create_destination_var (scalar_dest, vectype_out);
3214 VEC_quick_push (tree, vec_dsts, vec_dest);
3216 if (multi_step_cvt)
3218 for (i = VEC_length (tree, interm_types) - 1;
3219 VEC_iterate (tree, interm_types, i, intermediate_type); i--)
3221 vec_dest = vect_create_destination_var (scalar_dest,
3222 intermediate_type);
3223 VEC_quick_push (tree, vec_dsts, vec_dest);
3227 if (!slp_node)
3229 vec_oprnds0 = VEC_alloc (tree, heap,
3230 (multi_step_cvt ? vect_pow2 (multi_step_cvt) : 1));
3231 if (op_type == binary_op)
3232 vec_oprnds1 = VEC_alloc (tree, heap, 1);
3235 /* In case the vectorization factor (VF) is bigger than the number
3236 of elements that we can fit in a vectype (nunits), we have to generate
3237 more than one vector stmt - i.e - we need to "unroll" the
3238 vector stmt by a factor VF/nunits. */
3240 prev_stmt_info = NULL;
3241 for (j = 0; j < ncopies; j++)
3243 /* Handle uses. */
3244 if (j == 0)
3246 if (slp_node)
3247 vect_get_slp_defs (op0, op1, slp_node, &vec_oprnds0,
3248 &vec_oprnds1, -1);
3249 else
3251 vec_oprnd0 = vect_get_vec_def_for_operand (op0, stmt, NULL);
3252 VEC_quick_push (tree, vec_oprnds0, vec_oprnd0);
3253 if (op_type == binary_op)
3255 vec_oprnd1 = vect_get_vec_def_for_operand (op1, stmt, NULL);
3256 VEC_quick_push (tree, vec_oprnds1, vec_oprnd1);
3260 else
3262 vec_oprnd0 = vect_get_vec_def_for_stmt_copy (dt[0], vec_oprnd0);
3263 VEC_replace (tree, vec_oprnds0, 0, vec_oprnd0);
3264 if (op_type == binary_op)
3266 vec_oprnd1 = vect_get_vec_def_for_stmt_copy (dt[1], vec_oprnd1);
3267 VEC_replace (tree, vec_oprnds1, 0, vec_oprnd1);
3271 /* Arguments are ready. Create the new vector stmts. */
3272 tmp_vec_dsts = VEC_copy (tree, heap, vec_dsts);
3273 vect_create_vectorized_promotion_stmts (&vec_oprnds0, &vec_oprnds1,
3274 multi_step_cvt, stmt,
3275 tmp_vec_dsts,
3276 gsi, slp_node, code1, code2,
3277 decl1, decl2, op_type,
3278 &prev_stmt_info);
3281 VEC_free (tree, heap, vec_dsts);
3282 VEC_free (tree, heap, tmp_vec_dsts);
3283 VEC_free (tree, heap, interm_types);
3284 VEC_free (tree, heap, vec_oprnds0);
3285 VEC_free (tree, heap, vec_oprnds1);
3287 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
3288 return true;
3292 /* Function vectorizable_store.
3294 Check if STMT defines a non scalar data-ref (array/pointer/structure) that
3295 can be vectorized.
3296 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
3297 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
3298 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
3300 static bool
3301 vectorizable_store (gimple stmt, gimple_stmt_iterator *gsi, gimple *vec_stmt,
3302 slp_tree slp_node)
3304 tree scalar_dest;
3305 tree data_ref;
3306 tree op;
3307 tree vec_oprnd = NULL_TREE;
3308 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
3309 struct data_reference *dr = STMT_VINFO_DATA_REF (stmt_info), *first_dr = NULL;
3310 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
3311 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
3312 struct loop *loop = NULL;
3313 enum machine_mode vec_mode;
3314 tree dummy;
3315 enum dr_alignment_support alignment_support_scheme;
3316 tree def;
3317 gimple def_stmt;
3318 enum vect_def_type dt;
3319 stmt_vec_info prev_stmt_info = NULL;
3320 tree dataref_ptr = NULL_TREE;
3321 int nunits = TYPE_VECTOR_SUBPARTS (vectype);
3322 int ncopies;
3323 int j;
3324 gimple next_stmt, first_stmt = NULL;
3325 bool strided_store = false;
3326 unsigned int group_size, i;
3327 VEC(tree,heap) *dr_chain = NULL, *oprnds = NULL, *result_chain = NULL;
3328 bool inv_p;
3329 VEC(tree,heap) *vec_oprnds = NULL;
3330 bool slp = (slp_node != NULL);
3331 unsigned int vec_num;
3332 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
3334 if (loop_vinfo)
3335 loop = LOOP_VINFO_LOOP (loop_vinfo);
3337 /* Multiple types in SLP are handled by creating the appropriate number of
3338 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
3339 case of SLP. */
3340 if (slp)
3341 ncopies = 1;
3342 else
3343 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits;
3345 gcc_assert (ncopies >= 1);
3347 /* FORNOW. This restriction should be relaxed. */
3348 if (loop && nested_in_vect_loop_p (loop, stmt) && ncopies > 1)
3350 if (vect_print_dump_info (REPORT_DETAILS))
3351 fprintf (vect_dump, "multiple types in nested loop.");
3352 return false;
3355 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
3356 return false;
3358 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
3359 return false;
3361 /* Is vectorizable store? */
3363 if (!is_gimple_assign (stmt))
3364 return false;
3366 scalar_dest = gimple_assign_lhs (stmt);
3367 if (TREE_CODE (scalar_dest) != ARRAY_REF
3368 && TREE_CODE (scalar_dest) != INDIRECT_REF
3369 && TREE_CODE (scalar_dest) != COMPONENT_REF
3370 && TREE_CODE (scalar_dest) != IMAGPART_EXPR
3371 && TREE_CODE (scalar_dest) != REALPART_EXPR
3372 && TREE_CODE (scalar_dest) != MEM_REF)
3373 return false;
3375 gcc_assert (gimple_assign_single_p (stmt));
3376 op = gimple_assign_rhs1 (stmt);
3377 if (!vect_is_simple_use (op, loop_vinfo, bb_vinfo, &def_stmt, &def, &dt))
3379 if (vect_print_dump_info (REPORT_DETAILS))
3380 fprintf (vect_dump, "use not simple.");
3381 return false;
3384 /* The scalar rhs type needs to be trivially convertible to the vector
3385 component type. This should always be the case. */
3386 if (!useless_type_conversion_p (TREE_TYPE (vectype), TREE_TYPE (op)))
3388 if (vect_print_dump_info (REPORT_DETAILS))
3389 fprintf (vect_dump, "??? operands of different types");
3390 return false;
3393 vec_mode = TYPE_MODE (vectype);
3394 /* FORNOW. In some cases can vectorize even if data-type not supported
3395 (e.g. - array initialization with 0). */
3396 if (optab_handler (mov_optab, vec_mode) == CODE_FOR_nothing)
3397 return false;
3399 if (!STMT_VINFO_DATA_REF (stmt_info))
3400 return false;
3402 if (tree_int_cst_compare (DR_STEP (dr), size_zero_node) < 0)
3404 if (vect_print_dump_info (REPORT_DETAILS))
3405 fprintf (vect_dump, "negative step for store.");
3406 return false;
3409 if (STMT_VINFO_STRIDED_ACCESS (stmt_info))
3411 strided_store = true;
3412 first_stmt = DR_GROUP_FIRST_DR (stmt_info);
3413 if (!vect_strided_store_supported (vectype)
3414 && !PURE_SLP_STMT (stmt_info) && !slp)
3415 return false;
3417 if (first_stmt == stmt)
3419 /* STMT is the leader of the group. Check the operands of all the
3420 stmts of the group. */
3421 next_stmt = DR_GROUP_NEXT_DR (stmt_info);
3422 while (next_stmt)
3424 gcc_assert (gimple_assign_single_p (next_stmt));
3425 op = gimple_assign_rhs1 (next_stmt);
3426 if (!vect_is_simple_use (op, loop_vinfo, bb_vinfo, &def_stmt,
3427 &def, &dt))
3429 if (vect_print_dump_info (REPORT_DETAILS))
3430 fprintf (vect_dump, "use not simple.");
3431 return false;
3433 next_stmt = DR_GROUP_NEXT_DR (vinfo_for_stmt (next_stmt));
3438 if (!vec_stmt) /* transformation not required. */
3440 STMT_VINFO_TYPE (stmt_info) = store_vec_info_type;
3441 vect_model_store_cost (stmt_info, ncopies, dt, NULL);
3442 return true;
3445 /** Transform. **/
3447 if (strided_store)
3449 first_dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (first_stmt));
3450 group_size = DR_GROUP_SIZE (vinfo_for_stmt (first_stmt));
3452 DR_GROUP_STORE_COUNT (vinfo_for_stmt (first_stmt))++;
3454 /* FORNOW */
3455 gcc_assert (!loop || !nested_in_vect_loop_p (loop, stmt));
3457 /* We vectorize all the stmts of the interleaving group when we
3458 reach the last stmt in the group. */
3459 if (DR_GROUP_STORE_COUNT (vinfo_for_stmt (first_stmt))
3460 < DR_GROUP_SIZE (vinfo_for_stmt (first_stmt))
3461 && !slp)
3463 *vec_stmt = NULL;
3464 return true;
3467 if (slp)
3469 strided_store = false;
3470 /* VEC_NUM is the number of vect stmts to be created for this
3471 group. */
3472 vec_num = SLP_TREE_NUMBER_OF_VEC_STMTS (slp_node);
3473 first_stmt = VEC_index (gimple, SLP_TREE_SCALAR_STMTS (slp_node), 0);
3474 first_dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (first_stmt));
3476 else
3477 /* VEC_NUM is the number of vect stmts to be created for this
3478 group. */
3479 vec_num = group_size;
3481 else
3483 first_stmt = stmt;
3484 first_dr = dr;
3485 group_size = vec_num = 1;
3488 if (vect_print_dump_info (REPORT_DETAILS))
3489 fprintf (vect_dump, "transform store. ncopies = %d",ncopies);
3491 dr_chain = VEC_alloc (tree, heap, group_size);
3492 oprnds = VEC_alloc (tree, heap, group_size);
3494 alignment_support_scheme = vect_supportable_dr_alignment (first_dr, false);
3495 gcc_assert (alignment_support_scheme);
3497 /* In case the vectorization factor (VF) is bigger than the number
3498 of elements that we can fit in a vectype (nunits), we have to generate
3499 more than one vector stmt - i.e - we need to "unroll" the
3500 vector stmt by a factor VF/nunits. For more details see documentation in
3501 vect_get_vec_def_for_copy_stmt. */
3503 /* In case of interleaving (non-unit strided access):
3505 S1: &base + 2 = x2
3506 S2: &base = x0
3507 S3: &base + 1 = x1
3508 S4: &base + 3 = x3
3510 We create vectorized stores starting from base address (the access of the
3511 first stmt in the chain (S2 in the above example), when the last store stmt
3512 of the chain (S4) is reached:
3514 VS1: &base = vx2
3515 VS2: &base + vec_size*1 = vx0
3516 VS3: &base + vec_size*2 = vx1
3517 VS4: &base + vec_size*3 = vx3
3519 Then permutation statements are generated:
3521 VS5: vx5 = VEC_INTERLEAVE_HIGH_EXPR < vx0, vx3 >
3522 VS6: vx6 = VEC_INTERLEAVE_LOW_EXPR < vx0, vx3 >
3525 And they are put in STMT_VINFO_VEC_STMT of the corresponding scalar stmts
3526 (the order of the data-refs in the output of vect_permute_store_chain
3527 corresponds to the order of scalar stmts in the interleaving chain - see
3528 the documentation of vect_permute_store_chain()).
3530 In case of both multiple types and interleaving, above vector stores and
3531 permutation stmts are created for every copy. The result vector stmts are
3532 put in STMT_VINFO_VEC_STMT for the first copy and in the corresponding
3533 STMT_VINFO_RELATED_STMT for the next copies.
3536 prev_stmt_info = NULL;
3537 for (j = 0; j < ncopies; j++)
3539 gimple new_stmt;
3540 gimple ptr_incr;
3542 if (j == 0)
3544 if (slp)
3546 /* Get vectorized arguments for SLP_NODE. */
3547 vect_get_slp_defs (NULL_TREE, NULL_TREE, slp_node, &vec_oprnds,
3548 NULL, -1);
3550 vec_oprnd = VEC_index (tree, vec_oprnds, 0);
3552 else
3554 /* For interleaved stores we collect vectorized defs for all the
3555 stores in the group in DR_CHAIN and OPRNDS. DR_CHAIN is then
3556 used as an input to vect_permute_store_chain(), and OPRNDS as
3557 an input to vect_get_vec_def_for_stmt_copy() for the next copy.
3559 If the store is not strided, GROUP_SIZE is 1, and DR_CHAIN and
3560 OPRNDS are of size 1. */
3561 next_stmt = first_stmt;
3562 for (i = 0; i < group_size; i++)
3564 /* Since gaps are not supported for interleaved stores,
3565 GROUP_SIZE is the exact number of stmts in the chain.
3566 Therefore, NEXT_STMT can't be NULL_TREE. In case that
3567 there is no interleaving, GROUP_SIZE is 1, and only one
3568 iteration of the loop will be executed. */
3569 gcc_assert (next_stmt
3570 && gimple_assign_single_p (next_stmt));
3571 op = gimple_assign_rhs1 (next_stmt);
3573 vec_oprnd = vect_get_vec_def_for_operand (op, next_stmt,
3574 NULL);
3575 VEC_quick_push(tree, dr_chain, vec_oprnd);
3576 VEC_quick_push(tree, oprnds, vec_oprnd);
3577 next_stmt = DR_GROUP_NEXT_DR (vinfo_for_stmt (next_stmt));
3581 /* We should have catched mismatched types earlier. */
3582 gcc_assert (useless_type_conversion_p (vectype,
3583 TREE_TYPE (vec_oprnd)));
3584 dataref_ptr = vect_create_data_ref_ptr (first_stmt, NULL, NULL_TREE,
3585 &dummy, &ptr_incr, false,
3586 &inv_p);
3587 gcc_assert (bb_vinfo || !inv_p);
3589 else
3591 /* For interleaved stores we created vectorized defs for all the
3592 defs stored in OPRNDS in the previous iteration (previous copy).
3593 DR_CHAIN is then used as an input to vect_permute_store_chain(),
3594 and OPRNDS as an input to vect_get_vec_def_for_stmt_copy() for the
3595 next copy.
3596 If the store is not strided, GROUP_SIZE is 1, and DR_CHAIN and
3597 OPRNDS are of size 1. */
3598 for (i = 0; i < group_size; i++)
3600 op = VEC_index (tree, oprnds, i);
3601 vect_is_simple_use (op, loop_vinfo, bb_vinfo, &def_stmt, &def,
3602 &dt);
3603 vec_oprnd = vect_get_vec_def_for_stmt_copy (dt, op);
3604 VEC_replace(tree, dr_chain, i, vec_oprnd);
3605 VEC_replace(tree, oprnds, i, vec_oprnd);
3607 dataref_ptr =
3608 bump_vector_ptr (dataref_ptr, ptr_incr, gsi, stmt, NULL_TREE);
3611 if (strided_store)
3613 result_chain = VEC_alloc (tree, heap, group_size);
3614 /* Permute. */
3615 if (!vect_permute_store_chain (dr_chain, group_size, stmt, gsi,
3616 &result_chain))
3617 return false;
3620 next_stmt = first_stmt;
3621 for (i = 0; i < vec_num; i++)
3623 struct ptr_info_def *pi;
3625 if (i > 0)
3626 /* Bump the vector pointer. */
3627 dataref_ptr = bump_vector_ptr (dataref_ptr, ptr_incr, gsi, stmt,
3628 NULL_TREE);
3630 if (slp)
3631 vec_oprnd = VEC_index (tree, vec_oprnds, i);
3632 else if (strided_store)
3633 /* For strided stores vectorized defs are interleaved in
3634 vect_permute_store_chain(). */
3635 vec_oprnd = VEC_index (tree, result_chain, i);
3637 data_ref = build2 (MEM_REF, TREE_TYPE (vec_oprnd), dataref_ptr,
3638 build_int_cst (reference_alias_ptr_type
3639 (DR_REF (first_dr)), 0));
3640 pi = get_ptr_info (dataref_ptr);
3641 pi->align = TYPE_ALIGN_UNIT (vectype);
3642 if (aligned_access_p (first_dr))
3643 pi->misalign = 0;
3644 else if (DR_MISALIGNMENT (first_dr) == -1)
3646 TREE_TYPE (data_ref)
3647 = build_aligned_type (TREE_TYPE (data_ref),
3648 TYPE_ALIGN (TREE_TYPE (vectype)));
3649 pi->align = TYPE_ALIGN_UNIT (TREE_TYPE (vectype));
3650 pi->misalign = 0;
3652 else
3654 TREE_TYPE (data_ref)
3655 = build_aligned_type (TREE_TYPE (data_ref),
3656 TYPE_ALIGN (TREE_TYPE (vectype)));
3657 pi->misalign = DR_MISALIGNMENT (first_dr);
3660 /* Arguments are ready. Create the new vector stmt. */
3661 new_stmt = gimple_build_assign (data_ref, vec_oprnd);
3662 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3663 mark_symbols_for_renaming (new_stmt);
3665 if (slp)
3666 continue;
3668 if (j == 0)
3669 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
3670 else
3671 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
3673 prev_stmt_info = vinfo_for_stmt (new_stmt);
3674 next_stmt = DR_GROUP_NEXT_DR (vinfo_for_stmt (next_stmt));
3675 if (!next_stmt)
3676 break;
3680 VEC_free (tree, heap, dr_chain);
3681 VEC_free (tree, heap, oprnds);
3682 if (result_chain)
3683 VEC_free (tree, heap, result_chain);
3684 if (vec_oprnds)
3685 VEC_free (tree, heap, vec_oprnds);
3687 return true;
3690 /* Given a vector type VECTYPE returns a builtin DECL to be used
3691 for vector permutation and stores a mask into *MASK that implements
3692 reversal of the vector elements. If that is impossible to do
3693 returns NULL (and *MASK is unchanged). */
3695 static tree
3696 perm_mask_for_reverse (tree vectype, tree *mask)
3698 tree builtin_decl;
3699 tree mask_element_type, mask_type;
3700 tree mask_vec = NULL;
3701 int i;
3702 int nunits;
3703 if (!targetm.vectorize.builtin_vec_perm)
3704 return NULL;
3706 builtin_decl = targetm.vectorize.builtin_vec_perm (vectype,
3707 &mask_element_type);
3708 if (!builtin_decl || !mask_element_type)
3709 return NULL;
3711 mask_type = get_vectype_for_scalar_type (mask_element_type);
3712 nunits = TYPE_VECTOR_SUBPARTS (vectype);
3713 if (!mask_type
3714 || TYPE_VECTOR_SUBPARTS (vectype) != TYPE_VECTOR_SUBPARTS (mask_type))
3715 return NULL;
3717 for (i = 0; i < nunits; i++)
3718 mask_vec = tree_cons (NULL, build_int_cst (mask_element_type, i), mask_vec);
3719 mask_vec = build_vector (mask_type, mask_vec);
3721 if (!targetm.vectorize.builtin_vec_perm_ok (vectype, mask_vec))
3722 return NULL;
3723 if (mask)
3724 *mask = mask_vec;
3725 return builtin_decl;
3728 /* Given a vector variable X, that was generated for the scalar LHS of
3729 STMT, generate instructions to reverse the vector elements of X,
3730 insert them a *GSI and return the permuted vector variable. */
3732 static tree
3733 reverse_vec_elements (tree x, gimple stmt, gimple_stmt_iterator *gsi)
3735 tree vectype = TREE_TYPE (x);
3736 tree mask_vec, builtin_decl;
3737 tree perm_dest, data_ref;
3738 gimple perm_stmt;
3740 builtin_decl = perm_mask_for_reverse (vectype, &mask_vec);
3742 perm_dest = vect_create_destination_var (gimple_assign_lhs (stmt), vectype);
3744 /* Generate the permute statement. */
3745 perm_stmt = gimple_build_call (builtin_decl, 3, x, x, mask_vec);
3746 if (!useless_type_conversion_p (vectype,
3747 TREE_TYPE (TREE_TYPE (builtin_decl))))
3749 tree tem = create_tmp_reg (TREE_TYPE (TREE_TYPE (builtin_decl)), NULL);
3750 tem = make_ssa_name (tem, perm_stmt);
3751 gimple_call_set_lhs (perm_stmt, tem);
3752 vect_finish_stmt_generation (stmt, perm_stmt, gsi);
3753 perm_stmt = gimple_build_assign (NULL_TREE,
3754 build1 (VIEW_CONVERT_EXPR,
3755 vectype, tem));
3757 data_ref = make_ssa_name (perm_dest, perm_stmt);
3758 gimple_set_lhs (perm_stmt, data_ref);
3759 vect_finish_stmt_generation (stmt, perm_stmt, gsi);
3761 return data_ref;
3764 /* vectorizable_load.
3766 Check if STMT reads a non scalar data-ref (array/pointer/structure) that
3767 can be vectorized.
3768 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
3769 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
3770 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
3772 static bool
3773 vectorizable_load (gimple stmt, gimple_stmt_iterator *gsi, gimple *vec_stmt,
3774 slp_tree slp_node, slp_instance slp_node_instance)
3776 tree scalar_dest;
3777 tree vec_dest = NULL;
3778 tree data_ref = NULL;
3779 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
3780 stmt_vec_info prev_stmt_info;
3781 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
3782 struct loop *loop = NULL;
3783 struct loop *containing_loop = (gimple_bb (stmt))->loop_father;
3784 bool nested_in_vect_loop = false;
3785 struct data_reference *dr = STMT_VINFO_DATA_REF (stmt_info), *first_dr;
3786 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
3787 tree new_temp;
3788 enum machine_mode mode;
3789 gimple new_stmt = NULL;
3790 tree dummy;
3791 enum dr_alignment_support alignment_support_scheme;
3792 tree dataref_ptr = NULL_TREE;
3793 gimple ptr_incr;
3794 int nunits = TYPE_VECTOR_SUBPARTS (vectype);
3795 int ncopies;
3796 int i, j, group_size;
3797 tree msq = NULL_TREE, lsq;
3798 tree offset = NULL_TREE;
3799 tree realignment_token = NULL_TREE;
3800 gimple phi = NULL;
3801 VEC(tree,heap) *dr_chain = NULL;
3802 bool strided_load = false;
3803 gimple first_stmt;
3804 tree scalar_type;
3805 bool inv_p;
3806 bool negative;
3807 bool compute_in_loop = false;
3808 struct loop *at_loop;
3809 int vec_num;
3810 bool slp = (slp_node != NULL);
3811 bool slp_perm = false;
3812 enum tree_code code;
3813 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
3814 int vf;
3816 if (loop_vinfo)
3818 loop = LOOP_VINFO_LOOP (loop_vinfo);
3819 nested_in_vect_loop = nested_in_vect_loop_p (loop, stmt);
3820 vf = LOOP_VINFO_VECT_FACTOR (loop_vinfo);
3822 else
3823 vf = 1;
3825 /* Multiple types in SLP are handled by creating the appropriate number of
3826 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
3827 case of SLP. */
3828 if (slp)
3829 ncopies = 1;
3830 else
3831 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits;
3833 gcc_assert (ncopies >= 1);
3835 /* FORNOW. This restriction should be relaxed. */
3836 if (nested_in_vect_loop && ncopies > 1)
3838 if (vect_print_dump_info (REPORT_DETAILS))
3839 fprintf (vect_dump, "multiple types in nested loop.");
3840 return false;
3843 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
3844 return false;
3846 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
3847 return false;
3849 /* Is vectorizable load? */
3850 if (!is_gimple_assign (stmt))
3851 return false;
3853 scalar_dest = gimple_assign_lhs (stmt);
3854 if (TREE_CODE (scalar_dest) != SSA_NAME)
3855 return false;
3857 code = gimple_assign_rhs_code (stmt);
3858 if (code != ARRAY_REF
3859 && code != INDIRECT_REF
3860 && code != COMPONENT_REF
3861 && code != IMAGPART_EXPR
3862 && code != REALPART_EXPR
3863 && code != MEM_REF)
3864 return false;
3866 if (!STMT_VINFO_DATA_REF (stmt_info))
3867 return false;
3869 negative = tree_int_cst_compare (DR_STEP (dr), size_zero_node) < 0;
3870 if (negative && ncopies > 1)
3872 if (vect_print_dump_info (REPORT_DETAILS))
3873 fprintf (vect_dump, "multiple types with negative step.");
3874 return false;
3877 scalar_type = TREE_TYPE (DR_REF (dr));
3878 mode = TYPE_MODE (vectype);
3880 /* FORNOW. In some cases can vectorize even if data-type not supported
3881 (e.g. - data copies). */
3882 if (optab_handler (mov_optab, mode) == CODE_FOR_nothing)
3884 if (vect_print_dump_info (REPORT_DETAILS))
3885 fprintf (vect_dump, "Aligned load, but unsupported type.");
3886 return false;
3889 /* The vector component type needs to be trivially convertible to the
3890 scalar lhs. This should always be the case. */
3891 if (!useless_type_conversion_p (TREE_TYPE (scalar_dest), TREE_TYPE (vectype)))
3893 if (vect_print_dump_info (REPORT_DETAILS))
3894 fprintf (vect_dump, "??? operands of different types");
3895 return false;
3898 /* Check if the load is a part of an interleaving chain. */
3899 if (STMT_VINFO_STRIDED_ACCESS (stmt_info))
3901 strided_load = true;
3902 /* FORNOW */
3903 gcc_assert (! nested_in_vect_loop);
3905 /* Check if interleaving is supported. */
3906 if (!vect_strided_load_supported (vectype)
3907 && !PURE_SLP_STMT (stmt_info) && !slp)
3908 return false;
3911 if (negative)
3913 gcc_assert (!strided_load);
3914 alignment_support_scheme = vect_supportable_dr_alignment (dr, false);
3915 if (alignment_support_scheme != dr_aligned
3916 && alignment_support_scheme != dr_unaligned_supported)
3918 if (vect_print_dump_info (REPORT_DETAILS))
3919 fprintf (vect_dump, "negative step but alignment required.");
3920 return false;
3922 if (!perm_mask_for_reverse (vectype, NULL))
3924 if (vect_print_dump_info (REPORT_DETAILS))
3925 fprintf (vect_dump, "negative step and reversing not supported.");
3926 return false;
3930 if (!vec_stmt) /* transformation not required. */
3932 STMT_VINFO_TYPE (stmt_info) = load_vec_info_type;
3933 vect_model_load_cost (stmt_info, ncopies, NULL);
3934 return true;
3937 if (vect_print_dump_info (REPORT_DETAILS))
3938 fprintf (vect_dump, "transform load.");
3940 /** Transform. **/
3942 if (strided_load)
3944 first_stmt = DR_GROUP_FIRST_DR (stmt_info);
3945 /* Check if the chain of loads is already vectorized. */
3946 if (STMT_VINFO_VEC_STMT (vinfo_for_stmt (first_stmt)))
3948 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
3949 return true;
3951 first_dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (first_stmt));
3952 group_size = DR_GROUP_SIZE (vinfo_for_stmt (first_stmt));
3954 /* VEC_NUM is the number of vect stmts to be created for this group. */
3955 if (slp)
3957 strided_load = false;
3958 vec_num = SLP_TREE_NUMBER_OF_VEC_STMTS (slp_node);
3959 if (SLP_INSTANCE_LOAD_PERMUTATION (slp_node_instance))
3960 slp_perm = true;
3962 else
3963 vec_num = group_size;
3965 dr_chain = VEC_alloc (tree, heap, vec_num);
3967 else
3969 first_stmt = stmt;
3970 first_dr = dr;
3971 group_size = vec_num = 1;
3974 alignment_support_scheme = vect_supportable_dr_alignment (first_dr, false);
3975 gcc_assert (alignment_support_scheme);
3977 /* In case the vectorization factor (VF) is bigger than the number
3978 of elements that we can fit in a vectype (nunits), we have to generate
3979 more than one vector stmt - i.e - we need to "unroll" the
3980 vector stmt by a factor VF/nunits. In doing so, we record a pointer
3981 from one copy of the vector stmt to the next, in the field
3982 STMT_VINFO_RELATED_STMT. This is necessary in order to allow following
3983 stages to find the correct vector defs to be used when vectorizing
3984 stmts that use the defs of the current stmt. The example below
3985 illustrates the vectorization process when VF=16 and nunits=4 (i.e., we
3986 need to create 4 vectorized stmts):
3988 before vectorization:
3989 RELATED_STMT VEC_STMT
3990 S1: x = memref - -
3991 S2: z = x + 1 - -
3993 step 1: vectorize stmt S1:
3994 We first create the vector stmt VS1_0, and, as usual, record a
3995 pointer to it in the STMT_VINFO_VEC_STMT of the scalar stmt S1.
3996 Next, we create the vector stmt VS1_1, and record a pointer to
3997 it in the STMT_VINFO_RELATED_STMT of the vector stmt VS1_0.
3998 Similarly, for VS1_2 and VS1_3. This is the resulting chain of
3999 stmts and pointers:
4000 RELATED_STMT VEC_STMT
4001 VS1_0: vx0 = memref0 VS1_1 -
4002 VS1_1: vx1 = memref1 VS1_2 -
4003 VS1_2: vx2 = memref2 VS1_3 -
4004 VS1_3: vx3 = memref3 - -
4005 S1: x = load - VS1_0
4006 S2: z = x + 1 - -
4008 See in documentation in vect_get_vec_def_for_stmt_copy for how the
4009 information we recorded in RELATED_STMT field is used to vectorize
4010 stmt S2. */
4012 /* In case of interleaving (non-unit strided access):
4014 S1: x2 = &base + 2
4015 S2: x0 = &base
4016 S3: x1 = &base + 1
4017 S4: x3 = &base + 3
4019 Vectorized loads are created in the order of memory accesses
4020 starting from the access of the first stmt of the chain:
4022 VS1: vx0 = &base
4023 VS2: vx1 = &base + vec_size*1
4024 VS3: vx3 = &base + vec_size*2
4025 VS4: vx4 = &base + vec_size*3
4027 Then permutation statements are generated:
4029 VS5: vx5 = VEC_EXTRACT_EVEN_EXPR < vx0, vx1 >
4030 VS6: vx6 = VEC_EXTRACT_ODD_EXPR < vx0, vx1 >
4033 And they are put in STMT_VINFO_VEC_STMT of the corresponding scalar stmts
4034 (the order of the data-refs in the output of vect_permute_load_chain
4035 corresponds to the order of scalar stmts in the interleaving chain - see
4036 the documentation of vect_permute_load_chain()).
4037 The generation of permutation stmts and recording them in
4038 STMT_VINFO_VEC_STMT is done in vect_transform_strided_load().
4040 In case of both multiple types and interleaving, the vector loads and
4041 permutation stmts above are created for every copy. The result vector
4042 stmts are put in STMT_VINFO_VEC_STMT for the first copy and in the
4043 corresponding STMT_VINFO_RELATED_STMT for the next copies. */
4045 /* If the data reference is aligned (dr_aligned) or potentially unaligned
4046 on a target that supports unaligned accesses (dr_unaligned_supported)
4047 we generate the following code:
4048 p = initial_addr;
4049 indx = 0;
4050 loop {
4051 p = p + indx * vectype_size;
4052 vec_dest = *(p);
4053 indx = indx + 1;
4056 Otherwise, the data reference is potentially unaligned on a target that
4057 does not support unaligned accesses (dr_explicit_realign_optimized) -
4058 then generate the following code, in which the data in each iteration is
4059 obtained by two vector loads, one from the previous iteration, and one
4060 from the current iteration:
4061 p1 = initial_addr;
4062 msq_init = *(floor(p1))
4063 p2 = initial_addr + VS - 1;
4064 realignment_token = call target_builtin;
4065 indx = 0;
4066 loop {
4067 p2 = p2 + indx * vectype_size
4068 lsq = *(floor(p2))
4069 vec_dest = realign_load (msq, lsq, realignment_token)
4070 indx = indx + 1;
4071 msq = lsq;
4072 } */
4074 /* If the misalignment remains the same throughout the execution of the
4075 loop, we can create the init_addr and permutation mask at the loop
4076 preheader. Otherwise, it needs to be created inside the loop.
4077 This can only occur when vectorizing memory accesses in the inner-loop
4078 nested within an outer-loop that is being vectorized. */
4080 if (loop && nested_in_vect_loop_p (loop, stmt)
4081 && (TREE_INT_CST_LOW (DR_STEP (dr))
4082 % GET_MODE_SIZE (TYPE_MODE (vectype)) != 0))
4084 gcc_assert (alignment_support_scheme != dr_explicit_realign_optimized);
4085 compute_in_loop = true;
4088 if ((alignment_support_scheme == dr_explicit_realign_optimized
4089 || alignment_support_scheme == dr_explicit_realign)
4090 && !compute_in_loop)
4092 msq = vect_setup_realignment (first_stmt, gsi, &realignment_token,
4093 alignment_support_scheme, NULL_TREE,
4094 &at_loop);
4095 if (alignment_support_scheme == dr_explicit_realign_optimized)
4097 phi = SSA_NAME_DEF_STMT (msq);
4098 offset = size_int (TYPE_VECTOR_SUBPARTS (vectype) - 1);
4101 else
4102 at_loop = loop;
4104 if (negative)
4105 offset = size_int (-TYPE_VECTOR_SUBPARTS (vectype) + 1);
4107 prev_stmt_info = NULL;
4108 for (j = 0; j < ncopies; j++)
4110 /* 1. Create the vector pointer update chain. */
4111 if (j == 0)
4112 dataref_ptr = vect_create_data_ref_ptr (first_stmt,
4113 at_loop, offset,
4114 &dummy, &ptr_incr, false,
4115 &inv_p);
4116 else
4117 dataref_ptr =
4118 bump_vector_ptr (dataref_ptr, ptr_incr, gsi, stmt, NULL_TREE);
4120 for (i = 0; i < vec_num; i++)
4122 if (i > 0)
4123 dataref_ptr = bump_vector_ptr (dataref_ptr, ptr_incr, gsi, stmt,
4124 NULL_TREE);
4126 /* 2. Create the vector-load in the loop. */
4127 switch (alignment_support_scheme)
4129 case dr_aligned:
4130 case dr_unaligned_supported:
4132 struct ptr_info_def *pi;
4133 data_ref
4134 = build2 (MEM_REF, vectype, dataref_ptr,
4135 build_int_cst (reference_alias_ptr_type
4136 (DR_REF (first_dr)), 0));
4137 pi = get_ptr_info (dataref_ptr);
4138 pi->align = TYPE_ALIGN_UNIT (vectype);
4139 if (alignment_support_scheme == dr_aligned)
4141 gcc_assert (aligned_access_p (first_dr));
4142 pi->misalign = 0;
4144 else if (DR_MISALIGNMENT (first_dr) == -1)
4146 TREE_TYPE (data_ref)
4147 = build_aligned_type (TREE_TYPE (data_ref),
4148 TYPE_ALIGN (TREE_TYPE (vectype)));
4149 pi->align = TYPE_ALIGN_UNIT (TREE_TYPE (vectype));
4150 pi->misalign = 0;
4152 else
4154 TREE_TYPE (data_ref)
4155 = build_aligned_type (TREE_TYPE (data_ref),
4156 TYPE_ALIGN (TREE_TYPE (vectype)));
4157 pi->misalign = DR_MISALIGNMENT (first_dr);
4159 break;
4161 case dr_explicit_realign:
4163 tree ptr, bump;
4164 tree vs_minus_1 = size_int (TYPE_VECTOR_SUBPARTS (vectype) - 1);
4166 if (compute_in_loop)
4167 msq = vect_setup_realignment (first_stmt, gsi,
4168 &realignment_token,
4169 dr_explicit_realign,
4170 dataref_ptr, NULL);
4172 new_stmt = gimple_build_assign_with_ops
4173 (BIT_AND_EXPR, NULL_TREE, dataref_ptr,
4174 build_int_cst
4175 (TREE_TYPE (dataref_ptr),
4176 -(HOST_WIDE_INT)TYPE_ALIGN_UNIT (vectype)));
4177 ptr = make_ssa_name (SSA_NAME_VAR (dataref_ptr), new_stmt);
4178 gimple_assign_set_lhs (new_stmt, ptr);
4179 vect_finish_stmt_generation (stmt, new_stmt, gsi);
4180 data_ref
4181 = build2 (MEM_REF, vectype, ptr,
4182 build_int_cst (reference_alias_ptr_type
4183 (DR_REF (first_dr)), 0));
4184 vec_dest = vect_create_destination_var (scalar_dest, vectype);
4185 new_stmt = gimple_build_assign (vec_dest, data_ref);
4186 new_temp = make_ssa_name (vec_dest, new_stmt);
4187 gimple_assign_set_lhs (new_stmt, new_temp);
4188 gimple_set_vdef (new_stmt, gimple_vdef (stmt));
4189 gimple_set_vuse (new_stmt, gimple_vuse (stmt));
4190 vect_finish_stmt_generation (stmt, new_stmt, gsi);
4191 msq = new_temp;
4193 bump = size_binop (MULT_EXPR, vs_minus_1,
4194 TYPE_SIZE_UNIT (scalar_type));
4195 ptr = bump_vector_ptr (dataref_ptr, NULL, gsi, stmt, bump);
4196 new_stmt = gimple_build_assign_with_ops
4197 (BIT_AND_EXPR, NULL_TREE, ptr,
4198 build_int_cst
4199 (TREE_TYPE (ptr),
4200 -(HOST_WIDE_INT)TYPE_ALIGN_UNIT (vectype)));
4201 ptr = make_ssa_name (SSA_NAME_VAR (dataref_ptr), new_stmt);
4202 gimple_assign_set_lhs (new_stmt, ptr);
4203 vect_finish_stmt_generation (stmt, new_stmt, gsi);
4204 data_ref
4205 = build2 (MEM_REF, vectype, ptr,
4206 build_int_cst (reference_alias_ptr_type
4207 (DR_REF (first_dr)), 0));
4208 break;
4210 case dr_explicit_realign_optimized:
4211 new_stmt = gimple_build_assign_with_ops
4212 (BIT_AND_EXPR, NULL_TREE, dataref_ptr,
4213 build_int_cst
4214 (TREE_TYPE (dataref_ptr),
4215 -(HOST_WIDE_INT)TYPE_ALIGN_UNIT (vectype)));
4216 new_temp = make_ssa_name (SSA_NAME_VAR (dataref_ptr), new_stmt);
4217 gimple_assign_set_lhs (new_stmt, new_temp);
4218 vect_finish_stmt_generation (stmt, new_stmt, gsi);
4219 data_ref
4220 = build2 (MEM_REF, vectype, new_temp,
4221 build_int_cst (reference_alias_ptr_type
4222 (DR_REF (first_dr)), 0));
4223 break;
4224 default:
4225 gcc_unreachable ();
4227 vec_dest = vect_create_destination_var (scalar_dest, vectype);
4228 new_stmt = gimple_build_assign (vec_dest, data_ref);
4229 new_temp = make_ssa_name (vec_dest, new_stmt);
4230 gimple_assign_set_lhs (new_stmt, new_temp);
4231 vect_finish_stmt_generation (stmt, new_stmt, gsi);
4232 mark_symbols_for_renaming (new_stmt);
4234 /* 3. Handle explicit realignment if necessary/supported. Create in
4235 loop: vec_dest = realign_load (msq, lsq, realignment_token) */
4236 if (alignment_support_scheme == dr_explicit_realign_optimized
4237 || alignment_support_scheme == dr_explicit_realign)
4239 tree tmp;
4241 lsq = gimple_assign_lhs (new_stmt);
4242 if (!realignment_token)
4243 realignment_token = dataref_ptr;
4244 vec_dest = vect_create_destination_var (scalar_dest, vectype);
4245 tmp = build3 (REALIGN_LOAD_EXPR, vectype, msq, lsq,
4246 realignment_token);
4247 new_stmt = gimple_build_assign (vec_dest, tmp);
4248 new_temp = make_ssa_name (vec_dest, new_stmt);
4249 gimple_assign_set_lhs (new_stmt, new_temp);
4250 vect_finish_stmt_generation (stmt, new_stmt, gsi);
4252 if (alignment_support_scheme == dr_explicit_realign_optimized)
4254 gcc_assert (phi);
4255 if (i == vec_num - 1 && j == ncopies - 1)
4256 add_phi_arg (phi, lsq, loop_latch_edge (containing_loop),
4257 UNKNOWN_LOCATION);
4258 msq = lsq;
4262 /* 4. Handle invariant-load. */
4263 if (inv_p && !bb_vinfo)
4265 gcc_assert (!strided_load);
4266 gcc_assert (nested_in_vect_loop_p (loop, stmt));
4267 if (j == 0)
4269 int k;
4270 tree t = NULL_TREE;
4271 tree vec_inv, bitpos, bitsize = TYPE_SIZE (scalar_type);
4273 /* CHECKME: bitpos depends on endianess? */
4274 bitpos = bitsize_zero_node;
4275 vec_inv = build3 (BIT_FIELD_REF, scalar_type, new_temp,
4276 bitsize, bitpos);
4277 vec_dest =
4278 vect_create_destination_var (scalar_dest, NULL_TREE);
4279 new_stmt = gimple_build_assign (vec_dest, vec_inv);
4280 new_temp = make_ssa_name (vec_dest, new_stmt);
4281 gimple_assign_set_lhs (new_stmt, new_temp);
4282 vect_finish_stmt_generation (stmt, new_stmt, gsi);
4284 for (k = nunits - 1; k >= 0; --k)
4285 t = tree_cons (NULL_TREE, new_temp, t);
4286 /* FIXME: use build_constructor directly. */
4287 vec_inv = build_constructor_from_list (vectype, t);
4288 new_temp = vect_init_vector (stmt, vec_inv, vectype, gsi);
4289 new_stmt = SSA_NAME_DEF_STMT (new_temp);
4291 else
4292 gcc_unreachable (); /* FORNOW. */
4295 if (negative)
4297 new_temp = reverse_vec_elements (new_temp, stmt, gsi);
4298 new_stmt = SSA_NAME_DEF_STMT (new_temp);
4301 /* Collect vector loads and later create their permutation in
4302 vect_transform_strided_load (). */
4303 if (strided_load || slp_perm)
4304 VEC_quick_push (tree, dr_chain, new_temp);
4306 /* Store vector loads in the corresponding SLP_NODE. */
4307 if (slp && !slp_perm)
4308 VEC_quick_push (gimple, SLP_TREE_VEC_STMTS (slp_node), new_stmt);
4311 if (slp && !slp_perm)
4312 continue;
4314 if (slp_perm)
4316 if (!vect_transform_slp_perm_load (stmt, dr_chain, gsi, vf,
4317 slp_node_instance, false))
4319 VEC_free (tree, heap, dr_chain);
4320 return false;
4323 else
4325 if (strided_load)
4327 if (!vect_transform_strided_load (stmt, dr_chain, group_size, gsi))
4328 return false;
4330 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
4331 VEC_free (tree, heap, dr_chain);
4332 dr_chain = VEC_alloc (tree, heap, group_size);
4334 else
4336 if (j == 0)
4337 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
4338 else
4339 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
4340 prev_stmt_info = vinfo_for_stmt (new_stmt);
4345 if (dr_chain)
4346 VEC_free (tree, heap, dr_chain);
4348 return true;
4351 /* Function vect_is_simple_cond.
4353 Input:
4354 LOOP - the loop that is being vectorized.
4355 COND - Condition that is checked for simple use.
4357 Returns whether a COND can be vectorized. Checks whether
4358 condition operands are supportable using vec_is_simple_use. */
4360 static bool
4361 vect_is_simple_cond (tree cond, loop_vec_info loop_vinfo)
4363 tree lhs, rhs;
4364 tree def;
4365 enum vect_def_type dt;
4367 if (!COMPARISON_CLASS_P (cond))
4368 return false;
4370 lhs = TREE_OPERAND (cond, 0);
4371 rhs = TREE_OPERAND (cond, 1);
4373 if (TREE_CODE (lhs) == SSA_NAME)
4375 gimple lhs_def_stmt = SSA_NAME_DEF_STMT (lhs);
4376 if (!vect_is_simple_use (lhs, loop_vinfo, NULL, &lhs_def_stmt, &def,
4377 &dt))
4378 return false;
4380 else if (TREE_CODE (lhs) != INTEGER_CST && TREE_CODE (lhs) != REAL_CST
4381 && TREE_CODE (lhs) != FIXED_CST)
4382 return false;
4384 if (TREE_CODE (rhs) == SSA_NAME)
4386 gimple rhs_def_stmt = SSA_NAME_DEF_STMT (rhs);
4387 if (!vect_is_simple_use (rhs, loop_vinfo, NULL, &rhs_def_stmt, &def,
4388 &dt))
4389 return false;
4391 else if (TREE_CODE (rhs) != INTEGER_CST && TREE_CODE (rhs) != REAL_CST
4392 && TREE_CODE (rhs) != FIXED_CST)
4393 return false;
4395 return true;
4398 /* vectorizable_condition.
4400 Check if STMT is conditional modify expression that can be vectorized.
4401 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
4402 stmt using VEC_COND_EXPR to replace it, put it in VEC_STMT, and insert it
4403 at GSI.
4405 When STMT is vectorized as nested cycle, REDUC_DEF is the vector variable
4406 to be used at REDUC_INDEX (in then clause if REDUC_INDEX is 1, and in
4407 else caluse if it is 2).
4409 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
4411 bool
4412 vectorizable_condition (gimple stmt, gimple_stmt_iterator *gsi,
4413 gimple *vec_stmt, tree reduc_def, int reduc_index)
4415 tree scalar_dest = NULL_TREE;
4416 tree vec_dest = NULL_TREE;
4417 tree op = NULL_TREE;
4418 tree cond_expr, then_clause, else_clause;
4419 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
4420 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
4421 tree vec_cond_lhs = NULL_TREE, vec_cond_rhs = NULL_TREE;
4422 tree vec_then_clause = NULL_TREE, vec_else_clause = NULL_TREE;
4423 tree vec_compare, vec_cond_expr;
4424 tree new_temp;
4425 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
4426 enum machine_mode vec_mode;
4427 tree def;
4428 enum vect_def_type dt, dts[4];
4429 int nunits = TYPE_VECTOR_SUBPARTS (vectype);
4430 int ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits;
4431 enum tree_code code;
4432 stmt_vec_info prev_stmt_info = NULL;
4433 int j;
4435 /* FORNOW: unsupported in basic block SLP. */
4436 gcc_assert (loop_vinfo);
4438 gcc_assert (ncopies >= 1);
4439 if (reduc_index && ncopies > 1)
4440 return false; /* FORNOW */
4442 if (!STMT_VINFO_RELEVANT_P (stmt_info))
4443 return false;
4445 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def
4446 && !(STMT_VINFO_DEF_TYPE (stmt_info) == vect_nested_cycle
4447 && reduc_def))
4448 return false;
4450 /* FORNOW: SLP not supported. */
4451 if (STMT_SLP_TYPE (stmt_info))
4452 return false;
4454 /* FORNOW: not yet supported. */
4455 if (STMT_VINFO_LIVE_P (stmt_info))
4457 if (vect_print_dump_info (REPORT_DETAILS))
4458 fprintf (vect_dump, "value used after loop.");
4459 return false;
4462 /* Is vectorizable conditional operation? */
4463 if (!is_gimple_assign (stmt))
4464 return false;
4466 code = gimple_assign_rhs_code (stmt);
4468 if (code != COND_EXPR)
4469 return false;
4471 gcc_assert (gimple_assign_single_p (stmt));
4472 op = gimple_assign_rhs1 (stmt);
4473 cond_expr = TREE_OPERAND (op, 0);
4474 then_clause = TREE_OPERAND (op, 1);
4475 else_clause = TREE_OPERAND (op, 2);
4477 if (!vect_is_simple_cond (cond_expr, loop_vinfo))
4478 return false;
4480 /* We do not handle two different vector types for the condition
4481 and the values. */
4482 if (!types_compatible_p (TREE_TYPE (TREE_OPERAND (cond_expr, 0)),
4483 TREE_TYPE (vectype)))
4484 return false;
4486 if (TREE_CODE (then_clause) == SSA_NAME)
4488 gimple then_def_stmt = SSA_NAME_DEF_STMT (then_clause);
4489 if (!vect_is_simple_use (then_clause, loop_vinfo, NULL,
4490 &then_def_stmt, &def, &dt))
4491 return false;
4493 else if (TREE_CODE (then_clause) != INTEGER_CST
4494 && TREE_CODE (then_clause) != REAL_CST
4495 && TREE_CODE (then_clause) != FIXED_CST)
4496 return false;
4498 if (TREE_CODE (else_clause) == SSA_NAME)
4500 gimple else_def_stmt = SSA_NAME_DEF_STMT (else_clause);
4501 if (!vect_is_simple_use (else_clause, loop_vinfo, NULL,
4502 &else_def_stmt, &def, &dt))
4503 return false;
4505 else if (TREE_CODE (else_clause) != INTEGER_CST
4506 && TREE_CODE (else_clause) != REAL_CST
4507 && TREE_CODE (else_clause) != FIXED_CST)
4508 return false;
4511 vec_mode = TYPE_MODE (vectype);
4513 if (!vec_stmt)
4515 STMT_VINFO_TYPE (stmt_info) = condition_vec_info_type;
4516 return expand_vec_cond_expr_p (TREE_TYPE (op), vec_mode);
4519 /* Transform */
4521 /* Handle def. */
4522 scalar_dest = gimple_assign_lhs (stmt);
4523 vec_dest = vect_create_destination_var (scalar_dest, vectype);
4525 /* Handle cond expr. */
4526 for (j = 0; j < ncopies; j++)
4528 gimple new_stmt;
4529 if (j == 0)
4531 gimple gtemp;
4532 vec_cond_lhs =
4533 vect_get_vec_def_for_operand (TREE_OPERAND (cond_expr, 0),
4534 stmt, NULL);
4535 vect_is_simple_use (TREE_OPERAND (cond_expr, 0), loop_vinfo,
4536 NULL, &gtemp, &def, &dts[0]);
4537 vec_cond_rhs =
4538 vect_get_vec_def_for_operand (TREE_OPERAND (cond_expr, 1),
4539 stmt, NULL);
4540 vect_is_simple_use (TREE_OPERAND (cond_expr, 1), loop_vinfo,
4541 NULL, &gtemp, &def, &dts[1]);
4542 if (reduc_index == 1)
4543 vec_then_clause = reduc_def;
4544 else
4546 vec_then_clause = vect_get_vec_def_for_operand (then_clause,
4547 stmt, NULL);
4548 vect_is_simple_use (then_clause, loop_vinfo,
4549 NULL, &gtemp, &def, &dts[2]);
4551 if (reduc_index == 2)
4552 vec_else_clause = reduc_def;
4553 else
4555 vec_else_clause = vect_get_vec_def_for_operand (else_clause,
4556 stmt, NULL);
4557 vect_is_simple_use (else_clause, loop_vinfo,
4558 NULL, &gtemp, &def, &dts[3]);
4561 else
4563 vec_cond_lhs = vect_get_vec_def_for_stmt_copy (dts[0], vec_cond_lhs);
4564 vec_cond_rhs = vect_get_vec_def_for_stmt_copy (dts[1], vec_cond_rhs);
4565 vec_then_clause = vect_get_vec_def_for_stmt_copy (dts[2],
4566 vec_then_clause);
4567 vec_else_clause = vect_get_vec_def_for_stmt_copy (dts[3],
4568 vec_else_clause);
4571 /* Arguments are ready. Create the new vector stmt. */
4572 vec_compare = build2 (TREE_CODE (cond_expr), vectype,
4573 vec_cond_lhs, vec_cond_rhs);
4574 vec_cond_expr = build3 (VEC_COND_EXPR, vectype,
4575 vec_compare, vec_then_clause, vec_else_clause);
4577 new_stmt = gimple_build_assign (vec_dest, vec_cond_expr);
4578 new_temp = make_ssa_name (vec_dest, new_stmt);
4579 gimple_assign_set_lhs (new_stmt, new_temp);
4580 vect_finish_stmt_generation (stmt, new_stmt, gsi);
4581 if (j == 0)
4582 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
4583 else
4584 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
4586 prev_stmt_info = vinfo_for_stmt (new_stmt);
4589 return true;
4593 /* Make sure the statement is vectorizable. */
4595 bool
4596 vect_analyze_stmt (gimple stmt, bool *need_to_vectorize, slp_tree node)
4598 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
4599 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
4600 enum vect_relevant relevance = STMT_VINFO_RELEVANT (stmt_info);
4601 bool ok;
4602 tree scalar_type, vectype;
4604 if (vect_print_dump_info (REPORT_DETAILS))
4606 fprintf (vect_dump, "==> examining statement: ");
4607 print_gimple_stmt (vect_dump, stmt, 0, TDF_SLIM);
4610 if (gimple_has_volatile_ops (stmt))
4612 if (vect_print_dump_info (REPORT_UNVECTORIZED_LOCATIONS))
4613 fprintf (vect_dump, "not vectorized: stmt has volatile operands");
4615 return false;
4618 /* Skip stmts that do not need to be vectorized. In loops this is expected
4619 to include:
4620 - the COND_EXPR which is the loop exit condition
4621 - any LABEL_EXPRs in the loop
4622 - computations that are used only for array indexing or loop control.
4623 In basic blocks we only analyze statements that are a part of some SLP
4624 instance, therefore, all the statements are relevant. */
4626 if (!STMT_VINFO_RELEVANT_P (stmt_info)
4627 && !STMT_VINFO_LIVE_P (stmt_info))
4629 if (vect_print_dump_info (REPORT_DETAILS))
4630 fprintf (vect_dump, "irrelevant.");
4632 return true;
4635 switch (STMT_VINFO_DEF_TYPE (stmt_info))
4637 case vect_internal_def:
4638 break;
4640 case vect_reduction_def:
4641 case vect_nested_cycle:
4642 gcc_assert (!bb_vinfo && (relevance == vect_used_in_outer
4643 || relevance == vect_used_in_outer_by_reduction
4644 || relevance == vect_unused_in_scope));
4645 break;
4647 case vect_induction_def:
4648 case vect_constant_def:
4649 case vect_external_def:
4650 case vect_unknown_def_type:
4651 default:
4652 gcc_unreachable ();
4655 if (bb_vinfo)
4657 gcc_assert (PURE_SLP_STMT (stmt_info));
4659 scalar_type = TREE_TYPE (gimple_get_lhs (stmt));
4660 if (vect_print_dump_info (REPORT_DETAILS))
4662 fprintf (vect_dump, "get vectype for scalar type: ");
4663 print_generic_expr (vect_dump, scalar_type, TDF_SLIM);
4666 vectype = get_vectype_for_scalar_type (scalar_type);
4667 if (!vectype)
4669 if (vect_print_dump_info (REPORT_DETAILS))
4671 fprintf (vect_dump, "not SLPed: unsupported data-type ");
4672 print_generic_expr (vect_dump, scalar_type, TDF_SLIM);
4674 return false;
4677 if (vect_print_dump_info (REPORT_DETAILS))
4679 fprintf (vect_dump, "vectype: ");
4680 print_generic_expr (vect_dump, vectype, TDF_SLIM);
4683 STMT_VINFO_VECTYPE (stmt_info) = vectype;
4686 if (STMT_VINFO_RELEVANT_P (stmt_info))
4688 gcc_assert (!VECTOR_MODE_P (TYPE_MODE (gimple_expr_type (stmt))));
4689 gcc_assert (STMT_VINFO_VECTYPE (stmt_info));
4690 *need_to_vectorize = true;
4693 ok = true;
4694 if (!bb_vinfo
4695 && (STMT_VINFO_RELEVANT_P (stmt_info)
4696 || STMT_VINFO_DEF_TYPE (stmt_info) == vect_reduction_def))
4697 ok = (vectorizable_type_promotion (stmt, NULL, NULL, NULL)
4698 || vectorizable_type_demotion (stmt, NULL, NULL, NULL)
4699 || vectorizable_conversion (stmt, NULL, NULL, NULL)
4700 || vectorizable_shift (stmt, NULL, NULL, NULL)
4701 || vectorizable_operation (stmt, NULL, NULL, NULL)
4702 || vectorizable_assignment (stmt, NULL, NULL, NULL)
4703 || vectorizable_load (stmt, NULL, NULL, NULL, NULL)
4704 || vectorizable_call (stmt, NULL, NULL)
4705 || vectorizable_store (stmt, NULL, NULL, NULL)
4706 || vectorizable_reduction (stmt, NULL, NULL, NULL)
4707 || vectorizable_condition (stmt, NULL, NULL, NULL, 0));
4708 else
4710 if (bb_vinfo)
4711 ok = (vectorizable_shift (stmt, NULL, NULL, node)
4712 || vectorizable_operation (stmt, NULL, NULL, node)
4713 || vectorizable_assignment (stmt, NULL, NULL, node)
4714 || vectorizable_load (stmt, NULL, NULL, node, NULL)
4715 || vectorizable_store (stmt, NULL, NULL, node));
4718 if (!ok)
4720 if (vect_print_dump_info (REPORT_UNVECTORIZED_LOCATIONS))
4722 fprintf (vect_dump, "not vectorized: relevant stmt not ");
4723 fprintf (vect_dump, "supported: ");
4724 print_gimple_stmt (vect_dump, stmt, 0, TDF_SLIM);
4727 return false;
4730 if (bb_vinfo)
4731 return true;
4733 /* Stmts that are (also) "live" (i.e. - that are used out of the loop)
4734 need extra handling, except for vectorizable reductions. */
4735 if (STMT_VINFO_LIVE_P (stmt_info)
4736 && STMT_VINFO_TYPE (stmt_info) != reduc_vec_info_type)
4737 ok = vectorizable_live_operation (stmt, NULL, NULL);
4739 if (!ok)
4741 if (vect_print_dump_info (REPORT_UNVECTORIZED_LOCATIONS))
4743 fprintf (vect_dump, "not vectorized: live stmt not ");
4744 fprintf (vect_dump, "supported: ");
4745 print_gimple_stmt (vect_dump, stmt, 0, TDF_SLIM);
4748 return false;
4751 if (!PURE_SLP_STMT (stmt_info))
4753 /* Groups of strided accesses whose size is not a power of 2 are not
4754 vectorizable yet using loop-vectorization. Therefore, if this stmt
4755 feeds non-SLP-able stmts (i.e., this stmt has to be both SLPed and
4756 loop-based vectorized), the loop cannot be vectorized. */
4757 if (STMT_VINFO_STRIDED_ACCESS (stmt_info)
4758 && exact_log2 (DR_GROUP_SIZE (vinfo_for_stmt (
4759 DR_GROUP_FIRST_DR (stmt_info)))) == -1)
4761 if (vect_print_dump_info (REPORT_DETAILS))
4763 fprintf (vect_dump, "not vectorized: the size of group "
4764 "of strided accesses is not a power of 2");
4765 print_gimple_stmt (vect_dump, stmt, 0, TDF_SLIM);
4768 return false;
4772 return true;
4776 /* Function vect_transform_stmt.
4778 Create a vectorized stmt to replace STMT, and insert it at BSI. */
4780 bool
4781 vect_transform_stmt (gimple stmt, gimple_stmt_iterator *gsi,
4782 bool *strided_store, slp_tree slp_node,
4783 slp_instance slp_node_instance)
4785 bool is_store = false;
4786 gimple vec_stmt = NULL;
4787 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
4788 gimple orig_stmt_in_pattern, orig_scalar_stmt = stmt;
4789 bool done;
4791 switch (STMT_VINFO_TYPE (stmt_info))
4793 case type_demotion_vec_info_type:
4794 done = vectorizable_type_demotion (stmt, gsi, &vec_stmt, slp_node);
4795 gcc_assert (done);
4796 break;
4798 case type_promotion_vec_info_type:
4799 done = vectorizable_type_promotion (stmt, gsi, &vec_stmt, slp_node);
4800 gcc_assert (done);
4801 break;
4803 case type_conversion_vec_info_type:
4804 done = vectorizable_conversion (stmt, gsi, &vec_stmt, slp_node);
4805 gcc_assert (done);
4806 break;
4808 case induc_vec_info_type:
4809 gcc_assert (!slp_node);
4810 done = vectorizable_induction (stmt, gsi, &vec_stmt);
4811 gcc_assert (done);
4812 break;
4814 case shift_vec_info_type:
4815 done = vectorizable_shift (stmt, gsi, &vec_stmt, slp_node);
4816 gcc_assert (done);
4817 break;
4819 case op_vec_info_type:
4820 done = vectorizable_operation (stmt, gsi, &vec_stmt, slp_node);
4821 gcc_assert (done);
4822 break;
4824 case assignment_vec_info_type:
4825 done = vectorizable_assignment (stmt, gsi, &vec_stmt, slp_node);
4826 gcc_assert (done);
4827 break;
4829 case load_vec_info_type:
4830 done = vectorizable_load (stmt, gsi, &vec_stmt, slp_node,
4831 slp_node_instance);
4832 gcc_assert (done);
4833 break;
4835 case store_vec_info_type:
4836 done = vectorizable_store (stmt, gsi, &vec_stmt, slp_node);
4837 gcc_assert (done);
4838 if (STMT_VINFO_STRIDED_ACCESS (stmt_info) && !slp_node)
4840 /* In case of interleaving, the whole chain is vectorized when the
4841 last store in the chain is reached. Store stmts before the last
4842 one are skipped, and there vec_stmt_info shouldn't be freed
4843 meanwhile. */
4844 *strided_store = true;
4845 if (STMT_VINFO_VEC_STMT (stmt_info))
4846 is_store = true;
4848 else
4849 is_store = true;
4850 break;
4852 case condition_vec_info_type:
4853 gcc_assert (!slp_node);
4854 done = vectorizable_condition (stmt, gsi, &vec_stmt, NULL, 0);
4855 gcc_assert (done);
4856 break;
4858 case call_vec_info_type:
4859 gcc_assert (!slp_node);
4860 done = vectorizable_call (stmt, gsi, &vec_stmt);
4861 stmt = gsi_stmt (*gsi);
4862 break;
4864 case reduc_vec_info_type:
4865 done = vectorizable_reduction (stmt, gsi, &vec_stmt, slp_node);
4866 gcc_assert (done);
4867 break;
4869 default:
4870 if (!STMT_VINFO_LIVE_P (stmt_info))
4872 if (vect_print_dump_info (REPORT_DETAILS))
4873 fprintf (vect_dump, "stmt not supported.");
4874 gcc_unreachable ();
4878 /* Handle inner-loop stmts whose DEF is used in the loop-nest that
4879 is being vectorized, but outside the immediately enclosing loop. */
4880 if (vec_stmt
4881 && STMT_VINFO_LOOP_VINFO (stmt_info)
4882 && nested_in_vect_loop_p (LOOP_VINFO_LOOP (
4883 STMT_VINFO_LOOP_VINFO (stmt_info)), stmt)
4884 && STMT_VINFO_TYPE (stmt_info) != reduc_vec_info_type
4885 && (STMT_VINFO_RELEVANT (stmt_info) == vect_used_in_outer
4886 || STMT_VINFO_RELEVANT (stmt_info) ==
4887 vect_used_in_outer_by_reduction))
4889 struct loop *innerloop = LOOP_VINFO_LOOP (
4890 STMT_VINFO_LOOP_VINFO (stmt_info))->inner;
4891 imm_use_iterator imm_iter;
4892 use_operand_p use_p;
4893 tree scalar_dest;
4894 gimple exit_phi;
4896 if (vect_print_dump_info (REPORT_DETAILS))
4897 fprintf (vect_dump, "Record the vdef for outer-loop vectorization.");
4899 /* Find the relevant loop-exit phi-node, and reord the vec_stmt there
4900 (to be used when vectorizing outer-loop stmts that use the DEF of
4901 STMT). */
4902 if (gimple_code (stmt) == GIMPLE_PHI)
4903 scalar_dest = PHI_RESULT (stmt);
4904 else
4905 scalar_dest = gimple_assign_lhs (stmt);
4907 FOR_EACH_IMM_USE_FAST (use_p, imm_iter, scalar_dest)
4909 if (!flow_bb_inside_loop_p (innerloop, gimple_bb (USE_STMT (use_p))))
4911 exit_phi = USE_STMT (use_p);
4912 STMT_VINFO_VEC_STMT (vinfo_for_stmt (exit_phi)) = vec_stmt;
4917 /* Handle stmts whose DEF is used outside the loop-nest that is
4918 being vectorized. */
4919 if (STMT_VINFO_LIVE_P (stmt_info)
4920 && STMT_VINFO_TYPE (stmt_info) != reduc_vec_info_type)
4922 done = vectorizable_live_operation (stmt, gsi, &vec_stmt);
4923 gcc_assert (done);
4926 if (vec_stmt)
4928 STMT_VINFO_VEC_STMT (stmt_info) = vec_stmt;
4929 orig_stmt_in_pattern = STMT_VINFO_RELATED_STMT (stmt_info);
4930 if (orig_stmt_in_pattern)
4932 stmt_vec_info stmt_vinfo = vinfo_for_stmt (orig_stmt_in_pattern);
4933 /* STMT was inserted by the vectorizer to replace a computation idiom.
4934 ORIG_STMT_IN_PATTERN is a stmt in the original sequence that
4935 computed this idiom. We need to record a pointer to VEC_STMT in
4936 the stmt_info of ORIG_STMT_IN_PATTERN. See more details in the
4937 documentation of vect_pattern_recog. */
4938 if (STMT_VINFO_IN_PATTERN_P (stmt_vinfo))
4940 gcc_assert (STMT_VINFO_RELATED_STMT (stmt_vinfo)
4941 == orig_scalar_stmt);
4942 STMT_VINFO_VEC_STMT (stmt_vinfo) = vec_stmt;
4947 return is_store;
4951 /* Remove a group of stores (for SLP or interleaving), free their
4952 stmt_vec_info. */
4954 void
4955 vect_remove_stores (gimple first_stmt)
4957 gimple next = first_stmt;
4958 gimple tmp;
4959 gimple_stmt_iterator next_si;
4961 while (next)
4963 /* Free the attached stmt_vec_info and remove the stmt. */
4964 next_si = gsi_for_stmt (next);
4965 gsi_remove (&next_si, true);
4966 tmp = DR_GROUP_NEXT_DR (vinfo_for_stmt (next));
4967 free_stmt_vec_info (next);
4968 next = tmp;
4973 /* Function new_stmt_vec_info.
4975 Create and initialize a new stmt_vec_info struct for STMT. */
4977 stmt_vec_info
4978 new_stmt_vec_info (gimple stmt, loop_vec_info loop_vinfo,
4979 bb_vec_info bb_vinfo)
4981 stmt_vec_info res;
4982 res = (stmt_vec_info) xcalloc (1, sizeof (struct _stmt_vec_info));
4984 STMT_VINFO_TYPE (res) = undef_vec_info_type;
4985 STMT_VINFO_STMT (res) = stmt;
4986 STMT_VINFO_LOOP_VINFO (res) = loop_vinfo;
4987 STMT_VINFO_BB_VINFO (res) = bb_vinfo;
4988 STMT_VINFO_RELEVANT (res) = vect_unused_in_scope;
4989 STMT_VINFO_LIVE_P (res) = false;
4990 STMT_VINFO_VECTYPE (res) = NULL;
4991 STMT_VINFO_VEC_STMT (res) = NULL;
4992 STMT_VINFO_VECTORIZABLE (res) = true;
4993 STMT_VINFO_IN_PATTERN_P (res) = false;
4994 STMT_VINFO_RELATED_STMT (res) = NULL;
4995 STMT_VINFO_DATA_REF (res) = NULL;
4997 STMT_VINFO_DR_BASE_ADDRESS (res) = NULL;
4998 STMT_VINFO_DR_OFFSET (res) = NULL;
4999 STMT_VINFO_DR_INIT (res) = NULL;
5000 STMT_VINFO_DR_STEP (res) = NULL;
5001 STMT_VINFO_DR_ALIGNED_TO (res) = NULL;
5003 if (gimple_code (stmt) == GIMPLE_PHI
5004 && is_loop_header_bb_p (gimple_bb (stmt)))
5005 STMT_VINFO_DEF_TYPE (res) = vect_unknown_def_type;
5006 else
5007 STMT_VINFO_DEF_TYPE (res) = vect_internal_def;
5009 STMT_VINFO_SAME_ALIGN_REFS (res) = VEC_alloc (dr_p, heap, 5);
5010 STMT_VINFO_INSIDE_OF_LOOP_COST (res) = 0;
5011 STMT_VINFO_OUTSIDE_OF_LOOP_COST (res) = 0;
5012 STMT_SLP_TYPE (res) = loop_vect;
5013 DR_GROUP_FIRST_DR (res) = NULL;
5014 DR_GROUP_NEXT_DR (res) = NULL;
5015 DR_GROUP_SIZE (res) = 0;
5016 DR_GROUP_STORE_COUNT (res) = 0;
5017 DR_GROUP_GAP (res) = 0;
5018 DR_GROUP_SAME_DR_STMT (res) = NULL;
5019 DR_GROUP_READ_WRITE_DEPENDENCE (res) = false;
5021 return res;
5025 /* Create a hash table for stmt_vec_info. */
5027 void
5028 init_stmt_vec_info_vec (void)
5030 gcc_assert (!stmt_vec_info_vec);
5031 stmt_vec_info_vec = VEC_alloc (vec_void_p, heap, 50);
5035 /* Free hash table for stmt_vec_info. */
5037 void
5038 free_stmt_vec_info_vec (void)
5040 gcc_assert (stmt_vec_info_vec);
5041 VEC_free (vec_void_p, heap, stmt_vec_info_vec);
5045 /* Free stmt vectorization related info. */
5047 void
5048 free_stmt_vec_info (gimple stmt)
5050 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
5052 if (!stmt_info)
5053 return;
5055 VEC_free (dr_p, heap, STMT_VINFO_SAME_ALIGN_REFS (stmt_info));
5056 set_vinfo_for_stmt (stmt, NULL);
5057 free (stmt_info);
5061 /* Function get_vectype_for_scalar_type_and_size.
5063 Returns the vector type corresponding to SCALAR_TYPE and SIZE as supported
5064 by the target. */
5066 static tree
5067 get_vectype_for_scalar_type_and_size (tree scalar_type, unsigned size)
5069 enum machine_mode inner_mode = TYPE_MODE (scalar_type);
5070 enum machine_mode simd_mode;
5071 unsigned int nbytes = GET_MODE_SIZE (inner_mode);
5072 int nunits;
5073 tree vectype;
5075 if (nbytes == 0)
5076 return NULL_TREE;
5078 /* We can't build a vector type of elements with alignment bigger than
5079 their size. */
5080 if (nbytes < TYPE_ALIGN_UNIT (scalar_type))
5081 return NULL_TREE;
5083 /* If we'd build a vector type of elements whose mode precision doesn't
5084 match their types precision we'll get mismatched types on vector
5085 extracts via BIT_FIELD_REFs. This effectively means we disable
5086 vectorization of bool and/or enum types in some languages. */
5087 if (INTEGRAL_TYPE_P (scalar_type)
5088 && GET_MODE_BITSIZE (inner_mode) != TYPE_PRECISION (scalar_type))
5089 return NULL_TREE;
5091 if (GET_MODE_CLASS (inner_mode) != MODE_INT
5092 && GET_MODE_CLASS (inner_mode) != MODE_FLOAT)
5093 return NULL_TREE;
5095 /* If no size was supplied use the mode the target prefers. Otherwise
5096 lookup a vector mode of the specified size. */
5097 if (size == 0)
5098 simd_mode = targetm.vectorize.preferred_simd_mode (inner_mode);
5099 else
5100 simd_mode = mode_for_vector (inner_mode, size / nbytes);
5101 nunits = GET_MODE_SIZE (simd_mode) / nbytes;
5102 if (nunits <= 1)
5103 return NULL_TREE;
5105 vectype = build_vector_type (scalar_type, nunits);
5106 if (vect_print_dump_info (REPORT_DETAILS))
5108 fprintf (vect_dump, "get vectype with %d units of type ", nunits);
5109 print_generic_expr (vect_dump, scalar_type, TDF_SLIM);
5112 if (!vectype)
5113 return NULL_TREE;
5115 if (vect_print_dump_info (REPORT_DETAILS))
5117 fprintf (vect_dump, "vectype: ");
5118 print_generic_expr (vect_dump, vectype, TDF_SLIM);
5121 if (!VECTOR_MODE_P (TYPE_MODE (vectype))
5122 && !INTEGRAL_MODE_P (TYPE_MODE (vectype)))
5124 if (vect_print_dump_info (REPORT_DETAILS))
5125 fprintf (vect_dump, "mode not supported by target.");
5126 return NULL_TREE;
5129 return vectype;
5132 unsigned int current_vector_size;
5134 /* Function get_vectype_for_scalar_type.
5136 Returns the vector type corresponding to SCALAR_TYPE as supported
5137 by the target. */
5139 tree
5140 get_vectype_for_scalar_type (tree scalar_type)
5142 tree vectype;
5143 vectype = get_vectype_for_scalar_type_and_size (scalar_type,
5144 current_vector_size);
5145 if (vectype
5146 && current_vector_size == 0)
5147 current_vector_size = GET_MODE_SIZE (TYPE_MODE (vectype));
5148 return vectype;
5151 /* Function get_same_sized_vectype
5153 Returns a vector type corresponding to SCALAR_TYPE of size
5154 VECTOR_TYPE if supported by the target. */
5156 tree
5157 get_same_sized_vectype (tree scalar_type, tree vector_type)
5159 return get_vectype_for_scalar_type_and_size
5160 (scalar_type, GET_MODE_SIZE (TYPE_MODE (vector_type)));
5163 /* Function vect_is_simple_use.
5165 Input:
5166 LOOP_VINFO - the vect info of the loop that is being vectorized.
5167 BB_VINFO - the vect info of the basic block that is being vectorized.
5168 OPERAND - operand of a stmt in the loop or bb.
5169 DEF - the defining stmt in case OPERAND is an SSA_NAME.
5171 Returns whether a stmt with OPERAND can be vectorized.
5172 For loops, supportable operands are constants, loop invariants, and operands
5173 that are defined by the current iteration of the loop. Unsupportable
5174 operands are those that are defined by a previous iteration of the loop (as
5175 is the case in reduction/induction computations).
5176 For basic blocks, supportable operands are constants and bb invariants.
5177 For now, operands defined outside the basic block are not supported. */
5179 bool
5180 vect_is_simple_use (tree operand, loop_vec_info loop_vinfo,
5181 bb_vec_info bb_vinfo, gimple *def_stmt,
5182 tree *def, enum vect_def_type *dt)
5184 basic_block bb;
5185 stmt_vec_info stmt_vinfo;
5186 struct loop *loop = NULL;
5188 if (loop_vinfo)
5189 loop = LOOP_VINFO_LOOP (loop_vinfo);
5191 *def_stmt = NULL;
5192 *def = NULL_TREE;
5194 if (vect_print_dump_info (REPORT_DETAILS))
5196 fprintf (vect_dump, "vect_is_simple_use: operand ");
5197 print_generic_expr (vect_dump, operand, TDF_SLIM);
5200 if (TREE_CODE (operand) == INTEGER_CST || TREE_CODE (operand) == REAL_CST)
5202 *dt = vect_constant_def;
5203 return true;
5206 if (is_gimple_min_invariant (operand))
5208 *def = operand;
5209 *dt = vect_external_def;
5210 return true;
5213 if (TREE_CODE (operand) == PAREN_EXPR)
5215 if (vect_print_dump_info (REPORT_DETAILS))
5216 fprintf (vect_dump, "non-associatable copy.");
5217 operand = TREE_OPERAND (operand, 0);
5220 if (TREE_CODE (operand) != SSA_NAME)
5222 if (vect_print_dump_info (REPORT_DETAILS))
5223 fprintf (vect_dump, "not ssa-name.");
5224 return false;
5227 *def_stmt = SSA_NAME_DEF_STMT (operand);
5228 if (*def_stmt == NULL)
5230 if (vect_print_dump_info (REPORT_DETAILS))
5231 fprintf (vect_dump, "no def_stmt.");
5232 return false;
5235 if (vect_print_dump_info (REPORT_DETAILS))
5237 fprintf (vect_dump, "def_stmt: ");
5238 print_gimple_stmt (vect_dump, *def_stmt, 0, TDF_SLIM);
5241 /* Empty stmt is expected only in case of a function argument.
5242 (Otherwise - we expect a phi_node or a GIMPLE_ASSIGN). */
5243 if (gimple_nop_p (*def_stmt))
5245 *def = operand;
5246 *dt = vect_external_def;
5247 return true;
5250 bb = gimple_bb (*def_stmt);
5252 if ((loop && !flow_bb_inside_loop_p (loop, bb))
5253 || (!loop && bb != BB_VINFO_BB (bb_vinfo))
5254 || (!loop && gimple_code (*def_stmt) == GIMPLE_PHI))
5255 *dt = vect_external_def;
5256 else
5258 stmt_vinfo = vinfo_for_stmt (*def_stmt);
5259 *dt = STMT_VINFO_DEF_TYPE (stmt_vinfo);
5262 if (*dt == vect_unknown_def_type)
5264 if (vect_print_dump_info (REPORT_DETAILS))
5265 fprintf (vect_dump, "Unsupported pattern.");
5266 return false;
5269 if (vect_print_dump_info (REPORT_DETAILS))
5270 fprintf (vect_dump, "type of def: %d.",*dt);
5272 switch (gimple_code (*def_stmt))
5274 case GIMPLE_PHI:
5275 *def = gimple_phi_result (*def_stmt);
5276 break;
5278 case GIMPLE_ASSIGN:
5279 *def = gimple_assign_lhs (*def_stmt);
5280 break;
5282 case GIMPLE_CALL:
5283 *def = gimple_call_lhs (*def_stmt);
5284 if (*def != NULL)
5285 break;
5286 /* FALLTHRU */
5287 default:
5288 if (vect_print_dump_info (REPORT_DETAILS))
5289 fprintf (vect_dump, "unsupported defining stmt: ");
5290 return false;
5293 return true;
5296 /* Function vect_is_simple_use_1.
5298 Same as vect_is_simple_use_1 but also determines the vector operand
5299 type of OPERAND and stores it to *VECTYPE. If the definition of
5300 OPERAND is vect_uninitialized_def, vect_constant_def or
5301 vect_external_def *VECTYPE will be set to NULL_TREE and the caller
5302 is responsible to compute the best suited vector type for the
5303 scalar operand. */
5305 bool
5306 vect_is_simple_use_1 (tree operand, loop_vec_info loop_vinfo,
5307 bb_vec_info bb_vinfo, gimple *def_stmt,
5308 tree *def, enum vect_def_type *dt, tree *vectype)
5310 if (!vect_is_simple_use (operand, loop_vinfo, bb_vinfo, def_stmt, def, dt))
5311 return false;
5313 /* Now get a vector type if the def is internal, otherwise supply
5314 NULL_TREE and leave it up to the caller to figure out a proper
5315 type for the use stmt. */
5316 if (*dt == vect_internal_def
5317 || *dt == vect_induction_def
5318 || *dt == vect_reduction_def
5319 || *dt == vect_double_reduction_def
5320 || *dt == vect_nested_cycle)
5322 stmt_vec_info stmt_info = vinfo_for_stmt (*def_stmt);
5323 if (STMT_VINFO_IN_PATTERN_P (stmt_info))
5324 stmt_info = vinfo_for_stmt (STMT_VINFO_RELATED_STMT (stmt_info));
5325 *vectype = STMT_VINFO_VECTYPE (stmt_info);
5326 gcc_assert (*vectype != NULL_TREE);
5328 else if (*dt == vect_uninitialized_def
5329 || *dt == vect_constant_def
5330 || *dt == vect_external_def)
5331 *vectype = NULL_TREE;
5332 else
5333 gcc_unreachable ();
5335 return true;
5339 /* Function supportable_widening_operation
5341 Check whether an operation represented by the code CODE is a
5342 widening operation that is supported by the target platform in
5343 vector form (i.e., when operating on arguments of type VECTYPE_IN
5344 producing a result of type VECTYPE_OUT).
5346 Widening operations we currently support are NOP (CONVERT), FLOAT
5347 and WIDEN_MULT. This function checks if these operations are supported
5348 by the target platform either directly (via vector tree-codes), or via
5349 target builtins.
5351 Output:
5352 - CODE1 and CODE2 are codes of vector operations to be used when
5353 vectorizing the operation, if available.
5354 - DECL1 and DECL2 are decls of target builtin functions to be used
5355 when vectorizing the operation, if available. In this case,
5356 CODE1 and CODE2 are CALL_EXPR.
5357 - MULTI_STEP_CVT determines the number of required intermediate steps in
5358 case of multi-step conversion (like char->short->int - in that case
5359 MULTI_STEP_CVT will be 1).
5360 - INTERM_TYPES contains the intermediate type required to perform the
5361 widening operation (short in the above example). */
5363 bool
5364 supportable_widening_operation (enum tree_code code, gimple stmt,
5365 tree vectype_out, tree vectype_in,
5366 tree *decl1, tree *decl2,
5367 enum tree_code *code1, enum tree_code *code2,
5368 int *multi_step_cvt,
5369 VEC (tree, heap) **interm_types)
5371 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
5372 loop_vec_info loop_info = STMT_VINFO_LOOP_VINFO (stmt_info);
5373 struct loop *vect_loop = LOOP_VINFO_LOOP (loop_info);
5374 bool ordered_p;
5375 enum machine_mode vec_mode;
5376 enum insn_code icode1, icode2;
5377 optab optab1, optab2;
5378 tree vectype = vectype_in;
5379 tree wide_vectype = vectype_out;
5380 enum tree_code c1, c2;
5382 /* The result of a vectorized widening operation usually requires two vectors
5383 (because the widened results do not fit int one vector). The generated
5384 vector results would normally be expected to be generated in the same
5385 order as in the original scalar computation, i.e. if 8 results are
5386 generated in each vector iteration, they are to be organized as follows:
5387 vect1: [res1,res2,res3,res4], vect2: [res5,res6,res7,res8].
5389 However, in the special case that the result of the widening operation is
5390 used in a reduction computation only, the order doesn't matter (because
5391 when vectorizing a reduction we change the order of the computation).
5392 Some targets can take advantage of this and generate more efficient code.
5393 For example, targets like Altivec, that support widen_mult using a sequence
5394 of {mult_even,mult_odd} generate the following vectors:
5395 vect1: [res1,res3,res5,res7], vect2: [res2,res4,res6,res8].
5397 When vectorizing outer-loops, we execute the inner-loop sequentially
5398 (each vectorized inner-loop iteration contributes to VF outer-loop
5399 iterations in parallel). We therefore don't allow to change the order
5400 of the computation in the inner-loop during outer-loop vectorization. */
5402 if (STMT_VINFO_RELEVANT (stmt_info) == vect_used_by_reduction
5403 && !nested_in_vect_loop_p (vect_loop, stmt))
5404 ordered_p = false;
5405 else
5406 ordered_p = true;
5408 if (!ordered_p
5409 && code == WIDEN_MULT_EXPR
5410 && targetm.vectorize.builtin_mul_widen_even
5411 && targetm.vectorize.builtin_mul_widen_even (vectype)
5412 && targetm.vectorize.builtin_mul_widen_odd
5413 && targetm.vectorize.builtin_mul_widen_odd (vectype))
5415 if (vect_print_dump_info (REPORT_DETAILS))
5416 fprintf (vect_dump, "Unordered widening operation detected.");
5418 *code1 = *code2 = CALL_EXPR;
5419 *decl1 = targetm.vectorize.builtin_mul_widen_even (vectype);
5420 *decl2 = targetm.vectorize.builtin_mul_widen_odd (vectype);
5421 return true;
5424 switch (code)
5426 case WIDEN_MULT_EXPR:
5427 if (BYTES_BIG_ENDIAN)
5429 c1 = VEC_WIDEN_MULT_HI_EXPR;
5430 c2 = VEC_WIDEN_MULT_LO_EXPR;
5432 else
5434 c2 = VEC_WIDEN_MULT_HI_EXPR;
5435 c1 = VEC_WIDEN_MULT_LO_EXPR;
5437 break;
5439 CASE_CONVERT:
5440 if (BYTES_BIG_ENDIAN)
5442 c1 = VEC_UNPACK_HI_EXPR;
5443 c2 = VEC_UNPACK_LO_EXPR;
5445 else
5447 c2 = VEC_UNPACK_HI_EXPR;
5448 c1 = VEC_UNPACK_LO_EXPR;
5450 break;
5452 case FLOAT_EXPR:
5453 if (BYTES_BIG_ENDIAN)
5455 c1 = VEC_UNPACK_FLOAT_HI_EXPR;
5456 c2 = VEC_UNPACK_FLOAT_LO_EXPR;
5458 else
5460 c2 = VEC_UNPACK_FLOAT_HI_EXPR;
5461 c1 = VEC_UNPACK_FLOAT_LO_EXPR;
5463 break;
5465 case FIX_TRUNC_EXPR:
5466 /* ??? Not yet implemented due to missing VEC_UNPACK_FIX_TRUNC_HI_EXPR/
5467 VEC_UNPACK_FIX_TRUNC_LO_EXPR tree codes and optabs used for
5468 computing the operation. */
5469 return false;
5471 default:
5472 gcc_unreachable ();
5475 if (code == FIX_TRUNC_EXPR)
5477 /* The signedness is determined from output operand. */
5478 optab1 = optab_for_tree_code (c1, vectype_out, optab_default);
5479 optab2 = optab_for_tree_code (c2, vectype_out, optab_default);
5481 else
5483 optab1 = optab_for_tree_code (c1, vectype, optab_default);
5484 optab2 = optab_for_tree_code (c2, vectype, optab_default);
5487 if (!optab1 || !optab2)
5488 return false;
5490 vec_mode = TYPE_MODE (vectype);
5491 if ((icode1 = optab_handler (optab1, vec_mode)) == CODE_FOR_nothing
5492 || (icode2 = optab_handler (optab2, vec_mode)) == CODE_FOR_nothing)
5493 return false;
5495 /* Check if it's a multi-step conversion that can be done using intermediate
5496 types. */
5497 if (insn_data[icode1].operand[0].mode != TYPE_MODE (wide_vectype)
5498 || insn_data[icode2].operand[0].mode != TYPE_MODE (wide_vectype))
5500 int i;
5501 tree prev_type = vectype, intermediate_type;
5502 enum machine_mode intermediate_mode, prev_mode = vec_mode;
5503 optab optab3, optab4;
5505 if (!CONVERT_EXPR_CODE_P (code))
5506 return false;
5508 *code1 = c1;
5509 *code2 = c2;
5511 /* We assume here that there will not be more than MAX_INTERM_CVT_STEPS
5512 intermediate steps in promotion sequence. We try
5513 MAX_INTERM_CVT_STEPS to get to NARROW_VECTYPE, and fail if we do
5514 not. */
5515 *interm_types = VEC_alloc (tree, heap, MAX_INTERM_CVT_STEPS);
5516 for (i = 0; i < 3; i++)
5518 intermediate_mode = insn_data[icode1].operand[0].mode;
5519 intermediate_type = lang_hooks.types.type_for_mode (intermediate_mode,
5520 TYPE_UNSIGNED (prev_type));
5521 optab3 = optab_for_tree_code (c1, intermediate_type, optab_default);
5522 optab4 = optab_for_tree_code (c2, intermediate_type, optab_default);
5524 if (!optab3 || !optab4
5525 || ((icode1 = optab_handler (optab1, prev_mode))
5526 == CODE_FOR_nothing)
5527 || insn_data[icode1].operand[0].mode != intermediate_mode
5528 || ((icode2 = optab_handler (optab2, prev_mode))
5529 == CODE_FOR_nothing)
5530 || insn_data[icode2].operand[0].mode != intermediate_mode
5531 || ((icode1 = optab_handler (optab3, intermediate_mode))
5532 == CODE_FOR_nothing)
5533 || ((icode2 = optab_handler (optab4, intermediate_mode))
5534 == CODE_FOR_nothing))
5535 return false;
5537 VEC_quick_push (tree, *interm_types, intermediate_type);
5538 (*multi_step_cvt)++;
5540 if (insn_data[icode1].operand[0].mode == TYPE_MODE (wide_vectype)
5541 && insn_data[icode2].operand[0].mode == TYPE_MODE (wide_vectype))
5542 return true;
5544 prev_type = intermediate_type;
5545 prev_mode = intermediate_mode;
5548 return false;
5551 *code1 = c1;
5552 *code2 = c2;
5553 return true;
5557 /* Function supportable_narrowing_operation
5559 Check whether an operation represented by the code CODE is a
5560 narrowing operation that is supported by the target platform in
5561 vector form (i.e., when operating on arguments of type VECTYPE_IN
5562 and producing a result of type VECTYPE_OUT).
5564 Narrowing operations we currently support are NOP (CONVERT) and
5565 FIX_TRUNC. This function checks if these operations are supported by
5566 the target platform directly via vector tree-codes.
5568 Output:
5569 - CODE1 is the code of a vector operation to be used when
5570 vectorizing the operation, if available.
5571 - MULTI_STEP_CVT determines the number of required intermediate steps in
5572 case of multi-step conversion (like int->short->char - in that case
5573 MULTI_STEP_CVT will be 1).
5574 - INTERM_TYPES contains the intermediate type required to perform the
5575 narrowing operation (short in the above example). */
5577 bool
5578 supportable_narrowing_operation (enum tree_code code,
5579 tree vectype_out, tree vectype_in,
5580 enum tree_code *code1, int *multi_step_cvt,
5581 VEC (tree, heap) **interm_types)
5583 enum machine_mode vec_mode;
5584 enum insn_code icode1;
5585 optab optab1, interm_optab;
5586 tree vectype = vectype_in;
5587 tree narrow_vectype = vectype_out;
5588 enum tree_code c1;
5589 tree intermediate_type, prev_type;
5590 int i;
5592 switch (code)
5594 CASE_CONVERT:
5595 c1 = VEC_PACK_TRUNC_EXPR;
5596 break;
5598 case FIX_TRUNC_EXPR:
5599 c1 = VEC_PACK_FIX_TRUNC_EXPR;
5600 break;
5602 case FLOAT_EXPR:
5603 /* ??? Not yet implemented due to missing VEC_PACK_FLOAT_EXPR
5604 tree code and optabs used for computing the operation. */
5605 return false;
5607 default:
5608 gcc_unreachable ();
5611 if (code == FIX_TRUNC_EXPR)
5612 /* The signedness is determined from output operand. */
5613 optab1 = optab_for_tree_code (c1, vectype_out, optab_default);
5614 else
5615 optab1 = optab_for_tree_code (c1, vectype, optab_default);
5617 if (!optab1)
5618 return false;
5620 vec_mode = TYPE_MODE (vectype);
5621 if ((icode1 = optab_handler (optab1, vec_mode)) == CODE_FOR_nothing)
5622 return false;
5624 /* Check if it's a multi-step conversion that can be done using intermediate
5625 types. */
5626 if (insn_data[icode1].operand[0].mode != TYPE_MODE (narrow_vectype))
5628 enum machine_mode intermediate_mode, prev_mode = vec_mode;
5630 *code1 = c1;
5631 prev_type = vectype;
5632 /* We assume here that there will not be more than MAX_INTERM_CVT_STEPS
5633 intermediate steps in promotion sequence. We try
5634 MAX_INTERM_CVT_STEPS to get to NARROW_VECTYPE, and fail if we do
5635 not. */
5636 *interm_types = VEC_alloc (tree, heap, MAX_INTERM_CVT_STEPS);
5637 for (i = 0; i < 3; i++)
5639 intermediate_mode = insn_data[icode1].operand[0].mode;
5640 intermediate_type = lang_hooks.types.type_for_mode (intermediate_mode,
5641 TYPE_UNSIGNED (prev_type));
5642 interm_optab = optab_for_tree_code (c1, intermediate_type,
5643 optab_default);
5644 if (!interm_optab
5645 || ((icode1 = optab_handler (optab1, prev_mode))
5646 == CODE_FOR_nothing)
5647 || insn_data[icode1].operand[0].mode != intermediate_mode
5648 || ((icode1 = optab_handler (interm_optab, intermediate_mode))
5649 == CODE_FOR_nothing))
5650 return false;
5652 VEC_quick_push (tree, *interm_types, intermediate_type);
5653 (*multi_step_cvt)++;
5655 if (insn_data[icode1].operand[0].mode == TYPE_MODE (narrow_vectype))
5656 return true;
5658 prev_type = intermediate_type;
5659 prev_mode = intermediate_mode;
5662 return false;
5665 *code1 = c1;
5666 return true;