Skip to content

Commit 586baf8

Browse files
committed
benchmarks: fix benchmark eval options
1 parent 3113499 commit 586baf8

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

benchmark/benchmarks.jl

+8-1
Original file line numberDiff line numberDiff line change
@@ -74,11 +74,18 @@ function benchmark_evaluation()
7474
extra_kws...
7575
)
7676
suite[T]["evaluation$(extra_key)"] = @benchmarkable(
77-
[eval_tree_array(tree, X, $operators; turbo=$turbo, $extra_kws...) for tree in trees],
77+
[eval_tree_array(tree, X, $operators; kws...) for tree in trees],
7878
setup=(
7979
X=randn(MersenneTwister(0), $T, 5, $n);
8080
treesize=20;
8181
ntrees=100;
82+
kws=$(
83+
if @isdefined(EvalOptions)
84+
(; eval_options=EvalOptions(; turbo=turbo, extra_kws...))
85+
else
86+
(; turbo, extra_kws...)
87+
end
88+
);
8289
trees=[gen_random_tree_fixed_size(treesize, $operators, 5, $T) for _ in 1:ntrees]
8390
)
8491
)

0 commit comments

Comments
 (0)