Fix the tests for GpuMaxAndArgmax.

theano/gpuarray/dnn_redux.c

@@ -83,7 +83,7 @@ int APPLY_SPECIFIC(dnn_redux)(PyGpuArrayObject *input,
   if (indices != NULL) {
     if (theano_prep_output(indices, p, dims, GA_UINT, GA_C_ORDER, c) != 0)
       return 1;
-    indsize = PyGpuArray_SIZE(*indices);
+    indsize = PyGpuArray_SIZE(*indices) * 4;
   }
 
   if (p == input->ga.nd || rsz == 1) {

theano/gpuarray/tests/test_reduction.py

@@ -10,6 +10,8 @@ from theano.tests.unittest_tools import SkipTest
 from .config import mode_with_gpu, mode_without_gpu
 from .test_basic_ops import rand_gpuarray
 from .. import GpuArrayType
+from ..reduction import GpuMaxAndArgmax
+from ..dnn import GpuDnnReduction
 
 import math
 
@@ -54,13 +56,13 @@ def numpy_maxandargmax(X, axis=None):
 
 
 def check_if_gpu_maxandargmax_in_graph(theano_function):
-    assert len([node for node in theano_function.maker.fgraph.apply_nodes
-                if isinstance(node.op, theano.gpuarray.reduction.GpuMaxAndArgmax)]) > 0
+    assert any(isinstance(node.op, (GpuMaxAndArgmax, GpuDnnReduction))
+               for node in theano_function.maker.fgraph.apply_nodes)
 
 
 def check_if_gpu_maxandargmax_not_in_graph(theano_function):
-    assert len([node for node in theano_function.maker.fgraph.apply_nodes
-                if isinstance(node.op, theano.gpuarray.reduction.GpuMaxAndArgmax)]) == 0
+    assert all(not isinstance(node.op, (GpuMaxAndArgmax, GpuDnnReduction))
+               for node in theano_function.maker.fgraph.apply_nodes)
 
 
 class BaseTest: