已合并
[sync] PR-40339: modify _dynamo_device.py for benchmarks multi process set device error #40448
ascend-robot创建于 7月8日
[sync] PR-40339: modify _dynamo_device.py for benchmarks multi process set device error #40448
已合并
Pull Request已成功合入, 合并人@ascend-robot
（感谢 ascend-robot 的贡献）
7月8日 创建了 pull request，commit 18825bbf
7月8日 关联了issue：[Bug]: A5 跑benchmark的triton 后端 BERT_pytorch模型，plog报set_device卡死
atomgit-bot
7月8日 评论：
7月8日 评论：
变更摘要
本次变更是从上游 PR #40339 同步而来，主要修复了 torch_npu/utils/_dynamo_device.py 中 NpuInterface.exchange_device 方法在多进程 benchmark 场景下调用 set_device 报错的问题。修复方式是在切换设备前增加当前设备与目标设备的比较判断，仅当两者不同时才执行 set_device，避免不必要的设备设置操作引发错误。
主要改动
NpuInterface.exchange_device增加设备相等性检查：在调用set_device(device)前，新增if curr_device != device条件判断，仅当当前设备与目标设备不同时才执行设备切换，从而避免多进程场景下因重复/无效的设备设置导致的错误。- 保留原有设备返回逻辑不变：无论是否执行
set_device，方法始终返回切换前的当前设备编号curr_device，对外接口行为无变化。
ascend-robot
7月8日 评论：
7月8日 评论：
atomgit-bot
7月8日 评论：
7月8日 评论：
7月8日 添加了label：ascend-cla/yes
ascend-robot
7月8日 评论：
7月8日 评论：
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 |
|---|---|---|
| torch_npu/utils | ✅ htchu, ffmh (2/2) | ✅ htchu (1/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
ascend-ds-bot, thanks for your pull request. All authors of the commits have signed the CLA. 👍
7月8日 添加了label：ci-pipeline-running
ascend-robot
7月8日 评论：
7月8日 评论：
流水线 PR-pipeline_pytorch#42266 [ commitID：4020b856 ] 运行中
>>>代码风格自动修复执行成功（无修复内容）
| 阶段 | 任务名 | 状态 | 详情 |
|---|---|---|---|
| 编译构建 | Build_X86 | 🕚 | >>> |
| Build_ARM | 🕚 | >>> | |
| Build_LibTorch_x86 | 🕚 | >>> | |
| Build_LibTorch_ARM | 🕚 | >>> | |
| Build_X86_torchair | 🕚 | >>> | |
| Build_ARM_torchair | 🕚 | >>> | |
| patch_test | 🕚 | >>> | |
| 恶意代码检查 | Antipoison | 🕚 | >>> |
| 编码安全与规范检查 | CodeCheck | 🕚 | >>> |
| check_error | 🕚 | >>> | |
| CodeCheck_lintrunner | 🕚 | >>> | |
| 开源片段检查 | SCA | 🕚 | >>> |
| 开发者测试 | UT_X86_Part_01 | 🕚 | >>> |
| UT_X86_Part_02 | 🕚 | >>> | |
| UT_ARM_A3_Part_01 | 🕚 | >>> | |
| UT_ARM_A3_Part_02 | 🕚 | >>> | |
| UT_ARM_A2_Part_01 | 🕚 | >>> | |
| UT_ARM_A2_Part_02 | 🕚 | >>> | |
| UT_ARM_A2_Part_03 | 🕚 | >>> | |
| UT_inductor_Part_01 | 🕚 | >>> | |
| UT_inductor_Part_02 | 🕚 | >>> | |
| UT_inductor_Part_03 | 🕚 | >>> | |
| UT_inductor_Part_04 | 🕚 | >>> | |
| UT_DIST_ARM_Part_01 | 🕚 | >>> | |
| UT_DIST_ARM_Part_02 | 🕚 | >>> | |
| UT_DIST_ARM_Part_03 | 🕚 | >>> | |
| UT_DIST_ARM_Part_04 | 🕚 | >>> | |
| UT_ARM_A2_Select_Part_01 | 🕚 | >>> | |
| UT_ARM_A2_Select_Part_02 | 🕚 | >>> | |
| 流水线 | PR-pipeline_pytorch | 🕚 | >>> |
- compile、compile_inductor、compile_torchair : 运行流水线
- retry : 重试流水线所有失败子任务
- retry <任务名> : 仅重试指定失败子任务
- stop : 停止流水线
ascend-robot
7月8日 评论：
7月8日 评论：
ascend docs pipeline is running...
7月8日 添加了label：docs-ci-pipeline-running
ascend-robot
7月8日 评论：
7月8日 评论：
✅ 跳过 docs ci 检查，没有需要检查的文档文件
7月8日 删除了label：docs-ci-pipeline-running
7月8日 添加了label：docs-ci-pipeline-success
LucciC
7月9日 评论：
7月9日 评论：
retry
ascend-robot
7月9日 评论：
7月9日 评论：
重试流水线失败：PR-pipeline_pytorch 仅失败状态流水线支持重试操作。
7月9日 删除了label：ci-pipeline-running
7月9日 添加了label：ci-pipeline-passed
ascend-robot
7月9日 评论：
7月9日 评论：
流水线 PR-pipeline_pytorch#42266 （重试第1次） [ commitID：4020b856 ] 已完成
>>>代码风格自动修复执行成功（无修复内容）
| 阶段 | 任务名 | 状态 | 详情 |
|---|---|---|---|
| 编译构建 | Build_X86 | ✅ | >>> |
| Build_ARM | ✅ | >>> | |
| Build_LibTorch_x86 | ✅ | >>> | |
| Build_LibTorch_ARM | ✅ | >>> | |
| Build_X86_torchair | 🛑 | >>> | |
| Build_ARM_torchair | 🛑 | >>> | |
| patch_test | 🛑 | >>> | |
| 恶意代码检查 | Antipoison | ✅ | >>> |
| 编码安全与规范检查 | CodeCheck | ✅ | >>> |
| check_error | ✅ | >>> | |
| CodeCheck_lintrunner | ✅ | >>> | |
| 开源片段检查 | SCA | ✅ | >>> |
| 开发者测试 | UT_X86_Part_01 | 🛑 | >>> |
| UT_X86_Part_02 | 🛑 | >>> | |
| UT_ARM_A3_Part_01 | 🛑 | >>> | |
| UT_ARM_A3_Part_02 | 🛑 | >>> | |
| UT_ARM_A2_Part_01 | ✅ | >>> | |
| UT_ARM_A2_Part_02 | ✅ | >>> | |
| UT_ARM_A2_Part_03 | ✅ | >>> | |
| UT_inductor_Part_01 | 🛑 | >>> | |
| UT_inductor_Part_02 | 🛑 | >>> | |
| UT_inductor_Part_03 | 🛑 | >>> | |
| UT_inductor_Part_04 | 🛑 | >>> | |
| UT_DIST_ARM_Part_01 | 🛑 | >>> | |
| UT_DIST_ARM_Part_02 | 🛑 | >>> | |
| UT_DIST_ARM_Part_03 | 🛑 | >>> | |
| UT_DIST_ARM_Part_04 | 🛑 | >>> | |
| UT_ARM_A2_Select_Part_01 | ✅ | >>> | |
| UT_ARM_A2_Select_Part_02 | ✅ | >>> | |
| 流水线 | PR-pipeline_pytorch | ✅ | >>> |
- compile、compile_inductor、compile_torchair : 运行流水线
- retry : 重试流水线所有失败子任务
- retry <任务名> : 仅重试指定失败子任务
- stop : 停止流水线
7月9日 添加了label：approved
ffmh
7月9日 评论：
7月9日 评论：
/lgtm
7月9日 添加了label：lgtm
7月9日 合入了pull request
ascend-robot
7月9日 评论：
7月9日 评论：
流水线 pytorch_gitcode_PR_multiVersion#12290 [ commitID：4020b856 ] 已完成
1. Origin pull request:
https://gitcode.com/Ascend/pytorch/merge_requests/40339
2. Original pull request related issue(s):
https://gitcode.com/Ascend/pytorch/issues/2645
3. Original pull request related commit(s):