| 对 sparse_flash_mla_grad 和 sparse_lightning_indexer_kl_loss_grad 两个算子的文档进行修正,并对 touch 文档进行对齐 Co-authored-by: ThunLoilu<202313410048@nuist.edu.cn> # message auto-generated for no-merge-commit merge: !8471 merge master into master 对 sparse_flash_mla_grad 和 sparse_lightning_indexer_kl_loss_grad 两个算子的文档进行修正,并对 touch 文档进行对齐 Created-by: ThunLoilu Commit-by: ThunLoilu Merged-by: cann-robot Description: ## 描述 对 sparse_flash_mla_grad 和 sparse_lightning_indexer_kl_loss_grad 两个算子的文档进行多处修正与对齐,涉及以下 4 项改动: ### 1. 字段支持信息修正 修正 README 和 aclnn 文档中若干可选字段(sequsedQOptional、sequsedOriKvOptional、sequsedCmpKvOptional、oriTopkLengthOptional、cmpTopkLengthOptional、metadataOptional)的描述: - 去掉"预留字段"的表述,改为实际含义描述(如"表示每个batch实际使用的q长度") - 补充 Atlas A2/A3 暂不支持这些字段的说明 ### 2. Touch 文档与 aclnn 文档对齐 将 touch 接口文档(torch_extension/cann_ops_transformer/docs/zh/)与 aclnn 文档对齐,主要修改: - 新增 Ascend 950PR/Ascend 950DT 的产品支持标记(原为×,改为√) - 将 seqused_q/seqused_ori_kv/seqused_cmp_kv 等字段从"当前仅支持传 None"改为实际支持描述 - 补充 ori_mask_mode、cmp_mask_mode 各平台支持的取值说明 - 补充 ori_softmax_l1norm、cmp_softmax_l1norm 输出参数描述 - 新增确定性计算说明和各平台规格约束表格 ### 3. K 支持范围修正 aclnnSparseLightningIndexerKLLossGrad 文档中 Ascend 950PR/Ascend 950DT 的 K 支持范围从 "0~2048" 修正为 "1~2048"。 ### 4. 可选输入描述修正 对齐 sparse_lightning_indexer_kl_loss_grad 的 zh 文档中 seqused_q/seqused_k 的描述。 ## 关联的Issue 无 ## 测试 文档修改,不涉及代码逻辑,无需测试。 ## 文档更新 - attention/sparse_flash_mla_grad/README.md - attention/sparse_flash_mla_grad/docs/aclnnSparseFlashMlaGrad.md - attention/sparse_flash_mla_grad/docs/zh/sparse_flash_mla_grad.md - attention/sparse_lightning_indexer_kl_loss_grad/README.md - attention/sparse_lightning_indexer_kl_loss_grad/docs/aclnnSparseLightningIndexerKLLossGrad.md - attention/sparse_lightning_indexer_kl_loss_grad/docs/zh/sparse_lightning_indexer_kl_loss_grad.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] 🐛 Bug修复 - [ ] ✨ 新特性 - [ ] ⚡ 性能优化 - [ ] ♻️ 重构 - [ ] 🧪 测试 - [ ] 📦 构建/CI - [ ] 🔧 配置变更 - [x] 📝 文档更新 - [ ] ⬆️ 依赖升级 - [ ] 🔒 安全修复 - [ ] 🧹 代码清理 - [ ] ❓ 其他,请描述: See merge request: cann/ops-transformer!8471 | 8 天前 |