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: