From f7ff756060cb53941c1df54fbe1a537a615a98df Mon Sep 17 00:00:00 2001 From: Sergey Zagoruyko Date: Sun, 22 May 2016 22:19:14 +0200 Subject: add avg pooling back compat test --- test/test.lua | 2 ++ 1 file changed, 2 insertions(+) (limited to 'test') diff --git a/test/test.lua b/test/test.lua index ba1b5a1..3d2521e 100644 --- a/test/test.lua +++ b/test/test.lua @@ -820,6 +820,8 @@ function cudnntest.SpatialAveragePooling_single() local sconv = nn.SpatialAveragePooling(ki,kj,si,sj):cuda() local gconv = cudnn.SpatialAveragePooling(ki,kj,si,sj):cuda() + mytester:assert(cudnn.C.CUDNN_POOLING_AVERAGE ~= nil, 'back-compat broken') + local function test(sconv, gconv) local groundtruth = sconv:forward(input):clone() local groundgrad = sconv:backward(input, gradOutput) -- cgit v1.2.3