昇腾社区首页
中文
注册

MatmulReshapeFusionPass

融合模式

该融合规则在Matmul N轴等于1时,会对第二个输入增加Reshape节点。

使用约束

  • 该规则默认关闭,可参考如何关闭/开启融合规则自行开启。
  • 仅支持静态shape场景。
  • 支持数据类型:BFLOAT16、FLOAT16、FLOAT32。
  • x1的输入为[M, K],x2的输入未转置,N=1并在尾轴,输入为[K, 1]。
  • 数据类型为BFLOAT16或FLOAT16时,K<27392时,该融合规则生效。

支持的型号

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

Atlas A3 训练系列产品/Atlas A3 推理系列产品