Skip to content

Commit cde19ed

Browse files
Update test/cuda/layers.jl
1 parent 82d61a0 commit cde19ed

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
@@ -133,5 +133,5 @@ end
133133
@test_nowarn 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))
136-
@test Array(gs_cpu) gs_gpu
136+
@test gs_cpu Array(gs_gpu)
137137
end

0 commit comments

Comments
 (0)