MemoryConfig构造函数

函数原型

def __init__(self,

atomic_clean_policy=0,

static_memory_policy=0

)

功能说明

MemoryConfig类的构造函数,用于配置系统内存使用方式。

约束说明

无。

参数说明

参数名

输入/输出

描述

atomic_clean_policy

输入

是否集中清理网络中所有atomic算子占用的内存,取值包括:

  • 0:集中清理,默认为0。
  • 1:不集中清理,对网络每一个atomic算子进行单独清零。当网络中内存超限时,可尝试此种清理方式,但可能会导致一定的性能损耗。

static_memory_policy

输入

网络运行是否使用内存静态分配方式。

  • 0:动态分配内存,即按照实际大小动态分配。
  • 1:静态分配内存,即按照允许的最大静态内存与变量内存进行分配。
  • 2:动态扩展内存,即当前图所需内存超过前一张图的内存时,直接释放前一张图的内存。

默认值是0。

返回值

返回MemoryConfig类对象,作为NPURunConfig的参数传入。