Merge pull request #310 from JuliaDiff/ChrisRackauckas-patch-2

Fix AutoFiniteDiff caching
JuliaDiff · Dec 22, 2024 · 5deb9d9 · 5deb9d9
2 parents 601ff82 + f05b38b
commit 5deb9d9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/differentiation/jaches_products.jl b/src/differentiation/jaches_products.jl
@@ -268,8 +268,9 @@ function JacVec(f, u::AbstractArray, p = nothing, t = nothing; fu = nothing,
     cache, vecprod, vecprod! = if autodiff isa AutoFiniteDiff
         cache1 = similar(fu)
         cache2 = similar(u)
+        cache3 = similar(u)
 
-        (cache1, cache2), num_jacvec, num_jacvec!
+        (cache1, cache2, cache3), num_jacvec, num_jacvec!
     elseif autodiff isa AutoForwardDiff
         cache1 = Dual{
             typeof(ForwardDiff.Tag(tag, eltype(u))), eltype(u), 1