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

Update test

Browse Source
This commit is contained in:
Avik Pal 2018-08-03 19:32:21 +05:30
parent 6a41f823c8
commit 3f6c065523
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/cudnn.jl
View File

@ -14,7 +14,7 @@ using Flux.Tracker: TrackedArray, data
@test cpu(data(cy)) ≈ data(y) @test cpu(data(cy)) ≈ data(y)
g = ones(size(y)...) g = rand(size(y))
Flux.back!(y, g) Flux.back!(y, g)
Flux.back!(cy, gpu(g)) Flux.back!(cy, gpu(g))