Open
[Requirement|需求建议]: embeddingbag算子960回归主线 #6903
Wang XuAdd on  23 days ago
Wang Xu
Wang Xu
23 days ago Create

Backgroud(背景信息)

将 DV121中 EmbeddingBag 算子的相关能力反合至 ops-nn 主线,新增 ascend960dt 平台适配。本次需求包括算子的编译与二进制配置、2D 路径的 GatherDMA 优化模板,以及 arch35/arch92 双入口版本隔离,同时复用主线已有的 tiling、infershape 与 kernel 公共实现。

Origin(信息来源)

DV121EmbeddingBag 算子蓝区反合

Benefit / Necessity (价值/作用)

将算子的编译、注册和二进制配置扩展至 ascend960dt。为 DAV_9201 架构引入 EmbeddingBag 2D 路径的 GatherDMA 优化,使用 GatherDataCopy 按索引批量搬入权重行,替代逐行 DataCopyPad,提升大 bag 场景的搬运效率。通过 arch35/arch92 双入口与公共头复用主线实现,减少架构间的重复代码。隔离 arch92 专属 API,恢复 950 等既有平台的可编译性。对齐 kernel 侧 gatherIndexBuf_ 分配量与 host 侧 UB 预算口径,消除 UB 越界风险。

Design(设计方案)

nan

likedislike
Wang Xu
Wang Xu
23 days ago Comments:

/assign

likedislike
CANN-robotCANN-robotCollaborators
23 days ago Assign Apricityh as responsible person
Wang XuWang Xu
23 days ago Associated pull request