已合并
[feat]Bucketize新增Aicore通路支持 #4711
jinpenghe创建于 4月13日
[feat]Bucketize新增Aicore通路支持 #4711
已合并
jinpenghe创建于 4月13日
jinpenghe
jinpenghe
4月13日

【合入来源】

如有社区issue,请关联issue链接
请勿携带内部流程信息(需求链接、问题单、内部issue等)

【修改方案】

芯片版本大于等于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')
  • 通路结果:
    image.png

【CheckList】

PR提交人对以下CheckList自检项进行全量自检,自检通过或不涉及,均修改 [ ] 为 [x]

likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 jinpenghe 的贡献)
jinpenghejinpenghe
4月13日 创建了 pull request,commit ce054b47
ascend-robot
ascend-robot成员
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 /approve or /lgtm
  • Commenting /approve implies 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. 👍

likedislike
ascend-robotascend-robot成员
4月13日 添加了label:ascend-cla/yes
jinpenghe
jinpenghe
4月13日 评论:

compile

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

/lgtm

likedislike
ZhihaoLi
ZhihaoLi成员
4月29日 评论:

/approve

likedislike
ascend-robotascend-robot成员
4月29日 添加了label:approvedlgtm
ascend-robot
ascend-robot成员
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.

likedislike
ascend-robotascend-robot成员
4月29日 合入了pull request