mirror of
https://github.com/c-cube/moonpool.git
synced 2025-12-06 03:05:30 -05:00
add benchs/ with a basic fib test
This commit is contained in:
parent
39693b0f3f
commit
3d972cc0cd
2 changed files with 50 additions and 0 deletions
4
benchs/dune
Normal file
4
benchs/dune
Normal file
|
|
@ -0,0 +1,4 @@
|
|||
|
||||
(executables
|
||||
(names fib_rec)
|
||||
(libraries moonpool))
|
||||
46
benchs/fib_rec.ml
Normal file
46
benchs/fib_rec.ml
Normal 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 ()
|
||||
Loading…
Add table
Reference in a new issue