add more checks

theano/gpuarray/corr_gemm.c

@@ -417,6 +417,11 @@ PyGpuArrayObject* corrMM(PyGpuArrayObject *const bottom,
                 "GpuCorrMM images and kernel must have the same stack size\n");
         return NULL;
     }
+    if ((nFilters % numgroups) != 0) {
+        PyErr_SetString(PyExc_ValueError,
+                "GPUCorrMM the number of filters must be divisible by the number of groups\n");
+        return NULL;
+    }
     // implicit dilated filter
     const size_t dil_kH = (kH - 1) * dilH + 1;
     const size_t dil_kW = (kW - 1) * dilW + 1;

theano/gpuarray/dnn_fwd.c

@@ -34,6 +34,11 @@ APPLY_SPECIFIC(conv_fwd)(PyGpuArrayObject *input, PyGpuArrayObject *kerns,
 		    "images and kernel must have the same stack size");
     return 1;
   }
+  if ((PyGpuArray_DIMS(kerns)[0] % params->num_groups) != 0) {
+    PyErr_SetString(PyExc_ValueError,
+		    "Number of filters must be divisible by number of groups");
+    return 1;
+  }
 
   switch (input->ga.typecode) {
   case GA_DOUBLE:

theano/gpuarray/dnn_gi.c

@@ -33,6 +33,11 @@ APPLY_SPECIFIC(conv_gi)(PyGpuArrayObject *kerns, PyGpuArrayObject *output,
                     "stack size");
     return 1;
   }
+  if ((PyGpuArray_DIMS(kerns)[0] % params->num_groups) != 0) {
+    PyErr_SetString(PyExc_ValueError,
+		    "Number of filters must be divisible by number of groups");
+    return 1;
+  }
 
   switch (im->ga.typecode) {
   case GA_DOUBLE:

theano/gpuarray/dnn_gw.c

@@ -33,6 +33,11 @@ APPLY_SPECIFIC(conv_gw)(PyGpuArrayObject *input, PyGpuArrayObject *output,
                     "GpuDnnConv images and kernel must have the same stack size");
     return 1;
   }
+  if ((PyGpuArray_DIMS(output)[1] % params->num_groups) != 0) {
+    PyErr_SetString(PyExc_ValueError,
+		    "Number of output channels must be divisible by number of groups");
+    return 1;
+  }
 
   switch (input->ga.typecode) {
   case GA_DOUBLE:

theano/tensor/nnet/corr_gemm.c

@@ -161,6 +161,11 @@ PyArrayObject* corrMM(PyArrayObject* bottom,
                 "CorrMM images and kernel must have the same stack size\n");
         return NULL;
     }
+    if ((nFilters %% numgroups) != 0) {
+        PyErr_SetString(PyExc_ValueError,
+                "CorrMM the number of filters must be divisible by the number of groups\n");
+        return NULL;
+    }
     // implicit dilated filter
     const int dil_kH = (kH - 1) * dilH + 1;
     const int dil_kW = (kW - 1) * dilW + 1;