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