| [fa_v3]skip cpu check, register_sharding for npu_fusion_attention_v3
Co-authored-by: wangchao430<wangchao430@huawei.com>
# message auto-generated for no-merge-commit merge:
!30784 merge v2.7.0_fav3 into v2.7.1
[fa_v3]skip cpu check, register_sharding for npu_fusion_attention_v3
Created-by: wangchao430
Commit-by: wangchao430
Merged-by: ascend-robot
Description: <!--
PR描述模板更新日期:20260203
-->
# 【合入来源】
> <font color="red">**如有社区issue,请关联issue链接**</font>\
> <font color="red">**请勿携带内部流程信息(需求链接、问题单、内部issue等)**</font>
- [x] 需求
- [ ] 问题单
- [ ] 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]
- [x] 代码注释完备,正确记录错误日志
- [x] 代码实现进行了返回值、空指针等校验
- [x] PR标题正确使用类型标签,如:feat、fix、refactor、docs、test等
- [x] PR持续集成流水线(CI)执行通过,代码检查无异常
See merge request: Ascend/pytorch!30784 | 3 个月前 |