mlx.core.metal.device_info

目录

mlx.core.metal.device_info#

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

获取 GPU 设备和系统设置信息。

当前返回

  • 架构

  • 最大缓冲区大小

  • 最大推荐工作集大小

  • 内存大小

  • 资源限制

返回:

一个字典,键为字符串,值为字符串或整数。

返回类型:

dict