Skip to content

Commit ca283f7

Browse files
Update test/cuda/layers.jl
1 parent 9eac5b4 commit ca283f7

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
@@ -130,7 +130,7 @@ end
130130
b = Flux.Bilinear(10, 2, 3) |> gpu
131131
@test size(b(x,y)) == (3,9)
132132
@test sum(abs2, b(x,y)) 0f0
133-
@test_nowarn gs_gpu = gradient(() -> sum(abs2.(b(x,y))), params(b))
133+
gs_gpu = gradient(() -> sum(abs2.(b(x, y))), params(b))
134134
b_cpu, x_cpu, y_cpu = b |> cpu, x |> cpu, y |> cpu
135135
gs_cpu = gradient(() -> sum(abs2.(b_cpu(x_cpu, y_cpu))), params(b_cpu))
136136
@test gs_cpu Array(gs_gpu)

0 commit comments

Comments
 (0)