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

added onecold broadcast test

Browse Source
This commit is contained in:
Dhairya Gandhi 2019-02-04 17:57:34 +05:30
parent 30aa814c4d
commit 6654ebfc90
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
test/cuda/cuda.jl
View File

@ -38,6 +38,12 @@ Flux.back!(sum(l))
end end
@testset "onecold gpu" begin
x = rand(Float32, 10, 3) |> gpu;
y = Flux.onehotbatch(1:3, 1:10) |> gpu;
@test_nowarn Flux.onecold(x) .== Flux.onecold(y)
end
if CuArrays.libcudnn != nothing if CuArrays.libcudnn != nothing
@info "Testing Flux/CUDNN" @info "Testing Flux/CUDNN"
include("cudnn.jl") include("cudnn.jl")