simon/moonpool
1
0
Fork 0
mirror of https://github.com/c-cube/moonpool.git synced 2025-12-06 03:05:30 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

bench: add -niter param

Browse source
This commit is contained in:
Simon Cruanes 2023-06-07 21:23:15 -04:00
parent 65b3d8de06
commit c77b579caa
No known key found for this signature in database
GPG key ID: EBFFF6F283F3A2B4
2 changed files with 22 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
Makefile
View file

@ -19,13 +19,14 @@ watch:
DUNE_OPTS_BENCH?=--profile=release
N?=40
NITER?=3
bench-fib:
@echo running for N=$(N)
dune build $(DUNE_OPTS_BENCH) benchs/fib_rec.exe
hyperfine \
'./_build/default/benchs/fib_rec.exe -psize=1 -n $(N)' \
'./_build/default/benchs/fib_rec.exe -psize=8 -n $(N)' \
'./_build/default/benchs/fib_rec.exe -psize=20 -n $(N)' \
'./_build/default/benchs/fib_rec.exe -n $(N) -seq'
'./_build/default/benchs/fib_rec.exe -niter $(NITER) -psize=1 -n $(N)' \
'./_build/default/benchs/fib_rec.exe -niter $(NITER) -psize=8 -n $(N)' \
'./_build/default/benchs/fib_rec.exe -niter $(NITER) -psize=20 -n $(N)' \
'./_build/default/benchs/fib_rec.exe -niter $(NITER) -n $(N) -seq'
.PHONY: test clean

12
benchs/fib_rec.ml
View file

@ -16,31 +16,35 @@ let rec fib ~on x : int Fut.t =
let () = assert (List.init 10 fib_direct = [ 1; 1; 2; 3; 5; 8; 13; 21; 34; 55 ])
let run ~psize ~n ~seq () : unit =
let run ~psize ~n ~seq ~niter () : unit =
let pool = lazy (Pool.create ~min:psize ()) in
for _i = 1 to niter do
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
fib ~on:(Lazy.force pool) n |> Fut.wait_block_exn
)
in
Printf.printf "fib %d = %d\n%!" n res
done
let () =
let n = ref 40 in
let psize = ref 16 in
let seq = ref false in
let niter = ref 3 in
let opts =
[
"-psize", Arg.Set_int psize, " pool size";
"-n", Arg.Set_int n, " fib <n>";
"-seq", Arg.Set seq, " sequential";
"-niter", Arg.Set_int niter, " number of iterations";
]
|> Arg.align
in
Arg.parse opts ignore "";
run ~psize:!psize ~n:!n ~seq:!seq ()
run ~psize:!psize ~n:!n ~seq:!seq ~niter:!niter ()