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

harder test

Browse Source
This commit is contained in:
Mike J Innes 2018-03-01 16:37:52 +00:00
parent 8019f789f8
commit cf7dd34767
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

@ -15,7 +15,7 @@ x = Flux.onehotbatch([1, 2, 3], 1:3)
cx = gpu(x) cx = gpu(x)
@test cx isa Flux.OneHotMatrix && cx.data isa CuArray @test cx isa Flux.OneHotMatrix && cx.data isa CuArray
m = Chain(Dense(10, 5, σ), Dense(5, 2)) m = Chain(Dense(10, 5, tanh), Dense(5, 2), softmax)
cm = gpu(m) cm = gpu(m)
@test all(p isa TrackedArray && p.data isa CuArray for p in params(cm)) @test all(p isa TrackedArray && p.data isa CuArray for p in params(cm))