提交 d8389753 authored 作者: Frederic Bastien's avatar Frederic Bastien

fix test in FAST_COMPILE

上级 485f11e2
...@@ -1211,7 +1211,7 @@ class T_Rebroadcast(unittest.TestCase): ...@@ -1211,7 +1211,7 @@ class T_Rebroadcast(unittest.TestCase):
assert rebroadcast_nodes[0].op.axis == {0: True} assert rebroadcast_nodes[0].op.axis == {0: True}
def test_local_useless_eq(): def test_local_useless_eq():
mode = theano.compile.get_default_mode().including('local_useless_eq') mode = theano.compile.get_default_mode().including('canonicalize')
x=T.dmatrix() x=T.dmatrix()
y=T.dmatrix() y=T.dmatrix()
f=theano.function([x,y],T.eq(x,y), mode=mode) f=theano.function([x,y],T.eq(x,y), mode=mode)
...@@ -1231,7 +1231,7 @@ def test_local_useless_eq(): ...@@ -1231,7 +1231,7 @@ def test_local_useless_eq():
assert isinstance(topo2[-1].op,T.Alloc) assert isinstance(topo2[-1].op,T.Alloc)
def test_local_useless_neq(): def test_local_useless_neq():
mode = theano.compile.get_default_mode().including('local_useless_neq') mode = theano.compile.get_default_mode().including('canonicalize')
x=T.dmatrix() x=T.dmatrix()
y=T.dmatrix() y=T.dmatrix()
f=theano.function([x,y],T.neq(x,y), mode=mode) f=theano.function([x,y],T.neq(x,y), mode=mode)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论