已合并
fix DTensor #29015
fix DTensor #29015
已合并
jizewei创建于 1月4日
jizewei
1月4日

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. fix dtensor strategy
  3. adapt the newest npu_fusion_attention

Which issue(s) this PR fixes:

Fixes #

Special notes for your reviewers:

test_attention_ops.py
image.png
test_gather_swiglu.py
image.png
test_math_ops.py
image.png
test_matrix_ops.py
image.png
test_moe_ops.py
image.png

likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 jizewei 的贡献)
Jjizewei
1月4日 创建了 pull request,commit 2cbdbe28
ascend-robot
ascend-robot成员
1月4日 评论:

Thank your for your pull-request.

The full list of commands accepted by me can be found at here.

You can get sig-info at here

likedislike
ascend-robot
ascend-robot成员
1月4日 评论:

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

Module List Reviewers Committers
repo-Ascend/pytorch shaoyf, yangkaixin, wangchao285, culechan, wangchao430 dilililiwhy, chenhao_1209, yanpengquan, shaoyf, anyrenwei
likedislike
ascend-robotascend-robot成员
1月4日 添加了label:stat/needs-squash
此处折叠了56条消息 查看更多
王超成员
1月6日 评论:

/approve

likedislike
ascend-robotascend-robot成员
1月6日 添加了label:approved
ascend-robot
ascend-robot成员
1月6日 评论:

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.

likedislike
ascend-robotascend-robot成员
1月6日 合入了pull request
Jjizewei
1月6日 修改了pull request 的描述