相关的Issue

https://gitcode.com/mindspore/hyper-parallel/issues/188

原因（目的、解决的问题等）

在 Hyper-Parallel 框架中，分布式训练需要一个统一的设备拓扑抽象来管理多维并行策略。DeviceMesh 模块旨在：

• 提供设备拓扑抽象：将物理设备组织成多维逻辑网格，支持数据并行（DP）、张量并行（TP）、流水线并行（PP）等多种并行策略
• 统一通信组管理：为不同并行维度创建和缓存通信组，简化集合通信操作
• 支持子网格操作：允许从原始网格中提取子网格，满足混合并行场景需求
• 支持网格扁平化：提供 flatten 方法将多维网格转换为一维，便于全局通信操作

描述（做了什么，变更了什么）

核心实现

新增 DeviceMesh 类 (hyper_parallel/core/device_mesh.py)

• 实现 DeviceMesh 类，提供设备网格的拓扑抽象
• 实现 init_device_mesh 工厂函数，支持自动 rank_list 生成和缓存机制
• 支持多种操作：
  • 子网格获取：通过 __getitem__ 方法按维度名称获取子网格
  • 通信组获取：通过 get_group 方法获取指定维度的通信组
  • 本地排名计算：通过 get_local_rank 方法获取当前进程在指定维度的本地排名
  • 网格扁平化：通过 flatten 方法将多维网格转换为一维网格
• 实现完整的输入验证和错误处理
• 支持通信组缓存，避免重复创建

主要功能

测试用例

单元测试 (tests/mindspore/ut/test_device_mesh.py)

新增 5 个单元测试用例：

变更文件列表

新增文件（2个）：

代码统计：

• 新增代码：约 832 行
• 新增测试：5 个 UT

测试用例（新增、改动、可能影响的功能）

新增测试用例

单元测试（UT）：

• ✅ test_init_device_mesh_basic - init_device_mesh 基础功能测试
• ✅ test_device_mesh_getitem_valid - 子网格获取测试
• ✅ test_device_mesh_get_group_valid - 通信组获取测试
• ✅ test_device_mesh_get_local_rank - 本地排名计算测试
• ✅ test_device_mesh_flatten - 网格扁平化测试

测试执行方式

# 运行单元测试
cd /path/to/hyper-parallel
pytest tests/mindspore/ut/test_device_mesh.py -v

可能影响的功能

• Layout 系统：DeviceMesh 作为 Layout 的底层支撑，Layout 依赖 DeviceMesh 进行设备管理
• 分布式算子：分布式算子通过 DeviceMesh 获取通信组进行集合通信
• 现有功能：纯新增功能，不影响现有模块

测试覆盖范围

• ✅ 正常场景：覆盖所有主要方法的正常使用
• ✅ 缓存机制：验证 DeviceMesh 和通信组的缓存复用
• ✅ 多维网格：2D 和 3D 网格场景
• ✅ Mock 测试：使用 Mock 隔离平台依赖

测试状态

• ✅ 所有单元测试通过
• ✅ 代码质量检查通过（无 Linter 错误）