github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-25 14:58:46 -05:00
Code Issues Packages Projects Releases Wiki Activity

failed test case for ellipsis in einsum (#4779)

Browse Source 
from #4156
This commit is contained in:
chenyu
2024-05-30 11:14:42 -04:00
committed by GitHub
parent e89bc42cc7
commit 4cf0eadf8f
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_ops.py
View File

@@ -635,6 +635,10 @@ class TestOps(unittest.TestCase):
lambda a,b: Tensor.einsum('zqrs,tuqvr->zstuv', a,b), atol=1e-5)
# bilinear transformation
helper_test_op([(2,3),(5,3,7),(2,7)], lambda a,b,c: torch.einsum('ik,jkl,il->ij', [a,b,c]), lambda a,b,c: Tensor.einsum('ik,jkl,il->ij', [a,b,c]))
# test ellipsis # TODO: FIXME
with self.assertRaises(Exception):
helper_test_op([(16,29,256),(16,29,256)], lambda a,b: torch.einsum('...id, ...jd -> ...ij', [a,b]),
lambda a,b: Tensor.einsum('...id, ...jd -> ...ij', [a,b]))
def test_einsum_shape_check(self):
a = Tensor.zeros(3,8,10,5)