diff --git a/WeeklyReports/Hackathon_8th/NKNaN/[WeeklyReports]2025.4.7~2025.4.27.md b/WeeklyReports/Hackathon_8th/NKNaN/[WeeklyReports]2025.4.7~2025.4.27.md new file mode 100644 index 00000000..c9a78648 --- /dev/null +++ b/WeeklyReports/Hackathon_8th/NKNaN/[WeeklyReports]2025.4.7~2025.4.27.md @@ -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 + +### 导师点评