Merge pull request #2436 from sebastien-j/faster_tests

theano/sandbox/cuda/tests/test_fftconv.py

 import unittest
 import numpy
+import copy
 
 import theano
 from theano.tests import unittest_tools as utt
@@ -188,8 +189,12 @@ class TestConv3dFFT(unittest.TestCase):
                                                           pad_last_dim = pad)
         conv_fft = conv_fft.dimshuffle(0, 2, 3, 4, 1)
 
-        f_ref = theano.function([], conv_ref)
-        f_fft = theano.function([], conv_fft, mode=mode_with_gpu)
+        ref_mode = copy.copy(theano.compile.get_default_mode())
+        ref_mode.check_py_code = False
+        f_ref = theano.function([], conv_ref, mode=ref_mode)
+        mode = mode_with_gpu
+        mode.check_py_code = False
+        f_fft = theano.function([], conv_fft, mode=mode)
 
         res_ref = f_ref()
         res_fft = f_fft()
@@ -251,9 +256,12 @@ class TestConv3dFFT(unittest.TestCase):
 
         conv = theano.tensor.nnet.conv3D(V=inputs, W=filters,
                                          b=bias, d=(1,1,1))
+        ref_mode = copy.copy(theano.compile.get_default_mode())
+        ref_mode.check_py_code = False
         mode = mode_with_gpu.including('conv3d_fft')
+        mode.check_py_code = False
 
-        f_ref = theano.function([], conv)
+        f_ref = theano.function([], conv, mode=ref_mode)
         f_fft = theano.function([], conv, mode=mode)
 
         # make sure we inserted the fft trickery
@@ -281,9 +289,12 @@ class TestConv3dFFT(unittest.TestCase):
         conv = theano.tensor.nnet.convGrad3D(V=inputs, dCdH=dCdH,
                                              WShape=filters_shape,
                                              d=(1,1,1))
+        ref_mode = copy.copy(theano.compile.get_default_mode())
+        ref_mode.check_py_code = False
         mode = mode_with_gpu.including('convgrad3d_fft')
+        mode.check_py_code = False
 
-        f_ref = theano.function([], conv)
+        f_ref = theano.function([], conv, mode=ref_mode)
         f_fft = theano.function([], conv, mode=mode)
 
         # make sure we inserted the fft trickery

theano/sandbox/cuda/tests/test_gemmcorr3d.py

 import unittest
 import numpy
+import copy
 
 import theano
 from theano.tests import unittest_tools as utt
@@ -172,8 +173,11 @@ class TestCorr3DMM(unittest.TestCase):
         conv = theano.tensor.nnet.conv3D(V=inputs, W=filters,
                                          b=bias, d=(1, 1, 1))
         mode = mode_with_gpu.including('conv3d_gemm')
+        mode.check_py_code = False
+        ref_mode = copy.copy(theano.compile.get_default_mode())
+        ref_mode.check_py_code = False
 
-        f_ref = theano.function([], conv)
+        f_ref = theano.function([], conv, mode=ref_mode)
         f_gemm = theano.function([], conv, mode=mode)
 
         # make sure we inserted the gemm trickery

theano/sandbox/test_rng_mrg.py

 import os
 import sys
 import time
+import copy
 
 import numpy
 import theano
@@ -322,7 +323,8 @@ def test_GPU_nstreams_limit():
         if theano.config.mode == "FAST_COMPILE":
             mode = "FAST_RUN"
         else:
-            mode = None
+            mode = copy.copy(theano.compile.get_default_mode())
+            mode.check_py_code = False
         out = R.uniform(size=size, nstreams=nstreams, dtype='float32')
         f = theano.function([], out, mode=mode)
         return f()