Multiple fixes.

theano/sandbox/cuda/type.py

@@ -293,6 +293,9 @@ class CudaNdarrayType(Type):
             //fprintf(stderr, "c_extract CNDA object w refcnt %%p %%i\\n", py_%(name)s, (py_%(name)s->ob_refcnt));
             %(name)s = (CudaNdarray*)py_%(name)s;
             //std::cerr << "c_extract " << %(name)s << '\\n';
+        """ % locals()
+        if(check_input):
+            print >> sio, """
                 if (%(name)s->nd != %(nd)s)
                 {
                     PyErr_Format(PyExc_RuntimeError,
@@ -303,7 +306,6 @@ class CudaNdarrayType(Type):
                 }
                 //std::cerr << "c_extract " << %(name)s << " nd check passed\\n";
             """ % locals()
-        if(check_input):
             for i, b in enumerate(self.broadcastable):
                 if b:
                     print >> sio, """

theano/tensor/type.py

@@ -416,14 +416,21 @@ class TensorType(Type):
         return str(self)
         #"TensorType{%s, %s}" % (str(self.dtype), str(self.broadcastable))
 
-    def c_declare(self, name, sub):
+    def c_declare(self, name, sub, check_input=True):
         """Override `CLinkerType.c_declare` """
-        return """
+        if(check_input):
+            check = """
+            typedef %(dtype)s dtype_%(name)s;
+            """ % dict(sub, name=name, dtype=self.dtype_specs()[1])
+        else:
+            check = ""
+        declaration = """
         PyArrayObject* %(name)s;
         int type_num_%(name)s;
-        typedef %(dtype)s dtype_%(name)s;
         """ % dict(sub, name=name, dtype=self.dtype_specs()[1])
 
+        return declaration + check
+
     def c_init(self, name, sub):
         """Override `CLinkerType.c_init` """
         return """
@@ -485,7 +492,9 @@ class TensorType(Type):
             }
             """ % dict(sub, name=name, type_num=self.dtype_specs()[2])
         else:
-            check = ""
+            check = """
+            type_num_%(name)s = PyArray_TYPE((PyArrayObject*) py_%(name)s);
+            """
         return check + """
         %(name)s = (PyArrayObject*)(py_%(name)s);
         Py_XINCREF(%(name)s);