Integrate Timer Outputs

Author: avik-pal

Project.toml

@@ -27,6 +27,7 @@ SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
 SparseDiffTools = "47a9eef4-7e08-11e9-0b38-333d64bd3804"
 StaticArraysCore = "1e83bf80-4336-4d27-bf5d-d5a4f845583c"
 SumTypes = "8e1ec7a9-0e02-4297-b0fe-6433085c89f2"
+TimerOutputs = "a759f4b9-e2f1-59dc-863e-4aeb61b1ea8f"
 
 [weakdeps]
 BandedMatrices = "aae01518-5342-5314-be14-df237901396f"

src/NonlinearSolve.jl

@@ -8,21 +8,18 @@ import Reexport: @reexport
 import PrecompileTools: @recompile_invalidations, @compile_workload, @setup_workload
 
 @recompile_invalidations begin
-    using ADTypes, DiffEqBase, LazyArrays, LineSearches, LinearAlgebra, LinearSolve, Printf,
-        SciMLBase, SimpleNonlinearSolve, SparseArrays, SparseDiffTools, SumTypes
+    using ADTypes, ConcreteStructs, DiffEqBase, FastBroadcast, FastClosures, LazyArrays,
+        LineSearches, LinearAlgebra, LinearSolve, MaybeInplace, Printf, SciMLBase,
+        SimpleNonlinearSolve, SparseArrays, SparseDiffTools, SumTypes, TimerOutputs
 
     import ArrayInterface: undefmatrix, can_setindex, restructure, fast_scalar_indexing
-    import ConcreteStructs: @concrete
     import DiffEqBase: AbstractNonlinearTerminationMode,
         AbstractSafeNonlinearTerminationMode, AbstractSafeBestNonlinearTerminationMode,
         NonlinearSafeTerminationReturnCode, get_termination_mode
-    import FastBroadcast: @..
-    import FastClosures: @closure
     import FiniteDiff
     import ForwardDiff
     import ForwardDiff: Dual
     import LinearSolve: ComposePreconditioner, InvPreconditioner, needs_concrete_A
-    import MaybeInplace: @bb
     import RecursiveArrayTools: recursivecopy!, recursivefill!
 
     import SciMLBase: AbstractNonlinearAlgorithm, JacobianWrapper, AbstractNonlinearProblem,
@@ -121,14 +118,14 @@ const False = Val(false)
 # __alg_print_modifiers(_) = String[]
 
 include("abstract_types.jl")
+include("internal/helpers.jl")
 
 include("descent/newton.jl")
 include("descent/steepest.jl")
 include("descent/dogleg.jl")
 include("descent/damped_newton.jl")
 include("descent/geodesic_acceleration.jl")
 
-include("internal/helpers.jl")
 include("internal/operators.jl")
 include("internal/jacobian.jl")
 include("internal/forward_diff.jl")
@@ -140,6 +137,7 @@ include("internal/approx_initialization.jl")
 include("globalization/line_search.jl")
 include("globalization/trust_region.jl")
 
+include("core/generic.jl")
 include("core/approximate_jacobian.jl")
 include("core/generalized_first_order.jl")
 include("core/spectral_methods.jl")

src/abstract_types.jl

@@ -129,16 +129,10 @@ abstract type AbstractNonlinearSolveCache{iip} end
 
 SciMLBase.isinplace(::AbstractNonlinearSolveCache{iip}) where {iip} = iip
 
-import SciMLBase: set_u!
-function get_u end
-function get_fu end
-function set_fu! end
-
-function get_nf end
-function get_njacs end
-function get_nsteps end
-function get_nsolve end
-function get_nfactors end
+get_fu(cache::AbstractNonlinearSolveCache) = cache.fu
+get_u(cache::AbstractNonlinearSolveCache) = cache.u
+set_fu!(cache::AbstractNonlinearSolveCache, fu) = (cache.fu = fu)
+SciMLBase.set_u!(cache::AbstractNonlinearSolveCache, u) = (cache.u = u)
 
 """
     AbstractLinearSolverCache <: Function
@@ -202,3 +196,7 @@ abstract type AbstractTrustRegionMethodCache end
 function last_step_accepted(cache::AbstractTrustRegionMethodCache)
     return cache.last_step_accepted
 end
+
+abstract type AbstractNonlinearSolveJacobianCache{iip} <: Function end
+
+SciMLBase.isinplace(::AbstractNonlinearSolveJacobianCache{iip}) where {iip} = iip