unikernel/duniverse/dune_/bench/micro/dune_bench/dune

@@ -0,0 +1,6 @@
+(library
+ (name dune_bench)
+ (libraries stdune fiber dune_engine dune_rules)
+ (library_flags -linkall)
+ (preprocess
+  (pps ppx_bench)))

unikernel/duniverse/dune_/bench/micro/dune_bench/scheduler_bench.ml

@@ -0,0 +1,39 @@
+(* Benchmark the scheduler *)
+
+open Stdune
+open Dune_engine
+module Caml = Stdlib
+
+let config =
+  Dune_engine.Clflags.display := Short;
+  { Scheduler.Config.concurrency = 1
+  ; stats = None
+  ; print_ctrl_c_warning = false
+  ; watch_exclusions = []
+  }
+;;
+
+let setup =
+  lazy
+    (Path.set_root (Path.External.cwd ());
+     Path.Build.set_build_dir (Path.Outside_build_dir.of_string "_build"))
+;;
+
+let prog = Option.value_exn (Bin.which ~path:(Env_path.path Env.initial) "true")
+let run () = Process.run ~display:Quiet ~env:Env.initial Strict prog []
+
+let go ~jobs fiber =
+  Scheduler.Run.go ~on_event:(fun _ _ -> ()) { config with concurrency = jobs } fiber
+;;
+
+let%bench_fun "single" =
+  Lazy.force setup;
+  fun () -> go run ~jobs:1
+;;
+
+let l = List.init 100 ~f:ignore
+
+let%bench_fun ("many" [@indexed jobs = [ 1; 2; 4; 8 ]]) =
+  Lazy.force setup;
+  fun () -> go ~jobs (fun () -> Fiber.parallel_iter l ~f:run)
+;;