Set up fuse_level option for Enzyme compatibility

Author: MilesCranmer

src/ConstantOptimization.jl

@@ -56,7 +56,7 @@ function dispatch_optimize_constants(
             algorithm,
             options.optimizer_options,
             idx,
-            options.enable_enzyme ? Val(true) : Val(false),
+            options.v_enable_enzyme,
         )
     end
     if T <: Complex

src/InterfaceDynamicExpressions.jl

@@ -53,7 +53,10 @@ which speed up evaluation significantly.
     to the equation.
 """
 function eval_tree_array(tree::Node, X::AbstractArray, options::Options; kws...)
-    return eval_tree_array(tree, X, options.operators; turbo=options.v_turbo, kws...)
+    fuse_level = options.v_enable_enzyme === Val(true) ? Val(1) : Val(2)
+    return eval_tree_array(
+        tree, X, options.operators; turbo=options.v_turbo, fuse_level, kws...
+    )
 end
 
 """

src/Options.jl

@@ -732,6 +732,8 @@ function Options end
         mutation_weights
     end
 
+    v_enable_enzyme = Val(enable_enzyme)
+
     @assert print_precision > 0
 
     options = Options{
@@ -741,6 +743,7 @@ function Options end
         typeof(optimizer_options),
         typeof(tournament_selection_weights),
         turbo,
+        enable_enzyme,
     }(
         operators,
         bin_constraints,
@@ -802,7 +805,7 @@ function Options end
         nested_constraints,
         deterministic,
         define_helper_functions,
-        enable_enzyme,
+        v_enable_enzyme,
     )
 
     return options

src/OptionsStruct.jl

@@ -129,7 +129,9 @@ function ComplexityMapping(;
     )
 end
 
-struct Options{CT,OP<:AbstractOperatorEnum,use_recorder,OPT<:Optim.Options,W,_turbo}
+struct Options{
+    CT,OP<:AbstractOperatorEnum,use_recorder,OPT<:Optim.Options,W,_turbo,enable_enzyme
+}
     operators::OP
     bin_constraints::Vector{Tuple{Int,Int}}
     una_constraints::Vector{Int}
@@ -190,7 +192,7 @@ struct Options{CT,OP<:AbstractOperatorEnum,use_recorder,OPT<:Optim.Options,W,_tu
     nested_constraints::Union{Vector{Tuple{Int,Int,Vector{Tuple{Int,Int,Int}}}},Nothing}
     deterministic::Bool
     define_helper_functions::Bool
-    enable_enzyme::Bool
+    v_enable_enzyme::Val{enable_enzyme}
 end
 
 function Base.print(io::IO, options::Options)