已合并
[fa_v3]skip cpu check, register_sharding for npu_fusion_attention_v3 #30662
[fa_v3]skip cpu check, register_sharding for npu_fusion_attention_v3 #30662
已合并
王超创建于 2月9日
王超成员
2月9日

【合入来源】

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

【修改方案】

  1. torch_npu.npu_fusion_attention_v3适配支持DTensor策略注册。
  2. graph_tree中支持根据torch_npu._inductor.config.npugraph_trees.disable_cpu_input_check配置开启时跳过cpu Tensor的相关检测,从而支持torch_npu.npu_fusion_attention_v3入graph_tree

【资料变更】

不涉及

【接口变更】

不涉及

【功能验证】

新增DTensor测试用例

【CheckList】

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

likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 王超 的贡献)
王超成员
2月9日 创建了 pull request,commit 56c967e8
ascend-robot
ascend-robot成员
2月9日 评论:

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

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

Module List Reviewers Committers
torch_npu/_inductor ffmh, huangyunlong2022, shaoyf, htchu, wasd1111222 wangqiang160, chujinjin, adrian07110, shaoyf, dilililiwhy
repo-Ascend/pytorch shaoyf, wangmin0104, chuboning, liujunzhu, hbhu_bin li_jing_hw, guoqi1024, chujinjin, htchu, wangmin0104
torch_npu/distributed pengjingyou, liwei386, wangchao430, htchu, chujinjin wasd1111222, medivh-x, liujunzhu, dilililiwhy, zqwenn
torch_npu/npu probiotics_53, wanglijun55, li_jing_hw, dilililiwhy, htchu shaoyf, zqwenn, guoqi1024, wangmin0104, medivh-x
torch_npu/utils medivh-x, chujinjin, li_jing_hw, wangchao430, wangmin0104 medivh-x, li_jing_hw, wangchao430, wangqiang160, suhaibo
likedislike
ascend-robotascend-robot成员
2月9日 添加了label:stat/needs-squash
ascend-robotascend-robot成员
2月9日 添加了label:ascend-cla/yes
此处折叠了105条消息 查看更多
ascend-robotascend-robot成员
2月14日 添加了label:approved
liwei386成员
2月14日 评论:

/approve

likedislike
ascend-robotascend-robot成员
2月14日 添加了label:lgtm
ascend-robot
ascend-robot成员
2月14日 评论:

Review Guide

This pull-request passes review.
Committers who wrote a comment of /approve are: zqwenn, liwei386.
Reviewers who wrote a comment of /lgtm are: zqwenn, liwei386.

likedislike
ascend-robotascend-robot成员
2月14日 合入了pull request