- Fixed bugs in schedule verifier
[libfirm] / ir / be / bespill.c
1 /*
2  * Author:      Daniel Grund, Sebastian Hack, Matthias Braun
3  * Date:                29.09.2005
4  * Copyright:   (c) Universitaet Karlsruhe
5  * Licence:     This file protected by GPL -  GNU GENERAL PUBLIC LICENSE.
6  */
7 #ifdef HAVE_CONFIG_H
8 #include "config.h"
9 #endif
10
11 #include <stdlib.h>
12
13 #include "pset.h"
14 #include "irnode_t.h"
15 #include "ircons_t.h"
16 #include "iredges_t.h"
17 #include "ident_t.h"
18 #include "type_t.h"
19 #include "entity_t.h"
20 #include "debug.h"
21 #include "irgwalk.h"
22 #include "array.h"
23 #include "pdeq.h"
24 #include "unionfind.h"
25 #include "execfreq.h"
26
27 #include "belive_t.h"
28 #include "besched_t.h"
29 #include "bespill.h"
30 #include "belive_t.h"
31 #include "benode_t.h"
32 #include "bechordal_t.h"
33 #include "bejavacoal.h"
34
35 // only rematerialise when costs are less than REMAT_COST_LIMIT
36 #define REMAT_COST_LIMIT        4
37
38 /* This enables re-computation of values. Current state: Unfinished and buggy. */
39 #undef BUGGY_REMAT
40
41 typedef struct _reloader_t reloader_t;
42
43 struct _reloader_t {
44         reloader_t *next;
45         ir_node *reloader;
46 };
47
48 typedef struct _spill_info_t {
49         ir_node *spilled_node;
50         reloader_t *reloaders;
51
52         ir_node *spill;
53 } spill_info_t;
54
55 struct _spill_env_t {
56         const arch_register_class_t *cls;
57         const be_chordal_env_t *chordal_env;
58         struct obstack obst;
59         set *spills;                            /**< all spill_info_t's, which must be placed */
60         pset *mem_phis;                         /**< set of all special spilled phis. allocated and freed separately */
61
62         DEBUG_ONLY(firm_dbg_module_t *dbg;)
63 };
64
65 /**
66  * Compare two spill infos.
67  */
68 static int cmp_spillinfo(const void *x, const void *y, size_t size) {
69         const spill_info_t *xx = x;
70         const spill_info_t *yy = y;
71         return xx->spilled_node != yy->spilled_node;
72 }
73
74 /**
75  * Returns spill info for a specific value (the value that is to be spilled)
76  */
77 static spill_info_t *get_spillinfo(const spill_env_t *env, ir_node *value) {
78         spill_info_t info, *res;
79         int hash = HASH_PTR(value);
80
81         info.spilled_node = value;
82         res = set_find(env->spills, &info, sizeof(info), hash);
83
84         if (res == NULL) {
85                 info.reloaders = NULL;
86                 info.spill = NULL;
87                 res = set_insert(env->spills, &info, sizeof(info), hash);
88         }
89
90         return res;
91 }
92
93 DEBUG_ONLY(
94 /* Sets the debug module of a spill environment. */
95 void be_set_spill_env_dbg_module(spill_env_t *env, firm_dbg_module_t *dbg) {
96         env->dbg = dbg;
97 }
98 )
99
100 /* Creates a new spill environment. */
101 spill_env_t *be_new_spill_env(const be_chordal_env_t *chordal_env) {
102         spill_env_t *env        = xmalloc(sizeof(env[0]));
103         env->spills                     = new_set(cmp_spillinfo, 1024);
104         env->cls                        = chordal_env->cls;
105         env->chordal_env        = chordal_env;
106         env->mem_phis           = pset_new_ptr_default();
107         obstack_init(&env->obst);
108         return env;
109 }
110
111 /* Deletes a spill environment. */
112 void be_delete_spill_env(spill_env_t *env) {
113         del_set(env->spills);
114         del_pset(env->mem_phis);
115         obstack_free(&env->obst, NULL);
116         free(env);
117 }
118
119 /**
120  *  ____  _                  ____      _                 _
121  * |  _ \| | __ _  ___ ___  |  _ \ ___| | ___   __ _  __| |___
122  * | |_) | |/ _` |/ __/ _ \ | |_) / _ \ |/ _ \ / _` |/ _` / __|
123  * |  __/| | (_| | (_|  __/ |  _ <  __/ | (_) | (_| | (_| \__ \
124  * |_|   |_|\__,_|\___\___| |_| \_\___|_|\___/ \__,_|\__,_|___/
125  *
126  */
127
128 void be_add_reload(spill_env_t *env, ir_node *to_spill, ir_node *before) {
129         spill_info_t *info;
130         reloader_t *rel;
131
132         assert(sched_is_scheduled(before));
133         assert(arch_irn_consider_in_reg_alloc(env->chordal_env->birg->main_env->arch_env, env->cls, to_spill));
134
135         info = get_spillinfo(env, to_spill);
136
137         if(is_Phi(to_spill)) {
138                 int i, arity;
139                 // create spillinfos for the phi arguments
140                 for(i = 0, arity = get_irn_arity(to_spill); i < arity; ++i) {
141                         ir_node *arg = get_irn_n(to_spill, i);
142                         get_spillinfo(env, arg);
143                 }
144         }
145
146         rel           = obstack_alloc(&env->obst, sizeof(rel[0]));
147         rel->reloader = before;
148         rel->next     = info->reloaders;
149         info->reloaders = rel;
150         be_liveness_add_missing(env->chordal_env->lv);
151 }
152
153 void be_add_reload_on_edge(spill_env_t *env, ir_node *to_spill, ir_node *block, int pos) {
154         ir_node *predblock, *last;
155
156         /* simply add the reload to the beginning of the block if we only have 1 predecessor
157          * (we don't need to check for phis as there can't be any in a block with only 1 pred)
158          */
159         if(get_Block_n_cfgpreds(block) == 1) {
160                 assert(!is_Phi(sched_first(block)));
161                 be_add_reload(env, to_spill, sched_first(block));
162                 return;
163         }
164
165         /* We have to reload the value in pred-block */
166         predblock = get_Block_cfgpred_block(block, pos);
167         last = sched_last(predblock);
168
169         /* we might have projs and keepanys behind the jump... */
170         while(is_Proj(last) || be_is_Keep(last)) {
171                 last = sched_prev(last);
172                 assert(!sched_is_end(last));
173         }
174         assert(is_cfop(last));
175
176         // add the reload before the (cond-)jump
177         be_add_reload(env, to_spill, last);
178 }
179
180 void be_spill_phi(spill_env_t *env, ir_node *node) {
181         int i, arity;
182
183         assert(is_Phi(node));
184
185         pset_insert_ptr(env->mem_phis, node);
186
187         // create spillinfos for the phi arguments
188         get_spillinfo(env, node);
189         for(i = 0, arity = get_irn_arity(node); i < arity; ++i) {
190                 ir_node *arg = get_irn_n(node, i);
191                 get_spillinfo(env, arg);
192         }
193 }
194
195 /*
196  *   ____                _         ____        _ _ _
197  *  / ___|_ __ ___  __ _| |_ ___  / ___| _ __ (_) | |___
198  * | |   | '__/ _ \/ _` | __/ _ \ \___ \| '_ \| | | / __|
199  * | |___| | |  __/ (_| | ||  __/  ___) | |_) | | | \__ \
200  *  \____|_|  \___|\__,_|\__\___| |____/| .__/|_|_|_|___/
201  *                                      |_|
202  */
203
204 /**
205  * Schedules a node after an instruction. (That is the place after all projs and phis
206  * that are scheduled after the instruction)
207  * This function also skips phi nodes at the beginning of a block
208  */
209 static void sched_add_after_insn(ir_node *sched_after, ir_node *node) {
210         ir_node *next = sched_next(sched_after);
211         while(is_Proj(next) || is_Phi(next)) {
212                 next = sched_next(next);
213         }
214         assert(next != NULL);
215
216         if(sched_is_end(next)) {
217                 sched_add_after(sched_last(get_nodes_block(sched_after)), node);
218         } else {
219                 sched_add_before(next, node);
220         }
221 }
222
223 /**
224  * Creates a spill.
225  *
226  * @param senv      the spill environment
227  * @param irn       the node that should be spilled
228  * @param ctx_irn   an user of the spilled node
229  *
230  * @return a be_Spill node
231  */
232 static void spill_irn(spill_env_t *env, spill_info_t *spillinfo) {
233         const be_main_env_t *mainenv = env->chordal_env->birg->main_env;
234         ir_node *to_spill = spillinfo->spilled_node;
235
236         DBG((env->dbg, LEVEL_1, "%+F\n", to_spill));
237
238         /* Trying to spill an already spilled value, no need for a new spill
239          * node then, we can simply connect to the same one for this reload
240          *
241          * (although rematerialisation code should handle most of these cases
242          * this can still happen when spilling Phis)
243          */
244         if(be_is_Reload(to_spill)) {
245                 spillinfo->spill = get_irn_n(to_spill, be_pos_Reload_mem);
246                 return;
247         }
248
249         spillinfo->spill = be_spill(mainenv->arch_env, to_spill);
250         sched_add_after_insn(to_spill, spillinfo->spill);
251 }
252
253 static void spill_node(spill_env_t *env, spill_info_t *spillinfo);
254
255 /**
256  * If the first usage of a Phi result would be out of memory
257  * there is no sense in allocating a register for it.
258  * Thus we spill it and all its operands to the same spill slot.
259  * Therefore the phi/dataB becomes a phi/Memory
260  *
261  * @param senv      the spill environment
262  * @param phi       the Phi node that should be spilled
263  * @param ctx_irn   an user of the spilled node
264  */
265 static void spill_phi(spill_env_t *env, spill_info_t *spillinfo) {
266         ir_node *phi = spillinfo->spilled_node;
267         int i;
268         int arity = get_irn_arity(phi);
269         ir_node     *block    = get_nodes_block(phi);
270         ir_node     **ins;
271
272         assert(is_Phi(phi));
273
274         /* build a new PhiM */
275         ins = alloca(sizeof(ir_node*) * arity);
276         for(i = 0; i < arity; ++i) {
277                 ins[i] = get_irg_bad(env->chordal_env->irg);
278         }
279         spillinfo->spill = new_r_Phi(env->chordal_env->irg, block, arity, ins, mode_M);
280
281         for(i = 0; i < arity; ++i) {
282                 ir_node *arg = get_irn_n(phi, i);
283                 spill_info_t *arg_info = get_spillinfo(env, arg);
284
285                 spill_node(env, arg_info);
286
287                 set_irn_n(spillinfo->spill, i, arg_info->spill);
288         }
289 }
290
291 /**
292  * Spill a node.
293  *
294  * @param senv      the spill environment
295  * @param to_spill  the node that should be spilled
296  */
297 static void spill_node(spill_env_t *env, spill_info_t *spillinfo) {
298         ir_node *to_spill;
299
300         // the node should be tagged for spilling already...
301         if(spillinfo->spill != NULL)
302                 return;
303
304         to_spill = spillinfo->spilled_node;
305         if (is_Phi(to_spill) && pset_find_ptr(env->mem_phis, spillinfo->spilled_node)) {
306                 spill_phi(env, spillinfo);
307         } else {
308                 spill_irn(env, spillinfo);
309         }
310 }
311
312 /*
313  *
314  *  ____                      _            _       _ _
315  * |  _ \ ___ _ __ ___   __ _| |_ ___ _ __(_) __ _| (_)_______
316  * | |_) / _ \ '_ ` _ \ / _` | __/ _ \ '__| |/ _` | | |_  / _ \
317  * |  _ <  __/ | | | | | (_| | ||  __/ |  | | (_| | | |/ /  __/
318  * |_| \_\___|_| |_| |_|\__,_|\__\___|_|  |_|\__,_|_|_/___\___|
319  *
320  */
321
322 /**
323  * Tests whether value @p arg is available before node @p reloader
324  * @returns 1 if value is available, 0 otherwise
325  */
326 static int is_value_available(spill_env_t *env, ir_node *arg, ir_node *reloader) {
327         if(is_Unknown(arg) || arg == new_NoMem())
328                 return 1;
329
330         if(be_is_Spill(arg))
331                 return 1;
332
333         if(arg == get_irg_frame(env->chordal_env->irg))
334                 return 1;
335
336 #if 0
337         /* we want to remat before the insn reloader
338          * thus an arguments is alive if
339          *   - it interferes with the reloaders result
340          * or
341          *   - or it is (last-) used by reloader itself
342          */
343         int i, m;
344
345         if (values_interfere(reloader, arg))
346                 return 1;
347
348         for (i=0, m=get_irn_arity(reloader); i<m; ++i) {
349                 ir_node *rel_arg = get_irn_n(reloader, i);
350                 if (rel_arg == arg)
351                         return 1;
352         }
353
354         /* arg is not alive before reloader */
355         return 0;
356 #endif
357
358         return 0;
359 }
360
361 /**
362  * Checks whether the node can principally be rematerialized
363  */
364 static int is_remat_node(spill_env_t *env, ir_node *node) {
365         const arch_env_t *arch_env = env->chordal_env->birg->main_env->arch_env;
366
367         assert(!be_is_Spill(node));
368
369         if(be_is_Reload(node))
370                 return 1;
371
372         // TODO why does arch_irn_is say rematerializable anyway?
373         if(be_is_Barrier(node))
374                 return 0;
375
376         if(arch_irn_is(arch_env, node, rematerializable))
377                 return 1;
378
379         if(be_is_StackParam(node))
380                 return 1;
381
382         return 0;
383 }
384
385 /**
386  * Check if a node is rematerializable. This tests for the following conditions:
387  *
388  * - The node itself is rematerializable
389  * - All arguments of the node are available or also rematerialisable
390  * - The costs for the rematerialisation operation is less or equal a limit
391  *
392  * Returns the costs needed for rematerialisation or something
393  * > REMAT_COST_LIMIT if remat is not possible.
394  */
395 static int check_remat_conditions_costs(spill_env_t *env, ir_node *spilled, ir_node *reloader, int parentcosts) {
396         int i, arity;
397         int argremats;
398         int costs = 0;
399
400         if(!is_remat_node(env, spilled))
401                 return REMAT_COST_LIMIT;
402
403         if(be_is_Reload(spilled)) {
404                 costs += 2;
405         } else if(is_Proj(spilled)) {
406                 costs += 0;
407         } else {
408                 costs += 1;
409         }
410         if(parentcosts + costs >= REMAT_COST_LIMIT)
411                 return REMAT_COST_LIMIT;
412
413         argremats = 0;
414         for(i = 0, arity = get_irn_arity(spilled); i < arity; ++i) {
415                 ir_node *arg = get_irn_n(spilled, i);
416
417                 if(is_value_available(env, arg, reloader))
418                         continue;
419
420                 // we have to rematerialize the argument as well...
421                 if(argremats >= 1) {
422                         /* we only support rematerializing 1 argument at the moment,
423                          * so that we don't have to care about register pressure
424                          */
425                         return REMAT_COST_LIMIT;
426                 }
427                 argremats++;
428
429                 // TODO can we get more accurate costs than +1?
430                 costs += check_remat_conditions_costs(env, arg, reloader, parentcosts + costs);
431                 if(parentcosts + costs >= REMAT_COST_LIMIT)
432                         return REMAT_COST_LIMIT;
433         }
434
435         return costs;
436 }
437
438 static int check_remat_conditions(spill_env_t *env, ir_node *spilled, ir_node *reloader) {
439         int costs = check_remat_conditions_costs(env, spilled, reloader, 1);
440
441         return costs < REMAT_COST_LIMIT;
442 }
443
444 /**
445  * Re-materialize a node.
446  *
447  * @param senv      the spill environment
448  * @param spilled   the node that was spilled
449  * @param reloader  a irn that requires a reload
450  */
451 static ir_node *do_remat(spill_env_t *env, ir_node *spilled, ir_node *reloader) {
452         int i, arity;
453         ir_node *res;
454         ir_node *bl = get_nodes_block(reloader);
455         ir_node **ins;
456
457         ins = alloca(get_irn_arity(spilled) * sizeof(ins[0]));
458         for(i = 0, arity = get_irn_arity(spilled); i < arity; ++i) {
459                 ir_node *arg = get_irn_n(spilled, i);
460
461                 if(is_value_available(env, arg, reloader)) {
462                         ins[i] = arg;
463                 } else {
464                         ins[i] = do_remat(env, arg, reloader);
465                 }
466         }
467
468         /* create a copy of the node */
469         res = new_ir_node(get_irn_dbg_info(spilled), env->chordal_env->irg, bl,
470                 get_irn_op(spilled),
471                 get_irn_mode(spilled),
472                 get_irn_arity(spilled),
473                 ins);
474         copy_node_attr(spilled, res);
475
476         DBG((env->dbg, LEVEL_1, "Insert remat %+F before reloader %+F\n", res, reloader));
477
478         /* insert in schedule */
479         assert(!is_Block(reloader));
480         sched_add_before(reloader, res);
481
482         return res;
483 }
484
485 /*
486  *  ___                     _     ____      _                 _
487  * |_ _|_ __  ___  ___ _ __| |_  |  _ \ ___| | ___   __ _  __| |___
488  *  | || '_ \/ __|/ _ \ '__| __| | |_) / _ \ |/ _ \ / _` |/ _` / __|
489  *  | || | | \__ \  __/ |  | |_  |  _ <  __/ | (_) | (_| | (_| \__ \
490  * |___|_| |_|___/\___|_|   \__| |_| \_\___|_|\___/ \__,_|\__,_|___/
491  *
492  */
493
494 void be_insert_spills_reloads(spill_env_t *env) {
495         const arch_env_t *arch_env = env->chordal_env->birg->main_env->arch_env;
496         spill_info_t *si;
497
498         /* process each spilled node */
499         DBG((env->dbg, LEVEL_1, "Insert spills and reloads:\n"));
500         for(si = set_first(env->spills); si; si = set_next(env->spills)) {
501                 reloader_t *rld;
502                 ir_mode *mode = get_irn_mode(si->spilled_node);
503                 pset *values = pset_new_ptr(16);
504
505                 /* go through all reloads for this spill */
506                 for(rld = si->reloaders; rld; rld = rld->next) {
507                         ir_node *new_val;
508
509                         if (check_remat_conditions(env, si->spilled_node, rld->reloader)) {
510                                 new_val = do_remat(env, si->spilled_node, rld->reloader);
511                         } else {
512                                 /* make sure we have a spill */
513                                 spill_node(env, si);
514
515                                 /* do a reload */
516                                 new_val = be_reload(arch_env, env->cls, rld->reloader, mode, si->spill);
517                         }
518
519                         DBG((env->dbg, LEVEL_1, " %+F of %+F before %+F\n", new_val, si->spilled_node, rld->reloader));
520                         pset_insert_ptr(values, new_val);
521                 }
522
523                 if(pset_count(values) > 0) {
524                         /* introduce copies, rewire the uses */
525                         pset_insert_ptr(values, si->spilled_node);
526                         be_ssa_constr_set_ignore(env->chordal_env->dom_front, env->chordal_env->lv, values, env->mem_phis);
527                 }
528
529                 del_pset(values);
530         }
531
532         // reloads are placed now, but we might reuse the spill environment for further spilling decisions
533         del_set(env->spills);
534         env->spills = new_set(cmp_spillinfo, 1024);
535 }