fix for cache in solution

Author: ChrisRackauckas

Project.toml

@@ -1,7 +1,7 @@
 name = "LinearSolve"
 uuid = "7ed4a6bd-45f5-4d41-b270-4a48e9bafcae"
 authors = ["SciML"]
-version = "0.2.0"
+version = "0.2.1"
 
 [deps]
 ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"

docs/src/tutorials/caching_interface.md

@@ -38,7 +38,7 @@ sol1.u
   1.8385599677530706
 =#
 
-linsolve = LinearSolve.set_b(linsolve,b2)
+linsolve = LinearSolve.set_b(sol1.cache,b2)
 sol2 = solve(linsolve)
 
 sol2.u

src/factorization.jl

@@ -5,7 +5,7 @@ function SciMLBase.solve(cache::LinearCache, alg::AbstractFactorization; kwargs.
     end
 
     y = ldiv!(cache.u, cache.cacheval, cache.b)
-    SciMLBase.build_linear_solution(alg,y,nothing)
+    SciMLBase.build_linear_solution(alg,y,nothing,cache)
 end
 
 ## LU Factorizations
@@ -68,7 +68,7 @@ function SciMLBase.solve(cache::LinearCache, alg::UMFPACKFactorization)
     end
 
     y = ldiv!(cache.u, cache.cacheval, cache.b)
-    SciMLBase.build_linear_solution(alg,y,nothing)
+    SciMLBase.build_linear_solution(alg,y,nothing,cache)
 end
 
 Base.@kwdef struct KLUFactorization <: AbstractFactorization
@@ -104,7 +104,7 @@ function SciMLBase.solve(cache::LinearCache, alg::KLUFactorization)
     end
 
     y = ldiv!(cache.u, cache.cacheval, cache.b)
-    SciMLBase.build_linear_solution(alg,y,nothing)
+    SciMLBase.build_linear_solution(alg,y,nothing,cache)
 end
 
 ## QRFactorization

src/iterative_wrappers.jl

@@ -140,7 +140,7 @@ function SciMLBase.solve(cache::LinearCache, alg::KrylovJL; kwargs...)
         Krylov.solve!(args...; kwargs...)
     end
 
-    return SciMLBase.build_linear_solution(alg,cache.u,nothing)
+    return SciMLBase.build_linear_solution(alg,cache.u,nothing,cache)
 end
 
 ## IterativeSolvers.jl
@@ -235,5 +235,5 @@ function SciMLBase.solve(cache::LinearCache, alg::IterativeSolversJL; kwargs...)
     end
     cache.verbose && println()
 
-    return SciMLBase.build_linear_solution(alg,cache.u,nothing)
+    return SciMLBase.build_linear_solution(alg,cache.u,nothing,cache)
 end

src/pardiso.jl

@@ -94,7 +94,7 @@ function SciMLBase.solve(cache::LinearCache, alg::PardisoJL; kwargs...)
     Pardiso.pardiso(cache.cacheval, u, A, b)
     alg.release_phase !== nothing && Pardiso.set_phase!(cache.cacheval, alg.release_phase)
 
-    return SciMLBase.build_linear_solution(alg,cache.u,nothing)
+    return SciMLBase.build_linear_solution(alg,cache.u,nothing,cache)
 end
 
 export PardisoJL, MKLPardisoFactorize, MKLPardisoIterate