Codecov

Author: gdalle

DifferentiationInterface/src/second_order/hvp.jl

@@ -172,16 +172,12 @@ function _prepare_hvp_aux(
     grad_buffer = similar(x)
     rewrap = Rewrap(contexts...)
     # Inner gradient
-    inner_gradient_prep = let
-        xo = overloaded_input(pushforward, shuffled_gradient, outer(backend), x, tx)
-        prepare_gradient(f, inner(backend), xo, contexts...)
-    end
-    inner_gradient_in_prep = let
-        xo = overloaded_input(
-            pushforward, shuffled_gradient!, grad_buffer, outer(backend), x, tx
-        )
-        prepare_gradient(f, inner(backend), xo, contexts...)
-    end
+    xo = overloaded_input(pushforward, shuffled_gradient, outer(backend), x, tx)
+    xoi = overloaded_input(
+        pushforward, shuffled_gradient!, grad_buffer, outer(backend), x, tx
+    )
+    inner_gradient_prep = prepare_gradient(f, inner(backend), xo, contexts...)
+    inner_gradient_in_prep = prepare_gradient(f, inner(backend), xoi, contexts...)
     # Outer pushforward
     new_contexts = (
         FunctionContext(f),

DifferentiationInterface/test/Core/SimpleFiniteDiff/test.jl

@@ -11,6 +11,7 @@ LOGGING = get(ENV, "CI", "false") == "false"
 
 backends = [ #
     AutoSimpleFiniteDiff(; chunksize=5),
+    AutoForwardFromPrimitive(AutoSimpleFiniteDiff(; chunksize=4)),
     AutoReverseFromPrimitive(AutoSimpleFiniteDiff(; chunksize=4)),
 ]
 
@@ -68,7 +69,7 @@ end
     test_differentiation(
         second_order_hvp_backends;
         excluded=vcat(FIRST_ORDER, :hessian, :second_derivative),
-        logging=true,
+        logging=LOGGING,
     )
 
     test_differentiation(backends, complex_scenarios(); logging=LOGGING)