add benchs/ with a basic fib test

This commit is contained in:
Simon Cruanes 2023-06-07 21:03:26 -04:00
parent 39693b0f3f
commit 3d972cc0cd
No known key found for this signature in database
GPG key ID: EBFFF6F283F3A2B4
2 changed files with 50 additions and 0 deletions

4
benchs/dune Normal file
View file

@ -0,0 +1,4 @@
(executables
(names fib_rec)
(libraries moonpool))

46
benchs/fib_rec.ml Normal file
View file

@ -0,0 +1,46 @@
open Moonpool
let rec fib_direct x =
if x <= 1 then
1
else
fib_direct (x - 1) + fib_direct (x - 2)
let rec fib ~on x : int Fut.t =
if x <= 20 then
Fut.spawn ~on (fun () -> fib_direct x)
else
let open Fut.Infix_local in
let+ t1 = fib ~on (x - 1) and+ t2 = fib ~on (x - 2) in
t1 + t2
let () = assert (List.init 10 fib_direct = [ 1; 1; 2; 3; 5; 8; 13; 21; 34; 55 ])
let run ~psize ~n ~seq () : unit =
let res =
if seq then (
Printf.printf "compute fib %d sequentially\n%!" n;
fib_direct n
) else (
Printf.printf "compute fib %d with pool size=%d\n%!" n psize;
let pool = Pool.create ~min:psize () in
fib ~on:pool n |> Fut.wait_block_exn
)
in
Printf.printf "fib %d = %d\n%!" n res
let () =
let n = ref 40 in
let psize = ref 16 in
let seq = ref false in
let opts =
[
"-psize", Arg.Set_int psize, " pool size";
"-n", Arg.Set_int n, " fib <n>";
"-seq", Arg.Set seq, " sequential";
]
|> Arg.align
in
Arg.parse opts ignore "";
run ~psize:!psize ~n:!n ~seq:!seq ()