MemoryConfig构造函数
函数原型
def __init__(self,
atomic_clean_policy=0,
static_memory_policy=0
)
功能说明
MemoryConfig类的构造函数,用于配置系统内存使用方式。
约束说明
无。
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
atomic_clean_policy |
输入 |
是否集中清理网络中所有atomic算子占用的内存,取值包括:
|
static_memory_policy |
输入 |
网络运行是否使用内存静态分配方式。
默认值是0。 |
返回值
返回MemoryConfig类对象,作为NPURunConfig的参数传入。