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

better tests

Browse Source
This commit is contained in:
Dhairya Gandhi 2019-02-04 18:43:25 +05:30
parent 6654ebfc90
commit 2f916f9763
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
test/cuda/cuda.jl
View File

@ -39,9 +39,10 @@ Flux.back!(sum(l))
end end
@testset "onecold gpu" begin @testset "onecold gpu" begin
x = rand(Float32, 10, 3) |> gpu; x = zeros(Float32, 10, 3) |> gpu;
y = Flux.onehotbatch(1:3, 1:10) |> gpu; y = Flux.onehotbatch(ones(3), 1:10) |> gpu;
@test_nowarn Flux.onecold(x) .== Flux.onecold(y) res = Flux.onecold(x) .== Flux.onecold(y)
@test res isa CuArray
end end
if CuArrays.libcudnn != nothing if CuArrays.libcudnn != nothing