KrylovJL solvers returning incorrect answer when `u0` is a `SubArray` about linearsolve.jl

[fredrikekre]

This is caused by

LinearSolve.jl/src/iterative_wrappers.jl

Line 220 in 0886d05

solver.x = u

. Because the RHS is ::Vector then the solver struct enforces the same type also for x, and thus u0 will be converted to Vector in the line above. A potential fix is:

diff --git a/src/iterative_wrappers.jl b/src/iterative_wrappers.jl
index b471065..6de56a9 100644
--- a/src/iterative_wrappers.jl
+++ b/src/iterative_wrappers.jl
@@ -282,6 +282,11 @@ function SciMLBase.solve!(cache::LinearCache, alg::KrylovJL; kwargs...)
         ReturnCode.Success
     end

+    # Copy the solution to the allocated output vector
+    if cache.u !== cache.cacheval.x
+        cache.u .= cache.cacheval.x
+    end
+
     return SciMLBase.build_linear_solution(alg, cache.u, resid, cache;
         iters = stats.niter)
 end