TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

assert no scalar indexing for onecold

Browse Source
This commit is contained in:
Dhairya Gandhi 2019-02-09 22:38:49 +05:30
parent 35cd9761a8
commit 1ada9afe81
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
test/cuda/cuda.jl
View File

@ -39,6 +39,7 @@ Flux.back!(sum(l))
end end
@testset "onecold gpu" begin @testset "onecold gpu" begin
CuArrays.allowscalar(false)
y = Flux.onehotbatch(ones(3), 1:10) |> gpu; y = Flux.onehotbatch(ones(3), 1:10) |> gpu;
@test Flux.onecold(y) isa CuArray @test Flux.onecold(y) isa CuArray
@test y[3,:] isa CuArray @test y[3,:] isa CuArray