Update test/cuda/layers.jl

Co-authored-by: Dhairya Gandhi <[email protected]>
FluxML · Jul 13, 2021 · 51c7ccf · 51c7ccf
1 parent 74d72a2
commit 51c7ccf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/cuda/layers.jl b/test/cuda/layers.jl
@@ -291,6 +291,6 @@ end
     @test Array(y_g) == y
     gs = gradient(() -> sum(m(x)), params(m))
     gs_g = gradient(() -> sum(m_g(x_g)), params(m_g))
-    @test collect(gs_g[m_g.weight]) ≈ gs[m.weight]
+    @test Array(gs_g[m_g.weight]) ≈ gs[m.weight]
   end
 end