提交 547bf1f1 authored 作者: Frederic Bastien's avatar Frederic Bastien

fix test in FAST_COMPILE mode.

上级 799834f6
...@@ -710,6 +710,7 @@ def test_gemv1(): ...@@ -710,6 +710,7 @@ def test_gemv1():
topo = f.maker.env.toposort() topo = f.maker.env.toposort()
assert len(topo)==1 assert len(topo)==1
assert isinstance(topo[0].op, Gemv) assert isinstance(topo[0].op, Gemv)
if config.mode != 'FAST_COMPILE':
assert topo[0].op.inplace==True assert topo[0].op.inplace==True
...@@ -737,5 +738,5 @@ def test_gemv2(): ...@@ -737,5 +738,5 @@ def test_gemv2():
assert numpy.allclose(v2.value, numpy.dot(v1.value, m.value)+v2_orig) assert numpy.allclose(v2.value, numpy.dot(v1.value, m.value)+v2_orig)
topo = f.maker.env.toposort() topo = f.maker.env.toposort()
assert sum(isinstance(node.op, Gemv) for node in topo)==1 assert sum(isinstance(node.op, Gemv) for node in topo)==1
assert topo[0].op.inplace==True if config.mode != 'FAST_COMPILE':
assert topo[-1].op.inplace==True
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论