Use WOperator when jac_prototype isa AbstractDiffEqLinearOperator

SciML · Oct 25, 2019 · add5e24 · add5e24
1 parent ff5d989
commit add5e24
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/derivative_utils.jl b/src/derivative_utils.jl
@@ -492,7 +492,7 @@ function build_J_W(alg,u,uprev,p,t,dt,f,uEltypeNoUnits,::Val{true})
                                     f.jac_prototype isa SparseMatrixCSC
     J = similar(f.jac_prototype)
     W = similar(J)
-  elseif DiffEqBase.has_jac(f) && !DiffEqBase.has_Wfact(f) && f.jac_prototype !== nothing
+  elseif f.jac_prototype isa DiffEqBase.AbstractDiffEqLinearOperator
     W = WOperator(f, dt, true)
     J = W.J
   else
@@ -514,7 +514,7 @@ function build_J_W(alg,u,uprev,p,t,dt,f,uEltypeNoUnits,::Val{false})
       J = DiffEqArrayOperator(J)
     end
     W = WOperator(f.mass_matrix, dt, J, false)
-  elseif DiffEqBase.has_jac(f) && !DiffEqBase.has_Wfact(f) && f.jac_prototype !== nothing
+  elseif f.jac_prototype isa DiffEqBase.AbstractDiffEqLinearOperator
     W = WOperator(f, dt, true)
     J = W.J
   # https://github.com/JuliaDiffEq/OrdinaryDiffEq.jl/pull/672