master

cann-robot修复 Embedding 二维 indices 的输出形状推导

c3bc4a02Add on 23 days agoHistory

Files	Last commit	Last update
op_api	format cpp Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !6784 merge issue_fix into master format cpp Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 批量刷新cpp代码格式 ## 关联的Issue [#3791](https://gitcode.com/cann/ops-nn/issues/3791) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码格式化 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!6784	4 months ago
op_host	修复 Embedding 二维 indices 的输出形状推导 Co-authored-by: KaranocaVe<karanocave@qq.com> # message auto-generated for no-merge-commit merge: !11750 merge fix/2026-10-12-ops-nn-embedding-infershape-rank2-dim-order-01 into master 修复 Embedding 二维 indices 的输出形状推导 Created-by: KaranocaVe Commit-by: KaranocaVe Merged-by: cann-robot Description: ## 修复 Embedding 二维 indices 的输出形状推导 ### 问题 Embedding Host InferShape 将 embedding width 固定写入输出第二维。二维 indices 时会覆盖 indices 的第二维并遗漏输出末维，错误返回成功。 ### 修复 将 weight 的最后一维写入 output[rank(indices)]，并增加二维 indices 的 Host InferShape 回归测试。 ### 验证 定向 EmbeddingTest.* 共 2 项通过：一维输出 [9, 10]，二维输出 [2, 3, 10]。 关联 Issue: https://gitcode.com/cann/ops-nn/issues/6512 See merge request: cann/ops-nn!11750	23 days ago
CMakeLists.txt	[feat] 新增embedding算子 Co-authored-by: warming<wangming176@huawei.com> # message auto-generated for no-merge-commit merge: !2000 merge master into master [feat] 新增embedding算子 Created-by: wsunmoon Commit-by: warming Merged-by: cann-robot Description: ## 描述 950新增embedding算子，包含算子原型、tiling、kernel、SIMT模板等。该算子把数据集合映射到向量空间，进而将数据进行量化。 例如：输入二维权重张量为weight(m+1行，n列)，对于任意输入索引张量indices（如1行3列），输出out是一个3行n列的张量，如下： $$ \begin{aligned} & weight = \begin{bmatrix} x_{01} & x_{02} & ... & x_{0n}\\ x_{11} & x_{12} & ... & x_{1n}\\ x_{21} & x_{22} & ... & x_{2n}\\ ...\\ x_{m1} & x_{m2} & ... & x_{mn}\\ \end{bmatrix}\\ & indices = \begin{bmatrix} 0,2,m \end{bmatrix}\\ & out = \begin{bmatrix} x_{01} & x_{02} & ... & x_{0n}\\ x_{21} & x_{22} & ... & x_{2n}\\ x_{m1} & x_{m2} & ... & x_{mn}\\ \end{bmatrix}\\ \end{aligned} $$ ## 关联的Issue 关联Issue [#1360](https://gitcode.com/cann/ops-nn/issues/1360) ## 测试 UT，ST，线上冒烟均已通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!2000	7 months ago