MemoryConfig构造函数

函数原型

def __init__(self,

atomic_clean_policy=0,

static_memory_policy=None

)

功能说明

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

约束说明

无。

参数说明

参数名

输入/输出

描述

atomic_clean_policy

输入

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

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

static_memory_policy

输入

当前图运行所需内存超过前一张图运行所需内存时,是否直接释放前一张图内存。

0:不释放,采用默认的动态分配方式,当内存超限时才释放内存。默认是0。

1:释放,一旦当前图内超过前一张图运行所需内存,就直接释放前一张图内存。

返回值

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