simple on device failing test
@@ -93,9 +93,7 @@ class TestOps(unittest.TestCase):
def test_dot(self):
helper_test_op([(45,65), (65,100)], lambda x,y: x.matmul(y), Tensor.dot, atol=1e-4)
def test_matmul(self):
- helper_test_op([(65), (65,100)], lambda x,y: x.matmul(y), Tensor.dot, atol=1e-4)
- helper_test_op([(512), (512,802)], lambda x,y: x.matmul(y), Tensor.dot, atol=1e-4)
- helper_test_op([(802), (802,512)], lambda x,y: x.matmul(y), Tensor.dot, atol=1e-4)
+ helper_test_op([(65), (65,99)], lambda x,y: x.matmul(y), Tensor.dot, atol=1e-4)
def test_broadcastdot(self):
helper_test_op([(10,45,65), (65,45)], lambda x,y: x @ y, Tensor.dot, atol=1e-4)
def test_multidot(self):
