megengine.set_prealloc_config

set_prealloc_config(alignment=1, min_req=32 * 1024 * 1024, max_overhead=0, growth_factor=2.0, device_type=DeviceType.CUDA)

对原始设备的内存分配器指定预先分配的方案。

参数

• alignment (int) – 指定对齐的字节数。

• min_req (int) – 最小的需要的字节数。

• max_overhead (int) – 为满足需要的字节数花费的开销上限（单位：字节）。

• growth_factor – 需要的大小 / 现在分配了的大小`

• device_type – 设备类别

• alignment – int:

• min_req – int:

• max_overhead – int:

上一页

megengine.reset_max_memory_stats

下一页

megengine.coalesce_free_memory