Counter for jacobians

Author: avik-pal

src/jacobian.jl

@@ -34,6 +34,7 @@ jacobian!!(J, _) = J
 # and we don't want wasteful `copyto!`
 function jacobian!!(J::Union{AbstractMatrix{<:Number}, Nothing}, cache)
     @unpack f, uf, u, p, jac_cache, alg, fu_cache = cache
+    cache.stats.njacs += 1
     iip = isinplace(cache)
     if iip
         if has_jac(f)
@@ -53,8 +54,10 @@ function jacobian!!(J::Union{AbstractMatrix{<:Number}, Nothing}, cache)
     end
 end
 # Scalar case
-jacobian!!(::Number, cache) = last(value_derivative(cache.uf, cache.u))
-
+function jacobian!!(::Number, cache)
+    cache.stats.njacs += 1
+    return last(value_derivative(cache.uf, cache.u))
+end
 # Build Jacobian Caches
 function jacobian_caches(alg::AbstractNonlinearSolveAlgorithm, f::F, u, p, ::Val{iip};
         linsolve_kwargs = (;), lininit::Val{linsolve_init} = Val(true),