master

cann-robotfeat(scatter_elements): 支持确定性全局排序

f0d06fb6Add on Sep 10History

Files	Last commit	Last update
arch35	feat(scatter_elements): 支持确定性全局排序 Co-authored-by: yourealize<chenzhanxi1@huawei.com> # message auto-generated for no-merge-commit merge: !9766 merge codex/migrate-scatter-elements-global-sort into master feat(scatter_elements): 支持确定性全局排序 Created-by: yourealize Commit-by: yourealize Merged-by: cann-robot Description: ## 描述 ScatterElements 在 Ascend 950 确定性模式下新增全局排序路径，复用现有 SortLib，并保留 data -> y 拷贝语义。 主要修改： - Host Tiling 增加全局排序 tiling、workspace 及 2xxxxxx tiling key 路由。 - 新增 copy -> linear index -> SortLib radix sort -> grouped reduction kernel。 - FP16/BF16 分组内使用 FP32 累加；INT8/INT16 使用 INT32 累加，组间按输出 dtype 写回。 - permSize 跟随 countMode，采用 32/64 位同宽设计，不修改 SortLib。 - 新增 ScatterElements Host Tiling UT，覆盖排序路由、fallback、subset 和 64 位 permutation 边界。 本 PR 仅修改 index/scatter_elements，index/sort_lib 与目标基线一致。 ## 关联的Issue Closes #5735 ## 测试 - pre-commit run --from-ref upstream/master --to-ref HEAD：全部通过。 - clang-format、codespell、OAT Compliance Check：全部通过。 - rebase 前后 git range-diff：补丁完全等价。 - git diff --check upstream/master...HEAD：通过。 - ScatterElementsWithSortedTiling：5/5 通过。 - ScatterElementsV2WithSortedTiling：9/9 通过。 - 上游 SortLibTilingTest：10/10 通过。 - ScatterElements Ascend950PR_9599 kernel 配置：20/20 编译通过。 - 完整 Host UT：48/51 通过；其余 3 个失败为本机 CANN 9.1 下既有的 ScatterElementsV2 确定性 tiling 序列化期望差异，与本 PR 无关。 ## 文档更新 无需文档更新。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 - [x] AI辅助编写 See merge request: cann/ops-nn!9766	2 months ago
scatter_elements_apt.cpp	feat(scatter_elements): 支持确定性全局排序 Co-authored-by: yourealize<chenzhanxi1@huawei.com> # message auto-generated for no-merge-commit merge: !9766 merge codex/migrate-scatter-elements-global-sort into master feat(scatter_elements): 支持确定性全局排序 Created-by: yourealize Commit-by: yourealize Merged-by: cann-robot Description: ## 描述 ScatterElements 在 Ascend 950 确定性模式下新增全局排序路径，复用现有 SortLib，并保留 data -> y 拷贝语义。 主要修改： - Host Tiling 增加全局排序 tiling、workspace 及 2xxxxxx tiling key 路由。 - 新增 copy -> linear index -> SortLib radix sort -> grouped reduction kernel。 - FP16/BF16 分组内使用 FP32 累加；INT8/INT16 使用 INT32 累加，组间按输出 dtype 写回。 - permSize 跟随 countMode，采用 32/64 位同宽设计，不修改 SortLib。 - 新增 ScatterElements Host Tiling UT，覆盖排序路由、fallback、subset 和 64 位 permutation 边界。 本 PR 仅修改 index/scatter_elements，index/sort_lib 与目标基线一致。 ## 关联的Issue Closes #5735 ## 测试 - pre-commit run --from-ref upstream/master --to-ref HEAD：全部通过。 - clang-format、codespell、OAT Compliance Check：全部通过。 - rebase 前后 git range-diff：补丁完全等价。 - git diff --check upstream/master...HEAD：通过。 - ScatterElementsWithSortedTiling：5/5 通过。 - ScatterElementsV2WithSortedTiling：9/9 通过。 - 上游 SortLibTilingTest：10/10 通过。 - ScatterElements Ascend950PR_9599 kernel 配置：20/20 编译通过。 - 完整 Host UT：48/51 通过；其余 3 个失败为本机 CANN 9.1 下既有的 ScatterElementsV2 确定性 tiling 序列化期望差异，与本 PR 无关。 ## 文档更新 无需文档更新。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 - [x] AI辅助编写 See merge request: cann/ops-nn!9766	2 months ago