Add tests for ifelse on the gpu.

theano/sandbox/cuda/tests/test_opt.py

@@ -11,6 +11,8 @@ import theano
 from theano.tests import unittest_tools as utt
 
 import theano.sandbox.cuda as cuda
+from theano.sandbox.cuda import basic_ops
+
 if cuda.cuda_available == False:
     raise SkipTest('Optional package cuda disabled')
 
@@ -330,6 +332,17 @@ class test_local_gpu_tensordot(unittest.TestCase):
         f3(tensor1, tensor3)
 
 
+import theano.tests.test_ifelse
+
+
+class TestIfElse(theano.tests.test_ifelse.test_ifelse):
+    dtype = "float32"
+    mode = mode_with_gpu
+    cast_output = staticmethod(basic_ops.as_cuda_ndarray_variable)
+
+    def get_ifelse(self, n):
+        return theano.ifelse.IfElse(n, gpu=True, as_view=True)
+
 if __name__ == '__main__':
     test_gpualloc()
     test_opt_gpujoin_onlyajoin()