test: minor test fixes

LuxDL · Sep 6, 2024 · c548e64 · c548e64
1 parent 5afdd5a
commit c548e64
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/test/autodiff/nested_autodiff_tests.jl b/test/autodiff/nested_autodiff_tests.jl
@@ -29,7 +29,7 @@ function test_nested_ad_input_gradient_jacobian(aType, dev, ongpu, loss_fn, X, m
 
     allow_unstable() do
         test_gradients((x, ps) -> loss_fn(model, x, ps, st), X, ps;
-            atol=1.0f-3, rtol=1.0f-1, soft_fail=[AutoForwardDiff()],
+            atol=1.0f-2, rtol=1.0f-1, soft_fail=[AutoForwardDiff()],
             skip_backends=[AutoReverseDiff(), AutoTracker(), AutoEnzyme()])
     end
 end

diff --git a/test/layers/pooling_tests.jl b/test/layers/pooling_tests.jl
@@ -11,7 +11,7 @@
                 continue
             end
 
-            broken_backends = ltype == :LPPool ? [AutoTracker()] : []
+            broken_backends = ltype == :LPPool ? [AutoTracker(), AutoEnzyme()] : []
 
             adaptive_ltype = Symbol(:Adaptive, ltype)
             global_ltype = Symbol(:Global, ltype)