(self)
| 132 | |
| 133 | @unittest.skipIf(not mx.is_available(mx.gpu), "GPU is not available") |
| 134 | def test_device_info_gpu(self): |
| 135 | gpu_count = mx.device_count(mx.gpu) |
| 136 | for i in range(gpu_count): |
| 137 | info = mx.device_info(mx.Device(mx.gpu, i)) |
| 138 | self.assertIsInstance(info, dict) |
| 139 | self.assertIn("device_name", info) |
| 140 | self.assertTrue(len(info["device_name"]) > 0) |
| 141 | self.assertIn("architecture", info) |
| 142 | |
| 143 | def test_device_info_default(self): |
| 144 | info = mx.device_info() |