Format

Author: gdalle

DifferentiationInterface/ext/DifferentiationInterfaceSymbolicsExt/onearg.jl

@@ -18,7 +18,9 @@ function DI.prepare_pushforward_nokwarg(
     step_der_var = derivative(f(x_var + t_var * dx_var, context_vars...), t_var)
     pf_var = substitute(step_der_var, Dict(t_var => zero(eltype(x))))
 
-    res = build_function(pf_var, x_var, dx_var, context_vars...; expression=Val(false), cse=true)
+    res = build_function(
+        pf_var, x_var, dx_var, context_vars...; expression=Val(false), cse=true
+    )
     (pf_exe, pf_exe!) = if res isa Tuple
         res
     elseif res isa RuntimeGeneratedFunction
@@ -177,7 +179,9 @@ function DI.prepare_gradient_nokwarg(
     # Symbolic.gradient only accepts vectors
     grad_var = gradient(f(x_var, context_vars...), vec(x_var))
 
-    res = build_function(grad_var, vec(x_var), context_vars...; expression=Val(false), cse=true)
+    res = build_function(
+        grad_var, vec(x_var), context_vars...; expression=Val(false), cse=true
+    )
     (grad_exe, grad_exe!) = res
     return SymbolicsOneArgGradientPrep(_sig, grad_exe, grad_exe!)
 end
@@ -333,7 +337,9 @@ function DI.prepare_hessian_nokwarg(
         hessian(f(x_var, context_vars...), vec(x_var))
     end
 
-    res = build_function(hess_var, vec(x_var), context_vars...; expression=Val(false), cse=true)
+    res = build_function(
+        hess_var, vec(x_var), context_vars...; expression=Val(false), cse=true
+    )
     (hess_exe, hess_exe!) = res
 
     gradient_prep = DI.prepare_gradient_nokwarg(
@@ -420,7 +426,12 @@ function DI.prepare_hvp_nokwarg(
     hvp_vec_var = hess_var * vec(dx_var)
 
     res = build_function(
-        hvp_vec_var, vec(x_var), vec(dx_var), context_vars...; expression=Val(false), cse=true
+        hvp_vec_var,
+        vec(x_var),
+        vec(dx_var),
+        context_vars...;
+        expression=Val(false),
+        cse=true,
     )
     (hvp_exe, hvp_exe!) = res
 

DifferentiationInterface/ext/DifferentiationInterfaceSymbolicsExt/twoarg.jl

@@ -26,7 +26,9 @@ function DI.prepare_pushforward_nokwarg(
     step_der_var = derivative(y_var, t_var)
     pf_var = substitute(step_der_var, Dict(t_var => zero(eltype(x))))
 
-    res = build_function(pf_var, x_var, dx_var, context_vars...; expression=Val(false), cse=true)
+    res = build_function(
+        pf_var, x_var, dx_var, context_vars...; expression=Val(false), cse=true
+    )
     (pushforward_exe, pushforward_exe!) = res
     return SymbolicsTwoArgPushforwardPrep(_sig, pushforward_exe, pushforward_exe!)
 end