Files
mercury/tests/tabling/fib_uint32.m
2018-07-08 23:20:52 +02:00

84 lines
2.0 KiB
Mathematica

% vim: ts=4 sw=4 et ft=mercury
:- module fib_uint32.
:- interface.
:- import_module io.
:- pred main(io::di, io::uo) is cc_multi.
:- implementation.
:- import_module benchmarking.
:- import_module int.
:- import_module uint32.
:- import_module require.
:- pragma require_feature_set([memo]).
main(!IO) :-
perform_trials(20u32, !IO).
:- pred perform_trials(uint32::in, io::di, io::uo) is cc_multi.
perform_trials(N, !IO) :-
trial(N, Time, MTime),
trace [compiletime(flag("progress")), io(!S)] (
io.write_string("trial ", !S),
io.write_uint32(N, !S),
io.write_string(": ", !S),
io.write_int(Time, !S),
io.write_string("ms nonmemoed vs ", !S),
io.write_int(MTime, !S),
io.write_string("ms memoed\n", !S)
),
( if
(
Time > 10 * MTime,
MTime > 0 % untabled takes ten times as long
;
Time > 100, % untabled takes at least 100 ms
MTime < 1 % while tabled takes at most 1 ms
)
then
io.write_string("tabling works\n", !IO)
else if
Time > 10000 % Untabled takes at least 10 seconds
then
io.write_string("tabling does not appear to work\n", !IO)
else
% We couldn't get a measurable result with N,
% and it looks like we can afford a bigger trial
perform_trials(N + 3u32, !IO)
).
:- pred trial(uint32::in, int::out, int::out) is cc_multi.
trial(N, Time, MTime) :-
benchmark_det(fib, N, Res, 1, Time),
benchmark_det(mfib, N, MRes, 1, MTime),
require(unify(Res, MRes), "tabling produces wrong answer").
:- pred fib(uint32::in, uint32::out) is det.
fib(N, F) :-
( if N < 2u32 then
F = 1u32
else
fib(N - 1u32, F1),
fib(N - 2u32, F2),
F = F1 + F2
).
:- pred mfib(uint32::in, uint32::out) is det.
:- pragma memo(mfib/2).
mfib(N, F) :-
( if N < 2u32 then
F = 1u32
else
mfib(N - 1u32, F1),
mfib(N - 2u32, F2),
F = F1 + F2
).