已合并
fix incorrect dynamic scales calculation when tp_world_size is 0 #4740
zhong-zixin创建于 4月16日
fix incorrect dynamic scales calculation when tp_world_size is 0 #4740
已合并
Pull Request已成功合入, 合并人@ascend-robot
(感谢 zhong-zixin 的贡献)4月16日 创建了 pull request,commit 538515b8
ascend-robot
4月16日 评论:
4月16日 评论:
ascend-robot
4月16日 评论:
4月16日 评论:
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
PR Approval Progress
✅ Congratulations! All modules have met the lgtm and approve requirements.
Module Approval Details
| module | lgtm status | approve status |
|---|---|---|
| repo-Ascend/op-plugin | ✅ 李伟, wuyi_huawei, songkai111 (3/2) | ✅ 李伟 (1/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
zhong-zixin, thanks for your pull request. All authors of the commits have signed the CLA. 👍


4月16日 添加了label:ascend-cla/yes
ascend-robot
4月16日 评论:
4月16日 评论:
Ascend docs pipeline is running...


此处折叠了64条消息 查看更多
ascend-robot
4月21日 评论:
4月21日 评论:
The following users do not have permission to comment /lgtm or /approve on any module in this PR:
yang-binrong


4月21日 添加了label:approved
ascend-robot
4月21日 评论:
4月21日 评论:
Review Guide
This pull-request passes review.
Committers who wrote a comment of /approve are: 李伟.
Reviewers who wrote a comment of /lgtm are: songkai111, 李伟, wuyi_huawei.


4月21日 合入了pull request
【合入来源】
【修改方案】
tp_world_size 为 0/1 均表示无 TP 域通信,但 dispatch v2 dynamic_scales_shape 推导在 tp_world_size 传入 0/1 时逻辑不一致,需要修复。改动涉及 op_plugin/ops/opapi/MoeDistributeDispatchV2KernelOpApi.cpp 以及 meta_registrations.py 对应 dispatch v2 部分。
【资料变更】
不涉及
【接口变更】
不涉及
【功能验证】
改动后经过算子测试脚本验证,传入 tp_world_size 为 0 / 1 时均通过,结果如下:

已添加UT测试用例看护,新增测试用例自验证结果如下:

【CheckList】