WIP: Mooncake Direct Adjoints

Author: ChrisRackauckas

lib/OrdinaryDiffEqCore/Project.toml

@@ -39,9 +39,11 @@ TruncatedStacktraces = "781d530d-4396-4725-bb49-402e4bee1e77"
 
 [weakdeps]
 EnzymeCore = "f151be2c-9106-41f4-ab19-57ee4f262869"
+Mooncake = "da2b9cff-9c12-43a0-ae48-6db2b0edb7d6"
 
 [extensions]
 OrdinaryDiffEqCoreEnzymeCoreExt = "EnzymeCore"
+OrdinaryDiffEqCoreMooncakeExt = "Mooncake"
 
 [compat]
 ADTypes = "0.2, 1"
@@ -63,6 +65,7 @@ InteractiveUtils = "1.9"
 LinearAlgebra = "1.9"
 Logging = "1.9"
 MacroTools = "0.5"
+Mooncake = "0.4"
 MuladdMacro = "0.2.1"
 Polyester = "0.7"
 PrecompileTools = "1"

lib/OrdinaryDiffEqCore/ext/OrdinaryDiffEqCoreMooncakeExt.jl

@@ -0,0 +1,7 @@
+module OrdinaryDiffEqCoreMooncakeExt
+
+using OrdinaryDiffEqCore, Mooncake
+using Mooncake: @zero_adjoint, MinimalCtx
+@zero_adjoint MinimalCtx Tuple{typeof(OrdinaryDiffEqCore.ode_determine_initdt), Any, Any, Any, Any, Any, Any, Any, Any, Any}
+
+end