已合并
scatter_nd_update support uncontigous for dim0 #5510
scatter_nd_update support uncontigous for dim0 #5510
已合并
zhangqijia1创建于 6月1日
zhangqijia1
6月1日

描述

新增 scatter_nd_update 算子的 AscendC 实现,使其支持首轴(dim0)非连续的输入 tensor。

主要改动:

  • aclnn 接口(aclnn_scatter_nd_update)适配首轴非连续场景;
  • op_host 新增 arch32 tiling 实现(scatter_nd_update_tiling.cpp/.h)及算子原型定义(scatter_nd_update_def.cpp);
  • op_kernel 新增 arch32 kernel 实现,包含 large_index / linear_index / no_sort 等分支,并提供 common 头文件供复用;
  • 新增 ascend910b、ascend910_93 平台的 binary json 与 simplified key 配置。

关联的Issue

关联 Issue #2703

文档更新

无。

类型标签

likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 zhangqijia1 的贡献)
Zzhangqijia1
6月1日 创建了 pull request,commit ccc20525
CANN-robotCANN-robot成员
6月1日 添加了label:cann-cla/yes
CANN-robot
CANN-robot成员
6月1日 评论:

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
*/*/op_api/*.h Andy-lb, 胡碧霞 (2/2) Andy-lb (1/1)
*/*/op_host/*_def.cpp Andy-lb, 胡碧霞 (2/2) Andy-lb (1/1)
branch-9.0.0 Andy-lb, 胡碧霞 (2/2) Andy-lb (1/1)
index Andy-lb, 胡碧霞 (2/2) Andy-lb (1/1)

💡 Tip:

  • Committer can comment /approve or /lgtm
  • Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

gcw_SUaZx3UQ, thanks for your pull request. All authors of the commits have signed the CLA. 👍

likedislike
zhangqijia1
6月1日 评论:

compile

likedislike
Zzhangqijia1
6月1日 update merge request[project id: 7665709, iid: 5510, commit_id: 7ef266fe38af73d02491278b4429d7dccd1f8ac8] virtual merging success
此处折叠了145条消息 查看更多
Andy-lb成员
6月2日 评论:

/approve

likedislike
CANN-robotCANN-robot成员
6月2日 添加了label:approved
胡碧霞成员
6月2日 评论:

/lgtm

likedislike
CANN-robotCANN-robot成员
6月2日 添加了label:lgtm
CANN-robotCANN-robot成员
6月2日 合入了pull request