Switch cache vector

SciML · Oct 20, 2023 · 0640676 · 0640676
1 parent 5af2da0
commit 0640676
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/levenberg.jl b/src/levenberg.jl
@@ -218,8 +218,8 @@ function perform_step!(cache::LevenbergMarquardtCache{true})
 
     # Geodesic acceleration (step_size = v + a / 2).
     @unpack v, α_geodesic, h = cache
-    _vec(cache.du) .= _vec(u) .+ h .* _vec(v)
-    f(cache.fu_tmp, cache.du, p)
+    _vec(cache.u_tmp) .= _vec(u) .+ h .* _vec(v)
+    f(cache.fu_tmp, cache.u_tmp, p)
 
     # The following lines do: cache.a = -J \ cache.fu_tmp
     mul!(_vec(cache.Jv), J, _vec(v))