From 4cf0eadf8ff2a61864e996ed20df6797c5b868a3 Mon Sep 17 00:00:00 2001 From: chenyu Date: Thu, 30 May 2024 11:14:42 -0400 Subject: [PATCH] failed test case for ellipsis in einsum (#4779) from #4156 --- test/test_ops.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/test/test_ops.py b/test/test_ops.py index e875e98876..25706382c8 100644 --- a/test/test_ops.py +++ b/test/test_ops.py @@ -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)