2022-04-05 01:11:53 +00:00
|
|
|
module Benchmark
|
|
|
|
|
|
|
|
|
|
using Dates: Dates
|
|
|
|
|
using BenchmarkTools: @benchmark
|
|
|
|
|
|
|
|
|
|
# TODO: Replace JSON3
|
|
|
|
|
# FIX: Benchmarking
|
|
|
|
|
|
|
|
|
|
using ReCo
|
|
|
|
|
|
|
|
|
|
function run_benchmarks(
|
|
|
|
|
dir::String = "";
|
|
|
|
|
n_particles::Int64 = 1000,
|
|
|
|
|
v₀::Float64 = 20.0,
|
|
|
|
|
duration::Float64 = 2.0,
|
|
|
|
|
n_bundle_snapshots::Int64 = 0,
|
|
|
|
|
comment = ""
|
|
|
|
|
)
|
|
|
|
|
if length(dir) == 0
|
|
|
|
|
dir = init_sim(; n_particles = n_particles, v₀ = v₀, parent_dir = "benchmark")
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
benchmark = @benchmark run_sim(
|
|
|
|
|
$dir; duration = $duration, n_bundle_snapshots = $n_bundle_snapshots
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
display(benchmark)
|
|
|
|
|
|
2022-09-25 15:56:00 +00:00
|
|
|
open("../exports/benchmark.txt", "a+") do f
|
2022-04-05 01:11:53 +00:00
|
|
|
JSON3.pretty(
|
|
|
|
|
f,
|
|
|
|
|
Dict(
|
|
|
|
|
"n_particles" => n_particles,
|
|
|
|
|
"v₀" => v₀,
|
|
|
|
|
"duration" => duration,
|
|
|
|
|
"n_bundle_snapshots" => n_bundle_snapshots,
|
|
|
|
|
"comment" => comment,
|
|
|
|
|
"datetime" => Dates.now(),
|
|
|
|
|
"mean_time/ns" => mean(benchmark.times),
|
|
|
|
|
"allocs" => benchmark.allocs,
|
|
|
|
|
"memory" => benchmark.memory,
|
|
|
|
|
),
|
|
|
|
|
)
|
|
|
|
|
write(f, "\n")
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
return dir
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
end # module
|
