up

theano/sandbox/cuda/dnn.py

@@ -1442,6 +1442,13 @@ if (pool%(name)s != NULL) { cudnnDestroyPoolingDescriptor(pool%(name)s); }
 """ % dict(name=name)
 
     def c_code(self, node, name, inputs, outputs, sub):
+        if len(inputs) == 2:
+            warnings.warn("Theano GPUDnnPoolGrad internal changed.", stacklevel=3)
+            desc = inputs[1]
+            ws = desc.ws
+            stride = desc.stride
+            pad = desc.pad
+        else:
             ws = inputs[1]
             stride = inputs[2]
             pad = inputs[3]
@@ -1668,7 +1675,15 @@ if (pool%(name)s != NULL) { cudnnDestroyPoolingDescriptor(pool%(name)s); }
         # Here the name out and inp are based on the cudnn definition.
         # Not the definition of this class.
         # This make it complicated.
+        if len(inputs) == 4:
+            warnings.warn("Theano GPUDnnPoolGrad internal changed.", stacklevel=3)
+            out, inp, inp_grad, desc = inputs
+            ws = desc.ws
+            stride = desc.stride
+            pad = desc.pad
+        else:
             out, inp, inp_grad, ws, stride, pad = inputs
+            
         out_grad, = outputs
 
         if self.mode == 'max':

theano/sandbox/cuda/tests/test_dnn.py

@@ -372,6 +372,7 @@ def test_pooling_with_tensor_vars():
     out2 = pool_2d_i2n(x, ds=(2, 2), strides=(1, 1),
                        pad=(0, 0),
                        pool_function=T.max)
+    
     mode_without_gpu2 = mode_without_gpu.including()
     mode_without_gpu2.check_isfinite = False