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

remove duplicate allowscalar call

Browse Source
This commit is contained in:
Dhairya Gandhi 2019-02-11 20:32:23 +05:30
parent 1ada9afe81
commit d16ef75b1c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/cuda/cuda.jl
View File

@ -39,9 +39,9 @@ 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[:,:] isa Flux.OneHotMatrix{<:CuArray}
@test y[3,:] isa CuArray @test y[3,:] isa CuArray
end end