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: