描述

本 PR 适配 Addmm、Baddbmm 及 AddmmWeightNz 的 ACLNN 接口路由，使 A2/A3（DAV_2201）上的 broadcast bias 场景能够直接使用 GemmV3 完成矩阵乘、alpha/beta 缩放及 bias broadcast。

主要修改如下：

1. 清理 broadcast bias 路由限制

• 删除 CheckAddmmTensorShapeNeedBroadcast。
• 简化 CheckCubeMathTypeForAddMm，仅保留 cubeMathType 取值及平台能力校验。
• 不再因为 bias shape 与矩阵乘输出 shape 不一致而拒绝 cubeMathType=USE_FP32_ADD。
• 路由限制在 A2/A3（DAV_2201），不改变其他架构的执行逻辑。

2. 调整 Addmm 和 Baddbmm 的 GemmV3 路由

• cubeMathType=USE_FP32_ADD 时，允许 broadcast bias 进入 GemmV3。
• 区分“接口计算需要 FP32 中间结果”和“GemmV3 实际输出 FP32”，避免 FP16/BF16 输出错误使用 16in32out GemmV3。

3. 调整 AddmmWeightNz 的 GemmV3 路由

• 取消 A2/A3 上将 cubeMathType=USE_FP32_ADD 转换为 cubeMathType=KEEP_DTYPE 的逻辑。
• cubeMathType=USE_FP32_ADD 时，支持 AddmmWeightNz 直接路由至 GemmV3。
• 支持 broadcast bias 与 ND×NZ 矩阵输入组合进入 GemmV3。
• 支持 FP16/BF16 输入及对应的低精度输出。
• 保留原有 16in32out 场景，支持 GemmV3 输出 FP32。

4. 补齐 GemmV3 OpDef 类型声明

补充以下四组类型和格式组合：

前两组用于 ND×ND 低精度 bias 的 16in32out 场景，后两组用于 AddmmWeightNz 在 cubeMathType=USE_FP32_ADD 下保持低精度输出的场景。

5. 补齐 A2/A3 GemmV3 binary 配置

在以下配置中补充 ND×NZ 同类型输出 Kernel：

• ascend910b/gemm_v3_binary.json
• ascend910_93/gemm_v3_binary.json

新增配置：

未修改 ascend950 和 ascend350 的 binary 配置。

6. 支持的 bias broadcast shape

Addmm/AddmmWeightNz

输出 shape 为 [M,N]：

Baddbmm

输出 shape 为 [B,M,N]：

关联的Issue

Issue #4844

测试

1. ATK 精度泛化用例

在 A2/A3 环境执行2000条 ATK L0 双标杆精度泛化用例：

精度用例覆盖：

• FP16/BF16 矩阵输入；
• FP16、BF16及FP32输出；
• ND×ND 和 ND×NZ；
• cubeMathType=USE_FP32_ADD；
• FP16/BF16 16in32out；
• transA/transB 四种组合；
• N、M、scalar 和 per-batch scalar broadcast；
• 对齐与非对齐 shape；
• 大小 shape 及尾块场景。

双标杆精度首轮结果为 1998/2000 通过，两条失败用例定点复检均通过，未发现稳定复现的精度问题。

2. 性能验证

执行100条三通路性能对比用例：

对比以下三条通路：

1. GemmV3 + broadcast bias；
2. GemmV3 + 等价 full-shape bias；
3. ATB PPMatMul + Muls + Axpy。

GemmV3 与 PPMatMul 使用相同的原始 shape、物理布局、transpose 和 contiguous 条件。

性能结果：

• 小算子链/GemmV3 耗时比中位数：2.2852；
• 比值范围：1.1051～2.8523；
• GemmV3 broadcast 不慢于小算子拼接：100/100；
• 三条通路输出一致性检查：100/100通过。

文档更新

更新以下 ACLNN 接口文档：

• aclnnAddmm&aclnnInplaceAddmm.md
• aclnnBaddbmm&aclnnInplaceBaddbmm.md
• aclnnAddmmWeightNz.md

文档更新内容包括：

• A2/A3 上 cubeMathType=USE_FP32_ADD 对 broadcast bias 的支持说明；
• AddmmWeightNz 对 cubeMathType=USE_FP32_ADD 的支持说明；
• AddmmWeightNz 支持的 [M,1]、[1,1] 等 broadcast shape；
• FP16/BF16 输入、低精度输出和 FP32 中间计算行为说明。
• GemmV3 broadcast bias 的 Kernel 设计和精度、性能验证方法由对应设计文档及测试 README 说明。

类型标签

• Bug修复
• 新特性
• 性能优化
• 文档更新
• 其他，请描述：

AI/Agent生成声明

• AI辅助编写