Merge pull request #1169 from AstitvaAggarwal/master

ChrisRackauckas · web-flow · commit 367b691bdee2 · 2025-07-09T11:47:32.000-04:00
Mooncake overlay for erroring out before Mooncake.DerivedRule construction
diff --git a/ext/DiffEqBaseMooncakeExt.jl b/ext/DiffEqBaseMooncakeExt.jl
@@ -34,6 +34,8 @@ import Mooncake: rrule!!, CoDual, zero_fcodual, @is_primitive,
     typeof(DiffEqBase.set_mooncakeoriginator_if_mooncake), SciMLBase.ChainRulesOriginator
 }
 
+@mooncake_overlay DiffEqBase.set_mooncakeoriginator_if_mooncake(x::SciMLBase.ADOriginator) = SciMLBase.MooncakeOriginator()
+
 function rrule!!(
         f::CoDual{typeof(DiffEqBase.set_mooncakeoriginator_if_mooncake)},
         X::CoDual{SciMLBase.ChainRulesOriginator}

Original file line number	Diff line number	Diff line change
`@@ -34,6 +34,8 @@ import Mooncake: rrule!!, CoDual, zero_fcodual, @is_primitive,`
`34`	`34`	`typeof(DiffEqBase.set_mooncakeoriginator_if_mooncake), SciMLBase.ChainRulesOriginator`
`35`	`35`	`}`
`36`	`36`
	`37`	`+@mooncake_overlay DiffEqBase.set_mooncakeoriginator_if_mooncake(x::SciMLBase.ADOriginator) = SciMLBase.MooncakeOriginator()`
	`38`	`+`
`37`	`39`	`function rrule!!(`
`38`	`40`	`f::CoDual{typeof(DiffEqBase.set_mooncakeoriginator_if_mooncake)},`
`39`	`41`	`X::CoDual{SciMLBase.ChainRulesOriginator}`