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

Fix GPU test

Browse Source
This commit is contained in:
Avik Pal 2018-07-17 09:53:39 +05:30
parent da7fe93b31
commit 8874d9cccd
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

@ -16,7 +16,7 @@ using Flux.Tracker: TrackedArray, data
g = ones(size(y)...) g = ones(size(y)...)
Flux.back!(y, g) Flux.back!(y, g)
Flux.back!(cy, gpu(g))) Flux.back!(cy, gpu(g))
@test m.γ.grad ≈ cpu(cm.γ.grad) @test m.γ.grad ≈ cpu(cm.γ.grad)
@test m.β.grad ≈ cpu(cm.β.grad) @test m.β.grad ≈ cpu(cm.β.grad)