Star0

Star0

cann-robotfix(unsorted_segment_sum): 修复 uss 算子 golden 的三方比对参考实现

2648dc4fAdd on 22 days agoHistory

Files	Last commit	Last update
docs	docs: 更新 UnsortedSegmentSum 产品命名及确定性说明 Co-authored-by: aiteer<lujiale4@huawei.com> # message auto-generated for no-merge-commit merge: !11260 merge aclnn_uss_doc into master docs: 更新 UnsortedSegmentSum 产品命名及确定性说明 Created-by: aiteer Commit-by: aiteer Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> * 根据最新仓文档模板修改产品名称 * 删除算子层文档中的确定性约束说明 * 修正aclnn接口文档中参数取值支持的dtype，int64_t -> INT64 * 修正aclnn接口文档中确定性说明的语句 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> [#6251](https://gitcode.com/cann/ops-nn/issues/6251) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> * uss 的算子层文档 * uss 的aclnn接口文档 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!11260	1 month ago
examples	feat: 新增 UnsortedSegmentSum 算子支持 Aclnn 接口 Co-authored-by: aiteer<lujiale4@huawei.com> # message auto-generated for no-merge-commit merge: !10174 merge aclnn_unsortedsegmentsum into master feat: 新增 UnsortedSegmentSum 算子支持 Aclnn 接口 Created-by: aiteer Commit-by: aiteer Merged-by: cann-robot Description: ## 描述 * 新增 `UnsortedSegmentSum` 算子的 aclnn 单算子两段式接口，使其支持通过 `aclnnUnsortedSegmentSum` 直接调用（此前该算子仅支持图模式/GEIR 调用）。仅支持 Ascend 950PR/Ascend 950DT 产品。 * 新增 `aclnnUnsortedSegmentSum` 调用样例代码，向用户展示如何直调 aclnnUnsortedSegmentSum 两段式接口。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> [#6193](https://gitcode.com/cann/ops-nn/issues/6193) ## 测试 * 本地构建aclnnUnsortedSegmentSum用例验证aclnn直调。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> * 更新op_api_list.md文档，将aclnnUSS添加进去。 * 更新了算子README.md文件，增加约束说明与aclnn调用方式。 * 增加aclnnUnsortedSegmentSum.md文档，对aclnn接口进行说明。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!10174	1 month ago
framework	TF plugin修改1 Co-authored-by: sunchun<sunchun4@h-partners.com> # message auto-generated for no-merge-commit merge: !7547 merge nn1 into master TF plugin修改1 Created-by: sunchun Commit-by: sunchun Merged-by: cann-robot Description: ## 描述 TF plugin修改 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/4196 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!7547	3 months ago
op_api	fix(unsorted_segment_sum): 修复空tensor零填充路径支持uint32/uint64 Co-authored-by: aiteer<lujiale4@huawei.com> # message auto-generated for no-merge-commit merge: !11721 merge unsorted_segment_sum_fill_uint_fix into master fix(unsorted_segment_sum): 修复空tensor零填充路径支持uint32/uint64 Created-by: aiteer Commit-by: aiteer Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> * FillOutWithZeros 依赖的 Fill 算子缺少 UINT32/UINT64 二进制，在aclnn的第一段接口中的空tensor校验零填充路径支持uint32/uint64。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> [#6691](https://gitcode.com/cann/ops-nn/issues/6691) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!11721	28 days ago
op_graph	add InferDataType for GatherElements/IndexPutV2/UnsortedSegmentSum Co-authored-by: 贾剑勇<jiajianyong123@hisilicon.com> # message auto-generated for no-merge-commit merge: !10531 merge master into master add InferDataType for GatherElements/IndexPutV2/UnsortedSegmentSum Created-by: jia-jianyong Commit-by: 贾剑勇 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> add InferDataType for GatherElements/IndexPutV2/UnsortedSegmentSum ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/6172 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> GEIR验证ok，冒烟ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!10531	1 month ago
op_host	新增unsorted segment sum 960实现 Co-authored-by: m0_46386992<wanghongbin19@huawei.com> # message auto-generated for no-merge-commit merge: !11544 merge unsorted_segment_num_A6 into master 新增unsorted segment sum 960实现 Created-by: m0_46386992 Commit-by: m0_46386992 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> unsorted_segment_sum算子支持Ascend960 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/6880 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> - 本地测试 - 冒烟测试 ## 文档更新不涉及 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!11544	23 days ago
op_kernel	新增unsorted segment sum 960实现 Co-authored-by: m0_46386992<wanghongbin19@huawei.com> # message auto-generated for no-merge-commit merge: !11544 merge unsorted_segment_num_A6 into master 新增unsorted segment sum 960实现 Created-by: m0_46386992 Commit-by: m0_46386992 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> unsorted_segment_sum算子支持Ascend960 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/6880 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> - 本地测试 - 冒烟测试 ## 文档更新不涉及 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!11544	23 days ago
tests	fix(unsorted_segment_sum): 修复 uss 算子 golden 的三方比对参考实现 Co-authored-by: aiteer<lujiale4@huawei.com> # message auto-generated for no-merge-commit merge: !12090 merge unsorted_segment_sum_third_party into master fix(unsorted_segment_sum): 修复 uss 算子 golden 的三方比对参考实现 Created-by: aiteer Commit-by: aiteer Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> * UnsortedSegmentSum算子golden无法统计竞品性能，该pr修复 uss 算子 golden 的三方比对参考实现 * 本次竞品采集仅针对 aclnn 通路，kernel 通路不在本次范围 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> [#7002](https://gitcode.com/cann/ops-nn/issues/7002) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ttk三方测试 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!12090	22 days ago
CMakeLists.txt	新增unsorted segment sum 960实现 Co-authored-by: m0_46386992<wanghongbin19@huawei.com> # message auto-generated for no-merge-commit merge: !11544 merge unsorted_segment_num_A6 into master 新增unsorted segment sum 960实现 Created-by: m0_46386992 Commit-by: m0_46386992 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> unsorted_segment_sum算子支持Ascend960 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/6880 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> - 本地测试 - 冒烟测试 ## 文档更新不涉及 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!11544	23 days ago
README.md	docs: 更新 UnsortedSegmentSum 产品命名及确定性说明 Co-authored-by: aiteer<lujiale4@huawei.com> # message auto-generated for no-merge-commit merge: !11260 merge aclnn_uss_doc into master docs: 更新 UnsortedSegmentSum 产品命名及确定性说明 Created-by: aiteer Commit-by: aiteer Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> * 根据最新仓文档模板修改产品名称 * 删除算子层文档中的确定性约束说明 * 修正aclnn接口文档中参数取值支持的dtype，int64_t -> INT64 * 修正aclnn接口文档中确定性说明的语句 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> [#6251](https://gitcode.com/cann/ops-nn/issues/6251) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> * uss 的算子层文档 * uss 的aclnn接口文档 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!11260	1 month ago

UnsortedSegmentSum

产品支持情况

产品	是否支持
Ascend 950PR&950DT系列产品	√
Atlas A3系列产品	×
Atlas A2系列产品	×
Atlas 200I/500 A2推理产品	×
Atlas推理系列产品	×
Atlas训练系列产品	×

功能说明

算子功能: 对一个张量分段求和
公式说明： ${\sum}_{j...:segment_ids[j...]==i} x[j...]$ 其中，对满足segment_ids[j...] == i的所有位置j... 求和。若某个段i没有对应的元素，则y[i] = 0。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	输入数据	FLOAT32、FLOAT16、BFLOAT16、INT32、INT64、UINT32、UINT64	ND
segment_ids	输入	分段索引	INT32、INT64	ND
num_segments	输入	分段个数	INT32、INT64	ND
y	输出	输出值信息	FLOAT32、FLOAT16、BFLOAT16、INT32、INT64、UINT32、UINT64	ND

约束说明

无

调用说明

调用方式	样例代码	说明
aclnn调用	test_aclnn_unsorted_segment_sum	通过aclnnUnsortedSegmentSum接口方式调用UnsortedSegmentSum算子。
图模式调用	test_geir_unsorted_segment_sum	通过算子IR构图方式调用UnsortedSegmentSum算子。