From ca2c261083ca8c68de9a9a95a0e672f8129d268f Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Thu, 21 Nov 2024 14:01:13 -0500 Subject: [PATCH] chore: apply suggestions from code review Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com> --- lib/LuxLib/test/normalization/batchnorm_tests.jl | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/lib/LuxLib/test/normalization/batchnorm_tests.jl b/lib/LuxLib/test/normalization/batchnorm_tests.jl index 2ad299d796..95b2fb30ce 100644 --- a/lib/LuxLib/test/normalization/batchnorm_tests.jl +++ b/lib/LuxLib/test/normalization/batchnorm_tests.jl @@ -145,7 +145,8 @@ end @test nt.running_var isa aType && length(nt.running_var) == 6 @test_gradients(sumabs2first, batchnorm, x, scale, bias, Constant(running_mean), - Constant(running_var), Val(true), gelu, 0.9, 1e-5; atol=1.0f-3, rtol=1.0f-3, + Constant(running_var), Val(true), gelu, 0.9, 1e-5; atol=1.0f-3, + rtol=1.0f-3, broken_backends=[AutoEnzyme()]) end end