2 * Author: Daniel Grund, Sebastian Hack, Matthias Braun
4 * Copyright: (c) Universitaet Karlsruhe
5 * Licence: This file protected by GPL - GNU GENERAL PUBLIC LICENSE.
16 #include "iredges_t.h"
24 #include "unionfind.h"
28 #include "besched_t.h"
32 #include "bechordal_t.h"
33 #include "bejavacoal.h"
35 // only rematerialise when costs are less than REMAT_COST_LIMIT
36 #define REMAT_COST_LIMIT 4
38 /* This enables re-computation of values. Current state: Unfinished and buggy. */
41 typedef struct _reloader_t reloader_t;
48 typedef struct _spill_info_t {
49 ir_node *spilled_node;
50 reloader_t *reloaders;
56 const arch_register_class_t *cls;
57 const be_chordal_env_t *chordal_env;
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 */
62 DEBUG_ONLY(firm_dbg_module_t *dbg;)
66 * Compare two spill infos.
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;
75 * Returns spill info for a specific value (the value that is to be spilled)
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);
81 info.spilled_node = value;
82 res = set_find(env->spills, &info, sizeof(info), hash);
85 info.reloaders = NULL;
87 res = set_insert(env->spills, &info, sizeof(info), hash);
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) {
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);
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);
121 * | _ \| | __ _ ___ ___ | _ \ ___| | ___ __ _ __| |___
122 * | |_) | |/ _` |/ __/ _ \ | |_) / _ \ |/ _ \ / _` |/ _` / __|
123 * | __/| | (_| | (_| __/ | _ < __/ | (_) | (_| | (_| \__ \
124 * |_| |_|\__,_|\___\___| |_| \_\___|_|\___/ \__,_|\__,_|___/
128 void be_add_reload(spill_env_t *env, ir_node *to_spill, ir_node *before) {
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));
135 info = get_spillinfo(env, to_spill);
137 if(is_Phi(to_spill)) {
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);
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);
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;
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)
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));
165 /* We have to reload the value in pred-block */
166 predblock = get_Block_cfgpred_block(block, pos);
167 last = sched_last(predblock);
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));
174 assert(is_cfop(last));
176 // add the reload before the (cond-)jump
177 be_add_reload(env, to_spill, last);
180 void be_spill_phi(spill_env_t *env, ir_node *node) {
183 assert(is_Phi(node));
185 pset_insert_ptr(env->mem_phis, node);
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);
197 * / ___|_ __ ___ __ _| |_ ___ / ___| _ __ (_) | |___
198 * | | | '__/ _ \/ _` | __/ _ \ \___ \| '_ \| | | / __|
199 * | |___| | | __/ (_| | || __/ ___) | |_) | | | \__ \
200 * \____|_| \___|\__,_|\__\___| |____/| .__/|_|_|_|___/
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
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);
214 assert(next != NULL);
216 if(sched_is_end(next)) {
217 sched_add_after(sched_last(get_nodes_block(sched_after)), node);
219 sched_add_before(next, node);
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
230 * @return a be_Spill node
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;
236 DBG((env->dbg, LEVEL_1, "%+F\n", to_spill));
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
241 * (although rematerialisation code should handle most of these cases
242 * this can still happen when spilling Phis)
244 if(be_is_Reload(to_spill)) {
245 spillinfo->spill = get_irn_n(to_spill, be_pos_Reload_mem);
249 spillinfo->spill = be_spill(mainenv->arch_env, to_spill);
250 sched_add_after_insn(to_spill, spillinfo->spill);
253 static void spill_node(spill_env_t *env, spill_info_t *spillinfo);
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
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
265 static void spill_phi(spill_env_t *env, spill_info_t *spillinfo) {
266 ir_node *phi = spillinfo->spilled_node;
268 int arity = get_irn_arity(phi);
269 ir_node *block = get_nodes_block(phi);
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);
279 spillinfo->spill = new_r_Phi(env->chordal_env->irg, block, arity, ins, mode_M);
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);
285 spill_node(env, arg_info);
287 set_irn_n(spillinfo->spill, i, arg_info->spill);
294 * @param senv the spill environment
295 * @param to_spill the node that should be spilled
297 static void spill_node(spill_env_t *env, spill_info_t *spillinfo) {
300 // the node should be tagged for spilling already...
301 if(spillinfo->spill != NULL)
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);
308 spill_irn(env, spillinfo);
315 * | _ \ ___ _ __ ___ __ _| |_ ___ _ __(_) __ _| (_)_______
316 * | |_) / _ \ '_ ` _ \ / _` | __/ _ \ '__| |/ _` | | |_ / _ \
317 * | _ < __/ | | | | | (_| | || __/ | | | (_| | | |/ / __/
318 * |_| \_\___|_| |_| |_|\__,_|\__\___|_| |_|\__,_|_|_/___\___|
323 * Tests whether value @p arg is available before node @p reloader
324 * @returns 1 if value is available, 0 otherwise
326 static int is_value_available(spill_env_t *env, ir_node *arg, ir_node *reloader) {
327 if(is_Unknown(arg) || arg == new_NoMem())
333 if(arg == get_irg_frame(env->chordal_env->irg))
337 /* we want to remat before the insn reloader
338 * thus an arguments is alive if
339 * - it interferes with the reloaders result
341 * - or it is (last-) used by reloader itself
345 if (values_interfere(reloader, arg))
348 for (i=0, m=get_irn_arity(reloader); i<m; ++i) {
349 ir_node *rel_arg = get_irn_n(reloader, i);
354 /* arg is not alive before reloader */
362 * Checks whether the node can principally be rematerialized
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;
367 assert(!be_is_Spill(node));
369 if(be_is_Reload(node))
372 // TODO why does arch_irn_is say rematerializable anyway?
373 if(be_is_Barrier(node))
376 if(arch_irn_is(arch_env, node, rematerializable))
379 if(be_is_StackParam(node))
386 * Check if a node is rematerializable. This tests for the following conditions:
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
392 * Returns the costs needed for rematerialisation or something
393 * > REMAT_COST_LIMIT if remat is not possible.
395 static int check_remat_conditions_costs(spill_env_t *env, ir_node *spilled, ir_node *reloader, int parentcosts) {
400 if(!is_remat_node(env, spilled))
401 return REMAT_COST_LIMIT;
403 if(be_is_Reload(spilled)) {
405 } else if(is_Proj(spilled)) {
410 if(parentcosts + costs >= REMAT_COST_LIMIT)
411 return REMAT_COST_LIMIT;
414 for(i = 0, arity = get_irn_arity(spilled); i < arity; ++i) {
415 ir_node *arg = get_irn_n(spilled, i);
417 if(is_value_available(env, arg, reloader))
420 // we have to rematerialize the argument as well...
422 /* we only support rematerializing 1 argument at the moment,
423 * so that we don't have to care about register pressure
425 return REMAT_COST_LIMIT;
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;
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);
441 return costs < REMAT_COST_LIMIT;
445 * Re-materialize a node.
447 * @param senv the spill environment
448 * @param spilled the node that was spilled
449 * @param reloader a irn that requires a reload
451 static ir_node *do_remat(spill_env_t *env, ir_node *spilled, ir_node *reloader) {
454 ir_node *bl = get_nodes_block(reloader);
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);
461 if(is_value_available(env, arg, reloader)) {
464 ins[i] = do_remat(env, arg, reloader);
468 /* create a copy of the node */
469 res = new_ir_node(get_irn_dbg_info(spilled), env->chordal_env->irg, bl,
471 get_irn_mode(spilled),
472 get_irn_arity(spilled),
474 copy_node_attr(spilled, res);
476 DBG((env->dbg, LEVEL_1, "Insert remat %+F before reloader %+F\n", res, reloader));
477 ir_printf("Insert remat %+F for %+F before reloader %+F(%s)\n", res, spilled, reloader, get_irg_dump_name(get_irn_irg(reloader)));
479 /* insert in schedule */
480 assert(!is_Block(reloader));
481 sched_add_before(reloader, res);
488 * |_ _|_ __ ___ ___ _ __| |_ | _ \ ___| | ___ __ _ __| |___
489 * | || '_ \/ __|/ _ \ '__| __| | |_) / _ \ |/ _ \ / _` |/ _` / __|
490 * | || | | \__ \ __/ | | |_ | _ < __/ | (_) | (_| | (_| \__ \
491 * |___|_| |_|___/\___|_| \__| |_| \_\___|_|\___/ \__,_|\__,_|___/
495 void be_insert_spills_reloads(spill_env_t *env) {
496 const arch_env_t *arch_env = env->chordal_env->birg->main_env->arch_env;
499 /* process each spilled node */
500 DBG((env->dbg, LEVEL_1, "Insert spills and reloads:\n"));
501 for(si = set_first(env->spills); si; si = set_next(env->spills)) {
503 ir_mode *mode = get_irn_mode(si->spilled_node);
504 pset *values = pset_new_ptr(16);
506 /* go through all reloads for this spill */
507 for(rld = si->reloaders; rld; rld = rld->next) {
510 if (check_remat_conditions(env, si->spilled_node, rld->reloader)) {
511 new_val = do_remat(env, si->spilled_node, rld->reloader);
513 /* make sure we have a spill */
517 new_val = be_reload(arch_env, env->cls, rld->reloader, mode, si->spill);
520 DBG((env->dbg, LEVEL_1, " %+F of %+F before %+F\n", new_val, si->spilled_node, rld->reloader));
521 pset_insert_ptr(values, new_val);
524 if(pset_count(values) > 0) {
525 /* introduce copies, rewire the uses */
526 pset_insert_ptr(values, si->spilled_node);
527 be_ssa_constr_set_ignore(env->chordal_env->dom_front, env->chordal_env->lv, values, env->mem_phis);
533 // reloads are placed now, but we might reuse the spill environment for further spilling decisions
534 del_set(env->spills);
535 env->spills = new_set(cmp_spillinfo, 1024);