Skip to content

Commit f36d93c

Browse files
add auto-alg dispatch
1 parent b451452 commit f36d93c

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

ext/DiffEqBaseMooncakeExt.jl

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,19 @@ Mooncake.@from_rrule(
1616
true,
1717
)
1818

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+
Any,
28+
},
29+
true,
30+
)
31+
1932
Mooncake.@zero_adjoint Mooncake.MinimalCtx Tuple{typeof(DiffEqBase.numargs), Any}
2033
Mooncake.@mooncake_overlay DiffEqBase.set_mooncakeoriginator_if_mooncake(x::ADOriginator) = MooncakeOriginator
2134

0 commit comments

Comments
 (0)