From 4ef5ec00057d5247d991be71056814d554a5882d Mon Sep 17 00:00:00 2001 From: Manjunath Bhat Date: Fri, 12 Jul 2019 21:03:57 +0530 Subject: [PATCH] brackets corrected --- test/cuda/cudnn.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/cuda/cudnn.jl b/test/cuda/cudnn.jl index 7aca1208..0ae00814 100644 --- a/test/cuda/cudnn.jl +++ b/test/cuda/cudnn.jl @@ -15,7 +15,7 @@ trainmode(f, x...) = forward(f, x...)[1] @test cpu(data(cy)) ≈ data(y) g = gradient(()->sum(m(x)), params(m)) - cg = gradient(()->sum(cm(cx), params(cm)) + cg = gradient(()->sum(cm(cx)), params(cm)) @test g.grads[m.γ] ≈ cpu(cg.grads[cm.γ]) @test g.grads[m.β] ≈ cpu(cg.grads[cm.β]) @@ -35,7 +35,7 @@ trainmode(f, x...) = forward(f, x...)[1] @test cpu(data(cy)) ≈ data(y) g = gradient(()->sum(m(x)), params(m)) - cg = gradient(()->sum(cm(cx), params(cm)) + cg = gradient(()->sum(cm(cx)), params(cm)) @test g.grads[m.γ] ≈ cpu(cg.grads[cm.γ]) @test g.grads[m.β] ≈ cpu(cg.grads[cm.β])