| 实现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!1150 | 3 个月前 |