提交 4f79b6bc authored 作者: James Bergstra's avatar James Bergstra

updated gemm test to use function instead of eval_outputs

上级 97ec802e
...@@ -141,9 +141,10 @@ class t_gemm(TestCase): ...@@ -141,9 +141,10 @@ class t_gemm(TestCase):
"""test that dot args can be aliased""" """test that dot args can be aliased"""
Z = value(self.rand(2,2)) Z = value(self.rand(2,2))
A = value(self.rand(2,2)) A = value(self.rand(2,2))
eval_outputs([gemm(Z, 1.0, A, A, 1.0)]) f = inplace_func([A,Z], gemm(Z, 1.0, A, A, 1.0))
eval_outputs([gemm(Z, 1.0, A, A.T, 1.0)]) f(A.data, Z.data)
f = inplace_func([A,Z], gemm(Z, 1.0, A, A.T, 1.0))
f(A.data, Z.data)
def test_transposes(self): def test_transposes(self):
# three square matrices which are not contiguous # three square matrices which are not contiguous
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论