Add shape check in all version of CrossentropySoftmax1HotWithBiasDx

theano/sandbox/cuda/nnet.py

@@ -247,7 +247,7 @@ class GpuCrossentropySoftmax1HotWithBiasDx(GpuOp):
 
     def c_code_cache_version(self):
         # return ()
-        return (7,)
+        return (8,)
 
     def c_code(self, node, nodename, inp, out, sub):
         dnll, sm, y_idx = inp
@@ -288,6 +288,13 @@ class GpuCrossentropySoftmax1HotWithBiasDx(GpuOp):
                             "dnll.shape[0] != y_idx.shape[0]");
             %(fail)s;
         }
+        if (CudaNdarray_HOST_DIMS(%(sm)s)[0] !=
+            CudaNdarray_HOST_DIMS(%(y_idx)s)[0])
+        {
+            PyErr_SetString(PyExc_ValueError,
+                            "sm.shape[0] != y_idx.shape[0]");
+            %(fail)s;
+        }
         if ((NULL == %(dx)s)
             || (CudaNdarray_HOST_DIMS(%(dx)s)[0] !=
                 CudaNdarray_HOST_DIMS(%(sm)s)[0])

theano/sandbox/gpuarray/nnet.py

@@ -295,7 +295,7 @@ class GpuCrossentropySoftmax1HotWithBiasDx(Op):
 
     def c_code_cache_version(self):
         # return ()
-        return (7,)
+        return (8,)
 
     def c_headers(self):
         return ['cuda.h', '<gpuarray/extension.h>', '<numpy_compat.h>']
@@ -351,6 +351,13 @@ class GpuCrossentropySoftmax1HotWithBiasDx(Op):
                             "dnll.shape[0] != y_idx.shape[0]");
             %(fail)s;
         }
+        if (PyGpuArray_DIMS(%(sm)s)[0] !=
+            PyGpuArray__DIMS(%(y_idx)s)[0])
+        {
+            PyErr_SetString(PyExc_ValueError,
+                            "sm.shape[0] != y_idx.shape[0]");
+            %(fail)s;
+        }
         if ((NULL == %(dx)s)
             || (PyGpuArray_DIMS(%(dx)s)[0] !=
                 PyGpuArray_DIMS(%(sm)s)[0])

theano/tensor/nnet/nnet.py

@@ -1108,7 +1108,7 @@ class CrossentropySoftmax1HotWithBiasDx(gof.Op):
         return [g_dy, g_sm, g_y_idx]
 
     def c_code_cache_version(self):
-        return (4,)
+        return (5,)
 
     def c_code(self, node, name, inp, out, sub):
         dnll, sm, y_idx = inp
@@ -1167,6 +1167,13 @@ class CrossentropySoftmax1HotWithBiasDx(gof.Op):
                          (long int)PyArray_DIMS(%(y_idx)s)[0]);
             %(fail)s;
         }
+        if (PyArray_DIMS(%(sm)s)[0] !=
+            PyArray_DIMS(%(y_idx)s)[0])
+        {
+            PyErr_SetString(PyExc_ValueError,
+                            "sm.shape[0] != y_idx.shape[0]");
+            %(fail)s;
+        }
         if ((NULL == %(dx)s)
             || (PyArray_DIMS(%(dx)s)[0] != PyArray_DIMS(%(sm)s)[0])
             || (PyArray_DIMS(%(dx)s)[1] != PyArray_DIMS(%(sm)s)[1]))