0
代码
代码
Issues
Pull Requests
讨论
Wiki
分析
0
cann-robotcann-robotsortwithindex 公共文件更名
e683f9b4创建于 23 小时前历史提交
文件最后提交记录最后更新时间
examples
sort_with_index算子开源 Co-authored-by: chensi888<chensi79@huawei.com> # message auto-generated for no-merge-commit merge: !632 merge master into master sort_with_index算子开源 Created-by: chensi79 Commit-by: chensi888 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6324 个月前
op_api
sort_with_index算子开源 Co-authored-by: chensi888<chensi79@huawei.com> # message auto-generated for no-merge-commit merge: !632 merge master into master sort_with_index算子开源 Created-by: chensi79 Commit-by: chensi888 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6324 个月前
op_graph
实现SortedWithIndex算子支持索引INT64数据类型的功能 Co-authored-by: zhaopei16<zhaopei16@huawei.com> Co-authored-by: caoyan_huawei<caoyan11@huawei.com> # message auto-generated for no-merge-commit merge: !1150 merge sortWithIndexReuseSort into master 实现SortedWithIndex算子支持索引INT64数据类型的功能 Created-by: caoyan_huawei Commit-by: caoyan_huawei;zhaopei16 Merged-by: cann-robot Description: ## 描述 SortedWithIndex算子支持索引 INT64数据类型，当前需求主要实现两个功能： 1. 支持SortWithIndex算子在RadixSort算法模板下，当排序轴输入的数据个数超过int32范围(即：int32的最大值 2147483647)时的处理，其主体流程和Sort保持一致， 包括Cutsum，前缀和，统计直方图等关键逻辑。 2. 支持SortWiithIndex算子在MergeSort算法中小模板的输入索引为int64类型的场景处理。 ## 测试 UT全量测试通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!11503 个月前
op_host
修改 sort / sort_with_index / topkv2 error message Co-authored-by: ConanHuang<huangxiaobin1@huawei.com> # message auto-generated for no-merge-commit merge: !3030 merge master into master 修改 sort / sort_with_index / topkv2 error message Created-by: ConanHuang Commit-by: ConanHuang Merged-by: cann-robot Description: ## 描述 对 sort/sort_with_index/top_k_v2 算子的报错日志进行可维测性改造，将异常场景、设计参数及异常原因等信息描述出来 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/1731 ## 测试 1 三个算子累计800+ ST用例测试通过 2 二级冒烟通过 ## 文档更新 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：优化报错日志 See merge request: cann/ops-math!30306 天前
op_kernel
sortwithindex 公共文件更名 Co-authored-by: ConanHuang<huangxiaobin1@huawei.com> # message auto-generated for no-merge-commit merge: !3123 merge fix_codecheck into master sortwithindex 公共文件更名 Created-by: ConanHuang Commit-by: ConanHuang Merged-by: cann-robot Description: ## 描述 sortwithindex 公共文件更名 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/1840 ## 测试 sort_with_index 190+ ST用例均通过 ## 文档更新 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：优化文件命名 See merge request: cann/ops-math!312323 小时前
tests
优化SortwithIndex支持输出索引类型为int64 Co-authored-by: caoyan_huawei<caoyan11@huawei.com> # message auto-generated for no-merge-commit merge: !1499 merge topk_int64_tumoshi into master 优化SortwithIndex支持输出索引类型为int64 Created-by: caoyan_huawei Commit-by: caoyan_huawei Merged-by: cann-robot Description: ## 描述 优化SortWithIndex算子，区分SingleBlock和MultiCore场景，避免tiling和实际运行的模板不一致。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 ST全量测试通过 ## 文档更新 不涉及。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!14992 个月前
CMakeLists.txt
实现SortedWithIndex算子支持索引INT64数据类型的功能 Co-authored-by: zhaopei16<zhaopei16@huawei.com> Co-authored-by: caoyan_huawei<caoyan11@huawei.com> # message auto-generated for no-merge-commit merge: !1150 merge sortWithIndexReuseSort into master 实现SortedWithIndex算子支持索引INT64数据类型的功能 Created-by: caoyan_huawei Commit-by: caoyan_huawei;zhaopei16 Merged-by: cann-robot Description: ## 描述 SortedWithIndex算子支持索引 INT64数据类型，当前需求主要实现两个功能： 1. 支持SortWithIndex算子在RadixSort算法模板下，当排序轴输入的数据个数超过int32范围(即：int32的最大值 2147483647)时的处理，其主体流程和Sort保持一致， 包括Cutsum，前缀和，统计直方图等关键逻辑。 2. 支持SortWiithIndex算子在MergeSort算法中小模板的输入索引为int64类型的场景处理。 ## 测试 UT全量测试通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!11503 个月前
README.md
doc Tools扫描错误修改 Co-authored-by: gitee-yanglulu<yanglulul@h-partners.com> # message auto-generated for no-merge-commit merge: !1899 merge master into master doc Tools扫描错误修改 Created-by: gitee-yanglulu Commit-by: gitee-yanglulu Merged-by: cann-robot Description: doc Tools扫描错误修改 See merge request: cann/ops-math!18992 个月前
README.md

SortWithIndex

产品支持情况

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

功能说明

  • 算子功能：将输入tensor按照元素值大小进行排序，index值跟随对应元素值进行排序。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 待进行元素值排序的输入张量。 FLOAT16、FLOAT、BFLOAT16、INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64 ND
index 输入 待进行元素值排序的输入张量。 INT32、INT64 ND
axis 输入 用来作为排序标准的维度。 INT64 -
descending 输入 控制排序顺序，True为降序，False为升序。 BOOL -
stable 输入 是否稳定排序, True为稳定排序，False为非稳定排序。 BOOL -
y 输出 元素值排序后的输出结果。 FLOAT16、FLOAT、BFLOAT16、INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64 ND
sorted_index 输出 索引值随元素值排序后的输出结果。 INT32、INT64 ND

约束说明

  • 输入x和index的shape必须一致

调用说明

调用方式 调用样例 说明
图模式调用 test_geir_sort_with_index 通过算子IR构图方式调用SortWithIndex算子。