fix gpu test

FluxML · Dec 12, 2024 · e5e0896 · e5e0896
1 parent 74f1e08
commit e5e0896
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,