--disable_reuse_memory

产品支持情况

功能说明

内存复用开关。

关联参数

无。

参数取值

参数值：

• 1：关闭内存复用。
• 0：开启内存复用。

参数默认值：0

参数值约束：如果网络模型较大，关闭内存复用开关，调优时可能会造成内存不足，导致调优失败。

推荐配置及收益

无。

示例

--disable_reuse_memory=0

依赖约束

在内存复用场景下（默认开启内存复用），支持基于指定算子（节点名称/算子类型）单独分配内存。通过OP_NO_REUSE_MEM环境变量指定要单独分配的一个或多个节点，支持混合配置。配置多个节点时，中间通过英文逗号(“,”)隔开。详细说明请参见《环境变量参考》。

• 基于节点名称配置
  节点名称需要配置为转换为CANN平台网络后的节点名称，节点名称可以通过设置DUMP_GE_GRAPH环境变量，在导出的ge_onnx_xxx_Build.pbtxt最终图中查看“name”字段获取。

  export OP_NO_REUSE_MEM=gradients/logits/semantic/kernel/Regularizer/l2_regularizer_grad/Mul_1,resnet_v1_50/conv1_1/BatchNorm/AssignMovingAvg2

• 基于算子类型配置

  export OP_NO_REUSE_MEM=FusedMulAddN,BatchNorm

• 混合配置

  export OP_NO_REUSE_MEM=FusedMulAddN,resnet_v1_50/conv1_1/BatchNorm/AssignMovingAvg