--disable_reuse_memory

Applicability

Product

Supported

Atlas A3 training products/Atlas A3 inference products

Atlas A2 training products/Atlas A2 inference products

Atlas 200I/500 A2 inference products

Atlas inference products

Atlas training products

Description

Enables memory reuse.

See Also

None

Arguments

Arguments:

  • 1: disabled
  • 0: enabled

Default: 0

Restrictions: If the network model is large and the memory reuse function is disabled, the memory may be insufficient during model tuning. As a result, the model tuning fails.

Suggestions and Benefits

None

Examples

--disable_reuse_memory=0

Dependencies and Restrictions

Selects the operator to skip in memory reuse. (Memory reuse is enabled by default.) The operator (or operators separated by commas) specified by OP_NO_REUSE_MEM will use exclusively allocated memory. If multiple nodes are configured, separate them with commas (,). For details, see Environment Variables.

  • Configuration by node name:
    After the node name is converted to be compatible with the CANN platform, ensure that the node name is correctly configured. You can obtain the node name by setting DUMP_GE_GRAPH and viewing the name field in the exported ge_onnx_xxx_Build.pbtxt graph.
    export OP_NO_REUSE_MEM=gradients/logits/semantic/kernel/Regularizer/l2_regularizer_grad/Mul_1,resnet_v1_50/conv1_1/BatchNorm/AssignMovingAvg2
  • Configuration by operator type:
    export OP_NO_REUSE_MEM=FusedMulAddN,BatchNorm
  • Configuration by operator name and operator type:
    export OP_NO_REUSE_MEM=FusedMulAddN,resnet_v1_50/conv1_1/BatchNorm/AssignMovingAvg