Add distributed ops TransposeExtView · Pull Request !131 · MindSpore/hyper-parallel

1. Add distributed ops TransposeExtView
---
### 相关的Issue
TP并行调试问题记录：mint.transpose 报错，算子未适配 parallel layout infer func.

### 原因（目的、解决的问题等）
新增 TransposeExtView 算子的分布式支持。

### 描述（做了什么，变更了什么）
1. 新增 TransposeExtView 算子实现：添加 TransposeExtViewDistributedOp 类，支持根据 (dim0, dim1) 交换输入 alias_tensor_map 的两个维度来推导输出 layout。
2. 新增 TransposeExtView 算子注册（YAML 配置文件）。
3. 新增 TransposeExtView 算子 UT 测试用例。
4. 新增 TransposeExtView 算子 ST 测试用例。

### 测试用例（新增、改动、可能影响的功能）
1. 新增 TransposeExtView UT 测试用例，覆盖：
（1）基础维度交换场景下的 infer_layout 推导（3D tensor_map 交换 dim0/dim1）；
（2）负维度输入（dim0/dim1 为负数）等价归一化后的推导结果校验；
（3）dim0==dim1 的 no-op 场景（输出布局保持不变）；
（4）tuple alias 维度参与交换的推导（如 ("dp","cp") 与普通维度互换）；
（5）异常情况：dim 超出范围、dim 类型非法、extra_args 不符合 (dim0, dim1) 约束。

2. 新增 TransposeExtView ST 测试用例：
（1）test_transpose_ext_view_basic_3d_1
功能：测试 TransposeExtView 在 3D 输入下交换两个维度的正确性（并行 vs standalone 一致）
配置：
输入 layout：x_layout = layout("dp", "cp", "mp")（触发 DTensor 路径）
算子参数:dim0=0, dim1=2
后续算子：ReLU（配置 shard，确保输出布局在后续算子中继续被正确消费）
mesh：Layout((2, 2, 2), ("dp", "cp", "mp"))
验证：将并行输出 local_to_global 后与 standalone 输出对比一致。
（2）test_transpose_ext_view_negative_dims_2
功能：测试 TransposeExtView 支持负维度语义（并行 vs standalone 一致）
配置：
输入 layout：x_layout = layout("None", "None", "mp", "None")（4D 场景，覆盖非全维切分）
算子参数：dim0=-1, dim1=-3（等价于交换最后一维与倒数第三维）
后续算子：ReLU（按推导后的输出形状设置 shard）
mesh：Layout((2, 2, 2), ("dp", "cp", "mp"))
验证：将并行输出 local_to_global 后与 standalone 输出对比一致。
（3）test_transpose_ext_view_same_dims_noop_3
功能：测试 dim0==dim1 时算子为 no-op 的正确性（并行 vs standalone 一致）
配置：
输入 layout：x_layout = layout("dp", "cp", "mp")
算子参数：dim0=1, dim1=1
后续算子：ReLU（保持与输入一致的 shard）
mesh：Layout((2, 2, 2), ("dp", "cp", "mp"))
验证：将并行输出 local_to_global 后与 standalone 输出对比一致。

MindSpore/hyper-parallel关闭 .gitee-modal { width: 500px !important; }

搜索帮助

MindSpore/hyper-parallel
关闭