支持 AutoFuse 在多 reduce 融合场景下按 reduce 节点名保存和读取原始轴信息,避免多个 reduce 共用单份 reduce_original_axis/repeats 时互相覆盖。同时移除 reduce 原始轴不一致时的保守拒绝融合逻辑,让不同原始轴的 reduce 节点具备融合能力。
reduce_original_axis/repeats
请选择本次引入的变更类型:
无
git diff --check HEAD~1..HEAD
tc_af_tf1_BrctReduceElimination_0009
feat:
fix:
本 PR 保持 IfNormLikeReduce 直接返回 false,不放开已知存在精度风险的 norm-like reduce 融合,仅提供不同 reduce 原始轴的多 reduce 场景融合能力。
IfNormLikeReduce
Pull Request
描述
支持 AutoFuse 在多 reduce 融合场景下按 reduce 节点名保存和读取原始轴信息,避免多个 reduce 共用单份
reduce_original_axis/repeats时互相覆盖。同时移除 reduce 原始轴不一致时的保守拒绝融合逻辑,让不同原始轴的 reduce 节点具备融合能力。变更类型
请选择本次引入的变更类型:
关联的Issue
无
如何测试
git diff --check HEAD~1..HEAD:通过,无输出。tc_af_tf1_BrctReduceElimination_0009定位并确认 reduce 前原始轴 repeat 可按节点区分。核对清单
其他信息
本 PR 保持
IfNormLikeReduce直接返回 false,不放开已知存在精度风险的 norm-like reduce 融合,仅提供不同 reduce 原始轴的多 reduce 场景融合能力。