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 – request size / cur allocated`

• device_type – 设备类别

• alignment – int:

• min_req – int:

• max_overhead – int:

上一页

megengine.get_cuda_compute_capability

下一页

megengine.enable_debug_log