已开启
fix: preserve reduce original axes per node #2
ling-DT创建于 8月6日
fix: preserve reduce original axes per node #2
已开启
ling-DT创建于 8月6日
ling-DT成员
8月6日

Pull Request

描述

支持 AutoFuse 在多 reduce 融合场景下按 reduce 节点名保存和读取原始轴信息,避免多个 reduce 共用单份 reduce_original_axis/repeats 时互相覆盖。同时移除 reduce 原始轴不一致时的保守拒绝融合逻辑,让不同原始轴的 reduce 节点具备融合能力。

变更类型

请选择本次引入的变更类型:

关联的Issue

如何测试

  1. git diff --check HEAD~1..HEAD:通过,无输出。
  2. 提交阶段 pre-commit hooks:通过(trim trailing whitespace、end of files、clang-format、codespell、OAT 等)。
  3. 按用户要求,本次创建 PR 前未重新编译;此前使用 tc_af_tf1_BrctReduceElimination_0009 定位并确认 reduce 前原始轴 repeat 可按节点区分。

核对清单

其他信息

本 PR 保持 IfNormLikeReduce 直接返回 false,不放开已知存在精度风险的 norm-like reduce 融合,仅提供不同 reduce 原始轴的多 reduce 场景融合能力。

likedislike
合并受阻