We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b451452 commit f36d93cCopy full SHA for f36d93c
ext/DiffEqBaseMooncakeExt.jl
@@ -16,6 +16,19 @@ Mooncake.@from_rrule(
16
true,
17
)
18
19
+# Dispatch for auto-alg
20
+Mooncake.@from_rrule(
21
+ Mooncake.MinimalCtx,
22
+ Tuple{
23
+ typeof(DiffEqBase.solve_up),
24
+ DiffEqBase.AbstractDEProblem,
25
+ Union{Nothing, DiffEqBase.AbstractSensitivityAlgorithm},
26
+ Any,
27
28
+ },
29
+ true,
30
+ )
31
+
32
Mooncake.@zero_adjoint Mooncake.MinimalCtx Tuple{typeof(DiffEqBase.numargs), Any}
33
Mooncake.@mooncake_overlay DiffEqBase.set_mooncakeoriginator_if_mooncake(x::ADOriginator) = MooncakeOriginator
34
0 commit comments