split out logging to zero the derivative

Author: ChrisRackauckas

lib/OrdinaryDiffEqCore/ext/OrdinaryDiffEqCoreMooncakeExt.jl

@@ -5,5 +5,6 @@ using Mooncake: @zero_adjoint, MinimalCtx
 Mooncake.@zero_adjoint Mooncake.MinimalCtx Tuple{typeof(OrdinaryDiffEqCore.ode_determine_initdt), Any, Any, Any, Any, Any, Any, Any, Any, Any}
 Mooncake.@zero_adjoint Mooncake.MinimalCtx Tuple{typeof(OrdinaryDiffEqCore.SciMLBase.check_error), Any}
 Mooncake.@zero_adjoint Mooncake.MinimalCtx Tuple{typeof(OrdinaryDiffEqCore.fixed_t_for_floatingpoint_error!), Any, Any}
+Mooncake.@zero_adjoint Mooncake.MinimalCtx Tuple{typeof(OrdinaryDiffEqCore.final_progress), Any}
 
 end

lib/OrdinaryDiffEqCore/src/integrators/integrator_utils.jl

@@ -158,15 +158,19 @@ function _postamble!(integrator)
         resize!(integrator.sol.k, integrator.saveiter_dense)
     end
     if integrator.opts.progress
-        @logmsg(LogLevel(-1),
-            integrator.opts.progress_name,
-            _id=integrator.opts.progress_id,
-            message=integrator.opts.progress_message(integrator.dt, integrator.u,
-                integrator.p, integrator.t),
-            progress="done")
+
     end
 end
 
+function final_progress(integrator)
+    @logmsg(LogLevel(-1),
+    integrator.opts.progress_name,
+    _id=integrator.opts.progress_id,
+    message=integrator.opts.progress_message(integrator.dt, integrator.u,
+        integrator.p, integrator.t),
+    progress="done")
+end
+
 function solution_endpoint_match_cur_integrator!(integrator)
     if integrator.opts.save_end &&
        (integrator.saveiter == 0 ||