mirror of
https://github.com/Mercury-Language/mercury.git
synced 2025-12-13 04:44:39 +00:00
compiler/du_type_layout.m:
If the --allow-packing-remote-sectag option is set, then try to pack
an initial subsequence of subword-sized arguments next to remote sectags.
To allow the polymorphism transformation to put the type_infos and/or
typeclass_infos it adds to a function symbol's argument list at the
*front* of that argument list, pack arguments next to remote sectags
only in function symbols that won't have any such extra arguments
added to them.
Do not write all new code for the new optimization; instead, generalize
the code that already does a very similar job for packing args next to
local sectags.
Delete the code we used to have that picked the packed representation
over the base unpacked representation only if it reduced the
"rounded-to-even" number of words. A case could be made for its usefulness,
but in the presence of the new optimization the extra code complexity
it requires is not worth it (in my opinion).
Extend the code that informs users about possible argument order
rearrangements that yield better packing to take packing next to sectags
into account.
compiler/hlds_data.m:
Provide a representation for cons_tags that use the new optimization.
Instead of adding a new cons_tag, we do this by replacing several old
cons_tags that all represent pointers to memory cells with a single
cons_tag named remote_args_tag with an argument that selects among
the old cons_tags being replaced, and adding a new alternative inside
this new type. The new alternative is remote_args_shared with a
remote_sectag whose size is rsectag_subword(...).
Instead of representing the value of the "data" field in classes
on the Java and C# backends as a strange kind of secondary tag
that is added to a memory cell by a class constructor instead of
having to be explicitly added to the front of the argument vector
by the code of a unification, represent it more directly as separate
kind of remote_args_tag. Continuing to treat it as a sectag would have
been very confusing to readers of the code of ml_unify_gen_*.m in the
presence of the new optimization.
Replacing several cons_tags that were usually treated similarly with
one cons_tag simplifies many switches. Instead of an switch with that
branches to the same switch arm for single_functor_tag, unshared_tag
and shared_remote_tag, and then switches on these three tags again
to get e.g. the primary tag of each, the new code of the switch arm
is executed for just cons_tag value (remote_args_tag), and switches
on the various kinds of remote args tags only when it needs to.
In is also more natural to pass around the argument of remote_args_tag
than to pass around a variable of type cons_tag that can be bound to only
single_functor_tag, unshared_tag or shared_remote_tag.
Add an XXX about possible further steps along these lines, such as
making a new cons_tag named something like "user_const_tag" represent
all user-visible constants.
compiler/unify_gen_construct.m:
compiler/unify_gen_deconstruct.m:
compiler/unify_gen_test.m:
compiler/unify_gen_util.m:
compiler/ml_unify_gen_construct.m:
compiler/ml_unify_gen_deconstruct.m:
compiler/ml_unify_gen_test.m:
compiler/ml_unify_gen_util.m:
Implement X = f(Yi) unifications where f uses the new representation,
i.e. some of its arguments are stored next to a remote sectag.
Some of the Yi are stored in a tagword (a word that also contains a tag,
in this case the remote secondary tag), while some are stored in other
words in a memory cell. This means that such unifications have similarities
both to unifications involving arguments being packed next to local
sectags, and to unifications involving ordinary arguments in memory cells.
Therefore wherever possible, their implemenation uses suitably generalized
versions of existing code that did those two jobs for two separate kinds of
cons_tags.
Making such generalizations possible in some cases required shifting the
boundary between predicates, moving work from a caller to a callee
or vice versa.
In unify_gen_deconstruct.m, stop using uni_vals to represent *either* a var
*or* a word in a memory cell. While this enabled us to factor out some
common code, the predicate boundaries it lead to are unsuitable for the
generalizations we now need.
Consistently use unsigned ints to represent both the whole and the parts
of words containing packed arguments (and maybe sectags), except when
comparing ptag constants with the result of applying the "tag" unop
to a word, (since that unop returns an int, at least for now).
In a few cases, avoid the recomputation of some information that we
already know. The motivation is not efficiency, since the recomputation
we avoid is usually cheap, but the simplification of the code's correctness
argument.
Use more consistent terminology in things such as variable names.
Note the possibility of further future improvements in several places.
compiler/ml_foreign_proc_gen.m:
Delete a long unused predicate.
compiler/mlds.m:
Add an XXX documenting a possible improvement.
compiler/rtti.m:
Update the compiler's internal representation of RTTI data structures
to make them able to describe secondary tags that are smaller than
a full word.
compiler/rtti_out.m:
Conform to the changes above, and delete a long-unused predicate.
compiler/type_ctor_info.m:
Use the RTTI's du_hl_rep to represent cons_tags that distinguish
between function symbols using a field in a class.
compiler/ml_type_gen.m:
Provide a specialized form of a function for code in ml_unify_gen_*.m.
Conform to the changes above.
compiler/add_special_pred.m:
compiler/bytecode_gen.m:
compiler/export.m:
compiler/hlds_code_util.m:
compiler/lco.m:
compiler/ml_closure_gen.m:
compiler/ml_switch_gen.m:
compiler/ml_tag_switch.m:
compiler/rtti_to_mlds.m:
compiler/switch_util.m:
compiler/tag_switch.m:
Conform to the changes above.
runtime/mercury_type_info.h:
Update the runtime's representation of RTTI data structures to make them
able to describe remote secondary tags that are smaller than a full word.
runtime/mercury_deconstruct.[ch]:
runtime/mercury_deconstruct.h:
runtime/mercury_deconstruct_macros.h:
runtime/mercury_ml_expand_body.h:
runtime/mercury_ml_arg_body.h:
runtime/mercury_ml_deconstruct_body.h:
runtime/mercury_ml_functor_body.h:
These modules collectively implement the predicates in deconstruct.m
in the library, and provide access to its functionality to other C code,
e.g. in the debugger. Update these to be able to handle terms with the
new data representation optimization.
This update requires a significant change in the distribution of work
between these files for the predicates deconstruct.deconstruct and
deconstruct.limited_deconstruct. We used to have mercury_ml_expand_body.h
fill in the fields of their expand_info structures (whose types are
defined in mercury_deconstruct.h) with pointers to three vectors:
(a) a vector of arg_locns with one element per argument, with a NULL
pointer being equivalent to a vector with a given element in every slot;
(b) a vector of type_infos with one element per argument, constructed
dynamically (and later freed) if necessary; and (c) a vector of argument
words. Once upon a time, before double-word and sub-word arguments,
vector (c) also had one word per argument, but that hasn't been true
for a while; we added vector (a) help the consumers of the expand_info
decode the difference. The consumers of this info always used these
vectors to build up a Mercury term containing a list of univs,
with one univ for each argument.
This structure could be stretched to handle function symbols that store
*all* their arguments in a tagword next to a local sectag, but I found
that stretching it to cover function symbols that have *some* of their
arguments packed next to a remote sectag and *some other* of their
arguments in a memory cell as usual would have required a well-nigh
incomprehensibly complex, and therefore almost undebuggable, interface
between mercury_ml_expand_body.h and the other files above. This diff
therefore changes the interface to have mercury_ml_expand_body.h
build the list of univs directly. This make its code relatively simple
and self-contained, and it should be somewhat faster then the old code
as well, since it never needs to allocate, fill in and then free
vectors of type_infos (each such typeinfo now gets put into a univ
as soon as it is constructed). The downside is that if we ever wanted
to get all the arguments at once for a purpose other than constructing
a list of univs from them, it would nevertheless require constructing
that list of univs anyway as an intermediate data structure. I don't see
this downside is significant, because (a) I don't think such a use case
is very likely, and (b) even if one arises, debuggable but a bit slow
is probably preferable to faster but very hard to debug.
Reduce the level of indentation of some of these files to make the code
easier to edit. Do this by
- not adding an indent level from switch statements to their cases; and
- not adding an indent level when a case in a switch has a local block.
Move the break or return ending a case inside that case's block,
if it has one.
runtime/mercury_deep_copy_body.h:
runtime/mercury_table_type_body.h:
Update these to enable the copying or tabling of terms whose
representations uses the new optimization.
Use the techniques listed above to reduce the level of indentation
make the code easier to edit.
runtime/mercury_tabling.c:
runtime/mercury_term_size.c:
Conform to the changes above.
runtime/mercury_unify_compare_body.h:
Make this code compile after the changes above. It does need to work
correctly, since we only ever used this code to compare the speed
of unify-by-rtti with the speed of unify-by-compiler-generated-code,
and in real life, we always use the latter. (It hasn't been updated
to work right with previous arg packing changes either.)
library/construct.m:
Update to enable the code to construct terms whose representations
uses the new optimization.
Add some sanity checks.
library/private_builtin.m:
runtime/mercury_dotnet.cs.in:
java/runtime/Sectag_Locn.java:
Update the list of possible sectag kinds.
library/store.m:
Conform to the changes above.
trace/mercury_trace_vars.c:
Conform to the changes above.
tests/hard_coded/deconstruct_arg.{m,exp,exp2}:
Extend this test to test the deconstruction of terms whose
representations uses the new optimization.
Modify some of the existing terms being tested to make them more diverse,
in order to make the output easier to navigate.
tests/hard_coded/construct_packed.{m,exp}:
A new test case to test the construction of terms whose
representations uses the new optimization.
tests/debugger/browse_packed.{m,exp}:
A new test case to test access to the fields of terms whose
representations uses the new optimization.
tests/tabling/test_packed.{m,exp}:
A new test case to test the tabling of terms whose
representations uses the new optimization.
tests/debugger/Mmakefile:
tests/hard_coded/Mmakefile:
tests/tabling/Mmakefile:
Enable the new test cases.
1049 lines
42 KiB
Mathematica
1049 lines
42 KiB
Mathematica
%-----------------------------------------------------------------------------%
|
|
% vim: ft=mercury ts=4 sw=4 et
|
|
%-----------------------------------------------------------------------------%
|
|
% Copyright (C) 1994-2000,2002-2007, 2009-2011 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: tag_switch.m.
|
|
% Author: zs.
|
|
%
|
|
% Generate switches based on primary and secondary tags.
|
|
%
|
|
%-----------------------------------------------------------------------------%
|
|
|
|
:- module ll_backend.tag_switch.
|
|
:- interface.
|
|
|
|
:- import_module hlds.
|
|
:- import_module hlds.code_model.
|
|
:- import_module hlds.hlds_goal.
|
|
:- import_module ll_backend.code_info.
|
|
:- import_module ll_backend.code_loc_dep.
|
|
:- import_module ll_backend.llds.
|
|
:- import_module parse_tree.
|
|
:- import_module parse_tree.prog_data.
|
|
|
|
:- import_module list.
|
|
|
|
%-----------------------------------------------------------------------------%
|
|
|
|
% Generate intelligent indexing code for tag based switches.
|
|
%
|
|
:- pred generate_tag_switch(list(tagged_case)::in, rval::in, mer_type::in,
|
|
string::in, code_model::in, can_fail::in, hlds_goal_info::in, label::in,
|
|
branch_end::in, branch_end::out, llds_code::out,
|
|
code_info::in, code_info::out, code_loc_dep::in) is det.
|
|
|
|
%-----------------------------------------------------------------------------%
|
|
%-----------------------------------------------------------------------------%
|
|
|
|
:- implementation.
|
|
|
|
:- import_module backend_libs.
|
|
:- import_module backend_libs.builtin_ops.
|
|
:- import_module backend_libs.rtti.
|
|
:- import_module backend_libs.switch_util.
|
|
:- import_module hlds.hlds_data.
|
|
:- import_module hlds.hlds_llds.
|
|
:- import_module libs.
|
|
:- import_module libs.globals.
|
|
:- import_module libs.options.
|
|
:- import_module ll_backend.switch_case.
|
|
|
|
:- import_module assoc_list.
|
|
:- import_module cord.
|
|
:- import_module int.
|
|
:- import_module map.
|
|
:- import_module maybe.
|
|
:- import_module pair.
|
|
:- import_module require.
|
|
:- import_module string.
|
|
:- import_module uint8.
|
|
|
|
%-----------------------------------------------------------------------------%
|
|
|
|
% The idea is to generate two-level switches, first on the primary
|
|
% tag and then on the secondary tag. Since more than one function
|
|
% symbol can be eliminated by a failed primary tag test, this reduces
|
|
% the expected the number of comparisons required before finding the
|
|
% code corresponding to the actual value of the switch variable.
|
|
% We also get a speedup compared to non-tag switches by extracting
|
|
% the primary and secondary tags once instead of repeatedly for
|
|
% each functor test.
|
|
%
|
|
% We have four methods we can use for generating the code for the
|
|
% switches on both primary and secondary tags.
|
|
%
|
|
% 1. try-me-else chains have the form
|
|
%
|
|
% if (tag(var) != tag1) goto L1
|
|
% code for tag1
|
|
% goto end
|
|
% L1: if (tag(var) != tag2) goto L2
|
|
% code for tag2
|
|
% goto end
|
|
% L2: ...
|
|
% Ln: code for last possible tag value (or failure)
|
|
% goto end
|
|
%
|
|
% 2. try chains have the form
|
|
%
|
|
% if (tag(var) == tag1) goto L1
|
|
% if (tag(var) == tag2) goto L2
|
|
% ...
|
|
% code for last possible tag value (or failure)
|
|
% goto end
|
|
% L1: code for tag1
|
|
% goto end
|
|
% L2: code for tag2
|
|
% goto end
|
|
% ...
|
|
%
|
|
% 3. jump tables have the form
|
|
%
|
|
% goto tag(var) of L1, L2, ...
|
|
% L1: code for tag1
|
|
% goto end
|
|
% L2: code for tag2
|
|
% goto end
|
|
% ...
|
|
%
|
|
% 4. binary search switches have the form
|
|
%
|
|
% if (tag(var)) > 1) goto L23
|
|
% if (tag(var)) != 0) goto L1
|
|
% code for tag 0
|
|
% goto end
|
|
% L1: code for tag 1
|
|
% goto end
|
|
% L23: if (tag(var)) != 2) goto L3
|
|
% code for tag 2
|
|
% goto end
|
|
% L3: code for tag 3
|
|
% goto end
|
|
%
|
|
% Note that for a det switch with two tag values, try-me-else chains
|
|
% and try chains are equivalent.
|
|
%
|
|
% Which method is best depends
|
|
% - on the number of possible tag values,
|
|
% - on the costs of taken/untaken branches and table lookups on the given
|
|
% architecture, and
|
|
% - on the frequency with which the various alternatives are taken.
|
|
%
|
|
% While the first two are in principle known at compile time, the third
|
|
% is not (at least not without feedback from a profiler). Nevertheless,
|
|
% for switches on primary tags we can use the heuristic that the more
|
|
% secondary tags assigned to a primary tag, the more likely that the
|
|
% switch variable will have that primary tag at runtime.
|
|
%
|
|
% Try chains are good for switches with small numbers of alternatives
|
|
% on architectures where untaken branches are cheaper than taken
|
|
% branches.
|
|
%
|
|
% Try-me-else chains are good for switches with very small numbers of
|
|
% alternatives on architectures where taken branches are cheaper than
|
|
% untaken branches (which are rare these days).
|
|
%
|
|
% Jump tables are good for switches with large numbers of alternatives.
|
|
% The cost of jumping through a jump table is relatively high, since
|
|
% it involves a memory access and an indirect branch (which most
|
|
% current architectures do not handle well), but this cost is
|
|
% independent of the number of alternatives.
|
|
%
|
|
% Binary search switches are good for switches where the number of
|
|
% alternatives is large enough for the reduced expected number of
|
|
% branches executed to overcome the extra overhead of the subtraction
|
|
% required for some conditional branches (compared to try chains
|
|
% and try-me-else chains), but not large enough to make the
|
|
% expected cost of the expected number of comparisons exceed the
|
|
% expected cost of a jump table lookup and dispatch.
|
|
|
|
% For try-me-else chains, we want tag1 to be the most frequent case,
|
|
% tag2 the next most frequent case, etc.
|
|
%
|
|
% For det try chains, we want the last tag value to be the most
|
|
% frequent case, since it can be reached without taken jumps.
|
|
% We want tag1 to be the next most frequent, tag2 the next most
|
|
% frequent after that, etc.
|
|
%
|
|
% For semidet try chains, there is no last possible tag value (the
|
|
% code for failure occupies its position), so we want tag1 to be
|
|
% the most frequent case, tag 2 the next most frequent case, etc.
|
|
%
|
|
% For jump tables, the position of the labels in the computed goto
|
|
% must conform to their numerical value. The order of the code
|
|
% fragments does not really matter, although the last has a slight
|
|
% edge in that no goto is needed to reach the code following the
|
|
% switch. If there is no code following the switch (which happens
|
|
% very frequently), then even this advantage is nullified.
|
|
%
|
|
% For binary search switches, we want the case of the most frequently
|
|
% occurring tag to be the first, since this code is reached with no
|
|
% taken branches and ends with an unconditional branch, whereas
|
|
% reaching the code of the other cases requires at least one taken
|
|
% *conditional* branch. In general, at each binary decision we
|
|
% want the more frequently reached cases to be in the half that
|
|
% immediately follows the if statement implementing the decision.
|
|
|
|
:- type switch_method
|
|
---> try_me_else_chain
|
|
; try_chain
|
|
; jump_table
|
|
; binary_search.
|
|
|
|
%-----------------------------------------------------------------------------%
|
|
|
|
generate_tag_switch(TaggedCases, VarRval, VarType, VarName, CodeModel, CanFail,
|
|
SwitchGoalInfo, EndLabel, !MaybeEnd, Code, !CI, CLD0) :-
|
|
% We get registers for holding the primary and (if needed) the secondary
|
|
% tag. The tags are needed only by the switch, and no other code gets
|
|
% control between producing the tag values and all their uses, so
|
|
% we can immediately release the registers for use by the code of
|
|
% the various cases.
|
|
%
|
|
% We need to get and release the registers before we generate the code
|
|
% of the switch arms, since the set of free registers will in general be
|
|
% different before and after that action.
|
|
%
|
|
% We forgo using the primary tag register if the primary tag is needed
|
|
% only once, or if the "register" we get is likely to be slower than
|
|
% recomputing the tag from scratch.
|
|
some [!CLD] (
|
|
!:CLD = CLD0,
|
|
acquire_reg(reg_r, PtagReg, !CLD),
|
|
acquire_reg(reg_r, StagReg, !CLD),
|
|
release_reg(PtagReg, !CLD),
|
|
release_reg(StagReg, !CLD),
|
|
remember_position(!.CLD, BranchStart)
|
|
),
|
|
|
|
% Group the cases based on primary tag value and find out how many
|
|
% constructors share each primary tag value.
|
|
get_module_info(!.CI, ModuleInfo),
|
|
get_ptag_counts(VarType, ModuleInfo, MaxPrimary, PtagCountMap),
|
|
Params = represent_params(VarName, SwitchGoalInfo, CodeModel, BranchStart,
|
|
EndLabel),
|
|
group_cases_by_ptag(TaggedCases, represent_tagged_case_for_llds(Params),
|
|
map.init, CaseLabelMap0, !MaybeEnd, !CI, _, PtagCaseMap),
|
|
|
|
map.count(PtagCaseMap, PtagsUsed),
|
|
get_globals(!.CI, Globals),
|
|
globals.lookup_int_option(Globals, dense_switch_size, DenseSwitchSize),
|
|
globals.lookup_int_option(Globals, try_switch_size, TrySwitchSize),
|
|
globals.lookup_int_option(Globals, binary_switch_size, BinarySwitchSize),
|
|
( if PtagsUsed >= DenseSwitchSize then
|
|
PrimaryMethod = jump_table
|
|
else if PtagsUsed >= BinarySwitchSize then
|
|
PrimaryMethod = binary_search
|
|
else if PtagsUsed >= TrySwitchSize then
|
|
PrimaryMethod = try_chain
|
|
else
|
|
PrimaryMethod = try_me_else_chain
|
|
),
|
|
|
|
( if
|
|
PrimaryMethod \= jump_table,
|
|
PtagsUsed >= 2,
|
|
globals.lookup_int_option(Globals, num_real_r_regs, NumRealRegs),
|
|
(
|
|
NumRealRegs = 0
|
|
;
|
|
( if PtagReg = reg(reg_r, PtagRegNo) then
|
|
PtagRegNo =< NumRealRegs
|
|
else
|
|
unexpected($pred, "improper reg in tag switch")
|
|
)
|
|
)
|
|
then
|
|
PtagCode = singleton(
|
|
llds_instr(assign(PtagReg, unop(tag, VarRval)),
|
|
"compute tag to switch on")
|
|
),
|
|
PtagRval = lval(PtagReg)
|
|
else
|
|
PtagCode = empty,
|
|
PtagRval = unop(tag, VarRval)
|
|
),
|
|
|
|
% We generate EndCode (and if needed, FailCode) here because
|
|
% the last case within a primary tag may not be the last case overall.
|
|
EndCode = singleton(
|
|
llds_instr(label(EndLabel), "end of tag switch")
|
|
),
|
|
(
|
|
CanFail = cannot_fail,
|
|
MaybeFailLabel = no,
|
|
FailCode = empty
|
|
;
|
|
CanFail = can_fail,
|
|
get_next_label(FailLabel, !CI),
|
|
MaybeFailLabel = yes(FailLabel),
|
|
FailLabelCode = singleton(
|
|
llds_instr(label(FailLabel), "switch has failed")
|
|
),
|
|
% We must generate the failure code in the context in which
|
|
% none of the switch arms have been executed yet.
|
|
some [!CLD] (
|
|
reset_to_position(BranchStart, !.CI, !:CLD),
|
|
generate_failure(FailureCode, !CI, !.CLD)
|
|
),
|
|
FailCode = FailLabelCode ++ FailureCode
|
|
),
|
|
|
|
(
|
|
PrimaryMethod = binary_search,
|
|
order_ptags_by_value(ptag(0u8), ptag(MaxPrimary),
|
|
PtagCaseMap, PtagCaseList),
|
|
generate_primary_binary_search(PtagCaseList, 0u8, MaxPrimary, PtagRval,
|
|
StagReg, VarRval, MaybeFailLabel, PtagCountMap, CasesCode,
|
|
CaseLabelMap0, CaseLabelMap, !CI)
|
|
;
|
|
PrimaryMethod = jump_table,
|
|
order_ptags_by_value(ptag(0u8), ptag(MaxPrimary),
|
|
PtagCaseMap, PtagCaseList),
|
|
generate_primary_jump_table(PtagCaseList, 0u8, MaxPrimary, StagReg,
|
|
VarRval, MaybeFailLabel, PtagCountMap, Targets, TableCode,
|
|
CaseLabelMap0, CaseLabelMap, !CI),
|
|
SwitchCode = singleton(
|
|
llds_instr(computed_goto(PtagRval, Targets),
|
|
"switch on primary tag")
|
|
),
|
|
CasesCode = SwitchCode ++ TableCode
|
|
;
|
|
PrimaryMethod = try_chain,
|
|
order_ptags_by_count(PtagCountMap, PtagCaseMap, PtagCaseList0),
|
|
( if
|
|
CanFail = cannot_fail,
|
|
PtagCaseList0 = [MostFreqCase | OtherCases]
|
|
then
|
|
PtagCaseList = OtherCases ++ [MostFreqCase]
|
|
else
|
|
PtagCaseList = PtagCaseList0
|
|
),
|
|
generate_primary_try_chain(PtagCaseList, PtagRval, StagReg, VarRval,
|
|
MaybeFailLabel, PtagCountMap, empty, empty, CasesCode,
|
|
CaseLabelMap0, CaseLabelMap, !CI)
|
|
;
|
|
PrimaryMethod = try_me_else_chain,
|
|
order_ptags_by_count(PtagCountMap, PtagCaseMap, PtagCaseList),
|
|
generate_primary_try_me_else_chain(PtagCaseList, PtagRval, StagReg,
|
|
VarRval, MaybeFailLabel, PtagCountMap, CasesCode,
|
|
CaseLabelMap0, CaseLabelMap, !CI)
|
|
),
|
|
map.foldl(add_remaining_case, CaseLabelMap, empty, RemainingCasesCode),
|
|
Code = PtagCode ++ CasesCode ++ RemainingCasesCode ++ FailCode ++ EndCode.
|
|
|
|
%-----------------------------------------------------------------------------%
|
|
|
|
% Generate a switch on a primary tag value using a try-me-else chain.
|
|
%
|
|
:- pred generate_primary_try_me_else_chain(ptag_case_group_list(label)::in,
|
|
rval::in, lval::in, rval::in, maybe(label)::in,
|
|
ptag_count_map::in, llds_code::out,
|
|
case_label_map::in, case_label_map::out,
|
|
code_info::in, code_info::out) is det.
|
|
|
|
generate_primary_try_me_else_chain([], _, _, _, _, _, _,
|
|
!CaseLabelMap, !CI) :-
|
|
unexpected($pred, "empty switch").
|
|
generate_primary_try_me_else_chain([PtagGroup | PtagGroups], PtagRval, StagReg,
|
|
VarRval, MaybeFailLabel, PtagCountMap, Code, !CaseLabelMap, !CI) :-
|
|
PtagGroup = ptag_case_group_entry(MainPtag, OtherPtags, PtagCase),
|
|
PtagCase = ptag_case(StagLoc, StagGoalMap),
|
|
map.lookup(PtagCountMap, MainPtag, CountInfo),
|
|
CountInfo = StagLocPrime - MaxSecondary,
|
|
expect(unify(StagLoc, StagLocPrime), $pred,
|
|
"secondary tag locations differ"),
|
|
(
|
|
PtagGroups = [_ | _],
|
|
generate_primary_try_me_else_chain_case(PtagRval, StagReg,
|
|
MainPtag, OtherPtags, PtagCase, MaxSecondary, VarRval,
|
|
MaybeFailLabel, ThisTagCode, !CaseLabelMap, !CI),
|
|
generate_primary_try_me_else_chain(PtagGroups, PtagRval, StagReg,
|
|
VarRval, MaybeFailLabel, PtagCountMap, OtherTagsCode,
|
|
!CaseLabelMap, !CI),
|
|
Code = ThisTagCode ++ OtherTagsCode
|
|
;
|
|
PtagGroups = [],
|
|
(
|
|
MaybeFailLabel = yes(FailLabel),
|
|
generate_primary_try_me_else_chain_case(PtagRval, StagReg,
|
|
MainPtag, OtherPtags, PtagCase, MaxSecondary, VarRval,
|
|
MaybeFailLabel, ThisTagCode, !CaseLabelMap, !CI),
|
|
% FailLabel ought to be the next label anyway, so this goto
|
|
% will be optimized away (unless the layout of the failcode
|
|
% in the caller changes).
|
|
FailCode = singleton(
|
|
llds_instr(goto(code_label(FailLabel)),
|
|
"primary tag with no code to handle it")
|
|
),
|
|
Code = ThisTagCode ++ FailCode
|
|
;
|
|
MaybeFailLabel = no,
|
|
generate_primary_tag_code(StagGoalMap, MainPtag, OtherPtags,
|
|
MaxSecondary, StagReg, StagLoc, VarRval, MaybeFailLabel, Code,
|
|
!CaseLabelMap, !CI)
|
|
)
|
|
).
|
|
|
|
:- pred generate_primary_try_me_else_chain_case(rval::in, lval::in,
|
|
ptag::in, list(ptag)::in, ptag_case(label)::in, int::in, rval::in,
|
|
maybe(label)::in, llds_code::out, case_label_map::in, case_label_map::out,
|
|
code_info::in, code_info::out) is det.
|
|
|
|
generate_primary_try_me_else_chain_case(PtagRval, StagReg,
|
|
MainPtag, OtherPtags, PtagCase, MaxSecondary, VarRval, MaybeFailLabel,
|
|
Code, !CaseLabelMap, !CI) :-
|
|
get_next_label(ElseLabel, !CI),
|
|
MainPtag = ptag(MainPtagUint8),
|
|
TestRval0 = binop(ne(int_type_int), PtagRval,
|
|
const(llconst_int(uint8.cast_to_int(MainPtagUint8)))),
|
|
generate_primary_try_me_else_chain_other_ptags(OtherPtags, PtagRval,
|
|
TestRval0, TestRval),
|
|
TestCode = singleton(
|
|
llds_instr(if_val(TestRval, code_label(ElseLabel)),
|
|
"test primary tag only")
|
|
),
|
|
PtagCase = ptag_case(StagLoc, StagGoalMap),
|
|
generate_primary_tag_code(StagGoalMap, MainPtag, OtherPtags, MaxSecondary,
|
|
StagReg, StagLoc, VarRval, MaybeFailLabel, TagCode,
|
|
!CaseLabelMap, !CI),
|
|
ElseCode = singleton(
|
|
llds_instr(label(ElseLabel), "handle next primary tag")
|
|
),
|
|
Code = TestCode ++ TagCode ++ ElseCode.
|
|
|
|
:- pred generate_primary_try_me_else_chain_other_ptags(list(ptag)::in,
|
|
rval::in, rval::in, rval::out) is det.
|
|
|
|
generate_primary_try_me_else_chain_other_ptags([], _, TestRval, TestRval).
|
|
generate_primary_try_me_else_chain_other_ptags([OtherPtag | OtherPtags],
|
|
PtagRval, TestRval0, TestRval) :-
|
|
OtherPtag = ptag(OtherPtagUint8),
|
|
ThisTestRval = binop(ne(int_type_int), PtagRval,
|
|
const(llconst_int(uint8.cast_to_int(OtherPtagUint8)))),
|
|
TestRval1 = binop(logical_and, TestRval0, ThisTestRval),
|
|
generate_primary_try_me_else_chain_other_ptags(OtherPtags,
|
|
PtagRval, TestRval1, TestRval).
|
|
|
|
%-----------------------------------------------------------------------------%
|
|
|
|
% Generate a switch on a primary tag value using a try chain.
|
|
%
|
|
:- pred generate_primary_try_chain(ptag_case_group_list(label)::in,
|
|
rval::in, lval::in, rval::in, maybe(label)::in,
|
|
ptag_count_map::in, llds_code::in, llds_code::in, llds_code::out,
|
|
case_label_map::in, case_label_map::out,
|
|
code_info::in, code_info::out) is det.
|
|
|
|
generate_primary_try_chain([], _, _, _, _, _, _, _, _, !CaseLabelMap, !CI) :-
|
|
unexpected($pred, "empty list").
|
|
generate_primary_try_chain([PtagGroup | PtagGroups], PtagRval, StagReg,
|
|
VarRval, MaybeFailLabel, PtagCountMap, PrevTestsCode0, PrevCasesCode0,
|
|
Code, !CaseLabelMap, !CI) :-
|
|
PtagGroup = ptag_case_group_entry(MainPtag, OtherPtags, PtagCase),
|
|
PtagCase = ptag_case(StagLoc, StagGoalMap),
|
|
map.lookup(PtagCountMap, MainPtag, CountInfo),
|
|
CountInfo = StagLocPrime - MaxSecondary,
|
|
expect(unify(StagLoc, StagLocPrime), $pred,
|
|
"secondary tag locations differ"),
|
|
(
|
|
PtagGroups = [_ | _],
|
|
generate_primary_try_chain_case(PtagRval, StagReg,
|
|
MainPtag, OtherPtags, PtagCase, MaxSecondary, VarRval,
|
|
MaybeFailLabel,
|
|
PrevTestsCode0, PrevTestsCode1, PrevCasesCode0, PrevCasesCode1,
|
|
!CaseLabelMap, !CI),
|
|
generate_primary_try_chain(PtagGroups, PtagRval, StagReg, VarRval,
|
|
MaybeFailLabel, PtagCountMap, PrevTestsCode1, PrevCasesCode1,
|
|
Code, !CaseLabelMap, !CI)
|
|
;
|
|
PtagGroups = [],
|
|
(
|
|
MaybeFailLabel = yes(FailLabel),
|
|
generate_primary_try_chain_case(PtagRval, StagReg,
|
|
MainPtag, OtherPtags, PtagCase, MaxSecondary,
|
|
VarRval, MaybeFailLabel,
|
|
PrevTestsCode0, PrevTestsCode1, PrevCasesCode0, PrevCasesCode1,
|
|
!CaseLabelMap, !CI),
|
|
FailCode = singleton(
|
|
llds_instr(goto(code_label(FailLabel)),
|
|
"primary tag with no code to handle it")
|
|
),
|
|
Code = PrevTestsCode1 ++ FailCode ++ PrevCasesCode1
|
|
;
|
|
MaybeFailLabel = no,
|
|
make_ptag_comment("fallthrough to last primary tag value: ",
|
|
MainPtag, OtherPtags, Comment),
|
|
CommentCode = singleton(
|
|
llds_instr(comment(Comment), "")
|
|
),
|
|
generate_primary_tag_code(StagGoalMap, MainPtag, OtherPtags,
|
|
MaxSecondary, StagReg, StagLoc, VarRval, MaybeFailLabel,
|
|
TagCode, !CaseLabelMap, !CI),
|
|
Code = PrevTestsCode0 ++ CommentCode ++ TagCode ++ PrevCasesCode0
|
|
)
|
|
).
|
|
|
|
:- pred generate_primary_try_chain_case(rval::in, lval::in,
|
|
ptag::in, list(ptag)::in,
|
|
ptag_case(label)::in, int::in, rval::in, maybe(label)::in,
|
|
llds_code::in, llds_code::out, llds_code::in, llds_code::out,
|
|
case_label_map::in, case_label_map::out,
|
|
code_info::in, code_info::out) is det.
|
|
|
|
generate_primary_try_chain_case(PtagRval, StagReg, MainPtag, OtherPtags,
|
|
PtagCase, MaxSecondary, VarRval, MaybeFailLabel,
|
|
PrevTestsCode0, PrevTestsCode, PrevCasesCode0, PrevCasesCode,
|
|
!CaseLabelMap, !CI) :-
|
|
get_next_label(ThisPtagLabel, !CI),
|
|
MainPtag = ptag(MainPtagUint8),
|
|
TestRval0 = binop(eq(int_type_int), PtagRval,
|
|
const(llconst_int(uint8.cast_to_int(MainPtagUint8)))),
|
|
generate_primary_try_chain_other_ptags(OtherPtags, PtagRval,
|
|
TestRval0, TestRval),
|
|
TestCode = singleton(
|
|
llds_instr(if_val(TestRval, code_label(ThisPtagLabel)),
|
|
"test primary tag only")
|
|
),
|
|
make_ptag_comment("primary tag value: ", MainPtag, OtherPtags, Comment),
|
|
LabelCode = singleton(
|
|
llds_instr(label(ThisPtagLabel), Comment)
|
|
),
|
|
PtagCase = ptag_case(StagLoc, StagGoalMap),
|
|
generate_primary_tag_code(StagGoalMap, MainPtag, OtherPtags, MaxSecondary,
|
|
StagReg, StagLoc, VarRval, MaybeFailLabel, TagCode,
|
|
!CaseLabelMap, !CI),
|
|
PrevTestsCode = PrevTestsCode0 ++ TestCode,
|
|
PrevCasesCode = LabelCode ++ TagCode ++ PrevCasesCode0.
|
|
|
|
:- pred generate_primary_try_chain_other_ptags(list(ptag)::in,
|
|
rval::in, rval::in, rval::out) is det.
|
|
|
|
generate_primary_try_chain_other_ptags([], _, TestRval, TestRval).
|
|
generate_primary_try_chain_other_ptags([OtherPtag | OtherPtags],
|
|
PtagRval, TestRval0, TestRval) :-
|
|
OtherPtag = ptag(OtherPtagUint8),
|
|
ThisTestRval = binop(eq(int_type_int), PtagRval,
|
|
const(llconst_int(uint8.cast_to_int(OtherPtagUint8)))),
|
|
TestRval1 = binop(logical_or, TestRval0, ThisTestRval),
|
|
generate_primary_try_chain_other_ptags(OtherPtags,
|
|
PtagRval, TestRval1, TestRval).
|
|
|
|
%-----------------------------------------------------------------------------%
|
|
|
|
% Generate the cases for a primary tag using a dense jump table
|
|
% that has an entry for all possible primary tag values.
|
|
%
|
|
:- pred generate_primary_jump_table(ptag_case_list(label)::in,
|
|
uint8::in, uint8::in, lval::in, rval::in,
|
|
maybe(label)::in, ptag_count_map::in,
|
|
list(maybe(label))::out, llds_code::out,
|
|
case_label_map::in, case_label_map::out,
|
|
code_info::in, code_info::out) is det.
|
|
|
|
generate_primary_jump_table(PtagGroups, CurPrimary, MaxPrimary, StagReg,
|
|
VarRval, MaybeFailLabel, PtagCountMap, Targets, Code,
|
|
!CaseLabelMap, !CI) :-
|
|
( if CurPrimary > MaxPrimary then
|
|
expect(unify(PtagGroups, []), $pred,
|
|
"PtagGroups != [] when Cur > Max"),
|
|
Targets = [],
|
|
Code = empty
|
|
else
|
|
NextPrimary = CurPrimary + 1u8,
|
|
( if
|
|
PtagGroups = [PtagCaseEntry | PtagGroupsTail],
|
|
PtagCaseEntry = ptag_case_entry(ptag(CurPrimary), PrimaryInfo)
|
|
then
|
|
PrimaryInfo = ptag_case(StagLoc, StagGoalMap),
|
|
map.lookup(PtagCountMap, ptag(CurPrimary), CountInfo),
|
|
CountInfo = StagLocPrime - MaxSecondary,
|
|
expect(unify(StagLoc, StagLocPrime), $pred,
|
|
"secondary tag locations differ"),
|
|
get_next_label(NewLabel, !CI),
|
|
Comment = "start of a case in primary tag switch: ptag " ++
|
|
string.uint8_to_string(CurPrimary),
|
|
LabelCode = singleton(llds_instr(label(NewLabel), Comment)),
|
|
generate_primary_tag_code(StagGoalMap, ptag(CurPrimary), [],
|
|
MaxSecondary, StagReg, StagLoc, VarRval, MaybeFailLabel,
|
|
ThisTagCode, !CaseLabelMap, !CI),
|
|
generate_primary_jump_table(PtagGroupsTail, NextPrimary,
|
|
MaxPrimary, StagReg, VarRval, MaybeFailLabel, PtagCountMap,
|
|
TailTargets, TailCode, !CaseLabelMap, !CI),
|
|
Targets = [yes(NewLabel) | TailTargets],
|
|
Code = LabelCode ++ ThisTagCode ++ TailCode
|
|
else
|
|
generate_primary_jump_table(PtagGroups, NextPrimary, MaxPrimary,
|
|
StagReg, VarRval, MaybeFailLabel, PtagCountMap,
|
|
TailTargets, TailCode, !CaseLabelMap, !CI),
|
|
Targets = [MaybeFailLabel | TailTargets],
|
|
Code = TailCode
|
|
)
|
|
).
|
|
|
|
%-----------------------------------------------------------------------------%
|
|
|
|
% Generate the cases for a primary tag using a binary search.
|
|
% This invocation looks after primary tag values in the range
|
|
% MinPtag to MaxPtag (including both boundary values).
|
|
%
|
|
:- pred generate_primary_binary_search(ptag_case_list(label)::in,
|
|
uint8::in, uint8::in, rval::in, lval::in, rval::in, maybe(label)::in,
|
|
ptag_count_map::in, llds_code::out,
|
|
case_label_map::in, case_label_map::out,
|
|
code_info::in, code_info::out) is det.
|
|
|
|
generate_primary_binary_search(PtagGroups, MinPtag, MaxPtag, PtagRval, StagReg,
|
|
VarRval, MaybeFailLabel, PtagCountMap, Code, !CaseLabelMap, !CI) :-
|
|
( if MinPtag = MaxPtag then
|
|
CurPrimary = MinPtag,
|
|
(
|
|
PtagGroups = [],
|
|
% There is no code for this tag.
|
|
(
|
|
MaybeFailLabel = yes(FailLabel),
|
|
PtagStr = string.uint8_to_string(CurPrimary),
|
|
Comment = "no code for ptag " ++ PtagStr,
|
|
Code = singleton(
|
|
llds_instr(goto(code_label(FailLabel)), Comment)
|
|
)
|
|
;
|
|
MaybeFailLabel = no,
|
|
% The switch is cannot_fail, which means this case cannot
|
|
% happen.
|
|
Code = empty
|
|
)
|
|
;
|
|
PtagGroups = [ptag_case_entry(ptag(CurPrimaryPrime), PrimaryInfo)],
|
|
expect(unify(CurPrimary, CurPrimaryPrime), $pred,
|
|
"cur_primary mismatch"),
|
|
PrimaryInfo = ptag_case(StagLoc, StagGoalMap),
|
|
map.lookup(PtagCountMap, ptag(CurPrimary), CountInfo),
|
|
CountInfo = StagLocPrime - MaxSecondary,
|
|
expect(unify(StagLoc, StagLocPrime), $pred,
|
|
"secondary tag locations differ"),
|
|
generate_primary_tag_code(StagGoalMap, ptag(CurPrimary), [],
|
|
MaxSecondary, StagReg, StagLoc, VarRval, MaybeFailLabel, Code,
|
|
!CaseLabelMap, !CI)
|
|
;
|
|
PtagGroups = [_, _ | _],
|
|
unexpected($pred,
|
|
"caselist not singleton or empty when binary search ends")
|
|
)
|
|
else
|
|
LowRangeEnd = (MinPtag + MaxPtag) // 2u8,
|
|
HighRangeStart = LowRangeEnd + 1u8,
|
|
InLowGroup =
|
|
( pred(PtagGroup::in) is semidet :-
|
|
PtagGroup = ptag_case_entry(ptag(Ptag), _),
|
|
Ptag =< LowRangeEnd
|
|
),
|
|
list.filter(InLowGroup, PtagGroups, LowGroups, HighGroups),
|
|
get_next_label(NewLabel, !CI),
|
|
string.uint8_to_string(MinPtag) = LowStartStr,
|
|
string.uint8_to_string(LowRangeEnd) = LowEndStr,
|
|
string.uint8_to_string(HighRangeStart) = HighStartStr,
|
|
string.uint8_to_string(MaxPtag) = HighEndStr,
|
|
IfComment = "fallthrough for ptags " ++
|
|
LowStartStr ++ " to " ++ LowEndStr,
|
|
LabelComment = "code for ptags " ++
|
|
HighStartStr ++ " to " ++ HighEndStr,
|
|
% XXX ARG_PACK We should do the comparison on uint8s, not ints.
|
|
LowRangeEndConst = const(llconst_int(uint8.cast_to_int(LowRangeEnd))),
|
|
TestRval = binop(int_gt(int_type_int), PtagRval, LowRangeEndConst),
|
|
IfCode = singleton(
|
|
llds_instr(if_val(TestRval, code_label(NewLabel)), IfComment)
|
|
),
|
|
LabelCode = singleton(
|
|
llds_instr(label(NewLabel), LabelComment)
|
|
),
|
|
|
|
generate_primary_binary_search(LowGroups, MinPtag, LowRangeEnd,
|
|
PtagRval, StagReg, VarRval, MaybeFailLabel, PtagCountMap,
|
|
LowRangeCode, !CaseLabelMap, !CI),
|
|
generate_primary_binary_search(HighGroups, HighRangeStart, MaxPtag,
|
|
PtagRval, StagReg, VarRval, MaybeFailLabel, PtagCountMap,
|
|
HighRangeCode, !CaseLabelMap, !CI),
|
|
Code = IfCode ++ LowRangeCode ++ LabelCode ++ HighRangeCode
|
|
).
|
|
|
|
%-----------------------------------------------------------------------------%
|
|
|
|
% Generate the code corresponding to a primary tag.
|
|
% If this primary tag has secondary tags, decide whether we should
|
|
% use a jump table to implement the secondary switch.
|
|
%
|
|
:- pred generate_primary_tag_code(stag_goal_map(label)::in,
|
|
ptag::in, list(ptag)::in, int::in, lval::in, sectag_locn::in,
|
|
rval::in, maybe(label)::in, llds_code::out,
|
|
case_label_map::in, case_label_map::out,
|
|
code_info::in, code_info::out) is det.
|
|
|
|
generate_primary_tag_code(StagGoalMap, MainPtag, OtherPtags, MaxSecondary,
|
|
StagReg, StagLoc, Rval, MaybeFailLabel, Code, !CaseLabelMap, !CI) :-
|
|
map.to_assoc_list(StagGoalMap, StagGoalList),
|
|
(
|
|
( StagLoc = sectag_none
|
|
; StagLoc = sectag_none_direct_arg
|
|
),
|
|
% There is no secondary tag, so there is no switch on it.
|
|
(
|
|
StagGoalList = [],
|
|
unexpected($pred, "no goal for non-shared tag")
|
|
;
|
|
StagGoalList = [StagGoal],
|
|
( if StagGoal = -1 - CaseLabel then
|
|
generate_case_code_or_jump(CaseLabel, Code, !CaseLabelMap)
|
|
else
|
|
unexpected($pred, "badly formed goal for non-shared tag")
|
|
)
|
|
;
|
|
StagGoalList = [_, _ | _],
|
|
unexpected($pred, "more than one goal for non-shared tag")
|
|
)
|
|
;
|
|
(
|
|
StagLoc = sectag_remote_word,
|
|
OrigStagRval = lval(field(yes(MainPtag), Rval,
|
|
const(llconst_int(0)))),
|
|
Comment = "compute remote word sec tag to switch on"
|
|
;
|
|
StagLoc = sectag_remote_bits(_NumBits, Mask),
|
|
StagWordRval = lval(field(yes(MainPtag), Rval,
|
|
const(llconst_int(0)))),
|
|
OrigStagRval = binop(bitwise_and(int_type_uint),
|
|
StagWordRval, const(llconst_uint(Mask))),
|
|
Comment = "compute remote sec tag bits to switch on"
|
|
;
|
|
StagLoc = sectag_local_rest_of_word,
|
|
OrigStagRval = unop(unmkbody, Rval),
|
|
Comment = "compute local rest-of-word sec tag to switch on"
|
|
;
|
|
StagLoc = sectag_local_bits(_NumBits, Mask),
|
|
OrigStagRval = binop(bitwise_and(int_type_uint),
|
|
unop(unmkbody, Rval), const(llconst_uint(Mask))),
|
|
Comment = "compute local sec tag bits to switch on"
|
|
),
|
|
expect(unify(OtherPtags, []), $pred, ">1 ptag with secondary tag"),
|
|
|
|
% There is a secondary tag, so figure out how to switch on it.
|
|
get_globals(!.CI, Globals),
|
|
globals.lookup_int_option(Globals, dense_switch_size,
|
|
DenseSwitchSize),
|
|
globals.lookup_int_option(Globals, binary_switch_size,
|
|
BinarySwitchSize),
|
|
globals.lookup_int_option(Globals, try_switch_size, TrySwitchSize),
|
|
( if MaxSecondary >= DenseSwitchSize then
|
|
SecondaryMethod = jump_table
|
|
else if MaxSecondary >= BinarySwitchSize then
|
|
SecondaryMethod = binary_search
|
|
else if MaxSecondary >= TrySwitchSize then
|
|
SecondaryMethod = try_chain
|
|
else
|
|
SecondaryMethod = try_me_else_chain
|
|
),
|
|
|
|
( if
|
|
SecondaryMethod \= jump_table,
|
|
MaxSecondary >= 2,
|
|
globals.lookup_int_option(Globals, num_real_r_regs, NumRealRegs),
|
|
(
|
|
NumRealRegs = 0
|
|
;
|
|
( if StagReg = reg(reg_r, StagRegNo) then
|
|
StagRegNo =< NumRealRegs
|
|
else
|
|
unexpected($pred, "improper reg in tag switch")
|
|
)
|
|
)
|
|
then
|
|
StagCode = singleton(
|
|
llds_instr(assign(StagReg, OrigStagRval), Comment)
|
|
),
|
|
StagRval = lval(StagReg)
|
|
else
|
|
StagCode = empty,
|
|
StagRval = OrigStagRval
|
|
),
|
|
(
|
|
MaybeFailLabel = yes(FailLabel),
|
|
( if
|
|
list.length(StagGoalList, StagGoalCount),
|
|
FullGoalCount = MaxSecondary + 1,
|
|
FullGoalCount = StagGoalCount
|
|
then
|
|
MaybeSecFailLabel = no
|
|
else
|
|
MaybeSecFailLabel = yes(FailLabel)
|
|
)
|
|
;
|
|
MaybeFailLabel = no,
|
|
MaybeSecFailLabel = no
|
|
),
|
|
|
|
(
|
|
SecondaryMethod = jump_table,
|
|
generate_secondary_jump_table(StagGoalList, 0, MaxSecondary,
|
|
MaybeSecFailLabel, Targets),
|
|
Code = singleton(
|
|
llds_instr(computed_goto(StagRval, Targets),
|
|
"switch on secondary tag")
|
|
)
|
|
;
|
|
SecondaryMethod = binary_search,
|
|
generate_secondary_binary_search(StagGoalList, 0, MaxSecondary,
|
|
StagRval, MaybeSecFailLabel, Code, !CaseLabelMap, !CI)
|
|
;
|
|
SecondaryMethod = try_chain,
|
|
generate_secondary_try_chain(StagGoalList, StagRval,
|
|
MaybeSecFailLabel, empty, Codes, !CaseLabelMap),
|
|
Code = StagCode ++ Codes
|
|
;
|
|
SecondaryMethod = try_me_else_chain,
|
|
generate_secondary_try_me_else_chain(StagGoalList, StagRval,
|
|
MaybeSecFailLabel, Codes, !CaseLabelMap, !CI),
|
|
Code = StagCode ++ Codes
|
|
)
|
|
).
|
|
|
|
%-----------------------------------------------------------------------------%
|
|
|
|
% Generate a switch on a secondary tag value using a try-me-else chain.
|
|
%
|
|
:- pred generate_secondary_try_me_else_chain(stag_goal_list(label)::in,
|
|
rval::in, maybe(label)::in, llds_code::out,
|
|
case_label_map::in, case_label_map::out,
|
|
code_info::in, code_info::out) is det.
|
|
|
|
generate_secondary_try_me_else_chain([], _, _, _, !CaseLabelMap, !CI) :-
|
|
unexpected($pred, "empty switch").
|
|
generate_secondary_try_me_else_chain([Case | Cases], StagRval,
|
|
MaybeFailLabel, Code, !CaseLabelMap, !CI) :-
|
|
Case = Secondary - CaseLabel,
|
|
(
|
|
Cases = [_ | _],
|
|
generate_secondary_try_me_else_chain_case(CaseLabel, StagRval,
|
|
Secondary, ThisCode, !CaseLabelMap, !CI),
|
|
generate_secondary_try_me_else_chain(Cases, StagRval,
|
|
MaybeFailLabel, OtherCode, !CaseLabelMap, !CI),
|
|
Code = ThisCode ++ OtherCode
|
|
;
|
|
Cases = [],
|
|
(
|
|
MaybeFailLabel = yes(FailLabel),
|
|
generate_secondary_try_me_else_chain_case(CaseLabel, StagRval,
|
|
Secondary, ThisCode, !CaseLabelMap, !CI),
|
|
FailCode = singleton(
|
|
llds_instr(goto(code_label(FailLabel)),
|
|
"secondary tag does not match")
|
|
),
|
|
Code = ThisCode ++ FailCode
|
|
;
|
|
MaybeFailLabel = no,
|
|
generate_case_code_or_jump(CaseLabel, Code, !CaseLabelMap)
|
|
)
|
|
).
|
|
|
|
:- pred generate_secondary_try_me_else_chain_case(label::in, rval::in, int::in,
|
|
llds_code::out, case_label_map::in, case_label_map::out,
|
|
code_info::in, code_info::out) is det.
|
|
|
|
generate_secondary_try_me_else_chain_case(CaseLabel, StagRval, Secondary,
|
|
Code, !CaseLabelMap, !CI) :-
|
|
generate_case_code_or_jump(CaseLabel, CaseCode, !CaseLabelMap),
|
|
% XXX Optimize what we generate when CaseCode = goto(CaseLabel).
|
|
get_next_label(ElseLabel, !CI),
|
|
TestCode = singleton(
|
|
llds_instr(
|
|
if_val(binop(ne(int_type_int), StagRval,
|
|
const(llconst_int(Secondary))),
|
|
code_label(ElseLabel)),
|
|
"test sec tag only")
|
|
),
|
|
ElseLabelCode = singleton(
|
|
llds_instr(label(ElseLabel), "handle next secondary tag")
|
|
),
|
|
Code = TestCode ++ CaseCode ++ ElseLabelCode.
|
|
|
|
%-----------------------------------------------------------------------------%
|
|
|
|
% Generate a switch on a secondary tag value using a try chain.
|
|
%
|
|
:- pred generate_secondary_try_chain(stag_goal_list(label)::in, rval::in,
|
|
maybe(label)::in, llds_code::in, llds_code::out,
|
|
case_label_map::in, case_label_map::out) is det.
|
|
|
|
generate_secondary_try_chain([], _, _, _, _, !CaseLabelMap) :-
|
|
unexpected($pred, "empty switch").
|
|
generate_secondary_try_chain([Case | Cases], StagRval, MaybeFailLabel,
|
|
PrevTestsCode0, Code, !CaseLabelMap) :-
|
|
Case = Secondary - CaseLabel,
|
|
(
|
|
Cases = [_ | _],
|
|
generate_secondary_try_chain_case(CaseLabel, StagRval, Secondary,
|
|
PrevTestsCode0, PrevTestsCode1, !.CaseLabelMap),
|
|
generate_secondary_try_chain(Cases, StagRval,
|
|
MaybeFailLabel, PrevTestsCode1, Code, !CaseLabelMap)
|
|
;
|
|
Cases = [],
|
|
(
|
|
MaybeFailLabel = yes(FailLabel),
|
|
generate_secondary_try_chain_case(CaseLabel, StagRval, Secondary,
|
|
PrevTestsCode0, PrevTestsCode1, !.CaseLabelMap),
|
|
FailCode = singleton(
|
|
llds_instr(goto(code_label(FailLabel)),
|
|
"secondary tag with no code to handle it")
|
|
),
|
|
Code = PrevTestsCode1 ++ FailCode
|
|
;
|
|
MaybeFailLabel = no,
|
|
generate_case_code_or_jump(CaseLabel, ThisCode, !CaseLabelMap),
|
|
Code = PrevTestsCode0 ++ ThisCode
|
|
)
|
|
).
|
|
|
|
:- pred generate_secondary_try_chain_case(label::in, rval::in, int::in,
|
|
llds_code::in, llds_code::out, case_label_map::in) is det.
|
|
|
|
generate_secondary_try_chain_case(CaseLabel, StagRval, Secondary,
|
|
PrevTestsCode0, PrevTestsCode, CaseLabelMap) :-
|
|
map.lookup(CaseLabelMap, CaseLabel, CaseInfo0),
|
|
CaseInfo0 = case_label_info(Comment, _CaseCode, _CaseGenerated),
|
|
TestCode = singleton(
|
|
llds_instr(
|
|
if_val(binop(eq(int_type_int), StagRval,
|
|
const(llconst_int(Secondary))),
|
|
code_label(CaseLabel)),
|
|
"test sec tag only for " ++ Comment)
|
|
),
|
|
PrevTestsCode = PrevTestsCode0 ++ TestCode.
|
|
|
|
%-----------------------------------------------------------------------------%
|
|
|
|
% Generate the cases for a primary tag using a dense jump table
|
|
% that has an entry for all possible secondary tag values.
|
|
%
|
|
:- pred generate_secondary_jump_table(stag_goal_list(label)::in, int::in,
|
|
int::in, maybe(label)::in, list(maybe(label))::out) is det.
|
|
|
|
generate_secondary_jump_table(CaseList, CurSecondary, MaxSecondary,
|
|
MaybeFailLabel, Targets) :-
|
|
( if CurSecondary > MaxSecondary then
|
|
expect(unify(CaseList, []), $pred,
|
|
"caselist not empty when reaching limiting secondary tag"),
|
|
Targets = []
|
|
else
|
|
NextSecondary = CurSecondary + 1,
|
|
( if CaseList = [CurSecondary - CaseLabel | CaseListTail] then
|
|
generate_secondary_jump_table(CaseListTail, NextSecondary,
|
|
MaxSecondary, MaybeFailLabel, OtherTargets),
|
|
Targets = [yes(CaseLabel) | OtherTargets]
|
|
else
|
|
generate_secondary_jump_table(CaseList, NextSecondary,
|
|
MaxSecondary, MaybeFailLabel, OtherTargets),
|
|
Targets = [MaybeFailLabel | OtherTargets]
|
|
)
|
|
).
|
|
|
|
%-----------------------------------------------------------------------------%
|
|
|
|
% Generate the cases for a secondary tag using a binary search.
|
|
% This invocation looks after secondary tag values in the range
|
|
% MinPtag to MaxPtag (including both boundary values).
|
|
%
|
|
:- pred generate_secondary_binary_search(stag_goal_list(label)::in,
|
|
int::in, int::in, rval::in, maybe(label)::in, llds_code::out,
|
|
case_label_map::in, case_label_map::out,
|
|
code_info::in, code_info::out) is det.
|
|
|
|
generate_secondary_binary_search(StagGoals, MinStag, MaxStag, StagRval,
|
|
MaybeFailLabel, Code, !CaseLabelMap, !CI) :-
|
|
( if MinStag = MaxStag then
|
|
CurSec = MinStag,
|
|
(
|
|
StagGoals = [],
|
|
% There is no code for this tag.
|
|
(
|
|
MaybeFailLabel = yes(FailLabel),
|
|
string.int_to_string(CurSec, StagStr),
|
|
Comment = "no code for ptag " ++ StagStr,
|
|
Code = singleton(
|
|
llds_instr(goto(code_label(FailLabel)), Comment)
|
|
)
|
|
;
|
|
MaybeFailLabel = no,
|
|
Code = empty
|
|
)
|
|
;
|
|
StagGoals = [CurSecPrime - CaseLabel],
|
|
expect(unify(CurSec, CurSecPrime), $pred,
|
|
"cur_secondary mismatch"),
|
|
generate_case_code_or_jump(CaseLabel, Code, !CaseLabelMap)
|
|
;
|
|
StagGoals = [_, _ | _],
|
|
unexpected($pred,
|
|
"goallist not singleton or empty when binary search ends")
|
|
)
|
|
else
|
|
LowRangeEnd = (MinStag + MaxStag) // 2,
|
|
HighRangeStart = LowRangeEnd + 1,
|
|
InLowGroup = (pred(StagGoal::in) is semidet :-
|
|
StagGoal = Stag - _,
|
|
Stag =< LowRangeEnd
|
|
),
|
|
list.filter(InLowGroup, StagGoals, LowGoals, HighGoals),
|
|
get_next_label(NewLabel, !CI),
|
|
string.int_to_string(MinStag, LowStartStr),
|
|
string.int_to_string(LowRangeEnd, LowEndStr),
|
|
string.int_to_string(HighRangeStart, HighStartStr),
|
|
string.int_to_string(MaxStag, HighEndStr),
|
|
IfComment = "fallthrough for stags " ++
|
|
LowStartStr ++ " to " ++ LowEndStr,
|
|
LabelComment = "code for stags " ++
|
|
HighStartStr ++ " to " ++ HighEndStr,
|
|
LowRangeEndConst = const(llconst_int(LowRangeEnd)),
|
|
TestRval = binop(int_gt(int_type_int), StagRval, LowRangeEndConst),
|
|
IfCode = singleton(
|
|
llds_instr(if_val(TestRval, code_label(NewLabel)), IfComment)
|
|
),
|
|
LabelCode = singleton(
|
|
llds_instr(label(NewLabel), LabelComment)
|
|
),
|
|
|
|
generate_secondary_binary_search(LowGoals, MinStag, LowRangeEnd,
|
|
StagRval, MaybeFailLabel, LowRangeCode, !CaseLabelMap, !CI),
|
|
generate_secondary_binary_search(HighGoals, HighRangeStart, MaxStag,
|
|
StagRval, MaybeFailLabel, HighRangeCode, !CaseLabelMap, !CI),
|
|
|
|
Code = IfCode ++ LowRangeCode ++ LabelCode ++ HighRangeCode
|
|
).
|
|
|
|
%-----------------------------------------------------------------------------%
|
|
|
|
:- pred make_ptag_comment(string::in, ptag::in, list(ptag)::in,
|
|
string::out) is det.
|
|
|
|
make_ptag_comment(BaseStr, MainPtag, OtherPtags, Comment) :-
|
|
(
|
|
OtherPtags = [],
|
|
Comment = BaseStr ++ ptag_to_string(MainPtag)
|
|
;
|
|
OtherPtags = [_ | _],
|
|
Comment = BaseStr ++ ptag_to_string(MainPtag)
|
|
++ "(shared with " ++
|
|
string.join_list(", ", list.map(ptag_to_string, OtherPtags))
|
|
++ ")"
|
|
).
|
|
|
|
:- func ptag_to_string(ptag) = string.
|
|
|
|
ptag_to_string(ptag(Ptag)) = string.uint8_to_string(Ptag).
|
|
|
|
%-----------------------------------------------------------------------------%
|
|
:- end_module ll_backend.tag_switch.
|
|
%-----------------------------------------------------------------------------%
|