Merge pull request #39 from SciML/return_type

Add a return type wrapper

Author: ChrisRackauckas

Project.toml

@@ -25,16 +25,14 @@ KrylovKit = "0.5"
 RecursiveFactorization = "0.2"
 Reexport = "1"
 Requires = "1"
-SciMLBase = "1.18.6"
+SciMLBase = "1.22"
 Setfield = "0.7, 0.8"
 UnPack = "1"
 julia = "1.6"
 
 [extras]
-DiffEqProblemLibrary = "a077e3f3-b75c-5d7f-a0c6-6bc4c8ec64a9"
-OrdinaryDiffEq = "1dea7af3-3e70-54e6-95c3-0bf5283fa5ed"
 Pardiso = "46dd5b70-b6fb-5a00-ae2d-e8fea33afaf2"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [targets]
-test = ["Test", "OrdinaryDiffEq", "DiffEqProblemLibrary", "Pardiso"]
+test = ["Test","Pardiso"]

src/common.jl

@@ -66,7 +66,7 @@ function SciMLBase.init(prob::LinearProblem, alg::Union{SciMLLinearSolveAlgorith
                        )
     @unpack A, b, u0, p = prob
 
-    u0 = (u0 !== nothing) ? u0 : zero(b)   
+    u0 = (u0 !== nothing) ? u0 : zero(b)
     Pl = (Pl !== nothing) ? Pl : Identity()
     Pr = (Pr !== nothing) ? Pr : Identity()
 

src/factorization.jl

@@ -4,7 +4,8 @@ function SciMLBase.solve(cache::LinearCache, alg::AbstractFactorization)
         cache = set_cacheval(cache, fact)
     end
 
-    ldiv!(cache.u, cache.cacheval, cache.b)
+    y = ldiv!(cache.u, cache.cacheval, cache.b)
+    SciMLBase.build_linear_solution(alg,y,nothing)
 end
 
 ## LU Factorizations

src/pardiso.jl

@@ -91,5 +91,5 @@ 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 cache.u
+    return SciMLBase.build_linear_solution(alg,cache.u,nothing)
 end

src/wrappers.jl

@@ -46,7 +46,7 @@ function get_preconditioner(Pi, Po)
     ifPi = Pi !== Identity()
     ifPo = Po !== Identity()
 
-    P = 
+    P =
     if ifPi & ifPo
         ComposePreconditioner(Pi, Po)
     elseif ifPi | ifPo
@@ -200,7 +200,7 @@ function SciMLBase.solve(cache::LinearCache, alg::KrylovJL; kwargs...)
         Krylov.solve!(args...; kwargs...)
     end
 
-    return cache.u
+    return SciMLBase.build_linear_solution(alg,cache.u,nothing)
 end
 
 ## IterativeSolvers.jl
@@ -295,6 +295,5 @@ function SciMLBase.solve(cache::LinearCache, alg::IterativeSolversJL; kwargs...)
     end
     cache.verbose && println()
 
-    return cache.u
+    return SciMLBase.build_linear_solution(alg,cache.u,nothing)
 end
-