fix RNN gpu test

FluxML · Oct 16, 2021 · 0400686 · 0400686
1 parent 3dd5138
commit 0400686
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/cuda/curnn.jl b/test/cuda/curnn.jl
@@ -8,7 +8,7 @@ using Flux, CUDA, Test
   θ = gradient(() -> sum(m(x)), params(m))
   @test x isa CuArray
   @test θ[m.cell.Wi] isa CuArray
-  @test collect(m̄[].cell.Wi) == collect(θ[m.cell.Wi])
+  @test collect(m̄.cell.Wi) == collect(θ[m.cell.Wi])
 end
 
 @testset "RNN" begin