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

benchmark for memory usage of data structures

Browse source
This commit is contained in:
Fabian 2020-04-19 17:37:25 +02:00
parent 952b664a68
commit 53d5a80b96
3 changed files with 67 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
benchs/dune
View file

@ -1,11 +1,9 @@
(executables
(names run_benchs run_bench_hash)
(names run_benchs run_bench_hash run_objsize)
(libraries containers containers.data containers.iter
containers.thread benchmark gen iter qcheck oseq
batteries)
batteries base core_kernel sek)
(flags :standard -warn-error -3 -safe-string -color always -open CCShims_)
(ocamlopt_flags :standard -O3 -color always
-unbox-closures -unbox-closures-factor 20)
)

57
benchs/run_objsize.ml Normal file
View file

@ -0,0 +1,57 @@
module Deque = Core_kernel.Deque
module Int_map = Map.Make(CCInt)
module Int_set = Set.Make(CCInt)
let dup = CCPair.dup
let id = CCFun.id
let ns n = List.init n CCFun.id
let iter_range n f = List.iter f (ns n)
let gen_cons x xs =
let saw_x = ref false in
fun () ->
if !saw_x then (saw_x := true; Some x)
else xs ()
let front = Sek.front
let dummy = 0
let types = [
"Stdlib.List", (fun n -> Obj.magic @@ ns n);
"Stdlib.Array", (fun n -> Obj.magic @@ Array.init n id);
"Stdlib.Hashtbl", (fun n -> Obj.magic @@ Hashtbl.of_seq (OSeq.init ~n dup));
"Base.Hashtbl", (fun n -> Obj.magic @@ Base.Hashtbl.Poly.of_alist_exn (List.init n dup));
"Stdlib.Map", (fun n -> Obj.magic @@ Int_map.of_seq (OSeq.init ~n dup));
"Stdlib.Set", (fun n -> Obj.magic @@ Int_set.of_seq (OSeq.init ~n id));
"CCFun_vec", (fun n -> Obj.magic @@ CCFun_vec.of_list (ns n));
"CCRAL", (fun n -> Obj.magic @@ CCRAL.of_list (ns n));
"BatVect", (fun n -> Obj.magic @@ BatVect.of_list (ns n));
"Sek.Persistent", (fun n -> Obj.magic @@ List.fold_left (Sek.Persistent.push front) (Sek.Persistent.create dummy) (ns n));
"Sek.Ephemeral", (fun n -> Obj.magic @@ let c = Sek.Ephemeral.create dummy in iter_range n (Sek.Ephemeral.push front c); c);
"CCVector", (fun n -> Obj.magic @@ let c = CCVector.create () in iter_range n (CCVector.push c); c);
"Core_kernel.Deque", (fun n -> Obj.magic @@ let c = Deque.create () in iter_range n (Deque.enqueue_back c); c);
"Base.Queue", (fun n -> Obj.magic @@ let c = Base.Queue.create () in iter_range n (Base.Queue.enqueue c); c);
"Stdlib.Queue", (fun n -> Obj.magic @@ Queue.of_seq (OSeq.init ~n id));
"CCQueue", (fun n -> Obj.magic @@ CCDeque.of_list (ns n));
"Iter", (fun n -> Obj.magic @@ List.fold_right Iter.cons (ns n) Iter.empty);
"Gen", (fun n -> Obj.magic @@ List.fold_right gen_cons (ns n) Gen.empty);
"Stdlib.Seq", (fun n -> Obj.magic @@ List.fold_right OSeq.cons (ns n) OSeq.empty);
]
let () =
let sizes = [0; 1; 10; 100; 1000; 10000] in
Printf.printf "%-20s " "";
sizes |> List.iter (fun n -> Printf.printf "%6d " n);
Printf.printf "\n";
types
|> List.iter (fun (name, create) ->
Printf.printf "%-20s: " name;
sizes
|> List.iter (fun n ->
let obj = create n in
let size = Objsize.size_w obj in
(* let size = Obj.reachable_words (Obj.repr obj) in *)
Printf.printf "%6d " size
);
Printf.printf "\n"
)