已合并
fix d2h pinned memeory bug #32048
zzhongmin创建于 3月19日
fix d2h pinned memeory bug #32048
已合并
Pull Request已成功合入, 合并人@ascend-robot
（感谢 zzhongmin 的贡献）
AtlasAccount
3月19日 评论：
3月19日 评论：
ascend-robot
3月19日 评论：
3月19日 评论：
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 | ✅ liujunzhu, liangsongwei, suhaibo (3/2) | ✅ liujunzhu (1/1) |
| torch_npu/utils | ✅ liujunzhu, liangsongwei, suhaibo (3/2) | ✅ liujunzhu (1/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
zzhongmin, thanks for your pull request. All authors of the commits have signed the CLA. 👍
3月19日 添加了label：ascend-cla/yes
zzhongmin
3月19日 评论：
3月19日 评论：
compile
3月19日 添加了label：ci-pipeline-running
3月19日 删除了label：ci-pipeline-running
3月19日 添加了label：ci-pipeline-passed
ascend-robot
3月19日 评论：
3月19日 评论：
流水线 PR-pipeline_pytorch#11503 已完成
| 阶段 | 任务名 | 状态 | 详情 |
|---|---|---|---|
| 编译构建 | Build_X86 | ✅ | >>> |
| Build_ARM | ✅ | >>> | |
| Build_LibTorch_x86 | ✅ | >>> | |
| Build_X86_torchair | 🛑 | >>> | |
| Build_ARM_torchair | 🛑 | >>> | |
| Build_ARM_inductor | 🛑 | >>> | |
| 恶意代码检查 | Antipoison | ✅ | >>> |
| 编码安全与规范检查 | CodeCheck | ✅ | >>> |
| check_error | ✅ | >>> | |
| 开源片段检查 | SCA | ✅ | >>> |
| 开发者测试 | UT_X86_Part_01 | 🛑 | >>> |
| UT_X86_Part_02 | 🛑 | >>> | |
| UT_ARM_A3_Part_01 | 🛑 | >>> | |
| UT_ARM_A3_Part_02 | 🛑 | >>> | |
| UT_DIST_X86_Part_01 | 🛑 | >>> | |
| UT_DIST_X86_Part_02 | 🛑 | >>> | |
| UT_DIST_X86_Part_03 | 🛑 | >>> | |
| UT_DIST_X86_Part_04 | 🛑 | >>> | |
| UT_inductor_Part_01 | 🛑 | >>> | |
| UT_inductor_Part_02 | 🛑 | >>> | |
| UT_inductor_Part_03 | 🛑 | >>> | |
| UT_inductor_Part_04 | 🛑 | >>> | |
| UT_ARM_A2_Part_01 | ✅ | >>> | |
| UT_ARM_A2_Part_02 | ✅ | >>> | |
| 流水线 | PR-pipeline_pytorch | ✅ | >>> |
梁松伟
3月25日 评论：
3月25日 评论：
/lgtm
3月25日 添加了label：lgtm
liujunzhu
3月25日 评论：
3月25日 评论：
/approve
3月25日 添加了label：approved
ascend-robot
3月25日 评论：
3月25日 评论：
Review Guide
This pull-request passes review.
Committers who wrote a comment of /approve are: liujunzhu.
Reviewers who wrote a comment of /lgtm are: liujunzhu, suhaibo, liangsongwei.
3月25日 合入了pull request
【合入来源】
【修改方案】
--问题背景：在 NPU 序列化场景下，test_use_pinned_memory_for_d2h 期望 write_record 收到的是 pinned host storage
现有流程里，_reduce_ex_阶段在进入 _npu_save 前已把 storage 变成 CPU，导致原来的仅非 CPU storage 才走 pinned D2H 逻辑覆盖不到。
--解决方案：仅调整 NPU 保存路径，在 _npu_save 中保留原有 storage.device != cpu 的 pinned D2H 分支，新增 CPU fallback 当 use_pinned_memory_for_d2h=True 且当前 accelerator 为 npu 时，如果 storage 已是 CPU，也在写盘前转换为 pinned CPU storage，再 write_record。
【资料变更】
不涉及
【接口变更】
不涉及
【功能验证】
test_serialization.py -v -k test_use_pinned_memory_for_d2h用例通过
【CheckList】