Merge pull request #124 from MSeeker1340/jacobian

Hotfix for #123
SciML · Jul 18, 2018 · a29e55c · a29e55c
2 parents aa16264 + 93bc0e5
commit a29e55c
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/diffeqfunction.jl b/src/diffeqfunction.jl
@@ -148,7 +148,11 @@ function ODEFunction{iip,true}(f;
                  paramjac = nothing,
                  syms = nothing) where iip
                  if jac == nothing && isa(jac_prototype, AbstractDiffEqLinearOperator)
-                   jac = update_coefficients!
+                  if iip
+                    jac = update_coefficients! #(J,u,p,t)
+                  else
+                    jac = (u,p,t) -> update_coefficients!(deepcopy(jac_prototype),u,p,t)
+                  end
                  end
                  ODEFunction{iip,typeof(f),typeof(analytic),typeof(tgrad),
                  typeof(jac),typeof(jac_prototype),typeof(invW),typeof(invW_t),
@@ -166,7 +170,11 @@ function ODEFunction{iip,false}(f;
                  paramjac = nothing,
                  syms = nothing) where iip
                  if jac == nothing && isa(jac_prototype, AbstractDiffEqLinearOperator)
-                   jac = update_coefficients!
+                  if iip
+                    jac = update_coefficients! #(J,u,p,t)
+                  else
+                    jac = (u,p,t) -> update_coefficients!(deepcopy(jac_prototype),u,p,t)
+                  end
                  end
                  ODEFunction{iip,Any,Any,Any,
                  Any,Any,Any,Any,