Skip to content

Commit 762bfb1

Browse files
committed
make sure u is aliased
1 parent 96244fd commit 762bfb1

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

ext/LinearSolveForwardDiffExt.jl

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -167,7 +167,12 @@ function SciMLBase.solve!(cache::DualLinearCache, alg::SciMLLinearSolveAlgorithm
167167
partials = linearsolve_forwarddiff_solve(
168168
cache::DualLinearCache, cache.alg, args...; kwargs...)
169169
dual_sol = linearsolve_dual_solution(sol.u, partials, cache.dual_type)
170-
cache.dual_u = dual_sol
170+
171+
if cache.dual_u isa AbstractArray
172+
cache.dual_u[:] = dual_sol
173+
else
174+
cache.dual_u = dual_sol
175+
end
171176

172177
return SciMLBase.build_linear_solution(
173178
cache.alg, dual_sol, sol.resid, cache; sol.retcode, sol.iters, sol.stats

0 commit comments

Comments
 (0)