提交 8908b957 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

fix bug in test

上级 bbebd2d3
...@@ -308,9 +308,9 @@ class T_Scan(unittest.TestCase): ...@@ -308,9 +308,9 @@ class T_Scan(unittest.TestCase):
updates=updates, updates=updates,
allow_input_downcast=True) allow_input_downcast=True)
nodes = [x for x in my_f.maker.env.toposort() nodes = [x for x in my_f.maker.env.toposort()
if isinstance(x.owner.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]._scan_savemem_visited assert nodes[0].op._scan_savemem_visited
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论