描述

在 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

https://gitcode.com/cann/ops-nn/issues/4807

测试

文档更新

类型标签

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

AI/Agent生成声明

• AI辅助编写