Fix TestGpuCumsum

theano/gpuarray/tests/test_extra_ops.py

@@ -28,6 +28,21 @@ class TestGpuCumsum(theano.tensor.tests.test_extra_ops.TestCumsumOp):
             raise SkipTest("Cuda specific tests")
         self.max_threads_dim0 = test_ctx.maxlsize0
         self.max_grid_size1 = test_ctx.maxgsize2
+        self.op_class = GpuCumsum
+
+    def test_infer_shape(self):
+        x = T.tensor3('x')
+        a = np.random.random((3, 5, 2)).astype(theano.config.floatX)
+
+        for axis in range(-len(a.shape), len(a.shape)):
+            self._compile_and_check([x],
+                                    [cumsum(x, axis=axis)],
+                                    [a],
+                                    self.op_class)
+
+    def test_grad(self):
+        # no grad for GpuCumsum
+        pass
 
     def test_Strides1D(self):
         x = T.fvector('x')