提交 134d3bbf authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix test in FAST_COMPILE.

上级 0ecaf3d6
......@@ -2703,8 +2703,9 @@ def test_local_set_to_inc_subtensor():
g = s + 3
r = theano.tensor.set_subtensor(s, g)
moder = compile.get_default_mode().excluding('local_set_to_inc_subtensor')
modet = compile.get_default_mode().including('local_set_to_inc_subtensor')
f1 = theano.function([v], r, mode=moder)
f2 = theano.function([v], r)
f2 = theano.function([v], r, mode=modet)
advi1 = [n for n in f1.maker.fgraph.toposort()
if isinstance(n.op, tensor.AdvancedIncSubtensor1)]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论