Source file env.ml

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
open! Base

type t =
  | Max_workers
  | Length
  | Eager

let to_string = function
  | Max_workers -> "PARALLEL_BENCH_MAX_WORKERS"
  | Length -> "PARALLEL_BENCH_LENGTH"
  | Eager -> "PARALLEL_BENCH_EAGER"
;;

let get t ~default =
  match Sys.getenv (to_string t) with
  | None -> default
  | Some i -> Int.of_string i
;;

let max_workers = get Max_workers ~default:4096
let length = get Length ~default:1_000_000
let eager = get Eager ~default:0