提交 b80090c2 authored 作者: Frederic's avatar Frederic

[TESTS] fix tests in FAST_COMPILE mode.

上级 b1559dab
......@@ -641,7 +641,8 @@ def test_valid(conv_gemm=False):
shapes += get_shapes2(scales_img=(2, 2), img_stride=(2, 2))
shapes += get_shapes2(scales_kern=(2, 2), kern_stride=(2, 2))
else:
mode = cls = None
mode = theano_mode
cls = None
exec_conv(version, shapes, verbose, random, 'valid',
print_=print_, ones=ones, rtol=1.1e-5,
theano_mode=mode, cls=cls)
......@@ -717,7 +718,8 @@ def test_full(conv_gemm=False):
# dummy version; not used by GpuCorrMM so one version is enough
version = [-1]
else:
mode = cls = None
mode = theano_mode
cls = None
exec_conv(version, shapes, verbose, random, 'full',
theano_mode=mode, cls=cls)
......@@ -757,7 +759,8 @@ def test_subsample(conv_gemm=False):
# dummy version; not used by GpuCorrMM so one version is enough
version_valid = version_full = [-1]
else:
mode = cls = None
mode = theano_mode
cls = None
exec_conv(version_valid, shapes, verbose, random, 'valid',
print_=print_, ones=ones,
......
......@@ -2887,10 +2887,13 @@ class T_Tile(unittest.TestCase):
def test_local_useless_tile(self):
v = T.vector()
m = T.matrix()
mode = None
if theano.config.mode == "FAST_COMPILE":
mode = "FAST_RUN"
for var, data in [(v, [1, 2, 3]), (m, [[1, 2], [3, 4]])]:
# Currently, only a repeat patter == ndim is supported.
for ndim in [var.ndim]: # range(1, var.ndim):
f = theano.function([var], T.tile(var, (1,)*ndim))
f = theano.function([var], T.tile(var, (1,)*ndim), mode=mode)
topo = f.maker.fgraph.toposort()
assert len(topo) == 1
assert isinstance(topo[0].op, compile.DeepCopyOp)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论