已合并
Optimize the patch for DTensor #34610
yc_233创建于 4月28日
Optimize the patch for DTensor #34610
已合并
Pull Request已成功合入, 合并人@ascend-robot
(感谢 yc_233 的贡献)ascend-robot
4月28日 评论:
4月28日 评论:
ascend-robot
4月28日 评论:
4月28日 评论:
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 |
|---|---|---|
| test | ✅ wjlflyer, kisnwang (2/2) | ✅ wjlflyer (1/1) |
| torch_npu/distributed | ✅ wjlflyer, kisnwang (2/2) | ✅ wjlflyer (1/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
zhenyu10, thanks for your pull request. All authors of the commits have signed the CLA. 👍


4月28日 添加了label:ascend-cla/yes
此处折叠了42条消息 查看更多
4月30日 添加了label:lgtm
4月30日 添加了label:approved
ascend-robot
4月30日 评论:
4月30日 评论:
4月30日 合入了pull request
【合入来源】
社区 issue:https://gitcode.com/Ascend/pytorch/issues/1786
【修改方案】
torch_npu/distributed/tensor/_dtensor_patch.py。torch_npu/distributed/tensor/__init__.py中移除_dtensor_patch导入,统一依赖上游原生 DTensor 能力。test/distributed/tensor/test_dtensor_native_kwargs_strategy.py,重点验证:OpSchema.kwargs_strategy(flat/pytree)解析expand_to_full_mesh_op_strategy在args + kwargs场景下的策略展开行为【资料变更】
不涉及。
【接口变更】
不涉及。无新增客户可见接口,无跨仓接口协议变更。
【功能验证】
测试场景:
_dtensor_patch后,DTensor kwargs 策略仍能正常工作;expand_to_full_mesh_op_strategy在含 kwargs 策略输入时展开正确;测试方法:
test/distributed/tensor/test_dtensor_native_kwargs_strategy.py覆盖点:
UT看护:
【CheckList】