add test and run

Author: ChrisRackauckas

src/pardiso.jl

@@ -68,15 +68,13 @@ end
 
 function SciMLBase.solve(cache::LinearCache, alg::PardisoJL; kwargs...)
     @unpack A, b, u = cache
-    A = copy(convert(AbstractMatrix,A))
-
-    #if cache.isfresh
-    #    Pardiso.set_phase!(cache.cacheval, Pardiso.NUM_FACT)
-    #    Pardiso.pardiso(cache.cacheval, u, A, b)
-    #end
-    #Pardiso.set_phase!(cache.cacheval, Pardiso.SOLVE_ITERATIVE_REFINE)
+    A = convert(AbstractMatrix,A)
 
-    Pardiso.set_phase!(cache.cacheval, Pardiso.ANALYSIS_NUM_FACT_SOLVE_REFINE)
+    if cache.isfresh
+        Pardiso.set_phase!(cache.cacheval, Pardiso.NUM_FACT)
+        Pardiso.pardiso(cache.cacheval, u, A, b)
+    end
+    Pardiso.set_phase!(cache.cacheval, Pardiso.SOLVE_ITERATIVE_REFINE)
     Pardiso.pardiso(cache.cacheval, u, A, b)
 
     return SciMLBase.build_linear_solution(alg,cache.u,nothing,cache)

test/basictests.jl

@@ -1,5 +1,6 @@
 using LinearSolve, LinearAlgebra, SparseArrays
 using Test
+import Random
 
 n = 8
 A = Matrix(I,n,n)
@@ -232,4 +233,41 @@ end
     end
 end
 
+@testset "Sparse Precaching" begin
+    n = 4
+    Random.seed!(10)
+    A = sprand(n,n,0.8); A2 = 2.0 .* A
+    b1 = rand(n); b2 = rand(n)
+
+    prob = LinearProblem(copy(A), copy(b1))
+    linsolve = init(prob,UMFPACKFactorization())
+    sol11 = solve(linsolve)
+    linsolve = LinearSolve.set_b(sol1.cache,copy(b2))
+    sol12 = solve(linsolve)
+    linsolve = LinearSolve.set_A(sol2.cache,copy(A2))
+    sol13 = solve(linsolve)
+
+    prob = LinearProblem(copy(A), copy(b1))
+    linsolve = init(prob,KLUFactorization())
+    sol21 = solve(linsolve)
+    linsolve = LinearSolve.set_b(sol1.cache,copy(b2))
+    sol22 = solve(linsolve)
+    linsolve = LinearSolve.set_A(sol2.cache,copy(A2))
+    sol23 = solve(linsolve)
+
+    linsolve = init(prob,MKLPardisoFactorize())
+    sol31 = solve(linsolve)
+    linsolve = LinearSolve.set_b(sol1.cache,copy(b2))
+    sol32 = solve(linsolve)
+    linsolve = LinearSolve.set_A(sol2.cache,copy(A2))
+    sol33 = solve(linsolve)
+
+    @test sol11.u ≈ sol21.u
+    @test sol11.u ≈ sol31.u
+    @test sol12.u ≈ sol22.u
+    @test sol12.u ≈ sol32.u
+    @test sol13.u ≈ sol23.u
+    @test sol13.u ≈ sol33.u
+end
+
 end # testset