megengine.set_prealloc_config

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

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

参数

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

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

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

• growth_factor – 需要的大小和目前分配的大小的比值

• device_type – 设备类别

megengine.get_mem_status_bytes 查询版本信息