提交 bdcb0c0d authored 作者: Pierre Luc Carrier's avatar Pierre Luc Carrier

Correct bug in PushOutScanOutput unittests where optimization was not properly excluded

上级 58dc930e
...@@ -160,9 +160,9 @@ class TestPushOutScanOutputDot(object): ...@@ -160,9 +160,9 @@ class TestPushOutScanOutputDot(object):
f_opt = theano.function([v, m], T.jacobian(output, v)) f_opt = theano.function([v, m], T.jacobian(output, v))
default_mode = theano.compile.get_default_mode() default_mode = theano.compile.get_default_mode()
default_mode.excluding("scanOp_pushout_output") new_mode = default_mode.excluding("scanOp_pushout_output")
f_no_opt = theano.function([v, m], T.jacobian(output, v), f_no_opt = theano.function([v, m], T.jacobian(output, v),
mode=default_mode) mode=new_mode)
# Ensure that the optimization was performed correctly in f_opt # Ensure that the optimization was performed correctly in f_opt
# The inner function of scan should have only one output and it should # The inner function of scan should have only one output and it should
...@@ -251,8 +251,8 @@ class TestPushOutScanOutputDot(object): ...@@ -251,8 +251,8 @@ class TestPushOutScanOutputDot(object):
f_opt = theano.function([a, b], outputs) f_opt = theano.function([a, b], outputs)
default_mode = theano.compile.get_default_mode() default_mode = theano.compile.get_default_mode()
default_mode.excluding("scanOp_pushout_output") new_mode = default_mode.excluding("scanOp_pushout_output")
f_no_opt = theano.function([a, b], outputs, mode=default_mode) f_no_opt = theano.function([a, b], outputs, mode=new_mode)
# Ensure that the optimization was performed correctly in f_opt # Ensure that the optimization was performed correctly in f_opt
# The inner function of scan should have only one output and it should # The inner function of scan should have only one output and it should
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论