Skip to content

Commit 67d016b

Browse files
Update test/cuda/layers.jl
1 parent 58da113 commit 67d016b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/cuda/layers.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -132,5 +132,5 @@ end
132132
@test sum(abs2, b(x,y)) 0f0
133133
@test_nowarn gs_gpu = gradient(() -> sum(abs2.(b(x,y))), params(b))
134134
gs_cpu = gradient(() -> sum(abs2.(b(x |> cpu,y |> cpu) |> cpu)), params(b))
135-
@test gs_cpu gs_gpu
136-
end
135+
@test Array(gs_cpu) gs_gpu
136+
end

0 commit comments

Comments
 (0)