| Files | Last commit | Last update |
|---|---|---|
fix: 修复 moe_token_permute_with_ep_grad/moe_token_unpermute_grad UT脚本 Co-authored-by: fazhenyao<fazhenyao@h-partners.com> # message auto-generated for no-merge-commit merge: !13952 merge fix-issue-5796-moe-grad-ut into master fix: 修复 moe_token_permute_with_ep_grad/moe_token_unpermute_grad UT脚本 Created-by: fazhenyao123 Commit-by: fazhenyao Merged-by: cann-robot Description: ## 描述 修复 moe_token_permute_with_ep_grad 与 moe_token_unpermute_grad 的 kernel UT：原造数脚本存在类型转换和参数处理错误，部分 UT 未加载有效输入，也未比较输出，导致无法发现精度回归。 - 重写 CPU golden 造数，严格解析布尔参数，正确保存 FP32、FP16、BF16 二进制数据，并使用非恒等排列。 - UT 加载真实输入、初始化 tiling，并校验 token/probs 两路梯度；覆盖无 probs、多 top-k、局部 token 范围和 hidden size 尾块。 - 修正 unpermute UT 的编译类型配置，新增基于独立 autograd 参考的 Python 回归测试。 目标分支：master；基线提交：a4a986c76e2f8cbe6c8b1612d81bf6c288106389。改动仅涉及 6 个测试及测试配置文件。 ## 关联的Issue Fixes #5796 https://gitcode.com/cann/ops-transformer/issues/5796 ## 测试 - Python 回归测试：4 组全部通过，包含 FP32/FP16/BF16、参数校验和输出文件处理。 - A5_ci 上使用 Ascend910B1 tikicpulib 的独立 CPU 仿真构建：最终代码 7/7 个 kernel UT 通过（permute 3 个、unpermute 4 个）。此验证使用实际 kernel 源码与 UT，未执行完整仓库构建或 NPU 硬件测试。 - 4 项负向验证：分别破坏两个算子的 token/probs golden，UT 均能检测并失败；验证后已恢复原始 golden 生成脚本。 - clang-format 18.1.8、ruff 0.14.14 检查及 git diff --check 通过。 ## 文档更新 无文档更新。 ## 类型标签 - [x] 🐛 Bug修复 - [x] 🧪 测试 See merge request: cann/ops-transformer!13952 | 21 days ago |