From e5e0896ee9f8b3dec752b4719049bac95a672671 Mon Sep 17 00:00:00 2001 From: Carlo Lucibello Date: Thu, 12 Dec 2024 18:24:55 +0100 Subject: [PATCH] fix gpu test --- test/ext_common/recurrent_gpu_ad.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/ext_common/recurrent_gpu_ad.jl b/test/ext_common/recurrent_gpu_ad.jl index 59f40242d9..35fa983806 100644 --- a/test/ext_common/recurrent_gpu_ad.jl +++ b/test/ext_common/recurrent_gpu_ad.jl @@ -117,7 +117,7 @@ end x = [randn(Float32, d_in, batch_size) for _ in 1:len] h = zeros(Float32, d_out) @test test_gradients(r, x[1], h; test_gpu=true, - compare_finite_diff=false + compare_finite_diff=false, loss=cell_loss) broken = :gruv3cell_single ∈ BROKEN_TESTS @test test_gradients(r, x, h; test_gpu=true, compare_finite_diff=false,