fix: enable arbitrary benchmark aggregations (#639)

JuliaDiff · Nov 24, 2024 · 5576221 · 5576221
1 parent 6b55559
commit 5576221
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/DifferentiationInterfaceTest/src/tests/benchmark.jl b/DifferentiationInterfaceTest/src/tests/benchmark.jl
@@ -93,20 +93,19 @@ function record!(
     calls::Integer,
     aggregation,
 )
-    bench_agg = aggregation(bench)
     row = DifferentiationBenchmarkDataRow(;
         backend=backend,
         scenario=scenario,
         operator=Symbol(operator),
         prepared=prepared,
         calls=calls,
         samples=length(bench.samples),
-        evals=Int(bench_agg.evals),
-        time=bench_agg.time,
-        allocs=bench_agg.allocs,
-        bytes=bench_agg.bytes,
-        gc_fraction=bench_agg.gc_fraction,
-        compile_fraction=bench_agg.compile_fraction,
+        evals=Int(bench.samples[1].evals),
+        time=aggregation(getfield.(bench.samples, :time)),
+        allocs=aggregation(getfield.(bench.samples, :allocs)),
+        bytes=aggregation(getfield.(bench.samples, :bytes)),
+        gc_fraction=aggregation(getfield.(bench.samples, :gc_fraction)),
+        compile_fraction=aggregation(getfield.(bench.samples, :compile_fraction)),
     )
     return push!(data, row)
 end