已合并
[Test] Sync DeviceMesh tests with MeshLayout changes #43811
MengJiayi创建于 8月5日
[Test] Sync DeviceMesh tests with MeshLayout changes #43811
已合并
Pull Request已成功合入, 合并人@ascend-robot
（感谢 MengJiayi 的贡献）
8月5日 关联了issue：[Bug]: 用例PT_Device_Mesh只在PT213上执行失败
atomgit-bot
8月5日 评论：
8月5日 评论：
变更摘要
本 PR 是一个测试用例兼容性修复，将 test/distributed/test_device_mesh.py 与当前 DeviceMesh 和 _MeshLayout 的实现行为保持同步。主要涉及：删除已失效的 test_assert_invalid_mesh_tensor 测试（因 DeviceMesh 现已自动将 mesh tensor 转移到 CPU）、更新 test_flatten_mesh_3d 中扁平布局的 rank 校验方式（在调用 global_ranks() 前先调用 collapse()）、修正非法扁平维度切片的异常类型和错误信息预期（从 NotImplementedError 改为 KeyError），以及清理未使用的 import 和格式规范化。
主要改动
- 删除失效测试
test_assert_invalid_mesh_tensor：当前DeviceMesh在初始化前会自动将非 CPU 的 mesh tensor 转移到 CPU，旧测试预期抛出ValueError的前提已不成立，因此移除整个测试方法。 - 更新扁平布局 rank 校验调用链：在
test_flatten_mesh_3d中，dp_cp和dp_tp两种扁平布局的 rank 校验从_layout.global_ranks(8)改为_layout.collapse().global_ranks(8)，与当前_MeshLayout.global_ranks()的接口要求保持一致。 - 修正非法扁平维度切片的异常预期：将
mesh_3d["dp_tp", "cp"]的异常断言从NotImplementedError（消息"Currently, this only allows slicing out a contiguous flattened dim"）改为KeyError（消息"Mesh dim indices should be in ascending order"），并补充注释说明dp_tp分布在cp维度两侧，不满足升序切片要求。 - 清理无用 import 与格式规范化：移除了
unittest、timedelta、FakeTensorMode、FakeProcessGroup等未使用的 import，重整了部分 import 语句的格式，并修复了类注释前的空格及文件末尾的换行符。
不准确？
8月5日 关联了看板：FrameworkPTAdapter 版本issue看板
ascend-robot
8月5日 评论：
8月5日 评论：
atomgit-bot
8月5日 评论：
8月5日 评论：
8月5日 添加了label：ascend-cla/yes
ascend-robot
8月5日 评论：
8月5日 评论：
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 | ✅ wjlflyer, renyujin (2/2) | ✅ wjlflyer (1/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
Jiayi_Meng, thanks for your pull request. All authors of the commits have signed the CLA. 👍
ascend-robot
8月5日 评论：
8月5日 评论：
当前仓库存在以下 保护分支 ：
| Protected Branch | Version | Release |
|---|---|---|
| master | ||
| v2.12.0 | ||
| v2.11.0 | ||
| v2.10.0 | ||
| v2.9.0 | ||
| v2.7.1 | ||
| v2.9.0-26.1.0 | ||
| v2.7.1-26.1.0 | ||
| v2.10.0-26.1.0 | ||
| v2.12.0-26.1.0 | ||
| v2.11.0-26.1.0 | ||
| ci-test |
评论 /sync <branch1> <branch2> ... 可将当前 PR 修改同步到其它分支（创建同步 PR）：
a) 如果当前 PR 是 Open 状态，同步操作将延迟到 PR 被合并时执行
b) 如果当前 PR 已经 Merged，将立即执行同步操作
注意：
- /sync 命令可以指定同步到多个分支，仅最后一个 /sync 命令生效
- 如果创建的同步 PR 不正确，可通过向同步 PR 的源分支提交轻量级 PR 完善，或使用 /close 命令关闭
8月5日 添加了label：ci-pipeline-running
ascend-robot
8月5日 评论：
8月5日 评论：
ascend docs pipeline is running...
8月5日 添加了label：docs-ci-pipeline-running
ascend-robot
8月5日 评论：
8月5日 评论：
✅ 跳过 docs ci 检查，没有需要检查的文档文件
8月5日 删除了label：docs-ci-pipeline-running
8月5日 添加了label：docs-ci-pipeline-success
8月5日 删除了label：ci-pipeline-running
8月5日 添加了label：ci-pipeline-passed
ascend-robot
8月5日 评论：
8月5日 评论：
流水线 PR-pipeline_pytorch#55289 [ commitID：e2e777fb ] 已完成
>>>代码风格自动修复执行成功（无修复内容）
| 阶段 | 任务名 | 状态 | 详情 |
|---|---|---|---|
| 编译构建 | Build_X86 | ✅ | >>> |
| Build_ARM | ✅ | >>> | |
| Build_LibTorch_x86 | ✅ | >>> | |
| Build_LibTorch_ARM | ✅ | >>> | |
| Build_X86_torchair | 🛑 | >>> | |
| Build_ARM_torchair | 🛑 | >>> | |
| patch_test | 🛑 | >>> | |
| 恶意代码检查 | Antipoison | ✅ | >>> |
| 编码安全与规范检查 | codecheck_pre-commit | ✅ | >>> |
| check_error | ✅ | >>> | |
| lintrunner | ✅ | >>> | |
| 开源片段检查 | SCA | ✅ | >>> |
| 开发者测试 | 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 : 停止流水线
8月5日 添加了label：approvedlgtm
8月5日 合入了pull request
ascend-robot
8月5日 评论：
8月5日 评论：
流水线 pytorch_gitcode_PR_multiVersion#13664 [ commitID：e2e777fb ] 已完成
【合入来源】
【修改方案】
本 PR 将
test/distributed/test_device_mesh.py与当前 PyTorchDeviceMesh和_MeshLayout的实现行为保持同步，具体修改如下：删除已失效的
test_assert_invalid_mesh_tensor测试。ValueError。DeviceMesh会在初始化前自动将 mesh tensor 转移到 CPU，因此原测试预期已经失效。更新
test_flatten_mesh_3d中扁平布局的 rank 校验。dp_cp和dp_tp两种扁平布局，均先调用_MeshLayout.collapse()获取折叠后的一维布局，再调用global_ranks(8)。_MeshLayout.global_ranks()的接口要求保持一致。更新非法扁平维度切片的异常预期。
NotImplementedError修改为当前实现抛出的KeyError。Mesh dim indices should be in ascending order。dp_tp分布在cp维度的两侧，不满足升序切片要求。补齐测试文件末尾换行，不涉及功能行为变化。
涉及的组件交互关系如下：
test_device_mesh.py负责构造和切分DeviceMesh。DeviceMesh._flatten()创建扁平 mesh，并在根 mesh 中记录对应的_MeshLayout。_MeshLayout.collapse()将多层布局折叠为一维布局。global_ranks()基于折叠后的布局计算全局 rank，用于验证扁平 mesh 的 rank 映射是否正确。本修改属于测试用例兼容性修复，不涉及需求开发、架构重构、生产代码逻辑、组件时序或 DFX 能力变更，因此无需补充详细设计文档、时序图或类图。
【资料变更】
【接口变更】
【功能验证】
【CheckList】