diff options
author | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-05-22 23:19:14 +0300 |
---|---|---|
committer | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-05-22 23:19:14 +0300 |
commit | f7ff756060cb53941c1df54fbe1a537a615a98df (patch) | |
tree | 92e3f8ec06feae29651eec892ac46ed6782feffc | |
parent | 92d7e316a254902fc658895781c549603b849127 (diff) |
add avg pooling back compat test
-rw-r--r-- | test/test.lua | 2 |
1 files changed, 2 insertions, 0 deletions
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) |