(self)
| 50 | |
| 51 | @unittest.skipIf(not mx.metal.is_available(), "Metal is not available") |
| 52 | def test_wired_memory(self): |
| 53 | old_limit = mx.set_wired_limit(1000) |
| 54 | old_limit = mx.set_wired_limit(0) |
| 55 | self.assertEqual(old_limit, 1000) |
| 56 | |
| 57 | max_size = mx.device_info(mx.gpu)["max_recommended_working_set_size"] |
| 58 | with self.assertRaises(ValueError): |
| 59 | mx.set_wired_limit(max_size + 10) |
| 60 | |
| 61 | def test_active_memory_count(self): |
| 62 | mx.synchronize() |
nothing calls this directly
no test coverage detected