已合并
fix sharding strategy for npu_fusion_attention #27780
jizewei创建于 2025年12月10日
fix sharding strategy for npu_fusion_attention #27780
已合并
Pull Request已成功合入, 合并人@ascend-robot
(感谢 jizewei 的贡献)ascend-robot
2025年12月10日 评论:
2025年12月10日 评论:
AtlasAccount
2025年12月10日 评论:
2025年12月10日 评论:
2025年12月10日 添加了label:stat/needs-squash
ascend-robot
2025年12月10日 评论:
2025年12月10日 评论:
以下是根据您提交的修改文件推荐的Reviewer和Committer序列,需各模块评审通过后方可合入
| Module List | Reviewers | Committers |
|---|---|---|
| repo-Ascend/pytorch | chenhao_1209, qinchuanyu, guoguanghao, yuhaiyan, zhaozhijie | shaoyf, wasd1111222, wangqiang160, chenhao_1209, yanpengquan |


此处折叠了69条消息 查看更多
2025年12月15日 添加了label:MindSpore-Component
2025年12月15日 添加了label:approved
ascend-robot
2025年12月15日 评论:
2025年12月15日 评论:
Review Guide
This Pull-Request Passes Review.
Committers who writed a comment of /approve are: zqwenn, wangchao430.
Reviewers who writed a comment of /lgtm are: wangchao430, zqwenn.


2025年12月15日 合入了pull request
What type of PR is this?
/kind bug
What does this PR do / why do we need it:
fix npu_fusion_attention strategy:
register_shardingfor operation with Tensor kwargs. Apply a patch for previous pytorch version.head_numarg, sohead_numneeds to be reset before local computation.npu_fusion_attention_backward, redistribution handling of DTensors in kwargs is added(official pytorch ops rarely have Tensor in kwargs, so most DTensor processing only considers args and ignores kwargs).Which issue(s) this PR fixes:
Fixes #
Special notes for your reviewers: