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

Check that dot22 are inserted.

上级 cd9e62a0
...@@ -770,6 +770,8 @@ def test_dot_vm(): ...@@ -770,6 +770,8 @@ def test_dot_vm():
# Assert that the dot was optimized somehow # Assert that the dot was optimized somehow
assert sum([isinstance(node.op, T.Dot) for node in assert sum([isinstance(node.op, T.Dot) for node in
f.maker.env.toposort() ]) == 0 f.maker.env.toposort() ]) == 0
assert sum([isinstance(node.op, T.blas.Dot22) for node in
f.maker.env.toposort() ]) == 1
def test_dot_mv(): def test_dot_mv():
''' Test matrix dot vector ''' ''' Test matrix dot vector '''
...@@ -785,6 +787,8 @@ def test_dot_mv(): ...@@ -785,6 +787,8 @@ def test_dot_mv():
# Assert that the dot was optimized somehow # Assert that the dot was optimized somehow
assert sum([isinstance(node.op, T.Dot) for node in assert sum([isinstance(node.op, T.Dot) for node in
f.maker.env.toposort() ]) == 0 f.maker.env.toposort() ]) == 0
assert sum([isinstance(node.op, T.blas.Dot22) for node in
f.maker.env.toposort() ]) == 1
class TestGemv(TestCase): class TestGemv(TestCase):
def test_gemv1(self): def test_gemv1(self):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论