34761009创建于 3月31日历史提交

文件	最后提交记录	最后更新时间
examples	个人-AscendC实现Sort算子贡献 Co-authored-by: LePenseur<zhoujianhua@stu.hit.edu.cn> # message auto-generated for no-merge-commit merge: !435 merge feat/sort into master 个人-AscendC实现Sort算子贡献 Created-by: LePenseur Commit-by: LePenseur Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 使用AscendC对TBE实现的Sort算子进行重构，实现了AscendC实现的Sort算子对Atlas 200/500 A2推理产品和Atlas 800I/T A2硬件的适配。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> 关联Issue：https://gitcode.com/cann/ops-math/issues/288 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 采用aclnn调用测试 ![12.18sort测试截图.jpg](https://raw.gitcode.com/user-images/assets/7649531/418a889c-3c61-4ec2-8136-127be2865b3e/12.18sort测试截图.jpg '12.18sort测试截图.jpg') ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!435	5 个月前
op_graph	个人-AscendC实现Sort算子贡献 Co-authored-by: LePenseur<zhoujianhua@stu.hit.edu.cn> # message auto-generated for no-merge-commit merge: !435 merge feat/sort into master 个人-AscendC实现Sort算子贡献 Created-by: LePenseur Commit-by: LePenseur Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 使用AscendC对TBE实现的Sort算子进行重构，实现了AscendC实现的Sort算子对Atlas 200/500 A2推理产品和Atlas 800I/T A2硬件的适配。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> 关联Issue：https://gitcode.com/cann/ops-math/issues/288 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 采用aclnn调用测试 ![12.18sort测试截图.jpg](https://raw.gitcode.com/user-images/assets/7649531/418a889c-3c61-4ec2-8136-127be2865b3e/12.18sort测试截图.jpg '12.18sort测试截图.jpg') ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!435	5 个月前
op_host	个人-AscendC实现Sort算子贡献 Co-authored-by: LePenseur<zhoujianhua@stu.hit.edu.cn> # message auto-generated for no-merge-commit merge: !435 merge feat/sort into master 个人-AscendC实现Sort算子贡献 Created-by: LePenseur Commit-by: LePenseur Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 使用AscendC对TBE实现的Sort算子进行重构，实现了AscendC实现的Sort算子对Atlas 200/500 A2推理产品和Atlas 800I/T A2硬件的适配。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> 关联Issue：https://gitcode.com/cann/ops-math/issues/288 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 采用aclnn调用测试 ![12.18sort测试截图.jpg](https://raw.gitcode.com/user-images/assets/7649531/418a889c-3c61-4ec2-8136-127be2865b3e/12.18sort测试截图.jpg '12.18sort测试截图.jpg') ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!435	5 个月前
op_kernel	个人-AscendC实现Sort算子贡献 Co-authored-by: LePenseur<zhoujianhua@stu.hit.edu.cn> # message auto-generated for no-merge-commit merge: !435 merge feat/sort into master 个人-AscendC实现Sort算子贡献 Created-by: LePenseur Commit-by: LePenseur Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 使用AscendC对TBE实现的Sort算子进行重构，实现了AscendC实现的Sort算子对Atlas 200/500 A2推理产品和Atlas 800I/T A2硬件的适配。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> 关联Issue：https://gitcode.com/cann/ops-math/issues/288 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 采用aclnn调用测试 ![12.18sort测试截图.jpg](https://raw.gitcode.com/user-images/assets/7649531/418a889c-3c61-4ec2-8136-127be2865b3e/12.18sort测试截图.jpg '12.18sort测试截图.jpg') ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!435	5 个月前
tests	个人-AscendC实现Sort算子贡献 Co-authored-by: LePenseur<zhoujianhua@stu.hit.edu.cn> # message auto-generated for no-merge-commit merge: !435 merge feat/sort into master 个人-AscendC实现Sort算子贡献 Created-by: LePenseur Commit-by: LePenseur Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 使用AscendC对TBE实现的Sort算子进行重构，实现了AscendC实现的Sort算子对Atlas 200/500 A2推理产品和Atlas 800I/T A2硬件的适配。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> 关联Issue：https://gitcode.com/cann/ops-math/issues/288 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 采用aclnn调用测试 ![12.18sort测试截图.jpg](https://raw.gitcode.com/user-images/assets/7649531/418a889c-3c61-4ec2-8136-127be2865b3e/12.18sort测试截图.jpg '12.18sort测试截图.jpg') ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!435	5 个月前
CMakeLists.txt	个人-AscendC实现Sort算子贡献 Co-authored-by: LePenseur<zhoujianhua@stu.hit.edu.cn> # message auto-generated for no-merge-commit merge: !435 merge feat/sort into master 个人-AscendC实现Sort算子贡献 Created-by: LePenseur Commit-by: LePenseur Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 使用AscendC对TBE实现的Sort算子进行重构，实现了AscendC实现的Sort算子对Atlas 200/500 A2推理产品和Atlas 800I/T A2硬件的适配。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> 关联Issue：https://gitcode.com/cann/ops-math/issues/288 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 采用aclnn调用测试 ![12.18sort测试截图.jpg](https://raw.gitcode.com/user-images/assets/7649531/418a889c-3c61-4ec2-8136-127be2865b3e/12.18sort测试截图.jpg '12.18sort测试截图.jpg') ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!435	5 个月前
README.md	个人-AscendC实现Sort算子贡献 Co-authored-by: LePenseur<zhoujianhua@stu.hit.edu.cn> # message auto-generated for no-merge-commit merge: !435 merge feat/sort into master 个人-AscendC实现Sort算子贡献 Created-by: LePenseur Commit-by: LePenseur Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 使用AscendC对TBE实现的Sort算子进行重构，实现了AscendC实现的Sort算子对Atlas 200/500 A2推理产品和Atlas 800I/T A2硬件的适配。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> 关联Issue：https://gitcode.com/cann/ops-math/issues/288 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 采用aclnn调用测试 ![12.18sort测试截图.jpg](https://raw.gitcode.com/user-images/assets/7649531/418a889c-3c61-4ec2-8136-127be2865b3e/12.18sort测试截图.jpg '12.18sort测试截图.jpg') ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!435	5 个月前

Sort

贡献说明

贡献者	贡献方	贡献算子	贡献时间	贡献内容
Zhoujianhua	个人开发者	Sort	2025/12/18	新增Sort算子

支持的产品型号

Atlas A2 训练系列产品（Ascend 910B）
Atlas 200I/500 A2 推理产品（Ascend 310B）
Atlas 推理系列产品 AI Core（Ascend 310P）
Atlas 训练系列产品（Ascend 910）

算子描述

功能描述

Sort算子沿指定轴对输入张量进行排序，返回排序后的数值张量和对应的索引张量。

原型信息

算子类型(OpType)	Sort
	name	Type	data type	format
算子输入	x	tensor	float16、float	ND
算子输入	index	tensor	uint32	ND
算子输出	y1	tensor	float16、float	ND
算子输出	dstIndex	tensor	uint32	ND
核函数名	sort

约束与限制

支持 float16, float32 数据类型。
输入 shape 暂时只支持二维，可以指定其中任一维度排序，输出排序结果以及排序后的索引顺序（可选）。
支持升序和降序排序，排序的稳定性取决于sort接口。
不支持广播机制，仅支持对给定维度的独立排序。

算子使用

使用该算子前，请参考社区版CANN开发套件包安装文档完成开发运行环境的部署。

编译部署

进入到仓库目录
```
cd ${git_clone_path}/ops-math
```

执行编译

bash build.sh --experimental --ops=sort_v2 --soc=ascend910b --pkg

部署算子包

./build_out/cann-ops-math-custom_linux-aarch64.run

算子调用

执行调用

bash build.sh --run_example sort_v2 eager cust --vendor_name=custom

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_sort_v2	通过aclnnSortV2接口方式调用Sort算子。