fix gpu test

FluxML · Dec 15, 2024 · 4c8cbfc · 4c8cbfc
1 parent 991951c
commit 4c8cbfc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,