Skip to content

Commit 9b76d4a

Browse files
committed
Add missing turbo to benchmarks
1 parent fec16df commit 9b76d4a

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

benchmark/benchmarks.jl

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,9 @@ simple_tree = Node(
2727
),
2828
)
2929
for T in (ComplexF32, ComplexF64, Float32, Float64)
30-
!(T <: Real) && v_PACKAGE_VERSION < v"0.5.0" && continue
30+
if !(T <: Real) && v_PACKAGE_VERSION < v"0.5.0" && v_PACKAGE_VERSION != v"0.0.0"
31+
continue
32+
end
3133
evals = 10
3234
samples = 1_000
3335
n = 1_000
@@ -38,10 +40,12 @@ for T in (ComplexF32, ComplexF64, Float32, Float64)
3840
end
3941

4042
for turbo in (false, true)
41-
turbo && !(T in (Float32, Float64)) && continue
43+
if turbo && !(T in (Float32, Float64))
44+
continue
45+
end
4246
extra_key = turbo ? "_turbo" : ""
4347
SUITE["OperatorEnum"][T]["evaluation$(extra_key)"] = @benchmarkable(
44-
eval_tree_array(tree, X, $operators),
48+
eval_tree_array(tree, X, $operators; turbo=$turbo),
4549
evals=evals,
4650
samples=samples,
4751
seconds=5.0,
@@ -52,7 +56,7 @@ for T in (ComplexF32, ComplexF64, Float32, Float64)
5256
)
5357
if T <: Real
5458
SUITE["OperatorEnum"][T]["derivative$(extra_key)"] = @benchmarkable(
55-
eval_grad_tree_array(tree, X, $operators; variable=true),
59+
eval_grad_tree_array(tree, X, $operators; variable=true, turbo=$turbo),
5660
evals=evals,
5761
samples=samples,
5862
seconds=5.0,

0 commit comments

Comments
 (0)