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,27 @@
### 姓名

李睿文

### 实习项目

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

### 本周工作

1. **分析并开发 put_along_axis 算子的切分推导规则**

- 根据对输入 tensor 的各种可能的切分情况进行分析,得到经验性规则:1. x 的 axis 维度不能切分;2. 三个输入 tensor 的同一维度形状不一致时不能切分,只有当形状一致时才可以切分,且 index 和 value 的切分维度要与 x 的切分维度完全一致;3. out 与 x 的切分状态一致。

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


#### 问题疑惑与解答

暂无

### 下周工作

1. 完善 put_along_axis 算子的切分推导规则
2. 分析 einsum 算子的切分推导规则

### 导师点评
Loading