Files
mercury/compiler/livemap.m
Zoltan Somogyi 5e25af41f7 This diff changes the LLDS backend to make it easier to read and to maintain,
Estimated hours taken: 6
Branches: main

This diff changes the LLDS backend to make it easier to read and to maintain,
but contains no changes in algorithms whatsoever.

compiler/basic_block.m:
compiler/dupelim.m:
compiler/frameopt.m:
compiler/jumpopt.m:
compiler/layout_out.m:
compiler/livemap.m:
compiler/peephole.m:
compiler/reassign.m:
compiler/rtti_out.m:
compiler/use_local_vars.m:
	Convert these modules to our current coding standards. Use state
	variable notation when appropriate, reordering arguments as necessary.

compiler/llds_out.m:
	Convert these modules to our current coding standards. Use state
	variable notation when appropriate, reordering arguments as necessary.

	Delete predicates which are just specialized forms of foldl, using
	foldl (or foldl2 etc) directly instead.

	Factor out some common code.

compiler/livemap.m:
	Convert these modules to our current coding standards. Use state
	variable notation when appropriate, reordering arguments as necessary.

	Remove some special case handling that used to be required by the value
	numbering pass.

library/bintree_set.m:
	Provide a function version of the initialization predicate.
2003-11-11 03:35:13 +00:00

452 lines
15 KiB
Mathematica

%-----------------------------------------------------------------------------%
% Copyright (C) 1995-2003 The University of Melbourne.
% This file may only be copied under the terms of the GNU General
% Public License - see the file COPYING in the Mercury distribution.
%-----------------------------------------------------------------------------%
%
% File: livemap.m
%
% Main author: zs.
%
% This module builds up a map that gives the set of live lvals at each label.
%-----------------------------------------------------------------------------%
:- module ll_backend__livemap.
:- interface.
:- import_module ll_backend__llds.
:- import_module list, set, map, std_util.
:- type livemap == map(label, lvalset).
:- type lvalset == set(lval).
% Given a list of instructions defining a procedure, return a map
% giving the set of live non-field lvals at each label.
%
% We can compute this set only if the procedure contains no C code.
:- pred livemap__build(list(instruction)::in, maybe(livemap)::out) is det.
:- implementation.
:- import_module ll_backend__opt_util.
:- import_module require, string, bool.
%-----------------------------------------------------------------------------%
%-----------------------------------------------------------------------------%
% The method we follow is a backward scan of the instruction list,
% keeping track of the set of live lvals as we go. We update this set
% at each instruction. When we get to a label, we know that this set
% of lvals is live at that label.
%
% At instructions that can branch away, every lval that is live at
% any possible target is live before that instruction. Since some
% branches may be backward branches, we may not have seen the branch
% target when we process the branch. Therefore we have to repeat the
% scan, this time with more knowledge about more labels, until we
% get to a fixpoint.
livemap__build(Instrs, MaybeLivemap) :-
map__init(Livemap0),
list__reverse(Instrs, BackInstrs),
livemap__build_2(BackInstrs, Livemap0, MaybeLivemap).
:- pred livemap__build_2(list(instruction)::in, livemap::in,
maybe(livemap)::out) is det.
livemap__build_2(Backinstrs, Livemap0, MaybeLivemap) :-
set__init(Livevals0),
livemap__build_livemap(Backinstrs, Livevals0, no, ContainsUserCode,
Livemap0, Livemap1),
( ContainsUserCode = yes ->
MaybeLivemap = no
; livemap__equal_livemaps(Livemap0, Livemap1) ->
MaybeLivemap = yes(Livemap1)
;
livemap__build_2(Backinstrs, Livemap1, MaybeLivemap)
).
% Check whether the two livemaps agree on the set of live lvals
% at every label. They must agree on the set of labels as well.
% This is important. Livemap1 will be empty in the first call,
% so agreement only on the set of labels in Livemap1 is useless.
% The domain of Livemap2 should always be every label in the procedure.
% as should the domain of Livemap1 in every call after the first.
:- pred livemap__equal_livemaps(livemap::in, livemap::in) is semidet.
livemap__equal_livemaps(Livemap1, Livemap2) :-
map__keys(Livemap1, Labels),
map__keys(Livemap2, Labels),
livemap__equal_livemaps_keys(Labels, Livemap1, Livemap2).
:- pred livemap__equal_livemaps_keys(list(label)::in, livemap::in, livemap::in)
is semidet.
livemap__equal_livemaps_keys([], _Livemap1, _Livemap2).
livemap__equal_livemaps_keys([Label | Labels], Livemap1, Livemap2) :-
map__lookup(Livemap1, Label, Liveset1),
map__lookup(Livemap2, Label, Liveset2),
set__equal(Liveset1, Liveset2),
livemap__equal_livemaps_keys(Labels, Livemap1, Livemap2).
%-----------------------------------------------------------------------------%
%-----------------------------------------------------------------------------%
% Build up a map of what lvals are live at each label.
% The input instruction sequence is reversed.
:- pred livemap__build_livemap(list(instruction)::in, lvalset::in,
bool::in, bool::out, livemap::in, livemap::out) is det.
livemap__build_livemap([], _, !ContainsUserCode, !Livemap).
livemap__build_livemap([Instr0 | Instrs0], Livevals0,
!ContainsUserCode, !Livemap) :-
livemap__build_livemap_instr(Instr0, Instrs0, Instrs1,
Livevals0, Livevals1, !ContainsUserCode, !Livemap),
livemap__build_livemap(Instrs1, Livevals1, !ContainsUserCode, !Livemap).
:- pred livemap__build_livemap_instr(instruction::in, list(instruction)::in,
list(instruction)::out, lvalset::in, lvalset::out,
bool::in, bool::out, livemap::in, livemap::out) is det.
livemap__build_livemap_instr(Instr0, !Instrs, !Livevals, !ContainsUserCode,
!Livemap) :-
Instr0 = Uinstr0 - _,
(
Uinstr0 = comment(_)
;
Uinstr0 = livevals(_),
error("livevals found in backward scan in build_livemap")
;
Uinstr0 = block(_, _, _),
error("block found in backward scan in build_livemap")
;
Uinstr0 = assign(Lval, Rval),
% Make dead the variable assigned, but make any variables
% needed to access it live. Make the variables in the assigned
% expression live as well.
% The deletion has to be done first. If the assigned-to lval
% appears on the right hand side as well as the left, then we
% want make_live to put it back into the liveval set.
set__delete(!.Livevals, Lval, !:Livevals),
opt_util__lval_access_rvals(Lval, Rvals),
livemap__make_live_in_rvals([Rval | Rvals], !Livevals)
;
Uinstr0 = call(_, _, _, _, _, _),
livemap__look_for_livevals(!Instrs, !Livevals, "call", yes, _)
;
Uinstr0 = mkframe(_, _)
;
Uinstr0 = label(Label),
map__set(!.Livemap, Label, !.Livevals, !:Livemap)
;
Uinstr0 = goto(CodeAddr),
opt_util__livevals_addr(CodeAddr, LivevalsNeeded),
livemap__look_for_livevals(!Instrs, !Livevals, "goto",
LivevalsNeeded, Found),
( Found = yes ->
true
; CodeAddr = label(Label) ->
livemap__insert_label_livevals([Label], !.Livemap,
set__init, !:Livevals)
;
( CodeAddr = do_redo
; CodeAddr = do_fail
; CodeAddr = do_not_reached
)
->
true
;
error("unknown label type in build_livemap")
),
livemap__special_code_addr(CodeAddr, MaybeSpecial),
( MaybeSpecial = yes(Special) ->
set__insert(!.Livevals, Special, !:Livevals)
;
true
)
;
Uinstr0 = computed_goto(Rval, Labels),
livemap__make_live_in_rvals([Rval], set__init, !:Livevals),
livemap__insert_label_livevals(Labels, !.Livemap, !Livevals)
;
Uinstr0 = if_val(Rval, CodeAddr),
Livevals0 = !.Livevals,
livemap__look_for_livevals(!Instrs, !Livevals, "if_val",
no, Found),
(
Found = yes,
% This if_val was put here by middle_rec.
% We must make sure that the locations mentioned
% in the livevals annotation become live,
% since they will be needed at CodeAddr.
% The locations in Livevals0 may be needed
% in the fall-through continuation.
set__union(Livevals0, !Livevals)
;
Found = no,
livemap__make_live_in_rvals([Rval], !Livevals),
( CodeAddr = label(Label) ->
livemap__insert_label_livevals([Label],
!.Livemap, !Livevals)
;
true
)
),
livemap__special_code_addr(CodeAddr, MaybeSpecial),
( MaybeSpecial = yes(Special) ->
set__insert(!.Livevals, Special, !:Livevals)
;
true
)
;
Uinstr0 = incr_hp(Lval, _, _, Rval, _),
% Make dead the variable assigned, but make any variables
% needed to access it live. Make the variables in the size
% expression live as well.
% The use of the size rval occurs after the assignment
% to lval, but the two should never have any variables in
% common. This is why doing the deletion first works.
set__delete(!.Livevals, Lval, !:Livevals),
opt_util__lval_access_rvals(Lval, Rvals),
livemap__make_live_in_rvals([Rval | Rvals], !Livevals)
;
Uinstr0 = mark_hp(Lval),
set__delete(!.Livevals, Lval, !:Livevals),
opt_util__lval_access_rvals(Lval, Rvals),
livemap__make_live_in_rvals(Rvals, !Livevals)
;
Uinstr0 = restore_hp(Rval),
livemap__make_live_in_rvals([Rval], !Livevals)
;
Uinstr0 = free_heap(Rval),
livemap__make_live_in_rvals([Rval], !Livevals)
;
Uinstr0 = store_ticket(Lval),
set__delete(!.Livevals, Lval, !:Livevals),
opt_util__lval_access_rvals(Lval, Rvals),
livemap__make_live_in_rvals(Rvals, !Livevals)
;
Uinstr0 = reset_ticket(Rval, _Reason),
livemap__make_live_in_rval(Rval, !Livevals)
;
Uinstr0 = discard_ticket
;
Uinstr0 = prune_ticket
;
Uinstr0 = mark_ticket_stack(Lval),
set__delete(!.Livevals, Lval, !:Livevals),
opt_util__lval_access_rvals(Lval, Rvals),
livemap__make_live_in_rvals(Rvals, !Livevals)
;
Uinstr0 = prune_tickets_to(Rval),
livemap__make_live_in_rval(Rval, !Livevals)
;
Uinstr0 = incr_sp(_, _)
;
Uinstr0 = decr_sp(_)
;
Uinstr0 = init_sync_term(_, _)
;
Uinstr0 = fork(_, _, _)
;
Uinstr0 = join_and_terminate(_)
;
Uinstr0 = join_and_continue(_, _)
;
Uinstr0 = c_code(_, LiveLvalInfo),
livemap__build_live_lval_info(LiveLvalInfo,
!Livevals, !ContainsUserCode)
;
Uinstr0 = pragma_c(_, Components, _, _, _, _, _, _),
livemap__build_livemap_pragma_components(Components,
!Livevals, !ContainsUserCode)
).
:- pred livemap__build_livemap_pragma_components(list(pragma_c_component)::in,
lvalset::in, lvalset::out, bool::in, bool::out) is det.
livemap__build_livemap_pragma_components([], !Livevals, !ContainsUserCode).
livemap__build_livemap_pragma_components([Component | Components],
!Livevals, !ContainsUserCode) :-
(
Component = pragma_c_inputs(Inputs),
livemap__build_livemap_pragma_inputs(Inputs,
!Livevals)
;
Component = pragma_c_outputs(_)
;
Component = pragma_c_user_code(_, _),
!:ContainsUserCode = yes
;
Component = pragma_c_raw_code(_, LiveLvalInfo),
livemap__build_live_lval_info(LiveLvalInfo,
!Livevals, !ContainsUserCode)
;
Component = pragma_c_fail_to(_)
;
Component = pragma_c_noop
),
livemap__build_livemap_pragma_components(Components,
!Livevals, !ContainsUserCode).
:- pred livemap__build_live_lval_info(c_code_live_lvals::in,
lvalset::in, lvalset::out, bool::in, bool::out) is det.
livemap__build_live_lval_info(no_live_lvals_info, !Livevals, _, yes).
livemap__build_live_lval_info(live_lvals_info(LiveLvalSet), !Livevals,
!DontValueNumber) :-
set__to_sorted_list(LiveLvalSet, LiveLvals),
livemap__insert_proper_livevals(LiveLvals, !Livevals).
:- pred livemap__build_livemap_pragma_inputs(list(pragma_c_input)::in,
lvalset::in, lvalset::out) is det.
livemap__build_livemap_pragma_inputs([], !Livevals).
livemap__build_livemap_pragma_inputs([Input | Inputs], !Livevals) :-
Input = pragma_c_input(_, _, Rval, _),
( Rval = lval(Lval) ->
livemap__insert_proper_liveval(Lval, !Livevals)
;
true
),
livemap__build_livemap_pragma_inputs(Inputs, !Livevals).
:- pred livemap__look_for_livevals(list(instruction)::in,
list(instruction)::out, lvalset::in, lvalset::out, string::in,
bool::in, bool::out) is det.
livemap__look_for_livevals(Instrs0, Instrs, !Livevals,
Site, Compulsory, Found) :-
opt_util__skip_comments(Instrs0, Instrs1),
( Instrs1 = [livevals(Livevals1) - _ | Instrs2] ->
livemap__filter_livevals(Livevals1, !:Livevals),
Instrs = Instrs2,
Found = yes
; Compulsory = yes ->
string__append(Site, " not preceded by livevals", Msg),
error(Msg)
;
Instrs = Instrs1,
Found = no
).
% What lval (if any) is consulted when we branch to a code address?
:- pred livemap__special_code_addr(code_addr::in, maybe(lval)::out) is det.
livemap__special_code_addr(label(_), no).
livemap__special_code_addr(imported(_), no).
livemap__special_code_addr(succip, yes(succip)).
livemap__special_code_addr(do_succeed(_), yes(succip(lval(curfr)))).
livemap__special_code_addr(do_redo, yes(redoip(lval(maxfr)))).
livemap__special_code_addr(do_trace_redo_fail_shallow, no).
livemap__special_code_addr(do_trace_redo_fail_deep, no).
livemap__special_code_addr(do_fail, no).
livemap__special_code_addr(do_call_closure, no).
livemap__special_code_addr(do_call_class_method, no).
livemap__special_code_addr(do_not_reached, no).
%-----------------------------------------------------------------------------%
%-----------------------------------------------------------------------------%
:- pred livemap__make_live_in_rvals(list(rval)::in, lvalset::in, lvalset::out)
is det.
livemap__make_live_in_rvals([], !Live).
livemap__make_live_in_rvals([Rval | Rvals], !Live) :-
livemap__make_live_in_rval(Rval, !Live),
livemap__make_live_in_rvals(Rvals, !Live).
% Set all lvals found in this rval to live, with the exception of
% fields, since they are treated specially (the later stages consider
% them to be live even if they are not explicitly in the live set).
:- pred livemap__make_live_in_rval(rval::in, lvalset::in, lvalset::out) is det.
livemap__make_live_in_rval(lval(Lval), !Live) :-
% XXX maybe we should treat mem_refs the same way as field refs
( Lval = field(_, _, _) ->
true
;
set__insert(!.Live, Lval, !:Live)
),
opt_util__lval_access_rvals(Lval, AccessRvals),
livemap__make_live_in_rvals(AccessRvals, !Live).
livemap__make_live_in_rval(mkword(_, Rval), !Live) :-
livemap__make_live_in_rval(Rval, !Live).
livemap__make_live_in_rval(const(_), !Live).
livemap__make_live_in_rval(unop(_, Rval), !Live) :-
livemap__make_live_in_rval(Rval, !Live).
livemap__make_live_in_rval(binop(_, Rval1, Rval2), !Live) :-
livemap__make_live_in_rval(Rval1, !Live),
livemap__make_live_in_rval(Rval2, !Live).
livemap__make_live_in_rval(var(_), _, _) :-
error("var rval should not propagate to the optimizer").
livemap__make_live_in_rval(mem_addr(MemRef), !Live) :-
livemap__make_live_in_mem_ref(MemRef, !Live).
:- pred livemap__make_live_in_mem_ref(mem_ref::in, lvalset::in, lvalset::out)
is det.
livemap__make_live_in_mem_ref(stackvar_ref(_), !Live).
livemap__make_live_in_mem_ref(framevar_ref(_), !Live).
livemap__make_live_in_mem_ref(heap_ref(Rval, _, _), !Live) :-
livemap__make_live_in_rval(Rval, !Live).
%-----------------------------------------------------------------------------%
%-----------------------------------------------------------------------------%
:- pred livemap__filter_livevals(lvalset::in, lvalset::out) is det.
livemap__filter_livevals(Livevals0, Livevals) :-
set__to_sorted_list(Livevals0, Livelist),
set__init(Livevals1),
livemap__insert_proper_livevals(Livelist, Livevals1, Livevals).
:- pred livemap__insert_label_livevals(list(label)::in, livemap::in,
lvalset::in, lvalset::out) is det.
livemap__insert_label_livevals([], _, !Livevals).
livemap__insert_label_livevals([Label | Labels], Livemap, !Livevals) :-
( map__search(Livemap, Label, LabelLivevals) ->
set__to_sorted_list(LabelLivevals, Livelist),
livemap__insert_proper_livevals(Livelist, !Livevals)
;
true
),
livemap__insert_label_livevals(Labels, Livemap, !Livevals).
:- pred livemap__insert_proper_livevals(list(lval)::in, lvalset::in,
lvalset::out) is det.
livemap__insert_proper_livevals([], !Livevals).
livemap__insert_proper_livevals([Live | Livelist], !Livevals) :-
livemap__insert_proper_liveval(Live, !Livevals),
livemap__insert_proper_livevals(Livelist, !Livevals).
% Don't insert references to locations on the heap.
:- pred livemap__insert_proper_liveval(lval::in, lvalset::in, lvalset::out)
is det.
livemap__insert_proper_liveval(Live, !Livevals) :-
( Live = field(_, _, _) ->
true
;
set__insert(!.Livevals, Live, !:Livevals)
).
%-----------------------------------------------------------------------------%
%-----------------------------------------------------------------------------%