Update test/cuda/layers.jl

Author: CarloLucibello

test/cuda/layers.jl

@@ -131,6 +131,7 @@ end
   @test size(b(x,y)) == (3,9)
   @test sum(abs2, b(x,y)) ≈ 0f0
   @test_nowarn gs_gpu = gradient(() -> sum(abs2.(b(x,y))), params(b))
-  gs_cpu = gradient(() -> sum(abs2.(b(x |> cpu,y |> cpu) |> cpu)), params(b))
+  b_cpu, x_cpu, y_cpu = b |> cpu, x |> cpu, y |> cpu
+  gs_cpu = gradient(() -> sum(abs2.(b_cpu(x_cpu, y_cpu)), params(b_cpu))
   @test Array(gs_cpu) ≈ gs_gpu
 end