(self)
| 2362 | "This test requires a lot of memory", |
| 2363 | ) |
| 2364 | def test_large_binary(self): |
| 2365 | a = mx.ones([1000, 2147484], mx.int8) |
| 2366 | b = mx.ones([2147484], mx.int8) |
| 2367 | self.assertEqual((a + b)[0, 0].item(), 2) |
| 2368 | |
| 2369 | def test_eye(self): |
| 2370 | self.assertCmpNumpy([3], mx.eye, np.eye) |