crash fix. Fix gh-4929.

theano/gpuarray/dnn.py

@@ -1282,6 +1282,11 @@ def dnn_pool(img, ws, stride=(1, 1), mode='max', pad=(0, 0)):
 
     """
     img = gpu_contiguous(img)
+    if mode == "sum":
+        ret = GpuDnnPool(mode="average_inc_pad")(img, ws, stride, pad)
+        context_name = ret.type.context_name
+        window_elem = numpy.prod(ws).astype(ret.dtype)
+        return as_gpuarray_variable(ret * window_elem, context_name)
     return GpuDnnPool(mode=mode)(img, ws, stride, pad)
 
 

theano/gpuarray/tests/test_dnn.py

@@ -356,6 +356,18 @@ def test_pooling_opt():
 
     f(numpy.zeros((10, 10), dtype='float32'))
 
+    # Test sum pooling
+    f = theano.function(
+        [x],
+        pool_2d(x, ds=(2, 3), mode='sum',
+                ignore_border=True),
+        mode=mode_with_gpu)
+
+    assert any([isinstance(n.op, dnn.GpuDnnPool)
+                for n in f.maker.fgraph.toposort()])
+    data = numpy.random.rand(10, 10).astype('float32')
+    f(data)
+
 
 def test_dnn_tag():
     """

theano/sandbox/cuda/dnn.py

@@ -2005,6 +2005,11 @@ def dnn_pool(img, ws, stride=(1, 1), mode='max', pad=(0, 0)):
 
     """
     img = gpu_contiguous(img)
+    if mode == "sum":
+        ret = GpuDnnPool(mode="average_inc_pad")(img, ws, stride, pad)
+        window_elem = numpy.prod(ws).astype(ret.dtype)
+        return as_cuda_ndarray_variable(ret * window_elem)
+
     return GpuDnnPool(mode=mode)(img, ws, stride, pad)
 
 

theano/sandbox/cuda/tests/test_dnn.py

@@ -545,6 +545,17 @@ def test_pooling_opt():
                 for n in f.maker.fgraph.toposort()])
 
     f(numpy.zeros((10, 10), dtype='float32'))
+    # Test sum pooling
+    f = theano.function(
+        [x],
+        pool_2d(x, ds=(2, 3), mode='sum',
+                ignore_border=True),
+        mode=mode_with_gpu)
+
+    assert any([isinstance(n.op, dnn.GpuDnnPool)
+                for n in f.maker.fgraph.toposort()])
+    data = numpy.random.rand(10, 10).astype('float32')
+    f(data)
 
 
 class test_DnnSoftMax(test_nnet.test_SoftMax):