Skip to content

Commit 51c7ccf

Browse files
Update test/cuda/layers.jl
Co-authored-by: Dhairya Gandhi <dhairya@juliacomputing.com>
1 parent 74d72a2 commit 51c7ccf

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/cuda/layers.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -291,6 +291,6 @@ end
291291
@test Array(y_g) == y
292292
gs = gradient(() -> sum(m(x)), params(m))
293293
gs_g = gradient(() -> sum(m_g(x_g)), params(m_g))
294-
@test collect(gs_g[m_g.weight]) gs[m.weight]
294+
@test Array(gs_g[m_g.weight]) gs[m.weight]
295295
end
296296
end

0 commit comments

Comments
 (0)