master

ascend-robot[fa_v3]skip cpu check, register_sharding for npu_fusion_attention_v3

8f398a7b创建于 2月14日历史提交

文件	最后提交记录	最后更新时间
parallel	!18965 [2/2] Add tp cases Merge pull request !18965 from dilililiwhy/tp_case_part2	1 年前
test_attention_ops.py	[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: !30860 merge v2.99.0_fav3 into master [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!30860	7 个月前
test_dtensor_ops.py	modify test case Co-authored-by: pta-robot<pta_robot@163.com> Co-authored-by: gcw_Dgfy2aKk<dujunjie2@huawei.com> # message auto-generated for no-merge-commit merge: !29184 merge correct_testcase_260107 into master modify test case Created-by: gcw_Dgfy2aKk Commit-by: gcw_Dgfy2aKk;pta-robot Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> **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 **What does this PR do / why do we need it**: **Which issue(s) this PR fixes**: <!-- *Automatically closes linked issue when PR is merged. Usage: Fixes #<issue number>, or Fixes (paste link of issue). --> Fixes # **Special notes for your reviewers**: See merge request: Ascend/pytorch!29184	8 个月前
test_gather_swiglu.py	fix DTensor UT Co-authored-by: jizewei<jizewei@huawei.com> # message auto-generated for no-merge-commit merge: !28979 merge master_fix_dtensor_ut into master fix DTensor UT Created-by: jizewei Commit-by: jizewei Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> **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 bug **What does this PR do / why do we need it**: 1. change DTensor UT from 4 devices to 2 devices, because the CI environment only has 2 devices 2. adapt the newest npu_fusion_attention **Which issue(s) this PR fixes**: <!-- *Automatically closes linked issue when PR is merged. Usage: Fixes #<issue number>, or Fixes (paste link of issue). --> Fixes # **Special notes for your reviewers**: test_attention_ops.py  test_gather_swiglu.py  test_math_ops.py  test_matrix_ops.py  test_moe_ops.py  See merge request: Ascend/pytorch!28979	8 个月前
test_math_ops.py	Fix UT from test_register_sharding Co-authored-by: zhangguoguang<zhangguoguang2@huawei.com> # message auto-generated for no-merge-commit merge: !29145 merge fix_UT_master into master Fix UT from test_register_sharding Created-by: zhangguoguang Commit-by: zhangguoguang Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> **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 **What does this PR do / why do we need it**: **Which issue(s) this PR fixes**: <!-- *Automatically closes linked issue when PR is merged. Usage: Fixes #<issue number>, or Fixes (paste link of issue). --> Fixes # **Special notes for your reviewers**: See merge request: Ascend/pytorch!29145	8 个月前
test_matrix_ops.py	Fix UT from test_register_sharding Co-authored-by: zhangguoguang<zhangguoguang2@huawei.com> # message auto-generated for no-merge-commit merge: !29145 merge fix_UT_master into master Fix UT from test_register_sharding Created-by: zhangguoguang Commit-by: zhangguoguang Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> **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 **What does this PR do / why do we need it**: **Which issue(s) this PR fixes**: <!-- *Automatically closes linked issue when PR is merged. Usage: Fixes #<issue number>, or Fixes (paste link of issue). --> Fixes # **Special notes for your reviewers**: See merge request: Ascend/pytorch!29145	8 个月前
test_moe_ops.py	fix DTensor UT Co-authored-by: jizewei<jizewei@huawei.com> # message auto-generated for no-merge-commit merge: !28979 merge master_fix_dtensor_ut into master fix DTensor UT Created-by: jizewei Commit-by: jizewei Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> **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 bug **What does this PR do / why do we need it**: 1. change DTensor UT from 4 devices to 2 devices, because the CI environment only has 2 devices 2. adapt the newest npu_fusion_attention **Which issue(s) this PR fixes**: <!-- *Automatically closes linked issue when PR is merged. Usage: Fixes #<issue number>, or Fixes (paste link of issue). --> Fixes # **Special notes for your reviewers**: test_attention_ops.py  test_gather_swiglu.py  test_math_ops.py  test_matrix_ops.py  test_moe_ops.py  See merge request: Ascend/pytorch!28979	8 个月前
test_sharded_optim.py	modify test case Co-authored-by: pta-robot<pta_robot@163.com> Co-authored-by: gcw_Dgfy2aKk<dujunjie2@huawei.com> # message auto-generated for no-merge-commit merge: !29184 merge correct_testcase_260107 into master modify test case Created-by: gcw_Dgfy2aKk Commit-by: gcw_Dgfy2aKk;pta-robot Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> **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 **What does this PR do / why do we need it**: **Which issue(s) this PR fixes**: <!-- *Automatically closes linked issue when PR is merged. Usage: Fixes #<issue number>, or Fixes (paste link of issue). --> Fixes # **Special notes for your reviewers**: See merge request: Ascend/pytorch!29184	8 个月前