| 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 | 5 个月前 |
