Skip to content

Commit 0400686

Browse files
fix RNN gpu test
1 parent 3dd5138 commit 0400686

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/cuda/curnn.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ using Flux, CUDA, Test
88
θ = gradient(() -> sum(m(x)), params(m))
99
@test x isa CuArray
1010
@test θ[m.cell.Wi] isa CuArray
11-
@test collect(m̄[].cell.Wi) == collect(θ[m.cell.Wi])
11+
@test collect(m̄.cell.Wi) == collect(θ[m.cell.Wi])
1212
end
1313

1414
@testset "RNN" begin

0 commit comments

Comments
 (0)