mlx.core.metal.device_info

Contents

mlx.core.metal.device_info#

device_info() dict[str, Union[str, int]]#

Get information about the GPU device and system settings.

Currently returns:

  • architecture

  • max_buffer_size

  • max_recommended_working_set_size

  • memory_size

Returns:

A dictionary with string keys and string or integer values.

Return type:

dict