* asan.c (create_cond_insert_point): Maintain profile.
[official-gcc.git] / gcc / tree-tailcall.c
blob0e637147e8c57f7ee95f5769fe2c65e0d9162c74
1 /* Tail call optimization on trees.
2 Copyright (C) 2003-2017 Free Software Foundation, Inc.
4 This file is part of GCC.
6 GCC is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 3, or (at your option)
9 any later version.
11 GCC is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with GCC; see the file COPYING3. If not see
18 <http://www.gnu.org/licenses/>. */
20 #include "config.h"
21 #include "system.h"
22 #include "coretypes.h"
23 #include "backend.h"
24 #include "rtl.h"
25 #include "tree.h"
26 #include "gimple.h"
27 #include "cfghooks.h"
28 #include "tree-pass.h"
29 #include "ssa.h"
30 #include "cgraph.h"
31 #include "gimple-pretty-print.h"
32 #include "fold-const.h"
33 #include "stor-layout.h"
34 #include "gimple-iterator.h"
35 #include "gimplify-me.h"
36 #include "tree-cfg.h"
37 #include "tree-into-ssa.h"
38 #include "tree-dfa.h"
39 #include "except.h"
40 #include "dbgcnt.h"
41 #include "cfgloop.h"
42 #include "common/common-target.h"
43 #include "ipa-utils.h"
45 /* The file implements the tail recursion elimination. It is also used to
46 analyze the tail calls in general, passing the results to the rtl level
47 where they are used for sibcall optimization.
49 In addition to the standard tail recursion elimination, we handle the most
50 trivial cases of making the call tail recursive by creating accumulators.
51 For example the following function
53 int sum (int n)
55 if (n > 0)
56 return n + sum (n - 1);
57 else
58 return 0;
61 is transformed into
63 int sum (int n)
65 int acc = 0;
67 while (n > 0)
68 acc += n--;
70 return acc;
73 To do this, we maintain two accumulators (a_acc and m_acc) that indicate
74 when we reach the return x statement, we should return a_acc + x * m_acc
75 instead. They are initially initialized to 0 and 1, respectively,
76 so the semantics of the function is obviously preserved. If we are
77 guaranteed that the value of the accumulator never change, we
78 omit the accumulator.
80 There are three cases how the function may exit. The first one is
81 handled in adjust_return_value, the other two in adjust_accumulator_values
82 (the second case is actually a special case of the third one and we
83 present it separately just for clarity):
85 1) Just return x, where x is not in any of the remaining special shapes.
86 We rewrite this to a gimple equivalent of return m_acc * x + a_acc.
88 2) return f (...), where f is the current function, is rewritten in a
89 classical tail-recursion elimination way, into assignment of arguments
90 and jump to the start of the function. Values of the accumulators
91 are unchanged.
93 3) return a + m * f(...), where a and m do not depend on call to f.
94 To preserve the semantics described before we want this to be rewritten
95 in such a way that we finally return
97 a_acc + (a + m * f(...)) * m_acc = (a_acc + a * m_acc) + (m * m_acc) * f(...).
99 I.e. we increase a_acc by a * m_acc, multiply m_acc by m and
100 eliminate the tail call to f. Special cases when the value is just
101 added or just multiplied are obtained by setting a = 0 or m = 1.
103 TODO -- it is possible to do similar tricks for other operations. */
105 /* A structure that describes the tailcall. */
107 struct tailcall
109 /* The iterator pointing to the call statement. */
110 gimple_stmt_iterator call_gsi;
112 /* True if it is a call to the current function. */
113 bool tail_recursion;
115 /* The return value of the caller is mult * f + add, where f is the return
116 value of the call. */
117 tree mult, add;
119 /* Next tailcall in the chain. */
120 struct tailcall *next;
123 /* The variables holding the value of multiplicative and additive
124 accumulator. */
125 static tree m_acc, a_acc;
127 static bool optimize_tail_call (struct tailcall *, bool);
128 static void eliminate_tail_call (struct tailcall *);
130 /* Returns false when the function is not suitable for tail call optimization
131 from some reason (e.g. if it takes variable number of arguments). */
133 static bool
134 suitable_for_tail_opt_p (void)
136 if (cfun->stdarg)
137 return false;
139 return true;
141 /* Returns false when the function is not suitable for tail call optimization
142 for some reason (e.g. if it takes variable number of arguments).
143 This test must pass in addition to suitable_for_tail_opt_p in order to make
144 tail call discovery happen. */
146 static bool
147 suitable_for_tail_call_opt_p (void)
149 tree param;
151 /* alloca (until we have stack slot life analysis) inhibits
152 sibling call optimizations, but not tail recursion. */
153 if (cfun->calls_alloca)
154 return false;
156 /* If we are using sjlj exceptions, we may need to add a call to
157 _Unwind_SjLj_Unregister at exit of the function. Which means
158 that we cannot do any sibcall transformations. */
159 if (targetm_common.except_unwind_info (&global_options) == UI_SJLJ
160 && current_function_has_exception_handlers ())
161 return false;
163 /* Any function that calls setjmp might have longjmp called from
164 any called function. ??? We really should represent this
165 properly in the CFG so that this needn't be special cased. */
166 if (cfun->calls_setjmp)
167 return false;
169 /* ??? It is OK if the argument of a function is taken in some cases,
170 but not in all cases. See PR15387 and PR19616. Revisit for 4.1. */
171 for (param = DECL_ARGUMENTS (current_function_decl);
172 param;
173 param = DECL_CHAIN (param))
174 if (TREE_ADDRESSABLE (param))
175 return false;
177 return true;
180 /* Checks whether the expression EXPR in stmt AT is independent of the
181 statement pointed to by GSI (in a sense that we already know EXPR's value
182 at GSI). We use the fact that we are only called from the chain of
183 basic blocks that have only single successor. Returns the expression
184 containing the value of EXPR at GSI. */
186 static tree
187 independent_of_stmt_p (tree expr, gimple *at, gimple_stmt_iterator gsi,
188 bitmap to_move)
190 basic_block bb, call_bb, at_bb;
191 edge e;
192 edge_iterator ei;
194 if (is_gimple_min_invariant (expr))
195 return expr;
197 if (TREE_CODE (expr) != SSA_NAME)
198 return NULL_TREE;
200 if (bitmap_bit_p (to_move, SSA_NAME_VERSION (expr)))
201 return expr;
203 /* Mark the blocks in the chain leading to the end. */
204 at_bb = gimple_bb (at);
205 call_bb = gimple_bb (gsi_stmt (gsi));
206 for (bb = call_bb; bb != at_bb; bb = single_succ (bb))
207 bb->aux = &bb->aux;
208 bb->aux = &bb->aux;
210 while (1)
212 at = SSA_NAME_DEF_STMT (expr);
213 bb = gimple_bb (at);
215 /* The default definition or defined before the chain. */
216 if (!bb || !bb->aux)
217 break;
219 if (bb == call_bb)
221 for (; !gsi_end_p (gsi); gsi_next (&gsi))
222 if (gsi_stmt (gsi) == at)
223 break;
225 if (!gsi_end_p (gsi))
226 expr = NULL_TREE;
227 break;
230 if (gimple_code (at) != GIMPLE_PHI)
232 expr = NULL_TREE;
233 break;
236 FOR_EACH_EDGE (e, ei, bb->preds)
237 if (e->src->aux)
238 break;
239 gcc_assert (e);
241 expr = PHI_ARG_DEF_FROM_EDGE (at, e);
242 if (TREE_CODE (expr) != SSA_NAME)
244 /* The value is a constant. */
245 break;
249 /* Unmark the blocks. */
250 for (bb = call_bb; bb != at_bb; bb = single_succ (bb))
251 bb->aux = NULL;
252 bb->aux = NULL;
254 return expr;
257 enum par { FAIL, OK, TRY_MOVE };
259 /* Simulates the effect of an assignment STMT on the return value of the tail
260 recursive CALL passed in ASS_VAR. M and A are the multiplicative and the
261 additive factor for the real return value. */
263 static par
264 process_assignment (gassign *stmt,
265 gimple_stmt_iterator call, tree *m,
266 tree *a, tree *ass_var, bitmap to_move)
268 tree op0, op1 = NULL_TREE, non_ass_var = NULL_TREE;
269 tree dest = gimple_assign_lhs (stmt);
270 enum tree_code code = gimple_assign_rhs_code (stmt);
271 enum gimple_rhs_class rhs_class = get_gimple_rhs_class (code);
272 tree src_var = gimple_assign_rhs1 (stmt);
274 /* See if this is a simple copy operation of an SSA name to the function
275 result. In that case we may have a simple tail call. Ignore type
276 conversions that can never produce extra code between the function
277 call and the function return. */
278 if ((rhs_class == GIMPLE_SINGLE_RHS || gimple_assign_cast_p (stmt))
279 && src_var == *ass_var)
281 /* Reject a tailcall if the type conversion might need
282 additional code. */
283 if (gimple_assign_cast_p (stmt))
285 if (TYPE_MODE (TREE_TYPE (dest)) != TYPE_MODE (TREE_TYPE (src_var)))
286 return FAIL;
288 /* Even if the type modes are the same, if the precision of the
289 type is smaller than mode's precision,
290 reduce_to_bit_field_precision would generate additional code. */
291 if (INTEGRAL_TYPE_P (TREE_TYPE (dest))
292 && !type_has_mode_precision_p (TREE_TYPE (dest)))
293 return FAIL;
296 *ass_var = dest;
297 return OK;
300 switch (rhs_class)
302 case GIMPLE_BINARY_RHS:
303 op1 = gimple_assign_rhs2 (stmt);
305 /* Fall through. */
307 case GIMPLE_UNARY_RHS:
308 op0 = gimple_assign_rhs1 (stmt);
309 break;
311 default:
312 return FAIL;
315 /* Accumulator optimizations will reverse the order of operations.
316 We can only do that for floating-point types if we're assuming
317 that addition and multiplication are associative. */
318 if (!flag_associative_math)
319 if (FLOAT_TYPE_P (TREE_TYPE (DECL_RESULT (current_function_decl))))
320 return FAIL;
322 if (rhs_class == GIMPLE_UNARY_RHS
323 && op0 == *ass_var)
325 else if (op0 == *ass_var
326 && (non_ass_var = independent_of_stmt_p (op1, stmt, call,
327 to_move)))
329 else if (op1 == *ass_var
330 && (non_ass_var = independent_of_stmt_p (op0, stmt, call,
331 to_move)))
333 else
334 return TRY_MOVE;
336 switch (code)
338 case PLUS_EXPR:
339 *a = non_ass_var;
340 *ass_var = dest;
341 return OK;
343 case POINTER_PLUS_EXPR:
344 if (op0 != *ass_var)
345 return FAIL;
346 *a = non_ass_var;
347 *ass_var = dest;
348 return OK;
350 case MULT_EXPR:
351 *m = non_ass_var;
352 *ass_var = dest;
353 return OK;
355 case NEGATE_EXPR:
356 *m = build_minus_one_cst (TREE_TYPE (op0));
357 *ass_var = dest;
358 return OK;
360 case MINUS_EXPR:
361 if (*ass_var == op0)
362 *a = fold_build1 (NEGATE_EXPR, TREE_TYPE (non_ass_var), non_ass_var);
363 else
365 *m = build_minus_one_cst (TREE_TYPE (non_ass_var));
366 *a = fold_build1 (NEGATE_EXPR, TREE_TYPE (non_ass_var), non_ass_var);
369 *ass_var = dest;
370 return OK;
372 default:
373 return FAIL;
377 /* Propagate VAR through phis on edge E. */
379 static tree
380 propagate_through_phis (tree var, edge e)
382 basic_block dest = e->dest;
383 gphi_iterator gsi;
385 for (gsi = gsi_start_phis (dest); !gsi_end_p (gsi); gsi_next (&gsi))
387 gphi *phi = gsi.phi ();
388 if (PHI_ARG_DEF_FROM_EDGE (phi, e) == var)
389 return PHI_RESULT (phi);
391 return var;
394 /* Finds tailcalls falling into basic block BB. The list of found tailcalls is
395 added to the start of RET. */
397 static void
398 find_tail_calls (basic_block bb, struct tailcall **ret)
400 tree ass_var = NULL_TREE, ret_var, func, param;
401 gimple *stmt;
402 gcall *call = NULL;
403 gimple_stmt_iterator gsi, agsi;
404 bool tail_recursion;
405 struct tailcall *nw;
406 edge e;
407 tree m, a;
408 basic_block abb;
409 size_t idx;
410 tree var;
412 if (!single_succ_p (bb))
413 return;
415 for (gsi = gsi_last_bb (bb); !gsi_end_p (gsi); gsi_prev (&gsi))
417 stmt = gsi_stmt (gsi);
419 /* Ignore labels, returns, nops, clobbers and debug stmts. */
420 if (gimple_code (stmt) == GIMPLE_LABEL
421 || gimple_code (stmt) == GIMPLE_RETURN
422 || gimple_code (stmt) == GIMPLE_NOP
423 || gimple_code (stmt) == GIMPLE_PREDICT
424 || gimple_clobber_p (stmt)
425 || is_gimple_debug (stmt))
426 continue;
428 /* Check for a call. */
429 if (is_gimple_call (stmt))
431 call = as_a <gcall *> (stmt);
432 ass_var = gimple_call_lhs (call);
433 break;
436 /* Allow simple copies between local variables, even if they're
437 aggregates. */
438 if (is_gimple_assign (stmt)
439 && auto_var_in_fn_p (gimple_assign_lhs (stmt), cfun->decl)
440 && auto_var_in_fn_p (gimple_assign_rhs1 (stmt), cfun->decl))
441 continue;
443 /* If the statement references memory or volatile operands, fail. */
444 if (gimple_references_memory_p (stmt)
445 || gimple_has_volatile_ops (stmt))
446 return;
449 if (gsi_end_p (gsi))
451 edge_iterator ei;
452 /* Recurse to the predecessors. */
453 FOR_EACH_EDGE (e, ei, bb->preds)
454 find_tail_calls (e->src, ret);
456 return;
459 /* If the LHS of our call is not just a simple register or local
460 variable, we can't transform this into a tail or sibling call.
461 This situation happens, in (e.g.) "*p = foo()" where foo returns a
462 struct. In this case we won't have a temporary here, but we need
463 to carry out the side effect anyway, so tailcall is impossible.
465 ??? In some situations (when the struct is returned in memory via
466 invisible argument) we could deal with this, e.g. by passing 'p'
467 itself as that argument to foo, but it's too early to do this here,
468 and expand_call() will not handle it anyway. If it ever can, then
469 we need to revisit this here, to allow that situation. */
470 if (ass_var
471 && !is_gimple_reg (ass_var)
472 && !auto_var_in_fn_p (ass_var, cfun->decl))
473 return;
475 /* We found the call, check whether it is suitable. */
476 tail_recursion = false;
477 func = gimple_call_fndecl (call);
478 if (func
479 && !DECL_BUILT_IN (func)
480 && recursive_call_p (current_function_decl, func))
482 tree arg;
484 for (param = DECL_ARGUMENTS (func), idx = 0;
485 param && idx < gimple_call_num_args (call);
486 param = DECL_CHAIN (param), idx ++)
488 arg = gimple_call_arg (call, idx);
489 if (param != arg)
491 /* Make sure there are no problems with copying. The parameter
492 have a copyable type and the two arguments must have reasonably
493 equivalent types. The latter requirement could be relaxed if
494 we emitted a suitable type conversion statement. */
495 if (!is_gimple_reg_type (TREE_TYPE (param))
496 || !useless_type_conversion_p (TREE_TYPE (param),
497 TREE_TYPE (arg)))
498 break;
500 /* The parameter should be a real operand, so that phi node
501 created for it at the start of the function has the meaning
502 of copying the value. This test implies is_gimple_reg_type
503 from the previous condition, however this one could be
504 relaxed by being more careful with copying the new value
505 of the parameter (emitting appropriate GIMPLE_ASSIGN and
506 updating the virtual operands). */
507 if (!is_gimple_reg (param))
508 break;
511 if (idx == gimple_call_num_args (call) && !param)
512 tail_recursion = true;
515 /* Make sure the tail invocation of this function does not indirectly
516 refer to local variables. (Passing variables directly by value
517 is OK.) */
518 FOR_EACH_LOCAL_DECL (cfun, idx, var)
520 if (TREE_CODE (var) != PARM_DECL
521 && auto_var_in_fn_p (var, cfun->decl)
522 && may_be_aliased (var)
523 && (ref_maybe_used_by_stmt_p (call, var)
524 || call_may_clobber_ref_p (call, var)))
525 return;
528 /* Now check the statements after the call. None of them has virtual
529 operands, so they may only depend on the call through its return
530 value. The return value should also be dependent on each of them,
531 since we are running after dce. */
532 m = NULL_TREE;
533 a = NULL_TREE;
534 auto_bitmap to_move_defs;
535 auto_vec<gimple *> to_move_stmts;
537 abb = bb;
538 agsi = gsi;
539 while (1)
541 tree tmp_a = NULL_TREE;
542 tree tmp_m = NULL_TREE;
543 gsi_next (&agsi);
545 while (gsi_end_p (agsi))
547 ass_var = propagate_through_phis (ass_var, single_succ_edge (abb));
548 abb = single_succ (abb);
549 agsi = gsi_start_bb (abb);
552 stmt = gsi_stmt (agsi);
553 if (gimple_code (stmt) == GIMPLE_RETURN)
554 break;
556 if (gimple_code (stmt) == GIMPLE_LABEL
557 || gimple_code (stmt) == GIMPLE_NOP
558 || gimple_code (stmt) == GIMPLE_PREDICT
559 || gimple_clobber_p (stmt)
560 || is_gimple_debug (stmt))
561 continue;
563 if (gimple_code (stmt) != GIMPLE_ASSIGN)
564 return;
566 /* This is a gimple assign. */
567 par ret = process_assignment (as_a <gassign *> (stmt), gsi,
568 &tmp_m, &tmp_a, &ass_var, to_move_defs);
569 if (ret == FAIL)
570 return;
571 else if (ret == TRY_MOVE)
573 if (! tail_recursion)
574 return;
575 /* Do not deal with checking dominance, the real fix is to
576 do path isolation for the transform phase anyway, removing
577 the need to compute the accumulators with new stmts. */
578 if (abb != bb)
579 return;
580 for (unsigned opno = 1; opno < gimple_num_ops (stmt); ++opno)
582 tree op = gimple_op (stmt, opno);
583 if (independent_of_stmt_p (op, stmt, gsi, to_move_defs) != op)
584 return;
586 bitmap_set_bit (to_move_defs,
587 SSA_NAME_VERSION (gimple_assign_lhs (stmt)));
588 to_move_stmts.safe_push (stmt);
589 continue;
592 if (tmp_a)
594 tree type = TREE_TYPE (tmp_a);
595 if (a)
596 a = fold_build2 (PLUS_EXPR, type, fold_convert (type, a), tmp_a);
597 else
598 a = tmp_a;
600 if (tmp_m)
602 tree type = TREE_TYPE (tmp_m);
603 if (m)
604 m = fold_build2 (MULT_EXPR, type, fold_convert (type, m), tmp_m);
605 else
606 m = tmp_m;
608 if (a)
609 a = fold_build2 (MULT_EXPR, type, fold_convert (type, a), tmp_m);
613 /* See if this is a tail call we can handle. */
614 ret_var = gimple_return_retval (as_a <greturn *> (stmt));
616 /* We may proceed if there either is no return value, or the return value
617 is identical to the call's return. */
618 if (ret_var
619 && (ret_var != ass_var))
620 return;
622 /* If this is not a tail recursive call, we cannot handle addends or
623 multiplicands. */
624 if (!tail_recursion && (m || a))
625 return;
627 /* For pointers only allow additions. */
628 if (m && POINTER_TYPE_P (TREE_TYPE (DECL_RESULT (current_function_decl))))
629 return;
631 /* Move queued defs. */
632 if (tail_recursion)
634 unsigned i;
635 FOR_EACH_VEC_ELT (to_move_stmts, i, stmt)
637 gimple_stmt_iterator mgsi = gsi_for_stmt (stmt);
638 gsi_move_before (&mgsi, &gsi);
642 nw = XNEW (struct tailcall);
644 nw->call_gsi = gsi;
646 nw->tail_recursion = tail_recursion;
648 nw->mult = m;
649 nw->add = a;
651 nw->next = *ret;
652 *ret = nw;
655 /* Helper to insert PHI_ARGH to the phi of VAR in the destination of edge E. */
657 static void
658 add_successor_phi_arg (edge e, tree var, tree phi_arg)
660 gphi_iterator gsi;
662 for (gsi = gsi_start_phis (e->dest); !gsi_end_p (gsi); gsi_next (&gsi))
663 if (PHI_RESULT (gsi.phi ()) == var)
664 break;
666 gcc_assert (!gsi_end_p (gsi));
667 add_phi_arg (gsi.phi (), phi_arg, e, UNKNOWN_LOCATION);
670 /* Creates a GIMPLE statement which computes the operation specified by
671 CODE, ACC and OP1 to a new variable with name LABEL and inserts the
672 statement in the position specified by GSI. Returns the
673 tree node of the statement's result. */
675 static tree
676 adjust_return_value_with_ops (enum tree_code code, const char *label,
677 tree acc, tree op1, gimple_stmt_iterator gsi)
680 tree ret_type = TREE_TYPE (DECL_RESULT (current_function_decl));
681 tree result = make_temp_ssa_name (ret_type, NULL, label);
682 gassign *stmt;
684 if (POINTER_TYPE_P (ret_type))
686 gcc_assert (code == PLUS_EXPR && TREE_TYPE (acc) == sizetype);
687 code = POINTER_PLUS_EXPR;
689 if (types_compatible_p (TREE_TYPE (acc), TREE_TYPE (op1))
690 && code != POINTER_PLUS_EXPR)
691 stmt = gimple_build_assign (result, code, acc, op1);
692 else
694 tree tem;
695 if (code == POINTER_PLUS_EXPR)
696 tem = fold_build2 (code, TREE_TYPE (op1), op1, acc);
697 else
698 tem = fold_build2 (code, TREE_TYPE (op1),
699 fold_convert (TREE_TYPE (op1), acc), op1);
700 tree rhs = fold_convert (ret_type, tem);
701 rhs = force_gimple_operand_gsi (&gsi, rhs,
702 false, NULL, true, GSI_SAME_STMT);
703 stmt = gimple_build_assign (result, rhs);
706 gsi_insert_before (&gsi, stmt, GSI_NEW_STMT);
707 return result;
710 /* Creates a new GIMPLE statement that adjusts the value of accumulator ACC by
711 the computation specified by CODE and OP1 and insert the statement
712 at the position specified by GSI as a new statement. Returns new SSA name
713 of updated accumulator. */
715 static tree
716 update_accumulator_with_ops (enum tree_code code, tree acc, tree op1,
717 gimple_stmt_iterator gsi)
719 gassign *stmt;
720 tree var = copy_ssa_name (acc);
721 if (types_compatible_p (TREE_TYPE (acc), TREE_TYPE (op1)))
722 stmt = gimple_build_assign (var, code, acc, op1);
723 else
725 tree rhs = fold_convert (TREE_TYPE (acc),
726 fold_build2 (code,
727 TREE_TYPE (op1),
728 fold_convert (TREE_TYPE (op1), acc),
729 op1));
730 rhs = force_gimple_operand_gsi (&gsi, rhs,
731 false, NULL, false, GSI_CONTINUE_LINKING);
732 stmt = gimple_build_assign (var, rhs);
734 gsi_insert_after (&gsi, stmt, GSI_NEW_STMT);
735 return var;
738 /* Adjust the accumulator values according to A and M after GSI, and update
739 the phi nodes on edge BACK. */
741 static void
742 adjust_accumulator_values (gimple_stmt_iterator gsi, tree m, tree a, edge back)
744 tree var, a_acc_arg, m_acc_arg;
746 if (m)
747 m = force_gimple_operand_gsi (&gsi, m, true, NULL, true, GSI_SAME_STMT);
748 if (a)
749 a = force_gimple_operand_gsi (&gsi, a, true, NULL, true, GSI_SAME_STMT);
751 a_acc_arg = a_acc;
752 m_acc_arg = m_acc;
753 if (a)
755 if (m_acc)
757 if (integer_onep (a))
758 var = m_acc;
759 else
760 var = adjust_return_value_with_ops (MULT_EXPR, "acc_tmp", m_acc,
761 a, gsi);
763 else
764 var = a;
766 a_acc_arg = update_accumulator_with_ops (PLUS_EXPR, a_acc, var, gsi);
769 if (m)
770 m_acc_arg = update_accumulator_with_ops (MULT_EXPR, m_acc, m, gsi);
772 if (a_acc)
773 add_successor_phi_arg (back, a_acc, a_acc_arg);
775 if (m_acc)
776 add_successor_phi_arg (back, m_acc, m_acc_arg);
779 /* Adjust value of the return at the end of BB according to M and A
780 accumulators. */
782 static void
783 adjust_return_value (basic_block bb, tree m, tree a)
785 tree retval;
786 greturn *ret_stmt = as_a <greturn *> (gimple_seq_last_stmt (bb_seq (bb)));
787 gimple_stmt_iterator gsi = gsi_last_bb (bb);
789 gcc_assert (gimple_code (ret_stmt) == GIMPLE_RETURN);
791 retval = gimple_return_retval (ret_stmt);
792 if (!retval || retval == error_mark_node)
793 return;
795 if (m)
796 retval = adjust_return_value_with_ops (MULT_EXPR, "mul_tmp", m_acc, retval,
797 gsi);
798 if (a)
799 retval = adjust_return_value_with_ops (PLUS_EXPR, "acc_tmp", a_acc, retval,
800 gsi);
801 gimple_return_set_retval (ret_stmt, retval);
802 update_stmt (ret_stmt);
805 /* Subtract COUNT and FREQUENCY from the basic block and it's
806 outgoing edge. */
807 static void
808 decrease_profile (basic_block bb, profile_count count)
810 bb->count = bb->count - count;
811 if (!single_succ_p (bb))
813 gcc_assert (!EDGE_COUNT (bb->succs));
814 return;
818 /* Returns true if argument PARAM of the tail recursive call needs to be copied
819 when the call is eliminated. */
821 static bool
822 arg_needs_copy_p (tree param)
824 tree def;
826 if (!is_gimple_reg (param))
827 return false;
829 /* Parameters that are only defined but never used need not be copied. */
830 def = ssa_default_def (cfun, param);
831 if (!def)
832 return false;
834 return true;
837 /* Eliminates tail call described by T. TMP_VARS is a list of
838 temporary variables used to copy the function arguments. */
840 static void
841 eliminate_tail_call (struct tailcall *t)
843 tree param, rslt;
844 gimple *stmt, *call;
845 tree arg;
846 size_t idx;
847 basic_block bb, first;
848 edge e;
849 gphi *phi;
850 gphi_iterator gpi;
851 gimple_stmt_iterator gsi;
852 gimple *orig_stmt;
854 stmt = orig_stmt = gsi_stmt (t->call_gsi);
855 bb = gsi_bb (t->call_gsi);
857 if (dump_file && (dump_flags & TDF_DETAILS))
859 fprintf (dump_file, "Eliminated tail recursion in bb %d : ",
860 bb->index);
861 print_gimple_stmt (dump_file, stmt, 0, TDF_SLIM);
862 fprintf (dump_file, "\n");
865 gcc_assert (is_gimple_call (stmt));
867 first = single_succ (ENTRY_BLOCK_PTR_FOR_FN (cfun));
869 /* Remove the code after call_gsi that will become unreachable. The
870 possibly unreachable code in other blocks is removed later in
871 cfg cleanup. */
872 gsi = t->call_gsi;
873 gimple_stmt_iterator gsi2 = gsi_last_bb (gimple_bb (gsi_stmt (gsi)));
874 while (gsi_stmt (gsi2) != gsi_stmt (gsi))
876 gimple *t = gsi_stmt (gsi2);
877 /* Do not remove the return statement, so that redirect_edge_and_branch
878 sees how the block ends. */
879 if (gimple_code (t) != GIMPLE_RETURN)
881 gimple_stmt_iterator gsi3 = gsi2;
882 gsi_prev (&gsi2);
883 gsi_remove (&gsi3, true);
884 release_defs (t);
886 else
887 gsi_prev (&gsi2);
890 /* Number of executions of function has reduced by the tailcall. */
891 e = single_succ_edge (gsi_bb (t->call_gsi));
892 decrease_profile (EXIT_BLOCK_PTR_FOR_FN (cfun), e->count ());
893 decrease_profile (ENTRY_BLOCK_PTR_FOR_FN (cfun), e->count ());
894 if (e->dest != EXIT_BLOCK_PTR_FOR_FN (cfun))
895 decrease_profile (e->dest, e->count ());
897 /* Replace the call by a jump to the start of function. */
898 e = redirect_edge_and_branch (single_succ_edge (gsi_bb (t->call_gsi)),
899 first);
900 gcc_assert (e);
901 PENDING_STMT (e) = NULL;
903 /* Add phi node entries for arguments. The ordering of the phi nodes should
904 be the same as the ordering of the arguments. */
905 for (param = DECL_ARGUMENTS (current_function_decl),
906 idx = 0, gpi = gsi_start_phis (first);
907 param;
908 param = DECL_CHAIN (param), idx++)
910 if (!arg_needs_copy_p (param))
911 continue;
913 arg = gimple_call_arg (stmt, idx);
914 phi = gpi.phi ();
915 gcc_assert (param == SSA_NAME_VAR (PHI_RESULT (phi)));
917 add_phi_arg (phi, arg, e, gimple_location (stmt));
918 gsi_next (&gpi);
921 /* Update the values of accumulators. */
922 adjust_accumulator_values (t->call_gsi, t->mult, t->add, e);
924 call = gsi_stmt (t->call_gsi);
925 rslt = gimple_call_lhs (call);
926 if (rslt != NULL_TREE && TREE_CODE (rslt) == SSA_NAME)
928 /* Result of the call will no longer be defined. So adjust the
929 SSA_NAME_DEF_STMT accordingly. */
930 SSA_NAME_DEF_STMT (rslt) = gimple_build_nop ();
933 gsi_remove (&t->call_gsi, true);
934 release_defs (call);
937 /* Optimizes the tailcall described by T. If OPT_TAILCALLS is true, also
938 mark the tailcalls for the sibcall optimization. */
940 static bool
941 optimize_tail_call (struct tailcall *t, bool opt_tailcalls)
943 if (t->tail_recursion)
945 eliminate_tail_call (t);
946 return true;
949 if (opt_tailcalls)
951 gcall *stmt = as_a <gcall *> (gsi_stmt (t->call_gsi));
953 gimple_call_set_tail (stmt, true);
954 cfun->tail_call_marked = true;
955 if (dump_file && (dump_flags & TDF_DETAILS))
957 fprintf (dump_file, "Found tail call ");
958 print_gimple_stmt (dump_file, stmt, 0, dump_flags);
959 fprintf (dump_file, " in bb %i\n", (gsi_bb (t->call_gsi))->index);
963 return false;
966 /* Creates a tail-call accumulator of the same type as the return type of the
967 current function. LABEL is the name used to creating the temporary
968 variable for the accumulator. The accumulator will be inserted in the
969 phis of a basic block BB with single predecessor with an initial value
970 INIT converted to the current function return type. */
972 static tree
973 create_tailcall_accumulator (const char *label, basic_block bb, tree init)
975 tree ret_type = TREE_TYPE (DECL_RESULT (current_function_decl));
976 if (POINTER_TYPE_P (ret_type))
977 ret_type = sizetype;
979 tree tmp = make_temp_ssa_name (ret_type, NULL, label);
980 gphi *phi;
982 phi = create_phi_node (tmp, bb);
983 /* RET_TYPE can be a float when -ffast-maths is enabled. */
984 add_phi_arg (phi, fold_convert (ret_type, init), single_pred_edge (bb),
985 UNKNOWN_LOCATION);
986 return PHI_RESULT (phi);
989 /* Optimizes tail calls in the function, turning the tail recursion
990 into iteration. */
992 static unsigned int
993 tree_optimize_tail_calls_1 (bool opt_tailcalls)
995 edge e;
996 bool phis_constructed = false;
997 struct tailcall *tailcalls = NULL, *act, *next;
998 bool changed = false;
999 basic_block first = single_succ (ENTRY_BLOCK_PTR_FOR_FN (cfun));
1000 tree param;
1001 gimple *stmt;
1002 edge_iterator ei;
1004 if (!suitable_for_tail_opt_p ())
1005 return 0;
1006 if (opt_tailcalls)
1007 opt_tailcalls = suitable_for_tail_call_opt_p ();
1009 FOR_EACH_EDGE (e, ei, EXIT_BLOCK_PTR_FOR_FN (cfun)->preds)
1011 /* Only traverse the normal exits, i.e. those that end with return
1012 statement. */
1013 stmt = last_stmt (e->src);
1015 if (stmt
1016 && gimple_code (stmt) == GIMPLE_RETURN)
1017 find_tail_calls (e->src, &tailcalls);
1020 /* Construct the phi nodes and accumulators if necessary. */
1021 a_acc = m_acc = NULL_TREE;
1022 for (act = tailcalls; act; act = act->next)
1024 if (!act->tail_recursion)
1025 continue;
1027 if (!phis_constructed)
1029 /* Ensure that there is only one predecessor of the block
1030 or if there are existing degenerate PHI nodes. */
1031 if (!single_pred_p (first)
1032 || !gimple_seq_empty_p (phi_nodes (first)))
1033 first =
1034 split_edge (single_succ_edge (ENTRY_BLOCK_PTR_FOR_FN (cfun)));
1036 /* Copy the args if needed. */
1037 for (param = DECL_ARGUMENTS (current_function_decl);
1038 param;
1039 param = DECL_CHAIN (param))
1040 if (arg_needs_copy_p (param))
1042 tree name = ssa_default_def (cfun, param);
1043 tree new_name = make_ssa_name (param, SSA_NAME_DEF_STMT (name));
1044 gphi *phi;
1046 set_ssa_default_def (cfun, param, new_name);
1047 phi = create_phi_node (name, first);
1048 add_phi_arg (phi, new_name, single_pred_edge (first),
1049 EXPR_LOCATION (param));
1051 phis_constructed = true;
1054 if (act->add && !a_acc)
1055 a_acc = create_tailcall_accumulator ("add_acc", first,
1056 integer_zero_node);
1058 if (act->mult && !m_acc)
1059 m_acc = create_tailcall_accumulator ("mult_acc", first,
1060 integer_one_node);
1063 if (a_acc || m_acc)
1065 /* When the tail call elimination using accumulators is performed,
1066 statements adding the accumulated value are inserted at all exits.
1067 This turns all other tail calls to non-tail ones. */
1068 opt_tailcalls = false;
1071 for (; tailcalls; tailcalls = next)
1073 next = tailcalls->next;
1074 changed |= optimize_tail_call (tailcalls, opt_tailcalls);
1075 free (tailcalls);
1078 if (a_acc || m_acc)
1080 /* Modify the remaining return statements. */
1081 FOR_EACH_EDGE (e, ei, EXIT_BLOCK_PTR_FOR_FN (cfun)->preds)
1083 stmt = last_stmt (e->src);
1085 if (stmt
1086 && gimple_code (stmt) == GIMPLE_RETURN)
1087 adjust_return_value (e->src, m_acc, a_acc);
1091 if (changed)
1093 /* We may have created new loops. Make them magically appear. */
1094 loops_state_set (LOOPS_NEED_FIXUP);
1095 free_dominance_info (CDI_DOMINATORS);
1098 /* Add phi nodes for the virtual operands defined in the function to the
1099 header of the loop created by tail recursion elimination. Do so
1100 by triggering the SSA renamer. */
1101 if (phis_constructed)
1102 mark_virtual_operands_for_renaming (cfun);
1104 if (changed)
1105 return TODO_cleanup_cfg | TODO_update_ssa_only_virtuals;
1106 return 0;
1109 static bool
1110 gate_tail_calls (void)
1112 return flag_optimize_sibling_calls != 0 && dbg_cnt (tail_call);
1115 static unsigned int
1116 execute_tail_calls (void)
1118 return tree_optimize_tail_calls_1 (true);
1121 namespace {
1123 const pass_data pass_data_tail_recursion =
1125 GIMPLE_PASS, /* type */
1126 "tailr", /* name */
1127 OPTGROUP_NONE, /* optinfo_flags */
1128 TV_NONE, /* tv_id */
1129 ( PROP_cfg | PROP_ssa ), /* properties_required */
1130 0, /* properties_provided */
1131 0, /* properties_destroyed */
1132 0, /* todo_flags_start */
1133 0, /* todo_flags_finish */
1136 class pass_tail_recursion : public gimple_opt_pass
1138 public:
1139 pass_tail_recursion (gcc::context *ctxt)
1140 : gimple_opt_pass (pass_data_tail_recursion, ctxt)
1143 /* opt_pass methods: */
1144 opt_pass * clone () { return new pass_tail_recursion (m_ctxt); }
1145 virtual bool gate (function *) { return gate_tail_calls (); }
1146 virtual unsigned int execute (function *)
1148 return tree_optimize_tail_calls_1 (false);
1151 }; // class pass_tail_recursion
1153 } // anon namespace
1155 gimple_opt_pass *
1156 make_pass_tail_recursion (gcc::context *ctxt)
1158 return new pass_tail_recursion (ctxt);
1161 namespace {
1163 const pass_data pass_data_tail_calls =
1165 GIMPLE_PASS, /* type */
1166 "tailc", /* name */
1167 OPTGROUP_NONE, /* optinfo_flags */
1168 TV_NONE, /* tv_id */
1169 ( PROP_cfg | PROP_ssa ), /* properties_required */
1170 0, /* properties_provided */
1171 0, /* properties_destroyed */
1172 0, /* todo_flags_start */
1173 0, /* todo_flags_finish */
1176 class pass_tail_calls : public gimple_opt_pass
1178 public:
1179 pass_tail_calls (gcc::context *ctxt)
1180 : gimple_opt_pass (pass_data_tail_calls, ctxt)
1183 /* opt_pass methods: */
1184 virtual bool gate (function *) { return gate_tail_calls (); }
1185 virtual unsigned int execute (function *) { return execute_tail_calls (); }
1187 }; // class pass_tail_calls
1189 } // anon namespace
1191 gimple_opt_pass *
1192 make_pass_tail_calls (gcc::context *ctxt)
1194 return new pass_tail_calls (ctxt);