Added check inut for cuda and tensor

theano/sandbox/cuda/type.py

@@ -303,6 +303,7 @@ class CudaNdarrayType(Type):
             }
             //std::cerr << "c_extract " << %(name)s << " nd check passed\\n";
         """ % locals()
+        if(theano.compile.ops.Shape.check_input):
             for i, b in enumerate(self.broadcastable):
                 if b:
                     print >> sio, """
@@ -348,6 +349,12 @@ class CudaNdarrayType(Type):
             }
             //std::cerr << "c_extract done " << %(name)s << '\\n';
             """ % locals()
+        else:
+            print >> sio, """
+                assert(%(name)s);
+                Py_INCREF(py_%(name)s);
+            }
+            """ % locals()
         #print sio.getvalue()
         return sio.getvalue()
 

theano/tensor/type.py

@@ -433,7 +433,8 @@ class TensorType(Type):
 
     def c_extract(self, name, sub):
         """Override `CLinkerType.c_extract` """
-        return """
+        if(theano.compile.ops.Shape.check_input):
+            check = """
             %(name)s = NULL;
             if (py_%(name)s == Py_None) {
                 // We can either fail here or set %(name)s to NULL and rely on Ops
@@ -482,6 +483,10 @@ class TensorType(Type):
                              %(type_num)s, type_num_%(name)s);
                 %(fail)s
             }
+            """ % dict(sub, name=name, type_num=self.dtype_specs()[2])
+        else:
+            check = ""
+        return check + """
         %(name)s = (PyArrayObject*)(py_%(name)s);
         Py_XINCREF(%(name)s);
         """ % dict(sub, name=name, type_num=self.dtype_specs()[2])