| Optimize QBMM MX tensor API scalar path Co-authored-by: smdbha<panzhijie2@huawei.com> # message auto-generated for no-merge-commit merge: !77 merge master into master Optimize QBMM MX tensor API scalar path Created-by: smdbha Commit-by: smdbha Merged-by: cann-robot Description: ## 描述 本 PR 优化 QBMM MX Tensor API 的 scalar/unbatched 路径，新增 QuantMmUnbatchedMX kernel，并同步整理 QBMM MX 相关 block scheduler、block mmad 和文档接口。 主要修改包括： 1. 新增 kernel_qbmm_mx_unbatched.h，为 QBMM MX scalar/unbatched 场景提供独立 kernel 路径，避免复用 batch 调度路径带来的额外开销。 2. 调整 BlockSchedulerQuantBatchMatmulV3 的 tile 坐标获取方式，将 tail split 的 M/N offset 放入 BlockShape，减少调度状态耦合。 3. 优化 BlockMmad MX 路径中的 scale 拷贝与 K 方向切分处理，减少重复 scale 搬运，同时保持原有 Mmad 计算参数和计算语义不变。 4. 更新 kernel_qbmm_mx.h、kernel_qbmm_cube.h 对新 scheduler 接口的调用。 5. 新增 Align64/Align32/Align16 等公共对齐工具函数。 6. 同步更新 QBMM MX block mmad 和 scheduler 相关 API 文档。 该修改聚焦 QBMM MX Tensor API scalar/unbatched 路径优化，不改变现有矩阵乘计算逻辑。 ## 关联的Issue 关联 Issue：https://gitcode.com/cann/ops-tensor/issues/4 ## 测试 已完成： 1. 静态检查：git diff --check / git show --check 通过。 2. 代码检视：确认 scheduler 坐标迁移、scale 复用逻辑、unbatched 路径切片方式与现有 QBMM MX 计算逻辑一致。 3. 影响面检查：现有调用点已适配新的 GetTileCoord 接口，未发现会改变 batched/cube 既有计算结果的修改。 待补充： 1. 二级冒烟测试结果。 2. QBMM MX scalar/unbatched 典型用例精度与性能回归结果。 3. 现有 batched/cube 场景回归结果。 ## 文档更新 已更新以下文档： 1. docs/API/gemm/block/block_mmad_qbmm_mx.md 2. docs/API/gemm/block/block_scheduler.md 3. docs/API/gemm/block/block_scheduler_qbmm_mx_basic.md ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [x] 性能优化 - [x] 文档更新 - [x] 其他，请描述：QBMM MX scalar/unbatched 路径优化与内部接口整理 See merge request: cann/ops-tensor!77 | 8 天前 |
