Added new mechanism to construct output types that avoid importing Cuda.

theano/scan_module/scan_op.py

@@ -28,7 +28,7 @@ from theano import gof
 from theano.tensor import TensorType
 from theano import tensor
 from theano.tensor.opt import Shape_i
-from theano.sandbox import cuda
+#from theano.sandbox import cuda
 from theano.compile.profiling import ScanProfileStats
 
 import scan_utils
@@ -46,7 +46,9 @@ class Scan(Op):
     def __init__( self
                  , inputs
                  , outputs
-                 , info  ):
+                 , info
+                 , typeConstructor = None
+                ):
         """
         :param inputs: inputs of the inner function of scan
         :param outputs: outputs of the inner function of scan
@@ -66,39 +68,16 @@ class Scan(Op):
         self.output_types = []
         idx = 0
         jdx = 0
-        if self.gpu:
-            # mit_mot
-            while idx < self.n_mit_mot_outs:
-                # Not that for mit_mot there are several output slices per
-                # output sequence
-                o     = outputs[idx]
-                self.output_types.append(
-                    cuda.CudaNdarrayType(
-                        broadcastable = (False,) + o.type.broadcastable))
-                idx += len(self.mit_mot_out_slices[jdx])
-                jdx += 1
+        if typeConstructor is None:
+            typeConstructor = lambda broadcastable, dtype: TensorType(
+                broadcastable = broadcastable, dtype = dtype)
 
-            # mit_sot / sit_sot / nit_sot
-            end = idx + self.n_mit_sot + self.n_sit_sot + self.n_nit_sot
-            for o in outputs[idx:end]:
-                self.output_types.append(
-                    cuda.CudaNdarrayType( broadcastable = (False,) +
-                                    o.type.broadcastable))
-            # shared outputs
-            for o in outputs[end:]:
-                if isinstance(o.type, TensorType):
-                    self.output_types.append(cuda.CudaNdarrayType(
-                        broadcastable = o.type.broadcastable))
-                else:
-                    self.output_types.append( o.type )
-        else:
         while idx < self.n_mit_mot_outs:
             # Not that for mit_mot there are several output slices per
             # output sequence
             o     = outputs[idx]
             self.output_types.append(
-                    TensorType(
-                        broadcastable = (False,) + o.type.broadcastable
+                typeConstructor( broadcastable = (False,) + o.type.broadcastable
                                 , dtype = o.type.dtype)
                         )
             idx += len(self.mit_mot_out_slices[jdx])
@@ -108,17 +87,11 @@ class Scan(Op):
         end = idx + self.n_mit_sot + self.n_sit_sot + self.n_nit_sot
         for o in outputs[idx:end]:
             self.output_types.append(
-                    TensorType(
+                typeConstructor(
                     broadcastable = (False,) + o.type.broadcastable
                     , dtype = o.type.dtype ))
         # shared outputs + possibly the ending condition
         for o in outputs[end:]:
-                if cuda.cuda_available and isinstance(o.type,
-                                                      cuda.CudaNdarrayType):
-                    self.output_types.append( TensorType(
-                        broadcastable = o.type.broadcastable
-                        , dtype = theano.config.floatX) )
-                else:
             self.output_types.append( o.type )
 
         if self.as_while: