Fixes

Author: gdalle

DifferentiationInterface/ext/DifferentiationInterfaceReverseDiffExt/onearg.jl

@@ -87,7 +87,7 @@ end
 function DI.prepare_gradient(
     f, backend::AutoReverseDiff{compile}, x; strict::Val=Val(false)
 ) where {compile}
-    _sig = DI.signature(f, backend, x)
+    _sig = DI.signature(f, backend, x; strict)
     if compile
         tape = ReverseDiff.compile(GradientTape(f, x))
         return ReverseDiffGradientPrep(_sig, nothing, tape)

DifferentiationInterface/ext/DifferentiationInterfaceSparseMatrixColoringsExt/hessian.jl

@@ -62,10 +62,10 @@ function _prepare_sparse_hessian_aux(
         ntuple(b -> seeds[1 + ((a - 1) * B + (b - 1)) % N], Val(B)) for a in 1:A
     ]
     batched_results = [ntuple(b -> similar(x), Val(B)) for _ in batched_seeds]
-    hvp_prep = DI.prepare_hvp(f, dense_backend, x, batched_seeds[1], contexts...)
-    gradient_prep = DI.prepare_gradient(f, DI.inner(dense_backend), x, contexts...)
+    hvp_prep = DI.prepare_hvp(f, dense_backend, x, batched_seeds[1], contexts...; strict)
+    gradient_prep = DI.prepare_gradient(f, DI.inner(dense_backend), x, contexts...; strict)
     return SparseHessianPrep(
-        SIG,
+        _sig,
         batch_size_settings,
         coloring_result,
         compressed_matrix,

DifferentiationInterface/ext/DifferentiationInterfaceSparseMatrixColoringsExt/jacobian.jl

@@ -109,10 +109,10 @@ function _prepare_sparse_jacobian_aux_aux(
     ]
     batched_results = [ntuple(b -> similar(y), Val(B)) for _ in batched_seeds]
     pushforward_prep = DI.prepare_pushforward(
-        f_or_f!y..., dense_backend, x, batched_seeds[1], contexts...
+        f_or_f!y..., dense_backend, x, batched_seeds[1], contexts...; strict
     )
     return PushforwardSparseJacobianPrep(
-        SIG,
+        _sig,
         batch_size_settings,
         coloring_result,
         compressed_matrix,
@@ -143,10 +143,10 @@ function _prepare_sparse_jacobian_aux_aux(
     ]
     batched_results = [ntuple(b -> similar(x), Val(B)) for _ in batched_seeds]
     pullback_prep = DI.prepare_pullback(
-        f_or_f!y..., dense_backend, x, batched_seeds[1], contexts...
+        f_or_f!y..., dense_backend, x, batched_seeds[1], contexts...; strict
     )
     return PullbackSparseJacobianPrep(
-        SIG,
+        _sig,
         batch_size_settings,
         coloring_result,
         compressed_matrix,

DifferentiationInterface/ext/DifferentiationInterfaceSparseMatrixColoringsExt/jacobian_mixed.jl

@@ -132,18 +132,20 @@ function _prepare_mixed_sparse_jacobian_aux_aux(
         DI.forward_backend(dense_backend),
         x,
         batched_seeds_forward[1],
-        contexts...,
+        contexts...;
+        strict,
     )
     pullback_prep = DI.prepare_pullback(
         f_or_f!y...,
         DI.reverse_backend(dense_backend),
         x,
         batched_seeds_reverse[1],
-        contexts...,
+        contexts...;
+        strict,
     )
 
     return MixedModeSparseJacobianPrep(
-        SIG,
+        _sig,
         batch_size_settings_forward,
         batch_size_settings_reverse,
         coloring_result,

DifferentiationInterface/ext/DifferentiationInterfaceTrackerExt/DifferentiationInterfaceTrackerExt.jl

@@ -35,7 +35,7 @@ function DI.prepare_pullback_same_point(
     ty::NTuple,
     contexts::Vararg{DI.GeneralizedConstant,C},
 ) where {C}
-    _sig = DI.signature(f, prep, backend, x, ty, contexts...)
+    _sig = DI.signature(f, prep, backend, x, ty, contexts...; strict)
     DI.check_prep(f, prep, backend, x, ty, contexts...)
     y, pb = forward(f, x, map(DI.unwrap, contexts)...)
     return TrackerPullbackPrepSamePoint(_sig, y, pb)