Add more conditions for jac_prototype

SciML · Oct 22, 2019 · 7532bff · 7532bff
1 parent ac5d76f
commit 7532bff
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/derivative_utils.jl b/src/derivative_utils.jl
@@ -486,7 +486,8 @@ function build_J_W(alg,u,uprev,p,t,dt,f,uEltypeNoUnits,::Val{true})
       return nothing, nothing
     end
   end
-  if ArrayInterface.isstructured(f.jac_prototype) || f.jac_prototype isa SparseMatrixCSC
+  n = length(u)
+  if size(f.jac_prototype) != (n, n) || ArrayInterface.isstructured(f.jac_prototype) || 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