add test for GPUCorrMM

theano/gpuarray/tests/test_gemmcorr.py

@@ -11,6 +11,7 @@ from theano.tensor.nnet.corr import CorrMM, CorrMM_gradWeights, CorrMM_gradInput
 from ..type import gpuarray_shared_constructor
 from ..blas import GpuCorrMM, GpuCorrMM_gradWeights, GpuCorrMM_gradInputs
 from .config import mode_with_gpu, mode_without_gpu, ref_cast
+from theano.tensor.nnet.tests.test_abstract_conv import Grouped_conv_noOptim
 
 
 class TestCorrMM(unittest.TestCase):
@@ -219,3 +220,11 @@ class TestCorrMM(unittest.TestCase):
                             verify_grad=False)
         self.run_gradinput(inputs_shape=(1, 1024, 3, 1),
                            filters_shape=(1, 1, 1, 1024))
+
+
+class TestGroupGpuCorr2d(Grouped_conv_noOptim):
+    mode = theano.compile.get_mode("FAST_RUN")
+    conv2d = staticmethod(GpuCorrMM)
+    conv2d_gradw = staticmethod(GpuCorrMM_gradWeights)
+    conv2d_gradi = staticmethod(GpuCorrMM_gradInputs)
+    flip_filter = True