Skip to content

Commit 644bd1a

Browse files
committed
Merge branch 'master' of https://github.com/racinmat/Flux.jl
2 parents 2507126 + 931d409 commit 644bd1a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/cuda/cuda.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ end
4343
y = Flux.onehotbatch(ones(3), 1:2) |> gpu;
4444
@test (repr("text/plain", y); true)
4545

46-
gA = rand(3, 3) |> gpu;
46+
gA = rand(3, 2) |> gpu;
4747
@test gradient(A -> sum(A * y), gA)[1] isa CuArray
4848
end
4949

0 commit comments

Comments
 (0)