Skip to content

Commit b7d692f

Browse files
committed
remove DiffEqBase reference from mooncake extension
1 parent a34686f commit b7d692f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

ext/SciMLBaseMooncakeExt.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,12 +6,12 @@ import Mooncake: rrule!!, CoDual, zero_fcodual, @is_primitive,
66
@from_rrule, @zero_adjoint, @mooncake_overlay, MinimalCtx,
77
NoPullback
88

9-
@zero_adjoint MinimalCtx Tuple{typeof(DiffEqBase.numargs), Any}
9+
@zero_adjoint MinimalCtx Tuple{typeof(SciMLBase.numargs), Any}
1010
@is_primitive MinimalCtx Tuple{
11-
typeof(DiffEqBase.set_mooncakeoriginator_if_mooncake), SciMLBase.ChainRulesOriginator
11+
typeof(SciMLBase.set_mooncakeoriginator_if_mooncake), SciMLBase.ChainRulesOriginator
1212
}
1313

14-
@mooncake_overlay DiffEqBase.set_mooncakeoriginator_if_mooncake(x::SciMLBase.ADOriginator) = SciMLBase.MooncakeOriginator()
14+
@mooncake_overlay SciMLBase.set_mooncakeoriginator_if_mooncake(x::SciMLBase.ADOriginator) = SciMLBase.MooncakeOriginator()
1515

1616
function rrule!!(
1717
f::CoDual{typeof(SciMLBase.set_mooncakeoriginator_if_mooncake)},

0 commit comments

Comments
 (0)