Mark check_sizes as inactive

FluxML · May 14, 2024 · 5e19e3c · 5e19e3c
1 parent 546ff16
commit 5e19e3c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/losses/utils.jl b/src/losses/utils.jl
@@ -36,3 +36,4 @@ end
 _check_sizes(ŷ, y) = nothing  # pass-through, for constant label e.g. y = 1
 
 ChainRulesCore.@non_differentiable _check_sizes(ŷ::Any, y::Any)
+Enzyme.EnzymeRules.inactive_noinl(typeof(_check_sizes), ...) = true