Star854
1344
代码
代码
Issues1607
Pull Requests2209
项目讨论
Wiki
分析
Star854
1344
已合并
register strategy for npu custom ops #27051
jizewei创建于 2025年11月26日
register strategy for npu custom ops #27051
已合并
jizewei创建于 2025年11月26日
从jizewei/pytorchv2.9.0_register_strategy合入到Ascend/pytorchv2.9.0
J
jizewei成员
2025年11月26日

What type of PR is this?

Uncomment only one /kind <> line, hit enter to put that in a new line, and remove leading whitespaces from that line:

/kind bug
/kind task
/kind feature

/kind feature

What does this PR do / why do we need it:
register sharding strategies for npu custom ops to support DTensor:

  1. npu_add_rms_norm
  2. npu_moe_token_permute, npu_moe_token_permute_grad
  3. npu_moe_token_unpermute, npu_moe_token_unpermute_grad
  4. npu_rotary_mul, npu_rotary_mul_backward

Which issue(s) this PR fixes:

Fixes #

Special notes for your reviewers:

likedislike
Pull Request已成功合入, 合并人@ascend-robot
（感谢 jizewei 的贡献）
Jjizewei成员
2025年11月26日 创建了 pull request，commit b857ce0a
ascend-robot
ascend-robot成员
2025年11月26日 评论：

Thank your for your pull-request.

The full list of commands accepted by me can be found at here.

You can get sig-info at here

likedislike
AtlasAccount
AtlasAccount成员
2025年11月26日 评论：
ascend-robotascend-robot成员
2025年11月26日 添加了label：stat/needs-squash
ascend-robot
ascend-robot成员
2025年11月26日 评论：

以下是根据您提交的修改文件推荐的Reviewer和Committer序列，需各模块评审通过后方可合入

Module List Reviewers Committers
repo-Ascend/pytorch li_jing_hw, shaoyf, kuhn7, yanpengquan, chuboning aichunzheng, zqwenn, dilililiwhy, wangchao430, wangchao285
likedislike
此处折叠了47条消息 查看更多
ascend-robotascend-robot成员
2025年11月27日 添加了label：lgtm
Y
yanpengquan成员
2025年11月27日 评论：

/approve

likedislike
ascend-robotascend-robot成员
2025年11月27日 添加了label：approved
ascend-robot
ascend-robot成员
2025年11月27日 评论：

Review Guide

This Pull-Request Passes Review.
Committers who writed a comment of /approve are: zqwenn, yanpengquan.
Reviewers who writed a comment of /lgtm are: zqwenn, yanpengquan.

likedislike
ascend-robotascend-robot成员
2025年11月27日 合入了pull request