From 5e19e3c2deb14cb3cdc50f9f006456afd3d6425a Mon Sep 17 00:00:00 2001
From: Billy Moses <wmoses@google.com>
Date: Mon, 13 May 2024 21:50:07 -0700
Subject: [PATCH] Mark check_sizes as inactive

---
 src/losses/utils.jl | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/losses/utils.jl b/src/losses/utils.jl
index e42bdfbe2e..52e1907c22 100644
--- a/src/losses/utils.jl
+++ b/src/losses/utils.jl
@@ -36,3 +36,4 @@ end
 _check_sizes(ŷ, y) = nothing  # pass-through, for constant label e.g. y = 1
 
 ChainRulesCore.@non_differentiable _check_sizes(ŷ::Any, y::Any)
+Enzyme.EnzymeRules.inactive_noinl(typeof(_check_sizes), ...) = true