已合并
[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
已合并
wuyouqi1创建于 2月11日
已删除 :v2.9.0合入到Ascend/pytorchv2.9.0
wuyouqi1
wuyouqi1成员
2月11日

【合入来源】

(如有)请关联需求文档/issue链接

【修改方案】

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
image.png

【CheckList】

PR提交人对以下CheckList自检项进行全量自检,自检通过或不涉及,均修改 [ ] 为 [x]

likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 wuyouqi1 的贡献)
wuyouqi1wuyouqi1成员
2月11日 创建了 pull request,commit 801b152b
ascend-robot
ascend-robot成员
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 /approve or /lgtm
  • Commenting /approve implies both code review (lgtm) and intent to merge (approve)
likedislike
ascend-robotascend-robot成员
2月11日 添加了label:ascend-cla/yes
wuyouqi1
wuyouqi1成员
2月11日 评论:

compile

likedislike
此处折叠了189条消息 查看更多
ascend-robotascend-robot成员
3月12日 添加了label:approved
wangmin0104
wangmin0104成员
3月12日 评论:

/lgtm

likedislike
ascend-robotascend-robot成员
3月12日 添加了label:lgtm
ascend-robot
ascend-robot成员
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.

likedislike
ascend-robotascend-robot成员
3月12日 合入了pull request