* config/i386/i386.c (ix86_adjust_stack_and_probe_stack_clash): New.
[official-gcc.git] / gcc / tree-tailcall.c
blobc4b8cee4d27738a663d08a09ad0af28b5218ed45
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, int frequency)
810 edge e;
811 bb->count = bb->count - count;
812 bb->frequency -= frequency;
813 if (bb->frequency < 0)
814 bb->frequency = 0;
815 if (!single_succ_p (bb))
817 gcc_assert (!EDGE_COUNT (bb->succs));
818 return;
820 e = single_succ_edge (bb);
821 e->count -= count;
824 /* Returns true if argument PARAM of the tail recursive call needs to be copied
825 when the call is eliminated. */
827 static bool
828 arg_needs_copy_p (tree param)
830 tree def;
832 if (!is_gimple_reg (param))
833 return false;
835 /* Parameters that are only defined but never used need not be copied. */
836 def = ssa_default_def (cfun, param);
837 if (!def)
838 return false;
840 return true;
843 /* Eliminates tail call described by T. TMP_VARS is a list of
844 temporary variables used to copy the function arguments. */
846 static void
847 eliminate_tail_call (struct tailcall *t)
849 tree param, rslt;
850 gimple *stmt, *call;
851 tree arg;
852 size_t idx;
853 basic_block bb, first;
854 edge e;
855 gphi *phi;
856 gphi_iterator gpi;
857 gimple_stmt_iterator gsi;
858 gimple *orig_stmt;
860 stmt = orig_stmt = gsi_stmt (t->call_gsi);
861 bb = gsi_bb (t->call_gsi);
863 if (dump_file && (dump_flags & TDF_DETAILS))
865 fprintf (dump_file, "Eliminated tail recursion in bb %d : ",
866 bb->index);
867 print_gimple_stmt (dump_file, stmt, 0, TDF_SLIM);
868 fprintf (dump_file, "\n");
871 gcc_assert (is_gimple_call (stmt));
873 first = single_succ (ENTRY_BLOCK_PTR_FOR_FN (cfun));
875 /* Remove the code after call_gsi that will become unreachable. The
876 possibly unreachable code in other blocks is removed later in
877 cfg cleanup. */
878 gsi = t->call_gsi;
879 gimple_stmt_iterator gsi2 = gsi_last_bb (gimple_bb (gsi_stmt (gsi)));
880 while (gsi_stmt (gsi2) != gsi_stmt (gsi))
882 gimple *t = gsi_stmt (gsi2);
883 /* Do not remove the return statement, so that redirect_edge_and_branch
884 sees how the block ends. */
885 if (gimple_code (t) != GIMPLE_RETURN)
887 gimple_stmt_iterator gsi3 = gsi2;
888 gsi_prev (&gsi2);
889 gsi_remove (&gsi3, true);
890 release_defs (t);
892 else
893 gsi_prev (&gsi2);
896 /* Number of executions of function has reduced by the tailcall. */
897 e = single_succ_edge (gsi_bb (t->call_gsi));
898 decrease_profile (EXIT_BLOCK_PTR_FOR_FN (cfun), e->count, EDGE_FREQUENCY (e));
899 decrease_profile (ENTRY_BLOCK_PTR_FOR_FN (cfun), e->count,
900 EDGE_FREQUENCY (e));
901 if (e->dest != EXIT_BLOCK_PTR_FOR_FN (cfun))
902 decrease_profile (e->dest, e->count, EDGE_FREQUENCY (e));
904 /* Replace the call by a jump to the start of function. */
905 e = redirect_edge_and_branch (single_succ_edge (gsi_bb (t->call_gsi)),
906 first);
907 gcc_assert (e);
908 PENDING_STMT (e) = NULL;
910 /* Add phi node entries for arguments. The ordering of the phi nodes should
911 be the same as the ordering of the arguments. */
912 for (param = DECL_ARGUMENTS (current_function_decl),
913 idx = 0, gpi = gsi_start_phis (first);
914 param;
915 param = DECL_CHAIN (param), idx++)
917 if (!arg_needs_copy_p (param))
918 continue;
920 arg = gimple_call_arg (stmt, idx);
921 phi = gpi.phi ();
922 gcc_assert (param == SSA_NAME_VAR (PHI_RESULT (phi)));
924 add_phi_arg (phi, arg, e, gimple_location (stmt));
925 gsi_next (&gpi);
928 /* Update the values of accumulators. */
929 adjust_accumulator_values (t->call_gsi, t->mult, t->add, e);
931 call = gsi_stmt (t->call_gsi);
932 rslt = gimple_call_lhs (call);
933 if (rslt != NULL_TREE && TREE_CODE (rslt) == SSA_NAME)
935 /* Result of the call will no longer be defined. So adjust the
936 SSA_NAME_DEF_STMT accordingly. */
937 SSA_NAME_DEF_STMT (rslt) = gimple_build_nop ();
940 gsi_remove (&t->call_gsi, true);
941 release_defs (call);
944 /* Optimizes the tailcall described by T. If OPT_TAILCALLS is true, also
945 mark the tailcalls for the sibcall optimization. */
947 static bool
948 optimize_tail_call (struct tailcall *t, bool opt_tailcalls)
950 if (t->tail_recursion)
952 eliminate_tail_call (t);
953 return true;
956 if (opt_tailcalls)
958 gcall *stmt = as_a <gcall *> (gsi_stmt (t->call_gsi));
960 gimple_call_set_tail (stmt, true);
961 cfun->tail_call_marked = true;
962 if (dump_file && (dump_flags & TDF_DETAILS))
964 fprintf (dump_file, "Found tail call ");
965 print_gimple_stmt (dump_file, stmt, 0, dump_flags);
966 fprintf (dump_file, " in bb %i\n", (gsi_bb (t->call_gsi))->index);
970 return false;
973 /* Creates a tail-call accumulator of the same type as the return type of the
974 current function. LABEL is the name used to creating the temporary
975 variable for the accumulator. The accumulator will be inserted in the
976 phis of a basic block BB with single predecessor with an initial value
977 INIT converted to the current function return type. */
979 static tree
980 create_tailcall_accumulator (const char *label, basic_block bb, tree init)
982 tree ret_type = TREE_TYPE (DECL_RESULT (current_function_decl));
983 if (POINTER_TYPE_P (ret_type))
984 ret_type = sizetype;
986 tree tmp = make_temp_ssa_name (ret_type, NULL, label);
987 gphi *phi;
989 phi = create_phi_node (tmp, bb);
990 /* RET_TYPE can be a float when -ffast-maths is enabled. */
991 add_phi_arg (phi, fold_convert (ret_type, init), single_pred_edge (bb),
992 UNKNOWN_LOCATION);
993 return PHI_RESULT (phi);
996 /* Optimizes tail calls in the function, turning the tail recursion
997 into iteration. */
999 static unsigned int
1000 tree_optimize_tail_calls_1 (bool opt_tailcalls)
1002 edge e;
1003 bool phis_constructed = false;
1004 struct tailcall *tailcalls = NULL, *act, *next;
1005 bool changed = false;
1006 basic_block first = single_succ (ENTRY_BLOCK_PTR_FOR_FN (cfun));
1007 tree param;
1008 gimple *stmt;
1009 edge_iterator ei;
1011 if (!suitable_for_tail_opt_p ())
1012 return 0;
1013 if (opt_tailcalls)
1014 opt_tailcalls = suitable_for_tail_call_opt_p ();
1016 FOR_EACH_EDGE (e, ei, EXIT_BLOCK_PTR_FOR_FN (cfun)->preds)
1018 /* Only traverse the normal exits, i.e. those that end with return
1019 statement. */
1020 stmt = last_stmt (e->src);
1022 if (stmt
1023 && gimple_code (stmt) == GIMPLE_RETURN)
1024 find_tail_calls (e->src, &tailcalls);
1027 /* Construct the phi nodes and accumulators if necessary. */
1028 a_acc = m_acc = NULL_TREE;
1029 for (act = tailcalls; act; act = act->next)
1031 if (!act->tail_recursion)
1032 continue;
1034 if (!phis_constructed)
1036 /* Ensure that there is only one predecessor of the block
1037 or if there are existing degenerate PHI nodes. */
1038 if (!single_pred_p (first)
1039 || !gimple_seq_empty_p (phi_nodes (first)))
1040 first =
1041 split_edge (single_succ_edge (ENTRY_BLOCK_PTR_FOR_FN (cfun)));
1043 /* Copy the args if needed. */
1044 for (param = DECL_ARGUMENTS (current_function_decl);
1045 param;
1046 param = DECL_CHAIN (param))
1047 if (arg_needs_copy_p (param))
1049 tree name = ssa_default_def (cfun, param);
1050 tree new_name = make_ssa_name (param, SSA_NAME_DEF_STMT (name));
1051 gphi *phi;
1053 set_ssa_default_def (cfun, param, new_name);
1054 phi = create_phi_node (name, first);
1055 add_phi_arg (phi, new_name, single_pred_edge (first),
1056 EXPR_LOCATION (param));
1058 phis_constructed = true;
1061 if (act->add && !a_acc)
1062 a_acc = create_tailcall_accumulator ("add_acc", first,
1063 integer_zero_node);
1065 if (act->mult && !m_acc)
1066 m_acc = create_tailcall_accumulator ("mult_acc", first,
1067 integer_one_node);
1070 if (a_acc || m_acc)
1072 /* When the tail call elimination using accumulators is performed,
1073 statements adding the accumulated value are inserted at all exits.
1074 This turns all other tail calls to non-tail ones. */
1075 opt_tailcalls = false;
1078 for (; tailcalls; tailcalls = next)
1080 next = tailcalls->next;
1081 changed |= optimize_tail_call (tailcalls, opt_tailcalls);
1082 free (tailcalls);
1085 if (a_acc || m_acc)
1087 /* Modify the remaining return statements. */
1088 FOR_EACH_EDGE (e, ei, EXIT_BLOCK_PTR_FOR_FN (cfun)->preds)
1090 stmt = last_stmt (e->src);
1092 if (stmt
1093 && gimple_code (stmt) == GIMPLE_RETURN)
1094 adjust_return_value (e->src, m_acc, a_acc);
1098 if (changed)
1100 /* We may have created new loops. Make them magically appear. */
1101 loops_state_set (LOOPS_NEED_FIXUP);
1102 free_dominance_info (CDI_DOMINATORS);
1105 /* Add phi nodes for the virtual operands defined in the function to the
1106 header of the loop created by tail recursion elimination. Do so
1107 by triggering the SSA renamer. */
1108 if (phis_constructed)
1109 mark_virtual_operands_for_renaming (cfun);
1111 if (changed)
1112 return TODO_cleanup_cfg | TODO_update_ssa_only_virtuals;
1113 return 0;
1116 static bool
1117 gate_tail_calls (void)
1119 return flag_optimize_sibling_calls != 0 && dbg_cnt (tail_call);
1122 static unsigned int
1123 execute_tail_calls (void)
1125 return tree_optimize_tail_calls_1 (true);
1128 namespace {
1130 const pass_data pass_data_tail_recursion =
1132 GIMPLE_PASS, /* type */
1133 "tailr", /* name */
1134 OPTGROUP_NONE, /* optinfo_flags */
1135 TV_NONE, /* tv_id */
1136 ( PROP_cfg | PROP_ssa ), /* properties_required */
1137 0, /* properties_provided */
1138 0, /* properties_destroyed */
1139 0, /* todo_flags_start */
1140 0, /* todo_flags_finish */
1143 class pass_tail_recursion : public gimple_opt_pass
1145 public:
1146 pass_tail_recursion (gcc::context *ctxt)
1147 : gimple_opt_pass (pass_data_tail_recursion, ctxt)
1150 /* opt_pass methods: */
1151 opt_pass * clone () { return new pass_tail_recursion (m_ctxt); }
1152 virtual bool gate (function *) { return gate_tail_calls (); }
1153 virtual unsigned int execute (function *)
1155 return tree_optimize_tail_calls_1 (false);
1158 }; // class pass_tail_recursion
1160 } // anon namespace
1162 gimple_opt_pass *
1163 make_pass_tail_recursion (gcc::context *ctxt)
1165 return new pass_tail_recursion (ctxt);
1168 namespace {
1170 const pass_data pass_data_tail_calls =
1172 GIMPLE_PASS, /* type */
1173 "tailc", /* name */
1174 OPTGROUP_NONE, /* optinfo_flags */
1175 TV_NONE, /* tv_id */
1176 ( PROP_cfg | PROP_ssa ), /* properties_required */
1177 0, /* properties_provided */
1178 0, /* properties_destroyed */
1179 0, /* todo_flags_start */
1180 0, /* todo_flags_finish */
1183 class pass_tail_calls : public gimple_opt_pass
1185 public:
1186 pass_tail_calls (gcc::context *ctxt)
1187 : gimple_opt_pass (pass_data_tail_calls, ctxt)
1190 /* opt_pass methods: */
1191 virtual bool gate (function *) { return gate_tail_calls (); }
1192 virtual unsigned int execute (function *) { return execute_tail_calls (); }
1194 }; // class pass_tail_calls
1196 } // anon namespace
1198 gimple_opt_pass *
1199 make_pass_tail_calls (gcc::context *ctxt)
1201 return new pass_tail_calls (ctxt);