已合并
add torch.npu.memory._set_allocator_settings(only support expandable_segments) and testcases #25984
zhaoyu65创建于 2025年10月23日
add torch.npu.memory._set_allocator_settings(only support expandable_segments) and testcases #25984
已合并
Pull Request已成功合入, 合并人@ascend-robot
（感谢 zhaoyu65 的贡献）
ascend-robot
2025年10月23日 评论：
2025年10月23日 评论：
openLiBingCI
2025年10月23日 评论：
2025年10月23日 评论：
ascend-robot
2025年10月23日 评论：
2025年10月23日 评论：
以下是根据您提交的修改文件推荐的Reviewer和Committer序列，需各模块评审通过后方可合入
| Module List | Reviewers | Committers |
|---|---|---|
| repo-Ascend/pytorch | wangqiang160, liwei386, yanpengquan, guoguanghao, zhaozhijie | yangkaixin, rain-666, wasd1111222, liwei386, aichunzheng |
2025年10月23日 添加了label：ascend-cla/yes
ascend-robot
2025年10月23日 评论：
2025年10月23日 评论：
CLA Signature Pass
@zhaoyu65, thanks for your pull request. All authors of the commits have signed the CLA. 👍
zhaoyu65
2025年10月23日 评论：
2025年10月23日 评论：
compile
AtlasAccount
2025年10月23日 评论：
2025年10月23日 评论：
✅ 流水线任务触发成功，正在执行，请稍候。 任务链接[4addc81e4942442486796f6576332b46]
2025年10月23日 添加了label：ci-pipeline-running
AtlasAccount
2025年10月23日 评论：
2025年10月23日 评论：
get-log
2025年10月23日 删除了label：ci-pipeline-running
2025年10月23日 添加了label：ci-pipeline-running
AtlasAccount
2025年10月23日 评论：
2025年10月23日 评论：
/torch_npu/csrc/core/npu/NPUCachingAllocator.cpp
2025年10月23日 删除了label：ci-pipeline-running
2025年10月23日 添加了label：ci-pipeline-passed
AtlasAccount
2025年10月23日 评论：
2025年10月23日 评论：
2025年10月23日 删除了label：ci-pipeline-passed
2025年10月23日 添加了label：ci-pipeline-passed
2025年10月24日 添加了label：lgtm
li_jing_hw
2025年10月24日 评论：
2025年10月24日 评论：
/approve
2025年10月24日 添加了label：approved
ascend-robot
2025年10月24日 评论：
2025年10月24日 评论：
Review Guide
This Pull-Request Passes Review.
Committers who writed a comment of /approve are: wangchao430, li_jing_hw.
Reviewers who writed a comment of /lgtm are: wangchao430, li_jing_hw.
2025年10月24日 合入了pull request
What type of PR is this?
What does this PR do / why do we need it:
1、增加torch_npu.npu.memory._set_allocator_settings接口,仅支持expandable_segments:True或expandable_segments:False场景。
2、增加对应的测试用例。
Which issue(s) this PR fixes:
Fixes #
Special notes for your reviewers: