add MooncakeExt

Author: jClugstor

lib/NonlinearSolveBase/Project.toml

@@ -36,6 +36,7 @@ Enzyme = "7da242da-08ed-463a-9acd-ee780be4f1d9"
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
 LineSearch = "87fe0de2-c867-4266-b59a-2f0a94fc965b"
 LinearSolve = "7ed4a6bd-45f5-4d41-b270-4a48e9bafcae"
+Mooncake = "da2b9cff-9c12-43a0-ae48-6db2b0edb7d6"
 ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267"
 SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
 SparseMatrixColorings = "0a514795-09f3-496d-8182-132a7b665d35"
@@ -48,6 +49,7 @@ NonlinearSolveBaseEnzymeExt = ["ChainRulesCore", "Enzyme"]
 NonlinearSolveBaseForwardDiffExt = "ForwardDiff"
 NonlinearSolveBaseLineSearchExt = "LineSearch"
 NonlinearSolveBaseLinearSolveExt = "LinearSolve"
+NonlinearSolveBaseMooncakeExt = "Mooncake"
 NonlinearSolveBaseReverseDiffExt = "ReverseDiff"
 NonlinearSolveBaseSparseArraysExt = "SparseArrays"
 NonlinearSolveBaseSparseMatrixColoringsExt = "SparseMatrixColorings"
@@ -77,6 +79,7 @@ LinearAlgebra = "1.10"
 LinearSolve = "3.15"
 Markdown = "1.10"
 MaybeInplace = "0.1.4"
+Mooncake = "0.4"
 Preferences = "1.4"
 Printf = "1.10"
 RecursiveArrayTools = "3"

lib/NonlinearSolveBase/src/solve.jl

@@ -794,11 +794,11 @@ end
 function _solve_adjoint(prob, sensealg, u0, p, originator, args...; merge_callbacks = true,
         kwargs...)
     alg = extract_alg(args, kwargs, prob.kwargs)
-    if isnothing(alg) || !(alg isa AbstractNonlinearAlgorithm) # Default algorithm handling
-        _prob = get_concrete_problem(prob, true; u0 = u0,
+    if isnothing(alg) || !(alg isa AbstractDEAlgorithm) # Default algorithm handling
+        _prob = get_concrete_problem(prob, !(prob isa DiscreteProblem); u0 = u0,
             p = p, kwargs...)
     else
-        _prob = get_concrete_problem(prob, true; u0 = u0, p = p, kwargs...)
+        _prob = get_concrete_problem(prob, isadaptive(alg); u0 = u0, p = p, kwargs...)
     end
 
     if has_kwargs(_prob)