mlx.core.set_cache_limit

mlx.core.set_cache_limit#

set_cache_limit(limit: int) int#

设置空闲缓存限制。

如果使用的内存超过给定限制,下一次分配时将从缓存中回收空闲内存。要禁用缓存,请将限制设置为 0

缓存限制默认为内存限制。有关更多详细信息,请参阅 set_memory_limit()

参数:

limit (int) – 缓存限制(字节)。

返回值:

之前的缓存限制(字节)。

返回类型:

int