Skip to content

Commit 70b22f9

Browse files
Fix Runic formatting
Space after comma in a Tuple type, and explicit `return` on six single-expression function bodies (Runic requires the last statement of a function body to be an explicit return). Co-Authored-By: Claude Code <noreply@anthropic.com>
1 parent 9270f95 commit 70b22f9

1 file changed

Lines changed: 7 additions & 7 deletions

File tree

ext/SciMLSensitivityMooncakeExt.jl

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ end
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))
481481
end
482482

483483
function 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))
492492
end
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))
533533
end
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))
554554
end
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()
589589
end
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()
610610
end
611611

612612
end

0 commit comments

Comments
 (0)