昇腾社区首页
中文
注册

OP_NO_REUSE_MEM

功能描述

在内存复用场景下(默认开启内存复用),支持基于指定算子(节点名称/算子类型)单独分配内存。

通过该环境变量指定要单独分配的一个或多个节点,支持混合配置。配置多个节点时,中间通过逗号(“,”)隔开。

配置示例

  • 基于节点名称配置
    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