forked from thautwarm/MLStyle.jl
-
Notifications
You must be signed in to change notification settings - Fork 0
/
matrix_benchmark.jl
50 lines (44 loc) · 1.21 KB
/
matrix_benchmark.jl
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
# If you want to run this matrix-benchmarks, you should install
# a dependency `Benchmarkplotting` which is not registered on JuliaRegistries.
# pkg> add https://github.com/thautwarm/Benchmarkplotting.jl
using MLStyle
using ArgParse
include("matrix-benchmark/sampler.jl")
export ArbitrarySampler
versus_items = ("misc", "tuple", "array", "structfields", "vs-match", "datatype")
function parse_cmd()
s = ArgParseSettings()
@add_arg_table s begin
"versus"
help = join(", ", versus_items)
required = true
action => :store_arg
nargs = '+'
end
parse_args(ARGS, s)
end
check_versus(x) = x in versus_items
function benchmark(x)
filename = "matrix-benchmark/bench-$x.jl"
open(filename) do f
include_string(Main, read(f, String), filename);
end
end
action = @λ begin
[] -> nothing
[GuardBy(check_versus) && hd, tl...] ->
begin
benchmark(hd)
action(tl)
end
[hd, tl...] ->
begin
@warn "Unknown versus item: $hd"
action(hd)
end
end
function main()
parsed_args = parse_cmd()
Dict([a => b for (a, b) in parsed_args])["versus"] |> action
end
main()