SoftmaxSmallOpFusionPass

融合模式

该融合规则将ReduceMaxD、Sub等小算子组合识别并融合为SoftmaxV2算子。

该融合规则默认关闭。

融合为

使用约束

• 不支持动态shape场景。
• ReduceMaxD和ReduceSumD的输入参数约束：
  • 参数axes保持一致。
  • 参数keep_dims的值为true。
• 在reduce轴为1且AReduceSumFusionPass融合规则开启时，当前融合规则不生效。
• 在reduce轴超过一个并包含尾轴，数据类型为FLOAT32，且ReduceMaxDFusionPass融合规则开启时，当前融合规则不生效。
• 数据类型限制：
  • Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件：数据类型支持FLOAT32、FLOAT16、BFLOAT16。

支持的型号

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件