Bring in all split-stack work done over on gccgo branch.
[official-gcc.git] / gcc / tree-vect-stmts.c
blobf562ed2d528f7b845fa7df066af1f95281bd6601
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 "toplev.h"
42 #include "tree-vectorizer.h"
43 #include "langhooks.h"
46 /* Utility functions used by vect_mark_stmts_to_be_vectorized. */
48 /* Function vect_mark_relevant.
50 Mark STMT as "relevant for vectorization" and add it to WORKLIST. */
52 static void
53 vect_mark_relevant (VEC(gimple,heap) **worklist, gimple stmt,
54 enum vect_relevant relevant, bool live_p)
56 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
57 enum vect_relevant save_relevant = STMT_VINFO_RELEVANT (stmt_info);
58 bool save_live_p = STMT_VINFO_LIVE_P (stmt_info);
60 if (vect_print_dump_info (REPORT_DETAILS))
61 fprintf (vect_dump, "mark relevant %d, live %d.", relevant, live_p);
63 if (STMT_VINFO_IN_PATTERN_P (stmt_info))
65 gimple pattern_stmt;
67 /* This is the last stmt in a sequence that was detected as a
68 pattern that can potentially be vectorized. Don't mark the stmt
69 as relevant/live because it's not going to be vectorized.
70 Instead mark the pattern-stmt that replaces it. */
72 pattern_stmt = STMT_VINFO_RELATED_STMT (stmt_info);
74 if (vect_print_dump_info (REPORT_DETAILS))
75 fprintf (vect_dump, "last stmt in pattern. don't mark relevant/live.");
76 stmt_info = vinfo_for_stmt (pattern_stmt);
77 gcc_assert (STMT_VINFO_RELATED_STMT (stmt_info) == stmt);
78 save_relevant = STMT_VINFO_RELEVANT (stmt_info);
79 save_live_p = STMT_VINFO_LIVE_P (stmt_info);
80 stmt = pattern_stmt;
83 STMT_VINFO_LIVE_P (stmt_info) |= live_p;
84 if (relevant > STMT_VINFO_RELEVANT (stmt_info))
85 STMT_VINFO_RELEVANT (stmt_info) = relevant;
87 if (STMT_VINFO_RELEVANT (stmt_info) == save_relevant
88 && STMT_VINFO_LIVE_P (stmt_info) == save_live_p)
90 if (vect_print_dump_info (REPORT_DETAILS))
91 fprintf (vect_dump, "already marked relevant/live.");
92 return;
95 VEC_safe_push (gimple, heap, *worklist, stmt);
99 /* Function vect_stmt_relevant_p.
101 Return true if STMT in loop that is represented by LOOP_VINFO is
102 "relevant for vectorization".
104 A stmt is considered "relevant for vectorization" if:
105 - it has uses outside the loop.
106 - it has vdefs (it alters memory).
107 - control stmts in the loop (except for the exit condition).
109 CHECKME: what other side effects would the vectorizer allow? */
111 static bool
112 vect_stmt_relevant_p (gimple stmt, loop_vec_info loop_vinfo,
113 enum vect_relevant *relevant, bool *live_p)
115 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
116 ssa_op_iter op_iter;
117 imm_use_iterator imm_iter;
118 use_operand_p use_p;
119 def_operand_p def_p;
121 *relevant = vect_unused_in_scope;
122 *live_p = false;
124 /* cond stmt other than loop exit cond. */
125 if (is_ctrl_stmt (stmt)
126 && STMT_VINFO_TYPE (vinfo_for_stmt (stmt))
127 != loop_exit_ctrl_vec_info_type)
128 *relevant = vect_used_in_scope;
130 /* changing memory. */
131 if (gimple_code (stmt) != GIMPLE_PHI)
132 if (gimple_vdef (stmt))
134 if (vect_print_dump_info (REPORT_DETAILS))
135 fprintf (vect_dump, "vec_stmt_relevant_p: stmt has vdefs.");
136 *relevant = vect_used_in_scope;
139 /* uses outside the loop. */
140 FOR_EACH_PHI_OR_STMT_DEF (def_p, stmt, op_iter, SSA_OP_DEF)
142 FOR_EACH_IMM_USE_FAST (use_p, imm_iter, DEF_FROM_PTR (def_p))
144 basic_block bb = gimple_bb (USE_STMT (use_p));
145 if (!flow_bb_inside_loop_p (loop, bb))
147 if (vect_print_dump_info (REPORT_DETAILS))
148 fprintf (vect_dump, "vec_stmt_relevant_p: used out of loop.");
150 if (is_gimple_debug (USE_STMT (use_p)))
151 continue;
153 /* We expect all such uses to be in the loop exit phis
154 (because of loop closed form) */
155 gcc_assert (gimple_code (USE_STMT (use_p)) == GIMPLE_PHI);
156 gcc_assert (bb == single_exit (loop)->dest);
158 *live_p = true;
163 return (*live_p || *relevant);
167 /* Function exist_non_indexing_operands_for_use_p
169 USE is one of the uses attached to STMT. Check if USE is
170 used in STMT for anything other than indexing an array. */
172 static bool
173 exist_non_indexing_operands_for_use_p (tree use, gimple stmt)
175 tree operand;
176 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
178 /* USE corresponds to some operand in STMT. If there is no data
179 reference in STMT, then any operand that corresponds to USE
180 is not indexing an array. */
181 if (!STMT_VINFO_DATA_REF (stmt_info))
182 return true;
184 /* STMT has a data_ref. FORNOW this means that its of one of
185 the following forms:
186 -1- ARRAY_REF = var
187 -2- var = ARRAY_REF
188 (This should have been verified in analyze_data_refs).
190 'var' in the second case corresponds to a def, not a use,
191 so USE cannot correspond to any operands that are not used
192 for array indexing.
194 Therefore, all we need to check is if STMT falls into the
195 first case, and whether var corresponds to USE. */
197 if (!gimple_assign_copy_p (stmt))
198 return false;
199 if (TREE_CODE (gimple_assign_lhs (stmt)) == SSA_NAME)
200 return false;
201 operand = gimple_assign_rhs1 (stmt);
202 if (TREE_CODE (operand) != SSA_NAME)
203 return false;
205 if (operand == use)
206 return true;
208 return false;
213 Function process_use.
215 Inputs:
216 - a USE in STMT in a loop represented by LOOP_VINFO
217 - LIVE_P, RELEVANT - enum values to be set in the STMT_VINFO of the stmt
218 that defined USE. This is done by calling mark_relevant and passing it
219 the WORKLIST (to add DEF_STMT to the WORKLIST in case it is relevant).
221 Outputs:
222 Generally, LIVE_P and RELEVANT are used to define the liveness and
223 relevance info of the DEF_STMT of this USE:
224 STMT_VINFO_LIVE_P (DEF_STMT_info) <-- live_p
225 STMT_VINFO_RELEVANT (DEF_STMT_info) <-- relevant
226 Exceptions:
227 - case 1: If USE is used only for address computations (e.g. array indexing),
228 which does not need to be directly vectorized, then the liveness/relevance
229 of the respective DEF_STMT is left unchanged.
230 - case 2: If STMT is a reduction phi and DEF_STMT is a reduction stmt, we
231 skip DEF_STMT cause it had already been processed.
232 - case 3: If DEF_STMT and STMT are in different nests, then "relevant" will
233 be modified accordingly.
235 Return true if everything is as expected. Return false otherwise. */
237 static bool
238 process_use (gimple stmt, tree use, loop_vec_info loop_vinfo, bool live_p,
239 enum vect_relevant relevant, VEC(gimple,heap) **worklist)
241 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
242 stmt_vec_info stmt_vinfo = vinfo_for_stmt (stmt);
243 stmt_vec_info dstmt_vinfo;
244 basic_block bb, def_bb;
245 tree def;
246 gimple def_stmt;
247 enum vect_def_type dt;
249 /* case 1: we are only interested in uses that need to be vectorized. Uses
250 that are used for address computation are not considered relevant. */
251 if (!exist_non_indexing_operands_for_use_p (use, stmt))
252 return true;
254 if (!vect_is_simple_use (use, loop_vinfo, NULL, &def_stmt, &def, &dt))
256 if (vect_print_dump_info (REPORT_UNVECTORIZED_LOCATIONS))
257 fprintf (vect_dump, "not vectorized: unsupported use in stmt.");
258 return false;
261 if (!def_stmt || gimple_nop_p (def_stmt))
262 return true;
264 def_bb = gimple_bb (def_stmt);
265 if (!flow_bb_inside_loop_p (loop, def_bb))
267 if (vect_print_dump_info (REPORT_DETAILS))
268 fprintf (vect_dump, "def_stmt is out of loop.");
269 return true;
272 /* case 2: A reduction phi (STMT) defined by a reduction stmt (DEF_STMT).
273 DEF_STMT must have already been processed, because this should be the
274 only way that STMT, which is a reduction-phi, was put in the worklist,
275 as there should be no other uses for DEF_STMT in the loop. So we just
276 check that everything is as expected, and we are done. */
277 dstmt_vinfo = vinfo_for_stmt (def_stmt);
278 bb = gimple_bb (stmt);
279 if (gimple_code (stmt) == GIMPLE_PHI
280 && STMT_VINFO_DEF_TYPE (stmt_vinfo) == vect_reduction_def
281 && gimple_code (def_stmt) != GIMPLE_PHI
282 && STMT_VINFO_DEF_TYPE (dstmt_vinfo) == vect_reduction_def
283 && bb->loop_father == def_bb->loop_father)
285 if (vect_print_dump_info (REPORT_DETAILS))
286 fprintf (vect_dump, "reduc-stmt defining reduc-phi in the same nest.");
287 if (STMT_VINFO_IN_PATTERN_P (dstmt_vinfo))
288 dstmt_vinfo = vinfo_for_stmt (STMT_VINFO_RELATED_STMT (dstmt_vinfo));
289 gcc_assert (STMT_VINFO_RELEVANT (dstmt_vinfo) < vect_used_by_reduction);
290 gcc_assert (STMT_VINFO_LIVE_P (dstmt_vinfo)
291 || STMT_VINFO_RELEVANT (dstmt_vinfo) > vect_unused_in_scope);
292 return true;
295 /* case 3a: outer-loop stmt defining an inner-loop stmt:
296 outer-loop-header-bb:
297 d = def_stmt
298 inner-loop:
299 stmt # use (d)
300 outer-loop-tail-bb:
301 ... */
302 if (flow_loop_nested_p (def_bb->loop_father, bb->loop_father))
304 if (vect_print_dump_info (REPORT_DETAILS))
305 fprintf (vect_dump, "outer-loop def-stmt defining inner-loop stmt.");
307 switch (relevant)
309 case vect_unused_in_scope:
310 relevant = (STMT_VINFO_DEF_TYPE (stmt_vinfo) == vect_nested_cycle) ?
311 vect_used_in_scope : vect_unused_in_scope;
312 break;
314 case vect_used_in_outer_by_reduction:
315 gcc_assert (STMT_VINFO_DEF_TYPE (stmt_vinfo) != vect_reduction_def);
316 relevant = vect_used_by_reduction;
317 break;
319 case vect_used_in_outer:
320 gcc_assert (STMT_VINFO_DEF_TYPE (stmt_vinfo) != vect_reduction_def);
321 relevant = vect_used_in_scope;
322 break;
324 case vect_used_in_scope:
325 break;
327 default:
328 gcc_unreachable ();
332 /* case 3b: inner-loop stmt defining an outer-loop stmt:
333 outer-loop-header-bb:
335 inner-loop:
336 d = def_stmt
337 outer-loop-tail-bb (or outer-loop-exit-bb in double reduction):
338 stmt # use (d) */
339 else if (flow_loop_nested_p (bb->loop_father, def_bb->loop_father))
341 if (vect_print_dump_info (REPORT_DETAILS))
342 fprintf (vect_dump, "inner-loop def-stmt defining outer-loop stmt.");
344 switch (relevant)
346 case vect_unused_in_scope:
347 relevant = (STMT_VINFO_DEF_TYPE (stmt_vinfo) == vect_reduction_def
348 || STMT_VINFO_DEF_TYPE (stmt_vinfo) == vect_double_reduction_def) ?
349 vect_used_in_outer_by_reduction : vect_unused_in_scope;
350 break;
352 case vect_used_by_reduction:
353 relevant = vect_used_in_outer_by_reduction;
354 break;
356 case vect_used_in_scope:
357 relevant = vect_used_in_outer;
358 break;
360 default:
361 gcc_unreachable ();
365 vect_mark_relevant (worklist, def_stmt, relevant, live_p);
366 return true;
370 /* Function vect_mark_stmts_to_be_vectorized.
372 Not all stmts in the loop need to be vectorized. For example:
374 for i...
375 for j...
376 1. T0 = i + j
377 2. T1 = a[T0]
379 3. j = j + 1
381 Stmt 1 and 3 do not need to be vectorized, because loop control and
382 addressing of vectorized data-refs are handled differently.
384 This pass detects such stmts. */
386 bool
387 vect_mark_stmts_to_be_vectorized (loop_vec_info loop_vinfo)
389 VEC(gimple,heap) *worklist;
390 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
391 basic_block *bbs = LOOP_VINFO_BBS (loop_vinfo);
392 unsigned int nbbs = loop->num_nodes;
393 gimple_stmt_iterator si;
394 gimple stmt;
395 unsigned int i;
396 stmt_vec_info stmt_vinfo;
397 basic_block bb;
398 gimple phi;
399 bool live_p;
400 enum vect_relevant relevant, tmp_relevant;
401 enum vect_def_type def_type;
403 if (vect_print_dump_info (REPORT_DETAILS))
404 fprintf (vect_dump, "=== vect_mark_stmts_to_be_vectorized ===");
406 worklist = VEC_alloc (gimple, heap, 64);
408 /* 1. Init worklist. */
409 for (i = 0; i < nbbs; i++)
411 bb = bbs[i];
412 for (si = gsi_start_phis (bb); !gsi_end_p (si); gsi_next (&si))
414 phi = gsi_stmt (si);
415 if (vect_print_dump_info (REPORT_DETAILS))
417 fprintf (vect_dump, "init: phi relevant? ");
418 print_gimple_stmt (vect_dump, phi, 0, TDF_SLIM);
421 if (vect_stmt_relevant_p (phi, loop_vinfo, &relevant, &live_p))
422 vect_mark_relevant (&worklist, phi, relevant, live_p);
424 for (si = gsi_start_bb (bb); !gsi_end_p (si); gsi_next (&si))
426 stmt = gsi_stmt (si);
427 if (vect_print_dump_info (REPORT_DETAILS))
429 fprintf (vect_dump, "init: stmt relevant? ");
430 print_gimple_stmt (vect_dump, stmt, 0, TDF_SLIM);
433 if (vect_stmt_relevant_p (stmt, loop_vinfo, &relevant, &live_p))
434 vect_mark_relevant (&worklist, stmt, relevant, live_p);
438 /* 2. Process_worklist */
439 while (VEC_length (gimple, worklist) > 0)
441 use_operand_p use_p;
442 ssa_op_iter iter;
444 stmt = VEC_pop (gimple, worklist);
445 if (vect_print_dump_info (REPORT_DETAILS))
447 fprintf (vect_dump, "worklist: examine stmt: ");
448 print_gimple_stmt (vect_dump, stmt, 0, TDF_SLIM);
451 /* Examine the USEs of STMT. For each USE, mark the stmt that defines it
452 (DEF_STMT) as relevant/irrelevant and live/dead according to the
453 liveness and relevance properties of STMT. */
454 stmt_vinfo = vinfo_for_stmt (stmt);
455 relevant = STMT_VINFO_RELEVANT (stmt_vinfo);
456 live_p = STMT_VINFO_LIVE_P (stmt_vinfo);
458 /* Generally, the liveness and relevance properties of STMT are
459 propagated as is to the DEF_STMTs of its USEs:
460 live_p <-- STMT_VINFO_LIVE_P (STMT_VINFO)
461 relevant <-- STMT_VINFO_RELEVANT (STMT_VINFO)
463 One exception is when STMT has been identified as defining a reduction
464 variable; in this case we set the liveness/relevance as follows:
465 live_p = false
466 relevant = vect_used_by_reduction
467 This is because we distinguish between two kinds of relevant stmts -
468 those that are used by a reduction computation, and those that are
469 (also) used by a regular computation. This allows us later on to
470 identify stmts that are used solely by a reduction, and therefore the
471 order of the results that they produce does not have to be kept. */
473 def_type = STMT_VINFO_DEF_TYPE (stmt_vinfo);
474 tmp_relevant = relevant;
475 switch (def_type)
477 case vect_reduction_def:
478 switch (tmp_relevant)
480 case vect_unused_in_scope:
481 relevant = vect_used_by_reduction;
482 break;
484 case vect_used_by_reduction:
485 if (gimple_code (stmt) == GIMPLE_PHI)
486 break;
487 /* fall through */
489 default:
490 if (vect_print_dump_info (REPORT_DETAILS))
491 fprintf (vect_dump, "unsupported use of reduction.");
493 VEC_free (gimple, heap, worklist);
494 return false;
497 live_p = false;
498 break;
500 case vect_nested_cycle:
501 if (tmp_relevant != vect_unused_in_scope
502 && tmp_relevant != vect_used_in_outer_by_reduction
503 && tmp_relevant != vect_used_in_outer)
505 if (vect_print_dump_info (REPORT_DETAILS))
506 fprintf (vect_dump, "unsupported use of nested cycle.");
508 VEC_free (gimple, heap, worklist);
509 return false;
512 live_p = false;
513 break;
515 case vect_double_reduction_def:
516 if (tmp_relevant != vect_unused_in_scope
517 && tmp_relevant != vect_used_by_reduction)
519 if (vect_print_dump_info (REPORT_DETAILS))
520 fprintf (vect_dump, "unsupported use of double reduction.");
522 VEC_free (gimple, heap, worklist);
523 return false;
526 live_p = false;
527 break;
529 default:
530 break;
533 FOR_EACH_PHI_OR_STMT_USE (use_p, stmt, iter, SSA_OP_USE)
535 tree op = USE_FROM_PTR (use_p);
536 if (!process_use (stmt, op, loop_vinfo, live_p, relevant, &worklist))
538 VEC_free (gimple, heap, worklist);
539 return false;
542 } /* while worklist */
544 VEC_free (gimple, heap, worklist);
545 return true;
549 /* Get cost by calling cost target builtin. */
551 static inline
552 int vect_get_stmt_cost (enum vect_cost_for_stmt type_of_cost)
554 tree dummy_type = NULL;
555 int dummy = 0;
557 return targetm.vectorize.builtin_vectorization_cost (type_of_cost,
558 dummy_type, dummy);
562 /* Get cost for STMT. */
565 cost_for_stmt (gimple stmt)
567 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
569 switch (STMT_VINFO_TYPE (stmt_info))
571 case load_vec_info_type:
572 return vect_get_stmt_cost (scalar_load);
573 case store_vec_info_type:
574 return vect_get_stmt_cost (scalar_store);
575 case op_vec_info_type:
576 case condition_vec_info_type:
577 case assignment_vec_info_type:
578 case reduc_vec_info_type:
579 case induc_vec_info_type:
580 case type_promotion_vec_info_type:
581 case type_demotion_vec_info_type:
582 case type_conversion_vec_info_type:
583 case call_vec_info_type:
584 return vect_get_stmt_cost (scalar_stmt);
585 case undef_vec_info_type:
586 default:
587 gcc_unreachable ();
591 /* Function vect_model_simple_cost.
593 Models cost for simple operations, i.e. those that only emit ncopies of a
594 single op. Right now, this does not account for multiple insns that could
595 be generated for the single vector op. We will handle that shortly. */
597 void
598 vect_model_simple_cost (stmt_vec_info stmt_info, int ncopies,
599 enum vect_def_type *dt, slp_tree slp_node)
601 int i;
602 int inside_cost = 0, outside_cost = 0;
604 /* The SLP costs were already calculated during SLP tree build. */
605 if (PURE_SLP_STMT (stmt_info))
606 return;
608 inside_cost = ncopies * vect_get_stmt_cost (vector_stmt);
610 /* FORNOW: Assuming maximum 2 args per stmts. */
611 for (i = 0; i < 2; i++)
613 if (dt[i] == vect_constant_def || dt[i] == vect_external_def)
614 outside_cost += vect_get_stmt_cost (vector_stmt);
617 if (vect_print_dump_info (REPORT_COST))
618 fprintf (vect_dump, "vect_model_simple_cost: inside_cost = %d, "
619 "outside_cost = %d .", inside_cost, outside_cost);
621 /* Set the costs either in STMT_INFO or SLP_NODE (if exists). */
622 stmt_vinfo_set_inside_of_loop_cost (stmt_info, slp_node, inside_cost);
623 stmt_vinfo_set_outside_of_loop_cost (stmt_info, slp_node, outside_cost);
627 /* Function vect_cost_strided_group_size
629 For strided load or store, return the group_size only if it is the first
630 load or store of a group, else return 1. This ensures that group size is
631 only returned once per group. */
633 static int
634 vect_cost_strided_group_size (stmt_vec_info stmt_info)
636 gimple first_stmt = DR_GROUP_FIRST_DR (stmt_info);
638 if (first_stmt == STMT_VINFO_STMT (stmt_info))
639 return DR_GROUP_SIZE (stmt_info);
641 return 1;
645 /* Function vect_model_store_cost
647 Models cost for stores. In the case of strided accesses, one access
648 has the overhead of the strided access attributed to it. */
650 void
651 vect_model_store_cost (stmt_vec_info stmt_info, int ncopies,
652 enum vect_def_type dt, slp_tree slp_node)
654 int group_size;
655 unsigned int inside_cost = 0, outside_cost = 0;
656 struct data_reference *first_dr;
657 gimple first_stmt;
659 /* The SLP costs were already calculated during SLP tree build. */
660 if (PURE_SLP_STMT (stmt_info))
661 return;
663 if (dt == vect_constant_def || dt == vect_external_def)
664 outside_cost = vect_get_stmt_cost (scalar_to_vec);
666 /* Strided access? */
667 if (DR_GROUP_FIRST_DR (stmt_info))
669 if (slp_node)
671 first_stmt = VEC_index (gimple, SLP_TREE_SCALAR_STMTS (slp_node), 0);
672 group_size = 1;
674 else
676 first_stmt = DR_GROUP_FIRST_DR (stmt_info);
677 group_size = vect_cost_strided_group_size (stmt_info);
680 first_dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (first_stmt));
682 /* Not a strided access. */
683 else
685 group_size = 1;
686 first_dr = STMT_VINFO_DATA_REF (stmt_info);
689 /* Is this an access in a group of stores, which provide strided access?
690 If so, add in the cost of the permutes. */
691 if (group_size > 1)
693 /* Uses a high and low interleave operation for each needed permute. */
694 inside_cost = ncopies * exact_log2(group_size) * group_size
695 * vect_get_stmt_cost (vector_stmt);
697 if (vect_print_dump_info (REPORT_COST))
698 fprintf (vect_dump, "vect_model_store_cost: strided group_size = %d .",
699 group_size);
703 /* Costs of the stores. */
704 vect_get_store_cost (first_dr, ncopies, &inside_cost);
706 if (vect_print_dump_info (REPORT_COST))
707 fprintf (vect_dump, "vect_model_store_cost: inside_cost = %d, "
708 "outside_cost = %d .", inside_cost, outside_cost);
710 /* Set the costs either in STMT_INFO or SLP_NODE (if exists). */
711 stmt_vinfo_set_inside_of_loop_cost (stmt_info, slp_node, inside_cost);
712 stmt_vinfo_set_outside_of_loop_cost (stmt_info, slp_node, outside_cost);
716 /* Calculate cost of DR's memory access. */
717 void
718 vect_get_store_cost (struct data_reference *dr, int ncopies,
719 unsigned int *inside_cost)
721 int alignment_support_scheme = vect_supportable_dr_alignment (dr, false);
723 switch (alignment_support_scheme)
725 case dr_aligned:
727 *inside_cost += ncopies * vect_get_stmt_cost (vector_store);
729 if (vect_print_dump_info (REPORT_COST))
730 fprintf (vect_dump, "vect_model_store_cost: aligned.");
732 break;
735 case dr_unaligned_supported:
737 gimple stmt = DR_STMT (dr);
738 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
739 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
741 /* Here, we assign an additional cost for the unaligned store. */
742 *inside_cost += ncopies
743 * targetm.vectorize.builtin_vectorization_cost (unaligned_store,
744 vectype, DR_MISALIGNMENT (dr));
746 if (vect_print_dump_info (REPORT_COST))
747 fprintf (vect_dump, "vect_model_store_cost: unaligned supported by "
748 "hardware.");
750 break;
753 default:
754 gcc_unreachable ();
759 /* Function vect_model_load_cost
761 Models cost for loads. In the case of strided accesses, the last access
762 has the overhead of the strided access attributed to it. Since unaligned
763 accesses are supported for loads, we also account for the costs of the
764 access scheme chosen. */
766 void
767 vect_model_load_cost (stmt_vec_info stmt_info, int ncopies, slp_tree slp_node)
770 int group_size;
771 gimple first_stmt;
772 struct data_reference *dr = STMT_VINFO_DATA_REF (stmt_info), *first_dr;
773 unsigned int inside_cost = 0, outside_cost = 0;
775 /* The SLP costs were already calculated during SLP tree build. */
776 if (PURE_SLP_STMT (stmt_info))
777 return;
779 /* Strided accesses? */
780 first_stmt = DR_GROUP_FIRST_DR (stmt_info);
781 if (first_stmt && !slp_node)
783 group_size = vect_cost_strided_group_size (stmt_info);
784 first_dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (first_stmt));
786 /* Not a strided access. */
787 else
789 group_size = 1;
790 first_dr = dr;
793 /* Is this an access in a group of loads providing strided access?
794 If so, add in the cost of the permutes. */
795 if (group_size > 1)
797 /* Uses an even and odd extract operations for each needed permute. */
798 inside_cost = ncopies * exact_log2(group_size) * group_size
799 * vect_get_stmt_cost (vector_stmt);
801 if (vect_print_dump_info (REPORT_COST))
802 fprintf (vect_dump, "vect_model_load_cost: strided group_size = %d .",
803 group_size);
806 /* The loads themselves. */
807 vect_get_load_cost (first_dr, ncopies,
808 ((!DR_GROUP_FIRST_DR (stmt_info)) || group_size > 1 || slp_node),
809 &inside_cost, &outside_cost);
811 if (vect_print_dump_info (REPORT_COST))
812 fprintf (vect_dump, "vect_model_load_cost: inside_cost = %d, "
813 "outside_cost = %d .", inside_cost, outside_cost);
815 /* Set the costs either in STMT_INFO or SLP_NODE (if exists). */
816 stmt_vinfo_set_inside_of_loop_cost (stmt_info, slp_node, inside_cost);
817 stmt_vinfo_set_outside_of_loop_cost (stmt_info, slp_node, outside_cost);
821 /* Calculate cost of DR's memory access. */
822 void
823 vect_get_load_cost (struct data_reference *dr, int ncopies,
824 bool add_realign_cost, unsigned int *inside_cost,
825 unsigned int *outside_cost)
827 int alignment_support_scheme = vect_supportable_dr_alignment (dr, false);
829 switch (alignment_support_scheme)
831 case dr_aligned:
833 *inside_cost += ncopies * vect_get_stmt_cost (vector_load);
835 if (vect_print_dump_info (REPORT_COST))
836 fprintf (vect_dump, "vect_model_load_cost: aligned.");
838 break;
840 case dr_unaligned_supported:
842 gimple stmt = DR_STMT (dr);
843 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
844 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
846 /* Here, we assign an additional cost for the unaligned load. */
847 *inside_cost += ncopies
848 * targetm.vectorize.builtin_vectorization_cost (unaligned_load,
849 vectype, DR_MISALIGNMENT (dr));
850 if (vect_print_dump_info (REPORT_COST))
851 fprintf (vect_dump, "vect_model_load_cost: unaligned supported by "
852 "hardware.");
854 break;
856 case dr_explicit_realign:
858 *inside_cost += ncopies * (2 * vect_get_stmt_cost (vector_load)
859 + vect_get_stmt_cost (vector_stmt));
861 /* FIXME: If the misalignment remains fixed across the iterations of
862 the containing loop, the following cost should be added to the
863 outside costs. */
864 if (targetm.vectorize.builtin_mask_for_load)
865 *inside_cost += vect_get_stmt_cost (vector_stmt);
867 break;
869 case dr_explicit_realign_optimized:
871 if (vect_print_dump_info (REPORT_COST))
872 fprintf (vect_dump, "vect_model_load_cost: unaligned software "
873 "pipelined.");
875 /* Unaligned software pipeline has a load of an address, an initial
876 load, and possibly a mask operation to "prime" the loop. However,
877 if this is an access in a group of loads, which provide strided
878 access, then the above cost should only be considered for one
879 access in the group. Inside the loop, there is a load op
880 and a realignment op. */
882 if (add_realign_cost)
884 *outside_cost = 2 * vect_get_stmt_cost (vector_stmt);
885 if (targetm.vectorize.builtin_mask_for_load)
886 *outside_cost += vect_get_stmt_cost (vector_stmt);
889 *inside_cost += ncopies * (vect_get_stmt_cost (vector_load)
890 + vect_get_stmt_cost (vector_stmt));
891 break;
894 default:
895 gcc_unreachable ();
900 /* Function vect_init_vector.
902 Insert a new stmt (INIT_STMT) that initializes a new vector variable with
903 the vector elements of VECTOR_VAR. Place the initialization at BSI if it
904 is not NULL. Otherwise, place the initialization at the loop preheader.
905 Return the DEF of INIT_STMT.
906 It will be used in the vectorization of STMT. */
908 tree
909 vect_init_vector (gimple stmt, tree vector_var, tree vector_type,
910 gimple_stmt_iterator *gsi)
912 stmt_vec_info stmt_vinfo = vinfo_for_stmt (stmt);
913 tree new_var;
914 gimple init_stmt;
915 tree vec_oprnd;
916 edge pe;
917 tree new_temp;
918 basic_block new_bb;
920 new_var = vect_get_new_vect_var (vector_type, vect_simple_var, "cst_");
921 add_referenced_var (new_var);
922 init_stmt = gimple_build_assign (new_var, vector_var);
923 new_temp = make_ssa_name (new_var, init_stmt);
924 gimple_assign_set_lhs (init_stmt, new_temp);
926 if (gsi)
927 vect_finish_stmt_generation (stmt, init_stmt, gsi);
928 else
930 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_vinfo);
932 if (loop_vinfo)
934 struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
936 if (nested_in_vect_loop_p (loop, stmt))
937 loop = loop->inner;
939 pe = loop_preheader_edge (loop);
940 new_bb = gsi_insert_on_edge_immediate (pe, init_stmt);
941 gcc_assert (!new_bb);
943 else
945 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_vinfo);
946 basic_block bb;
947 gimple_stmt_iterator gsi_bb_start;
949 gcc_assert (bb_vinfo);
950 bb = BB_VINFO_BB (bb_vinfo);
951 gsi_bb_start = gsi_after_labels (bb);
952 gsi_insert_before (&gsi_bb_start, init_stmt, GSI_SAME_STMT);
956 if (vect_print_dump_info (REPORT_DETAILS))
958 fprintf (vect_dump, "created new init_stmt: ");
959 print_gimple_stmt (vect_dump, init_stmt, 0, TDF_SLIM);
962 vec_oprnd = gimple_assign_lhs (init_stmt);
963 return vec_oprnd;
967 /* Function vect_get_vec_def_for_operand.
969 OP is an operand in STMT. This function returns a (vector) def that will be
970 used in the vectorized stmt for STMT.
972 In the case that OP is an SSA_NAME which is defined in the loop, then
973 STMT_VINFO_VEC_STMT of the defining stmt holds the relevant def.
975 In case OP is an invariant or constant, a new stmt that creates a vector def
976 needs to be introduced. */
978 tree
979 vect_get_vec_def_for_operand (tree op, gimple stmt, tree *scalar_def)
981 tree vec_oprnd;
982 gimple vec_stmt;
983 gimple def_stmt;
984 stmt_vec_info def_stmt_info = NULL;
985 stmt_vec_info stmt_vinfo = vinfo_for_stmt (stmt);
986 tree vectype = STMT_VINFO_VECTYPE (stmt_vinfo);
987 unsigned int nunits = TYPE_VECTOR_SUBPARTS (vectype);
988 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_vinfo);
989 tree vec_inv;
990 tree vec_cst;
991 tree t = NULL_TREE;
992 tree def;
993 int i;
994 enum vect_def_type dt;
995 bool is_simple_use;
996 tree vector_type;
998 if (vect_print_dump_info (REPORT_DETAILS))
1000 fprintf (vect_dump, "vect_get_vec_def_for_operand: ");
1001 print_generic_expr (vect_dump, op, TDF_SLIM);
1004 is_simple_use = vect_is_simple_use (op, loop_vinfo, NULL, &def_stmt, &def,
1005 &dt);
1006 gcc_assert (is_simple_use);
1007 if (vect_print_dump_info (REPORT_DETAILS))
1009 if (def)
1011 fprintf (vect_dump, "def = ");
1012 print_generic_expr (vect_dump, def, TDF_SLIM);
1014 if (def_stmt)
1016 fprintf (vect_dump, " def_stmt = ");
1017 print_gimple_stmt (vect_dump, def_stmt, 0, TDF_SLIM);
1021 switch (dt)
1023 /* Case 1: operand is a constant. */
1024 case vect_constant_def:
1026 vector_type = get_vectype_for_scalar_type (TREE_TYPE (op));
1027 gcc_assert (vector_type);
1029 if (scalar_def)
1030 *scalar_def = op;
1032 /* Create 'vect_cst_ = {cst,cst,...,cst}' */
1033 if (vect_print_dump_info (REPORT_DETAILS))
1034 fprintf (vect_dump, "Create vector_cst. nunits = %d", nunits);
1036 for (i = nunits - 1; i >= 0; --i)
1038 t = tree_cons (NULL_TREE, op, t);
1040 vec_cst = build_vector (vector_type, t);
1041 return vect_init_vector (stmt, vec_cst, vector_type, NULL);
1044 /* Case 2: operand is defined outside the loop - loop invariant. */
1045 case vect_external_def:
1047 vector_type = get_vectype_for_scalar_type (TREE_TYPE (def));
1048 gcc_assert (vector_type);
1049 nunits = TYPE_VECTOR_SUBPARTS (vector_type);
1051 if (scalar_def)
1052 *scalar_def = def;
1054 /* Create 'vec_inv = {inv,inv,..,inv}' */
1055 if (vect_print_dump_info (REPORT_DETAILS))
1056 fprintf (vect_dump, "Create vector_inv.");
1058 for (i = nunits - 1; i >= 0; --i)
1060 t = tree_cons (NULL_TREE, def, t);
1063 /* FIXME: use build_constructor directly. */
1064 vec_inv = build_constructor_from_list (vector_type, t);
1065 return vect_init_vector (stmt, vec_inv, vector_type, NULL);
1068 /* Case 3: operand is defined inside the loop. */
1069 case vect_internal_def:
1071 if (scalar_def)
1072 *scalar_def = NULL/* FIXME tuples: def_stmt*/;
1074 /* Get the def from the vectorized stmt. */
1075 def_stmt_info = vinfo_for_stmt (def_stmt);
1076 vec_stmt = STMT_VINFO_VEC_STMT (def_stmt_info);
1077 gcc_assert (vec_stmt);
1078 if (gimple_code (vec_stmt) == GIMPLE_PHI)
1079 vec_oprnd = PHI_RESULT (vec_stmt);
1080 else if (is_gimple_call (vec_stmt))
1081 vec_oprnd = gimple_call_lhs (vec_stmt);
1082 else
1083 vec_oprnd = gimple_assign_lhs (vec_stmt);
1084 return vec_oprnd;
1087 /* Case 4: operand is defined by a loop header phi - reduction */
1088 case vect_reduction_def:
1089 case vect_double_reduction_def:
1090 case vect_nested_cycle:
1092 struct loop *loop;
1094 gcc_assert (gimple_code (def_stmt) == GIMPLE_PHI);
1095 loop = (gimple_bb (def_stmt))->loop_father;
1097 /* Get the def before the loop */
1098 op = PHI_ARG_DEF_FROM_EDGE (def_stmt, loop_preheader_edge (loop));
1099 return get_initial_def_for_reduction (stmt, op, scalar_def);
1102 /* Case 5: operand is defined by loop-header phi - induction. */
1103 case vect_induction_def:
1105 gcc_assert (gimple_code (def_stmt) == GIMPLE_PHI);
1107 /* Get the def from the vectorized stmt. */
1108 def_stmt_info = vinfo_for_stmt (def_stmt);
1109 vec_stmt = STMT_VINFO_VEC_STMT (def_stmt_info);
1110 gcc_assert (vec_stmt && gimple_code (vec_stmt) == GIMPLE_PHI);
1111 vec_oprnd = PHI_RESULT (vec_stmt);
1112 return vec_oprnd;
1115 default:
1116 gcc_unreachable ();
1121 /* Function vect_get_vec_def_for_stmt_copy
1123 Return a vector-def for an operand. This function is used when the
1124 vectorized stmt to be created (by the caller to this function) is a "copy"
1125 created in case the vectorized result cannot fit in one vector, and several
1126 copies of the vector-stmt are required. In this case the vector-def is
1127 retrieved from the vector stmt recorded in the STMT_VINFO_RELATED_STMT field
1128 of the stmt that defines VEC_OPRND.
1129 DT is the type of the vector def VEC_OPRND.
1131 Context:
1132 In case the vectorization factor (VF) is bigger than the number
1133 of elements that can fit in a vectype (nunits), we have to generate
1134 more than one vector stmt to vectorize the scalar stmt. This situation
1135 arises when there are multiple data-types operated upon in the loop; the
1136 smallest data-type determines the VF, and as a result, when vectorizing
1137 stmts operating on wider types we need to create 'VF/nunits' "copies" of the
1138 vector stmt (each computing a vector of 'nunits' results, and together
1139 computing 'VF' results in each iteration). This function is called when
1140 vectorizing such a stmt (e.g. vectorizing S2 in the illustration below, in
1141 which VF=16 and nunits=4, so the number of copies required is 4):
1143 scalar stmt: vectorized into: STMT_VINFO_RELATED_STMT
1145 S1: x = load VS1.0: vx.0 = memref0 VS1.1
1146 VS1.1: vx.1 = memref1 VS1.2
1147 VS1.2: vx.2 = memref2 VS1.3
1148 VS1.3: vx.3 = memref3
1150 S2: z = x + ... VSnew.0: vz0 = vx.0 + ... VSnew.1
1151 VSnew.1: vz1 = vx.1 + ... VSnew.2
1152 VSnew.2: vz2 = vx.2 + ... VSnew.3
1153 VSnew.3: vz3 = vx.3 + ...
1155 The vectorization of S1 is explained in vectorizable_load.
1156 The vectorization of S2:
1157 To create the first vector-stmt out of the 4 copies - VSnew.0 -
1158 the function 'vect_get_vec_def_for_operand' is called to
1159 get the relevant vector-def for each operand of S2. For operand x it
1160 returns the vector-def 'vx.0'.
1162 To create the remaining copies of the vector-stmt (VSnew.j), this
1163 function is called to get the relevant vector-def for each operand. It is
1164 obtained from the respective VS1.j stmt, which is recorded in the
1165 STMT_VINFO_RELATED_STMT field of the stmt that defines VEC_OPRND.
1167 For example, to obtain the vector-def 'vx.1' in order to create the
1168 vector stmt 'VSnew.1', this function is called with VEC_OPRND='vx.0'.
1169 Given 'vx0' we obtain the stmt that defines it ('VS1.0'); from the
1170 STMT_VINFO_RELATED_STMT field of 'VS1.0' we obtain the next copy - 'VS1.1',
1171 and return its def ('vx.1').
1172 Overall, to create the above sequence this function will be called 3 times:
1173 vx.1 = vect_get_vec_def_for_stmt_copy (dt, vx.0);
1174 vx.2 = vect_get_vec_def_for_stmt_copy (dt, vx.1);
1175 vx.3 = vect_get_vec_def_for_stmt_copy (dt, vx.2); */
1177 tree
1178 vect_get_vec_def_for_stmt_copy (enum vect_def_type dt, tree vec_oprnd)
1180 gimple vec_stmt_for_operand;
1181 stmt_vec_info def_stmt_info;
1183 /* Do nothing; can reuse same def. */
1184 if (dt == vect_external_def || dt == vect_constant_def )
1185 return vec_oprnd;
1187 vec_stmt_for_operand = SSA_NAME_DEF_STMT (vec_oprnd);
1188 def_stmt_info = vinfo_for_stmt (vec_stmt_for_operand);
1189 gcc_assert (def_stmt_info);
1190 vec_stmt_for_operand = STMT_VINFO_RELATED_STMT (def_stmt_info);
1191 gcc_assert (vec_stmt_for_operand);
1192 vec_oprnd = gimple_get_lhs (vec_stmt_for_operand);
1193 if (gimple_code (vec_stmt_for_operand) == GIMPLE_PHI)
1194 vec_oprnd = PHI_RESULT (vec_stmt_for_operand);
1195 else
1196 vec_oprnd = gimple_get_lhs (vec_stmt_for_operand);
1197 return vec_oprnd;
1201 /* Get vectorized definitions for the operands to create a copy of an original
1202 stmt. See vect_get_vec_def_for_stmt_copy () for details. */
1204 static void
1205 vect_get_vec_defs_for_stmt_copy (enum vect_def_type *dt,
1206 VEC(tree,heap) **vec_oprnds0,
1207 VEC(tree,heap) **vec_oprnds1)
1209 tree vec_oprnd = VEC_pop (tree, *vec_oprnds0);
1211 vec_oprnd = vect_get_vec_def_for_stmt_copy (dt[0], vec_oprnd);
1212 VEC_quick_push (tree, *vec_oprnds0, vec_oprnd);
1214 if (vec_oprnds1 && *vec_oprnds1)
1216 vec_oprnd = VEC_pop (tree, *vec_oprnds1);
1217 vec_oprnd = vect_get_vec_def_for_stmt_copy (dt[1], vec_oprnd);
1218 VEC_quick_push (tree, *vec_oprnds1, vec_oprnd);
1223 /* Get vectorized definitions for OP0 and OP1, or SLP_NODE if it is not
1224 NULL. */
1226 static void
1227 vect_get_vec_defs (tree op0, tree op1, gimple stmt,
1228 VEC(tree,heap) **vec_oprnds0, VEC(tree,heap) **vec_oprnds1,
1229 slp_tree slp_node)
1231 if (slp_node)
1232 vect_get_slp_defs (slp_node, vec_oprnds0, vec_oprnds1, -1);
1233 else
1235 tree vec_oprnd;
1237 *vec_oprnds0 = VEC_alloc (tree, heap, 1);
1238 vec_oprnd = vect_get_vec_def_for_operand (op0, stmt, NULL);
1239 VEC_quick_push (tree, *vec_oprnds0, vec_oprnd);
1241 if (op1)
1243 *vec_oprnds1 = VEC_alloc (tree, heap, 1);
1244 vec_oprnd = vect_get_vec_def_for_operand (op1, stmt, NULL);
1245 VEC_quick_push (tree, *vec_oprnds1, vec_oprnd);
1251 /* Function vect_finish_stmt_generation.
1253 Insert a new stmt. */
1255 void
1256 vect_finish_stmt_generation (gimple stmt, gimple vec_stmt,
1257 gimple_stmt_iterator *gsi)
1259 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
1260 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
1261 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
1263 gcc_assert (gimple_code (stmt) != GIMPLE_LABEL);
1265 gsi_insert_before (gsi, vec_stmt, GSI_SAME_STMT);
1267 set_vinfo_for_stmt (vec_stmt, new_stmt_vec_info (vec_stmt, loop_vinfo,
1268 bb_vinfo));
1270 if (vect_print_dump_info (REPORT_DETAILS))
1272 fprintf (vect_dump, "add new stmt: ");
1273 print_gimple_stmt (vect_dump, vec_stmt, 0, TDF_SLIM);
1276 gimple_set_location (vec_stmt, gimple_location (gsi_stmt (*gsi)));
1279 /* Checks if CALL can be vectorized in type VECTYPE. Returns
1280 a function declaration if the target has a vectorized version
1281 of the function, or NULL_TREE if the function cannot be vectorized. */
1283 tree
1284 vectorizable_function (gimple call, tree vectype_out, tree vectype_in)
1286 tree fndecl = gimple_call_fndecl (call);
1288 /* We only handle functions that do not read or clobber memory -- i.e.
1289 const or novops ones. */
1290 if (!(gimple_call_flags (call) & (ECF_CONST | ECF_NOVOPS)))
1291 return NULL_TREE;
1293 if (!fndecl
1294 || TREE_CODE (fndecl) != FUNCTION_DECL
1295 || !DECL_BUILT_IN (fndecl))
1296 return NULL_TREE;
1298 return targetm.vectorize.builtin_vectorized_function (fndecl, vectype_out,
1299 vectype_in);
1302 /* Function vectorizable_call.
1304 Check if STMT performs a function call that can be vectorized.
1305 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
1306 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
1307 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
1309 static bool
1310 vectorizable_call (gimple stmt, gimple_stmt_iterator *gsi, gimple *vec_stmt)
1312 tree vec_dest;
1313 tree scalar_dest;
1314 tree op, type;
1315 tree vec_oprnd0 = NULL_TREE, vec_oprnd1 = NULL_TREE;
1316 stmt_vec_info stmt_info = vinfo_for_stmt (stmt), prev_stmt_info;
1317 tree vectype_out, vectype_in;
1318 int nunits_in;
1319 int nunits_out;
1320 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
1321 tree fndecl, new_temp, def, rhs_type;
1322 gimple def_stmt;
1323 enum vect_def_type dt[2] = {vect_unknown_def_type, vect_unknown_def_type};
1324 gimple new_stmt = NULL;
1325 int ncopies, j;
1326 VEC(tree, heap) *vargs = NULL;
1327 enum { NARROW, NONE, WIDEN } modifier;
1328 size_t i, nargs;
1330 /* FORNOW: unsupported in basic block SLP. */
1331 gcc_assert (loop_vinfo);
1333 if (!STMT_VINFO_RELEVANT_P (stmt_info))
1334 return false;
1336 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
1337 return false;
1339 /* FORNOW: SLP not supported. */
1340 if (STMT_SLP_TYPE (stmt_info))
1341 return false;
1343 /* Is STMT a vectorizable call? */
1344 if (!is_gimple_call (stmt))
1345 return false;
1347 if (TREE_CODE (gimple_call_lhs (stmt)) != SSA_NAME)
1348 return false;
1350 if (stmt_could_throw_p (stmt))
1351 return false;
1353 vectype_out = STMT_VINFO_VECTYPE (stmt_info);
1355 /* Process function arguments. */
1356 rhs_type = NULL_TREE;
1357 vectype_in = NULL_TREE;
1358 nargs = gimple_call_num_args (stmt);
1360 /* Bail out if the function has more than two arguments, we
1361 do not have interesting builtin functions to vectorize with
1362 more than two arguments. No arguments is also not good. */
1363 if (nargs == 0 || nargs > 2)
1364 return false;
1366 for (i = 0; i < nargs; i++)
1368 tree opvectype;
1370 op = gimple_call_arg (stmt, i);
1372 /* We can only handle calls with arguments of the same type. */
1373 if (rhs_type
1374 && !types_compatible_p (rhs_type, TREE_TYPE (op)))
1376 if (vect_print_dump_info (REPORT_DETAILS))
1377 fprintf (vect_dump, "argument types differ.");
1378 return false;
1380 if (!rhs_type)
1381 rhs_type = TREE_TYPE (op);
1383 if (!vect_is_simple_use_1 (op, loop_vinfo, NULL,
1384 &def_stmt, &def, &dt[i], &opvectype))
1386 if (vect_print_dump_info (REPORT_DETAILS))
1387 fprintf (vect_dump, "use not simple.");
1388 return false;
1391 if (!vectype_in)
1392 vectype_in = opvectype;
1393 else if (opvectype
1394 && opvectype != vectype_in)
1396 if (vect_print_dump_info (REPORT_DETAILS))
1397 fprintf (vect_dump, "argument vector types differ.");
1398 return false;
1401 /* If all arguments are external or constant defs use a vector type with
1402 the same size as the output vector type. */
1403 if (!vectype_in)
1404 vectype_in = get_same_sized_vectype (rhs_type, vectype_out);
1405 if (vec_stmt)
1406 gcc_assert (vectype_in);
1407 if (!vectype_in)
1409 if (vect_print_dump_info (REPORT_DETAILS))
1411 fprintf (vect_dump, "no vectype for scalar type ");
1412 print_generic_expr (vect_dump, rhs_type, TDF_SLIM);
1415 return false;
1418 /* FORNOW */
1419 nunits_in = TYPE_VECTOR_SUBPARTS (vectype_in);
1420 nunits_out = TYPE_VECTOR_SUBPARTS (vectype_out);
1421 if (nunits_in == nunits_out / 2)
1422 modifier = NARROW;
1423 else if (nunits_out == nunits_in)
1424 modifier = NONE;
1425 else if (nunits_out == nunits_in / 2)
1426 modifier = WIDEN;
1427 else
1428 return false;
1430 /* For now, we only vectorize functions if a target specific builtin
1431 is available. TODO -- in some cases, it might be profitable to
1432 insert the calls for pieces of the vector, in order to be able
1433 to vectorize other operations in the loop. */
1434 fndecl = vectorizable_function (stmt, vectype_out, vectype_in);
1435 if (fndecl == NULL_TREE)
1437 if (vect_print_dump_info (REPORT_DETAILS))
1438 fprintf (vect_dump, "function is not vectorizable.");
1440 return false;
1443 gcc_assert (!gimple_vuse (stmt));
1445 if (modifier == NARROW)
1446 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_out;
1447 else
1448 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_in;
1450 /* Sanity check: make sure that at least one copy of the vectorized stmt
1451 needs to be generated. */
1452 gcc_assert (ncopies >= 1);
1454 if (!vec_stmt) /* transformation not required. */
1456 STMT_VINFO_TYPE (stmt_info) = call_vec_info_type;
1457 if (vect_print_dump_info (REPORT_DETAILS))
1458 fprintf (vect_dump, "=== vectorizable_call ===");
1459 vect_model_simple_cost (stmt_info, ncopies, dt, NULL);
1460 return true;
1463 /** Transform. **/
1465 if (vect_print_dump_info (REPORT_DETAILS))
1466 fprintf (vect_dump, "transform operation.");
1468 /* Handle def. */
1469 scalar_dest = gimple_call_lhs (stmt);
1470 vec_dest = vect_create_destination_var (scalar_dest, vectype_out);
1472 prev_stmt_info = NULL;
1473 switch (modifier)
1475 case NONE:
1476 for (j = 0; j < ncopies; ++j)
1478 /* Build argument list for the vectorized call. */
1479 if (j == 0)
1480 vargs = VEC_alloc (tree, heap, nargs);
1481 else
1482 VEC_truncate (tree, vargs, 0);
1484 for (i = 0; i < nargs; i++)
1486 op = gimple_call_arg (stmt, i);
1487 if (j == 0)
1488 vec_oprnd0
1489 = vect_get_vec_def_for_operand (op, stmt, NULL);
1490 else
1492 vec_oprnd0 = gimple_call_arg (new_stmt, i);
1493 vec_oprnd0
1494 = vect_get_vec_def_for_stmt_copy (dt[i], vec_oprnd0);
1497 VEC_quick_push (tree, vargs, vec_oprnd0);
1500 new_stmt = gimple_build_call_vec (fndecl, vargs);
1501 new_temp = make_ssa_name (vec_dest, new_stmt);
1502 gimple_call_set_lhs (new_stmt, new_temp);
1504 vect_finish_stmt_generation (stmt, new_stmt, gsi);
1505 mark_symbols_for_renaming (new_stmt);
1507 if (j == 0)
1508 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
1509 else
1510 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
1512 prev_stmt_info = vinfo_for_stmt (new_stmt);
1515 break;
1517 case NARROW:
1518 for (j = 0; j < ncopies; ++j)
1520 /* Build argument list for the vectorized call. */
1521 if (j == 0)
1522 vargs = VEC_alloc (tree, heap, nargs * 2);
1523 else
1524 VEC_truncate (tree, vargs, 0);
1526 for (i = 0; i < nargs; i++)
1528 op = gimple_call_arg (stmt, i);
1529 if (j == 0)
1531 vec_oprnd0
1532 = vect_get_vec_def_for_operand (op, stmt, NULL);
1533 vec_oprnd1
1534 = vect_get_vec_def_for_stmt_copy (dt[i], vec_oprnd0);
1536 else
1538 vec_oprnd1 = gimple_call_arg (new_stmt, 2*i);
1539 vec_oprnd0
1540 = vect_get_vec_def_for_stmt_copy (dt[i], vec_oprnd1);
1541 vec_oprnd1
1542 = vect_get_vec_def_for_stmt_copy (dt[i], vec_oprnd0);
1545 VEC_quick_push (tree, vargs, vec_oprnd0);
1546 VEC_quick_push (tree, vargs, vec_oprnd1);
1549 new_stmt = gimple_build_call_vec (fndecl, vargs);
1550 new_temp = make_ssa_name (vec_dest, new_stmt);
1551 gimple_call_set_lhs (new_stmt, new_temp);
1553 vect_finish_stmt_generation (stmt, new_stmt, gsi);
1554 mark_symbols_for_renaming (new_stmt);
1556 if (j == 0)
1557 STMT_VINFO_VEC_STMT (stmt_info) = new_stmt;
1558 else
1559 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
1561 prev_stmt_info = vinfo_for_stmt (new_stmt);
1564 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
1566 break;
1568 case WIDEN:
1569 /* No current target implements this case. */
1570 return false;
1573 VEC_free (tree, heap, vargs);
1575 /* Update the exception handling table with the vector stmt if necessary. */
1576 if (maybe_clean_or_replace_eh_stmt (stmt, *vec_stmt))
1577 gimple_purge_dead_eh_edges (gimple_bb (stmt));
1579 /* The call in STMT might prevent it from being removed in dce.
1580 We however cannot remove it here, due to the way the ssa name
1581 it defines is mapped to the new definition. So just replace
1582 rhs of the statement with something harmless. */
1584 type = TREE_TYPE (scalar_dest);
1585 new_stmt = gimple_build_assign (gimple_call_lhs (stmt),
1586 fold_convert (type, integer_zero_node));
1587 set_vinfo_for_stmt (new_stmt, stmt_info);
1588 set_vinfo_for_stmt (stmt, NULL);
1589 STMT_VINFO_STMT (stmt_info) = new_stmt;
1590 gsi_replace (gsi, new_stmt, false);
1591 SSA_NAME_DEF_STMT (gimple_assign_lhs (new_stmt)) = new_stmt;
1593 return true;
1597 /* Function vect_gen_widened_results_half
1599 Create a vector stmt whose code, type, number of arguments, and result
1600 variable are CODE, OP_TYPE, and VEC_DEST, and its arguments are
1601 VEC_OPRND0 and VEC_OPRND1. The new vector stmt is to be inserted at BSI.
1602 In the case that CODE is a CALL_EXPR, this means that a call to DECL
1603 needs to be created (DECL is a function-decl of a target-builtin).
1604 STMT is the original scalar stmt that we are vectorizing. */
1606 static gimple
1607 vect_gen_widened_results_half (enum tree_code code,
1608 tree decl,
1609 tree vec_oprnd0, tree vec_oprnd1, int op_type,
1610 tree vec_dest, gimple_stmt_iterator *gsi,
1611 gimple stmt)
1613 gimple new_stmt;
1614 tree new_temp;
1616 /* Generate half of the widened result: */
1617 if (code == CALL_EXPR)
1619 /* Target specific support */
1620 if (op_type == binary_op)
1621 new_stmt = gimple_build_call (decl, 2, vec_oprnd0, vec_oprnd1);
1622 else
1623 new_stmt = gimple_build_call (decl, 1, vec_oprnd0);
1624 new_temp = make_ssa_name (vec_dest, new_stmt);
1625 gimple_call_set_lhs (new_stmt, new_temp);
1627 else
1629 /* Generic support */
1630 gcc_assert (op_type == TREE_CODE_LENGTH (code));
1631 if (op_type != binary_op)
1632 vec_oprnd1 = NULL;
1633 new_stmt = gimple_build_assign_with_ops (code, vec_dest, vec_oprnd0,
1634 vec_oprnd1);
1635 new_temp = make_ssa_name (vec_dest, new_stmt);
1636 gimple_assign_set_lhs (new_stmt, new_temp);
1638 vect_finish_stmt_generation (stmt, new_stmt, gsi);
1640 return new_stmt;
1644 /* Check if STMT performs a conversion operation, that can be vectorized.
1645 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
1646 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
1647 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
1649 static bool
1650 vectorizable_conversion (gimple stmt, gimple_stmt_iterator *gsi,
1651 gimple *vec_stmt, slp_tree slp_node)
1653 tree vec_dest;
1654 tree scalar_dest;
1655 tree op0;
1656 tree vec_oprnd0 = NULL_TREE, vec_oprnd1 = NULL_TREE;
1657 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
1658 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
1659 enum tree_code code, code1 = ERROR_MARK, code2 = ERROR_MARK;
1660 tree decl1 = NULL_TREE, decl2 = NULL_TREE;
1661 tree new_temp;
1662 tree def;
1663 gimple def_stmt;
1664 enum vect_def_type dt[2] = {vect_unknown_def_type, vect_unknown_def_type};
1665 gimple new_stmt = NULL;
1666 stmt_vec_info prev_stmt_info;
1667 int nunits_in;
1668 int nunits_out;
1669 tree vectype_out, vectype_in;
1670 int ncopies, j;
1671 tree rhs_type;
1672 tree builtin_decl;
1673 enum { NARROW, NONE, WIDEN } modifier;
1674 int i;
1675 VEC(tree,heap) *vec_oprnds0 = NULL;
1676 tree vop0;
1677 VEC(tree,heap) *dummy = NULL;
1678 int dummy_int;
1680 /* Is STMT a vectorizable conversion? */
1682 /* FORNOW: unsupported in basic block SLP. */
1683 gcc_assert (loop_vinfo);
1685 if (!STMT_VINFO_RELEVANT_P (stmt_info))
1686 return false;
1688 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
1689 return false;
1691 if (!is_gimple_assign (stmt))
1692 return false;
1694 if (TREE_CODE (gimple_assign_lhs (stmt)) != SSA_NAME)
1695 return false;
1697 code = gimple_assign_rhs_code (stmt);
1698 if (code != FIX_TRUNC_EXPR && code != FLOAT_EXPR)
1699 return false;
1701 /* Check types of lhs and rhs. */
1702 scalar_dest = gimple_assign_lhs (stmt);
1703 vectype_out = STMT_VINFO_VECTYPE (stmt_info);
1705 op0 = gimple_assign_rhs1 (stmt);
1706 rhs_type = TREE_TYPE (op0);
1707 /* Check the operands of the operation. */
1708 if (!vect_is_simple_use_1 (op0, loop_vinfo, NULL,
1709 &def_stmt, &def, &dt[0], &vectype_in))
1711 if (vect_print_dump_info (REPORT_DETAILS))
1712 fprintf (vect_dump, "use not simple.");
1713 return false;
1715 /* If op0 is an external or constant defs use a vector type of
1716 the same size as the output vector type. */
1717 if (!vectype_in)
1718 vectype_in = get_same_sized_vectype (rhs_type, vectype_out);
1719 if (vec_stmt)
1720 gcc_assert (vectype_in);
1721 if (!vectype_in)
1723 if (vect_print_dump_info (REPORT_DETAILS))
1725 fprintf (vect_dump, "no vectype for scalar type ");
1726 print_generic_expr (vect_dump, rhs_type, TDF_SLIM);
1729 return false;
1732 /* FORNOW */
1733 nunits_in = TYPE_VECTOR_SUBPARTS (vectype_in);
1734 nunits_out = TYPE_VECTOR_SUBPARTS (vectype_out);
1735 if (nunits_in == nunits_out / 2)
1736 modifier = NARROW;
1737 else if (nunits_out == nunits_in)
1738 modifier = NONE;
1739 else if (nunits_out == nunits_in / 2)
1740 modifier = WIDEN;
1741 else
1742 return false;
1744 if (modifier == NARROW)
1745 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_out;
1746 else
1747 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_in;
1749 /* Multiple types in SLP are handled by creating the appropriate number of
1750 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
1751 case of SLP. */
1752 if (slp_node)
1753 ncopies = 1;
1755 /* Sanity check: make sure that at least one copy of the vectorized stmt
1756 needs to be generated. */
1757 gcc_assert (ncopies >= 1);
1759 /* Supportable by target? */
1760 if ((modifier == NONE
1761 && !targetm.vectorize.builtin_conversion (code, vectype_out, vectype_in))
1762 || (modifier == WIDEN
1763 && !supportable_widening_operation (code, stmt,
1764 vectype_out, vectype_in,
1765 &decl1, &decl2,
1766 &code1, &code2,
1767 &dummy_int, &dummy))
1768 || (modifier == NARROW
1769 && !supportable_narrowing_operation (code, vectype_out, vectype_in,
1770 &code1, &dummy_int, &dummy)))
1772 if (vect_print_dump_info (REPORT_DETAILS))
1773 fprintf (vect_dump, "conversion not supported by target.");
1774 return false;
1777 if (modifier != NONE)
1779 /* FORNOW: SLP not supported. */
1780 if (STMT_SLP_TYPE (stmt_info))
1781 return false;
1784 if (!vec_stmt) /* transformation not required. */
1786 STMT_VINFO_TYPE (stmt_info) = type_conversion_vec_info_type;
1787 return true;
1790 /** Transform. **/
1791 if (vect_print_dump_info (REPORT_DETAILS))
1792 fprintf (vect_dump, "transform conversion.");
1794 /* Handle def. */
1795 vec_dest = vect_create_destination_var (scalar_dest, vectype_out);
1797 if (modifier == NONE && !slp_node)
1798 vec_oprnds0 = VEC_alloc (tree, heap, 1);
1800 prev_stmt_info = NULL;
1801 switch (modifier)
1803 case NONE:
1804 for (j = 0; j < ncopies; j++)
1806 if (j == 0)
1807 vect_get_vec_defs (op0, NULL, stmt, &vec_oprnds0, NULL, slp_node);
1808 else
1809 vect_get_vec_defs_for_stmt_copy (dt, &vec_oprnds0, NULL);
1811 builtin_decl =
1812 targetm.vectorize.builtin_conversion (code,
1813 vectype_out, vectype_in);
1814 FOR_EACH_VEC_ELT (tree, vec_oprnds0, i, vop0)
1816 /* Arguments are ready. create the new vector stmt. */
1817 new_stmt = gimple_build_call (builtin_decl, 1, vop0);
1818 new_temp = make_ssa_name (vec_dest, new_stmt);
1819 gimple_call_set_lhs (new_stmt, new_temp);
1820 vect_finish_stmt_generation (stmt, new_stmt, gsi);
1821 if (slp_node)
1822 VEC_quick_push (gimple, SLP_TREE_VEC_STMTS (slp_node), new_stmt);
1825 if (j == 0)
1826 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
1827 else
1828 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
1829 prev_stmt_info = vinfo_for_stmt (new_stmt);
1831 break;
1833 case WIDEN:
1834 /* In case the vectorization factor (VF) is bigger than the number
1835 of elements that we can fit in a vectype (nunits), we have to
1836 generate more than one vector stmt - i.e - we need to "unroll"
1837 the vector stmt by a factor VF/nunits. */
1838 for (j = 0; j < ncopies; j++)
1840 if (j == 0)
1841 vec_oprnd0 = vect_get_vec_def_for_operand (op0, stmt, NULL);
1842 else
1843 vec_oprnd0 = vect_get_vec_def_for_stmt_copy (dt[0], vec_oprnd0);
1845 /* Generate first half of the widened result: */
1846 new_stmt
1847 = vect_gen_widened_results_half (code1, decl1,
1848 vec_oprnd0, vec_oprnd1,
1849 unary_op, vec_dest, gsi, stmt);
1850 if (j == 0)
1851 STMT_VINFO_VEC_STMT (stmt_info) = new_stmt;
1852 else
1853 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
1854 prev_stmt_info = vinfo_for_stmt (new_stmt);
1856 /* Generate second half of the widened result: */
1857 new_stmt
1858 = vect_gen_widened_results_half (code2, decl2,
1859 vec_oprnd0, vec_oprnd1,
1860 unary_op, vec_dest, gsi, stmt);
1861 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
1862 prev_stmt_info = vinfo_for_stmt (new_stmt);
1864 break;
1866 case NARROW:
1867 /* In case the vectorization factor (VF) is bigger than the number
1868 of elements that we can fit in a vectype (nunits), we have to
1869 generate more than one vector stmt - i.e - we need to "unroll"
1870 the vector stmt by a factor VF/nunits. */
1871 for (j = 0; j < ncopies; j++)
1873 /* Handle uses. */
1874 if (j == 0)
1876 vec_oprnd0 = vect_get_vec_def_for_operand (op0, stmt, NULL);
1877 vec_oprnd1 = vect_get_vec_def_for_stmt_copy (dt[0], vec_oprnd0);
1879 else
1881 vec_oprnd0 = vect_get_vec_def_for_stmt_copy (dt[0], vec_oprnd1);
1882 vec_oprnd1 = vect_get_vec_def_for_stmt_copy (dt[0], vec_oprnd0);
1885 /* Arguments are ready. Create the new vector stmt. */
1886 new_stmt = gimple_build_assign_with_ops (code1, vec_dest, vec_oprnd0,
1887 vec_oprnd1);
1888 new_temp = make_ssa_name (vec_dest, new_stmt);
1889 gimple_assign_set_lhs (new_stmt, new_temp);
1890 vect_finish_stmt_generation (stmt, new_stmt, gsi);
1892 if (j == 0)
1893 STMT_VINFO_VEC_STMT (stmt_info) = new_stmt;
1894 else
1895 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
1897 prev_stmt_info = vinfo_for_stmt (new_stmt);
1900 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
1903 if (vec_oprnds0)
1904 VEC_free (tree, heap, vec_oprnds0);
1906 return true;
1910 /* Function vectorizable_assignment.
1912 Check if STMT performs an assignment (copy) that can be vectorized.
1913 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
1914 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
1915 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
1917 static bool
1918 vectorizable_assignment (gimple stmt, gimple_stmt_iterator *gsi,
1919 gimple *vec_stmt, slp_tree slp_node)
1921 tree vec_dest;
1922 tree scalar_dest;
1923 tree op;
1924 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
1925 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
1926 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
1927 tree new_temp;
1928 tree def;
1929 gimple def_stmt;
1930 enum vect_def_type dt[2] = {vect_unknown_def_type, vect_unknown_def_type};
1931 unsigned int nunits = TYPE_VECTOR_SUBPARTS (vectype);
1932 int ncopies;
1933 int i, j;
1934 VEC(tree,heap) *vec_oprnds = NULL;
1935 tree vop;
1936 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
1937 gimple new_stmt = NULL;
1938 stmt_vec_info prev_stmt_info = NULL;
1939 enum tree_code code;
1940 tree vectype_in;
1942 /* Multiple types in SLP are handled by creating the appropriate number of
1943 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
1944 case of SLP. */
1945 if (slp_node)
1946 ncopies = 1;
1947 else
1948 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits;
1950 gcc_assert (ncopies >= 1);
1952 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
1953 return false;
1955 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
1956 return false;
1958 /* Is vectorizable assignment? */
1959 if (!is_gimple_assign (stmt))
1960 return false;
1962 scalar_dest = gimple_assign_lhs (stmt);
1963 if (TREE_CODE (scalar_dest) != SSA_NAME)
1964 return false;
1966 code = gimple_assign_rhs_code (stmt);
1967 if (gimple_assign_single_p (stmt)
1968 || code == PAREN_EXPR
1969 || CONVERT_EXPR_CODE_P (code))
1970 op = gimple_assign_rhs1 (stmt);
1971 else
1972 return false;
1974 if (!vect_is_simple_use_1 (op, loop_vinfo, bb_vinfo,
1975 &def_stmt, &def, &dt[0], &vectype_in))
1977 if (vect_print_dump_info (REPORT_DETAILS))
1978 fprintf (vect_dump, "use not simple.");
1979 return false;
1982 /* We can handle NOP_EXPR conversions that do not change the number
1983 of elements or the vector size. */
1984 if (CONVERT_EXPR_CODE_P (code)
1985 && (!vectype_in
1986 || TYPE_VECTOR_SUBPARTS (vectype_in) != nunits
1987 || (GET_MODE_SIZE (TYPE_MODE (vectype))
1988 != GET_MODE_SIZE (TYPE_MODE (vectype_in)))))
1989 return false;
1991 if (!vec_stmt) /* transformation not required. */
1993 STMT_VINFO_TYPE (stmt_info) = assignment_vec_info_type;
1994 if (vect_print_dump_info (REPORT_DETAILS))
1995 fprintf (vect_dump, "=== vectorizable_assignment ===");
1996 vect_model_simple_cost (stmt_info, ncopies, dt, NULL);
1997 return true;
2000 /** Transform. **/
2001 if (vect_print_dump_info (REPORT_DETAILS))
2002 fprintf (vect_dump, "transform assignment.");
2004 /* Handle def. */
2005 vec_dest = vect_create_destination_var (scalar_dest, vectype);
2007 /* Handle use. */
2008 for (j = 0; j < ncopies; j++)
2010 /* Handle uses. */
2011 if (j == 0)
2012 vect_get_vec_defs (op, NULL, stmt, &vec_oprnds, NULL, slp_node);
2013 else
2014 vect_get_vec_defs_for_stmt_copy (dt, &vec_oprnds, NULL);
2016 /* Arguments are ready. create the new vector stmt. */
2017 FOR_EACH_VEC_ELT (tree, vec_oprnds, i, vop)
2019 if (CONVERT_EXPR_CODE_P (code))
2020 vop = build1 (VIEW_CONVERT_EXPR, vectype, vop);
2021 new_stmt = gimple_build_assign (vec_dest, vop);
2022 new_temp = make_ssa_name (vec_dest, new_stmt);
2023 gimple_assign_set_lhs (new_stmt, new_temp);
2024 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2025 if (slp_node)
2026 VEC_quick_push (gimple, SLP_TREE_VEC_STMTS (slp_node), new_stmt);
2029 if (slp_node)
2030 continue;
2032 if (j == 0)
2033 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
2034 else
2035 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
2037 prev_stmt_info = vinfo_for_stmt (new_stmt);
2040 VEC_free (tree, heap, vec_oprnds);
2041 return true;
2044 /* Function vectorizable_operation.
2046 Check if STMT performs a binary or unary operation that can be vectorized.
2047 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
2048 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
2049 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
2051 static bool
2052 vectorizable_operation (gimple stmt, gimple_stmt_iterator *gsi,
2053 gimple *vec_stmt, slp_tree slp_node)
2055 tree vec_dest;
2056 tree scalar_dest;
2057 tree op0, op1 = NULL;
2058 tree vec_oprnd1 = NULL_TREE;
2059 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
2060 tree vectype;
2061 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
2062 enum tree_code code;
2063 enum machine_mode vec_mode;
2064 tree new_temp;
2065 int op_type;
2066 optab optab;
2067 int icode;
2068 enum machine_mode optab_op2_mode;
2069 tree def;
2070 gimple def_stmt;
2071 enum vect_def_type dt[2] = {vect_unknown_def_type, vect_unknown_def_type};
2072 gimple new_stmt = NULL;
2073 stmt_vec_info prev_stmt_info;
2074 int nunits_in;
2075 int nunits_out;
2076 tree vectype_out;
2077 int ncopies;
2078 int j, i;
2079 VEC(tree,heap) *vec_oprnds0 = NULL, *vec_oprnds1 = NULL;
2080 tree vop0, vop1;
2081 unsigned int k;
2082 bool scalar_shift_arg = false;
2083 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
2084 int vf;
2086 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
2087 return false;
2089 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
2090 return false;
2092 /* Is STMT a vectorizable binary/unary operation? */
2093 if (!is_gimple_assign (stmt))
2094 return false;
2096 if (TREE_CODE (gimple_assign_lhs (stmt)) != SSA_NAME)
2097 return false;
2099 code = gimple_assign_rhs_code (stmt);
2101 /* For pointer addition, we should use the normal plus for
2102 the vector addition. */
2103 if (code == POINTER_PLUS_EXPR)
2104 code = PLUS_EXPR;
2106 /* Support only unary or binary operations. */
2107 op_type = TREE_CODE_LENGTH (code);
2108 if (op_type != unary_op && op_type != binary_op)
2110 if (vect_print_dump_info (REPORT_DETAILS))
2111 fprintf (vect_dump, "num. args = %d (not unary/binary op).", op_type);
2112 return false;
2115 scalar_dest = gimple_assign_lhs (stmt);
2116 vectype_out = STMT_VINFO_VECTYPE (stmt_info);
2118 op0 = gimple_assign_rhs1 (stmt);
2119 if (!vect_is_simple_use_1 (op0, loop_vinfo, bb_vinfo,
2120 &def_stmt, &def, &dt[0], &vectype))
2122 if (vect_print_dump_info (REPORT_DETAILS))
2123 fprintf (vect_dump, "use not simple.");
2124 return false;
2126 /* If op0 is an external or constant def use a vector type with
2127 the same size as the output vector type. */
2128 if (!vectype)
2129 vectype = get_same_sized_vectype (TREE_TYPE (op0), vectype_out);
2130 if (vec_stmt)
2131 gcc_assert (vectype);
2132 if (!vectype)
2134 if (vect_print_dump_info (REPORT_DETAILS))
2136 fprintf (vect_dump, "no vectype for scalar type ");
2137 print_generic_expr (vect_dump, TREE_TYPE (op0), TDF_SLIM);
2140 return false;
2143 nunits_out = TYPE_VECTOR_SUBPARTS (vectype_out);
2144 nunits_in = TYPE_VECTOR_SUBPARTS (vectype);
2145 if (nunits_out != nunits_in)
2146 return false;
2148 if (op_type == binary_op)
2150 op1 = gimple_assign_rhs2 (stmt);
2151 if (!vect_is_simple_use (op1, loop_vinfo, bb_vinfo, &def_stmt, &def,
2152 &dt[1]))
2154 if (vect_print_dump_info (REPORT_DETAILS))
2155 fprintf (vect_dump, "use not simple.");
2156 return false;
2160 if (loop_vinfo)
2161 vf = LOOP_VINFO_VECT_FACTOR (loop_vinfo);
2162 else
2163 vf = 1;
2165 /* Multiple types in SLP are handled by creating the appropriate number of
2166 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
2167 case of SLP. */
2168 if (slp_node)
2169 ncopies = 1;
2170 else
2171 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_in;
2173 gcc_assert (ncopies >= 1);
2175 /* If this is a shift/rotate, determine whether the shift amount is a vector,
2176 or scalar. If the shift/rotate amount is a vector, use the vector/vector
2177 shift optabs. */
2178 if (code == LSHIFT_EXPR || code == RSHIFT_EXPR || code == LROTATE_EXPR
2179 || code == RROTATE_EXPR)
2181 /* vector shifted by vector */
2182 if (dt[1] == vect_internal_def)
2184 optab = optab_for_tree_code (code, vectype, optab_vector);
2185 if (vect_print_dump_info (REPORT_DETAILS))
2186 fprintf (vect_dump, "vector/vector shift/rotate found.");
2189 /* See if the machine has a vector shifted by scalar insn and if not
2190 then see if it has a vector shifted by vector insn */
2191 else if (dt[1] == vect_constant_def || dt[1] == vect_external_def)
2193 optab = optab_for_tree_code (code, vectype, optab_scalar);
2194 if (optab
2195 && optab_handler (optab, TYPE_MODE (vectype)) != CODE_FOR_nothing)
2197 scalar_shift_arg = true;
2198 if (vect_print_dump_info (REPORT_DETAILS))
2199 fprintf (vect_dump, "vector/scalar shift/rotate found.");
2201 else
2203 optab = optab_for_tree_code (code, vectype, optab_vector);
2204 if (optab
2205 && (optab_handler (optab, TYPE_MODE (vectype))
2206 != CODE_FOR_nothing))
2208 if (vect_print_dump_info (REPORT_DETAILS))
2209 fprintf (vect_dump, "vector/vector shift/rotate found.");
2211 /* Unlike the other binary operators, shifts/rotates have
2212 the rhs being int, instead of the same type as the lhs,
2213 so make sure the scalar is the right type if we are
2214 dealing with vectors of short/char. */
2215 if (dt[1] == vect_constant_def)
2216 op1 = fold_convert (TREE_TYPE (vectype), op1);
2221 else
2223 if (vect_print_dump_info (REPORT_DETAILS))
2224 fprintf (vect_dump, "operand mode requires invariant argument.");
2225 return false;
2228 else
2229 optab = optab_for_tree_code (code, vectype, optab_default);
2231 /* Supportable by target? */
2232 if (!optab)
2234 if (vect_print_dump_info (REPORT_DETAILS))
2235 fprintf (vect_dump, "no optab.");
2236 return false;
2238 vec_mode = TYPE_MODE (vectype);
2239 icode = (int) optab_handler (optab, vec_mode);
2240 if (icode == CODE_FOR_nothing)
2242 if (vect_print_dump_info (REPORT_DETAILS))
2243 fprintf (vect_dump, "op not supported by target.");
2244 /* Check only during analysis. */
2245 if (GET_MODE_SIZE (vec_mode) != UNITS_PER_WORD
2246 || (vf < vect_min_worthwhile_factor (code)
2247 && !vec_stmt))
2248 return false;
2249 if (vect_print_dump_info (REPORT_DETAILS))
2250 fprintf (vect_dump, "proceeding using word mode.");
2253 /* Worthwhile without SIMD support? Check only during analysis. */
2254 if (!VECTOR_MODE_P (TYPE_MODE (vectype))
2255 && vf < vect_min_worthwhile_factor (code)
2256 && !vec_stmt)
2258 if (vect_print_dump_info (REPORT_DETAILS))
2259 fprintf (vect_dump, "not worthwhile without SIMD support.");
2260 return false;
2263 if (!vec_stmt) /* transformation not required. */
2265 STMT_VINFO_TYPE (stmt_info) = op_vec_info_type;
2266 if (vect_print_dump_info (REPORT_DETAILS))
2267 fprintf (vect_dump, "=== vectorizable_operation ===");
2268 vect_model_simple_cost (stmt_info, ncopies, dt, NULL);
2269 return true;
2272 /** Transform. **/
2274 if (vect_print_dump_info (REPORT_DETAILS))
2275 fprintf (vect_dump, "transform binary/unary operation.");
2277 /* Handle def. */
2278 vec_dest = vect_create_destination_var (scalar_dest, vectype);
2280 /* Allocate VECs for vector operands. In case of SLP, vector operands are
2281 created in the previous stages of the recursion, so no allocation is
2282 needed, except for the case of shift with scalar shift argument. In that
2283 case we store the scalar operand in VEC_OPRNDS1 for every vector stmt to
2284 be created to vectorize the SLP group, i.e., SLP_NODE->VEC_STMTS_SIZE.
2285 In case of loop-based vectorization we allocate VECs of size 1. We
2286 allocate VEC_OPRNDS1 only in case of binary operation. */
2287 if (!slp_node)
2289 vec_oprnds0 = VEC_alloc (tree, heap, 1);
2290 if (op_type == binary_op)
2291 vec_oprnds1 = VEC_alloc (tree, heap, 1);
2293 else if (scalar_shift_arg)
2294 vec_oprnds1 = VEC_alloc (tree, heap, slp_node->vec_stmts_size);
2296 /* In case the vectorization factor (VF) is bigger than the number
2297 of elements that we can fit in a vectype (nunits), we have to generate
2298 more than one vector stmt - i.e - we need to "unroll" the
2299 vector stmt by a factor VF/nunits. In doing so, we record a pointer
2300 from one copy of the vector stmt to the next, in the field
2301 STMT_VINFO_RELATED_STMT. This is necessary in order to allow following
2302 stages to find the correct vector defs to be used when vectorizing
2303 stmts that use the defs of the current stmt. The example below
2304 illustrates the vectorization process when VF=16 and nunits=4 (i.e.,
2305 we need to create 4 vectorized stmts):
2307 before vectorization:
2308 RELATED_STMT VEC_STMT
2309 S1: x = memref - -
2310 S2: z = x + 1 - -
2312 step 1: vectorize stmt S1 (done in vectorizable_load. See more details
2313 there):
2314 RELATED_STMT VEC_STMT
2315 VS1_0: vx0 = memref0 VS1_1 -
2316 VS1_1: vx1 = memref1 VS1_2 -
2317 VS1_2: vx2 = memref2 VS1_3 -
2318 VS1_3: vx3 = memref3 - -
2319 S1: x = load - VS1_0
2320 S2: z = x + 1 - -
2322 step2: vectorize stmt S2 (done here):
2323 To vectorize stmt S2 we first need to find the relevant vector
2324 def for the first operand 'x'. This is, as usual, obtained from
2325 the vector stmt recorded in the STMT_VINFO_VEC_STMT of the stmt
2326 that defines 'x' (S1). This way we find the stmt VS1_0, and the
2327 relevant vector def 'vx0'. Having found 'vx0' we can generate
2328 the vector stmt VS2_0, and as usual, record it in the
2329 STMT_VINFO_VEC_STMT of stmt S2.
2330 When creating the second copy (VS2_1), we obtain the relevant vector
2331 def from the vector stmt recorded in the STMT_VINFO_RELATED_STMT of
2332 stmt VS1_0. This way we find the stmt VS1_1 and the relevant
2333 vector def 'vx1'. Using 'vx1' we create stmt VS2_1 and record a
2334 pointer to it in the STMT_VINFO_RELATED_STMT of the vector stmt VS2_0.
2335 Similarly when creating stmts VS2_2 and VS2_3. This is the resulting
2336 chain of stmts and pointers:
2337 RELATED_STMT VEC_STMT
2338 VS1_0: vx0 = memref0 VS1_1 -
2339 VS1_1: vx1 = memref1 VS1_2 -
2340 VS1_2: vx2 = memref2 VS1_3 -
2341 VS1_3: vx3 = memref3 - -
2342 S1: x = load - VS1_0
2343 VS2_0: vz0 = vx0 + v1 VS2_1 -
2344 VS2_1: vz1 = vx1 + v1 VS2_2 -
2345 VS2_2: vz2 = vx2 + v1 VS2_3 -
2346 VS2_3: vz3 = vx3 + v1 - -
2347 S2: z = x + 1 - VS2_0 */
2349 prev_stmt_info = NULL;
2350 for (j = 0; j < ncopies; j++)
2352 /* Handle uses. */
2353 if (j == 0)
2355 if (op_type == binary_op && scalar_shift_arg)
2357 /* Vector shl and shr insn patterns can be defined with scalar
2358 operand 2 (shift operand). In this case, use constant or loop
2359 invariant op1 directly, without extending it to vector mode
2360 first. */
2361 optab_op2_mode = insn_data[icode].operand[2].mode;
2362 if (!VECTOR_MODE_P (optab_op2_mode))
2364 if (vect_print_dump_info (REPORT_DETAILS))
2365 fprintf (vect_dump, "operand 1 using scalar mode.");
2366 vec_oprnd1 = op1;
2367 VEC_quick_push (tree, vec_oprnds1, vec_oprnd1);
2368 if (slp_node)
2370 /* Store vec_oprnd1 for every vector stmt to be created
2371 for SLP_NODE. We check during the analysis that all
2372 the shift arguments are the same.
2373 TODO: Allow different constants for different vector
2374 stmts generated for an SLP instance. */
2375 for (k = 0; k < slp_node->vec_stmts_size - 1; k++)
2376 VEC_quick_push (tree, vec_oprnds1, vec_oprnd1);
2381 /* vec_oprnd1 is available if operand 1 should be of a scalar-type
2382 (a special case for certain kind of vector shifts); otherwise,
2383 operand 1 should be of a vector type (the usual case). */
2384 if (op_type == binary_op && !vec_oprnd1)
2385 vect_get_vec_defs (op0, op1, stmt, &vec_oprnds0, &vec_oprnds1,
2386 slp_node);
2387 else
2388 vect_get_vec_defs (op0, NULL_TREE, stmt, &vec_oprnds0, NULL,
2389 slp_node);
2391 else
2392 vect_get_vec_defs_for_stmt_copy (dt, &vec_oprnds0, &vec_oprnds1);
2394 /* Arguments are ready. Create the new vector stmt. */
2395 FOR_EACH_VEC_ELT (tree, vec_oprnds0, i, vop0)
2397 vop1 = ((op_type == binary_op)
2398 ? VEC_index (tree, vec_oprnds1, i) : NULL);
2399 new_stmt = gimple_build_assign_with_ops (code, vec_dest, vop0, vop1);
2400 new_temp = make_ssa_name (vec_dest, new_stmt);
2401 gimple_assign_set_lhs (new_stmt, new_temp);
2402 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2403 if (slp_node)
2404 VEC_quick_push (gimple, SLP_TREE_VEC_STMTS (slp_node), new_stmt);
2407 if (slp_node)
2408 continue;
2410 if (j == 0)
2411 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
2412 else
2413 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
2414 prev_stmt_info = vinfo_for_stmt (new_stmt);
2417 VEC_free (tree, heap, vec_oprnds0);
2418 if (vec_oprnds1)
2419 VEC_free (tree, heap, vec_oprnds1);
2421 return true;
2425 /* Get vectorized definitions for loop-based vectorization. For the first
2426 operand we call vect_get_vec_def_for_operand() (with OPRND containing
2427 scalar operand), and for the rest we get a copy with
2428 vect_get_vec_def_for_stmt_copy() using the previous vector definition
2429 (stored in OPRND). See vect_get_vec_def_for_stmt_copy() for details.
2430 The vectors are collected into VEC_OPRNDS. */
2432 static void
2433 vect_get_loop_based_defs (tree *oprnd, gimple stmt, enum vect_def_type dt,
2434 VEC (tree, heap) **vec_oprnds, int multi_step_cvt)
2436 tree vec_oprnd;
2438 /* Get first vector operand. */
2439 /* All the vector operands except the very first one (that is scalar oprnd)
2440 are stmt copies. */
2441 if (TREE_CODE (TREE_TYPE (*oprnd)) != VECTOR_TYPE)
2442 vec_oprnd = vect_get_vec_def_for_operand (*oprnd, stmt, NULL);
2443 else
2444 vec_oprnd = vect_get_vec_def_for_stmt_copy (dt, *oprnd);
2446 VEC_quick_push (tree, *vec_oprnds, vec_oprnd);
2448 /* Get second vector operand. */
2449 vec_oprnd = vect_get_vec_def_for_stmt_copy (dt, vec_oprnd);
2450 VEC_quick_push (tree, *vec_oprnds, vec_oprnd);
2452 *oprnd = vec_oprnd;
2454 /* For conversion in multiple steps, continue to get operands
2455 recursively. */
2456 if (multi_step_cvt)
2457 vect_get_loop_based_defs (oprnd, stmt, dt, vec_oprnds, multi_step_cvt - 1);
2461 /* Create vectorized demotion statements for vector operands from VEC_OPRNDS.
2462 For multi-step conversions store the resulting vectors and call the function
2463 recursively. */
2465 static void
2466 vect_create_vectorized_demotion_stmts (VEC (tree, heap) **vec_oprnds,
2467 int multi_step_cvt, gimple stmt,
2468 VEC (tree, heap) *vec_dsts,
2469 gimple_stmt_iterator *gsi,
2470 slp_tree slp_node, enum tree_code code,
2471 stmt_vec_info *prev_stmt_info)
2473 unsigned int i;
2474 tree vop0, vop1, new_tmp, vec_dest;
2475 gimple new_stmt;
2476 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
2478 vec_dest = VEC_pop (tree, vec_dsts);
2480 for (i = 0; i < VEC_length (tree, *vec_oprnds); i += 2)
2482 /* Create demotion operation. */
2483 vop0 = VEC_index (tree, *vec_oprnds, i);
2484 vop1 = VEC_index (tree, *vec_oprnds, i + 1);
2485 new_stmt = gimple_build_assign_with_ops (code, vec_dest, vop0, vop1);
2486 new_tmp = make_ssa_name (vec_dest, new_stmt);
2487 gimple_assign_set_lhs (new_stmt, new_tmp);
2488 vect_finish_stmt_generation (stmt, new_stmt, gsi);
2490 if (multi_step_cvt)
2491 /* Store the resulting vector for next recursive call. */
2492 VEC_replace (tree, *vec_oprnds, i/2, new_tmp);
2493 else
2495 /* This is the last step of the conversion sequence. Store the
2496 vectors in SLP_NODE or in vector info of the scalar statement
2497 (or in STMT_VINFO_RELATED_STMT chain). */
2498 if (slp_node)
2499 VEC_quick_push (gimple, SLP_TREE_VEC_STMTS (slp_node), new_stmt);
2500 else
2502 if (!*prev_stmt_info)
2503 STMT_VINFO_VEC_STMT (stmt_info) = new_stmt;
2504 else
2505 STMT_VINFO_RELATED_STMT (*prev_stmt_info) = new_stmt;
2507 *prev_stmt_info = vinfo_for_stmt (new_stmt);
2512 /* For multi-step demotion operations we first generate demotion operations
2513 from the source type to the intermediate types, and then combine the
2514 results (stored in VEC_OPRNDS) in demotion operation to the destination
2515 type. */
2516 if (multi_step_cvt)
2518 /* At each level of recursion we have have of the operands we had at the
2519 previous level. */
2520 VEC_truncate (tree, *vec_oprnds, (i+1)/2);
2521 vect_create_vectorized_demotion_stmts (vec_oprnds, multi_step_cvt - 1,
2522 stmt, vec_dsts, gsi, slp_node,
2523 code, prev_stmt_info);
2528 /* Function vectorizable_type_demotion
2530 Check if STMT performs a binary or unary operation that involves
2531 type demotion, and if it can be vectorized.
2532 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
2533 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
2534 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
2536 static bool
2537 vectorizable_type_demotion (gimple stmt, gimple_stmt_iterator *gsi,
2538 gimple *vec_stmt, slp_tree slp_node)
2540 tree vec_dest;
2541 tree scalar_dest;
2542 tree op0;
2543 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
2544 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
2545 enum tree_code code, code1 = ERROR_MARK;
2546 tree def;
2547 gimple def_stmt;
2548 enum vect_def_type dt[2] = {vect_unknown_def_type, vect_unknown_def_type};
2549 stmt_vec_info prev_stmt_info;
2550 int nunits_in;
2551 int nunits_out;
2552 tree vectype_out;
2553 int ncopies;
2554 int j, i;
2555 tree vectype_in;
2556 int multi_step_cvt = 0;
2557 VEC (tree, heap) *vec_oprnds0 = NULL;
2558 VEC (tree, heap) *vec_dsts = NULL, *interm_types = NULL, *tmp_vec_dsts = NULL;
2559 tree last_oprnd, intermediate_type;
2561 /* FORNOW: not supported by basic block SLP vectorization. */
2562 gcc_assert (loop_vinfo);
2564 if (!STMT_VINFO_RELEVANT_P (stmt_info))
2565 return false;
2567 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
2568 return false;
2570 /* Is STMT a vectorizable type-demotion operation? */
2571 if (!is_gimple_assign (stmt))
2572 return false;
2574 if (TREE_CODE (gimple_assign_lhs (stmt)) != SSA_NAME)
2575 return false;
2577 code = gimple_assign_rhs_code (stmt);
2578 if (!CONVERT_EXPR_CODE_P (code))
2579 return false;
2581 scalar_dest = gimple_assign_lhs (stmt);
2582 vectype_out = STMT_VINFO_VECTYPE (stmt_info);
2584 /* Check the operands of the operation. */
2585 op0 = gimple_assign_rhs1 (stmt);
2586 if (! ((INTEGRAL_TYPE_P (TREE_TYPE (scalar_dest))
2587 && INTEGRAL_TYPE_P (TREE_TYPE (op0)))
2588 || (SCALAR_FLOAT_TYPE_P (TREE_TYPE (scalar_dest))
2589 && SCALAR_FLOAT_TYPE_P (TREE_TYPE (op0))
2590 && CONVERT_EXPR_CODE_P (code))))
2591 return false;
2592 if (!vect_is_simple_use_1 (op0, loop_vinfo, NULL,
2593 &def_stmt, &def, &dt[0], &vectype_in))
2595 if (vect_print_dump_info (REPORT_DETAILS))
2596 fprintf (vect_dump, "use not simple.");
2597 return false;
2599 /* If op0 is an external def use a vector type with the
2600 same size as the output vector type if possible. */
2601 if (!vectype_in)
2602 vectype_in = get_same_sized_vectype (TREE_TYPE (op0), vectype_out);
2603 if (vec_stmt)
2604 gcc_assert (vectype_in);
2605 if (!vectype_in)
2607 if (vect_print_dump_info (REPORT_DETAILS))
2609 fprintf (vect_dump, "no vectype for scalar type ");
2610 print_generic_expr (vect_dump, TREE_TYPE (op0), TDF_SLIM);
2613 return false;
2616 nunits_in = TYPE_VECTOR_SUBPARTS (vectype_in);
2617 nunits_out = TYPE_VECTOR_SUBPARTS (vectype_out);
2618 if (nunits_in >= nunits_out)
2619 return false;
2621 /* Multiple types in SLP are handled by creating the appropriate number of
2622 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
2623 case of SLP. */
2624 if (slp_node)
2625 ncopies = 1;
2626 else
2627 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_out;
2628 gcc_assert (ncopies >= 1);
2630 /* Supportable by target? */
2631 if (!supportable_narrowing_operation (code, vectype_out, vectype_in,
2632 &code1, &multi_step_cvt, &interm_types))
2633 return false;
2635 if (!vec_stmt) /* transformation not required. */
2637 STMT_VINFO_TYPE (stmt_info) = type_demotion_vec_info_type;
2638 if (vect_print_dump_info (REPORT_DETAILS))
2639 fprintf (vect_dump, "=== vectorizable_demotion ===");
2640 vect_model_simple_cost (stmt_info, ncopies, dt, NULL);
2641 return true;
2644 /** Transform. **/
2645 if (vect_print_dump_info (REPORT_DETAILS))
2646 fprintf (vect_dump, "transform type demotion operation. ncopies = %d.",
2647 ncopies);
2649 /* In case of multi-step demotion, we first generate demotion operations to
2650 the intermediate types, and then from that types to the final one.
2651 We create vector destinations for the intermediate type (TYPES) received
2652 from supportable_narrowing_operation, and store them in the correct order
2653 for future use in vect_create_vectorized_demotion_stmts(). */
2654 if (multi_step_cvt)
2655 vec_dsts = VEC_alloc (tree, heap, multi_step_cvt + 1);
2656 else
2657 vec_dsts = VEC_alloc (tree, heap, 1);
2659 vec_dest = vect_create_destination_var (scalar_dest, vectype_out);
2660 VEC_quick_push (tree, vec_dsts, vec_dest);
2662 if (multi_step_cvt)
2664 for (i = VEC_length (tree, interm_types) - 1;
2665 VEC_iterate (tree, interm_types, i, intermediate_type); i--)
2667 vec_dest = vect_create_destination_var (scalar_dest,
2668 intermediate_type);
2669 VEC_quick_push (tree, vec_dsts, vec_dest);
2673 /* In case the vectorization factor (VF) is bigger than the number
2674 of elements that we can fit in a vectype (nunits), we have to generate
2675 more than one vector stmt - i.e - we need to "unroll" the
2676 vector stmt by a factor VF/nunits. */
2677 last_oprnd = op0;
2678 prev_stmt_info = NULL;
2679 for (j = 0; j < ncopies; j++)
2681 /* Handle uses. */
2682 if (slp_node)
2683 vect_get_slp_defs (slp_node, &vec_oprnds0, NULL, -1);
2684 else
2686 VEC_free (tree, heap, vec_oprnds0);
2687 vec_oprnds0 = VEC_alloc (tree, heap,
2688 (multi_step_cvt ? vect_pow2 (multi_step_cvt) * 2 : 2));
2689 vect_get_loop_based_defs (&last_oprnd, stmt, dt[0], &vec_oprnds0,
2690 vect_pow2 (multi_step_cvt) - 1);
2693 /* Arguments are ready. Create the new vector stmts. */
2694 tmp_vec_dsts = VEC_copy (tree, heap, vec_dsts);
2695 vect_create_vectorized_demotion_stmts (&vec_oprnds0,
2696 multi_step_cvt, stmt, tmp_vec_dsts,
2697 gsi, slp_node, code1,
2698 &prev_stmt_info);
2701 VEC_free (tree, heap, vec_oprnds0);
2702 VEC_free (tree, heap, vec_dsts);
2703 VEC_free (tree, heap, tmp_vec_dsts);
2704 VEC_free (tree, heap, interm_types);
2706 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
2707 return true;
2711 /* Create vectorized promotion statements for vector operands from VEC_OPRNDS0
2712 and VEC_OPRNDS1 (for binary operations). For multi-step conversions store
2713 the resulting vectors and call the function recursively. */
2715 static void
2716 vect_create_vectorized_promotion_stmts (VEC (tree, heap) **vec_oprnds0,
2717 VEC (tree, heap) **vec_oprnds1,
2718 int multi_step_cvt, gimple stmt,
2719 VEC (tree, heap) *vec_dsts,
2720 gimple_stmt_iterator *gsi,
2721 slp_tree slp_node, enum tree_code code1,
2722 enum tree_code code2, tree decl1,
2723 tree decl2, int op_type,
2724 stmt_vec_info *prev_stmt_info)
2726 int i;
2727 tree vop0, vop1, new_tmp1, new_tmp2, vec_dest;
2728 gimple new_stmt1, new_stmt2;
2729 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
2730 VEC (tree, heap) *vec_tmp;
2732 vec_dest = VEC_pop (tree, vec_dsts);
2733 vec_tmp = VEC_alloc (tree, heap, VEC_length (tree, *vec_oprnds0) * 2);
2735 FOR_EACH_VEC_ELT (tree, *vec_oprnds0, i, vop0)
2737 if (op_type == binary_op)
2738 vop1 = VEC_index (tree, *vec_oprnds1, i);
2739 else
2740 vop1 = NULL_TREE;
2742 /* Generate the two halves of promotion operation. */
2743 new_stmt1 = vect_gen_widened_results_half (code1, decl1, vop0, vop1,
2744 op_type, vec_dest, gsi, stmt);
2745 new_stmt2 = vect_gen_widened_results_half (code2, decl2, vop0, vop1,
2746 op_type, vec_dest, gsi, stmt);
2747 if (is_gimple_call (new_stmt1))
2749 new_tmp1 = gimple_call_lhs (new_stmt1);
2750 new_tmp2 = gimple_call_lhs (new_stmt2);
2752 else
2754 new_tmp1 = gimple_assign_lhs (new_stmt1);
2755 new_tmp2 = gimple_assign_lhs (new_stmt2);
2758 if (multi_step_cvt)
2760 /* Store the results for the recursive call. */
2761 VEC_quick_push (tree, vec_tmp, new_tmp1);
2762 VEC_quick_push (tree, vec_tmp, new_tmp2);
2764 else
2766 /* Last step of promotion sequience - store the results. */
2767 if (slp_node)
2769 VEC_quick_push (gimple, SLP_TREE_VEC_STMTS (slp_node), new_stmt1);
2770 VEC_quick_push (gimple, SLP_TREE_VEC_STMTS (slp_node), new_stmt2);
2772 else
2774 if (!*prev_stmt_info)
2775 STMT_VINFO_VEC_STMT (stmt_info) = new_stmt1;
2776 else
2777 STMT_VINFO_RELATED_STMT (*prev_stmt_info) = new_stmt1;
2779 *prev_stmt_info = vinfo_for_stmt (new_stmt1);
2780 STMT_VINFO_RELATED_STMT (*prev_stmt_info) = new_stmt2;
2781 *prev_stmt_info = vinfo_for_stmt (new_stmt2);
2786 if (multi_step_cvt)
2788 /* For multi-step promotion operation we first generate we call the
2789 function recurcively for every stage. We start from the input type,
2790 create promotion operations to the intermediate types, and then
2791 create promotions to the output type. */
2792 *vec_oprnds0 = VEC_copy (tree, heap, vec_tmp);
2793 vect_create_vectorized_promotion_stmts (vec_oprnds0, vec_oprnds1,
2794 multi_step_cvt - 1, stmt,
2795 vec_dsts, gsi, slp_node, code1,
2796 code2, decl2, decl2, op_type,
2797 prev_stmt_info);
2800 VEC_free (tree, heap, vec_tmp);
2804 /* Function vectorizable_type_promotion
2806 Check if STMT performs a binary or unary operation that involves
2807 type promotion, and if it can be vectorized.
2808 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
2809 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
2810 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
2812 static bool
2813 vectorizable_type_promotion (gimple stmt, gimple_stmt_iterator *gsi,
2814 gimple *vec_stmt, slp_tree slp_node)
2816 tree vec_dest;
2817 tree scalar_dest;
2818 tree op0, op1 = NULL;
2819 tree vec_oprnd0=NULL, vec_oprnd1=NULL;
2820 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
2821 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
2822 enum tree_code code, code1 = ERROR_MARK, code2 = ERROR_MARK;
2823 tree decl1 = NULL_TREE, decl2 = NULL_TREE;
2824 int op_type;
2825 tree def;
2826 gimple def_stmt;
2827 enum vect_def_type dt[2] = {vect_unknown_def_type, vect_unknown_def_type};
2828 stmt_vec_info prev_stmt_info;
2829 int nunits_in;
2830 int nunits_out;
2831 tree vectype_out;
2832 int ncopies;
2833 int j, i;
2834 tree vectype_in;
2835 tree intermediate_type = NULL_TREE;
2836 int multi_step_cvt = 0;
2837 VEC (tree, heap) *vec_oprnds0 = NULL, *vec_oprnds1 = NULL;
2838 VEC (tree, heap) *vec_dsts = NULL, *interm_types = NULL, *tmp_vec_dsts = NULL;
2840 /* FORNOW: not supported by basic block SLP vectorization. */
2841 gcc_assert (loop_vinfo);
2843 if (!STMT_VINFO_RELEVANT_P (stmt_info))
2844 return false;
2846 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
2847 return false;
2849 /* Is STMT a vectorizable type-promotion operation? */
2850 if (!is_gimple_assign (stmt))
2851 return false;
2853 if (TREE_CODE (gimple_assign_lhs (stmt)) != SSA_NAME)
2854 return false;
2856 code = gimple_assign_rhs_code (stmt);
2857 if (!CONVERT_EXPR_CODE_P (code)
2858 && code != WIDEN_MULT_EXPR)
2859 return false;
2861 scalar_dest = gimple_assign_lhs (stmt);
2862 vectype_out = STMT_VINFO_VECTYPE (stmt_info);
2864 /* Check the operands of the operation. */
2865 op0 = gimple_assign_rhs1 (stmt);
2866 if (! ((INTEGRAL_TYPE_P (TREE_TYPE (scalar_dest))
2867 && INTEGRAL_TYPE_P (TREE_TYPE (op0)))
2868 || (SCALAR_FLOAT_TYPE_P (TREE_TYPE (scalar_dest))
2869 && SCALAR_FLOAT_TYPE_P (TREE_TYPE (op0))
2870 && CONVERT_EXPR_CODE_P (code))))
2871 return false;
2872 if (!vect_is_simple_use_1 (op0, loop_vinfo, NULL,
2873 &def_stmt, &def, &dt[0], &vectype_in))
2875 if (vect_print_dump_info (REPORT_DETAILS))
2876 fprintf (vect_dump, "use not simple.");
2877 return false;
2879 /* If op0 is an external or constant def use a vector type with
2880 the same size as the output vector type. */
2881 if (!vectype_in)
2882 vectype_in = get_same_sized_vectype (TREE_TYPE (op0), vectype_out);
2883 if (vec_stmt)
2884 gcc_assert (vectype_in);
2885 if (!vectype_in)
2887 if (vect_print_dump_info (REPORT_DETAILS))
2889 fprintf (vect_dump, "no vectype for scalar type ");
2890 print_generic_expr (vect_dump, TREE_TYPE (op0), TDF_SLIM);
2893 return false;
2896 nunits_in = TYPE_VECTOR_SUBPARTS (vectype_in);
2897 nunits_out = TYPE_VECTOR_SUBPARTS (vectype_out);
2898 if (nunits_in <= nunits_out)
2899 return false;
2901 /* Multiple types in SLP are handled by creating the appropriate number of
2902 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
2903 case of SLP. */
2904 if (slp_node)
2905 ncopies = 1;
2906 else
2907 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits_in;
2909 gcc_assert (ncopies >= 1);
2911 op_type = TREE_CODE_LENGTH (code);
2912 if (op_type == binary_op)
2914 op1 = gimple_assign_rhs2 (stmt);
2915 if (!vect_is_simple_use (op1, loop_vinfo, NULL, &def_stmt, &def, &dt[1]))
2917 if (vect_print_dump_info (REPORT_DETAILS))
2918 fprintf (vect_dump, "use not simple.");
2919 return false;
2923 /* Supportable by target? */
2924 if (!supportable_widening_operation (code, stmt, vectype_out, vectype_in,
2925 &decl1, &decl2, &code1, &code2,
2926 &multi_step_cvt, &interm_types))
2927 return false;
2929 /* Binary widening operation can only be supported directly by the
2930 architecture. */
2931 gcc_assert (!(multi_step_cvt && op_type == binary_op));
2933 if (!vec_stmt) /* transformation not required. */
2935 STMT_VINFO_TYPE (stmt_info) = type_promotion_vec_info_type;
2936 if (vect_print_dump_info (REPORT_DETAILS))
2937 fprintf (vect_dump, "=== vectorizable_promotion ===");
2938 vect_model_simple_cost (stmt_info, 2*ncopies, dt, NULL);
2939 return true;
2942 /** Transform. **/
2944 if (vect_print_dump_info (REPORT_DETAILS))
2945 fprintf (vect_dump, "transform type promotion operation. ncopies = %d.",
2946 ncopies);
2948 /* Handle def. */
2949 /* In case of multi-step promotion, we first generate promotion operations
2950 to the intermediate types, and then from that types to the final one.
2951 We store vector destination in VEC_DSTS in the correct order for
2952 recursive creation of promotion operations in
2953 vect_create_vectorized_promotion_stmts(). Vector destinations are created
2954 according to TYPES recieved from supportable_widening_operation(). */
2955 if (multi_step_cvt)
2956 vec_dsts = VEC_alloc (tree, heap, multi_step_cvt + 1);
2957 else
2958 vec_dsts = VEC_alloc (tree, heap, 1);
2960 vec_dest = vect_create_destination_var (scalar_dest, vectype_out);
2961 VEC_quick_push (tree, vec_dsts, vec_dest);
2963 if (multi_step_cvt)
2965 for (i = VEC_length (tree, interm_types) - 1;
2966 VEC_iterate (tree, interm_types, i, intermediate_type); i--)
2968 vec_dest = vect_create_destination_var (scalar_dest,
2969 intermediate_type);
2970 VEC_quick_push (tree, vec_dsts, vec_dest);
2974 if (!slp_node)
2976 vec_oprnds0 = VEC_alloc (tree, heap,
2977 (multi_step_cvt ? vect_pow2 (multi_step_cvt) : 1));
2978 if (op_type == binary_op)
2979 vec_oprnds1 = VEC_alloc (tree, heap, 1);
2982 /* In case the vectorization factor (VF) is bigger than the number
2983 of elements that we can fit in a vectype (nunits), we have to generate
2984 more than one vector stmt - i.e - we need to "unroll" the
2985 vector stmt by a factor VF/nunits. */
2987 prev_stmt_info = NULL;
2988 for (j = 0; j < ncopies; j++)
2990 /* Handle uses. */
2991 if (j == 0)
2993 if (slp_node)
2994 vect_get_slp_defs (slp_node, &vec_oprnds0, &vec_oprnds1, -1);
2995 else
2997 vec_oprnd0 = vect_get_vec_def_for_operand (op0, stmt, NULL);
2998 VEC_quick_push (tree, vec_oprnds0, vec_oprnd0);
2999 if (op_type == binary_op)
3001 vec_oprnd1 = vect_get_vec_def_for_operand (op1, stmt, NULL);
3002 VEC_quick_push (tree, vec_oprnds1, vec_oprnd1);
3006 else
3008 vec_oprnd0 = vect_get_vec_def_for_stmt_copy (dt[0], vec_oprnd0);
3009 VEC_replace (tree, vec_oprnds0, 0, vec_oprnd0);
3010 if (op_type == binary_op)
3012 vec_oprnd1 = vect_get_vec_def_for_stmt_copy (dt[1], vec_oprnd1);
3013 VEC_replace (tree, vec_oprnds1, 0, vec_oprnd1);
3017 /* Arguments are ready. Create the new vector stmts. */
3018 tmp_vec_dsts = VEC_copy (tree, heap, vec_dsts);
3019 vect_create_vectorized_promotion_stmts (&vec_oprnds0, &vec_oprnds1,
3020 multi_step_cvt, stmt,
3021 tmp_vec_dsts,
3022 gsi, slp_node, code1, code2,
3023 decl1, decl2, op_type,
3024 &prev_stmt_info);
3027 VEC_free (tree, heap, vec_dsts);
3028 VEC_free (tree, heap, tmp_vec_dsts);
3029 VEC_free (tree, heap, interm_types);
3030 VEC_free (tree, heap, vec_oprnds0);
3031 VEC_free (tree, heap, vec_oprnds1);
3033 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
3034 return true;
3038 /* Function vectorizable_store.
3040 Check if STMT defines a non scalar data-ref (array/pointer/structure) that
3041 can be vectorized.
3042 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
3043 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
3044 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
3046 static bool
3047 vectorizable_store (gimple stmt, gimple_stmt_iterator *gsi, gimple *vec_stmt,
3048 slp_tree slp_node)
3050 tree scalar_dest;
3051 tree data_ref;
3052 tree op;
3053 tree vec_oprnd = NULL_TREE;
3054 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
3055 struct data_reference *dr = STMT_VINFO_DATA_REF (stmt_info), *first_dr = NULL;
3056 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
3057 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
3058 struct loop *loop = NULL;
3059 enum machine_mode vec_mode;
3060 tree dummy;
3061 enum dr_alignment_support alignment_support_scheme;
3062 tree def;
3063 gimple def_stmt;
3064 enum vect_def_type dt;
3065 stmt_vec_info prev_stmt_info = NULL;
3066 tree dataref_ptr = NULL_TREE;
3067 int nunits = TYPE_VECTOR_SUBPARTS (vectype);
3068 int ncopies;
3069 int j;
3070 gimple next_stmt, first_stmt = NULL;
3071 bool strided_store = false;
3072 unsigned int group_size, i;
3073 VEC(tree,heap) *dr_chain = NULL, *oprnds = NULL, *result_chain = NULL;
3074 bool inv_p;
3075 VEC(tree,heap) *vec_oprnds = NULL;
3076 bool slp = (slp_node != NULL);
3077 unsigned int vec_num;
3078 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
3080 if (loop_vinfo)
3081 loop = LOOP_VINFO_LOOP (loop_vinfo);
3083 /* Multiple types in SLP are handled by creating the appropriate number of
3084 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
3085 case of SLP. */
3086 if (slp)
3087 ncopies = 1;
3088 else
3089 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits;
3091 gcc_assert (ncopies >= 1);
3093 /* FORNOW. This restriction should be relaxed. */
3094 if (loop && nested_in_vect_loop_p (loop, stmt) && ncopies > 1)
3096 if (vect_print_dump_info (REPORT_DETAILS))
3097 fprintf (vect_dump, "multiple types in nested loop.");
3098 return false;
3101 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
3102 return false;
3104 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
3105 return false;
3107 /* Is vectorizable store? */
3109 if (!is_gimple_assign (stmt))
3110 return false;
3112 scalar_dest = gimple_assign_lhs (stmt);
3113 if (TREE_CODE (scalar_dest) != ARRAY_REF
3114 && TREE_CODE (scalar_dest) != INDIRECT_REF
3115 && TREE_CODE (scalar_dest) != COMPONENT_REF
3116 && TREE_CODE (scalar_dest) != IMAGPART_EXPR
3117 && TREE_CODE (scalar_dest) != REALPART_EXPR
3118 && TREE_CODE (scalar_dest) != MEM_REF)
3119 return false;
3121 gcc_assert (gimple_assign_single_p (stmt));
3122 op = gimple_assign_rhs1 (stmt);
3123 if (!vect_is_simple_use (op, loop_vinfo, bb_vinfo, &def_stmt, &def, &dt))
3125 if (vect_print_dump_info (REPORT_DETAILS))
3126 fprintf (vect_dump, "use not simple.");
3127 return false;
3130 /* The scalar rhs type needs to be trivially convertible to the vector
3131 component type. This should always be the case. */
3132 if (!useless_type_conversion_p (TREE_TYPE (vectype), TREE_TYPE (op)))
3134 if (vect_print_dump_info (REPORT_DETAILS))
3135 fprintf (vect_dump, "??? operands of different types");
3136 return false;
3139 vec_mode = TYPE_MODE (vectype);
3140 /* FORNOW. In some cases can vectorize even if data-type not supported
3141 (e.g. - array initialization with 0). */
3142 if (optab_handler (mov_optab, vec_mode) == CODE_FOR_nothing)
3143 return false;
3145 if (!STMT_VINFO_DATA_REF (stmt_info))
3146 return false;
3148 if (tree_int_cst_compare (DR_STEP (dr), size_zero_node) < 0)
3150 if (vect_print_dump_info (REPORT_DETAILS))
3151 fprintf (vect_dump, "negative step for store.");
3152 return false;
3155 if (STMT_VINFO_STRIDED_ACCESS (stmt_info))
3157 strided_store = true;
3158 first_stmt = DR_GROUP_FIRST_DR (stmt_info);
3159 if (!vect_strided_store_supported (vectype)
3160 && !PURE_SLP_STMT (stmt_info) && !slp)
3161 return false;
3163 if (first_stmt == stmt)
3165 /* STMT is the leader of the group. Check the operands of all the
3166 stmts of the group. */
3167 next_stmt = DR_GROUP_NEXT_DR (stmt_info);
3168 while (next_stmt)
3170 gcc_assert (gimple_assign_single_p (next_stmt));
3171 op = gimple_assign_rhs1 (next_stmt);
3172 if (!vect_is_simple_use (op, loop_vinfo, bb_vinfo, &def_stmt,
3173 &def, &dt))
3175 if (vect_print_dump_info (REPORT_DETAILS))
3176 fprintf (vect_dump, "use not simple.");
3177 return false;
3179 next_stmt = DR_GROUP_NEXT_DR (vinfo_for_stmt (next_stmt));
3184 if (!vec_stmt) /* transformation not required. */
3186 STMT_VINFO_TYPE (stmt_info) = store_vec_info_type;
3187 vect_model_store_cost (stmt_info, ncopies, dt, NULL);
3188 return true;
3191 /** Transform. **/
3193 if (strided_store)
3195 first_dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (first_stmt));
3196 group_size = DR_GROUP_SIZE (vinfo_for_stmt (first_stmt));
3198 DR_GROUP_STORE_COUNT (vinfo_for_stmt (first_stmt))++;
3200 /* FORNOW */
3201 gcc_assert (!loop || !nested_in_vect_loop_p (loop, stmt));
3203 /* We vectorize all the stmts of the interleaving group when we
3204 reach the last stmt in the group. */
3205 if (DR_GROUP_STORE_COUNT (vinfo_for_stmt (first_stmt))
3206 < DR_GROUP_SIZE (vinfo_for_stmt (first_stmt))
3207 && !slp)
3209 *vec_stmt = NULL;
3210 return true;
3213 if (slp)
3215 strided_store = false;
3216 /* VEC_NUM is the number of vect stmts to be created for this
3217 group. */
3218 vec_num = SLP_TREE_NUMBER_OF_VEC_STMTS (slp_node);
3219 first_stmt = VEC_index (gimple, SLP_TREE_SCALAR_STMTS (slp_node), 0);
3220 first_dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (first_stmt));
3222 else
3223 /* VEC_NUM is the number of vect stmts to be created for this
3224 group. */
3225 vec_num = group_size;
3227 else
3229 first_stmt = stmt;
3230 first_dr = dr;
3231 group_size = vec_num = 1;
3234 if (vect_print_dump_info (REPORT_DETAILS))
3235 fprintf (vect_dump, "transform store. ncopies = %d",ncopies);
3237 dr_chain = VEC_alloc (tree, heap, group_size);
3238 oprnds = VEC_alloc (tree, heap, group_size);
3240 alignment_support_scheme = vect_supportable_dr_alignment (first_dr, false);
3241 gcc_assert (alignment_support_scheme);
3243 /* In case the vectorization factor (VF) is bigger than the number
3244 of elements that we can fit in a vectype (nunits), we have to generate
3245 more than one vector stmt - i.e - we need to "unroll" the
3246 vector stmt by a factor VF/nunits. For more details see documentation in
3247 vect_get_vec_def_for_copy_stmt. */
3249 /* In case of interleaving (non-unit strided access):
3251 S1: &base + 2 = x2
3252 S2: &base = x0
3253 S3: &base + 1 = x1
3254 S4: &base + 3 = x3
3256 We create vectorized stores starting from base address (the access of the
3257 first stmt in the chain (S2 in the above example), when the last store stmt
3258 of the chain (S4) is reached:
3260 VS1: &base = vx2
3261 VS2: &base + vec_size*1 = vx0
3262 VS3: &base + vec_size*2 = vx1
3263 VS4: &base + vec_size*3 = vx3
3265 Then permutation statements are generated:
3267 VS5: vx5 = VEC_INTERLEAVE_HIGH_EXPR < vx0, vx3 >
3268 VS6: vx6 = VEC_INTERLEAVE_LOW_EXPR < vx0, vx3 >
3271 And they are put in STMT_VINFO_VEC_STMT of the corresponding scalar stmts
3272 (the order of the data-refs in the output of vect_permute_store_chain
3273 corresponds to the order of scalar stmts in the interleaving chain - see
3274 the documentation of vect_permute_store_chain()).
3276 In case of both multiple types and interleaving, above vector stores and
3277 permutation stmts are created for every copy. The result vector stmts are
3278 put in STMT_VINFO_VEC_STMT for the first copy and in the corresponding
3279 STMT_VINFO_RELATED_STMT for the next copies.
3282 prev_stmt_info = NULL;
3283 for (j = 0; j < ncopies; j++)
3285 gimple new_stmt;
3286 gimple ptr_incr;
3288 if (j == 0)
3290 if (slp)
3292 /* Get vectorized arguments for SLP_NODE. */
3293 vect_get_slp_defs (slp_node, &vec_oprnds, NULL, -1);
3295 vec_oprnd = VEC_index (tree, vec_oprnds, 0);
3297 else
3299 /* For interleaved stores we collect vectorized defs for all the
3300 stores in the group in DR_CHAIN and OPRNDS. DR_CHAIN is then
3301 used as an input to vect_permute_store_chain(), and OPRNDS as
3302 an input to vect_get_vec_def_for_stmt_copy() for the next copy.
3304 If the store is not strided, GROUP_SIZE is 1, and DR_CHAIN and
3305 OPRNDS are of size 1. */
3306 next_stmt = first_stmt;
3307 for (i = 0; i < group_size; i++)
3309 /* Since gaps are not supported for interleaved stores,
3310 GROUP_SIZE is the exact number of stmts in the chain.
3311 Therefore, NEXT_STMT can't be NULL_TREE. In case that
3312 there is no interleaving, GROUP_SIZE is 1, and only one
3313 iteration of the loop will be executed. */
3314 gcc_assert (next_stmt
3315 && gimple_assign_single_p (next_stmt));
3316 op = gimple_assign_rhs1 (next_stmt);
3318 vec_oprnd = vect_get_vec_def_for_operand (op, next_stmt,
3319 NULL);
3320 VEC_quick_push(tree, dr_chain, vec_oprnd);
3321 VEC_quick_push(tree, oprnds, vec_oprnd);
3322 next_stmt = DR_GROUP_NEXT_DR (vinfo_for_stmt (next_stmt));
3326 /* We should have catched mismatched types earlier. */
3327 gcc_assert (useless_type_conversion_p (vectype,
3328 TREE_TYPE (vec_oprnd)));
3329 dataref_ptr = vect_create_data_ref_ptr (first_stmt, NULL, NULL_TREE,
3330 &dummy, &ptr_incr, false,
3331 &inv_p);
3332 gcc_assert (bb_vinfo || !inv_p);
3334 else
3336 /* For interleaved stores we created vectorized defs for all the
3337 defs stored in OPRNDS in the previous iteration (previous copy).
3338 DR_CHAIN is then used as an input to vect_permute_store_chain(),
3339 and OPRNDS as an input to vect_get_vec_def_for_stmt_copy() for the
3340 next copy.
3341 If the store is not strided, GROUP_SIZE is 1, and DR_CHAIN and
3342 OPRNDS are of size 1. */
3343 for (i = 0; i < group_size; i++)
3345 op = VEC_index (tree, oprnds, i);
3346 vect_is_simple_use (op, loop_vinfo, bb_vinfo, &def_stmt, &def,
3347 &dt);
3348 vec_oprnd = vect_get_vec_def_for_stmt_copy (dt, op);
3349 VEC_replace(tree, dr_chain, i, vec_oprnd);
3350 VEC_replace(tree, oprnds, i, vec_oprnd);
3352 dataref_ptr =
3353 bump_vector_ptr (dataref_ptr, ptr_incr, gsi, stmt, NULL_TREE);
3356 if (strided_store)
3358 result_chain = VEC_alloc (tree, heap, group_size);
3359 /* Permute. */
3360 if (!vect_permute_store_chain (dr_chain, group_size, stmt, gsi,
3361 &result_chain))
3362 return false;
3365 next_stmt = first_stmt;
3366 for (i = 0; i < vec_num; i++)
3368 struct ptr_info_def *pi;
3370 if (i > 0)
3371 /* Bump the vector pointer. */
3372 dataref_ptr = bump_vector_ptr (dataref_ptr, ptr_incr, gsi, stmt,
3373 NULL_TREE);
3375 if (slp)
3376 vec_oprnd = VEC_index (tree, vec_oprnds, i);
3377 else if (strided_store)
3378 /* For strided stores vectorized defs are interleaved in
3379 vect_permute_store_chain(). */
3380 vec_oprnd = VEC_index (tree, result_chain, i);
3382 data_ref = build2 (MEM_REF, TREE_TYPE (vec_oprnd), dataref_ptr,
3383 build_int_cst (reference_alias_ptr_type
3384 (DR_REF (first_dr)), 0));
3385 pi = get_ptr_info (dataref_ptr);
3386 pi->align = TYPE_ALIGN_UNIT (vectype);
3387 if (aligned_access_p (first_dr))
3388 pi->misalign = 0;
3389 else if (DR_MISALIGNMENT (first_dr) == -1)
3391 TREE_TYPE (data_ref)
3392 = build_aligned_type (TREE_TYPE (data_ref),
3393 TYPE_ALIGN (TREE_TYPE (vectype)));
3394 pi->align = TYPE_ALIGN_UNIT (TREE_TYPE (vectype));
3395 pi->misalign = 0;
3397 else
3399 TREE_TYPE (data_ref)
3400 = build_aligned_type (TREE_TYPE (data_ref),
3401 TYPE_ALIGN (TREE_TYPE (vectype)));
3402 pi->misalign = DR_MISALIGNMENT (first_dr);
3405 /* Arguments are ready. Create the new vector stmt. */
3406 new_stmt = gimple_build_assign (data_ref, vec_oprnd);
3407 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3408 mark_symbols_for_renaming (new_stmt);
3410 if (slp)
3411 continue;
3413 if (j == 0)
3414 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
3415 else
3416 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
3418 prev_stmt_info = vinfo_for_stmt (new_stmt);
3419 next_stmt = DR_GROUP_NEXT_DR (vinfo_for_stmt (next_stmt));
3420 if (!next_stmt)
3421 break;
3425 VEC_free (tree, heap, dr_chain);
3426 VEC_free (tree, heap, oprnds);
3427 if (result_chain)
3428 VEC_free (tree, heap, result_chain);
3429 if (vec_oprnds)
3430 VEC_free (tree, heap, vec_oprnds);
3432 return true;
3435 /* Given a vector type VECTYPE returns a builtin DECL to be used
3436 for vector permutation and stores a mask into *MASK that implements
3437 reversal of the vector elements. If that is impossible to do
3438 returns NULL (and *MASK is unchanged). */
3440 static tree
3441 perm_mask_for_reverse (tree vectype, tree *mask)
3443 tree builtin_decl;
3444 tree mask_element_type, mask_type;
3445 tree mask_vec = NULL;
3446 int i;
3447 int nunits;
3448 if (!targetm.vectorize.builtin_vec_perm)
3449 return NULL;
3451 builtin_decl = targetm.vectorize.builtin_vec_perm (vectype,
3452 &mask_element_type);
3453 if (!builtin_decl || !mask_element_type)
3454 return NULL;
3456 mask_type = get_vectype_for_scalar_type (mask_element_type);
3457 nunits = TYPE_VECTOR_SUBPARTS (vectype);
3458 if (TYPE_VECTOR_SUBPARTS (vectype) != TYPE_VECTOR_SUBPARTS (mask_type))
3459 return NULL;
3461 for (i = 0; i < nunits; i++)
3462 mask_vec = tree_cons (NULL, build_int_cst (mask_element_type, i), mask_vec);
3463 mask_vec = build_vector (mask_type, mask_vec);
3465 if (!targetm.vectorize.builtin_vec_perm_ok (vectype, mask_vec))
3466 return NULL;
3467 if (mask)
3468 *mask = mask_vec;
3469 return builtin_decl;
3472 /* Given a vector variable X, that was generated for the scalar LHS of
3473 STMT, generate instructions to reverse the vector elements of X,
3474 insert them a *GSI and return the permuted vector variable. */
3476 static tree
3477 reverse_vec_elements (tree x, gimple stmt, gimple_stmt_iterator *gsi)
3479 tree vectype = TREE_TYPE (x);
3480 tree mask_vec, builtin_decl;
3481 tree perm_dest, data_ref;
3482 gimple perm_stmt;
3484 builtin_decl = perm_mask_for_reverse (vectype, &mask_vec);
3486 perm_dest = vect_create_destination_var (gimple_assign_lhs (stmt), vectype);
3488 /* Generate the permute statement. */
3489 perm_stmt = gimple_build_call (builtin_decl, 3, x, x, mask_vec);
3490 data_ref = make_ssa_name (perm_dest, perm_stmt);
3491 gimple_call_set_lhs (perm_stmt, data_ref);
3492 vect_finish_stmt_generation (stmt, perm_stmt, gsi);
3494 return data_ref;
3497 /* vectorizable_load.
3499 Check if STMT reads a non scalar data-ref (array/pointer/structure) that
3500 can be vectorized.
3501 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
3502 stmt to replace it, put it in VEC_STMT, and insert it at BSI.
3503 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
3505 static bool
3506 vectorizable_load (gimple stmt, gimple_stmt_iterator *gsi, gimple *vec_stmt,
3507 slp_tree slp_node, slp_instance slp_node_instance)
3509 tree scalar_dest;
3510 tree vec_dest = NULL;
3511 tree data_ref = NULL;
3512 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
3513 stmt_vec_info prev_stmt_info;
3514 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
3515 struct loop *loop = NULL;
3516 struct loop *containing_loop = (gimple_bb (stmt))->loop_father;
3517 bool nested_in_vect_loop = false;
3518 struct data_reference *dr = STMT_VINFO_DATA_REF (stmt_info), *first_dr;
3519 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
3520 tree new_temp;
3521 enum machine_mode mode;
3522 gimple new_stmt = NULL;
3523 tree dummy;
3524 enum dr_alignment_support alignment_support_scheme;
3525 tree dataref_ptr = NULL_TREE;
3526 gimple ptr_incr;
3527 int nunits = TYPE_VECTOR_SUBPARTS (vectype);
3528 int ncopies;
3529 int i, j, group_size;
3530 tree msq = NULL_TREE, lsq;
3531 tree offset = NULL_TREE;
3532 tree realignment_token = NULL_TREE;
3533 gimple phi = NULL;
3534 VEC(tree,heap) *dr_chain = NULL;
3535 bool strided_load = false;
3536 gimple first_stmt;
3537 tree scalar_type;
3538 bool inv_p;
3539 bool negative;
3540 bool compute_in_loop = false;
3541 struct loop *at_loop;
3542 int vec_num;
3543 bool slp = (slp_node != NULL);
3544 bool slp_perm = false;
3545 enum tree_code code;
3546 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
3547 int vf;
3549 if (loop_vinfo)
3551 loop = LOOP_VINFO_LOOP (loop_vinfo);
3552 nested_in_vect_loop = nested_in_vect_loop_p (loop, stmt);
3553 vf = LOOP_VINFO_VECT_FACTOR (loop_vinfo);
3555 else
3556 vf = 1;
3558 /* Multiple types in SLP are handled by creating the appropriate number of
3559 vectorized stmts for each SLP node. Hence, NCOPIES is always 1 in
3560 case of SLP. */
3561 if (slp)
3562 ncopies = 1;
3563 else
3564 ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits;
3566 gcc_assert (ncopies >= 1);
3568 /* FORNOW. This restriction should be relaxed. */
3569 if (nested_in_vect_loop && ncopies > 1)
3571 if (vect_print_dump_info (REPORT_DETAILS))
3572 fprintf (vect_dump, "multiple types in nested loop.");
3573 return false;
3576 if (!STMT_VINFO_RELEVANT_P (stmt_info) && !bb_vinfo)
3577 return false;
3579 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def)
3580 return false;
3582 /* Is vectorizable load? */
3583 if (!is_gimple_assign (stmt))
3584 return false;
3586 scalar_dest = gimple_assign_lhs (stmt);
3587 if (TREE_CODE (scalar_dest) != SSA_NAME)
3588 return false;
3590 code = gimple_assign_rhs_code (stmt);
3591 if (code != ARRAY_REF
3592 && code != INDIRECT_REF
3593 && code != COMPONENT_REF
3594 && code != IMAGPART_EXPR
3595 && code != REALPART_EXPR
3596 && code != MEM_REF)
3597 return false;
3599 if (!STMT_VINFO_DATA_REF (stmt_info))
3600 return false;
3602 negative = tree_int_cst_compare (DR_STEP (dr), size_zero_node) < 0;
3603 if (negative && ncopies > 1)
3605 if (vect_print_dump_info (REPORT_DETAILS))
3606 fprintf (vect_dump, "multiple types with negative step.");
3607 return false;
3610 scalar_type = TREE_TYPE (DR_REF (dr));
3611 mode = TYPE_MODE (vectype);
3613 /* FORNOW. In some cases can vectorize even if data-type not supported
3614 (e.g. - data copies). */
3615 if (optab_handler (mov_optab, mode) == CODE_FOR_nothing)
3617 if (vect_print_dump_info (REPORT_DETAILS))
3618 fprintf (vect_dump, "Aligned load, but unsupported type.");
3619 return false;
3622 /* The vector component type needs to be trivially convertible to the
3623 scalar lhs. This should always be the case. */
3624 if (!useless_type_conversion_p (TREE_TYPE (scalar_dest), TREE_TYPE (vectype)))
3626 if (vect_print_dump_info (REPORT_DETAILS))
3627 fprintf (vect_dump, "??? operands of different types");
3628 return false;
3631 /* Check if the load is a part of an interleaving chain. */
3632 if (STMT_VINFO_STRIDED_ACCESS (stmt_info))
3634 strided_load = true;
3635 /* FORNOW */
3636 gcc_assert (! nested_in_vect_loop);
3638 /* Check if interleaving is supported. */
3639 if (!vect_strided_load_supported (vectype)
3640 && !PURE_SLP_STMT (stmt_info) && !slp)
3641 return false;
3644 if (negative)
3646 gcc_assert (!strided_load);
3647 alignment_support_scheme = vect_supportable_dr_alignment (dr, false);
3648 if (alignment_support_scheme != dr_aligned
3649 && alignment_support_scheme != dr_unaligned_supported)
3651 if (vect_print_dump_info (REPORT_DETAILS))
3652 fprintf (vect_dump, "negative step but alignment required.");
3653 return false;
3655 if (!perm_mask_for_reverse (vectype, NULL))
3657 if (vect_print_dump_info (REPORT_DETAILS))
3658 fprintf (vect_dump, "negative step and reversing not supported.");
3659 return false;
3663 if (!vec_stmt) /* transformation not required. */
3665 STMT_VINFO_TYPE (stmt_info) = load_vec_info_type;
3666 vect_model_load_cost (stmt_info, ncopies, NULL);
3667 return true;
3670 if (vect_print_dump_info (REPORT_DETAILS))
3671 fprintf (vect_dump, "transform load.");
3673 /** Transform. **/
3675 if (strided_load)
3677 first_stmt = DR_GROUP_FIRST_DR (stmt_info);
3678 /* Check if the chain of loads is already vectorized. */
3679 if (STMT_VINFO_VEC_STMT (vinfo_for_stmt (first_stmt)))
3681 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
3682 return true;
3684 first_dr = STMT_VINFO_DATA_REF (vinfo_for_stmt (first_stmt));
3685 group_size = DR_GROUP_SIZE (vinfo_for_stmt (first_stmt));
3687 /* VEC_NUM is the number of vect stmts to be created for this group. */
3688 if (slp)
3690 strided_load = false;
3691 vec_num = SLP_TREE_NUMBER_OF_VEC_STMTS (slp_node);
3692 if (SLP_INSTANCE_LOAD_PERMUTATION (slp_node_instance))
3693 slp_perm = true;
3695 else
3696 vec_num = group_size;
3698 dr_chain = VEC_alloc (tree, heap, vec_num);
3700 else
3702 first_stmt = stmt;
3703 first_dr = dr;
3704 group_size = vec_num = 1;
3707 alignment_support_scheme = vect_supportable_dr_alignment (first_dr, false);
3708 gcc_assert (alignment_support_scheme);
3710 /* In case the vectorization factor (VF) is bigger than the number
3711 of elements that we can fit in a vectype (nunits), we have to generate
3712 more than one vector stmt - i.e - we need to "unroll" the
3713 vector stmt by a factor VF/nunits. In doing so, we record a pointer
3714 from one copy of the vector stmt to the next, in the field
3715 STMT_VINFO_RELATED_STMT. This is necessary in order to allow following
3716 stages to find the correct vector defs to be used when vectorizing
3717 stmts that use the defs of the current stmt. The example below
3718 illustrates the vectorization process when VF=16 and nunits=4 (i.e., we
3719 need to create 4 vectorized stmts):
3721 before vectorization:
3722 RELATED_STMT VEC_STMT
3723 S1: x = memref - -
3724 S2: z = x + 1 - -
3726 step 1: vectorize stmt S1:
3727 We first create the vector stmt VS1_0, and, as usual, record a
3728 pointer to it in the STMT_VINFO_VEC_STMT of the scalar stmt S1.
3729 Next, we create the vector stmt VS1_1, and record a pointer to
3730 it in the STMT_VINFO_RELATED_STMT of the vector stmt VS1_0.
3731 Similarly, for VS1_2 and VS1_3. This is the resulting chain of
3732 stmts and pointers:
3733 RELATED_STMT VEC_STMT
3734 VS1_0: vx0 = memref0 VS1_1 -
3735 VS1_1: vx1 = memref1 VS1_2 -
3736 VS1_2: vx2 = memref2 VS1_3 -
3737 VS1_3: vx3 = memref3 - -
3738 S1: x = load - VS1_0
3739 S2: z = x + 1 - -
3741 See in documentation in vect_get_vec_def_for_stmt_copy for how the
3742 information we recorded in RELATED_STMT field is used to vectorize
3743 stmt S2. */
3745 /* In case of interleaving (non-unit strided access):
3747 S1: x2 = &base + 2
3748 S2: x0 = &base
3749 S3: x1 = &base + 1
3750 S4: x3 = &base + 3
3752 Vectorized loads are created in the order of memory accesses
3753 starting from the access of the first stmt of the chain:
3755 VS1: vx0 = &base
3756 VS2: vx1 = &base + vec_size*1
3757 VS3: vx3 = &base + vec_size*2
3758 VS4: vx4 = &base + vec_size*3
3760 Then permutation statements are generated:
3762 VS5: vx5 = VEC_EXTRACT_EVEN_EXPR < vx0, vx1 >
3763 VS6: vx6 = VEC_EXTRACT_ODD_EXPR < vx0, vx1 >
3766 And they are put in STMT_VINFO_VEC_STMT of the corresponding scalar stmts
3767 (the order of the data-refs in the output of vect_permute_load_chain
3768 corresponds to the order of scalar stmts in the interleaving chain - see
3769 the documentation of vect_permute_load_chain()).
3770 The generation of permutation stmts and recording them in
3771 STMT_VINFO_VEC_STMT is done in vect_transform_strided_load().
3773 In case of both multiple types and interleaving, the vector loads and
3774 permutation stmts above are created for every copy. The result vector
3775 stmts are put in STMT_VINFO_VEC_STMT for the first copy and in the
3776 corresponding STMT_VINFO_RELATED_STMT for the next copies. */
3778 /* If the data reference is aligned (dr_aligned) or potentially unaligned
3779 on a target that supports unaligned accesses (dr_unaligned_supported)
3780 we generate the following code:
3781 p = initial_addr;
3782 indx = 0;
3783 loop {
3784 p = p + indx * vectype_size;
3785 vec_dest = *(p);
3786 indx = indx + 1;
3789 Otherwise, the data reference is potentially unaligned on a target that
3790 does not support unaligned accesses (dr_explicit_realign_optimized) -
3791 then generate the following code, in which the data in each iteration is
3792 obtained by two vector loads, one from the previous iteration, and one
3793 from the current iteration:
3794 p1 = initial_addr;
3795 msq_init = *(floor(p1))
3796 p2 = initial_addr + VS - 1;
3797 realignment_token = call target_builtin;
3798 indx = 0;
3799 loop {
3800 p2 = p2 + indx * vectype_size
3801 lsq = *(floor(p2))
3802 vec_dest = realign_load (msq, lsq, realignment_token)
3803 indx = indx + 1;
3804 msq = lsq;
3805 } */
3807 /* If the misalignment remains the same throughout the execution of the
3808 loop, we can create the init_addr and permutation mask at the loop
3809 preheader. Otherwise, it needs to be created inside the loop.
3810 This can only occur when vectorizing memory accesses in the inner-loop
3811 nested within an outer-loop that is being vectorized. */
3813 if (loop && nested_in_vect_loop_p (loop, stmt)
3814 && (TREE_INT_CST_LOW (DR_STEP (dr))
3815 % GET_MODE_SIZE (TYPE_MODE (vectype)) != 0))
3817 gcc_assert (alignment_support_scheme != dr_explicit_realign_optimized);
3818 compute_in_loop = true;
3821 if ((alignment_support_scheme == dr_explicit_realign_optimized
3822 || alignment_support_scheme == dr_explicit_realign)
3823 && !compute_in_loop)
3825 msq = vect_setup_realignment (first_stmt, gsi, &realignment_token,
3826 alignment_support_scheme, NULL_TREE,
3827 &at_loop);
3828 if (alignment_support_scheme == dr_explicit_realign_optimized)
3830 phi = SSA_NAME_DEF_STMT (msq);
3831 offset = size_int (TYPE_VECTOR_SUBPARTS (vectype) - 1);
3834 else
3835 at_loop = loop;
3837 if (negative)
3838 offset = size_int (-TYPE_VECTOR_SUBPARTS (vectype) + 1);
3840 prev_stmt_info = NULL;
3841 for (j = 0; j < ncopies; j++)
3843 /* 1. Create the vector pointer update chain. */
3844 if (j == 0)
3845 dataref_ptr = vect_create_data_ref_ptr (first_stmt,
3846 at_loop, offset,
3847 &dummy, &ptr_incr, false,
3848 &inv_p);
3849 else
3850 dataref_ptr =
3851 bump_vector_ptr (dataref_ptr, ptr_incr, gsi, stmt, NULL_TREE);
3853 for (i = 0; i < vec_num; i++)
3855 if (i > 0)
3856 dataref_ptr = bump_vector_ptr (dataref_ptr, ptr_incr, gsi, stmt,
3857 NULL_TREE);
3859 /* 2. Create the vector-load in the loop. */
3860 switch (alignment_support_scheme)
3862 case dr_aligned:
3863 case dr_unaligned_supported:
3865 struct ptr_info_def *pi;
3866 data_ref
3867 = build2 (MEM_REF, vectype, dataref_ptr,
3868 build_int_cst (reference_alias_ptr_type
3869 (DR_REF (first_dr)), 0));
3870 pi = get_ptr_info (dataref_ptr);
3871 pi->align = TYPE_ALIGN_UNIT (vectype);
3872 if (alignment_support_scheme == dr_aligned)
3874 gcc_assert (aligned_access_p (first_dr));
3875 pi->misalign = 0;
3877 else if (DR_MISALIGNMENT (first_dr) == -1)
3879 TREE_TYPE (data_ref)
3880 = build_aligned_type (TREE_TYPE (data_ref),
3881 TYPE_ALIGN (TREE_TYPE (vectype)));
3882 pi->align = TYPE_ALIGN_UNIT (TREE_TYPE (vectype));
3883 pi->misalign = 0;
3885 else
3887 TREE_TYPE (data_ref)
3888 = build_aligned_type (TREE_TYPE (data_ref),
3889 TYPE_ALIGN (TREE_TYPE (vectype)));
3890 pi->misalign = DR_MISALIGNMENT (first_dr);
3892 break;
3894 case dr_explicit_realign:
3896 tree ptr, bump;
3897 tree vs_minus_1 = size_int (TYPE_VECTOR_SUBPARTS (vectype) - 1);
3899 if (compute_in_loop)
3900 msq = vect_setup_realignment (first_stmt, gsi,
3901 &realignment_token,
3902 dr_explicit_realign,
3903 dataref_ptr, NULL);
3905 new_stmt = gimple_build_assign_with_ops
3906 (BIT_AND_EXPR, NULL_TREE, dataref_ptr,
3907 build_int_cst
3908 (TREE_TYPE (dataref_ptr),
3909 -(HOST_WIDE_INT)TYPE_ALIGN_UNIT (vectype)));
3910 ptr = make_ssa_name (SSA_NAME_VAR (dataref_ptr), new_stmt);
3911 gimple_assign_set_lhs (new_stmt, ptr);
3912 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3913 data_ref
3914 = build2 (MEM_REF, vectype, ptr,
3915 build_int_cst (reference_alias_ptr_type
3916 (DR_REF (first_dr)), 0));
3917 vec_dest = vect_create_destination_var (scalar_dest, vectype);
3918 new_stmt = gimple_build_assign (vec_dest, data_ref);
3919 new_temp = make_ssa_name (vec_dest, new_stmt);
3920 gimple_assign_set_lhs (new_stmt, new_temp);
3921 gimple_set_vdef (new_stmt, gimple_vdef (stmt));
3922 gimple_set_vuse (new_stmt, gimple_vuse (stmt));
3923 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3924 msq = new_temp;
3926 bump = size_binop (MULT_EXPR, vs_minus_1,
3927 TYPE_SIZE_UNIT (scalar_type));
3928 ptr = bump_vector_ptr (dataref_ptr, NULL, gsi, stmt, bump);
3929 new_stmt = gimple_build_assign_with_ops
3930 (BIT_AND_EXPR, NULL_TREE, ptr,
3931 build_int_cst
3932 (TREE_TYPE (ptr),
3933 -(HOST_WIDE_INT)TYPE_ALIGN_UNIT (vectype)));
3934 ptr = make_ssa_name (SSA_NAME_VAR (dataref_ptr), new_stmt);
3935 gimple_assign_set_lhs (new_stmt, ptr);
3936 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3937 data_ref
3938 = build2 (MEM_REF, vectype, ptr,
3939 build_int_cst (reference_alias_ptr_type
3940 (DR_REF (first_dr)), 0));
3941 break;
3943 case dr_explicit_realign_optimized:
3944 new_stmt = gimple_build_assign_with_ops
3945 (BIT_AND_EXPR, NULL_TREE, dataref_ptr,
3946 build_int_cst
3947 (TREE_TYPE (dataref_ptr),
3948 -(HOST_WIDE_INT)TYPE_ALIGN_UNIT (vectype)));
3949 new_temp = make_ssa_name (SSA_NAME_VAR (dataref_ptr), new_stmt);
3950 gimple_assign_set_lhs (new_stmt, new_temp);
3951 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3952 data_ref
3953 = build2 (MEM_REF, vectype, new_temp,
3954 build_int_cst (reference_alias_ptr_type
3955 (DR_REF (first_dr)), 0));
3956 break;
3957 default:
3958 gcc_unreachable ();
3960 vec_dest = vect_create_destination_var (scalar_dest, vectype);
3961 new_stmt = gimple_build_assign (vec_dest, data_ref);
3962 new_temp = make_ssa_name (vec_dest, new_stmt);
3963 gimple_assign_set_lhs (new_stmt, new_temp);
3964 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3965 mark_symbols_for_renaming (new_stmt);
3967 /* 3. Handle explicit realignment if necessary/supported. Create in
3968 loop: vec_dest = realign_load (msq, lsq, realignment_token) */
3969 if (alignment_support_scheme == dr_explicit_realign_optimized
3970 || alignment_support_scheme == dr_explicit_realign)
3972 tree tmp;
3974 lsq = gimple_assign_lhs (new_stmt);
3975 if (!realignment_token)
3976 realignment_token = dataref_ptr;
3977 vec_dest = vect_create_destination_var (scalar_dest, vectype);
3978 tmp = build3 (REALIGN_LOAD_EXPR, vectype, msq, lsq,
3979 realignment_token);
3980 new_stmt = gimple_build_assign (vec_dest, tmp);
3981 new_temp = make_ssa_name (vec_dest, new_stmt);
3982 gimple_assign_set_lhs (new_stmt, new_temp);
3983 vect_finish_stmt_generation (stmt, new_stmt, gsi);
3985 if (alignment_support_scheme == dr_explicit_realign_optimized)
3987 gcc_assert (phi);
3988 if (i == vec_num - 1 && j == ncopies - 1)
3989 add_phi_arg (phi, lsq, loop_latch_edge (containing_loop),
3990 UNKNOWN_LOCATION);
3991 msq = lsq;
3995 /* 4. Handle invariant-load. */
3996 if (inv_p && !bb_vinfo)
3998 gcc_assert (!strided_load);
3999 gcc_assert (nested_in_vect_loop_p (loop, stmt));
4000 if (j == 0)
4002 int k;
4003 tree t = NULL_TREE;
4004 tree vec_inv, bitpos, bitsize = TYPE_SIZE (scalar_type);
4006 /* CHECKME: bitpos depends on endianess? */
4007 bitpos = bitsize_zero_node;
4008 vec_inv = build3 (BIT_FIELD_REF, scalar_type, new_temp,
4009 bitsize, bitpos);
4010 vec_dest =
4011 vect_create_destination_var (scalar_dest, NULL_TREE);
4012 new_stmt = gimple_build_assign (vec_dest, vec_inv);
4013 new_temp = make_ssa_name (vec_dest, new_stmt);
4014 gimple_assign_set_lhs (new_stmt, new_temp);
4015 vect_finish_stmt_generation (stmt, new_stmt, gsi);
4017 for (k = nunits - 1; k >= 0; --k)
4018 t = tree_cons (NULL_TREE, new_temp, t);
4019 /* FIXME: use build_constructor directly. */
4020 vec_inv = build_constructor_from_list (vectype, t);
4021 new_temp = vect_init_vector (stmt, vec_inv, vectype, gsi);
4022 new_stmt = SSA_NAME_DEF_STMT (new_temp);
4024 else
4025 gcc_unreachable (); /* FORNOW. */
4028 if (negative)
4030 new_temp = reverse_vec_elements (new_temp, stmt, gsi);
4031 new_stmt = SSA_NAME_DEF_STMT (new_temp);
4034 /* Collect vector loads and later create their permutation in
4035 vect_transform_strided_load (). */
4036 if (strided_load || slp_perm)
4037 VEC_quick_push (tree, dr_chain, new_temp);
4039 /* Store vector loads in the corresponding SLP_NODE. */
4040 if (slp && !slp_perm)
4041 VEC_quick_push (gimple, SLP_TREE_VEC_STMTS (slp_node), new_stmt);
4044 if (slp && !slp_perm)
4045 continue;
4047 if (slp_perm)
4049 if (!vect_transform_slp_perm_load (stmt, dr_chain, gsi, vf,
4050 slp_node_instance, false))
4052 VEC_free (tree, heap, dr_chain);
4053 return false;
4056 else
4058 if (strided_load)
4060 if (!vect_transform_strided_load (stmt, dr_chain, group_size, gsi))
4061 return false;
4063 *vec_stmt = STMT_VINFO_VEC_STMT (stmt_info);
4064 VEC_free (tree, heap, dr_chain);
4065 dr_chain = VEC_alloc (tree, heap, group_size);
4067 else
4069 if (j == 0)
4070 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
4071 else
4072 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
4073 prev_stmt_info = vinfo_for_stmt (new_stmt);
4078 if (dr_chain)
4079 VEC_free (tree, heap, dr_chain);
4081 return true;
4084 /* Function vect_is_simple_cond.
4086 Input:
4087 LOOP - the loop that is being vectorized.
4088 COND - Condition that is checked for simple use.
4090 Returns whether a COND can be vectorized. Checks whether
4091 condition operands are supportable using vec_is_simple_use. */
4093 static bool
4094 vect_is_simple_cond (tree cond, loop_vec_info loop_vinfo)
4096 tree lhs, rhs;
4097 tree def;
4098 enum vect_def_type dt;
4100 if (!COMPARISON_CLASS_P (cond))
4101 return false;
4103 lhs = TREE_OPERAND (cond, 0);
4104 rhs = TREE_OPERAND (cond, 1);
4106 if (TREE_CODE (lhs) == SSA_NAME)
4108 gimple lhs_def_stmt = SSA_NAME_DEF_STMT (lhs);
4109 if (!vect_is_simple_use (lhs, loop_vinfo, NULL, &lhs_def_stmt, &def,
4110 &dt))
4111 return false;
4113 else if (TREE_CODE (lhs) != INTEGER_CST && TREE_CODE (lhs) != REAL_CST
4114 && TREE_CODE (lhs) != FIXED_CST)
4115 return false;
4117 if (TREE_CODE (rhs) == SSA_NAME)
4119 gimple rhs_def_stmt = SSA_NAME_DEF_STMT (rhs);
4120 if (!vect_is_simple_use (rhs, loop_vinfo, NULL, &rhs_def_stmt, &def,
4121 &dt))
4122 return false;
4124 else if (TREE_CODE (rhs) != INTEGER_CST && TREE_CODE (rhs) != REAL_CST
4125 && TREE_CODE (rhs) != FIXED_CST)
4126 return false;
4128 return true;
4131 /* vectorizable_condition.
4133 Check if STMT is conditional modify expression that can be vectorized.
4134 If VEC_STMT is also passed, vectorize the STMT: create a vectorized
4135 stmt using VEC_COND_EXPR to replace it, put it in VEC_STMT, and insert it
4136 at GSI.
4138 When STMT is vectorized as nested cycle, REDUC_DEF is the vector variable
4139 to be used at REDUC_INDEX (in then clause if REDUC_INDEX is 1, and in
4140 else caluse if it is 2).
4142 Return FALSE if not a vectorizable STMT, TRUE otherwise. */
4144 bool
4145 vectorizable_condition (gimple stmt, gimple_stmt_iterator *gsi,
4146 gimple *vec_stmt, tree reduc_def, int reduc_index)
4148 tree scalar_dest = NULL_TREE;
4149 tree vec_dest = NULL_TREE;
4150 tree op = NULL_TREE;
4151 tree cond_expr, then_clause, else_clause;
4152 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
4153 tree vectype = STMT_VINFO_VECTYPE (stmt_info);
4154 tree vec_cond_lhs = NULL_TREE, vec_cond_rhs = NULL_TREE;
4155 tree vec_then_clause = NULL_TREE, vec_else_clause = NULL_TREE;
4156 tree vec_compare, vec_cond_expr;
4157 tree new_temp;
4158 loop_vec_info loop_vinfo = STMT_VINFO_LOOP_VINFO (stmt_info);
4159 enum machine_mode vec_mode;
4160 tree def;
4161 enum vect_def_type dt, dts[4];
4162 int nunits = TYPE_VECTOR_SUBPARTS (vectype);
4163 int ncopies = LOOP_VINFO_VECT_FACTOR (loop_vinfo) / nunits;
4164 enum tree_code code;
4165 stmt_vec_info prev_stmt_info = NULL;
4166 int j;
4168 /* FORNOW: unsupported in basic block SLP. */
4169 gcc_assert (loop_vinfo);
4171 gcc_assert (ncopies >= 1);
4172 if (reduc_index && ncopies > 1)
4173 return false; /* FORNOW */
4175 if (!STMT_VINFO_RELEVANT_P (stmt_info))
4176 return false;
4178 if (STMT_VINFO_DEF_TYPE (stmt_info) != vect_internal_def
4179 && !(STMT_VINFO_DEF_TYPE (stmt_info) == vect_nested_cycle
4180 && reduc_def))
4181 return false;
4183 /* FORNOW: SLP not supported. */
4184 if (STMT_SLP_TYPE (stmt_info))
4185 return false;
4187 /* FORNOW: not yet supported. */
4188 if (STMT_VINFO_LIVE_P (stmt_info))
4190 if (vect_print_dump_info (REPORT_DETAILS))
4191 fprintf (vect_dump, "value used after loop.");
4192 return false;
4195 /* Is vectorizable conditional operation? */
4196 if (!is_gimple_assign (stmt))
4197 return false;
4199 code = gimple_assign_rhs_code (stmt);
4201 if (code != COND_EXPR)
4202 return false;
4204 gcc_assert (gimple_assign_single_p (stmt));
4205 op = gimple_assign_rhs1 (stmt);
4206 cond_expr = TREE_OPERAND (op, 0);
4207 then_clause = TREE_OPERAND (op, 1);
4208 else_clause = TREE_OPERAND (op, 2);
4210 if (!vect_is_simple_cond (cond_expr, loop_vinfo))
4211 return false;
4213 /* We do not handle two different vector types for the condition
4214 and the values. */
4215 if (!types_compatible_p (TREE_TYPE (TREE_OPERAND (cond_expr, 0)),
4216 TREE_TYPE (vectype)))
4217 return false;
4219 if (TREE_CODE (then_clause) == SSA_NAME)
4221 gimple then_def_stmt = SSA_NAME_DEF_STMT (then_clause);
4222 if (!vect_is_simple_use (then_clause, loop_vinfo, NULL,
4223 &then_def_stmt, &def, &dt))
4224 return false;
4226 else if (TREE_CODE (then_clause) != INTEGER_CST
4227 && TREE_CODE (then_clause) != REAL_CST
4228 && TREE_CODE (then_clause) != FIXED_CST)
4229 return false;
4231 if (TREE_CODE (else_clause) == SSA_NAME)
4233 gimple else_def_stmt = SSA_NAME_DEF_STMT (else_clause);
4234 if (!vect_is_simple_use (else_clause, loop_vinfo, NULL,
4235 &else_def_stmt, &def, &dt))
4236 return false;
4238 else if (TREE_CODE (else_clause) != INTEGER_CST
4239 && TREE_CODE (else_clause) != REAL_CST
4240 && TREE_CODE (else_clause) != FIXED_CST)
4241 return false;
4244 vec_mode = TYPE_MODE (vectype);
4246 if (!vec_stmt)
4248 STMT_VINFO_TYPE (stmt_info) = condition_vec_info_type;
4249 return expand_vec_cond_expr_p (TREE_TYPE (op), vec_mode);
4252 /* Transform */
4254 /* Handle def. */
4255 scalar_dest = gimple_assign_lhs (stmt);
4256 vec_dest = vect_create_destination_var (scalar_dest, vectype);
4258 /* Handle cond expr. */
4259 for (j = 0; j < ncopies; j++)
4261 gimple new_stmt;
4262 if (j == 0)
4264 gimple gtemp;
4265 vec_cond_lhs =
4266 vect_get_vec_def_for_operand (TREE_OPERAND (cond_expr, 0),
4267 stmt, NULL);
4268 vect_is_simple_use (TREE_OPERAND (cond_expr, 0), loop_vinfo,
4269 NULL, &gtemp, &def, &dts[0]);
4270 vec_cond_rhs =
4271 vect_get_vec_def_for_operand (TREE_OPERAND (cond_expr, 1),
4272 stmt, NULL);
4273 vect_is_simple_use (TREE_OPERAND (cond_expr, 1), loop_vinfo,
4274 NULL, &gtemp, &def, &dts[1]);
4275 if (reduc_index == 1)
4276 vec_then_clause = reduc_def;
4277 else
4279 vec_then_clause = vect_get_vec_def_for_operand (then_clause,
4280 stmt, NULL);
4281 vect_is_simple_use (then_clause, loop_vinfo,
4282 NULL, &gtemp, &def, &dts[2]);
4284 if (reduc_index == 2)
4285 vec_else_clause = reduc_def;
4286 else
4288 vec_else_clause = vect_get_vec_def_for_operand (else_clause,
4289 stmt, NULL);
4290 vect_is_simple_use (else_clause, loop_vinfo,
4291 NULL, &gtemp, &def, &dts[3]);
4294 else
4296 vec_cond_lhs = vect_get_vec_def_for_stmt_copy (dts[0], vec_cond_lhs);
4297 vec_cond_rhs = vect_get_vec_def_for_stmt_copy (dts[1], vec_cond_rhs);
4298 vec_then_clause = vect_get_vec_def_for_stmt_copy (dts[2],
4299 vec_then_clause);
4300 vec_else_clause = vect_get_vec_def_for_stmt_copy (dts[3],
4301 vec_else_clause);
4304 /* Arguments are ready. Create the new vector stmt. */
4305 vec_compare = build2 (TREE_CODE (cond_expr), vectype,
4306 vec_cond_lhs, vec_cond_rhs);
4307 vec_cond_expr = build3 (VEC_COND_EXPR, vectype,
4308 vec_compare, vec_then_clause, vec_else_clause);
4310 new_stmt = gimple_build_assign (vec_dest, vec_cond_expr);
4311 new_temp = make_ssa_name (vec_dest, new_stmt);
4312 gimple_assign_set_lhs (new_stmt, new_temp);
4313 vect_finish_stmt_generation (stmt, new_stmt, gsi);
4314 if (j == 0)
4315 STMT_VINFO_VEC_STMT (stmt_info) = *vec_stmt = new_stmt;
4316 else
4317 STMT_VINFO_RELATED_STMT (prev_stmt_info) = new_stmt;
4319 prev_stmt_info = vinfo_for_stmt (new_stmt);
4322 return true;
4326 /* Make sure the statement is vectorizable. */
4328 bool
4329 vect_analyze_stmt (gimple stmt, bool *need_to_vectorize, slp_tree node)
4331 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
4332 bb_vec_info bb_vinfo = STMT_VINFO_BB_VINFO (stmt_info);
4333 enum vect_relevant relevance = STMT_VINFO_RELEVANT (stmt_info);
4334 bool ok;
4335 tree scalar_type, vectype;
4337 if (vect_print_dump_info (REPORT_DETAILS))
4339 fprintf (vect_dump, "==> examining statement: ");
4340 print_gimple_stmt (vect_dump, stmt, 0, TDF_SLIM);
4343 if (gimple_has_volatile_ops (stmt))
4345 if (vect_print_dump_info (REPORT_UNVECTORIZED_LOCATIONS))
4346 fprintf (vect_dump, "not vectorized: stmt has volatile operands");
4348 return false;
4351 /* Skip stmts that do not need to be vectorized. In loops this is expected
4352 to include:
4353 - the COND_EXPR which is the loop exit condition
4354 - any LABEL_EXPRs in the loop
4355 - computations that are used only for array indexing or loop control.
4356 In basic blocks we only analyze statements that are a part of some SLP
4357 instance, therefore, all the statements are relevant. */
4359 if (!STMT_VINFO_RELEVANT_P (stmt_info)
4360 && !STMT_VINFO_LIVE_P (stmt_info))
4362 if (vect_print_dump_info (REPORT_DETAILS))
4363 fprintf (vect_dump, "irrelevant.");
4365 return true;
4368 switch (STMT_VINFO_DEF_TYPE (stmt_info))
4370 case vect_internal_def:
4371 break;
4373 case vect_reduction_def:
4374 case vect_nested_cycle:
4375 gcc_assert (!bb_vinfo && (relevance == vect_used_in_outer
4376 || relevance == vect_used_in_outer_by_reduction
4377 || relevance == vect_unused_in_scope));
4378 break;
4380 case vect_induction_def:
4381 case vect_constant_def:
4382 case vect_external_def:
4383 case vect_unknown_def_type:
4384 default:
4385 gcc_unreachable ();
4388 if (bb_vinfo)
4390 gcc_assert (PURE_SLP_STMT (stmt_info));
4392 scalar_type = TREE_TYPE (gimple_get_lhs (stmt));
4393 if (vect_print_dump_info (REPORT_DETAILS))
4395 fprintf (vect_dump, "get vectype for scalar type: ");
4396 print_generic_expr (vect_dump, scalar_type, TDF_SLIM);
4399 vectype = get_vectype_for_scalar_type (scalar_type);
4400 if (!vectype)
4402 if (vect_print_dump_info (REPORT_DETAILS))
4404 fprintf (vect_dump, "not SLPed: unsupported data-type ");
4405 print_generic_expr (vect_dump, scalar_type, TDF_SLIM);
4407 return false;
4410 if (vect_print_dump_info (REPORT_DETAILS))
4412 fprintf (vect_dump, "vectype: ");
4413 print_generic_expr (vect_dump, vectype, TDF_SLIM);
4416 STMT_VINFO_VECTYPE (stmt_info) = vectype;
4419 if (STMT_VINFO_RELEVANT_P (stmt_info))
4421 gcc_assert (!VECTOR_MODE_P (TYPE_MODE (gimple_expr_type (stmt))));
4422 gcc_assert (STMT_VINFO_VECTYPE (stmt_info));
4423 *need_to_vectorize = true;
4426 ok = true;
4427 if (!bb_vinfo
4428 && (STMT_VINFO_RELEVANT_P (stmt_info)
4429 || STMT_VINFO_DEF_TYPE (stmt_info) == vect_reduction_def))
4430 ok = (vectorizable_type_promotion (stmt, NULL, NULL, NULL)
4431 || vectorizable_type_demotion (stmt, NULL, NULL, NULL)
4432 || vectorizable_conversion (stmt, NULL, NULL, NULL)
4433 || vectorizable_operation (stmt, NULL, NULL, NULL)
4434 || vectorizable_assignment (stmt, NULL, NULL, NULL)
4435 || vectorizable_load (stmt, NULL, NULL, NULL, NULL)
4436 || vectorizable_call (stmt, NULL, NULL)
4437 || vectorizable_store (stmt, NULL, NULL, NULL)
4438 || vectorizable_reduction (stmt, NULL, NULL, NULL)
4439 || vectorizable_condition (stmt, NULL, NULL, NULL, 0));
4440 else
4442 if (bb_vinfo)
4443 ok = (vectorizable_operation (stmt, NULL, NULL, node)
4444 || vectorizable_assignment (stmt, NULL, NULL, node)
4445 || vectorizable_load (stmt, NULL, NULL, node, NULL)
4446 || vectorizable_store (stmt, NULL, NULL, node));
4449 if (!ok)
4451 if (vect_print_dump_info (REPORT_UNVECTORIZED_LOCATIONS))
4453 fprintf (vect_dump, "not vectorized: relevant stmt not ");
4454 fprintf (vect_dump, "supported: ");
4455 print_gimple_stmt (vect_dump, stmt, 0, TDF_SLIM);
4458 return false;
4461 if (bb_vinfo)
4462 return true;
4464 /* Stmts that are (also) "live" (i.e. - that are used out of the loop)
4465 need extra handling, except for vectorizable reductions. */
4466 if (STMT_VINFO_LIVE_P (stmt_info)
4467 && STMT_VINFO_TYPE (stmt_info) != reduc_vec_info_type)
4468 ok = vectorizable_live_operation (stmt, NULL, NULL);
4470 if (!ok)
4472 if (vect_print_dump_info (REPORT_UNVECTORIZED_LOCATIONS))
4474 fprintf (vect_dump, "not vectorized: live stmt not ");
4475 fprintf (vect_dump, "supported: ");
4476 print_gimple_stmt (vect_dump, stmt, 0, TDF_SLIM);
4479 return false;
4482 if (!PURE_SLP_STMT (stmt_info))
4484 /* Groups of strided accesses whose size is not a power of 2 are not
4485 vectorizable yet using loop-vectorization. Therefore, if this stmt
4486 feeds non-SLP-able stmts (i.e., this stmt has to be both SLPed and
4487 loop-based vectorized), the loop cannot be vectorized. */
4488 if (STMT_VINFO_STRIDED_ACCESS (stmt_info)
4489 && exact_log2 (DR_GROUP_SIZE (vinfo_for_stmt (
4490 DR_GROUP_FIRST_DR (stmt_info)))) == -1)
4492 if (vect_print_dump_info (REPORT_DETAILS))
4494 fprintf (vect_dump, "not vectorized: the size of group "
4495 "of strided accesses is not a power of 2");
4496 print_gimple_stmt (vect_dump, stmt, 0, TDF_SLIM);
4499 return false;
4503 return true;
4507 /* Function vect_transform_stmt.
4509 Create a vectorized stmt to replace STMT, and insert it at BSI. */
4511 bool
4512 vect_transform_stmt (gimple stmt, gimple_stmt_iterator *gsi,
4513 bool *strided_store, slp_tree slp_node,
4514 slp_instance slp_node_instance)
4516 bool is_store = false;
4517 gimple vec_stmt = NULL;
4518 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
4519 gimple orig_stmt_in_pattern;
4520 bool done;
4522 switch (STMT_VINFO_TYPE (stmt_info))
4524 case type_demotion_vec_info_type:
4525 done = vectorizable_type_demotion (stmt, gsi, &vec_stmt, slp_node);
4526 gcc_assert (done);
4527 break;
4529 case type_promotion_vec_info_type:
4530 done = vectorizable_type_promotion (stmt, gsi, &vec_stmt, slp_node);
4531 gcc_assert (done);
4532 break;
4534 case type_conversion_vec_info_type:
4535 done = vectorizable_conversion (stmt, gsi, &vec_stmt, slp_node);
4536 gcc_assert (done);
4537 break;
4539 case induc_vec_info_type:
4540 gcc_assert (!slp_node);
4541 done = vectorizable_induction (stmt, gsi, &vec_stmt);
4542 gcc_assert (done);
4543 break;
4545 case op_vec_info_type:
4546 done = vectorizable_operation (stmt, gsi, &vec_stmt, slp_node);
4547 gcc_assert (done);
4548 break;
4550 case assignment_vec_info_type:
4551 done = vectorizable_assignment (stmt, gsi, &vec_stmt, slp_node);
4552 gcc_assert (done);
4553 break;
4555 case load_vec_info_type:
4556 done = vectorizable_load (stmt, gsi, &vec_stmt, slp_node,
4557 slp_node_instance);
4558 gcc_assert (done);
4559 break;
4561 case store_vec_info_type:
4562 done = vectorizable_store (stmt, gsi, &vec_stmt, slp_node);
4563 gcc_assert (done);
4564 if (STMT_VINFO_STRIDED_ACCESS (stmt_info) && !slp_node)
4566 /* In case of interleaving, the whole chain is vectorized when the
4567 last store in the chain is reached. Store stmts before the last
4568 one are skipped, and there vec_stmt_info shouldn't be freed
4569 meanwhile. */
4570 *strided_store = true;
4571 if (STMT_VINFO_VEC_STMT (stmt_info))
4572 is_store = true;
4574 else
4575 is_store = true;
4576 break;
4578 case condition_vec_info_type:
4579 gcc_assert (!slp_node);
4580 done = vectorizable_condition (stmt, gsi, &vec_stmt, NULL, 0);
4581 gcc_assert (done);
4582 break;
4584 case call_vec_info_type:
4585 gcc_assert (!slp_node);
4586 done = vectorizable_call (stmt, gsi, &vec_stmt);
4587 break;
4589 case reduc_vec_info_type:
4590 done = vectorizable_reduction (stmt, gsi, &vec_stmt, slp_node);
4591 gcc_assert (done);
4592 break;
4594 default:
4595 if (!STMT_VINFO_LIVE_P (stmt_info))
4597 if (vect_print_dump_info (REPORT_DETAILS))
4598 fprintf (vect_dump, "stmt not supported.");
4599 gcc_unreachable ();
4603 /* Handle inner-loop stmts whose DEF is used in the loop-nest that
4604 is being vectorized, but outside the immediately enclosing loop. */
4605 if (vec_stmt
4606 && STMT_VINFO_LOOP_VINFO (stmt_info)
4607 && nested_in_vect_loop_p (LOOP_VINFO_LOOP (
4608 STMT_VINFO_LOOP_VINFO (stmt_info)), stmt)
4609 && STMT_VINFO_TYPE (stmt_info) != reduc_vec_info_type
4610 && (STMT_VINFO_RELEVANT (stmt_info) == vect_used_in_outer
4611 || STMT_VINFO_RELEVANT (stmt_info) ==
4612 vect_used_in_outer_by_reduction))
4614 struct loop *innerloop = LOOP_VINFO_LOOP (
4615 STMT_VINFO_LOOP_VINFO (stmt_info))->inner;
4616 imm_use_iterator imm_iter;
4617 use_operand_p use_p;
4618 tree scalar_dest;
4619 gimple exit_phi;
4621 if (vect_print_dump_info (REPORT_DETAILS))
4622 fprintf (vect_dump, "Record the vdef for outer-loop vectorization.");
4624 /* Find the relevant loop-exit phi-node, and reord the vec_stmt there
4625 (to be used when vectorizing outer-loop stmts that use the DEF of
4626 STMT). */
4627 if (gimple_code (stmt) == GIMPLE_PHI)
4628 scalar_dest = PHI_RESULT (stmt);
4629 else
4630 scalar_dest = gimple_assign_lhs (stmt);
4632 FOR_EACH_IMM_USE_FAST (use_p, imm_iter, scalar_dest)
4634 if (!flow_bb_inside_loop_p (innerloop, gimple_bb (USE_STMT (use_p))))
4636 exit_phi = USE_STMT (use_p);
4637 STMT_VINFO_VEC_STMT (vinfo_for_stmt (exit_phi)) = vec_stmt;
4642 /* Handle stmts whose DEF is used outside the loop-nest that is
4643 being vectorized. */
4644 if (STMT_VINFO_LIVE_P (stmt_info)
4645 && STMT_VINFO_TYPE (stmt_info) != reduc_vec_info_type)
4647 done = vectorizable_live_operation (stmt, gsi, &vec_stmt);
4648 gcc_assert (done);
4651 if (vec_stmt)
4653 STMT_VINFO_VEC_STMT (stmt_info) = vec_stmt;
4654 orig_stmt_in_pattern = STMT_VINFO_RELATED_STMT (stmt_info);
4655 if (orig_stmt_in_pattern)
4657 stmt_vec_info stmt_vinfo = vinfo_for_stmt (orig_stmt_in_pattern);
4658 /* STMT was inserted by the vectorizer to replace a computation idiom.
4659 ORIG_STMT_IN_PATTERN is a stmt in the original sequence that
4660 computed this idiom. We need to record a pointer to VEC_STMT in
4661 the stmt_info of ORIG_STMT_IN_PATTERN. See more details in the
4662 documentation of vect_pattern_recog. */
4663 if (STMT_VINFO_IN_PATTERN_P (stmt_vinfo))
4665 gcc_assert (STMT_VINFO_RELATED_STMT (stmt_vinfo) == stmt);
4666 STMT_VINFO_VEC_STMT (stmt_vinfo) = vec_stmt;
4671 return is_store;
4675 /* Remove a group of stores (for SLP or interleaving), free their
4676 stmt_vec_info. */
4678 void
4679 vect_remove_stores (gimple first_stmt)
4681 gimple next = first_stmt;
4682 gimple tmp;
4683 gimple_stmt_iterator next_si;
4685 while (next)
4687 /* Free the attached stmt_vec_info and remove the stmt. */
4688 next_si = gsi_for_stmt (next);
4689 gsi_remove (&next_si, true);
4690 tmp = DR_GROUP_NEXT_DR (vinfo_for_stmt (next));
4691 free_stmt_vec_info (next);
4692 next = tmp;
4697 /* Function new_stmt_vec_info.
4699 Create and initialize a new stmt_vec_info struct for STMT. */
4701 stmt_vec_info
4702 new_stmt_vec_info (gimple stmt, loop_vec_info loop_vinfo,
4703 bb_vec_info bb_vinfo)
4705 stmt_vec_info res;
4706 res = (stmt_vec_info) xcalloc (1, sizeof (struct _stmt_vec_info));
4708 STMT_VINFO_TYPE (res) = undef_vec_info_type;
4709 STMT_VINFO_STMT (res) = stmt;
4710 STMT_VINFO_LOOP_VINFO (res) = loop_vinfo;
4711 STMT_VINFO_BB_VINFO (res) = bb_vinfo;
4712 STMT_VINFO_RELEVANT (res) = vect_unused_in_scope;
4713 STMT_VINFO_LIVE_P (res) = false;
4714 STMT_VINFO_VECTYPE (res) = NULL;
4715 STMT_VINFO_VEC_STMT (res) = NULL;
4716 STMT_VINFO_VECTORIZABLE (res) = true;
4717 STMT_VINFO_IN_PATTERN_P (res) = false;
4718 STMT_VINFO_RELATED_STMT (res) = NULL;
4719 STMT_VINFO_DATA_REF (res) = NULL;
4721 STMT_VINFO_DR_BASE_ADDRESS (res) = NULL;
4722 STMT_VINFO_DR_OFFSET (res) = NULL;
4723 STMT_VINFO_DR_INIT (res) = NULL;
4724 STMT_VINFO_DR_STEP (res) = NULL;
4725 STMT_VINFO_DR_ALIGNED_TO (res) = NULL;
4727 if (gimple_code (stmt) == GIMPLE_PHI
4728 && is_loop_header_bb_p (gimple_bb (stmt)))
4729 STMT_VINFO_DEF_TYPE (res) = vect_unknown_def_type;
4730 else
4731 STMT_VINFO_DEF_TYPE (res) = vect_internal_def;
4733 STMT_VINFO_SAME_ALIGN_REFS (res) = VEC_alloc (dr_p, heap, 5);
4734 STMT_VINFO_INSIDE_OF_LOOP_COST (res) = 0;
4735 STMT_VINFO_OUTSIDE_OF_LOOP_COST (res) = 0;
4736 STMT_SLP_TYPE (res) = loop_vect;
4737 DR_GROUP_FIRST_DR (res) = NULL;
4738 DR_GROUP_NEXT_DR (res) = NULL;
4739 DR_GROUP_SIZE (res) = 0;
4740 DR_GROUP_STORE_COUNT (res) = 0;
4741 DR_GROUP_GAP (res) = 0;
4742 DR_GROUP_SAME_DR_STMT (res) = NULL;
4743 DR_GROUP_READ_WRITE_DEPENDENCE (res) = false;
4745 return res;
4749 /* Create a hash table for stmt_vec_info. */
4751 void
4752 init_stmt_vec_info_vec (void)
4754 gcc_assert (!stmt_vec_info_vec);
4755 stmt_vec_info_vec = VEC_alloc (vec_void_p, heap, 50);
4759 /* Free hash table for stmt_vec_info. */
4761 void
4762 free_stmt_vec_info_vec (void)
4764 gcc_assert (stmt_vec_info_vec);
4765 VEC_free (vec_void_p, heap, stmt_vec_info_vec);
4769 /* Free stmt vectorization related info. */
4771 void
4772 free_stmt_vec_info (gimple stmt)
4774 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
4776 if (!stmt_info)
4777 return;
4779 VEC_free (dr_p, heap, STMT_VINFO_SAME_ALIGN_REFS (stmt_info));
4780 set_vinfo_for_stmt (stmt, NULL);
4781 free (stmt_info);
4785 /* Function get_vectype_for_scalar_type.
4787 Returns the vector type corresponding to SCALAR_TYPE as supported
4788 by the target. */
4790 tree
4791 get_vectype_for_scalar_type (tree scalar_type)
4793 enum machine_mode inner_mode = TYPE_MODE (scalar_type);
4794 unsigned int nbytes = GET_MODE_SIZE (inner_mode);
4795 int nunits;
4796 tree vectype;
4798 if (nbytes == 0
4799 || (nbytes >= targetm.vectorize.units_per_simd_word (inner_mode)))
4800 return NULL_TREE;
4802 /* We can't build a vector type of elements with alignment bigger than
4803 their size. */
4804 if (nbytes < TYPE_ALIGN_UNIT (scalar_type))
4805 return NULL_TREE;
4807 /* If we'd build a vector type of elements whose mode precision doesn't
4808 match their types precision we'll get mismatched types on vector
4809 extracts via BIT_FIELD_REFs. This effectively means we disable
4810 vectorization of bool and/or enum types in some languages. */
4811 if (INTEGRAL_TYPE_P (scalar_type)
4812 && GET_MODE_BITSIZE (inner_mode) != TYPE_PRECISION (scalar_type))
4813 return NULL_TREE;
4815 /* FORNOW: Only a single vector size per mode
4816 (TARGET_VECTORIZE_UNITS_PER_SIMD_WORD) is expected. */
4817 nunits = targetm.vectorize.units_per_simd_word (inner_mode) / nbytes;
4819 vectype = build_vector_type (scalar_type, nunits);
4820 if (vect_print_dump_info (REPORT_DETAILS))
4822 fprintf (vect_dump, "get vectype with %d units of type ", nunits);
4823 print_generic_expr (vect_dump, scalar_type, TDF_SLIM);
4826 if (!vectype)
4827 return NULL_TREE;
4829 if (vect_print_dump_info (REPORT_DETAILS))
4831 fprintf (vect_dump, "vectype: ");
4832 print_generic_expr (vect_dump, vectype, TDF_SLIM);
4835 if (!VECTOR_MODE_P (TYPE_MODE (vectype))
4836 && !INTEGRAL_MODE_P (TYPE_MODE (vectype)))
4838 if (vect_print_dump_info (REPORT_DETAILS))
4839 fprintf (vect_dump, "mode not supported by target.");
4840 return NULL_TREE;
4843 return vectype;
4846 /* Function get_same_sized_vectype
4848 Returns a vector type corresponding to SCALAR_TYPE of size
4849 VECTOR_TYPE if supported by the target. */
4851 tree
4852 get_same_sized_vectype (tree scalar_type, tree vector_type ATTRIBUTE_UNUSED)
4854 return get_vectype_for_scalar_type (scalar_type);
4857 /* Function vect_is_simple_use.
4859 Input:
4860 LOOP_VINFO - the vect info of the loop that is being vectorized.
4861 BB_VINFO - the vect info of the basic block that is being vectorized.
4862 OPERAND - operand of a stmt in the loop or bb.
4863 DEF - the defining stmt in case OPERAND is an SSA_NAME.
4865 Returns whether a stmt with OPERAND can be vectorized.
4866 For loops, supportable operands are constants, loop invariants, and operands
4867 that are defined by the current iteration of the loop. Unsupportable
4868 operands are those that are defined by a previous iteration of the loop (as
4869 is the case in reduction/induction computations).
4870 For basic blocks, supportable operands are constants and bb invariants.
4871 For now, operands defined outside the basic block are not supported. */
4873 bool
4874 vect_is_simple_use (tree operand, loop_vec_info loop_vinfo,
4875 bb_vec_info bb_vinfo, gimple *def_stmt,
4876 tree *def, enum vect_def_type *dt)
4878 basic_block bb;
4879 stmt_vec_info stmt_vinfo;
4880 struct loop *loop = NULL;
4882 if (loop_vinfo)
4883 loop = LOOP_VINFO_LOOP (loop_vinfo);
4885 *def_stmt = NULL;
4886 *def = NULL_TREE;
4888 if (vect_print_dump_info (REPORT_DETAILS))
4890 fprintf (vect_dump, "vect_is_simple_use: operand ");
4891 print_generic_expr (vect_dump, operand, TDF_SLIM);
4894 if (TREE_CODE (operand) == INTEGER_CST || TREE_CODE (operand) == REAL_CST)
4896 *dt = vect_constant_def;
4897 return true;
4900 if (is_gimple_min_invariant (operand))
4902 *def = operand;
4903 *dt = vect_external_def;
4904 return true;
4907 if (TREE_CODE (operand) == PAREN_EXPR)
4909 if (vect_print_dump_info (REPORT_DETAILS))
4910 fprintf (vect_dump, "non-associatable copy.");
4911 operand = TREE_OPERAND (operand, 0);
4914 if (TREE_CODE (operand) != SSA_NAME)
4916 if (vect_print_dump_info (REPORT_DETAILS))
4917 fprintf (vect_dump, "not ssa-name.");
4918 return false;
4921 *def_stmt = SSA_NAME_DEF_STMT (operand);
4922 if (*def_stmt == NULL)
4924 if (vect_print_dump_info (REPORT_DETAILS))
4925 fprintf (vect_dump, "no def_stmt.");
4926 return false;
4929 if (vect_print_dump_info (REPORT_DETAILS))
4931 fprintf (vect_dump, "def_stmt: ");
4932 print_gimple_stmt (vect_dump, *def_stmt, 0, TDF_SLIM);
4935 /* Empty stmt is expected only in case of a function argument.
4936 (Otherwise - we expect a phi_node or a GIMPLE_ASSIGN). */
4937 if (gimple_nop_p (*def_stmt))
4939 *def = operand;
4940 *dt = vect_external_def;
4941 return true;
4944 bb = gimple_bb (*def_stmt);
4946 if ((loop && !flow_bb_inside_loop_p (loop, bb))
4947 || (!loop && bb != BB_VINFO_BB (bb_vinfo))
4948 || (!loop && gimple_code (*def_stmt) == GIMPLE_PHI))
4949 *dt = vect_external_def;
4950 else
4952 stmt_vinfo = vinfo_for_stmt (*def_stmt);
4953 *dt = STMT_VINFO_DEF_TYPE (stmt_vinfo);
4956 if (*dt == vect_unknown_def_type)
4958 if (vect_print_dump_info (REPORT_DETAILS))
4959 fprintf (vect_dump, "Unsupported pattern.");
4960 return false;
4963 if (vect_print_dump_info (REPORT_DETAILS))
4964 fprintf (vect_dump, "type of def: %d.",*dt);
4966 switch (gimple_code (*def_stmt))
4968 case GIMPLE_PHI:
4969 *def = gimple_phi_result (*def_stmt);
4970 break;
4972 case GIMPLE_ASSIGN:
4973 *def = gimple_assign_lhs (*def_stmt);
4974 break;
4976 case GIMPLE_CALL:
4977 *def = gimple_call_lhs (*def_stmt);
4978 if (*def != NULL)
4979 break;
4980 /* FALLTHRU */
4981 default:
4982 if (vect_print_dump_info (REPORT_DETAILS))
4983 fprintf (vect_dump, "unsupported defining stmt: ");
4984 return false;
4987 return true;
4990 /* Function vect_is_simple_use_1.
4992 Same as vect_is_simple_use_1 but also determines the vector operand
4993 type of OPERAND and stores it to *VECTYPE. If the definition of
4994 OPERAND is vect_uninitialized_def, vect_constant_def or
4995 vect_external_def *VECTYPE will be set to NULL_TREE and the caller
4996 is responsible to compute the best suited vector type for the
4997 scalar operand. */
4999 bool
5000 vect_is_simple_use_1 (tree operand, loop_vec_info loop_vinfo,
5001 bb_vec_info bb_vinfo, gimple *def_stmt,
5002 tree *def, enum vect_def_type *dt, tree *vectype)
5004 if (!vect_is_simple_use (operand, loop_vinfo, bb_vinfo, def_stmt, def, dt))
5005 return false;
5007 /* Now get a vector type if the def is internal, otherwise supply
5008 NULL_TREE and leave it up to the caller to figure out a proper
5009 type for the use stmt. */
5010 if (*dt == vect_internal_def
5011 || *dt == vect_induction_def
5012 || *dt == vect_reduction_def
5013 || *dt == vect_double_reduction_def
5014 || *dt == vect_nested_cycle)
5016 stmt_vec_info stmt_info = vinfo_for_stmt (*def_stmt);
5017 if (STMT_VINFO_IN_PATTERN_P (stmt_info))
5018 stmt_info = vinfo_for_stmt (STMT_VINFO_RELATED_STMT (stmt_info));
5019 *vectype = STMT_VINFO_VECTYPE (stmt_info);
5020 gcc_assert (*vectype != NULL_TREE);
5022 else if (*dt == vect_uninitialized_def
5023 || *dt == vect_constant_def
5024 || *dt == vect_external_def)
5025 *vectype = NULL_TREE;
5026 else
5027 gcc_unreachable ();
5029 return true;
5033 /* Function supportable_widening_operation
5035 Check whether an operation represented by the code CODE is a
5036 widening operation that is supported by the target platform in
5037 vector form (i.e., when operating on arguments of type VECTYPE_IN
5038 producing a result of type VECTYPE_OUT).
5040 Widening operations we currently support are NOP (CONVERT), FLOAT
5041 and WIDEN_MULT. This function checks if these operations are supported
5042 by the target platform either directly (via vector tree-codes), or via
5043 target builtins.
5045 Output:
5046 - CODE1 and CODE2 are codes of vector operations to be used when
5047 vectorizing the operation, if available.
5048 - DECL1 and DECL2 are decls of target builtin functions to be used
5049 when vectorizing the operation, if available. In this case,
5050 CODE1 and CODE2 are CALL_EXPR.
5051 - MULTI_STEP_CVT determines the number of required intermediate steps in
5052 case of multi-step conversion (like char->short->int - in that case
5053 MULTI_STEP_CVT will be 1).
5054 - INTERM_TYPES contains the intermediate type required to perform the
5055 widening operation (short in the above example). */
5057 bool
5058 supportable_widening_operation (enum tree_code code, gimple stmt,
5059 tree vectype_out, tree vectype_in,
5060 tree *decl1, tree *decl2,
5061 enum tree_code *code1, enum tree_code *code2,
5062 int *multi_step_cvt,
5063 VEC (tree, heap) **interm_types)
5065 stmt_vec_info stmt_info = vinfo_for_stmt (stmt);
5066 loop_vec_info loop_info = STMT_VINFO_LOOP_VINFO (stmt_info);
5067 struct loop *vect_loop = LOOP_VINFO_LOOP (loop_info);
5068 bool ordered_p;
5069 enum machine_mode vec_mode;
5070 enum insn_code icode1, icode2;
5071 optab optab1, optab2;
5072 tree vectype = vectype_in;
5073 tree wide_vectype = vectype_out;
5074 enum tree_code c1, c2;
5076 /* The result of a vectorized widening operation usually requires two vectors
5077 (because the widened results do not fit int one vector). The generated
5078 vector results would normally be expected to be generated in the same
5079 order as in the original scalar computation, i.e. if 8 results are
5080 generated in each vector iteration, they are to be organized as follows:
5081 vect1: [res1,res2,res3,res4], vect2: [res5,res6,res7,res8].
5083 However, in the special case that the result of the widening operation is
5084 used in a reduction computation only, the order doesn't matter (because
5085 when vectorizing a reduction we change the order of the computation).
5086 Some targets can take advantage of this and generate more efficient code.
5087 For example, targets like Altivec, that support widen_mult using a sequence
5088 of {mult_even,mult_odd} generate the following vectors:
5089 vect1: [res1,res3,res5,res7], vect2: [res2,res4,res6,res8].
5091 When vectorizing outer-loops, we execute the inner-loop sequentially
5092 (each vectorized inner-loop iteration contributes to VF outer-loop
5093 iterations in parallel). We therefore don't allow to change the order
5094 of the computation in the inner-loop during outer-loop vectorization. */
5096 if (STMT_VINFO_RELEVANT (stmt_info) == vect_used_by_reduction
5097 && !nested_in_vect_loop_p (vect_loop, stmt))
5098 ordered_p = false;
5099 else
5100 ordered_p = true;
5102 if (!ordered_p
5103 && code == WIDEN_MULT_EXPR
5104 && targetm.vectorize.builtin_mul_widen_even
5105 && targetm.vectorize.builtin_mul_widen_even (vectype)
5106 && targetm.vectorize.builtin_mul_widen_odd
5107 && targetm.vectorize.builtin_mul_widen_odd (vectype))
5109 if (vect_print_dump_info (REPORT_DETAILS))
5110 fprintf (vect_dump, "Unordered widening operation detected.");
5112 *code1 = *code2 = CALL_EXPR;
5113 *decl1 = targetm.vectorize.builtin_mul_widen_even (vectype);
5114 *decl2 = targetm.vectorize.builtin_mul_widen_odd (vectype);
5115 return true;
5118 switch (code)
5120 case WIDEN_MULT_EXPR:
5121 if (BYTES_BIG_ENDIAN)
5123 c1 = VEC_WIDEN_MULT_HI_EXPR;
5124 c2 = VEC_WIDEN_MULT_LO_EXPR;
5126 else
5128 c2 = VEC_WIDEN_MULT_HI_EXPR;
5129 c1 = VEC_WIDEN_MULT_LO_EXPR;
5131 break;
5133 CASE_CONVERT:
5134 if (BYTES_BIG_ENDIAN)
5136 c1 = VEC_UNPACK_HI_EXPR;
5137 c2 = VEC_UNPACK_LO_EXPR;
5139 else
5141 c2 = VEC_UNPACK_HI_EXPR;
5142 c1 = VEC_UNPACK_LO_EXPR;
5144 break;
5146 case FLOAT_EXPR:
5147 if (BYTES_BIG_ENDIAN)
5149 c1 = VEC_UNPACK_FLOAT_HI_EXPR;
5150 c2 = VEC_UNPACK_FLOAT_LO_EXPR;
5152 else
5154 c2 = VEC_UNPACK_FLOAT_HI_EXPR;
5155 c1 = VEC_UNPACK_FLOAT_LO_EXPR;
5157 break;
5159 case FIX_TRUNC_EXPR:
5160 /* ??? Not yet implemented due to missing VEC_UNPACK_FIX_TRUNC_HI_EXPR/
5161 VEC_UNPACK_FIX_TRUNC_LO_EXPR tree codes and optabs used for
5162 computing the operation. */
5163 return false;
5165 default:
5166 gcc_unreachable ();
5169 if (code == FIX_TRUNC_EXPR)
5171 /* The signedness is determined from output operand. */
5172 optab1 = optab_for_tree_code (c1, vectype_out, optab_default);
5173 optab2 = optab_for_tree_code (c2, vectype_out, optab_default);
5175 else
5177 optab1 = optab_for_tree_code (c1, vectype, optab_default);
5178 optab2 = optab_for_tree_code (c2, vectype, optab_default);
5181 if (!optab1 || !optab2)
5182 return false;
5184 vec_mode = TYPE_MODE (vectype);
5185 if ((icode1 = optab_handler (optab1, vec_mode)) == CODE_FOR_nothing
5186 || (icode2 = optab_handler (optab2, vec_mode)) == CODE_FOR_nothing)
5187 return false;
5189 /* Check if it's a multi-step conversion that can be done using intermediate
5190 types. */
5191 if (insn_data[icode1].operand[0].mode != TYPE_MODE (wide_vectype)
5192 || insn_data[icode2].operand[0].mode != TYPE_MODE (wide_vectype))
5194 int i;
5195 tree prev_type = vectype, intermediate_type;
5196 enum machine_mode intermediate_mode, prev_mode = vec_mode;
5197 optab optab3, optab4;
5199 if (!CONVERT_EXPR_CODE_P (code))
5200 return false;
5202 *code1 = c1;
5203 *code2 = c2;
5205 /* We assume here that there will not be more than MAX_INTERM_CVT_STEPS
5206 intermediate steps in promotion sequence. We try
5207 MAX_INTERM_CVT_STEPS to get to NARROW_VECTYPE, and fail if we do
5208 not. */
5209 *interm_types = VEC_alloc (tree, heap, MAX_INTERM_CVT_STEPS);
5210 for (i = 0; i < 3; i++)
5212 intermediate_mode = insn_data[icode1].operand[0].mode;
5213 intermediate_type = lang_hooks.types.type_for_mode (intermediate_mode,
5214 TYPE_UNSIGNED (prev_type));
5215 optab3 = optab_for_tree_code (c1, intermediate_type, optab_default);
5216 optab4 = optab_for_tree_code (c2, intermediate_type, optab_default);
5218 if (!optab3 || !optab4
5219 || ((icode1 = optab_handler (optab1, prev_mode))
5220 == CODE_FOR_nothing)
5221 || insn_data[icode1].operand[0].mode != intermediate_mode
5222 || ((icode2 = optab_handler (optab2, prev_mode))
5223 == CODE_FOR_nothing)
5224 || insn_data[icode2].operand[0].mode != intermediate_mode
5225 || ((icode1 = optab_handler (optab3, intermediate_mode))
5226 == CODE_FOR_nothing)
5227 || ((icode2 = optab_handler (optab4, intermediate_mode))
5228 == CODE_FOR_nothing))
5229 return false;
5231 VEC_quick_push (tree, *interm_types, intermediate_type);
5232 (*multi_step_cvt)++;
5234 if (insn_data[icode1].operand[0].mode == TYPE_MODE (wide_vectype)
5235 && insn_data[icode2].operand[0].mode == TYPE_MODE (wide_vectype))
5236 return true;
5238 prev_type = intermediate_type;
5239 prev_mode = intermediate_mode;
5242 return false;
5245 *code1 = c1;
5246 *code2 = c2;
5247 return true;
5251 /* Function supportable_narrowing_operation
5253 Check whether an operation represented by the code CODE is a
5254 narrowing operation that is supported by the target platform in
5255 vector form (i.e., when operating on arguments of type VECTYPE_IN
5256 and producing a result of type VECTYPE_OUT).
5258 Narrowing operations we currently support are NOP (CONVERT) and
5259 FIX_TRUNC. This function checks if these operations are supported by
5260 the target platform directly via vector tree-codes.
5262 Output:
5263 - CODE1 is the code of a vector operation to be used when
5264 vectorizing the operation, if available.
5265 - MULTI_STEP_CVT determines the number of required intermediate steps in
5266 case of multi-step conversion (like int->short->char - in that case
5267 MULTI_STEP_CVT will be 1).
5268 - INTERM_TYPES contains the intermediate type required to perform the
5269 narrowing operation (short in the above example). */
5271 bool
5272 supportable_narrowing_operation (enum tree_code code,
5273 tree vectype_out, tree vectype_in,
5274 enum tree_code *code1, int *multi_step_cvt,
5275 VEC (tree, heap) **interm_types)
5277 enum machine_mode vec_mode;
5278 enum insn_code icode1;
5279 optab optab1, interm_optab;
5280 tree vectype = vectype_in;
5281 tree narrow_vectype = vectype_out;
5282 enum tree_code c1;
5283 tree intermediate_type, prev_type;
5284 int i;
5286 switch (code)
5288 CASE_CONVERT:
5289 c1 = VEC_PACK_TRUNC_EXPR;
5290 break;
5292 case FIX_TRUNC_EXPR:
5293 c1 = VEC_PACK_FIX_TRUNC_EXPR;
5294 break;
5296 case FLOAT_EXPR:
5297 /* ??? Not yet implemented due to missing VEC_PACK_FLOAT_EXPR
5298 tree code and optabs used for computing the operation. */
5299 return false;
5301 default:
5302 gcc_unreachable ();
5305 if (code == FIX_TRUNC_EXPR)
5306 /* The signedness is determined from output operand. */
5307 optab1 = optab_for_tree_code (c1, vectype_out, optab_default);
5308 else
5309 optab1 = optab_for_tree_code (c1, vectype, optab_default);
5311 if (!optab1)
5312 return false;
5314 vec_mode = TYPE_MODE (vectype);
5315 if ((icode1 = optab_handler (optab1, vec_mode)) == CODE_FOR_nothing)
5316 return false;
5318 /* Check if it's a multi-step conversion that can be done using intermediate
5319 types. */
5320 if (insn_data[icode1].operand[0].mode != TYPE_MODE (narrow_vectype))
5322 enum machine_mode intermediate_mode, prev_mode = vec_mode;
5324 *code1 = c1;
5325 prev_type = vectype;
5326 /* We assume here that there will not be more than MAX_INTERM_CVT_STEPS
5327 intermediate steps in promotion sequence. We try
5328 MAX_INTERM_CVT_STEPS to get to NARROW_VECTYPE, and fail if we do
5329 not. */
5330 *interm_types = VEC_alloc (tree, heap, MAX_INTERM_CVT_STEPS);
5331 for (i = 0; i < 3; i++)
5333 intermediate_mode = insn_data[icode1].operand[0].mode;
5334 intermediate_type = lang_hooks.types.type_for_mode (intermediate_mode,
5335 TYPE_UNSIGNED (prev_type));
5336 interm_optab = optab_for_tree_code (c1, intermediate_type,
5337 optab_default);
5338 if (!interm_optab
5339 || ((icode1 = optab_handler (optab1, prev_mode))
5340 == CODE_FOR_nothing)
5341 || insn_data[icode1].operand[0].mode != intermediate_mode
5342 || ((icode1 = optab_handler (interm_optab, intermediate_mode))
5343 == CODE_FOR_nothing))
5344 return false;
5346 VEC_quick_push (tree, *interm_types, intermediate_type);
5347 (*multi_step_cvt)++;
5349 if (insn_data[icode1].operand[0].mode == TYPE_MODE (narrow_vectype))
5350 return true;
5352 prev_type = intermediate_type;
5353 prev_mode = intermediate_mode;
5356 return false;
5359 *code1 = c1;
5360 return true;