已合并
[feat]Bucketize新增Aicore通路支持 #4711
jinpenghe创建于 4月13日
[feat]Bucketize新增Aicore通路支持 #4711
已合并
Pull Request已成功合入, 合并人@ascend-robot
(感谢 jinpenghe 的贡献)ascend-robot
4月13日 评论:
4月13日 评论:
ascend-robot
4月13日 评论:
4月13日 评论:
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 |
|---|---|---|
| repo-Ascend/op-plugin | ✅ adrian07110, liangsongwei (2/2) | ✅ adrian07110 (1/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
jinpenghe, thanks for your pull request. All authors of the commits have signed the CLA. 👍


4月13日 添加了label:ascend-cla/yes
jinpenghe
4月13日 评论:
4月13日 评论:
compile


此处折叠了237条消息 查看更多
梁松伟
4月29日 评论:
4月29日 评论:
/lgtm


ZhihaoLi
4月29日 评论:
4月29日 评论:
/approve


4月29日 添加了label:approvedlgtm
ascend-robot
4月29日 评论:
4月29日 评论:
Review Guide
This pull-request passes review.
Committers who wrote a comment of /approve are: adrian07110.
Reviewers who wrote a comment of /lgtm are: liangsongwei, adrian07110.


4月29日 合入了pull request
【合入来源】
【修改方案】
芯片版本大于等于950, 输入x和boundaries不为double情况下走aclnnBucketize
【资料变更】
不涉及
【接口变更】
不涉及
【功能验证】
values = torch.tensor( [
[-5, 10000, 1],
[150, 10, 2],
[5, 100, 3]
], dtype=torch.float32, device=device)
bins = torch.tensor([0.0, 10.0, 100.0], dtype=torch.float32, device=device)
function operator())
输入值 : tensor([[-5.0000e+00, 1.0000e+04, 1.0000e+00],
[ 1.5000e+02, 1.0000e+01, 2.0000e+00],
[ 5.0000e+00, 1.0000e+02, 3.0000e+00]], device='npu:0')
分桶边界 : tensor([ 0., 10., 100.], device='npu:0')
bucketize 结果 : tensor([[0, 3, 1],
[3, 1, 1],
[1, 2, 1]], device='npu:0')
【CheckList】