提交 96eaafeb authored 作者: sebastien-j's avatar sebastien-j

"FAST_RUN" for reference function

上级 0899f5e4
...@@ -189,9 +189,7 @@ class TestConv3dFFT(unittest.TestCase): ...@@ -189,9 +189,7 @@ class TestConv3dFFT(unittest.TestCase):
pad_last_dim = pad) pad_last_dim = pad)
conv_fft = conv_fft.dimshuffle(0, 2, 3, 4, 1) conv_fft = conv_fft.dimshuffle(0, 2, 3, 4, 1)
ref_mode = copy.copy(theano.compile.get_default_mode()) f_ref = theano.function([], conv_ref, mode="FAST_RUN")
ref_mode.check_py_code = False
f_ref = theano.function([], conv_ref, mode=ref_mode)
mode = mode_with_gpu mode = mode_with_gpu
mode.check_py_code = False mode.check_py_code = False
f_fft = theano.function([], conv_fft, mode=mode) f_fft = theano.function([], conv_fft, mode=mode)
...@@ -256,12 +254,10 @@ class TestConv3dFFT(unittest.TestCase): ...@@ -256,12 +254,10 @@ class TestConv3dFFT(unittest.TestCase):
conv = theano.tensor.nnet.conv3D(V=inputs, W=filters, conv = theano.tensor.nnet.conv3D(V=inputs, W=filters,
b=bias, d=(1,1,1)) 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 = mode_with_gpu.including('conv3d_fft')
mode.check_py_code = False mode.check_py_code = False
f_ref = theano.function([], conv, mode=ref_mode) f_ref = theano.function([], conv, mode="FAST_RUN")
f_fft = theano.function([], conv, mode=mode) f_fft = theano.function([], conv, mode=mode)
# make sure we inserted the fft trickery # make sure we inserted the fft trickery
...@@ -289,12 +285,10 @@ class TestConv3dFFT(unittest.TestCase): ...@@ -289,12 +285,10 @@ class TestConv3dFFT(unittest.TestCase):
conv = theano.tensor.nnet.convGrad3D(V=inputs, dCdH=dCdH, conv = theano.tensor.nnet.convGrad3D(V=inputs, dCdH=dCdH,
WShape=filters_shape, WShape=filters_shape,
d=(1,1,1)) 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 = mode_with_gpu.including('convgrad3d_fft')
mode.check_py_code = False mode.check_py_code = False
f_ref = theano.function([], conv, mode=ref_mode) f_ref = theano.function([], conv, mode="FAST_RUN")
f_fft = theano.function([], conv, mode=mode) f_fft = theano.function([], conv, mode=mode)
# make sure we inserted the fft trickery # make sure we inserted the fft trickery
......
...@@ -174,10 +174,8 @@ class TestCorr3DMM(unittest.TestCase): ...@@ -174,10 +174,8 @@ class TestCorr3DMM(unittest.TestCase):
b=bias, d=(1, 1, 1)) b=bias, d=(1, 1, 1))
mode = mode_with_gpu.including('conv3d_gemm') mode = mode_with_gpu.including('conv3d_gemm')
mode.check_py_code = False 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, mode=ref_mode) f_ref = theano.function([], conv, mode="FAST_RUN")
f_gemm = theano.function([], conv, mode=mode) f_gemm = theano.function([], conv, mode=mode)
# make sure we inserted the gemm trickery # make sure we inserted the gemm trickery
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论