Add a `mx.metal.device_info`

ml-explore / mlx

MLX: An array framework for Apple silicon

https://ml-explore.github.io/mlx/

MIT License

14.83k stars 845 forks source link

Closed awni closed 2 weeks ago

awni commented 2 weeks ago

A convenience function to get some useful device properties from Python.

mx.metal.device_info()

Currently:

{'max_recommended_working_set_size': 188743680000, 'max_buffer_length': 115964116992, 'architecture': 'applegpu_g14d'}