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

Test that the optimization have been applied.

上级 e6d70b6a
...@@ -83,6 +83,8 @@ def test_gemv1(): ...@@ -83,6 +83,8 @@ def test_gemv1():
gpu_f.maker.env.toposort() ]) == 1 gpu_f.maker.env.toposort() ]) == 1
assert sum([isinstance(node.op, blasop.GpuGemm) for node in assert sum([isinstance(node.op, blasop.GpuGemm) for node in
gpu_f2.maker.env.toposort() ]) == 1 gpu_f2.maker.env.toposort() ]) == 1
assert any([node.op is cuda.blas.gpu_gemm_inplace for node in gpu_f2.maker.env.toposort()])
assert any([node.op is cuda.blas.gpu_gemm_inplace for node in gpu_f.maker.env.toposort()])
def test_gemv2(): def test_gemv2():
...@@ -105,6 +107,8 @@ def test_gemv2(): ...@@ -105,6 +107,8 @@ def test_gemv2():
gpu_f.maker.env.toposort() ]) == 1 gpu_f.maker.env.toposort() ]) == 1
assert sum([isinstance(node.op, blasop.GpuGemm) for node in assert sum([isinstance(node.op, blasop.GpuGemm) for node in
gpu_f2.maker.env.toposort() ]) == 1 gpu_f2.maker.env.toposort() ]) == 1
assert any([node.op is cuda.blas.gpu_gemm_inplace for node in gpu_f2.maker.env.toposort()])
assert any([node.op is cuda.blas.gpu_gemm_inplace for node in gpu_f.maker.env.toposort()])
if __name__=='__main__': if __name__=='__main__':
test_dot_vm() test_dot_vm()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论