已合并
segment_sum算子支持simd确定性计算以及性能优化 #1831
z30075199创建于 2月11日
segment_sum算子支持simd确定性计算以及性能优化 #1831
已合并
Pull Request已成功合入, 合并人@CANN-robot
（感谢 z30075199 的贡献）
CANN-robot
2月11日 评论：
2月11日 评论：
Thanks for your pull-request.
The full list of commands accepted by me can be found at here。
You can get sig-info at here
PR Approval Progress
✅ Congratulations! All modules have met the lgtm and approve requirements.
Module Approval Details
| module | lgtm status | approve status |
|---|---|---|
| index | ✅ wangyongguang, chenqi317 (2/2) | ✅ wangyongguang, chenqi317 (2/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CANN-robot2月11日 将FelixTang7,crystalhu,fanqirui,zhou-qilong,chenqi317,liubo75,tangweiwei2,wangyongguang,zhang-wu设为评审人
2月11日 将FelixTang7,crystalhu,fanqirui,zhou-qilong,chenqi317,liubo75,tangweiwei2,wangyongguang,zhang-wu设为评审人
2月11日 添加了label：cann-cla/yes
CANN-robot2月11日 将FelixTang7,crystalhu,zhou-qilong,chenqi317,liubo75,tangweiwei2,wangyongguang,zhang-wu设为审查人
2月11日 将FelixTang7,crystalhu,zhou-qilong,chenqi317,liubo75,tangweiwei2,wangyongguang,zhang-wu设为审查人
此处折叠了64条消息 查看更多
2月13日 添加了label：lgtm
CANN-robot
2月13日 评论：
2月13日 评论：
Review Guide
This pull-request passes review.
Committers who wrote a comment of /approve are: wangyongguang, chenqi317.
Reviewers who wrote a comment of /lgtm are: wangyongguang, chenqi317.
2月13日 关闭了关联的issue
2月13日 合入了pull request
描述
1、针对小尾轴场景的shape，将simt模板进行方案重构以优化性能：
使用segment_ids分核，利用simd计算核内唯一索引的个数和位置，simt对每个唯一索引对应的所有行进行累加。
2、新增SIMD确定性分支，相同输入产生相同输出。
关联的Issue
https://gitcode.com/cann/ops-nn/issues/1055
测试
文档更新
类型标签