已合并
[fix][2.9.0]add getMemoryFraction attribute for torch._C #30794
wuyouqi1创建于 2月11日
[fix][2.9.0]add getMemoryFraction attribute for torch._C #30794
已合并
从已删除 :v2.9.0合入到Ascend/pytorchv2.9.0
Pull Request已成功合入, 合并人@ascend-robot
(感谢 wuyouqi1 的贡献)AtlasAccount
2月11日 评论:
2月11日 评论:
ascend-robot
2月11日 评论:
2月11日 评论:
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/pytorch | ✅ guoqi1024, wangmin0104 (2/2) | ✅ guoqi1024 (1/1) |
| test | ✅ guoqi1024, wangmin0104 (2/2) | ✅ guoqi1024 (1/1) |
| torch_npu/dynamo | ✅ guoqi1024, wangmin0104 (2/2) | ✅ guoqi1024 (1/1) |
| torch_npu/npu | ✅ guoqi1024, wangmin0104 (2/2) | ✅ guoqi1024 (1/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)


2月11日 添加了label:ascend-cla/yes
wuyouqi1
2月11日 评论:
2月11日 评论:
compile


此处折叠了189条消息 查看更多
3月12日 添加了label:approved
wangmin0104
3月12日 评论:
3月12日 评论:
/lgtm


3月12日 添加了label:lgtm
ascend-robot
3月12日 评论:
3月12日 评论:
Review Guide
This pull-request passes review.
Committers who wrote a comment of /approve are: guoqi1024.
Reviewers who wrote a comment of /lgtm are: guoqi1024, wangmin0104.


3月12日 合入了pull request
【合入来源】
【修改方案】
1、NPUCachingAllocator中新增getMemoryFraction方法及虚函数声明,Module中注册_npu_getMemoryFraction绑定;
2、在torch_npu.npu和torch_npu.npu.memory中实现并暴露get_per_process_memory_fraction接口,支持指定设备,调用底层 C++ 接口;
3、新增该接口的测试用例;
4、将该接口及底层Cpp绑定加入Dynamo追踪规则,完成适配。
【资料变更】
【接口变更】
【功能验证】
【CheckList】