提交 2f223eff authored 作者: Frederic's avatar Frederic

fix test in fast_compile mode.

上级 b8165faa
...@@ -312,7 +312,8 @@ class T_Scan(unittest.TestCase): ...@@ -312,7 +312,8 @@ class T_Scan(unittest.TestCase):
nodes = [x for x in my_f.maker.env.toposort() nodes = [x for x in my_f.maker.env.toposort()
if isinstance(x.op, theano.scan_module.scan_op.Scan)] if isinstance(x.op, theano.scan_module.scan_op.Scan)]
# This assertation fails if savemem optimization failed on scan # This assertation fails if savemem optimization failed on scan
assert nodes[0].op._scan_savemem_visited if theano.config.mode != "FAST_COMPILE":
assert nodes[0].op._scan_savemem_visited
rng = numpy.random.RandomState(utt.fetch_seed()) rng = numpy.random.RandomState(utt.fetch_seed())
my_f(rng.uniform(size=(3,)), my_f(rng.uniform(size=(3,)),
4, 4,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论