Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
### 姓名

李睿文

### 实习项目

混合专家架构自动切分推导优化

### 本周工作

1. **完善 take_along_axis 算子的切分推导规则**

- 进一步分析了输入 tensor 在切分维度形状不一致时的切分情况,补充得到更完整的经验性规则:1. x 的 axis 维度不能切分;2. index 的切分要与 x 切分的维度一致,且可以再加上切分 axis 维;3. out 与 index 切分状态一致;4. 若 axis 维进行切分,该切分维度上 x 和 index 的形状可以不一致,若非 axis 维进行切分,该切分维度上 x 和 index 的形状必须一致。

- pr 链接:https://github.com/PaddlePaddle/Paddle/pull/72063


2. **分析 put_along_axis 算子的切分推导规则**

- 将输入 tensor 的各种可能的切分情况,以及其他输入参数的可能情况进行组合得到不同的 case,再根据不同 case 的切分后输出是否与切分前输出是否一致,判断输入切分状态是否合理以及输出的切分状态。


#### 问题疑惑与解答

暂无

### 下周工作

1. 继续完善 put_along_axis 算子的切分推导规则并提 pr

### 导师点评
Loading