Star1

Star1

cann-robot新增SIMT SORT实现

6de02d52创建于 6月22日历史提交

文件	最后提交记录	最后更新时间
examples	新增SIMT SORT实现 Co-authored-by: y00580292<yezhao1@huawei.com> # message auto-generated for no-merge-commit merge: !3458 merge master into master 新增SIMT SORT实现 Created-by: yezhao1 Commit-by: y00580292 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3458	2 个月前
op_graph	新增SIMT SORT实现 Co-authored-by: y00580292<yezhao1@huawei.com> # message auto-generated for no-merge-commit merge: !3458 merge master into master 新增SIMT SORT实现 Created-by: yezhao1 Commit-by: y00580292 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3458	2 个月前
op_host	新增SIMT SORT实现 Co-authored-by: y00580292<yezhao1@huawei.com> # message auto-generated for no-merge-commit merge: !3458 merge master into master 新增SIMT SORT实现 Created-by: yezhao1 Commit-by: y00580292 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3458	2 个月前
op_kernel	新增SIMT SORT实现 Co-authored-by: y00580292<yezhao1@huawei.com> # message auto-generated for no-merge-commit merge: !3458 merge master into master 新增SIMT SORT实现 Created-by: yezhao1 Commit-by: y00580292 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3458	2 个月前
tests	新增SIMT SORT实现 Co-authored-by: y00580292<yezhao1@huawei.com> # message auto-generated for no-merge-commit merge: !3458 merge master into master 新增SIMT SORT实现 Created-by: yezhao1 Commit-by: y00580292 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3458	2 个月前
CMakeLists.txt	新增SIMT SORT实现 Co-authored-by: y00580292<yezhao1@huawei.com> # message auto-generated for no-merge-commit merge: !3458 merge master into master 新增SIMT SORT实现 Created-by: yezhao1 Commit-by: y00580292 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3458	2 个月前
README.md	新增SIMT SORT实现 Co-authored-by: y00580292<yezhao1@huawei.com> # message auto-generated for no-merge-commit merge: !3458 merge master into master 新增SIMT SORT实现 Created-by: yezhao1 Commit-by: y00580292 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3458	2 个月前

RadixSort

贡献说明

贡献者	贡献算子	贡献时间	贡献内容
CANN-BOT SIMT	radix_sort	2026/06/22	新增radix_sort算子

产品支持情况

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

功能说明

算子功能：使用 LSD（Least Significant Digit）基数排序算法对一维整数 Tensor 执行稳定排序，同时返回排序后的值和对应的原始索引。
计算公式：

算法按字节（8-bit）从低位到高位逐趟扫描，每趟使用 256 个桶进行直方图统计 → 前缀和 → 逆序散射。有符号整数通过 XOR 符号位翻转映射到无符号域排序。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	待排序的一维整数数组。	INT32、UINT32、INT64、UINT64	ND
sorted_values	输出	排序后的值，与输入具有相同的类型和形状。	INT32、UINT32、INT64、UINT64	ND
sorted_indices	输出	排序后每个元素对应的原始索引。	INT64	ND
descending	属性	排序方向。false 为升序，true 为降序。默认值为 false。	BOOL	-

约束说明

仅支持一维（1D）输入。
使用 LSD 基数排序算法（每趟 8-bit，256 bins）。
排序结果保证稳定性（stability），重复值的原始索引顺序保持不变。
支持的数据类型：int32、uint32、int64、uint64。

调用说明

调用方式	调用样例	说明
aclnn调用	不支持	参见算子调用完成算子编译和验证。
图模式调用	test_geir_radix_sort	参见算子调用完成算子编译和验证。