The module 'mlx.core.metal' has no attribute 'device_info'?

[Lance729]

I want to get the device info show in official doc,, because I want to test the gpu and cpu, so I follow the document said, but when I run mlx.core.device_info(mx.DeviceType.gpu), there is an error

AttributeError: module 'mlx.core.metal' has no attribute 'device_info'

❓ How can I get the device info?

I work on:

• m5 platform
• mlx 0.7

[KOKOSde]

You need mx.device_info(mx.gpu), not mx.metal.device_info(...). mx.metal only exposes backend helpers like availability checks, so that attribute error is expected.

[aicayzer]

@KOKOSde has it — mx.device_info(mx.gpu) is what you want, not mx.metal.device_info(...).

Worth flagging though: you're on mlx 0.7 and the current release is around 0.29 — that's a lot of API and perf changes between. If you're poking at gpu vs cpu it's worth upgrading first (pip install -U mlx), otherwise the docs you're looking at won't match what's installed.

Once upgraded:

>>> import mlx.core as mx
>>> mx.device_info(mx.gpu)
{'architecture': '...', 'memory_size': ..., 'max_buffer_length': ..., 'max_recommended_working_set_size': ...}

No args also works (mx.device_info()) and returns the default device's info.