已合并
[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
wuyouqi1成员
2月11日
2月11日
# 【合入来源】
> (如有)请关联需求文档/issue链接
- [ ] 需求
- [ ] 问题单
- [x] issue/工单 ： torch._C不支持_npu_getMemoryFraction
- [ ] 重构优化
- [ ] 资料更新
# 【修改方案】
1、NPUCachingAllocator中新增getMemoryFraction方法及虚函数声明，Module中注册_npu_getMemoryFraction绑定；
2、在torch_npu.npu和torch_npu.npu.memory中实现并暴露get_per_process_memory_fraction接口，支持指定设备，调用底层 C++ 接口；
3、新增该接口的测试用例；
4、将该接口及底层Cpp绑定加入Dynamo追踪规则，完成适配。
# 【资料变更】
> 不涉及
# 【接口变更】
> 否
# 【功能验证】
> UT用例测试通过：python test_npu.py -v -k test_get_per_process_memory_fraction
# 【CheckList】
> PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x]
- [x] 代码注释完备，正确记录错误日志
- [x] 代码实现进行了返回值、空指针等校验
- [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等
- [x] PR持续集成流水线（CI）执行通过，代码检查无异常
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