def __init__(self,
atomic_clean_policy=0,
static_memory_policy=None
)
MemoryConfig类的构造函数,用于配置系统内存使用方式。
无。
参数名 |
输入/输出 |
描述 |
---|---|---|
atomic_clean_policy |
输入 |
是否集中清理网络中所有atomic算子占用的内存,取值包括:
|
static_memory_policy |
输入 |
当前图运行所需内存超过前一张图运行所需内存时,是否直接释放前一张图内存。 0:不释放,采用默认的动态分配方式,当内存超限时才释放内存。默认是0。 1:释放,一旦当前图内超过前一张图运行所需内存,就直接释放前一张图内存。 |
返回MemoryConfig类对象,作为NPURunConfig的参数传入。