Skip to content

Commit 4be8443

Browse files
bors[bot]darsnack
andauthored
Merge #1550
1550: Add test for show(io, ::OneHotArray) on GPU r=darsnack a=darsnack Adds a test for #1534. ### PR Checklist - [x] Tests are added - [ ] ~~Entry in NEWS.md~~ - [ ] ~~Documentation, if applicable~~ - [ ] ~~API changes require approval from a committer (different from the author, if applicable)~~ Co-authored-by: Kyle Daruwalla <daruwalla@wisc.edu>
2 parents 7e00a0e + fc89dbc commit 4be8443

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

test/cuda/cuda.jl

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,11 @@ using LinearAlgebra: I, cholesky, Cholesky
3838

3939
end
4040

41+
@testset "onehot gpu" begin
42+
y = Flux.onehotbatch(ones(3), 1:2) |> gpu;
43+
@test (repr("text/plain", y); true)
44+
end
45+
4146
@testset "onecold gpu" begin
4247
y = Flux.onehotbatch(ones(3), 1:10) |> gpu;
4348
l = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j']

0 commit comments

Comments
 (0)