3535# function of u0/p, but its body has a try/catch Mooncake's forward-mode compiler can't
3636# trace. Zero-derivative keeps Mooncake from ever looking inside.
3737@zero_derivative (
38- MinimalCtx, Tuple{typeof (SciMLSensitivity. automatic_sensealg_choice),Vararg},
38+ MinimalCtx, Tuple{typeof (SciMLSensitivity. automatic_sensealg_choice), Vararg},
3939)
4040
4141# Mooncake-native gradient for the DAE/ODE init path. Avoids pulling Zygote
@@ -477,7 +477,7 @@ function rrule!!(
477477 sensealg:: CoDual{<:_MooncakeUnsupportedAnotherADSensealg} ,
478478 :: CoDual , :: CoDual , :: CoDual... ,
479479 )
480- _mooncake_another_ad_backend_unsupported (Mooncake. primal (sensealg))
480+ return _mooncake_another_ad_backend_unsupported (Mooncake. primal (sensealg))
481481end
482482
483483function rrule!! (
@@ -488,7 +488,7 @@ function rrule!!(
488488 sensealg:: CoDual{<:_MooncakeUnsupportedAnotherADSensealg} ,
489489 :: CoDual , :: CoDual , :: CoDual... ,
490490 )
491- _mooncake_another_ad_backend_unsupported (Mooncake. primal (sensealg))
491+ return _mooncake_another_ad_backend_unsupported (Mooncake. primal (sensealg))
492492end
493493
494494# Forward-mode primitive for `DiffEqBase._solve_adjoint` under ReverseDiffAdjoint and
@@ -529,7 +529,7 @@ function frule!!(
529529 :: Dual{<:SciMLBase.MooncakeOriginator} ,
530530 :: Dual... ,
531531 )
532- _mooncake_solve_adjoint_forward_mode_unsupported (Mooncake. primal (sensealg))
532+ return _mooncake_solve_adjoint_forward_mode_unsupported (Mooncake. primal (sensealg))
533533end
534534
535535@is_primitive (
@@ -550,7 +550,7 @@ function frule!!(
550550 :: Dual{<:SciMLBase.MooncakeOriginator} ,
551551 :: Dual... ,
552552 )
553- _mooncake_solve_adjoint_forward_mode_unsupported (Mooncake. primal (sensealg))
553+ return _mooncake_solve_adjoint_forward_mode_unsupported (Mooncake. primal (sensealg))
554554end
555555
556556# ForwardSensitivity's HVP is also unsupported, but not for the ReverseDiff/Tracker
@@ -585,7 +585,7 @@ function frule!!(
585585 :: Dual{<:SciMLBase.MooncakeOriginator} ,
586586 :: Dual... ,
587587 )
588- _mooncake_forward_sensitivity_hvp_unsupported ()
588+ return _mooncake_forward_sensitivity_hvp_unsupported ()
589589end
590590
591591@is_primitive (
@@ -606,7 +606,7 @@ function frule!!(
606606 :: Dual{<:SciMLBase.MooncakeOriginator} ,
607607 :: Dual... ,
608608 )
609- _mooncake_forward_sensitivity_hvp_unsupported ()
609+ return _mooncake_forward_sensitivity_hvp_unsupported ()
610610end
611611
612612end
0 commit comments