Add benchmark file

Author: MilesCranmer

benchmark/Project.toml

@@ -0,0 +1,4 @@
+[deps]
+BenchmarkTools = "6e4b80f9-dd63-53aa-95a3-0cdb28fa8baf"
+DynamicExpressions = "a40a106e-89c9-4ca8-8020-a735e8728b6b"
+Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"

benchmark/benchmarks.jl

@@ -0,0 +1,67 @@
+using DynamicExpressions, BenchmarkTools, Random
+
+const v_PACKAGE_VERSION = try
+    VersionNumber(PACKAGE_VERSION)
+catch
+    VersionNumber("v0.0.0")
+end
+
+const SUITE = BenchmarkGroup()
+
+SUITE["OperatorEnum"] = BenchmarkGroup()
+
+operators = OperatorEnum(;
+    binary_operators=[+, -, /, *], unary_operators=[cos, exp], enable_autodiff=true
+)
+simple_tree = Node(
+    2,
+    Node(
+        1,
+        Node(3, Node(1, Node(; val=1.0f0), Node(; feature=2)), Node(2, Node(; val=-1.0f0))),
+        Node(1, Node(; feature=3), Node(; feature=4)),
+    ),
+    Node(
+        4,
+        Node(3, Node(1, Node(; val=1.0f0), Node(; feature=2)), Node(2, Node(; val=-1.0f0))),
+        Node(1, Node(; feature=3), Node(; feature=4)),
+    ),
+)
+for T in (ComplexF32, ComplexF64, Float32, Float64)
+    !(T <: Real) && v_PACKAGE_VERSION < v"0.5.0" && continue
+    evals = 10
+    samples = 1_000
+    n = 1_000
+
+    #! format: off
+    if !haskey(SUITE["OperatorEnum"], T)
+        SUITE["OperatorEnum"][T] = BenchmarkGroup()
+    end
+
+    for turbo in (false, true)
+        turbo && !(T in (Float32, Float64)) && continue
+        extra_key = turbo ? "_turbo" : ""
+        SUITE["OperatorEnum"][T]["evaluation$(extra_key)"] = @benchmarkable(
+            eval_tree_array(tree, X, $operators),
+            evals=evals,
+            samples=samples,
+            seconds=5.0,
+            setup=(
+                X=randn(MersenneTwister(0), $T, 5, $n);
+                tree=convert(Node{$T}, copy_node($simple_tree))
+            )
+        )
+        if T <: Real
+            SUITE["OperatorEnum"][T]["derivative$(extra_key)"] = @benchmarkable(
+                eval_grad_tree_array(tree, X, $operators; variable=true),
+                evals=evals,
+                samples=samples,
+                seconds=5.0,
+                setup=(
+                    X=randn(MersenneTwister(0), $T, 5, $n);
+                    tree=convert(Node{$T}, copy_node($simple_tree))
+                )
+            )
+        end
+    end
+    #! format: on
+end