mlx.core.set_memory_limit

set_memory_limit(limit: int) → int

设置内存限制。

内存限制是图评估期间使用的最大内存量的指导值。如果超出内存限制且没有更多 RAM（包括可用时的交换空间），则分配将导致异常。

当 Metal 可用时，内存限制默认为设备报告的最大建议工作集大小的 1.5 倍。

参数:

limit (int) – 内存限制，以字节为单位。

返回值:

之前的内存限制，以字节为单位。

返回类型:

int