ops-nn/activation/softmax_grad_ext/op_host · CANN/ops-nn

cann-robotfeat(softmax_grad_ext): 新增 Ascend950 融合规则及 infershape 并补充 UT

文件	最后提交记录	最后更新时间
arch35	softmaxgradext算子订正tiling校验拦截报错 Co-authored-by: qq_52874943<gaomingyuan3@h-partners.com> # message auto-generated for no-merge-commit merge: !8108 merge master into master softmaxgradext算子订正tiling校验拦截报错 Created-by: qq_52874943 Commit-by: qq_52874943 Merged-by: cann-robot Description: ## 描述 softmaxgradext算子订正tiling校验拦截报错 ## 关联的Issue 关联Issue[#4441](https://gitcode.com/cann/ops-nn/issues/4441) ## 测试本地已完成新精度对比与性能测试，通过静态检查、蓝区冒烟、算子ST。 ## 文档更新修正文档中算子命名不规范问题，打开Ascend950支持选项。 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!8108	17 天前
config	修复代码中的告警问题 Co-authored-by: yuanbin_22<yuanbin22@huawei.com> # message auto-generated for no-merge-commit merge: !4995 merge CleanCode0519 into master 修复代码中的告警问题 Created-by: yuanbin_22 Commit-by: yuanbin_22 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 修复代码中的告警问题 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/1323 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!4995	2 个月前
softmax_grad_ext_def.cpp	修复代码中的告警问题 Co-authored-by: yuanbin_22<yuanbin22@huawei.com> # message auto-generated for no-merge-commit merge: !4995 merge CleanCode0519 into master 修复代码中的告警问题 Created-by: yuanbin_22 Commit-by: yuanbin_22 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 修复代码中的告警问题 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/1323 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!4995	2 个月前
softmax_grad_ext_infershape.cpp	feat(softmax_grad_ext): 新增 Ascend950 融合规则及 infershape 并补充 UT Co-authored-by: tianye<tianye82@h-partners.com> # message auto-generated for no-merge-commit merge: !8089 merge br_fusion_softmaxgradext into master feat(softmax_grad_ext): 新增 Ascend950 融合规则及 infershape 并补充 UT Created-by: transformer910 Commit-by: tianye Merged-by: cann-robot Description: ## 描述在 activation/softmax_grad_ext 下新增 SoftmaxGradExt 融合规则，使用新框架 PatternFusionPass 写法重写 canndev 原版 v1/v2 两个 pass，逻辑保持一致；同时补充 infershape 实现及完整 UT。主要改动 ========= 1. 融合规则 (op_graph/fusion_pass/) - softmax_grad_ext_fusion_pass.h/.cpp - SoftmaxGradExtFusionPass (v1)：匹配 mul=Mul(grad,x1) -> sum=ReduceSum(mul) -> sub=Sub(grad,sum) -> mul1=Mul(x2,x1) -> mulGrad=Mul(mul1,sub)，替换为单个 SoftmaxGradExt(grad, x1, x2)。 - SoftmaxGradExtV2FusionPass (v2)：4 个变体，覆盖 mul1/mulGrad 输入顺序的 4 种排列，其余结构与 v1 一致。 - axis 从匹配到的 ReduceSum const 输入 (index 1) 读取， keep_dims 从属性读取，透传到 SoftmaxGradExt 的 axes/keep_dims 属性，与 canndev 原版 ops::GetConstIntData(op, 1, axis) 逻辑一致。 - 平台校验：仅 Ascend950 生效；未知 shape 跳过。 - 注册阶段：CustomPassStage::kAfterInferShape。 - 成环问题修复：pattern 中 ReduceSum 的 axes 使用 CreateConst（内部 Const 节点）而非 CreateInput（Data 边界节点）。CreateInput 会创建第 4 个 Data 节点，其 required-but-unconnected input anchor 在某些 GE 库版本下导致 TopologicalSorting 判定 pattern graph 不可达节点，报 "cycle in pattern graph"。改回 CreateConst 后 pattern graph 为标准 3-Data + Const 结构，与 canndev 原版一致。 2. InferShape (op_host/) - softmax_grad_ext_infershape.cpp - y.shape = grad.shape (input0) - y.dtype = grad.dtype (input0) 3. UT - tests/ut/op_graph/test_softmax_grad_ext_fusion_pass.cpp (18 用例) - v1/v2 各变体 fp16/fp32、不同 axis 与 keep_dims 组合 - 未知 shape、非 950 平台跳过 - v1↔v2 互不误匹配 - Mul 输入顺序调换不误匹配 - 控制边成环场景 (WillCauseCycleIfFuse 检测) - axes const 共享场景 - mulGrad 控制消费者场景 - DumpToFile 输出图、node_count 校验 - tests/ut/op_host/test_softmax_grad_ext_infershape.cpp (3 用例) - fp16 3D / fp32 4D shape 推导、dtype 推导 4. 其他 - op_graph/CMakeLists.txt：add_graph_plugin_sources() - tests/ut/op_graph/CMakeLists.txt：注册 OP_GRAPH_UT - tests/ut/op_host/CMakeLists.txt：启用 OP_INFERSHAPE_MODULE - tests/ut/op_host/arch35/test_softmax_grad_ext_tiling.cpp：注释 dlog_setlevel 调用（预存编译错误，与 rms_norm_grad 修复方式一致）验证 ==== - infershape UT: 3 passed - fusion UT: 18 passed - 编译包 (--ops=softmax_grad_ext --soc=ascend950): 成功 - clang-format / trailing-whitespace / end-of-file: 通过 pre-commit 检查 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!8089	4 天前