Skip tests of average_exc_pad with cudnn v3

theano/sandbox/cuda/tests/test_dnn.py

@@ -235,8 +235,14 @@ def test_pooling():
     if not cuda.dnn.dnn_available():
         raise SkipTest(cuda.dnn.dnn_available.msg)
 
+    # 'average_exc_pad' is disabled for versions < 4004
+    if cuda.dnn.version() < (4004, 4004):
+        modes = ('max', 'average_inc_pad')
+    else:
+        modes = ('max', 'average_inc_pad', 'average_exc_pad')
+
     x = T.ftensor4()
-    for mode, pad in product(('max', 'average_inc_pad', 'average_exc_pad'),
+    for mode, pad in product(modes,
                              ((0, 0), (1, 0), (1, 0), (2, 3), (3, 2))):
         if mode == 'max':
             func = T.max
@@ -346,9 +352,15 @@ def test_pooling3d():
     if not cuda.dnn.dnn_available() or cuda.dnn.version() < (3000, 3000):
         raise SkipTest(cuda.dnn.dnn_available.msg)
 
+    # 'average_exc_pad' is disabled for versions < 4004
+    if cuda.dnn.version() < (4004, 4004):
+        modes = ('max', 'average_inc_pad')
+    else:
+        modes = ('max', 'average_inc_pad', 'average_exc_pad')
+
     x = T.TensorType(broadcastable=(False, False, False, False, False),
                      dtype='float32')()
-    for mode, pad in product(('max', 'average_inc_pad', 'average_exc_pad'),
+    for mode, pad in product(modes,
                              ((0, 0, 0), (1, 0, 0), (0, 1, 0), (0, 0, 1),
                               (2, 3, 2), (3, 2, 2), (2, 2, 3))):
         if mode == 'max':
@@ -940,10 +952,17 @@ class TestDnnInferShapes(utt.InferShapeTester):
             numpy.random.rand(2, 3, 4, 5),
             dtype='float32'
         )
+
+        # 'average_exc_pad' is disabled for versions < 4004
+        if cuda.dnn.version() < (4004, 4004):
+            modes = ['max', 'average_inc_pad']
+        else:
+            modes = ['max', 'average_inc_pad', 'average_exc_pad']
+
         for params in product(
             [(1, 1), (2, 2), (3, 3)],
             [(1, 1), (2, 2), (3, 3)],
-            ['max', 'average_inc_pad', 'average_exc_pad']
+            modes
         ):
             desc = dnn.GpuDnnPoolDesc(
                 ws=params[0],

theano/sandbox/gpuarray/tests/test_dnn.py

@@ -166,8 +166,14 @@ def test_pooling():
     if not dnn.dnn_available(test_ctx_name):
         raise SkipTest(dnn.dnn_available.msg)
 
+    # 'average_exc_pad' is disabled for versions < 4004
+    if dnn.version() < 4004:
+        modes = ('max', 'average_inc_pad')
+    else:
+        modes = ('max', 'average_inc_pad', 'average_exc_pad')
+
     x = T.ftensor4()
-    for mode, pad in product(('max', 'average_inc_pad', 'average_exc_pad'),
+    for mode, pad in product(modes,
                              ((0, 0), (1, 0), (1, 0), (2, 3), (3, 2))):
         if mode == 'max':
             func = T.max
@@ -506,10 +512,17 @@ class TestDnnInferShapes(utt.InferShapeTester):
             numpy.random.rand(2, 3, 4, 5),
             dtype='float32'
         )
+
+        # 'average_exc_pad' is disabled for versions < 4004
+        if dnn.version() < 4004:
+            modes = ['max', 'average_inc_pad']
+        else:
+            modes = ['max', 'average_inc_pad', 'average_exc_pad']
+
         for params in product(
             [(1, 1), (2, 2), (3, 3)],
             [(1, 1), (2, 2), (3, 3)],
-            ['max', 'average_inc_pad', 'average_exc_pad']
+            modes
         ):
             desc = dnn.GpuDnnPoolDesc(
                 ws=params[0],