Mark check_sizes as inactive

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