Fix GPU test

2018-07-17 09:53:39 +05:30 · 2018-07-17 09:53:39 +05:30 · 8874d9cccd
parent da7fe93b31
commit 8874d9cccd
1 changed files with 1 additions and 1 deletions
--- a/test/cuda/cudnn.jl
+++ b/test/cuda/cudnn.jl
@ -16,7 +16,7 @@ using Flux.Tracker: TrackedArray, data

    g = ones(size(y)...)
    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)